Hp-socket/HP-Socket-ma.../HP-Socket/Src/HttpClient.cpp

/*
 * Copyright: JessMA Open Source (ldcsaa@gmail.com)
 *
 * Version	: 4.1.3
 * Author	: Bruce Liang
 * Website	: http://www.jessma.org
 * Project	: https://github.com/ldcsaa
 * Blog		: http://www.cnblogs.com/ldcsaa
 * Wiki		: http://www.oschina.net/p/hp-socket
 * QQ Group	: 75375912
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#include "stdafx.h"
#include "HttpClient.h"

template<class R, class T, USHORT default_port> BOOL CHttpClientT<R, T, default_port>::CheckParams()
{
	if(m_enLocalVersion != HV_1_1 && m_enLocalVersion != HV_1_0)
	{
		SetLastError(SE_INVALID_PARAM, __FUNCTION__, ERROR_INVALID_PARAMETER);
		return FALSE;
	}

	return __super::CheckParams();
}

template<class R, class T, USHORT default_port> BOOL CHttpClientT<R, T, default_port>::SendRequest(LPCSTR lpszMethod, LPCSTR lpszPath, const THeader lpHeaders[], int iHeaderCount, const BYTE* pBody, int iLength)
{
	USES_CONVERSION;

	WSABUF szBuffer[2];
	CStringA strHeader;

	LPCSTR lpszHost	= nullptr;
	USHORT usPort	= 0;

	GetRemoteHost(&lpszHost, &usPort);
	if(usPort == default_port) usPort = 0;

	::MakeRequestLine(lpszMethod, lpszPath, m_enLocalVersion, strHeader);
	::MakeHeaderLines(lpHeaders, iHeaderCount, &m_objHttp.GetCookieMap(), iLength, TRUE, lpszHost, usPort, strHeader);
	::MakeHttpPacket(strHeader, pBody, iLength, szBuffer);

	return SendPackets(szBuffer, 2);
}

template<class R, class T, USHORT default_port> BOOL CHttpClientT<R, T, default_port>::SendLocalFile(LPCSTR lpszFileName, LPCSTR lpszMethod, LPCSTR lpszPath, const THeader lpHeaders[] = nullptr, int iHeaderCount = 0)
{
	if(
		strcmp(lpszMethod, HTTP_METHOD_POST)  != 0	&&
		strcmp(lpszMethod, HTTP_METHOD_PUT)	  != 0	&&
		strcmp(lpszMethod, HTTP_METHOD_PATCH) != 0 	)
	{
		::SetLastError(ERROR_INVALID_PARAMETER);
		return FALSE;
	}

	CAtlFile file;
	CAtlFileMapping<> fmap;

	HRESULT hr = ::ReadSmallFile(CA2T(lpszFileName), file, fmap);

	if(FAILED(hr))
	{
		::SetLastError(HRESULT_CODE(hr));
		return FALSE;
	}

	return SendRequest(lpszMethod, lpszPath, lpHeaders, iHeaderCount, (BYTE*)(char*)fmap, (int)fmap.GetMappingSize());
}

template<class R, class T, USHORT default_port> BOOL CHttpClientT<R, T, default_port>::SendWSMessage(BOOL bFinal, BYTE iReserved, BYTE iOperationCode, const BYTE lpszMask[4], BYTE* pData, int iLength, ULONGLONG ullBodyLen)
{
	WSABUF szBuffer[2];
	BYTE szHeader[HTTP_MAX_WS_HEADER_LEN];

	if(!::MakeWSPacket(bFinal, iReserved, iOperationCode, lpszMask, pData, iLength, ullBodyLen, szHeader, szBuffer))
		return FALSE;

	return SendPackets(szBuffer, 2);
}

// ------------------------------------------------------------------------------------------------------------- //

template<class T, USHORT default_port> BOOL CHttpSyncClientT<T, default_port>::Start(LPCTSTR lpszRemoteAddress, USHORT usPort, BOOL bAsyncConnect, LPCTSTR lpszBindAddress)
{
	CleanupRequestResult();

	if(!__super::Start(lpszRemoteAddress, usPort, TRUE, lpszBindAddress))
		return FALSE;

	DWORD dwWait = m_dwConnectTimeout != 0 ? m_dwConnectTimeout : INFINITE;
	BOOL isOK	 = ::WaitForSingleObject(m_evWait, dwWait) == WAIT_OBJECT_0;

	if(!isOK || m_enProgress != HSRP_DONE)
	{
		int ec = m_enProgress == HSRP_WAITING ? WSAETIMEDOUT : WSAECONNREFUSED;

		if(!isOK) Stop();

		SetLastError(SE_CONNECT_SERVER, __FUNCTION__, ec);
		return FALSE;
	}

	return TRUE;
}

template<class T, USHORT default_port> BOOL CHttpSyncClientT<T, default_port>::SendRequest(LPCSTR lpszMethod, LPCSTR lpszPath, const THeader lpHeaders[], int iHeaderCount, const BYTE* pBody, int iLength)
{
	CleanupRequestResult();

	if(!__super::SendRequest(lpszMethod, lpszPath, lpHeaders, iHeaderCount, pBody, iLength))
		return FALSE;

	DWORD dwWait = m_dwRequestTimeout != 0 ? m_dwRequestTimeout : INFINITE;
	BOOL isOK	 = ::WaitForSingleObject(m_evWait, dwWait) == WAIT_OBJECT_0;

	if(!isOK || m_enProgress != HSRP_DONE)
	{
		int ec = m_enProgress == HSRP_WAITING ? WSAETIMEDOUT :
				(m_enProgress == HSRP_CLOSE ? WSAECONNABORTED : ERROR_INVALID_DATA);

		if(!isOK) Stop();

		SetLastError(SE_DATA_SEND, __FUNCTION__, ec);
		return FALSE;
	}

	return TRUE;
}

template<class T, USHORT default_port> BOOL CHttpSyncClientT<T, default_port>::SendWSMessage(BOOL bFinal, BYTE iReserved, BYTE iOperationCode, const BYTE lpszMask[4], BYTE* pData, int iLength, ULONGLONG ullBodyLen)
{
	CleanupRequestResult();

	if(!__super::SendWSMessage(bFinal, iReserved, iOperationCode, lpszMask, pData, iLength, ullBodyLen))
		return FALSE;

	DWORD dwWait = m_dwRequestTimeout != 0 ? m_dwRequestTimeout : INFINITE;
	BOOL isOK	 = ::WaitForSingleObject(m_evWait, dwWait) == WAIT_OBJECT_0;

	if(!isOK || m_enProgress != HSRP_DONE)
	{
		int ec = m_enProgress == HSRP_WAITING ? WSAETIMEDOUT :
				(m_enProgress == HSRP_CLOSE ? WSAECONNABORTED : ERROR_INVALID_DATA);

		if(!isOK) Stop();

		SetLastError(SE_DATA_SEND, __FUNCTION__, ec);
		return FALSE;
	}

	return TRUE;
}

template<class T, USHORT default_port> BOOL CHttpSyncClientT<T, default_port>::OpenUrl(LPCSTR lpszMethod, LPCSTR lpszUrl, const THeader lpHeaders[], int iHeaderCount, const BYTE* pBody, int iLength, BOOL bForceReconnect)
{
	BOOL		bHttps;
	USHORT		usPort;
	CStringA	strHost;
	CStringA	strPath;

	if(!::ParseUrl(lpszUrl, bHttps, strHost, usPort, strPath))
	{
		SetLastError(SE_CONNECT_SERVER, __FUNCTION__, WSAEADDRNOTAVAIL);
		return FALSE;
	}

	if((bHttps && default_port == HTTP_DEFAULT_PORT) || (!bHttps && default_port == HTTPS_DEFAULT_PORT))
	{
		SetLastError(SE_CONNECT_SERVER, __FUNCTION__, WSAEPROTOTYPE);
		return FALSE;
	}

	if(HasStarted())
	{
		BOOL bNeedStop = bForceReconnect;

		if(!bNeedStop)
		{
			LPCSTR lpszHost	= nullptr;
			USHORT usPort2	= 0;

			GetRemoteHost(&lpszHost, &usPort2);

			if(strHost.CompareNoCase(lpszHost) != 0 || usPort != usPort2)
				bNeedStop = TRUE;

		}

		if(bNeedStop) Stop();
	}

	EnServiceState state = GetState();

	if(state != SS_STARTED)
	{
		if(state == SS_STARTING)
		{
			do 
			{
				::Sleep(50);
				state = GetState();
			} while(state != SS_STARTED && state != SS_STOPPED);
		}
		else
		{
			while(state != SS_STOPPED)
			{
				::Sleep(50);
				state = GetState();
			}

			Start(CA2T(strHost), usPort, FALSE, nullptr);
			state = GetState();
		}

		if(state == SS_STOPPED)
			return FALSE;
	}

	return SendRequest(lpszMethod, strPath, lpHeaders, iHeaderCount, pBody, iLength);
}

template<class T, USHORT default_port> BOOL CHttpSyncClientT<T, default_port>::CleanupRequestResult()
{
	m_szBuffer.Free();
	m_objHttp2.Reset();
	
	ResetRequestEvent();

	return TRUE;
}

template<class T, USHORT default_port> void CHttpSyncClientT<T, default_port>::ResetRequestEvent()
{
	m_enProgress = HSRP_WAITING;
	m_evWait.Reset();
}

template<class T, USHORT default_port> void CHttpSyncClientT<T, default_port>::SetRequestEvent(EnHttpSyncRequestProgress enProgress)
{
	if(m_enProgress == HSRP_WAITING)
	{
		m_enProgress = enProgress;
		m_evWait.Set();}
}

template<class T, USHORT default_port> BOOL CHttpSyncClientT<T, default_port>::GetResponseBody(LPCBYTE* lpszBody, int* iLength)
{
	ASSERT(lpszBody && iLength);

	*lpszBody = m_szBuffer.Ptr();
	*iLength  = (int)m_szBuffer.Size();

	return TRUE;
}

template<class T, USHORT default_port> EnHandleResult CHttpSyncClientT<T, default_port>::OnHandShake(ITcpClient* pSender, CONNID dwConnID)
{
	SetRequestEvent(HSRP_DONE);

	return HR_OK;
}

template<class T, USHORT default_port> EnHandleResult CHttpSyncClientT<T, default_port>::OnClose(ITcpClient* pSender, CONNID dwConnID, EnSocketOperation enOperation, int iErrorCode)
{
	SetRequestEvent(HSRP_CLOSE);

	return HR_OK;
}

template<class T, USHORT default_port> EnHttpParseResult CHttpSyncClientT<T, default_port>::OnHeadersComplete(IHttpClient* pSender, CONNID dwConnID)
{
	return HPR_OK;
}

template<class T, USHORT default_port> EnHttpParseResult CHttpSyncClientT<T, default_port>::OnBody(IHttpClient* pSender, CONNID dwConnID, const BYTE* pData, int iLength)
{
	m_szBuffer.Cat(pData, iLength);

	return HPR_OK;
}

template<class T, USHORT default_port> EnHttpParseResult CHttpSyncClientT<T, default_port>::OnMessageComplete(IHttpClient* pSender, CONNID dwConnID)
{
	m_objHttp2.CopyData(m_objHttp);

	if(GetUpgradeType() == HUT_NONE)
		SetRequestEvent(HSRP_DONE);

	return HPR_OK;
}

template<class T, USHORT default_port> EnHttpParseResult CHttpSyncClientT<T, default_port>::OnUpgrade(IHttpClient* pSender, CONNID dwConnID, EnHttpUpgradeType enUpgradeType)
{
	if(enUpgradeType == HUT_WEB_SOCKET)
	{
		SetRequestEvent(HSRP_DONE);
		return HPR_OK;
	}
	else
	{
		SetRequestEvent(HSRP_ERROR);
		return HPR_ERROR;
	}
}

template<class T, USHORT default_port> EnHttpParseResult CHttpSyncClientT<T, default_port>::OnParseError(IHttpClient* pSender, CONNID dwConnID, int iErrorCode, LPCSTR lpszErrorDesc)
{
	m_objHttp2.CopyData(m_objHttp);
	SetRequestEvent(HSRP_ERROR);

	return HPR_OK;
}

template<class T, USHORT default_port> EnHandleResult CHttpSyncClientT<T, default_port>::OnWSMessageBody(IHttpClient* pSender, CONNID dwConnID, const BYTE* pData, int iLength)
{
	m_szBuffer.Cat(pData, iLength);

	return HR_OK;
}

template<class T, USHORT default_port> EnHandleResult CHttpSyncClientT<T, default_port>::OnWSMessageComplete(IHttpClient* pSender, CONNID dwConnID)
{
	m_objHttp2.CopyWSContext(m_objHttp);
	SetRequestEvent(HSRP_DONE);

	return HR_OK;
}

// ------------------------------------------------------------------------------------------------------------- //

template class CHttpClientT<IHttpRequester, CTcpClient, HTTP_DEFAULT_PORT>;
template class CHttpSyncClientT<CTcpClient, HTTP_DEFAULT_PORT>;

#ifdef _SSL_SUPPORT

#include "SSLClient.h"

template class CHttpClientT<IHttpRequester, CSSLClient, HTTPS_DEFAULT_PORT>;
template class CHttpSyncClientT<CSSLClient, HTTPS_DEFAULT_PORT>;

#endif