reverseproxy.go 49 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646
  1. package main
  2. import (
  3. "encoding/json"
  4. "net/http"
  5. "path/filepath"
  6. "sort"
  7. "strconv"
  8. "strings"
  9. "time"
  10. "imuslab.com/zoraxy/mod/auth"
  11. "imuslab.com/zoraxy/mod/dynamicproxy"
  12. "imuslab.com/zoraxy/mod/dynamicproxy/loadbalance"
  13. "imuslab.com/zoraxy/mod/dynamicproxy/permissionpolicy"
  14. "imuslab.com/zoraxy/mod/dynamicproxy/rewrite"
  15. "imuslab.com/zoraxy/mod/netutils"
  16. "imuslab.com/zoraxy/mod/uptime"
  17. "imuslab.com/zoraxy/mod/utils"
  18. )
  19. var (
  20. dynamicProxyRouter *dynamicproxy.Router
  21. )
  22. // Add user customizable reverse proxy
  23. func ReverseProxtInit() {
  24. /*
  25. Load Reverse Proxy Global Settings
  26. */
  27. inboundPort := *defaultInboundPort
  28. autoStartReverseProxy := *defaultEnableInboundTraffic
  29. if sysdb.KeyExists("settings", "inbound") {
  30. //Read settings from database
  31. sysdb.Read("settings", "inbound", &inboundPort)
  32. if netutils.CheckIfPortOccupied(inboundPort) {
  33. autoStartReverseProxy = false
  34. SystemWideLogger.Println("Inbound port ", inboundPort, " is occupied. Change the listening port in the webmin panel and press \"Start Service\" to start reverse proxy service")
  35. } else {
  36. SystemWideLogger.Println("Serving inbound port ", inboundPort)
  37. }
  38. } else {
  39. //Default port
  40. if netutils.CheckIfPortOccupied(inboundPort) {
  41. autoStartReverseProxy = false
  42. SystemWideLogger.Println("Port 443 is occupied. Change the listening port in the webmin panel and press \"Start Service\" to start reverse proxy service")
  43. }
  44. SystemWideLogger.Println("Inbound port not set. Using default (443)")
  45. }
  46. useTls := true
  47. sysdb.Read("settings", "usetls", &useTls)
  48. if useTls {
  49. SystemWideLogger.Println("TLS mode enabled. Serving proxy request with TLS")
  50. } else {
  51. SystemWideLogger.Println("TLS mode disabled. Serving proxy request with plain http")
  52. }
  53. forceLatestTLSVersion := false
  54. sysdb.Read("settings", "forceLatestTLS", &forceLatestTLSVersion)
  55. if forceLatestTLSVersion {
  56. SystemWideLogger.Println("Force latest TLS mode enabled. Minimum TLS LS version is set to v1.2")
  57. } else {
  58. SystemWideLogger.Println("Force latest TLS mode disabled. Minimum TLS version is set to v1.0")
  59. }
  60. developmentMode := false
  61. sysdb.Read("settings", "devMode", &developmentMode)
  62. if useTls {
  63. SystemWideLogger.Println("Development mode enabled. Using no-store Cache Control policy")
  64. } else {
  65. SystemWideLogger.Println("Development mode disabled. Proxying with default Cache Control policy")
  66. }
  67. listenOnPort80 := true
  68. if netutils.CheckIfPortOccupied(80) {
  69. listenOnPort80 = false
  70. }
  71. sysdb.Read("settings", "listenP80", &listenOnPort80)
  72. if listenOnPort80 {
  73. SystemWideLogger.Println("Port 80 listener enabled")
  74. } else {
  75. SystemWideLogger.Println("Port 80 listener disabled")
  76. }
  77. forceHttpsRedirect := true
  78. sysdb.Read("settings", "redirect", &forceHttpsRedirect)
  79. if forceHttpsRedirect {
  80. SystemWideLogger.Println("Force HTTPS mode enabled")
  81. //Port 80 listener must be enabled to perform http -> https redirect
  82. listenOnPort80 = true
  83. } else {
  84. SystemWideLogger.Println("Force HTTPS mode disabled")
  85. }
  86. /*
  87. Create a new proxy object
  88. The DynamicProxy is the parent of all reverse proxy handlers,
  89. use for managemening and provide functions to access proxy handlers
  90. */
  91. dprouter, err := dynamicproxy.NewDynamicProxy(dynamicproxy.RouterOption{
  92. HostUUID: nodeUUID,
  93. HostVersion: SYSTEM_VERSION,
  94. Port: inboundPort,
  95. UseTls: useTls,
  96. ForceTLSLatest: forceLatestTLSVersion,
  97. NoCache: developmentMode,
  98. ListenOnPort80: listenOnPort80,
  99. ForceHttpsRedirect: forceHttpsRedirect,
  100. TlsManager: tlsCertManager,
  101. RedirectRuleTable: redirectTable,
  102. GeodbStore: geodbStore,
  103. StatisticCollector: statisticCollector,
  104. WebDirectory: *path_webserver,
  105. AccessController: accessController,
  106. AutheliaRouter: autheliaRouter,
  107. LoadBalancer: loadBalancer,
  108. Logger: SystemWideLogger,
  109. })
  110. if err != nil {
  111. SystemWideLogger.PrintAndLog("proxy-config", "Unable to create dynamic proxy router", err)
  112. return
  113. }
  114. dynamicProxyRouter = dprouter
  115. /*
  116. Load all conf from files
  117. */
  118. confs, _ := filepath.Glob("./conf/proxy/*.config")
  119. for _, conf := range confs {
  120. err := LoadReverseProxyConfig(conf)
  121. if err != nil {
  122. SystemWideLogger.PrintAndLog("proxy-config", "Failed to load config file: "+filepath.Base(conf), err)
  123. return
  124. }
  125. }
  126. if dynamicProxyRouter.Root == nil {
  127. //Root config not set (new deployment?), use internal static web server as root
  128. defaultRootRouter, err := GetDefaultRootConfig()
  129. if err != nil {
  130. SystemWideLogger.PrintAndLog("proxy-config", "Failed to generate default root routing", err)
  131. return
  132. }
  133. dynamicProxyRouter.SetProxyRouteAsRoot(defaultRootRouter)
  134. }
  135. //Start Service
  136. //Not sure why but delay must be added if you have another
  137. //reverse proxy server in front of this service
  138. if autoStartReverseProxy {
  139. time.Sleep(300 * time.Millisecond)
  140. dynamicProxyRouter.StartProxyService()
  141. SystemWideLogger.Println("Dynamic Reverse Proxy service started")
  142. }
  143. //Add all proxy services to uptime monitor
  144. //Create a uptime monitor service
  145. go func() {
  146. //This must be done in go routine to prevent blocking on system startup
  147. uptimeMonitor, _ = uptime.NewUptimeMonitor(&uptime.Config{
  148. Targets: GetUptimeTargetsFromReverseProxyRules(dynamicProxyRouter),
  149. Interval: 300, //5 minutes
  150. MaxRecordsStore: 288, //1 day
  151. OnlineStateNotify: loadBalancer.NotifyHostOnlineState, //Notify the load balancer for online state
  152. Logger: SystemWideLogger, //Logger
  153. })
  154. SystemWideLogger.Println("Uptime Monitor background service started")
  155. }()
  156. }
  157. func ReverseProxyHandleOnOff(w http.ResponseWriter, r *http.Request) {
  158. enable, _ := utils.PostPara(r, "enable") //Support root, vdir and subd
  159. if enable == "true" {
  160. err := dynamicProxyRouter.StartProxyService()
  161. if err != nil {
  162. utils.SendErrorResponse(w, err.Error())
  163. return
  164. }
  165. } else {
  166. //Check if it is loopback
  167. if dynamicProxyRouter.IsProxiedSubdomain(r) {
  168. //Loopback routing. Turning it off will make the user lost control
  169. //of the whole system. Do not allow shutdown
  170. utils.SendErrorResponse(w, "Unable to shutdown in loopback rp mode. Remove proxy rules for management interface and retry.")
  171. return
  172. }
  173. err := dynamicProxyRouter.StopProxyService()
  174. if err != nil {
  175. utils.SendErrorResponse(w, err.Error())
  176. return
  177. }
  178. }
  179. utils.SendOK(w)
  180. }
  181. func ReverseProxyHandleAddEndpoint(w http.ResponseWriter, r *http.Request) {
  182. eptype, err := utils.PostPara(r, "type") //Support root and host
  183. if err != nil {
  184. utils.SendErrorResponse(w, "type not defined")
  185. return
  186. }
  187. endpoint, err := utils.PostPara(r, "ep")
  188. if err != nil {
  189. utils.SendErrorResponse(w, "endpoint not defined")
  190. return
  191. }
  192. tls, _ := utils.PostPara(r, "tls")
  193. if tls == "" {
  194. tls = "false"
  195. }
  196. useTLS := (tls == "true")
  197. //Bypass global TLS value / allow direct access from port 80?
  198. bypassGlobalTLS, _ := utils.PostPara(r, "bypassGlobalTLS")
  199. if bypassGlobalTLS == "" {
  200. bypassGlobalTLS = "false"
  201. }
  202. useBypassGlobalTLS := bypassGlobalTLS == "true"
  203. //Enable TLS validation?
  204. skipTlsValidation, _ := utils.PostBool(r, "tlsval")
  205. //Get access rule ID
  206. accessRuleID, _ := utils.PostPara(r, "access")
  207. if accessRuleID == "" {
  208. accessRuleID = "default"
  209. }
  210. if !accessController.AccessRuleExists(accessRuleID) {
  211. utils.SendErrorResponse(w, "invalid access rule ID selected")
  212. return
  213. }
  214. // Require basic auth?
  215. requireBasicAuth, _ := utils.PostBool(r, "bauth")
  216. //Use sticky session?
  217. useStickySession, _ := utils.PostBool(r, "stickysess")
  218. // Require Rate Limiting?
  219. requireRateLimit := false
  220. proxyRateLimit := 1000
  221. requireRateLimit, err = utils.PostBool(r, "rate")
  222. if err != nil {
  223. requireRateLimit = false
  224. }
  225. if requireRateLimit {
  226. proxyRateLimit, err = utils.PostInt(r, "ratenum")
  227. if err != nil {
  228. proxyRateLimit = 0
  229. }
  230. if err != nil {
  231. utils.SendErrorResponse(w, "invalid rate limit number")
  232. return
  233. }
  234. if proxyRateLimit <= 0 {
  235. utils.SendErrorResponse(w, "rate limit number must be greater than 0")
  236. return
  237. }
  238. }
  239. // Bypass WebSocket Origin Check
  240. strbpwsorg, _ := utils.PostPara(r, "bpwsorg")
  241. if strbpwsorg == "" {
  242. strbpwsorg = "false"
  243. }
  244. bypassWebsocketOriginCheck := (strbpwsorg == "true")
  245. //Prase the basic auth to correct structure
  246. cred, _ := utils.PostPara(r, "cred")
  247. basicAuthCredentials := []*dynamicproxy.BasicAuthCredentials{}
  248. if requireBasicAuth {
  249. preProcessCredentials := []*dynamicproxy.BasicAuthUnhashedCredentials{}
  250. err = json.Unmarshal([]byte(cred), &preProcessCredentials)
  251. if err != nil {
  252. utils.SendErrorResponse(w, "invalid user credentials")
  253. return
  254. }
  255. //Check if there are empty password credentials
  256. for _, credObj := range preProcessCredentials {
  257. if strings.TrimSpace(credObj.Password) == "" {
  258. utils.SendErrorResponse(w, credObj.Username+" has empty password")
  259. return
  260. }
  261. }
  262. //Convert and hash the passwords
  263. for _, credObj := range preProcessCredentials {
  264. basicAuthCredentials = append(basicAuthCredentials, &dynamicproxy.BasicAuthCredentials{
  265. Username: credObj.Username,
  266. PasswordHash: auth.Hash(credObj.Password),
  267. })
  268. }
  269. }
  270. tagStr, _ := utils.PostPara(r, "tags")
  271. tags := []string{}
  272. if tagStr != "" {
  273. tags = strings.Split(tagStr, ",")
  274. for i := range tags {
  275. tags[i] = strings.TrimSpace(tags[i])
  276. }
  277. }
  278. // Remove empty tags
  279. filteredTags := []string{}
  280. for _, tag := range tags {
  281. if tag != "" {
  282. filteredTags = append(filteredTags, tag)
  283. }
  284. }
  285. tags = filteredTags
  286. var proxyEndpointCreated *dynamicproxy.ProxyEndpoint
  287. if eptype == "host" {
  288. rootOrMatchingDomain, err := utils.PostPara(r, "rootname")
  289. if err != nil {
  290. utils.SendErrorResponse(w, "hostname not defined")
  291. return
  292. }
  293. rootOrMatchingDomain = strings.TrimSpace(rootOrMatchingDomain)
  294. //Check if it contains ",", if yes, split the remainings as alias
  295. aliasHostnames := []string{}
  296. if strings.Contains(rootOrMatchingDomain, ",") {
  297. matchingDomains := strings.Split(rootOrMatchingDomain, ",")
  298. if len(matchingDomains) > 1 {
  299. rootOrMatchingDomain = matchingDomains[0]
  300. for _, aliasHostname := range matchingDomains[1:] {
  301. //Filter out any space
  302. aliasHostnames = append(aliasHostnames, strings.TrimSpace(aliasHostname))
  303. }
  304. }
  305. }
  306. //Generate a default authenticaion provider
  307. authMethod := dynamicproxy.AuthMethodNone
  308. if requireBasicAuth {
  309. authMethod = dynamicproxy.AuthMethodBasic
  310. }
  311. thisAuthenticationProvider := dynamicproxy.AuthenticationProvider{
  312. AuthMethod: authMethod,
  313. BasicAuthCredentials: basicAuthCredentials,
  314. BasicAuthExceptionRules: []*dynamicproxy.BasicAuthExceptionRule{},
  315. }
  316. //Generate a proxy endpoint object
  317. thisProxyEndpoint := dynamicproxy.ProxyEndpoint{
  318. //I/O
  319. ProxyType: dynamicproxy.ProxyTypeHost,
  320. RootOrMatchingDomain: rootOrMatchingDomain,
  321. MatchingDomainAlias: aliasHostnames,
  322. ActiveOrigins: []*loadbalance.Upstream{
  323. {
  324. OriginIpOrDomain: endpoint,
  325. RequireTLS: useTLS,
  326. SkipCertValidations: skipTlsValidation,
  327. SkipWebSocketOriginCheck: bypassWebsocketOriginCheck,
  328. Weight: 1,
  329. },
  330. },
  331. InactiveOrigins: []*loadbalance.Upstream{},
  332. UseStickySession: useStickySession,
  333. //TLS
  334. BypassGlobalTLS: useBypassGlobalTLS,
  335. AccessFilterUUID: accessRuleID,
  336. //VDir
  337. VirtualDirectories: []*dynamicproxy.VirtualDirectoryEndpoint{},
  338. //Custom headers
  339. //Auth
  340. AuthenticationProvider: &thisAuthenticationProvider,
  341. //Header Rewrite
  342. HeaderRewriteRules: dynamicproxy.GetDefaultHeaderRewriteRules(),
  343. //Default Site
  344. DefaultSiteOption: 0,
  345. DefaultSiteValue: "",
  346. // Rate Limit
  347. RequireRateLimit: requireRateLimit,
  348. RateLimit: int64(proxyRateLimit),
  349. Tags: tags,
  350. }
  351. preparedEndpoint, err := dynamicProxyRouter.PrepareProxyRoute(&thisProxyEndpoint)
  352. if err != nil {
  353. utils.SendErrorResponse(w, "unable to prepare proxy route to target endpoint: "+err.Error())
  354. return
  355. }
  356. dynamicProxyRouter.AddProxyRouteToRuntime(preparedEndpoint)
  357. proxyEndpointCreated = &thisProxyEndpoint
  358. } else if eptype == "root" {
  359. //Get the default site options and target
  360. dsOptString, err := utils.PostPara(r, "defaultSiteOpt")
  361. if err != nil {
  362. utils.SendErrorResponse(w, "default site action not defined")
  363. return
  364. }
  365. var defaultSiteOption int = 1
  366. opt, err := strconv.Atoi(dsOptString)
  367. if err != nil {
  368. utils.SendErrorResponse(w, "invalid default site option")
  369. return
  370. }
  371. defaultSiteOption = opt
  372. dsVal, err := utils.PostPara(r, "defaultSiteVal")
  373. if err != nil && (defaultSiteOption == 1 || defaultSiteOption == 2) {
  374. //Reverse proxy or redirect, must require value to be set
  375. utils.SendErrorResponse(w, "target not defined")
  376. return
  377. }
  378. //Write the root options to file
  379. rootRoutingEndpoint := dynamicproxy.ProxyEndpoint{
  380. ProxyType: dynamicproxy.ProxyTypeRoot,
  381. RootOrMatchingDomain: "/",
  382. ActiveOrigins: []*loadbalance.Upstream{
  383. {
  384. OriginIpOrDomain: endpoint,
  385. RequireTLS: useTLS,
  386. SkipCertValidations: true,
  387. SkipWebSocketOriginCheck: true,
  388. Weight: 1,
  389. },
  390. },
  391. InactiveOrigins: []*loadbalance.Upstream{},
  392. BypassGlobalTLS: false,
  393. DefaultSiteOption: defaultSiteOption,
  394. DefaultSiteValue: dsVal,
  395. }
  396. preparedRootProxyRoute, err := dynamicProxyRouter.PrepareProxyRoute(&rootRoutingEndpoint)
  397. if err != nil {
  398. utils.SendErrorResponse(w, "unable to prepare root routing: "+err.Error())
  399. return
  400. }
  401. err = dynamicProxyRouter.SetProxyRouteAsRoot(preparedRootProxyRoute)
  402. if err != nil {
  403. utils.SendErrorResponse(w, "unable to update default site: "+err.Error())
  404. return
  405. }
  406. proxyEndpointCreated = &rootRoutingEndpoint
  407. } else {
  408. //Invalid eptype
  409. utils.SendErrorResponse(w, "invalid endpoint type")
  410. return
  411. }
  412. //Save the config to file
  413. err = SaveReverseProxyConfig(proxyEndpointCreated)
  414. if err != nil {
  415. SystemWideLogger.PrintAndLog("proxy-config", "Unable to save new proxy rule to file", err)
  416. return
  417. }
  418. //Update utm if exists
  419. UpdateUptimeMonitorTargets()
  420. utils.SendOK(w)
  421. }
  422. /*
  423. ReverseProxyHandleEditEndpoint handles proxy endpoint edit
  424. (host only, for root use Default Site page to edit)
  425. This endpoint do not handle basic auth credential update.
  426. The credential will be loaded from old config and reused
  427. */
  428. func ReverseProxyHandleEditEndpoint(w http.ResponseWriter, r *http.Request) {
  429. rootNameOrMatchingDomain, err := utils.PostPara(r, "rootname")
  430. if err != nil {
  431. utils.SendErrorResponse(w, "Target proxy rule not defined")
  432. return
  433. }
  434. tls, _ := utils.PostPara(r, "tls")
  435. if tls == "" {
  436. tls = "false"
  437. }
  438. useStickySession, _ := utils.PostBool(r, "ss")
  439. //Load bypass TLS option
  440. bpgtls, _ := utils.PostPara(r, "bpgtls")
  441. if bpgtls == "" {
  442. bpgtls = "false"
  443. }
  444. bypassGlobalTLS := (bpgtls == "true")
  445. //Disable uptime monitor
  446. disbleUtm, err := utils.PostBool(r, "dutm")
  447. if err != nil {
  448. disbleUtm = false
  449. }
  450. // Auth Provider
  451. authProviderTypeStr, _ := utils.PostPara(r, "authprovider")
  452. if authProviderTypeStr == "" {
  453. authProviderTypeStr = "0"
  454. }
  455. authProviderType, err := strconv.Atoi(authProviderTypeStr)
  456. if err != nil {
  457. utils.SendErrorResponse(w, "Invalid auth provider type")
  458. return
  459. }
  460. // Rate Limiting?
  461. rl, _ := utils.PostPara(r, "rate")
  462. if rl == "" {
  463. rl = "false"
  464. }
  465. requireRateLimit := (rl == "true")
  466. rlnum, _ := utils.PostPara(r, "ratenum")
  467. if rlnum == "" {
  468. rlnum = "0"
  469. }
  470. proxyRateLimit, err := strconv.ParseInt(rlnum, 10, 64)
  471. if err != nil {
  472. utils.SendErrorResponse(w, "invalid rate limit number")
  473. return
  474. }
  475. if requireRateLimit && proxyRateLimit <= 0 {
  476. utils.SendErrorResponse(w, "rate limit number must be greater than 0")
  477. return
  478. } else if proxyRateLimit < 0 {
  479. proxyRateLimit = 1000
  480. }
  481. //Load the previous basic auth credentials from current proxy rules
  482. targetProxyEntry, err := dynamicProxyRouter.LoadProxy(rootNameOrMatchingDomain)
  483. if err != nil {
  484. utils.SendErrorResponse(w, "Target proxy config not found or could not be loaded")
  485. return
  486. }
  487. tagStr, _ := utils.PostPara(r, "tags")
  488. tags := []string{}
  489. if tagStr != "" {
  490. tags = strings.Split(tagStr, ",")
  491. for i := range tags {
  492. tags[i] = strings.TrimSpace(tags[i])
  493. }
  494. }
  495. //Generate a new proxyEndpoint from the new config
  496. newProxyEndpoint := dynamicproxy.CopyEndpoint(targetProxyEntry)
  497. newProxyEndpoint.BypassGlobalTLS = bypassGlobalTLS
  498. if newProxyEndpoint.AuthenticationProvider == nil {
  499. newProxyEndpoint.AuthenticationProvider = &dynamicproxy.AuthenticationProvider{
  500. AuthMethod: dynamicproxy.AuthMethodNone,
  501. BasicAuthCredentials: []*dynamicproxy.BasicAuthCredentials{},
  502. BasicAuthExceptionRules: []*dynamicproxy.BasicAuthExceptionRule{},
  503. }
  504. }
  505. if authProviderType == 1 {
  506. newProxyEndpoint.AuthenticationProvider.AuthMethod = dynamicproxy.AuthMethodBasic
  507. } else if authProviderType == 2 {
  508. newProxyEndpoint.AuthenticationProvider.AuthMethod = dynamicproxy.AuthMethodAuthelia
  509. } else if authProviderType == 3 {
  510. newProxyEndpoint.AuthenticationProvider.AuthMethod = dynamicproxy.AuthMethodOauth2
  511. } else {
  512. newProxyEndpoint.AuthenticationProvider.AuthMethod = dynamicproxy.AuthMethodNone
  513. }
  514. newProxyEndpoint.RequireRateLimit = requireRateLimit
  515. newProxyEndpoint.RateLimit = proxyRateLimit
  516. newProxyEndpoint.UseStickySession = useStickySession
  517. newProxyEndpoint.DisableUptimeMonitor = disbleUtm
  518. newProxyEndpoint.Tags = tags
  519. //Prepare to replace the current routing rule
  520. readyRoutingRule, err := dynamicProxyRouter.PrepareProxyRoute(newProxyEndpoint)
  521. if err != nil {
  522. utils.SendErrorResponse(w, err.Error())
  523. return
  524. }
  525. targetProxyEntry.Remove()
  526. dynamicProxyRouter.AddProxyRouteToRuntime(readyRoutingRule)
  527. //Save it to file
  528. SaveReverseProxyConfig(newProxyEndpoint)
  529. //Update uptime monitor targets
  530. UpdateUptimeMonitorTargets()
  531. utils.SendOK(w)
  532. }
  533. func ReverseProxyHandleAlias(w http.ResponseWriter, r *http.Request) {
  534. rootNameOrMatchingDomain, err := utils.PostPara(r, "ep")
  535. if err != nil {
  536. utils.SendErrorResponse(w, "Invalid ep given")
  537. return
  538. }
  539. //No need to check for type as root (/) can be set to default route
  540. //and hence, you will not need alias
  541. //Load the previous alias from current proxy rules
  542. targetProxyEntry, err := dynamicProxyRouter.LoadProxy(rootNameOrMatchingDomain)
  543. if err != nil {
  544. utils.SendErrorResponse(w, "Target proxy config not found or could not be loaded")
  545. return
  546. }
  547. newAliasJSON, err := utils.PostPara(r, "alias")
  548. if err != nil {
  549. //No new set of alias given
  550. utils.SendErrorResponse(w, "new alias not given")
  551. return
  552. }
  553. //Write new alias to runtime and file
  554. newAlias := []string{}
  555. err = json.Unmarshal([]byte(newAliasJSON), &newAlias)
  556. if err != nil {
  557. SystemWideLogger.PrintAndLog("proxy-config", "Unable to parse new alias list", err)
  558. utils.SendErrorResponse(w, "Invalid alias list given")
  559. return
  560. }
  561. //Set the current alias
  562. newProxyEndpoint := dynamicproxy.CopyEndpoint(targetProxyEntry)
  563. newProxyEndpoint.MatchingDomainAlias = newAlias
  564. // Prepare to replace the current routing rule
  565. readyRoutingRule, err := dynamicProxyRouter.PrepareProxyRoute(newProxyEndpoint)
  566. if err != nil {
  567. utils.SendErrorResponse(w, err.Error())
  568. return
  569. }
  570. targetProxyEntry.Remove()
  571. dynamicProxyRouter.AddProxyRouteToRuntime(readyRoutingRule)
  572. // Save it to file
  573. err = SaveReverseProxyConfig(newProxyEndpoint)
  574. if err != nil {
  575. utils.SendErrorResponse(w, "Alias update failed")
  576. SystemWideLogger.PrintAndLog("proxy-config", "Unable to save alias update", err)
  577. }
  578. utils.SendOK(w)
  579. }
  580. func DeleteProxyEndpoint(w http.ResponseWriter, r *http.Request) {
  581. ep, err := utils.PostPara(r, "ep")
  582. if err != nil {
  583. utils.SendErrorResponse(w, "Invalid ep given")
  584. return
  585. }
  586. //Remove the config from runtime
  587. err = dynamicProxyRouter.RemoveProxyEndpointByRootname(ep)
  588. if err != nil {
  589. utils.SendErrorResponse(w, err.Error())
  590. return
  591. }
  592. //Remove the config from file
  593. err = RemoveReverseProxyConfig(ep)
  594. if err != nil {
  595. utils.SendErrorResponse(w, err.Error())
  596. return
  597. }
  598. //Update uptime monitor
  599. UpdateUptimeMonitorTargets()
  600. utils.SendOK(w)
  601. }
  602. /*
  603. Handle update request for basic auth credential
  604. Require paramter: ep (Endpoint) and pytype (proxy Type)
  605. if request with GET, the handler will return current credentials
  606. on this endpoint by its username
  607. if request is POST, the handler will write the results to proxy config
  608. */
  609. func UpdateProxyBasicAuthCredentials(w http.ResponseWriter, r *http.Request) {
  610. if r.Method == http.MethodGet {
  611. ep, err := utils.GetPara(r, "ep")
  612. if err != nil {
  613. utils.SendErrorResponse(w, "Invalid ep given")
  614. return
  615. }
  616. //Load the target proxy object from router
  617. targetProxy, err := dynamicProxyRouter.LoadProxy(ep)
  618. if err != nil {
  619. utils.SendErrorResponse(w, err.Error())
  620. return
  621. }
  622. usernames := []string{}
  623. for _, cred := range targetProxy.AuthenticationProvider.BasicAuthCredentials {
  624. usernames = append(usernames, cred.Username)
  625. }
  626. js, _ := json.Marshal(usernames)
  627. utils.SendJSONResponse(w, string(js))
  628. } else if r.Method == http.MethodPost {
  629. //Write to target
  630. ep, err := utils.PostPara(r, "ep")
  631. if err != nil {
  632. utils.SendErrorResponse(w, "Invalid ep given")
  633. return
  634. }
  635. creds, err := utils.PostPara(r, "creds")
  636. if err != nil {
  637. utils.SendErrorResponse(w, "Invalid ptype given")
  638. return
  639. }
  640. //Load the target proxy object from router
  641. targetProxy, err := dynamicProxyRouter.LoadProxy(ep)
  642. if err != nil {
  643. utils.SendErrorResponse(w, err.Error())
  644. return
  645. }
  646. //Try to marshal the content of creds into the suitable structure
  647. newCredentials := []*dynamicproxy.BasicAuthUnhashedCredentials{}
  648. err = json.Unmarshal([]byte(creds), &newCredentials)
  649. if err != nil {
  650. utils.SendErrorResponse(w, "Malformed credential data")
  651. return
  652. }
  653. //Merge the credentials into the original config
  654. //If a new username exists in old config with no pw given, keep the old pw hash
  655. //If a new username is found with new password, hash it and push to credential slice
  656. mergedCredentials := []*dynamicproxy.BasicAuthCredentials{}
  657. for _, credential := range newCredentials {
  658. if credential.Password == "" {
  659. //Check if exists in the old credential files
  660. keepUnchange := false
  661. for _, oldCredEntry := range targetProxy.AuthenticationProvider.BasicAuthCredentials {
  662. if oldCredEntry.Username == credential.Username {
  663. //Exists! Reuse the old hash
  664. mergedCredentials = append(mergedCredentials, &dynamicproxy.BasicAuthCredentials{
  665. Username: oldCredEntry.Username,
  666. PasswordHash: oldCredEntry.PasswordHash,
  667. })
  668. keepUnchange = true
  669. }
  670. }
  671. if !keepUnchange {
  672. //This is a new username with no pw given
  673. utils.SendErrorResponse(w, "Access password for "+credential.Username+" is empty!")
  674. return
  675. }
  676. } else {
  677. //This username have given password
  678. mergedCredentials = append(mergedCredentials, &dynamicproxy.BasicAuthCredentials{
  679. Username: credential.Username,
  680. PasswordHash: auth.Hash(credential.Password),
  681. })
  682. }
  683. }
  684. targetProxy.AuthenticationProvider.BasicAuthCredentials = mergedCredentials
  685. //Save it to file
  686. SaveReverseProxyConfig(targetProxy)
  687. //Replace runtime configuration
  688. targetProxy.UpdateToRuntime()
  689. utils.SendOK(w)
  690. } else {
  691. http.Error(w, "invalid usage", http.StatusMethodNotAllowed)
  692. }
  693. }
  694. // List, Update or Remove the exception paths for basic auth.
  695. func ListProxyBasicAuthExceptionPaths(w http.ResponseWriter, r *http.Request) {
  696. if r.Method != http.MethodGet {
  697. http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
  698. }
  699. ep, err := utils.GetPara(r, "ep")
  700. if err != nil {
  701. utils.SendErrorResponse(w, "Invalid ep given")
  702. return
  703. }
  704. //Load the target proxy object from router
  705. targetProxy, err := dynamicProxyRouter.LoadProxy(ep)
  706. if err != nil {
  707. utils.SendErrorResponse(w, err.Error())
  708. return
  709. }
  710. //List all the exception paths for this proxy
  711. results := targetProxy.AuthenticationProvider.BasicAuthExceptionRules
  712. if results == nil {
  713. //It is a config from a really old version of zoraxy. Overwrite it with empty array
  714. results = []*dynamicproxy.BasicAuthExceptionRule{}
  715. }
  716. js, _ := json.Marshal(results)
  717. utils.SendJSONResponse(w, string(js))
  718. return
  719. }
  720. func AddProxyBasicAuthExceptionPaths(w http.ResponseWriter, r *http.Request) {
  721. ep, err := utils.PostPara(r, "ep")
  722. if err != nil {
  723. utils.SendErrorResponse(w, "Invalid ep given")
  724. return
  725. }
  726. matchingPrefix, err := utils.PostPara(r, "prefix")
  727. if err != nil {
  728. utils.SendErrorResponse(w, "Invalid matching prefix given")
  729. return
  730. }
  731. //Load the target proxy object from router
  732. targetProxy, err := dynamicProxyRouter.LoadProxy(ep)
  733. if err != nil {
  734. utils.SendErrorResponse(w, err.Error())
  735. return
  736. }
  737. //Check if the prefix starts with /. If not, prepend it
  738. if !strings.HasPrefix(matchingPrefix, "/") {
  739. matchingPrefix = "/" + matchingPrefix
  740. }
  741. //Add a new exception rule if it is not already exists
  742. alreadyExists := false
  743. for _, thisExceptionRule := range targetProxy.AuthenticationProvider.BasicAuthExceptionRules {
  744. if thisExceptionRule.PathPrefix == matchingPrefix {
  745. alreadyExists = true
  746. break
  747. }
  748. }
  749. if alreadyExists {
  750. utils.SendErrorResponse(w, "This matching path already exists")
  751. return
  752. }
  753. targetProxy.AuthenticationProvider.BasicAuthExceptionRules = append(targetProxy.AuthenticationProvider.BasicAuthExceptionRules, &dynamicproxy.BasicAuthExceptionRule{
  754. PathPrefix: strings.TrimSpace(matchingPrefix),
  755. })
  756. //Save configs to runtime and file
  757. targetProxy.UpdateToRuntime()
  758. SaveReverseProxyConfig(targetProxy)
  759. utils.SendOK(w)
  760. }
  761. func RemoveProxyBasicAuthExceptionPaths(w http.ResponseWriter, r *http.Request) {
  762. // Delete a rule
  763. ep, err := utils.PostPara(r, "ep")
  764. if err != nil {
  765. utils.SendErrorResponse(w, "Invalid ep given")
  766. return
  767. }
  768. matchingPrefix, err := utils.PostPara(r, "prefix")
  769. if err != nil {
  770. utils.SendErrorResponse(w, "Invalid matching prefix given")
  771. return
  772. }
  773. // Load the target proxy object from router
  774. targetProxy, err := dynamicProxyRouter.LoadProxy(ep)
  775. if err != nil {
  776. utils.SendErrorResponse(w, err.Error())
  777. return
  778. }
  779. newExceptionRuleList := []*dynamicproxy.BasicAuthExceptionRule{}
  780. matchingExists := false
  781. for _, thisExceptionalRule := range targetProxy.AuthenticationProvider.BasicAuthExceptionRules {
  782. if thisExceptionalRule.PathPrefix != matchingPrefix {
  783. newExceptionRuleList = append(newExceptionRuleList, thisExceptionalRule)
  784. } else {
  785. matchingExists = true
  786. }
  787. }
  788. if !matchingExists {
  789. utils.SendErrorResponse(w, "target matching rule not exists")
  790. return
  791. }
  792. targetProxy.AuthenticationProvider.BasicAuthExceptionRules = newExceptionRuleList
  793. // Save configs to runtime and file
  794. targetProxy.UpdateToRuntime()
  795. SaveReverseProxyConfig(targetProxy)
  796. utils.SendOK(w)
  797. }
  798. // Report the current status of the reverse proxy server
  799. func ReverseProxyStatus(w http.ResponseWriter, r *http.Request) {
  800. js, _ := json.Marshal(dynamicProxyRouter)
  801. utils.SendJSONResponse(w, string(js))
  802. }
  803. // Toggle a certain rule on and off
  804. func ReverseProxyToggleRuleSet(w http.ResponseWriter, r *http.Request) {
  805. //No need to check for type as root cannot be turned off
  806. ep, err := utils.PostPara(r, "ep")
  807. if err != nil {
  808. utils.SendErrorResponse(w, "invalid ep given")
  809. return
  810. }
  811. targetProxyRule, err := dynamicProxyRouter.LoadProxy(ep)
  812. if err != nil {
  813. utils.SendErrorResponse(w, "invalid endpoint given")
  814. return
  815. }
  816. enableStr, err := utils.PostPara(r, "enable")
  817. if err != nil {
  818. enableStr = "true"
  819. }
  820. //Flip the enable and disabled tag state
  821. ruleDisabled := enableStr == "false"
  822. targetProxyRule.Disabled = ruleDisabled
  823. err = SaveReverseProxyConfig(targetProxyRule)
  824. if err != nil {
  825. utils.SendErrorResponse(w, "unable to save updated rule")
  826. return
  827. }
  828. //Update uptime monitor
  829. UpdateUptimeMonitorTargets()
  830. utils.SendOK(w)
  831. }
  832. func ReverseProxyListDetail(w http.ResponseWriter, r *http.Request) {
  833. eptype, err := utils.PostPara(r, "type") //Support root and host
  834. if err != nil {
  835. utils.SendErrorResponse(w, "type not defined")
  836. return
  837. }
  838. if eptype == "host" {
  839. epname, err := utils.PostPara(r, "epname")
  840. if err != nil {
  841. utils.SendErrorResponse(w, "epname not defined")
  842. return
  843. }
  844. epname = strings.ToLower(strings.TrimSpace(epname))
  845. endpointRaw, ok := dynamicProxyRouter.ProxyEndpoints.Load(epname)
  846. if !ok {
  847. utils.SendErrorResponse(w, "proxy rule not found")
  848. return
  849. }
  850. targetEndpoint := dynamicproxy.CopyEndpoint(endpointRaw.(*dynamicproxy.ProxyEndpoint))
  851. js, _ := json.Marshal(targetEndpoint)
  852. utils.SendJSONResponse(w, string(js))
  853. } else if eptype == "root" {
  854. js, _ := json.Marshal(dynamicProxyRouter.Root)
  855. utils.SendJSONResponse(w, string(js))
  856. } else {
  857. utils.SendErrorResponse(w, "Invalid type given")
  858. }
  859. }
  860. func ReverseProxyList(w http.ResponseWriter, r *http.Request) {
  861. eptype, err := utils.PostPara(r, "type") //Support root and host
  862. if err != nil {
  863. utils.SendErrorResponse(w, "type not defined")
  864. return
  865. }
  866. if eptype == "host" {
  867. results := []*dynamicproxy.ProxyEndpoint{}
  868. dynamicProxyRouter.ProxyEndpoints.Range(func(key, value interface{}) bool {
  869. thisEndpoint := dynamicproxy.CopyEndpoint(value.(*dynamicproxy.ProxyEndpoint))
  870. //Clear the auth passwords before showing to front-end
  871. cleanedCredentials := []*dynamicproxy.BasicAuthCredentials{}
  872. for _, user := range thisEndpoint.AuthenticationProvider.BasicAuthCredentials {
  873. cleanedCredentials = append(cleanedCredentials, &dynamicproxy.BasicAuthCredentials{
  874. Username: user.Username,
  875. PasswordHash: "",
  876. })
  877. }
  878. thisEndpoint.AuthenticationProvider.BasicAuthCredentials = cleanedCredentials
  879. results = append(results, thisEndpoint)
  880. return true
  881. })
  882. sort.Slice(results, func(i, j int) bool {
  883. return results[i].RootOrMatchingDomain < results[j].RootOrMatchingDomain
  884. })
  885. js, _ := json.Marshal(results)
  886. utils.SendJSONResponse(w, string(js))
  887. } else if eptype == "root" {
  888. js, _ := json.Marshal(dynamicProxyRouter.Root)
  889. utils.SendJSONResponse(w, string(js))
  890. } else {
  891. utils.SendErrorResponse(w, "Invalid type given")
  892. }
  893. }
  894. // Handle port 80 incoming traffics
  895. func HandleUpdatePort80Listener(w http.ResponseWriter, r *http.Request) {
  896. if r.Method == http.MethodGet {
  897. //Load the current status
  898. currentEnabled := false
  899. err := sysdb.Read("settings", "listenP80", &currentEnabled)
  900. if err != nil {
  901. utils.SendErrorResponse(w, err.Error())
  902. return
  903. }
  904. js, _ := json.Marshal(currentEnabled)
  905. utils.SendJSONResponse(w, string(js))
  906. } else if r.Method == http.MethodPost {
  907. enabled, err := utils.PostPara(r, "enable")
  908. if err != nil {
  909. utils.SendErrorResponse(w, "enable state not set")
  910. return
  911. }
  912. if enabled == "true" {
  913. sysdb.Write("settings", "listenP80", true)
  914. SystemWideLogger.Println("Enabling port 80 listener")
  915. dynamicProxyRouter.UpdatePort80ListenerState(true)
  916. } else if enabled == "false" {
  917. sysdb.Write("settings", "listenP80", false)
  918. SystemWideLogger.Println("Disabling port 80 listener")
  919. dynamicProxyRouter.UpdatePort80ListenerState(false)
  920. } else {
  921. utils.SendErrorResponse(w, "invalid mode given: "+enabled)
  922. }
  923. utils.SendOK(w)
  924. } else {
  925. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  926. }
  927. }
  928. // Handle https redirect
  929. func HandleUpdateHttpsRedirect(w http.ResponseWriter, r *http.Request) {
  930. if r.Method == http.MethodGet {
  931. currentRedirectToHttps := false
  932. //Load the current status
  933. err := sysdb.Read("settings", "redirect", &currentRedirectToHttps)
  934. if err != nil {
  935. utils.SendErrorResponse(w, err.Error())
  936. return
  937. }
  938. js, _ := json.Marshal(currentRedirectToHttps)
  939. utils.SendJSONResponse(w, string(js))
  940. } else if r.Method == http.MethodPost {
  941. useRedirect, err := utils.PostBool(r, "set")
  942. if err != nil {
  943. utils.SendErrorResponse(w, "status not set")
  944. return
  945. }
  946. if dynamicProxyRouter.Option.Port == 80 {
  947. utils.SendErrorResponse(w, "This option is not available when listening on port 80")
  948. return
  949. }
  950. if useRedirect {
  951. sysdb.Write("settings", "redirect", true)
  952. SystemWideLogger.Println("Updating force HTTPS redirection to true")
  953. dynamicProxyRouter.UpdateHttpToHttpsRedirectSetting(true)
  954. } else {
  955. sysdb.Write("settings", "redirect", false)
  956. SystemWideLogger.Println("Updating force HTTPS redirection to false")
  957. dynamicProxyRouter.UpdateHttpToHttpsRedirectSetting(false)
  958. }
  959. utils.SendOK(w)
  960. } else {
  961. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  962. }
  963. }
  964. // Handle checking if the current user is accessing via the reverse proxied interface
  965. // Of the management interface.
  966. func HandleManagementProxyCheck(w http.ResponseWriter, r *http.Request) {
  967. isProxied := dynamicProxyRouter.IsProxiedSubdomain(r)
  968. js, _ := json.Marshal(isProxied)
  969. utils.SendJSONResponse(w, string(js))
  970. }
  971. func HandleDevelopmentModeChange(w http.ResponseWriter, r *http.Request) {
  972. enableDevelopmentModeStr, err := utils.GetPara(r, "enable")
  973. if err != nil {
  974. //Load the current development mode toggle state
  975. js, _ := json.Marshal(dynamicProxyRouter.Option.NoCache)
  976. utils.SendJSONResponse(w, string(js))
  977. } else {
  978. //Write changes to runtime
  979. enableDevelopmentMode := false
  980. if enableDevelopmentModeStr == "true" {
  981. enableDevelopmentMode = true
  982. }
  983. //Write changes to runtime
  984. dynamicProxyRouter.Option.NoCache = enableDevelopmentMode
  985. //Write changes to database
  986. sysdb.Write("settings", "devMode", enableDevelopmentMode)
  987. utils.SendOK(w)
  988. }
  989. }
  990. // Handle incoming port set. Change the current proxy incoming port
  991. func HandleIncomingPortSet(w http.ResponseWriter, r *http.Request) {
  992. newIncomingPort, err := utils.PostPara(r, "incoming")
  993. if err != nil {
  994. utils.SendErrorResponse(w, "invalid incoming port given")
  995. return
  996. }
  997. newIncomingPortInt, err := strconv.Atoi(newIncomingPort)
  998. if err != nil {
  999. utils.SendErrorResponse(w, "Invalid incoming port given")
  1000. return
  1001. }
  1002. rootProxyTargetOrigin := ""
  1003. if len(dynamicProxyRouter.Root.ActiveOrigins) > 0 {
  1004. rootProxyTargetOrigin = dynamicProxyRouter.Root.ActiveOrigins[0].OriginIpOrDomain
  1005. }
  1006. //Check if it is identical as proxy root (recursion!)
  1007. if dynamicProxyRouter.Root == nil || rootProxyTargetOrigin == "" {
  1008. //Check if proxy root is set before checking recursive listen
  1009. //Fixing issue #43
  1010. utils.SendErrorResponse(w, "Set Proxy Root before changing inbound port")
  1011. return
  1012. }
  1013. proxyRoot := strings.TrimSuffix(rootProxyTargetOrigin, "/")
  1014. if strings.EqualFold(proxyRoot, "localhost:"+strconv.Itoa(newIncomingPortInt)) || strings.EqualFold(proxyRoot, "127.0.0.1:"+strconv.Itoa(newIncomingPortInt)) {
  1015. //Listening port is same as proxy root
  1016. //Not allow recursive settings
  1017. utils.SendErrorResponse(w, "Recursive listening port! Check your proxy root settings.")
  1018. return
  1019. }
  1020. //Stop and change the setting of the reverse proxy service
  1021. if dynamicProxyRouter.Running {
  1022. dynamicProxyRouter.StopProxyService()
  1023. dynamicProxyRouter.Option.Port = newIncomingPortInt
  1024. time.Sleep(1 * time.Second) //Fixed start fail issue
  1025. dynamicProxyRouter.StartProxyService()
  1026. } else {
  1027. //Only change setting but not starting the proxy service
  1028. dynamicProxyRouter.Option.Port = newIncomingPortInt
  1029. }
  1030. sysdb.Write("settings", "inbound", newIncomingPortInt)
  1031. utils.SendOK(w)
  1032. }
  1033. /* Handle Custom Header Rules */
  1034. //List all the custom header defined in this proxy rule
  1035. func HandleCustomHeaderList(w http.ResponseWriter, r *http.Request) {
  1036. epType, err := utils.GetPara(r, "type")
  1037. if err != nil {
  1038. utils.SendErrorResponse(w, "endpoint type not defined")
  1039. return
  1040. }
  1041. domain, err := utils.GetPara(r, "domain")
  1042. if err != nil {
  1043. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1044. return
  1045. }
  1046. var targetProxyEndpoint *dynamicproxy.ProxyEndpoint
  1047. if epType == "root" {
  1048. targetProxyEndpoint = dynamicProxyRouter.Root
  1049. } else {
  1050. ep, err := dynamicProxyRouter.LoadProxy(domain)
  1051. if err != nil {
  1052. utils.SendErrorResponse(w, "target endpoint not exists")
  1053. return
  1054. }
  1055. targetProxyEndpoint = ep
  1056. }
  1057. //List all custom headers
  1058. customHeaderList := targetProxyEndpoint.HeaderRewriteRules.UserDefinedHeaders
  1059. if customHeaderList == nil {
  1060. customHeaderList = []*rewrite.UserDefinedHeader{}
  1061. }
  1062. js, _ := json.Marshal(customHeaderList)
  1063. utils.SendJSONResponse(w, string(js))
  1064. }
  1065. // Add a new header to the target endpoint
  1066. func HandleCustomHeaderAdd(w http.ResponseWriter, r *http.Request) {
  1067. rewriteType, err := utils.PostPara(r, "type")
  1068. if err != nil {
  1069. utils.SendErrorResponse(w, "rewriteType not defined")
  1070. return
  1071. }
  1072. domain, err := utils.PostPara(r, "domain")
  1073. if err != nil {
  1074. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1075. return
  1076. }
  1077. direction, err := utils.PostPara(r, "direction")
  1078. if err != nil {
  1079. utils.SendErrorResponse(w, "HTTP modifiy direction not set")
  1080. return
  1081. }
  1082. name, err := utils.PostPara(r, "name")
  1083. if err != nil {
  1084. utils.SendErrorResponse(w, "HTTP header name not set")
  1085. return
  1086. }
  1087. value, err := utils.PostPara(r, "value")
  1088. if err != nil && rewriteType == "add" {
  1089. utils.SendErrorResponse(w, "HTTP header value not set")
  1090. return
  1091. }
  1092. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1093. if err != nil {
  1094. utils.SendErrorResponse(w, "target endpoint not exists")
  1095. return
  1096. }
  1097. //Create a Custom Header Definition type
  1098. var rewriteDirection rewrite.HeaderDirection
  1099. if direction == "toOrigin" {
  1100. rewriteDirection = rewrite.HeaderDirection_ZoraxyToUpstream
  1101. } else if direction == "toClient" {
  1102. rewriteDirection = rewrite.HeaderDirection_ZoraxyToDownstream
  1103. } else {
  1104. //Unknown direction
  1105. utils.SendErrorResponse(w, "header rewrite direction not supported")
  1106. return
  1107. }
  1108. isRemove := false
  1109. if rewriteType == "remove" {
  1110. isRemove = true
  1111. }
  1112. headerRewriteDefinition := rewrite.UserDefinedHeader{
  1113. Key: name,
  1114. Value: value,
  1115. Direction: rewriteDirection,
  1116. IsRemove: isRemove,
  1117. }
  1118. //Create a new custom header object
  1119. err = targetProxyEndpoint.AddUserDefinedHeader(&headerRewriteDefinition)
  1120. if err != nil {
  1121. utils.SendErrorResponse(w, "unable to add header rewrite rule: "+err.Error())
  1122. return
  1123. }
  1124. //Save it (no need reload as header are not handled by dpcore)
  1125. err = SaveReverseProxyConfig(targetProxyEndpoint)
  1126. if err != nil {
  1127. utils.SendErrorResponse(w, "unable to save update")
  1128. return
  1129. }
  1130. utils.SendOK(w)
  1131. }
  1132. // Remove a header from the target endpoint
  1133. func HandleCustomHeaderRemove(w http.ResponseWriter, r *http.Request) {
  1134. domain, err := utils.PostPara(r, "domain")
  1135. if err != nil {
  1136. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1137. return
  1138. }
  1139. name, err := utils.PostPara(r, "name")
  1140. if err != nil {
  1141. utils.SendErrorResponse(w, "HTTP header name not set")
  1142. return
  1143. }
  1144. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1145. if err != nil {
  1146. utils.SendErrorResponse(w, "target endpoint not exists")
  1147. return
  1148. }
  1149. err = targetProxyEndpoint.RemoveUserDefinedHeader(name)
  1150. if err != nil {
  1151. utils.SendErrorResponse(w, "unable to remove header rewrite rule: "+err.Error())
  1152. return
  1153. }
  1154. err = SaveReverseProxyConfig(targetProxyEndpoint)
  1155. if err != nil {
  1156. utils.SendErrorResponse(w, "unable to save update")
  1157. return
  1158. }
  1159. utils.SendOK(w)
  1160. }
  1161. func HandleHostOverwrite(w http.ResponseWriter, r *http.Request) {
  1162. domain, err := utils.PostPara(r, "domain")
  1163. if err != nil {
  1164. domain, err = utils.GetPara(r, "domain")
  1165. if err != nil {
  1166. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1167. return
  1168. }
  1169. }
  1170. //Get the proxy endpoint object dedicated to this domain
  1171. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1172. if err != nil {
  1173. utils.SendErrorResponse(w, "target endpoint not exists")
  1174. return
  1175. }
  1176. if r.Method == http.MethodGet {
  1177. //Get the current host header
  1178. js, _ := json.Marshal(targetProxyEndpoint.HeaderRewriteRules.RequestHostOverwrite)
  1179. utils.SendJSONResponse(w, string(js))
  1180. } else if r.Method == http.MethodPost {
  1181. //Set the new host header
  1182. newHostname, _ := utils.PostPara(r, "hostname")
  1183. //As this will require change in the proxy instance we are running
  1184. //we need to clone and respawn this proxy endpoint
  1185. newProxyEndpoint := targetProxyEndpoint.Clone()
  1186. newProxyEndpoint.HeaderRewriteRules.RequestHostOverwrite = newHostname
  1187. //Save proxy endpoint
  1188. err = SaveReverseProxyConfig(newProxyEndpoint)
  1189. if err != nil {
  1190. utils.SendErrorResponse(w, err.Error())
  1191. return
  1192. }
  1193. //Spawn a new endpoint with updated dpcore
  1194. preparedEndpoint, err := dynamicProxyRouter.PrepareProxyRoute(newProxyEndpoint)
  1195. if err != nil {
  1196. utils.SendErrorResponse(w, err.Error())
  1197. return
  1198. }
  1199. //Remove the old endpoint
  1200. err = targetProxyEndpoint.Remove()
  1201. if err != nil {
  1202. utils.SendErrorResponse(w, err.Error())
  1203. return
  1204. }
  1205. //Add the newly prepared endpoint to runtime
  1206. err = dynamicProxyRouter.AddProxyRouteToRuntime(preparedEndpoint)
  1207. if err != nil {
  1208. utils.SendErrorResponse(w, err.Error())
  1209. return
  1210. }
  1211. //Print log message
  1212. if newHostname != "" {
  1213. SystemWideLogger.Println("Updated " + domain + " hostname overwrite to: " + newHostname)
  1214. } else {
  1215. SystemWideLogger.Println("Removed " + domain + " hostname overwrite")
  1216. }
  1217. utils.SendOK(w)
  1218. } else {
  1219. //Invalid method
  1220. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  1221. }
  1222. }
  1223. // HandleHopByHop get and set the hop by hop remover state
  1224. // note that it shows the DISABLE STATE of hop-by-hop remover, not the enable state
  1225. func HandleHopByHop(w http.ResponseWriter, r *http.Request) {
  1226. domain, err := utils.PostPara(r, "domain")
  1227. if err != nil {
  1228. domain, err = utils.GetPara(r, "domain")
  1229. if err != nil {
  1230. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1231. return
  1232. }
  1233. }
  1234. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1235. if err != nil {
  1236. utils.SendErrorResponse(w, "target endpoint not exists")
  1237. return
  1238. }
  1239. if r.Method == http.MethodGet {
  1240. //Get the current hop by hop header state
  1241. js, _ := json.Marshal(!targetProxyEndpoint.HeaderRewriteRules.DisableHopByHopHeaderRemoval)
  1242. utils.SendJSONResponse(w, string(js))
  1243. } else if r.Method == http.MethodPost {
  1244. //Set the hop by hop header state
  1245. enableHopByHopRemover, _ := utils.PostBool(r, "removeHopByHop")
  1246. //As this will require change in the proxy instance we are running
  1247. //we need to clone and respawn this proxy endpoint
  1248. newProxyEndpoint := targetProxyEndpoint.Clone()
  1249. //Storage file use false as default, so disable removal = not enable remover
  1250. newProxyEndpoint.HeaderRewriteRules.DisableHopByHopHeaderRemoval = !enableHopByHopRemover
  1251. //Save proxy endpoint
  1252. err = SaveReverseProxyConfig(newProxyEndpoint)
  1253. if err != nil {
  1254. utils.SendErrorResponse(w, err.Error())
  1255. return
  1256. }
  1257. //Spawn a new endpoint with updated dpcore
  1258. preparedEndpoint, err := dynamicProxyRouter.PrepareProxyRoute(newProxyEndpoint)
  1259. if err != nil {
  1260. utils.SendErrorResponse(w, err.Error())
  1261. return
  1262. }
  1263. //Remove the old endpoint
  1264. err = targetProxyEndpoint.Remove()
  1265. if err != nil {
  1266. utils.SendErrorResponse(w, err.Error())
  1267. return
  1268. }
  1269. //Add the newly prepared endpoint to runtime
  1270. err = dynamicProxyRouter.AddProxyRouteToRuntime(preparedEndpoint)
  1271. if err != nil {
  1272. utils.SendErrorResponse(w, err.Error())
  1273. return
  1274. }
  1275. //Print log message
  1276. if enableHopByHopRemover {
  1277. SystemWideLogger.Println("Enabled hop-by-hop headers removal on " + domain)
  1278. } else {
  1279. SystemWideLogger.Println("Disabled hop-by-hop headers removal on " + domain)
  1280. }
  1281. utils.SendOK(w)
  1282. } else {
  1283. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  1284. }
  1285. }
  1286. // Handle view or edit HSTS states
  1287. func HandleHSTSState(w http.ResponseWriter, r *http.Request) {
  1288. domain, err := utils.PostPara(r, "domain")
  1289. if err != nil {
  1290. domain, err = utils.GetPara(r, "domain")
  1291. if err != nil {
  1292. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1293. return
  1294. }
  1295. }
  1296. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1297. if err != nil {
  1298. utils.SendErrorResponse(w, "target endpoint not exists")
  1299. return
  1300. }
  1301. if r.Method == http.MethodGet {
  1302. //Return current HSTS enable state
  1303. hstsAge := targetProxyEndpoint.HeaderRewriteRules.HSTSMaxAge
  1304. js, _ := json.Marshal(hstsAge)
  1305. utils.SendJSONResponse(w, string(js))
  1306. return
  1307. } else if r.Method == http.MethodPost {
  1308. newMaxAge, err := utils.PostInt(r, "maxage")
  1309. if err != nil {
  1310. utils.SendErrorResponse(w, "maxage not defeined")
  1311. return
  1312. }
  1313. if newMaxAge == 0 || newMaxAge >= 31536000 {
  1314. targetProxyEndpoint.HeaderRewriteRules.HSTSMaxAge = int64(newMaxAge)
  1315. err = SaveReverseProxyConfig(targetProxyEndpoint)
  1316. if err != nil {
  1317. utils.SendErrorResponse(w, "save HSTS state failed: "+err.Error())
  1318. return
  1319. }
  1320. targetProxyEndpoint.UpdateToRuntime()
  1321. } else {
  1322. utils.SendErrorResponse(w, "invalid max age given")
  1323. return
  1324. }
  1325. utils.SendOK(w)
  1326. return
  1327. }
  1328. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  1329. }
  1330. // HandlePermissionPolicy handle read or write to permission policy
  1331. func HandlePermissionPolicy(w http.ResponseWriter, r *http.Request) {
  1332. domain, err := utils.PostPara(r, "domain")
  1333. if err != nil {
  1334. domain, err = utils.GetPara(r, "domain")
  1335. if err != nil {
  1336. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1337. return
  1338. }
  1339. }
  1340. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1341. if err != nil {
  1342. utils.SendErrorResponse(w, "target endpoint not exists")
  1343. return
  1344. }
  1345. if r.Method == http.MethodGet {
  1346. type CurrentPolicyState struct {
  1347. PPEnabled bool
  1348. CurrentPolicy *permissionpolicy.PermissionsPolicy
  1349. }
  1350. currentPolicy := permissionpolicy.GetDefaultPermissionPolicy()
  1351. if targetProxyEndpoint.HeaderRewriteRules.PermissionPolicy != nil {
  1352. currentPolicy = targetProxyEndpoint.HeaderRewriteRules.PermissionPolicy
  1353. }
  1354. result := CurrentPolicyState{
  1355. PPEnabled: targetProxyEndpoint.HeaderRewriteRules.EnablePermissionPolicyHeader,
  1356. CurrentPolicy: currentPolicy,
  1357. }
  1358. js, _ := json.Marshal(result)
  1359. utils.SendJSONResponse(w, string(js))
  1360. return
  1361. } else if r.Method == http.MethodPost {
  1362. //Update the enable state of permission policy
  1363. enableState, err := utils.PostBool(r, "enable")
  1364. if err != nil {
  1365. utils.SendErrorResponse(w, "invalid enable state given")
  1366. return
  1367. }
  1368. targetProxyEndpoint.HeaderRewriteRules.EnablePermissionPolicyHeader = enableState
  1369. SaveReverseProxyConfig(targetProxyEndpoint)
  1370. targetProxyEndpoint.UpdateToRuntime()
  1371. utils.SendOK(w)
  1372. return
  1373. } else if r.Method == http.MethodPut {
  1374. //Store the new permission policy
  1375. newPermissionPolicyJSONString, err := utils.PostPara(r, "pp")
  1376. if err != nil {
  1377. utils.SendErrorResponse(w, "missing pp (permission policy) paramter")
  1378. return
  1379. }
  1380. //Parse the permission policy from JSON string
  1381. newPermissionPolicy := permissionpolicy.GetDefaultPermissionPolicy()
  1382. err = json.Unmarshal([]byte(newPermissionPolicyJSONString), &newPermissionPolicy)
  1383. if err != nil {
  1384. utils.SendErrorResponse(w, "permission policy parse error: "+err.Error())
  1385. return
  1386. }
  1387. //Save it to file
  1388. targetProxyEndpoint.HeaderRewriteRules.PermissionPolicy = newPermissionPolicy
  1389. SaveReverseProxyConfig(targetProxyEndpoint)
  1390. targetProxyEndpoint.UpdateToRuntime()
  1391. utils.SendOK(w)
  1392. return
  1393. }
  1394. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  1395. }
  1396. func HandleWsHeaderBehavior(w http.ResponseWriter, r *http.Request) {
  1397. domain, err := utils.PostPara(r, "domain")
  1398. if err != nil {
  1399. domain, err = utils.GetPara(r, "domain")
  1400. if err != nil {
  1401. utils.SendErrorResponse(w, "domain or matching rule not defined")
  1402. return
  1403. }
  1404. }
  1405. targetProxyEndpoint, err := dynamicProxyRouter.LoadProxy(domain)
  1406. if err != nil {
  1407. utils.SendErrorResponse(w, "target endpoint not exists")
  1408. return
  1409. }
  1410. if r.Method == http.MethodGet {
  1411. js, _ := json.Marshal(targetProxyEndpoint.EnableWebsocketCustomHeaders)
  1412. utils.SendJSONResponse(w, string(js))
  1413. } else if r.Method == http.MethodPost {
  1414. enableWsHeader, err := utils.PostBool(r, "enable")
  1415. if err != nil {
  1416. utils.SendErrorResponse(w, "invalid enable state given")
  1417. return
  1418. }
  1419. targetProxyEndpoint.EnableWebsocketCustomHeaders = enableWsHeader
  1420. SaveReverseProxyConfig(targetProxyEndpoint)
  1421. targetProxyEndpoint.UpdateToRuntime()
  1422. utils.SendOK(w)
  1423. } else {
  1424. http.Error(w, "405 - Method not allowed", http.StatusMethodNotAllowed)
  1425. }
  1426. }