From c27825c4364ec8abd1e0a93587fa907dbf0f6ad8 Mon Sep 17 00:00:00 2001
From: LI Qing <geding.lq@antfin.com>
Date: Wed, 6 Jan 2021 17:52:49 +0800
Subject: [PATCH] Add the check for AESM service before run

---
 tools/occlum | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tools/occlum b/tools/occlum
index 10ae85ea..f4cf5f07 100755
--- a/tools/occlum
+++ b/tools/occlum
@@ -102,6 +102,14 @@ check_has_run() {
     fi
 }
 
+check_aesm_service() {
+    for i in $(seq 1 3); do
+        pgrep "aesm_service" > /dev/null && return || sleep 5
+    done
+    echo "Error: AESM service is not started yet. Need to start it first"
+    exit 1
+}
+
 cmd_new() {
     if [ -z $@ ]; then
         echo "Error: target directory is not set"
@@ -254,6 +262,7 @@ cmd_build() {
 cmd_run() {
     check_has_built
     check_has_run
+    check_aesm_service
 
     SGX_MODE=$(cat $instance_dir/.sgx_mode)
     if [[ -n $SGX_MODE && "$SGX_MODE" != "HW" ]]; then
@@ -274,6 +283,7 @@ cmd_run() {
 
 cmd_start() {
     check_has_built
+    check_aesm_service
 
     SGX_MODE=$(cat $instance_dir/.sgx_mode)
     if [[ -n $SGX_MODE && "$SGX_MODE" != "HW" ]]; then