feat: impl httpclient.safeCurl

eggjs · Jul 11, 2024 · 914e66a · 914e66a
1 parent 52a03a8
commit 914e66a
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 0 deletions.
diff --git a/lib/core/httpclient.js b/lib/core/httpclient.js
@@ -44,6 +44,17 @@ class HttpClient extends urllib.HttpClient2 {
   async curl(...args) {
     return await this.request(...args);
   }
+
+  async safeCurl(url, options = {}) {
+    const ssrfConfig = this.app.config.security.ssrf;
+    if (ssrfConfig?.checkAddress) {
+      options.checkAddress = ssrfConfig.checkAddress;
+    } else {
+      this.app.logger.warn('[egg-security] please configure `config.security.ssrf` first');
+    }
+
+    return this.curl(url, options);
+  }
 }
 
 function normalizeConfig(app) {

diff --git a/lib/core/httpclient_next.js b/lib/core/httpclient_next.js
@@ -1,5 +1,6 @@
 const { HttpClient } = require('urllib-next');
 const ms = require('humanize-ms');
+const SSRF_HTTPCLIENT = Symbol('SSRF_HTTPCLIENT');
 
 class HttpClientNext extends HttpClient {
   constructor(app, options) {
@@ -33,6 +34,21 @@ class HttpClientNext extends HttpClient {
   async curl(...args) {
     return await this.request(...args);
   }
+
+  async safeCurl(url, options = {}) {
+    if (!this[SSRF_HTTPCLIENT]) {
+      const ssrfConfig = this.app.config.security.ssrf;
+      if (ssrfConfig?.checkAddress) {
+        options.checkAddress = ssrfConfig.checkAddress;
+      } else {
+        this.app.logger.warn('[egg-security] please configure `config.security.ssrf` first');
+      }
+      this[SSRF_HTTPCLIENT] = new HttpClientNext(this.app, {
+        checkAddress: ssrfConfig.checkAddress,
+      });
+    }
+    return this[SSRF_HTTPCLIENT].request(url, options);
+  }
 }
 
 function normalizeConfig(app) {

diff --git a/test/lib/core/httpclient.test.js b/test/lib/core/httpclient.test.js
@@ -98,6 +98,26 @@ describe('test/lib/core/httpclient.test.js', () => {
     });
   });
 
+  it('should support safeCurl', async () => {
+    let ip;
+    let family;
+    let host;
+    mm(client.app.config, 'security', {
+      ssrf: {
+        checkAddress(aIp, aFamilay, aHost) {
+          ip = aIp;
+          family = aFamilay;
+          host = aHost;
+          return true;
+        },
+      },
+    });
+    await client.safeCurl(url);
+    assert(ip);
+    assert(family);
+    assert(host);
+  });
+
   describe('HttpClientNext', () => {
     it('should request ok with log', async () => {
       const args = {
@@ -145,6 +165,26 @@ describe('test/lib/core/httpclient.test.js', () => {
         return true;
       });
     });
+
+    it('should support safeCurl', async () => {
+      let ip;
+      let family;
+      let host;
+      mm(clientNext.app.config, 'security', {
+        ssrf: {
+          checkAddress(aIp, aFamilay, aHost) {
+            ip = aIp;
+            family = aFamilay;
+            host = aHost;
+            return true;
+          },
+        },
+      });
+      await clientNext.safeCurl(url);
+      assert(ip);
+      assert(family);
+      assert(host);
+    });
   });
 
   describe('httpclient.httpAgent.timeout < 30000', () => {