aboutsummaryrefslogtreecommitdiffstats
path: root/pkg
diff options
context:
space:
mode:
authorAleksandr Nogikh <nogikh@google.com>2024-07-01 18:28:45 +0200
committerAleksandr Nogikh <nogikh@google.com>2024-07-04 12:34:14 +0000
commitecfab6a2e698b37bff1bf0b2996ff7954cf78303 (patch)
treebf6c9152ea479b8df13bbd84b3b91c4eeaf00754 /pkg
parent891bf15f3ceb3f69eaf590882ba3f245811c1698 (diff)
pkg/rpcserver: capitalize external interface methods
Make it explicit which methods of Runner refer to its implementation and which are supposed to be invoked by its users.
Diffstat (limited to 'pkg')
-rw-r--r--pkg/rpcserver/rpcserver.go26
-rw-r--r--pkg/rpcserver/runner.go18
2 files changed, 22 insertions, 22 deletions
diff --git a/pkg/rpcserver/rpcserver.go b/pkg/rpcserver/rpcserver.go
index da4faa19a..d9efb9d0d 100644
--- a/pkg/rpcserver/rpcserver.go
+++ b/pkg/rpcserver/rpcserver.go
@@ -199,20 +199,20 @@ func (serv *Server) MachineInfo(name string) []byte {
serv.mu.Lock()
runner := serv.runners[name]
serv.mu.Unlock()
- if runner == nil || !runner.alive() {
+ if runner == nil || !runner.Alive() {
return []byte("VM is not alive")
}
- return runner.getMachineInfo()
+ return runner.MachineInfo()
}
func (serv *Server) RunnerStatus(name string) []byte {
serv.mu.Lock()
runner := serv.runners[name]
serv.mu.Unlock()
- if runner == nil || !runner.alive() {
+ if runner == nil || !runner.Alive() {
return []byte("VM is not alive")
}
- return runner.queryStatus()
+ return runner.QueryStatus()
}
func (serv *Server) handleConn(conn *flatrpc.Conn) {
@@ -259,7 +259,7 @@ func (serv *Server) handleConn(conn *flatrpc.Conn) {
opts.Features = serv.cfg.Features
}
- err = runner.handshake(conn, opts)
+ err = runner.Handshake(conn, opts)
if err != nil {
log.Logf(1, "%v", err)
return
@@ -270,7 +270,7 @@ func (serv *Server) handleConn(conn *flatrpc.Conn) {
serv.mu.Unlock()
if serv.triagedCorpus.Load() {
- if err := runner.sendCorpusTriaged(); err != nil {
+ if err := runner.SendCorpusTriaged(); err != nil {
log.Logf(2, "%v", err)
return
}
@@ -333,7 +333,7 @@ func (serv *Server) connectionLoop(runner *Runner) error {
// buffer too much (we don't want to grow it larger than what will be needed
// to send programs).
n := min(len(maxSignal), 50000)
- if err := runner.sendSignalUpdate(maxSignal[:n]); err != nil {
+ if err := runner.SendSignalUpdate(maxSignal[:n]); err != nil {
return err
}
maxSignal = maxSignal[n:]
@@ -343,7 +343,7 @@ func (serv *Server) connectionLoop(runner *Runner) error {
serv.StatNumFuzzing.Add(1)
defer serv.StatNumFuzzing.Add(-1)
- return runner.connectionLoop()
+ return runner.ConnectionLoop()
}
func checkRevisions(a *flatrpc.ConnectRequest, target *prog.Target) {
@@ -462,7 +462,7 @@ func (serv *Server) StopFuzzing(name string) {
runner := serv.runners[name]
serv.info[name] = VMState{StateStopping, time.Now()}
serv.mu.Unlock()
- runner.stop()
+ runner.Stop()
}
func (serv *Server) ShutdownInstance(name string, crashed bool) ([]ExecRecord, []byte) {
@@ -471,20 +471,20 @@ func (serv *Server) ShutdownInstance(name string, crashed bool) ([]ExecRecord, [
delete(serv.runners, name)
serv.info[name] = VMState{StateOffline, time.Now()}
serv.mu.Unlock()
- return runner.shutdown(crashed), runner.getMachineInfo()
+ return runner.Shutdown(crashed), runner.MachineInfo()
}
func (serv *Server) DistributeSignalDelta(plus signal.Signal) {
plusRaw := plus.ToRaw()
serv.foreachRunnerAsync(func(runner *Runner) {
- runner.sendSignalUpdate(plusRaw)
+ runner.SendSignalUpdate(plusRaw)
})
}
func (serv *Server) TriagedCorpus() {
serv.triagedCorpus.Store(true)
serv.foreachRunnerAsync(func(runner *Runner) {
- runner.sendCorpusTriaged()
+ runner.SendCorpusTriaged()
})
}
@@ -495,7 +495,7 @@ func (serv *Server) foreachRunnerAsync(fn func(runner *Runner)) {
serv.mu.Lock()
defer serv.mu.Unlock()
for _, runner := range serv.runners {
- if runner.alive() {
+ if runner.Alive() {
go fn(runner)
}
}
diff --git a/pkg/rpcserver/runner.go b/pkg/rpcserver/runner.go
index 3886e5ffc..ea45aa89a 100644
--- a/pkg/rpcserver/runner.go
+++ b/pkg/rpcserver/runner.go
@@ -77,7 +77,7 @@ type handshakeResult struct {
Canonicalizer *cover.CanonicalizerInstance
}
-func (runner *Runner) handshake(conn *flatrpc.Conn, cfg *handshakeConfig) error {
+func (runner *Runner) Handshake(conn *flatrpc.Conn, cfg *handshakeConfig) error {
connectReply := &flatrpc.ConnectReply{
Debug: runner.debug,
Cover: runner.cover,
@@ -118,7 +118,7 @@ func (runner *Runner) handshake(conn *flatrpc.Conn, cfg *handshakeConfig) error
return nil
}
-func (runner *Runner) connectionLoop() error {
+func (runner *Runner) ConnectionLoop() error {
runner.mu.Lock()
stopped := runner.stopped
if !stopped {
@@ -390,7 +390,7 @@ func (runner *Runner) convertCallInfo(call *flatrpc.CallInfo) {
}
}
-func (runner *Runner) sendSignalUpdate(plus []uint64) error {
+func (runner *Runner) SendSignalUpdate(plus []uint64) error {
msg := &flatrpc.HostMessage{
Msg: &flatrpc.HostMessages{
Type: flatrpc.HostMessagesRawSignalUpdate,
@@ -402,7 +402,7 @@ func (runner *Runner) sendSignalUpdate(plus []uint64) error {
return flatrpc.Send(runner.conn, msg)
}
-func (runner *Runner) sendCorpusTriaged() error {
+func (runner *Runner) SendCorpusTriaged() error {
msg := &flatrpc.HostMessage{
Msg: &flatrpc.HostMessages{
Type: flatrpc.HostMessagesRawCorpusTriaged,
@@ -412,7 +412,7 @@ func (runner *Runner) sendCorpusTriaged() error {
return flatrpc.Send(runner.conn, msg)
}
-func (runner *Runner) stop() {
+func (runner *Runner) Stop() {
runner.mu.Lock()
runner.stopped = true
conn := runner.conn
@@ -422,7 +422,7 @@ func (runner *Runner) stop() {
}
}
-func (runner *Runner) shutdown(crashed bool) []ExecRecord {
+func (runner *Runner) Shutdown(crashed bool) []ExecRecord {
runner.mu.Lock()
runner.stopped = true
finished := runner.finished
@@ -442,13 +442,13 @@ func (runner *Runner) shutdown(crashed bool) []ExecRecord {
return runner.lastExec.Collect()
}
-func (runner *Runner) getMachineInfo() []byte {
+func (runner *Runner) MachineInfo() []byte {
runner.mu.Lock()
defer runner.mu.Unlock()
return runner.machineInfo
}
-func (runner *Runner) queryStatus() []byte {
+func (runner *Runner) QueryStatus() []byte {
resc := make(chan []byte, 1)
timeout := time.After(time.Minute)
select {
@@ -464,7 +464,7 @@ func (runner *Runner) queryStatus() []byte {
}
}
-func (runner *Runner) alive() bool {
+func (runner *Runner) Alive() bool {
runner.mu.Lock()
defer runner.mu.Unlock()
return runner.conn != nil && !runner.stopped