ronin-support 0.5.2 → 1.0.0.beta2

data/lib/ronin/support/binary/cstring.rb

@@ -0,0 +1,259 @@
+# 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/memory'
+
+module Ronin
+  module Support
+    module Binary
+      #
+      # Represents a null terminated C string.
+      #
+      # @api public
+      #
+      # @since 1.0.0
+      #
+      class CString < Memory
+
+        # Null byte
+        NULL = "\0".encode(Encoding::ASCII_8BIT).freeze
+
+        #
+        # Initializes the C string.
+        #
+        # @param [String, ByteSlice, nil] value
+        #   The contents of the C string.
+        #
+        def initialize(value=nil)
+          case value
+          when String
+            if value.include?(NULL)
+              super(value)
+            else
+              # ensure the C String ends in or contains a NULL byte.
+              super("#{value}#{NULL}")
+            end
+          when nil
+            # initialize with a single \0 byte
+            super(1)
+          else
+            super(value)
+          end
+        end
+
+        #
+        # Creates a C string.
+        #
+        # @param [Array<Integer>, Array<String>] values
+        #   The bytes or characters for the new C string.
+        #
+        # @return [CString]
+        #   The newly created C string.
+        #
+        # @example Create a C string from a series of bytes:
+        #   Binary::CString[0x41, 0x41, 0x41, 0x00, 0x00, 0x00]
+        #
+        # @example Create a C string from a series of chars:
+        #   Binary::CString['A', 'A', 'A']
+        #
+        # @example Create a C string from a String:
+        #   Binary::CString["AAA"]
+        #
+        # @see #initialize
+        #
+        def self.[](*values)
+          buffer = String.new
+
+          values.each do |element|
+            buffer << case element
+                      when Integer then element.chr
+                      else              element.to_s
+                      end
+          end
+
+          # ensure the C String ends in or contains a NULL byte.
+          buffer << NULL unless buffer.include?(NULL)
+
+          return new(buffer)
+        end
+
+        #
+        # Concatinates a character, byte, or String to the C string.
+        #
+        # @param [String, Integer, #to_s] value
+        #   The other String to concat to the C string.
+        #
+        # @return [self]
+        #
+        def concat(value)
+          value = case value
+                  when Integer then value.chr
+                  else              value.to_s
+                  end
+          value_size = value.bytesize
+
+          unless value.include?(NULL)
+            value = "#{value}#{NULL}"
+            value_size += 1
+          end
+
+          self[null_index,value_size] = value
+          return self
+        end
+
+        alias << concat
+
+        #
+        # Creates a new C string by adding two C strings together.
+        #
+        # @param [#to_s, Integer] other
+        #   The other String or an offset.
+        #
+        # @return [CString, ByteSlice]
+        #   The new combined C string, or a {ByteSlice} if an offset Integer
+        #   was given.
+        #
+        def +(other)
+          case other
+          when Integer then super(other)
+          else              CString.new(to_s + other.to_s)
+          end
+        end
+
+        #
+        # Enumerates over each characters within the C string.
+        #
+        # @yield [char]
+        #   If a block is given, it will be passed each character within the C
+        #   string.
+        #
+        # @yieldparam [String] char
+        #   A character within the C string.
+        #
+        # @return [Enumerator]
+        #   If no block is given, an Enumerator will be returned.
+        #
+        def each_char
+          return enum_for(__method__) unless block_given?
+
+          @string.each_char do |char|
+            break if char == NULL
+            yield char
+          end
+        end
+
+        #
+        # The characters within the C string.
+        #
+        # @return [Array<String>]
+        #   The Array of characters within the C string.
+        #
+        def chars
+          each_char.to_a
+        end
+
+        #
+        # Enumerates over each byte within the C string.
+        #
+        # @yield [byte]
+        #   If a block is given, it will be passed each character within the C
+        #   string.
+        #
+        # @yieldparam [Integer] byte
+        #   A byte within the C string.
+        #
+        # @return [Enumerator]
+        #   If no block is given, an Enumerator will be returned.
+        #
+        def each_byte(&block)
+          return enum_for(__method__) unless block_given?
+
+          @string.each_byte do |byte|
+            break if byte == 0x00
+            yield byte
+          end
+        end
+
+        #
+        # The bytes within the C string.
+        #
+        # @return [Array<Integer>]
+        #   The Array of bytes within the C string.
+        #
+        def bytes
+          each_byte.to_a
+        end
+
+        #
+        # Searches for the char, byte, substring, or regexp within the C string.
+        #
+        # @param [String, Integer, Regexp] key
+        #   The char, byte, substring, or Regexp to search for.
+        #
+        # @return [Integer, nil]
+        #   The index of the char, byte, substring, or Regexp.
+        #
+        def index(key)
+          if (index = @string.index(key))
+            if index < null_index
+              return index
+            end
+          end
+        end
+
+        #
+        # The length of the C string.
+        #
+        # @return [Integer]
+        #   The number of non-null characters in the String.
+        #
+        def length
+          null_index || size
+        end
+
+        alias len length
+
+        #
+        # Converts the C stirng into a regular String.
+        #
+        # @return [String]
+        #   The C string without it's terminating null byte.
+        #
+        def to_s
+          if (length = null_index)
+            @string[0,length]
+          else
+            @string.to_s
+          end
+        end
+
+        private
+
+        #
+        # Finds the index of the first null byte (`\0`).
+        #
+        # @return [Integer, nil]
+        #
+        def null_index
+          @string.index(NULL)
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,98 @@
+# 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 aggregate types.
+        #
+        # @api private
+        #
+        # @since 1.0.0
+        #
+        class AggregateType < Type
+
+          #
+          # The number of elements within the aggregate type.
+          #
+          # @return [Integer, Float::INFINITY]
+          #
+          # @abstract
+          #
+          def length
+            raise(NotImplementedError,"#{self.class}##{__method__} was not implemented")
+          end
+
+          #
+          # Packs the value into the aggregate type's binary format.
+          #
+          # @param [Integer, Float, String] value
+          #   The value to pack.
+          #
+          # @return [String]
+          #   The packed binary data.
+          #
+          # @raise [NotImplementedError]
+          #   {#pack_string} was not set.
+          #
+          # @api public
+          #
+          def pack(value)
+            if @pack_string
+              values = []
+              enqueue_value(values,value)
+
+              return values.pack(@pack_string)
+            else
+              raise(NotImplementedError,"#{self.class} does not define a #pack_string")
+            end
+          end
+
+          #
+          # Unpacks the binary data.
+          #
+          # @param [String] data
+          #   The binary data to unpack.
+          #
+          # @return [Integer, Float, String, nil]
+          #   The unpacked value.
+          #
+          # @raise [NotImplementedError]
+          #   {#pack_string} was not set.
+          #
+          # @api public
+          #
+          def unpack(data)
+            if @pack_string
+              values = data.unpack(@pack_string)
+
+              return dequeue_value(values)
+            else
+              raise(NotImplementedError,"#{self.class} does not define a #pack_string")
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/arch/arm/big_endian.rb

