@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/python3
import sys
import os
import subprocess
@@ -1,9 +1,8 @@
#!/bin/bash
SGX_DRIVER_FILE=/dev/isgx
-test -e "$SGX_DRIVER_FILE"
-SGX_FLAG=$?
-if [ $SGX_FLAG ]
+test -e "$SGX_DRIVER_FILE" && SGX_FLAG=1 || SGX_FLAG=0
+if [ $SGX_FLAG == "1" ]
then
docker run --device /dev/isgx -dt --name zt_container zt_image bash -c "start-aesmd && exec bash"
docker exec -t zt_container bash -c "cd /pir/ && ./run_ztlsoram.sh"