text: // Copyright (C) 2017-2018 Baidu, Inc. All Rights Reserved.
  //
  // Redistribution and use in source and binary forms, with or without
  // modification, are permitted provided that the following conditions
  // are met:
  //
  //  * Redistributions of source code must retain the above copyright
  //    notice, this list of conditions and the following disclaimer.
  //  * Redistributions in binary form must reproduce the above copyright
  //    notice, this list of conditions and the following disclaimer in
  //    the documentation and/or other materials provided with the
  //    distribution.
  //  * Neither the name of Baidu, Inc., nor the names of its
  //    contributors may be used to endorse or promote products derived
  //    from this software without specific prior written permission.
  //
  // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  
  #ifndef MESSAGEHANDLER_H
  #define MESSAGEHANDLER_H
  
  #include <string>
  #include <stdio.h>
  #include <limits.h>
  #include <unistd.h>
  #include <iostream>
  #include <iomanip>
  
  #include "Enclave.h"
  #include "NetworkManagerServer.h"
  #include "Messages.pb.h"
  #include "UtilityFunctions.h"
  #include "remote_attestation_result.h"
  #include "LogBase.h"
  #include "../GeneralSettings.h"
  
  using namespace std;
  using namespace util;
  
  class MessageHandler {
  
  public:
      MessageHandler(int port = Settings::rh_port);
      virtual ~MessageHandler();
  
      sgx_ra_msg3_t* getMSG3();
      int init();
      void start();
      vector<string> incomingHandler(string v, int type);
  
  private:
      sgx_status_t initEnclave();
      uint32_t getExtendedEPID_GID();
      sgx_status_t getEnclaveStatus();
  
      void assembleAttestationMSG(Messages::AttestationMessage msg, ra_samp_response_header_t **pp_att_msg);
      string handleAttestationResult(Messages::AttestationMessage msg);
      void assembleMSG2(Messages::MessageMSG2 msg, sgx_ra_msg2_t **pp_msg2);
      string handleMSG2(Messages::MessageMSG2 msg);
      string handleMSG0(Messages::MessageMsg0 msg);
      string generateMSG1();
      string handleVerification();
      string generateMSG0();
      string createInitMsg(int type, string msg);
  
  protected:
      Enclave *enclave = NULL;
  
  private:
      int busy_retry_time = 4;
      NetworkManagerServer *nm = NULL;
  
  };
  
  #endif
  
  