@@ -0,0 +1,77 @@
+# 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/big_endian'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        module Arch
+          module ARM
+            module BigEndian
+              include CTypes::BigEndian
+
+              # The size of a pointer in bytes on ARM (big-endian).
+              ADDRESS_SIZE = 4
+
+              # The `long` type.
+              LONG = CTypes::BigEndian::INT32
+
+              # The `unsigned long` type.
+              ULONG = CTypes::BigEndian::UINT32
+
+              # The "machine word" type.
+              MACHINE_WORD = CTypes::BigEndian::UINT32
+
+              # The `void *` type.
+              POINTER = MACHINE_WORD
+
+              # The ARM (big-endian) types.
+              TYPES = CTypes::BigEndian::TYPES.merge(
+                long:  self::LONG,
+                ulong: self::ULONG,
+
+                machine_word: self::MACHINE_WORD,
+                pointer:      self::POINTER
+              )
+
+              #
+              # Fetches the type from {TYPES}.
+              #
+              # @param [Symbol] name
+              #   The type name to lookup.
+              #
+              # @return [Type]
+              #   The type object from {TYPES}.
+              #
+              # @raise [ArgumentError]
+              #   The type name was unknown.
+              #
+              def self.[](name)
+                TYPES.fetch(name) do
+                  raise(ArgumentError,"unknown ARM (big-endian) type: #{name.inspect}")
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/arch/arm.rb

