module PacketFu

# File lib/packetfu/common.rb, line 51
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
  self.clear_packet_groups
  @packet_classes.sort_by! { |x| x.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/common.rb, line 46
def self.classes
  constants.map { |const| const_get(const) if const_get(const).kind_of? Class}.compact
end

# File lib/packetfu/common.rb, line 95
def self.clear_packet_groups
  @packet_class_prefixes = nil
  @packet_classes_by_layer = nil
  @packet_classes_by_layer_without_application = nil
end

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

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

# File lib/packetfu/common.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/common.rb, line 72
def self.packet_classes
  @packet_classes || []
end

# File lib/packetfu/common.rb, line 83
def self.packet_classes_by_layer
  return [] if @packet_classes.nil?
  self.reset_packet_groups unless @packet_classes_by_layer
  @packet_classes_by_layer
end

# File lib/packetfu/common.rb, line 89
def self.packet_classes_by_layer_without_application
  return [] if @packet_classes.nil?
  self.reset_packet_groups unless @packet_classes_by_layer_without_application
  @packet_classes_by_layer_without_application
end

# File lib/packetfu/common.rb, line 77
def self.packet_prefixes
  return [] if @packet_classes.nil?
  self.reset_packet_groups unless @packet_class_prefixes
  @packet_class_prefixes
end

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

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

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

# File lib/packetfu/common.rb, line 101
def self.reset_packet_groups
              @packet_class_prefixes = @packet_classes.map {|p| p.to_s.split("::").last.to_s.downcase.gsub(/packet$/,"")}
  @packet_classes_by_layer = @packet_classes.sort_by { |pclass| pclass.layer }.reverse
  @packet_classes_by_layer_without_application = @packet_classes_by_layer.reject { |pclass| pclass.layer_symbol == :application }
end

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

# File lib/packetfu/common.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