pwntools 0.1.0 → 1.0.0

data/test/util/cyclic_test.rb

@@ -1,10 +1,11 @@
 # encoding: ASCII-8BIT
 
 require 'test_helper'
+
 require 'pwnlib/util/cyclic'
 
 class CyclicTest < MiniTest::Test
-  include ::Pwnlib::Util::Cyclic::ClassMethods
+  include ::Pwnlib::Util::Cyclic
 
   def test_cyclic
     assert_equal('AAABAACABBABCACBACCBBBCBCCC', cyclic(alphabet: 'ABC', n: 3))

data/test/util/fiddling_test.rb

@@ -1,10 +1,11 @@
 # encoding: ASCII-8BIT
 
 require 'test_helper'
+
 require 'pwnlib/util/fiddling'
 
 class FiddlingTest < MiniTest::Test
-  include ::Pwnlib::Util::Fiddling::ClassMethods
+  include ::Pwnlib::Util::Fiddling
 
   def test_enhex
     assert_equal('4141313233', enhex('AA123'))

data/test/util/getdents_test.rb

@@ -0,0 +1,32 @@
+# encoding: ASCII-8BIT
+
+require 'test_helper'
+
+require 'pwnlib/context'
+require 'pwnlib/util/getdents'
+
+class GetdentsTest < MiniTest::Test
+  include ::Pwnlib::Context
+  include ::Pwnlib::Util::Getdents
+
+  def test_parse
+    context.local(arch: 'i386') do
+      assert_equal("REG README.md\nDIR lib\n",
+                   parse("\x92\x22\x0e\x01\x8f\x4a\xb3\x41" \
+                         "\x18\x00\x52\x45\x41\x44\x4d\x45" \
+                         "\x2e\x6d\x64\x00\x30\x00\x00\x08" \
+                         "\xb5\x10\x34\x01\xff\xff\xff\x7f" \
+                         "\x10\x00\x6c\x69\x62\x00\x00\x04"))
+    end
+    context.local(arch: 'amd64') do
+      assert_equal("REG README.md\nDIR lib\n",
+                   parse("\x92\x22\x0e\x01\x00\x00\x00\x00" \
+                         "\x3d\xf6\x7c\x45\x8f\x4a\xb3\x41" \
+                         "\x20\x00\x52\x45\x41\x44\x4d\x45" \
+                         "\x2e\x6d\x64\x00\x30\x00\x00\x08" \
+                         "\xb5\x10\x34\x01\x00\x00\x00\x00" \
+                         "\xff\xff\xff\xff\xff\xff\xff\x7f" \
+                         "\x18\x00\x6c\x69\x62\x00\x00\x04"))
+    end
+  end
+end

data/test/util/hexdump_test.rb

@@ -1,15 +1,13 @@
-require 'rainbow'
+# encoding: UTF-8
+
+# This test use UTF-8 encoding for strings since the output for hexdump contains lots of UTF-8 characters.
 
 require 'test_helper'
+
 require 'pwnlib/util/hexdump'
 
 class HexDumpTest < MiniTest::Test
-  include ::Pwnlib::Util::HexDump::ClassMethods
-
-  def setup
-    # Default to disable coloring for easier testing.
-    Rainbow.enabled = false
-  end
+  include ::Pwnlib::Util::HexDump
 
   def assert_lines_equal(s1, s2)
     s1l = s1.chomp.lines
@@ -151,13 +149,13 @@ class HexDumpTest < MiniTest::Test
     orig_style = HIGHLIGHT_STYLE
     begin
       $VERBOSE = nil
-      ::Pwnlib::Util::HexDump::ClassMethods.const_set(:HIGHLIGHT_STYLE, ->(s) { "#{s}!" })
+      ::Pwnlib::Util::HexDump.const_set(:HIGHLIGHT_STYLE, ->(s) { "#{s}!" })
       assert_lines_equal(<<-'EOS', hexdump('abcdefghi', highlight: 'aeiou'))
 00000000  61! 62 63 64  65! 66 67 68  69!                        │a!bcd│e!fgh│i!│
 00000009
         EOS
     ensure
-      ::Pwnlib::Util::HexDump::ClassMethods.const_set(:HIGHLIGHT_STYLE, orig_style)
+      ::Pwnlib::Util::HexDump.const_set(:HIGHLIGHT_STYLE, orig_style)
       $VERBOSE = orig_verbose
     end
   end

data/test/util/lists_test.rb

