ronin-support 0.5.2 → 1.0.0.beta2

data/lib/ronin/support/binary/ctypes/enum_type.rb

@@ -0,0 +1,181 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all enum types.
+        #
+        class EnumType < Type
+
+          # The underlying integer type.
+          #
+          # @return [IntType]
+          attr_reader :int_type
+
+          # The enum's mapping of symbols to their integer values.
+          #
+          # @return [Hash{Symbol => Integer}]
+          attr_reader :mapping
+
+          # The reverse mapping of the enum's integers to their symbol values.
+          #
+          # @return [Hash{Integer => Symbol}]
+          attr_reader :reverse_mapping
+
+          #
+          # Initializes the enum type.
+          #
+          # @param [IntType] int_type
+          #   The underlying int type for the enum.
+          #
+          # @param [Hash{Symbol => Integer}] mapping
+          #   The mapping of Symbols to Integer values for the enum.
+          #
+          def initialize(int_type,mapping)
+            super(pack_string: nil)
+
+            @int_type = int_type
+            @mapping  = mapping
+            @reverse_mapping = @mapping.invert
+          end
+
+          #
+          # The uninitinalized value for the enum.
+          #
+          # @return [Symbol, Integer]
+          #   The enum Symbol which maps to `0` or `0`.
+          #
+          def uninitialized_value
+            default_value = @int_type.uninitialized_value
+
+            @reverse_mapping.fetch(default_value,default_value)
+          end
+
+          #
+          # The size of the enum type.
+          #
+          # @return [Integer]
+          #   The size of the underlying {#int_type} in bytes.
+          #
+          def size
+            @int_type.size
+          end
+
+          #
+          # The alignment of the enum type.
+          #
+          # @return [Integer]
+          #   The alignment of the underlying {#int_type} in bytes.
+          #
+          def alignment
+            @int_type.alignment
+          end
+
+          #
+          # Creates a copy of the enum type with different alignment.
+          #
+          # @param [Integer] new_alignment
+          #   The new alignment for the enum type.
+          #
+          # @return [EnumType]
+          #   The new enum type.
+          #
+          def align(new_alignment)
+            self.class.new(@int_type.align(new_alignment),@mapping)
+          end
+
+          #
+          # Packs an enum value.
+          #
+          # @param [Symbol, Integer] value
+          #   The enum value. Can be either a Symbol from the enum or an
+          #   Integer.
+          #
+          # @return [String]
+          #   The packed enum value.
+          #
+          # @raise [ArgumentError]
+          #   The enum value either was not a Symbol or an Integer, or was a
+          #   Symbol value not in {#mapping}.
+          #
+          def pack(value)
+            int = case value
+                  when Integer then value
+                  when Symbol
+                    @mapping.fetch(value) do
+                      raise(ArgumentError,"invalid enum value: #{value.inspect}")
+                    end
+                  else
+                    raise(ArgumentError,"enum value must be a Symbol or an Integer: #{value.inspect}")
+                  end
+
+            @int_type.pack(int)
+          end
+
+          #
+          # Unpacks a previously packed enum value.
+          #
+          # @param [String] data
+          #   The packed enum value.
+          #
+          # @return [Symbol, Integer]
+          #   The enum Symbol value or an Integer if the Integer was not in
+          #   {#reverse_mapping}.
+          #
+          def unpack(data)
+            int = @int_type.unpack(data)
+
+            return @reverse_mapping.fetch(int,int)
+          end
+
+          #
+          # Enqueues a value onto the flat list of values.
+          #
+          # @param [Array] values
+          #   The flat array of values.
+          #
+          # @param [Symbol, Integer] value
+          #   The value to enqueue.
+          #
+          def enqueue_value(values,value)
+            @int_type.enqueue_value(values,value)
+          end
+
+          #
+          # Dequeues a value from the flat list of values.
+          #
+          # @param [Array] values
+          #   The flat array of values.
+          #
+          # @return [Symbol, Integer]
+          #   The dequeued value.
+          #
+          def dequeue_value(values)
+            @int_type.dequeue_value(values)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/float32_type.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/float_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all `Float32` types.
+        #
+        class Float32Type < FloatType
+
+          #
+          # Initializes the `Float32` type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(size: 4, **kwargs)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/float64_type.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/float_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all `Float64` types.
+        #
+        class Float64Type < FloatType
+
+          #
+          # Initializes the `Float64` type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(size: 8, **kwargs)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/float_type.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/scalar_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all floating-point types.
+        #
+        class FloatType < ScalarType
+
+          #
+          # Initializes the float type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [Integer] :size
+          #   The size in bytes of the type.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(signed: true, **kwargs)
+          end
+
+          #
+          # Returns an uninitialized Float value.
+          #
+          # @return [0.0]
+          #
+          def uninitialized_value
+            0.0
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/int16_type.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/int_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all `Int16` types.
+        #
+        class Int16Type < IntType
+
+          #
+          # Initializes the `Int16` type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(signed: true, size: 2, **kwargs)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/int32_type.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/int_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all `Int32` types.
+        #
+        class Int32Type < IntType
+
+          #
+          # Initializes the `Int32` type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(signed: true, size: 4, **kwargs)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/int64_type.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/int_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all `Int64` types.
+        #
+        class Int64Type < IntType
+
+          #
+          # Initializes the `Int64` type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(signed: true, size: 8, **kwargs)
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/{support.rb → binary/ctypes/int8_type.rb}

