No SameSite cookie vulnerability detection

Author: uurien

packages/dd-trace/src/appsec/iast/analyzers/analyzers.js

@@ -5,6 +5,7 @@ module.exports = {
   'INSECURE_COOKIE_ANALYZER': require('./insecure-cookie-analyzer'),
   'LDAP_ANALYZER': require('./ldap-injection-analyzer'),
   'NO_HTTPONLY_COOKIE_ANALYZER': require('./no-httponly-cookie-analyzer'),
+  'NO_SAMESITE_COOKIE_ANALYZER': require('./no-samesite-cookie-analyzer'),
   'PATH_TRAVERSAL_ANALYZER': require('./path-traversal-analyzer'),
   'SQL_INJECTION_ANALYZER': require('./sql-injection-analyzer'),
   'SSRF': require('./ssrf-analyzer'),

packages/dd-trace/src/appsec/iast/analyzers/no-samesite-cookie-analyzer.js

@@ -0,0 +1,12 @@
+'use strict'
+
+const { NO_SAMESITE_COOKIE } = require('../vulnerabilities')
+const CookieAnalyzer = require('./cookie-analyzer')
+
+class NoSamesiteCookieAnalyzer extends CookieAnalyzer {
+  constructor () {
+    super(NO_SAMESITE_COOKIE, 'SameSite=strict')
+  }
+}
+
+module.exports = new NoSamesiteCookieAnalyzer()

packages/dd-trace/src/appsec/iast/vulnerabilities.js

@@ -3,6 +3,7 @@ module.exports = {
   INSECURE_COOKIE: 'INSECURE_COOKIE',
   LDAP_INJECTION: 'LDAP_INJECTION',
   NO_HTTPONLY_COOKIE: 'NO_HTTPONLY_COOKIE',
+  NO_SAMESITE_COOKIE: 'NO_SAMESITE_COOKIE',
   PATH_TRAVERSAL: 'PATH_TRAVERSAL',
   SQL_INJECTION: 'SQL_INJECTION',
   SSRF: 'SSRF',

packages/dd-trace/test/appsec/iast/analyzers/no-samesite-cookie-analyzer.express.plugin.spec.js

@@ -0,0 +1,75 @@
+'use strict'
+
+const { prepareTestServerForIastInExpress } = require('../utils')
+const fs = require('fs')
+const os = require('os')
+const path = require('path')
+describe('no SameSite cookie vulnerability', () => {
+  let setCookieFunctions
+  const setCookieFunctionsFilename = 'set-cookie-express-functions.js'
+  const setCookieFunctionsPath = path.join(os.tmpdir(), setCookieFunctionsFilename)
+
+  before(() => {
+    fs.copyFileSync(path.join(__dirname, 'resources', 'set-cookie-express-functions.js'), setCookieFunctionsPath)
+    setCookieFunctions = require(setCookieFunctionsPath)
+  })
+
+  after(() => {
+    fs.unlinkSync(setCookieFunctionsPath)
+  })
+
+  withVersions('express', 'express', '>=4.15.0', version => {
+    // Oldest express4 versions do not support sameSite property in cookie method
+    prepareTestServerForIastInExpress('in express', version,
+      (testThatRequestHasVulnerability, testThatRequestHasNoVulnerability) => {
+        testThatRequestHasVulnerability((req, res) => {
+          setCookieFunctions.insecureWithResCookieMethod('insecure', 'cookie', res)
+        }, 'NO_SAMESITE_COOKIE', {
+          occurrences: 1,
+          location: {
+            path: setCookieFunctionsFilename,
+            line: 4
+          }
+        })
+
+        testThatRequestHasVulnerability((req, res) => {
+          setCookieFunctions.insecureWithResCookieMethod('insecure', 'cookie', res)
+          setCookieFunctions.insecureWithResCookieMethod('insecure2', 'cookie2', res)
+        }, 'NO_SAMESITE_COOKIE', {
+          occurrences: 2,
+          location: {
+            path: setCookieFunctionsFilename,
+            line: 4
+          }
+        })
+
+        testThatRequestHasVulnerability((req, res) => {
+          setCookieFunctions.insecureWithResHeaderMethod('insecure', 'cookie', res)
+        }, 'NO_SAMESITE_COOKIE', {
+          occurrences: 1,
+          location: {
+            path: setCookieFunctionsFilename,
+            line: 8
+          }
+        })
+
+        testThatRequestHasVulnerability((req, res) => {
+          setCookieFunctions.insecureWithResHeaderMethodWithArray('insecure', 'cookie', 'insecure2', 'cookie2', res)
+        }, 'NO_SAMESITE_COOKIE', {
+          occurrences: 2,
+          location: {
+            path: setCookieFunctionsFilename,
+            line: 12
+          }
+        })
+
+        testThatRequestHasNoVulnerability((req, res) => {
+          res.cookie('secure', 'cookie', { secure: true, httpOnly: true, sameSite: 'strict' })
+          res.cookie('secure', 'cookie', { secure: true, httpOnly: true, sameSite: true })
+          res.clearCookie('deleteinsecure')
+          res.header('set-cookie', 'other=secure; Secure; HttpOnly; SameSite=strict')
+          res.header('set-cookie', ['other=safe; ; SameSite=strict', 'more=safe2; Secure; HttpOnly; SameSite=strict'])
+        }, 'NO_SAMESITE_COOKIE')
+      })
+  })
+})

packages/dd-trace/test/appsec/iast/analyzers/no-samesite-cookie-analyzer.spec.js

@@ -0,0 +1,66 @@
+'use strict'
+
+const { prepareTestServerForIast } = require('../utils')
+const Analyzer = require('../../../../src/appsec/iast/analyzers/vulnerability-analyzer')
+const { NO_SAMESITE_COOKIE } = require('../../../../src/appsec/iast/vulnerabilities')
+const analyzer = new Analyzer()
+
+describe('no SameSite cookie analyzer', () => {
+  it('Expected vulnerability identifier', () => {
+    expect(NO_SAMESITE_COOKIE).to.be.equals('NO_SAMESITE_COOKIE')
+  })
+
+  // In these test, even when we are having multiple vulnerabilities, all the vulnerabilities
+  // are in the same cookies method, and it is expected to detect both even when the max operations is 1
+  const iastConfig = {
+    enabled: true,
+    requestSampling: 100,
+    maxConcurrentRequests: 1,
+    maxContextOperations: 1
+  }
+  prepareTestServerForIast('no HttpOnly cookie analyzer',
+    (testThatRequestHasVulnerability, testThatRequestHasNoVulnerability) => {
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', 'key=value')
+      }, NO_SAMESITE_COOKIE, 1, function (vulnerabilities) {
+        expect(vulnerabilities[0].evidence.value).to.be.equals('key')
+        expect(vulnerabilities[0].hash).to.be.equals(analyzer._createHash('NO_SAMESITE_COOKIE:key'))
+      })
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value'])
+      }, NO_SAMESITE_COOKIE, 1)
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value; SameSite=Lax'])
+      }, NO_SAMESITE_COOKIE, 1)
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value; SameSite=None'])
+      }, NO_SAMESITE_COOKIE, 1)
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value', 'key2=value2'])
+      }, NO_SAMESITE_COOKIE, 2)
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value', 'key2=value2; Secure'])
+      }, NO_SAMESITE_COOKIE, 2)
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value', 'key2=value2; SameSite=strict'])
+      }, NO_SAMESITE_COOKIE, 1)
+
+      testThatRequestHasVulnerability((req, res) => {
+        res.setHeader('set-cookie', ['key=value; SameSite=strict', 'key2=value2; Secure'])
+      }, NO_SAMESITE_COOKIE, 1)
+
+      testThatRequestHasNoVulnerability((req, res) => {
+        res.setHeader('set-cookie', 'key=value; SameSite=strict')
+      }, NO_SAMESITE_COOKIE)
+
+      testThatRequestHasNoVulnerability((req, res) => {
+        res.setHeader('set-cookie', 'key=')
+      }, NO_SAMESITE_COOKIE)
+    }, iastConfig)
+})