Implement the Free() RPC (#343)

cisco · Apr 6, 2023 · 623acd0 · 623acd0
1 parent 14d0b7e
commit 623acd0
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/cmd/interop/src/mls_client_impl.cpp b/cmd/interop/src/mls_client_impl.cpp
@@ -427,6 +427,20 @@ MLSClientImpl::HandleBranch(ServerContext* /* context */,
   });
 }
 
+Status
+MLSClientImpl::Free(ServerContext* /* context */,
+                    const FreeRequest* request,
+                    FreeResponse* /* response */)
+{
+  const auto state_id = request->state_id();
+  if (state_cache.count(state_id) == 0) {
+    return Status(StatusCode::NOT_FOUND, "Unknown state");
+  }
+
+  remove_state(state_id);
+  return Status::OK;
+}
+
 // Cached join transactions
 uint32_t
 MLSClientImpl::store_join(KeyPackageWithSecrets&& kp_priv)

diff --git a/cmd/interop/src/mls_client_impl.h b/cmd/interop/src/mls_client_impl.h
@@ -129,6 +129,11 @@ class MLSClientImpl final : public MLSClient::Service
                       const HandleBranchRequest* request,
                       HandleBranchResponse* response) override;
 
+  // Cleanup
+  Status Free(ServerContext* context,
+              const FreeRequest* request,
+              FreeResponse* response) override;
+
 private:
   // Wrapper for methods that rely on state
   template<typename Req, typename F>