mirror of
https://github.com/reactos/syzkaller.git
synced 2024-10-08 01:23:20 +00:00
116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
// 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
|
|
}
|