开源的socket服务端客户端,支持C# C++

ServerDlg.cpp 18KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621
  1. // ServerDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "Server.h"
  5. #include "ServerDlg.h"
  6. #include "afxdialogex.h"
  7. // CServerDlg dialog
  8. const CString SPECIAL_SERVER_NAME = _T("hpsocket.org");
  9. int SPECIAL_SERVER_INDEX = -1;
  10. const LPCTSTR CServerDlg::ADDRESS = _T("0.0.0.0");
  11. const USHORT CServerDlg::HTTP_PORT = 8080;
  12. const USHORT CServerDlg::HTTPS_PORT = 8443;
  13. CServerDlg::CServerDlg(CWnd* pParent /*=NULL*/)
  14. : CDialogEx(CServerDlg::IDD, pParent)
  15. , m_HttpListener(HTTP_NAME)
  16. , m_HttpsListener(HTTPS_NAME)
  17. , m_HttpServer(&m_HttpListener)
  18. , m_HttpsServer(&m_HttpsListener)
  19. {
  20. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  21. }
  22. void CServerDlg::DoDataExchange(CDataExchange* pDX)
  23. {
  24. CDialogEx::DoDataExchange(pDX);
  25. DDX_Control(pDX, IDC_INFO, m_Info);
  26. DDX_Control(pDX, IDC_START, m_Start);
  27. DDX_Control(pDX, IDC_STOP, m_Stop);
  28. DDX_Control(pDX, IDC_CONN_ID, m_ConnID);
  29. DDX_Control(pDX, IDC_DISCONNECT, m_DisConn);
  30. DDX_Control(pDX, IDC_RELEASE, m_Release);
  31. DDX_Control(pDX, IDC_HTTP, m_RadioHttp);
  32. DDX_Control(pDX, IDC_HTTPS, m_RadioHttps);
  33. }
  34. BEGIN_MESSAGE_MAP(CServerDlg, CDialogEx)
  35. ON_WM_PAINT()
  36. ON_WM_VKEYTOITEM()
  37. ON_WM_QUERYDRAGICON()
  38. ON_MESSAGE(USER_INFO_MSG, OnUserInfoMsg)
  39. ON_BN_CLICKED(IDC_START, &CServerDlg::OnBnClickedStart)
  40. ON_BN_CLICKED(IDC_STOP, &CServerDlg::OnBnClickedStop)
  41. ON_BN_CLICKED(IDC_DISCONNECT, &CServerDlg::OnBnClickedDisconnect)
  42. ON_EN_CHANGE(IDC_CONN_ID, &CServerDlg::OnEnChangeConnId)
  43. ON_BN_CLICKED(IDC_RELEASE, &CServerDlg::OnBnClickedRelease)
  44. END_MESSAGE_MAP()
  45. // CServerDlg message handlers
  46. BOOL CServerDlg::OnInitDialog()
  47. {
  48. CDialogEx::OnInitDialog();
  49. // Set the icon for this dialog. The framework does this automatically
  50. // when the application's main window is not a dialog
  51. SetIcon(m_hIcon, TRUE); // Set big icon
  52. SetIcon(m_hIcon, FALSE); // Set small icon
  53. // TODO: Add extra initialization here
  54. CString strTitle;
  55. CString strOriginTitle;
  56. GetWindowText(strOriginTitle);
  57. strTitle.Format(_T("%s - (%s:%d/%d)"), strOriginTitle, ADDRESS, HTTP_PORT, HTTPS_PORT);
  58. SetWindowText(strTitle);
  59. m_RadioHttp.SetCheck(BST_CHECKED);
  60. ::SetMainWnd(this);
  61. ::SetInfoList(&m_Info);
  62. SetAppState(ST_STOPPED);
  63. return TRUE; // return TRUE unless you set the focus to a control
  64. }
  65. // If you add a minimize button to your dialog, you will need the code below
  66. // to draw the icon. For MFC applications using the document/view model,
  67. // this is automatically done for you by the framework.
  68. void CServerDlg::OnPaint()
  69. {
  70. if (IsIconic())
  71. {
  72. CPaintDC dc(this); // device context for painting
  73. SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  74. // Center icon in client rectangle
  75. int cxIcon = GetSystemMetrics(SM_CXICON);
  76. int cyIcon = GetSystemMetrics(SM_CYICON);
  77. CRect rect;
  78. GetClientRect(&rect);
  79. int x = (rect.Width() - cxIcon + 1) / 2;
  80. int y = (rect.Height() - cyIcon + 1) / 2;
  81. // Draw the icon
  82. dc.DrawIcon(x, y, m_hIcon);
  83. }
  84. else
  85. {
  86. CDialogEx::OnPaint();
  87. }
  88. }
  89. // The system calls this function to obtain the cursor to display while the user drags
  90. // the minimized window.
  91. HCURSOR CServerDlg::OnQueryDragIcon()
  92. {
  93. return static_cast<HCURSOR>(m_hIcon);
  94. }
  95. BOOL CServerDlg::PreTranslateMessage(MSG* pMsg)
  96. {
  97. if (
  98. pMsg->message == WM_KEYDOWN
  99. &&( pMsg->wParam == VK_ESCAPE
  100. || pMsg->wParam == VK_CANCEL
  101. || pMsg->wParam == VK_RETURN
  102. ))
  103. return TRUE;
  104. return CDialog::PreTranslateMessage(pMsg);
  105. }
  106. void CServerDlg::SetAppState(EnAppState state)
  107. {
  108. m_enState = state;
  109. if(this->GetSafeHwnd() == nullptr)
  110. return;
  111. m_Start.EnableWindow(m_enState == ST_STOPPED);
  112. m_Stop.EnableWindow(m_enState == ST_STARTED);
  113. m_RadioHttp.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  114. m_RadioHttps.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  115. m_DisConn.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  116. m_Release.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  117. }
  118. int CALLBACK CServerDlg::SIN_ServerNameCallback(LPCTSTR lpszServerName)
  119. {
  120. if(::IsIPAddress(lpszServerName))
  121. return 0;
  122. int len = lstrlen(lpszServerName);
  123. int diff = len - SPECIAL_SERVER_NAME.GetLength();
  124. if(diff < 0)
  125. return 0;
  126. if(SPECIAL_SERVER_NAME.CompareNoCase(lpszServerName + diff) != 0)
  127. return 0;
  128. return SPECIAL_SERVER_INDEX;
  129. }
  130. void CServerDlg::OnBnClickedStart()
  131. {
  132. SetAppState(ST_STARTING);
  133. g_SSL.Cleanup();
  134. if(g_SSL.Initialize(SSL_SM_SERVER, SSL_VM_NONE, g_s_lpszPemCertFile, g_s_lpszPemKeyFile, g_s_lpszKeyPasswod, g_s_lpszCAPemCertFileOrPath, SIN_ServerNameCallback))
  135. SPECIAL_SERVER_INDEX = g_SSL.AddServerContext(SSL_VM_NONE, g_s_lpszPemCertFile2, g_s_lpszPemKeyFile2, g_s_lpszKeyPasswod2, g_s_lpszCAPemCertFileOrPath2);
  136. else
  137. {
  138. ::LogServerStartFail(::GetLastError(), _T("initialize SSL env fail"));
  139. SetAppState(ST_STOPPED);
  140. return;
  141. }
  142. if(m_HttpServer.Start(ADDRESS, HTTP_PORT))
  143. {
  144. ::LogServerStart(ADDRESS, HTTP_PORT, HTTP_NAME);
  145. if(m_HttpsServer.Start(ADDRESS, HTTPS_PORT))
  146. {
  147. ::LogServerStart(ADDRESS, HTTPS_PORT, HTTPS_NAME);
  148. SetAppState(ST_STARTED);
  149. }
  150. else
  151. {
  152. m_HttpServer.Stop();
  153. ::LogServerStartFail(m_HttpsServer.GetLastError(), m_HttpsServer.GetLastErrorDesc(), HTTPS_NAME);
  154. SetAppState(ST_STOPPED);
  155. }
  156. }
  157. else
  158. {
  159. ::LogServerStartFail(m_HttpServer.GetLastError(), m_HttpServer.GetLastErrorDesc(), HTTP_NAME);
  160. SetAppState(ST_STOPPED);
  161. }
  162. }
  163. void CServerDlg::OnBnClickedStop()
  164. {
  165. SetAppState(ST_STOPPING);
  166. if(!m_HttpsServer.Stop())
  167. ASSERT(FALSE);
  168. ::LogServerStop(HTTPS_NAME);
  169. if(!m_HttpServer.Stop())
  170. ASSERT(FALSE);
  171. ::LogServerStop(HTTP_NAME);
  172. SetAppState(ST_STOPPED);
  173. }
  174. void CServerDlg::OnBnClickedDisconnect()
  175. {
  176. CString strConnID;
  177. m_ConnID.GetWindowText(strConnID);
  178. CONNID dwConnID = (CONNID)_ttoi(strConnID);
  179. BOOL bHttp = m_RadioHttp.GetCheck() == BST_CHECKED;
  180. LPCTSTR lpszName = bHttp ? HTTP_NAME : HTTPS_NAME;
  181. IHttpServer* pServer = bHttp ? (IHttpServer*)&m_HttpServer : (IHttpServer*)&m_HttpsServer;
  182. if(pServer->Disconnect(dwConnID))
  183. ::LogDisconnect(dwConnID, lpszName);
  184. else
  185. ::LogDisconnectFail(dwConnID, lpszName);
  186. }
  187. void CServerDlg::OnBnClickedRelease()
  188. {
  189. CString strConnID;
  190. m_ConnID.GetWindowText(strConnID);
  191. CONNID dwConnID = (CONNID)_ttoi(strConnID);
  192. BOOL bHttp = m_RadioHttp.GetCheck() == BST_CHECKED;
  193. LPCTSTR lpszName = bHttp ? HTTP_NAME : HTTPS_NAME;
  194. IHttpServer* pServer = bHttp ? (IHttpServer*)&m_HttpServer : (IHttpServer*)&m_HttpsServer;
  195. if(pServer->Release(dwConnID))
  196. ::LogRelease(dwConnID, lpszName);
  197. else
  198. ::LogReleaseFail(dwConnID, lpszName);
  199. }
  200. void CServerDlg::OnEnChangeConnId()
  201. {
  202. m_RadioHttp.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  203. m_RadioHttps.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  204. m_DisConn.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  205. m_Release.EnableWindow(m_enState == ST_STARTED && m_ConnID.GetWindowTextLength() > 0);
  206. }
  207. int CServerDlg::OnVKeyToItem(UINT nKey, CListBox* pListBox, UINT nIndex)
  208. {
  209. if(nKey == 'C')
  210. pListBox->ResetContent();
  211. return __super::OnVKeyToItem(nKey, pListBox, nIndex);
  212. }
  213. LRESULT CServerDlg::OnUserInfoMsg(WPARAM wp, LPARAM lp)
  214. {
  215. info_msg* msg = (info_msg*)wp;
  216. ::LogInfoMsg(msg);
  217. return 0;
  218. }
  219. // ------------------------------------------------------------------------------------------------------------- //
  220. EnHandleResult CHttpServerListenerImpl::OnPrepareListen(ITcpServer* pSender, SOCKET soListen)
  221. {
  222. TCHAR szAddress[40];
  223. int iAddressLen = sizeof(szAddress) / sizeof(TCHAR);
  224. USHORT usPort;
  225. pSender->GetListenAddress(szAddress, iAddressLen, usPort);
  226. ::PostOnPrepareListen(szAddress, usPort, m_strName);
  227. return HR_OK;
  228. }
  229. EnHandleResult CHttpServerListenerImpl::OnAccept(ITcpServer* pSender, CONNID dwConnID, SOCKET soClient)
  230. {
  231. BOOL bPass = TRUE;
  232. TCHAR szAddress[40];
  233. int iAddressLen = sizeof(szAddress) / sizeof(TCHAR);
  234. USHORT usPort;
  235. pSender->GetRemoteAddress(dwConnID, szAddress, iAddressLen, usPort);
  236. ::PostOnAccept(dwConnID, szAddress, usPort, bPass, m_strName);
  237. return bPass ? HR_OK : HR_ERROR;
  238. }
  239. EnHandleResult CHttpServerListenerImpl::OnHandShake(ITcpServer* pSender, CONNID dwConnID)
  240. {
  241. ::PostOnHandShake(dwConnID, m_strName);
  242. return HR_OK;
  243. }
  244. EnHandleResult CHttpServerListenerImpl::OnReceive(ITcpServer* pSender, CONNID dwConnID, const BYTE* pData, int iLength)
  245. {
  246. ::PostOnReceive(dwConnID, pData, iLength, m_strName);
  247. if(pSender->Send(dwConnID, pData, iLength))
  248. return HR_OK;
  249. else
  250. return HR_ERROR;
  251. }
  252. EnHandleResult CHttpServerListenerImpl::OnSend(ITcpServer* pSender, CONNID dwConnID, const BYTE* pData, int iLength)
  253. {
  254. ::PostOnSend(dwConnID, pData, iLength, m_strName);
  255. return HR_OK;
  256. }
  257. EnHandleResult CHttpServerListenerImpl::OnClose(ITcpServer* pSender, CONNID dwConnID, EnSocketOperation enOperation, int iErrorCode)
  258. {
  259. iErrorCode == SE_OK ? ::PostOnClose(dwConnID, m_strName) :
  260. ::PostOnError(dwConnID, enOperation, iErrorCode, m_strName) ;
  261. CBufferPtr* pBuffer = nullptr;
  262. pSender->GetConnectionExtra(dwConnID, (PVOID*)&pBuffer);
  263. if(pBuffer) delete pBuffer;
  264. return HR_OK;
  265. }
  266. EnHandleResult CHttpServerListenerImpl::OnShutdown(ITcpServer* pSender)
  267. {
  268. ::PostOnShutdown(m_strName);
  269. return HR_OK;
  270. }
  271. // ------------------------------------------------------------------------------------------------------------- //
  272. EnHttpParseResult CHttpServerListenerImpl::OnMessageBegin(IHttpServer* pSender, CONNID dwConnID)
  273. {
  274. ::PostOnMessageBegin(dwConnID, m_strName);
  275. return HPR_OK;
  276. }
  277. EnHttpParseResult CHttpServerListenerImpl::OnRequestLine(IHttpServer* pSender, CONNID dwConnID, LPCSTR lpszMethod, LPCSTR lpszUrl)
  278. {
  279. ::PostOnRequestLine(dwConnID, lpszMethod, pSender->GetUrlFieldSet(dwConnID), lpszUrl, m_strName);
  280. return HPR_OK;
  281. }
  282. EnHttpParseResult CHttpServerListenerImpl::OnHeader(IHttpServer* pSender, CONNID dwConnID, LPCSTR lpszName, LPCSTR lpszValue)
  283. {
  284. ::PostOnHeader(dwConnID, lpszName, lpszValue, m_strName);
  285. return HPR_OK;
  286. }
  287. EnHttpParseResult CHttpServerListenerImpl::OnHeadersComplete(IHttpServer* pSender, CONNID dwConnID)
  288. {
  289. CStringA strSummary = GetHeaderSummary(pSender, dwConnID, " ", 0, TRUE);
  290. ::PostOnHeadersComplete(dwConnID, strSummary, m_strName);
  291. return HPR_OK;
  292. }
  293. EnHttpParseResult CHttpServerListenerImpl::OnBody(IHttpServer* pSender, CONNID dwConnID, const BYTE* pData, int iLength)
  294. {
  295. ::PostOnBody(dwConnID, pData, iLength, m_strName);
  296. return HPR_OK;
  297. }
  298. EnHttpParseResult CHttpServerListenerImpl::OnChunkHeader(IHttpServer* pSender, CONNID dwConnID, int iLength)
  299. {
  300. ::PostOnChunkHeader(dwConnID, iLength, m_strName);
  301. return HPR_OK;
  302. }
  303. EnHttpParseResult CHttpServerListenerImpl::OnChunkComplete(IHttpServer* pSender, CONNID dwConnID)
  304. {
  305. ::PostOnChunkComplete(dwConnID, m_strName);
  306. return HPR_OK;
  307. }
  308. EnHttpParseResult CHttpServerListenerImpl::OnMessageComplete(IHttpServer* pSender, CONNID dwConnID)
  309. {
  310. ::PostOnMessageComplete(dwConnID, m_strName);
  311. if(pSender->IsUpgrade(dwConnID))
  312. return HPR_OK;
  313. CStringA strBody = GetHeaderSummary(pSender, dwConnID, " ", 0, FALSE);
  314. int iBodyLength = strBody.GetLength();
  315. BOOL bSkipBody = FALSE;
  316. if(strcmp(pSender->GetMethod(dwConnID), HTTP_METHOD_HEAD) == 0)
  317. bSkipBody = TRUE;
  318. CStringA strContentLength;
  319. strContentLength.Format("%u", iBodyLength);
  320. DWORD dwSeq = 1;
  321. LPCSTR lpszReqSequence = nullptr;
  322. if(pSender->GetCookie(dwConnID, "__reqSequence", &lpszReqSequence))
  323. dwSeq += atoi(lpszReqSequence);
  324. CStringA strSeqCookie;
  325. strSeqCookie.Format("__reqSequence=%u; path=/", dwSeq);
  326. THeader header[] = {{"Content-Type", "text/plain"}, {"Content-Length", strContentLength}, {"Set-Cookie", strSeqCookie}};
  327. int iHeaderCount = sizeof(header) / sizeof(THeader);
  328. if(bSkipBody)
  329. {
  330. strBody.Empty();
  331. iBodyLength = 0;
  332. }
  333. pSender->SendResponse( dwConnID,
  334. HSC_OK,
  335. "HP Http Server OK",
  336. header, iHeaderCount,
  337. (const BYTE*)(LPCSTR)strBody,
  338. iBodyLength);
  339. if(!pSender->IsKeepAlive(dwConnID))
  340. pSender->Release(dwConnID);
  341. return HPR_OK;
  342. }
  343. EnHttpParseResult CHttpServerListenerImpl::OnUpgrade(IHttpServer* pSender, CONNID dwConnID, EnHttpUpgradeType enUpgradeType)
  344. {
  345. ::PostOnUpgrade(dwConnID, enUpgradeType, m_strName);
  346. if(enUpgradeType == HUT_HTTP_TUNNEL)
  347. {
  348. pSender->SendResponse(dwConnID, HSC_OK, "Connection Established");
  349. }
  350. else if(enUpgradeType == HUT_WEB_SOCKET)
  351. {
  352. THeader header[] = {{"Connection", UPGRADE_HEADER},
  353. {UPGRADE_HEADER, WEB_SOCKET_HEADER_VALUE},
  354. {nullptr, nullptr},
  355. {nullptr, nullptr}};
  356. LPCSTR lpszAccept = nullptr;
  357. if(!pSender->GetHeader(dwConnID, "Sec-WebSocket-Key", &lpszAccept))
  358. return HPR_ERROR;
  359. CStringA strAccept;
  360. ::MakeSecWebSocketAccept(lpszAccept, strAccept);
  361. header[2].name = "Sec-WebSocket-Accept";
  362. header[2].value = strAccept;
  363. LPCSTR lpszProtocol = nullptr;
  364. if(pSender->GetHeader(dwConnID, "Sec-WebSocket-Protocol", &lpszProtocol))
  365. {
  366. int i = 0;
  367. CStringA strProtocol(lpszProtocol);
  368. CStringA strFirst = strProtocol.Tokenize(", ", i);
  369. if(!strFirst.IsEmpty())
  370. {
  371. header[3].name = "Sec-WebSocket-Protocol";
  372. header[3].value = strFirst;
  373. }
  374. }
  375. int iHeaderCount = sizeof(header) / sizeof(THeader);
  376. pSender->SendResponse(dwConnID, HSC_SWITCHING_PROTOCOLS, nullptr, header, iHeaderCount);
  377. pSender->SetConnectionExtra(dwConnID, new CBufferPtr);
  378. }
  379. else
  380. ASSERT(FALSE);
  381. return HPR_OK;
  382. }
  383. EnHttpParseResult CHttpServerListenerImpl::OnParseError(IHttpServer* pSender, CONNID dwConnID, int iErrorCode, LPCSTR lpszErrorDesc)
  384. {
  385. ::PostOnParseError(dwConnID, iErrorCode, lpszErrorDesc, m_strName);
  386. return HPR_OK;
  387. }
  388. // ------------------------------------------------------------------------------------------------------------- //
  389. EnHandleResult CHttpServerListenerImpl::OnWSMessageHeader(IHttpServer* pSender, CONNID dwConnID, BOOL bFinal, BYTE iReserved, BYTE iOperationCode, const BYTE lpszMask[4], ULONGLONG ullBodyLen)
  390. {
  391. ::PostOnWSMessageHeader(dwConnID, bFinal, iReserved, iOperationCode, lpszMask, ullBodyLen, m_strName);
  392. return HR_OK;
  393. }
  394. EnHandleResult CHttpServerListenerImpl::OnWSMessageBody(IHttpServer* pSender, CONNID dwConnID, const BYTE* pData, int iLength)
  395. {
  396. ::PostOnWSMessageBody(dwConnID, pData, iLength, m_strName);
  397. CBufferPtr* pBuffer = nullptr;
  398. pSender->GetConnectionExtra(dwConnID, (PVOID*)&pBuffer);
  399. VERIFY(pBuffer);
  400. pBuffer->Cat(pData, iLength);
  401. return HR_OK;
  402. }
  403. EnHandleResult CHttpServerListenerImpl::OnWSMessageComplete(IHttpServer* pSender, CONNID dwConnID)
  404. {
  405. ::PostOnWSMessageComplete(dwConnID, m_strName);
  406. CBufferPtr* pBuffer = nullptr;
  407. pSender->GetConnectionExtra(dwConnID, (PVOID*)&pBuffer);
  408. VERIFY(pBuffer);
  409. BOOL bFinal;
  410. BYTE iReserved, iOperationCode;
  411. VERIFY(pSender->GetWSMessageState(dwConnID, &bFinal, &iReserved, &iOperationCode, nullptr, nullptr, nullptr));
  412. pSender->SendWSMessage(dwConnID, bFinal, iReserved, iOperationCode, nullptr, pBuffer->Ptr(), (int)pBuffer->Size());
  413. pBuffer->Free();
  414. if(iOperationCode == 0x8)
  415. pSender->Disconnect(dwConnID);
  416. return HR_OK;
  417. }
  418. // ------------------------------------------------------------------------------------------------------------- //
  419. CStringA CHttpServerListenerImpl::GetHeaderSummary(IHttpServer* pSender, CONNID dwConnID, LPCSTR lpszSep, int iSepCount, BOOL bWithContentLength)
  420. {
  421. CStringA SEP1;
  422. for(int i = 0; i < iSepCount; i++)
  423. SEP1 += lpszSep;
  424. CStringA SEP2(SEP1);
  425. SEP2 += lpszSep;
  426. CStringA strResult;
  427. USHORT usUrlFieldSet = pSender->GetUrlFieldSet(dwConnID);
  428. strResult.AppendFormat("%s[URL Fields]%s", SEP1, CRLF);
  429. strResult.AppendFormat("%s%8s: %s%s", SEP2, "SCHEMA", pSender->GetUrlField(dwConnID, HUF_SCHEMA), CRLF);
  430. strResult.AppendFormat("%s%8s: %s%s", SEP2, "HOST", pSender->GetUrlField(dwConnID, HUF_HOST), CRLF);
  431. strResult.AppendFormat("%s%8s: %s%s", SEP2, "PORT", pSender->GetUrlField(dwConnID, HUF_PORT), CRLF);
  432. strResult.AppendFormat("%s%8s: %s%s", SEP2, "PATH", pSender->GetUrlField(dwConnID, HUF_PATH), CRLF);
  433. strResult.AppendFormat("%s%8s: %s%s", SEP2, "QUERY", pSender->GetUrlField(dwConnID, HUF_QUERY), CRLF);
  434. strResult.AppendFormat("%s%8s: %s%s", SEP2, "FRAGMENT", pSender->GetUrlField(dwConnID, HUF_FRAGMENT), CRLF);
  435. strResult.AppendFormat("%s%8s: %s%s", SEP2, "USERINFO", pSender->GetUrlField(dwConnID, HUF_USERINFO), CRLF);
  436. DWORD dwHeaderCount = 0;
  437. pSender->GetAllHeaders(dwConnID, nullptr, dwHeaderCount);
  438. strResult.AppendFormat("%s[Request Headers]%s", SEP1, CRLF);
  439. if(dwHeaderCount == 0)
  440. strResult.AppendFormat("%s(no header)%s", SEP2, CRLF);
  441. else
  442. {
  443. unique_ptr<THeader[]> headers(new THeader[dwHeaderCount]);
  444. VERIFY(pSender->GetAllHeaders(dwConnID, headers.get(), dwHeaderCount));
  445. for(DWORD i = 0; i < dwHeaderCount; i++)
  446. strResult.AppendFormat("%s%s: %s%s", SEP2, headers[i].name, headers[i].value, CRLF);
  447. }
  448. DWORD dwCookieCount = 0;
  449. pSender->GetAllCookies(dwConnID, nullptr, dwCookieCount);
  450. strResult.AppendFormat("%s[Cookies]%s", SEP1, CRLF);
  451. if(dwCookieCount == 0)
  452. strResult.AppendFormat("%s(no cookie)%s", SEP2, CRLF);
  453. else
  454. {
  455. unique_ptr<TCookie[]> cookies(new TCookie[dwCookieCount]);
  456. VERIFY(pSender->GetAllCookies(dwConnID, cookies.get(), dwCookieCount));
  457. for(DWORD i = 0; i < dwCookieCount; i++)
  458. strResult.AppendFormat("%s%s: %s%s", SEP2, cookies[i].name, cookies[i].value, CRLF);
  459. }
  460. CStringA strVersion;
  461. ::HttpVersionToString((EnHttpVersion)pSender->GetVersion(dwConnID), strVersion);
  462. EnHttpUpgradeType enUpgType = pSender->GetUpgradeType(dwConnID);
  463. LPCSTR lpszUpgrade = enUpgType != HUT_NONE ? "true" : "false";
  464. LPCSTR lpszKeepAlive = pSender->IsKeepAlive(dwConnID) ? "true" : "false";
  465. strResult.AppendFormat("%s[Basic Info]%s", SEP1, CRLF);
  466. strResult.AppendFormat("%s%13s: %s%s", SEP2, "Version", strVersion, CRLF);
  467. strResult.AppendFormat("%s%13s: %s%s", SEP2, "Method", pSender->GetMethod(dwConnID), CRLF);
  468. strResult.AppendFormat("%s%13s: %s%s", SEP2, "IsUpgrade", lpszUpgrade, CRLF);
  469. if(enUpgType != HUT_NONE)
  470. strResult.AppendFormat("%s%13s: %d%s", SEP2, "UpgradeType", enUpgType, CRLF);
  471. strResult.AppendFormat("%s%13s: %s%s", SEP2, "IsKeepAlive", lpszKeepAlive, CRLF);
  472. if(bWithContentLength)
  473. strResult.AppendFormat("%s%13s: %lld%s", SEP2, "ContentLength", pSender->GetContentLength(dwConnID), CRLF);
  474. strResult.AppendFormat("%s%13s: %s%s", SEP2, "ContentType", pSender->GetContentType(dwConnID), CRLF);
  475. return strResult;
  476. }