Changeset 33708 for gs2-extensions/malware-checker/trunk/java/src/main/java/org/greenstone/virustotal/ScanFile.java

Timestamp:

2019-11-20T11:30:47+13:00 (4 years ago)

Author:

davidb

Message:

Changed code so api key can be in separate file, and passed in on the command-line to the Java code

File:

• gs2-extensions/malware-checker/trunk/java/src/main/java/org/greenstone/virustotal/ScanFile.java (modified) (2 diffs)

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

@@ -15 +15 @@
 public class ScanFile {
     public static void printUsage(){
-        System.err.println("Usage: ScanFile [all] file");
+        System.err.println("Usage: ScanFile [-all] api-key file");
         System.exit(1);
     }
 
     public static void main(String[] args) {
+    String apiKey = null;
     String inputFileName = null;
+    
     boolean printall=false;
-    if (args.length<1 || args.length>2){
+    if (args.length<1 || args.length>3){
         printUsage();
 
     }
-    else if(args.length==2){
-        if(args[0].equals("all")){
-            inputFileName = args[1];
+    else if(args.length==3){
+        if(args[0].equals("-all")){
+            apiKey = args[1];
+            inputFileName = args[2];
             printall=true;
         }
@@ -36 +39 @@
             
     }
-    
-    else{
-        inputFileName = args[0];
+    else if(args.length==2){
+        apiKey = args[0];
+        inputFileName = args[1];
+    }
+    else {
+        printUsage();
     }
 
     try {
     
-            VirusTotalConfig.getConfigInstance().setVirusTotalAPIKey("09311639ef4193fc22faa35d493a5bf12dc44844a55f03812e40bd1a16fd7b6f");
+            VirusTotalConfig.getConfigInstance().setVirusTotalAPIKey(apiKey);
+        
             VirustotalPublicV2 virusTotalRef = new VirustotalPublicV2Impl();