CC enclave app example: check that handshake is successful.

Change-Id: I49e0bd8affcd66a9b1959f46bb6406968593597b
project-oak · Jan 15, 2025 · 7548c99 · 7548c99
1 parent 930f945
commit 7548c99
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cc/containers/hello_world_enclave_app/app_service.cc b/cc/containers/hello_world_enclave_app/app_service.cc
@@ -63,8 +63,14 @@ grpc::Status EnclaveApplicationImpl::LegacySession(
 grpc::Status EnclaveApplicationImpl::OakSession(
     grpc::ServerContext* context,
     grpc::ServerReaderWriter<SessionResponse, SessionRequest>* stream) {
+  // Set up channel and do handshake.
   auto channel = session_server_.NewChannel(
       std::make_unique<transport::GrpcSyncSessionServerTransport>(stream));
+  if (!channel.ok()) {
+    return FromAbsl(channel.status());
+  }
+
+  // Handshake done, process requests until closed.
   while (true) {
     absl::StatusOr<std::string> request = (*channel)->Receive();
     if (!request.ok()) {