Close connections on enclave requests (#122)

capeprivacy · Aug 9, 2022 · a789ad2 · a789ad2
1 parent 4f49869
commit a789ad2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/capetest/capetest.go b/capetest/capetest.go
@@ -80,6 +80,7 @@ func CapeTest(testReq TestRequest, endpoint string, insecure bool, pcrSlice []st
 		}
 		return nil, err
 	}
+	defer conn.Close()
 
 	nonce, err := crypto.GetNonce()
 	if err != nil {

diff --git a/cmd/cape/cmd/deploy.go b/cmd/cape/cmd/deploy.go
@@ -220,6 +220,7 @@ func doDeploy(url string, name string, reader io.Reader, insecure bool, pcrSlice
 		}
 		return "", nil, err
 	}
+	defer conn.Close()
 
 	p := runner.Protocol{Websocket: conn}
 

diff --git a/cmd/cape/cmd/run.go b/cmd/cape/cmd/run.go
@@ -163,6 +163,8 @@ func doRun(url string, functionID string, data []byte, insecure bool, funcHash [
 		}
 		return nil, err
 	}
+	defer c.Close()
+
 	nonce, err := crypto.GetNonce()
 	if err != nil {
 		return nil, err
-Original file line number
+Diff line change
@@ Expand Up @@
     		}
     		return "", nil, err
     	}
+    	defer conn.Close()
     	p := runner.Protocol{Websocket: conn}
@@ Expand Down @@