all: add /addinput rest api to upload new prog during runtime

ex to upload prog to syzkaller:
curl -s --noproxy 0.0.0.0 -F "file=@prog" http://0.0.0.0:8888/addinput

Author: jiangenj

pkg/manager/http.go

@@ -24,7 +24,9 @@ import (
 
 	"github.com/google/syzkaller/pkg/corpus"
 	"github.com/google/syzkaller/pkg/cover"
+	"github.com/google/syzkaller/pkg/db"
 	"github.com/google/syzkaller/pkg/fuzzer"
+	"github.com/google/syzkaller/pkg/hash"
 	"github.com/google/syzkaller/pkg/html/pages"
 	"github.com/google/syzkaller/pkg/log"
 	"github.com/google/syzkaller/pkg/mgrconfig"
@@ -55,6 +57,7 @@ type HTTPServer struct {
 	Pool        *vm.Dispatcher
 	Pools       map[string]*vm.Dispatcher
 	TogglePause func(paused bool)
+	CorpusDB    *db.DB
 
 	// Can be set dynamically after calling Serve.
 	Corpus          atomic.Pointer[corpus.Corpus]
@@ -81,6 +84,7 @@ func (serv *HTTPServer) Serve() {
 	handle("/vms", serv.httpVMs)
 	handle("/vm", serv.httpVM)
 	handle("/metrics", promhttp.HandlerFor(prometheus.DefaultGatherer, promhttp.HandlerOpts{}).ServeHTTP)
+	handle("/addinput", serv.httpAddInput)
 	handle("/syscalls", serv.httpSyscalls)
 	handle("/corpus", serv.httpCorpus)
 	handle("/corpus.db", serv.httpDownloadCorpus)
@@ -734,6 +738,42 @@ func (serv *HTTPServer) modulesInfo(w http.ResponseWriter, r *http.Request) {
 	serv.jsonPage(w, r, "modules", cover.Modules)
 }
 
+func (serv *HTTPServer) httpAddInput(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "multipart/form-data")
+	if r.Method != http.MethodPost {
+		http.Error(w, "only POST method supported", http.StatusMethodNotAllowed)
+		return
+	}
+	err := r.ParseMultipartForm(20 << 20)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to parse form: %v", err), http.StatusBadRequest)
+		return
+	}
+	file, _, err := r.FormFile("file")
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to retrieve file from form-data: %v", err), http.StatusBadRequest)
+		return
+	}
+	defer file.Close()
+	data, err := io.ReadAll(file)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to read file: %v", err), http.StatusBadRequest)
+		return
+	}
+	p, err := serv.Cfg.Target.Deserialize(data, prog.NonStrict)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to deserialize prog: %v", err), http.StatusBadRequest)
+		return
+	}
+	data = p.Serialize()
+	sig := hash.String(data)
+	if serv.CorpusDB != nil {
+		serv.CorpusDB.Save(sig, data, 0)
+		serv.CorpusDB.Flush()
+		w.Write([]byte(fmt.Sprintf("uploaded:\n%v", data)))
+	}
+}
+
 var alphaNumRegExp = regexp.MustCompile(`^[a-zA-Z0-9]*$`)
 
 func isAlphanumeric(s string) bool {

syz-manager/manager.go

@@ -283,6 +283,7 @@ func RunManager(mode *Mode, cfg *mgrconfig.Config) {
 	mgr.http = &manager.HTTPServer{
 		Cfg:        cfg,
 		StartTime:  time.Now(),
+		CorpusDB:   mgr.corpusDB,
 		CrashStore: mgr.crashStore,
 	}