update

ai_anti_malware/ai_anti_malware.cpp

@@ -313,24 +313,26 @@ auto doMalwareScan(int argc, char* argv[]) -> void {
 }
 
 int doSandbox(int argc, char* argv[]) {
-
-    std::string filePath = "Z:\\wangkun_muma\\opengl32.dll.bin";
-    auto peInfo = getPeInfo(filePath);
-    if (peInfo == nullptr) {
-        std::cout << "无法加载PE文件: " << filePath << std::endl;
-        return 1;
+    if (argc < 3) {
+        std::cout << "用法: " << argv[0] << " <文件路径> <地址>" << std::endl;
+        return;
     }
 
+    std::string filePath = argv[1];
+    auto peInfo = getPeInfo(filePath);
+    if (peInfo == nullptr) {
+        return 0;
+    }
     Sandbox se;
     se.InitEnv(peInfo);
-    se.Run(0x10002F20);
+    se.Run();
     return 0;
 }
 
 int main(int argc, char* argv[]) {
     // doMl(argc, argv);
     // doPredict(argc, argv);
-    // doMalwareScan(argc, argv);
-    doSandbox(argc, argv);
+    doMalwareScan(argc, argv);
+    // doSandbox(argc, argv);
     return 0;
 }