Add primitive enclave init entry point

Add user defined enclave init entry points to primitives API

PiperOrigin-RevId: 255032718
Change-Id: I36a73c53c35f1864966d4fe05ddce5a6a3a5cd48

Author: anubhavsr

asylo/platform/core/BUILD

@@ -118,6 +118,8 @@ cc_library(
         "//asylo/platform/posix/io:io_manager",
         "//asylo/platform/posix/signal:signal_manager",
         "//asylo/platform/posix/threading:thread_manager",
+        "//asylo/platform/primitives",
+        "//asylo/platform/primitives:trusted_runtime",
         "//asylo/util:logging",
         "//asylo/util:status",
         "@com_google_absl//absl/memory",

asylo/platform/core/trusted_application.cc

@@ -44,10 +44,13 @@
 #include "asylo/platform/posix/io/random_devices.h"
 #include "asylo/platform/posix/signal/signal_manager.h"
 #include "asylo/platform/posix/threading/thread_manager.h"
+#include "asylo/platform/primitives/primitive_status.h"
+#include "asylo/platform/primitives/trusted_runtime.h"
 #include "asylo/util/posix_error_space.h"
 #include "asylo/util/status.h"
 #include "asylo/util/status_macros.h"
 
+using asylo::primitives::PrimitiveStatus;
 using EnclaveState = ::asylo::TrustedApplication::State;
 using google::protobuf::RepeatedPtrField;
 
@@ -572,3 +575,5 @@ int __asylo_transfer_secure_snapshot_key(const char *input, size_t input_len,
 }  // extern "C"
 
 }  // namespace asylo
+
+extern "C" PrimitiveStatus enc_init() { return PrimitiveStatus::OkStatus(); }

asylo/platform/host_call/test/BUILD

@@ -88,6 +88,8 @@ sim_enclave_test(
     ],
     deps = [
         ":host_call_test_lib",
+        "//asylo/platform/primitives",
+        "//asylo/platform/primitives:trusted_runtime",
         "//asylo/platform/primitives/test:sim_test_backend",
     ],
 )

asylo/platform/host_call/test/test_enclave.cc

@@ -19,11 +19,15 @@
 #include "asylo/platform/host_call/test/enclave_test_selectors.h"
 #include "asylo/platform/host_call/trusted/host_call_dispatcher.h"
 #include "asylo/platform/host_call/trusted/host_calls.h"
+#include "asylo/platform/primitives/primitive_status.h"
 #include "asylo/platform/primitives/trusted_primitives.h"
+#include "asylo/platform/primitives/trusted_runtime.h"
 #include "asylo/platform/system_call/system_call.h"
 #include "asylo/platform/system_call/type_conversions/types_functions.h"
 #include "asylo/util/status_macros.h"
 