@@ -0,0 +1,21 @@
+# encoding: ASCII-8BIT
+
+require 'test_helper'
+
+require 'pwnlib/util/lists'
+
+class FiddlingTest < MiniTest::Test
+  include ::Pwnlib::Util::Lists
+
+  def test_slice
+    assert_equal(%w(A B C D), slice(1, 'ABCD'))
+    assert_equal(%w(AB CD E), slice(2, 'ABCDE'))
+    assert_equal(%w(AB CD), slice(2, 'ABCDE', underfull_action: :drop))
+    assert_equal(%w(AB CD EX), slice(2, 'ABCDE', underfull_action: :fill, fill_value: 'X'))
+    assert_equal(%w(AB CD EF), slice(2, 'ABCDEF', underfull_action: :fill, fill_value: 'X'))
+    err = assert_raises(ArgumentError) { slice(2, 'ABCDE', underfull_action: :pusheen) }
+    assert_equal('underfull_action expect to be one of :ignore, :drop, and :fill', err.message)
+    err = assert_raises(ArgumentError) { slice(2, 'ABCDE', underfull_action: :fill, fill_value: nil) }
+    assert_equal('fill_value must be a character', err.message)
+  end
+end

data/test/util/packing_test.rb

@@ -1,10 +1,11 @@
 # encoding: ASCII-8BIT
 
 require 'test_helper'
+
 require 'pwnlib/util/packing'
 
 class PackingTest < MiniTest::Test
