zsteg 0.0.0 → 0.0.1

data/lib/zsteg.rb

@@ -10,3 +10,5 @@ require 'zsteg/result'
 require 'zsteg/file_cmd'
 
 require 'zsteg/checker/wbstego'
+
+require 'zsteg/masker'

data/spec/checker_spec.rb

@@ -0,0 +1,47 @@
+require 'spec_helper'
+
+include ZSteg
+
+describe Checker do
+  each_sample do |fname|
+    describe fname do
+      describe "#check" do
+        before :all do
+          @checker = Checker.new(fname)
+          orig_stdout, @out = $stdout, ""
+          begin
+            $stdout = StringIO.new(@out)
+            @results = @checker.check
+          ensure
+            $stdout = orig_stdout
+          end
+        end
+
+        it "should be quiet by default" do
+          @out.should == ""
+        end
+
+        it "returned results should be equal to #results" do
+          @results.should == @checker.results
+        end
+
+        it "should return array of results" do
+          @results.should be_instance_of(Array)
+        end
+
+        describe "results" do
+          it "should not have text results shorter than #{Checker::MIN_TEXT_LENGTH}" do
+            @results.each do |result|
+              case result
+              when Result::WholeText
+                result.text.size.should(be >= Checker::MIN_WHOLETEXT_LENGTH, result.inspect)
+              when Result::Text
+                result.text.size.should(be >= Checker::MIN_TEXT_LENGTH, result.inspect)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/spec/easybmp_spec.rb

@@ -0,0 +1,9 @@
+require 'spec_helper'
+
+sample("EasyBMP.bmp") do |fname|
+  describe fname do
+    it "contains hidden message" do
+      cli(fname, "-o","xy").should include("EasyBMPstego")
+    end
+  end
+end

data/spec/hackquest_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+
+sample("hackquest/square.bmp") do |fname|
+  describe fname do
+    subject{ cli(fname, "-b 10") }
+
+    it { should include "thesecretpasswordis:jedimaster" }
+  end
+end
+
+sample("hackquest/crypt.bmp") do |fname|
+  describe fname do
+    subject{ cli(fname, "-b 10") }
+
+    it { should include "111Hello" }
+  end
+end
+

data/spec/mask_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+
+sample("Steganography_original.png") do |fname|
+  describe fname do
+    it "extracts hidden image" do
+      tname = "tmp/mask.tmp.png"
+      File.unlink(tname) if File.exist?(tname)
+      cli(ZSteg::MaskCLI, fname, "-m 00000011 -O #{tname}")
+      img1 = ZPNG::Image.load tname
+      img2 = ZPNG::Image.load fname.sub(/\.png$/,".00000011.png")
+      img1.should == img2
+    end
+  end
+end

data/spec/polictf2012_spec.rb

@@ -0,0 +1,48 @@
+require 'spec_helper'
+
+sample('polictf2012_f200.bmp') do |fname|
+  describe fname do
+    it "should detect hidden BMP" do
+      cli(fname).should include("PC bitmap, Windows 3.x format, 500 x 277 x 24")
+    end
+
+    describe "hidden BMP #1" do
+      subject(:data){ cli(fname, "--extract", "4b,lsb,bY") }
+      its(:size){ should == 416816 }
+
+      it "should have BMP header" do
+        data[0,2].should == "BM"
+      end
+
+      it "should have 7zip after BMP" do
+        data.index("7z").should == 2005
+      end
+
+      describe "deeper" do
+        let(:tname){  File.join("tmp", File.basename(fname) + ".bmp") }
+        before(:all){ File.open(tname, "wb"){ |f| f<<data } }
+
+        it "should detect 7z & BMP" do
+          out = cli(tname)
+          out.should include('7-zip archive')
+          out.should include('PC bitmap, Windows 3.x format, 100 x 55 x 24')
+        end
+
+        describe "hidden BMP #2" do
+          subject(:data2){ cli(tname, "--extract", "2b,lsb,bY") }
+          its(:size){ should == 103875 }
+
+          describe "deeper" do
+            let(:tname2){ File.join("tmp", File.basename(tname) + ".bmp") }
+            before(:all){ File.open(tname2, "wb"){ |f| f<<data2 } }
+
+            it "should detect text" do
+              out = cli(tname2)
+              out.should include('sticazziantanieancoraunavoltacomesefossestato')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/spec/prime_spec.rb

