Added Program#free_register (closes #57).

Author: postmodern

lib/ronin/asm/program.rb

@@ -219,6 +219,21 @@ def allocate_register(name)
         @allocated_registers << name
       end
 
+      #
+      # Mark the register as de-allocated.
+      #
+      # @param [Symbol] name
+      #   The register name.
+      #
+      # @return [Set<Symbol>]
+      #   The set of allocated registers.
+      #
+      # @since 1.0.0
+      #
+      def free_register(name)
+        @allocated_registers.delete(name)
+      end
+
       #
       # Accesses a register.
       #

spec/program_spec.rb

@@ -197,6 +197,19 @@
     end
   end
 
+  describe "#free_register" do
+    let(:name) { :ebx }
+
+    before do
+      subject.allocate_register(name)
+      subject.free_register(name)
+    end
+
+    it "must remove the register name to #allocated_registers" do
+      expect(subject.allocated_registers).to_not include(name)
+    end
+  end
+
   describe "#register" do
     it "must return a Register" do
       expect(subject.register(:eax)).to be_kind_of(Ronin::ASM::Register)