daffainfo
/
nuclei
mirror of https://github.com/daffainfo/nuclei.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #250 from projectdiscovery/update-delete-fix 

Don't delete previous nuclei directory
dev
Ice3man 2020-08-23 11:00:49 -07:00 committed by GitHub
parent db5bd924a2 9cf12c6d9a
commit 3eb37df130
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
v2/internal/runner/update.go → v2/internal/runner/config.go
View File

@ -108,7 +108,6 @@ func (r *Runner) updateTemplates() error {
home = r.options.TemplatesDirectory home = r.options.TemplatesDirectory
} }
r.templatesConfig = &nucleiConfig{TemplatesDirectory: path.Join(home, "nuclei-templates")} r.templatesConfig = &nucleiConfig{TemplatesDirectory: path.Join(home, "nuclei-templates")}
os.RemoveAll(r.templatesConfig.TemplatesDirectory)
// Download the repository and also write the revision to a HEAD file. // Download the repository and also write the revision to a HEAD file.
version, asset, err := r.getLatestReleaseFromGithub() version, asset, err := r.getLatestReleaseFromGithub()
@ -169,7 +168,6 @@ func (r *Runner) updateTemplates() error {
r.templatesConfig.TemplatesDirectory = path.Join(home, "nuclei-templates") r.templatesConfig.TemplatesDirectory = path.Join(home, "nuclei-templates")
} }
r.templatesConfig.CurrentVersion = version.String() r.templatesConfig.CurrentVersion = version.String()
os.RemoveAll(r.templatesConfig.TemplatesDirectory)
gologger.Verbosef("Downloading nuclei-templates (v%s) to %s\n", "update-templates", version.String(), r.templatesConfig.TemplatesDirectory) gologger.Verbosef("Downloading nuclei-templates (v%s) to %s\n", "update-templates", version.String(), r.templatesConfig.TemplatesDirectory)
@ -268,17 +266,18 @@ func (r *Runner) downloadReleaseAndUnzip(downloadURL string) error {
templateDirectory := path.Join(r.templatesConfig.TemplatesDirectory, finalPath) templateDirectory := path.Join(r.templatesConfig.TemplatesDirectory, finalPath)
os.MkdirAll(templateDirectory, os.ModePerm) os.MkdirAll(templateDirectory, os.ModePerm)
f, err := os.Create(path.Join(templateDirectory, name)) f, err := os.OpenFile(path.Join(templateDirectory, name), os.O_CREATE|os.O_WRONLY, 0777)
if err != nil { if err != nil {
return fmt.Errorf("Could not create uncompressed file: %s", err) return fmt.Errorf("Could not create uncompressed file: %s", err)
} }
defer f.Close()
reader, err := file.Open() reader, err := file.Open()
if err != nil { if err != nil {
f.Close()
return fmt.Errorf("Could not open archive to extract file: %s", err) return fmt.Errorf("Could not open archive to extract file: %s", err)
} }
io.Copy(f, reader) io.Copy(f, reader)
f.Close()
} }
return nil return nil
} }