CSRManagerServer.cpp

/*
    Copyright (C) 2019-Present SKALE Labs

    This file is part of sgxwallet.

    sgxwallet is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published
    by the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    sgxwallet is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with sgxwallet.  If not, see <https://www.gnu.org/licenses/>.

    @file CSRManager.cpp
    @author Stan Kladko
    @date 2019
*/

#include <fstream>
#include <iostream>

#include <jsonrpccpp/server/connectors/httpserver.h>

#include "CSRManagerServer.h"
#include "SGXException.h"
#include "sgxwallet_common.h"

#include "Log.h"
#include "common.h"

shared_ptr<CSRManagerServer> CSRManagerServer::cs = nullptr;
shared_ptr<jsonrpc::HttpServer> CSRManagerServer::hs3 = nullptr;

CSRManagerServer::CSRManagerServer(AbstractServerConnector &connector,
                                   serverVersion_t type)
    : abstractCSRManagerServer(connector, type) {}

Json::Value getUnsignedCSRsImpl() {
  INIT_RESULT(result)

  try {
    vector<string> hashes_vect =
        LevelDB::getCsrDb()->writeKeysToVector1(MAX_CSR_NUM);
    for (int i = 0; i < (int)hashes_vect.size(); i++) {
      result["hashes"][i] = hashes_vect.at(i);
    }
  }
  HANDLE_SGX_EXCEPTION(result);

  RETURN_SUCCESS(result)
}

Json::Value signByHashImpl(const string &hash, int status) {
  INIT_RESULT(result)

  try {
    if (!(status == 0 || status == 2)) {
      throw SGXException(-111, "Invalid csr status");
    }

    string csr_db_key = "CSR:HASH:" + hash;
    shared_ptr<string> csr_ptr = LevelDB::getCsrDb()->readString(csr_db_key);
    if (csr_ptr == nullptr) {
      throw SGXException(KEY_SHARE_DOES_NOT_EXIST, "HASH DOES NOT EXIST IN DB");
    }

    if (status == 0) {
      string csr_name = "sgx_data/cert/" + hash + ".csr";
      ofstream outfile(csr_name);
      outfile << *csr_ptr << endl;
      outfile.close();
      if (access(csr_name.c_str(), F_OK) != 0) {
        LevelDB::getCsrDb()->deleteKey(csr_db_key);
        throw SGXException(FILE_NOT_FOUND, "Csr does not exist");
      }

      string signClientCert =
          "cd sgx_data/cert && ./create_client_cert " + hash;

      if (system(signClientCert.c_str()) == 0) {
        spdlog::info("CLIENT CERTIFICATE IS SUCCESSFULLY GENERATED");
      } else {
        spdlog::info("CLIENT CERTIFICATE GENERATION FAILED");
        LevelDB::getCsrDb()->deleteKey(csr_db_key);
        string status_db_key = "CSR:HASH:" + hash + "STATUS:";
        LevelDB::getCsrStatusDb()->deleteKey(status_db_key);
        LevelDB::getCsrStatusDb()->writeDataUnique(status_db_key, "-1");
        throw SGXException(FAIL_TO_CREATE_CERTIFICATE,
                           "CLIENT CERTIFICATE GENERATION FAILED");
      }
    }

    LevelDB::getCsrDb()->deleteKey(csr_db_key);
    string status_db_key = "CSR:HASH:" + hash + "STATUS:";
    LevelDB::getCsrStatusDb()->deleteKey(status_db_key);
    LevelDB::getCsrStatusDb()->writeDataUnique(status_db_key,
                                               to_string(status));

    result["status"] = status;
  }
  HANDLE_SGX_EXCEPTION(result)

  RETURN_SUCCESS(result)
}

Json::Value CSRManagerServer::getUnsignedCSRs() {
  return getUnsignedCSRsImpl();
}

Json::Value CSRManagerServer::signByHash(const string &hash, int status) {
  return signByHashImpl(hash, status);
}

void CSRManagerServer::initCSRManagerServer() {
  hs3 = make_shared<jsonrpc::HttpServer>(BASE_PORT + 2);
  hs3->BindLocalhost();
  cs = make_shared<CSRManagerServer>(
      *hs3, JSONRPC_SERVER_V2); // server (json-rpc 2.0)

  spdlog::info("Starting csr manager server on port {} ...", BASE_PORT + 2);

  if (!cs->StartListening()) {
    spdlog::info("CSR manager server could not start listening");
    throw SGXException(CSR_MANAGER_SERVER_FAILED_TO_START,
                       "CSRManager server could not start listening.");
  } else {
    spdlog::info("CSR manager server started on port {}", BASE_PORT + 2);
  }
};

int CSRManagerServer::exitServer() {
  spdlog::info("Stoping CSRManager server");

  if (cs && !cs->StopListening()) {
    spdlog::error("CSRManager server could not be stopped. Will forcefully "
                  "terminate the app");
  } else {
    spdlog::info("CSRManager server stopped");
  }

  return 0;
}