app.py 49 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076
  1. from flask import Flask, render_template, request, redirect, flash, url_for, jsonify, send_file, session
  2. from werkzeug.security import check_password_hash, generate_password_hash
  3. from functools import wraps
  4. import subprocess, os, re, time, json, zipfile, io, datetime, urllib.request
  5. import config
  6. from core import file_manager as fm
  7. from core import dns_manager as dns
  8. from core import adblock_manager as ab
  9. from core import cluster_manager as cm
  10. from core import stats_manager as stats
  11. app = Flask(__name__)
  12. app.secret_key = "bind_pro_v108_modular_masterpiece"
  13. # Εκκίνηση του Background Daemon για τα Analytics
  14. stats.start_stats_daemon()
  15. # --- DECORATORS & AUTH ---
  16. def login_required(f):
  17. @wraps(f)
  18. def dec(*args, **kwargs):
  19. if 'username' not in session: return redirect(url_for('login'))
  20. return f(*args, **kwargs)
  21. return dec
  22. def admin_required(f):
  23. @wraps(f)
  24. def dec(*args, **kwargs):
  25. if session.get('role') != 'admin': return redirect(url_for('dashboard'))
  26. return f(*args, **kwargs)
  27. return dec
  28. def api_required(f):
  29. @wraps(f)
  30. def dec(*args, **kwargs):
  31. key = request.headers.get('X-API-Key')
  32. if not key or key != fm.load_panel_settings().get('api_key'):
  33. return jsonify({"status": "error", "message": "Unauthorized"}), 401
  34. return f(*args, **kwargs)
  35. return dec
  36. def check_zone_access(domain):
  37. if session.get('role') == 'admin': return True
  38. return fm.load_panel_settings().get('zone_owners', {}).get(domain) == session.get('username')
  39. # --- AUTH ROUTES ---
  40. @app.route('/login', methods=['GET', 'POST'])
  41. def login():
  42. if request.method == 'POST':
  43. u, p = request.form.get('username'), request.form.get('password')
  44. users = fm.load_users()
  45. if u in users and check_password_hash(users[u]['password'], p):
  46. session.update({'username': u, 'role': users[u]['role']})
  47. return redirect(url_for('dashboard'))
  48. flash("Invalid Credentials", "error")
  49. return render_template('login.html')
  50. @app.route('/logout')
  51. def logout():
  52. session.clear()
  53. return redirect(url_for('login'))
  54. # --- MAIN DASHBOARD & ZONES ---
  55. @app.route('/')
  56. @login_required
  57. def dashboard():
  58. conf = fm.read_file(config.LOCAL_CONF)
  59. zones_raw = re.findall(r'zone\s+"([^"]+)"\s*\{.*?type\s+(master|slave|forward|hint);', conf, re.DOTALL)
  60. zones_raw = [z for z in zones_raw if z[0] != "rpz.blacklist"]
  61. settings = fm.load_panel_settings()
  62. owners, users = settings.get('zone_owners', {}), fm.load_users()
  63. if session['role'] == 'admin':
  64. zones = zones_raw
  65. else:
  66. zones = [z for z in zones_raw if owners.get(z[0]) == session['username']]
  67. try: status = subprocess.check_output(["systemctl", "is-active", config.BIND_SERVICE]).decode().strip()
  68. except: status = "inactive"
  69. view_mode = users.get(session['username'], {}).get('view_mode', 'cards')
  70. return render_template('dashboard.html', zones=zones, status=status, templates=settings['templates'],
  71. owners=owners, all_usernames=list(users.keys()), view_mode=view_mode)
  72. @app.route('/zones/add', methods=['POST'])
  73. @login_required
  74. def add_zone():
  75. raw_domain = request.form.get('domain', '').strip()
  76. domain = re.sub(r'[^a-zA-Z0-9\.\-\/]', '', raw_domain).replace('../', '')
  77. z_type, direction = request.form.get('type'), request.form.get('direction', 'forward')
  78. if z_type != "hint" and direction == 'reverse' and not domain.endswith('.arpa'):
  79. domain = re.sub(r'/.*$', '', domain)
  80. parts = domain.strip('.').split('.')
  81. if len(parts) == 4 and parts[3] == '0': parts = parts[:3]
  82. domain = ".".join(reversed(parts)) + ".in-addr.arpa"
  83. if z_type == "hint": domain = "."
  84. conf = fm.read_file(config.LOCAL_CONF)
  85. default_zones = fm.read_file(os.path.join(config.BIND_PATH, "named.conf.default-zones"))
  86. if f'zone "{domain}"' in conf or f'zone "{domain}"' in default_zones:
  87. flash(f"Zone '{domain}' exists!", "error")
  88. return redirect(url_for('dashboard'))
  89. if z_type == "hint":
  90. hint_path = os.path.join(config.BIND_PATH, "db.root")
  91. try:
  92. urllib.request.urlretrieve("https://www.internic.net/domain/named.root", hint_path)
  93. subprocess.run(["chown", "bind:bind", hint_path])
  94. except:
  95. fm.write_file(hint_path, ". 3600000 IN NS A.ROOT-SERVERS.NET.\nA.ROOT-SERVERS.NET. 3600000 A 198.41.0.4")
  96. conf += f'\nzone "." {{\n type hint;\n file "{hint_path}";\n}};\n'
  97. elif z_type == "master":
  98. path = os.path.join(config.BIND_PATH, f"db.{domain}")
  99. conf += f'\nzone "{domain}" {{\n type master;\n file "{path}";\n}};\n'
  100. t_name = request.form.get('template_name')
  101. if t_name:
  102. t = fm.load_panel_settings()['templates'].get(t_name)
  103. if t: fm.write_file(path, dns.apply_template(t['content'], domain, t))
  104. else:
  105. s_ip = fm.load_panel_settings().get('server_public_ip', '') or '127.0.0.1'
  106. fm.write_file(path, f"$TTL 86400\n@ IN SOA ns1.{domain}. admin.{domain}. (\n {time.strftime('%Y%m%d01')}\n 3600\n 1800\n 604800\n 86400\n)\n@ IN NS ns1.{domain}.\nns1 IN A {s_ip}\n")
  107. elif z_type == "forward":
  108. ips = "".join([f" {x.strip()};" for x in request.form.getlist('forwarders[]') if x.strip()])
  109. conf += f'\nzone "{domain}" {{\n type forward;\n forwarders {{{ips} }};\n}};\n'
  110. else:
  111. path = os.path.join(config.CACHE_PATH, f"db.{domain}")
  112. ips = "".join([f" {x.strip()};" for x in request.form.getlist('master_ips[]') if x.strip()])
  113. conf += f'\nzone "{domain}" {{\n type slave;\n file "{path}";\n masters {{{ips} }};\n masterfile-format text;\n}};\n'
  114. fm.write_file(config.LOCAL_CONF, conf)
  115. s = fm.load_panel_settings()
  116. s['zone_owners'][domain] = session['username']
  117. fm.save_panel_settings(s)
  118. subprocess.run(["rndc", "reconfig"])
  119. if z_type == "master": cm.broadcast_to_cluster('add_slave', domain)
  120. flash(f"Zone '{domain}' created!", "success")
  121. return redirect(url_for('dashboard'))
  122. @app.route('/zones/manage/<path:domain>')
  123. @login_required
  124. def manage_zone(domain):
  125. clean = "." if domain == "root_zone" else domain.strip('/')
  126. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  127. conf = fm.read_file(config.LOCAL_CONF)
  128. block = dns.get_zone_block(conf, clean)
  129. if not block: return redirect(url_for('dashboard'))
  130. path = dns.get_zone_file_path(clean)
  131. try:
  132. cmd = f"tail -n 5000 {config.QUERY_LOG_PATH} | grep -iE '({clean}|transfer|notify|axfr)' | tail -n 30"
  133. logs = subprocess.check_output(cmd, shell=True, text=True).strip().split('\n')
  134. except: logs = []
  135. records = dns.parse_zone_file(path, clean)
  136. has_records = any(len(r)>0 for r in records.values())
  137. m_ips = (re.search(r'masters\s*\{\s*(.*?)\s*\};', block, re.S | re.I) or type('m',(),{'group':lambda x,y:''})()).group(1).replace(';',' ')
  138. f_ips = (re.search(r'forwarders\s*\{\s*(.*?)\s*\};', block, re.S | re.I) or type('m',(),{'group':lambda x,y:''})()).group(1).replace(';',' ')
  139. z_type = "hint" if "type hint" in block else "slave" if "type slave" in block else "forward" if "type forward" in block else "master"
  140. audit_logs = fm.get_audit_log(clean)
  141. backups = dns.get_backups(clean)
  142. dnssec_enabled = "dnssec-policy default" in block
  143. ds_records = dns.get_ds_records(clean) if dnssec_enabled else []
  144. return render_template('manage_zone.html', domain=clean, type=z_type, records=records, has_records=has_records,
  145. path=path, logs=[l for l in logs if l], master_ips=m_ips, forwarders=f_ips,
  146. audit_logs=audit_logs, backups=backups, dnssec_enabled=dnssec_enabled, ds_records=ds_records)
  147. @app.route('/zones/delete/<path:domain>')
  148. @login_required
  149. def delete_zone(domain):
  150. clean = "." if domain == "root_zone" else domain.strip('/')
  151. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  152. conf = fm.read_file(config.LOCAL_CONF)
  153. path = dns.get_zone_file_path(clean)
  154. block = dns.get_zone_block(conf, clean)
  155. if block:
  156. fm.write_file(config.LOCAL_CONF, conf.replace(block, ""))
  157. s = fm.load_panel_settings()
  158. s['zone_owners'].pop(clean, None)
  159. fm.save_panel_settings(s)
  160. subprocess.run(["rndc", "reconfig"])
  161. try: os.remove(path)
  162. except: pass
  163. if "type master" in block: cm.broadcast_to_cluster('delete_zone', clean)
  164. flash("Zone Deleted", "success")
  165. return redirect(url_for('dashboard'))
  166. # --- ZONE ACTIONS (RECORDS, IP, DNSSEC) ---
  167. @app.route('/zones/manage/<path:domain>/add_record', methods=['POST'])
  168. @login_required
  169. def add_record(domain):
  170. clean = "." if domain == "root_zone" else domain
  171. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  172. path, rtype = dns.get_zone_file_path(clean), request.form.get('rtype')
  173. name = dns.sanitize_record_name(request.form.get('name'), clean)
  174. val = dns.construct_record_value(request.form, rtype)
  175. new_content = dns.increment_serial(fm.read_file(path)) + f"{name}\tIN\t{rtype}\t{val}\n"
  176. is_valid, err = dns.check_zone_syntax(clean, new_content)
  177. if not is_valid:
  178. flash(f"Linter Error: {err}", "error")
  179. return redirect(url_for('manage_zone', domain=domain))
  180. dns.backup_zone(clean, fm.read_file(path))
  181. fm.write_file(path, new_content)
  182. subprocess.run(["rndc", "reload", clean])
  183. fm.log_audit(session['username'], clean, f"Added record {rtype} {name}")
  184. flash("Record added", "success")
  185. return redirect(url_for('manage_zone', domain=domain))
  186. @app.route('/zones/manage/<path:domain>/delete_record', methods=['POST'])
  187. @login_required
  188. def delete_record(domain):
  189. clean = "." if domain == "root_zone" else domain
  190. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  191. path, raw = dns.get_zone_file_path(clean), request.form.get('raw_line').strip()
  192. lines = fm.read_file(path).split('\n')
  193. for i, l in enumerate(lines):
  194. if l.strip() == raw:
  195. del lines[i]
  196. break
  197. new_content = dns.increment_serial('\n'.join(lines))
  198. is_valid, err = dns.check_zone_syntax(clean, new_content)
  199. if not is_valid:
  200. flash(f"Linter Error: {err}", "error")
  201. return redirect(url_for('manage_zone', domain=domain))
  202. dns.backup_zone(clean, fm.read_file(path))
  203. fm.write_file(path, new_content)
  204. subprocess.run(["rndc", "reload", clean])
  205. fm.log_audit(session['username'], clean, "Deleted a record")
  206. flash("Record deleted", "success")
  207. return redirect(url_for('manage_zone', domain=domain))
  208. @app.route('/zones/manage/<path:domain>/edit_record', methods=['POST'])
  209. @login_required
  210. def edit_record(domain):
  211. clean = "." if domain == "root_zone" else domain
  212. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  213. path, old_raw, rtype = dns.get_zone_file_path(clean), request.form.get('old_raw').strip(), request.form.get('rtype')
  214. name = dns.sanitize_record_name(request.form.get('name'), clean)
  215. val = dns.construct_record_value(request.form, rtype)
  216. lines = fm.read_file(path).split('\n')
  217. for i, l in enumerate(lines):
  218. if l.strip() == old_raw:
  219. lines[i] = f"{name}\tIN\t{rtype}\t{val}"
  220. break
  221. new_content = dns.increment_serial('\n'.join(lines))
  222. is_valid, err = dns.check_zone_syntax(clean, new_content)
  223. if not is_valid:
  224. flash(f"Linter Error: {err}", "error")
  225. return redirect(url_for('manage_zone', domain=domain))
  226. dns.backup_zone(clean, fm.read_file(path))
  227. fm.write_file(path, new_content)
  228. subprocess.run(["rndc", "reload", clean])
  229. fm.log_audit(session['username'], clean, f"Edited record {rtype} {name}")
  230. flash("Record updated", "success")
  231. return redirect(url_for('manage_zone', domain=domain))
  232. @app.route('/zones/manage/<path:domain>/rollback/<ts>', methods=['POST'])
  233. @login_required
  234. def rollback_zone(domain, ts):
  235. clean = "." if domain == "root_zone" else domain
  236. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  237. backup_file = os.path.join(config.BACKUP_DIR, f"{clean}_{ts}.db")
  238. if os.path.exists(backup_file):
  239. path = dns.get_zone_file_path(clean)
  240. dns.backup_zone(clean, fm.read_file(path))
  241. fm.write_file(path, fm.read_file(backup_file))
  242. subprocess.run(["rndc", "reload", clean])
  243. fm.log_audit(session['username'], clean, f"Rolled back to {ts}")
  244. flash("Zone rolled back", "success")
  245. else:
  246. flash("Backup not found", "error")
  247. return redirect(url_for('manage_zone', domain=domain))
  248. @app.route('/zones/manage/<path:domain>/dnssec', methods=['POST'])
  249. @login_required
  250. def toggle_dnssec(domain):
  251. clean = "." if domain == "root_zone" else domain
  252. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  253. conf = fm.read_file(config.LOCAL_CONF)
  254. block = dns.get_zone_block(conf, clean)
  255. # BUG FIX: Αν δεν βρει τη ζώνη, να μην κρασάρει, αλλά να γυρίσει πίσω
  256. if not block:
  257. flash("Zone configuration block not found!", "error")
  258. return redirect(url_for('manage_zone', domain=domain))
  259. if "dnssec-policy" in block:
  260. new_block = re.sub(r'\s*dnssec-policy default;\s*inline-signing yes;', '', block)
  261. fm.log_audit(session['username'], clean, "Disabled DNSSEC")
  262. else:
  263. new_block = block.replace('type master;', 'type master;\n dnssec-policy default;\n inline-signing yes;')
  264. subprocess.run(["chown", "-R", "bind:bind", config.BIND_PATH])
  265. fm.log_audit(session['username'], clean, "Enabled DNSSEC")
  266. fm.write_file(config.LOCAL_CONF, conf.replace(block, new_block))
  267. subprocess.run(["rndc", "reconfig"])
  268. subprocess.run(["rndc", "reload", clean])
  269. return redirect(url_for('manage_zone', domain=domain))
  270. @app.route('/zones/manage/<path:domain>/edit_ips', methods=['POST'])
  271. @login_required
  272. def edit_zone_ips(domain):
  273. clean = "." if domain == "root_zone" else domain
  274. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  275. conf, block = fm.read_file(config.LOCAL_CONF), dns.get_zone_block(fm.read_file(config.LOCAL_CONF), clean)
  276. ips = "".join([f" {x.strip()};" for x in request.form.getlist('ips[]') if x.strip()]) or " none;"
  277. if "type slave" in block: new_block = re.sub(r'masters\s*\{.*?\};', f'masters {{{ips} }};', block, flags=re.S|re.I)
  278. elif "type forward" in block: new_block = re.sub(r'forwarders\s*\{.*?\};', f'forwarders {{{ips} }};', block, flags=re.S|re.I)
  279. fm.write_file(config.LOCAL_CONF, conf.replace(block, new_block))
  280. subprocess.run(["rndc", "reconfig"])
  281. return redirect(url_for('manage_zone', domain=domain))
  282. @app.route('/zones/assign', methods=['POST'])
  283. @admin_required
  284. def assign_zone():
  285. d, o = request.form.get('domain'), request.form.get('owner')
  286. s = fm.load_panel_settings()
  287. s['zone_owners'][d] = o
  288. fm.save_panel_settings(s)
  289. flash("Assigned", "success")
  290. return redirect(url_for('dashboard'))
  291. @app.route('/zones/manage/<path:domain>/cluster_sync', methods=['POST'])
  292. @login_required
  293. def cluster_sync_zone(domain):
  294. clean = "." if domain == "root_zone" else domain.strip('/')
  295. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  296. conf, block = fm.read_file(config.LOCAL_CONF), dns.get_zone_block(fm.read_file(config.LOCAL_CONF), clean)
  297. if block and "type master" in block:
  298. cm.broadcast_to_cluster('add_slave', clean)
  299. time.sleep(1.5)
  300. subprocess.run(["rndc", "notify", clean])
  301. flash(f"Cluster Sync triggered for {clean}!", "success")
  302. else:
  303. flash("Only Master zones can be synced.", "error")
  304. return redirect(url_for('dashboard'))
  305. @app.route('/zones/manage/<path:domain>/sync')
  306. @login_required
  307. def sync_zone(domain):
  308. clean = "." if domain == "root_zone" else domain
  309. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  310. try:
  311. subprocess.run(["rndc", "retransfer", clean], check=True)
  312. flash(f"Sync initiated", "success")
  313. except:
  314. flash("Sync failed", "error")
  315. return redirect(url_for('manage_zone', domain=domain))
  316. @app.route('/zones/manage/<path:domain>/convert_to_master')
  317. @login_required
  318. def convert_to_master(domain):
  319. clean = "." if domain == "root_zone" else domain
  320. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  321. conf, block = fm.read_file(config.LOCAL_CONF), dns.get_zone_block(fm.read_file(config.LOCAL_CONF), clean)
  322. if not block or "type slave" not in block: return redirect(url_for('dashboard'))
  323. path_old = dns.get_zone_file_path(clean)
  324. res = subprocess.run(["named-compilezone", "-f", "raw", "-F", "text", "-o", "-", clean, path_old], capture_output=True, text=True)
  325. records_text = res.stdout if res.returncode == 0 else fm.read_file(path_old)
  326. path_new = os.path.join(config.BIND_PATH, f"db.{clean}" if clean != "." else "db.root")
  327. fm.write_file(config.LOCAL_CONF, conf.replace(block, f'zone "{clean}" {{\n type master;\n file "{path_new}";\n}};'))
  328. fm.write_file(path_new, records_text)
  329. subprocess.run(["rndc", "reconfig"])
  330. fm.log_audit(session['username'], clean, "Converted to Master")
  331. return redirect(url_for('manage_zone', domain=domain))
  332. @app.route('/zones/manage/<path:domain>/convert_to_slave', methods=['POST'])
  333. @login_required
  334. def convert_to_slave(domain):
  335. clean = "." if domain == "root_zone" else domain
  336. if not check_zone_access(clean): return redirect(url_for('dashboard'))
  337. conf, block = fm.read_file(config.LOCAL_CONF), dns.get_zone_block(fm.read_file(config.LOCAL_CONF), clean)
  338. if not block or "type master" not in block: return redirect(url_for('dashboard'))
  339. path_old = dns.get_zone_file_path(clean)
  340. path_new = os.path.join(config.CACHE_PATH, f"db.{clean}" if clean != "." else "db.root")
  341. ips = "".join([f" {x.strip()};" for x in request.form.getlist('master_ips[]') if x.strip()])
  342. fm.write_file(config.LOCAL_CONF, conf.replace(block, f'zone "{clean}" {{\n type slave;\n file "{path_new}";\n masters {{{ips} }};\n masterfile-format text;\n}};'))
  343. try: os.remove(path_old)
  344. except: pass
  345. subprocess.run(["rndc", "reconfig"])
  346. fm.log_audit(session['username'], clean, "Converted to Slave")
  347. return redirect(url_for('manage_zone', domain=domain))
  348. # --- ADBLOCK & ANALYTICS VIEWS ---
  349. @app.route('/adanalytics')
  350. @admin_required
  351. def adanalytics_page():
  352. s = fm.load_panel_settings()
  353. return render_template('adanalytics.html', log_enabled=(s.get('query_log_enabled') == 'yes'))
  354. @app.route('/adblock')
  355. @admin_required
  356. def adblock_dashboard():
  357. settings = fm.load_panel_settings()
  358. adblock_lists = fm.load_adblock_lists()
  359. custom_rules = fm.load_custom_rules()
  360. rpz_enabled = 'zone "rpz.blacklist"' in fm.read_file(config.LOCAL_CONF) and 'response-policy' in fm.read_file(config.OPTIONS_FILE)
  361. return render_template('adblock.html', settings=settings, adblock_lists=adblock_lists, custom_rules=custom_rules, rpz_enabled=rpz_enabled)
  362. @app.route('/analytics')
  363. @admin_required
  364. def analytics():
  365. return render_template('analytics.html')
  366. # --- CLUSTER & SETTINGS VIEWS ---
  367. @app.route('/cluster', methods=['GET', 'POST'])
  368. @admin_required
  369. def cluster_management():
  370. settings = fm.load_panel_settings()
  371. if request.method == 'POST':
  372. action = request.form.get('action')
  373. if action == 'save_ip':
  374. settings['server_public_ip'] = request.form.get('server_public_ip', '').strip()
  375. flash("Master IP Configuration Saved", "success")
  376. elif action == 'add_node':
  377. node_url = request.form.get('node_url', '').strip()
  378. if node_url and not node_url.startswith('http'): node_url = 'http://' + node_url
  379. node_url = node_url.rstrip('/')
  380. new_node = {"name": request.form.get('node_name', '').strip(), "url": node_url, "api_key": request.form.get('node_api_key', '').strip()}
  381. if new_node['name'] and new_node['url'] and new_node['api_key']:
  382. settings['cluster_nodes'] = [n for n in settings.get('cluster_nodes', []) if n['name'].strip() != new_node['name']]
  383. settings.setdefault('cluster_nodes', []).append(new_node)
  384. flash("Cluster Node Added Successfully!", "success")
  385. elif action == 'delete_node':
  386. try:
  387. node_idx = int(request.form.get('node_index'))
  388. settings['cluster_nodes'].pop(node_idx)
  389. flash("Cluster Node Removed", "success")
  390. except: flash("Failed to remove node", "error")
  391. elif action == 'regenerate_api':
  392. settings['api_key'] = secrets.token_hex(32)
  393. flash("API Key Regenerated. Update your nodes!", "success")
  394. elif action == 'save_master':
  395. m_url = request.form.get('master_url', '').strip()
  396. if m_url and not m_url.startswith('http'): m_url = 'http://' + m_url
  397. m_url = m_url.rstrip('/')
  398. settings['master_node'] = {"name": request.form.get('master_name', '').strip(), "url": m_url}
  399. flash("Master Node Linked", "success")
  400. elif action == 'unlink_master':
  401. settings['master_node'] = {}
  402. flash("Master Node Unlinked", "success")
  403. fm.save_panel_settings(settings)
  404. return redirect(url_for('cluster_management'))
  405. return render_template('cluster.html', settings=settings)
  406. @app.route('/settings')
  407. @admin_required
  408. def settings():
  409. conf = fm.read_file(config.OPTIONS_FILE)
  410. def parse_opt(regex, default='any'):
  411. m = re.search(regex, conf, re.DOTALL | re.I)
  412. return [x.strip() for x in m.group(1).split(';') if x.strip()] if m else [default]
  413. b_opt = {
  414. 'listen_v4': parse_opt(r'listen-on\s*port\s+53\s*\{\s*(.*?)\s*\};'),
  415. 'allow_query': parse_opt(r'allow-query\s*\{\s*(.*?)\s*\};'),
  416. 'allow_transfer': parse_opt(r'allow-transfer\s*\{\s*(.*?)\s*\};', 'none'),
  417. 'also_notify': parse_opt(r'also-notify\s*\{\s*(.*?)\s*\};', ''),
  418. 'dnssec': (re.search(r'dnssec-validation\s+(.*?);', conf) or type('m',(),{'group':lambda x,y:'auto'})()).group(1)
  419. }
  420. return render_template('general_config.html', b_opt=b_opt, p_data=fm.load_panel_settings())
  421. @app.route('/settings/save', methods=['POST'])
  422. @admin_required
  423. def save_settings():
  424. def join_ips(field):
  425. vals = [x.strip() for x in request.form.getlist(field) if x.strip()]
  426. return "".join([f" {x};" for x in vals]) if vals else ""
  427. lv4 = join_ips('listen_v4[]') or " any;"
  428. aq = join_ips('allow_query[]') or " any;"
  429. at = join_ips('allow_transfer[]') or " none;"
  430. an_raw = join_ips('also_notify[]')
  431. an_line = f" also-notify {{{an_raw} }};\n" if an_raw else ""
  432. dnssec = request.form.get('dnssec', 'auto')
  433. query_log = request.form.get('query_log_enabled', 'no')
  434. log_block = ""
  435. if query_log == 'yes':
  436. subprocess.run(["mkdir", "-p", "/var/log/named"])
  437. subprocess.run(["touch", "/var/log/named/queries.log"])
  438. subprocess.run(["chown", "-R", "bind:bind", "/var/log/named"])
  439. log_block = """
  440. logging {
  441. channel queries_log {
  442. file "/var/log/named/queries.log" versions 3 size 5m;
  443. severity info;
  444. print-time yes;
  445. print-category yes;
  446. print-severity yes;
  447. };
  448. category queries { queries_log; };
  449. category xfer-in { queries_log; };
  450. category xfer-out { queries_log; };
  451. category notify { queries_log; };
  452. category rpz { queries_log; };
  453. };"""
  454. old_opts = fm.read_file(config.OPTIONS_FILE)
  455. rpz_line = ' response-policy { zone "rpz.blacklist"; };\n' if 'response-policy' in old_opts else ''
  456. opt_content = f"options {{\n directory \"/var/cache/bind\";\n listen-on port 53 {{{lv4} }};\n allow-query {{{aq} }};\n allow-transfer {{{at} }};\n{an_line}{rpz_line} dnssec-validation {dnssec};\n querylog {'yes' if query_log=='yes' else 'no'};\n auth-nxdomain no;\n}};\n{log_block}"
  457. fm.write_file(config.OPTIONS_FILE, opt_content.strip())
  458. t_names, t_refresh, t_retry, t_expire, t_ttl, t_content = request.form.getlist('t_name[]'), request.form.getlist('t_refresh[]'), request.form.getlist('t_retry[]'), request.form.getlist('t_expire[]'), request.form.getlist('t_ttl[]'), request.form.getlist('t_content[]')
  459. new_templates = {}
  460. for idx, name in enumerate(t_names):
  461. name = name.strip()
  462. if not name: continue
  463. keys, vals = request.form.getlist(f'custom_key_{idx}[]'), request.form.getlist(f'custom_val_{idx}[]')
  464. custom_vars = {keys[i].strip().upper(): vals[i] for i in range(len(keys)) if keys[i].strip()}
  465. new_templates[name] = {"refresh": t_refresh[idx], "retry": t_retry[idx], "expire": t_expire[idx], "ttl": t_ttl[idx], "content": t_content[idx], "custom_vars": custom_vars}
  466. s = fm.load_panel_settings()
  467. s.update({"templates": new_templates, "stats_retention": int(request.form.get('stats_retention', 30)), "query_log_enabled": query_log})
  468. fm.save_panel_settings(s)
  469. subprocess.run(["systemctl", "restart", config.BIND_SERVICE])
  470. flash("Settings Saved", "success")
  471. return redirect(url_for('settings'))
  472. @app.route('/settings/view_mode', methods=['POST'])
  473. @login_required
  474. def update_view_mode():
  475. mode = request.json.get('mode', 'cards')
  476. users = fm.load_users()
  477. if session['username'] in users:
  478. users[session['username']]['view_mode'] = mode
  479. fm.save_users(users)
  480. return jsonify({"status": "success"})
  481. # --- USERS VIEWS ---
  482. @app.route('/users')
  483. @admin_required
  484. def manage_users():
  485. return render_template('users.html', users=fm.load_users(), all_templates=list(fm.load_panel_settings()['templates'].keys()))
  486. @app.route('/users/save', methods=['POST'])
  487. @admin_required
  488. def save_user():
  489. action, old_u, u, p, r = request.form.get('action'), request.form.get('old_username'), request.form.get('username'), request.form.get('password'), request.form.get('role')
  490. assigned = request.form.getlist('assigned_templates[]')
  491. users = fm.load_users()
  492. if action == 'add':
  493. if u in users: flash("User exists", "error")
  494. else: users[u] = {"password": generate_password_hash(p), "role": r, "assigned_templates": assigned, "view_mode": "cards"}
  495. elif action == 'edit' and old_u in users:
  496. pw = generate_password_hash(p) if p else users[old_u]['password']
  497. users[old_u] = {"password": pw, "role": r, "assigned_templates": assigned, "view_mode": users[old_u].get('view_mode','cards')}
  498. if u != old_u: users[u] = users.pop(old_u)
  499. fm.save_users(users)
  500. flash("User saved", "success")
  501. return redirect(url_for('manage_users'))
  502. @app.route('/users/delete/<username>')
  503. @admin_required
  504. def delete_user(username):
  505. users = fm.load_users()
  506. if username != 'admin' and username in users:
  507. del users[username]
  508. fm.save_users(users)
  509. flash("User deleted", "success")
  510. return redirect(url_for('manage_users'))
  511. # --- ADBLOCK API (MASTER) ---
  512. @app.route('/api/adblock/update', methods=['POST'])
  513. @admin_required
  514. def route_update_adblock():
  515. result = ab.generate_and_apply_rpz()
  516. if result.get("status") == "success":
  517. # BUG FIX: Ειδοποιούμε το Cluster ότι κάναμε update!
  518. cm.broadcast_adblock_to_cluster('enable')
  519. return jsonify(result)
  520. else:
  521. return jsonify(result), 500
  522. @app.route('/api/adblock/disable', methods=['POST'])
  523. @admin_required
  524. def route_disable_adblock():
  525. result = ab.disable_rpz()
  526. cm.broadcast_adblock_to_cluster('disable')
  527. return jsonify(result)
  528. @app.route('/api/adblock/sync_cluster', methods=['POST'])
  529. @admin_required
  530. def route_sync_adblock_cluster():
  531. rpz_enabled = 'zone "rpz.blacklist"' in fm.read_file(config.LOCAL_CONF)
  532. action = 'enable' if rpz_enabled else 'disable'
  533. settings = fm.load_panel_settings()
  534. nodes = settings.get('cluster_nodes', [])
  535. lists = fm.load_adblock_lists()
  536. rules = fm.load_custom_rules()
  537. count = settings.get('adblock_domains_count', 0)
  538. update_time = settings.get('adblock_last_update', 'Never')
  539. success_count = 0
  540. for node in nodes:
  541. try:
  542. url = f"{node['url'].rstrip('/')}/api/v1/cluster/adblock"
  543. headers = {"X-API-Key": node['api_key'], "Content-Type": "application/json"}
  544. payload = {"action": action, "lists": lists, "rules": rules, "count": count, "time": update_time}
  545. proxy_handler = urllib.request.ProxyHandler({})
  546. opener = urllib.request.build_opener(proxy_handler)
  547. req = urllib.request.Request(url, data=json.dumps(payload).encode('utf-8'), headers=headers, method='POST')
  548. with opener.open(req, timeout=10) as response:
  549. if response.status == 200: success_count += 1
  550. except Exception as e:
  551. print(f"Cluster AdBlock Sync FAIL: {node['name']} -> {e}")
  552. return jsonify({"status": "success", "message": f"Sent AdBlock config to {success_count}/{len(nodes)} cluster nodes."})
  553. @app.route('/api/adblock/toggle_list', methods=['POST'])
  554. @admin_required
  555. def toggle_adblock_list():
  556. list_id = request.json.get('id')
  557. state = request.json.get('state')
  558. lists = fm.load_adblock_lists()
  559. for l in lists:
  560. if l['id'] == list_id:
  561. l['enabled'] = state
  562. break
  563. fm.save_adblock_lists(lists)
  564. return jsonify({"status": "success"})
  565. @app.route('/api/adblock/add_custom', methods=['POST'])
  566. @admin_required
  567. def add_custom_adblock():
  568. name = request.form.get('name', '').strip()
  569. url = request.form.get('url', '').strip()
  570. desc = request.form.get('desc', 'Custom User Blocklist').strip()
  571. if not name or not url: return redirect(url_for('adblock_dashboard'))
  572. lists = fm.load_adblock_lists()
  573. new_id = "custom_" + str(int(time.time()))
  574. lists.append({"id": new_id, "name": name, "desc": desc, "url": url, "enabled": True, "category": "Custom", "is_custom": True, "info": url})
  575. fm.save_adblock_lists(lists)
  576. flash("Custom list added", "success")
  577. return redirect(url_for('adblock_dashboard'))
  578. @app.route('/api/adblock/edit_custom', methods=['POST'])
  579. @admin_required
  580. def edit_custom_adblock():
  581. list_id = request.form.get('list_id')
  582. name = request.form.get('name', '').strip()
  583. url = request.form.get('url', '').strip()
  584. desc = request.form.get('desc', '').strip()
  585. lists = fm.load_adblock_lists()
  586. for l in lists:
  587. if l['id'] == list_id and l.get('is_custom'):
  588. l['name'], l['desc'], l['url'], l['info'] = name, desc, url, url
  589. break
  590. fm.save_adblock_lists(lists)
  591. flash("Custom list updated", "success")
  592. return redirect(url_for('adblock_dashboard'))
  593. @app.route('/api/adblock/delete_list/<list_id>', methods=['GET'])
  594. @admin_required
  595. def delete_adblock_list(list_id):
  596. lists = [l for l in fm.load_adblock_lists() if l['id'] != list_id or not l['is_custom']]
  597. fm.save_adblock_lists(lists)
  598. flash("List deleted", "success")
  599. return redirect(url_for('adblock_dashboard'))
  600. @app.route('/api/adblock/rules/add', methods=['POST'])
  601. @admin_required
  602. def add_adblock_rule():
  603. rtype, domain = request.form.get('type'), request.form.get('domain', '').strip().lower()
  604. domain = re.sub(r'^https?://', '', domain).rstrip('/')
  605. rules = fm.load_custom_rules()
  606. if domain not in rules[rtype]: rules[rtype].append(domain)
  607. fm.save_custom_rules(rules)
  608. return jsonify({"status": "success"})
  609. @app.route('/api/adblock/rules/edit', methods=['POST'])
  610. @admin_required
  611. def edit_adblock_rule():
  612. rtype = request.form.get('type')
  613. old_domain = request.form.get('old_domain', '').strip()
  614. new_domain = re.sub(r'^https?://', '', request.form.get('new_domain', '').strip().lower()).rstrip('/')
  615. rules = fm.load_custom_rules()
  616. if old_domain in rules.get(rtype, []):
  617. rules[rtype][rules[rtype].index(old_domain)] = new_domain
  618. fm.save_custom_rules(rules)
  619. return jsonify({"status": "success"})
  620. @app.route('/api/adblock/rules/delete', methods=['POST'])
  621. @admin_required
  622. def delete_adblock_rule():
  623. rtype, domain = request.form.get('type'), request.form.get('domain', '').strip()
  624. rules = fm.load_custom_rules()
  625. rules[rtype] = [d for d in rules.get(rtype, []) if d != domain]
  626. fm.save_custom_rules(rules)
  627. return jsonify({"status": "success"})
  628. # --- STATS API ---
  629. @app.route('/api/adblock/stats_live')
  630. @admin_required
  631. def api_adblock_stats_live():
  632. with stats.stats_lock:
  633. try:
  634. with open(config.STATS_FILE, "r") as f: st = json.load(f)
  635. except: st = {}
  636. now = datetime.datetime.now().strftime("%Y-%m-%d")
  637. day_data = st.get(now, {})
  638. total_q = day_data.get("queries", 0)
  639. total_b = day_data.get("blocked_queries", 0)
  640. top_domains = sorted(day_data.get("blocked_domains", {}).items(), key=lambda x: x[1], reverse=True)[:10]
  641. top_clients = sorted(day_data.get("blocked_clients", {}).items(), key=lambda x: x[1], reverse=True)[:10]
  642. percentage = round((total_b / total_q * 100), 2) if total_q > 0 else 0
  643. return jsonify({
  644. "total_queries": total_q,
  645. "total_blocked": total_b,
  646. "percentage": percentage,
  647. "top_domains": top_domains,
  648. "top_clients": top_clients,
  649. "timeline": day_data.get("timeline", {}),
  650. "block_timeline": day_data.get("block_timeline", {})
  651. })
  652. @app.route('/api/stats')
  653. @admin_required
  654. def api_stats():
  655. with stats.stats_lock: return fm.read_file(config.STATS_FILE) or "{}"
  656. @app.route('/api/ip_queries/<ip>')
  657. @admin_required
  658. def get_ip_queries(ip):
  659. if not re.match(r'^[\d\.\:a-fA-F]+$', ip): return jsonify({"error": "Invalid IP format"})
  660. try:
  661. cmd = f"grep -i '{ip}#' {config.QUERY_LOG_PATH} | tail -n 50"
  662. proc = subprocess.run(cmd, shell=True, capture_output=True, text=True)
  663. lines = proc.stdout.strip().split('\n')
  664. queries = []
  665. for line in reversed(lines):
  666. if not line: continue
  667. m = re.search(r'query:\s+([^\s]+)\s+IN\s+([A-Z0-9]+)', line, re.I)
  668. if m:
  669. time_str = " ".join(line.split()[:2]) if len(line.split()) > 2 else ""
  670. queries.append({"time": time_str, "domain": m.group(1), "type": m.group(2)})
  671. return jsonify({"ip": ip, "queries": queries})
  672. except Exception as e: return jsonify({"error": str(e)})
  673. # --- UTILS API ---
  674. @app.route('/api/zones/test_dig', methods=['POST'])
  675. @login_required
  676. def test_dig_transfer():
  677. domain, target_ip = request.json.get('domain'), request.json.get('target_ip', '127.0.0.1')
  678. if not domain: return jsonify({"error": "No domain provided"}), 400
  679. block = dns.get_zone_block(fm.read_file(config.LOCAL_CONF), domain)
  680. if block:
  681. m = re.search(r'masters\s*\{\s*([^;]+);', block)
  682. if m: target_ip = m.group(1).strip()
  683. try:
  684. output = subprocess.run(["dig", f"@{target_ip}", domain, "AXFR"], capture_output=True, text=True, timeout=8).stdout
  685. if "Transfer failed" in output or "connection refused" in output.lower() or not output.strip():
  686. return jsonify({"success": False, "raw": output, "message": f"Transfer failed from Master ({target_ip})"})
  687. analysis, total_count = {}, 0
  688. for line in output.split('\n'):
  689. if not line or line.startswith((';', '\n')): continue
  690. for p in line.split():
  691. if p in ['A', 'AAAA', 'MX', 'TXT', 'CNAME', 'NS', 'SOA', 'SRV', 'PTR']:
  692. analysis[p] = analysis.get(p, 0) + 1
  693. total_count += 1
  694. break
  695. return jsonify({"success": True, "total": total_count, "analysis": analysis, "raw": output, "target": target_ip})
  696. except Exception as e: return jsonify({"success": False, "message": str(e)})
  697. @app.route('/api/service/status')
  698. @admin_required
  699. def get_service_status():
  700. try:
  701. res = subprocess.run(["systemctl", "is-active", config.BIND_SERVICE], capture_output=True, text=True)
  702. is_active = res.stdout.strip() == "active"
  703. uptime = "Unknown"
  704. if is_active:
  705. res_up = subprocess.run(["systemctl", "show", config.BIND_SERVICE, "--property=ActiveEnterTimestamp"], capture_output=True, text=True)
  706. uptime = res_up.stdout.replace("ActiveEnterTimestamp=", "").strip()
  707. return jsonify({"status": "active" if is_active else "inactive", "uptime": uptime})
  708. except: return jsonify({"status": "error", "uptime": "N/A"})
  709. @app.route('/service/<action>')
  710. @admin_required
  711. def service_control(action):
  712. try:
  713. if action == 'reload': subprocess.run(["rndc", "reload"], check=True)
  714. else: subprocess.run(["systemctl", action, config.BIND_SERVICE], check=True)
  715. except: pass
  716. return redirect(request.referrer or url_for('dashboard'))
  717. @app.route('/backup/export')
  718. @admin_required
  719. def export_backup():
  720. memory_file = io.BytesIO()
  721. zip_f = zipfile.ZipFile(memory_file, 'w', zipfile.ZIP_DEFLATED)
  722. core_files = [
  723. config.LOCAL_CONF, config.OPTIONS_FILE, config.PANEL_SETTINGS,
  724. config.USERS_FILE, config.STATS_FILE, config.ADBLOCK_LISTS_FILE,
  725. config.ADBLOCK_CUSTOM_RULES_FILE
  726. ]
  727. for f in core_files:
  728. if os.path.exists(f):
  729. zip_f.write(f, os.path.basename(f))
  730. # BUG FIX: Διπλός έλεγχος os.path.exists() για να μην κρασάρει το zip
  731. for f in os.listdir(config.BIND_PATH):
  732. if f.startswith("db.") and f != "db.rpz.blacklist":
  733. full_path = os.path.join(config.BIND_PATH, f)
  734. if os.path.exists(full_path):
  735. zip_f.write(full_path, f)
  736. zip_f.close()
  737. memory_file.seek(0)
  738. return send_file(memory_file, download_name=f"bind_backup_{time.strftime('%Y%m%d_%H%M%S')}.zip", as_attachment=True)
  739. @app.route('/backup/import', methods=['POST'])
  740. @admin_required
  741. def import_backup():
  742. file = request.files.get('backup_file')
  743. if file and file.filename.endswith('.zip'):
  744. with zipfile.ZipFile(file, 'r') as zf: zf.extractall(config.BIND_PATH)
  745. subprocess.run(["systemctl", "restart", config.BIND_SERVICE])
  746. flash("Restored", "success")
  747. return redirect(url_for('dashboard'))
  748. # --- CLUSTER SLAVE RECEIVERS (V1 API) ---
  749. @app.route('/api/v1/cluster/adblock', methods=['POST'])
  750. @api_required
  751. def api_cluster_adblock():
  752. data = request.json
  753. action = data.get('action')
  754. master_ip = request.headers.get('X-Forwarded-For', request.remote_addr).split(',')[0].strip()
  755. if 'lists' in data: fm.save_adblock_lists(data['lists'])
  756. if 'rules' in data: fm.save_custom_rules(data['rules'])
  757. s = fm.load_panel_settings()
  758. if 'count' in data: s['adblock_domains_count'] = data['count']
  759. if 'time' in data: s['adblock_last_update'] = data['time']
  760. fm.save_panel_settings(s)
  761. conf, opts = fm.read_file(config.LOCAL_CONF), fm.read_file(config.OPTIONS_FILE)
  762. if action == 'enable':
  763. if 'zone "rpz.blacklist"' not in conf:
  764. path = os.path.join(config.CACHE_PATH, "db.rpz.blacklist")
  765. conf += f'\nzone "rpz.blacklist" {{\n type slave;\n file "{path}";\n masters {{ {master_ip}; }};\n allow-query {{ any; }};\n}};\n'
  766. fm.write_file(config.LOCAL_CONF, conf)
  767. if "response-policy" not in opts:
  768. opts = re.sub(r'(options\s*\{)', r'\1\n response-policy { zone "rpz.blacklist"; };', opts)
  769. fm.write_file(config.OPTIONS_FILE, opts)
  770. else:
  771. block = dns.get_zone_block(conf, "rpz.blacklist")
  772. if block: fm.write_file(config.LOCAL_CONF, conf.replace(block, ""))
  773. opts = re.sub(r'\s*response-policy\s*\{\s*zone\s*"rpz\.blacklist";\s*\};\s*', '\n', opts)
  774. fm.write_file(config.OPTIONS_FILE, opts)
  775. subprocess.run(["systemctl", "restart", config.BIND_SERVICE])
  776. return jsonify({"status": "success", "message": "RPZ synced on slave."})
  777. # Στο app.py του SLAVE
  778. @app.route('/api/v1/cluster/zone', methods=['POST'])
  779. @api_required
  780. def api_add_slave():
  781. # force=True για να διαβάσει το JSON ακόμα κι αν το content-type λείπει
  782. data = request.get_json(force=True, silent=True) or {}
  783. domain = data.get('domain')
  784. if not domain:
  785. return jsonify({"status": "error", "message": "No domain provided"}), 400
  786. # Καθαρισμός domain
  787. domain = re.sub(r'[^a-zA-Z0-9\.\-\/]', '', domain).replace('../', '')
  788. # Λήψη των Masters (είτε από το payload είτε από την IP του Master)
  789. masters_list = data.get('masters', [])
  790. if not masters_list:
  791. master_ip = request.headers.get('X-Forwarded-For', request.remote_addr).split(',')[0].strip()
  792. masters_list = [master_ip]
  793. masters_string = "".join([f" {ip};" for ip in masters_list])
  794. conf = fm.read_file(config.LOCAL_CONF)
  795. # Αν υπάρχει ήδη, δεν την ξαναφτιάχνουμε
  796. if f'zone "{domain}"' in conf:
  797. return jsonify({"status": "success", "message": "Already exists"}), 200
  798. path = os.path.join(config.CACHE_PATH, f"db.{domain}")
  799. new_zone = f'\nzone "{domain}" {{\n type slave;\n file "{path}";\n masters {{{masters_string} }};\n masterfile-format text;\n}};\n'
  800. fm.write_file(config.LOCAL_CONF, conf + new_zone)
  801. subprocess.run(["rndc", "reconfig"])
  802. return jsonify({"status": "success", "message": f"Zone {domain} added"})
  803. @app.route('/api/v1/cluster/zone/<domain>', methods=['DELETE'])
  804. @api_required
  805. def api_delete_zone(domain):
  806. clean = re.sub(r'[^a-zA-Z0-9\.\-\/]', '', domain).replace('../', '')
  807. conf = fm.read_file(config.LOCAL_CONF)
  808. block = dns.get_zone_block(conf, clean)
  809. if not block: return jsonify({"status": "error", "message": "Zone not found"}), 404
  810. path = dns.get_zone_file_path(clean)
  811. fm.write_file(config.LOCAL_CONF, conf.replace(block, ""))
  812. subprocess.run(["rndc", "reconfig"])
  813. try: os.remove(path)
  814. except: pass
  815. fm.log_audit("API_SYNC", clean, "Auto-deleted via Cluster Sync")
  816. return jsonify({"status": "success", "message": f"Zone {clean} deleted."})
  817. @app.route('/api/v1/cluster/ping', methods=['GET'])
  818. @api_required
  819. def api_ping():
  820. # ΑΦΑΙΡΕΘΗΚΕ ΤΟ AUTO-LINKING! Το Ping πλέον απλά επιβεβαιώνει ότι το API Key είναι σωστό.
  821. return jsonify({"status": "success", "message": "Node is synchronized"})
  822. @app.route('/cluster/check_node', methods=['POST'])
  823. @login_required
  824. def check_cluster_node():
  825. node_name = request.json.get('node_name')
  826. node = next((n for n in fm.load_panel_settings().get('cluster_nodes', []) if n['name'] == node_name), None)
  827. if not node: return jsonify({"status": "error"})
  828. try:
  829. url = f"{node['url'].rstrip('/')}/api/v1/cluster/ping"
  830. proxy_handler = urllib.request.ProxyHandler({})
  831. opener = urllib.request.build_opener(proxy_handler)
  832. req = urllib.request.Request(url, headers={"X-API-Key": node['api_key']})
  833. with opener.open(req, timeout=10) as response:
  834. if response.status == 200: return jsonify({"status": "online"})
  835. except urllib.error.HTTPError as e:
  836. print(f"\n[CLUSTER HTTP ERROR] Node: {node_name} -> Κωδικός: {e.code}\n")
  837. if e.code == 401: return jsonify({"status": "auth_error"})
  838. except Exception as e:
  839. print(f"\n[CLUSTER NETWORK ERROR] Node: {node_name} -> Αιτία: {str(e)}\n")
  840. return jsonify({"status": "offline"})
  841. # --- REMOTE SERVICE CONTROL (SLAVE RECEIVER) ---
  842. @app.route('/api/v1/cluster/service', methods=['POST'])
  843. @api_required
  844. def api_cluster_service():
  845. # Fix: Ασφαλής ανάγνωση του JSON, χωρίς να σκάει
  846. data = request.get_json(force=True, silent=True) or {}
  847. action = data.get('action')
  848. if action in ['start', 'stop', 'restart', 'reload']:
  849. try:
  850. # Fix: Πιάνουμε το πραγματικό error του systemctl αν αποτύχει
  851. res = subprocess.run(["systemctl", action, config.BIND_SERVICE], capture_output=True, text=True)
  852. if res.returncode == 0:
  853. return jsonify({"status": "success", "message": f"Service {action}ed successfully"})
  854. else:
  855. return jsonify({"status": "error", "message": f"Systemctl failed: {res.stderr.strip()}"})
  856. except Exception as e:
  857. return jsonify({"status": "error", "message": str(e)})
  858. elif action == 'status':
  859. try:
  860. res = subprocess.run(["systemctl", "is-active", config.BIND_SERVICE], capture_output=True, text=True)
  861. is_active = res.stdout.strip() == "active"
  862. return jsonify({"status": "success", "service_status": "active" if is_active else "inactive"})
  863. except:
  864. return jsonify({"status": "error", "service_status": "unknown"})
  865. return jsonify({"status": "error", "message": "Invalid action provided"})
  866. # --- REMOTE SERVICE PROXY (MASTER TO SLAVE) ---
  867. @app.route('/cluster/node_service', methods=['POST'])
  868. @login_required
  869. def cluster_node_service():
  870. import urllib.request, urllib.error
  871. try:
  872. data = request.get_json(force=True, silent=True) or {}
  873. node = next((n for n in fm.load_panel_settings().get('cluster_nodes', []) if n['name'] == data.get('node_name')), None)
  874. if not node: return jsonify({"status": "error", "message": "Node not found."})
  875. url = f"{node['url'].rstrip('/')}/api/v1/cluster/service"
  876. req = urllib.request.Request(url, data=json.dumps({"action": data.get('action')}).encode('utf-8'), headers={"X-API-Key": node['api_key'], "Content-Type": "application/json"}, method='POST')
  877. proxy_handler = urllib.request.ProxyHandler({})
  878. https_handler = urllib.request.HTTPSHandler(context=config.ssl_ctx)
  879. opener = urllib.request.build_opener(proxy_handler, https_handler)
  880. with opener.open(req, timeout=15) as response:
  881. return jsonify(json.loads(response.read().decode('utf-8')))
  882. except urllib.error.HTTPError as e:
  883. return jsonify({"status": "error", "message": f"Slave HTTP Error {e.code}."})
  884. except Exception as e:
  885. return jsonify({"status": "error", "message": f"Connection failed: {str(e)}"})
  886. # --- GLOBAL CLUSTER SYNC (MASTER ACTION) ---
  887. @app.route('/api/cluster/sync_all', methods=['POST'])
  888. @admin_required
  889. def sync_all_cluster():
  890. conf = fm.read_file(config.LOCAL_CONF)
  891. import re
  892. zones = re.findall(r'zone\s+"([^"]+)"', conf)
  893. ignore = ["rpz.blacklist", ".", "localhost", "127.in-addr.arpa", "0.in-addr.arpa", "255.in-addr.arpa"]
  894. valid_zones = [z for z in zones if z not in ignore]
  895. if not valid_zones:
  896. return jsonify({"status": "error", "message": "No zones found to sync."})
  897. def bulk_sync():
  898. import urllib.request, json, ssl, time
  899. settings = fm.load_panel_settings()
  900. nodes = settings.get('cluster_nodes', [])
  901. ctx = ssl.create_default_context()
  902. ctx.check_hostname = False
  903. ctx.verify_mode = ssl.CERT_NONE
  904. for node in nodes:
  905. for z in valid_zones:
  906. masters = dns.get_zone_masters_list(z)
  907. if not masters:
  908. master_ip = settings.get('server_public_ip', '').strip()
  909. if master_ip: masters = [master_ip]
  910. try:
  911. url = f"{node['url'].rstrip('/')}/api/v1/cluster/zone"
  912. headers = {"X-API-Key": node['api_key'], "Content-Type": "application/json"}
  913. payload = {"domain": z, "masters": masters}
  914. req = urllib.request.Request(url, data=json.dumps(payload).encode('utf-8'), headers=headers, method='POST')
  915. urllib.request.urlopen(req, timeout=5, context=ctx)
  916. print(f"Cluster Sync OK: {node['name']} -> {z}")
  917. except Exception as e:
  918. print(f"Cluster Sync FAIL: {node['name']} -> {z} ({e})")
  919. # ΠΟΛΥ ΣΗΜΑΝΤΙΚΟ: Περιμένουμε μισό δευτερόλεπτο για να προλάβει ο Slave να γράψει το αρχείο!
  920. time.sleep(0.5)
  921. import threading
  922. threading.Thread(target=bulk_sync, daemon=True).start()
  923. return jsonify({"status": "success", "message": f"Sync started! Sending {len(valid_zones)} zones in the background. Please wait ~10 seconds."})
  924. if __name__ == '__main__':
  925. # Είναι σημαντικό το threaded=True για να μην κάνει "block" το UI στο self-ping
  926. app.run(host='0.0.0.0', port=5000, debug=False, threaded=True)