Module: Datadog::AppSec::WAF

Defined in:

lib/datadog/appsec/waf.rb,
lib/datadog/appsec/waf/version.rb

Defined Under Namespace

Modules: LibDDWAF, VERSION Classes: Context, Handle, Result

Class Method Summary

• .logger=(logger) ⇒ Object
• .object_to_ruby(obj) ⇒ Object
• .ruby_to_object(val) ⇒ Object
• .version ⇒ Object

Class Method Details

.logger=(logger) ⇒ Object

# File 'lib/datadog/appsec/waf.rb', line 348

def self.logger=(logger)
  @log_cb = proc do |level, func, file, line, message, len|
    logger.debug { { level: level, func: func, file: file, line: line, message: message.read_bytes(len) }.inspect }
  end

  Datadog::AppSec::WAF::LibDDWAF.ddwaf_set_log_cb(@log_cb, :ddwaf_log_trace)
end

.object_to_ruby(obj) ⇒ Object

# File 'lib/datadog/appsec/waf.rb', line 320

def self.object_to_ruby(obj)
  case obj[:type]
  when :ddwaf_obj_invalid
    nil
  when :ddwaf_obj_string
    obj[:valueUnion][:stringValue].read_bytes(obj[:nbEntries])
  when :ddwaf_obj_signed
    obj[:valueUnion][:intValue]
  when :ddwaf_obj_unsigned
    obj[:valueUnion][:uintValue]
  when :ddwaf_obj_array
    (0...obj[:nbEntries]).each.with_object([]) do |i, a|
      ptr = obj[:valueUnion][:array] + i * LibDDWAF::Object.size
      e = object_to_ruby(LibDDWAF::Object.new(ptr))
      a << e
    end
  when :ddwaf_obj_map
    (0...obj[:nbEntries]).each.with_object({}) do |i, h|
      ptr = obj[:valueUnion][:array] + i * Datadog::AppSec::WAF::LibDDWAF::Object.size
      o = Datadog::AppSec::WAF::LibDDWAF::Object.new(ptr)
      l = o[:parameterNameLength]
      k = o[:parameterName].read_bytes(l)
      v = object_to_ruby(LibDDWAF::Object.new(ptr))
      h[k] = v
    end
  end
end

.ruby_to_object(val) ⇒ Object

# File 'lib/datadog/appsec/waf.rb', line 243

def self.ruby_to_object(val)
  case val
  when Array
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_array(obj)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    val.each do |e|
      res = LibDDWAF.ddwaf_object_array_add(obj, ruby_to_object(e))
      unless res
        fail LibDDWAF::Error, "Could not add to map object: #{k.inspect} => #{v.inspect}"
      end
    end

    obj
  when Hash
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_map(obj)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    val.each do |k, v|
      res = LibDDWAF.ddwaf_object_map_addl(obj, k.to_s, k.to_s.size, ruby_to_object(v))
      unless res
        fail LibDDWAF::Error, "Could not add to map object: #{k.inspect} => #{v.inspect}"
      end
    end

    obj
  when String
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_stringl(obj, val, val.size)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    obj
  when Symbol
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_stringl(obj, val.to_s, val.size)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    obj
  when Integer
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_string(obj, val.to_s)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    obj
  when Float
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_string(obj, val.to_s)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    obj
  when TrueClass, FalseClass
    obj = LibDDWAF::Object.new
    res = LibDDWAF.ddwaf_object_string(obj, val.to_s)
    if res.null?
      fail LibDDWAF::Error, "Could not convert into object: #{val}"
    end

    obj
  else
    ruby_to_object(''.freeze)
  end
end

.version ⇒ Object

# File 'lib/datadog/appsec/waf.rb', line 236

def self.version
  version = LibDDWAF::Version.new
  LibDDWAF.ddwaf_get_version(version.pointer)

  [version[:major], version[:minor], version[:patch]]
end