Browse Source

Partial refactor - Transforms class

dettanym 4 years ago
parent
commit
a864eb1bed
2 changed files with 452 additions and 0 deletions
  1. 398 0
      App/LAInitiator/Transforms.cpp
  2. 54 0
      Include/Transforms.h

+ 398 - 0
App/LAInitiator/Transforms.cpp

@@ -0,0 +1,398 @@
+//
+// Created by miti on 2020-01-01.
+//
+
+#include "ProtobufLAMessages.pb.h"
+#include "../Decryptor/Decryptor_u.h"
+#include "sgx_eid.h"
+#include "sgx_urts.h"
+#define __STDC_FORMAT_MACROS
+#include <inttypes.h>
+#include <stdio.h>
+#include "sgx_report.h"
+#define SGX_REPORT_BODY_RESERVED1 28
+#define SGX_REPORT_BODY_RESERVED2 32
+#define SGX_REPORT_BODY_RESERVED3 96
+#define SGX_REPORT_BODY_RESERVED4 60
+
+int Transforms::fit_32_into_uint8_t(google::protobuf::uint32 temp32, uint8_t* temp8)
+{
+    if(temp32 > UINT8_MAX)
+        return -1;
+    else
+    {
+        //		 *temp8 = *(uint8_t*)&temp32; // Probably works irrespective of endianness but not sure.
+        *temp8 = (uint8_t)temp32;
+        return 0;
+    }
+}
+
+int Transforms::fit_32_into_uint16_t(google::protobuf::uint32 temp32, uint16_t* temp16)
+{
+    if(temp32 > UINT16_MAX)
+        return -1;
+    else
+    {
+        //              *temp8 = *(uint8_t*)&temp32; // Probably works irrespective of endianness but not sure.
+        *temp16 = (uint16_t)temp32;
+        return 0;
+    }
+}
+
+void Transforms::encode_ec256_public_key_to_protobuf(protobuf_sgx_ec256_public_t* protobuf_g_a , sgx_ec256_public_t* g_a)
+{
+    int counter; google::protobuf::uint32 temp32;
+    for(counter=0;counter<SGX_ECP256_KEY_SIZE;counter++)
+    {
+        temp32 = g_a->gx[counter];
+        protobuf_g_a->add_gx(temp32);
+        temp32 = g_a->gy[counter];
+        protobuf_g_a->add_gy(temp32);
+    }
+}
+
+int Transforms::decode_ec256_public_key_from_protobuf(protobuf_sgx_ec256_public_t* protobuf_g_a , sgx_ec256_public_t* g_a)
+{
+    printf("\n ec256 pub key\n");
+    int counter; google::protobuf::uint32 temp32;
+    for(counter=0;counter<SGX_ECP256_KEY_SIZE;counter++)
+    {
+        temp32 = protobuf_g_a->gx(counter);
+        if(fit_32_into_uint8_t(temp32, &(g_a->gx[counter]))!=0)
+            return -1;
+        printf("%02x ",g_a->gx[counter]);
+        temp32 = protobuf_g_a->gy(counter);
+        if(fit_32_into_uint8_t(temp32, &(g_a->gy[counter]))!=0)
+            return -1;
+        printf("%02x ",g_a->gy[counter]);
+    }
+    return 0;
+}
+
+void Transforms::encode_attributes_to_protobuf(protobuf_sgx_attributes_t* protobuf_attributes, sgx_attributes_t* attributes)
+{
+    protobuf_attributes->set_flags(attributes->flags); // 64 bit
+    protobuf_attributes->set_xfrm(attributes->xfrm); // 64 bit
+}
+
+int Transforms::decode_attributes_from_protobuf(protobuf_sgx_attributes_t* protobuf_attributes, sgx_attributes_t* attributes)
+{
+    attributes->flags = protobuf_attributes->flags();
+    printf("\n flags %" PRIu64 " \n", attributes->flags);
+    attributes->xfrm = protobuf_attributes->xfrm();
+    printf("\n xfrm %" PRIu64 " \n", attributes->xfrm);
+    return 0;
+}
+
+void Transforms::encode_report_to_protobuf(protobuf_sgx_report_t* protobuf_report, sgx_report_t* report)
+{
+    printf("message 3 ------------------------------------------------------------ \n"); fflush(stdout);
+    int counter; google::protobuf::uint32 temp32;
+    printf("Key ID\n");
+    for(counter=0;counter<SGX_KEYID_SIZE;counter++)
+    {
+        temp32=report->key_id.id[counter];
+        printf("%02x ", temp32);
+        protobuf_report->add_key_id(temp32);
+    }
+
+    printf("MAC\n");
+    for(counter=0;counter<SGX_MAC_SIZE;counter++)
+    {
+        temp32=report->mac[counter];
+        printf("%02x ", temp32);
+        protobuf_report->add_mac(temp32);
+    }
+
+    protobuf_report->mutable_body()->set_misc_select(report->body.misc_select); // 32 bit
+    protobuf_report->mutable_body()->set_isv_svn(report->body.isv_svn); // 16 bit
+    protobuf_report->mutable_body()->set_isv_prod_id(report->body.isv_prod_id); // 16 bit
+    encode_attributes_to_protobuf(protobuf_report->mutable_body()->mutable_attributes(), &(report->body.attributes));
+
+    for(counter=0;counter<SGX_CPUSVN_SIZE;counter++)
+    {
+        temp32=report->body.cpu_svn.svn[counter];
+        protobuf_report->mutable_body()->add_cpu_svn(temp32);
+    }
+
+    for(counter=0;counter<SGX_REPORT_BODY_RESERVED1;counter++)
+    {
+        temp32=report->body.reserved1[counter]; // TODO: Could be optimized out - if these are determined to be 0s.
+        protobuf_report->mutable_body()->add_reserved1(temp32);
+    }
+
+    for(counter=0;counter<SGX_REPORT_BODY_RESERVED2;counter++)
+    {
+        temp32=report->body.reserved2[counter]; // TODO: Could be optimized out - if these are determined to be 0s.
+        protobuf_report->mutable_body()->add_reserved2(temp32);
+    }
+
+    for(counter=0;counter<SGX_REPORT_BODY_RESERVED3;counter++)
+    {
+        temp32=report->body.reserved3[counter]; // TODO: Could be optimized out - if these are determined to be 0s.
+        protobuf_report->mutable_body()->add_reserved3(temp32);
+    }
+
+    for(counter=0;counter<SGX_REPORT_BODY_RESERVED4;counter++)
+    {
+        temp32=report->body.reserved4[counter]; // TODO: Could be optimized out - if these are determined to be 0s.
+        protobuf_report->mutable_body()->add_reserved4(temp32);
+    }
+
+    printf("\n enclave measurement\n");
+    for(counter=0;counter<SGX_HASH_SIZE;counter++)
+    {
+        temp32=report->body.mr_enclave.m[counter];
+        printf("%02x ", temp32);
+        protobuf_report->mutable_body()->add_mr_enclave(temp32);
+    }
+
+    printf("\n signer measurement\n");
+    for(counter=0;counter<SGX_HASH_SIZE;counter++)
+    {
+        temp32=report->body.mr_signer.m[counter];
+        printf("%02x ", temp32);
+        protobuf_report->mutable_body()->add_mr_signer(temp32);
+    }
+
+    printf("\n report data\n");
+    for(counter=0;counter<SGX_REPORT_DATA_SIZE;counter++)
+    {
+        temp32=report->body.report_data.d[counter];
+        printf("%02x ", temp32);
+        protobuf_report->mutable_body()->add_report_data(temp32);
+    }
+}
+
+int Transforms::decode_report_from_protobuf(protobuf_sgx_report_t* protobuf_report, sgx_report_t* report)
+{
+    int counter; google::protobuf::uint32 temp32;
+
+    printf("\n decoding report body:\n keyid\n");
+    for(counter=0;counter<SGX_KEYID_SIZE;counter++)
+    {
+        temp32=protobuf_report->key_id(counter);
+        if(fit_32_into_uint8_t(temp32, &(report->key_id.id[counter]))!=0)
+            return -1;
+        printf("%02x  ", report->key_id.id[counter]);
+    }
+
+    printf("\nreport mac\n");
+    for(counter=0;counter<SGX_MAC_SIZE;counter++)
+    {
+        temp32=protobuf_report->mac(counter);
+        if(fit_32_into_uint8_t(temp32, &(report->mac[counter]))!=0)
+            return -1;
+        printf("%02x  ", report->mac[counter]);
+    }
+
+    report->body.misc_select=protobuf_report->mutable_body()->misc_select(); // 32 bit
+    temp32=protobuf_report->mutable_body()->isv_svn();
+    if(fit_32_into_uint16_t(temp32, &(report->body.isv_svn))!=0)
+        return -1;
+    printf("\nmisc select %02x \n", report->body.misc_select);
+
+    temp32=protobuf_report->mutable_body()->isv_prod_id();
+    if(fit_32_into_uint16_t(temp32, &(report->body.isv_prod_id))!=0)
+        return -1;
+    printf("\nprod id %02x \n", report->body.isv_prod_id);
+
+    decode_attributes_from_protobuf(protobuf_report->mutable_body()->mutable_attributes(), &(report->body.attributes));
+
+    printf("\n cpu svn\n");
+    for(counter=0;counter<SGX_CPUSVN_SIZE;counter++)
+    {
+        temp32=protobuf_report->mutable_body()->cpu_svn(counter);
+        if(fit_32_into_uint8_t(temp32, &(report->body.cpu_svn.svn[counter]))!=0)
+            return -1;
+        printf("%02x ", report->body.cpu_svn.svn[counter]);
+
+    }
+/*
+  printf("\n reserved1 \n");
+  for(counter=0;counter<SGX_REPORT_BODY_RESERVED1;counter++)
+  {
+    temp32=protobuf_report->mutable_body()->reserved1(counter);
+    if(fit_32_into_uint8_t(temp32, &(report->body.reserved1[counter]))!=0)
+      return -1;
+    printf("%02x  ", report->body.reserved1[counter]);
+  }
+
+  printf("\n reserved2 \n");
+  for(counter=0;counter<SGX_REPORT_BODY_RESERVED2;counter++)
+  {
+    temp32=protobuf_report->mutable_body()->reserved2(counter);
+    if(fit_32_into_uint8_t(temp32, &(report->body.reserved2[counter]))!=0)
+      return -1;
+    printf("%02x  ", report->body.reserved2[counter]);
+  }
+
+  printf("\n reserved3 \n");
+  for(counter=0;counter<SGX_REPORT_BODY_RESERVED3;counter++)
+  {
+    temp32=protobuf_report->mutable_body()->reserved3(counter);
+    if(fit_32_into_uint8_t(temp32, &(report->body.reserved3[counter]))!=0)
+      return -1;
+    printf("%02x  ", report->body.reserved3[counter]);
+  }
+
+  printf("\n reserved4 \n");
+  for(counter=0;counter<SGX_REPORT_BODY_RESERVED4;counter++)
+  {
+    temp32=protobuf_report->mutable_body()->reserved4(counter);
+    if(fit_32_into_uint8_t(temp32, &(report->body.reserved4[counter]))!=0)
+      return -1;
+    printf("%02x  ", report->body.reserved4[counter]);
+
+  }
+*/
+    printf("\n mrenclave \n");
+    for(counter=0;counter<SGX_HASH_SIZE;counter++)
+    {
+        temp32=protobuf_report->mutable_body()->mr_enclave(counter);
+        if(fit_32_into_uint8_t(temp32, &(report->body.mr_enclave.m[counter]))!=0)
+            return -1;
+        printf("%02x ", report->body.mr_enclave.m[counter]);
+    }
+
+    printf("\n mrsigner \n");
+    for(counter=0;counter<SGX_HASH_SIZE;counter++)
+    {
+        temp32=protobuf_report->mutable_body()->mr_signer(counter);
+        if(fit_32_into_uint8_t(temp32, &(report->body.mr_signer.m[counter]))!=0)
+            return -1;
+        printf("%02x ", report->body.mr_signer.m[counter]);
+    }
+
+    printf("\n report data\n");
+    for(counter=0;counter<SGX_REPORT_DATA_SIZE;counter++)
+    {
+        temp32=protobuf_report->mutable_body()->report_data(counter);
+        if(fit_32_into_uint8_t(temp32, &(report->body.report_data.d[counter]))!=0)
+            return -1;
+        printf("%02x  ", report->body.report_data.d[counter]);
+    }
+    return 0;
+}
+
+int Transforms::print_initialized_msg1( protobuf_sgx_dh_msg1_t& protobuf_dhmsg1, sgx_dh_msg1_t* native_dhmsg1)
+{
+    int counter;
+    printf("Printing msg1:\n");
+    printf("gx\n");
+    for(counter=0;counter<SGX_ECP256_KEY_SIZE;counter++)
+    {
+        printf("%02x ", protobuf_dhmsg1.g_a().gx(counter));
+//      printf("%02x ", native_dhmsg1->g_a.gx[counter]);
+    }
+    printf("\ngy\n");
+    for(counter=0;counter<SGX_ECP256_KEY_SIZE;counter++)
+    {
+        printf("%02x ", protobuf_dhmsg1.g_a().gy(counter));
+//      printf("%02x ", native_dhmsg1->g_a.gy[counter]);
+    }
+
+    printf("\nmrenclave for target\n");
+    for(counter=0;counter<SGX_HASH_SIZE;counter++)
+    {
+        printf("%02x ", protobuf_dhmsg1.target().mr_enclave(counter));
+//      printf("%02x ", native_dhmsg1->target.mr_enclave.m[counter]);
+
+    }
+    /*
+    printf("\nreserved1 in target\n");
+    for(counter=0;counter<SGX_TARGET_INFO_RESERVED1_BYTES;counter++)
+    {
+       printf("%" PRIu32 " ", protobuf_dhmsg1.target().reserved1(counter));
+       printf("%02x  ", native_dhmsg1->target.reserved1[counter]);
+
+    }
+    printf("\nreserved2 in target\n");
+    for(counter=0;counter<SGX_TARGET_INFO_RESERVED2_BYTES;counter++)
+    {
+       printf("%" PRIu32 " ", protobuf_dhmsg1.target().reserved2(counter));
+       printf("%02x  ", native_dhmsg1->target.reserved2[counter]);
+    }
+
+    printf("\n %" PRIu64 "\n", native_dhmsg1->target.attributes.flags);
+    printf("\n %" PRIu64 "\n", protobuf_dhmsg1.target().attributes().flags());
+    printf("\n %" PRIu64 "\n", native_dhmsg1->target.attributes.xfrm);
+    printf("\n %" PRIu64 "\n", protobuf_dhmsg1.target().attributes().xfrm());
+    printf("\n %" PRIu32 "\n", native_dhmsg1->target.misc_select);
+    printf("\n %" PRIu32 "\n", protobuf_dhmsg1.target().misc_select());
+    */
+    fflush(stdout);
+    return 0;
+}
+
+// PUBLIC:
+void Transforms::encode_msg1_to_protobuf( protobuf_sgx_dh_msg1_t& protobuf_dhmsg1, sgx_dh_msg1_t* native_dhmsg1)
+{
+    int counter; google::protobuf::uint32 temp32; // google::protobuf::uint64 temp64;
+
+    encode_ec256_public_key_to_protobuf(protobuf_dhmsg1.mutable_g_a(), &(native_dhmsg1->g_a));
+
+    for(counter=0;counter<SGX_HASH_SIZE;counter++)
+    {
+        temp32=native_dhmsg1->target.mr_enclave.m[counter];
+        protobuf_dhmsg1.mutable_target()->add_mr_enclave(temp32);
+    }
+
+    for(counter=0;counter<SGX_TARGET_INFO_RESERVED1_BYTES;counter++)
+    {
+        temp32=native_dhmsg1->target.reserved1[counter];
+        protobuf_dhmsg1.mutable_target()->add_reserved1(temp32);
+    }
+
+    for(counter=0;counter<SGX_TARGET_INFO_RESERVED2_BYTES;counter++)
+    {
+        temp32=native_dhmsg1->target.reserved2[counter];
+        protobuf_dhmsg1.mutable_target()->add_reserved2(temp32);
+    }
+
+    encode_attributes_to_protobuf(protobuf_dhmsg1.mutable_target()->mutable_attributes(), &(native_dhmsg1->target.attributes));
+
+    temp32=native_dhmsg1->target.misc_select ;
+    protobuf_dhmsg1.mutable_target()->set_misc_select(temp32);
+}
+
+void Transforms::encode_msg3_to_protobuf(protobuf_sgx_dh_msg3_t& protobuf_dhmsg3, sgx_dh_msg3_t* native_dhmsg3)
+{
+    int counter; google::protobuf::uint32 temp32;
+    for(counter=0;counter<SGX_DH_MAC_SIZE;counter++)
+    {
+        temp32=native_dhmsg3->cmac[counter];
+        protobuf_dhmsg3.add_cmac(temp32);
+    }
+
+    encode_report_to_protobuf(protobuf_dhmsg3.mutable_msg3_body()->mutable_report(), &(native_dhmsg3->msg3_body.report));
+    int max_counter=native_dhmsg3->msg3_body.additional_prop_length;
+    unsigned char*temp;
+    for(counter=0,temp=native_dhmsg3->msg3_body.additional_prop;counter<max_counter;counter++,temp++)
+    {
+        protobuf_dhmsg3.mutable_msg3_body()->add_additional_prop(*temp);
+    }
+}
+
+int Transforms::decode_msg2_from_protobuf(protobuf_sgx_dh_msg2_t& protobuf_dhmsg2, sgx_dh_msg2_t* native_dhmsg2)
+{
+    int counter; google::protobuf::uint32 temp32; //google::protobuf::uint64 temp64;
+    printf("RECEIVED the following msg2\n"); fflush(stdout);
+    printf("\ncmac\n");
+    for(counter=0;counter<SGX_DH_MAC_SIZE;counter++)
+    {
+        temp32=protobuf_dhmsg2.cmac(counter);
+        if(fit_32_into_uint8_t(temp32, &(native_dhmsg2->cmac[counter]))!=0)
+            return -1;
+        printf("%02x  ",native_dhmsg2->cmac[counter]);
+    }
+
+    if(decode_ec256_public_key_from_protobuf(protobuf_dhmsg2.mutable_g_b(), &(native_dhmsg2->g_b)) !=0)
+        return -1;
+
+    if(decode_report_from_protobuf(protobuf_dhmsg2.mutable_report(), &(native_dhmsg2->report)) !=0)
+        return -1;
+
+    return 0;
+}
+

