dashboard/dashapi: AI patching jobs

Author: dvyukov

dashboard/app/ai.go

@@ -0,0 +1,78 @@
+// Copyright 2025 syzkaller project authors. All rights reserved.
+// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
+
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"io"
+
+	"github.com/google/syzkaller/dashboard/dashapi"
+)
+
+/*
+import (
+	"bytes"
+	"compress/gzip"
+	"context"
+	"crypto/subtle"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"math/rand"
+	"net/http"
+	"net/mail"
+	"net/url"
+	"reflect"
+	"regexp"
+	"sort"
+	"strings"
+	"time"
+	"unicode/utf8"
+
+	"github.com/google/syzkaller/dashboard/dashapi"
+	"github.com/google/syzkaller/pkg/asset"
+	"github.com/google/syzkaller/pkg/auth"
+	"github.com/google/syzkaller/pkg/coveragedb"
+	"github.com/google/syzkaller/pkg/debugtracer"
+	"github.com/google/syzkaller/pkg/email"
+	"github.com/google/syzkaller/pkg/gcs"
+	"github.com/google/syzkaller/pkg/hash"
+	"github.com/google/syzkaller/pkg/subsystem"
+	"github.com/google/syzkaller/sys/targets"
+	"github.com/google/uuid"
+	"google.golang.org/appengine/v2"
+	db "google.golang.org/appengine/v2/datastore"
+	"google.golang.org/appengine/v2/log"
+	aemail "google.golang.org/appengine/v2/mail"
+	"google.golang.org/appengine/v2/user"
+)
+*/
+
+func apiAIJobPoll(c context.Context, payload io.Reader) (interface{}, error) {
+	req := new(dashapi.AIJobPollReq)
+	if err := json.NewDecoder(payload).Decode(req); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal request: %w", err)
+	}
+	resp := &dashapi.AIJobPollResp{}
+	return resp, nil
+}
+
+func apiAIJobDone(c context.Context, payload io.Reader) (interface{}, error) {
+	req := new(dashapi.AIJobDoneReq)
+	if err := json.NewDecoder(payload).Decode(req); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal request: %w", err)
+	}
+	return nil, nil
+}
+
+func apiAIJournal(c context.Context, payload io.Reader) (interface{}, error) {
+	req := new(dashapi.AIJournalReq)
+	if err := json.NewDecoder(payload).Decode(req); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal request: %w", err)
+	}
+	return nil, nil
+}

dashboard/app/api.go

@@ -60,6 +60,9 @@ var apiHandlers = map[string]APIHandler{
 	"save_discussion":       apiSaveDiscussion,
 	"create_upload_url":     apiCreateUploadURL,
 	"send_email":            apiSendEmail,
+	"ai_job_poll":           apiAIJobPoll,
+	"ai_job_done":           apiAIJobDone,
+	"ai_journal":            apiAIJournal,
 	"save_coverage":         gcsPayloadHandler(apiSaveCoverage),
 	"upload_build":          nsHandler(apiUploadBuild),
 	"builder_poll":          nsHandler(apiBuilderPoll),

dashboard/dashapi/dashapi.go

@@ -17,6 +17,7 @@ import (
 	"reflect"
 	"time"
 
+	"github.com/google/syzkaller/pkg/aflow/journal"
 	"github.com/google/syzkaller/pkg/auth"
 )
 
@@ -988,6 +989,70 @@ type JobInfo struct {
 	OnMergeBase      bool
 }
 
+type AIJobPollReq struct {
+	Namespaces []string
+	Workflows  []AIWorkflow
+}
+
+type AIWorkflow struct {
+	Type AIWorkflowType
+	Name string
+}
+
+type AIWorkflowType string
+
+const (
+	AIPatching AIWorkflowType = "patching"
+)
+
+type AIJobPollResp struct {
+	ID       string
+	Workflow AIWorkflow
+	Patching *AIPatchingJob
+}
+
+type AIJobDoneReq struct {
+	ID       string
+	Error    string
+	Patching *AIPatchingResult
+}
+
+type AIPatchingJob struct {
+	ReproOpts       string
+	ReproSyz        string
+	ReproC          string
+	KernelConfig    string
+	SyzkallerCommit string
+}
+
+type AIPatchingResult struct {
+	PatchDescription string
+	PatchDiff        string
+}
+
+type AIJournalReq struct {
+	JobID string
+	Event journal.Event
+	//!!! what to do with retried/duplicated entries
+	//!!! Attempt int?
+}
+
+func (dash *Dashboard) AIJobPoll(req *AIJobPollReq) (*AIJobPollResp, error) {
+	resp := new(AIJobPollResp)
+	if err := dash.Query("ai_job_poll", req, resp); err != nil {
+		return nil, err
+	}
+	return resp, nil
+}
+
+func (dash *Dashboard) AIJobDone(req *AIJobDoneReq) error {
+	return dash.Query("ai_job_done", req, nil)
+}
+
+func (dash *Dashboard) AIJournal(req *AIJournalReq) error {
+	return dash.Query("ai_journal", req, nil)
+}
+
 func (dash *Dashboard) Query(method string, req, reply interface{}) error {
 	if dash.logger != nil {
 		dash.logger("API(%v): %#v", method, req)