asmrepl 1.0.3 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1fea751dab8edd03b3cecfb3ea60417c34a9464f19c6e2bc2c2b67b4dcc209cd
-  data.tar.gz: db10a7a8e9a385bcafa3d517dc851774f52a657184334c493627e7cd09d7f870
+  metadata.gz: 7045e3fe0facfdc15e21bb3dfebcaf5232eb68c144ac07e6bf61588979ca3c7f
+  data.tar.gz: 158f3aea1088ae25397dccfc6bf593449df3482f031c025c6dbf33ce372f6a53
 SHA512:
-  metadata.gz: cf9202d1ec0d1700f404f6b8e3c353b56b1985b9a36b4e7520857ac228815439d424c63936b840b56c4a0d073a5814ac699e4afa146ffa73e360b39ec526f05f
-  data.tar.gz: 7c999f1a3b51ea5156e4224296b2df27ca509ae5a4becec8f9d2c57ef1a6af4338993137efe98e98f75ffed1fff791a25c530e5bdd75d36b18f025a8e8aa4a50
+  metadata.gz: d039b567f9cb938f56f86c1ea02a9c16a3c266748fb16d37b5c710c66618010dba32b13755d778c9dded6a56bb521fb61198281297543aad9f3f6c39028510b4
+  data.tar.gz: 7f37562a9568489d61d9eb3144310d3c372fd5fe7ab07b3b0786c268bc8a2b20077a52736723d0f684930924614083cc4df2a817e6f91f8f6b7afae438aeb0fc

data/README.md

@@ -2,11 +2,6 @@
 
 This is a REPL for assembly language.
 
-## Linux requirements
-```
-$ sudo apt-get install libcapstone-dev
-```
-
 ## Usage
 
 Install the gem:
@@ -27,16 +22,16 @@ When the REPL starts, it will display all register values and flags:
 
 ```
 ================== CPU STATE ===================
-rax  000000000000000000  r8   0x0000600001782be0
-rbx  000000000000000000  r9   0x00007fbf9b0068c0
-rcx  0x0000000109dae951  r10  000000000000000000
-rdx  0x000000000000000c  r11  0x000000010999c000
-rdi  0x00007ff7b6b2bbf0  r12  000000000000000000
-rsi  0x00000001096315fd  r13  0x00007ff7b6b2bdc0
-rbp  0x00007ff7b6b2bc40  r14  000000000000000000
-rsp  0x00007ff7b6b2bc38  r15  000000000000000000
-
-rip     0x000000010999c001
+rax  000000000000000000  r8   0x00007f89d0f04640
+rbx  000000000000000000  r9   0x0000000000000004
+rcx  0x00007f89d0f04a50  r10  000000000000000000
+rdx  0x..fc611d3f0aa2900d4  r11  0x00000001033a4000
+rdi  0x00007ff7bd126148  r12  000000000000000000
+rsi  000000000000000000  r13  0x00007ff7bd125dc0
+rbp  0x00007ff7bd125c40  r14  000000000000000000
+rsp  0x00007ff7bd125c38  r15  000000000000000000
+
+rip     0x00000001033a4001
 rflags  0x0000000000000246
 cs      0x000000000000002b
 fs      000000000000000000
@@ -44,46 +39,58 @@ gs      000000000000000000
 
 FLAGS: ["PF", "ZF", "IF"]
 
->> 
+(rip 0x00000001033a4001)>
 ```
 
 Then you can issue commands and inspect register values.  Let's write to the
 `rax` register and inspect its value:
 
 ```
->> mov rax, 5
->> rax
+(rip 0x00000001033a4001)> mov rax, 5
+=============== REGISTER CHANGES ===============
+rax     000000000000000000 => 0x0000000000000005
+
+(rip 0x00000001033a4009)> rax
 0x0000000000000005
->> 
+(rip 0x00000001033a4009)>
 ```
 
 Now let's write to the `rbx` register and add the two values:
 
 ```
->> mov rbx, 3
->> add rax, rbx
->> rax
+(rip 0x00000001033a4009)> mov rbx, 3
+=============== REGISTER CHANGES ===============
+rbx     000000000000000000 => 0x0000000000000003
+
+(rip 0x00000001033a4011)> add rax, rbx
+=============== REGISTER CHANGES ===============
+rax     0x0000000000000005 => 0x0000000000000008
+rflags  0x0000000000000246 => 0x0000000000000202
+
+FLAGS: ["IF"]
+
+(rip 0x00000001033a4015)> rax
 0x0000000000000008
->> rbx
+(rip 0x00000001033a4015)> rbx
 0x0000000000000003
->> 
+(rip 0x00000001033a4015)>
 ```
 
 Finally, lets check all values in the CPU:
 
 ```
->> cpu
+(rip 0x00000001033a4015)> cpu
 ================== CPU STATE ===================
-rax  0x0000000000000008  r8   0x0000600001d848a0
-rbx  0x0000000000000003  r9   0x00007fced316f850
-rcx  0x00000001017da951  r10  000000000000000000
-rdx  0x000000000000000c  r11  0x00000001013cc000
-rdi  0x00007ff7bf0fdbf0  r12  000000000000000000
-rsi  0x000000010105f5fd  r13  0x00007ff7bf0fddc0
-rbp  0x00007ff7bf0fdc40  r14  000000000000000000
-rsp  0x00007ff7bf0fdc38  r15  000000000000000000
-
-rip     0x00000001013cc029
+rax  0x0000000000000008  r8   0x00007f89d0f04640
+rbx  0x0000000000000003  r9   0x0000000000000004
+rcx  0x00007f89d0f04a50  r10  000000000000000000
+rdx  0x..fc611d3f0aa2900d4  r11  0x00000001033a4000
+rdi  0x00007ff7bd126148  r12  000000000000000000
+rsi  000000000000000000  r13  0x00007ff7bd125dc0
+rbp  0x00007ff7bd125c40  r14  000000000000000000
+rsp  0x00007ff7bd125c38  r15  000000000000000000
+
+rip     0x00000001033a4015
 rflags  0x0000000000000202
 cs      0x000000000000002b
 fs      000000000000000000
@@ -91,5 +98,5 @@ gs      000000000000000000
 
 FLAGS: ["IF"]
 
->> 
+(rip 0x00000001033a4015)>
 ```