-  include ::Pwnlib::Util::Packing::ClassMethods
+  include ::Pwnlib::Util::Packing
 
   def test_pack
     assert_equal('ABC',
@@ -115,8 +116,8 @@ class PackingTest < MiniTest::Test
   def test_up_rand
     srand(217)
     [8, 16, 32, 64].each do |sz|
-      u = ->(*x) { public_send("u#{sz}", *x) }
-      p = ->(*x) { public_send("p#{sz}", *x) }
+      u = ->(*x) { ::Pwnlib::Util::Packing.public_send("u#{sz}", *x) }
+      p = ->(*x) { ::Pwnlib::Util::Packing.public_send("p#{sz}", *x) }
       100.times do
         limit = (1 << sz)
         val = rand(0...limit)

metadata

@@ -1,16 +1,73 @@
 --- !ruby/object:Gem::Specification
 name: pwntools
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 1.0.0
 platform: ruby
 authors:
 - peter50216@gmail.com
 - david942j@gmail.com
+- hanhan0912@gmail.com
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-03-23 00:00:00.000000000 Z
+date: 2017-10-17 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: crabstone
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+- !ruby/object:Gem::Dependency
+  name: dentaku
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.11
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.11
+- !ruby/object:Gem::Dependency
+  name: elftools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+- !ruby/object:Gem::Dependency
+  name: keystone-engine
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
 - !ruby/object:Gem::Dependency
   name: rainbow
   requirement: !ruby/object:Gem::Requirement
@@ -43,16 +100,16 @@ dependencies:
   name: minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - '='
       - !ruby/object:Gem::Version
-        version: '5.8'
+        version: 5.10.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - '='
       - !ruby/object:Gem::Version
-        version: '5.8'
+        version: 5.10.1
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -73,42 +130,42 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '12.0'
+        version: '12.1'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '12.0'
+        version: '12.1'
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.46'
+        version: '0.49'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.46'
+        version: '0.49'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.12'
+        version: '0.15'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.12'
+        version: '0.15'
 - !ruby/object:Gem::Dependency
   name: tty-platform
   requirement: !ruby/object:Gem::Requirement
@@ -142,7 +199,10 @@ description: |2
     Implement useful/easy functions first,
     try to be of ruby style and don't follow original pwntools everywhere.
     Would still try to have similar name whenever possible.
-email: 
+email:
+- peter50216@gmail.com
+- david942j@gmail.com
+- hanhan0912@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
@@ -150,43 +210,134 @@ files:
 - README.md
 - Rakefile
 - lib/pwn.rb
+- lib/pwnlib/abi.rb
+- lib/pwnlib/asm.rb
 - lib/pwnlib/constants/constant.rb
 - lib/pwnlib/constants/constants.rb
 - lib/pwnlib/constants/linux/amd64.rb
 - lib/pwnlib/constants/linux/i386.rb
 - lib/pwnlib/context.rb
 - lib/pwnlib/dynelf.rb
+- lib/pwnlib/elf/elf.rb
 - lib/pwnlib/ext/array.rb
 - lib/pwnlib/ext/helper.rb
 - lib/pwnlib/ext/integer.rb
 - lib/pwnlib/ext/string.rb
+- lib/pwnlib/logger.rb
 - lib/pwnlib/memleak.rb
 - lib/pwnlib/pwn.rb
 - lib/pwnlib/reg_sort.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/common.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/infloop.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/memcpy.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/mov.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/nop.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/popad.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/pushstr.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/pushstr_array.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/ret.rb
+- lib/pwnlib/shellcraft/generators/amd64/common/setregs.rb
+- lib/pwnlib/shellcraft/generators/amd64/linux/execve.rb
+- lib/pwnlib/shellcraft/generators/amd64/linux/linux.rb
+- lib/pwnlib/shellcraft/generators/amd64/linux/ls.rb
+- lib/pwnlib/shellcraft/generators/amd64/linux/sh.rb
+- lib/pwnlib/shellcraft/generators/amd64/linux/syscall.rb
+- lib/pwnlib/shellcraft/generators/helper.rb
+- lib/pwnlib/shellcraft/generators/i386/common/common.rb
+- lib/pwnlib/shellcraft/generators/i386/common/infloop.rb
+- lib/pwnlib/shellcraft/generators/i386/common/mov.rb
+- lib/pwnlib/shellcraft/generators/i386/common/nop.rb
+- lib/pwnlib/shellcraft/generators/i386/common/pushstr.rb
+- lib/pwnlib/shellcraft/generators/i386/common/pushstr_array.rb
+- lib/pwnlib/shellcraft/generators/i386/common/setregs.rb
+- lib/pwnlib/shellcraft/generators/i386/linux/execve.rb
+- lib/pwnlib/shellcraft/generators/i386/linux/linux.rb
+- lib/pwnlib/shellcraft/generators/i386/linux/ls.rb
+- lib/pwnlib/shellcraft/generators/i386/linux/sh.rb
+- lib/pwnlib/shellcraft/generators/i386/linux/syscall.rb
+- lib/pwnlib/shellcraft/generators/x86/common/common.rb
+- lib/pwnlib/shellcraft/generators/x86/common/infloop.rb
+- lib/pwnlib/shellcraft/generators/x86/common/mov.rb
+- lib/pwnlib/shellcraft/generators/x86/common/pushstr.rb
+- lib/pwnlib/shellcraft/generators/x86/common/pushstr_array.rb
+- lib/pwnlib/shellcraft/generators/x86/common/setregs.rb
+- lib/pwnlib/shellcraft/generators/x86/linux/execve.rb
+- lib/pwnlib/shellcraft/generators/x86/linux/linux.rb
+- lib/pwnlib/shellcraft/generators/x86/linux/ls.rb
+- lib/pwnlib/shellcraft/generators/x86/linux/sh.rb
+- lib/pwnlib/shellcraft/generators/x86/linux/syscall.rb
+- lib/pwnlib/shellcraft/registers.rb
+- lib/pwnlib/shellcraft/shellcraft.rb
+- lib/pwnlib/timer.rb
+- lib/pwnlib/tubes/buffer.rb
+- lib/pwnlib/tubes/sock.rb
+- lib/pwnlib/tubes/tube.rb
 - lib/pwnlib/util/cyclic.rb
 - lib/pwnlib/util/fiddling.rb
+- lib/pwnlib/util/getdents.rb
 - lib/pwnlib/util/hexdump.rb
+- lib/pwnlib/util/lists.rb
 - lib/pwnlib/util/packing.rb
+- lib/pwnlib/util/ruby.rb
 - lib/pwnlib/version.rb
+- test/abi_test.rb
+- test/asm_test.rb
 - test/constants/constant_test.rb
 - test/constants/constants_test.rb
 - test/context_test.rb
+- test/data/echo.rb
+- test/data/elfs/Makefile
+- test/data/elfs/amd64.frelro.elf
+- test/data/elfs/amd64.frelro.pie.elf
+- test/data/elfs/amd64.nrelro.elf
+- test/data/elfs/amd64.prelro.elf
+- test/data/elfs/i386.frelro.pie.elf
+- test/data/elfs/i386.prelro.elf
+- test/data/elfs/source.cpp
+- test/data/flag
+- test/data/lib32/ld.so.2
+- test/data/lib32/libc.so.6
+- test/data/lib64/ld.so.2
+- test/data/lib64/libc.so.6
 - test/data/victim.c
 - test/data/victim32
 - test/data/victim64
 - test/dynelf_test.rb
+- test/elf/elf_test.rb
 - test/ext_test.rb
 - test/files/use_pwn.rb
 - test/files/use_pwnlib.rb
 - test/full_file_test.rb
+- test/logger_test.rb
 - test/memleak_test.rb
 - test/reg_sort_test.rb
+- test/shellcraft/infloop_test.rb
+- test/shellcraft/linux/ls_test.rb
+- test/shellcraft/linux/sh_test.rb
+- test/shellcraft/linux/syscalls/execve_test.rb
+- test/shellcraft/linux/syscalls/syscall_test.rb
+- test/shellcraft/memcpy_test.rb
+- test/shellcraft/mov_test.rb
+- test/shellcraft/nop_test.rb
+- test/shellcraft/popad_test.rb
+- test/shellcraft/pushstr_array_test.rb
+- test/shellcraft/pushstr_test.rb
+- test/shellcraft/registers_test.rb
+- test/shellcraft/ret_test.rb
+- test/shellcraft/setregs_test.rb
+- test/shellcraft/shellcraft_test.rb
 - test/test_helper.rb
+- test/timer_test.rb
+- test/tubes/buffer_test.rb
+- test/tubes/sock_test.rb
+- test/tubes/tube_test.rb
 - test/util/cyclic_test.rb
 - test/util/fiddling_test.rb
+- test/util/getdents_test.rb
 - test/util/hexdump_test.rb
+- test/util/lists_test.rb
 - test/util/packing_test.rb
-homepage: 
+homepage: https://github.com/peter50216/pwntools-ruby
 licenses:
 - MIT
 metadata: {}
@@ -211,21 +362,60 @@ signing_key:
 specification_version: 4
 summary: pwntools
 test_files:
-- test/files/use_pwn.rb
-- test/files/use_pwnlib.rb
-- test/dynelf_test.rb
-- test/ext_test.rb
-- test/test_helper.rb
+- test/data/elfs/amd64.frelro.pie.elf
+- test/data/elfs/Makefile
+- test/data/elfs/amd64.frelro.elf
+- test/data/elfs/i386.frelro.pie.elf
+- test/data/elfs/amd64.nrelro.elf
+- test/data/elfs/i386.prelro.elf
+- test/data/elfs/amd64.prelro.elf
+- test/data/elfs/source.cpp
+- test/data/victim32
 - test/data/victim.c
 - test/data/victim64
-- test/data/victim32
-- test/reg_sort_test.rb
-- test/full_file_test.rb
+- test/data/lib32/ld.so.2
+- test/data/lib32/libc.so.6
+- test/data/lib64/ld.so.2
+- test/data/lib64/libc.so.6
+- test/data/flag
+- test/data/echo.rb
+- test/asm_test.rb
+- test/test_helper.rb
 - test/context_test.rb
-- test/constants/constant_test.rb
+- test/shellcraft/pushstr_array_test.rb
+- test/shellcraft/memcpy_test.rb
+- test/shellcraft/registers_test.rb
+- test/shellcraft/ret_test.rb
+- test/shellcraft/popad_test.rb
+- test/shellcraft/nop_test.rb
+- test/shellcraft/linux/ls_test.rb
+- test/shellcraft/linux/sh_test.rb
+- test/shellcraft/linux/syscalls/syscall_test.rb
+- test/shellcraft/linux/syscalls/execve_test.rb
+- test/shellcraft/shellcraft_test.rb
+- test/shellcraft/pushstr_test.rb
+- test/shellcraft/setregs_test.rb
+- test/shellcraft/infloop_test.rb
+- test/shellcraft/mov_test.rb
 - test/constants/constants_test.rb
+- test/constants/constant_test.rb
+- test/reg_sort_test.rb
+- test/logger_test.rb
 - test/memleak_test.rb
-- test/util/packing_test.rb
+- test/util/fiddling_test.rb
 - test/util/hexdump_test.rb
+- test/util/getdents_test.rb
+- test/util/packing_test.rb
 - test/util/cyclic_test.rb
-- test/util/fiddling_test.rb
+- test/util/lists_test.rb
+- test/timer_test.rb
+- test/tubes/sock_test.rb
+- test/tubes/buffer_test.rb
+- test/tubes/tube_test.rb
+- test/abi_test.rb
+- test/files/use_pwnlib.rb
+- test/files/use_pwn.rb
+- test/ext_test.rb
+- test/full_file_test.rb
+- test/dynelf_test.rb
+- test/elf/elf_test.rb