executor.go 141 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548
  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/sdwan"
  27. "debtik/internal/services"
  28. "debtik/internal/system"
  29. "debtik/internal/vpn"
  30. "debtik/internal/vrrp"
  31. )
  32. type Executor struct {
  33. authMgr *auth.Manager
  34. netMgr *network.Manager
  35. routingMgr *routing.RoutingManager
  36. dynRoutingMgr *dynamicrouting.Manager
  37. fwMgr *firewall.FirewallManager
  38. dhcpMgr *dhcp.Manager
  39. dhcpServMgr *dhcp.ServerManager
  40. dnsMgr *dns.Manager
  41. bridgeMgr *bridge.BridgeManager
  42. vpnMgr *vpn.Manager
  43. vrrpMgr *vrrp.Manager
  44. bwtestMgr *bwtest.Manager
  45. servicesMgr *services.Manager
  46. certMgr *certs.CertManager
  47. proxyMgr *reverseproxy.Manager
  48. backupMgr *backup.Manager
  49. sdwanMgr *sdwan.Manager
  50. }
  51. func NewExecutor(
  52. authMgr *auth.Manager,
  53. netMgr *network.Manager,
  54. routingMgr *routing.RoutingManager,
  55. dynRoutingMgr *dynamicrouting.Manager,
  56. fwMgr *firewall.FirewallManager,
  57. dhcpMgr *dhcp.Manager,
  58. dhcpServMgr *dhcp.ServerManager,
  59. dnsMgr *dns.Manager,
  60. bridgeMgr *bridge.BridgeManager,
  61. vpnMgr *vpn.Manager,
  62. vrrpMgr *vrrp.Manager,
  63. bwtestMgr *bwtest.Manager,
  64. servicesMgr *services.Manager,
  65. certMgr *certs.CertManager,
  66. proxyMgr *reverseproxy.Manager,
  67. backupMgr *backup.Manager,
  68. sdwanMgr *sdwan.Manager,
  69. ) *Executor {
  70. if backupMgr == nil {
  71. backupMgr = backup.NewManager("", "", "", nil)
  72. }
  73. if sdwanMgr == nil {
  74. sdwanMgr = sdwan.NewManager()
  75. }
  76. return &Executor{
  77. authMgr: authMgr,
  78. netMgr: netMgr,
  79. routingMgr: routingMgr,
  80. dynRoutingMgr: dynRoutingMgr,
  81. fwMgr: fwMgr,
  82. dhcpMgr: dhcpMgr,
  83. dhcpServMgr: dhcpServMgr,
  84. dnsMgr: dnsMgr,
  85. bridgeMgr: bridgeMgr,
  86. vpnMgr: vpnMgr,
  87. vrrpMgr: vrrpMgr,
  88. bwtestMgr: bwtestMgr,
  89. servicesMgr: servicesMgr,
  90. certMgr: certMgr,
  91. proxyMgr: proxyMgr,
  92. backupMgr: backupMgr,
  93. sdwanMgr: sdwanMgr,
  94. }
  95. }
  96. func (e *Executor) Execute(rawCmd string, user *auth.PublicUser) (output string, shouldExit bool) {
  97. // A malformed command must never take down the router or its tunnels.
  98. defer func() {
  99. if recover() != nil {
  100. logging.Log("cli,error", "Command handler panicked")
  101. output = ColorRed + "Internal command error; the router is still running.\n" + ColorReset
  102. shouldExit = false
  103. }
  104. }()
  105. cmd := strings.TrimSpace(rawCmd)
  106. if cmd == "" {
  107. return "", false
  108. }
  109. cmd = strings.TrimPrefix(cmd, "/")
  110. parts, err := splitCommand(cmd)
  111. if err != nil {
  112. return ColorRed + "Syntax error: " + err.Error() + "\n" + ColorReset, false
  113. }
  114. if len(parts) == 0 {
  115. return "", false
  116. }
  117. mainCmd := strings.ToLower(parts[0])
  118. switch mainCmd {
  119. case "quit", "exit":
  120. return ColorYellow + "Connection closed. Goodbye!\n" + ColorReset, true
  121. case "help", "?":
  122. return e.showHelp(), false
  123. case "clear":
  124. return "\033[2J\033[H", false
  125. case "ping":
  126. return e.handlePing(parts[1:]), false
  127. case "traceroute":
  128. return e.handleTraceroute(parts[1:]), false
  129. case "log":
  130. return e.handleLog(parts[1:]), false
  131. case "system":
  132. return e.handleSystem(parts[1:], user), false
  133. case "interface":
  134. return e.handleInterface(parts[1:], user), false
  135. case "ip":
  136. return e.handleIP(parts[1:], user), false
  137. case "routing":
  138. return e.handleRouting(parts[1:], user), false
  139. case "tool":
  140. if user == nil || user.Role == auth.RoleRead {
  141. return ColorRed + "Permission denied: tools require write privileges\n" + ColorReset, false
  142. }
  143. return e.handleTool(parts[1:]), false
  144. case "user":
  145. return e.handleUser(parts[1:], user), false
  146. case "certificate", "certificates":
  147. return e.handleCertificate(parts[1:], user), false
  148. case "reverse-proxy", "reverseproxy":
  149. return e.handleIPReverseProxy(parts[1:], user), false
  150. case "sdwan", "sd-wan":
  151. return e.handleIPSDWAN(parts[1:], user), false
  152. default:
  153. return fmt.Sprintf(ColorRed+"Unknown command or syntax error: %q. Type 'help' or '?' for command list.\n"+ColorReset, cmd), false
  154. }
  155. }
  156. func parseArgs(parts []string) ([]string, map[string]string) {
  157. var pos []string
  158. params := make(map[string]string)
  159. for _, p := range parts {
  160. if idx := strings.Index(p, "="); idx != -1 {
  161. k := strings.ToLower(p[:idx])
  162. v := p[idx+1:]
  163. params[k] = v
  164. } else {
  165. pos = append(pos, p)
  166. }
  167. }
  168. return pos, params
  169. }
  170. // ==================== LOGGING ====================
  171. func (e *Executor) handleLog(parts []string) string {
  172. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" || parts[0] == "show" {
  173. limit := 50
  174. filter := ""
  175. args := parts
  176. if len(args) > 0 {
  177. args = args[1:]
  178. }
  179. _, params := parseArgs(args)
  180. if lStr, ok := params["limit"]; ok {
  181. if lVal, err := strconv.Atoi(lStr); err == nil && lVal > 0 {
  182. limit = lVal
  183. }
  184. }
  185. if fStr, ok := params["where"]; ok {
  186. filter = fStr
  187. } else if fStr, ok := params["filter"]; ok {
  188. filter = fStr
  189. }
  190. logger := logging.GetLogger()
  191. entries := logger.List(limit, filter)
  192. if len(entries) == 0 {
  193. return ColorYellow + "No log records found.\n" + ColorReset
  194. }
  195. tbl := NewTable("#", "TIME", "TOPICS", "MESSAGE")
  196. for _, logEntry := range entries {
  197. timeStr := logEntry.Time.Format("Jan/02 15:04:05")
  198. tbl.AddRow(
  199. fmt.Sprintf("%d", logEntry.ID),
  200. timeStr,
  201. logEntry.Topics,
  202. logEntry.Message,
  203. )
  204. }
  205. return tbl.String()
  206. }
  207. return ColorRed + "Usage: /log print [limit=N] [where=filter]\n" + ColorReset
  208. }
  209. // ==================== INTERFACES ====================
  210. func (e *Executor) handleInterface(parts []string, user *auth.PublicUser) string {
  211. if len(parts) == 0 {
  212. return e.printAllInterfaces()
  213. }
  214. sub := strings.ToLower(parts[0])
  215. switch sub {
  216. case "print", "pr", "show", "list":
  217. return e.printAllInterfaces()
  218. case "vlan":
  219. return e.handleInterfaceVLAN(parts[1:], user)
  220. case "bridge":
  221. return e.handleInterfaceBridge(parts[1:], user)
  222. case "wireguard":
  223. return e.handleInterfaceWireGuard(parts[1:], user)
  224. case "aerotun":
  225. return e.handleInterfaceAeroTUN(parts[1:], user)
  226. case "eoip":
  227. return e.handleInterfaceEoIP(parts[1:], user)
  228. case "vxlan":
  229. return e.handleInterfaceVXLAN(parts[1:], user)
  230. case "vrrp":
  231. return e.handleInterfaceVRRP(parts[1:], user)
  232. case "pppoe-client":
  233. return e.handleInterfacePPPoE(parts[1:], user)
  234. case "enable":
  235. if len(parts) < 2 {
  236. return ColorRed + "Usage: /interface enable <name>\n" + ColorReset
  237. }
  238. if user.Role == auth.RoleRead {
  239. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  240. }
  241. name := parts[1]
  242. if err := e.netMgr.SetInterfaceState(name, true); err != nil {
  243. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  244. }
  245. return ColorGreen + fmt.Sprintf("Interface %s enabled successfully.\n", name) + ColorReset
  246. case "disable":
  247. if len(parts) < 2 {
  248. return ColorRed + "Usage: /interface disable <name>\n" + ColorReset
  249. }
  250. if user.Role == auth.RoleRead {
  251. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  252. }
  253. name := parts[1]
  254. if err := e.netMgr.SetInterfaceState(name, false); err != nil {
  255. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  256. }
  257. return ColorGreen + fmt.Sprintf("Interface %s disabled successfully.\n", name) + ColorReset
  258. case "set":
  259. if len(parts) < 2 {
  260. return ColorRed + "Usage: /interface set <name> mtu=<val>\n" + ColorReset
  261. }
  262. if user.Role == auth.RoleRead {
  263. return ColorRed + "Permission denied: viewer role cannot modify interfaces\n" + ColorReset
  264. }
  265. name := parts[1]
  266. _, params := parseArgs(parts[2:])
  267. if mtuStr, ok := params["mtu"]; ok {
  268. mtu, err := strconv.Atoi(mtuStr)
  269. if err != nil || mtu < 68 || mtu > 9000 {
  270. return ColorRed + "Invalid MTU value (68-9000)\n" + ColorReset
  271. }
  272. if err := e.netMgr.SetInterfaceMTU(name, mtu); err != nil {
  273. return ColorRed + "Error setting MTU: " + err.Error() + ColorReset + "\n"
  274. }
  275. return ColorGreen + fmt.Sprintf("Interface %s MTU updated to %d.\n", name, mtu) + ColorReset
  276. }
  277. return ColorRed + "Usage: /interface set <name> mtu=<val>\n" + ColorReset
  278. default:
  279. return ColorRed + "Unknown interface subcommand: " + sub + "\n" + ColorReset
  280. }
  281. }
  282. func (e *Executor) printAllInterfaces() string {
  283. ifaces, err := e.netMgr.ListInterfaces()
  284. if err != nil {
  285. return ColorRed + "Error listing interfaces: " + err.Error() + ColorReset + "\n"
  286. }
  287. tbl := NewTable("FLAGS", "NAME", "TYPE", "STATE", "MTU", "MAC ADDRESS", "IP ADDRESSES")
  288. for _, iface := range ifaces {
  289. isUp := strings.ToUpper(iface.State) == "UP"
  290. flags := ""
  291. if isUp {
  292. flags += "R"
  293. } else {
  294. flags += "X"
  295. }
  296. stateStr := ColorGreen + "UP" + ColorReset
  297. if !isUp {
  298. stateStr = ColorRed + "DOWN" + ColorReset
  299. }
  300. allIPs := append([]string{}, iface.IPv4...)
  301. allIPs = append(allIPs, iface.IPv6...)
  302. ips := strings.Join(allIPs, ", ")
  303. if ips == "" {
  304. ips = "-"
  305. }
  306. tbl.AddRow(flags, iface.Name, string(iface.Type), stateStr, fmt.Sprintf("%d", iface.MTU), iface.MAC, ips)
  307. }
  308. return tbl.String()
  309. }
  310. // --- VLAN ---
  311. func (e *Executor) handleInterfaceVLAN(parts []string, user *auth.PublicUser) string {
  312. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  313. vlans, err := e.netMgr.ListVLANs()
  314. if err != nil {
  315. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  316. }
  317. tbl := NewTable("#", "NAME", "VLAN-ID", "INTERFACE", "MTU", "STATE")
  318. for idx, v := range vlans {
  319. isUp := strings.ToUpper(v.State) == "UP"
  320. st := ColorGreen + "UP" + ColorReset
  321. if !isUp {
  322. st = ColorRed + "DOWN" + ColorReset
  323. }
  324. tbl.AddRow(fmt.Sprintf("%d", idx), v.Name, fmt.Sprintf("%d", v.VlanID), v.ParentName, fmt.Sprintf("%d", v.MTU), st)
  325. }
  326. return tbl.String()
  327. }
  328. sub := strings.ToLower(parts[0])
  329. if user.Role == auth.RoleRead {
  330. return ColorRed + "Permission denied: viewer role cannot modify VLANs\n" + ColorReset
  331. }
  332. switch sub {
  333. case "add":
  334. _, params := parseArgs(parts[1:])
  335. name := params["name"]
  336. parent := params["interface"]
  337. vlanIDStr := params["vlan-id"]
  338. if vlanIDStr == "" {
  339. vlanIDStr = params["vlan_id"]
  340. }
  341. vlanID, err := strconv.Atoi(vlanIDStr)
  342. if err != nil || vlanID < 1 || vlanID > 4094 {
  343. return ColorRed + "Invalid vlan-id (1-4094)\n" + ColorReset
  344. }
  345. if parent == "" {
  346. return ColorRed + "Missing parent interface (interface=...)\n" + ColorReset
  347. }
  348. if name == "" {
  349. name = fmt.Sprintf("%s.%d", parent, vlanID)
  350. }
  351. req := network.CreateVLANRequest{
  352. ParentName: parent,
  353. VlanID: vlanID,
  354. Name: name,
  355. AutoUp: true,
  356. }
  357. if _, err := e.netMgr.CreateVLAN(req); err != nil {
  358. return ColorRed + "Error creating VLAN: " + err.Error() + ColorReset + "\n"
  359. }
  360. return ColorGreen + fmt.Sprintf("VLAN interface %s created.\n", name) + ColorReset
  361. case "remove", "rem", "delete":
  362. pos, _ := parseArgs(parts[1:])
  363. if len(pos) == 0 {
  364. return ColorRed + "Usage: /interface vlan remove <name>\n" + ColorReset
  365. }
  366. name := pos[0]
  367. if err := e.netMgr.DeleteVLAN(name); err != nil {
  368. return ColorRed + "Error deleting VLAN: " + err.Error() + ColorReset + "\n"
  369. }
  370. return ColorGreen + fmt.Sprintf("VLAN %s removed.\n", name) + ColorReset
  371. default:
  372. return ColorRed + "Usage: /interface vlan [print|add|remove]\n" + ColorReset
  373. }
  374. }
  375. // --- Bridge ---
  376. func (e *Executor) handleInterfaceBridge(parts []string, user *auth.PublicUser) string {
  377. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  378. if e.bridgeMgr == nil {
  379. return ColorRed + "Bridge manager not available.\n" + ColorReset
  380. }
  381. bridges := e.bridgeMgr.ListBridges()
  382. tbl := NewTable("FLAGS", "NAME", "MTU", "MAC-ADDRESS", "RUNNING", "PORTS")
  383. for _, b := range bridges {
  384. flags := ""
  385. if b.Disabled {
  386. flags += "X"
  387. } else {
  388. flags += "R"
  389. }
  390. tbl.AddRow(flags, b.Name, fmt.Sprintf("%d", b.MTU), b.MACAddress, fmt.Sprintf("%t", b.Running), fmt.Sprintf("%d", b.PortsCount))
  391. }
  392. return tbl.String()
  393. }
  394. sub := strings.ToLower(parts[0])
  395. if sub == "port" {
  396. return e.handleBridgePort(parts[1:], user)
  397. }
  398. if user.Role == auth.RoleRead {
  399. return ColorRed + "Permission denied: viewer role cannot modify bridge\n" + ColorReset
  400. }
  401. switch sub {
  402. case "add":
  403. _, params := parseArgs(parts[1:])
  404. name := params["name"]
  405. if name == "" {
  406. return ColorRed + "Usage: /interface bridge add name=<name>\n" + ColorReset
  407. }
  408. b := bridge.BridgeInstance{
  409. Name: name,
  410. Disabled: false,
  411. }
  412. if err := e.bridgeMgr.SaveBridge(b); err != nil {
  413. return ColorRed + "Error adding bridge: " + err.Error() + ColorReset + "\n"
  414. }
  415. return ColorGreen + fmt.Sprintf("Bridge %s added.\n", name) + ColorReset
  416. case "remove", "rem", "delete":
  417. pos, _ := parseArgs(parts[1:])
  418. if len(pos) == 0 {
  419. return ColorRed + "Usage: /interface bridge remove <id|name>\n" + ColorReset
  420. }
  421. if err := e.bridgeMgr.DeleteBridge(pos[0]); err != nil {
  422. return ColorRed + "Error deleting bridge: " + err.Error() + ColorReset + "\n"
  423. }
  424. return ColorGreen + fmt.Sprintf("Bridge %s removed.\n", pos[0]) + ColorReset
  425. case "enable":
  426. pos, _ := parseArgs(parts[1:])
  427. if len(pos) == 0 {
  428. return ColorRed + "Usage: /interface bridge enable <id|name>\n" + ColorReset
  429. }
  430. if err := e.bridgeMgr.SetBridgeState(pos[0], false); err != nil {
  431. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  432. }
  433. return ColorGreen + fmt.Sprintf("Bridge %s enabled.\n", pos[0]) + ColorReset
  434. case "disable":
  435. pos, _ := parseArgs(parts[1:])
  436. if len(pos) == 0 {
  437. return ColorRed + "Usage: /interface bridge disable <id|name>\n" + ColorReset
  438. }
  439. if err := e.bridgeMgr.SetBridgeState(pos[0], true); err != nil {
  440. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  441. }
  442. return ColorGreen + fmt.Sprintf("Bridge %s disabled.\n", pos[0]) + ColorReset
  443. default:
  444. return ColorRed + "Usage: /interface bridge [print|add|remove|enable|disable|port]\n" + ColorReset
  445. }
  446. }
  447. func (e *Executor) handleBridgePort(parts []string, user *auth.PublicUser) string {
  448. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  449. ports := e.bridgeMgr.ListPorts()
  450. tbl := NewTable("FLAGS", "INTERFACE", "BRIDGE", "STATUS", "PVID")
  451. for _, p := range ports {
  452. flags := ""
  453. if p.Disabled {
  454. flags += "X"
  455. } else {
  456. flags += "H"
  457. }
  458. tbl.AddRow(flags, p.Interface, p.Bridge, p.Status, fmt.Sprintf("%d", p.PVID))
  459. }
  460. return tbl.String()
  461. }
  462. if user.Role == auth.RoleRead {
  463. return ColorRed + "Permission denied: viewer role cannot modify bridge ports\n" + ColorReset
  464. }
  465. sub := strings.ToLower(parts[0])
  466. switch sub {
  467. case "add":
  468. _, params := parseArgs(parts[1:])
  469. br := params["bridge"]
  470. iface := params["interface"]
  471. if br == "" || iface == "" {
  472. return ColorRed + "Usage: /interface bridge port add bridge=<bridge> interface=<iface>\n" + ColorReset
  473. }
  474. port := bridge.BridgePort{
  475. Bridge: br,
  476. Interface: iface,
  477. Disabled: false,
  478. }
  479. if err := e.bridgeMgr.SavePort(port); err != nil {
  480. return ColorRed + "Error adding bridge port: " + err.Error() + ColorReset + "\n"
  481. }
  482. return ColorGreen + fmt.Sprintf("Port %s attached to bridge %s.\n", iface, br) + ColorReset
  483. case "remove", "rem", "delete":
  484. pos, _ := parseArgs(parts[1:])
  485. if len(pos) == 0 {
  486. return ColorRed + "Usage: /interface bridge port remove <id>\n" + ColorReset
  487. }
  488. if err := e.bridgeMgr.DeletePort(pos[0]); err != nil {
  489. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  490. }
  491. return ColorGreen + fmt.Sprintf("Bridge port %s removed.\n", pos[0]) + ColorReset
  492. default:
  493. return ColorRed + "Usage: /interface bridge port [print|add|remove]\n" + ColorReset
  494. }
  495. }
  496. // --- WireGuard ---
  497. func (e *Executor) handleInterfaceWireGuard(parts []string, user *auth.PublicUser) string {
  498. if e.vpnMgr == nil || e.vpnMgr.WireGuard == nil {
  499. return ColorRed + "WireGuard subsystem not available.\n" + ColorReset
  500. }
  501. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  502. ifaces := e.vpnMgr.WireGuard.ListInterfaces()
  503. tbl := NewTable("FLAGS", "NAME", "LISTEN-PORT", "PUBLIC-KEY", "STATUS")
  504. for _, w := range ifaces {
  505. flags := ""
  506. if w.Disabled {
  507. flags += "X"
  508. } else {
  509. flags += "R"
  510. }
  511. st := "running"
  512. if !w.Running {
  513. st = "stopped"
  514. }
  515. tbl.AddRow(flags, w.Name, fmt.Sprintf("%d", w.ListenPort), w.PublicKey, st)
  516. }
  517. return tbl.String()
  518. }
  519. sub := strings.ToLower(parts[0])
  520. if sub == "peers" || sub == "peer" {
  521. return e.handleWireGuardPeers(parts[1:], user)
  522. }
  523. if user.Role == auth.RoleRead {
  524. return ColorRed + "Permission denied: viewer role cannot modify wireguard\n" + ColorReset
  525. }
  526. switch sub {
  527. case "add":
  528. _, params := parseArgs(parts[1:])
  529. name := params["name"]
  530. portStr := params["listen-port"]
  531. privKey := params["private-key"]
  532. port := 51820
  533. if portStr != "" {
  534. if pVal, err := strconv.Atoi(portStr); err == nil && pVal > 0 {
  535. port = pVal
  536. }
  537. }
  538. if name == "" {
  539. return ColorRed + "Usage: /interface wireguard add name=<name> [listen-port=<port>]\n" + ColorReset
  540. }
  541. req := vpn.SaveWireGuardInterfaceRequest{
  542. WireGuardInterface: vpn.WireGuardInterface{
  543. Name: name,
  544. ListenPort: port,
  545. PrivateKey: privKey,
  546. Disabled: false,
  547. },
  548. }
  549. if err := e.vpnMgr.WireGuard.SaveInterface(req); err != nil {
  550. return ColorRed + "Error adding WireGuard interface: " + err.Error() + ColorReset + "\n"
  551. }
  552. return ColorGreen + fmt.Sprintf("WireGuard interface %s added.\n", name) + ColorReset
  553. case "remove", "rem", "delete":
  554. pos, _ := parseArgs(parts[1:])
  555. if len(pos) == 0 {
  556. return ColorRed + "Usage: /interface wireguard remove <name>\n" + ColorReset
  557. }
  558. if err := e.vpnMgr.WireGuard.DeleteInterface(pos[0]); err != nil {
  559. return ColorRed + "Error deleting WireGuard interface: " + err.Error() + ColorReset + "\n"
  560. }
  561. return ColorGreen + fmt.Sprintf("WireGuard interface %s removed.\n", pos[0]) + ColorReset
  562. case "enable":
  563. pos, _ := parseArgs(parts[1:])
  564. if len(pos) == 0 {
  565. return ColorRed + "Usage: /interface wireguard enable <name>\n" + ColorReset
  566. }
  567. if err := e.vpnMgr.WireGuard.SetInterfaceEnabled(pos[0], true); err != nil {
  568. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  569. }
  570. return ColorGreen + fmt.Sprintf("WireGuard interface %s enabled.\n", pos[0]) + ColorReset
  571. case "disable":
  572. pos, _ := parseArgs(parts[1:])
  573. if len(pos) == 0 {
  574. return ColorRed + "Usage: /interface wireguard disable <name>\n" + ColorReset
  575. }
  576. if err := e.vpnMgr.WireGuard.SetInterfaceEnabled(pos[0], false); err != nil {
  577. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  578. }
  579. return ColorGreen + fmt.Sprintf("WireGuard interface %s disabled.\n", pos[0]) + ColorReset
  580. default:
  581. return ColorRed + "Usage: /interface wireguard [print|add|remove|enable|disable|peers]\n" + ColorReset
  582. }
  583. }
  584. func (e *Executor) handleWireGuardPeers(parts []string, user *auth.PublicUser) string {
  585. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  586. _, params := parseArgs(parts)
  587. peers := e.vpnMgr.WireGuard.ListPeers(params["interface"])
  588. tbl := NewTable("INTERFACE", "PUBLIC-KEY", "ENDPOINT", "ALLOWED-IPS", "LAST-HANDSHAKE")
  589. for _, p := range peers {
  590. hs := "-"
  591. if p.LatestHandshake != "" {
  592. hs = p.LatestHandshake
  593. }
  594. ep := p.EndpointAddress
  595. if p.EndpointPort > 0 {
  596. ep = fmt.Sprintf("%s:%d", p.EndpointAddress, p.EndpointPort)
  597. }
  598. tbl.AddRow(p.Interface, p.PublicKey, ep, strings.Join(p.AllowedIPs, ","), hs)
  599. }
  600. return tbl.String()
  601. }
  602. if user.Role == auth.RoleRead {
  603. return ColorRed + "Permission denied: viewer role cannot modify wireguard peers\n" + ColorReset
  604. }
  605. sub := strings.ToLower(parts[0])
  606. switch sub {
  607. case "add":
  608. _, params := parseArgs(parts[1:])
  609. iface := params["interface"]
  610. pubKey := params["public-key"]
  611. endpoint := params["endpoint"]
  612. allowed := params["allowed-address"]
  613. if allowed == "" {
  614. allowed = params["allowed-ips"]
  615. }
  616. if iface == "" || pubKey == "" {
  617. return ColorRed + "Usage: /interface wireguard peers add interface=<iface> public-key=<key> [endpoint=host:port] [allowed-address=cidr]\n" + ColorReset
  618. }
  619. var allowedList []string
  620. if allowed != "" {
  621. allowedList = strings.Split(allowed, ",")
  622. }
  623. epHost := endpoint
  624. epPort := 0
  625. if strings.Contains(endpoint, ":") {
  626. parts := strings.Split(endpoint, ":")
  627. epHost = parts[0]
  628. if p, err := strconv.Atoi(parts[1]); err == nil {
  629. epPort = p
  630. }
  631. }
  632. req := vpn.SaveWireGuardPeerRequest{
  633. WireGuardPeer: vpn.WireGuardPeer{
  634. Interface: iface,
  635. PublicKey: pubKey,
  636. EndpointAddress: epHost,
  637. EndpointPort: epPort,
  638. AllowedIPs: allowedList,
  639. Disabled: false,
  640. },
  641. }
  642. if err := e.vpnMgr.WireGuard.SavePeer(req); err != nil {
  643. return ColorRed + "Error saving WireGuard peer: " + err.Error() + ColorReset + "\n"
  644. }
  645. return ColorGreen + "WireGuard peer added successfully.\n" + ColorReset
  646. case "remove", "rem", "delete":
  647. _, params := parseArgs(parts[1:])
  648. iface := params["interface"]
  649. pubKey := params["public-key"]
  650. if iface == "" || pubKey == "" {
  651. return ColorRed + "Usage: /interface wireguard peers remove interface=<iface> public-key=<key>\n" + ColorReset
  652. }
  653. if err := e.vpnMgr.WireGuard.DeletePeer(iface, pubKey); err != nil {
  654. return ColorRed + "Error deleting WireGuard peer: " + err.Error() + ColorReset + "\n"
  655. }
  656. return ColorGreen + "WireGuard peer removed successfully.\n" + ColorReset
  657. default:
  658. return ColorRed + "Usage: /interface wireguard peers [print|add|remove]\n" + ColorReset
  659. }
  660. }
  661. // --- EoIP ---
  662. func (e *Executor) handleInterfaceEoIP(parts []string, user *auth.PublicUser) string {
  663. if e.vpnMgr == nil || e.vpnMgr.EoIP == nil {
  664. return ColorRed + "EoIP subsystem not available.\n" + ColorReset
  665. }
  666. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  667. tunnels := e.vpnMgr.EoIP.List()
  668. tbl := NewTable("FLAGS", "NAME", "REMOTE-ADDRESS", "TUNNEL-ID", "STATUS")
  669. for _, t := range tunnels {
  670. flags := ""
  671. if t.Disabled {
  672. flags += "X"
  673. } else {
  674. flags += "R"
  675. }
  676. st := "stopped"
  677. if t.Running {
  678. st = "running"
  679. }
  680. tbl.AddRow(flags, t.Name, t.RemoteAddress, fmt.Sprintf("%d", t.TunnelID), st)
  681. }
  682. return tbl.String()
  683. }
  684. if user.Role == auth.RoleRead {
  685. return ColorRed + "Permission denied: viewer role cannot modify EoIP\n" + ColorReset
  686. }
  687. sub := strings.ToLower(parts[0])
  688. switch sub {
  689. case "add":
  690. _, params := parseArgs(parts[1:])
  691. name := params["name"]
  692. remote := params["remote-address"]
  693. tidStr := params["tunnel-id"]
  694. tid, err := strconv.Atoi(tidStr)
  695. if err != nil || tid < 1 || tid > 65535 {
  696. return ColorRed + "Invalid tunnel-id (1-65535)\n" + ColorReset
  697. }
  698. if name == "" || remote == "" {
  699. return ColorRed + "Usage: /interface eoip add name=<name> remote-address=<ip> tunnel-id=<id>\n" + ColorReset
  700. }
  701. inst := &vpn.EoIPInstance{
  702. Name: name,
  703. RemoteAddress: remote,
  704. TunnelID: tid,
  705. Disabled: false,
  706. }
  707. if err := e.vpnMgr.EoIP.Save(inst); err != nil {
  708. return ColorRed + "Error adding EoIP tunnel: " + err.Error() + ColorReset + "\n"
  709. }
  710. return ColorGreen + fmt.Sprintf("EoIP tunnel %s added.\n", name) + ColorReset
  711. case "remove", "rem", "delete":
  712. pos, _ := parseArgs(parts[1:])
  713. if len(pos) == 0 {
  714. return ColorRed + "Usage: /interface eoip remove <id>\n" + ColorReset
  715. }
  716. if err := e.vpnMgr.EoIP.Delete(pos[0]); err != nil {
  717. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  718. }
  719. return ColorGreen + fmt.Sprintf("EoIP tunnel %s removed.\n", pos[0]) + ColorReset
  720. case "enable":
  721. pos, _ := parseArgs(parts[1:])
  722. if len(pos) == 0 {
  723. return ColorRed + "Usage: /interface eoip enable <id>\n" + ColorReset
  724. }
  725. if err := e.vpnMgr.EoIP.SetState(pos[0], false); err != nil {
  726. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  727. }
  728. return ColorGreen + fmt.Sprintf("EoIP tunnel %s enabled.\n", pos[0]) + ColorReset
  729. case "disable":
  730. pos, _ := parseArgs(parts[1:])
  731. if len(pos) == 0 {
  732. return ColorRed + "Usage: /interface eoip disable <id>\n" + ColorReset
  733. }
  734. if err := e.vpnMgr.EoIP.SetState(pos[0], true); err != nil {
  735. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  736. }
  737. return ColorGreen + fmt.Sprintf("EoIP tunnel %s disabled.\n", pos[0]) + ColorReset
  738. default:
  739. return ColorRed + "Usage: /interface eoip [print|add|remove|enable|disable]\n" + ColorReset
  740. }
  741. }
  742. // --- VXLAN ---
  743. func (e *Executor) handleInterfaceVXLAN(parts []string, user *auth.PublicUser) string {
  744. if e.vpnMgr == nil || e.vpnMgr.VXLAN == nil {
  745. return ColorRed + "VXLAN subsystem not available.\n" + ColorReset
  746. }
  747. if len(parts) > 0 {
  748. sub := strings.ToLower(parts[0])
  749. if sub == "vteps" || sub == "vtep" {
  750. return e.handleInterfaceVXLANVteps(parts[1:], user)
  751. }
  752. }
  753. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  754. tunnels := e.vpnMgr.VXLAN.List()
  755. tbl := NewTable("FLAGS", "NAME", "VNI", "PORT", "MTU", "MAC-ADDRESS", "STATUS")
  756. for _, t := range tunnels {
  757. flags := ""
  758. if t.Disabled {
  759. flags += "X"
  760. }
  761. if t.Running {
  762. flags += "R"
  763. }
  764. st := "stopped"
  765. if !t.Disabled {
  766. if t.Running {
  767. st = "running"
  768. } else {
  769. st = "down"
  770. }
  771. }
  772. tbl.AddRow(flags, t.Name, fmt.Sprintf("%d", t.VNI), fmt.Sprintf("%d", t.Port), fmt.Sprintf("%d", t.MTU), t.MACAddress, st)
  773. }
  774. return tbl.String()
  775. }
  776. if user != nil && user.Role == auth.RoleRead {
  777. return ColorRed + "Permission denied: viewer role cannot modify VXLAN\n" + ColorReset
  778. }
  779. sub := strings.ToLower(parts[0])
  780. switch sub {
  781. case "add":
  782. _, params := parseArgs(parts[1:])
  783. vniStr := params["vni"]
  784. vni, err := strconv.Atoi(vniStr)
  785. if err != nil || vni < 1 || vni > 16777215 {
  786. return ColorRed + "Invalid or missing vni (1-16777215)\n" + ColorReset
  787. }
  788. name := params["name"]
  789. if name == "" {
  790. name = fmt.Sprintf("vxlan%d", vni)
  791. }
  792. port := 4789
  793. if pStr := params["port"]; pStr != "" {
  794. if p, err := strconv.Atoi(pStr); err == nil && p > 0 && p <= 65535 {
  795. port = p
  796. }
  797. }
  798. mtu := 1500
  799. if mStr := params["mtu"]; mStr != "" {
  800. if m, err := strconv.Atoi(mStr); err == nil && m >= 576 {
  801. mtu = m
  802. }
  803. }
  804. learning := true
  805. if lStr := params["learning"]; lStr != "" {
  806. lLower := strings.ToLower(lStr)
  807. if lLower == "no" || lLower == "false" || lLower == "0" {
  808. learning = false
  809. }
  810. }
  811. inst := &vpn.VXLANInstance{
  812. Name: name,
  813. VNI: vni,
  814. Port: port,
  815. MTU: mtu,
  816. MACAddress: params["mac-address"],
  817. IPAddress: params["ip-address"],
  818. LocalAddress: params["local-address"],
  819. RemoteAddress: params["remote-address"],
  820. Group: params["group"],
  821. ParentInterface: params["interface"],
  822. Learning: learning,
  823. Comment: params["comment"],
  824. Disabled: false,
  825. }
  826. if err := e.vpnMgr.VXLAN.Save(inst); err != nil {
  827. return ColorRed + "Error adding VXLAN interface: " + err.Error() + ColorReset + "\n"
  828. }
  829. return ColorGreen + fmt.Sprintf("VXLAN interface %s added.\n", name) + ColorReset
  830. case "set":
  831. pos, params := parseArgs(parts[1:])
  832. if len(pos) == 0 {
  833. return ColorRed + "Usage: /interface vxlan set <name|id> [name=...] [vni=...] [port=...] [mtu=...] [comment=...]\n" + ColorReset
  834. }
  835. inst, found := e.vpnMgr.VXLAN.Get(pos[0])
  836. if !found {
  837. return ColorRed + fmt.Sprintf("VXLAN interface %q not found\n", pos[0]) + ColorReset
  838. }
  839. if newName := params["name"]; newName != "" {
  840. inst.Name = newName
  841. }
  842. if vniStr := params["vni"]; vniStr != "" {
  843. if v, err := strconv.Atoi(vniStr); err == nil && v >= 1 && v <= 16777215 {
  844. inst.VNI = v
  845. }
  846. }
  847. if pStr := params["port"]; pStr != "" {
  848. if p, err := strconv.Atoi(pStr); err == nil && p > 0 {
  849. inst.Port = p
  850. }
  851. }
  852. if mStr := params["mtu"]; mStr != "" {
  853. if m, err := strconv.Atoi(mStr); err == nil && m >= 576 {
  854. inst.MTU = m
  855. }
  856. }
  857. if mac := params["mac-address"]; mac != "" {
  858. inst.MACAddress = mac
  859. }
  860. if ip := params["ip-address"]; ip != "" {
  861. inst.IPAddress = ip
  862. }
  863. if loc := params["local-address"]; loc != "" {
  864. inst.LocalAddress = loc
  865. }
  866. if rem := params["remote-address"]; rem != "" {
  867. inst.RemoteAddress = rem
  868. }
  869. if grp := params["group"]; grp != "" {
  870. inst.Group = grp
  871. }
  872. if iface := params["interface"]; iface != "" {
  873. inst.ParentInterface = iface
  874. }
  875. if cmt, ok := params["comment"]; ok {
  876. inst.Comment = cmt
  877. }
  878. if err := e.vpnMgr.VXLAN.Save(inst); err != nil {
  879. return ColorRed + "Error updating VXLAN interface: " + err.Error() + ColorReset + "\n"
  880. }
  881. return ColorGreen + fmt.Sprintf("VXLAN interface %s updated.\n", inst.Name) + ColorReset
  882. case "remove", "rem", "delete":
  883. pos, _ := parseArgs(parts[1:])
  884. if len(pos) == 0 {
  885. return ColorRed + "Usage: /interface vxlan remove <name|id>\n" + ColorReset
  886. }
  887. if err := e.vpnMgr.VXLAN.Delete(pos[0]); err != nil {
  888. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  889. }
  890. return ColorGreen + fmt.Sprintf("VXLAN interface %s removed.\n", pos[0]) + ColorReset
  891. case "enable":
  892. pos, _ := parseArgs(parts[1:])
  893. if len(pos) == 0 {
  894. return ColorRed + "Usage: /interface vxlan enable <name|id>\n" + ColorReset
  895. }
  896. if err := e.vpnMgr.VXLAN.SetState(pos[0], false); err != nil {
  897. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  898. }
  899. return ColorGreen + fmt.Sprintf("VXLAN interface %s enabled.\n", pos[0]) + ColorReset
  900. case "disable":
  901. pos, _ := parseArgs(parts[1:])
  902. if len(pos) == 0 {
  903. return ColorRed + "Usage: /interface vxlan disable <name|id>\n" + ColorReset
  904. }
  905. if err := e.vpnMgr.VXLAN.SetState(pos[0], true); err != nil {
  906. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  907. }
  908. return ColorGreen + fmt.Sprintf("VXLAN interface %s disabled.\n", pos[0]) + ColorReset
  909. default:
  910. return ColorRed + "Usage: /interface vxlan [print|add|set|remove|enable|disable|vteps]\n" + ColorReset
  911. }
  912. }
  913. // --- VXLAN VTEPs ---
  914. func (e *Executor) handleInterfaceVXLANVteps(parts []string, user *auth.PublicUser) string {
  915. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  916. _, params := parseArgs(parts)
  917. vteps := e.vpnMgr.VXLAN.ListVteps(params["interface"])
  918. tbl := NewTable("INTERFACE", "REMOTE-IP", "PORT", "COMMENT")
  919. for _, v := range vteps {
  920. pStr := fmt.Sprintf("%d", v.Port)
  921. if v.Port <= 0 {
  922. pStr = "default"
  923. }
  924. tbl.AddRow(v.Interface, v.RemoteIP, pStr, v.Comment)
  925. }
  926. return tbl.String()
  927. }
  928. if user != nil && user.Role == auth.RoleRead {
  929. return ColorRed + "Permission denied: viewer role cannot modify VXLAN VTEPs\n" + ColorReset
  930. }
  931. sub := strings.ToLower(parts[0])
  932. switch sub {
  933. case "add":
  934. _, params := parseArgs(parts[1:])
  935. iface := params["interface"]
  936. remote := params["remote-ip"]
  937. if remote == "" {
  938. remote = params["remote-address"]
  939. }
  940. if iface == "" || remote == "" {
  941. return ColorRed + "Usage: /interface vxlan vteps add interface=<iface> remote-ip=<ip> [port=4789] [comment=...]\n" + ColorReset
  942. }
  943. port := 0
  944. if pStr := params["port"]; pStr != "" {
  945. if p, err := strconv.Atoi(pStr); err == nil && p > 0 {
  946. port = p
  947. }
  948. }
  949. vtep := &vpn.VXLANVtep{
  950. Interface: iface,
  951. RemoteIP: remote,
  952. Port: port,
  953. Comment: params["comment"],
  954. }
  955. if err := e.vpnMgr.VXLAN.AddVtep(iface, vtep); err != nil {
  956. return ColorRed + "Error adding VTEP: " + err.Error() + ColorReset + "\n"
  957. }
  958. return ColorGreen + fmt.Sprintf("VTEP %s added to interface %s.\n", remote, iface) + ColorReset
  959. case "remove", "rem", "delete":
  960. pos, params := parseArgs(parts[1:])
  961. iface := params["interface"]
  962. remote := params["remote-ip"]
  963. if remote == "" {
  964. remote = params["remote-address"]
  965. }
  966. if iface != "" && remote != "" {
  967. if err := e.vpnMgr.VXLAN.DeleteVtep(iface, remote); err != nil {
  968. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  969. }
  970. return ColorGreen + fmt.Sprintf("VTEP %s removed from %s.\n", remote, iface) + ColorReset
  971. }
  972. if len(pos) >= 2 {
  973. if err := e.vpnMgr.VXLAN.DeleteVtep(pos[0], pos[1]); err != nil {
  974. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  975. }
  976. return ColorGreen + fmt.Sprintf("VTEP %s removed from %s.\n", pos[1], pos[0]) + ColorReset
  977. }
  978. return ColorRed + "Usage: /interface vxlan vteps remove interface=<iface> remote-ip=<ip> OR <iface> <remote-ip>\n" + ColorReset
  979. default:
  980. return ColorRed + "Usage: /interface vxlan vteps [print|add|remove]\n" + ColorReset
  981. }
  982. }
  983. // --- VRRP ---
  984. func (e *Executor) handleInterfaceVRRP(parts []string, user *auth.PublicUser) string {
  985. if e.vrrpMgr == nil {
  986. return ColorRed + "VRRP subsystem not available.\n" + ColorReset
  987. }
  988. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  989. instances := e.vrrpMgr.List()
  990. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "VRID", "PRIORITY", "STATE", "VIP")
  991. for _, inst := range instances {
  992. flags := ""
  993. if inst.Disabled {
  994. flags += "X"
  995. } else {
  996. flags += "R"
  997. }
  998. tbl.AddRow(flags, inst.Name, inst.Interface, fmt.Sprintf("%d", inst.VRID), fmt.Sprintf("%d", inst.Priority), inst.State, strings.Join(inst.VirtualIPs, ","))
  999. }
  1000. return tbl.String()
  1001. }
  1002. if user.Role == auth.RoleRead {
  1003. return ColorRed + "Permission denied: viewer role cannot modify VRRP\n" + ColorReset
  1004. }
  1005. sub := strings.ToLower(parts[0])
  1006. switch sub {
  1007. case "add":
  1008. _, params := parseArgs(parts[1:])
  1009. name := params["name"]
  1010. iface := params["interface"]
  1011. vridStr := params["vrid"]
  1012. prioStr := params["priority"]
  1013. vip := params["vip"]
  1014. if vip == "" {
  1015. vip = params["virtual-ip"]
  1016. }
  1017. vrid, err := strconv.Atoi(vridStr)
  1018. if err != nil || vrid < 1 || vrid > 255 {
  1019. return ColorRed + "Invalid vrid (1-255)\n" + ColorReset
  1020. }
  1021. prio := 100
  1022. if prioStr != "" {
  1023. if pVal, err := strconv.Atoi(prioStr); err == nil && pVal >= 1 && pVal <= 254 {
  1024. prio = pVal
  1025. }
  1026. }
  1027. if name == "" || iface == "" || vip == "" {
  1028. return ColorRed + "Usage: /interface vrrp add name=<name> interface=<iface> vrid=<id> vip=<ip/cidr> [priority=100]\n" + ColorReset
  1029. }
  1030. inst := vrrp.VRRPInstance{
  1031. Name: name,
  1032. Interface: iface,
  1033. VRID: vrid,
  1034. Priority: prio,
  1035. VirtualIPs: []string{vip},
  1036. Disabled: false,
  1037. }
  1038. if _, err := e.vrrpMgr.Save(inst); err != nil {
  1039. return ColorRed + "Error adding VRRP: " + err.Error() + ColorReset + "\n"
  1040. }
  1041. return ColorGreen + fmt.Sprintf("VRRP instance %s added.\n", name) + ColorReset
  1042. case "remove", "rem", "delete":
  1043. pos, _ := parseArgs(parts[1:])
  1044. if len(pos) == 0 {
  1045. return ColorRed + "Usage: /interface vrrp remove <id>\n" + ColorReset
  1046. }
  1047. if err := e.vrrpMgr.Delete(pos[0]); err != nil {
  1048. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1049. }
  1050. return ColorGreen + fmt.Sprintf("VRRP instance %s removed.\n", pos[0]) + ColorReset
  1051. case "enable":
  1052. pos, _ := parseArgs(parts[1:])
  1053. if len(pos) == 0 {
  1054. return ColorRed + "Usage: /interface vrrp enable <id>\n" + ColorReset
  1055. }
  1056. if err := e.vrrpMgr.ToggleState(pos[0], false); err != nil {
  1057. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1058. }
  1059. return ColorGreen + fmt.Sprintf("VRRP instance %s enabled.\n", pos[0]) + ColorReset
  1060. case "disable":
  1061. pos, _ := parseArgs(parts[1:])
  1062. if len(pos) == 0 {
  1063. return ColorRed + "Usage: /interface vrrp disable <id>\n" + ColorReset
  1064. }
  1065. if err := e.vrrpMgr.ToggleState(pos[0], true); err != nil {
  1066. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1067. }
  1068. return ColorGreen + fmt.Sprintf("VRRP instance %s disabled.\n", pos[0]) + ColorReset
  1069. default:
  1070. return ColorRed + "Usage: /interface vrrp [print|add|remove|enable|disable]\n" + ColorReset
  1071. }
  1072. }
  1073. // --- PPPoE Client ---
  1074. func (e *Executor) handleInterfacePPPoE(parts []string, user *auth.PublicUser) string {
  1075. if e.vpnMgr == nil || e.vpnMgr.PPPoE == nil {
  1076. return ColorRed + "PPPoE subsystem not available.\n" + ColorReset
  1077. }
  1078. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1079. clients := e.vpnMgr.PPPoE.List()
  1080. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "USER", "STATUS", "IP")
  1081. for _, c := range clients {
  1082. flags := ""
  1083. if c.Disabled {
  1084. flags += "X"
  1085. } else {
  1086. flags += "R"
  1087. }
  1088. st := "stopped"
  1089. if c.Running {
  1090. st = "running"
  1091. }
  1092. tbl.AddRow(flags, c.Name, c.Interface, c.Username, st, c.AssignedIP)
  1093. }
  1094. return tbl.String()
  1095. }
  1096. if user.Role == auth.RoleRead {
  1097. return ColorRed + "Permission denied: viewer role cannot modify PPPoE\n" + ColorReset
  1098. }
  1099. sub := strings.ToLower(parts[0])
  1100. switch sub {
  1101. case "add":
  1102. _, params := parseArgs(parts[1:])
  1103. name := params["name"]
  1104. iface := params["interface"]
  1105. usr := params["user"]
  1106. pwd := params["password"]
  1107. if name == "" || iface == "" || usr == "" {
  1108. return ColorRed + "Usage: /interface pppoe-client add name=<name> interface=<iface> user=<usr> password=<pwd>\n" + ColorReset
  1109. }
  1110. inst := vpn.PPPoEInstance{
  1111. Name: name,
  1112. Interface: iface,
  1113. Username: usr,
  1114. Password: pwd,
  1115. Disabled: false,
  1116. }
  1117. if err := e.vpnMgr.PPPoE.Save(inst); err != nil {
  1118. return ColorRed + "Error adding PPPoE client: " + err.Error() + ColorReset + "\n"
  1119. }
  1120. return ColorGreen + fmt.Sprintf("PPPoE client %s added.\n", name) + ColorReset
  1121. case "remove", "rem", "delete":
  1122. pos, _ := parseArgs(parts[1:])
  1123. if len(pos) == 0 {
  1124. return ColorRed + "Usage: /interface pppoe-client remove <name>\n" + ColorReset
  1125. }
  1126. if err := e.vpnMgr.PPPoE.Delete(pos[0]); err != nil {
  1127. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1128. }
  1129. return ColorGreen + fmt.Sprintf("PPPoE client %s removed.\n", pos[0]) + ColorReset
  1130. case "connect":
  1131. pos, _ := parseArgs(parts[1:])
  1132. if len(pos) == 0 {
  1133. return ColorRed + "Usage: /interface pppoe-client connect <name>\n" + ColorReset
  1134. }
  1135. if err := e.vpnMgr.PPPoE.Connect(pos[0]); err != nil {
  1136. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1137. }
  1138. return ColorGreen + fmt.Sprintf("Connecting PPPoE client %s...\n", pos[0]) + ColorReset
  1139. case "disconnect":
  1140. pos, _ := parseArgs(parts[1:])
  1141. if len(pos) == 0 {
  1142. return ColorRed + "Usage: /interface pppoe-client disconnect <name>\n" + ColorReset
  1143. }
  1144. if err := e.vpnMgr.PPPoE.Disconnect(pos[0]); err != nil {
  1145. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1146. }
  1147. return ColorGreen + fmt.Sprintf("Disconnected PPPoE client %s.\n", pos[0]) + ColorReset
  1148. default:
  1149. return ColorRed + "Usage: /interface pppoe-client [print|add|remove|connect|disconnect]\n" + ColorReset
  1150. }
  1151. }
  1152. // ==================== IP SUBSYSTEM ====================
  1153. func (e *Executor) handleIP(parts []string, user *auth.PublicUser) string {
  1154. if len(parts) == 0 {
  1155. return ColorYellow + "IP subsystem. Available: address, route, dhcp-client, dhcp-server, dns, firewall, service\n" + ColorReset
  1156. }
  1157. sub := strings.ToLower(parts[0])
  1158. switch sub {
  1159. case "address", "addr":
  1160. return e.handleIPAddress(parts[1:], user)
  1161. case "route":
  1162. return e.handleIPRoute(parts[1:], user)
  1163. case "dhcp-client":
  1164. return e.handleIPDHCPClient(parts[1:], user)
  1165. case "dhcp-server":
  1166. return e.handleIPDHCPServer(parts[1:], user)
  1167. case "dns":
  1168. return e.handleIPDNS(parts[1:], user)
  1169. case "firewall":
  1170. return e.handleIPFirewall(parts[1:], user)
  1171. case "service", "services":
  1172. return e.handleIPService(parts[1:], user)
  1173. case "reverse-proxy", "reverseproxy", "proxy":
  1174. return e.handleIPReverseProxy(parts[1:], user)
  1175. case "sdwan", "sd-wan":
  1176. return e.handleIPSDWAN(parts[1:], user)
  1177. default:
  1178. return ColorRed + "Unknown IP command: " + sub + "\n" + ColorReset
  1179. }
  1180. }
  1181. // --- IP Address ---
  1182. func (e *Executor) handleIPAddress(parts []string, user *auth.PublicUser) string {
  1183. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1184. ifaces, err := e.netMgr.ListInterfaces()
  1185. if err != nil {
  1186. return ColorRed + "Error listing IP addresses: " + err.Error() + ColorReset + "\n"
  1187. }
  1188. tbl := NewTable("#", "ADDRESS", "NETWORK", "INTERFACE", "STATUS")
  1189. idx := 0
  1190. for _, iface := range ifaces {
  1191. for _, cidr := range append(iface.IPv4, iface.IPv6...) {
  1192. ip, ipNet, parseErr := net.ParseCIDR(cidr)
  1193. netStr := "-"
  1194. if parseErr == nil && ipNet != nil {
  1195. netStr = ipNet.IP.String()
  1196. } else if ip != nil {
  1197. netStr = ip.String()
  1198. }
  1199. tbl.AddRow(fmt.Sprintf("%d", idx), cidr, netStr, iface.Name, "enabled")
  1200. idx++
  1201. }
  1202. }
  1203. return tbl.String()
  1204. }
  1205. if user.Role == auth.RoleRead {
  1206. return ColorRed + "Permission denied: viewer role cannot modify IP addresses\n" + ColorReset
  1207. }
  1208. sub := strings.ToLower(parts[0])
  1209. switch sub {
  1210. case "add":
  1211. _, params := parseArgs(parts[1:])
  1212. addr := params["address"]
  1213. iface := params["interface"]
  1214. if addr == "" || iface == "" {
  1215. return ColorRed + "Usage: /ip address add address=<ip/mask> interface=<iface>\n" + ColorReset
  1216. }
  1217. if err := e.netMgr.AddInterfaceIP(iface, addr); err != nil {
  1218. return ColorRed + "Error adding IP address: " + err.Error() + ColorReset + "\n"
  1219. }
  1220. return ColorGreen + fmt.Sprintf("IP address %s added to interface %s.\n", addr, iface) + ColorReset
  1221. case "remove", "rem", "delete":
  1222. _, params := parseArgs(parts[1:])
  1223. addr := params["address"]
  1224. iface := params["interface"]
  1225. if addr == "" || iface == "" {
  1226. return ColorRed + "Usage: /ip address remove address=<ip/mask> interface=<iface>\n" + ColorReset
  1227. }
  1228. if err := e.netMgr.DeleteInterfaceIP(iface, addr); err != nil {
  1229. return ColorRed + "Error removing IP address: " + err.Error() + ColorReset + "\n"
  1230. }
  1231. return ColorGreen + fmt.Sprintf("IP address %s removed from interface %s.\n", addr, iface) + ColorReset
  1232. default:
  1233. return ColorRed + "Usage: /ip address [print|add|remove]\n" + ColorReset
  1234. }
  1235. }
  1236. // --- IP Route ---
  1237. func (e *Executor) handleIPRoute(parts []string, user *auth.PublicUser) string {
  1238. if e.routingMgr == nil {
  1239. return ColorRed + "Routing manager not available\n" + ColorReset
  1240. }
  1241. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1242. routes := e.routingMgr.ListRoutes()
  1243. tbl := NewTable("FLAGS", "DST-ADDRESS", "GATEWAY", "DISTANCE", "TABLE", "STATUS")
  1244. for _, r := range routes {
  1245. flags := ""
  1246. if r.Disabled {
  1247. flags += "X"
  1248. } else {
  1249. flags += "A"
  1250. }
  1251. if r.Static {
  1252. flags += "S"
  1253. } else {
  1254. flags += "D"
  1255. }
  1256. st := "active"
  1257. if r.Disabled {
  1258. st = "disabled"
  1259. }
  1260. tbl.AddRow(flags, r.DstAddress, r.Gateway, fmt.Sprintf("%d", r.Distance), r.RoutingTable, st)
  1261. }
  1262. return tbl.String()
  1263. }
  1264. sub := strings.ToLower(parts[0])
  1265. switch sub {
  1266. case "rule":
  1267. rules := e.routingMgr.ListPolicyRules()
  1268. tbl := NewTable("#", "SRC", "DST", "TABLE", "PRIORITY")
  1269. for idx, r := range rules {
  1270. tbl.AddRow(fmt.Sprintf("%d", idx), r.SrcAddress, r.DstAddress, r.Table, fmt.Sprintf("%d", r.Priority))
  1271. }
  1272. return tbl.String()
  1273. case "table":
  1274. tables := e.routingMgr.ListTables()
  1275. tbl := NewTable("#", "NAME", "TABLE-ID", "COMMENT")
  1276. for idx, t := range tables {
  1277. tbl.AddRow(fmt.Sprintf("%d", idx), t.Name, fmt.Sprintf("%d", t.TableID), t.Comment)
  1278. }
  1279. return tbl.String()
  1280. }
  1281. if user.Role == auth.RoleRead {
  1282. return ColorRed + "Permission denied: viewer role cannot modify routes\n" + ColorReset
  1283. }
  1284. switch sub {
  1285. case "add":
  1286. _, params := parseArgs(parts[1:])
  1287. dst := params["dst-address"]
  1288. if dst == "" {
  1289. dst = params["dst"]
  1290. }
  1291. gw := params["gateway"]
  1292. iface := params["interface"]
  1293. if gw == "" && iface != "" {
  1294. gw = iface
  1295. }
  1296. if gw != "" && iface == "" && net.ParseIP(gw) == nil {
  1297. iface = gw
  1298. }
  1299. table := params["routing-table"]
  1300. if table == "" {
  1301. table = "main"
  1302. }
  1303. dist := 1
  1304. if dStr, ok := params["distance"]; ok {
  1305. if dVal, err := strconv.Atoi(dStr); err == nil && dVal >= 1 {
  1306. dist = dVal
  1307. }
  1308. }
  1309. if dst == "" || (gw == "" && iface == "") {
  1310. return ColorRed + "Usage: /ip route add dst-address=<cidr> gateway=<ip|interface> [distance=1] [routing-table=main]\n" + ColorReset
  1311. }
  1312. r := routing.RouteItem{
  1313. DstAddress: dst,
  1314. Gateway: gw,
  1315. Interface: iface,
  1316. Distance: dist,
  1317. RoutingTable: table,
  1318. Static: true,
  1319. Disabled: false,
  1320. }
  1321. if err := e.routingMgr.SaveRoute(r); err != nil {
  1322. return ColorRed + "Error adding route: " + err.Error() + ColorReset + "\n"
  1323. }
  1324. return ColorGreen + fmt.Sprintf("Route to %s via %s added.\n", dst, gw) + ColorReset
  1325. case "remove", "rem", "delete":
  1326. pos, _ := parseArgs(parts[1:])
  1327. if len(pos) == 0 {
  1328. return ColorRed + "Usage: /ip route remove <id>\n" + ColorReset
  1329. }
  1330. if err := e.routingMgr.DeleteRoute(pos[0]); err != nil {
  1331. return ColorRed + "Error removing route: " + err.Error() + ColorReset + "\n"
  1332. }
  1333. return ColorGreen + fmt.Sprintf("Route %s removed.\n", pos[0]) + ColorReset
  1334. case "enable":
  1335. pos, _ := parseArgs(parts[1:])
  1336. if len(pos) == 0 {
  1337. return ColorRed + "Usage: /ip route enable <id>\n" + ColorReset
  1338. }
  1339. if err := e.routingMgr.SetRouteState(pos[0], false); err != nil {
  1340. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1341. }
  1342. return ColorGreen + fmt.Sprintf("Route %s enabled.\n", pos[0]) + ColorReset
  1343. case "disable":
  1344. pos, _ := parseArgs(parts[1:])
  1345. if len(pos) == 0 {
  1346. return ColorRed + "Usage: /ip route disable <id>\n" + ColorReset
  1347. }
  1348. if err := e.routingMgr.SetRouteState(pos[0], true); err != nil {
  1349. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1350. }
  1351. return ColorGreen + fmt.Sprintf("Route %s disabled.\n", pos[0]) + ColorReset
  1352. default:
  1353. return ColorRed + "Usage: /ip route [print|add|remove|enable|disable|rule|table]\n" + ColorReset
  1354. }
  1355. }
  1356. // --- IP DHCP Client ---
  1357. func (e *Executor) handleIPDHCPClient(parts []string, user *auth.PublicUser) string {
  1358. if e.dhcpMgr == nil {
  1359. return ColorRed + "DHCP client manager not available.\n" + ColorReset
  1360. }
  1361. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1362. clients := e.dhcpMgr.ListClients()
  1363. tbl := NewTable("FLAGS", "INTERFACE", "USE-PEER-DNS", "STATUS", "ADDRESS", "GATEWAY")
  1364. for _, c := range clients {
  1365. flags := ""
  1366. if !c.Enabled {
  1367. flags += "X"
  1368. } else {
  1369. flags += "R"
  1370. }
  1371. addr := c.IPAddress
  1372. if addr == "" {
  1373. addr = "-"
  1374. }
  1375. gw := c.Gateway
  1376. if gw == "" {
  1377. gw = "-"
  1378. }
  1379. tbl.AddRow(flags, c.Interface, fmt.Sprintf("%t", c.UsePeerDNS), c.Status, addr, gw)
  1380. }
  1381. return tbl.String()
  1382. }
  1383. if user.Role == auth.RoleRead {
  1384. return ColorRed + "Permission denied: viewer role cannot modify DHCP clients\n" + ColorReset
  1385. }
  1386. sub := strings.ToLower(parts[0])
  1387. switch sub {
  1388. case "add":
  1389. _, params := parseArgs(parts[1:])
  1390. iface := params["interface"]
  1391. if iface == "" {
  1392. return ColorRed + "Usage: /ip dhcp-client add interface=<iface> [use-peer-dns=yes]\n" + ColorReset
  1393. }
  1394. peerDNS := true
  1395. if val, ok := params["use-peer-dns"]; ok {
  1396. peerDNS = (strings.ToLower(val) == "yes" || strings.ToLower(val) == "true")
  1397. }
  1398. req := dhcp.CreateDHCPClientRequest{
  1399. Interface: iface,
  1400. UsePeerDNS: peerDNS,
  1401. Enabled: true,
  1402. }
  1403. if _, err := e.dhcpMgr.AddClient(req); err != nil {
  1404. return ColorRed + "Error adding DHCP client: " + err.Error() + ColorReset + "\n"
  1405. }
  1406. return ColorGreen + fmt.Sprintf("DHCP client added on interface %s.\n", iface) + ColorReset
  1407. case "remove", "rem", "delete":
  1408. pos, _ := parseArgs(parts[1:])
  1409. if len(pos) == 0 {
  1410. return ColorRed + "Usage: /ip dhcp-client remove <id>\n" + ColorReset
  1411. }
  1412. if err := e.dhcpMgr.DeleteClient(pos[0]); err != nil {
  1413. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1414. }
  1415. return ColorGreen + fmt.Sprintf("DHCP client %s removed.\n", pos[0]) + ColorReset
  1416. case "renew":
  1417. pos, _ := parseArgs(parts[1:])
  1418. if len(pos) == 0 {
  1419. return ColorRed + "Usage: /ip dhcp-client renew <id>\n" + ColorReset
  1420. }
  1421. if err := e.dhcpMgr.RenewLease(pos[0]); err != nil {
  1422. return ColorRed + "Error renewing lease: " + err.Error() + ColorReset + "\n"
  1423. }
  1424. return ColorGreen + fmt.Sprintf("DHCP lease renewal requested for %s.\n", pos[0]) + ColorReset
  1425. case "release":
  1426. pos, _ := parseArgs(parts[1:])
  1427. if len(pos) == 0 {
  1428. return ColorRed + "Usage: /ip dhcp-client release <id>\n" + ColorReset
  1429. }
  1430. if err := e.dhcpMgr.ReleaseLease(pos[0]); err != nil {
  1431. return ColorRed + "Error releasing lease: " + err.Error() + ColorReset + "\n"
  1432. }
  1433. return ColorGreen + fmt.Sprintf("DHCP lease released for %s.\n", pos[0]) + ColorReset
  1434. default:
  1435. return ColorRed + "Usage: /ip dhcp-client [print|add|remove|renew|release]\n" + ColorReset
  1436. }
  1437. }
  1438. // --- IP DHCP Server ---
  1439. func (e *Executor) handleIPDHCPServer(parts []string, user *auth.PublicUser) string {
  1440. if e.dhcpServMgr == nil {
  1441. return ColorRed + "DHCP server manager not available.\n" + ColorReset
  1442. }
  1443. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1444. servers := e.dhcpServMgr.ListServers()
  1445. tbl := NewTable("FLAGS", "NAME", "INTERFACE", "ADDRESS-POOL", "LEASE-TIME")
  1446. for _, s := range servers {
  1447. flags := ""
  1448. if s.Disabled {
  1449. flags += "X"
  1450. } else {
  1451. flags += "R"
  1452. }
  1453. tbl.AddRow(flags, s.Name, s.Interface, s.AddressPool, s.LeaseTime)
  1454. }
  1455. return tbl.String()
  1456. }
  1457. sub := strings.ToLower(parts[0])
  1458. switch sub {
  1459. case "lease":
  1460. leases := e.dhcpServMgr.ListLeases()
  1461. tbl := NewTable("IP-ADDRESS", "MAC-ADDRESS", "HOSTNAME", "EXPIRES-IN", "STATUS")
  1462. for _, l := range leases {
  1463. tbl.AddRow(l.Address, l.MACAddress, l.HostName, l.ExpiresIn, l.Status)
  1464. }
  1465. return tbl.String()
  1466. case "network":
  1467. networks := e.dhcpServMgr.ListNetworks()
  1468. tbl := NewTable("SUBNET", "GATEWAY", "DNS-SERVERS", "DOMAIN")
  1469. for _, n := range networks {
  1470. tbl.AddRow(n.Address, n.Gateway, strings.Join(n.DNSServers, ","), n.Domain)
  1471. }
  1472. return tbl.String()
  1473. }
  1474. if user.Role == auth.RoleRead {
  1475. return ColorRed + "Permission denied: viewer role cannot modify DHCP server\n" + ColorReset
  1476. }
  1477. switch sub {
  1478. case "add":
  1479. _, params := parseArgs(parts[1:])
  1480. name := params["name"]
  1481. iface := params["interface"]
  1482. pool := params["address-pool"]
  1483. if pool == "" {
  1484. start := params["range-start"]
  1485. end := params["range-end"]
  1486. if start != "" && end != "" {
  1487. pool = fmt.Sprintf("%s-%s", start, end)
  1488. }
  1489. }
  1490. leaseTime := params["lease-time"]
  1491. if leaseTime == "" {
  1492. leaseTime = "12h"
  1493. }
  1494. if name == "" || iface == "" || pool == "" {
  1495. return ColorRed + "Usage: /ip dhcp-server add name=<name> interface=<iface> address-pool=<start-end> [lease-time=12h]\n" + ColorReset
  1496. }
  1497. inst := dhcp.DHCPServerInstance{
  1498. Name: name,
  1499. Interface: iface,
  1500. AddressPool: pool,
  1501. LeaseTime: leaseTime,
  1502. Disabled: false,
  1503. }
  1504. if _, err := e.dhcpServMgr.SaveServer(inst); err != nil {
  1505. return ColorRed + "Error adding DHCP server: " + err.Error() + ColorReset + "\n"
  1506. }
  1507. return ColorGreen + fmt.Sprintf("DHCP server %s added on interface %s.\n", name, iface) + ColorReset
  1508. case "remove", "rem", "delete":
  1509. pos, _ := parseArgs(parts[1:])
  1510. if len(pos) == 0 {
  1511. return ColorRed + "Usage: /ip dhcp-server remove <id>\n" + ColorReset
  1512. }
  1513. if err := e.dhcpServMgr.DeleteServer(pos[0]); err != nil {
  1514. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1515. }
  1516. return ColorGreen + fmt.Sprintf("DHCP server %s removed.\n", pos[0]) + ColorReset
  1517. case "enable":
  1518. pos, _ := parseArgs(parts[1:])
  1519. if len(pos) == 0 {
  1520. return ColorRed + "Usage: /ip dhcp-server enable <id>\n" + ColorReset
  1521. }
  1522. if err := e.dhcpServMgr.ToggleServerState(pos[0], false); err != nil {
  1523. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1524. }
  1525. return ColorGreen + fmt.Sprintf("DHCP server %s enabled.\n", pos[0]) + ColorReset
  1526. case "disable":
  1527. pos, _ := parseArgs(parts[1:])
  1528. if len(pos) == 0 {
  1529. return ColorRed + "Usage: /ip dhcp-server disable <id>\n" + ColorReset
  1530. }
  1531. if err := e.dhcpServMgr.ToggleServerState(pos[0], true); err != nil {
  1532. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1533. }
  1534. return ColorGreen + fmt.Sprintf("DHCP server %s disabled.\n", pos[0]) + ColorReset
  1535. default:
  1536. return ColorRed + "Usage: /ip dhcp-server [print|add|remove|enable|disable|lease|network]\n" + ColorReset
  1537. }
  1538. }
  1539. // --- IP DNS ---
  1540. func (e *Executor) handleIPDNS(parts []string, user *auth.PublicUser) string {
  1541. if e.dnsMgr == nil {
  1542. return ColorRed + "DNS manager not available.\n" + ColorReset
  1543. }
  1544. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1545. settings := e.dnsMgr.GetSettings()
  1546. out := " servers: " + strings.Join(settings.Servers, ", ") + "\n"
  1547. out += " dynamic-servers: " + strings.Join(settings.DynamicServers, ", ") + "\n"
  1548. out += fmt.Sprintf(" allow-remote-requests: %t\n", settings.AllowRemoteRequests)
  1549. out += fmt.Sprintf(" cache-size: %d entries\n", settings.CacheSize)
  1550. out += fmt.Sprintf(" domain: %s\n", settings.Domain)
  1551. return out
  1552. }
  1553. sub := strings.ToLower(parts[0])
  1554. switch sub {
  1555. case "cache":
  1556. if len(parts) > 1 && parts[1] == "flush" {
  1557. if user.Role == auth.RoleRead {
  1558. return ColorRed + "Permission denied\n" + ColorReset
  1559. }
  1560. if err := e.dnsMgr.FlushCache(); err != nil {
  1561. return ColorRed + "Error flushing DNS cache: " + err.Error() + ColorReset + "\n"
  1562. }
  1563. return ColorGreen + "DNS cache flushed.\n" + ColorReset
  1564. }
  1565. return ColorYellow + "Usage: /ip dns cache flush\n" + ColorReset
  1566. case "static":
  1567. records := e.dnsMgr.ListStaticRecords()
  1568. tbl := NewTable("#", "NAME", "ADDRESS", "TTL", "STATUS")
  1569. for idx, r := range records {
  1570. st := "enabled"
  1571. if r.Disabled {
  1572. st = "disabled"
  1573. }
  1574. tbl.AddRow(fmt.Sprintf("%d", idx), r.Name, r.Address, fmt.Sprintf("%d", r.TTL), st)
  1575. }
  1576. return tbl.String()
  1577. case "set":
  1578. if user.Role == auth.RoleRead {
  1579. return ColorRed + "Permission denied\n" + ColorReset
  1580. }
  1581. _, params := parseArgs(parts[1:])
  1582. if sStr, ok := params["servers"]; ok {
  1583. srvs := strings.Split(sStr, ",")
  1584. cfg := e.dnsMgr.GetSettings()
  1585. cfg.Servers = srvs
  1586. if err := e.dnsMgr.SaveSettings(cfg); err != nil {
  1587. return ColorRed + "Error saving DNS settings: " + err.Error() + ColorReset + "\n"
  1588. }
  1589. return ColorGreen + "DNS servers updated.\n" + ColorReset
  1590. }
  1591. return ColorRed + "Usage: /ip dns set servers=<ip1,ip2>\n" + ColorReset
  1592. default:
  1593. return ColorRed + "Usage: /ip dns [print|set|cache flush|static]\n" + ColorReset
  1594. }
  1595. }
  1596. // --- IP Firewall ---
  1597. func (e *Executor) handleIPFirewall(parts []string, user *auth.PublicUser) string {
  1598. if e.fwMgr == nil {
  1599. return ColorRed + "Firewall manager not available.\n" + ColorReset
  1600. }
  1601. if len(parts) == 0 {
  1602. return ColorYellow + "Firewall subsystem. Available: filter, nat, mangle, address-list\n" + ColorReset
  1603. }
  1604. tableType := strings.ToLower(parts[0])
  1605. switch tableType {
  1606. case "filter":
  1607. return e.handleFirewallFilter(parts[1:], user)
  1608. case "nat":
  1609. return e.handleFirewallNAT(parts[1:], user)
  1610. case "mangle":
  1611. return e.handleFirewallMangle(parts[1:], user)
  1612. case "address-list":
  1613. return e.handleFirewallAddressList(parts[1:], user)
  1614. default:
  1615. return ColorRed + "Unknown firewall table: " + tableType + "\n" + ColorReset
  1616. }
  1617. }
  1618. func (e *Executor) handleFirewallFilter(parts []string, user *auth.PublicUser) string {
  1619. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1620. rules := e.fwMgr.ListFilterRules()
  1621. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "PROTO", "SRC-ADDR", "DST-ADDR", "DST-PORT", "COMMENT")
  1622. for _, r := range rules {
  1623. flags := ""
  1624. if r.Disabled {
  1625. flags += "X"
  1626. } else {
  1627. flags += " "
  1628. }
  1629. tbl.AddRow(flags, r.Chain, r.Action, r.Protocol, r.SrcAddress, r.DstAddress, r.DstPort, r.Comment)
  1630. }
  1631. return tbl.String()
  1632. }
  1633. if user.Role == auth.RoleRead {
  1634. return ColorRed + "Permission denied: viewer role cannot modify firewall\n" + ColorReset
  1635. }
  1636. sub := strings.ToLower(parts[0])
  1637. switch sub {
  1638. case "add":
  1639. _, params := parseArgs(parts[1:])
  1640. chain := params["chain"]
  1641. action := params["action"]
  1642. if chain == "" || action == "" {
  1643. return ColorRed + "Usage: /ip firewall filter add chain=<input|forward|output> action=<accept|drop|reject> [protocol=...] [dst-port=...]\n" + ColorReset
  1644. }
  1645. r := firewall.FilterRule{
  1646. Chain: chain,
  1647. Action: action,
  1648. Protocol: params["protocol"],
  1649. SrcAddress: params["src-address"],
  1650. DstAddress: params["dst-address"],
  1651. DstPort: params["dst-port"],
  1652. Comment: params["comment"],
  1653. Disabled: false,
  1654. }
  1655. if err := e.fwMgr.SaveFilterRule(r); err != nil {
  1656. return ColorRed + "Error adding filter rule: " + err.Error() + ColorReset + "\n"
  1657. }
  1658. return ColorGreen + "Firewall filter rule added.\n" + ColorReset
  1659. case "remove", "rem", "delete":
  1660. pos, _ := parseArgs(parts[1:])
  1661. if len(pos) == 0 {
  1662. return ColorRed + "Usage: /ip firewall filter remove <id>\n" + ColorReset
  1663. }
  1664. if err := e.fwMgr.DeleteFilterRule(pos[0]); err != nil {
  1665. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1666. }
  1667. return ColorGreen + fmt.Sprintf("Filter rule %s removed.\n", pos[0]) + ColorReset
  1668. case "enable":
  1669. pos, _ := parseArgs(parts[1:])
  1670. if len(pos) == 0 {
  1671. return ColorRed + "Usage: /ip firewall filter enable <id>\n" + ColorReset
  1672. }
  1673. if err := e.fwMgr.SetFilterRuleState(pos[0], false); err != nil {
  1674. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1675. }
  1676. return ColorGreen + fmt.Sprintf("Filter rule %s enabled.\n", pos[0]) + ColorReset
  1677. case "disable":
  1678. pos, _ := parseArgs(parts[1:])
  1679. if len(pos) == 0 {
  1680. return ColorRed + "Usage: /ip firewall filter disable <id>\n" + ColorReset
  1681. }
  1682. if err := e.fwMgr.SetFilterRuleState(pos[0], true); err != nil {
  1683. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1684. }
  1685. return ColorGreen + fmt.Sprintf("Filter rule %s disabled.\n", pos[0]) + ColorReset
  1686. default:
  1687. return ColorRed + "Usage: /ip firewall filter [print|add|remove|enable|disable]\n" + ColorReset
  1688. }
  1689. }
  1690. func (e *Executor) handleFirewallNAT(parts []string, user *auth.PublicUser) string {
  1691. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1692. rules := e.fwMgr.ListNATRules()
  1693. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "OUT-IFACE", "TO-ADDRESSES", "TO-PORTS", "COMMENT")
  1694. for _, r := range rules {
  1695. flags := ""
  1696. if r.Disabled {
  1697. flags += "X"
  1698. } else {
  1699. flags += " "
  1700. }
  1701. tbl.AddRow(flags, r.Chain, r.Action, r.OutInterface, r.ToAddresses, r.ToPorts, r.Comment)
  1702. }
  1703. return tbl.String()
  1704. }
  1705. if user.Role == auth.RoleRead {
  1706. return ColorRed + "Permission denied: viewer role cannot modify firewall\n" + ColorReset
  1707. }
  1708. sub := strings.ToLower(parts[0])
  1709. switch sub {
  1710. case "add":
  1711. _, params := parseArgs(parts[1:])
  1712. chain := params["chain"]
  1713. action := params["action"]
  1714. if chain == "" || action == "" {
  1715. return ColorRed + "Usage: /ip firewall nat add chain=<srcnat|dstnat> action=<masquerade|dst-nat|src-nat> [out-interface=...] [to-addresses=...]\n" + ColorReset
  1716. }
  1717. r := firewall.NATRule{
  1718. Chain: chain,
  1719. Action: action,
  1720. OutInterface: params["out-interface"],
  1721. ToAddresses: params["to-addresses"],
  1722. ToPorts: params["to-ports"],
  1723. Comment: params["comment"],
  1724. Disabled: false,
  1725. }
  1726. if err := e.fwMgr.SaveNATRule(r); err != nil {
  1727. return ColorRed + "Error adding NAT rule: " + err.Error() + ColorReset + "\n"
  1728. }
  1729. return ColorGreen + "Firewall NAT rule added.\n" + ColorReset
  1730. case "remove", "rem", "delete":
  1731. pos, _ := parseArgs(parts[1:])
  1732. if len(pos) == 0 {
  1733. return ColorRed + "Usage: /ip firewall nat remove <id>\n" + ColorReset
  1734. }
  1735. if err := e.fwMgr.DeleteNATRule(pos[0]); err != nil {
  1736. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1737. }
  1738. return ColorGreen + fmt.Sprintf("NAT rule %s removed.\n", pos[0]) + ColorReset
  1739. case "enable":
  1740. pos, _ := parseArgs(parts[1:])
  1741. if len(pos) == 0 {
  1742. return ColorRed + "Usage: /ip firewall nat enable <id>\n" + ColorReset
  1743. }
  1744. if err := e.fwMgr.SetNATRuleState(pos[0], false); err != nil {
  1745. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1746. }
  1747. return ColorGreen + fmt.Sprintf("NAT rule %s enabled.\n", pos[0]) + ColorReset
  1748. case "disable":
  1749. pos, _ := parseArgs(parts[1:])
  1750. if len(pos) == 0 {
  1751. return ColorRed + "Usage: /ip firewall nat disable <id>\n" + ColorReset
  1752. }
  1753. if err := e.fwMgr.SetNATRuleState(pos[0], true); err != nil {
  1754. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1755. }
  1756. return ColorGreen + fmt.Sprintf("NAT rule %s disabled.\n", pos[0]) + ColorReset
  1757. default:
  1758. return ColorRed + "Usage: /ip firewall nat [print|add|remove|enable|disable]\n" + ColorReset
  1759. }
  1760. }
  1761. func (e *Executor) handleFirewallMangle(parts []string, user *auth.PublicUser) string {
  1762. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1763. rules := e.fwMgr.ListMangleRules()
  1764. tbl := NewTable("FLAGS", "CHAIN", "ACTION", "PASSTHROUGH", "COMMENT")
  1765. for _, r := range rules {
  1766. flags := ""
  1767. if r.Disabled {
  1768. flags += "X"
  1769. } else {
  1770. flags += " "
  1771. }
  1772. tbl.AddRow(flags, r.Chain, r.Action, fmt.Sprintf("%t", r.Passthrough), r.Comment)
  1773. }
  1774. return tbl.String()
  1775. }
  1776. if user.Role == auth.RoleRead {
  1777. return ColorRed + "Permission denied\n" + ColorReset
  1778. }
  1779. sub := strings.ToLower(parts[0])
  1780. switch sub {
  1781. case "add":
  1782. _, params := parseArgs(parts[1:])
  1783. chain := params["chain"]
  1784. action := params["action"]
  1785. if chain == "" || action == "" {
  1786. return ColorRed + "Usage: /ip firewall mangle add chain=... action=...\n" + ColorReset
  1787. }
  1788. r := firewall.MangleRule{
  1789. Chain: chain,
  1790. Action: action,
  1791. Passthrough: true,
  1792. Comment: params["comment"],
  1793. Disabled: false,
  1794. }
  1795. if err := e.fwMgr.SaveMangleRule(r); err != nil {
  1796. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1797. }
  1798. return ColorGreen + "Mangle rule added.\n" + ColorReset
  1799. case "remove", "rem", "delete":
  1800. pos, _ := parseArgs(parts[1:])
  1801. if len(pos) == 0 {
  1802. return ColorRed + "Usage: /ip firewall mangle remove <id>\n" + ColorReset
  1803. }
  1804. if err := e.fwMgr.DeleteMangleRule(pos[0]); err != nil {
  1805. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1806. }
  1807. return ColorGreen + fmt.Sprintf("Mangle rule %s removed.\n", pos[0]) + ColorReset
  1808. default:
  1809. return ColorRed + "Usage: /ip firewall mangle [print|add|remove]\n" + ColorReset
  1810. }
  1811. }
  1812. func (e *Executor) handleFirewallAddressList(parts []string, user *auth.PublicUser) string {
  1813. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1814. entries := e.fwMgr.ListAddressLists()
  1815. tbl := NewTable("FLAGS", "LIST", "ADDRESS", "COMMENT")
  1816. for _, entry := range entries {
  1817. flags := ""
  1818. if entry.Disabled {
  1819. flags += "X"
  1820. } else {
  1821. flags += " "
  1822. }
  1823. tbl.AddRow(flags, entry.Name, entry.Address, entry.Comment)
  1824. }
  1825. return tbl.String()
  1826. }
  1827. if user.Role == auth.RoleRead {
  1828. return ColorRed + "Permission denied\n" + ColorReset
  1829. }
  1830. sub := strings.ToLower(parts[0])
  1831. switch sub {
  1832. case "add":
  1833. _, params := parseArgs(parts[1:])
  1834. list := params["list"]
  1835. addr := params["address"]
  1836. if list == "" || addr == "" {
  1837. return ColorRed + "Usage: /ip firewall address-list add list=<name> address=<cidr> [comment=...]\n" + ColorReset
  1838. }
  1839. entry := firewall.AddressListEntry{
  1840. Name: list,
  1841. Address: addr,
  1842. Comment: params["comment"],
  1843. Disabled: false,
  1844. }
  1845. if err := e.fwMgr.SaveAddressListEntry(entry); err != nil {
  1846. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1847. }
  1848. return ColorGreen + fmt.Sprintf("Added %s to address list %s.\n", addr, list) + ColorReset
  1849. case "remove", "rem", "delete":
  1850. pos, _ := parseArgs(parts[1:])
  1851. if len(pos) == 0 {
  1852. return ColorRed + "Usage: /ip firewall address-list remove <id>\n" + ColorReset
  1853. }
  1854. if err := e.fwMgr.DeleteAddressListEntry(pos[0]); err != nil {
  1855. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1856. }
  1857. return ColorGreen + fmt.Sprintf("Address list entry %s removed.\n", pos[0]) + ColorReset
  1858. default:
  1859. return ColorRed + "Usage: /ip firewall address-list [print|add|remove]\n" + ColorReset
  1860. }
  1861. }
  1862. // --- IP Service ---
  1863. func (e *Executor) handleIPService(parts []string, user *auth.PublicUser) string {
  1864. if e.servicesMgr == nil {
  1865. return ColorRed + "Services manager not available.\n" + ColorReset
  1866. }
  1867. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  1868. svcs := e.servicesMgr.ListServices()
  1869. tbl := NewTable("NAME", "PORT", "DISABLED", "CERTIFICATE", "COMMENT")
  1870. for _, s := range svcs {
  1871. disStr := ColorGreen + "no" + ColorReset
  1872. if s.Disabled {
  1873. disStr = ColorRed + "yes" + ColorReset
  1874. }
  1875. certStr := "-"
  1876. if s.Certificate != "" {
  1877. certStr = ColorCyan + s.Certificate + ColorReset
  1878. }
  1879. tbl.AddRow(s.Name, fmt.Sprintf("%d", s.Port), disStr, certStr, s.Comment)
  1880. }
  1881. return tbl.String()
  1882. }
  1883. if user.Role != auth.RoleFull {
  1884. return ColorRed + "Permission denied: admin role required to manage services\n" + ColorReset
  1885. }
  1886. sub := strings.ToLower(parts[0])
  1887. switch sub {
  1888. case "enable":
  1889. pos, _ := parseArgs(parts[1:])
  1890. if len(pos) == 0 {
  1891. return ColorRed + "Usage: /ip service enable <name>\n" + ColorReset
  1892. }
  1893. name := pos[0]
  1894. if err := e.servicesMgr.SetServiceState(name, false); err != nil {
  1895. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1896. }
  1897. return ColorGreen + fmt.Sprintf("Service %s enabled.\n", name) + ColorReset
  1898. case "disable":
  1899. pos, _ := parseArgs(parts[1:])
  1900. if len(pos) == 0 {
  1901. return ColorRed + "Usage: /ip service disable <name>\n" + ColorReset
  1902. }
  1903. name := pos[0]
  1904. if err := e.servicesMgr.SetServiceState(name, true); err != nil {
  1905. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1906. }
  1907. return ColorGreen + fmt.Sprintf("Service %s disabled.\n", name) + ColorReset
  1908. case "set":
  1909. pos, params := parseArgs(parts[1:])
  1910. if len(pos) == 0 {
  1911. return ColorRed + "Usage: /ip service set <name> [port=<port>] [disabled=yes|no] [certificate=<cert_name>]\n" + ColorReset
  1912. }
  1913. name := pos[0]
  1914. svc, ok := e.servicesMgr.GetService(name)
  1915. if !ok {
  1916. return ColorRed + fmt.Sprintf("Service %q not found\n", name) + ColorReset
  1917. }
  1918. port := svc.Port
  1919. disabled := svc.Disabled
  1920. certName := svc.Certificate
  1921. if pStr, ok := params["port"]; ok {
  1922. if pVal, err := strconv.Atoi(pStr); err == nil && pVal > 0 && pVal < 65536 {
  1923. port = pVal
  1924. }
  1925. }
  1926. if dStr, ok := params["disabled"]; ok {
  1927. disabled = (strings.ToLower(dStr) == "yes" || strings.ToLower(dStr) == "true")
  1928. }
  1929. if cStr, ok := params["certificate"]; ok {
  1930. certName = cStr
  1931. }
  1932. if err := e.servicesMgr.SetService(name, port, disabled, certName); err != nil {
  1933. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  1934. }
  1935. return ColorGreen + fmt.Sprintf("Service %s updated (port=%d, disabled=%t, certificate=%q).\n", name, port, disabled, certName) + ColorReset
  1936. default:
  1937. return ColorRed + "Usage: /ip service [print|enable|disable|set]\n" + ColorReset
  1938. }
  1939. }
  1940. // ==================== DYNAMIC ROUTING ====================
  1941. func (e *Executor) handleRouting(parts []string, user *auth.PublicUser) string {
  1942. if e.dynRoutingMgr == nil {
  1943. return ColorRed + "Dynamic routing subsystem not available.\n" + ColorReset
  1944. }
  1945. if len(parts) == 0 {
  1946. return ColorYellow + "Routing subsystem. Available: bgp, ospf, filter\n" + ColorReset
  1947. }
  1948. sub := strings.ToLower(parts[0])
  1949. switch sub {
  1950. case "bgp":
  1951. return e.handleRoutingBGP(parts[1:], user)
  1952. case "ospf":
  1953. return e.handleRoutingOSPF(parts[1:], user)
  1954. case "filter":
  1955. return e.handleRoutingFilter(parts[1:], user)
  1956. default:
  1957. return ColorRed + "Unknown routing protocol: " + sub + "\n" + ColorReset
  1958. }
  1959. }
  1960. func (e *Executor) handleRoutingBGP(parts []string, user *auth.PublicUser) string {
  1961. if len(parts) == 0 || parts[0] == "instance" {
  1962. inst := e.dynRoutingMgr.GetBGPInstance()
  1963. out := fmt.Sprintf(" enabled: %t\n", inst.Enabled)
  1964. out += fmt.Sprintf(" as: %d\n", inst.AS)
  1965. out += fmt.Sprintf(" router-id: %s\n", inst.RouterID)
  1966. return out
  1967. }
  1968. sub := strings.ToLower(parts[0])
  1969. switch sub {
  1970. case "peer", "connection", "peers":
  1971. peers := e.dynRoutingMgr.ListBGPPeers()
  1972. tbl := NewTable("FLAGS", "NAME", "REMOTE-AS", "REMOTE-ADDRESS", "STATUS", "UPTIME")
  1973. for _, p := range peers {
  1974. flags := ""
  1975. if p.Disabled {
  1976. flags += "X"
  1977. } else {
  1978. flags += "E"
  1979. }
  1980. st := p.State
  1981. if st == "" {
  1982. st = "Active"
  1983. }
  1984. tbl.AddRow(flags, p.Name, fmt.Sprintf("%d", p.RemoteAS), p.RemoteAddress, st, p.Uptime)
  1985. }
  1986. return tbl.String()
  1987. case "network", "networks":
  1988. nets := e.dynRoutingMgr.ListBGPNetworks()
  1989. tbl := NewTable("NETWORK", "COMMENT")
  1990. for _, n := range nets {
  1991. tbl.AddRow(n.Network, n.Comment)
  1992. }
  1993. return tbl.String()
  1994. case "routes":
  1995. routes := e.dynRoutingMgr.GetBGPRoutes()
  1996. tbl := NewTable("PREFIX", "NEXT-HOP", "PATH", "METRIC", "WEIGHT")
  1997. for _, r := range routes {
  1998. tbl.AddRow(r.Prefix, r.NextHop, r.Path, fmt.Sprintf("%d", r.Metric), fmt.Sprintf("%d", r.Weight))
  1999. }
  2000. return tbl.String()
  2001. default:
  2002. return ColorRed + "Usage: /routing bgp [instance|peer|network|routes]\n" + ColorReset
  2003. }
  2004. }
  2005. func parseCLIEnableDisable(val string, defaultVal bool) bool {
  2006. v := strings.ToLower(strings.TrimSpace(val))
  2007. if v == "yes" || v == "true" || v == "1" || v == "enable" || v == "enabled" {
  2008. return true
  2009. }
  2010. if v == "no" || v == "false" || v == "0" || v == "disable" || v == "disabled" {
  2011. return false
  2012. }
  2013. return defaultVal
  2014. }
  2015. func (e *Executor) handleRoutingOSPF(parts []string, user *auth.PublicUser) string {
  2016. if len(parts) == 0 {
  2017. return e.handleOSPFInstance([]string{"print"}, user)
  2018. }
  2019. sub := strings.ToLower(parts[0])
  2020. switch sub {
  2021. case "instance", "instances":
  2022. return e.handleOSPFInstance(parts[1:], user)
  2023. case "area", "areas":
  2024. if len(parts) > 1 && (strings.EqualFold(parts[1], "range") || strings.EqualFold(parts[1], "ranges")) {
  2025. return e.handleOSPFAreaRange(parts[2:], user)
  2026. }
  2027. return e.handleOSPFArea(parts[1:], user)
  2028. case "area-range", "area-ranges", "range", "ranges":
  2029. return e.handleOSPFAreaRange(parts[1:], user)
  2030. case "interface", "interfaces", "template", "templates", "interface-template", "interface-templates":
  2031. return e.handleOSPFInterface(parts[1:], user)
  2032. case "static-neighbor", "static-neighbors":
  2033. return e.handleOSPFStaticNeighbor(parts[1:], user)
  2034. case "neighbor", "neighbors":
  2035. return e.handleOSPFNeighborLive(parts[1:], user)
  2036. case "lsa", "lsadb":
  2037. return e.handleOSPFLSA(parts[1:], user)
  2038. case "route", "routes":
  2039. return e.handleOSPFRoute(parts[1:], user)
  2040. default:
  2041. return ColorRed + "Usage: /routing ospf [instance|area|area-range|interface|static-neighbor|neighbor|lsa|routes]\n" + ColorReset
  2042. }
  2043. }
  2044. func (e *Executor) handleOSPFInstance(parts []string, user *auth.PublicUser) string {
  2045. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2046. instances := e.dynRoutingMgr.ListOSPFInstances()
  2047. tbl := NewTable("#", "NAME", "ROUTER-ID", "VERSION", "REDISTRIBUTE", "DEFAULT-METRIC", "STATUS")
  2048. for idx, inst := range instances {
  2049. st := "enabled"
  2050. if inst.Disabled {
  2051. st = "disabled"
  2052. }
  2053. var redist []string
  2054. if inst.RedistributeConnected {
  2055. redist = append(redist, "connected")
  2056. }
  2057. if inst.RedistributeStatic {
  2058. redist = append(redist, "static")
  2059. }
  2060. if inst.RedistributeBGP {
  2061. redist = append(redist, "bgp")
  2062. }
  2063. redistStr := "-"
  2064. if len(redist) > 0 {
  2065. redistStr = strings.Join(redist, ",")
  2066. }
  2067. metricStr := "-"
  2068. if inst.DefaultMetric > 0 {
  2069. metricStr = fmt.Sprintf("%d", inst.DefaultMetric)
  2070. }
  2071. tbl.AddRow(fmt.Sprintf("%d", idx), inst.Name, inst.RouterID, fmt.Sprintf("%d", inst.Version), redistStr, metricStr, st)
  2072. }
  2073. return tbl.String()
  2074. }
  2075. if user.Role == auth.RoleRead {
  2076. return ColorRed + "Permission denied: viewer role cannot modify OSPF\n" + ColorReset
  2077. }
  2078. sub := strings.ToLower(parts[0])
  2079. switch sub {
  2080. case "add":
  2081. _, params := parseArgs(parts[1:])
  2082. name := params["name"]
  2083. if name == "" {
  2084. name = "Default"
  2085. }
  2086. routerID := params["router-id"]
  2087. version := 2
  2088. if vStr, ok := params["version"]; ok && vStr != "" {
  2089. if v, err := strconv.Atoi(vStr); err == nil && (v == 2 || v == 3) {
  2090. version = v
  2091. }
  2092. }
  2093. redistConn := parseCLIEnableDisable(params["redistribute-connected"], false)
  2094. redistStat := parseCLIEnableDisable(params["redistribute-static"], false)
  2095. redistBGP := parseCLIEnableDisable(params["redistribute-bgp"], false)
  2096. metric := 0
  2097. if mStr, ok := params["default-metric"]; ok && mStr != "" {
  2098. metric, _ = strconv.Atoi(mStr)
  2099. }
  2100. disabled := parseCLIEnableDisable(params["disabled"], false)
  2101. vrf := params["vrf"]
  2102. if vrf == "" {
  2103. vrf = "main"
  2104. }
  2105. rt := params["routing-table"]
  2106. if rt == "" {
  2107. rt = "main"
  2108. }
  2109. inst := dynamicrouting.OSPFInstance{
  2110. Name: name,
  2111. Version: version,
  2112. VRF: vrf,
  2113. RouterID: routerID,
  2114. RoutingTable: rt,
  2115. Enabled: !disabled,
  2116. Disabled: disabled,
  2117. RedistributeConnected: redistConn,
  2118. RedistributeStatic: redistStat,
  2119. RedistributeBGP: redistBGP,
  2120. DefaultMetric: metric,
  2121. Comment: params["comment"],
  2122. }
  2123. if err := e.dynRoutingMgr.SaveOSPFInstance(&inst); err != nil {
  2124. return ColorRed + "Error saving OSPF instance: " + err.Error() + ColorReset + "\n"
  2125. }
  2126. return ColorGreen + fmt.Sprintf("OSPF instance %s saved.\n", name) + ColorReset
  2127. case "set":
  2128. pos, params := parseArgs(parts[1:])
  2129. if len(pos) == 0 && params["numbers"] == "" {
  2130. return ColorRed + "Usage: /routing ospf instance set <number|name> [router-id=...] [redistribute-connected=yes/no] ...\n" + ColorReset
  2131. }
  2132. target := ""
  2133. if len(pos) > 0 {
  2134. target = pos[0]
  2135. } else {
  2136. target = params["numbers"]
  2137. }
  2138. instances := e.dynRoutingMgr.ListOSPFInstances()
  2139. var targetInst *dynamicrouting.OSPFInstance
  2140. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(instances) {
  2141. targetInst = &instances[idx]
  2142. } else {
  2143. for i := range instances {
  2144. if strings.EqualFold(instances[i].Name, target) || instances[i].ID == target {
  2145. targetInst = &instances[i]
  2146. break
  2147. }
  2148. }
  2149. }
  2150. if targetInst == nil {
  2151. return ColorRed + fmt.Sprintf("OSPF instance %q not found\n", target) + ColorReset
  2152. }
  2153. if v, ok := params["name"]; ok && v != "" {
  2154. targetInst.Name = v
  2155. }
  2156. if v, ok := params["router-id"]; ok {
  2157. targetInst.RouterID = v
  2158. }
  2159. if v, ok := params["redistribute-connected"]; ok {
  2160. targetInst.RedistributeConnected = parseCLIEnableDisable(v, false)
  2161. }
  2162. if v, ok := params["redistribute-static"]; ok {
  2163. targetInst.RedistributeStatic = parseCLIEnableDisable(v, false)
  2164. }
  2165. if v, ok := params["redistribute-bgp"]; ok {
  2166. targetInst.RedistributeBGP = parseCLIEnableDisable(v, false)
  2167. }
  2168. if v, ok := params["default-metric"]; ok {
  2169. targetInst.DefaultMetric, _ = strconv.Atoi(v)
  2170. }
  2171. if v, ok := params["disabled"]; ok {
  2172. targetInst.Disabled = parseCLIEnableDisable(v, false)
  2173. targetInst.Enabled = !targetInst.Disabled
  2174. }
  2175. if v, ok := params["comment"]; ok {
  2176. targetInst.Comment = v
  2177. }
  2178. if err := e.dynRoutingMgr.SaveOSPFInstance(targetInst); err != nil {
  2179. return ColorRed + "Error updating OSPF instance: " + err.Error() + ColorReset + "\n"
  2180. }
  2181. return ColorGreen + fmt.Sprintf("OSPF instance %s updated.\n", targetInst.Name) + ColorReset
  2182. case "remove", "rem", "delete":
  2183. pos, params := parseArgs(parts[1:])
  2184. target := ""
  2185. if len(pos) > 0 {
  2186. target = pos[0]
  2187. } else if v, ok := params["numbers"]; ok {
  2188. target = v
  2189. }
  2190. if target == "" {
  2191. return ColorRed + "Usage: /routing ospf instance remove <number|name>\n" + ColorReset
  2192. }
  2193. instances := e.dynRoutingMgr.ListOSPFInstances()
  2194. idToDelete := ""
  2195. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(instances) {
  2196. idToDelete = instances[idx].ID
  2197. } else {
  2198. for _, inst := range instances {
  2199. if strings.EqualFold(inst.Name, target) || inst.ID == target {
  2200. idToDelete = inst.ID
  2201. break
  2202. }
  2203. }
  2204. }
  2205. if idToDelete == "" {
  2206. return ColorRed + fmt.Sprintf("OSPF instance %q not found\n", target) + ColorReset
  2207. }
  2208. if err := e.dynRoutingMgr.DeleteOSPFInstance(idToDelete); err != nil {
  2209. return ColorRed + "Error deleting OSPF instance: " + err.Error() + ColorReset + "\n"
  2210. }
  2211. return ColorGreen + fmt.Sprintf("OSPF instance %s removed.\n", target) + ColorReset
  2212. case "enable":
  2213. pos, _ := parseArgs(parts[1:])
  2214. if len(pos) == 0 {
  2215. return ColorRed + "Usage: /routing ospf instance enable <number|name>\n" + ColorReset
  2216. }
  2217. return e.setOSPFInstanceStateByTarget(pos[0], false)
  2218. case "disable":
  2219. pos, _ := parseArgs(parts[1:])
  2220. if len(pos) == 0 {
  2221. return ColorRed + "Usage: /routing ospf instance disable <number|name>\n" + ColorReset
  2222. }
  2223. return e.setOSPFInstanceStateByTarget(pos[0], true)
  2224. default:
  2225. return ColorRed + "Usage: /routing ospf instance [print|add|set|remove|enable|disable]\n" + ColorReset
  2226. }
  2227. }
  2228. func (e *Executor) handleOSPFArea(parts []string, user *auth.PublicUser) string {
  2229. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2230. areas := e.dynRoutingMgr.ListOSPFAreas()
  2231. tbl := NewTable("#", "NAME", "AREA-ID", "INSTANCE", "TYPE", "DEFAULT-COST", "NO-SUMMARIES", "STATUS")
  2232. for idx, a := range areas {
  2233. st := "enabled"
  2234. if a.Disabled {
  2235. st = "disabled"
  2236. }
  2237. costStr := "-"
  2238. if a.DefaultCost > 0 {
  2239. costStr = fmt.Sprintf("%d", a.DefaultCost)
  2240. }
  2241. tbl.AddRow(fmt.Sprintf("%d", idx), a.Name, a.AreaID, a.Instance, a.AreaType, costStr, fmt.Sprintf("%t", a.NoSummaries), st)
  2242. }
  2243. return tbl.String()
  2244. }
  2245. if user.Role == auth.RoleRead {
  2246. return ColorRed + "Permission denied: viewer role cannot modify OSPF\n" + ColorReset
  2247. }
  2248. sub := strings.ToLower(parts[0])
  2249. switch sub {
  2250. case "add":
  2251. _, params := parseArgs(parts[1:])
  2252. name := params["name"]
  2253. if name == "" {
  2254. return ColorRed + "Usage: /routing ospf area add name=<name> [area-id=...] [instance=...] [type=default|stub|nssa]\n" + ColorReset
  2255. }
  2256. areaID := params["area-id"]
  2257. if areaID == "" {
  2258. areaID = "0.0.0.0"
  2259. }
  2260. instName := params["instance"]
  2261. if instName == "" {
  2262. instName = "Default"
  2263. }
  2264. areaType := params["type"]
  2265. if areaType == "" {
  2266. areaType = "default"
  2267. }
  2268. defaultCost := 0
  2269. if cStr, ok := params["default-cost"]; ok && cStr != "" {
  2270. defaultCost, _ = strconv.Atoi(cStr)
  2271. }
  2272. noSummaries := parseCLIEnableDisable(params["no-summaries"], false)
  2273. disabled := parseCLIEnableDisable(params["disabled"], false)
  2274. area := dynamicrouting.OSPFArea{
  2275. Name: name,
  2276. AreaID: areaID,
  2277. Instance: instName,
  2278. AreaType: areaType,
  2279. DefaultCost: defaultCost,
  2280. NoSummaries: noSummaries,
  2281. Disabled: disabled,
  2282. Comment: params["comment"],
  2283. }
  2284. if err := e.dynRoutingMgr.SaveOSPFArea(area); err != nil {
  2285. return ColorRed + "Error saving OSPF area: " + err.Error() + ColorReset + "\n"
  2286. }
  2287. return ColorGreen + fmt.Sprintf("OSPF area %s saved.\n", name) + ColorReset
  2288. case "set":
  2289. pos, params := parseArgs(parts[1:])
  2290. target := ""
  2291. if len(pos) > 0 {
  2292. target = pos[0]
  2293. } else if v, ok := params["numbers"]; ok {
  2294. target = v
  2295. }
  2296. if target == "" {
  2297. return ColorRed + "Usage: /routing ospf area set <number|name> [area-id=...] [type=...] ...\n" + ColorReset
  2298. }
  2299. areas := e.dynRoutingMgr.ListOSPFAreas()
  2300. var targetArea *dynamicrouting.OSPFArea
  2301. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(areas) {
  2302. targetArea = &areas[idx]
  2303. } else {
  2304. for i := range areas {
  2305. if strings.EqualFold(areas[i].Name, target) || areas[i].ID == target {
  2306. targetArea = &areas[i]
  2307. break
  2308. }
  2309. }
  2310. }
  2311. if targetArea == nil {
  2312. return ColorRed + fmt.Sprintf("OSPF area %q not found\n", target) + ColorReset
  2313. }
  2314. if v, ok := params["name"]; ok && v != "" {
  2315. targetArea.Name = v
  2316. }
  2317. if v, ok := params["area-id"]; ok && v != "" {
  2318. targetArea.AreaID = v
  2319. }
  2320. if v, ok := params["instance"]; ok && v != "" {
  2321. targetArea.Instance = v
  2322. }
  2323. if v, ok := params["type"]; ok && v != "" {
  2324. targetArea.AreaType = v
  2325. }
  2326. if v, ok := params["default-cost"]; ok {
  2327. targetArea.DefaultCost, _ = strconv.Atoi(v)
  2328. }
  2329. if v, ok := params["no-summaries"]; ok {
  2330. targetArea.NoSummaries = parseCLIEnableDisable(v, false)
  2331. }
  2332. if v, ok := params["disabled"]; ok {
  2333. targetArea.Disabled = parseCLIEnableDisable(v, false)
  2334. }
  2335. if v, ok := params["comment"]; ok {
  2336. targetArea.Comment = v
  2337. }
  2338. if err := e.dynRoutingMgr.SaveOSPFArea(*targetArea); err != nil {
  2339. return ColorRed + "Error updating OSPF area: " + err.Error() + ColorReset + "\n"
  2340. }
  2341. return ColorGreen + fmt.Sprintf("OSPF area %s updated.\n", targetArea.Name) + ColorReset
  2342. case "remove", "rem", "delete":
  2343. pos, params := parseArgs(parts[1:])
  2344. target := ""
  2345. if len(pos) > 0 {
  2346. target = pos[0]
  2347. } else if v, ok := params["numbers"]; ok {
  2348. target = v
  2349. }
  2350. if target == "" {
  2351. return ColorRed + "Usage: /routing ospf area remove <number|name>\n" + ColorReset
  2352. }
  2353. areas := e.dynRoutingMgr.ListOSPFAreas()
  2354. idToDelete := ""
  2355. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(areas) {
  2356. idToDelete = areas[idx].ID
  2357. } else {
  2358. for _, a := range areas {
  2359. if strings.EqualFold(a.Name, target) || a.ID == target {
  2360. idToDelete = a.ID
  2361. break
  2362. }
  2363. }
  2364. }
  2365. if idToDelete == "" {
  2366. return ColorRed + fmt.Sprintf("OSPF area %q not found\n", target) + ColorReset
  2367. }
  2368. if err := e.dynRoutingMgr.DeleteOSPFArea(idToDelete); err != nil {
  2369. return ColorRed + "Error deleting OSPF area: " + err.Error() + ColorReset + "\n"
  2370. }
  2371. return ColorGreen + fmt.Sprintf("OSPF area %s removed.\n", target) + ColorReset
  2372. case "enable":
  2373. pos, _ := parseArgs(parts[1:])
  2374. if len(pos) == 0 {
  2375. return ColorRed + "Usage: /routing ospf area enable <number|name>\n" + ColorReset
  2376. }
  2377. return e.setOSPFAreaStateByTarget(pos[0], false)
  2378. case "disable":
  2379. pos, _ := parseArgs(parts[1:])
  2380. if len(pos) == 0 {
  2381. return ColorRed + "Usage: /routing ospf area disable <number|name>\n" + ColorReset
  2382. }
  2383. return e.setOSPFAreaStateByTarget(pos[0], true)
  2384. default:
  2385. return ColorRed + "Usage: /routing ospf area [print|add|set|remove|enable|disable]\n" + ColorReset
  2386. }
  2387. }
  2388. func (e *Executor) handleOSPFAreaRange(parts []string, user *auth.PublicUser) string {
  2389. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2390. ranges := e.dynRoutingMgr.ListOSPFAreaRanges()
  2391. tbl := NewTable("#", "PREFIX", "AREA", "COST", "ADVERTISE", "STATUS")
  2392. for idx, r := range ranges {
  2393. st := "enabled"
  2394. if r.Disabled {
  2395. st = "disabled"
  2396. }
  2397. tbl.AddRow(fmt.Sprintf("%d", idx), r.Prefix, r.Area, fmt.Sprintf("%d", r.Cost), fmt.Sprintf("%t", r.Advertise), st)
  2398. }
  2399. return tbl.String()
  2400. }
  2401. if user.Role == auth.RoleRead {
  2402. return ColorRed + "Permission denied: viewer role cannot modify OSPF\n" + ColorReset
  2403. }
  2404. sub := strings.ToLower(parts[0])
  2405. switch sub {
  2406. case "add":
  2407. _, params := parseArgs(parts[1:])
  2408. prefix := params["prefix"]
  2409. if prefix == "" {
  2410. return ColorRed + "Usage: /routing ospf area range add prefix=<cidr> [area=...] [cost=...] [advertise=yes/no]\n" + ColorReset
  2411. }
  2412. area := params["area"]
  2413. if area == "" {
  2414. area = "Default"
  2415. }
  2416. cost := 1
  2417. if cStr, ok := params["cost"]; ok && cStr != "" {
  2418. cost, _ = strconv.Atoi(cStr)
  2419. }
  2420. adv := true
  2421. if aStr, ok := params["advertise"]; ok && aStr != "" {
  2422. adv = parseCLIEnableDisable(aStr, true)
  2423. }
  2424. disabled := parseCLIEnableDisable(params["disabled"], false)
  2425. ar := dynamicrouting.OSPFAreaRange{
  2426. Prefix: prefix,
  2427. Area: area,
  2428. Cost: cost,
  2429. Advertise: adv,
  2430. Disabled: disabled,
  2431. Comment: params["comment"],
  2432. }
  2433. if err := e.dynRoutingMgr.SaveOSPFAreaRange(ar); err != nil {
  2434. return ColorRed + "Error saving area range: " + err.Error() + ColorReset + "\n"
  2435. }
  2436. return ColorGreen + fmt.Sprintf("Area range %s added.\n", prefix) + ColorReset
  2437. case "set":
  2438. pos, params := parseArgs(parts[1:])
  2439. target := ""
  2440. if len(pos) > 0 {
  2441. target = pos[0]
  2442. } else if v, ok := params["numbers"]; ok {
  2443. target = v
  2444. }
  2445. if target == "" {
  2446. return ColorRed + "Usage: /routing ospf area range set <number|prefix> [cost=...] [advertise=yes/no]\n" + ColorReset
  2447. }
  2448. ranges := e.dynRoutingMgr.ListOSPFAreaRanges()
  2449. var targetRange *dynamicrouting.OSPFAreaRange
  2450. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(ranges) {
  2451. targetRange = &ranges[idx]
  2452. } else {
  2453. for i := range ranges {
  2454. if strings.EqualFold(ranges[i].Prefix, target) || ranges[i].ID == target {
  2455. targetRange = &ranges[i]
  2456. break
  2457. }
  2458. }
  2459. }
  2460. if targetRange == nil {
  2461. return ColorRed + fmt.Sprintf("Area range %q not found\n", target) + ColorReset
  2462. }
  2463. if v, ok := params["prefix"]; ok && v != "" {
  2464. targetRange.Prefix = v
  2465. }
  2466. if v, ok := params["area"]; ok && v != "" {
  2467. targetRange.Area = v
  2468. }
  2469. if v, ok := params["cost"]; ok {
  2470. targetRange.Cost, _ = strconv.Atoi(v)
  2471. }
  2472. if v, ok := params["advertise"]; ok {
  2473. targetRange.Advertise = parseCLIEnableDisable(v, true)
  2474. }
  2475. if v, ok := params["disabled"]; ok {
  2476. targetRange.Disabled = parseCLIEnableDisable(v, false)
  2477. }
  2478. if v, ok := params["comment"]; ok {
  2479. targetRange.Comment = v
  2480. }
  2481. if err := e.dynRoutingMgr.SaveOSPFAreaRange(*targetRange); err != nil {
  2482. return ColorRed + "Error updating area range: " + err.Error() + ColorReset + "\n"
  2483. }
  2484. return ColorGreen + fmt.Sprintf("Area range %s updated.\n", targetRange.Prefix) + ColorReset
  2485. case "remove", "rem", "delete":
  2486. pos, params := parseArgs(parts[1:])
  2487. target := ""
  2488. if len(pos) > 0 {
  2489. target = pos[0]
  2490. } else if v, ok := params["numbers"]; ok {
  2491. target = v
  2492. }
  2493. if target == "" {
  2494. return ColorRed + "Usage: /routing ospf area range remove <number|prefix>\n" + ColorReset
  2495. }
  2496. ranges := e.dynRoutingMgr.ListOSPFAreaRanges()
  2497. idToDelete := ""
  2498. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(ranges) {
  2499. idToDelete = ranges[idx].ID
  2500. } else {
  2501. for _, r := range ranges {
  2502. if strings.EqualFold(r.Prefix, target) || r.ID == target {
  2503. idToDelete = r.ID
  2504. break
  2505. }
  2506. }
  2507. }
  2508. if idToDelete == "" {
  2509. return ColorRed + fmt.Sprintf("Area range %q not found\n", target) + ColorReset
  2510. }
  2511. if err := e.dynRoutingMgr.DeleteOSPFAreaRange(idToDelete); err != nil {
  2512. return ColorRed + "Error deleting area range: " + err.Error() + ColorReset + "\n"
  2513. }
  2514. return ColorGreen + fmt.Sprintf("Area range %s removed.\n", target) + ColorReset
  2515. default:
  2516. return ColorRed + "Usage: /routing ospf area range [print|add|set|remove]\n" + ColorReset
  2517. }
  2518. }
  2519. func (e *Executor) handleOSPFInterface(parts []string, user *auth.PublicUser) string {
  2520. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2521. ifaces := e.dynRoutingMgr.ListOSPFInterfaces()
  2522. tbl := NewTable("#", "INTERFACE", "AREA", "NETWORKS", "TYPE", "COST", "PRIORITY", "PASSIVE", "AUTH", "STATUS")
  2523. for idx, i := range ifaces {
  2524. st := "enabled"
  2525. if i.Disabled {
  2526. st = "disabled"
  2527. }
  2528. ifaceName := i.Interfaces
  2529. if ifaceName == "" {
  2530. ifaceName = i.Interface
  2531. }
  2532. if ifaceName == "" {
  2533. ifaceName = "*"
  2534. }
  2535. nets := i.Networks
  2536. if nets == "" {
  2537. nets = i.Network
  2538. }
  2539. if nets == "" {
  2540. nets = "-"
  2541. }
  2542. authStr := i.Auth
  2543. if authStr == "" {
  2544. authStr = i.AuthType
  2545. }
  2546. if authStr == "" {
  2547. authStr = "none"
  2548. }
  2549. tbl.AddRow(fmt.Sprintf("%d", idx), ifaceName, i.Area, nets, i.NetworkType, fmt.Sprintf("%d", i.Cost), fmt.Sprintf("%d", i.Priority), fmt.Sprintf("%t", i.Passive), authStr, st)
  2550. }
  2551. return tbl.String()
  2552. }
  2553. if user.Role == auth.RoleRead {
  2554. return ColorRed + "Permission denied: viewer role cannot modify OSPF\n" + ColorReset
  2555. }
  2556. sub := strings.ToLower(parts[0])
  2557. switch sub {
  2558. case "add":
  2559. _, params := parseArgs(parts[1:])
  2560. ifaceName := params["interface"]
  2561. if ifaceName == "" {
  2562. ifaceName = params["interfaces"]
  2563. }
  2564. if ifaceName == "" {
  2565. ifaceName = "*"
  2566. }
  2567. area := params["area"]
  2568. if area == "" {
  2569. area = "Default"
  2570. }
  2571. nets := params["networks"]
  2572. if nets == "" {
  2573. nets = params["network"]
  2574. }
  2575. netType := params["network-type"]
  2576. if netType == "" {
  2577. netType = "broadcast"
  2578. }
  2579. cost := 1
  2580. if cStr, ok := params["cost"]; ok && cStr != "" {
  2581. cost, _ = strconv.Atoi(cStr)
  2582. }
  2583. pri := 128
  2584. if pStr, ok := params["priority"]; ok && pStr != "" {
  2585. pri, _ = strconv.Atoi(pStr)
  2586. }
  2587. hello := 10
  2588. if hStr, ok := params["hello-interval"]; ok && hStr != "" {
  2589. hello, _ = strconv.Atoi(hStr)
  2590. }
  2591. dead := 40
  2592. if dStr, ok := params["dead-interval"]; ok && dStr != "" {
  2593. dead, _ = strconv.Atoi(dStr)
  2594. }
  2595. passive := parseCLIEnableDisable(params["passive"], false)
  2596. authType := params["auth"]
  2597. if authType == "" {
  2598. authType = params["auth-type"]
  2599. }
  2600. authKey := params["auth-key"]
  2601. disabled := parseCLIEnableDisable(params["disabled"], false)
  2602. iface := dynamicrouting.OSPFInterface{
  2603. Interfaces: ifaceName,
  2604. Interface: ifaceName,
  2605. Area: area,
  2606. Networks: nets,
  2607. Network: nets,
  2608. NetworkType: netType,
  2609. Cost: cost,
  2610. Priority: pri,
  2611. HelloInterval: hello,
  2612. DeadInterval: dead,
  2613. Passive: passive,
  2614. Auth: authType,
  2615. AuthType: authType,
  2616. AuthKey: authKey,
  2617. Disabled: disabled,
  2618. Comment: params["comment"],
  2619. }
  2620. if err := e.dynRoutingMgr.SaveOSPFInterface(iface); err != nil {
  2621. return ColorRed + "Error saving OSPF interface template: " + err.Error() + ColorReset + "\n"
  2622. }
  2623. return ColorGreen + fmt.Sprintf("OSPF interface template %s saved.\n", ifaceName) + ColorReset
  2624. case "set":
  2625. pos, params := parseArgs(parts[1:])
  2626. target := ""
  2627. if len(pos) > 0 {
  2628. target = pos[0]
  2629. } else if v, ok := params["numbers"]; ok {
  2630. target = v
  2631. }
  2632. if target == "" {
  2633. return ColorRed + "Usage: /routing ospf interface set <number|id> [cost=...] [priority=...] ...\n" + ColorReset
  2634. }
  2635. ifaces := e.dynRoutingMgr.ListOSPFInterfaces()
  2636. var targetIface *dynamicrouting.OSPFInterface
  2637. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(ifaces) {
  2638. targetIface = &ifaces[idx]
  2639. } else {
  2640. for i := range ifaces {
  2641. if ifaces[i].ID == target || ifaces[i].Interfaces == target || ifaces[i].Interface == target {
  2642. targetIface = &ifaces[i]
  2643. break
  2644. }
  2645. }
  2646. }
  2647. if targetIface == nil {
  2648. return ColorRed + fmt.Sprintf("OSPF interface template %q not found\n", target) + ColorReset
  2649. }
  2650. if v, ok := params["interface"]; ok && v != "" {
  2651. targetIface.Interface = v
  2652. targetIface.Interfaces = v
  2653. } else if v, ok := params["interfaces"]; ok && v != "" {
  2654. targetIface.Interface = v
  2655. targetIface.Interfaces = v
  2656. }
  2657. if v, ok := params["area"]; ok && v != "" {
  2658. targetIface.Area = v
  2659. }
  2660. if v, ok := params["networks"]; ok {
  2661. targetIface.Networks = v
  2662. targetIface.Network = v
  2663. } else if v, ok := params["network"]; ok {
  2664. targetIface.Networks = v
  2665. targetIface.Network = v
  2666. }
  2667. if v, ok := params["network-type"]; ok && v != "" {
  2668. targetIface.NetworkType = v
  2669. }
  2670. if v, ok := params["cost"]; ok {
  2671. targetIface.Cost, _ = strconv.Atoi(v)
  2672. }
  2673. if v, ok := params["priority"]; ok {
  2674. targetIface.Priority, _ = strconv.Atoi(v)
  2675. }
  2676. if v, ok := params["hello-interval"]; ok {
  2677. targetIface.HelloInterval, _ = strconv.Atoi(v)
  2678. }
  2679. if v, ok := params["dead-interval"]; ok {
  2680. targetIface.DeadInterval, _ = strconv.Atoi(v)
  2681. }
  2682. if v, ok := params["passive"]; ok {
  2683. targetIface.Passive = parseCLIEnableDisable(v, false)
  2684. }
  2685. if v, ok := params["auth"]; ok {
  2686. targetIface.Auth = v
  2687. targetIface.AuthType = v
  2688. } else if v, ok := params["auth-type"]; ok {
  2689. targetIface.Auth = v
  2690. targetIface.AuthType = v
  2691. }
  2692. if v, ok := params["auth-key"]; ok {
  2693. targetIface.AuthKey = v
  2694. }
  2695. if v, ok := params["disabled"]; ok {
  2696. targetIface.Disabled = parseCLIEnableDisable(v, false)
  2697. }
  2698. if v, ok := params["comment"]; ok {
  2699. targetIface.Comment = v
  2700. }
  2701. if err := e.dynRoutingMgr.SaveOSPFInterface(*targetIface); err != nil {
  2702. return ColorRed + "Error updating OSPF interface template: " + err.Error() + ColorReset + "\n"
  2703. }
  2704. return ColorGreen + fmt.Sprintf("OSPF interface template %s updated.\n", targetIface.ID) + ColorReset
  2705. case "remove", "rem", "delete":
  2706. pos, params := parseArgs(parts[1:])
  2707. target := ""
  2708. if len(pos) > 0 {
  2709. target = pos[0]
  2710. } else if v, ok := params["numbers"]; ok {
  2711. target = v
  2712. }
  2713. if target == "" {
  2714. return ColorRed + "Usage: /routing ospf interface remove <number|id>\n" + ColorReset
  2715. }
  2716. ifaces := e.dynRoutingMgr.ListOSPFInterfaces()
  2717. idToDelete := ""
  2718. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(ifaces) {
  2719. idToDelete = ifaces[idx].ID
  2720. } else {
  2721. for _, i := range ifaces {
  2722. if i.ID == target || i.Interfaces == target || i.Interface == target {
  2723. idToDelete = i.ID
  2724. break
  2725. }
  2726. }
  2727. }
  2728. if idToDelete == "" {
  2729. return ColorRed + fmt.Sprintf("OSPF interface template %q not found\n", target) + ColorReset
  2730. }
  2731. if err := e.dynRoutingMgr.DeleteOSPFInterface(idToDelete); err != nil {
  2732. return ColorRed + "Error deleting OSPF interface template: " + err.Error() + ColorReset + "\n"
  2733. }
  2734. return ColorGreen + fmt.Sprintf("OSPF interface template %s removed.\n", target) + ColorReset
  2735. case "enable":
  2736. pos, _ := parseArgs(parts[1:])
  2737. if len(pos) == 0 {
  2738. return ColorRed + "Usage: /routing ospf interface enable <number|id>\n" + ColorReset
  2739. }
  2740. return e.setOSPFInterfaceStateByTarget(pos[0], false)
  2741. case "disable":
  2742. pos, _ := parseArgs(parts[1:])
  2743. if len(pos) == 0 {
  2744. return ColorRed + "Usage: /routing ospf interface disable <number|id>\n" + ColorReset
  2745. }
  2746. return e.setOSPFInterfaceStateByTarget(pos[0], true)
  2747. default:
  2748. return ColorRed + "Usage: /routing ospf interface [print|add|set|remove|enable|disable]\n" + ColorReset
  2749. }
  2750. }
  2751. func (e *Executor) handleOSPFStaticNeighbor(parts []string, user *auth.PublicUser) string {
  2752. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  2753. nbrs := e.dynRoutingMgr.ListOSPFStaticNeighbors()
  2754. tbl := NewTable("#", "ADDRESS", "AREA", "POLL-INTERVAL", "COST", "STATUS")
  2755. for idx, n := range nbrs {
  2756. st := "enabled"
  2757. if n.Disabled {
  2758. st = "disabled"
  2759. }
  2760. tbl.AddRow(fmt.Sprintf("%d", idx), n.Address, n.Area, fmt.Sprintf("%d", n.PollInterval), fmt.Sprintf("%d", n.Cost), st)
  2761. }
  2762. return tbl.String()
  2763. }
  2764. if user.Role == auth.RoleRead {
  2765. return ColorRed + "Permission denied: viewer role cannot modify OSPF\n" + ColorReset
  2766. }
  2767. sub := strings.ToLower(parts[0])
  2768. switch sub {
  2769. case "add":
  2770. _, params := parseArgs(parts[1:])
  2771. addr := params["address"]
  2772. if addr == "" {
  2773. return ColorRed + "Usage: /routing ospf static-neighbor add address=<ip> [area=...] [poll-interval=...] [cost=...]\n" + ColorReset
  2774. }
  2775. area := params["area"]
  2776. if area == "" {
  2777. area = "Default"
  2778. }
  2779. poll := 120
  2780. if pStr, ok := params["poll-interval"]; ok && pStr != "" {
  2781. poll, _ = strconv.Atoi(pStr)
  2782. }
  2783. cost := 1
  2784. if cStr, ok := params["cost"]; ok && cStr != "" {
  2785. cost, _ = strconv.Atoi(cStr)
  2786. }
  2787. disabled := parseCLIEnableDisable(params["disabled"], false)
  2788. snbr := dynamicrouting.OSPFStaticNeighbor{
  2789. Address: addr,
  2790. Area: area,
  2791. PollInterval: poll,
  2792. Cost: cost,
  2793. Disabled: disabled,
  2794. Comment: params["comment"],
  2795. }
  2796. if err := e.dynRoutingMgr.SaveOSPFStaticNeighbor(snbr); err != nil {
  2797. return ColorRed + "Error saving static neighbor: " + err.Error() + ColorReset + "\n"
  2798. }
  2799. return ColorGreen + fmt.Sprintf("OSPF static neighbor %s saved.\n", addr) + ColorReset
  2800. case "set":
  2801. pos, params := parseArgs(parts[1:])
  2802. target := ""
  2803. if len(pos) > 0 {
  2804. target = pos[0]
  2805. } else if v, ok := params["numbers"]; ok {
  2806. target = v
  2807. }
  2808. if target == "" {
  2809. return ColorRed + "Usage: /routing ospf static-neighbor set <number|address> [poll-interval=...] [cost=...]\n" + ColorReset
  2810. }
  2811. nbrs := e.dynRoutingMgr.ListOSPFStaticNeighbors()
  2812. var targetNbr *dynamicrouting.OSPFStaticNeighbor
  2813. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(nbrs) {
  2814. targetNbr = &nbrs[idx]
  2815. } else {
  2816. for i := range nbrs {
  2817. if nbrs[i].Address == target || nbrs[i].ID == target {
  2818. targetNbr = &nbrs[i]
  2819. break
  2820. }
  2821. }
  2822. }
  2823. if targetNbr == nil {
  2824. return ColorRed + fmt.Sprintf("Static neighbor %q not found\n", target) + ColorReset
  2825. }
  2826. if v, ok := params["address"]; ok && v != "" {
  2827. targetNbr.Address = v
  2828. }
  2829. if v, ok := params["area"]; ok && v != "" {
  2830. targetNbr.Area = v
  2831. }
  2832. if v, ok := params["poll-interval"]; ok {
  2833. targetNbr.PollInterval, _ = strconv.Atoi(v)
  2834. }
  2835. if v, ok := params["cost"]; ok {
  2836. targetNbr.Cost, _ = strconv.Atoi(v)
  2837. }
  2838. if v, ok := params["disabled"]; ok {
  2839. targetNbr.Disabled = parseCLIEnableDisable(v, false)
  2840. }
  2841. if v, ok := params["comment"]; ok {
  2842. targetNbr.Comment = v
  2843. }
  2844. if err := e.dynRoutingMgr.SaveOSPFStaticNeighbor(*targetNbr); err != nil {
  2845. return ColorRed + "Error updating static neighbor: " + err.Error() + ColorReset + "\n"
  2846. }
  2847. return ColorGreen + fmt.Sprintf("OSPF static neighbor %s updated.\n", targetNbr.Address) + ColorReset
  2848. case "remove", "rem", "delete":
  2849. pos, params := parseArgs(parts[1:])
  2850. target := ""
  2851. if len(pos) > 0 {
  2852. target = pos[0]
  2853. } else if v, ok := params["numbers"]; ok {
  2854. target = v
  2855. }
  2856. if target == "" {
  2857. return ColorRed + "Usage: /routing ospf static-neighbor remove <number|address>\n" + ColorReset
  2858. }
  2859. nbrs := e.dynRoutingMgr.ListOSPFStaticNeighbors()
  2860. idToDelete := ""
  2861. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(nbrs) {
  2862. idToDelete = nbrs[idx].ID
  2863. } else {
  2864. for _, n := range nbrs {
  2865. if n.Address == target || n.ID == target {
  2866. idToDelete = n.ID
  2867. break
  2868. }
  2869. }
  2870. }
  2871. if idToDelete == "" {
  2872. return ColorRed + fmt.Sprintf("Static neighbor %q not found\n", target) + ColorReset
  2873. }
  2874. if err := e.dynRoutingMgr.DeleteOSPFStaticNeighbor(idToDelete); err != nil {
  2875. return ColorRed + "Error deleting static neighbor: " + err.Error() + ColorReset + "\n"
  2876. }
  2877. return ColorGreen + fmt.Sprintf("OSPF static neighbor %s removed.\n", target) + ColorReset
  2878. case "enable":
  2879. pos, _ := parseArgs(parts[1:])
  2880. if len(pos) == 0 {
  2881. return ColorRed + "Usage: /routing ospf static-neighbor enable <number|address>\n" + ColorReset
  2882. }
  2883. return e.setOSPFStaticNeighborStateByTarget(pos[0], false)
  2884. case "disable":
  2885. pos, _ := parseArgs(parts[1:])
  2886. if len(pos) == 0 {
  2887. return ColorRed + "Usage: /routing ospf static-neighbor disable <number|address>\n" + ColorReset
  2888. }
  2889. return e.setOSPFStaticNeighborStateByTarget(pos[0], true)
  2890. default:
  2891. return ColorRed + "Usage: /routing ospf static-neighbor [print|add|set|remove|enable|disable]\n" + ColorReset
  2892. }
  2893. }
  2894. func (e *Executor) handleOSPFNeighborLive(parts []string, user *auth.PublicUser) string {
  2895. neighbors := e.dynRoutingMgr.GetOSPFNeighbors()
  2896. tbl := NewTable("ROUTER-ID", "ADDRESS", "INTERFACE", "STATE", "PRIORITY", "DEAD-TIMER")
  2897. for _, n := range neighbors {
  2898. tbl.AddRow(n.NeighborID, n.Address, n.Interface, n.State, fmt.Sprintf("%d", n.Priority), n.DeadTimer)
  2899. }
  2900. return tbl.String()
  2901. }
  2902. func (e *Executor) handleOSPFLSA(parts []string, user *auth.PublicUser) string {
  2903. lsas := e.dynRoutingMgr.GetOSPFLSA()
  2904. tbl := NewTable("TYPE", "LS-ID", "ROUTER-ID", "AGE", "SEQUENCE")
  2905. for _, l := range lsas {
  2906. tbl.AddRow(l.Type, l.ID, l.Originator, fmt.Sprintf("%d", l.Age), l.Sequence)
  2907. }
  2908. return tbl.String()
  2909. }
  2910. func (e *Executor) handleOSPFRoute(parts []string, user *auth.PublicUser) string {
  2911. routes := e.dynRoutingMgr.GetOSPFRoutes()
  2912. tbl := NewTable("PREFIX", "NEXT-HOP", "INTERFACE", "COST", "TYPE")
  2913. for _, r := range routes {
  2914. tbl.AddRow(r.Prefix, r.NextHop, r.Interface, fmt.Sprintf("%d", r.Cost), r.RouteType)
  2915. }
  2916. return tbl.String()
  2917. }
  2918. func (e *Executor) setOSPFInstanceStateByTarget(target string, disabled bool) string {
  2919. instances := e.dynRoutingMgr.ListOSPFInstances()
  2920. id := ""
  2921. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(instances) {
  2922. id = instances[idx].ID
  2923. } else {
  2924. for _, inst := range instances {
  2925. if strings.EqualFold(inst.Name, target) || inst.ID == target {
  2926. id = inst.ID
  2927. break
  2928. }
  2929. }
  2930. }
  2931. if id == "" {
  2932. return ColorRed + fmt.Sprintf("OSPF instance %q not found\n", target) + ColorReset
  2933. }
  2934. if err := e.dynRoutingMgr.SetOSPFInstanceState(id, disabled); err != nil {
  2935. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2936. }
  2937. action := "enabled"
  2938. if disabled {
  2939. action = "disabled"
  2940. }
  2941. return ColorGreen + fmt.Sprintf("OSPF instance %s %s.\n", target, action) + ColorReset
  2942. }
  2943. func (e *Executor) setOSPFAreaStateByTarget(target string, disabled bool) string {
  2944. areas := e.dynRoutingMgr.ListOSPFAreas()
  2945. id := ""
  2946. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(areas) {
  2947. id = areas[idx].ID
  2948. } else {
  2949. for _, a := range areas {
  2950. if strings.EqualFold(a.Name, target) || a.ID == target {
  2951. id = a.ID
  2952. break
  2953. }
  2954. }
  2955. }
  2956. if id == "" {
  2957. return ColorRed + fmt.Sprintf("OSPF area %q not found\n", target) + ColorReset
  2958. }
  2959. if err := e.dynRoutingMgr.SetOSPFAreaState(id, disabled); err != nil {
  2960. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2961. }
  2962. action := "enabled"
  2963. if disabled {
  2964. action = "disabled"
  2965. }
  2966. return ColorGreen + fmt.Sprintf("OSPF area %s %s.\n", target, action) + ColorReset
  2967. }
  2968. func (e *Executor) setOSPFInterfaceStateByTarget(target string, disabled bool) string {
  2969. ifaces := e.dynRoutingMgr.ListOSPFInterfaces()
  2970. id := ""
  2971. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(ifaces) {
  2972. id = ifaces[idx].ID
  2973. } else {
  2974. for _, i := range ifaces {
  2975. if i.ID == target || i.Interfaces == target || i.Interface == target {
  2976. id = i.ID
  2977. break
  2978. }
  2979. }
  2980. }
  2981. if id == "" {
  2982. return ColorRed + fmt.Sprintf("OSPF interface template %q not found\n", target) + ColorReset
  2983. }
  2984. if err := e.dynRoutingMgr.SetOSPFInterfaceState(id, disabled); err != nil {
  2985. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  2986. }
  2987. action := "enabled"
  2988. if disabled {
  2989. action = "disabled"
  2990. }
  2991. return ColorGreen + fmt.Sprintf("OSPF interface template %s %s.\n", target, action) + ColorReset
  2992. }
  2993. func (e *Executor) setOSPFStaticNeighborStateByTarget(target string, disabled bool) string {
  2994. nbrs := e.dynRoutingMgr.ListOSPFStaticNeighbors()
  2995. id := ""
  2996. if idx, err := strconv.Atoi(target); err == nil && idx >= 0 && idx < len(nbrs) {
  2997. id = nbrs[idx].ID
  2998. } else {
  2999. for _, n := range nbrs {
  3000. if n.Address == target || n.ID == target {
  3001. id = n.ID
  3002. break
  3003. }
  3004. }
  3005. }
  3006. if id == "" {
  3007. return ColorRed + fmt.Sprintf("Static neighbor %q not found\n", target) + ColorReset
  3008. }
  3009. if err := e.dynRoutingMgr.SetOSPFStaticNeighborState(id, disabled); err != nil {
  3010. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  3011. }
  3012. action := "enabled"
  3013. if disabled {
  3014. action = "disabled"
  3015. }
  3016. return ColorGreen + fmt.Sprintf("Static neighbor %s %s.\n", target, action) + ColorReset
  3017. }
  3018. func (e *Executor) handleRoutingFilter(parts []string, user *auth.PublicUser) string {
  3019. rules := e.dynRoutingMgr.ListRoutingFilters()
  3020. tbl := NewTable("CHAIN", "RULE", "ACTION", "STATUS")
  3021. for _, r := range rules {
  3022. st := "enabled"
  3023. if r.Disabled {
  3024. st = "disabled"
  3025. }
  3026. tbl.AddRow(r.Chain, r.Rule, r.Action, st)
  3027. }
  3028. return tbl.String()
  3029. }
  3030. // ==================== TOOLS ====================
  3031. func (e *Executor) handleTool(parts []string) string {
  3032. if len(parts) == 0 {
  3033. return ColorYellow + "Tools available: ping, traceroute, bandwidth-test\n" + ColorReset
  3034. }
  3035. sub := strings.ToLower(parts[0])
  3036. switch sub {
  3037. case "ping":
  3038. return e.handlePing(parts[1:])
  3039. case "traceroute":
  3040. return e.handleTraceroute(parts[1:])
  3041. case "bandwidth-test", "btest":
  3042. return e.handleBandwidthTest(parts[1:])
  3043. default:
  3044. return ColorRed + "Unknown tool: " + sub + "\n" + ColorReset
  3045. }
  3046. }
  3047. func (e *Executor) handlePing(args []string) string {
  3048. if len(args) == 0 {
  3049. return ColorRed + "Usage: ping <host> [count=4]\n" + ColorReset
  3050. }
  3051. pos, params := parseArgs(args)
  3052. if len(pos) == 0 {
  3053. return ColorRed + "Missing host target for ping\n" + ColorReset
  3054. }
  3055. target := pos[0]
  3056. count := "4"
  3057. if c, ok := params["count"]; ok {
  3058. count = c
  3059. }
  3060. cmd := exec.Command("ping", "-c", count, target)
  3061. out, err := cmd.CombinedOutput()
  3062. if err != nil && len(out) == 0 {
  3063. return ColorRed + "Ping execution failed: " + err.Error() + ColorReset + "\n"
  3064. }
  3065. return string(out)
  3066. }
  3067. func (e *Executor) handleTraceroute(args []string) string {
  3068. if len(args) == 0 {
  3069. return ColorRed + "Usage: traceroute <host>\n" + ColorReset
  3070. }
  3071. pos, _ := parseArgs(args)
  3072. if len(pos) == 0 {
  3073. return ColorRed + "Missing host target for traceroute\n" + ColorReset
  3074. }
  3075. target := pos[0]
  3076. cmd := exec.Command("traceroute", "-n", "-m", "15", target)
  3077. out, err := cmd.CombinedOutput()
  3078. if err != nil {
  3079. cmdFallback := exec.Command("tracepath", "-n", target)
  3080. fbOut, fbErr := cmdFallback.CombinedOutput()
  3081. if fbErr == nil {
  3082. return string(fbOut)
  3083. }
  3084. return ColorRed + "Traceroute failed: " + err.Error() + "\n" + string(out) + ColorReset + "\n"
  3085. }
  3086. return string(out)
  3087. }
  3088. func (e *Executor) handleBandwidthTest(args []string) string {
  3089. if e.bwtestMgr == nil {
  3090. return ColorRed + "Bandwidth test manager not available.\n" + ColorReset
  3091. }
  3092. pos, params := parseArgs(args)
  3093. if len(pos) == 0 {
  3094. return ColorRed + "Usage: /tool bandwidth-test <target-ip> [direction=both|transmit|receive] [protocol=udp|tcp] [duration=5s]\n" + ColorReset
  3095. }
  3096. target := pos[0]
  3097. dir := "both"
  3098. if d, ok := params["direction"]; ok {
  3099. dir = d
  3100. }
  3101. switch dir {
  3102. case "transmit":
  3103. dir = "tx"
  3104. case "receive":
  3105. dir = "rx"
  3106. case "tx", "rx", "both":
  3107. default:
  3108. return cliError(fmt.Errorf("direction must be both, tx/transmit, or rx/receive"))
  3109. }
  3110. proto := "udp"
  3111. if p, ok := params["protocol"]; ok {
  3112. proto = p
  3113. }
  3114. if proto != "udp" && proto != "tcp" {
  3115. return cliError(fmt.Errorf("protocol must be udp or tcp"))
  3116. }
  3117. dur := 5 * time.Second
  3118. if dStr, ok := params["duration"]; ok {
  3119. if dVal, err := time.ParseDuration(dStr); err == nil && dVal >= time.Second {
  3120. dur = dVal
  3121. } else {
  3122. return cliError(fmt.Errorf("duration must be at least 1s"))
  3123. }
  3124. }
  3125. req := bwtest.SessionRequest{
  3126. Target: target,
  3127. Direction: dir,
  3128. Protocol: proto,
  3129. Duration: int(dur.Seconds()),
  3130. }
  3131. ctx, cancel := context.WithTimeout(context.Background(), dur+8*time.Second)
  3132. defer cancel()
  3133. if err := e.bwtestMgr.StartTest(ctx, req); err != nil {
  3134. return ColorRed + "Failed to initiate bandwidth test: " + err.Error() + ColorReset + "\n"
  3135. }
  3136. st := e.bwtestMgr.GetTestStatus()
  3137. ticker := time.NewTicker(100 * time.Millisecond)
  3138. defer ticker.Stop()
  3139. for st.Running {
  3140. select {
  3141. case <-ctx.Done():
  3142. e.bwtestMgr.StopTest()
  3143. return cliError(fmt.Errorf("bandwidth test did not finish before the deadline; stop requested"))
  3144. case <-ticker.C:
  3145. st = e.bwtestMgr.GetTestStatus()
  3146. }
  3147. }
  3148. if st.Error != "" {
  3149. return cliError(fmt.Errorf("bandwidth test failed: %s", st.Error))
  3150. }
  3151. out := fmt.Sprintf("Bandwidth test to %s completed:\n", target)
  3152. if st.Summary != nil {
  3153. out += fmt.Sprintf(" Tx Avg : %.2f Mbps\n", st.Summary.AvgTxMbps)
  3154. out += fmt.Sprintf(" Rx Avg : %.2f Mbps\n", st.Summary.AvgRxMbps)
  3155. out += fmt.Sprintf(" Loss : %.1f%%\n", st.Summary.AvgLossPct)
  3156. out += fmt.Sprintf(" Jitter : %.2f ms\n", st.Summary.AvgJitterMs)
  3157. } else if st.CurrentSample != nil {
  3158. out += fmt.Sprintf(" Tx Rate : %.2f Mbps\n", st.CurrentSample.TxMbps)
  3159. out += fmt.Sprintf(" Rx Rate : %.2f Mbps\n", st.CurrentSample.RxMbps)
  3160. } else {
  3161. out += " Test finished.\n"
  3162. }
  3163. return out
  3164. }
  3165. // ==================== USER MANAGEMENT ====================
  3166. func (e *Executor) handleUser(parts []string, actor *auth.PublicUser) string {
  3167. if e.authMgr == nil {
  3168. return ColorRed + "Auth manager not available.\n" + ColorReset
  3169. }
  3170. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  3171. users := e.authMgr.ListUsers()
  3172. tbl := NewTable("FLAGS", "NAME", "ROLE", "LAST-LOGIN", "COMMENT")
  3173. for _, u := range users {
  3174. flags := ""
  3175. if u.Disabled {
  3176. flags += "X"
  3177. } else {
  3178. flags += " "
  3179. }
  3180. lastLogin := "never"
  3181. if !u.LastLoginAt.IsZero() {
  3182. lastLogin = u.LastLoginAt.Format("Jan/02 15:04:05")
  3183. }
  3184. tbl.AddRow(flags, u.Username, string(u.Role), lastLogin, u.Comment)
  3185. }
  3186. return tbl.String()
  3187. }
  3188. if actor.Role != auth.RoleFull {
  3189. return ColorRed + "Permission denied: admin role required to manage users\n" + ColorReset
  3190. }
  3191. sub := strings.ToLower(parts[0])
  3192. switch sub {
  3193. case "add":
  3194. _, params := parseArgs(parts[1:])
  3195. name := params["name"]
  3196. pwd := params["password"]
  3197. roleStr := params["role"]
  3198. if roleStr == "" {
  3199. roleStr = "admin"
  3200. }
  3201. comment := params["comment"]
  3202. if name == "" || pwd == "" {
  3203. return ColorRed + "Usage: /user add name=<username> password=<pwd> [role=admin|operator|viewer] [comment=...]\n" + ColorReset
  3204. }
  3205. req := auth.CreateUserRequest{
  3206. Username: name,
  3207. Password: pwd,
  3208. Role: auth.Role(roleStr),
  3209. Comment: comment,
  3210. }
  3211. if err := e.authMgr.CreateUser(req); err != nil {
  3212. return ColorRed + "Error creating user: " + err.Error() + ColorReset + "\n"
  3213. }
  3214. return ColorGreen + fmt.Sprintf("User %s created.\n", name) + ColorReset
  3215. case "remove", "rem", "delete":
  3216. pos, _ := parseArgs(parts[1:])
  3217. if len(pos) == 0 {
  3218. return ColorRed + "Usage: /user remove <username>\n" + ColorReset
  3219. }
  3220. name := pos[0]
  3221. if err := e.authMgr.DeleteUser(name, actor.Username); err != nil {
  3222. return ColorRed + "Error removing user: " + err.Error() + ColorReset + "\n"
  3223. }
  3224. return ColorGreen + fmt.Sprintf("User %s removed.\n", name) + ColorReset
  3225. case "set":
  3226. pos, params := parseArgs(parts[1:])
  3227. if len(pos) == 0 {
  3228. return ColorRed + "Usage: /user set <username> [password=...] [role=...] [disabled=yes|no]\n" + ColorReset
  3229. }
  3230. name := pos[0]
  3231. pub, err := e.authMgr.GetPublicUser(name)
  3232. if err != nil || pub == nil {
  3233. return ColorRed + "User not found\n" + ColorReset
  3234. }
  3235. role := pub.Role
  3236. if rStr, ok := params["role"]; ok {
  3237. role = auth.Role(rStr)
  3238. }
  3239. dis := pub.Disabled
  3240. if dStr, ok := params["disabled"]; ok {
  3241. dis = (strings.ToLower(dStr) == "yes" || strings.ToLower(dStr) == "true")
  3242. }
  3243. cmt := pub.Comment
  3244. if c, ok := params["comment"]; ok {
  3245. cmt = c
  3246. }
  3247. req := auth.UpdateUserRequest{
  3248. Role: role,
  3249. Disabled: dis,
  3250. Comment: cmt,
  3251. }
  3252. if err := e.authMgr.UpdateUser(name, req); err != nil {
  3253. return ColorRed + "Error updating user: " + err.Error() + ColorReset + "\n"
  3254. }
  3255. if pwd, ok := params["password"]; ok && pwd != "" {
  3256. if err := e.authMgr.ChangePassword(name, "", pwd, true); err != nil {
  3257. return ColorRed + "Error resetting password: " + err.Error() + ColorReset + "\n"
  3258. }
  3259. }
  3260. return ColorGreen + fmt.Sprintf("User %s updated.\n", name) + ColorReset
  3261. default:
  3262. return ColorRed + "Usage: /user [print|add|remove|set]\n" + ColorReset
  3263. }
  3264. }
  3265. // ==================== SYSTEM ====================
  3266. func (e *Executor) handleSystem(parts []string, user *auth.PublicUser) string {
  3267. if len(parts) == 0 {
  3268. return ColorYellow + "System commands: resource print, reboot, identity print, identity set name=...\n" + ColorReset
  3269. }
  3270. sub := strings.ToLower(parts[0])
  3271. switch sub {
  3272. case "resource":
  3273. st, err := system.GetStatus()
  3274. if err != nil {
  3275. return ColorRed + "Error retrieving system resources: " + err.Error() + ColorReset + "\n"
  3276. }
  3277. freeMemMB := uint64(0)
  3278. if st.MemTotalMB > st.MemUsedMB {
  3279. freeMemMB = st.MemTotalMB - st.MemUsedMB
  3280. }
  3281. out := " uptime: " + formatUptime(int64(st.UptimeSeconds)) + "\n"
  3282. out += fmt.Sprintf(" version: DebTik 1.0 (Kernel %s)\n", st.Kernel)
  3283. out += fmt.Sprintf(" free-memory: %d MiB\n", freeMemMB)
  3284. out += fmt.Sprintf(" total-memory: %d MiB\n", st.MemTotalMB)
  3285. out += fmt.Sprintf(" cpu-count: %d\n", st.CPUCores)
  3286. out += fmt.Sprintf(" cpu-load: %.1f%%\n", st.CPUPct)
  3287. out += fmt.Sprintf(" os: %s\n", st.OS)
  3288. return out
  3289. case "identity":
  3290. if len(parts) > 1 && parts[1] == "set" {
  3291. if user.Role != auth.RoleFull {
  3292. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  3293. }
  3294. _, params := parseArgs(parts[2:])
  3295. name := params["name"]
  3296. if name == "" {
  3297. return ColorRed + "Usage: /system identity set name=<new-name>\n" + ColorReset
  3298. }
  3299. if err := exec.Command("hostnamectl", "set-hostname", name).Run(); err != nil {
  3300. return ColorRed + "Error setting identity: " + err.Error() + ColorReset + "\n"
  3301. }
  3302. return ColorGreen + fmt.Sprintf("System identity set to %q.\n", name) + ColorReset
  3303. }
  3304. st, _ := system.GetStatus()
  3305. h := "DebTik"
  3306. if st != nil && st.Hostname != "" {
  3307. h = st.Hostname
  3308. }
  3309. return fmt.Sprintf(" name: %s\n", h)
  3310. case "reboot":
  3311. if user.Role != auth.RoleFull {
  3312. return ColorRed + "Permission denied: admin role required to reboot\n" + ColorReset
  3313. }
  3314. go func() {
  3315. time.Sleep(1 * time.Second)
  3316. _ = exec.Command("systemctl", "reboot").Run()
  3317. }()
  3318. return ColorYellow + "System reboot initiated! Connection will terminate.\n" + ColorReset
  3319. case "backup":
  3320. if e.backupMgr == nil {
  3321. e.backupMgr = backup.NewManager("", "", "", nil)
  3322. }
  3323. if len(parts) < 2 {
  3324. return ColorYellow + "Usage: /system backup [save [name=...] [comment=...]] | [load name=...] | [print]\n" + ColorReset
  3325. }
  3326. action := parts[1]
  3327. switch action {
  3328. case "save":
  3329. if user.Role != auth.RoleFull {
  3330. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  3331. }
  3332. _, params := parseArgs(parts[2:])
  3333. name := params["name"]
  3334. comment := params["comment"]
  3335. savedPath, manifest, err := e.backupMgr.SaveBackup(name, comment)
  3336. if err != nil {
  3337. return ColorRed + "Error creating backup: " + err.Error() + ColorReset + "\n"
  3338. }
  3339. fi, _ := os.Stat(savedPath)
  3340. size := int64(0)
  3341. if fi != nil {
  3342. size = fi.Size()
  3343. }
  3344. 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
  3345. case "print":
  3346. backups, err := e.backupMgr.ListLocalBackups()
  3347. if err != nil {
  3348. return ColorRed + "Error listing backups: " + err.Error() + ColorReset + "\n"
  3349. }
  3350. if len(backups) == 0 {
  3351. return "No backup archives found.\n"
  3352. }
  3353. b := strings.Builder{}
  3354. b.WriteString(ColorCyan + fmt.Sprintf("%-3s %-42s %-10s %-20s %s\n", "#", "NAME", "SIZE", "CREATED", "COMMENT") + ColorReset)
  3355. for i, bk := range backups {
  3356. sizeStr := fmt.Sprintf("%d B", bk.Size)
  3357. if bk.Size >= 1024*1024 {
  3358. sizeStr = fmt.Sprintf("%.1f MB", float64(bk.Size)/(1024*1024))
  3359. } else if bk.Size >= 1024 {
  3360. sizeStr = fmt.Sprintf("%.1f KB", float64(bk.Size)/1024)
  3361. }
  3362. b.WriteString(fmt.Sprintf("%-3d %-42s %-10s %-20s %s\n",
  3363. i,
  3364. bk.Filename,
  3365. sizeStr,
  3366. bk.ModTime.Format("2006-01-02 15:04:05"),
  3367. bk.Comment,
  3368. ))
  3369. }
  3370. return b.String()
  3371. case "load":
  3372. if user.Role != auth.RoleFull {
  3373. return ColorRed + "Permission denied: admin role required to restore configuration\n" + ColorReset
  3374. }
  3375. _, params := parseArgs(parts[2:])
  3376. name := params["name"]
  3377. if name == "" {
  3378. return ColorRed + "Usage: /system backup load name=<backup-filename.zip>\n" + ColorReset
  3379. }
  3380. fullPath, err := e.backupMgr.GetLocalBackup(name)
  3381. if err != nil {
  3382. return ColorRed + "Backup file not found: " + err.Error() + ColorReset + "\n"
  3383. }
  3384. f, err := os.Open(fullPath)
  3385. if err != nil {
  3386. return ColorRed + "Cannot read backup file: " + err.Error() + ColorReset + "\n"
  3387. }
  3388. defer f.Close()
  3389. fi, err := f.Stat()
  3390. if err != nil {
  3391. return ColorRed + "Cannot stat backup file: " + err.Error() + ColorReset + "\n"
  3392. }
  3393. res, err := e.backupMgr.RestoreBackup(f, fi.Size())
  3394. if err != nil {
  3395. return ColorRed + "Restore failed: " + err.Error() + ColorReset + "\n"
  3396. }
  3397. e.backupMgr.TriggerRestart(1 * time.Second)
  3398. 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
  3399. default:
  3400. return ColorRed + "Unknown backup action: " + action + ". Use save, load, or print.\n" + ColorReset
  3401. }
  3402. default:
  3403. return ColorRed + "Unknown system subcommand: " + sub + "\n" + ColorReset
  3404. }
  3405. }
  3406. func (e *Executor) showHelp() string {
  3407. b := strings.Builder{}
  3408. b.WriteString(ColorCyan + ColorBold + "DebTik CLI Command Directory:\n" + ColorReset)
  3409. b.WriteString(" /interface - Network interface manager (print, enable, disable, set)\n")
  3410. b.WriteString(" /interface vlan - VLAN virtual interfaces (print, add, remove)\n")
  3411. b.WriteString(" /interface bridge - L2 Bridge and ports (print, add, remove, port)\n")
  3412. b.WriteString(" /interface aerotun - Encrypted tunnels (print detail=yes, add, set, enable, disable, start, stop, remove, genkey)\n")
  3413. b.WriteString(" /interface wireguard - WireGuard tunnels & peers (print, add, peers)\n")
  3414. b.WriteString(" /interface eoip - Ethernet over IP GRE tunnels (print, add, remove)\n")
  3415. b.WriteString(" /interface vxlan - VXLAN overlay tunnels & VTEPs (print, add, set, remove, enable, disable, vteps)\n")
  3416. b.WriteString(" /interface vrrp - Virtual Router Redundancy Protocol (print, add, remove)\n")
  3417. b.WriteString(" /interface pppoe-client - PPPoE WAN Client sessions (print, add, connect)\n")
  3418. b.WriteString(" /ip\n")
  3419. b.WriteString(" /ip address - IP address assignments (print, add, remove)\n")
  3420. b.WriteString(" /ip route - Static routes & tables (print, add, remove, rule)\n")
  3421. b.WriteString(" /ip dhcp-client - DHCP client on WAN links (print, add, renew, release)\n")
  3422. b.WriteString(" /ip dhcp-server - Local DHCP server & leases (print, add, lease, network)\n")
  3423. b.WriteString(" /ip dns - Router DNS resolver (print, set, cache flush)\n")
  3424. b.WriteString(" /ip firewall - Netfilter / iptables firewall\n")
  3425. b.WriteString(" /ip firewall filter - Traffic filtering rules (print, add, remove)\n")
  3426. b.WriteString(" /ip firewall nat - Source & destination NAT rules (print, add, remove)\n")
  3427. b.WriteString(" /ip firewall mangle - Packet marking and QoS (print, add, remove)\n")
  3428. b.WriteString(" /ip firewall address-list - Dynamic & static address groups (print, add, remove)\n")
  3429. b.WriteString(" /ip service - Management services: Telnet, SSH, Web (print, enable, disable, set)\n")
  3430. b.WriteString(" /ip reverse-proxy - HTTP/HTTPS Reverse Proxy & Path Router (print, add, set, remove, enable, disable)\n")
  3431. b.WriteString(" /ip sdwan - Software-Defined WAN multi-link aggregation (print, status, add, set, remove, enable, disable, member)\n")
  3432. b.WriteString(" /routing - Dynamic routing engine\n")
  3433. b.WriteString(" /routing bgp - BGP instance, peers, routes (instance, peer, routes)\n")
  3434. b.WriteString(" /routing ospf - OSPF areas, interfaces, neighbors (instance, area, neighbor)\n")
  3435. b.WriteString(" /routing filter - Route filtering policies (print)\n")
  3436. b.WriteString(" /tool - Diagnostics & performance testing\n")
  3437. b.WriteString(" /tool ping <host> - ICMP ping test\n")
  3438. b.WriteString(" /tool traceroute <host> - Route trace analysis\n")
  3439. b.WriteString(" /tool bandwidth-test <host> - High performance throughput & jitter test\n")
  3440. b.WriteString(" /log - System activity logs (print [limit=N] [where=filter])\n")
  3441. b.WriteString(" /user - Local user accounts (print, add, remove, set)\n")
  3442. b.WriteString(" /system - Router hardware and core controls\n")
  3443. b.WriteString(" /system resource print - CPU, RAM, disk, uptime metrics\n")
  3444. b.WriteString(" /system identity - Hostname identity (print, set name=...)\n")
  3445. b.WriteString(" /system backup - Configuration backups (save, load, print)\n")
  3446. b.WriteString(" /system reboot - Safe system restart\n")
  3447. b.WriteString(" quit / exit - Disconnect session\n")
  3448. return b.String()
  3449. }
  3450. // ==================== CERTIFICATES (/certificate) ====================
  3451. func (e *Executor) handleCertificate(parts []string, user *auth.PublicUser) string {
  3452. if e.certMgr == nil {
  3453. return ColorRed + "Certificates manager not available.\n" + ColorReset
  3454. }
  3455. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  3456. list := e.certMgr.ListCertificates()
  3457. tbl := NewTable("#", "NAME", "COMMON-NAME", "TYPE", "ISSUER", "EXPIRES", "SANS")
  3458. for idx, c := range list {
  3459. tStr := ColorGreen + c.Type + ColorReset
  3460. if c.IsCA {
  3461. tStr = ColorCyan + "ca" + ColorReset
  3462. }
  3463. expStr := c.ValidTo.Format("Jan/02/2006")
  3464. if time.Now().After(c.ValidTo) {
  3465. expStr = ColorRed + expStr + " (EXPIRED)" + ColorReset
  3466. }
  3467. sansStr := strings.Join(c.SANs, ",")
  3468. if sansStr == "" {
  3469. sansStr = "-"
  3470. }
  3471. tbl.AddRow(fmt.Sprintf("%d", idx), c.Name, c.CommonName, tStr, c.Issuer, expStr, sansStr)
  3472. }
  3473. return tbl.String()
  3474. }
  3475. if user.Role != auth.RoleFull {
  3476. return ColorRed + "Permission denied: admin role required to manage certificates\n" + ColorReset
  3477. }
  3478. sub := strings.ToLower(parts[0])
  3479. switch sub {
  3480. case "add-ca":
  3481. _, params := parseArgs(parts[1:])
  3482. name := params["name"]
  3483. cn := params["common-name"]
  3484. if name == "" || cn == "" {
  3485. return ColorRed + "Usage: /certificate add-ca name=<name> common-name=<cn> [days=<days>] [key-type=RSA-2048|RSA-4096|ECDSA-P256]\n" + ColorReset
  3486. }
  3487. days := 3650
  3488. if dStr, ok := params["days"]; ok {
  3489. if dVal, err := strconv.Atoi(dStr); err == nil && dVal > 0 {
  3490. days = dVal
  3491. }
  3492. }
  3493. keyType := params["key-type"]
  3494. if keyType == "" {
  3495. keyType = "RSA-2048"
  3496. }
  3497. info, err := e.certMgr.CreateCA(certs.CreateCARequest{
  3498. Name: name,
  3499. CommonName: cn,
  3500. Days: days,
  3501. KeyType: keyType,
  3502. Comment: params["comment"],
  3503. })
  3504. if err != nil {
  3505. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  3506. }
  3507. return ColorGreen + fmt.Sprintf("Certificate Authority %q created successfully.\n", info.Name) + ColorReset
  3508. case "add":
  3509. _, params := parseArgs(parts[1:])
  3510. name := params["name"]
  3511. cn := params["common-name"]
  3512. if name == "" || cn == "" {
  3513. return ColorRed + "Usage: /certificate add name=<name> common-name=<cn> [ca=<ca_name>] [sans=<san1,san2>] [days=<days>] [key-type=...]\n" + ColorReset
  3514. }
  3515. days := 365
  3516. if dStr, ok := params["days"]; ok {
  3517. if dVal, err := strconv.Atoi(dStr); err == nil && dVal > 0 {
  3518. days = dVal
  3519. }
  3520. }
  3521. var sans []string
  3522. if sStr, ok := params["sans"]; ok {
  3523. for _, s := range strings.Split(sStr, ",") {
  3524. if s = strings.TrimSpace(s); s != "" {
  3525. sans = append(sans, s)
  3526. }
  3527. }
  3528. }
  3529. keyType := params["key-type"]
  3530. if keyType == "" {
  3531. keyType = "RSA-2048"
  3532. }
  3533. info, err := e.certMgr.CreateCertificate(certs.CreateCertRequest{
  3534. Name: name,
  3535. CommonName: cn,
  3536. CAName: params["ca"],
  3537. SANs: sans,
  3538. Days: days,
  3539. KeyType: keyType,
  3540. IsServer: true,
  3541. Comment: params["comment"],
  3542. })
  3543. if err != nil {
  3544. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  3545. }
  3546. return ColorGreen + fmt.Sprintf("Certificate %q created successfully (Issuer: %s).\n", info.Name, info.Issuer) + ColorReset
  3547. case "issue-letsencrypt":
  3548. _, params := parseArgs(parts[1:])
  3549. domain := params["domain"]
  3550. email := params["email"]
  3551. if domain == "" || email == "" {
  3552. return ColorRed + "Usage: /certificate issue-letsencrypt domain=<domain> email=<email> [name=<name>] [staging=yes|no]\n" + ColorReset
  3553. }
  3554. name := params["name"]
  3555. if name == "" {
  3556. name = domain
  3557. }
  3558. staging := strings.ToLower(params["staging"]) == "yes" || strings.ToLower(params["staging"]) == "true"
  3559. ctx, cancel := context.WithTimeout(context.Background(), 3*time.Minute)
  3560. defer cancel()
  3561. info, err := e.certMgr.IssueLetsEncrypt(ctx, certs.LetsEncryptRequest{
  3562. Name: name,
  3563. Domain: domain,
  3564. Email: email,
  3565. Staging: staging,
  3566. Comment: params["comment"],
  3567. })
  3568. if err != nil {
  3569. return ColorRed + "Let's Encrypt Error: " + err.Error() + ColorReset + "\n"
  3570. }
  3571. return ColorGreen + fmt.Sprintf("Let's Encrypt certificate %q issued successfully!\n", info.Name) + ColorReset
  3572. case "export":
  3573. pos, _ := parseArgs(parts[1:])
  3574. if len(pos) == 0 {
  3575. return ColorRed + "Usage: /certificate export <name>\n" + ColorReset
  3576. }
  3577. certPEM, keyPEM, err := e.certMgr.ExportPEM(pos[0])
  3578. if err != nil {
  3579. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  3580. }
  3581. return fmt.Sprintf("----- CERTIFICATE -----\n%s\n----- PRIVATE KEY -----\n%s\n", certPEM, keyPEM)
  3582. case "remove", "rem", "delete":
  3583. pos, _ := parseArgs(parts[1:])
  3584. if len(pos) == 0 {
  3585. return ColorRed + "Usage: /certificate remove <name>\n" + ColorReset
  3586. }
  3587. name := pos[0]
  3588. if e.servicesMgr != nil {
  3589. if sslSvc, ok := e.servicesMgr.GetService("www-ssl"); ok {
  3590. if strings.EqualFold(sslSvc.Certificate, name) && !sslSvc.Disabled {
  3591. return ColorRed + "Cannot remove certificate: actively used by enabled www-ssl service\n" + ColorReset
  3592. }
  3593. }
  3594. }
  3595. if err := e.certMgr.DeleteCertificate(name); err != nil {
  3596. return ColorRed + "Error: " + err.Error() + ColorReset + "\n"
  3597. }
  3598. return ColorGreen + fmt.Sprintf("Certificate %s removed.\n", name) + ColorReset
  3599. default:
  3600. return ColorRed + "Usage: /certificate [print|add-ca|add|issue-letsencrypt|export|remove]\n" + ColorReset
  3601. }
  3602. }
  3603. func formatUptime(seconds int64) string {
  3604. d := seconds / 86400
  3605. h := (seconds % 86400) / 3600
  3606. m := (seconds % 3600) / 60
  3607. s := seconds % 60
  3608. if d > 0 {
  3609. return fmt.Sprintf("%dd %02dh %02dm %02ds", d, h, m, s)
  3610. }
  3611. return fmt.Sprintf("%02dh %02dm %02ds", h, m, s)
  3612. }
  3613. func formatBytes(bytes int64) string {
  3614. const unit = 1024
  3615. if bytes < unit {
  3616. return fmt.Sprintf("%d B", bytes)
  3617. }
  3618. div, exp := int64(unit), 0
  3619. for n := bytes / unit; n >= unit; n /= unit {
  3620. div *= unit
  3621. exp++
  3622. }
  3623. return fmt.Sprintf("%.1f %ciB", float64(bytes)/float64(div), "KMGTPE"[exp])
  3624. }
  3625. func (e *Executor) handleIPReverseProxy(parts []string, user *auth.PublicUser) string {
  3626. if e.proxyMgr == nil {
  3627. return ColorRed + "Reverse proxy service not available\n" + ColorReset
  3628. }
  3629. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" {
  3630. rules := e.proxyMgr.ListRules()
  3631. cfg := e.proxyMgr.GetConfig()
  3632. var out strings.Builder
  3633. stateStr := ColorGreen + "enabled" + ColorReset
  3634. if !cfg.Enabled {
  3635. stateStr = ColorRed + "disabled" + ColorReset
  3636. }
  3637. out.WriteString(fmt.Sprintf("Flags: X - disabled\nReverse Proxy Master: %s | HTTP Port: %d | HTTPS Port: %d\n", stateStr, cfg.HTTPPort, cfg.HTTPSPort))
  3638. tbl := NewTable("Flags", "Name", "Domains", "Path", "Target", "Proto", "SSL Cert", "Requests")
  3639. for _, r := range rules {
  3640. flag := " "
  3641. if !r.Enabled {
  3642. flag = "X"
  3643. }
  3644. doms := strings.Join(r.Domains, ",")
  3645. if doms == "" {
  3646. doms = "*"
  3647. }
  3648. cert := r.Certificate
  3649. if cert == "" {
  3650. cert = "none"
  3651. }
  3652. reqCount := fmt.Sprintf("%d", r.Stats.Requests)
  3653. tbl.AddRow(flag, r.Name, doms, r.Path, r.TargetURL, r.Protocol, cert, reqCount)
  3654. }
  3655. out.WriteString(tbl.String())
  3656. return out.String()
  3657. }
  3658. if parts[0] == "status" {
  3659. cfg := e.proxyMgr.GetConfig()
  3660. var totalReq uint64
  3661. for _, r := range cfg.Rules {
  3662. totalReq += r.Stats.Requests
  3663. }
  3664. 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",
  3665. cfg.Enabled, cfg.HTTPPort, cfg.HTTPSPort, len(cfg.Rules), totalReq, cfg.DefaultCert)
  3666. }
  3667. if user.Role != auth.RoleFull {
  3668. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  3669. }
  3670. sub := strings.ToLower(parts[0])
  3671. switch sub {
  3672. case "add":
  3673. _, params := parseArgs(parts[1:])
  3674. name := params["name"]
  3675. target := params["target"]
  3676. if target == "" {
  3677. target = params["target-url"]
  3678. }
  3679. if name == "" || target == "" {
  3680. 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
  3681. }
  3682. var domains []string
  3683. if dStr, ok := params["domains"]; ok {
  3684. for _, d := range strings.Split(dStr, ",") {
  3685. if d = strings.TrimSpace(d); d != "" {
  3686. domains = append(domains, d)
  3687. }
  3688. }
  3689. }
  3690. pfx := params["path"]
  3691. if pfx == "" {
  3692. pfx = "/"
  3693. }
  3694. proto := params["protocol"]
  3695. if proto == "" {
  3696. proto = "both"
  3697. }
  3698. stripPath := (params["strip-path"] == "yes" || params["strip-path"] == "true")
  3699. httpToHttps := (params["http-to-https"] == "yes" || params["http-to-https"] == "true")
  3700. insecureTLS := (params["insecure-tls"] == "yes" || params["insecure-tls"] == "true")
  3701. rule, err := e.proxyMgr.SaveRule(&reverseproxy.Rule{
  3702. Name: name,
  3703. Enabled: true,
  3704. Domains: domains,
  3705. Path: pfx,
  3706. TargetURL: target,
  3707. StripPath: stripPath,
  3708. Protocol: proto,
  3709. Certificate: params["certificate"],
  3710. HTTPToHTTPS: httpToHttps,
  3711. InsecureTLS: insecureTLS,
  3712. Comment: params["comment"],
  3713. })
  3714. if err != nil {
  3715. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  3716. }
  3717. return ColorGreen + fmt.Sprintf("Proxy rule %q (ID: %s) added successfully.\n", rule.Name, rule.ID) + ColorReset
  3718. case "remove", "delete":
  3719. if len(parts) < 2 {
  3720. return ColorRed + "Usage: /ip reverse-proxy remove <id>\n" + ColorReset
  3721. }
  3722. id := parts[1]
  3723. if err := e.proxyMgr.DeleteRule(id); err != nil {
  3724. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  3725. }
  3726. return ColorGreen + fmt.Sprintf("Rule %q removed.\n", id) + ColorReset
  3727. case "enable":
  3728. if len(parts) < 2 {
  3729. return ColorRed + "Usage: /ip reverse-proxy enable <id>\n" + ColorReset
  3730. }
  3731. id := parts[1]
  3732. if err := e.proxyMgr.SetRuleState(id, true); err != nil {
  3733. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  3734. }
  3735. return ColorGreen + fmt.Sprintf("Rule %q enabled.\n", id) + ColorReset
  3736. case "disable":
  3737. if len(parts) < 2 {
  3738. return ColorRed + "Usage: /ip reverse-proxy disable <id>\n" + ColorReset
  3739. }
  3740. id := parts[1]
  3741. if err := e.proxyMgr.SetRuleState(id, false); err != nil {
  3742. return ColorRed + "Error: " + err.Error() + "\n" + ColorReset
  3743. }
  3744. return ColorGreen + fmt.Sprintf("Rule %q disabled.\n", id) + ColorReset
  3745. default:
  3746. return ColorRed + "Usage: /ip reverse-proxy [print|status|add|set|remove|enable|disable]\n" + ColorReset
  3747. }
  3748. }
  3749. // ==================== SD-WAN ====================
  3750. func (e *Executor) handleIPSDWAN(parts []string, user *auth.PublicUser) string {
  3751. if e.sdwanMgr == nil {
  3752. return ColorRed + "SD-WAN service not available\n" + ColorReset
  3753. }
  3754. if len(parts) > 0 && (strings.ToLower(parts[0]) == "member" || strings.ToLower(parts[0]) == "members") {
  3755. return e.handleSDWANMember(parts[1:], user)
  3756. }
  3757. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" || parts[0] == "show" {
  3758. groups := e.sdwanMgr.List()
  3759. var out strings.Builder
  3760. telemetry := e.sdwanMgr.GetTelemetrySummary()
  3761. out.WriteString(fmt.Sprintf("SD-WAN Groups: %d active / %d total | Members: %d healthy / %d total\n",
  3762. telemetry.ActiveGroups, telemetry.TotalGroups, telemetry.HealthyMembers, telemetry.TotalMembers))
  3763. tbl := NewTable("Flags", "Name", "Mode", "Members", "Probe(s)", "Auto-NAT", "Comment")
  3764. for _, g := range groups {
  3765. flag := " "
  3766. if g.Disabled {
  3767. flag = "X"
  3768. } else if g.Running {
  3769. flag = "R"
  3770. }
  3771. autoNat := "no"
  3772. if g.AutoNAT {
  3773. autoNat = "yes"
  3774. }
  3775. tbl.AddRow(flag, g.Name, string(g.Mode), fmt.Sprintf("%d", len(g.Members)), fmt.Sprintf("%ds", g.ProbeInterval), autoNat, g.Comment)
  3776. }
  3777. out.WriteString(tbl.String())
  3778. return out.String()
  3779. }
  3780. if parts[0] == "status" {
  3781. t := e.sdwanMgr.GetTelemetrySummary()
  3782. var out strings.Builder
  3783. out.WriteString(ColorCyan + ColorBold + "DebTik SD-WAN Telemetry Status:\n" + ColorReset)
  3784. out.WriteString(fmt.Sprintf(" Total Groups: %d\n", t.TotalGroups))
  3785. out.WriteString(fmt.Sprintf(" Active Groups: %d\n", t.ActiveGroups))
  3786. out.WriteString(fmt.Sprintf(" Total Members: %d\n", t.TotalMembers))
  3787. out.WriteString(fmt.Sprintf(" Healthy Links: %d\n", t.HealthyMembers))
  3788. out.WriteString(fmt.Sprintf(" Degraded Links: %d\n", t.DegradedMembers))
  3789. out.WriteString(fmt.Sprintf(" Down Links: %d\n\n", t.DownMembers))
  3790. groups := e.sdwanMgr.List()
  3791. for _, g := range groups {
  3792. out.WriteString(fmt.Sprintf("Group: %s [%s] (Members: %d)\n", g.Name, g.Mode, len(g.Members)))
  3793. for _, m := range g.Members {
  3794. statusCol := ColorGreen + string(m.Status) + ColorReset
  3795. if m.Status == sdwan.StatusDegraded {
  3796. statusCol = ColorYellow + string(m.Status) + ColorReset
  3797. } else if m.Status == sdwan.StatusDown {
  3798. statusCol = ColorRed + string(m.Status) + ColorReset
  3799. } else if m.Status == sdwan.StatusDisabled {
  3800. statusCol = ColorDim + string(m.Status) + ColorReset
  3801. }
  3802. out.WriteString(fmt.Sprintf(" - %s (GW: %s) -> Status: %s | Latency: %.1fms | Loss: %.1f%% | Jitter: %.1fms | Table: %d\n",
  3803. m.Interface, m.Gateway, statusCol, m.LiveLatency, m.PacketLoss, m.Jitter, m.TableID))
  3804. }
  3805. }
  3806. return out.String()
  3807. }
  3808. if user != nil && user.Role != auth.RoleFull {
  3809. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  3810. }
  3811. sub := strings.ToLower(parts[0])
  3812. switch sub {
  3813. case "add":
  3814. _, params := parseArgs(parts[1:])
  3815. name := params["name"]
  3816. if name == "" {
  3817. return ColorRed + "Usage: /ip sdwan add name=<name> [mode=weighted|pcc|lowest-latency|lowest-loss|failover] [auto-nat=yes|no] [probe-interval=3] [comment=\"...\"]\n" + ColorReset
  3818. }
  3819. mode := sdwan.SDWANMode(params["mode"])
  3820. if mode == "" {
  3821. mode = sdwan.ModeWeighted
  3822. }
  3823. autoNat := (params["auto-nat"] == "yes" || params["auto-nat"] == "true")
  3824. probeInt := 3
  3825. if pStr, ok := params["probe-interval"]; ok {
  3826. if v, err := strconv.Atoi(pStr); err == nil && v > 0 {
  3827. probeInt = v
  3828. }
  3829. }
  3830. group := &sdwan.SDWANGroup{
  3831. Name: name,
  3832. Mode: mode,
  3833. HashFields: params["hash-fields"],
  3834. AutoNAT: autoNat,
  3835. ProbeInterval: probeInt,
  3836. Comment: params["comment"],
  3837. Disabled: (params["disabled"] == "yes" || params["disabled"] == "true"),
  3838. }
  3839. if err := e.sdwanMgr.Save(group); err != nil {
  3840. return ColorRed + "Error adding SD-WAN group: " + err.Error() + "\n" + ColorReset
  3841. }
  3842. return ColorGreen + fmt.Sprintf("SD-WAN group %q created.\n", name) + ColorReset
  3843. case "set":
  3844. pos, params := parseArgs(parts[1:])
  3845. if len(pos) == 0 && params["name"] == "" && params["id"] == "" {
  3846. return ColorRed + "Usage: /ip sdwan set <id|name> [mode=...] [auto-nat=yes|no] [probe-interval=N] [comment=...]\n" + ColorReset
  3847. }
  3848. idOrName := ""
  3849. if len(pos) > 0 {
  3850. idOrName = pos[0]
  3851. } else if params["name"] != "" {
  3852. idOrName = params["name"]
  3853. } else {
  3854. idOrName = params["id"]
  3855. }
  3856. group, found := e.sdwanMgr.Get(idOrName)
  3857. if !found {
  3858. return ColorRed + fmt.Sprintf("SD-WAN group %q not found\n", idOrName) + ColorReset
  3859. }
  3860. if m, ok := params["mode"]; ok {
  3861. group.Mode = sdwan.SDWANMode(m)
  3862. }
  3863. if hf, ok := params["hash-fields"]; ok {
  3864. group.HashFields = hf
  3865. }
  3866. if an, ok := params["auto-nat"]; ok {
  3867. group.AutoNAT = (an == "yes" || an == "true")
  3868. }
  3869. if pi, ok := params["probe-interval"]; ok {
  3870. if v, err := strconv.Atoi(pi); err == nil && v > 0 {
  3871. group.ProbeInterval = v
  3872. }
  3873. }
  3874. if c, ok := params["comment"]; ok {
  3875. group.Comment = c
  3876. }
  3877. if err := e.sdwanMgr.Save(group); err != nil {
  3878. return ColorRed + "Error updating SD-WAN group: " + err.Error() + "\n" + ColorReset
  3879. }
  3880. return ColorGreen + fmt.Sprintf("SD-WAN group %q updated.\n", group.Name) + ColorReset
  3881. case "remove", "delete", "del":
  3882. if len(parts) < 2 {
  3883. return ColorRed + "Usage: /ip sdwan remove <id|name>\n" + ColorReset
  3884. }
  3885. idOrName := parts[1]
  3886. if err := e.sdwanMgr.Delete(idOrName); err != nil {
  3887. return ColorRed + "Error deleting SD-WAN group: " + err.Error() + "\n" + ColorReset
  3888. }
  3889. return ColorGreen + fmt.Sprintf("SD-WAN group %q removed.\n", idOrName) + ColorReset
  3890. case "enable":
  3891. if len(parts) < 2 {
  3892. return ColorRed + "Usage: /ip sdwan enable <id|name>\n" + ColorReset
  3893. }
  3894. idOrName := parts[1]
  3895. if err := e.sdwanMgr.SetState(idOrName, false); err != nil {
  3896. return ColorRed + "Error enabling SD-WAN group: " + err.Error() + "\n" + ColorReset
  3897. }
  3898. return ColorGreen + fmt.Sprintf("SD-WAN group %q enabled.\n", idOrName) + ColorReset
  3899. case "disable":
  3900. if len(parts) < 2 {
  3901. return ColorRed + "Usage: /ip sdwan disable <id|name>\n" + ColorReset
  3902. }
  3903. idOrName := parts[1]
  3904. if err := e.sdwanMgr.SetState(idOrName, true); err != nil {
  3905. return ColorRed + "Error disabling SD-WAN group: " + err.Error() + "\n" + ColorReset
  3906. }
  3907. return ColorGreen + fmt.Sprintf("SD-WAN group %q disabled.\n", idOrName) + ColorReset
  3908. default:
  3909. return ColorRed + "Usage: /ip sdwan [print|status|add|set|remove|enable|disable|member]\n" + ColorReset
  3910. }
  3911. }
  3912. func (e *Executor) handleSDWANMember(parts []string, user *auth.PublicUser) string {
  3913. if len(parts) == 0 || parts[0] == "print" || parts[0] == "pr" || parts[0] == "show" {
  3914. _, params := parseArgs(parts)
  3915. groupFilter := params["group"]
  3916. groups := e.sdwanMgr.List()
  3917. tbl := NewTable("Flags", "Group", "Interface", "Gateway", "Weight", "Priority", "Target", "Status", "Latency", "Loss", "Jitter")
  3918. for _, g := range groups {
  3919. if groupFilter != "" && !strings.EqualFold(g.Name, groupFilter) && !strings.EqualFold(g.ID, groupFilter) {
  3920. continue
  3921. }
  3922. for _, m := range g.Members {
  3923. flag := " "
  3924. if m.Disabled || g.Disabled {
  3925. flag = "X"
  3926. }
  3927. statusCol := string(m.Status)
  3928. tbl.AddRow(
  3929. flag,
  3930. g.Name,
  3931. m.Interface,
  3932. m.Gateway,
  3933. fmt.Sprintf("%d", m.Weight),
  3934. fmt.Sprintf("%d", m.Priority),
  3935. m.TargetProbe,
  3936. statusCol,
  3937. fmt.Sprintf("%.1fms", m.LiveLatency),
  3938. fmt.Sprintf("%.1f%%", m.PacketLoss),
  3939. fmt.Sprintf("%.1fms", m.Jitter),
  3940. )
  3941. }
  3942. }
  3943. return tbl.String()
  3944. }
  3945. if user != nil && user.Role != auth.RoleFull {
  3946. return ColorRed + "Permission denied: admin role required\n" + ColorReset
  3947. }
  3948. sub := strings.ToLower(parts[0])
  3949. switch sub {
  3950. case "add":
  3951. _, params := parseArgs(parts[1:])
  3952. groupName := params["group"]
  3953. iface := params["interface"]
  3954. if groupName == "" || iface == "" {
  3955. return ColorRed + "Usage: /ip sdwan member add group=<group> interface=<iface> [gateway=<gw>] [weight=50] [priority=1] [target-probe=1.1.1.1] [max-latency=250] [max-loss=20] [comment=\"...\"]\n" + ColorReset
  3956. }
  3957. weight := 50
  3958. if wStr, ok := params["weight"]; ok {
  3959. if v, err := strconv.Atoi(wStr); err == nil && v > 0 {
  3960. weight = v
  3961. }
  3962. }
  3963. priority := 1
  3964. if pStr, ok := params["priority"]; ok {
  3965. if v, err := strconv.Atoi(pStr); err == nil && v > 0 {
  3966. priority = v
  3967. }
  3968. }
  3969. maxLat := 250
  3970. if lStr, ok := params["max-latency"]; ok {
  3971. if v, err := strconv.Atoi(lStr); err == nil && v > 0 {
  3972. maxLat = v
  3973. }
  3974. }
  3975. maxLoss := 20
  3976. if mlStr, ok := params["max-loss"]; ok {
  3977. if v, err := strconv.Atoi(mlStr); err == nil && v >= 0 {
  3978. maxLoss = v
  3979. }
  3980. }
  3981. targetProbe := params["target-probe"]
  3982. if targetProbe == "" {
  3983. targetProbe = "1.1.1.1"
  3984. }
  3985. mem := &sdwan.SDWANMember{
  3986. Interface: iface,
  3987. Gateway: params["gateway"],
  3988. Weight: weight,
  3989. Priority: priority,
  3990. TargetProbe: targetProbe,
  3991. MaxLatency: maxLat,
  3992. MaxLoss: maxLoss,
  3993. Comment: params["comment"],
  3994. Disabled: (params["disabled"] == "yes" || params["disabled"] == "true"),
  3995. }
  3996. if err := e.sdwanMgr.AddMember(groupName, mem); err != nil {
  3997. return ColorRed + "Error adding member: " + err.Error() + "\n" + ColorReset
  3998. }
  3999. return ColorGreen + fmt.Sprintf("Interface %s added to SD-WAN group %s.\n", iface, groupName) + ColorReset
  4000. case "set":
  4001. _, params := parseArgs(parts[1:])
  4002. groupName := params["group"]
  4003. iface := params["interface"]
  4004. if groupName == "" || iface == "" {
  4005. return ColorRed + "Usage: /ip sdwan member set group=<group> interface=<iface> [gateway=...] [weight=...] [priority=...] [target-probe=...]\n" + ColorReset
  4006. }
  4007. group, found := e.sdwanMgr.Get(groupName)
  4008. if !found {
  4009. return ColorRed + fmt.Sprintf("SD-WAN group %q not found\n", groupName) + ColorReset
  4010. }
  4011. var targetMember *sdwan.SDWANMember
  4012. for _, m := range group.Members {
  4013. if m.Interface == iface || m.ID == iface {
  4014. targetMember = m
  4015. break
  4016. }
  4017. }
  4018. if targetMember == nil {
  4019. return ColorRed + fmt.Sprintf("Member %q not found in group %s\n", iface, groupName) + ColorReset
  4020. }
  4021. if gw, ok := params["gateway"]; ok {
  4022. targetMember.Gateway = gw
  4023. }
  4024. if wStr, ok := params["weight"]; ok {
  4025. if v, err := strconv.Atoi(wStr); err == nil && v > 0 {
  4026. targetMember.Weight = v
  4027. }
  4028. }
  4029. if pStr, ok := params["priority"]; ok {
  4030. if v, err := strconv.Atoi(pStr); err == nil && v > 0 {
  4031. targetMember.Priority = v
  4032. }
  4033. }
  4034. if tp, ok := params["target-probe"]; ok {
  4035. targetMember.TargetProbe = tp
  4036. }
  4037. if lStr, ok := params["max-latency"]; ok {
  4038. if v, err := strconv.Atoi(lStr); err == nil && v > 0 {
  4039. targetMember.MaxLatency = v
  4040. }
  4041. }
  4042. if mlStr, ok := params["max-loss"]; ok {
  4043. if v, err := strconv.Atoi(mlStr); err == nil && v >= 0 {
  4044. targetMember.MaxLoss = v
  4045. }
  4046. }
  4047. if c, ok := params["comment"]; ok {
  4048. targetMember.Comment = c
  4049. }
  4050. if dStr, ok := params["disabled"]; ok {
  4051. targetMember.Disabled = (dStr == "yes" || dStr == "true")
  4052. }
  4053. if err := e.sdwanMgr.UpdateMember(groupName, targetMember); err != nil {
  4054. return ColorRed + "Error updating member: " + err.Error() + "\n" + ColorReset
  4055. }
  4056. return ColorGreen + fmt.Sprintf("Member %s in SD-WAN group %s updated.\n", iface, groupName) + ColorReset
  4057. case "remove", "delete", "del":
  4058. _, params := parseArgs(parts[1:])
  4059. groupName := params["group"]
  4060. iface := params["interface"]
  4061. if groupName == "" && len(parts) >= 3 {
  4062. groupName = parts[1]
  4063. iface = parts[2]
  4064. }
  4065. if groupName == "" || iface == "" {
  4066. return ColorRed + "Usage: /ip sdwan member remove group=<group> interface=<iface>\n" + ColorReset
  4067. }
  4068. if err := e.sdwanMgr.DeleteMember(groupName, iface); err != nil {
  4069. return ColorRed + "Error removing member: " + err.Error() + "\n" + ColorReset
  4070. }
  4071. return ColorGreen + fmt.Sprintf("Member %s removed from SD-WAN group %s.\n", iface, groupName) + ColorReset
  4072. case "enable":
  4073. _, params := parseArgs(parts[1:])
  4074. groupName := params["group"]
  4075. iface := params["interface"]
  4076. if groupName == "" || iface == "" {
  4077. return ColorRed + "Usage: /ip sdwan member enable group=<group> interface=<iface>\n" + ColorReset
  4078. }
  4079. group, found := e.sdwanMgr.Get(groupName)
  4080. if !found {
  4081. return ColorRed + fmt.Sprintf("SD-WAN group %q not found\n", groupName) + ColorReset
  4082. }
  4083. var targetMember *sdwan.SDWANMember
  4084. for _, m := range group.Members {
  4085. if m.Interface == iface || m.ID == iface {
  4086. targetMember = m
  4087. break
  4088. }
  4089. }
  4090. if targetMember == nil {
  4091. return ColorRed + fmt.Sprintf("Member %q not found in group %s\n", iface, groupName) + ColorReset
  4092. }
  4093. targetMember.Disabled = false
  4094. if err := e.sdwanMgr.UpdateMember(groupName, targetMember); err != nil {
  4095. return ColorRed + "Error enabling member: " + err.Error() + "\n" + ColorReset
  4096. }
  4097. return ColorGreen + fmt.Sprintf("Member %s in SD-WAN group %s enabled.\n", iface, groupName) + ColorReset
  4098. case "disable":
  4099. _, params := parseArgs(parts[1:])
  4100. groupName := params["group"]
  4101. iface := params["interface"]
  4102. if groupName == "" || iface == "" {
  4103. return ColorRed + "Usage: /ip sdwan member disable group=<group> interface=<iface>\n" + ColorReset
  4104. }
  4105. group, found := e.sdwanMgr.Get(groupName)
  4106. if !found {
  4107. return ColorRed + fmt.Sprintf("SD-WAN group %q not found\n", groupName) + ColorReset
  4108. }
  4109. var targetMember *sdwan.SDWANMember
  4110. for _, m := range group.Members {
  4111. if m.Interface == iface || m.ID == iface {
  4112. targetMember = m
  4113. break
  4114. }
  4115. }
  4116. if targetMember == nil {
  4117. return ColorRed + fmt.Sprintf("Member %q not found in group %s\n", iface, groupName) + ColorReset
  4118. }
  4119. targetMember.Disabled = true
  4120. if err := e.sdwanMgr.UpdateMember(groupName, targetMember); err != nil {
  4121. return ColorRed + "Error disabling member: " + err.Error() + "\n" + ColorReset
  4122. }
  4123. return ColorGreen + fmt.Sprintf("Member %s in SD-WAN group %s disabled.\n", iface, groupName) + ColorReset
  4124. default:
  4125. return ColorRed + "Usage: /ip sdwan member [print|add|set|remove|enable|disable]\n" + ColorReset
  4126. }
  4127. }
  4128. // Helpers for sorting/filtering
  4129. func sortStrings(s []string) {
  4130. sort.Strings(s)
  4131. }