283 lines
10 KiB
C++
283 lines
10 KiB
C++
|
// TestEcho-SSL-Console Client
|
|||
|
|
// 基于HP-Socket的SSL客户端控制台程序
|
|||
|
|
|
|||
|
|
#include "pch.h"
|
|||
|
|
|
|||
|
|
// SSL证书配置（来自helper.cpp）
|
|||
|
|
static const char* g_c_lpszPemCert = R"(-----BEGIN CERTIFICATE-----
|
|||
|
|
MIIDszCCApugAwIBAgIBATANBgkqhkiG9w0BAQsFADB7MQswCQYDVQQGEwJDTjEL
|
|||
|
|
MAkGA1UECAwCR0QxCzAJBgNVBAcMAkdaMQwwCgYDVQQKDANTU1QxDzANBgNVBAsM
|
|||
|
|
Bkplc3NtYTETMBEGA1UEAwwKamVzc21hLm9yZzEeMBwGCSqGSIb3DQEJARYPbGRj
|
|||
|
|
c2FhQDIxY24uY29tMCAXDTI0MDYyNjA1MjUwOFoYDzIyNDMwNzA5MDUyNTA4WjBu
|
|||
|
|
MQswCQYDVQQGEwJDTjELMAkGA1UECAwCR0QxDDAKBgNVBAoMA1NTVDEPMA0GA1UE
|
|||
|
|
CwwGSmVzc21hMRMwEQYDVQQDDApqZXNzbWEub3JnMR4wHAYJKoZIhvcNAQkBFg9s
|
|||
|
|
ZGNzYWFAMjFjbi5jb20wggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQCD
|
|||
|
|
+MyrJEKCheRoOpMRjR78S8hr9W7XN0/EZWyVKwXRT7EE0aGiQdH/W2a+qpWRMa6E
|
|||
|
|
Qi47zdBnt0P8ZoFiItQhuhwUJ064afpVoaHHX25UdbF8r+sRTofadughETBBj2Cf
|
|||
|
|
qh0ia6EOB0QvpJpywWmGZPoMtypjbUiTb/YGOJh2qsVr67MN/E48vt7qt0VxF9SE
|
|||
|
|
pucvqhraTBljWCeRVCae2c0yBSpq/n+7NhamK7+g3xxCKWRz4pN3wrIoEsXTboTh
|
|||
|
|
z940caDgthCc23VJ080DN44jZg6c87huKIuxbebJqw2HCM4DwrW+OSzTLszpFAXZ
|
|||
|
|
yarllOzWnBut20zmYnl1AgMBAAGjTTBLMAkGA1UdEwQCMAAwHQYDVR0OBBYEFJ5E
|
|||
|
|
RJmJ4pUzEbcU9Yge6nr0oi51MB8GA1UdIwQYMBaAFN49z48DywmoD4cNTQgC6nn2
|
|||
|
|
QJoUMA0GCSqGSIb3DQEBCwUAA4IBAQBpoSFfDDDKMAy95tSROpYu5WSWQXe6B7kl
|
|||
|
|
PGJAF6mWe/4b7jHQqDUVkEmFmbMWUAtpTC3P01TrV77dhIosAnC/B76fb7Pto8W4
|
|||
|
|
cjGpWAT0sSegZuhnLtguTGlnR0vVSh/yRRDEtjN8loWpu3BLWVHYOKnn62QGfY0B
|
|||
|
|
sRGrfZsKvwB+1w+HOvGopnWv6UYwrzEKthjPMR65rOsoManOv24ua8baJmq0gqF9
|
|||
|
|
752kD8n703uWUBx79/QlNIPMZC1iUIi1mEjyrTgSag6+3sWAIKihaoF/Nf9d01nw
|
|||
|
|
iL16EIT5dJ0QJWDCeIxhuTZckw+gL1pBeQU7pqzKHPnvo+8GBnTG
|
|||
|
|
-----END CERTIFICATE-----
|
|||
|
|
)";
|
|||
|
|
|
|||
|
|
static const char* g_c_lpszPemKey = R"(-----BEGIN ENCRYPTED PRIVATE KEY-----
|
|||
|
|
MIIFLTBXBgkqhkiG9w0BBQ0wSjApBgkqhkiG9w0BBQwwHAQIK2UJW9QXIj4CAggA
|
|||
|
|
MAwGCCqGSIb3DQIJBQAwHQYJYIZIAWUDBAEqBBCDDZQLhAdT91jd6v/5H0+GBIIE
|
|||
|
|
0PH6tKl+nPi8sU0ryjxDIrHwrT/ZFah+3TAHGE/YFAOZnzRyCFHQTvUZX4p8eSmw
|
|||
|
|
WOpt5NBUPJ3mT0Ctt7lGBRy4AXSyBrFSamlTruM3P1e3ijluYjMbweZFfCWPq8c/
|
|||
|
|
jPjbcUkXe6mD96aPSTt/jIunexS8AKovu8c/bFLyTLDk38lATc+GnXQQJ0KiXCRu
|
|||
|
|
vpjVSKcv2Br6cWqaNTZ71FvH1RmSD6K6givc0w65pKruHYTMApIRR8YC5Y0vx0gD
|
|||
|
|
6nS12LV/EJEtxTfZFlrzZTRWZISPIzYGuTfS+3bPePlxpbwzhN6vmvgjKhdk+3lN
|
|||
|
|
3W3ZfqODNhoOKG+mG5Fdj7vR2PU1UND6UUd3+FrzWkXikmalAAwKzRLnyTR1T2rl
|
|||
|
|
RhM0Qe/HZianeEQTHpCw27gOz1OMw2EKfIEHM6W2BKGOTY5ls5dqgMfP1ZoQUrOr
|
|||
|
|
59tJo4GpWYFGCuHhTEa5OS/gsgnzymGrkuEwPsdSQaBdzV7lFGTv2/ryKX+vNm9V
|
|||
|
|
CmKw0nHzOVP19+WL4vPDtbRnLUk8KV9Mg7PdSbGbNcMmTEBk8ju8OvjIUIWZbRTa
|
|||
|
|
n5C6fhD1DYZcczmlCILYgXyJISu7EDf3z9cKRAf5VbRAedDMB/xHWmrmlxUJ37Kt
|
|||
|
|
tVgaCD0U6Q3q+3y6OOwugc8UbSo4yA/DbLlG0/U7afwQaNxTLa4HGBQljpoNStIt
|
|||
|
|
Vgfy2olqHXaf2doSQtsYEl9MHa6neuGfZQMtonDkejnx4KKU+cMhe+KijEUwieYx
|
|||
|
|
7aoPB71b82XODquDPAL5zOegj0eYgKn5iXyOx5W44S34zfclxtxxgfsDJ3qJ9qoL
|
|||
|
|
sSenrQ3xAYHJSZRcqEgO31XhoEnkyt1V7G0Bk4/GUMD6uQudr3nsw/ulJpAlNK15
|
|||
|
|
ZxTSKWrtwOWdwcTj6B14K6wcqMFVNF1Ydbv/qp0b5q5S/orYHzRIPcFmdOAIsjyO
|
|||
|
|
6na7+D31BH/4pf+TASBNqRNRw5CBqNcGcfiXk11AywxUnmD5ZvC/C0pTpTD/9qC4
|
|||
|
|
LucWJ0sNAtPq8suFjKqQ+wMvq3rUh050NRm2cm2nUJLxafTnr0v3+kKYbVW8pSWB
|
|||
|
|
NMelZMVGF1MDYBujg8Mw/xuMhPeLozCZeKmo7eu7aDMXzQMZLfAEJAzU9Du8H4nq
|
|||
|
|
GgQVUgEkS5rdbjZGkHP0FuM8m8lueKEPDYwHCJv9Be5Z/uxp9OO/Lmdlha0J7gJu
|
|||
|
|
pihNkAYVxRst96b5okXKooYi/TZxAdThoPYH28VwinGR1I3/8I3M5DbUPIgHhDeB
|
|||
|
|
ga3u7jt7ZNDUgavukUD0S7WioRb5ooXrXGZ1xmzKLCmMdCDC5S32fQS0wRGfVoMl
|
|||
|
|
hWbaT+0uak+fOpqVRxSNyE3Ek788ua5iPHaTSXJSoe5lv7OQKDSZ/+wFeLmDPf4M
|
|||
|
|
BHL2gBLD6RNkz5cWgy14sQcJKNAnyptU4EGPyURZcB8APtB/ITAS2Az/JSxvSBgq
|
|||
|
|
g/L1FujnP2QEpWpVKkTNxsF867bUPN34KrlPKYjNqcKA2pD4fkFoKSeeNtOEWa++
|
|||
|
|
d6q9y+mDD97SnIFAAhDFlukzXtyl4MU6uiqRldFiuEt3KzvV19n8M+NyyYIFhfdg
|
|||
|
|
6TkYEbMJPQ/Y3EGNmyMqbFdJzrdl/B8pr7JQnikTfUZZ
|
|||
|
|
-----END ENCRYPTED PRIVATE KEY-----
|
|||
|
|
)";
|
|||
|
|
|
|||
|
|
static const char* g_c_lpszCAPemCert = R"(-----BEGIN CERTIFICATE-----
|
|||
|
|
MIID2TCCAsGgAwIBAgIUM8TTtPU+ejzffYXCcs/zZsU7OuIwDQYJKoZIhvcNAQEL
|
|||
|
|
BQAwezELMAkGA1UEBhMCQ04xCzAJBgNVBAgMAkdEMQswCQYDVQQHDAJHWjEMMAoG
|
|||
|
|
A1UECgwDU1NUMQ8wDQYDVQQLDAZKZXNzbWExEzARBgNVBAMMCmplc3NtYS5vcmcx
|
|||
|
|
HjAcBgkqhkiG9w0BCQEWD2xkY3NhYUAyMWNuLmNvbTAgFw0yNDA2MjYwNTA0NDNa
|
|||
|
|
GA8yMjcwMTEyNDA1MDQ0M1owezELMAkGA1UEBhMCQ04xCzAJBgNVBAgMAkdEMQsw
|
|||
|
|
CQYDVQQHDAJHWjEMMAoGA1UECgwDU1NUMQ8wDQYDVQQLDAZKZXNzbWExEzARBgNV
|
|||
|
|
BAMMCmplc3NtYS5vcmcxHjAcBgkqhkiG9w0BCQEWD2xkY3NhYUAyMWNuLmNvbTCC
|
|||
|
|
ASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAML+v79+aLQt0Za0dTIZHI5B
|
|||
|
|
NDs0g5G8bhdOTlW/kNWflaziZ3GY6d6nJSkQ5e29kyFKxlOD6Gls6bOJ86U71u4R
|
|||
|
|
bCmoFvRTDH4q2cJ/+PbiioLpNveDG6lnRCs9JNRQoJrkpRo6urnVnAdsIf6UFjLI
|
|||
|
|
dlByNMPGYJ0V8/oKJG5Vu5gcbZV0jVA5+tswkH/zquexEXoKvp18mcwl+pNc/LwW
|
|||
|
|
0WnGj0uoJjxHg4GsS78PASjhxMR/2d/1OpgPauldFaNHjVPtaLqJnuejwA6M6Sz8
|
|||
|
|
iFPybAQAMpHL9W8kf08jtbnFvnm4ibUkQL5h+OJoIEQa9AVZOSoFG2/g5Zcn8X8C
|
|||
|
|
AwEAAaNTMFEwHQYDVR0OBBYEFN49z48DywmoD4cNTQgC6nn2QJoUMB8GA1UdIwQY
|
|||
|
|
MBaAFN49z48DywmoD4cNTQgC6nn2QJoUMA8GA1UdEwEB/wQFMAMBAf8wDQYJKoZI
|
|||
|
|
hvcNAQELBQADggEBALJnYrYBSZLyYX14FQ04zxG3AX0CtQzNOOa7LDrr+H8Ly+nK
|
|||
|
|
qS87gg2njMVZH1zM2demtMwydR/F2Ui8ggaduMvc9h5YgQKEwYl8KarJEY03oZoe
|
|||
|
|
zbQGBxCXpDOtMs1vujzcl/iZbSzwEDF3g4la5U8q4MlmfGFKz9CJbvoxecqYA206
|
|||
|
|
nNbW2XZsW/xMiQv6iAw5iP/LOR9HAyxcvXIsL790nfcgnTYLmyP254Dj4outc6R+
|
|||
|
|
PA+f/c1FvkbUBTR5vJt2tsvHcNU218rY2hyOIhDmZeUWprqBO19sUk3scLbVPr3+
|
|||
|
|
WEWEl2XaCekKuPtAnMgVQuFsocXGyiuIhkOe5Z4=
|
|||
|
|
-----END CERTIFICATE-----
|
|||
|
|
)";
|
|||
|
|
|
|||
|
|
static const int g_c_iVerifyMode = SSL_VM_PEER;
|
|||
|
|
static const char* g_c_lpszKeyPassword = "123456";
|
|||
|
|
static const LPCTSTR DEFAULT_ADDRESS = _T("127.0.0.1");
|
|||
|
|
static const USHORT DEFAULT_PORT = 5555;
|
|||
|
|
|
|||
|
|
inline static bool writeFile(const std::string& path, const std::vector<unsigned char>& data) {
|
|||
|
|
std::ofstream f(path, std::ios::binary);
|
|||
|
|
if (!f) return false;
|
|||
|
|
// 写入数据
|
|||
|
|
f.write(reinterpret_cast<const char*>(data.data()), data.size());
|
|||
|
|
// 检查是否全部写入成功
|
|||
|
|
if (!f) return false;
|
|||
|
|
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
// 客户端监听器类
|
|||
|
|
class CClientListener : public CTcpClientListener
|
|||
|
|
{
|
|||
|
|
private:
|
|||
|
|
bool m_bConnected;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
CClientListener() : m_bConnected(false) {}
|
|||
|
|
|
|||
|
|
bool IsConnected() const { return m_bConnected; }
|
|||
|
|
|
|||
|
|
virtual EnHandleResult OnConnect(ITcpClient* pSender, CONNID dwConnID) override
|
|||
|
|
{
|
|||
|
|
TCHAR szAddress[100];
|
|||
|
|
int iAddressLen = sizeof(szAddress) / sizeof(TCHAR);
|
|||
|
|
USHORT usPort;
|
|||
|
|
pSender->GetLocalAddress(szAddress, iAddressLen, usPort);
|
|||
|
|
|
|||
|
|
std::wcout << L"[客户端] 连接成功: 本地地址=" << szAddress << L":" << usPort << std::endl;
|
|||
|
|
m_bConnected = true;
|
|||
|
|
|
|||
|
|
return HR_OK;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
virtual EnHandleResult OnHandShake(ITcpClient* pSender, CONNID dwConnID) override
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] SSL握手完成" << std::endl;
|
|||
|
|
return HR_OK;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
virtual EnHandleResult OnReceive(ITcpClient* pSender, CONNID dwConnID, const BYTE* pData, int iLength) override
|
|||
|
|
{
|
|||
|
|
std::string strData((char*)pData, iLength);
|
|||
|
|
std::cout << "[客户端] 收到服务器数据，长度为：" << iLength << std::endl;
|
|||
|
|
|
|||
|
|
return HR_OK;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
virtual EnHandleResult OnSend(ITcpClient* pSender, CONNID dwConnID, const BYTE* pData, int iLength) override
|
|||
|
|
{
|
|||
|
|
return HR_OK;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
virtual EnHandleResult OnClose(ITcpClient* pSender, CONNID dwConnID, EnSocketOperation enOperation, int iErrorCode) override
|
|||
|
|
{
|
|||
|
|
if (iErrorCode == SE_OK)
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] 连接已关闭" << std::endl;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] 连接错误: 错误码=" << iErrorCode << std::endl;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
m_bConnected = false;
|
|||
|
|
return HR_OK;
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
void PrintMenu()
|
|||
|
|
{
|
|||
|
|
std::wcout << L"\n----------------------------------------" << std::endl;
|
|||
|
|
std::wcout << L"命令菜单:" << std::endl;
|
|||
|
|
std::wcout << L" 1 - 发送 'hello' 到服务器" << std::endl;
|
|||
|
|
std::wcout << L" Q - 断开连接并退出" << std::endl;
|
|||
|
|
std::wcout << L"----------------------------------------" << std::endl;
|
|||
|
|
std::wcout << L"请输入命令: ";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int main()
|
|||
|
|
{
|
|||
|
|
// 设置控制台UTF-8编码和locale
|
|||
|
|
SetConsoleOutputCP(CP_UTF8);
|
|||
|
|
std::locale::global(std::locale(""));
|
|||
|
|
std::wcout.imbue(std::locale(""));
|
|||
|
|
|
|||
|
|
std::wcout << L"========================================" << std::endl;
|
|||
|
|
std::wcout << L" HP-Socket SSL Client 控制台演示程序" << std::endl;
|
|||
|
|
std::wcout << L"========================================" << std::endl;
|
|||
|
|
std::wcout << std::endl;
|
|||
|
|
|
|||
|
|
// 创建监听器和客户端
|
|||
|
|
CClientListener listener;
|
|||
|
|
CSSLClientPtr client(&listener);
|
|||
|
|
|
|||
|
|
// 初始化SSL上下文
|
|||
|
|
std::wcout << L"[客户端] 正在初始化SSL环境..." << std::endl;
|
|||
|
|
if (!client->SetupSSLContextByMemory(g_c_iVerifyMode, g_c_lpszPemCert, g_c_lpszPemKey,
|
|||
|
|
g_c_lpszKeyPassword, g_c_lpszCAPemCert))
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] SSL初始化失败！错误码: " << client->GetLastError() << std::endl;
|
|||
|
|
std::wcout << L"按任意键退出..." << std::endl;
|
|||
|
|
_getch();
|
|||
|
|
return 1;
|
|||
|
|
}
|
|||
|
|
std::wcout << L"[客户端] SSL环境初始化成功" << std::endl;
|
|||
|
|
|
|||
|
|
// 连接服务器
|
|||
|
|
std::wcout << L"[客户端] 正在连接服务器 " << DEFAULT_ADDRESS << L":" << DEFAULT_PORT << L" ..." << std::endl;
|
|||
|
|
if (!client->Start(DEFAULT_ADDRESS, DEFAULT_PORT, FALSE))
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] 连接失败！错误码: " << client->GetLastError() << std::endl;
|
|||
|
|
std::wcout << L"按任意键退出..." << std::endl;
|
|||
|
|
_getch();
|
|||
|
|
return 1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 等待连接完成
|
|||
|
|
int waitCount = 0;
|
|||
|
|
while (!listener.IsConnected() && waitCount < 50)
|
|||
|
|
{
|
|||
|
|
Sleep(100);
|
|||
|
|
waitCount++;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!listener.IsConnected())
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] 连接超时！" << std::endl;
|
|||
|
|
client->Stop();
|
|||
|
|
client->CleanupSSLContext();
|
|||
|
|
std::wcout << L"按任意键退出..." << std::endl;
|
|||
|
|
_getch();
|
|||
|
|
return 1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
std::wcout << L"[客户端] 已连接到服务器，可以开始发送消息" << std::endl;
|
|||
|
|
|
|||
|
|
// 主循环 - 等待用户输入命令
|
|||
|
|
PrintMenu();
|
|||
|
|
|
|||
|
|
bool bRunning = true;
|
|||
|
|
while (bRunning)
|
|||
|
|
{
|
|||
|
|
if (_kbhit())
|
|||
|
|
{
|
|||
|
|
char ch = _getch();
|
|||
|
|
std::cout << ch << std::endl;
|
|||
|
|
|
|||
|
|
if (ch == '1')
|
|||
|
|
{
|
|||
|
|
// 发送 "hello" 到服务器
|
|||
|
|
std::string message = "hello";
|
|||
|
|
if (client->Send((const BYTE*)message.c_str(), (int)message.length()))
|
|||
|
|
{
|
|||
|
|
std::cout << "[客户端] 发送消息: \"" << message << "\"" << std::endl;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] 发送失败！错误码: " << client->GetLastError() << std::endl;
|
|||
|
|
}
|
|||
|
|
PrintMenu();
|
|||
|
|
}
|
|||
|
|
else if (ch == 'q' || ch == 'Q')
|
|||
|
|
{
|
|||
|
|
bRunning = false;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
std::wcout << L"无效命令，请重新输入。" << std::endl;
|
|||
|
|
PrintMenu();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
Sleep(100);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 断开连接
|
|||
|
|
std::wcout << L"[客户端] 正在断开连接..." << std::endl;
|
|||
|
|
if (client->Stop())
|
|||
|
|
{
|
|||
|
|
std::wcout << L"[客户端] 已断开连接" << std::endl;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 清理SSL上下文
|
|||
|
|
client->CleanupSSLContext();
|
|||
|
|
|
|||
|
|
std::wcout << L"按任意键退出..." << std::endl;
|
|||
|
|
_getch();
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|