ronin-support 0.5.2 → 1.0.0.beta2

data/lib/ronin/support/binary/core_ext/integer.rb

@@ -0,0 +1,305 @@
+# 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'
+require 'ronin/support/binary/bit_flip'
+
+class Integer
+
+  #
+  # Enumerates over every bit flip in the integer.
+  #
+  # @param [Integer, Range(Integer)] bits
+  #   The number of bits to flip or a range of bit indexes to flip.
+  #
+  # @yield [int]
+  #   If a block is given, it will be passed each bit-flipped integer.
+  #
+  # @yieldparam [Integer] int
+  #   The integer but with one of it's bits flipped.
+  #
+  # @return [Enumerator]
+  #   If no block is given, an Enumerator object will be returned.
+  #
+  # @raise [ArgumentError]
+  #   The given bits must be either a Range or an Integer.
+  #
+  # @example bit-flip all eight bits:
+  #   0x41.each_bit_flip(8) { |int| puts "%.8b" % int }
+  #
+  # @example bit-flip bits 8-16:
+  #   0xffff.each_bit_flip(8...16) { |int| puts "%.16b" % int }
+  #
+  # @api public
+  #
+  def each_bit_flip(bits,&block)
+    Ronin::Support::Binary::BitFlip::Integer.each_bit_flip(self,bits,&block)
+  end
+
+  #
+  # Returns every bit flip in the integer.
+  #
+  # @param [Integer, Range(Integer)] bits
+  #   The number of bits to flip or a range of bit indexes to flip.
+  #
+  # @return [Array<Integer>]
+  #   The bit-flipped integers.
+  #
+  # @raise [ArgumentError]
+  #   The given bits must be either a Range or an Integer.
+  #
+  # @example bit-flip all eight bits:
+  #   0x41.bit_flips(8)
+  #
+  # @example bit-flip bits 8-16:
+  #   0xffff.bit_flips(8...16)
+  #
+  # @api public
+  #
+  def bit_flips(bits)
+    Ronin::Support::Binary::BitFlip::Integer.bit_flips(self,bits)
+  end
+
+  alias flip_bits bit_flips
+
+  #
+  # Packs the Integer into a String.
+  #
+  # @param [String, Symbol] argument
+  #   The `Array#pack` String or {Ronin::Support::Binary::Template} type.
+  #
+  # @param [Hash{Symbol => Object}] kwargs
+  #   Additional keyword arguments for
+  #   {Ronin::Support::Binary::CTypes.platform}.
+  #
+  # @option kwargs [:little, :big, :net, nil] :endian
+  #   The desired endianness of the binary format.
+  #
+  # @option kwargs [:x86, :x86_64,
+  #                 :ppc, :ppc64,
+  #                 :mips, :mips_le, :mips_be,
+  #                 :mips64, :mips64_le, :mips64_be,
+  #                 :arm, :arm_le, :arm_be,
+  #                 :arm64, :arm64_le, :arm64_be] :arch
+  #   The desired architecture of the binary format.
+  #
+  # @option kwargs [:linux, :macos, :windows,
+  #                 :bsd, :freebsd, :openbsd, :netbsd] :os
+  #   The Operating System name to lookup.
+  #
+  # @return [String]
+  #   The packed Integer.
+  #
+  # @raise [ArgumentError]
+  #   The given argument was not a `String`, `Symbol`, or valid type name.
+  #
+  # @example using a `Array#pack` format string:
+  #   0x41.pack('V')
+  #   # => "A\0\0\0"
+  #
+  # @example using {Ronin::Support::Binary::CTypes} types:
+  #   0x41.pack(:uint32_le)
+  #   # => "A\x00\x00\x00"
+  #
+  # @example specifying the endian-ness:
+  #   0x41.pack(:uint32, endian: :big)
+  #   # => "\x00\x00\x00A"
+  #
+  # @example specifying the architecture:
+  #   0x41.pack(:ulong, arch: :arm64)
+  #   # => "A\x00\x00\x00\x00\x00\x00\x00"
+  #
+  # @example specifying the architecture and Operating System (OS):
+  #   0x41.pack(:size_t, arch: :arm64, os: :linux)
+  #   # => "A\x00\x00\x00\x00\x00\x00\x00"
+  #
+  # @see http://rubydoc.info/stdlib/core/Array:pack
+  # @see Ronin::Support::Binary::Template
+  #
+  # @api public
+  #
+  def pack(argument, **kwargs)
+    case argument
+    when String
+      [self].pack(argument)
+    when Symbol
+      types = Ronin::Support::Binary::CTypes.platform(**kwargs)
+      type  = types[argument]
+      type.pack(self)
+    else
+      raise(ArgumentError,"invalid pack argument: #{argument}")
+    end
+  end
+
+  #
+  # Converts the integer into an 8-bit unsigned integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 8-bits.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_uint8
+    self & 0xff
+  end
+
+  alias to_u8 to_uint8
+
+  #
+  # Converts the integer into an 16-bit unsigned integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 16-bits.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_uint16
+    self & 0xffff
+  end
+
+  alias to_u16 to_uint16
+
+  #
+  # Converts the integer into an 32-bit unsigned integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 32-bits.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_uint32
+    self & 0xffffffff
+  end
+
+  alias to_u32 to_uint32
+
+  #
+  # Converts the integer into an 64-bit unsigned integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 64-bits.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_uint64
+    self & 0xffffffffffffffff
+  end
+
+  alias to_u64 to_uint64
+
+  #
+  # Converts the integer into an 8-bit signed integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 8-bits with signed preserved.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_int8
+    int = self & 0xff
+
+    if int[7] == 1
+      # interpret the new signed bit
+      int -= 0x100
+    end
+
+    return int
+  end
+
+  alias to_i8 to_int8
+
+  #
+  # Converts the integer into an 16-bit signed integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 16-bits with signed preserved.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_int16
+    int = self & 0xffff
+
+    if int[15] == 1
+      # interpret the new signed bit
+      int -= 0x10000
+    end
+
+    return int
+  end
+
+  alias to_i16 to_int16
+
+  #
+  # Converts the integer into an 32-bit signed integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 32-bits with signed preserved.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_int32
+    int = self & 0xffffffff
+
+    if int[31] == 1
+      # interpret the new signed bit
+      int -= 0x100000000
+    end
+
+    return int
+  end
+
+  alias to_i32 to_int32
+
+  #
+  # Converts the integer into an 64-bit signed integer.
+  #
+  # @return [Integer]
+  #   The integer truncated to 64-bits with signed preserved.
+  #
+  # @api public
+  #
+  # @since 1.0.0
+  #
+  def to_int64
+    int = self & 0xffffffffffffffff
+
+    if int[63] == 1
+      # interpret the new signed bit
+      int -= 0x10000000000000000
+    end
+
+    return int
+  end
+
+  alias to_i64 to_int64
+
+end

