source: gs2-extensions/malware-checker/trunk/java/src/main/java/org/greenstone/virustotal/ScanFile.java

package org.greenstone.virustotal;
import com.kanishka.virustotal.dto.ScanInfo;
import com.kanishka.virustotal.exception.APIKeyNotFoundException;
import com.kanishka.virustotal.exception.UnauthorizedAccessException;
import com.kanishka.virustotalv2.VirusTotalConfig;
import com.kanishka.virustotalv2.VirustotalPublicV2;
import com.kanishka.virustotalv2.VirustotalPublicV2Impl;
import java.io.*;
import java.util.*; 
import java.lang.*;   
import java.io.UnsupportedEncodingException;



public class ScanFile {
    public static void printUsage(){
        System.err.println("Usage: ScanFile [-all] api-key file");
        System.exit(1);
    }

    public static void main(String[] args) {
    String apiKey = null;
    String inputFileName = null;
    String outputFileName = null;
    
    boolean printall=false;
    if (args.length<1 || args.length>3){
        printUsage();

    }
    else if(args.length==3){
        if(args[0].equals("-all")){
            apiKey = args[1];
            inputFileName = args[2];
            printall=true;
        }
        else{
            printUsage();       
        }
            
    }
    else if(args.length==2){
        apiKey = args[0];
        inputFileName = args[1];
    }
    else {
        printUsage();
    }

    //outputFileName = ".virustotal-resource-id-" + inputFileName;
    outputFileName = inputFileName + "-VIRUSTOTAL-RESOURCEID";
    
    try {
    
            VirusTotalConfig.getConfigInstance().setVirusTotalAPIKey(apiKey);
        
            VirustotalPublicV2 virusTotalRef = new VirustotalPublicV2Impl();

            ScanInfo scanInformation = virusTotalRef.scanFile(new File(inputFileName));
        int responseCode = scanInformation.getResponseCode();
        if(responseCode!=1){
        System.err.println("Error:Failed to preform scan.");        
        System.out.println("Verbose Msg :\t" + scanInformation.getVerboseMessage());
        System.exit(6);
        }
        if (printall) {
            System.out.println("___SCAN INFORMATION___");
                System.out.println("MD5 :\t" + scanInformation.getMd5());
                System.out.println("Perma Link :\t" + scanInformation.getPermalink());
                System.out.println("Resource :\t" + scanInformation.getResource());
                System.out.println("Scan Date :\t" + scanInformation.getScanDate());
                System.out.println("Scan Id :\t" + scanInformation.getScanId());
                System.out.println("SHA1 :\t" + scanInformation.getSha1());
                System.out.println("SHA256 :\t" + scanInformation.getSha256());
                System.out.println("Verbose Msg :\t" + scanInformation.getVerboseMessage());
                System.out.println("Response Code :\t" + scanInformation.getResponseCode());
            System.out.println("done.");

        }
        else {
        String resource_id = scanInformation.getResource();
               
        System.out.println(resource_id);

        PrintWriter writer = new PrintWriter(outputFileName, "UTF-8");
        writer.println(resource_id);
        writer.close();     
        }
       
        } catch (APIKeyNotFoundException ex) {
            System.err.println("API Key not found! " + ex.getMessage());
        System.exit(2);
        } catch (UnsupportedEncodingException ex) {
            System.err.println("Unsupported Encoding Format!" + ex.getMessage());
        System.exit(3);
        } catch (UnauthorizedAccessException ex) {
            System.err.println("Invalid API Key " + ex.getMessage());
            System.exit(4);
        } catch (IOException ex) {
                System.err.println("IO error:" + ex.getMessage());
        System.exit(5);
        } catch (Exception ex) {
            System.err.println("Something Bad Happened! " + ex.getMessage());
        System.exit(6);
        }
    }
}