server_manager.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793
  1. package dhcp
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "log"
  6. "net"
  7. "os"
  8. "os/exec"
  9. "path/filepath"
  10. "strconv"
  11. "strings"
  12. "sync"
  13. "time"
  14. )
  15. const (
  16. DHCPServerStoreFile = "/etc/proxmox-router/dhcp_server.json"
  17. DnsmasqConfFile = "/etc/dnsmasq.d/debtik_dhcp.conf"
  18. DnsmasqLeasesFile = "/var/lib/misc/dnsmasq.leases"
  19. )
  20. type ServerManager struct {
  21. mu sync.RWMutex
  22. storePath string
  23. confPath string
  24. leasesPath string
  25. servers map[string]DHCPServerInstance
  26. networks map[string]DHCPNetwork
  27. staticLeases map[string]DHCPLease
  28. options map[string]DHCPOption
  29. }
  30. func NewServerManager() *ServerManager {
  31. sm := &ServerManager{
  32. storePath: DHCPServerStoreFile,
  33. confPath: DnsmasqConfFile,
  34. leasesPath: DnsmasqLeasesFile,
  35. servers: make(map[string]DHCPServerInstance),
  36. networks: make(map[string]DHCPNetwork),
  37. staticLeases: make(map[string]DHCPLease),
  38. options: make(map[string]DHCPOption),
  39. }
  40. _ = os.MkdirAll(filepath.Dir(sm.storePath), 0755)
  41. _ = os.MkdirAll(filepath.Dir(sm.confPath), 0755)
  42. if err := sm.load(); err != nil {
  43. log.Printf("[DHCP-Server] Error loading config: %v", err)
  44. }
  45. // Initial render and reload dnsmasq
  46. if err := sm.applyDnsmasqConfigLocked(); err != nil {
  47. log.Printf("[DHCP-Server] Error applying dnsmasq config: %v", err)
  48. }
  49. return sm
  50. }
  51. func (sm *ServerManager) load() error {
  52. sm.mu.Lock()
  53. defer sm.mu.Unlock()
  54. data, err := os.ReadFile(sm.storePath)
  55. if err != nil {
  56. if os.IsNotExist(err) {
  57. // Populate default option presets (MikroTik standard presets)
  58. sm.initDefaultOptions()
  59. return nil
  60. }
  61. return err
  62. }
  63. var store DHCPServerConfigStore
  64. if err := json.Unmarshal(data, &store); err != nil {
  65. return err
  66. }
  67. for _, s := range store.Servers {
  68. sm.servers[s.ID] = s
  69. }
  70. for _, n := range store.Networks {
  71. sm.networks[n.ID] = n
  72. }
  73. for _, l := range store.StaticLeases {
  74. sm.staticLeases[l.ID] = l
  75. }
  76. for _, o := range store.Options {
  77. sm.options[o.ID] = o
  78. }
  79. if len(sm.options) == 0 {
  80. sm.initDefaultOptions()
  81. }
  82. return nil
  83. }
  84. func (sm *ServerManager) initDefaultOptions() {
  85. presets := []DHCPOption{
  86. {ID: "opt-ntp", Name: "ntp-servers", Code: 42, Value: "192.168.88.1", Comment: "NTP Time Servers (Option 42)"},
  87. {ID: "opt-tftp-name", Name: "tftp-server-name", Code: 66, Value: "192.168.88.50", Comment: "TFTP Server Name / IP (Option 66)"},
  88. {ID: "opt-bootfile", Name: "bootfile-name", Code: 67, Value: "pxelinux.0", Comment: "PXE Boot File (Option 67)"},
  89. {ID: "opt-unifi", Name: "unifi-controller", Code: 43, Value: "01:04:c0:a8:58:0a", Comment: "UniFi Inform Controller (Option 43 Hex IP)"},
  90. {ID: "opt-voip", Name: "cisco-voip-tftp", Code: 150, Value: "192.168.88.50", Comment: "Cisco VoIP TFTP (Option 150)"},
  91. {ID: "opt-capwap", Name: "capwap-ac", Code: 138, Value: "192.168.88.10", Comment: "CAPWAP Access Controller (Option 138)"},
  92. {ID: "opt-wpad", Name: "wpad-url", Code: 252, Value: "\"http://wpad/wpad.dat\"", Comment: "Web Proxy Auto-Discovery (Option 252)"},
  93. }
  94. for _, p := range presets {
  95. sm.options[p.ID] = p
  96. }
  97. }
  98. func (sm *ServerManager) saveLocked() error {
  99. store := DHCPServerConfigStore{
  100. Servers: make([]DHCPServerInstance, 0, len(sm.servers)),
  101. Networks: make([]DHCPNetwork, 0, len(sm.networks)),
  102. StaticLeases: make([]DHCPLease, 0, len(sm.staticLeases)),
  103. Options: make([]DHCPOption, 0, len(sm.options)),
  104. }
  105. for _, s := range sm.servers {
  106. copyS := s
  107. copyS.Running = false
  108. copyS.ActiveLeases = 0
  109. store.Servers = append(store.Servers, copyS)
  110. }
  111. for _, n := range sm.networks {
  112. store.Networks = append(store.Networks, n)
  113. }
  114. for _, l := range sm.staticLeases {
  115. store.StaticLeases = append(store.StaticLeases, l)
  116. }
  117. for _, o := range sm.options {
  118. store.Options = append(store.Options, o)
  119. }
  120. data, err := json.MarshalIndent(store, "", " ")
  121. if err != nil {
  122. return err
  123. }
  124. return os.WriteFile(sm.storePath, data, 0644)
  125. }
  126. func (sm *ServerManager) applyDnsmasqConfigLocked() error {
  127. var sb strings.Builder
  128. sb.WriteString("# ==================================================================\n")
  129. sb.WriteString("# DebTik DHCP Server Configuration (MikroTik RouterOS style)\n")
  130. sb.WriteString("# Auto-generated by DebTik Network Appliance. DO NOT EDIT DIRECTLY.\n")
  131. sb.WriteString("# ==================================================================\n\n")
  132. sb.WriteString("dhcp-authoritative\n")
  133. sb.WriteString("dhcp-leasefile=/var/lib/misc/dnsmasq.leases\n")
  134. sb.WriteString("log-dhcp\n")
  135. sb.WriteString("bind-dynamic\n\n")
  136. activeServerCount := 0
  137. for _, srv := range sm.servers {
  138. if srv.Disabled {
  139. continue
  140. }
  141. activeServerCount++
  142. tag := sanitizeTag(srv.Name)
  143. sb.WriteString(fmt.Sprintf("# --- DHCP Server: %s on %s ---\n", srv.Name, srv.Interface))
  144. sb.WriteString(fmt.Sprintf("interface=%s\n", srv.Interface))
  145. leaseTime := srv.LeaseTime
  146. if leaseTime == "" {
  147. leaseTime = "12h"
  148. }
  149. // Parse Address Pool range (e.g. 192.168.88.10-192.168.88.254)
  150. startIP, endIP, netmask := parseAddressPool(srv.AddressPool)
  151. if startIP != "" && endIP != "" {
  152. if netmask != "" {
  153. sb.WriteString(fmt.Sprintf("dhcp-range=set:%s,interface:%s,%s,%s,%s,%s\n", tag, srv.Interface, startIP, endIP, netmask, leaseTime))
  154. } else {
  155. sb.WriteString(fmt.Sprintf("dhcp-range=set:%s,interface:%s,%s,%s,%s\n", tag, srv.Interface, startIP, endIP, leaseTime))
  156. }
  157. }
  158. sb.WriteString("\n")
  159. }
  160. // Subnet Networks & Options
  161. sb.WriteString("# --- DHCP Networks & Options (/ip dhcp-server network) ---\n")
  162. for _, netw := range sm.networks {
  163. sb.WriteString(fmt.Sprintf("# Subnet: %s\n", netw.Address))
  164. _, ipNet, err := net.ParseCIDR(netw.Address)
  165. networkTag := ""
  166. if err == nil && ipNet != nil {
  167. networkTag = fmt.Sprintf("net_%s", strings.ReplaceAll(ipNet.IP.String(), ".", "_"))
  168. // Match server range tag or subnet tag
  169. sb.WriteString(fmt.Sprintf("dhcp-range=set:%s,%s,static\n", networkTag, ipNet.IP.String()))
  170. }
  171. // Determine target tag (either network tag or global/matched)
  172. targetTag := ""
  173. if networkTag != "" {
  174. targetTag = fmt.Sprintf("tag:%s,", networkTag)
  175. }
  176. // Option 3: Gateway (Router)
  177. if netw.Gateway != "" {
  178. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:router,%s\n", targetTag, netw.Gateway))
  179. }
  180. // Option 6: DNS Servers
  181. if len(netw.DNSServers) > 0 {
  182. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:dns-server,%s\n", targetTag, strings.Join(netw.DNSServers, ",")))
  183. }
  184. // Option 15: Domain Name
  185. if netw.Domain != "" {
  186. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:domain-name,%s\n", targetTag, netw.Domain))
  187. }
  188. // Option 42: NTP Servers (MikroTik /ip dhcp-server network ntp-server)
  189. if len(netw.NTPServers) > 0 {
  190. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:ntp-server,%s\n", targetTag, strings.Join(netw.NTPServers, ",")))
  191. }
  192. // Option 66: TFTP Server Name / IP (next-server)
  193. if netw.NextServer != "" {
  194. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:tftp-server,%s\n", targetTag, netw.NextServer))
  195. }
  196. // Option 67: Boot File Name (PXE)
  197. if netw.BootFileName != "" {
  198. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:bootfile-name,%s\n", targetTag, netw.BootFileName))
  199. if netw.NextServer != "" {
  200. sb.WriteString(fmt.Sprintf("dhcp-boot=%s%s,,%s\n", targetTag, netw.BootFileName, netw.NextServer))
  201. } else {
  202. sb.WriteString(fmt.Sprintf("dhcp-boot=%s%s\n", targetTag, netw.BootFileName))
  203. }
  204. }
  205. // Option 44: WINS Servers
  206. if len(netw.WINSServers) > 0 {
  207. sb.WriteString(fmt.Sprintf("dhcp-option=%soption:netbios-ns,%s\n", targetTag, strings.Join(netw.WINSServers, ",")))
  208. }
  209. // Linked custom DHCP options
  210. for _, optID := range netw.DHCPOptions {
  211. if opt, exists := sm.options[optID]; exists {
  212. sb.WriteString(fmt.Sprintf("dhcp-option=%s%d,%s # %s\n", targetTag, opt.Code, opt.Value, opt.Name))
  213. }
  214. }
  215. sb.WriteString("\n")
  216. }
  217. // Static Leases & Reservations (/ip dhcp-server lease)
  218. sb.WriteString("# --- Static Leases & Reservations (/ip dhcp-server lease) ---\n")
  219. for _, l := range sm.staticLeases {
  220. if l.Disabled {
  221. continue
  222. }
  223. mac := strings.ToLower(strings.TrimSpace(l.MACAddress))
  224. ip := strings.TrimSpace(l.Address)
  225. hostname := strings.TrimSpace(l.HostName)
  226. if hostname == "" {
  227. hostname = "reserved"
  228. }
  229. if mac != "" && ip != "" {
  230. sb.WriteString(fmt.Sprintf("dhcp-host=%s,%s,%s,infinite\n", mac, ip, hostname))
  231. }
  232. }
  233. // Write config to disk
  234. if err := os.WriteFile(sm.confPath, []byte(sb.String()), 0644); err != nil {
  235. return fmt.Errorf("failed to write dnsmasq config: %w", err)
  236. }
  237. // Restart or reload dnsmasq
  238. if activeServerCount > 0 {
  239. _ = exec.Command("systemctl", "restart", "dnsmasq").Run()
  240. } else {
  241. _ = exec.Command("systemctl", "reload", "dnsmasq").Run()
  242. }
  243. return nil
  244. }
  245. func (sm *ServerManager) ListServers() []DHCPServerInstance {
  246. sm.mu.RLock()
  247. defer sm.mu.RUnlock()
  248. activeLeases := sm.countActiveLeasesByServer()
  249. list := make([]DHCPServerInstance, 0, len(sm.servers))
  250. for _, s := range sm.servers {
  251. copyS := s
  252. copyS.Running = !s.Disabled && isInterfaceUp(s.Interface)
  253. copyS.ActiveLeases = activeLeases[s.Name]
  254. list = append(list, copyS)
  255. }
  256. return list
  257. }
  258. func (sm *ServerManager) GetServer(id string) (DHCPServerInstance, bool) {
  259. sm.mu.RLock()
  260. defer sm.mu.RUnlock()
  261. s, ok := sm.servers[id]
  262. if !ok {
  263. return DHCPServerInstance{}, false
  264. }
  265. s.Running = !s.Disabled && isInterfaceUp(s.Interface)
  266. return s, true
  267. }
  268. func (sm *ServerManager) SaveServer(inst DHCPServerInstance) (DHCPServerInstance, error) {
  269. sm.mu.Lock()
  270. defer sm.mu.Unlock()
  271. if inst.ID == "" {
  272. inst.ID = fmt.Sprintf("dhcp-%d", time.Now().UnixNano())
  273. }
  274. if inst.Name == "" {
  275. inst.Name = fmt.Sprintf("dhcp%d", len(sm.servers)+1)
  276. }
  277. if inst.LeaseTime == "" {
  278. inst.LeaseTime = "12h"
  279. }
  280. inst.Authoritative = true
  281. sm.servers[inst.ID] = inst
  282. if err := sm.saveLocked(); err != nil {
  283. return inst, err
  284. }
  285. _ = sm.applyDnsmasqConfigLocked()
  286. return inst, nil
  287. }
  288. func (sm *ServerManager) DeleteServer(id string) error {
  289. sm.mu.Lock()
  290. defer sm.mu.Unlock()
  291. delete(sm.servers, id)
  292. if err := sm.saveLocked(); err != nil {
  293. return err
  294. }
  295. return sm.applyDnsmasqConfigLocked()
  296. }
  297. func (sm *ServerManager) ToggleServerState(id string, disabled bool) error {
  298. sm.mu.Lock()
  299. defer sm.mu.Unlock()
  300. s, exists := sm.servers[id]
  301. if !exists {
  302. return fmt.Errorf("server not found: %s", id)
  303. }
  304. s.Disabled = disabled
  305. sm.servers[id] = s
  306. if err := sm.saveLocked(); err != nil {
  307. return err
  308. }
  309. return sm.applyDnsmasqConfigLocked()
  310. }
  311. // ======================== NETWORKS ========================
  312. func (sm *ServerManager) ListNetworks() []DHCPNetwork {
  313. sm.mu.RLock()
  314. defer sm.mu.RUnlock()
  315. list := make([]DHCPNetwork, 0, len(sm.networks))
  316. for _, n := range sm.networks {
  317. list = append(list, n)
  318. }
  319. return list
  320. }
  321. func (sm *ServerManager) SaveNetwork(netw DHCPNetwork) (DHCPNetwork, error) {
  322. sm.mu.Lock()
  323. defer sm.mu.Unlock()
  324. if netw.ID == "" {
  325. netw.ID = fmt.Sprintf("net-%d", time.Now().UnixNano())
  326. }
  327. if netw.Address == "" {
  328. return netw, fmt.Errorf("subnet address (CIDR) is required")
  329. }
  330. sm.networks[netw.ID] = netw
  331. if err := sm.saveLocked(); err != nil {
  332. return netw, err
  333. }
  334. _ = sm.applyDnsmasqConfigLocked()
  335. return netw, nil
  336. }
  337. func (sm *ServerManager) DeleteNetwork(id string) error {
  338. sm.mu.Lock()
  339. defer sm.mu.Unlock()
  340. delete(sm.networks, id)
  341. if err := sm.saveLocked(); err != nil {
  342. return err
  343. }
  344. return sm.applyDnsmasqConfigLocked()
  345. }
  346. // ======================== LEASES ========================
  347. func (sm *ServerManager) ListLeases() []DHCPLease {
  348. sm.mu.RLock()
  349. defer sm.mu.RUnlock()
  350. dynamicLeases := sm.readDnsmasqLeasesFile()
  351. staticMap := make(map[string]DHCPLease)
  352. for _, st := range sm.staticLeases {
  353. staticMap[strings.ToLower(st.MACAddress)] = st
  354. }
  355. result := make([]DHCPLease, 0)
  356. seenMACs := make(map[string]bool)
  357. // Combine dynamic leases
  358. for _, dl := range dynamicLeases {
  359. macLower := strings.ToLower(dl.MACAddress)
  360. seenMACs[macLower] = true
  361. if st, isStatic := staticMap[macLower]; isStatic {
  362. dl.Dynamic = false
  363. dl.Status = "static"
  364. dl.Comment = st.Comment
  365. dl.Disabled = st.Disabled
  366. dl.ExpiresIn = "never (static)"
  367. } else {
  368. dl.Dynamic = true
  369. dl.Status = "bound"
  370. }
  371. result = append(result, dl)
  372. }
  373. // Add static reservations that haven't acquired lease yet
  374. for macLower, st := range staticMap {
  375. if !seenMACs[macLower] {
  376. st.Dynamic = false
  377. st.Status = "waiting"
  378. st.ExpiresIn = "never (static)"
  379. result = append(result, st)
  380. }
  381. }
  382. return result
  383. }
  384. func (sm *ServerManager) readDnsmasqLeasesFile() []DHCPLease {
  385. data, err := os.ReadFile(sm.leasesPath)
  386. if err != nil {
  387. return nil
  388. }
  389. lines := strings.Split(string(data), "\n")
  390. leases := make([]DHCPLease, 0, len(lines))
  391. now := time.Now()
  392. for _, line := range lines {
  393. line = strings.TrimSpace(line)
  394. if line == "" {
  395. continue
  396. }
  397. // Format: <timestamp> <mac> <ip> <hostname> <client-id>
  398. parts := strings.Fields(line)
  399. if len(parts) < 3 {
  400. continue
  401. }
  402. expTimestamp, _ := strconv.ParseInt(parts[0], 10, 64)
  403. mac := parts[1]
  404. ip := parts[2]
  405. hostname := "*"
  406. if len(parts) >= 4 && parts[3] != "*" {
  407. hostname = parts[3]
  408. }
  409. clientID := ""
  410. if len(parts) >= 5 && parts[4] != "*" {
  411. clientID = parts[4]
  412. }
  413. expiresAt := time.Unix(expTimestamp, 0)
  414. expiresIn := "expired"
  415. if expiresAt.After(now) {
  416. dur := expiresAt.Sub(now).Truncate(time.Minute)
  417. hours := int(dur.Hours())
  418. mins := int(dur.Minutes()) % 60
  419. expiresIn = fmt.Sprintf("%dh %dm", hours, mins)
  420. }
  421. // Determine server by matching IP against configured networks/pools
  422. serverName := sm.matchServerForIP(ip)
  423. leases = append(leases, DHCPLease{
  424. ID: fmt.Sprintf("lease-%s", strings.ReplaceAll(mac, ":", "")),
  425. Address: ip,
  426. MACAddress: mac,
  427. HostName: hostname,
  428. ClientID: clientID,
  429. Server: serverName,
  430. ExpiresAt: expiresAt,
  431. ExpiresIn: expiresIn,
  432. Dynamic: true,
  433. Status: "bound",
  434. })
  435. }
  436. return leases
  437. }
  438. func (sm *ServerManager) matchServerForIP(ipStr string) string {
  439. targetIP := net.ParseIP(ipStr)
  440. if targetIP == nil {
  441. return "dhcp1"
  442. }
  443. for _, srv := range sm.servers {
  444. startIP, endIP, _ := parseAddressPool(srv.AddressPool)
  445. if startIP != "" && endIP != "" {
  446. s := net.ParseIP(startIP)
  447. e := net.ParseIP(endIP)
  448. if s != nil && e != nil && compareIPs(targetIP, s) >= 0 && compareIPs(targetIP, e) <= 0 {
  449. return srv.Name
  450. }
  451. }
  452. }
  453. return "dhcp1"
  454. }
  455. func (sm *ServerManager) countActiveLeasesByServer() map[string]int {
  456. counts := make(map[string]int)
  457. leases := sm.readDnsmasqLeasesFile()
  458. for _, l := range leases {
  459. counts[l.Server]++
  460. }
  461. return counts
  462. }
  463. func (sm *ServerManager) SaveStaticLease(lease DHCPLease) (DHCPLease, error) {
  464. sm.mu.Lock()
  465. defer sm.mu.Unlock()
  466. if lease.ID == "" {
  467. lease.ID = fmt.Sprintf("static-%d", time.Now().UnixNano())
  468. }
  469. if lease.MACAddress == "" || lease.Address == "" {
  470. return lease, fmt.Errorf("both MAC address and IP address are required")
  471. }
  472. lease.MACAddress = strings.ToLower(lease.MACAddress)
  473. lease.Dynamic = false
  474. lease.Status = "static"
  475. sm.staticLeases[lease.ID] = lease
  476. if err := sm.saveLocked(); err != nil {
  477. return lease, err
  478. }
  479. _ = sm.applyDnsmasqConfigLocked()
  480. return lease, nil
  481. }
  482. func (sm *ServerManager) DeleteStaticLease(id string) error {
  483. sm.mu.Lock()
  484. defer sm.mu.Unlock()
  485. delete(sm.staticLeases, id)
  486. if err := sm.saveLocked(); err != nil {
  487. return err
  488. }
  489. return sm.applyDnsmasqConfigLocked()
  490. }
  491. func (sm *ServerManager) MakeStaticLease(mac, ip, hostname, comment string) (DHCPLease, error) {
  492. sm.mu.Lock()
  493. defer sm.mu.Unlock()
  494. macLower := strings.ToLower(strings.TrimSpace(mac))
  495. // Check if already in static leases
  496. for _, st := range sm.staticLeases {
  497. if strings.ToLower(st.MACAddress) == macLower {
  498. st.Address = ip
  499. if hostname != "" && hostname != "*" {
  500. st.HostName = hostname
  501. }
  502. if comment != "" {
  503. st.Comment = comment
  504. }
  505. sm.staticLeases[st.ID] = st
  506. _ = sm.saveLocked()
  507. _ = sm.applyDnsmasqConfigLocked()
  508. return st, nil
  509. }
  510. }
  511. id := fmt.Sprintf("static-%d", time.Now().UnixNano())
  512. st := DHCPLease{
  513. ID: id,
  514. MACAddress: macLower,
  515. Address: ip,
  516. HostName: hostname,
  517. Server: "all",
  518. Comment: comment,
  519. Dynamic: false,
  520. Status: "static",
  521. }
  522. sm.staticLeases[id] = st
  523. if err := sm.saveLocked(); err != nil {
  524. return st, err
  525. }
  526. _ = sm.applyDnsmasqConfigLocked()
  527. return st, nil
  528. }
  529. // ======================== OPTIONS ========================
  530. func (sm *ServerManager) ListOptions() []DHCPOption {
  531. sm.mu.RLock()
  532. defer sm.mu.RUnlock()
  533. list := make([]DHCPOption, 0, len(sm.options))
  534. for _, o := range sm.options {
  535. list = append(list, o)
  536. }
  537. return list
  538. }
  539. func (sm *ServerManager) SaveOption(opt DHCPOption) (DHCPOption, error) {
  540. sm.mu.Lock()
  541. defer sm.mu.Unlock()
  542. if opt.ID == "" {
  543. opt.ID = fmt.Sprintf("opt-%d", time.Now().UnixNano())
  544. }
  545. if opt.Name == "" || opt.Code <= 0 || opt.Code > 254 {
  546. return opt, fmt.Errorf("valid option name and code (1-254) are required")
  547. }
  548. sm.options[opt.ID] = opt
  549. if err := sm.saveLocked(); err != nil {
  550. return opt, err
  551. }
  552. _ = sm.applyDnsmasqConfigLocked()
  553. return opt, nil
  554. }
  555. func (sm *ServerManager) DeleteOption(id string) error {
  556. sm.mu.Lock()
  557. defer sm.mu.Unlock()
  558. delete(sm.options, id)
  559. if err := sm.saveLocked(); err != nil {
  560. return err
  561. }
  562. return sm.applyDnsmasqConfigLocked()
  563. }
  564. // ======================== SETUP WIZARD (MikroTik style) ========================
  565. func (sm *ServerManager) RunSetupWizard(req DHCPServerSetupRequest) error {
  566. sm.mu.Lock()
  567. defer sm.mu.Unlock()
  568. if req.Interface == "" {
  569. return fmt.Errorf("interface is required")
  570. }
  571. if req.AddressSpace == "" {
  572. req.AddressSpace = "192.168.88.0/24"
  573. }
  574. if req.Gateway == "" {
  575. req.Gateway = "192.168.88.1"
  576. }
  577. if req.AddressPool == "" {
  578. req.AddressPool = "192.168.88.10-192.168.88.254"
  579. }
  580. if len(req.DNSServers) == 0 {
  581. req.DNSServers = []string{req.Gateway, "1.1.1.1"}
  582. }
  583. if req.LeaseTime == "" {
  584. req.LeaseTime = "12h"
  585. }
  586. srvID := fmt.Sprintf("dhcp-%s", req.Interface)
  587. srv := DHCPServerInstance{
  588. ID: srvID,
  589. Name: fmt.Sprintf("dhcp-%s", req.Interface),
  590. Interface: req.Interface,
  591. AddressPool: req.AddressPool,
  592. LeaseTime: req.LeaseTime,
  593. Authoritative: true,
  594. Disabled: false,
  595. Comment: "Created by DHCP Setup Wizard",
  596. }
  597. sm.servers[srvID] = srv
  598. netID := fmt.Sprintf("net-%s", strings.ReplaceAll(req.AddressSpace, "/", "_"))
  599. netw := DHCPNetwork{
  600. ID: netID,
  601. Address: req.AddressSpace,
  602. Gateway: req.Gateway,
  603. DNSServers: req.DNSServers,
  604. Domain: req.Domain,
  605. NTPServers: req.NTPServers,
  606. NextServer: req.NextServer,
  607. BootFileName: req.BootFileName,
  608. Comment: fmt.Sprintf("Network for %s", req.Interface),
  609. }
  610. sm.networks[netID] = netw
  611. if err := sm.saveLocked(); err != nil {
  612. return err
  613. }
  614. return sm.applyDnsmasqConfigLocked()
  615. }
  616. // ======================== SUMMARY ========================
  617. func (sm *ServerManager) GetSummary() DHCPServerSummary {
  618. sm.mu.RLock()
  619. defer sm.mu.RUnlock()
  620. summary := DHCPServerSummary{
  621. TotalServers: len(sm.servers),
  622. TotalNetworks: len(sm.networks),
  623. StaticLeases: len(sm.staticLeases),
  624. TotalOptions: len(sm.options),
  625. }
  626. for _, s := range sm.servers {
  627. if !s.Disabled && isInterfaceUp(s.Interface) {
  628. summary.ActiveServers++
  629. }
  630. }
  631. leases := sm.readDnsmasqLeasesFile()
  632. summary.ActiveLeases = len(leases)
  633. return summary
  634. }
  635. // ======================== HELPERS ========================
  636. func sanitizeTag(name string) string {
  637. res := strings.Map(func(r rune) rune {
  638. if (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || (r >= '0' && r <= '9') || r == '_' {
  639. return r
  640. }
  641. return '_'
  642. }, name)
  643. return strings.ToLower(res)
  644. }
  645. func parseAddressPool(pool string) (startIP, endIP, netmask string) {
  646. pool = strings.TrimSpace(pool)
  647. parts := strings.Split(pool, "-")
  648. if len(parts) == 2 {
  649. return strings.TrimSpace(parts[0]), strings.TrimSpace(parts[1]), ""
  650. }
  651. return "", "", ""
  652. }
  653. func isInterfaceUp(ifaceName string) bool {
  654. iface, err := net.InterfaceByName(ifaceName)
  655. if err != nil {
  656. return false
  657. }
  658. return (iface.Flags & net.FlagUp) != 0
  659. }
  660. func compareIPs(a, b net.IP) int {
  661. a4 := a.To4()
  662. b4 := b.To4()
  663. if a4 == nil || b4 == nil {
  664. return 0
  665. }
  666. for i := 0; i < 4; i++ {
  667. if a4[i] < b4[i] {
  668. return -1
  669. }
  670. if a4[i] > b4[i] {
  671. return 1
  672. }
  673. }
  674. return 0
  675. }