bugfix in mergedirectories for go-fuzz

Author: Yevgeny Pats

client/agent.go

@@ -124,10 +124,8 @@ func (c *FuzzitClient) RunFuzzer(job Job, jobId string, updateDB bool) error {
 		}
 	}
 
-	if _, err := os.Stat("corpus"); os.IsNotExist(err) {
-		if err := os.Mkdir("corpus", 0644); err != nil {
-			return err
-		}
+	if err := createDirIfNotExist("corpus"); err != nil {
+		return err
 	}
 
 	if jobId != "" {
@@ -137,9 +135,6 @@ func (c *FuzzitClient) RunFuzzer(job Job, jobId string, updateDB bool) error {
 		}
 
 		log.Println("downloading additional corpus")
-		if err := os.Mkdir("additional-corpus", 0644); err != nil {
-			return err
-		}
 		if err := c.downloadAndExtract(
 			"additional-corpus",
 			fmt.Sprintf("orgs/%s/targets/%s/jobs/%s/additional-corpus", c.Org, c.currentJob.TargetId, c.jobId)); err != nil {
@@ -149,6 +144,11 @@ func (c *FuzzitClient) RunFuzzer(job Job, jobId string, updateDB bool) error {
 				return err
 			}
 		}
+
+		if err := createDirIfNotExist("additional-corpus"); err != nil {
+			return err
+		}
+
 	}
 
 	var err error

client/client.go

@@ -11,7 +11,7 @@ import (
 )
 
 const FuzzitEndpoint = "https://app.fuzzit.dev"
-const Version = "v2.4.71"
+const Version = "v2.4.72"
 
 type Target struct {
 	Name         string `firestore:"target_name"`

client/utils.go

@@ -180,13 +180,25 @@ func mergeDirectories(dst string, src string) error {
 		}
 		if !info.IsDir() {
 			fileName := info.Name()
-			err = os.Rename(filepath.Join(src, fileName), filepath.Join(dst, fileName))
-			if err != nil {
-				return err
+			dstPath := filepath.Join(dst, fileName)
+			if _, err := os.Stat(path); os.IsNotExist(err) {
+				err = os.Rename(filepath.Join(src, fileName), dstPath)
+				if err != nil {
+					return err
+				}
 			}
 		}
 		return nil
 	})
 
 	return err
 }
+
+func createDirIfNotExist(path string) error {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		if err := os.Mkdir(path, 0644); err != nil {
+			return err
+		}
+	}
+	return nil
+}