mirror of https://github.com/daffainfo/nuclei.git
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"io"
|
|
"io/fs"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) < 3 {
|
|
log.Fatalf("Usage: %s <templates-directory> <checksum-file>\n", os.Args[0])
|
|
}
|
|
checksumFile := os.Args[2]
|
|
templatesDirectory := os.Args[1]
|
|
|
|
file, err := os.Create(checksumFile)
|
|
if err != nil {
|
|
log.Fatalf("Could not create file: %s\n", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
err = filepath.WalkDir(templatesDirectory, func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil || d.IsDir() {
|
|
return nil
|
|
}
|
|
pathIndex := path[strings.Index(path, "nuclei-templates/")+17:]
|
|
pathIndex = strings.TrimPrefix(pathIndex, "nuclei-templates/")
|
|
// Ignore items starting with dots
|
|
if strings.HasPrefix(pathIndex, ".") {
|
|
return nil
|
|
}
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
h := sha1.New()
|
|
_, _ = io.Copy(h, bytes.NewReader(data))
|
|
hash := hex.EncodeToString(h.Sum(nil))
|
|
|
|
_, _ = file.WriteString(pathIndex)
|
|
_, _ = file.WriteString(":")
|
|
_, _ = file.WriteString(hash)
|
|
_, _ = file.WriteString("\n")
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
log.Fatalf("Could not walk directory: %s\n", err)
|
|
}
|
|
}
|