Add missing references

wetware · Sep 8, 2023 · 0c58956 · 0c58956
1 parent caf600e
commit 0c58956
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/cap/capstore/capstore.go b/cap/capstore/capstore.go
@@ -22,7 +22,7 @@ func (c CapStore) Set(ctx context.Context, id string, cap capnp.Client) error {
 		if err := cs.SetId(id); err != nil {
 			return err
 		}
-		return cs.SetCap(cap)
+		return cs.SetCap(cap.AddRef())
 	})
 	defer release()
 
@@ -43,5 +43,5 @@ func (c CapStore) Get(ctx context.Context, id string) (capnp.Client, error) {
 		return capnp.Client{}, err
 	}
 
-	return res.Cap(), nil
+	return res.Cap().AddRef(), nil
 }
diff --git a/cap/capstore/server/capstore.go b/cap/capstore/server/capstore.go
@@ -30,7 +30,7 @@ func (c *CapStore) Set(ctx context.Context, call api.CapStore_set) error {
 	cap := call.Args().Cap()
 	c.Logger.Info("set capability", "id", id)
 
-	c.Map.Store(id, cap)
+	c.Map.Store(id, cap.AddRef())
 	return nil
 }
 
@@ -56,5 +56,5 @@ func (c *CapStore) Get(ctx context.Context, call api.CapStore_get) error {
 		return err
 	}
 
-	return res.SetCap(cap)
+	return res.SetCap(cap.AddRef())
 }
diff --git a/rom/internal/main.wasm b/rom/internal/main.wasm