data/lib/ronin/support/binary/core_ext/io.rb

@@ -0,0 +1,38 @@
+# 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/stream/methods'
+require 'ronin/support/binary/ctypes'
+
+class IO
+
+  include Ronin::Support::Binary::Stream::Methods
+
+  #
+  # The type system which `read_*`/`write_*`
+  # {Ronin::Support::Binary::Stream::Methods methods} will use.
+  #
+  # @api private
+  #
+  # @see Ronin::Support::Binary::Stream::Methods
+  #
+  def type_system
+    Ronin::Support::Binary::CTypes
+  end
+
+end

data/lib/ronin/support/binary/core_ext/string.rb

@@ -0,0 +1,162 @@
+# 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/core_ext/integer'
+require 'ronin/support/binary/template'
+require 'ronin/support/binary/ctypes'
+require 'ronin/support/binary/bit_flip'
+
+class String
+
+  alias unpack_original unpack
+
+  #
+  # Unpacks the String.
+  #
+  # @param [String, Array<Symbol, (Symbol, Integer)>] arguments
+  #   The `String#unpack` format string or a list of
+  #   {Ronin::Support::Binary::Template} types.
+  #
+  # @param [Hash{Symbol => Object}] kwargs
+  #   Additional keyword arguments for
+  #   {Ronin::Support::Binary::Template#initialize}.
+  #
+  # @option kwargs [:little, :big, :net, nil] :endian
+  #   The desired endianness of the packed data.
+  #
+  # @option kwargs [:x86, :x86_64, :ppc, :ppc64,
+  #                 :arm, :arm_be, :arm64, :arm64_be,
+  #                 :mips, :mips_le, :mips64, :mips64_le, nil] :arch
+  #   The desired architecture that the data was packed for.
+  #
+  # @return [Array]
+  #   The values unpacked from the String.
+  #
+  # @raise [ArgumentError]
+  #   One of the arguments was not a known {Ronin::Support::Binary::Template}
+  #   type.
+  #
+  # @example using {Ronin::Support::Binary::Template} types:
+  #   "A\0\0\0hello\0".unpack(:uint32_le, :string)
+  #   # => [10, "hello"]
+  #
+  # @example using a `String#unpack` format string:
+  #   "A\0\0\0".unpack('V')
+  #   # => 65
+  #
+  # @see https://rubydoc.info/stdlib/core/String:unpack
+  # @see Ronin::Support::Binary::Template
+  #
+  # @since 0.5.0
+  #
+  # @api public
+  #
+  def unpack(*arguments,**kwargs)
+    if (arguments.length == 1 && arguments.first.kind_of?(String))
+      unpack_original(arguments.first)
+    else
+      format = Ronin::Support::Binary::Template.new(arguments,**kwargs)
+      unpack_original(format.pack_string)
+    end
+  end
+
+  alias unpack1_original unpack1
+
+  #
+  # Unpacks a single value from the String.
+  #
+  # @param [String, Symbol] argument
+  #   The `String#unpack` format String (ex: `L<`) or a
+  #   {Ronin::Support::Binary::CTypes} type name (ex: `:uint32_le`).
+  #
+  # @param [Hash{Symbol => Object}] kwargs
+  #   Additional keyword arguments for
+  #   {Ronin::Support::Binary::Template#initialize}.
+  #
+  # @option kwargs [:little, :big, :net, nil] :endian
+  #   The desired endianness of the packed data.
+  #
+  # @option kwargs [:x86, :x86_64, :ppc, :ppc64,
+  #                 :arm, :arm_be, :arm64, :arm64_be,
+  #                 :mips, :mips_le, :mips64, :mips64_le, nil] :arch
+  #   The desired architecture that the data was packed for.
+  #
+  # @return [Integer, Float, String]
+  #   The unpacked value.
+  #
+  # @raise [ArgumentError]
+  #   The given argument was not a String or a Symbol, or the given C type is
+  #   unknown.
+  #
+  # @since 1.0.0
+  #
+  # @api public
+  #
+  def unpack1(argument,**kwargs)
+    case argument
+    when String
+      unpack1_original(argument)
+    when Symbol
+      platform = Ronin::Support::Binary::CTypes.platform(**kwargs)
+      type     = platform[argument]
+
+      unpack1_original(type.pack_string)
+    else
+      raise(ArgumentError,"argument must be either a String or a Symbol: #{argument.inspect}")
+    end
+  end
+
+  #
+  # Enumerates over every bit flip of every byte in the string.
+  #
+  # @yield [string]
+  #   If a block is given, it will be passed each bit-flipped string.
+  #
+  # @yieldparam [String] string
+  #   The String, but with one of it's bits flipped.
+  #
+  # @return [Enumerator]
+  #   If no block is given, an Enumerator object will be returned.
+  #
+  # @example bit-flip all bytes in the String:
+  #   "foo".each_bit_flip { |string| puts string }
+  #
+  # @api public
+  #
+  def each_bit_flip(&block)
+    Ronin::Support::Binary::BitFlip::String.each_bit_flip(self,&block)
+  end
+
+  #
+  # Returns every bit flip of every byte in the string.
+  #
+  # @return [Array<String>]
+  #   The bit-flipped strings.
+  #
+  # @example bit-flip all bytes in the String:
+  #   "foo".bit_flips
+  #
+  # @api public
+  #
+  def bit_flips
+    Ronin::Support::Binary::BitFlip::String.bit_flips(self)
+  end
+
+  alias flip_bits bit_flips
+
+end

data/lib/ronin/{formatting/extensions/text.rb → support/binary/core_ext.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,7 +16,8 @@
 # along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
 #
 
-require 'ronin/formatting/extensions/text/string'
-require 'ronin/formatting/extensions/text/array'
+require 'ronin/support/binary/core_ext/string'
+require 'ronin/support/binary/core_ext/array'
+require 'ronin/support/binary/core_ext/io'
 
-require 'chars/extensions'
+require 'hexdump/core_ext'