修改扫描逻辑

scagent/core/JsonRpc.go

@@ -47,40 +47,32 @@ func (f *FileRpcService) GetFilePath(args *Args, replay *string) error {
 
 	// 判断返回
 	// 读取所有的
-	if args.Scope == "all" {
-		// 递归遍历目录及其子目录
-		err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
-			// 处理遍历过程中的错误（如权限不足）
-			if err != nil {
-				logger.Error("访问路径出错", zap.Error(err))
-				return filepath.SkipDir // 跳过当前目录，继续遍历其他目录
-			}
 
-			// 只收集文件（排除目录）
-			if !d.IsDir() {
-				allFiles = append(allFiles, path)
-			}
-			return nil
-		})
-		// 打印错误日志
-		if err != nil {
-			logger.Error("遍历目录出错", zap.Error(err))
-			return err
-		}
-	}
-	// 读取当前的目录
-	if args.Scope == "dir" {
-		// 读取当前的目录
-		dirEntries, err := os.ReadDir(rootDir)
+	// 递归遍历目录及其子目录
+	err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
+		// 处理遍历过程中的错误（如权限不足）
 		if err != nil {
-			logger.Error("读取目录出错", zap.Error(err))
-			return err
+			logger.Error("访问路径出错", zap.Error(err))
+			//
+			if args.Scope == "all" {
+				return filepath.SkipDir
+			} else {
+				return nil
+			}
 		}
-		// 遍历目录下的文件
-		for _, entry := range dirEntries {
-			allFiles = append(allFiles, entry.Name())
+
+		// 只收集文件（排除目录）
+		if !d.IsDir() {
+			allFiles = append(allFiles, path)
 		}
+		return nil
+	})
+	// 打印错误日志
+	if err != nil {
+		logger.Error("遍历目录出错", zap.Error(err))
+		return err
 	}
+
 	// 递归遍历目录及其子目录
 	// err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
 	// 	// 处理遍历过程中的错误（如权限不足）