manager: enable keepalives on rcp connections

VMs die frequently and it causes orphan tcp rpc connections on manager.

2 files changed, 12 insertions, 1 deletions

diff --git a/syz-hub/hub.go b/syz-hub/hub.go
index 037f01d47..a203149c5 100644
--- a/syz-hub/hub.go
+++ b/syz-hub/hub.go
@@ -65,7 +65,16 @@ func main() {
 	Logf(0, "serving rpc on tcp://%v", ln.Addr())
 	s := rpc.NewServer()
 	s.Register(hub)
-	s.Accept(ln)
+	for {
+		conn, err := ln.Accept()
+		if err != nil {
+			Logf(0, "failed to accept an rpc connection: %v", err)
+			continue
+		}
+		conn.SetKeepAlive(true)
+		conn.SetKeepAlivePeriod(time.Minute)
+		go s.ServeConn(conn)
+	}
 }
 
 func (hub *Hub) Connect(a *HubConnectArgs, r *int) error {
diff --git a/syz-manager/manager.go b/syz-manager/manager.go
index 34de9b54c..0041432e0 100644
--- a/syz-manager/manager.go
+++ b/syz-manager/manager.go
@@ -169,6 +169,8 @@ func RunManager(cfg *config.Config, syscalls map[int]bool, suppressions []*regex
 				Logf(0, "failed to accept an rpc connection: %v", err)
 				continue
 			}
+			conn.SetKeepAlive(true)
+			conn.SetKeepAlivePeriod(time.Minute)
 			go s.ServeCodec(jsonrpc.NewServerCodec(conn))
 		}
 	}()