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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
// Copyright 2016 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 (
"encoding/json"
"flag"
"fmt"
"io/ioutil"
"sync"
. "github.com/google/syzkaller/pkg/log"
. "github.com/google/syzkaller/rpctype"
"github.com/google/syzkaller/syz-hub/state"
)
var (
flagConfig = flag.String("config", "", "config file")
cfg *Config
)
type Config struct {
Http string
Rpc string
Workdir string
Managers []struct {
Name string
Key string
}
}
type Hub struct {
mu sync.Mutex
st *state.State
keys map[string]string
}
func main() {
flag.Parse()
cfg = readConfig(*flagConfig)
EnableLogCaching(1000, 1<<20)
st, err := state.Make(cfg.Workdir)
if err != nil {
Fatalf("failed to load state: %v", err)
}
hub := &Hub{
st: st,
keys: make(map[string]string),
}
for _, mgr := range cfg.Managers {
hub.keys[mgr.Name] = mgr.Key
}
hub.initHttp(cfg.Http)
s, err := NewRpcServer(cfg.Rpc, hub)
if err != nil {
Fatalf("failed to create rpc server: %v", err)
}
Logf(0, "serving rpc on tcp://%v", s.Addr())
s.Serve()
}
func (hub *Hub) Connect(a *HubConnectArgs, r *int) error {
if key, ok := hub.keys[a.Name]; !ok || key != a.Key {
Logf(0, "connect from unauthorized manager %v", a.Name)
return fmt.Errorf("unauthorized manager")
}
hub.mu.Lock()
defer hub.mu.Unlock()
Logf(0, "connect from %v: fresh=%v calls=%v corpus=%v", a.Name, a.Fresh, len(a.Calls), len(a.Corpus))
if err := hub.st.Connect(a.Name, a.Fresh, a.Calls, a.Corpus); err != nil {
Logf(0, "connect error: %v", err)
return err
}
return nil
}
func (hub *Hub) Sync(a *HubSyncArgs, r *HubSyncRes) error {
if key, ok := hub.keys[a.Name]; !ok || key != a.Key {
Logf(0, "sync from unauthorized manager %v", a.Name)
return fmt.Errorf("unauthorized manager")
}
hub.mu.Lock()
defer hub.mu.Unlock()
inputs, more, err := hub.st.Sync(a.Name, a.Add, a.Del)
if err != nil {
Logf(0, "sync error: %v", err)
return err
}
r.Inputs = inputs
r.More = more
Logf(0, "sync from %v: add=%v del=%v new=%v pending=%v", a.Name, len(a.Add), len(a.Del), len(inputs), more)
return nil
}
func readConfig(filename string) *Config {
if filename == "" {
Fatalf("supply config in -config flag")
}
data, err := ioutil.ReadFile(filename)
if err != nil {
Fatalf("failed to read config file: %v", err)
}
cfg := new(Config)
if err := json.Unmarshal(data, cfg); err != nil {
Fatalf("failed to parse config file: %v", err)
}
return cfg
}
|
