Finished rspec

Author: wchen-r7

spec/tools/egghunter_spec.rb

@@ -5,31 +5,100 @@
 
 describe Egghunter do
 
-  subject do
-    Egghunter::Driver.new
-  end
+  describe Egghunter::Driver do
+
+    subject do
+      Egghunter::Driver.new
+    end
+
+    let(:egg) {
+      'W00T'
+    }
+
+    describe '#run' do
+
+      def get_stdout(&block)
+        out = $stdout
+        $stdout = fake = StringIO.new
+        begin
+          yield
+        ensure
+          $stdout = out
+        end
+        fake.string
+      end
+
+      let(:default_opts) {
+        { :platform => 'windows', :format => 'c', :eggtag => egg, :arch => 'x86' }
+      }
+
+      before(:each) do
+        allow(Egghunter::OptsConsole).to receive(:parse).with(any_args).and_return(options)
+      end
 
-  describe '#run' do
+      context 'when the platform is windows' do
+        let(:options) { default_opts }
 
-    context 'when the platform is windows' do
-      it 'returns a windows egghunter' do
+        it 'returns a windows egghunter' do
+          output = get_stdout { subject.run }
+          expect(output).to include("\\x66\\x81\\xca\\xff")
+        end
+      end
+
+      context 'when the platform is linux' do
+        let(:options) do
+          { :platform => 'linux', :format => 'c', :eggtag => egg, :arch => 'x86' }
+        end
+
+        it 'returns a linux egghunter' do
+          output = get_stdout { subject.run }
+          expect(output).to include("\\xfc\\x66\\x81\\xc9\\xff")
+        end
+      end
+
+      context 'when the egg is WOOT' do
+        let(:options) { default_opts }
+
+        it 'includes W00T in the egghunter' do
+          output = get_stdout { subject.run }
+          expect(output).to include("\\x57\\x30\\x30\\x54")
+        end
       end
     end
+  end
+
 
-    context 'when the platform is linux' do
-      it 'returns a linux egghunter' do
+  describe Egghunter::OptsConsole do
+    subject do
+      Egghunter::OptsConsole
+    end
+
+    context 'when no options are given' do
+      it 'raises OptionParser::MissingArgument' do
+        expect{subject.parse([])}.to raise_error(OptionParser::MissingArgument)
       end
     end
 
-    context 'when the output format is java' do
-      it 'returns java format egghunter' do
+    context 'when no format is specified and --list-formats isn\'t used' do
+      it 'raises OptionParser::MissingArgument' do
+        args = '-e AAAA'.split
+        expect{subject.parse(args)}.to raise_error(OptionParser::MissingArgument)
       end
     end
 
-    context 'when the egg is WOOT' do
-      it 'includes W00TW00T in the egghunter' do
+    context 'when no egg is specified and --list-formats isn\'t used' do
+      it 'raises OptionParser::MissingArgument' do
+        args = '-f python'.split
+        expect{subject.parse(args)}.to raise_error(OptionParser::MissingArgument)
       end
     end
 
+    context 'when :depsize is a string' do
+      it 'raises OptionParser::InvalidOption' do
+        args = '-e AAAA -f c --depsize STRING'.split
+        expect{subject.parse(args)}.to raise_error(OptionParser::InvalidOption)
+      end
+    end
   end
+
 end

tools/egghunter.rb

@@ -93,7 +93,7 @@ def self.parse(args)
         raise OptionParser::MissingArgument, '-e is required'
       elsif options[:format] && !::Msf::Simple::Buffer.transform_formats.include?(options[:format])
         raise OptionParser::InvalidOption, "#{options[:format]} is not a valid format"
-      elsif options[:depsize] && options[:depsize] =~ /^\d+$/
+      elsif options[:depsize] && options[:depsize] !~ /^\d+$/
         raise OptionParser::InvalidOption, "--depsize must be a Fixnum"
       end