| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 | 
							- #ifndef BELL_TLS_SOCKET_H
 
- #define BELL_TLS_SOCKET_H
 
- #include "BellLogger.h"
 
- #include "BellSocket.h"
 
- #include <cstring>
 
- #include <ctype.h>
 
- #include <fstream>
 
- #include <iostream>
 
- #include <memory>
 
- #include <netdb.h>
 
- #include <netinet/in.h>
 
- #include <netinet/tcp.h>
 
- #include <sstream>
 
- #include <stdlib.h>
 
- #include <string>
 
- #include <sys/socket.h>
 
- #include <sys/types.h>
 
- #include <unistd.h>
 
- #include <vector>
 
- #ifdef BELL_USE_MBEDTLS
 
- #include "mbedtls/net_sockets.h"
 
- #include "mbedtls/ssl.h"
 
- #include "mbedtls/entropy.h"
 
- #include "mbedtls/ctr_drbg.h"
 
- #include "mbedtls/debug.h"
 
- #else
 
- #include <openssl/bio.h>
 
- #include <openssl/err.h>
 
- #include <openssl/ssl.h>
 
- #endif
 
- namespace bell {
 
- class TLSSocket : public bell::Socket {
 
- private:
 
- #ifdef BELL_USE_MBEDTLS
 
-     mbedtls_net_context server_fd;
 
-     mbedtls_entropy_context entropy;
 
-     mbedtls_ctr_drbg_context ctr_drbg;
 
-     mbedtls_ssl_context ssl;
 
-     mbedtls_ssl_config conf;
 
- #else
 
-   BIO *sbio, *out;
 
-   int len;
 
-   char tmpbuf[1024];
 
-   SSL_CTX *ctx;
 
-   SSL *ssl;
 
-   int sockFd;
 
-   int sslFd;
 
- #endif
 
-   bool isClosed = false;
 
- public:
 
-   TLSSocket();
 
-   ~TLSSocket() { close(); };
 
-   void open(std::string host, uint16_t port);
 
-   size_t read(uint8_t *buf, size_t len);
 
-   size_t write(uint8_t *buf, size_t len);
 
-   size_t poll();
 
-   void close();
 
- };
 
- } // namespace bell
 
- #endif
 
 
  |