www.pudn.com > WZ_UpdateModule_DLL.rar > FtpUpdate.cpp
#include "WZ_UM.h"
#include "ReadScript.h"
WZ_DECLAREDLL CWZFtpUpdateModule::CWZFtpUpdateModule ()
{
m_hUpdateThread = NULL;
m_hCheckThread = NULL;
m_nCheckThreadID = 0;
m_bConnected = FALSE;
}
WZ_DECLAREDLL CWZFtpUpdateModule::CWZFtpUpdateModule (LPSTR lpszIPADDR, WORD wPORT, LPSTR lpszID, LPSTR lpszPASS)
: CWZUpdateModule (lpszIPADDR, wPORT, lpszID, lpszPASS)
{
m_hUpdateThread = NULL;
m_hCheckThread = NULL;
m_nCheckThreadID = 0;
m_bConnected = FALSE;
}
WZ_DECLAREDLL CWZFtpUpdateModule::~CWZFtpUpdateModule ()
{
}
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::Connect ()
{
AddEvent( EVENT_CONNECTING );
m_iCurrentState = STATE_CONNECTING;
// error check
if (! strcmp (m_szIPADDR, ""))
{
m_iCurrentState = STATE_DISCONNECTED;
return FALSE;
}
if (! m_wPORT)
m_wPORT = PORT_FTP;
m_hINET_SESSION = InternetOpen("UPDATE_FTP", INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
if (m_hINET_SESSION == NULL)
{
// "인터넷 연결 실패"
m_iCurrentState = STATE_DISCONNECTED;
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError());
return FALSE;
}
m_hFTP_CONNECTION =InternetConnect(m_hINET_SESSION, m_szIPADDR, m_wPORT, m_szID, m_szPASS, INTERNET_SERVICE_FTP, INTERNET_FLAG_PASSIVE, 0);
if (m_hFTP_CONNECTION==NULL)
{
// "FTP 에 접속할 수 없습니다."
InternetCloseHandle(m_hINET_SESSION);
m_iCurrentState = STATE_DISCONNECTED;
AddEvent( EVENT_CONNECT_FAIL );
return FALSE;
}
m_iCurrentState = STATE_CONNECTED;
m_bConnected = TRUE;
AddEvent( EVENT_CONNECT_SUCCESS );
//Connect Check Thread
/*
m_nCheckThreadID = 2;
m_hCheckThread = (HANDLE) _beginthreadex (NULL, NULL, CheckConnectThread, (LPVOID) this, 0, &m_nCheckThreadID);
if (m_hCheckThread == INVALID_HANDLE_VALUE)
{
if (m_lpFN_EventCallbackProc)
{
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError());
return FALSE;
}
}
*/
return TRUE;
}
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::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();
}
//이 부분은 StartUpdate를 호출했을때와
//호추하지 않았을 때로 구분해야 되지 않을까?
WZ_DECLAREDLL VOID CWZFtpUpdateModule::Disconnect ()
{
m_iCurrentState = STATE_DISCONNECTED;
m_bCancelUpdate = TRUE;
m_bUseEventProc = FALSE;
BOOL bPass = FALSE;
WaitForSingleObject( m_hUpdateThread, 1000 );
if( m_hFTP_CONNECTION )
{
InternetCloseHandle(m_hFTP_CONNECTION);
m_hFTP_CONNECTION = NULL;
}
if( m_hINET_SESSION )
{
InternetCloseHandle(m_hINET_SESSION);
m_hINET_SESSION = NULL;
}
m_bCancelUpdate = FALSE;
m_bUseEventProc = TRUE;
if( m_bConnected )
{
AddEvent( EVENT_DISCONNECT );
m_bConnected = FALSE;
}
}
//lpszLocalPath : 파일이름이 포함되지 않은 순수한 로컬 경로
//lpszRemoteFilePath : 파일이름이 포함된 경로
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::DownloadFile (LPSTR lpszRemoteFilePath, LPSTR lpszLocalPath)
{
eRESULT_CONSTANT eResult;
HANDLE hLocalFile = INVALID_HANDLE_VALUE;
HINTERNET hFTP_FILE = NULL;
//1. 원격서버 경로를 분석한다.
CHAR lpszRemoteDir[MAX_PATH];
CHAR lpszRemoteFileName[MAX_PATH];
GetSplitFilePath( lpszRemoteFilePath, lpszRemoteDir, lpszRemoteFileName );
// 2. 함수이름 : MakeLocalFileName
CHAR lpszRealLocalFilePath [MAX_PATH];
GetLocalFileName( lpszLocalPath, lpszRemoteFileName, lpszRealLocalFilePath );
//3. 파일 상태 구조체에 현재 받을 파일이름 저장
ZeroMemory(m_strtUpdatingFileInfo.szFileName, sizeof(m_strtUpdatingFileInfo.szFileName) );
strcpy( m_strtUpdatingFileInfo.szFileName, lpszRemoteFileName );
__try
{
hLocalFile = MakeLocalFile( lpszRealLocalFilePath );
if(hLocalFile == NULL)
return FALSE;
//////////////////////////////////////////////////////////////////////////
// 원격지 작업 폴더 설정
SetRemoteDirectory( lpszRemoteDir );
//////////////////////////////////////////////////////////////////////////
// 원격지 파일정보 얻기
DWORD dwFileLength = 0;
FILETIME fileTime;
BOOL bRet = GetRemoteFileInfo( lpszRemoteFileName, dwFileLength, fileTime );
if( bRet == FALSE ) // TRUE를 하는것은 FALSE를 리턴할 경우 ERROR를 발생하기 때문이다.
return FALSE; // 규약에 파일의 크기가 0인것은 다운로드 하지 않는다는 것을 포함하자.
//////////////////////////////////////////////////////////////////////////
// 로컬에 서버에서 다운 받으려는 파일이 있는지 비교.. 파일 이름/날짜 비교
//////////////////////////////////////////////////////////////////////////
// 3 . 원격지 파일 열기
hFTP_FILE = OpenRemoteFile( lpszRemoteFileName );
if( hFTP_FILE == NULL )
{
DeleteFile (lpszRealLocalFilePath);
return FALSE; // 파일 하나만 실패해도 다운로드를 멈춤다.
}
// 상태 알림 사용자 함수가 지정되었다면 여기로 현재의 상태를 알린다.(다운로드 시작)
AddEvent( EVENT_FILEDOWNLOAD_START );
//////////////////////////////////////////////////////////////////////////
// 원격지 파일 얻기
eResult = GetRemoteFile( hFTP_FILE, hLocalFile, dwFileLength );
if( eResult != RESULT_DOWNLOAD_SUCCESS )
{
// 파일 얻기에 실패하면, 현재 만들어진 로컬 파일도 삭제한다.
DeleteFile (lpszRealLocalFilePath);
return FALSE;
}
//로컬파일에 시간 설정
SetFileTime(hLocalFile, NULL, NULL, &fileTime);
// 상태 알림 사용자 함수가 지정되었다면 여기로 현재의 상태를 알린다.(다운로드 끝)
AddEvent( EVENT_FILEDOWNLOAD_END );
}
__finally
{
if (hFTP_FILE)
InternetCloseHandle(hFTP_FILE);
if (hLocalFile != INVALID_HANDLE_VALUE)
CloseHandle(hLocalFile);
}
return TRUE;
}
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::GetSplitFilePath(const char* pszFilePath, char* pszDir, char* pszFile)
{
char pszFileName[MAX_PATH];
char pszFileExt[MAX_PATH];
//여기서 서버폴더 조건이 따른다.
//서버의 폴더는 루트에서 하위 1단계 까지만 존재한다.
_splitpath(pszFilePath, NULL, pszDir, pszFileName, pszFileExt );
if(pszDir == NULL || (strcmp(pszDir, "") == 0) )
wsprintf(pszDir, "\\");
wsprintf( pszFile, "%s%s", pszFileName, pszFileExt);
return TRUE;
}
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::GetLocalFileName(const char* pszLocalPath, const char* pszRemoteFileName, char* pszLocalFilePath)
{
// 여기서 로컬 파일이름이 없다면 원격파일이름과 동일하게 만든다.
if ( (!pszLocalPath) || (! strcmp(pszLocalPath, "")))
wsprintf(pszLocalFilePath, ".\\%s", pszRemoteFileName);
else
{
// 끝에 \ 이 있는지 없는지 검사해서 적용한다.
if (pszLocalPath[strlen(pszLocalPath)-1] != '\\')
wsprintf (pszLocalFilePath, "%s\\%s", pszLocalPath, pszRemoteFileName );
else
wsprintf (pszLocalFilePath, "%s%s", pszLocalPath, pszRemoteFileName);
}
return TRUE;
}
WZ_DECLAREDLL HANDLE CWZFtpUpdateModule::MakeLocalFile(const char* pszLocalFilePath)
{
//if( _access(pszLocalFilePath, 0) == 0 ) //만일현제 로컬파일이 존재하면.
//{
// BOOL bRet = DeleteFile (pszLocalFilePath);
// if(bRet == FALSE)
// {
// MessageBox (NULL, "파일삭제 실패", "메세지", MB_OK);
// }
//}
HANDLE hLocalFile = CreateFile(pszLocalFilePath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
if (hLocalFile == INVALID_HANDLE_VALUE)
{
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError());
MessageBox (HWND_DESKTOP, this->GetLastErrorMsg(), "Local File Create Error", MB_OK);
return NULL;
}
return hLocalFile;
}
//pszRemoteDir은 루트에 .\ 이런 것이 안붙는다.
//pszRemoteDir은 루트에서 부터의 경로이다.
//루트는 아무것도 안 붙는다.
//만일 .\AgentServer\\sss 경로에 있는 sss 폴더를 세팅하고 싶으면..
//pszRemoteDir은 AgentServer\\sss 이 되어야 한다.
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::SetRemoteDirectory(const char* pszRemoteDir)
{
DWORD dwSize = 128;
BOOL bSuccess = FALSE;
//1. 루트로 설정
bSuccess = FtpSetCurrentDirectory( m_hFTP_CONNECTION, "/" );
bSuccess = FtpSetCurrentDirectory( m_hFTP_CONNECTION, pszRemoteDir );
if(!bSuccess)
{
//MessageBox( NULL, szDir, "현재디렉토리", MB_OK );
// 제일 처음 빼고는 당연히 실패한다.
// 실패하는 이유는 현재디렉토리는 " /00.00.04 " 이런식으로 설정되어 있는데,
// 내가 설정할 디렉토리는 ".\00.00.04\ " 이런식으로 설정하려고 한다.
// 만일 이렇게 하면 /00.00.04/00.00.04 디렉토리를 찾게 되므로
//DWORD dwError;
//char szMsg[128];
//DWORD nSize = sizeof(szMsg);
//InternetGetLastResponseInfo( &dwError, szMsg, &nSize );
//MessageBox( NULL, szMsg, NULL, MB_OK );
}
else
{
//MessageBox( NULL, szDir, "현재디렉토리", MB_OK );
//MessageBox( NULL, lpszRemoteDir, "바꿀 디렉토리", MB_OK );
}
return bSuccess;
}
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::GetRemoteFileInfo(const char* pszRemoteFile, DWORD& dwFileLength, FILETIME& fileTime)
{
dwFileLength = 0;
WIN32_FIND_DATA wfd;
//고생했다... --+++
//FtpFindFirstFile 이 함수는 (두번째 인자에 무엇을 넣든..) 무조건 현재 세팅된 디렉토리의 제일첫번째 파일(또는 디렉토리)
//이름을 가지고 온다. 첫번째 파일(디렉토리) 정보는 wfd에 저장된다.
//파일이 없으면 NULL 이 리턴된다.
HINTERNET hFind = ::FtpFindFirstFile(m_hFTP_CONNECTION, "*.*", &wfd, 0, 0);
if( hFind == NULL )
{
INT iErrno = GetLastError();
if (iErrno != ERROR_INTERNET_EXTENDED_ERROR)
{
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, iErrno);
return -1; //핸들에 문제가 있는 경우이다.
}
}
else
{
BOOL bRet = FALSE;
do
{
// 원격지의 파일이름과, 현재 찾은 파일이름이 같다면...
if( strcmp(wfd.cFileName, pszRemoteFile) == 0 )
{
dwFileLength = wfd.nFileSizeLow;
fileTime = wfd.ftLastWriteTime;
bRet = TRUE;
break;
}
}while( bRet = InternetFindNextFile( hFind, &wfd ) );
if(bRet == FALSE) //파일을 찾지 못한경우
{
if( ERROR_NO_MORE_FILES == GetLastError() )
{
TCHAR szMsg[128];
_snprintf( szMsg, 128, "%s 파일이 없습니다.", pszRemoteFile );
MessageBox( NULL, szMsg, NULL, MB_OK );
}
return FALSE;
}
else // 파일은 찾았는데..
{
if( wfd.dwFileAttributes != FILE_ATTRIBUTE_NORMAL )
{
char szError[128];
wsprintf(szError, "버전정보 파일을 읽을수 없습니다. 파일 속성 = %d", wfd.dwFileAttributes );
MessageBox( NULL, szError, NULL, MB_OK );
InternetCloseHandle(hFind);
return FALSE;
}
}
InternetCloseHandle(hFind);
}
return TRUE;
}
WZ_DECLAREDLL HINTERNET CWZFtpUpdateModule::OpenRemoteFile(const char* pszRemoteFile)
{
HINTERNET hRemoteFile;
INT iCount = MAX_RETRY_DOWNLOADING;
while(iCount--)
{
hRemoteFile = ::FtpOpenFile(m_hFTP_CONNECTION, pszRemoteFile, GENERIC_READ, FTP_TRANSFER_TYPE_BINARY |
INTERNET_FLAG_RELOAD | INTERNET_FLAG_DONT_CACHE, 0);//(DWORD)this);
if( hRemoteFile == NULL )
{
// 에러
m_strtUpdatingFileInfo.iErrorCode = GetLastError();
Sleep(100);
}
else
{
break;
}
}
if( iCount < 1 )
{
// 파일을 찾을수 없다.
AddEvent( EVENT_FILE_NOTFOUND );
return NULL;
}
return hRemoteFile;
}
//원격지에 있는 파일을 가져온다.(다운로드 한다.)
WZ_DECLAREDLL eRESULT_CONSTANT CWZFtpUpdateModule::GetRemoteFile(const HINTERNET hRemoteFile, const HANDLE hLocalFile, DWORD dwFileLength)
{
DWORD dwNowRecvBytes, dwNowWrittenBytes, dwTotalWrittenBytes;
dwNowRecvBytes = dwNowWrittenBytes = dwTotalWrittenBytes = 0;
m_strtUpdatingFileInfo.uWholeFileSize = dwFileLength;
m_strtUpdatingFileInfo.uNowFileSize = 0;
int nZeroCount = 0;
if(dwFileLength > 0)
{
while(TRUE)
{
if ( !InternetReadFile(hRemoteFile, m_FTP_cBUFFER, sizeof(m_FTP_cBUFFER), &dwNowRecvBytes) )
{
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError());
MessageBox (HWND_DESKTOP, "InternetReadFile() error", "에러에러", MB_OK);
ZeroMemory( m_FTP_cBUFFER, MAX_BUFFER );
return RESULT_DOWNLOAD_FAIL;
}
if( dwNowRecvBytes == 0)
nZeroCount++;
if( nZeroCount >= 2)
{
AddEvent( EVENT_FILEDOWNLOAD_FAIL );
return RESULT_DOWNLOAD_FAIL;
}
// 만일 이 루틴을 실행하는 도중에 FTP서버와 연결이 끊어지면, 저 함수에서 FALSE를 리턴하는 것이
// 아니라, dwNowRecvBytes가 0이 리턴된다. 그래도, 에러가 아닐수도 있으므로, 2번 0이 나오면,
// 루틴을 빠져 나오는 것으로 한다.
if ( !WriteFile (hLocalFile, m_FTP_cBUFFER, dwNowRecvBytes, &dwNowWrittenBytes, NULL) )
{
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError());
MessageBox (HWND_DESKTOP, "WriteFile() error", "에러에러", MB_OK);
return RESULT_DOWNLOAD_FAIL;
}
dwTotalWrittenBytes += dwNowWrittenBytes;
m_strtUpdatingFileInfo.uNowFileSize = dwTotalWrittenBytes;
AddEvent( EVENT_FILEDOWNLOAD_WORKING );
// 현재까지 쓰여진 파일의 용량이 전체 파일의 용량보다 같거나 크다면...
if ( dwTotalWrittenBytes >= dwFileLength)
break;
if( m_bCancelUpdate )
{
m_bCancelUpdate = FALSE;
return RESULT_DOWNLOAD_STOP;
}
}
}
else if(dwFileLength == 0)
{
return RESULT_DOWNLOAD_SUCCESS;
}
else
{
// 원격지 파일 크기가 0보다 작다.
SetLastErrorMsg(CWZUpdateModule::ERROR_SYSTEM, GetLastError());
return RESULT_DOWNLOAD_FAIL;
}
return RESULT_DOWNLOAD_SUCCESS;
}
//파일 리스트에 있는 파일들을 다운 받는다.
//원래 리스트파일 이름은 고정인데.. 그냥 인자로 써준다.
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::DownloadFilesOnFileList( char* pszRemoteDirPath )
{
// 로컬에 리스트 파일을 다운 받았으면, 반드시 실행파일 경로에 있다. 따라서 이름만 써준다.
if(( SMDFile = fopen( DEFAULT_FILELIST_FILE, "r" ) ) == NULL )
{
MessageBox (HWND_DESKTOP, "can't open update-list file", "error", MB_OK | MB_ICONERROR);
return FALSE;
}
string strLocalRelativeDirPath; //로컬 저장 디렉토리 상대경로
// 파일 리스트를 분석해서, 파일 리스트에 있는 파일들을 다운 받는 구문..
while( TRUE )
{
SMDToken Token;
//GetToken하게 되면 SMDFile에 정의된 파일을 읽어서 Token과 TokenString에 값을 리턴하게 된다.
Token = (*GetToken)();
if( Token == END )
break;
if( Token == NAME ) // 파일이름인 경우
{
// 원격지 호스트의 전체경로이름
CHAR szCurrentRemoteFilePath[MAX_PATH];
ZeroMemory( szCurrentRemoteFilePath, MAX_PATH );
wsprintf(szCurrentRemoteFilePath, "%s\\%s", pszRemoteDirPath, TokenString);
string strAbsoluteDirPath;
GetLocalDownAbsoluteDir( strLocalRelativeDirPath, strAbsoluteDirPath ) ;
//////////////////////////////////////////////////////////////////////////
// 현재 이 파일을 다운 받은적이 있는지 체크..
// 1. 서버파일과 똑같은 로컬파일이 있는지 체크
string strLocalFilePath;
strLocalFilePath = strAbsoluteDirPath;
strLocalFilePath += TokenString;
BOOL bRet = CanDownloadFile( strLocalFilePath, TokenString );
if(bRet == TRUE) //다운로드 해도 된다면..
{
if ( !DownloadFile(szCurrentRemoteFilePath, const_cast(strAbsoluteDirPath.data()) ) )
return FALSE;
}
else //다운로드 하지 않아도 된다면.. 이미 지난번에 다운 받은 파일이다.
continue;
// 파일 다운로드가 성공했으므로 파일 개수를 하나 증가시켜 준다.
m_strtUpdatingFileInfo.uNowFileNum++;
}
if( Token == NUMBER ) // 폴더이름인 경우
{
// 0이면 폴더를 생성하라는 명령이므로, 0 다음에 나오는 값을 읽어야 한다.
// 그 값이 폴더 명이다.
Token = (*GetToken)();
strLocalRelativeDirPath = "\\";
strLocalRelativeDirPath += TokenString;
string strAbsoluteDirPath;
GetLocalDownAbsoluteDir( strLocalRelativeDirPath, strAbsoluteDirPath ) ;
CreateDirectory (strAbsoluteDirPath.data(), NULL); // 로컬 디렉토리 생성
}
}
return TRUE;
}
//로컬 파일과 서버파일을 비교하여 다운 받을지 안 받을지를 결정한다.
//TRUE : 파일을 다운 받는다.
//FALSE : 파일을 다운 받지 않는다.
WZ_DECLAREDLL BOOL CWZFtpUpdateModule::CanDownloadFile(string strLocalFilePath, string strRemoteFileName)
{
//////////////////////////////////////////////////////////////////////////
// 현재 이 파일을 다운 받은적이 있는지 체크..
// 1. 서버파일과 똑같은 로컬파일이 있는지 체크
HANDLE hFile = CreateFile( strLocalFilePath.data(), GENERIC_READ, FILE_SHARE_READ,
NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
if( hFile != INVALID_HANDLE_VALUE ) //파일이 존재한다.
{
// 2. 원격지 파일 정보
DWORD dwFileLength;
FILETIME fileTime, localFileTime;
GetRemoteFileInfo( strRemoteFileName.data(), dwFileLength, fileTime);
//3. 로컬파일 정보 읽어오기
GetFileTime( hFile, NULL, NULL, &localFileTime );
// 정보를 다 읽었으니, 핸들을 닫는다.
CloseHandle( hFile );
// 두 파일의 시간이 동일하지 않다.
if( ( fileTime.dwHighDateTime != localFileTime.dwHighDateTime ) ||
( fileTime.dwLowDateTime != localFileTime.dwLowDateTime ) )
{
return TRUE;
}
else //시간마져 동일하다면... 파일의 크기 비교
{
FILE* pFile = fopen(strLocalFilePath.data(), "r");
long nLen = _filelength( _fileno(pFile) );
fclose(pFile);
return FALSE;
if( nLen == dwFileLength) //파일의 크기가 같다면..
return FALSE; //파일을 다운 받지 않는다.
else
return TRUE;
}
}
else //파일이 존재 하지 않으면..
{
return TRUE;
}
//////////////////////////////////////////////////////////////////////////
}
//서버와의 연결 상태를 체크하는 쓰레드
//연결을 체크하는 함수가 없기 때문에 현재 서버의 디렉토리를 가져오는 함수로 체크한다.
WZ_DECLAREDLL unsigned __stdcall CWZFtpUpdateModule::CheckConnectThread(LPVOID lpParam)
{
CWZFtpUpdateModule* lpWZUM = (CWZFtpUpdateModule*) lpParam;
while( lpWZUM->m_bConnected )
{
Sleep(1000);
HINTERNET hConnection = InternetConnect( lpWZUM->m_hINET_SESSION, lpWZUM->m_szIPADDR, lpWZUM->m_wPORT,
lpWZUM->m_szID, lpWZUM->m_szPASS, INTERNET_SERVICE_FTP, INTERNET_FLAG_PASSIVE, 0);
if (hConnection == NULL) //Ftp에 연결할수 없습니다.
{
lpWZUM->m_bConnected = FALSE;
char szError[512];
wsprintf( szError, "서버와 연결할수 없습니다. ErrorCode = %d", GetLastError() );
MessageBox(NULL, szError, "알림", MB_OK);
lpWZUM->AddEvent( EVENT_DISCONNECT );
}
else
{
InternetCloseHandle( hConnection );
}
}
return 0;
}