From d39ef2f9994801180e552101c3b63a7b4e4a8579 Mon Sep 17 00:00:00 2001
From: Aleksandr Nogikh <nogikh@google.com>
Date: Tue, 4 Feb 2025 11:39:48 +0100
Subject: pkg/manager: fix RPCServer ops in RunDiffFuzzer()

We must not only call Listen(), but also Serve().
---
 pkg/manager/diff.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'pkg')

diff --git a/pkg/manager/diff.go b/pkg/manager/diff.go
index 8525015da..f472cbf18 100644
--- a/pkg/manager/diff.go
+++ b/pkg/manager/diff.go
@@ -294,8 +294,15 @@ func (kc *kernelContext) Loop() error {
 	if err := kc.serv.Listen(); err != nil {
 		return fmt.Errorf("failed to start rpc server: %w", err)
 	}
-	kc.pool.Loop(kc.ctx)
-	return nil
+	eg, ctx := errgroup.WithContext(kc.ctx)
+	eg.Go(func() error {
+		return kc.serv.Serve(ctx)
+	})
+	eg.Go(func() error {
+		kc.pool.Loop(ctx)
+		return nil
+	})
+	return eg.Wait()
 }
 
 func (kc *kernelContext) MaxSignal() signal.Signal {
-- 
cgit mrf-deployment