Fix cmgrd artifact fetching

ArmyCyberInstitute · Jan 12, 2021 · 38b6425 · 38b6425
1 parent e12bf55
commit 38b6425
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmd/cmgrd/main.go b/cmd/cmgrd/main.go
@@ -22,6 +22,8 @@ type state struct {
 	mgr *cmgr.Manager
 }
 
+var artifact_dir string
+
 func main() {
 	var iface string
 	var port int
@@ -36,6 +38,10 @@ func main() {
 		os.Exit(0)
 	}
 
+	artifact_dir, _ = os.LookupEnv(cmgr.ARTIFACT_DIR_ENV)
+	if artifact_dir == "" {
+		artifact_dir = "."
+	}
 	mgr := cmgr.NewManager(cmgr.INFO)
 	if mgr == nil {
 		log.Fatal("failed to initialize cmgr library")
@@ -285,7 +291,7 @@ func (s state) artifactsHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	f, err := os.Open(fmt.Sprintf("%d.tar.gz", build))
+	f, err := os.Open(fmt.Sprintf("%s/%d.tar.gz", artifact_dir, build))
 
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)