executor.go 95 KB

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