@@ -0,0 +1,9 @@
+require 'spec_helper'
+
+sample("prime.png") do |fname|
+  describe fname do
+    it "contains prime-encoded message" do
+      cli(fname, "--prime").should include("48300:TG9yZW0gaXBzdW")
+    end
+  end
+end

data/spec/r3g2b3_spec.rb

@@ -0,0 +1,9 @@
+require 'spec_helper'
+
+sample("r3g2b3.png") do |fname|
+  describe fname do
+    it "contains hidden message" do
+      cli(fname, "-c","r3g2b3").should include("astley.3gp")
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1,6 +1,7 @@
 $:.unshift(File.expand_path("../lib", File.dirname(__FILE__)))
 require 'zsteg'
 require 'zsteg/cli'
+require 'zsteg/mask_cli'
 
 SAMPLES_DIR = File.expand_path("../samples", File.dirname(__FILE__))
 
@@ -11,21 +12,37 @@ def each_sample glob="*.png"
 end
 
 def sample fname
-  File.join(SAMPLES_DIR, fname)
+  fname = File.join(SAMPLES_DIR, fname)
+  if block_given?
+    yield fname.sub(Dir.pwd+'/','')
+  end
+
+  fname
 end
 
 def cli *args
   @@cli_cache ||= {}
-  @@cli_cache[args] ||=
+  args.map! do |arg|
+    if arg.is_a?(String) && arg[' ']
+      # split strings with spaces into arrays
+      arg.split(' ')
+    else
+      arg
+    end
+  end
+  args.flatten!
+  @@cli_cache[args.inspect] ||=
     begin
+      klass = args.first.is_a?(Class) ? args.shift : ZSteg::CLI
+      args << "--no-color" unless args.any?{|x| x['color']}
       orig_stdout, out = $stdout, ""
       begin
         $stdout = StringIO.new(out)
-        ZSteg::CLI.new(args).run
+        klass.new(args).run
       ensure
         $stdout = orig_stdout
       end
-      out
+      out.strip
     end
 end
 

data/spec/wbstego_spec.rb

@@ -1,10 +1,28 @@
 require 'spec_helper'
 
-each_sample("wbsteg*.bmp") do |fname|
+each_sample("wbstego/*") do |fname|
   describe fname do
-    subject{ cli(fname) }
+    subject{
+      if fname['.png']
+        cli(fname, "1b,lsb", "-o", "all")
+      else
+        cli(fname, "1b,lsb")
+      end
+    }
 
     it { should include("wbStego") }
