57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
/*
|
|
* IXSocketMbedTLS.h
|
|
* Author: Benjamin Sergeant
|
|
* Copyright (c) 2019 Machine Zone, Inc. All rights reserved.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "IXSocket.h"
|
|
#include "IXSocketTLSOptions.h"
|
|
#include <mbedtls/ctr_drbg.h>
|
|
#include <mbedtls/debug.h>
|
|
#include <mbedtls/entropy.h>
|
|
#include <mbedtls/error.h>
|
|
#include <mbedtls/net.h>
|
|
#include <mbedtls/platform.h>
|
|
#include <mbedtls/x509.h>
|
|
#include <mbedtls/x509_crt.h>
|
|
#include <mutex>
|
|
|
|
namespace ix
|
|
{
|
|
class SocketMbedTLS final : public Socket
|
|
{
|
|
public:
|
|
SocketMbedTLS(const SocketTLSOptions& tlsOptions, int fd = -1);
|
|
~SocketMbedTLS();
|
|
|
|
virtual bool accept(std::string& errMsg) final;
|
|
|
|
virtual bool connect(const std::string& host,
|
|
int port,
|
|
std::string& errMsg,
|
|
const CancellationRequest& isCancellationRequested) final;
|
|
virtual void close() final;
|
|
|
|
virtual ssize_t send(char* buffer, size_t length) final;
|
|
virtual ssize_t send(const std::string& buffer) final;
|
|
virtual ssize_t recv(void* buffer, size_t length) final;
|
|
|
|
private:
|
|
mbedtls_ssl_context _ssl;
|
|
mbedtls_ssl_config _conf;
|
|
mbedtls_entropy_context _entropy;
|
|
mbedtls_ctr_drbg_context _ctr_drbg;
|
|
mbedtls_x509_crt _cacert;
|
|
mbedtls_x509_crt _cert;
|
|
|
|
std::mutex _mutex;
|
|
SocketTLSOptions _tlsOptions;
|
|
|
|
bool init(const std::string& host, bool isClient, std::string& errMsg);
|
|
void initMBedTLS();
|
|
};
|
|
|
|
} // namespace ix
|