[Assignment-7] add SGX sample code from VM

Assignment 7 - SGX Hands-on/SGX101_sample_code-master/PasswordWallet/app/app.cpp

@@ -0,0 +1,225 @@
+#include "enclave_u.h"
+#include "sgx_urts.h"
+
+#include <cstring>
+#include <fstream>
+#include <getopt.h>
+
+#include "app.h"
+#include "utils.h"
+#include "wallet.h"
+#include "enclave.h"
+
+using namespace std;
+
+
+// OCALLs implementation
+int ocall_save_wallet(const uint8_t* sealed_data, const size_t sealed_size) {
+    ofstream file(WALLET_FILE, ios::out | ios::binary);
+    if (file.fail()) {return 1;}
+    file.write((const char*) sealed_data, sealed_size);
+    file.close();
+    return 0;
+}
+
+int ocall_load_wallet(uint8_t* sealed_data, const size_t sealed_size) {
+    ifstream file(WALLET_FILE, ios::in | ios::binary);
+    if (file.fail()) {return 1;}
+    file.read((char*) sealed_data, sealed_size);
+    file.close();
+    return 0;
+}
+
+int ocall_is_wallet(void) {
+    ifstream file(WALLET_FILE, ios::in | ios::binary);
+    if (file.fail()) {return 0;} // failure means no wallet found
+    file.close();
+    return 1;
+}
+
+int main(int argc, char** argv) {
+
+    sgx_enclave_id_t eid = 0;
+    sgx_launch_token_t token = {0};
+    int updated, ret;
+    sgx_status_t ecall_status, enclave_status;
+
+    enclave_status = sgx_create_enclave(ENCLAVE_FILE, SGX_DEBUG_FLAG, &token, &updated, &eid, NULL);
+    if(enclave_status != SGX_SUCCESS) {
+        error_print("Fail to initialize enclave."); 
+        return -1;
+    }
+    info_print("Enclave successfully initilised.");
+
+    const char* options = "hvn:p:c:sax:y:z:r:";
+    opterr=0; // prevent 'getopt' from printing err messages
+    char err_message[100];
+    int opt, stop=0;
+    int h_flag=0, v_flag=0, s_flag=0, a_flag=0;
+    char * n_value=NULL, *p_value=NULL, *c_value=NULL, *x_value=NULL, *y_value=NULL, *z_value=NULL, *r_value=NULL;
+  
+    // read user input
+    while ((opt = getopt(argc, argv, options)) != -1) {
+        switch (opt) {
+            // help
+            case 'h':
+                h_flag = 1;
+                break;
+
+            // create new wallet
+            case 'n':
+                n_value = optarg;
+                break;
+
+            // master-password
+            case 'p':
+                p_value = optarg;
+                break;
+
+            // change master-password
+            case 'c':
+                c_value = optarg;
+                break;
+
+            // show wallet
+            case 's':
+                s_flag = 1;
+                break;
+
+            // add item
+            case 'a': // add item flag
+                a_flag = 1;
+                break;
+            case 'x': // item's title
+                x_value = optarg;
+                break;
+            case 'y': // item's username
+                y_value = optarg;
+                break;
+            case 'z': // item's password
+                z_value = optarg;
+                break;
+
+            // remove item
+            case 'r':
+                r_value = optarg;
+                break;
+
+            // exceptions
+            case '?':
+                if (optopt == 'n' || optopt == 'p' || optopt == 'c' || optopt == 'r' ||
+                    optopt == 'x' || optopt == 'y' || optopt == 'z'
+                ) {
+                    sprintf(err_message, "Option -%c requires an argument.", optopt);
+                }
+                else if (isprint(optopt)) {
+                    sprintf(err_message, "Unknown option `-%c'.", optopt);
+                }
+                else {
+                    sprintf(err_message, "Unknown option character `\\x%x'.",optopt);
+                }
+                stop = 1;
+                error_print(err_message);
+                error_print("Program exiting.");
+                break;
+
+            default:
+                error_print("Unknown option.");
+        }
+    }
+
+    // perform actions
+    if (stop != 1) {
+        // show help
+        if (h_flag) {
+            show_help();
+        }
+
+        // create new wallet
+        else if(n_value!=NULL) {
+            ecall_status = ecall_create_wallet(eid, &ret, n_value);
+            if (ecall_status != SGX_SUCCESS || is_error(ret)) {
+                error_print("Fail to create new wallet.");
+            }
+            else {
+                info_print("Wallet successfully created.");
+            }
+        }
+
+        // change master-password
+        else if (p_value!=NULL && c_value!=NULL) {
+            ecall_status = ecall_change_master_password(eid, &ret, p_value, c_value);
+            if (ecall_status != SGX_SUCCESS || is_error(ret)) {
+                error_print("Fail change master-password.");
+            }
+            else {
+                info_print("Master-password successfully changed.");
+            }
+        }
+
+        // show wallet
+        else if(p_value!=NULL && s_flag) {
+            wallet_t* wallet = (wallet_t*)malloc(sizeof(wallet_t));
+            ecall_status = ecall_show_wallet(eid, &ret, p_value, wallet, sizeof(wallet_t));
+            if (ecall_status != SGX_SUCCESS || is_error(ret)) {
+                error_print("Fail to retrieve wallet.");
+            }
+            else {
+                info_print("Wallet successfully retrieved.");
+                print_wallet(wallet);
+            }
+            free(wallet);
+        }
+
+        // add item
+        else if (p_value!=NULL && a_flag && x_value!=NULL && y_value!=NULL && z_value!=NULL) {
+            item_t* new_item = (item_t*)malloc(sizeof(item_t));
+            strcpy(new_item->title, x_value); 
+            strcpy(new_item->username, y_value); 
+            strcpy(new_item->password, z_value);
+            ecall_status = ecall_add_item(eid, &ret, p_value, new_item, sizeof(item_t));
+            if (ecall_status != SGX_SUCCESS || is_error(ret)) {
+                error_print("Fail to add new item to wallet.");
+            }
+            else {
+                info_print("Item successfully added to the wallet.");
+            }
+            free(new_item);
+        }
+
+        // remove item
+        else if (p_value!=NULL && r_value!=NULL) {
+            char* p_end;
+            int index = (int)strtol(r_value, &p_end, 10);
+            if (r_value == p_end) {
+                error_print("Option -r requires an integer argument.");
+            }
+            else {
+                ecall_status = ecall_remove_item(eid, &ret, p_value, index);
+                if (ecall_status != SGX_SUCCESS || is_error(ret)) {
+                    error_print("Fail to remove item.");
+                }
+                else {
+                    info_print("Item successfully removed from the wallet.");
+                }
+            }
+        }
+
+        // display help
+        else {
+            error_print("Wrong inputs.");
+            show_help();
+        }
+    }
+    
+    // destroy enclave
+    enclave_status = sgx_destroy_enclave(eid);
+    if(enclave_status != SGX_SUCCESS) {
+        error_print("Fail to destroy enclave."); 
+        return -1;
+    }
+    info_print("Enclave successfully destroyed.");
+
+    info_print("Program exit success.");
+    return 0;
+}