-    it { should include("SuperSecretMessage") }
+    it { should include("SuperSecretMessage") } if fname['noenc']
+    if fname['even']
+      it { should include("even=true") }
+    else
+      it { should include("even=false") }
+    end
+
+    %w'blowfish twofish cast128 rijndael'.each do |enc|
+      it { should match(/#{enc}/i) } if fname[enc]
+    end
+
+    it { should include("mix=true") } if fname['mix']
+    it { should include("enc=\"wbSteg") } if fname['enc'] && !fname['noenc']
   end
 end

data/spec/wechall_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+WECHALL = {
+  'stegano1.bmp' => "Look what the hex-edit revealed: passwd:steganoI"
+}
+
+each_sample("wechall/*.bmp") do |fname|
+  describe fname do
+    subject{ cli(fname) }
+
+    it { should include WECHALL[File.basename(fname)] }
+  end
+end
+
+sample("wechall/5ZMGcCLxpcpsru03.png") do |fname|
+  describe fname do
+    it "extracts hidden image" do
+      tname = "tmp/wechall.tmp.png"
+      File.unlink(tname) if File.exist?(tname)
+      cli(ZSteg::MaskCLI, fname, "--green 00000010 -O #{tname}")
+      img1 = ZPNG::Image.load tname
+      img2 = ZPNG::Image.load fname.sub(/\.png$/,".g00000010.png")
+      img1.should == img2
+    end
+  end
+end

data/zsteg.gemspec

@@ -0,0 +1,121 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "zsteg"
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Andrey \"Zed\" Zaikin"]
+  s.date = "2013-01-15"
+  s.email = "zed.0xff@gmail.com"
+  s.executables = ["zsteg", "zsteg-mask"]
+  s.extra_rdoc_files = [
+    "README.md",
+    "README.md.tpl",
+    "TODO"
+  ]
+  s.files = [
+    "Gemfile",
+    "Gemfile.lock",
+    "README.md",
+    "README.md.tpl",
+    "Rakefile",
+    "TODO",
+    "VERSION",
+    "bin/zsteg",
+    "bin/zsteg-mask",
+    "cmp_bmp.rb",
+    "cmp_png.rb",
+    "lib/zsteg.rb",
+    "lib/zsteg/checker.rb",
+    "lib/zsteg/checker/wbstego.rb",
+    "lib/zsteg/cli.rb",
+    "lib/zsteg/extractor.rb",
+    "lib/zsteg/extractor/byte_extractor.rb",
+    "lib/zsteg/extractor/color_extractor.rb",
+    "lib/zsteg/file_cmd.rb",
+    "lib/zsteg/mask_cli.rb",
+    "lib/zsteg/masker.rb",
+    "lib/zsteg/result.rb",
+    "pngsteg.gemspec",
+    "samples/hackquest/crypt.bmp",
+    "samples/hackquest/square.bmp",
+    "samples/wbstego/wbsteg_blowfish_pass_1.bmp",
+    "samples/wbstego/wbsteg_cast128_pass_1.bmp",
+    "samples/wbstego/wbsteg_enc_pass_pass.bmp",
+    "samples/wbstego/wbsteg_enc_pass_pass_even.bmp",
+    "samples/wbstego/wbsteg_mix_pass_1.bmp",
+    "samples/wbstego/wbsteg_mix_pass_1_even.bmp",
+    "samples/wbstego/wbsteg_mix_pass_foobar.bmp",
+    "samples/wbstego/wbsteg_mix_pass_pass.bmp",
+    "samples/wbstego/wbsteg_mixenc_pass_pass_even.bmp",
+    "samples/wbstego/wbsteg_noenc.bmp",
+    "samples/wbstego/wbsteg_noenc.png",
+    "samples/wbstego/wbsteg_noenc_.bmp",
+    "samples/wbstego/wbsteg_noenc_17.bmp",
+    "samples/wbstego/wbsteg_noenc__.bmp",
+    "samples/wbstego/wbsteg_noenc_even.bmp",
+    "samples/wbstego/wbsteg_noenc_even2.bmp",
+    "samples/wbstego/wbsteg_noenc_even_17.bmp",
+    "samples/wbstego/wbsteg_noenc_even_17_.bmp",
+    "samples/wbstego/wbsteg_noenc_ext_ABC.bmp",
+    "samples/wbstego/wbsteg_rijndael_pass_1.bmp",
+    "samples/wbstego/wbsteg_rijndael_pass_pass.bmp",
+    "samples/wbstego/wbsteg_rijndael_pass_pass_even.bmp",
+    "samples/wbstego/wbsteg_twofish_pass_1.bmp",
+    "samples/wechall/5ZMGcCLxpcpsru03.g00000010.png",
+    "samples/wechall/5ZMGcCLxpcpsru03.png",
+    "samples/wechall/stegano1.bmp",
+    "spec/camouflage_spec.rb",
+    "spec/cats_spec.rb",
+    "spec/checker_spec.rb",
+    "spec/easybmp_spec.rb",
+    "spec/flowers_spec.rb",
+    "spec/hackquest_spec.rb",
+    "spec/mask_spec.rb",
+    "spec/openstego_spec.rb",
+    "spec/polictf2012_spec.rb",
+    "spec/prime_spec.rb",
+    "spec/r3g2b3_spec.rb",
+    "spec/simple_spec.rb",
+    "spec/spec_helper.rb",
+    "spec/wbstego_spec.rb",
+    "spec/wechall_spec.rb",
+    "spec/zlib_spec.rb",
+    "tmp/.keep",
+    "zsteg.gemspec"
+  ]
+  s.homepage = "http://github.com/zed-0xff/zsteg"
+  s.licenses = ["MIT"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.24"
+  s.summary = "Detect stegano-hidden data in PNG & BMP files."
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<zpng>, [">= 0.2.2"])
+      s.add_runtime_dependency(%q<iostruct>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, [">= 2.8.0"])
+      s.add_development_dependency(%q<bundler>, [">= 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.8.4"])
+    else
+      s.add_dependency(%q<zpng>, [">= 0.2.2"])
+      s.add_dependency(%q<iostruct>, [">= 0"])
+      s.add_dependency(%q<rspec>, [">= 2.8.0"])
+      s.add_dependency(%q<bundler>, [">= 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+    end
+  else
+    s.add_dependency(%q<zpng>, [">= 0.2.2"])
+    s.add_dependency(%q<iostruct>, [">= 0"])
+    s.add_dependency(%q<rspec>, [">= 2.8.0"])
+    s.add_dependency(%q<bundler>, [">= 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
+  end
+end
+