749 lines
24 KiB
C++
749 lines
24 KiB
C++
|
|
|
|
///////////////////////////////////////////////////////////////////
|
|
// INCLUDE
|
|
///////////////////////////////////////////////////////////////////
|
|
|
|
#include "STDAFX.H"
|
|
#include "HTTPCLIENT.H"
|
|
#include "COMMONMACROS.H"
|
|
#include "BASESET.H"
|
|
#include "DEBUGUTIL.H"
|
|
#include "SAFESTRINGMACROS.H"
|
|
|
|
#include "../NetworkLib/Log.h"
|
|
|
|
#ifdef _DEBUG
|
|
#define new new(_NORMAL_BLOCK,__FILE__,__LINE__)
|
|
#endif
|
|
|
|
///////////////////////////////////////////////////////////////////
|
|
// IMPLEMENTATION
|
|
///////////////////////////////////////////////////////////////////
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::CHttpClient
|
|
//*---------------------------------------------------------------
|
|
// DESC : 생성자
|
|
// PARM : N/A
|
|
// RETV : N/A
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
CHttpClient::CHttpClient()
|
|
{
|
|
Initialize();
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::~CHttpClient
|
|
//*---------------------------------------------------------------
|
|
// DESC : 소멸자
|
|
// PARM : N/A
|
|
// RETV : N/A
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
CHttpClient::~CHttpClient()
|
|
{
|
|
Finalize();
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::Initialize
|
|
//*---------------------------------------------------------------
|
|
// DESC : 객체 관련 자원들을 초기화
|
|
// PARM : N/A
|
|
// RETV : N/A
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
VOID CHttpClient::Initialize()
|
|
{
|
|
m_hInternet = NULL;
|
|
m_hInternetConnection = NULL;
|
|
m_hHttpRequest = NULL;
|
|
m_eHttpConnectionMode = EV_HTTP_CONNECTION_NONE;
|
|
m_wHttpServerPortNumber = DF_DEFAULT_HTTP_PORTNUMBER;
|
|
::memset(m_szHttpServerAddress, 0, sizeof(m_szHttpServerAddress));
|
|
::memset(m_szUserName, 0, sizeof(m_szUserName));
|
|
::memset(m_szPassword, 0, sizeof(m_szPassword));
|
|
m_RefCount = 0;
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::Finalize
|
|
//*---------------------------------------------------------------
|
|
// DESC : 객체 관련 자원들을 해제
|
|
// PARM : N/A
|
|
// RETV : N/A
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
VOID CHttpClient::Finalize()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::Open
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 세션을 서버에 연결 / One-Req,One-Rsp 모드의 경우 URL자원을 바로 가져옴
|
|
// PARM : 1 . lpszHttpServerAddressUrl - One-Req,One-Rsp : HTTP 자원의 전체 URL / Keep-Alive : HTTP 서버의 주소 (도메인이름 (http:// 빼야함), 점10진방식)
|
|
// 2 . eHttpConnectionMode - EV_HTTP_CONNECTION_ONEREQONERSP : One-Req,One-Rsp 방식 (기본값) / EV_HTTP_CONNECTION_KEEPALIVE : Keep-Alive 방식
|
|
// 3 . wHttpServerPortNumber - (Keep-Alive 모드만 사용) HTTP 서버의 포트번호 (기본값 80)
|
|
// 4 . lpszUserName - (Keep-Alive 모드만 사용) HTTP 서버의 사용자 계정명 (기본값 NULL)
|
|
// 5 . lpszPassword - (Keep-Alive 모드만 사용) HTTP 서버의 사용자 계정에 따른 비밀번호(기본값 NULL)
|
|
// RETV : TRUE - 성공 / FALSE - 실패
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
BOOL CHttpClient::Open(
|
|
LPCTSTR lpszHttpServerAddressUrl,
|
|
EF_HTTP_CONNECTION_TYPE eHttpConnectionMode,
|
|
INTERNET_PORT wHttpServerPortNumber,
|
|
LPCTSTR lpszUserName,
|
|
LPCTSTR lpszPassword
|
|
)
|
|
{
|
|
// DN_ASSERT(!IsOpen(), "Invalid!");
|
|
|
|
if (lpszHttpServerAddressUrl == NULL || !_tcscmp(lpszHttpServerAddressUrl, _T(""))) {
|
|
// DN_TRACE(_T("lpszHttpServerAddressUrl == NULL || !_tcscmp(lpszHttpServerAddressUrl, _T(\"\")) - CHttpClient::Open()\n"));
|
|
return FALSE;
|
|
}
|
|
|
|
if (IsOpen()) {
|
|
Close();
|
|
}
|
|
|
|
switch(eHttpConnectionMode) {
|
|
case EV_HTTP_CONNECTION_ONEREQONERSP: // One-Req,One-Rsp 일 경우
|
|
{
|
|
m_hInternet = ::InternetOpen(_T("Microsoft Internet Explorer"), INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, NULL);
|
|
if (m_hInternet == NULL) {
|
|
// DN_TRACE(_T("::InternetOpen() == NULL : %d - CHttpClient::Open()\n"), GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
m_hInternetConnection = ::InternetOpenUrl(m_hInternet, lpszHttpServerAddressUrl, NULL, 0, INTERNET_FLAG_DONT_CACHE | INTERNET_FLAG_NO_CACHE_WRITE, 0);
|
|
if (m_hInternetConnection == NULL) {
|
|
::InternetCloseHandle(m_hInternet);
|
|
m_hInternet = NULL;
|
|
// DN_TRACE(_T("::InternetOpenUrl() == NULL : %d - CHttpClient::Open()\n"), GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
/*
|
|
DWORD dwNumberOfBytesAvailable = 0;
|
|
DWORD dwSize = sizeof(dwNumberOfBytesAvailable);
|
|
DWORD dwStatus = 0;
|
|
if (::HttpQueryInfo(m_hInternetConnection, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &dwNumberOfBytesAvailable, &dwSize, &dwStatus) ==FALSE
|
|
|| dwNumberOfBytesAvailable == 0)
|
|
{
|
|
// DN_TRACE(_T("dwNumberOfBytesAvailable : %u - CHttpClient::Open()\n"), dwNumberOfBytesAvailable);
|
|
return FALSE;
|
|
}
|
|
*/
|
|
|
|
m_eHttpConnectionMode = EV_HTTP_CONNECTION_ONEREQONERSP;
|
|
}
|
|
return TRUE;
|
|
case EV_HTTP_CONNECTION_KEEPALIVE: // Keep-Alive 일 경우
|
|
{
|
|
// STRNCPY(m_szHttpServerAddress, lpszHttpServerAddressUrl, COUNT_OF(m_szHttpServerAddress));
|
|
_tcsncpy_s(m_szHttpServerAddress, lpszHttpServerAddressUrl, _countof(m_szHttpServerAddress));
|
|
// _tcsncpy(m_szHttpServerAddress, lpszHttpServerAddressUrl, COUNT_OF(m_szHttpServerAddress));
|
|
m_wHttpServerPortNumber = wHttpServerPortNumber;
|
|
|
|
if (lpszUserName == NULL) {
|
|
m_szUserName[0] = _T('\0');
|
|
}
|
|
else {
|
|
// STRNCPY(m_szUserName, lpszUserName, COUNT_OF(m_szUserName));
|
|
_tcsncpy_s(m_szUserName, lpszUserName, _countof(m_szUserName));
|
|
// _tcsncpy(m_szUserName, lpszUserName, COUNT_OF(m_szUserName));
|
|
}
|
|
|
|
if (lpszPassword == NULL) {
|
|
m_szPassword[0] = _T('\0');
|
|
}
|
|
else {
|
|
// STRNCPY(m_szPassword, lpszPassword, COUNT_OF(m_szPassword));
|
|
_tcsncpy_s(m_szPassword, lpszPassword, _countof(m_szPassword));
|
|
// _tcsncpy(m_szPassword, lpszPassword, COUNT_OF(m_szPassword));
|
|
}
|
|
m_hInternet = ::InternetOpen(_T("Microsoft Internet Explorer"), INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, NULL);
|
|
|
|
if (m_hInternet == NULL) {
|
|
// DN_TRACE(_T("::InternetOpen() == NULL : %d - CHttpClient::Open()\n"), GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
m_hInternetConnection = ::InternetConnect(
|
|
m_hInternet,
|
|
m_szHttpServerAddress,
|
|
m_wHttpServerPortNumber
|
|
,m_szUserName, m_szPassword ,INTERNET_SERVICE_HTTP ,0 ,0);
|
|
if (m_hInternetConnection == NULL) {
|
|
// DN_TRACE(_T("::InternetConnect() == NULL : %d - CHttpClient::Open()\n"), GetLastError());
|
|
return FALSE;
|
|
}
|
|
|
|
m_eHttpConnectionMode = EV_HTTP_CONNECTION_KEEPALIVE;
|
|
}
|
|
return TRUE;
|
|
default: // 에러
|
|
{
|
|
m_eHttpConnectionMode = EV_HTTP_CONNECTION_NONE;
|
|
// DN_TRACE(_T("Unknown Http Connection Type : %d - CHttpClient::Open()\n"), eHttpConnectionMode);
|
|
}
|
|
break;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::IsOpen
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 세션이 연결되어 있는지 확인
|
|
// PARM : N/A
|
|
// RETV : TRUE - 성공 / FALSE - 실패
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
BOOL CHttpClient::IsOpen()
|
|
{
|
|
if ((m_hInternet == NULL) ||
|
|
((m_eHttpConnectionMode != EV_HTTP_CONNECTION_ONEREQONERSP) && (m_eHttpConnectionMode != EV_HTTP_CONNECTION_KEEPALIVE))
|
|
)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::Close
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 세션 연결 종료, 자원 해제
|
|
// PARM : N/A
|
|
// RETV : N/A
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
VOID CHttpClient::Close()
|
|
{
|
|
if (m_hHttpRequest) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
}
|
|
if (m_hInternetConnection) {
|
|
::InternetCloseHandle(m_hInternetConnection);
|
|
m_hInternetConnection = NULL;
|
|
}
|
|
if (m_hInternet) {
|
|
::InternetCloseHandle(m_hInternet);
|
|
m_hInternet = NULL;
|
|
}
|
|
m_eHttpConnectionMode = EV_HTTP_CONNECTION_NONE;
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::Close
|
|
//*---------------------------------------------------------------
|
|
// DESC : (Keep-Alive 모드만 사용) HTTP 세션 연결을 재수립
|
|
// PARM : N/A
|
|
// RETV : TRUE - 성공 / FALSE - 실패
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
BOOL CHttpClient::ReOpen()
|
|
{
|
|
// DN_ASSERT(IsOpen(), "Invalid!");
|
|
// DN_ASSERT(EV_HTTP_CONNECTION_KEEPALIVE == m_eHttpConnectionMode, "Invalid!");
|
|
// DN_ASSERT(_tcscmp(m_szHttpServerAddress, _T("")), "Invalid!");
|
|
|
|
if (!IsOpen() ||
|
|
EV_HTTP_CONNECTION_KEEPALIVE != m_eHttpConnectionMode ||
|
|
!_tcscmp(m_szHttpServerAddress, _T(""))
|
|
)
|
|
{
|
|
g_Log.Log(LogType::_ERROR, L"IsOpen() : %d / m_eHttpConnectionMode : %d / m_szHttpServerAddress : %s - CHttpClient::Open()\n", IsOpen(), m_eHttpConnectionMode, m_szHttpServerAddress);
|
|
// DN_TRACE(_T("IsOpen() : %d / m_eHttpConnectionMode : %d / m_szHttpServerAddress : %s - CHttpClient::Open()\n"), IsOpen(), m_eHttpConnectionMode, m_szHttpServerAddress);
|
|
// DN_RETURN(FALSE);
|
|
return FALSE;
|
|
}
|
|
|
|
Close();
|
|
|
|
return Open(m_szHttpServerAddress, EV_HTTP_CONNECTION_KEEPALIVE, m_wHttpServerPortNumber, m_szUserName, m_szPassword);
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::SendRequest
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 서버에 연결 후 서버의 상대적 위치의 자원을 요청 (Keep-Alive 모드만 사용)
|
|
// PARM : 1 . lpszUrl - HTTP서버로 부터 상대적인 자원위치를 나타내는 URL
|
|
// 2 . bCheckResultDataLength - 요청 성공 시 자원의 길이를 받을 것인지 여부
|
|
// 3 . bAsyncRequest -
|
|
// RETV : 0 미만 : 실패 / 0 이상 : 성공, bCheckResultDataLength == TRUE 이고 0 이상의 값이면 받을 자원의 크기
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
INT CHttpClient::SendRequest(LPCTSTR lpszUrl, BOOL bCheckResultDataLength, BOOL bAsyncRequest)
|
|
{
|
|
// DN_ASSERT(IsOpen(), "Invalid!");
|
|
// DN_ASSERT(NULL != lpszUrl, "Invalid!");
|
|
|
|
DWORD dwNumberOfBytesAvailable = 0;
|
|
|
|
if (lpszUrl == NULL || !_tcscmp(lpszUrl, _T(""))) {
|
|
// DN_TRACE(_T("lpszUrl == NULL || !_tcscmp(lpszUrl, _T(\"\")) - CHttpClient::SendRequest()"));
|
|
return -1;
|
|
}
|
|
|
|
if (IsOpen() == FALSE && ReOpen() == FALSE) {
|
|
// DN_TRACE(_T("IsOpen() == FALSE && ReOpen() == FALSE"));
|
|
return -1;
|
|
}
|
|
|
|
if (m_hHttpRequest) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
}
|
|
m_hHttpRequest = ::HttpOpenRequest(m_hInternetConnection, _T("GET"), lpszUrl, _T("HTTP/1.1"), NULL, NULL,
|
|
INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_KEEP_CONNECTION
|
|
| INTERNET_FLAG_DONT_CACHE
|
|
| INTERNET_FLAG_NO_AUTH
|
|
| INTERNET_FLAG_PRAGMA_NOCACHE
|
|
| INTERNET_FLAG_NO_COOKIES,
|
|
0
|
|
);
|
|
if (m_hHttpRequest == NULL) {
|
|
// DN_TRACE(_T("::HttpOpenRequest() == NULL : %d - CHttpClient::SendRequest()"), GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
BOOL bRetVal;
|
|
if (!bAsyncRequest) {
|
|
bRetVal = ::HttpSendRequest(m_hHttpRequest, NULL, 0, NULL, 0);
|
|
}
|
|
else {
|
|
bRetVal = ::HttpSendRequestEx(m_hHttpRequest, NULL, NULL, HSR_ASYNC, 0);
|
|
}
|
|
if (bRetVal == FALSE) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
// DN_TRACE(_T("::HttpSendRequest() == FALSE : %d, m_eHttpConnectionMode : %d - CHttpClient::RecvResponse()\n"), GetLastError(), m_eHttpConnectionMode);
|
|
return -1;
|
|
}
|
|
|
|
if (bCheckResultDataLength) {
|
|
DWORD dwSize = sizeof(dwNumberOfBytesAvailable);
|
|
DWORD dwStatus = 0;
|
|
if (::HttpQueryInfo(m_hHttpRequest, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, &dwNumberOfBytesAvailable, &dwSize, &dwStatus) == FALSE
|
|
|| dwNumberOfBytesAvailable == 0)
|
|
{
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return (INT)dwNumberOfBytesAvailable;
|
|
}
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::SendRequest
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 서버에 연결 후 서버의 상대적 위치의 자원을 요청 (Keep-Alive 모드만 사용)
|
|
// PARM : 1 . lpszUrl - HTTP서버로 부터 상대적인 자원위치를 나타내는 URL
|
|
// 2 . szPostData - PostData
|
|
// RETV : 0 미만 : 실패 / 0 이상 : 성공, bCheckResultDataLength == TRUE 이고 0 이상의 값이면 받을 자원의 크기
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
INT CHttpClient::SendRequsetPostA(LPCTSTR lpszUrl, LPCSTR szPostData)
|
|
{
|
|
if (lpszUrl == NULL || !_tcscmp(lpszUrl, _T(""))) {
|
|
return -1;
|
|
}
|
|
|
|
if (IsOpen() == FALSE && ReOpen() == FALSE) {
|
|
return -1;
|
|
}
|
|
|
|
if (m_hHttpRequest) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
}
|
|
m_hHttpRequest = ::HttpOpenRequest(m_hInternetConnection, _T("POST"), lpszUrl, _T("HTTP/1.1"), NULL, NULL,
|
|
INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_KEEP_CONNECTION
|
|
| INTERNET_FLAG_DONT_CACHE
|
|
| INTERNET_FLAG_NO_AUTH
|
|
| INTERNET_FLAG_PRAGMA_NOCACHE
|
|
| INTERNET_FLAG_NO_COOKIES,
|
|
0
|
|
);
|
|
if (m_hHttpRequest == NULL) {
|
|
return -1;
|
|
}
|
|
int nPostLength = 0, nHeaderLength = 0;
|
|
char* strHeader = NULL;
|
|
if( szPostData )
|
|
{
|
|
strHeader = "Content-Type: application/x-www-form-urlencoded";
|
|
nPostLength = (int)strlen(szPostData);
|
|
nHeaderLength = (int)strlen(strHeader);
|
|
}
|
|
|
|
BOOL bRetVal = ::HttpSendRequestA(m_hHttpRequest, strHeader, nHeaderLength, (LPVOID)szPostData, nPostLength);
|
|
if (bRetVal == FALSE) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
// DN_TRACE(_T("::HttpSendRequest() == FALSE : %d, m_eHttpConnectionMode : %d - CHttpClient::RecvResponse()\n"), GetLastError(), m_eHttpConnectionMode);
|
|
return -1;
|
|
}
|
|
INT nStatus =0;
|
|
DWORD dwBuffSize = sizeof (INT) ;
|
|
|
|
if ( ::HttpQueryInfo (
|
|
m_hHttpRequest
|
|
, HTTP_QUERY_STATUS_CODE // Get the HTTP status code
|
|
| HTTP_QUERY_FLAG_NUMBER
|
|
, static_cast<void *> (&nStatus)
|
|
, &dwBuffSize // Buffer size (byte)
|
|
, NULL // Does not use a header index
|
|
) == FALSE )
|
|
{
|
|
return -1;
|
|
}
|
|
return nStatus ;
|
|
}
|
|
|
|
INT CHttpClient::SendRequestGet(LPCTSTR lpszUrl)
|
|
{
|
|
DWORD dwNumberOfBytesAvailable = 0;
|
|
|
|
if (lpszUrl == NULL || !_tcscmp(lpszUrl, _T(""))) {
|
|
// DN_TRACE(_T("lpszUrl == NULL || !_tcscmp(lpszUrl, _T(\"\")) - CHttpClient::SendRequest()"));
|
|
return -1;
|
|
}
|
|
|
|
if (IsOpen() == FALSE && ReOpen() == FALSE) {
|
|
// DN_TRACE(_T("IsOpen() == FALSE && ReOpen() == FALSE"));
|
|
return -1;
|
|
}
|
|
|
|
if (m_hHttpRequest) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
}
|
|
m_hHttpRequest = ::HttpOpenRequest(m_hInternetConnection, _T("GET"), lpszUrl, _T("HTTP/1.1"), NULL, NULL,
|
|
INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_KEEP_CONNECTION
|
|
| INTERNET_FLAG_DONT_CACHE
|
|
//| INTERNET_FLAG_NO_AUTH
|
|
| INTERNET_FLAG_PRAGMA_NOCACHE
|
|
| INTERNET_FLAG_NO_COOKIES,
|
|
0
|
|
);
|
|
if (m_hHttpRequest == NULL) {
|
|
// DN_TRACE(_T("::HttpOpenRequest() == NULL : %d - CHttpClient::SendRequest()"), GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
BOOL bRetVal = ::HttpSendRequest(m_hHttpRequest, NULL, 0, NULL, 0);
|
|
if (bRetVal == FALSE) {
|
|
::InternetCloseHandle(m_hHttpRequest);
|
|
m_hHttpRequest = NULL;
|
|
// DN_TRACE(_T("::HttpSendRequest() == FALSE : %d, m_eHttpConnectionMode : %d - CHttpClient::RecvResponse()\n"), GetLastError(), m_eHttpConnectionMode);
|
|
return -1;
|
|
}
|
|
INT nStatus =0;
|
|
DWORD dwBuffSize = sizeof (INT) ;
|
|
|
|
if ( ::HttpQueryInfo (
|
|
m_hHttpRequest
|
|
, HTTP_QUERY_STATUS_CODE // Get the HTTP status code
|
|
| HTTP_QUERY_FLAG_NUMBER
|
|
, static_cast<void *> (&nStatus)
|
|
, &dwBuffSize // Buffer size (byte)
|
|
, NULL // Does not use a header index
|
|
) == FALSE )
|
|
{
|
|
return -1;
|
|
}
|
|
return nStatus;
|
|
}
|
|
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::RecvResponse
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 서버로의 자원요청 성공 후 데이터를 받음 (One-Req,One-Rsp : Open() 이후 / Keep-Alive : SendRequest() 이후)
|
|
// PARM : 1 . lpbtBuffer - 데이터를 받을 버퍼의 포인터
|
|
// 2 . uBufferSize - 데이터를 받을 버퍼의 크기
|
|
// RETV : 0 미만 : 실패/ 0 : 성공, 수신완료 / 0 초과 : 성공, 수신할 데이터 남아있음
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
INT CHttpClient::RecvResponse(LPBYTE lpbtBuffer, INT nBufferSize)
|
|
{
|
|
// DN_ASSERT(IsOpen(), "Invalid!");
|
|
// DN_ASSERT(NULL != lpbtBuffer, "Invalid!");
|
|
// DN_ASSERT(0 < uBufferSize, "Invalid!");
|
|
|
|
if (lpbtBuffer == NULL || nBufferSize == 0) {
|
|
// DN_TRACE(_T("lpbtBuffer == NULL || uBufferSize == 0 - CHttpClient::RecvResponse()\n"));
|
|
return -1;
|
|
}
|
|
|
|
if (!IsOpen()) {
|
|
// DN_TRACE(_T("!IsOpen() - CHttpClient::RecvResponse()\n"));
|
|
return -1;
|
|
}
|
|
|
|
HINTERNET hInternetFile = NULL;
|
|
|
|
switch(m_eHttpConnectionMode) {
|
|
case EV_HTTP_CONNECTION_ONEREQONERSP: // One-Req,One-Rsp 일 경우
|
|
{
|
|
hInternetFile = m_hInternetConnection;
|
|
}
|
|
break;
|
|
case EV_HTTP_CONNECTION_KEEPALIVE: // Keep-Alive 일 경우
|
|
{
|
|
hInternetFile = m_hHttpRequest;
|
|
}
|
|
break;
|
|
default: // 에러
|
|
{
|
|
// DN_TRACE(_T("Unknown Http Connection Type : %d - CHttpClient::RecvResponse()\n"), m_eHttpConnectionMode);
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
if (hInternetFile == NULL) {
|
|
// DN_TRACE(_T("hInternetFile == NULL - CHttpClient::RecvResponse()\n"));
|
|
return -1;
|
|
}
|
|
|
|
DWORD dwNumberOfBytesAvailable, dwIoBytes, dwBufferOffset;
|
|
dwNumberOfBytesAvailable = dwIoBytes = dwBufferOffset = 0;
|
|
|
|
BOOL bRetVal = ::InternetQueryDataAvailable(hInternetFile, &dwNumberOfBytesAvailable, 0, 0);
|
|
if (bRetVal == FALSE) {
|
|
g_Log.Log(LogType::_ERROR, L"::InternetQueryDataAvailable() : %d - CHttpClient::RecvResponse()\n", GetLastError());
|
|
// DN_TRACE(_T("::InternetQueryDataAvailable() : %d - CHttpClient::RecvResponse()\n"), GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
int nCount = 0;
|
|
while(dwNumberOfBytesAvailable > 0) {
|
|
if ( nBufferSize - dwBufferOffset < dwNumberOfBytesAvailable )
|
|
{
|
|
if( nBufferSize == dwBufferOffset )
|
|
{
|
|
// 버퍼 완전 부족..
|
|
g_Log.Log(LogType::_ERROR, L"Buffer Empty!! nBufferSize:%d, dwBufferOffset:%u, dwNumberOfBytesAvailable:%u \n", nBufferSize, dwBufferOffset, dwNumberOfBytesAvailable);
|
|
return 1;
|
|
}
|
|
// 버퍼가 부족하면 부족한 버퍼에 맞게 받을 크기를 다시 계산
|
|
dwNumberOfBytesAvailable = nBufferSize - dwBufferOffset;
|
|
g_Log.Log(LogType::_ERROR, L"(nBufferSize:%d - dwBufferOffset:%u) < dwNumberOfBytesAvailable:%u - CHttpClient::RecvResponse()\n", nBufferSize, dwBufferOffset, dwNumberOfBytesAvailable);
|
|
// DN_TRACE(_T("((INT)uBufferSize - (INT)dwBufferOffset) < (INT)dwNumberOfBytesAvailable - CHttpClient::RecvResponse()\n"));
|
|
}
|
|
|
|
if ((INT)dwNumberOfBytesAvailable <= 0) {
|
|
// 버퍼부족 - 에러가 아니고 버퍼를 더 확보하라는 의미
|
|
// DN_TRACE(_T("(INT)dwNumberOfBytesAvailable <= 0 - CHttpClient::RecvResponse()\n"));
|
|
return 1;
|
|
}
|
|
|
|
bRetVal = ::InternetReadFile(hInternetFile, lpbtBuffer + dwBufferOffset, dwNumberOfBytesAvailable, &dwIoBytes);
|
|
if (bRetVal == FALSE) {
|
|
// DN_TRACE(_T("::InternetReadFile() == FALSE : %d - CHttpClient::RecvResponse()\n"), GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
dwBufferOffset += dwIoBytes;
|
|
|
|
bRetVal = ::InternetQueryDataAvailable(hInternetFile, &dwNumberOfBytesAvailable, 0, 0);
|
|
if (bRetVal == FALSE) {
|
|
// DN_TRACE(_T("::InternetQueryDataAvailable() == FALSE : %d - CHttpClient::RecvResponse()\n"), GetLastError());
|
|
return -1;
|
|
}
|
|
++nCount;
|
|
if( nCount > 20 )
|
|
{
|
|
// 무한 루프 추적용..
|
|
g_Log.Log(LogType::_ERROR, L"RecvResponse Loof nBufferSize:%d, dwBufferOffset:%u, dwNumberOfBytesAvailable:%u\n", nBufferSize, dwBufferOffset, dwNumberOfBytesAvailable);
|
|
Close();
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
if (dwNumberOfBytesAvailable == 0) {
|
|
// 모든 결과 데이터 수신 성공
|
|
return 0;
|
|
}
|
|
|
|
// 알 수 없는 에러 - 이런 결과가 생기는지 확인해 보아야 함
|
|
return -1;
|
|
}
|
|
|
|
INT CHttpClient::RecvDownFile(LPCTSTR lpszFileName, LPBINDSTATUSCALLBACK pCallBack)
|
|
{
|
|
FILE* downfile = NULL;
|
|
downfile = _tfopen(lpszFileName, _T("w+b"));
|
|
|
|
if (!IsOpen()) {
|
|
fclose(downfile);
|
|
return -1;
|
|
}
|
|
|
|
HINTERNET hInternetFile = NULL;
|
|
|
|
switch(m_eHttpConnectionMode) {
|
|
case EV_HTTP_CONNECTION_ONEREQONERSP: // One-Req,One-Rsp 일 경우
|
|
{
|
|
hInternetFile = m_hInternetConnection;
|
|
}
|
|
break;
|
|
case EV_HTTP_CONNECTION_KEEPALIVE: // Keep-Alive 일 경우
|
|
{
|
|
hInternetFile = m_hHttpRequest;
|
|
}
|
|
break;
|
|
default: // 에러
|
|
{
|
|
// DN_TRACE(_T("Unknown Http Connection Type : %d - CHttpClient::RecvResponse()\n"), m_eHttpConnectionMode);
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
if (hInternetFile == NULL) {
|
|
fclose(downfile);
|
|
return -1;
|
|
}
|
|
|
|
DWORD dwNumberOfBytesAvailable, dwTotalBytes, dwIoBytes, dwCurTotalBytes;
|
|
dwNumberOfBytesAvailable = dwTotalBytes = dwIoBytes = dwCurTotalBytes = 0;
|
|
|
|
DWORD dwBufLen = sizeof(dwTotalBytes);
|
|
|
|
BOOL bRetVal = ::HttpQueryInfo(hInternetFile, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, (LPVOID)&dwTotalBytes, &dwBufLen, 0);
|
|
// BOOL bRetVal = ::InternetQueryDataAvailable(hInternetFile, &dwNumberOfBytesAvailable, 0, 0);
|
|
if (bRetVal == FALSE) {
|
|
fclose(downfile);
|
|
return -1;
|
|
}
|
|
char buf[1024*10]={0,};
|
|
|
|
dwNumberOfBytesAvailable = dwTotalBytes;
|
|
|
|
while(dwNumberOfBytesAvailable > 0)
|
|
{
|
|
bRetVal = ::InternetReadFile(hInternetFile, buf, sizeof(buf), &dwIoBytes);
|
|
if (bRetVal == FALSE) {
|
|
fclose(downfile);
|
|
return -1;
|
|
}
|
|
|
|
fwrite(buf, 1, dwIoBytes, downfile);
|
|
dwCurTotalBytes += dwIoBytes;
|
|
if( pCallBack )
|
|
pCallBack->OnProgress( dwCurTotalBytes, dwTotalBytes, BINDSTATUS_DOWNLOADINGDATA, NULL);
|
|
|
|
bRetVal = ::InternetQueryDataAvailable(hInternetFile, &dwNumberOfBytesAvailable, 0, 0);
|
|
if (bRetVal == FALSE) {
|
|
fclose(downfile);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (dwNumberOfBytesAvailable == 0) {
|
|
// 모든 결과 데이터 수신 성공
|
|
if( pCallBack )
|
|
pCallBack->OnProgress( dwTotalBytes, dwTotalBytes, BINDSTATUS_ENDDOWNLOADDATA, NULL);
|
|
fclose(downfile);
|
|
return 0;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
//*---------------------------------------------------------------
|
|
// TYPE : FUNCTION
|
|
// NAME : CHttpClient::SetOption
|
|
//*---------------------------------------------------------------
|
|
// DESC : HTTP 세션에 필요한 인터넷 옵션들을 지정 (wininet.lib 을 사용하는 경우만 가능하고 winhttp.lib 을 사용하는 경우는 지정불가)
|
|
// PARM : 1 . dwMaxConnCount - HTTP 요청의 최대개수 지정
|
|
// 2 . dwTimout - HTTP 요청의 타임아웃
|
|
// RETV : 0 미만 : 실패/ 0 : 성공, 수신완료 / 0 초과 : 성공, 수신할 데이터 남아있음
|
|
// PRGM : B4nFter
|
|
//*---------------------------------------------------------------
|
|
DWORD CHttpClient::SetOption(DWORD dwMaxConnCount, DWORD dwTimout)
|
|
{
|
|
// DN_ASSERT(0 < dwMaxConnCount, "Invalid!");
|
|
// DN_ASSERT((1*1000) < dwTimout, "1초이내로 송신이 완료되야 하는 것이 맞는가?");
|
|
|
|
// HTTP 규약에 의해서 HTTP서버에 동시에 요청할 수 있는 개수의 제약이 있음 (1.0:4개, 1.1:2개)
|
|
// 이것을 초과해서 요청 시 다음 것은 block 됨
|
|
// P.S.> 다음 기사 참고 → INFO: WinInet Limits Connections Per Server (Q183110)
|
|
DWORD dwArgVal = dwMaxConnCount;
|
|
if (FALSE == ::InternetSetOption(NULL, INTERNET_OPTION_MAX_CONNS_PER_1_0_SERVER, &dwArgVal, sizeof(dwArgVal)))
|
|
{
|
|
// DN_RETURN(::GetLastError());
|
|
return(::GetLastError());
|
|
}
|
|
|
|
dwArgVal = dwMaxConnCount;
|
|
if (FALSE == ::InternetSetOption(NULL, INTERNET_OPTION_MAX_CONNS_PER_SERVER, &dwArgVal, sizeof(dwArgVal)))
|
|
{
|
|
// DN_RETURN(::GetLastError());
|
|
return(::GetLastError());
|
|
}
|
|
|
|
dwArgVal = dwTimout;
|
|
if (FALSE == ::InternetSetOption(NULL, INTERNET_OPTION_SEND_TIMEOUT, &dwArgVal, sizeof(dwArgVal)))
|
|
{
|
|
// DN_RETURN(::GetLastError());
|
|
return(::GetLastError());
|
|
}
|
|
|
|
dwArgVal = dwTimout;
|
|
if (FALSE == ::InternetSetOption(NULL, INTERNET_OPTION_RECEIVE_TIMEOUT , &dwArgVal, sizeof(dwArgVal)))
|
|
{
|
|
// DN_RETURN(::GetLastError());
|
|
return(::GetLastError());
|
|
}
|
|
|
|
// Windows Server 2003 x64 (정확히는 Requires Internet Explorer 7.0.) 부터는 IDN을 사용하는데, MBCS이 깨짐
|
|
// 그래서 IDN을 off시키면 system code page를 사용해서 보내야 깨지지 않음
|
|
// Windows XP SP2 and Windows Server 2003 SP1 에서는 지원하지 않음
|
|
/*
|
|
dwArgVal = 0;
|
|
if (FALSE == ::InternetSetOption(NULL, INTERNET_OPTION_IDN, &dwArgVal, sizeof(dwArgVal)))
|
|
{
|
|
dwArgVal = ::GetLastError();
|
|
if (ERROR_CALL_NOT_IMPLEMENTED != dwArgVal)
|
|
{
|
|
// DN_RETURN(dwArgVal);
|
|
return dwArgVal;
|
|
}
|
|
}
|
|
*/
|
|
|
|
return(NOERROR);
|
|
}
|
|
|