Allow the output path to be overriden.

lib/ronin/asm/shellcode.rb

@@ -52,13 +52,19 @@ class Shellcode < Program
       # @param [Hash] options
       #   Additional options.
       #
+      # @option options [String] :output
+      #   The optional output to write the shellcode to. If no `:output` is
+      #   given a tempfile will be used.
+      #
       # @return [String]
       #   The raw object-code of the Shellcode.
       #
       # @see Program#assemble
       #
       def assemble(options={})
-        output = Tempfile.new(['ronin-shellcode', '.bin']).path
+        output = options.fetch(:output) do
+          Tempfile.new(['ronin-shellcode', '.bin']).path
+        end
 
         super(output,options.merge(format: :bin))
 

spec/asm/shellcode_spec.rb

@@ -27,6 +27,18 @@
       subject.assemble.should == shellcode
     end
 
+    context "with :output" do
+      let(:output) do
+        Tempfile.new(['ronin-shellcode-custom-path', '.bin']).path
+      end
+
+      it "should write to the custom path" do
+        subject.assemble(output: output).should == shellcode
+
+        File.binread(output)
+      end
+    end
+
     context "with :syntax is :intel" do
       it "assemble down to raw machine code" do
         subject.assemble(syntax: :intel).should == shellcode