data/asmrepl.gemspec

@@ -20,5 +20,5 @@ Gem::Specification.new do |s|
   s.add_development_dependency 'minitest', '~> 5.14'
   s.add_development_dependency 'crabstone', '~> 4.0'
   s.add_development_dependency 'rake', '~> 13.0'
-  s.add_dependency 'fisk', '~> 2'
+  s.add_dependency 'fisk', '~> 2.3.1'
 end

data/lib/asmrepl/repl.rb

@@ -27,13 +27,40 @@ module ASMREPL
     end
 
     def display_state state
-      puts " CPU STATE ".center(48, "=")
+      puts bold(" CPU STATE ".center(48, "="))
       puts state
       puts
       puts "FLAGS: #{state.flags.inspect}"
       puts
     end
 
+    def display_state_change last_state, state
+      puts bold(" REGISTER CHANGES ".center(48, "="))
+      show_flags = false
+
+      state.fields.each do |field|
+        next if field == "rip"
+
+        if last_state[field] != state[field]
+          print "#{field.ljust(6)}  "
+          print sprintf("%#018x", last_state[field])
+          print " => "
+          puts bold(sprintf("%#018x", state[field]))
+        end
+      end
+
+      if last_state.flags != state.flags
+        puts
+        puts "FLAGS: #{state.flags.inspect}"
+      end
+
+      puts
+    end
+
+    def bold string
+      "\e[1m#{string}\e[0m"
+    end
+
     def start
       pid = fork {
         CFuncs.traceme
@@ -42,13 +69,18 @@ module ASMREPL
 
       tracer = CFuncs::Tracer.new pid
       should_cpu = true
+      last_state = nil
+
       while tracer.wait
         state = tracer.state
 
         # Show CPU state once on boot
-        if should_cpu
+        if last_state.nil?
           display_state state
-          should_cpu = false
+          last_state = state
+        else
+          display_state_change last_state, state
+          last_state = state
         end
 
         # Move the JIT buffer to the current instruction pointer
@@ -58,7 +90,8 @@ module ASMREPL
         begin
           loop do
             cmd = nil
-            text = Reline.readmultiline(">> ", use_history) do |multiline_input|
+            prompt = sprintf("(rip %#018x)> ", state.rip)
+            text = Reline.readmultiline(prompt, use_history) do |multiline_input|
               if multiline_input =~ /\A\s*(\w+)\s*\Z/
                 register = $1
                 cmd = [:read, register]
@@ -71,8 +104,21 @@ module ASMREPL
             case cmd
             in :run
               break if text.chomp.empty?
-              binary = @assembler.assemble @parser.parse text.chomp
-              binary.bytes.each { |byte| @buffer.putc byte }
+              begin
+                parser_result = @parser.parse text.chomp
+              rescue
+                puts "Invalid intruction"
+                next
+              end
+
+              begin
+                binary = @assembler.assemble parser_result
+                binary.bytes.each { |byte| @buffer.putc byte }
+              rescue Fisk::Errors::InvalidInstructionError => e
+                # Print an error message when the instruction is invalid
+                puts e.message
+                next
+              end
               break
             in [:read, "cpu"]
               display_state state
@@ -88,6 +134,7 @@ module ASMREPL
             end
           end
         rescue Interrupt
+          puts ""
           exit 0
         end
         tracer.continue

data/lib/asmrepl/version.rb

@@ -1,3 +1,3 @@
 module ASMREPL
-  VERSION = '1.0.3'
+  VERSION = '1.1.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: asmrepl
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.1.0
 platform: ruby
 authors:
 - Aaron Patterson
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-12-01 00:00:00.000000000 Z
+date: 2021-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest
@@ -58,14 +58,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: 2.3.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: 2.3.1
 description: Tired of writing assembly and them assembling it? Now you can write assembly
   and evaluate it!
 email: tenderlove@ruby-lang.org