[sample] Enhance black/white list feature of samples/proxy

flomesh-io · Jul 5, 2024 · 666abb2 · 666abb2
1 parent ad5a517
commit 666abb2
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/samples/proxy/main.js b/samples/proxy/main.js
@@ -98,9 +98,13 @@ if (config.proxy.http) {
               return 'tunnel'
             } else {
               var url = new URL(head.path)
-              $host = `${url.hostname}:${url.port}`
-              log('HTTP', head.method, head.path)
-              return 'forward'
+              if (isTargetAllowed(url.hostname)) {
+                $host = `${url.hostname}:${url.port}`
+                log('HTTP', head.method, head.path)
+                return 'forward'
+              } else {
+                return 'forbidden'
+              }
             }
           }
         }, {
@@ -109,7 +113,10 @@ if (config.proxy.http) {
           ),
           'forward': ($=>$
             .muxHTTP(() => $host).to($=>$.connect(() => $host))
-          )
+          ),
+          'forbidden': ($=>$
+            .replaceMessage(new Message({ status: 403 }))
+          ),
         }
       )
     )