FilesExpand file tree

 main

/

FindFile_Java.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

107 lines (93 loc) · 3.72 KB

 main

/

FindFile_Java.go

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

107 lines (93 loc) · 3.72 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

package FindFile

import (

Rule2 "CodeScan/CommonVul/Rule"

"bufio"

"fmt"

"os"

"path/filepath"

"strings"

)

// FindFile FindFile函数用于在指定目录下查找符合规则的.java文件，并将包含规则的行写入到输出文件中

// 参数dir表示要搜索的目录路径

// 参数outputfile表示输出结果文件的路径（工具运行的目录）

// 参数rules表示要匹配的规则列表

func FindFileByJava(dir string, outputfile string, rules []string) {

var fileList []string

// 使用filepath.Walk遍历目标目录，跳过黑名单中的目录，收集所有.java文件的路径

err := filepath.Walk(dir, func(path string, f os.FileInfo, err error) error {

if err != nil {

return err

}

//如果f是一个文件夹

if f.IsDir() {

//继续进行遍历，如果在黑名单中的话就进行跳过

if Rule2.MatchRule(path, Rule2.PathBlackJava) {

return filepath.SkipDir

}

//如果文件存在的话就进行遍历 否则就进行判断，如果是java或者jsp后缀就添加到文件列表

} else if strings.HasSuffix(f.Name(), ".java") || strings.HasSuffix(f.Name(), ".jsp") {

fileList = append(fileList, path)

}

return nil

})

if err != nil {

fmt.Printf("error walking the path %v: %v\n", dir, err)

return

}

// 检查遍历目录过程中的错误

Check(err)

// 创建或打开输出文件，以追加模式写入

basedir := "./results/"

err1 := os.MkdirAll(basedir, os.ModePerm)

if err1 != nil {

fmt.Println("Error creating directory:", err)

return

}

outputfile = basedir + outputfile

outputFile, err := os.OpenFile(outputfile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)

Check(err)

defer outputFile.Close() // 确保文件在函数返回前被关闭

// 遍历收集到的.java文件列表，对每个文件进行扫描，查找包含规则的行

for _, file := range fileList {

f, err := os.Open(file)

Check(err)

defer f.Close() // 确保文件在处理完后被关闭

// 使用bufio.Scanner读取文件内容，为大文件读取优化

scanner := bufio.NewScanner(f)

buf := make([]byte, 0, 64*1024)

scanner.Buffer(buf, 10*1024*1024) // 设置一个更大的最大行大小

lineNumber := 1 // 行号，用于标识匹配行的位置

var lastFile string // 记录上一次输出的文件，用于控制输出格式

for scanner.Scan() {

line := strings.TrimSpace(scanner.Text()) // 去除行首尾空白字符

for _, rule := range rules {

// 检查当前行是否包含规则，且规则匹配函数Rule.MatchRule返回true

if strings.Contains(strings.ToLower(line), strings.ToLower(rule)) {

if Rule2.MatchRule(line, Rule2.LineBlack) {

break // 如果规则匹配，则跳出内层循环，处理下一行

} //xxx x

if !Rule2.RemoveStaticVar(strings.ToLower(line), strings.ToLower(rule)) {

break // 如果是静态变量则不做匹配

}

// 如果当前行是新文件的第一行且包含规则，则输出文件完整信息

if lastFile != file {

_, err := outputFile.WriteString(fmt.Sprintf("====================================================================\n\n"))

_, err = outputFile.WriteString(fmt.Sprintf("file [%s]\n%d : %s\n\n", file, lineNumber, line))

Check(err)

lastFile = file

} else {

// 如果当前行不是新文件的第一行，仅输出行号和内容

_, err := outputFile.WriteString(fmt.Sprintf("====================================================================\n\n"))

_, err = outputFile.WriteString(fmt.Sprintf("%d : %s\n\n", lineNumber, line))

Check(err)

}

}

}

lineNumber++

}

// 检查扫描过程是否出错

if err := scanner.Err(); err != nil {

fmt.Fprintln(os.Stderr, "reading standard input:", err)

}

}

}