require File.dirname(__FILE__) + '/../spec_helper'

class ClamAV

  describe "ClamAV" do

    before(:all) do
      @clam = ClamAV.instance
      @clam.loaddb
    end

    FILES = {
      'robots.txt' => CL_CLEAN,
      'eicar.com'  => 'Eicar-Test-Signature', # EICAR
      'test.txt'   => 'Eicar-Test-Signature', # EICAR in text/plain
      'clam.cab'      => 'ClamAV-Test-File',
      'clam.exe'      => 'ClamAV-Test-File',
      'clam.exe.bz2'  => 'ClamAV-Test-File',
      'clam.zip'      => 'ClamAV-Test-File',
      'clam-v2.rar'   => 'ClamAV-Test-File',
      'clam-v3.rar'   => 'ClamAV-Test-File',
      'clam-p.rar'    => CL_CLEAN,  # encripted RAR
      'clam-ph.rar'    => CL_CLEAN, # encripted RAR with encrypted both file data and headers
      'program.doc'   => 'W97M.Class.EB',
      'Программа.doc' => 'W97M.Class.EB', # filename in UTF-8
    }

    FILES_ENCRYPTED = {
      'clam-p.rar'    => 'Encrypted.RAR',  # encripted RAR
      'clam-ph.rar'   => 'Encrypted.RAR', # encripted RAR with encrypted both file data and headers
      'clam-v3.rar'   => 'ClamAV-Test-File'
    }

    context "with default options" do

      it "should be instance of Clamav" do
        @clam.should be_instance_of(ClamAV)
      end

      it "should return engine version" do
        @clam.version.should >= '0.95'
      end

      FILES.each do |file, result|
        it "should scan #{file} with result #{result.to_s}" do
          @clam.scanfile(File.join(File.dirname(__FILE__), "../clamav-testfiles/", file)).should == result
        end
      end

      it "should return signatures count" do
        # strange, why not 757156? bug?
        @clam.signo.should >= 756450 # on 17/04/10
      end

      if ClamAV.instance.version >= '0.96'
        it "should return all signatures count" do
          (default = @clam.countsigs).should >= 757156 # on 17/04/10
          (all = @clam.countsigs(CL_COUNTSIGS_ALL)).should >= 757156
          default.should == all
        end
      end

      it "should not reload db when fresh" do
        @clam.reload.should == 0
      end

    end

    context "scan with custom options" do

      FILES_ENCRYPTED.each do |file, result|
        it "should scan encrypted #{file} with result #{result.to_s}" do
          @clam.scanfile(File.join(File.dirname(__FILE__), "../clamav-testfiles/", file),
            CL_SCAN_STDOPT | CL_SCAN_BLOCKENCRYPTED).should match(/#{result}$/)
        end
      end

      it "should scan OLE2 file with not detect" do
        @clam.scanfile(File.join(File.dirname(__FILE__), "../clamav-testfiles/", 'program.doc'),
          CL_SCAN_RAW).should == CL_CLEAN
      end

    end

    context "with custom db options" do

      before(:all) do
        @clam.loaddb(CL_DB_STDOPT | CL_DB_PUA)
      end

      after(:all) do
        @clam.loaddb
      end

      it "should detect PUA" do
        @clam.scanfile(File.join(File.dirname(__FILE__), "../clamav-testfiles/",
          'jquery.tooltip.pack.js')).should == 'PUA.Script.Packed-2'
      end

    end

    context "limit CL_ENGINE_MAX_FILES" do

      before(:all) do
        # save default
        @mem = @clam.getlimit(CL_ENGINE_MAX_FILES)
      end

      after(:each) do
        # restore default
        @clam.setlimit(CL_ENGINE_MAX_FILES, @mem)
      end

      it "should get default limit" do
        @clam.getlimit(CL_ENGINE_MAX_FILES).should == 10000
      end

      it "should set limit" do
        @clam.setlimit(CL_ENGINE_MAX_FILES, 1).should == CL_SUCCESS
      end

      it "should do not check archive with two files" do
        @clam.setlimit(CL_ENGINE_MAX_FILES, 1)
        @clam.scanfile(File.join(File.dirname(__FILE__), "../clamav-testfiles/", 'multi.zip')).
          should == CL_CLEAN
      end

    end

    context "get/set" do

      it "should get db time" do
        Time.at(@clam.getlimit(CL_ENGINE_DB_TIME)).should >= Time.now - 60*60*24 # 1 day
      end

      it "should get tmpdir == nil" do
        @clam.getstring(CL_ENGINE_TMPDIR).should be_nil
      end

      it "should set tmpdir" do
        @clam.setstring(CL_ENGINE_TMPDIR, '/tmp').should == CL_SUCCESS
        @clam.getstring(CL_ENGINE_TMPDIR).should == '/tmp'
      end

    end

  end
end