executor.go 79 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556
  1. package cli
  2. import (
  3. "context"
  4. "fmt"
  5. "net"
  6. "os/exec"
  7. "sort"
  8. "strconv"
  9. "strings"
  10. "time"
  11. "proxmox-router/internal/auth"
  12. "proxmox-router/internal/bridge"
  13. "proxmox-router/internal/bwtest"
  14. "proxmox-router/internal/certs"
  15. "proxmox-router/internal/dhcp"
  16. "proxmox-router/internal/dns"
  17. "proxmox-router/internal/dynamicrouting"
  18. "proxmox-router/internal/firewall"
  19. "proxmox-router/internal/logging"
  20. "proxmox-router/internal/network"
  21. "proxmox-router/internal/routing"
  22. "proxmox-router/internal/services"
  23. "proxmox-router/internal/system"
  24. "proxmox-router/internal/vpn"
  25. "proxmox-router/internal/vrrp"
  26. )
  27. type Executor struct {
  28. authMgr *auth.Manager
  29. netMgr *network.Manager
  30. routingMgr *routing.RoutingManager
  31. dynRoutingMgr *dynamicrouting.Manager
  32. fwMgr *firewall.FirewallManager
  33. dhcpMgr *dhcp.Manager
  34. dhcpServMgr *dhcp.ServerManager
  35. dnsMgr *dns.Manager
  36. bridgeMgr *bridge.BridgeManager
  37. vpnMgr *vpn.Manager
  38. vrrpMgr *vrrp.Manager
  39. bwtestMgr *bwtest.Manager
  40. servicesMgr *services.Manager
  41. certMgr *certs.CertManager
  42. }
  43. func NewExecutor(
  44. authMgr *auth.Manager,
  45. netMgr *network.Manager,
  46. routingMgr *routing.RoutingManager,
  47. dynRoutingMgr *dynamicrouting.Manager,
  48. fwMgr *firewall.FirewallManager,
  49. dhcpMgr *dhcp.Manager,
  50. dhcpServMgr *dhcp.ServerManager,
  51. dnsMgr *dns.Manager,
  52. bridgeMgr *bridge.BridgeManager,
  53. vpnMgr *vpn.Manager,
  54. vrrpMgr *vrrp.Manager,
  55. bwtestMgr *bwtest.Manager,
  56. servicesMgr *services.Manager,
  57. certMgr *certs.CertManager,
  58. ) *Executor {
  59. return &Executor{
  60. authMgr: authMgr,
  61. netMgr: netMgr,
  62. routingMgr: routingMgr,
  63. dynRoutingMgr: dynRoutingMgr,
  64. fwMgr: fwMgr,
  65. dhcpMgr: dhcpMgr,
  66. dhcpServMgr: dhcpServMgr,
  67. dnsMgr: dnsMgr,
  68. bridgeMgr: bridgeMgr,
  69. vpnMgr: vpnMgr,
  70. vrrpMgr: vrrpMgr,
  71. bwtestMgr: bwtestMgr,
  72. servicesMgr: servicesMgr,
  73. certMgr: certMgr,
  74. }
  75. }
  76. func (e *Executor) Execute(rawCmd string, user *auth.PublicUser) (output string, shouldExit bool) {
  77. // A malformed command must never take down the router or its tunnels.
  78. defer func() {
  79. if recover() != nil {
  80. logging.Log("cli,error", "Command handler panicked")
  81. output = ColorRed + "Internal command error; the router is still running.\n" + ColorReset
  82. shouldExit = false
  83. }
  84. }()
  85. cmd := strings.TrimSpace(rawCmd)
  86. if cmd == "" {
  87. return "", false
  88. }
  89. cmd = strings.TrimPrefix(cmd, "/")
  90. parts, err := splitCommand(cmd)
  91. if err != nil {
  92. return ColorRed + "Syntax error: " + err.Error() + "\n" + ColorReset, false
  93. }
  94. if len(parts) == 0 {
  95. return "", false
  96. }
  97. mainCmd := strings.ToLower(parts[0])
  98. switch mainCmd {
  99. case "quit", "exit":
  100. return ColorYellow + "Connection closed. Goodbye!\n" + ColorReset, true
  101. case "help", "?":
  102. return e.showHelp(), false
  103. case "clear":
  104. return "\033[2J\033[H", false
  105. case "ping":
  106. return e.handlePing(parts[1:]), false
  107. case "traceroute":
  108. return e.handleTraceroute(parts[1:]), false
  109. case "log":
  110. return e.handleLog(parts[1:]), false
  111. case "system":
  112. return e.handleSystem(parts[1:], user), false
  113. case "interface":
  114. return e.handleInterface(parts[1:], user), false
  115. case "ip":
  116. return e.handleIP(parts[1:], user), false
  117. case "routing":
  118. return e.handleRouting(parts[1:], user), false
  119. case "tool":
  120. if user == nil || user.Role == auth.RoleRead {
  121. return ColorRed + "Permission denied: tools require write privileges\n" + ColorReset, false
  122. }
  123. return e.handleTool(parts[1:]), false
  124. case "user":
  125. return e.handleUser(parts[1:], user), false
  126. case "certificate", "certificates":
  127. return e.handleCertificate(parts[1:], user), false
  128. default:
  129. return fmt.Sprintf(ColorRed+"Unknown command or syntax error: %q. Type 'help' or '?' for command list.\n"+ColorReset, cmd), false
  130. }
  131. }
  132. func parseArgs(parts []string) ([]string, map[string]string) {
  133. var pos []string
  134. params := make(map[string]string)
  135. for _, p := range parts {
  136. if idx := strings.Index(p, "="); idx != -1 {
  137. k := strings.ToLower(p[:idx])
  138. v := p[idx+1:]
  139. params[k] = v
  140. } else {
  141. pos = append(pos, p)
  142. }
  143. }
  144. return pos, params
  145. }
  146. // ==================== LOGGING ====================
  147. func (e *Executor) handleLog(parts []string) string {
  148. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" || parts[0] == "show" {
  149. limit := 50
  150. filter := ""
  151. args := parts
  152. if len(args) > 0 {
  153. args = args[1:]
  154. }
  155. _, params := parseArgs(args)
  156. if lStr, ok := params["limit"]; ok {
  157. if lVal, err := strconv.Atoi(lStr); err == nil && lVal > 0 {
  158. limit = lVal
  159. }
  160. }
  161. if fStr, ok := params["where"]; ok {
  162. filter = fStr
  163. } else if fStr, ok := params["filter"]; ok {
  164. filter = fStr
  165. }
  166. logger := logging.GetLogger()
  167. entries := logger.List(limit, filter)
  168. if len(entries) == 0 {
  169. return ColorYellow + "No log records found.\n" + ColorReset
  170. }
  171. tbl := NewTable("#", "TIME", "TOPICS", "MESSAGE")
  172. for _, logEntry := range entries {
  173. timeStr := logEntry.Time.Format("Jan/02 15:04:05")
  174. tbl.AddRow(
  175. fmt.Sprintf("%d", logEntry.ID),
  176. timeStr,
  177. logEntry.Topics,
  178. logEntry.Message,
  179. )
  180. }
  181. return tbl.String()
  182. }
  183. return ColorRed + "Usage: /log print [limit=N] [where=filter]\n" + ColorReset
  184. }
  185. // ==================== INTERFACES ====================
  186. func (e *Executor) handleInterface(parts []string, user *auth.PublicUser) string {
  187. if len(parts) == 0 {
  188. return e.printAllInterfaces()
  189. }
  190. sub := strings.ToLower(parts[0])
  191. switch sub {
  192. case "print", "pr", "show", "list":
  193. return e.printAllInterfaces()
  194. case "vlan":
  195. return e.handleInterfaceVLAN(parts[1:], user)
  196. case "bridge":
  197. return e.handleInterfaceBridge(parts[1:], user)
  198. case "wireguard":
  199. return e.handleInterfaceWireGuard(parts[1:], user)
  200. case "aerotun":
  201. return e.handleInterfaceAeroTUN(parts[1:], user)
  202. case "eoip":
  203. return e.handleInterfaceEoIP(parts[1:], user)
  204. case "vrrp":
  205. return e.handleInterfaceVRRP(parts[1:], user)
  206. case "pppoe-client":
  207. return e.handleInterfacePPPoE(parts[1:], user)
  208. case "enable":
  209. if len(parts) < 2 {
  210. return ColorRed + "Usage: /interface enable <name>\n" + ColorReset
  211. }
  212. if user.Role == auth.RoleRead {
  213. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  214. }
  215. name := parts[1]
  216. if err := e.netMgr.SetInterfaceState(name, true); err != nil {
  217. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  218. }
  219. return ColorGreen + fmt.Sprintf("Interface %s enabled successfully.\n", name) + ColorReset
  220. case "disable":
  221. if len(parts) < 2 {
  222. return ColorRed + "Usage: /interface disable <name>\n" + ColorReset
  223. }
  224. if user.Role == auth.RoleRead {
  225. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  226. }
  227. name := parts[1]
  228. if err := e.netMgr.SetInterfaceState(name, false); err != nil {
  229. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  230. }
  231. return ColorGreen + fmt.Sprintf("Interface %s disabled successfully.\n", name) + ColorReset
  232. case "set":
  233. if len(parts) < 2 {
  234. return ColorRed + "Usage: /interface set <name> mtu=<val>\n" + ColorReset
  235. }
  236. if user.Role == auth.RoleRead {
  237. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  238. }
  239. name := parts[1]
  240. _, params := parseArgs(parts[2:])
  241. if mtuStr, ok := params["mtu"]; ok {
  242. mtu, err := strconv.Atoi(mtuStr)
  243. if err != nil || mtu < 68 || mtu > 9000 {
  244. return ColorRed + "Invalid MTU value (68-9000)\n" + ColorReset
  245. }
  246. if err := e.netMgr.SetInterfaceMTU(name, mtu); err != nil {
  247. return ColorRed + "Error setting MTU: " + err.Error() + ColorReset + "\n"
  248. }
  249. return ColorGreen + fmt.Sprintf("Interface %s MTU updated to %d.\n", name, mtu) + ColorReset
  250. }
  251. return ColorRed + "Usage: /interface set <name> mtu=<val>\n" + ColorReset
  252. default:
  253. return ColorRed + "Unknown interface subcommand: " + sub + "\n" + ColorReset
  254. }
  255. }
  256. func (e *Executor) printAllInterfaces() string {
  257. ifaces, err := e.netMgr.ListInterfaces()
  258. if err != nil {
  259. return ColorRed + "Error listing interfaces: " + err.Error() + ColorReset + "\n"
  260. }
  261. tbl := NewTable("FLAGS", "NAME", "TYPE", "STATE", "MTU", "MAC ADDRESS", "IP ADDRESSES")
  262. for _, iface := range ifaces {
  263. isUp := strings.ToUpper(iface.State) == "UP"
  264. flags := ""
  265. if isUp {
  266. flags += "R"
  267. } else {
  268. flags += "X"
  269. }
  270. stateStr := ColorGreen + "UP" + ColorReset
  271. if !isUp {
  272. stateStr = ColorRed + "DOWN" + ColorReset
  273. }
  274. allIPs := append([]string{}, iface.IPv4...)
  275. allIPs = append(allIPs, iface.IPv6...)
  276. ips := strings.Join(allIPs, ", ")
  277. if ips == "" {
  278. ips = "-"
  279. }
  280. tbl.AddRow(flags, iface.Name, string(iface.Type), stateStr, fmt.Sprintf("%d", iface.MTU), iface.MAC, ips)
  281. }
  282. return tbl.String()
  283. }
  284. // --- VLAN ---
  285. func (e *Executor) handleInterfaceVLAN(parts []string, user *auth.PublicUser) string {
  286. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  287. vlans, err := e.netMgr.ListVLANs()
  288. if err != nil {
  289. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  290. }
  291. tbl := NewTable("#", "NAME", "VLAN-ID", "INTERFACE", "MTU", "STATE")
  292. for idx, v := range vlans {
  293. isUp := strings.ToUpper(v.State) == "UP"
  294. st := ColorGreen + "UP" + ColorReset
  295. if !isUp {
  296. st = ColorRed + "DOWN" + ColorReset
  297. }
  298. tbl.AddRow(fmt.Sprintf("%d", idx), v.Name, fmt.Sprintf("%d", v.VlanID), v.ParentName, fmt.Sprintf("%d", v.MTU), st)
  299. }
  300. return tbl.String()
  301. }
  302. sub := strings.ToLower(parts[0])
  303. if user.Role == auth.RoleRead {
  304. return ColorRed + "Permission denied: viewer role cannot modify VLANs\n" + ColorReset
  305. }
  306. switch sub {
  307. case "add":
  308. _, params := parseArgs(parts[1:])
  309. name := params["name"]
  310. parent := params["interface"]
  311. vlanIDStr := params["vlan-id"]
  312. if vlanIDStr == "" {
  313. vlanIDStr = params["vlan_id"]
  314. }
  315. vlanID, err := strconv.Atoi(vlanIDStr)
  316. if err != nil || vlanID < 1 || vlanID > 4094 {
  317. return ColorRed + "Invalid vlan-id (1-4094)\n" + ColorReset
  318. }
  319. if parent == "" {
  320. return ColorRed + "Missing parent interface (interface=...)\n" + ColorReset
  321. }
  322. if name == "" {
  323. name = fmt.Sprintf("%s.%d", parent, vlanID)
  324. }
  325. req := network.CreateVLANRequest{
  326. ParentName: parent,
  327. VlanID: vlanID,
  328. Name: name,
  329. AutoUp: true,
  330. }
  331. if _, err := e.netMgr.CreateVLAN(req); err != nil {
  332. return ColorRed + "Error creating VLAN: " + err.Error() + ColorReset + "\n"
  333. }
  334. return ColorGreen + fmt.Sprintf("VLAN interface %s created.\n", name) + ColorReset
  335. case "remove", "rem", "delete":
  336. pos, _ := parseArgs(parts[1:])
  337. if len(pos) == 0 {
  338. return ColorRed + "Usage: /interface vlan remove <name>\n" + ColorReset
  339. }
  340. name := pos[0]
  341. if err := e.netMgr.DeleteVLAN(name); err != nil {
  342. return ColorRed + "Error deleting VLAN: " + err.Error() + ColorReset + "\n"
  343. }
  344. return ColorGreen + fmt.Sprintf("VLAN %s removed.\n", name) + ColorReset
  345. default:
  346. return ColorRed + "Usage: /interface vlan [print|add|remove]\n" + ColorReset
  347. }
  348. }
  349. // --- Bridge ---
  350. func (e *Executor) handleInterfaceBridge(parts []string, user *auth.PublicUser) string {
  351. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  352. if e.bridgeMgr == nil {
  353. return ColorRed + "Bridge manager not available.\n" + ColorReset
  354. }
  355. bridges := e.bridgeMgr.ListBridges()
  356. tbl := NewTable("FLAGS", "NAME", "MTU", "MAC-ADDRESS", "RUNNING", "PORTS")
  357. for _, b := range bridges {
  358. flags := ""
  359. if b.Disabled {
  360. flags += "X"
  361. } else {
  362. flags += "R"
  363. }
  364. tbl.AddRow(flags, b.Name, fmt.Sprintf("%d", b.MTU), b.MACAddress, fmt.Sprintf("%t", b.Running), fmt.Sprintf("%d", b.PortsCount))
  365. }
  366. return tbl.String()
  367. }
  368. sub := strings.ToLower(parts[0])
  369. if sub == "port" {
  370. return e.handleBridgePort(parts[1:], user)
  371. }
  372. if user.Role == auth.RoleRead {
  373. return ColorRed + "Permission denied: viewer role cannot modify bridge\n" + ColorReset
  374. }
  375. switch sub {
  376. case "add":
  377. _, params := parseArgs(parts[1:])
  378. name := params["name"]
  379. if name == "" {
  380. return ColorRed + "Usage: /interface bridge add name=<name>\n" + ColorReset
  381. }
  382. b := bridge.BridgeInstance{
  383. Name: name,
  384. Disabled: false,
  385. }
  386. if err := e.bridgeMgr.SaveBridge(b); err != nil {
  387. return ColorRed + "Error adding bridge: " + err.Error() + ColorReset + "\n"
  388. }
  389. return ColorGreen + fmt.Sprintf("Bridge %s added.\n", name) + ColorReset
  390. case "remove", "rem", "delete":
  391. pos, _ := parseArgs(parts[1:])
  392. if len(pos) == 0 {
  393. return ColorRed + "Usage: /interface bridge remove <id|name>\n" + ColorReset
  394. }
  395. if err := e.bridgeMgr.DeleteBridge(pos[0]); err != nil {
  396. return ColorRed + "Error deleting bridge: " + err.Error() + ColorReset + "\n"
  397. }
  398. return ColorGreen + fmt.Sprintf("Bridge %s removed.\n", pos[0]) + ColorReset
  399. case "enable":
  400. pos, _ := parseArgs(parts[1:])
  401. if len(pos) == 0 {
  402. return ColorRed + "Usage: /interface bridge enable <id|name>\n" + ColorReset
  403. }
  404. if err := e.bridgeMgr.SetBridgeState(pos[0], false); err != nil {
  405. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  406. }
  407. return ColorGreen + fmt.Sprintf("Bridge %s enabled.\n", pos[0]) + ColorReset
  408. case "disable":
  409. pos, _ := parseArgs(parts[1:])
  410. if len(pos) == 0 {
  411. return ColorRed + "Usage: /interface bridge disable <id|name>\n" + ColorReset
  412. }
  413. if err := e.bridgeMgr.SetBridgeState(pos[0], true); err != nil {
  414. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  415. }
  416. return ColorGreen + fmt.Sprintf("Bridge %s disabled.\n", pos[0]) + ColorReset
  417. default:
  418. return ColorRed + "Usage: /interface bridge [print|add|remove|enable|disable|port]\n" + ColorReset
  419. }
  420. }
  421. func (e *Executor) handleBridgePort(parts []string, user *auth.PublicUser) string {
  422. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  423. ports := e.bridgeMgr.ListPorts()
  424. tbl := NewTable("FLAGS", "INTERFACE", "BRIDGE", "STATUS", "PVID")
  425. for _, p := range ports {
  426. flags := ""
  427. if p.Disabled {
  428. flags += "X"
  429. } else {
  430. flags += "H"
  431. }
  432. tbl.AddRow(flags, p.Interface, p.Bridge, p.Status, fmt.Sprintf("%d", p.PVID))
  433. }
  434. return tbl.String()
  435. }
  436. if user.Role == auth.RoleRead {
  437. return ColorRed + "Permission denied: viewer role cannot modify bridge ports\n" + ColorReset
  438. }
  439. sub := strings.ToLower(parts[0])
  440. switch sub {
  441. case "add":
  442. _, params := parseArgs(parts[1:])
  443. br := params["bridge"]
  444. iface := params["interface"]
  445. if br == "" || iface == "" {
  446. return ColorRed + "Usage: /interface bridge port add bridge=<bridge> interface=<iface>\n" + ColorReset
  447. }
  448. port := bridge.BridgePort{
  449. Bridge: br,
  450. Interface: iface,
  451. Disabled: false,
  452. }
  453. if err := e.bridgeMgr.SavePort(port); err != nil {
  454. return ColorRed + "Error adding bridge port: " + err.Error() + ColorReset + "\n"
  455. }
  456. return ColorGreen + fmt.Sprintf("Port %s attached to bridge %s.\n", iface, br) + ColorReset
  457. case "remove", "rem", "delete":
  458. pos, _ := parseArgs(parts[1:])
  459. if len(pos) == 0 {
  460. return ColorRed + "Usage: /interface bridge port remove <id>\n" + ColorReset
  461. }
  462. if err := e.bridgeMgr.DeletePort(pos[0]); err != nil {
  463. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  464. }
  465. return ColorGreen + fmt.Sprintf("Bridge port %s removed.\n", pos[0]) + ColorReset
  466. default:
  467. return ColorRed + "Usage: /interface bridge port [print|add|remove]\n" + ColorReset
  468. }
  469. }
  470. // --- WireGuard ---
  471. func (e *Executor) handleInterfaceWireGuard(parts []string, user *auth.PublicUser) string {
  472. if e.vpnMgr == nil || e.vpnMgr.WireGuard == nil {
  473. return ColorRed + "WireGuard subsystem not available.\n" + ColorReset
  474. }
  475. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  476. ifaces := e.vpnMgr.WireGuard.ListInterfaces()
  477. tbl := NewTable("FLAGS", "NAME", "LISTEN-PORT", "PUBLIC-KEY", "STATUS")
  478. for _, w := range ifaces {
  479. flags := ""
  480. if w.Disabled {
  481. flags += "X"
  482. } else {
  483. flags += "R"
  484. }
  485. st := "running"
  486. if !w.Running {
  487. st = "stopped"
  488. }
  489. tbl.AddRow(flags, w.Name, fmt.Sprintf("%d", w.ListenPort), w.PublicKey, st)
  490. }
  491. return tbl.String()
  492. }
  493. sub := strings.ToLower(parts[0])
  494. if sub == "peers" || sub == "peer" {
  495. return e.handleWireGuardPeers(parts[1:], user)
  496. }
  497. if user.Role == auth.RoleRead {
  498. return ColorRed + "Permission denied: viewer role cannot modify wireguard\n" + ColorReset
  499. }
  500. switch sub {
  501. case "add":
  502. _, params := parseArgs(parts[1:])
  503. name := params["name"]
  504. portStr := params["listen-port"]
  505. privKey := params["private-key"]
  506. port := 51820
  507. if portStr != "" {
  508. if pVal, err := strconv.Atoi(portStr); err == nil && pVal > 0 {
  509. port = pVal
  510. }
  511. }
  512. if name == "" {
  513. return ColorRed + "Usage: /interface wireguard add name=<name> [listen-port=<port>]\n" + ColorReset
  514. }
  515. req := vpn.SaveWireGuardInterfaceRequest{
  516. WireGuardInterface: vpn.WireGuardInterface{
  517. Name: name,
  518. ListenPort: port,
  519. PrivateKey: privKey,
  520. Disabled: false,
  521. },
  522. }
  523. if err := e.vpnMgr.WireGuard.SaveInterface(req); err != nil {
  524. return ColorRed + "Error adding WireGuard interface: " + err.Error() + ColorReset + "\n"
  525. }
  526. return ColorGreen + fmt.Sprintf("WireGuard interface %s added.\n", name) + ColorReset
  527. case "remove", "rem", "delete":
  528. pos, _ := parseArgs(parts[1:])
  529. if len(pos) == 0 {
  530. return ColorRed + "Usage: /interface wireguard remove <name>\n" + ColorReset
  531. }
  532. if err := e.vpnMgr.WireGuard.DeleteInterface(pos[0]); err != nil {
  533. return ColorRed + "Error deleting WireGuard interface: " + err.Error() + ColorReset + "\n"
  534. }
  535. return ColorGreen + fmt.Sprintf("WireGuard interface %s removed.\n", pos[0]) + ColorReset
  536. case "enable":
  537. pos, _ := parseArgs(parts[1:])
  538. if len(pos) == 0 {
  539. return ColorRed + "Usage: /interface wireguard enable <name>\n" + ColorReset
  540. }
  541. if err := e.vpnMgr.WireGuard.SetInterfaceEnabled(pos[0], true); err != nil {
  542. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  543. }
  544. return ColorGreen + fmt.Sprintf("WireGuard interface %s enabled.\n", pos[0]) + ColorReset
  545. case "disable":
  546. pos, _ := parseArgs(parts[1:])
  547. if len(pos) == 0 {
  548. return ColorRed + "Usage: /interface wireguard disable <name>\n" + ColorReset
  549. }
  550. if err := e.vpnMgr.WireGuard.SetInterfaceEnabled(pos[0], false); err != nil {
  551. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  552. }
  553. return ColorGreen + fmt.Sprintf("WireGuard interface %s disabled.\n", pos[0]) + ColorReset
  554. default:
  555. return ColorRed + "Usage: /interface wireguard [print|add|remove|enable|disable|peers]\n" + ColorReset
  556. }
  557. }
  558. func (e *Executor) handleWireGuardPeers(parts []string, user *auth.PublicUser) string {
  559. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  560. _, params := parseArgs(parts)
  561. peers := e.vpnMgr.WireGuard.ListPeers(params["interface"])
  562. tbl := NewTable("INTERFACE", "PUBLIC-KEY", "ENDPOINT", "ALLOWED-IPS", "LAST-HANDSHAKE")
  563. for _, p := range peers {
  564. hs := "-"
  565. if p.LatestHandshake != "" {
  566. hs = p.LatestHandshake
  567. }
  568. ep := p.EndpointAddress
  569. if p.EndpointPort > 0 {
  570. ep = fmt.Sprintf("%s:%d", p.EndpointAddress, p.EndpointPort)
  571. }
  572. tbl.AddRow(p.Interface, p.PublicKey, ep, strings.Join(p.AllowedIPs, ","), hs)
  573. }
  574. return tbl.String()
  575. }
  576. if user.Role == auth.RoleRead {
  577. return ColorRed + "Permission denied: viewer role cannot modify wireguard peers\n" + ColorReset
  578. }
  579. sub := strings.ToLower(parts[0])
  580. switch sub {
  581. case "add":
  582. _, params := parseArgs(parts[1:])
  583. iface := params["interface"]
  584. pubKey := params["public-key"]
  585. endpoint := params["endpoint"]
  586. allowed := params["allowed-address"]
  587. if allowed == "" {
  588. allowed = params["allowed-ips"]
  589. }
  590. if iface == "" || pubKey == "" {
  591. return ColorRed + "Usage: /interface wireguard peers add interface=<iface> public-key=<key> [endpoint=host:port] [allowed-address=cidr]\n" + ColorReset
  592. }
  593. var allowedList []string
  594. if allowed != "" {
  595. allowedList = strings.Split(allowed, ",")
  596. }
  597. epHost := endpoint
  598. epPort := 0
  599. if strings.Contains(endpoint, ":") {
  600. parts := strings.Split(endpoint, ":")
  601. epHost = parts[0]
  602. if p, err := strconv.Atoi(parts[1]); err == nil {
  603. epPort = p
  604. }
  605. }
  606. req := vpn.SaveWireGuardPeerRequest{
  607. WireGuardPeer: vpn.WireGuardPeer{
  608. Interface: iface,
  609. PublicKey: pubKey,
  610. EndpointAddress: epHost,
  611. EndpointPort: epPort,
  612. AllowedIPs: allowedList,
  613. Disabled: false,
  614. },
  615. }
  616. if err := e.vpnMgr.WireGuard.SavePeer(req); err != nil {
  617. return ColorRed + "Error saving WireGuard peer: " + err.Error() + ColorReset + "\n"
  618. }
  619. return ColorGreen + "WireGuard peer added successfully.\n" + ColorReset
  620. case "remove", "rem", "delete":
  621. _, params := parseArgs(parts[1:])
  622. iface := params["interface"]
  623. pubKey := params["public-key"]
  624. if iface == "" || pubKey == "" {
  625. return ColorRed + "Usage: /interface wireguard peers remove interface=<iface> public-key=<key>\n" + ColorReset
  626. }
  627. if err := e.vpnMgr.WireGuard.DeletePeer(iface, pubKey); err != nil {
  628. return ColorRed + "Error deleting WireGuard peer: " + err.Error() + ColorReset + "\n"
  629. }
  630. return ColorGreen + "WireGuard peer removed successfully.\n" + ColorReset
  631. default:
  632. return ColorRed + "Usage: /interface wireguard peers [print|add|remove]\n" + ColorReset
  633. }
  634. }
  635. // --- EoIP ---
  636. func (e *Executor) handleInterfaceEoIP(parts []string, user *auth.PublicUser) string {
  637. if e.vpnMgr == nil || e.vpnMgr.EoIP == nil {
  638. return ColorRed + "EoIP subsystem not available.\n" + ColorReset
  639. }
  640. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  641. tunnels := e.vpnMgr.EoIP.List()
  642. tbl := NewTable("FLAGS", "NAME", "REMOTE-ADDRESS", "TUNNEL-ID", "STATUS")
  643. for _, t := range tunnels {
  644. flags := ""
  645. if t.Disabled {
  646. flags += "X"
  647. } else {
  648. flags += "R"
  649. }
  650. st := "stopped"
  651. if t.Running {
  652. st = "running"
  653. }
  654. tbl.AddRow(flags, t.Name, t.RemoteAddress, fmt.Sprintf("%d", t.TunnelID), st)
  655. }
  656. return tbl.String()
  657. }
  658. if user.Role == auth.RoleRead {
  659. return ColorRed + "Permission denied: viewer role cannot modify EoIP\n" + ColorReset
  660. }
  661. sub := strings.ToLower(parts[0])
  662. switch sub {
  663. case "add":
  664. _, params := parseArgs(parts[1:])
  665. name := params["name"]
  666. remote := params["remote-address"]
  667. tidStr := params["tunnel-id"]
  668. tid, err := strconv.Atoi(tidStr)
  669. if err != nil || tid < 1 || tid > 65535 {
  670. return ColorRed + "Invalid tunnel-id (1-65535)\n" + ColorReset
  671. }
  672. if name == "" || remote == "" {
  673. return ColorRed + "Usage: /interface eoip add name=<name> remote-address=<ip> tunnel-id=<id>\n" + ColorReset
  674. }
  675. inst := &vpn.EoIPInstance{
  676. Name: name,
  677. RemoteAddress: remote,
  678. TunnelID: tid,
  679. Disabled: false,
  680. }
  681. if err := e.vpnMgr.EoIP.Save(inst); err != nil {
  682. return ColorRed + "Error adding EoIP tunnel: " + err.Error() + ColorReset + "\n"
  683. }
  684. return ColorGreen + fmt.Sprintf("EoIP tunnel %s added.\n", name) + ColorReset
  685. case "remove", "rem", "delete":
  686. pos, _ := parseArgs(parts[1:])
  687. if len(pos) == 0 {
  688. return ColorRed + "Usage: /interface eoip remove <id>\n" + ColorReset
  689. }
  690. if err := e.vpnMgr.EoIP.Delete(pos[0]); err != nil {
  691. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  692. }
  693. return ColorGreen + fmt.Sprintf("EoIP tunnel %s removed.\n", pos[0]) + ColorReset
  694. case "enable":
  695. pos, _ := parseArgs(parts[1:])
  696. if len(pos) == 0 {
  697. return ColorRed + "Usage: /interface eoip enable <id>\n" + ColorReset
  698. }
  699. if err := e.vpnMgr.EoIP.SetState(pos[0], false); err != nil {
  700. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  701. }
  702. return ColorGreen + fmt.Sprintf("EoIP tunnel %s enabled.\n", pos[0]) + ColorReset
  703. case "disable":
  704. pos, _ := parseArgs(parts[1:])
  705. if len(pos) == 0 {
  706. return ColorRed + "Usage: /interface eoip disable <id>\n" + ColorReset
  707. }
  708. if err := e.vpnMgr.EoIP.SetState(pos[0], true); err != nil {
  709. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  710. }
  711. return ColorGreen + fmt.Sprintf("EoIP tunnel %s disabled.\n", pos[0]) + ColorReset
  712. default:
  713. return ColorRed + "Usage: /interface eoip [print|add|remove|enable|disable]\n" + ColorReset
  714. }
  715. }
  716. // --- VRRP ---
  717. func (e *Executor) handleInterfaceVRRP(parts []string, user *auth.PublicUser) string {
  718. if e.vrrpMgr == nil {
  719. return ColorRed + "VRRP subsystem not available.\n" + ColorReset
  720. }
  721. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  722. instances := e.vrrpMgr.List()
  723. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "VRID", "PRIORITY", "STATE", "VIP")
  724. for _, inst := range instances {
  725. flags := ""
  726. if inst.Disabled {
  727. flags += "X"
  728. } else {
  729. flags += "R"
  730. }
  731. tbl.AddRow(flags, inst.Name, inst.Interface, fmt.Sprintf("%d", inst.VRID), fmt.Sprintf("%d", inst.Priority), inst.State, strings.Join(inst.VirtualIPs, ","))
  732. }
  733. return tbl.String()
  734. }
  735. if user.Role == auth.RoleRead {
  736. return ColorRed + "Permission denied: viewer role cannot modify VRRP\n" + ColorReset
  737. }
  738. sub := strings.ToLower(parts[0])
  739. switch sub {
  740. case "add":
  741. _, params := parseArgs(parts[1:])
  742. name := params["name"]
  743. iface := params["interface"]
  744. vridStr := params["vrid"]
  745. prioStr := params["priority"]
  746. vip := params["vip"]
  747. if vip == "" {
  748. vip = params["virtual-ip"]
  749. }
  750. vrid, err := strconv.Atoi(vridStr)
  751. if err != nil || vrid < 1 || vrid > 255 {
  752. return ColorRed + "Invalid vrid (1-255)\n" + ColorReset
  753. }
  754. prio := 100
  755. if prioStr != "" {
  756. if pVal, err := strconv.Atoi(prioStr); err == nil && pVal >= 1 && pVal <= 254 {
  757. prio = pVal
  758. }
  759. }
  760. if name == "" || iface == "" || vip == "" {
  761. return ColorRed + "Usage: /interface vrrp add name=<name> interface=<iface> vrid=<id> vip=<ip/cidr> [priority=100]\n" + ColorReset
  762. }
  763. inst := vrrp.VRRPInstance{
  764. Name: name,
  765. Interface: iface,
  766. VRID: vrid,
  767. Priority: prio,
  768. VirtualIPs: []string{vip},
  769. Disabled: false,
  770. }
  771. if _, err := e.vrrpMgr.Save(inst); err != nil {
  772. return ColorRed + "Error adding VRRP: " + err.Error() + ColorReset + "\n"
  773. }
  774. return ColorGreen + fmt.Sprintf("VRRP instance %s added.\n", name) + ColorReset
  775. case "remove", "rem", "delete":
  776. pos, _ := parseArgs(parts[1:])
  777. if len(pos) == 0 {
  778. return ColorRed + "Usage: /interface vrrp remove <id>\n" + ColorReset
  779. }
  780. if err := e.vrrpMgr.Delete(pos[0]); err != nil {
  781. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  782. }
  783. return ColorGreen + fmt.Sprintf("VRRP instance %s removed.\n", pos[0]) + ColorReset
  784. case "enable":
  785. pos, _ := parseArgs(parts[1:])
  786. if len(pos) == 0 {
  787. return ColorRed + "Usage: /interface vrrp enable <id>\n" + ColorReset
  788. }
  789. if err := e.vrrpMgr.ToggleState(pos[0], false); err != nil {
  790. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  791. }
  792. return ColorGreen + fmt.Sprintf("VRRP instance %s enabled.\n", pos[0]) + ColorReset
  793. case "disable":
  794. pos, _ := parseArgs(parts[1:])
  795. if len(pos) == 0 {
  796. return ColorRed + "Usage: /interface vrrp disable <id>\n" + ColorReset
  797. }
  798. if err := e.vrrpMgr.ToggleState(pos[0], true); err != nil {
  799. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  800. }
  801. return ColorGreen + fmt.Sprintf("VRRP instance %s disabled.\n", pos[0]) + ColorReset
  802. default:
  803. return ColorRed + "Usage: /interface vrrp [print|add|remove|enable|disable]\n" + ColorReset
  804. }
  805. }
  806. // --- PPPoE Client ---
  807. func (e *Executor) handleInterfacePPPoE(parts []string, user *auth.PublicUser) string {
  808. if e.vpnMgr == nil || e.vpnMgr.PPPoE == nil {
  809. return ColorRed + "PPPoE subsystem not available.\n" + ColorReset
  810. }
  811. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  812. clients := e.vpnMgr.PPPoE.List()
  813. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "USER", "STATUS", "IP")
  814. for _, c := range clients {
  815. flags := ""
  816. if c.Disabled {
  817. flags += "X"
  818. } else {
  819. flags += "R"
  820. }
  821. st := "stopped"
  822. if c.Running {
  823. st = "running"
  824. }
  825. tbl.AddRow(flags, c.Name, c.Interface, c.Username, st, c.AssignedIP)
  826. }
  827. return tbl.String()
  828. }
  829. if user.Role == auth.RoleRead {
  830. return ColorRed + "Permission denied: viewer role cannot modify PPPoE\n" + ColorReset
  831. }
  832. sub := strings.ToLower(parts[0])
  833. switch sub {
  834. case "add":
  835. _, params := parseArgs(parts[1:])
  836. name := params["name"]
  837. iface := params["interface"]
  838. usr := params["user"]
  839. pwd := params["password"]
  840. if name == "" || iface == "" || usr == "" {
  841. return ColorRed + "Usage: /interface pppoe-client add name=<name> interface=<iface> user=<usr> password=<pwd>\n" + ColorReset
  842. }
  843. inst := vpn.PPPoEInstance{
  844. Name: name,
  845. Interface: iface,
  846. Username: usr,
  847. Password: pwd,
  848. Disabled: false,
  849. }
  850. if err := e.vpnMgr.PPPoE.Save(inst); err != nil {
  851. return ColorRed + "Error adding PPPoE client: " + err.Error() + ColorReset + "\n"
  852. }
  853. return ColorGreen + fmt.Sprintf("PPPoE client %s added.\n", name) + ColorReset
  854. case "remove", "rem", "delete":
  855. pos, _ := parseArgs(parts[1:])
  856. if len(pos) == 0 {
  857. return ColorRed + "Usage: /interface pppoe-client remove <name>\n" + ColorReset
  858. }
  859. if err := e.vpnMgr.PPPoE.Delete(pos[0]); err != nil {
  860. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  861. }
  862. return ColorGreen + fmt.Sprintf("PPPoE client %s removed.\n", pos[0]) + ColorReset
  863. case "connect":
  864. pos, _ := parseArgs(parts[1:])
  865. if len(pos) == 0 {
  866. return ColorRed + "Usage: /interface pppoe-client connect <name>\n" + ColorReset
  867. }
  868. if err := e.vpnMgr.PPPoE.Connect(pos[0]); err != nil {
  869. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  870. }
  871. return ColorGreen + fmt.Sprintf("Connecting PPPoE client %s...\n", pos[0]) + ColorReset
  872. case "disconnect":
  873. pos, _ := parseArgs(parts[1:])
  874. if len(pos) == 0 {
  875. return ColorRed + "Usage: /interface pppoe-client disconnect <name>\n" + ColorReset
  876. }
  877. if err := e.vpnMgr.PPPoE.Disconnect(pos[0]); err != nil {
  878. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  879. }
  880. return ColorGreen + fmt.Sprintf("Disconnected PPPoE client %s.\n", pos[0]) + ColorReset
  881. default:
  882. return ColorRed + "Usage: /interface pppoe-client [print|add|remove|connect|disconnect]\n" + ColorReset
  883. }
  884. }
  885. // ==================== IP SUBSYSTEM ====================
  886. func (e *Executor) handleIP(parts []string, user *auth.PublicUser) string {
  887. if len(parts) == 0 {
  888. return ColorYellow + "IP subsystem. Available: address, route, dhcp-client, dhcp-server, dns, firewall, service\n" + ColorReset
  889. }
  890. sub := strings.ToLower(parts[0])
  891. switch sub {
  892. case "address", "addr":
  893. return e.handleIPAddress(parts[1:], user)
  894. case "route":
  895. return e.handleIPRoute(parts[1:], user)
  896. case "dhcp-client":
  897. return e.handleIPDHCPClient(parts[1:], user)
  898. case "dhcp-server":
  899. return e.handleIPDHCPServer(parts[1:], user)
  900. case "dns":
  901. return e.handleIPDNS(parts[1:], user)
  902. case "firewall":
  903. return e.handleIPFirewall(parts[1:], user)
  904. case "service", "services":
  905. return e.handleIPService(parts[1:], user)
  906. default:
  907. return ColorRed + "Unknown IP command: " + sub + "\n" + ColorReset
  908. }
  909. }
  910. // --- IP Address ---
  911. func (e *Executor) handleIPAddress(parts []string, user *auth.PublicUser) string {
  912. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  913. ifaces, err := e.netMgr.ListInterfaces()
  914. if err != nil {
  915. return ColorRed + "Error listing IP addresses: " + err.Error() + ColorReset + "\n"
  916. }
  917. tbl := NewTable("#", "ADDRESS", "NETWORK", "INTERFACE", "STATUS")
  918. idx := 0
  919. for _, iface := range ifaces {
  920. for _, cidr := range append(iface.IPv4, iface.IPv6...) {
  921. ip, ipNet, parseErr := net.ParseCIDR(cidr)
  922. netStr := "-"
  923. if parseErr == nil && ipNet != nil {
  924. netStr = ipNet.IP.String()
  925. } else if ip != nil {
  926. netStr = ip.String()
  927. }
  928. tbl.AddRow(fmt.Sprintf("%d", idx), cidr, netStr, iface.Name, "enabled")
  929. idx++
  930. }
  931. }
  932. return tbl.String()
  933. }
  934. if user.Role == auth.RoleRead {
  935. return ColorRed + "Permission denied: viewer role cannot modify IP addresses\n" + ColorReset
  936. }
  937. sub := strings.ToLower(parts[0])
  938. switch sub {
  939. case "add":
  940. _, params := parseArgs(parts[1:])
  941. addr := params["address"]
  942. iface := params["interface"]
  943. if addr == "" || iface == "" {
  944. return ColorRed + "Usage: /ip address add address=<ip/mask> interface=<iface>\n" + ColorReset
  945. }
  946. if err := e.netMgr.AddInterfaceIP(iface, addr); err != nil {
  947. return ColorRed + "Error adding IP address: " + err.Error() + ColorReset + "\n"
  948. }
  949. return ColorGreen + fmt.Sprintf("IP address %s added to interface %s.\n", addr, iface) + ColorReset
  950. case "remove", "rem", "delete":
  951. _, params := parseArgs(parts[1:])
  952. addr := params["address"]
  953. iface := params["interface"]
  954. if addr == "" || iface == "" {
  955. return ColorRed + "Usage: /ip address remove address=<ip/mask> interface=<iface>\n" + ColorReset
  956. }
  957. if err := e.netMgr.DeleteInterfaceIP(iface, addr); err != nil {
  958. return ColorRed + "Error removing IP address: " + err.Error() + ColorReset + "\n"
  959. }
  960. return ColorGreen + fmt.Sprintf("IP address %s removed from interface %s.\n", addr, iface) + ColorReset
  961. default:
  962. return ColorRed + "Usage: /ip address [print|add|remove]\n" + ColorReset
  963. }
  964. }
  965. // --- IP Route ---
  966. func (e *Executor) handleIPRoute(parts []string, user *auth.PublicUser) string {
  967. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  968. routes := e.routingMgr.ListRoutes()
  969. tbl := NewTable("FLAGS", "DST-ADDRESS", "GATEWAY", "DISTANCE", "TABLE", "STATUS")
  970. for _, r := range routes {
  971. flags := ""
  972. if r.Disabled {
  973. flags += "X"
  974. } else {
  975. flags += "A"
  976. }
  977. if r.Static {
  978. flags += "S"
  979. } else {
  980. flags += "D"
  981. }
  982. st := "active"
  983. if r.Disabled {
  984. st = "disabled"
  985. }
  986. tbl.AddRow(flags, r.DstAddress, r.Gateway, fmt.Sprintf("%d", r.Distance), r.RoutingTable, st)
  987. }
  988. return tbl.String()
  989. }
  990. sub := strings.ToLower(parts[0])
  991. switch sub {
  992. case "rule":
  993. rules := e.routingMgr.ListPolicyRules()
  994. tbl := NewTable("#", "SRC", "DST", "TABLE", "PRIORITY")
  995. for idx, r := range rules {
  996. tbl.AddRow(fmt.Sprintf("%d", idx), r.SrcAddress, r.DstAddress, r.Table, fmt.Sprintf("%d", r.Priority))
  997. }
  998. return tbl.String()
  999. case "table":
  1000. tables := e.routingMgr.ListTables()
  1001. tbl := NewTable("#", "NAME", "TABLE-ID", "COMMENT")
  1002. for idx, t := range tables {
  1003. tbl.AddRow(fmt.Sprintf("%d", idx), t.Name, fmt.Sprintf("%d", t.TableID), t.Comment)
  1004. }
  1005. return tbl.String()
  1006. }
  1007. if user.Role == auth.RoleRead {
  1008. return ColorRed + "Permission denied: viewer role cannot modify routes\n" + ColorReset
  1009. }
  1010. switch sub {
  1011. case "add":
  1012. _, params := parseArgs(parts[1:])
  1013. dst := params["dst-address"]
  1014. if dst == "" {
  1015. dst = params["dst"]
  1016. }
  1017. gw := params["gateway"]
  1018. table := params["routing-table"]
  1019. if table == "" {
  1020. table = "main"
  1021. }
  1022. dist := 1
  1023. if dStr, ok := params["distance"]; ok {
  1024. if dVal, err := strconv.Atoi(dStr); err == nil && dVal >= 1 {
  1025. dist = dVal
  1026. }
  1027. }
  1028. if dst == "" || gw == "" {
  1029. return ColorRed + "Usage: /ip route add dst-address=<cidr> gateway=<ip> [distance=1] [routing-table=main]\n" + ColorReset
  1030. }
  1031. r := routing.RouteItem{
  1032. DstAddress: dst,
  1033. Gateway: gw,
  1034. Distance: dist,
  1035. RoutingTable: table,
  1036. Static: true,
  1037. Disabled: false,
  1038. }
  1039. if err := e.routingMgr.SaveRoute(r); err != nil {
  1040. return ColorRed + "Error adding route: " + err.Error() + ColorReset + "\n"
  1041. }
  1042. return ColorGreen + fmt.Sprintf("Route to %s via %s added.\n", dst, gw) + ColorReset
  1043. case "remove", "rem", "delete":
  1044. pos, _ := parseArgs(parts[1:])
  1045. if len(pos) == 0 {
  1046. return ColorRed + "Usage: /ip route remove <id>\n" + ColorReset
  1047. }
  1048. if err := e.routingMgr.DeleteRoute(pos[0]); err != nil {
  1049. return ColorRed + "Error removing route: " + err.Error() + ColorReset + "\n"
  1050. }
  1051. return ColorGreen + fmt.Sprintf("Route %s removed.\n", pos[0]) + ColorReset
  1052. case "enable":
  1053. pos, _ := parseArgs(parts[1:])
  1054. if len(pos) == 0 {
  1055. return ColorRed + "Usage: /ip route enable <id>\n" + ColorReset
  1056. }
  1057. if err := e.routingMgr.SetRouteState(pos[0], false); err != nil {
  1058. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1059. }
  1060. return ColorGreen + fmt.Sprintf("Route %s enabled.\n", pos[0]) + ColorReset
  1061. case "disable":
  1062. pos, _ := parseArgs(parts[1:])
  1063. if len(pos) == 0 {
  1064. return ColorRed + "Usage: /ip route disable <id>\n" + ColorReset
  1065. }
  1066. if err := e.routingMgr.SetRouteState(pos[0], true); err != nil {
  1067. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1068. }
  1069. return ColorGreen + fmt.Sprintf("Route %s disabled.\n", pos[0]) + ColorReset
  1070. default:
  1071. return ColorRed + "Usage: /ip route [print|add|remove|enable|disable|rule|table]\n" + ColorReset
  1072. }
  1073. }
  1074. // --- IP DHCP Client ---
  1075. func (e *Executor) handleIPDHCPClient(parts []string, user *auth.PublicUser) string {
  1076. if e.dhcpMgr == nil {
  1077. return ColorRed + "DHCP client manager not available.\n" + ColorReset
  1078. }
  1079. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1080. clients := e.dhcpMgr.ListClients()
  1081. tbl := NewTable("FLAGS", "INTERFACE", "USE-PEER-DNS", "STATUS", "ADDRESS", "GATEWAY")
  1082. for _, c := range clients {
  1083. flags := ""
  1084. if !c.Enabled {
  1085. flags += "X"
  1086. } else {
  1087. flags += "R"
  1088. }
  1089. addr := c.IPAddress
  1090. if addr == "" {
  1091. addr = "-"
  1092. }
  1093. gw := c.Gateway
  1094. if gw == "" {
  1095. gw = "-"
  1096. }
  1097. tbl.AddRow(flags, c.Interface, fmt.Sprintf("%t", c.UsePeerDNS), c.Status, addr, gw)
  1098. }
  1099. return tbl.String()
  1100. }
  1101. if user.Role == auth.RoleRead {
  1102. return ColorRed + "Permission denied: viewer role cannot modify DHCP clients\n" + ColorReset
  1103. }
  1104. sub := strings.ToLower(parts[0])
  1105. switch sub {
  1106. case "add":
  1107. _, params := parseArgs(parts[1:])
  1108. iface := params["interface"]
  1109. if iface == "" {
  1110. return ColorRed + "Usage: /ip dhcp-client add interface=<iface> [use-peer-dns=yes]\n" + ColorReset
  1111. }
  1112. peerDNS := true
  1113. if val, ok := params["use-peer-dns"]; ok {
  1114. peerDNS = (strings.ToLower(val) == "yes" || strings.ToLower(val) == "true")
  1115. }
  1116. req := dhcp.CreateDHCPClientRequest{
  1117. Interface: iface,
  1118. UsePeerDNS: peerDNS,
  1119. Enabled: true,
  1120. }
  1121. if _, err := e.dhcpMgr.AddClient(req); err != nil {
  1122. return ColorRed + "Error adding DHCP client: " + err.Error() + ColorReset + "\n"
  1123. }
  1124. return ColorGreen + fmt.Sprintf("DHCP client added on interface %s.\n", iface) + ColorReset
  1125. case "remove", "rem", "delete":
  1126. pos, _ := parseArgs(parts[1:])
  1127. if len(pos) == 0 {
  1128. return ColorRed + "Usage: /ip dhcp-client remove <id>\n" + ColorReset
  1129. }
  1130. if err := e.dhcpMgr.DeleteClient(pos[0]); err != nil {
  1131. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1132. }
  1133. return ColorGreen + fmt.Sprintf("DHCP client %s removed.\n", pos[0]) + ColorReset
  1134. case "renew":
  1135. pos, _ := parseArgs(parts[1:])
  1136. if len(pos) == 0 {
  1137. return ColorRed + "Usage: /ip dhcp-client renew <id>\n" + ColorReset
  1138. }
  1139. if err := e.dhcpMgr.RenewLease(pos[0]); err != nil {
  1140. return ColorRed + "Error renewing lease: " + err.Error() + ColorReset + "\n"
  1141. }
  1142. return ColorGreen + fmt.Sprintf("DHCP lease renewal requested for %s.\n", pos[0]) + ColorReset
  1143. case "release":
  1144. pos, _ := parseArgs(parts[1:])
  1145. if len(pos) == 0 {
  1146. return ColorRed + "Usage: /ip dhcp-client release <id>\n" + ColorReset
  1147. }
  1148. if err := e.dhcpMgr.ReleaseLease(pos[0]); err != nil {
  1149. return ColorRed + "Error releasing lease: " + err.Error() + ColorReset + "\n"
  1150. }
  1151. return ColorGreen + fmt.Sprintf("DHCP lease released for %s.\n", pos[0]) + ColorReset
  1152. default:
  1153. return ColorRed + "Usage: /ip dhcp-client [print|add|remove|renew|release]\n" + ColorReset
  1154. }
  1155. }
  1156. // --- IP DHCP Server ---
  1157. func (e *Executor) handleIPDHCPServer(parts []string, user *auth.PublicUser) string {
  1158. if e.dhcpServMgr == nil {
  1159. return ColorRed + "DHCP server manager not available.\n" + ColorReset
  1160. }
  1161. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1162. servers := e.dhcpServMgr.ListServers()
  1163. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "ADDRESS-POOL", "LEASE-TIME")
  1164. for _, s := range servers {
  1165. flags := ""
  1166. if s.Disabled {
  1167. flags += "X"
  1168. } else {
  1169. flags += "R"
  1170. }
  1171. tbl.AddRow(flags, s.Name, s.Interface, s.AddressPool, s.LeaseTime)
  1172. }
  1173. return tbl.String()
  1174. }
  1175. sub := strings.ToLower(parts[0])
  1176. switch sub {
  1177. case "lease":
  1178. leases := e.dhcpServMgr.ListLeases()
  1179. tbl := NewTable("IP-ADDRESS", "MAC-ADDRESS", "HOSTNAME", "EXPIRES-IN", "STATUS")
  1180. for _, l := range leases {
  1181. tbl.AddRow(l.Address, l.MACAddress, l.HostName, l.ExpiresIn, l.Status)
  1182. }
  1183. return tbl.String()
  1184. case "network":
  1185. networks := e.dhcpServMgr.ListNetworks()
  1186. tbl := NewTable("SUBNET", "GATEWAY", "DNS-SERVERS", "DOMAIN")
  1187. for _, n := range networks {
  1188. tbl.AddRow(n.Address, n.Gateway, strings.Join(n.DNSServers, ","), n.Domain)
  1189. }
  1190. return tbl.String()
  1191. }
  1192. if user.Role == auth.RoleRead {
  1193. return ColorRed + "Permission denied: viewer role cannot modify DHCP server\n" + ColorReset
  1194. }
  1195. switch sub {
  1196. case "add":
  1197. _, params := parseArgs(parts[1:])
  1198. name := params["name"]
  1199. iface := params["interface"]
  1200. pool := params["address-pool"]
  1201. if pool == "" {
  1202. start := params["range-start"]
  1203. end := params["range-end"]
  1204. if start != "" && end != "" {
  1205. pool = fmt.Sprintf("%s-%s", start, end)
  1206. }
  1207. }
  1208. leaseTime := params["lease-time"]
  1209. if leaseTime == "" {
  1210. leaseTime = "12h"
  1211. }
  1212. if name == "" || iface == "" || pool == "" {
  1213. return ColorRed + "Usage: /ip dhcp-server add name=<name> interface=<iface> address-pool=<start-end> [lease-time=12h]\n" + ColorReset
  1214. }
  1215. inst := dhcp.DHCPServerInstance{
  1216. Name: name,
  1217. Interface: iface,
  1218. AddressPool: pool,
  1219. LeaseTime: leaseTime,
  1220. Disabled: false,
  1221. }
  1222. if _, err := e.dhcpServMgr.SaveServer(inst); err != nil {
  1223. return ColorRed + "Error adding DHCP server: " + err.Error() + ColorReset + "\n"
  1224. }
  1225. return ColorGreen + fmt.Sprintf("DHCP server %s added on interface %s.\n", name, iface) + ColorReset
  1226. case "remove", "rem", "delete":
  1227. pos, _ := parseArgs(parts[1:])
  1228. if len(pos) == 0 {
  1229. return ColorRed + "Usage: /ip dhcp-server remove <id>\n" + ColorReset
  1230. }
  1231. if err := e.dhcpServMgr.DeleteServer(pos[0]); err != nil {
  1232. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1233. }
  1234. return ColorGreen + fmt.Sprintf("DHCP server %s removed.\n", pos[0]) + ColorReset
  1235. case "enable":
  1236. pos, _ := parseArgs(parts[1:])
  1237. if len(pos) == 0 {
  1238. return ColorRed + "Usage: /ip dhcp-server enable <id>\n" + ColorReset
  1239. }
  1240. if err := e.dhcpServMgr.ToggleServerState(pos[0], false); err != nil {
  1241. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1242. }
  1243. return ColorGreen + fmt.Sprintf("DHCP server %s enabled.\n", pos[0]) + ColorReset
  1244. case "disable":
  1245. pos, _ := parseArgs(parts[1:])
  1246. if len(pos) == 0 {
  1247. return ColorRed + "Usage: /ip dhcp-server disable <id>\n" + ColorReset
  1248. }
  1249. if err := e.dhcpServMgr.ToggleServerState(pos[0], true); err != nil {
  1250. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1251. }
  1252. return ColorGreen + fmt.Sprintf("DHCP server %s disabled.\n", pos[0]) + ColorReset
  1253. default:
  1254. return ColorRed + "Usage: /ip dhcp-server [print|add|remove|enable|disable|lease|network]\n" + ColorReset
  1255. }
  1256. }
  1257. // --- IP DNS ---
  1258. func (e *Executor) handleIPDNS(parts []string, user *auth.PublicUser) string {
  1259. if e.dnsMgr == nil {
  1260. return ColorRed + "DNS manager not available.\n" + ColorReset
  1261. }
  1262. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1263. settings := e.dnsMgr.GetSettings()
  1264. out := " servers: " + strings.Join(settings.Servers, ", ") + "\n"
  1265. out += " dynamic-servers: " + strings.Join(settings.DynamicServers, ", ") + "\n"
  1266. out += fmt.Sprintf(" allow-remote-requests: %t\n", settings.AllowRemoteRequests)
  1267. out += fmt.Sprintf(" cache-size: %d entries\n", settings.CacheSize)
  1268. out += fmt.Sprintf(" domain: %s\n", settings.Domain)
  1269. return out
  1270. }
  1271. sub := strings.ToLower(parts[0])
  1272. switch sub {
  1273. case "cache":
  1274. if len(parts) > 1 && parts[1] == "flush" {
  1275. if user.Role == auth.RoleRead {
  1276. return ColorRed + "Permission denied\n" + ColorReset
  1277. }
  1278. if err := e.dnsMgr.FlushCache(); err != nil {
  1279. return ColorRed + "Error flushing DNS cache: " + err.Error() + ColorReset + "\n"
  1280. }
  1281. return ColorGreen + "DNS cache flushed.\n" + ColorReset
  1282. }
  1283. return ColorYellow + "Usage: /ip dns cache flush\n" + ColorReset
  1284. case "static":
  1285. records := e.dnsMgr.ListStaticRecords()
  1286. tbl := NewTable("#", "NAME", "ADDRESS", "TTL", "STATUS")
  1287. for idx, r := range records {
  1288. st := "enabled"
  1289. if r.Disabled {
  1290. st = "disabled"
  1291. }
  1292. tbl.AddRow(fmt.Sprintf("%d", idx), r.Name, r.Address, fmt.Sprintf("%d", r.TTL), st)
  1293. }
  1294. return tbl.String()
  1295. case "set":
  1296. if user.Role == auth.RoleRead {
  1297. return ColorRed + "Permission denied\n" + ColorReset
  1298. }
  1299. _, params := parseArgs(parts[1:])
  1300. if sStr, ok := params["servers"]; ok {
  1301. srvs := strings.Split(sStr, ",")
  1302. cfg := e.dnsMgr.GetSettings()
  1303. cfg.Servers = srvs
  1304. if err := e.dnsMgr.SaveSettings(cfg); err != nil {
  1305. return ColorRed + "Error saving DNS settings: " + err.Error() + ColorReset + "\n"
  1306. }
  1307. return ColorGreen + "DNS servers updated.\n" + ColorReset
  1308. }
  1309. return ColorRed + "Usage: /ip dns set servers=<ip1,ip2>\n" + ColorReset
  1310. default:
  1311. return ColorRed + "Usage: /ip dns [print|set|cache flush|static]\n" + ColorReset
  1312. }
  1313. }
  1314. // --- IP Firewall ---
  1315. func (e *Executor) handleIPFirewall(parts []string, user *auth.PublicUser) string {
  1316. if e.fwMgr == nil {
  1317. return ColorRed + "Firewall manager not available.\n" + ColorReset
  1318. }
  1319. if len(parts) == 0 {
  1320. return ColorYellow + "Firewall subsystem. Available: filter, nat, mangle, address-list\n" + ColorReset
  1321. }
  1322. tableType := strings.ToLower(parts[0])
  1323. switch tableType {
  1324. case "filter":
  1325. return e.handleFirewallFilter(parts[1:], user)
  1326. case "nat":
  1327. return e.handleFirewallNAT(parts[1:], user)
  1328. case "mangle":
  1329. return e.handleFirewallMangle(parts[1:], user)
  1330. case "address-list":
  1331. return e.handleFirewallAddressList(parts[1:], user)
  1332. default:
  1333. return ColorRed + "Unknown firewall table: " + tableType + "\n" + ColorReset
  1334. }
  1335. }
  1336. func (e *Executor) handleFirewallFilter(parts []string, user *auth.PublicUser) string {
  1337. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1338. rules := e.fwMgr.ListFilterRules()
  1339. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "PROTO", "SRC-ADDR", "DST-ADDR", "DST-PORT", "COMMENT")
  1340. for _, r := range rules {
  1341. flags := ""
  1342. if r.Disabled {
  1343. flags += "X"
  1344. } else {
  1345. flags += " "
  1346. }
  1347. tbl.AddRow(flags, r.Chain, r.Action, r.Protocol, r.SrcAddress, r.DstAddress, r.DstPort, r.Comment)
  1348. }
  1349. return tbl.String()
  1350. }
  1351. if user.Role == auth.RoleRead {
  1352. return ColorRed + "Permission denied: viewer role cannot modify firewall\n" + ColorReset
  1353. }
  1354. sub := strings.ToLower(parts[0])
  1355. switch sub {
  1356. case "add":
  1357. _, params := parseArgs(parts[1:])
  1358. chain := params["chain"]
  1359. action := params["action"]
  1360. if chain == "" || action == "" {
  1361. return ColorRed + "Usage: /ip firewall filter add chain=<input|forward|output> action=<accept|drop|reject> [protocol=...] [dst-port=...]\n" + ColorReset
  1362. }
  1363. r := firewall.FilterRule{
  1364. Chain: chain,
  1365. Action: action,
  1366. Protocol: params["protocol"],
  1367. SrcAddress: params["src-address"],
  1368. DstAddress: params["dst-address"],
  1369. DstPort: params["dst-port"],
  1370. Comment: params["comment"],
  1371. Disabled: false,
  1372. }
  1373. if err := e.fwMgr.SaveFilterRule(r); err != nil {
  1374. return ColorRed + "Error adding filter rule: " + err.Error() + ColorReset + "\n"
  1375. }
  1376. return ColorGreen + "Firewall filter rule added.\n" + ColorReset
  1377. case "remove", "rem", "delete":
  1378. pos, _ := parseArgs(parts[1:])
  1379. if len(pos) == 0 {
  1380. return ColorRed + "Usage: /ip firewall filter remove <id>\n" + ColorReset
  1381. }
  1382. if err := e.fwMgr.DeleteFilterRule(pos[0]); err != nil {
  1383. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1384. }
  1385. return ColorGreen + fmt.Sprintf("Filter rule %s removed.\n", pos[0]) + ColorReset
  1386. case "enable":
  1387. pos, _ := parseArgs(parts[1:])
  1388. if len(pos) == 0 {
  1389. return ColorRed + "Usage: /ip firewall filter enable <id>\n" + ColorReset
  1390. }
  1391. if err := e.fwMgr.SetFilterRuleState(pos[0], false); err != nil {
  1392. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1393. }
  1394. return ColorGreen + fmt.Sprintf("Filter rule %s enabled.\n", pos[0]) + ColorReset
  1395. case "disable":
  1396. pos, _ := parseArgs(parts[1:])
  1397. if len(pos) == 0 {
  1398. return ColorRed + "Usage: /ip firewall filter disable <id>\n" + ColorReset
  1399. }
  1400. if err := e.fwMgr.SetFilterRuleState(pos[0], true); err != nil {
  1401. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1402. }
  1403. return ColorGreen + fmt.Sprintf("Filter rule %s disabled.\n", pos[0]) + ColorReset
  1404. default:
  1405. return ColorRed + "Usage: /ip firewall filter [print|add|remove|enable|disable]\n" + ColorReset
  1406. }
  1407. }
  1408. func (e *Executor) handleFirewallNAT(parts []string, user *auth.PublicUser) string {
  1409. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1410. rules := e.fwMgr.ListNATRules()
  1411. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "OUT-IFACE", "TO-ADDRESSES", "TO-PORTS", "COMMENT")
  1412. for _, r := range rules {
  1413. flags := ""
  1414. if r.Disabled {
  1415. flags += "X"
  1416. } else {
  1417. flags += " "
  1418. }
  1419. tbl.AddRow(flags, r.Chain, r.Action, r.OutInterface, r.ToAddresses, r.ToPorts, r.Comment)
  1420. }
  1421. return tbl.String()
  1422. }
  1423. if user.Role == auth.RoleRead {
  1424. return ColorRed + "Permission denied: viewer role cannot modify firewall\n" + ColorReset
  1425. }
  1426. sub := strings.ToLower(parts[0])
  1427. switch sub {
  1428. case "add":
  1429. _, params := parseArgs(parts[1:])
  1430. chain := params["chain"]
  1431. action := params["action"]
  1432. if chain == "" || action == "" {
  1433. return ColorRed + "Usage: /ip firewall nat add chain=<srcnat|dstnat> action=<masquerade|dst-nat|src-nat> [out-interface=...] [to-addresses=...]\n" + ColorReset
  1434. }
  1435. r := firewall.NATRule{
  1436. Chain: chain,
  1437. Action: action,
  1438. OutInterface: params["out-interface"],
  1439. ToAddresses: params["to-addresses"],
  1440. ToPorts: params["to-ports"],
  1441. Comment: params["comment"],
  1442. Disabled: false,
  1443. }
  1444. if err := e.fwMgr.SaveNATRule(r); err != nil {
  1445. return ColorRed + "Error adding NAT rule: " + err.Error() + ColorReset + "\n"
  1446. }
  1447. return ColorGreen + "Firewall NAT rule added.\n" + ColorReset
  1448. case "remove", "rem", "delete":
  1449. pos, _ := parseArgs(parts[1:])
  1450. if len(pos) == 0 {
  1451. return ColorRed + "Usage: /ip firewall nat remove <id>\n" + ColorReset
  1452. }
  1453. if err := e.fwMgr.DeleteNATRule(pos[0]); err != nil {
  1454. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1455. }
  1456. return ColorGreen + fmt.Sprintf("NAT rule %s removed.\n", pos[0]) + ColorReset
  1457. case "enable":
  1458. pos, _ := parseArgs(parts[1:])
  1459. if len(pos) == 0 {
  1460. return ColorRed + "Usage: /ip firewall nat enable <id>\n" + ColorReset
  1461. }
  1462. if err := e.fwMgr.SetNATRuleState(pos[0], false); err != nil {
  1463. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1464. }
  1465. return ColorGreen + fmt.Sprintf("NAT rule %s enabled.\n", pos[0]) + ColorReset
  1466. case "disable":
  1467. pos, _ := parseArgs(parts[1:])
  1468. if len(pos) == 0 {
  1469. return ColorRed + "Usage: /ip firewall nat disable <id>\n" + ColorReset
  1470. }
  1471. if err := e.fwMgr.SetNATRuleState(pos[0], true); err != nil {
  1472. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1473. }
  1474. return ColorGreen + fmt.Sprintf("NAT rule %s disabled.\n", pos[0]) + ColorReset
  1475. default:
  1476. return ColorRed + "Usage: /ip firewall nat [print|add|remove|enable|disable]\n" + ColorReset
  1477. }
  1478. }
  1479. func (e *Executor) handleFirewallMangle(parts []string, user *auth.PublicUser) string {
  1480. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1481. rules := e.fwMgr.ListMangleRules()
  1482. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "PASSTHROUGH", "COMMENT")
  1483. for _, r := range rules {
  1484. flags := ""
  1485. if r.Disabled {
  1486. flags += "X"
  1487. } else {
  1488. flags += " "
  1489. }
  1490. tbl.AddRow(flags, r.Chain, r.Action, fmt.Sprintf("%t", r.Passthrough), r.Comment)
  1491. }
  1492. return tbl.String()
  1493. }
  1494. if user.Role == auth.RoleRead {
  1495. return ColorRed + "Permission denied\n" + ColorReset
  1496. }
  1497. sub := strings.ToLower(parts[0])
  1498. switch sub {
  1499. case "add":
  1500. _, params := parseArgs(parts[1:])
  1501. chain := params["chain"]
  1502. action := params["action"]
  1503. if chain == "" || action == "" {
  1504. return ColorRed + "Usage: /ip firewall mangle add chain=... action=...\n" + ColorReset
  1505. }
  1506. r := firewall.MangleRule{
  1507. Chain: chain,
  1508. Action: action,
  1509. Passthrough: true,
  1510. Comment: params["comment"],
  1511. Disabled: false,
  1512. }
  1513. if err := e.fwMgr.SaveMangleRule(r); err != nil {
  1514. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1515. }
  1516. return ColorGreen + "Mangle rule added.\n" + ColorReset
  1517. case "remove", "rem", "delete":
  1518. pos, _ := parseArgs(parts[1:])
  1519. if len(pos) == 0 {
  1520. return ColorRed + "Usage: /ip firewall mangle remove <id>\n" + ColorReset
  1521. }
  1522. if err := e.fwMgr.DeleteMangleRule(pos[0]); err != nil {
  1523. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1524. }
  1525. return ColorGreen + fmt.Sprintf("Mangle rule %s removed.\n", pos[0]) + ColorReset
  1526. default:
  1527. return ColorRed + "Usage: /ip firewall mangle [print|add|remove]\n" + ColorReset
  1528. }
  1529. }
  1530. func (e *Executor) handleFirewallAddressList(parts []string, user *auth.PublicUser) string {
  1531. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1532. entries := e.fwMgr.ListAddressLists()
  1533. tbl := NewTable("FLAGS", "LIST", "ADDRESS", "COMMENT")
  1534. for _, entry := range entries {
  1535. flags := ""
  1536. if entry.Disabled {
  1537. flags += "X"
  1538. } else {
  1539. flags += " "
  1540. }
  1541. tbl.AddRow(flags, entry.Name, entry.Address, entry.Comment)
  1542. }
  1543. return tbl.String()
  1544. }
  1545. if user.Role == auth.RoleRead {
  1546. return ColorRed + "Permission denied\n" + ColorReset
  1547. }
  1548. sub := strings.ToLower(parts[0])
  1549. switch sub {
  1550. case "add":
  1551. _, params := parseArgs(parts[1:])
  1552. list := params["list"]
  1553. addr := params["address"]
  1554. if list == "" || addr == "" {
  1555. return ColorRed + "Usage: /ip firewall address-list add list=<name> address=<cidr> [comment=...]\n" + ColorReset
  1556. }
  1557. entry := firewall.AddressListEntry{
  1558. Name: list,
  1559. Address: addr,
  1560. Comment: params["comment"],
  1561. Disabled: false,
  1562. }
  1563. if err := e.fwMgr.SaveAddressListEntry(entry); err != nil {
  1564. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1565. }
  1566. return ColorGreen + fmt.Sprintf("Added %s to address list %s.\n", addr, list) + ColorReset
  1567. case "remove", "rem", "delete":
  1568. pos, _ := parseArgs(parts[1:])
  1569. if len(pos) == 0 {
  1570. return ColorRed + "Usage: /ip firewall address-list remove <id>\n" + ColorReset
  1571. }
  1572. if err := e.fwMgr.DeleteAddressListEntry(pos[0]); err != nil {
  1573. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1574. }
  1575. return ColorGreen + fmt.Sprintf("Address list entry %s removed.\n", pos[0]) + ColorReset
  1576. default:
  1577. return ColorRed + "Usage: /ip firewall address-list [print|add|remove]\n" + ColorReset
  1578. }
  1579. }
  1580. // --- IP Service ---
  1581. func (e *Executor) handleIPService(parts []string, user *auth.PublicUser) string {
  1582. if e.servicesMgr == nil {
  1583. return ColorRed + "Services manager not available.\n" + ColorReset
  1584. }
  1585. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1586. svcs := e.servicesMgr.ListServices()
  1587. tbl := NewTable("NAME", "PORT", "DISABLED", "CERTIFICATE", "COMMENT")
  1588. for _, s := range svcs {
  1589. disStr := ColorGreen + "no" + ColorReset
  1590. if s.Disabled {
  1591. disStr = ColorRed + "yes" + ColorReset
  1592. }
  1593. certStr := "-"
  1594. if s.Certificate != "" {
  1595. certStr = ColorCyan + s.Certificate + ColorReset
  1596. }
  1597. tbl.AddRow(s.Name, fmt.Sprintf("%d", s.Port), disStr, certStr, s.Comment)
  1598. }
  1599. return tbl.String()
  1600. }
  1601. if user.Role != auth.RoleFull {
  1602. return ColorRed + "Permission denied: admin role required to manage services\n" + ColorReset
  1603. }
  1604. sub := strings.ToLower(parts[0])
  1605. switch sub {
  1606. case "enable":
  1607. pos, _ := parseArgs(parts[1:])
  1608. if len(pos) == 0 {
  1609. return ColorRed + "Usage: /ip service enable <name>\n" + ColorReset
  1610. }
  1611. name := pos[0]
  1612. if err := e.servicesMgr.SetServiceState(name, false); err != nil {
  1613. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1614. }
  1615. return ColorGreen + fmt.Sprintf("Service %s enabled.\n", name) + ColorReset
  1616. case "disable":
  1617. pos, _ := parseArgs(parts[1:])
  1618. if len(pos) == 0 {
  1619. return ColorRed + "Usage: /ip service disable <name>\n" + ColorReset
  1620. }
  1621. name := pos[0]
  1622. if err := e.servicesMgr.SetServiceState(name, true); err != nil {
  1623. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1624. }
  1625. return ColorGreen + fmt.Sprintf("Service %s disabled.\n", name) + ColorReset
  1626. case "set":
  1627. pos, params := parseArgs(parts[1:])
  1628. if len(pos) == 0 {
  1629. return ColorRed + "Usage: /ip service set <name> [port=<port>] [disabled=yes|no] [certificate=<cert_name>]\n" + ColorReset
  1630. }
  1631. name := pos[0]
  1632. svc, ok := e.servicesMgr.GetService(name)
  1633. if !ok {
  1634. return ColorRed + fmt.Sprintf("Service %q not found\n", name) + ColorReset
  1635. }
  1636. port := svc.Port
  1637. disabled := svc.Disabled
  1638. certName := svc.Certificate
  1639. if pStr, ok := params["port"]; ok {
  1640. if pVal, err := strconv.Atoi(pStr); err == nil && pVal > 0 && pVal < 65536 {
  1641. port = pVal
  1642. }
  1643. }
  1644. if dStr, ok := params["disabled"]; ok {
  1645. disabled = (strings.ToLower(dStr) == "yes" || strings.ToLower(dStr) == "true")
  1646. }
  1647. if cStr, ok := params["certificate"]; ok {
  1648. certName = cStr
  1649. }
  1650. if err := e.servicesMgr.SetService(name, port, disabled, certName); err != nil {
  1651. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1652. }
  1653. return ColorGreen + fmt.Sprintf("Service %s updated (port=%d, disabled=%t, certificate=%q).\n", name, port, disabled, certName) + ColorReset
  1654. default:
  1655. return ColorRed + "Usage: /ip service [print|enable|disable|set]\n" + ColorReset
  1656. }
  1657. }
  1658. // ==================== DYNAMIC ROUTING ====================
  1659. func (e *Executor) handleRouting(parts []string, user *auth.PublicUser) string {
  1660. if e.dynRoutingMgr == nil {
  1661. return ColorRed + "Dynamic routing subsystem not available.\n" + ColorReset
  1662. }
  1663. if len(parts) == 0 {
  1664. return ColorYellow + "Routing subsystem. Available: bgp, ospf, filter\n" + ColorReset
  1665. }
  1666. sub := strings.ToLower(parts[0])
  1667. switch sub {
  1668. case "bgp":
  1669. return e.handleRoutingBGP(parts[1:], user)
  1670. case "ospf":
  1671. return e.handleRoutingOSPF(parts[1:], user)
  1672. case "filter":
  1673. return e.handleRoutingFilter(parts[1:], user)
  1674. default:
  1675. return ColorRed + "Unknown routing protocol: " + sub + "\n" + ColorReset
  1676. }
  1677. }
  1678. func (e *Executor) handleRoutingBGP(parts []string, user *auth.PublicUser) string {
  1679. if len(parts) == 0 || parts[0] == "instance" {
  1680. inst := e.dynRoutingMgr.GetBGPInstance()
  1681. out := fmt.Sprintf(" enabled: %t\n", inst.Enabled)
  1682. out += fmt.Sprintf(" as: %d\n", inst.AS)
  1683. out += fmt.Sprintf(" router-id: %s\n", inst.RouterID)
  1684. return out
  1685. }
  1686. sub := strings.ToLower(parts[0])
  1687. switch sub {
  1688. case "peer", "connection", "peers":
  1689. peers := e.dynRoutingMgr.ListBGPPeers()
  1690. tbl := NewTable("FLAGS", "NAME", "REMOTE-AS", "REMOTE-ADDRESS", "STATUS", "UPTIME")
  1691. for _, p := range peers {
  1692. flags := ""
  1693. if p.Disabled {
  1694. flags += "X"
  1695. } else {
  1696. flags += "E"
  1697. }
  1698. st := p.State
  1699. if st == "" {
  1700. st = "Active"
  1701. }
  1702. tbl.AddRow(flags, p.Name, fmt.Sprintf("%d", p.RemoteAS), p.RemoteAddress, st, p.Uptime)
  1703. }
  1704. return tbl.String()
  1705. case "network", "networks":
  1706. nets := e.dynRoutingMgr.ListBGPNetworks()
  1707. tbl := NewTable("NETWORK", "COMMENT")
  1708. for _, n := range nets {
  1709. tbl.AddRow(n.Network, n.Comment)
  1710. }
  1711. return tbl.String()
  1712. case "routes":
  1713. routes := e.dynRoutingMgr.GetBGPRoutes()
  1714. tbl := NewTable("PREFIX", "NEXT-HOP", "PATH", "METRIC", "WEIGHT")
  1715. for _, r := range routes {
  1716. tbl.AddRow(r.Prefix, r.NextHop, r.Path, fmt.Sprintf("%d", r.Metric), fmt.Sprintf("%d", r.Weight))
  1717. }
  1718. return tbl.String()
  1719. default:
  1720. return ColorRed + "Usage: /routing bgp [instance|peer|network|routes]\n" + ColorReset
  1721. }
  1722. }
  1723. func (e *Executor) handleRoutingOSPF(parts []string, user *auth.PublicUser) string {
  1724. if len(parts) == 0 || parts[0] == "instance" {
  1725. instances := e.dynRoutingMgr.ListOSPFInstances()
  1726. tbl := NewTable("NAME", "ROUTER-ID", "VERSION", "STATUS")
  1727. for _, inst := range instances {
  1728. st := "enabled"
  1729. if inst.Disabled {
  1730. st = "disabled"
  1731. }
  1732. tbl.AddRow(inst.Name, inst.RouterID, fmt.Sprintf("%d", inst.Version), st)
  1733. }
  1734. return tbl.String()
  1735. }
  1736. sub := strings.ToLower(parts[0])
  1737. switch sub {
  1738. case "area", "areas":
  1739. areas := e.dynRoutingMgr.ListOSPFAreas()
  1740. tbl := NewTable("NAME", "AREA-ID", "INSTANCE", "TYPE")
  1741. for _, a := range areas {
  1742. tbl.AddRow(a.Name, a.AreaID, a.Instance, a.AreaType)
  1743. }
  1744. return tbl.String()
  1745. case "interface", "interfaces":
  1746. ifaces := e.dynRoutingMgr.ListOSPFInterfaces()
  1747. tbl := NewTable("INTERFACE", "AREA", "COST", "PRIORITY", "NETWORK-TYPE")
  1748. for _, i := range ifaces {
  1749. tbl.AddRow(i.Interface, i.Area, fmt.Sprintf("%d", i.Cost), fmt.Sprintf("%d", i.Priority), i.NetworkType)
  1750. }
  1751. return tbl.String()
  1752. case "neighbor", "neighbors":
  1753. neighbors := e.dynRoutingMgr.GetOSPFNeighbors()
  1754. tbl := NewTable("ROUTER-ID", "ADDRESS", "INTERFACE", "STATE", "PRIORITY")
  1755. for _, n := range neighbors {
  1756. tbl.AddRow(n.NeighborID, n.Address, n.Interface, n.State, fmt.Sprintf("%d", n.Priority))
  1757. }
  1758. return tbl.String()
  1759. default:
  1760. return ColorRed + "Usage: /routing ospf [instance|area|interface|neighbor]\n" + ColorReset
  1761. }
  1762. }
  1763. func (e *Executor) handleRoutingFilter(parts []string, user *auth.PublicUser) string {
  1764. rules := e.dynRoutingMgr.ListRoutingFilters()
  1765. tbl := NewTable("CHAIN", "RULE", "ACTION", "STATUS")
  1766. for _, r := range rules {
  1767. st := "enabled"
  1768. if r.Disabled {
  1769. st = "disabled"
  1770. }
  1771. tbl.AddRow(r.Chain, r.Rule, r.Action, st)
  1772. }
  1773. return tbl.String()
  1774. }
  1775. // ==================== TOOLS ====================
  1776. func (e *Executor) handleTool(parts []string) string {
  1777. if len(parts) == 0 {
  1778. return ColorYellow + "Tools available: ping, traceroute, bandwidth-test\n" + ColorReset
  1779. }
  1780. sub := strings.ToLower(parts[0])
  1781. switch sub {
  1782. case "ping":
  1783. return e.handlePing(parts[1:])
  1784. case "traceroute":
  1785. return e.handleTraceroute(parts[1:])
  1786. case "bandwidth-test", "btest":
  1787. return e.handleBandwidthTest(parts[1:])
  1788. default:
  1789. return ColorRed + "Unknown tool: " + sub + "\n" + ColorReset
  1790. }
  1791. }
  1792. func (e *Executor) handlePing(args []string) string {
  1793. if len(args) == 0 {
  1794. return ColorRed + "Usage: ping <host> [count=4]\n" + ColorReset
  1795. }
  1796. pos, params := parseArgs(args)
  1797. if len(pos) == 0 {
  1798. return ColorRed + "Missing host target for ping\n" + ColorReset
  1799. }
  1800. target := pos[0]
  1801. count := "4"
  1802. if c, ok := params["count"]; ok {
  1803. count = c
  1804. }
  1805. cmd := exec.Command("ping", "-c", count, target)
  1806. out, err := cmd.CombinedOutput()
  1807. if err != nil && len(out) == 0 {
  1808. return ColorRed + "Ping execution failed: " + err.Error() + ColorReset + "\n"
  1809. }
  1810. return string(out)
  1811. }
  1812. func (e *Executor) handleTraceroute(args []string) string {
  1813. if len(args) == 0 {
  1814. return ColorRed + "Usage: traceroute <host>\n" + ColorReset
  1815. }
  1816. pos, _ := parseArgs(args)
  1817. if len(pos) == 0 {
  1818. return ColorRed + "Missing host target for traceroute\n" + ColorReset
  1819. }
  1820. target := pos[0]
  1821. cmd := exec.Command("traceroute", "-n", "-m", "15", target)
  1822. out, err := cmd.CombinedOutput()
  1823. if err != nil {
  1824. cmdFallback := exec.Command("tracepath", "-n", target)
  1825. fbOut, fbErr := cmdFallback.CombinedOutput()
  1826. if fbErr == nil {
  1827. return string(fbOut)
  1828. }
  1829. return ColorRed + "Traceroute failed: " + err.Error() + "\n" + string(out) + ColorReset + "\n"
  1830. }
  1831. return string(out)
  1832. }
  1833. func (e *Executor) handleBandwidthTest(args []string) string {
  1834. if e.bwtestMgr == nil {
  1835. return ColorRed + "Bandwidth test manager not available.\n" + ColorReset
  1836. }
  1837. pos, params := parseArgs(args)
  1838. if len(pos) == 0 {
  1839. return ColorRed + "Usage: /tool bandwidth-test <target-ip> [direction=both|transmit|receive] [protocol=udp|tcp] [duration=5s]\n" + ColorReset
  1840. }
  1841. target := pos[0]
  1842. dir := "both"
  1843. if d, ok := params["direction"]; ok {
  1844. dir = d
  1845. }
  1846. switch dir {
  1847. case "transmit":
  1848. dir = "tx"
  1849. case "receive":
  1850. dir = "rx"
  1851. case "tx", "rx", "both":
  1852. default:
  1853. return cliError(fmt.Errorf("direction must be both, tx/transmit, or rx/receive"))
  1854. }
  1855. proto := "udp"
  1856. if p, ok := params["protocol"]; ok {
  1857. proto = p
  1858. }
  1859. if proto != "udp" && proto != "tcp" {
  1860. return cliError(fmt.Errorf("protocol must be udp or tcp"))
  1861. }
  1862. dur := 5 * time.Second
  1863. if dStr, ok := params["duration"]; ok {
  1864. if dVal, err := time.ParseDuration(dStr); err == nil && dVal >= time.Second {
  1865. dur = dVal
  1866. } else {
  1867. return cliError(fmt.Errorf("duration must be at least 1s"))
  1868. }
  1869. }
  1870. req := bwtest.SessionRequest{
  1871. Target: target,
  1872. Direction: dir,
  1873. Protocol: proto,
  1874. Duration: int(dur.Seconds()),
  1875. }
  1876. ctx, cancel := context.WithTimeout(context.Background(), dur+8*time.Second)
  1877. defer cancel()
  1878. if err := e.bwtestMgr.StartTest(ctx, req); err != nil {
  1879. return ColorRed + "Failed to initiate bandwidth test: " + err.Error() + ColorReset + "\n"
  1880. }
  1881. st := e.bwtestMgr.GetTestStatus()
  1882. ticker := time.NewTicker(100 * time.Millisecond)
  1883. defer ticker.Stop()
  1884. for st.Running {
  1885. select {
  1886. case <-ctx.Done():
  1887. e.bwtestMgr.StopTest()
  1888. return cliError(fmt.Errorf("bandwidth test did not finish before the deadline; stop requested"))
  1889. case <-ticker.C:
  1890. st = e.bwtestMgr.GetTestStatus()
  1891. }
  1892. }
  1893. if st.Error != "" {
  1894. return cliError(fmt.Errorf("bandwidth test failed: %s", st.Error))
  1895. }
  1896. out := fmt.Sprintf("Bandwidth test to %s completed:\n", target)
  1897. if st.Summary != nil {
  1898. out += fmt.Sprintf(" Tx Avg : %.2f Mbps\n", st.Summary.AvgTxMbps)
  1899. out += fmt.Sprintf(" Rx Avg : %.2f Mbps\n", st.Summary.AvgRxMbps)
  1900. out += fmt.Sprintf(" Loss : %.1f%%\n", st.Summary.AvgLossPct)
  1901. out += fmt.Sprintf(" Jitter : %.2f ms\n", st.Summary.AvgJitterMs)
  1902. } else if st.CurrentSample != nil {
  1903. out += fmt.Sprintf(" Tx Rate : %.2f Mbps\n", st.CurrentSample.TxMbps)
  1904. out += fmt.Sprintf(" Rx Rate : %.2f Mbps\n", st.CurrentSample.RxMbps)
  1905. } else {
  1906. out += " Test finished.\n"
  1907. }
  1908. return out
  1909. }
  1910. // ==================== USER MANAGEMENT ====================
  1911. func (e *Executor) handleUser(parts []string, actor *auth.PublicUser) string {
  1912. if e.authMgr == nil {
  1913. return ColorRed + "Auth manager not available.\n" + ColorReset
  1914. }
  1915. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1916. users := e.authMgr.ListUsers()
  1917. tbl := NewTable("FLAGS", "NAME", "ROLE", "LAST-LOGIN", "COMMENT")
  1918. for _, u := range users {
  1919. flags := ""
  1920. if u.Disabled {
  1921. flags += "X"
  1922. } else {
  1923. flags += " "
  1924. }
  1925. lastLogin := "never"
  1926. if !u.LastLoginAt.IsZero() {
  1927. lastLogin = u.LastLoginAt.Format("Jan/02 15:04:05")
  1928. }
  1929. tbl.AddRow(flags, u.Username, string(u.Role), lastLogin, u.Comment)
  1930. }
  1931. return tbl.String()
  1932. }
  1933. if actor.Role != auth.RoleFull {
  1934. return ColorRed + "Permission denied: admin role required to manage users\n" + ColorReset
  1935. }
  1936. sub := strings.ToLower(parts[0])
  1937. switch sub {
  1938. case "add":
  1939. _, params := parseArgs(parts[1:])
  1940. name := params["name"]
  1941. pwd := params["password"]
  1942. roleStr := params["role"]
  1943. if roleStr == "" {
  1944. roleStr = "admin"
  1945. }
  1946. comment := params["comment"]
  1947. if name == "" || pwd == "" {
  1948. return ColorRed + "Usage: /user add name=<username> password=<pwd> [role=admin|operator|viewer] [comment=...]\n" + ColorReset
  1949. }
  1950. req := auth.CreateUserRequest{
  1951. Username: name,
  1952. Password: pwd,
  1953. Role: auth.Role(roleStr),
  1954. Comment: comment,
  1955. }
  1956. if err := e.authMgr.CreateUser(req); err != nil {
  1957. return ColorRed + "Error creating user: " + err.Error() + ColorReset + "\n"
  1958. }
  1959. return ColorGreen + fmt.Sprintf("User %s created.\n", name) + ColorReset
  1960. case "remove", "rem", "delete":
  1961. pos, _ := parseArgs(parts[1:])
  1962. if len(pos) == 0 {
  1963. return ColorRed + "Usage: /user remove <username>\n" + ColorReset
  1964. }
  1965. name := pos[0]
  1966. if err := e.authMgr.DeleteUser(name, actor.Username); err != nil {
  1967. return ColorRed + "Error removing user: " + err.Error() + ColorReset + "\n"
  1968. }
  1969. return ColorGreen + fmt.Sprintf("User %s removed.\n", name) + ColorReset
  1970. case "set":
  1971. pos, params := parseArgs(parts[1:])
  1972. if len(pos) == 0 {
  1973. return ColorRed + "Usage: /user set <username> [password=...] [role=...] [disabled=yes|no]\n" + ColorReset
  1974. }
  1975. name := pos[0]
  1976. pub, err := e.authMgr.GetPublicUser(name)
  1977. if err != nil || pub == nil {
  1978. return ColorRed + "User not found\n" + ColorReset
  1979. }
  1980. role := pub.Role
  1981. if rStr, ok := params["role"]; ok {
  1982. role = auth.Role(rStr)
  1983. }
  1984. dis := pub.Disabled
  1985. if dStr, ok := params["disabled"]; ok {
  1986. dis = (strings.ToLower(dStr) == "yes" || strings.ToLower(dStr) == "true")
  1987. }
  1988. cmt := pub.Comment
  1989. if c, ok := params["comment"]; ok {
  1990. cmt = c
  1991. }
  1992. req := auth.UpdateUserRequest{
  1993. Role: role,
  1994. Disabled: dis,
  1995. Comment: cmt,
  1996. }
  1997. if err := e.authMgr.UpdateUser(name, req); err != nil {
  1998. return ColorRed + "Error updating user: " + err.Error() + ColorReset + "\n"
  1999. }
  2000. if pwd, ok := params["password"]; ok && pwd != "" {
  2001. if err := e.authMgr.ChangePassword(name, "", pwd, true); err != nil {
  2002. return ColorRed + "Error resetting password: " + err.Error() + ColorReset + "\n"
  2003. }
  2004. }
  2005. return ColorGreen + fmt.Sprintf("User %s updated.\n", name) + ColorReset
  2006. default:
  2007. return ColorRed + "Usage: /user [print|add|remove|set]\n" + ColorReset
  2008. }
  2009. }
  2010. // ==================== SYSTEM ====================
  2011. func (e *Executor) handleSystem(parts []string, user *auth.PublicUser) string {
  2012. if len(parts) == 0 {
  2013. return ColorYellow + "System commands: resource print, reboot, identity print, identity set name=...\n" + ColorReset
  2014. }
  2015. sub := strings.ToLower(parts[0])
  2016. switch sub {
  2017. case "resource":
  2018. st, err := system.GetStatus()
  2019. if err != nil {
  2020. return ColorRed + "Error retrieving system resources: " + err.Error() + ColorReset + "\n"
  2021. }
  2022. freeMemMB := uint64(0)
  2023. if st.MemTotalMB > st.MemUsedMB {
  2024. freeMemMB = st.MemTotalMB - st.MemUsedMB
  2025. }
  2026. out := " uptime: " + formatUptime(int64(st.UptimeSeconds)) + "\n"
  2027. out += fmt.Sprintf(" version: DebTik 1.0 (Kernel %s)\n", st.Kernel)
  2028. out += fmt.Sprintf(" free-memory: %d MiB\n", freeMemMB)
  2029. out += fmt.Sprintf(" total-memory: %d MiB\n", st.MemTotalMB)
  2030. out += fmt.Sprintf(" cpu-count: %d\n", st.CPUCores)
  2031. out += fmt.Sprintf(" cpu-load: %.1f%%\n", st.CPUPct)
  2032. out += fmt.Sprintf(" os: %s\n", st.OS)
  2033. return out
  2034. case "identity":
  2035. if len(parts) > 1 && parts[1] == "set" {
  2036. if user.Role != auth.RoleFull {
  2037. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  2038. }
  2039. _, params := parseArgs(parts[2:])
  2040. name := params["name"]
  2041. if name == "" {
  2042. return ColorRed + "Usage: /system identity set name=<new-name>\n" + ColorReset
  2043. }
  2044. if err := exec.Command("hostnamectl", "set-hostname", name).Run(); err != nil {
  2045. return ColorRed + "Error setting identity: " + err.Error() + ColorReset + "\n"
  2046. }
  2047. return ColorGreen + fmt.Sprintf("System identity set to %q.\n", name) + ColorReset
  2048. }
  2049. st, _ := system.GetStatus()
  2050. h := "DebTik"
  2051. if st != nil && st.Hostname != "" {
  2052. h = st.Hostname
  2053. }
  2054. return fmt.Sprintf(" name: %s\n", h)
  2055. case "reboot":
  2056. if user.Role != auth.RoleFull {
  2057. return ColorRed + "Permission denied: admin role required to reboot\n" + ColorReset
  2058. }
  2059. go func() {
  2060. time.Sleep(1 * time.Second)
  2061. _ = exec.Command("systemctl", "reboot").Run()
  2062. }()
  2063. return ColorYellow + "System reboot initiated! Connection will terminate.\n" + ColorReset
  2064. default:
  2065. return ColorRed + "Unknown system subcommand: " + sub + "\n" + ColorReset
  2066. }
  2067. }
  2068. func (e *Executor) showHelp() string {
  2069. b := strings.Builder{}
  2070. b.WriteString(ColorCyan + ColorBold + "DebTik CLI Command Directory:\n" + ColorReset)
  2071. b.WriteString(" /interface - Network interface manager (print, enable, disable, set)\n")
  2072. b.WriteString(" /interface vlan - VLAN virtual interfaces (print, add, remove)\n")
  2073. b.WriteString(" /interface bridge - L2 Bridge and ports (print, add, remove, port)\n")
  2074. b.WriteString(" /interface aerotun - Encrypted tunnels (print detail=yes, add, set, enable, disable, start, stop, remove, genkey)\n")
  2075. b.WriteString(" /interface wireguard - WireGuard tunnels & peers (print, add, peers)\n")
  2076. b.WriteString(" /interface eoip - Ethernet over IP GRE tunnels (print, add, remove)\n")
  2077. b.WriteString(" /interface vrrp - Virtual Router Redundancy Protocol (print, add, remove)\n")
  2078. b.WriteString(" /interface pppoe-client - PPPoE WAN Client sessions (print, add, connect)\n")
  2079. b.WriteString(" /ip\n")
  2080. b.WriteString(" /ip address - IP address assignments (print, add, remove)\n")
  2081. b.WriteString(" /ip route - Static routes & tables (print, add, remove, rule)\n")
  2082. b.WriteString(" /ip dhcp-client - DHCP client on WAN links (print, add, renew, release)\n")
  2083. b.WriteString(" /ip dhcp-server - Local DHCP server & leases (print, add, lease, network)\n")
  2084. b.WriteString(" /ip dns - Router DNS resolver (print, set, cache flush)\n")
  2085. b.WriteString(" /ip firewall - Netfilter / iptables firewall\n")
  2086. b.WriteString(" /ip firewall filter - Traffic filtering rules (print, add, remove)\n")
  2087. b.WriteString(" /ip firewall nat - Source & destination NAT rules (print, add, remove)\n")
  2088. b.WriteString(" /ip firewall mangle - Packet marking and QoS (print, add, remove)\n")
  2089. b.WriteString(" /ip firewall address-list - Dynamic & static address groups (print, add, remove)\n")
  2090. b.WriteString(" /ip service - Management services: Telnet, SSH, Web (print, enable, disable, set)\n")
  2091. b.WriteString(" /routing - Dynamic routing engine\n")
  2092. b.WriteString(" /routing bgp - BGP instance, peers, routes (instance, peer, routes)\n")
  2093. b.WriteString(" /routing ospf - OSPF areas, interfaces, neighbors (instance, area, neighbor)\n")
  2094. b.WriteString(" /routing filter - Route filtering policies (print)\n")
  2095. b.WriteString(" /tool - Diagnostics & performance testing\n")
  2096. b.WriteString(" /tool ping <host> - ICMP ping test\n")
  2097. b.WriteString(" /tool traceroute <host> - Route trace analysis\n")
  2098. b.WriteString(" /tool bandwidth-test <host> - High performance throughput & jitter test\n")
  2099. b.WriteString(" /log - System activity logs (print [limit=N] [where=filter])\n")
  2100. b.WriteString(" /user - Local user accounts (print, add, remove, set)\n")
  2101. b.WriteString(" /system - Router hardware and core controls\n")
  2102. b.WriteString(" /system resource print - CPU, RAM, disk, uptime metrics\n")
  2103. b.WriteString(" /system identity - Hostname identity (print, set name=...)\n")
  2104. b.WriteString(" /system reboot - Safe system restart\n")
  2105. b.WriteString(" quit / exit - Disconnect session\n")
  2106. return b.String()
  2107. }
  2108. // ==================== CERTIFICATES (/certificate) ====================
  2109. func (e *Executor) handleCertificate(parts []string, user *auth.PublicUser) string {
  2110. if e.certMgr == nil {
  2111. return ColorRed + "Certificates manager not available.\n" + ColorReset
  2112. }
  2113. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2114. list := e.certMgr.ListCertificates()
  2115. tbl := NewTable("#", "NAME", "COMMON-NAME", "TYPE", "ISSUER", "EXPIRES", "SANS")
  2116. for idx, c := range list {
  2117. tStr := ColorGreen + c.Type + ColorReset
  2118. if c.IsCA {
  2119. tStr = ColorCyan + "ca" + ColorReset
  2120. }
  2121. expStr := c.ValidTo.Format("Jan/02/2006")
  2122. if time.Now().After(c.ValidTo) {
  2123. expStr = ColorRed + expStr + " (EXPIRED)" + ColorReset
  2124. }
  2125. sansStr := strings.Join(c.SANs, ",")
  2126. if sansStr == "" {
  2127. sansStr = "-"
  2128. }
  2129. tbl.AddRow(fmt.Sprintf("%d", idx), c.Name, c.CommonName, tStr, c.Issuer, expStr, sansStr)
  2130. }
  2131. return tbl.String()
  2132. }
  2133. if user.Role != auth.RoleFull {
  2134. return ColorRed + "Permission denied: admin role required to manage certificates\n" + ColorReset
  2135. }
  2136. sub := strings.ToLower(parts[0])
  2137. switch sub {
  2138. case "add-ca":
  2139. _, params := parseArgs(parts[1:])
  2140. name := params["name"]
  2141. cn := params["common-name"]
  2142. if name == "" || cn == "" {
  2143. return ColorRed + "Usage: /certificate add-ca name=<name> common-name=<cn> [days=<days>] [key-type=RSA-2048|RSA-4096|ECDSA-P256]\n" + ColorReset
  2144. }
  2145. days := 3650
  2146. if dStr, ok := params["days"]; ok {
  2147. if dVal, err := strconv.Atoi(dStr); err == nil && dVal > 0 {
  2148. days = dVal
  2149. }
  2150. }
  2151. keyType := params["key-type"]
  2152. if keyType == "" {
  2153. keyType = "RSA-2048"
  2154. }
  2155. info, err := e.certMgr.CreateCA(certs.CreateCARequest{
  2156. Name: name,
  2157. CommonName: cn,
  2158. Days: days,
  2159. KeyType: keyType,
  2160. Comment: params["comment"],
  2161. })
  2162. if err != nil {
  2163. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2164. }
  2165. return ColorGreen + fmt.Sprintf("Certificate Authority %q created successfully.\n", info.Name) + ColorReset
  2166. case "add":
  2167. _, params := parseArgs(parts[1:])
  2168. name := params["name"]
  2169. cn := params["common-name"]
  2170. if name == "" || cn == "" {
  2171. return ColorRed + "Usage: /certificate add name=<name> common-name=<cn> [ca=<ca_name>] [sans=<san1,san2>] [days=<days>] [key-type=...]\n" + ColorReset
  2172. }
  2173. days := 365
  2174. if dStr, ok := params["days"]; ok {
  2175. if dVal, err := strconv.Atoi(dStr); err == nil && dVal > 0 {
  2176. days = dVal
  2177. }
  2178. }
  2179. var sans []string
  2180. if sStr, ok := params["sans"]; ok {
  2181. for _, s := range strings.Split(sStr, ",") {
  2182. if s = strings.TrimSpace(s); s != "" {
  2183. sans = append(sans, s)
  2184. }
  2185. }
  2186. }
  2187. keyType := params["key-type"]
  2188. if keyType == "" {
  2189. keyType = "RSA-2048"
  2190. }
  2191. info, err := e.certMgr.CreateCertificate(certs.CreateCertRequest{
  2192. Name: name,
  2193. CommonName: cn,
  2194. CAName: params["ca"],
  2195. SANs: sans,
  2196. Days: days,
  2197. KeyType: keyType,
  2198. IsServer: true,
  2199. Comment: params["comment"],
  2200. })
  2201. if err != nil {
  2202. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2203. }
  2204. return ColorGreen + fmt.Sprintf("Certificate %q created successfully (Issuer: %s).\n", info.Name, info.Issuer) + ColorReset
  2205. case "issue-letsencrypt":
  2206. _, params := parseArgs(parts[1:])
  2207. domain := params["domain"]
  2208. email := params["email"]
  2209. if domain == "" || email == "" {
  2210. return ColorRed + "Usage: /certificate issue-letsencrypt domain=<domain> email=<email> [name=<name>] [staging=yes|no]\n" + ColorReset
  2211. }
  2212. name := params["name"]
  2213. if name == "" {
  2214. name = domain
  2215. }
  2216. staging := strings.ToLower(params["staging"]) == "yes" || strings.ToLower(params["staging"]) == "true"
  2217. ctx, cancel := context.WithTimeout(context.Background(), 3*time.Minute)
  2218. defer cancel()
  2219. info, err := e.certMgr.IssueLetsEncrypt(ctx, certs.LetsEncryptRequest{
  2220. Name: name,
  2221. Domain: domain,
  2222. Email: email,
  2223. Staging: staging,
  2224. Comment: params["comment"],
  2225. })
  2226. if err != nil {
  2227. return ColorRed + "Let's Encrypt Error: " + err.Error() + ColorReset + "\n"
  2228. }
  2229. return ColorGreen + fmt.Sprintf("Let's Encrypt certificate %q issued successfully!\n", info.Name) + ColorReset
  2230. case "export":
  2231. pos, _ := parseArgs(parts[1:])
  2232. if len(pos) == 0 {
  2233. return ColorRed + "Usage: /certificate export <name>\n" + ColorReset
  2234. }
  2235. certPEM, keyPEM, err := e.certMgr.ExportPEM(pos[0])
  2236. if err != nil {
  2237. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2238. }
  2239. return fmt.Sprintf("----- CERTIFICATE -----\n%s\n----- PRIVATE KEY -----\n%s\n", certPEM, keyPEM)
  2240. case "remove", "rem", "delete":
  2241. pos, _ := parseArgs(parts[1:])
  2242. if len(pos) == 0 {
  2243. return ColorRed + "Usage: /certificate remove <name>\n" + ColorReset
  2244. }
  2245. name := pos[0]
  2246. if e.servicesMgr != nil {
  2247. if sslSvc, ok := e.servicesMgr.GetService("www-ssl"); ok {
  2248. if strings.EqualFold(sslSvc.Certificate, name) && !sslSvc.Disabled {
  2249. return ColorRed + "Cannot remove certificate: actively used by enabled www-ssl service\n" + ColorReset
  2250. }
  2251. }
  2252. }
  2253. if err := e.certMgr.DeleteCertificate(name); err != nil {
  2254. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  2255. }
  2256. return ColorGreen + fmt.Sprintf("Certificate %s removed.\n", name) + ColorReset
  2257. default:
  2258. return ColorRed + "Usage: /certificate [print|add-ca|add|issue-letsencrypt|export|remove]\n" + ColorReset
  2259. }
  2260. }
  2261. func formatUptime(seconds int64) string {
  2262. d := seconds / 86400
  2263. h := (seconds % 86400) / 3600
  2264. m := (seconds % 3600) / 60
  2265. s := seconds % 60
  2266. if d > 0 {
  2267. return fmt.Sprintf("%dd %02dh %02dm %02ds", d, h, m, s)
  2268. }
  2269. return fmt.Sprintf("%02dh %02dm %02ds", h, m, s)
  2270. }
  2271. func formatBytes(bytes int64) string {
  2272. const unit = 1024
  2273. if bytes < unit {
  2274. return fmt.Sprintf("%d B", bytes)
  2275. }
  2276. div, exp := int64(unit), 0
  2277. for n := bytes / unit; n >= unit; n /= unit {
  2278. div *= unit
  2279. exp++
  2280. }
  2281. return fmt.Sprintf("%.1f %ciB", float64(bytes)/float64(div), "KMGTPE"[exp])
  2282. }
  2283. // Helpers for sorting/filtering
  2284. func sortStrings(s []string) {
  2285. sort.Strings(s)
  2286. }