fixed xrayPocs功能存在问题 #72 2022-08-03

hktalent · hktalent · commit 6285b3df4824 · 2022-08-03T11:50:02.000+08:00
diff --git a/pocs_yml/utils/load.go b/pocs_yml/utils/load.go
@@ -39,7 +39,8 @@ func SelectPoc(Pocs embed.FS, pocname string) []string {
 	}
 	var foundFiles []string
 	for _, entry := range entries {
-		if strings.HasPrefix(entry.Name(), pocname+"-") {
+		//if strings.HasPrefix(entry.Name(), pocname+"-") {
+		if -1 < strings.Index(entry.Name(), pocname) {
 			foundFiles = append(foundFiles, entry.Name())
 		}
 	}

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,8 @@ func SelectPoc(Pocs embed.FS, pocname string) []string {`
`39`	`39`	`}`
`40`	`40`	`var foundFiles []string`
`41`	`41`	`for _, entry := range entries {`
`42`		`- if strings.HasPrefix(entry.Name(), pocname+"-") {`
	`42`	`+ //if strings.HasPrefix(entry.Name(), pocname+"-") {`
	`43`	`+ if -1 < strings.Index(entry.Name(), pocname) {`
`43`	`44`	`foundFiles = append(foundFiles, entry.Name())`
`44`	`45`	`}`
`45`	`46`	`}`