@@ -0,0 +1,75 @@
+# 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/little_endian'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        module Arch
+          module ARM
+            include LittleEndian
+
+            # The size of a pointer in bytes on ARM.
+            ADDRESS_SIZE = 4
+
+            # The `long` type.
+            LONG = LittleEndian::INT32
+
+            # The `unsigned long` type.
+            ULONG = LittleEndian::UINT32
+
+            # The "machine word" type.
+            MACHINE_WORD = LittleEndian::UINT32
+
+            # The `void *` type.
+            POINTER = MACHINE_WORD
+
+            # The ARM types.
+            TYPES = LittleEndian::TYPES.merge(
+              long:  self::LONG,
+              ulong: self::ULONG,
+
+              machine_word: self::MACHINE_WORD,
+              pointer:      self::POINTER
+            )
+
+            #
+            # Fetches the type from {TYPES}.
+            #
+            # @param [Symbol] name
+            #   The type name to lookup.
+            #
+            # @return [Type]
+            #   The type object from {TYPES}.
+            #
+            # @raise [ArgumentError]
+            #   The type name was unknown.
+            #
+            def self.[](name)
+              TYPES.fetch(name) do
+                raise(ArgumentError,"unknown ARM type: #{name.inspect}")
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/arch/arm64/big_endian.rb

@@ -0,0 +1,77 @@
+# 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/big_endian'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        module Arch
+          module ARM64
+            module BigEndian
+              include CTypes::BigEndian
+
+              # The size of a pointer in bytes on ARM64 (big-endian).
+              ADDRESS_SIZE = 8
+
+              # The `long` type.
+              LONG = CTypes::BigEndian::INT64
+
+              # The `unsigned long` type.
+              ULONG = CTypes::BigEndian::UINT64
+
+              # The "machine word" type.
+              MACHINE_WORD = CTypes::BigEndian::UINT64
+
+              # The `void *` type.
+              POINTER = MACHINE_WORD
+
+              # The ARM64 types.
+              TYPES = CTypes::BigEndian::TYPES.merge(
+                long:  self::LONG,
+                ulong: self::ULONG,
+
+                machine_word: self::MACHINE_WORD,
+                pointer:      self::POINTER
+              )
+
+              #
+              # Fetches the type from {TYPES}.
+              #
+              # @param [Symbol] name
+              #   The type name to lookup.
+              #
+              # @return [Type]
+              #   The type object from {TYPES}.
+              #
+              # @raise [ArgumentError]
+              #   The type name was unknown.
+              #
+              def self.[](name)
+                self::TYPES.fetch(name) do
+                  raise(ArgumentError,"unknown ARM64 (big-endian) type: #{name.inspect}")
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/binary/ctypes/arch/arm64.rb

@@ -0,0 +1,75 @@
+# 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/little_endian'
+
+module Ronin
+  module Support
+    module Binary
+      module CTypes
+        module Arch
+          module ARM64
+            include LittleEndian
+
+            # The size of a pointer in bytes on ARM64.
+            ADDRESS_SIZE = 8
+
+            # The `long` type.
+            LONG = LittleEndian::INT64
+
+            # The `unsigned long` type.
+            ULONG = LittleEndian::UINT64
+
+            # The "machine word" type.
+            MACHINE_WORD = LittleEndian::UINT64
+
+            # The `void *` type.
+            POINTER = MACHINE_WORD
+
+            # The ARM64 types.
+            TYPES = LittleEndian::TYPES.merge(
+              long:  self::LONG,
+              ulong: self::ULONG,
+
+              machine_word: self::MACHINE_WORD,
+              pointer:      self::POINTER
+            )
+
+            #
+            # Fetches the type from {TYPES}.
+            #
+            # @param [Symbol] name
+            #   The type name to lookup.
+            #
+            # @return [Type]
+            #   The type object from {TYPES}.
+            #
+            # @raise [ArgumentError]
+            #   The type name was unknown.
+            #
+            def self.[](name)
+              self::TYPES.fetch(name) do
+                raise(ArgumentError,"unknown ARM64 type: #{name.inspect}")
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end