91 lines
2.1 KiB
C++
91 lines
2.1 KiB
C++
#pragma once
|
||
|
||
#include "pch.h"
|
||
#include "CertificateManager.h"
|
||
|
||
namespace SSLClient {
|
||
|
||
/// <summary>
|
||
/// SSL客户端类
|
||
/// 封装SSL连接、数据发送接收功能
|
||
/// </summary>
|
||
class SSLClientConnection
|
||
{
|
||
public:
|
||
SSLClientConnection();
|
||
~SSLClientConnection();
|
||
|
||
// 禁止拷贝
|
||
SSLClientConnection(const SSLClientConnection&) = delete;
|
||
SSLClientConnection& operator=(const SSLClientConnection&) = delete;
|
||
|
||
/// <summary>
|
||
/// 初始化SSL环境
|
||
/// </summary>
|
||
/// <param name="clientCert">客户端证书PEM</param>
|
||
/// <param name="clientKey">客户端私钥PEM</param>
|
||
/// <param name="caCert">CA证书PEM</param>
|
||
/// <param name="keyPassword">私钥密码</param>
|
||
/// <returns>成功返回true</returns>
|
||
bool Initialize(const char* clientCert, const char* clientKey,
|
||
const char* caCert, const char* keyPassword);
|
||
|
||
/// <summary>
|
||
/// 连接到服务器
|
||
/// </summary>
|
||
/// <param name="address">服务器地址</param>
|
||
/// <param name="port">服务器端口</param>
|
||
/// <returns>成功返回true</returns>
|
||
bool Connect(const char* address, int port);
|
||
|
||
/// <summary>
|
||
/// 发送数据
|
||
/// </summary>
|
||
/// <param name="data">要发送的数据</param>
|
||
/// <returns>成功返回true</returns>
|
||
bool Send(const std::string& data);
|
||
|
||
/// <summary>
|
||
/// 接收数据（非阻塞）
|
||
/// </summary>
|
||
/// <param name="buffer">接收缓冲区</param>
|
||
/// <param name="bufferSize">缓冲区大小</param>
|
||
/// <returns>接收到的字节数，-1表示错误，0表示没有数据</returns>
|
||
int Receive(char* buffer, int bufferSize);
|
||
|
||
/// <summary>
|
||
/// 断开连接
|
||
/// </summary>
|
||
void Disconnect();
|
||
|
||
/// <summary>
|
||
/// 检查是否已连接
|
||
/// </summary>
|
||
bool IsConnected() const { return m_isConnected; }
|
||
|
||
/// <summary>
|
||
/// 获取使用的加密套件
|
||
/// </summary>
|
||
const char* GetCipherSuite() const;
|
||
|
||
private:
|
||
/// <summary>
|
||
/// 初始化Winsock
|
||
/// </summary>
|
||
bool InitializeWinsock();
|
||
|
||
/// <summary>
|
||
/// 清理资源
|
||
/// </summary>
|
||
void Cleanup();
|
||
|
||
private:
|
||
SSL_CTX* m_sslContext;
|
||
SSL* m_ssl;
|
||
SOCKET m_socket;
|
||
bool m_isConnected;
|
||
CertificateManager m_certManager;
|
||
};
|
||
|
||
} // namespace SSLClient
|