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