W3MFC web服务器源码

/* Module : W3MFCClient.cpp Purpose: Implementation for the CW3MFCClient class Created: PJN / 22-04-1999 History PJN / 19-02-2006 1. Replaced all calls to ZeroMemory and CopyMemory with memset and memcpy PJN / 27-06-2006 1. Combined the functionality of the _W3MFC_DATA class into the main CW3MFCClient class. PJN / 19-11-2007 1. CHttpClient class has been renamed to CW3MFCClient PJN / 18-02-2008 1. Fixed a memory leak in CW3MFCClient::_TransmitFile PJN / 31-05-2008 1. Code now compiles cleanly using Code Analysis (/analyze) PJN / 23-05-2009 1. Reworked all token parsing code to use CString::Tokenize PJN / 07-09-2009 1. Fixed a debug mode ASSERT when calling TRACE in CW3MFCClient::PostLog Copyright (c) 1999 - 2009 by PJ Naughter (Web: www.naughter.com, Email: pjna@naughter.com) All rights reserved. Copyright / Usage Details: You are allowed to include the source code in any product (commercial, shareware, freeware or otherwise) when your product is released in binary form. You are allowed to modify the source code in any way you want except you cannot modify the copyright details at the top of each module. If you want to distribute source code with your application, then you are only allowed to distribute versions released by the author. This is to maintain a single distribution point for the source code. */ //////////////// Includes ///////////////////////////////////////////////////// #include "stdafx.h" #include "W3MFC.h" #include "W3MFCResponseHeader.h" #include "W3MFCClient.h" #ifndef __ATLENC_H__ #pragma message("To avoid this message, please put atlenc.h in your pre compiled header (usually stdafx.h)") #include <atlenc.h> #endif //////////////// Macros / Defines ///////////////////////////////////////////// #ifdef _DEBUG #define new DEBUG_NEW #endif //////////////// Implementation /////////////////////////////////////////////// IMPLEMENT_DYNCREATE(CW3MFCClient, CThreadPoolClient) CW3MFCClient::CW3MFCClient() : m_lpfnTransmitFile(NULL), m_lpfnTransmitPackets(NULL), m_pServer(NULL), m_hMSWsock(NULL), #ifdef W3MFC_SSL_SUPPORT m_pSSLContext(NULL), #endif #ifndef W3MFC_NO_ISAPI_SUPPORT m_dwDataSentViaWriteClient(0), m_nHttpStatusCodeSent(0), #endif m_bResponseKeepAlive(FALSE) { //Only use the TransmitFile and TransmitPacket API's if we are running on Win2K or later OSVERSIONINFO osvi; osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO); BOOL bNT = (GetVersionEx(&osvi) && osvi.dwPlatformId == VER_PLATFORM_WIN32_NT); if (bNT && (osvi.dwMajorVersion >= 5)) { m_hMSWsock = LoadLibrary(_T("MSWSOCK.DLL")); if (m_hMSWsock) { m_lpfnTransmitFile = reinterpret_cast<LPFN_TRANSMITFILE>(GetProcAddress(m_hMSWsock, "TransmitFile")); m_lpfnTransmitPackets = reinterpret_cast<LPFN_TRANSMITPACKETS>(GetProcAddress(m_hMSWsock, "TransmitPackets")); } } } CW3MFCClient::~CW3MFCClient() { if (m_hMSWsock) { FreeLibrary(m_hMSWsock); m_hMSWsock = NULL; } } #ifndef W3MFC_NO_ISAPI_SUPPORT BOOL CW3MFCClient::_TransmitFile(CW3MFCSocket& socket, CW3MFCResponseHeader& responseHdr, HSE_TF_INFO* pInfo) { //Assume the worst BOOL bSuccess = FALSE; if (pInfo->pfnHseIO) { SetLastError(ERROR_INVALID_PARAMETER); //we do not support ascynchronous notifications return FALSE; } if (pInfo->Offset) { SetLastError(ERROR_INVALID_PARAMETER); //we do not support partials sends thro TransmitFile because that requires overlapped IO return FALSE; } if (pInfo->dwFlags & HSE_IO_SEND_HEADERS) { if (m_lpfnTransmitPackets && m_lpfnTransmitFile) { TRANSMIT_PACKETS_ELEMENT tpe[2]; tpe[0].pBuffer = static_cast<void*>(const_cast<LPSTR>(pInfo->pszStatusCode)); tpe[0].dwElFlags = TP_ELEMENT_MEMORY; tpe[0].cLength = static_cast<ULONG>(strlen(pInfo->pszStatusCode)); CStringA sHeader(responseHdr.GetData()); tpe[1].cLength = sHeader.GetLength(); tpe[1].pBuffer = sHeader.GetBuffer(); tpe[1].dwElFlags = TP_ELEMENT_MEMORY; //Call the TransmitPackets function bSuccess = m_lpfnTransmitPackets(socket, tpe, 2, 0, NULL, TF_USE_KERNEL_APC); sHeader.ReleaseBuffer(); //And the TransmitFile function if (bSuccess) bSuccess = m_lpfnTransmitFile(socket, pInfo->hFile, pInfo->BytesToWrite, 0, NULL, NULL, TF_USE_KERNEL_APC); } else SetLastError(ERROR_CALL_NOT_IMPLEMENTED); } else { if (m_lpfnTransmitFile) { TRANSMIT_FILE_BUFFERS tfb; tfb.Head = pInfo->pHead; tfb.HeadLength = pInfo->HeadLength; tfb.Tail = pInfo->pTail; tfb.TailLength = pInfo->TailLength; //Call the TransmitFile function bSuccess = m_lpfnTransmitFile(socket, pInfo->hFile, pInfo->BytesToWrite, 0, NULL, &tfb, TF_USE_KERNEL_APC); } else SetLastError(ERROR_CALL_NOT_IMPLEMENTED); } return bSuccess; } #endif BOOL CW3MFCClient::_TransmitFile(CW3MFCSocket& socket, CW3MFCResponseHeader& responseHdr, HANDLE hFile, DWORD dwSize) { //Validate our parameter ASSERT(m_lpfnTransmitFile); //Setup the TFB ready for the call to the "TransmitFile" function TRANSMIT_FILE_BUFFERS tfb; CStringA sData(responseHdr.GetData()); tfb.HeadLength = sData.GetLength(); tfb.Head = sData.GetBuffer(); tfb.Tail = NULL; tfb.TailLength = 0; //Call the TransmitFile function BOOL bSuccess = m_lpfnTransmitFile(socket, hFile, dwSize, 0, NULL, &tfb, TF_USE_KERNEL_APC); sData.ReleaseBuffer(); return bSuccess; } BOOL CW3MFCClient::_TransmitBuffer(CW3MFCSocket& socket, CW3MFCResponseHeader& responseHdr, BYTE* byData, DWORD dwSize) { //Validate our parameters ASSERT(m_lpfnTransmitPackets); //Setup the TFB ready for the call to the "TransmitPackets" function TRANSMIT_PACKETS_ELEMENT tpe[2]; CStringA sData(responseHdr.GetData()); tpe[0].pBuffer = sData.GetBuffer(); tpe[0].dwElFlags = TP_ELEMENT_MEMORY; tpe[1].pBuffer = reinterpret_cast<void*>(byData); tpe[1].cLength = dwSize; tpe[0].dwElFlags = TP_ELEMENT_MEMORY; //Call the TransmitPackets function BOOL bSuccess = m_lpfnTransmitPackets(socket, tpe, 2, 0, NULL, TF_USE_KERNEL_APC); sData.ReleaseBuffer(); return bSuccess; } BOOL CW3MFCClient::AllowThisConnection() { return TRUE; } #ifdef W3MFC_SSL_SUPPORT BOOL CW3MFCClient::InitializeSSLConnection() { SSL* pSSL = SSL_new(m_pSSLContext->operator SSL_CTX*()); if (pSSL == NULL) { //Report the error CString sError; sError.Format(_T("CW3MFCClient::InitializeSSLConnection, Failed to create create SSL connection object")); AFXASSUME(m_pServer); m_pServer->OnError(sError); m_pServer->LogSSLErrors(); return FALSE; } else m_SSL.Attach(pSSL); //Associate the raw socket with the SSL connection object if (SSL_set_fd(m_SSL, m_Socket) != 1) { //Report the error CString sError; sError.Format(_T("CW3MFCClient::InitializeSSLConnection, Failed to create create SSL connection object")); AFXASSUME(m_pServer); m_pServer->OnError(sError); m_pServer->LogSSLErrors(); return FALSE; } return TRUE; } BOOL CW3MFCClient::DoSSLNegotiation() { //Validate our parameters AFXASSUME(m_pServer); CW3MFCServerSettings* pSettings = m_pServer->GetSettings(); AFXASSUME(pSettings); BOOL bNegotiationComplete = FALSE; while (!bNegotiationComplete) { int nSSLAccept = SSL_acce