server_manager.go 20 KB

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