www.pudn.com > WZ_UpdateModule_DLL.rar > HttpUpdate.cpp


 
#include "WZ_UM.h" 
 
 
WZ_DECLAREDLL CWZHttpUpdateModule::CWZHttpUpdateModule () 
{ 
	m_enUpdateType = USING_HTTP; 
} 
 
 
WZ_DECLAREDLL CWZHttpUpdateModule::CWZHttpUpdateModule (LPSTR lpszIPADDR, WORD wPORT, LPSTR lpszID, LPSTR lpszPASS) 
	: CWZUpdateModule (lpszIPADDR, wPORT, lpszID, lpszPASS) 
{ 
	m_enUpdateType = USING_HTTP; 
} 
 
 
WZ_DECLAREDLL CWZHttpUpdateModule::~CWZHttpUpdateModule () 
{ 
 
} 
 
 
WZ_DECLAREDLL BOOL CWZHttpUpdateModule::Connect () 
{ 
	m_iCurrentState = STATE_CONNECTING; 
 
	// error check 
	if (! strcmp (m_szIPADDR, "")) { 
		m_iCurrentState = STATE_DISCONNECTED; 
		return FALSE; 
	} 
	if (! m_wPORT) 
		m_wPORT = PORT_HTTP; 
 
 
	m_hINET_SESSION = InternetOpen ("UPDATE_HTTP", INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0); 
	if (! m_hINET_SESSION) { 
		// "인터넷 연결 실패" 
		m_iCurrentState = STATE_DISCONNECTED; 
		SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
		return FALSE; 
	} 
	 
	m_hHTTP_CONNECTION =InternetConnect(m_hINET_SESSION, m_szIPADDR, m_wPORT, m_szID, m_szPASS, INTERNET_SERVICE_HTTP,0,0); 
	if (m_hHTTP_CONNECTION==NULL) { 
		// "HTTP 에 접속할 수 없습니다." 
		InternetCloseHandle(m_hINET_SESSION); 
		m_iCurrentState = STATE_DISCONNECTED; 
		SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
		return FALSE; 
	} 
 
	m_iCurrentState = STATE_CONNECTED; 
	return TRUE; 
} 
 
 
WZ_DECLAREDLL BOOL CWZHttpUpdateModule::Connect (LPSTR lpszIPADDR, WORD wPORT, LPSTR lpszID, LPSTR lpszPASS) 
{ 
	strcpy (m_szIPADDR, lpszIPADDR); 
	m_wPORT = wPORT; 
 
	if (!lpszID) 
		strcpy (m_szID, ""); 
	else 
		strcpy (m_szID, lpszID); 
 
	if (!lpszPASS) 
		strcpy (m_szPASS, ""); 
	else 
		strcpy (m_szPASS, lpszPASS); 
	 
	return this->Connect(); 
} 
 
 
WZ_DECLAREDLL VOID CWZHttpUpdateModule::Disconnect () 
{ 
	m_iCurrentState = STATE_DISCONNECTED; 
 
	if( m_hHTTP_CONNECTION ) 
	{		 
		InternetCloseHandle(m_hHTTP_CONNECTION); 
		m_hHTTP_CONNECTION = NULL; 
	} 
	if( m_hINET_SESSION ) 
	{ 
		InternetCloseHandle(m_hINET_SESSION); 
		m_hINET_SESSION = NULL; 
	} 
} 
 
 
WZ_DECLAREDLL BOOL CWZHttpUpdateModule::DownloadFile (LPSTR lpszRemoteFilePath, LPSTR lpszLocalFilePath) 
{ 
	CHAR	lpszRemoteFileName [MAX_PATH]; 
	CHAR	lpszRemoteFileEXE [MAX_PATH]; 
	CHAR	lpszRealLocalFilePath [MAX_PATH]; 
 
	HINTERNET hHTTP_FILE_REQ = NULL; 
	HANDLE	hLocalFile = INVALID_HANDLE_VALUE; 
 
 
	// 여기서 로컬 파일이름이 없다면 원격파일이름과 동일하게 만든다. 
	_splitpath(lpszRemoteFilePath, NULL, NULL, lpszRemoteFileName, lpszRemoteFileEXE ); 
	if ((!lpszLocalFilePath) || (! strcmp(lpszLocalFilePath, ""))) 
		wsprintf (lpszRealLocalFilePath, ".\\%s%s", lpszRemoteFileName, lpszRemoteFileEXE); 
	else { 
		if (lpszLocalFilePath[strlen(lpszLocalFilePath)-1] != '\\') 
			wsprintf (lpszRealLocalFilePath, "%s\\%s%s", lpszLocalFilePath, lpszRemoteFileName, lpszRemoteFileEXE); 
		else 
			wsprintf (lpszRealLocalFilePath, "%s%s%s", lpszLocalFilePath, lpszRemoteFileName, lpszRemoteFileEXE); 
	} 
	 
	wsprintf(m_strtUpdatingFileInfo.szFileName, "%s%s", lpszRemoteFileName, lpszRemoteFileEXE);	// 파일 상태 구조체에 현재 받을 파일이름 저장 
 
	__try { 
		// 1 . 로컬 파일 만들기 
		hLocalFile = CreateFile (lpszRealLocalFilePath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0); 
		if (hLocalFile == INVALID_HANDLE_VALUE) { 
			SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
			return FALSE; 
		} 
 
		// 2 . 원격지 파일 열기 
		INT iCount = MAX_RETRY_DOWNLOADING; 
		while(iCount--) 
		{ 
			if (!hHTTP_FILE_REQ) 
				hHTTP_FILE_REQ = HttpOpenRequest(m_hHTTP_CONNECTION, "GET", lpszRemoteFilePath, NULL, NULL, NULL, 0, 0); 
 
			if (! HttpSendRequest (hHTTP_FILE_REQ, NULL, 0, NULL, 0)) { 
				Sleep(100); 
			} 
			else break; 
		} 
		if( iCount  < 1 ) 
		{ 
			// 파일을 찾을수 없다. 
			SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, m_strtUpdatingFileInfo.iErrorCode); 
			DeleteFile (lpszRealLocalFilePath); 
			return FALSE; 
		} 
 
		// 3 . 파일 크기 얻기 
		DWORD dwFileLength = 0; 
		CHAR szBuffer [16]; 
		DWORD dwBufferSize = sizeof (szBuffer), dwIndex = 0;	// dwIndex 를 0 으로 (중요 !!) 
		BOOL bResult = HttpQueryInfo(hHTTP_FILE_REQ, HTTP_QUERY_CONTENT_LENGTH, szBuffer, &dwBufferSize, &dwIndex); 
		if (!bResult) { 
			SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
			return FALSE; 
		} 
		dwFileLength = atoi (szBuffer); 
 
 
		 
		DWORD dwNowRecvBytes, dwNowWrittenBytes, dwTotalWrittenBytes, dwAvailableSize; 
		dwNowRecvBytes = dwNowWrittenBytes = dwTotalWrittenBytes = dwAvailableSize = 0; 
 
 
		if(dwFileLength >= 0) 
		{ 
			m_strtUpdatingFileInfo.uWholeFileSize = dwFileLength; 
			m_strtUpdatingFileInfo.uNowFileSize = 0; 
			AddEvent( EVENT_FILEDOWNLOAD_START ); 
 
			while(TRUE) 
			{ 
				if (! InternetQueryDataAvailable(hHTTP_FILE_REQ, & dwAvailableSize, 0, 0) ) { 
					SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
					return FALSE; 
				} 
 
				if (dwAvailableSize > sizeof (m_HTTP_cBUFFER))  
					dwAvailableSize = sizeof (m_HTTP_cBUFFER); 
 
				if (! InternetReadFile(hHTTP_FILE_REQ, m_HTTP_cBUFFER, dwAvailableSize, &dwNowRecvBytes) ) { 
					SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
					return FALSE; 
				} 
				 
				if (! WriteFile (hLocalFile, m_HTTP_cBUFFER, dwNowRecvBytes, &dwNowWrittenBytes, NULL)) { 
					SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
					return FALSE; 
				} 
 
				dwTotalWrittenBytes += dwNowWrittenBytes; 
				m_strtUpdatingFileInfo.uNowFileSize = dwTotalWrittenBytes; 
				 
				AddEvent( EVENT_FILEDOWNLOAD_WORKING ); 
 
				if (!dwNowRecvBytes) 
					break; 
 
				if(dwTotalWrittenBytes >= dwFileLength) 
					break; 
				 
				// 업데이트 작업이 중간에 취소되었다면 나간다. 
				if( m_bCancelUpdate ) 
				{ 
					DeleteFile (lpszRealLocalFilePath); 
					m_bCancelUpdate = FALSE; 
					return FALSE; 
				} 
			} 
		} 
		else 
		{ 
			// 원격지 파일 크기가 0보다 작다. (에러가 아닐 수도 있음) 
			SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError()); 
		} 
 
		AddEvent( EVENT_FILEDOWNLOAD_END ); 
		 
		//m_strtUpdatingFileInfo.uNowFileNum++;		// 이건 다운로드 파일이 성공하면 밖에서 해준다. 
	} 
	__finally { 
		if (hHTTP_FILE_REQ) 
			InternetCloseHandle(hHTTP_FILE_REQ); 
		if (hLocalFile != INVALID_HANDLE_VALUE) 
			CloseHandle(hLocalFile); 
	} 
	 
	return TRUE; 
}