+ 54 - 0
Include/Transforms.h

@@ -0,0 +1,54 @@
+//
+// Created by miti on 2020-01-01.
+//
+
+#ifndef DECRYPTORAPP_TRANSFORMS_H
+#define DECRYPTORAPP_TRANSFORMS_H
+
+#include "sgx_eid.h"
+#include "error_codes.h"
+#include "datatypes.h"
+#include "sgx_urts.h"
+#include "sgx_dh.h"
+
+// For msg1
+#define SGX_TARGET_INFO_RESERVED1_BYTES 4
+#define SGX_TARGET_INFO_RESERVED2_BYTES 456
+#define SGX_ECP256_KEY_SIZE 32
+#define SGX_HASH_SIZE 32 /* SHA256 */
+// For msg2
+#define SGX_REPORT_DATA_SIZE 64
+#define SGX_KEYID_SIZE 32
+#define SGX_DH_MAC_SIZE 16
+#define SGX_REPORT_BODY_RESERVED1 28
+#define SGX_REPORT_BODY_RESERVED2 32
+#define SGX_REPORT_BODY_RESERVED3 96
+#define SGX_REPORT_BODY_RESERVED4 60
+
+
+#include <stdio.h>
+// For google protobufs and deserialization/serialization
+#include "ProtobufLAMessages.pb.h"
+#include <google/protobuf/io/coded_stream.h>
+#include <google/protobuf/io/zero_copy_stream_impl.h>
+using namespace google::protobuf::io;
+#include <inttypes.h>
+
+
+class Transforms {
+    static int fit_32_into_uint8_t(google::protobuf::uint32 temp32, uint8_t* temp8);
+    static int fit_32_into_uint16_t(google::protobuf::uint32 temp32, uint16_t* temp16);
+    static void  encode_ec256_public_key_to_protobuf(protobuf_sgx_ec256_public_t* protobuf_g_a , sgx_ec256_public_t* g_a);
+    static int decode_ec256_public_key_from_protobuf(protobuf_sgx_ec256_public_t* protobuf_g_a , sgx_ec256_public_t* g_a);
+    static void  encode_attributes_to_protobuf(protobuf_sgx_attributes_t* protobuf_attributes, sgx_attributes_t* attributes);
+    static void  encode_report_to_protobuf(protobuf_sgx_report_t* protobuf_report, sgx_report_t* report);
+    static int decode_attributes_from_protobuf(protobuf_sgx_attributes_t* protobuf_attributes, sgx_attributes_t* attributes);
+    static int decode_report_from_protobuf(protobuf_sgx_report_t* protobuf_report, sgx_report_t* report);
+public:
+    static void encode_msg1_to_protobuf( protobuf_sgx_dh_msg1_t& protobuf_dhmsg1, sgx_dh_msg1_t* native_dhmsg1);
+    static void encode_msg3_to_protobuf(protobuf_sgx_dh_msg3_t& protobuf_dhmsg3, sgx_dh_msg3_t* native_dhmsg3);
+    static int decode_msg2_from_protobuf(protobuf_sgx_dh_msg2_t& protobuf_dhmsg2, sgx_dh_msg2_t* native_dhmsg2);
+    static int print_initialized_msg1( protobuf_sgx_dh_msg1_t& protobuf_dhmsg1);
+};
+
+#endif //DECRYPTORAPP_TRANSFORMS_H