From 4d9e57eb150fee8d24ff32fb4a8a414c77d246e6 Mon Sep 17 00:00:00 2001
From: Aleksandr Nogikh <nogikh@google.com>
Date: Thu, 10 Apr 2025 15:09:06 +0200
Subject: syz-cluster: provide API for uploading artifacts archive

The archive would be a useful source of debugging information.

Provide an HTTP endpoint that accepts a multipart form request with
the archived data.

Provide an *api.Client method to encapsulate the encoding of the data.

Add a test.
---
 syz-cluster/pkg/controller/api.go | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'syz-cluster/pkg/controller/api.go')

diff --git a/syz-cluster/pkg/controller/api.go b/syz-cluster/pkg/controller/api.go
index bcab7813f..6dbdc4889 100644
--- a/syz-cluster/pkg/controller/api.go
+++ b/syz-cluster/pkg/controller/api.go
@@ -43,6 +43,7 @@ func (c APIServer) Mux() *http.ServeMux {
 	mux.HandleFunc("/sessions/upload", c.uploadSession)
 	mux.HandleFunc("/sessions/{session_id}/series", c.getSessionSeries)
 	mux.HandleFunc("/sessions/{session_id}/skip", c.skipSession)
+	mux.HandleFunc("/tests/upload_artifacts", c.uploadTestArtifact)
 	mux.HandleFunc("/tests/upload", c.uploadTest)
 	mux.HandleFunc("/trees", c.getTrees)
 	return mux
@@ -116,6 +117,36 @@ func (c APIServer) uploadTest(w http.ResponseWriter, r *http.Request) {
 	api.ReplyJSON[interface{}](w, nil)
 }
 
+func (c APIServer) uploadTestArtifact(w http.ResponseWriter, r *http.Request) {
+	const maxMemory = 16 * 1000 * 1000 // 16 MB.
+	if err := r.ParseMultipartForm(maxMemory); err != nil {
+		http.Error(w, "could not parse the multipart form", http.StatusBadRequest)
+		return
+	}
+	defer r.MultipartForm.RemoveAll()
+
+	file, _, err := r.FormFile("content")
+	if err != nil {
+		if err == http.ErrMissingFile {
+			http.Error(w, "the 'content' file must be present", http.StatusBadRequest)
+			return
+		}
+		http.Error(w, fmt.Sprintf("failed to query the file: %s", err), http.StatusInternalServerError)
+		return
+	}
+	defer file.Close()
+
+	err = c.testService.SaveArtifacts(r.Context(),
+		r.FormValue("session"),
+		r.FormValue("test"),
+		file)
+	if err != nil {
+		http.Error(w, fmt.Sprint(err), http.StatusInternalServerError)
+		return
+	}
+	api.ReplyJSON[interface{}](w, nil)
+}
+
 func (c APIServer) uploadFinding(w http.ResponseWriter, r *http.Request) {
 	req := api.ParseJSON[api.NewFinding](w, r)
 	if req == nil {
-- 
cgit mrf-deployment