1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
// Copyright 2020 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"
"flag"
"os"
"github.com/google/syzkaller/dashboard/dashapi"
"github.com/google/syzkaller/pkg/kcidb"
"github.com/google/syzkaller/pkg/tool"
)
func main() {
const (
origin = "syzbot"
projectID = "kernelci-production"
topicName = "playground_kernelci_new"
)
var (
flagRestURI = flag.String("rest", "", "REST API endpoint for KCIDB")
flagToken = flag.String("token", "", "KCIDB API token")
flagDashClient = flag.String("client", "", "dashboard client")
flagDashAddr = flag.String("addr", "", "dashboard address")
flagDashKey = flag.String("key", "", "dashboard API key")
flagBug = flag.String("bug", "", "bug ID to upload to KCIDB")
flagInput = flag.String("input", "", "input JSON file with bug report")
flagOutput = flag.String("output", "", "output JSON file for KCIDB data")
)
flag.Parse()
var bug *dashapi.BugReport
// If input file is specified, read from file instead of calling API
if *flagInput != "" {
data, err := os.ReadFile(*flagInput)
if err != nil {
tool.Fail(err)
}
bug = &dashapi.BugReport{}
if err := json.Unmarshal(data, bug); err != nil {
tool.Fail(err)
}
} else {
// Original behavior: fetch from dashboard API
dash, err := dashapi.New(*flagDashClient, *flagDashAddr, *flagDashKey)
if err != nil {
tool.Fail(err)
}
bug, err = dash.LoadBug(*flagBug)
if err != nil {
tool.Fail(err)
}
}
kcidb.Validate = true
client, err := kcidb.NewClient(context.Background(), origin, *flagRestURI, *flagToken)
if err != nil {
tool.Fail(err)
}
defer client.Close()
// If output file is specified, write to file instead of submitting.
if *flagOutput != "" {
if err := client.PublishToFile(bug, *flagOutput); err != nil {
tool.Fail(err)
}
} else {
// Original behavior: submit to REST API
if err := client.Publish(bug); err != nil {
tool.Fail(err)
}
}
}
|
