module PacketFu

# File lib/packetfu.rb, line 73
def self.add_packet_class(klass)
  raise "Need a class" unless klass.kind_of? Class
  if klass.name !~ /[A-Za-z0-9]Packet/
    raise "Packet classes should be named 'ProtoPacket'"
  end
  @packet_classes ||= []
  @packet_classes << klass
  @packet_classes_dirty = true
  @packet_classes.sort! {|x,y| x.name <=> y.name}
end

# File lib/packetfu/version.rb, line 30
def self.at_least?(str)
  this_version = binarize_version(self.version)
  ask_version = binarize_version(str)
  this_version >= ask_version
end

# File lib/packetfu/version.rb, line 12
def self.binarize_version(str)
  if(str.respond_to?(:split) && str =~ /^[0-9]+(\.([0-9]+)(\.[0-9]+)?)?\..+$/)
    bin_major,bin_minor,bin_teeny = str.split(/\x2e/).map {|x| x.to_i}
    bin_version = (bin_major.to_i << 16) + (bin_minor.to_i << 8) + bin_teeny.to_i
  else
    raise ArgumentError, "Compare version malformed. Should be \x22x.y.z\x22"
  end
end

# File lib/packetfu.rb, line 68
def self.classes
  constants.map { |const| const_get(const) if const_get(const).kind_of? Class}.compact
end

# File lib/packetfu.rb, line 29
def self.force_binary(str)
  str.force_encoding Encoding::BINARY if str.respond_to? :force_encoding
end

# File lib/packetfu.rb, line 110
def self.inspect_style
  @inspect_style ||= :dissect
end

# File lib/packetfu.rb, line 115
def self.inspect_style=(arg)
  @inspect_style = case arg
    when :hex, :pretty
      :hex
    when :dissect, :verbose
      :dissect
    when :default, :ugly
      :default
    else
      :dissect
    end
end

# File lib/packetfu/version.rb, line 45
def self.newer_than?(str)
  return false if str == self.version
  !self.older_than?(str)
end

# File lib/packetfu/version.rb, line 37
def self.older_than?(str)
  return false if str == self.version
  this_version = binarize_version(self.version)
  ask_version = binarize_version(str)
  this_version < ask_version
end

# File lib/packetfu.rb, line 94
def self.packet_classes
  @packet_classes || []
end

# File lib/packetfu.rb, line 99
def self.packet_prefixes
  return [] if @packet_classes.nil?
  return @packet_class_prefixes if @packet_classes_dirty == false
  @packet_classes_dirty = false
  @packet_class_prefixes = @packet_classes.map {|p| p.to_s.split("::").last.to_s.downcase.gsub(/packet$/,"")}
  return @packet_class_prefixes
end

# File lib/packetfu.rb, line 63
def self.pcaprub_loaded?
  @pcaprub_loaded
end

# File lib/packetfu.rb, line 41
def self.pcaprub_platform_require
  begin
    require 'pcaprub'
  rescue LoadError
    return false
  end
  @pcaprub_loaded = true 
end

# File lib/packetfu.rb, line 85
def self.remove_packet_class(klass)
  raise "Need a class" unless klass.kind_of? Class
  @packet_classes ||= []
  @packet_classes.delete klass
  @packet_classes_dirty = true
  @packet_classes 
end

# File lib/packetfu.rb, line 16
def self.require_protos(cwd)
  protos_dir = File.join(cwd, "packetfu", "protos")
  Dir.new(protos_dir).each do |fname|
    next unless fname[/\.rb$/]
    begin 
      require File.join(protos_dir,fname)
    rescue
      warn "Warning: Could not load `#{fname}'. Skipping."
    end
  end
end

# File lib/packetfu/version.rb, line 7
def self.version
  VERSION
end

# File lib/packetfu.rb, line 130
def toggle_inspect
  case @inspect_style
  when :hex, :pretty
    @inspect_style = :dissect
  when :dissect, :verbose
    @inspect_style = :default
  when :default, :ugly
    @inspect_style = :hex
  else
    @inspect_style = :dissect
  end
end