Assignment 7 - SGX Hands-on/SGX101_sample_code-master/PasswordWallet/app/app.h

@@ -0,0 +1,13 @@
+#ifndef APP_H_
+#define APP_H_
+
+
+/***************************************************
+ * config.
+ ***************************************************/
+#define APP_NAME "sgx-wallet"
+#define ENCLAVE_FILE "enclave.signed.so"
+#define WALLET_FILE "wallet.seal"
+
+
+#endif // APP_H_

Assignment 7 - SGX Hands-on/SGX101_sample_code-master/PasswordWallet/app/utils.cpp

@@ -0,0 +1,101 @@
+#include <stdio.h>
+#include <cstring>
+
+#include "utils.h"
+#include "app.h"
+#include "wallet.h"
+#include "enclave.h"
+
+void info_print(const char* str) {
+    printf("[INFO] %s\n", str);
+}
+
+void warning_print(const char* str) {
+    printf("[WARNING] %s\n", str);
+}
+
+void error_print(const char* str) {
+    printf("[ERROR] %s\n", str);
+}
+
+void print_wallet(const wallet_t* wallet) {
+    printf("\n-----------------------------------------\n\n");
+    printf("Simple password wallet based on Intel SGX.\n\n");
+    printf("Number of items: %lu\n\n", wallet->size);
+    for (int i = 0; i < wallet->size; ++i) {
+        printf("#%d -- %s\n", i, wallet->items[i].title);
+        printf("[username:] %s\n", wallet->items[i].username);
+        printf("[password:] %s\n", wallet->items[i].password);
+        printf("\n");
+    }
+    printf("\n------------------------------------------\n\n");
+}
+
+int is_error(int error_code) {
+    char err_message[100];
+
+    // check error case
+    switch(error_code) {
+        case RET_SUCCESS:
+            return 0;
+
+        case ERR_PASSWORD_OUT_OF_RANGE:
+            sprintf(err_message, "Password should be at least 8 characters long and at most %d.", MAX_ITEM_SIZE);
+            break;
+
+        case ERR_WALLET_ALREADY_EXISTS:
+            sprintf(err_message, "Wallet already exists: delete file '%s' first.", WALLET_FILE);
+            break;
+
+        case ERR_CANNOT_SAVE_WALLET:
+            strcpy(err_message, "Coud not save wallet.");
+            break;
+
+        case ERR_CANNOT_LOAD_WALLET:
+            strcpy(err_message, "Coud not load wallet.");
+            break;
+
+        case ERR_WRONG_MASTER_PASSWORD:
+            strcpy(err_message, "Wrong master password."); 
+            break;
+
+        case ERR_WALLET_FULL:
+            sprintf(err_message, "Wallet full (maximum number of item: %d).", MAX_ITEMS);
+            break;
+
+        case ERR_ITEM_DOES_NOT_EXIST: 
+            strcpy(err_message, "Item does not exist."); 
+            break;
+
+        case ERR_ITEM_TOO_LONG:
+            sprintf(err_message, "Item too longth (maximum size: %d).", MAX_ITEM_SIZE); 
+            break;
+
+        case ERR_FAIL_SEAL:
+            sprintf(err_message, "Fail to seal wallet."); 
+            break;
+
+        case ERR_FAIL_UNSEAL:
+            sprintf(err_message, "Fail to unseal wallet."); 
+            break;
+
+        default:
+            sprintf(err_message, "Unknown error."); 
+    }
+
+    // print error message
+    error_print(err_message);
+    return 1;
+}
+
+void show_help() {
+	const char* command = "[-h Show this screen] [-v Show version] [-s Show wallet] " \
+		"[-n master-password] [-p master-password -c new-master-password]" \
+		"[-p master-password -a -x items_title -y items_username -z toitems_password]" \
+		"[-p master-password -r items_index]";
+	printf("\nusage: %s %s\n\n", APP_NAME, command);
+}
+
+
+
+

Assignment 7 - SGX Hands-on/SGX101_sample_code-master/PasswordWallet/app/utils.h

@@ -0,0 +1,21 @@
+#ifndef UTIL_H_
+#define UTIL_H_
+
+#include "wallet.h"
+
+void info_print(const char* str);
+
+void warning_print(const char* str);
+
+void error_print(const char* str);
+
+void print_wallet(const wallet_t* wallet);
+
+int is_error(int error_code);
+
+void show_help();
+
+void show_version();
+
+
+#endif // UTIL_H_