@@ -1,7 +1,6 @@
+# frozen_string_literal: true
 #
-# Copyright (c) 2006-2021 Hal Brodigan (postmodern.mod3 at gmail.com)
-#
-# This file is part of ronin-support.
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # ronin-support is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Lesser General Public License as published
@@ -17,26 +16,26 @@
 # along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-require 'ronin/ui/output'
-require 'ronin/network'
+require 'ronin/support/binary/ctypes/int_type'
 
 module Ronin
-  #
-  # The main namespace for `ronin-support`.
-  #
   module Support
-    include UI::Output::Helpers
-    include Network::DNS
-    include Network::TCP
-    include Network::UDP
-    include Network::SSL
-    include Network::UNIX
-    include Network::SMTP
-    include Network::ESMTP
-    include Network::POP3
-    include Network::IMAP
-    include Network::Telnet
-    include Network::FTP
-    include Network::HTTP
+    module Binary
+      module CTypes
+        #
+        # Base class for all `int8_t` types.
+        #
+        class Int8Type < IntType
+
+          #
+          # Initializes the integer type.
+          #
+          def initialize
+            super(signed: true, size: 1, endian: nil, pack_string: 'c')
+          end
+
+        end
+      end
+    end
   end
 end

data/lib/ronin/support/binary/ctypes/int_type.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# ronin-support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# ronin-support is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
+#
+
+require 'ronin/support/binary/ctypes/scalar_type'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        #
+        # Base class for all signed integer types.
+        #
+        class IntType < ScalarType
+
+          #
+          # Initializes the integer type.
+          #
+          # @param [Hash{Symbol => Object}] kwargs
+          #   Additional keyword arguments for {Type#initialize}.
+          #
+          # @option kwargs [Integer] :size
+          #   The size in bytes of the type.
+          #
+          # @option kwargs [:little, :big, nil] :endian
+          #   The endian-ness of the integer type.
+          #
+          # @option kwargs [String] :pack_string
+          #   The String for `Array#pack` or `String#unpack`.
+          #
+          def initialize(**kwargs)
+            super(signed: true, **kwargs)
+          end
+
+          #
+          # Returns an uninitialized signed integer value.
+          #
+          # @return [0]
+          #
+          def uninitialized_value
+            0
+          end
+
+        end
+      end
+    end
+  end
+end