|
@@ -0,0 +1,133 @@
|
|
|
+#!/usr/bin/env python2
|
|
|
+import sys,os
|
|
|
+import subprocess
|
|
|
+import re
|
|
|
+
|
|
|
+
|
|
|
+def gen_dockerfile( image_name, app_name, bin_name, proj_dir) :
|
|
|
+ if not os.path.exists(proj_dir + "/Tools/build") :
|
|
|
+ os.makedirs(proj_dir + "/Tools/build")
|
|
|
+ df =open(proj_dir + "/Tools/build/Dockerfile" + '.' + app_name, 'w')
|
|
|
+ df.write('# This file is auto-generated, any edits will be overwritten\n')
|
|
|
+
|
|
|
+ df.write('\n')
|
|
|
+ # Choose the base image from the user input
|
|
|
+ df.write('FROM '+ image_name + '\n')
|
|
|
+ df.write('\n')
|
|
|
+
|
|
|
+ # SWITCH to ROOT
|
|
|
+ df.write('# SWITCH to root \n')
|
|
|
+ df.write('USER root \n')
|
|
|
+ df.write('\n')
|
|
|
+
|
|
|
+ # DOWNLOAD dependencies
|
|
|
+ df.write('# Download dependencies\n')
|
|
|
+ df.write('RUN apt-get update && \ \n')
|
|
|
+ df.write(' apt-get install -y openssl libjemalloc-dev python python-pip python-dev \n')
|
|
|
+ df.write('RUN pip install protobuf && \ \n')
|
|
|
+ df.write(' pip install pycrypto\n')
|
|
|
+
|
|
|
+ df.write('# Temporal fixes for Dependencies Issue #1: libcrypto.so.1.0.0 and libssl.so.1.0.0 have different locations \n')
|
|
|
+ df.write('RUN ln -s /usr/lib/x86_64-linux-gnu/libcrypto.so.1.0.0 /lib/x86_64-linux-gnu/libcrypto.so.1.0.0 && \ \n')
|
|
|
+ df.write(' ln -s /usr/lib/x86_64-linux-gnu/libssl.so.1.0.0 /lib/x86_64-linux-gnu/libssl.so.1.0.0 \n')
|
|
|
+
|
|
|
+ # SETUP Directory Structure
|
|
|
+ print "cwd: "+ proj_dir
|
|
|
+ df.write('# Setup Directory Structure \n')
|
|
|
+ # df.write('RUN mkdir -p ' + proj_dir + '\n')
|
|
|
+ df.write('RUN mkdir -p ' + proj_dir + '/LibOS/shim/test/apps/' + app_name + '\n')
|
|
|
+ df.write('RUN mkdir -p ' + proj_dir + '/Pal/src/host/Linux-SGX/signer \n')
|
|
|
+ df.write('RUN mkdir -p ' + proj_dir + '/Runtime \n')
|
|
|
+ df.write('RUN mkdir /gbin \n')
|
|
|
+
|
|
|
+ # COPY system files
|
|
|
+ df.write('# Copy system files \n')
|
|
|
+ df.write('COPY Runtime/* ' + proj_dir + '/Runtime/ \n')
|
|
|
+ df.write('COPY Pal/src/Makefile.Host ' + proj_dir + '/Pal/src/Makefile.Host \n')
|
|
|
+ df.write('COPY Pal/src/host/Linux-SGX/signer/* ' + proj_dir + '/Pal/src/host/Linux-SGX/signer/ \n')
|
|
|
+
|
|
|
+ # COPY tools for building app instance
|
|
|
+ df.write('# Copy tools for building app instance\n')
|
|
|
+ df.write('COPY Tools/build/tools/* /gbin/ \n')
|
|
|
+ df.write('COPY Tools/gen_manifest /gbin/ \n')
|
|
|
+
|
|
|
+ # Generating manifest file for target app
|
|
|
+ df.write('# Generating manifest for target app \n')
|
|
|
+ df.write('RUN /gbin/gen_manifest ' + app_name + ' ' + bin_name + ' ' + proj_dir + '\n')
|
|
|
+
|
|
|
+ # Sign Enclave
|
|
|
+ df.write('# Signing Enclave \n')
|
|
|
+ df.write('RUN cd ' + proj_dir + '/LibOS/shim/test/apps/' + app_name + ' && \ \n'
|
|
|
+ ' '+ proj_dir + '/Pal/src/host/Linux-SGX/signer/pal-sgx-sign -libpal ' + proj_dir +
|
|
|
+ '/Pal/src/host/Linux-SGX/../../../../Runtime/libpal-Linux-SGX.so -key ' + proj_dir +
|
|
|
+ '/Pal/src/host/Linux-SGX/signer/enclave-key.pem -output ' + app_name + '.manifest.sgx ' +
|
|
|
+ '-manifest ' + app_name + '.manifest \n')
|
|
|
+ # Remove signing key
|
|
|
+ df.write('# Removing key after signing \n')
|
|
|
+ # TODO
|
|
|
+
|
|
|
+ # Overwrite Entry Point
|
|
|
+ df.write('ENTRYPOINT ["/bin/bash", "/gbin/app_exec"] \n')
|
|
|
+ df.close()
|
|
|
+
|
|
|
+def make_exec(path) :
|
|
|
+ mode = os.stat(path).st_mode
|
|
|
+ mode |= (mode & 0o444) >> 2 # copy R bits to X
|
|
|
+ os.chmod(path, mode)
|
|
|
+
|
|
|
+def gen_app_executor(app_name, bin_cmd, proj_dir) :
|
|
|
+ if not os.path.exists(proj_dir + "/Tools/build/tools") :
|
|
|
+ os.makedirs(proj_dir + "/Tools/build/tools")
|
|
|
+
|
|
|
+ e_path = proj_dir + "/Tools/build/tools/app_exec"
|
|
|
+ print "e_path: " + e_path
|
|
|
+ ef = open(e_path, "w")
|
|
|
+ make_exec(e_path)
|
|
|
+ ef.write('#!/bin/bash \n \n')
|
|
|
+ ef.write('cd ' + proj_dir + '/LibOS/shim/test/apps/' + app_name +'\n')
|
|
|
+ ef.write('# Generate EINITOKEN \n')
|
|
|
+ ef.write(proj_dir + '/Pal/src/host/Linux-SGX/signer/pal-sgx-get-token -output '
|
|
|
+ + app_name + '.token -sig ' + app_name + '.sig \n')
|
|
|
+ ef.write('# Run the application \n')
|
|
|
+ ef.write('SGX=1 ./' + app_name + '.manifest.sgx ' + bin_cmd + '\n')
|
|
|
+
|
|
|
+ ef.close()
|
|
|
+
|
|
|
+if __name__ == "__main__":
|
|
|
+ if len(sys.argv) < 3:
|
|
|
+ print "Usage: gsce run [Image name] "
|
|
|
+ exit()
|
|
|
+
|
|
|
+ image_name = sys.argv[-1]
|
|
|
+ image_match = re.match(r'([^:]*)(:*)(.*)', image_name)
|
|
|
+ if image_match :
|
|
|
+ app_name = image_match.group(1)
|
|
|
+ print "app_name: " + app_name
|
|
|
+ inspect_cmd = "sudo docker inspect --format '{{.Config.Cmd}}' " + image_name
|
|
|
+ res = subprocess.check_output(inspect_cmd, shell=True).strip()
|
|
|
+ print res
|
|
|
+ match = re.match(r'\[([^\s]*)\s*(.*)\]', res)
|
|
|
+ bin_name = match.group(1)
|
|
|
+ if match.group(2) :
|
|
|
+ bin_cmd = match.group(2)
|
|
|
+ else :
|
|
|
+ bin_cmd = ""
|
|
|
+ print "bin_name: " + bin_name + " bin_cmd: " + bin_cmd
|
|
|
+ # Store the rest arguments as docker run arguments
|
|
|
+ docker_str = " " + " ".join(sys.argv[2:-1])
|
|
|
+
|
|
|
+ # print image_cmd
|
|
|
+ proj_dir = os.path.abspath(os.getcwd() + "/../")
|
|
|
+
|
|
|
+ # STEP 1: Generating Dockerfile
|
|
|
+ gen_dockerfile(image_name, app_name, bin_name, proj_dir)
|
|
|
+
|
|
|
+ # STEP 2: Generating entry point execute script
|
|
|
+ gen_app_executor(app_name, bin_cmd, proj_dir)
|
|
|
+
|
|
|
+ # STEP 3: Building new docker image with generated Dockerfile
|
|
|
+ os.chdir('..')
|
|
|
+ os.system("sudo docker build -f Tools/build/Dockerfile." + app_name + " -t gsc_" + app_name + " .\n")
|
|
|
+
|
|
|
+ # STEP 4: Run GSC with the target app
|
|
|
+ os.system("sudo docker run -i -t" + docker_str +" --device=/dev/gsgx --device=/dev/isgx -v /var/run/aesmd/aesm.socket:/var/run/aesmd/aesm.socket gsc_"+app_name+"\n")
|