flexnbd-c/tests/acceptance/flexnbd/constants.rb

# encoding: utf-8

module FlexNBD

  def self.binary( str )
    if str.respond_to? :force_encoding
      str.force_encoding "ASCII-8BIT"
    else
      str
    end
  end

  # eeevil is his one and only name...
  def self.read_constants
    parents = []
    current = File.expand_path(".")
    while current != "/"
      parents << current
      current = File.expand_path( File.join( current, ".." ) )
    end

    source_root = parents.find do |dirname|
      File.directory?( File.join( dirname, "src" ) )
    end

    fail "No source root!" unless source_root

    headers = Dir[File.join( source_root, "src", "{common,proxy,server}","*.h" ) ]

    headers.each do |header_filename|
      txt_lines = File.readlines( header_filename )
      txt_lines.each do |line|
        if line =~ /^#\s*define\s+([A-Z0-9_]+)\s+(\d+)\s*$/
          # Bodge until I can figure out what to do with #ifdefs
          const_set($1, $2.to_i) unless constants.include?( $1 )
        end
      end
    end

  end

  read_constants()

  REQUEST_MAGIC = binary("\x25\x60\x95\x13") unless defined?(REQUEST_MAGIC)
  REPLY_MAGIC = binary("\x67\x44\x66\x98") unless defined?(REPLY_MAGIC)

end # module FlexNBD