dd-trace/packages/dd-trace/src/appsec/iast/taint-tracking/index.js

Datadog APM tracing client for JavaScript

'use strict'

const kafkaContextPlugin = require('../context/kafka-ctx-plugin')
const {
  createTransaction,
  removeTransaction,
  setMaxTransactions,
  enableTaintOperations,
  disableTaintOperations,
} = require('./operations')

const taintTrackingPlugin = require('./plugin')
const kafkaConsumerPlugin = require('./plugins/kafka')

module.exports = {
  enableTaintTracking (config, telemetryVerbosity) {
    enableTaintOperations(telemetryVerbosity)
    taintTrackingPlugin.enable(config)

    kafkaContextPlugin.enable(config)
    kafkaConsumerPlugin.enable(config)

    setMaxTransactions(config.maxConcurrentRequests)
  },
  disableTaintTracking () {
    disableTaintOperations()
    taintTrackingPlugin.disable()

    kafkaContextPlugin.disable()
    kafkaConsumerPlugin.disable()
  },
  setMaxTransactions,
  createTransaction,
  removeTransaction,
  taintTrackingPlugin,
}