+using asylo::primitives::PrimitiveStatus;
+
 namespace asylo {
 namespace host_call {
 namespace {
@@ -559,3 +563,5 @@ extern "C" primitives::PrimitiveStatus asylo_enclave_fini() {
 
 }  // namespace host_call
 }  // namespace asylo
+
+extern "C" PrimitiveStatus enc_init() { return PrimitiveStatus::OkStatus(); }

asylo/platform/primitives/BUILD

@@ -60,7 +60,7 @@ cc_library(
         "//asylo/platform/primitives/util:trusted_memory.h",
     ],
     copts = ASYLO_DEFAULT_COPTS,
-    deps = select({
+    deps = [":primitives"] + select({
         "//asylo/platform/primitives:sgx": [
             "//asylo/platform/primitives/sgx:trusted_sgx",
         ],
@@ -84,6 +84,7 @@ cc_library(
     name = "trusted_runtime",
     hdrs = ["trusted_runtime.h"],
     copts = ASYLO_DEFAULT_COPTS,
+    deps = [":primitives"],
 )
 
 # Test ParameterStack implementation.

asylo/platform/primitives/examples/BUILD

@@ -40,6 +40,7 @@ sim_enclave(
     copts = ASYLO_DEFAULT_COPTS,
     deps = [
         "//asylo/platform/primitives",
+        "//asylo/platform/primitives:trusted_runtime",
         "//asylo/util:status_macros",
     ],
 )

asylo/platform/primitives/examples/hello_enclave.cc

@@ -22,8 +22,13 @@
 #include "asylo/platform/primitives/extent.h"
 #include "asylo/platform/primitives/primitive_status.h"
 #include "asylo/platform/primitives/trusted_primitives.h"
+#include "asylo/platform/primitives/trusted_runtime.h"
 #include "asylo/util/status_macros.h"
 
+using asylo::primitives::EntryHandler;
+using asylo::primitives::PrimitiveStatus;
+using asylo::primitives::TrustedPrimitives;
+
 namespace asylo {
 namespace primitives {
 
@@ -60,10 +65,6 @@ PrimitiveStatus Hello(void *context, TrustedParameterStack *params) {
 }  // namespace
 
 extern "C" PrimitiveStatus asylo_enclave_init() {
-  ASYLO_RETURN_IF_ERROR(TrustedPrimitives::RegisterEntryHandler(
-      kAbortEnclaveSelector, EntryHandler{Abort}));
-  ASYLO_RETURN_IF_ERROR(TrustedPrimitives::RegisterEntryHandler(
-      kHelloEnclaveSelector, EntryHandler{Hello}));
   return PrimitiveStatus::OkStatus();
 }
 
@@ -73,3 +74,13 @@ extern "C" PrimitiveStatus asylo_enclave_fini() {
 
 }  // namespace primitives
 }  // namespace asylo
+
+extern "C" PrimitiveStatus enc_init() {
+  ASYLO_RETURN_IF_ERROR(TrustedPrimitives::RegisterEntryHandler(
+      asylo::primitives::kAbortEnclaveSelector,
+      EntryHandler{asylo::primitives::Abort}));
+  ASYLO_RETURN_IF_ERROR(TrustedPrimitives::RegisterEntryHandler(
+      asylo::primitives::kHelloEnclaveSelector,
+      EntryHandler{asylo::primitives::Hello}));
+  return PrimitiveStatus::OkStatus();
+}

asylo/platform/primitives/test/BUILD

@@ -53,6 +53,8 @@ sim_enclave(
     copts = ASYLO_DEFAULT_COPTS,
     deps = [
         ":test_selectors",
+        "//asylo/platform/primitives",
+        "//asylo/platform/primitives:trusted_runtime",
     ],
 )
 

asylo/platform/primitives/test/test_enclave.cc

@@ -23,8 +23,11 @@
 #include "asylo/platform/primitives/primitive_status.h"
 #include "asylo/platform/primitives/test/test_selectors.h"
 #include "asylo/platform/primitives/trusted_primitives.h"
+#include "asylo/platform/primitives/trusted_runtime.h"
 #include "asylo/util/status_macros.h"
 
+using asylo::primitives::PrimitiveStatus;
+
 namespace asylo {
 namespace primitives {
 
@@ -254,3 +257,5 @@ extern "C" PrimitiveStatus asylo_enclave_fini() {
 
 }  // namespace primitives
 }  // namespace asylo
+
+extern "C" PrimitiveStatus enc_init() { return PrimitiveStatus::OkStatus(); }

asylo/platform/primitives/trusted_runtime.h

@@ -23,8 +23,13 @@
 #include <cstdlib>
 #include <cstring>
 
+#include "asylo/platform/primitives/primitive_status.h"
+
 extern "C" {
 
+// Prototype of the user-defined enclave initialization function.
+asylo::primitives::PrimitiveStatus enc_init();
+
 // Emulates the Unix `sbrk` system call. See sbrk(2). This functions must be
 // exported by each backend to support linking against libc.
 void *enclave_sbrk(intptr_t increment);