www.pudn.com > WZ_UpdateModule_DLL.rar > WZ_UM.h


////////////////////////////////////////////////////////////////////////// 
// 	Wenzen Update Module Ver 0.2  
// 
//	Build By Cho Jae Sik 
// 
//  First Release 20051219 
// 
////////////////////////////////////////////////////////////////////////// 
 
 
#pragma warning (disable : 4786) 
 
#ifndef	_WZ_UPDATE_MODULE_DLL_ 
#define	_WZ_UPDATE_MODULE_DLL_ 
 
#include  
#include  
#include 		// _beginthread() 
#include 				// _access() 
#include  
#include  
#include "UpdateDefine.h" 
#include "UpdateInfo.h" 
using namespace std; 
 
 
typedef VOID (* LPFUNC_RECEIVEEVENT) (	// 사용자 정의 상황알림 콜백함수 
	INT iEvent,						// 이벤트 종류 
	LPVOID lpOBJ,					// 이벤트를 넘긴 객체 포인터 (CWZFtpUpdateModule * / CWZHttpUpdateModule *) 
	UPDATING_FILEINFO_STRUCT & strtUpdatingInfo	// 현재의 업데이트 상황정보 구조체 
); 
 
 
/////////////////////////////////////////////////////////////////////////////////////// 
/* 
	기본 업데이트 클래스 : 이 클래스에서 FTP 와 HTTP 업데이트 클래스가 상속 받는다. 
*/ 
 
class WZ_DECLAREDLL CWZUpdateModule 
{ 
protected : 
	static enum {	// 이벤트 종류 
		STATE_CONNECTING,					// 현재 접속중 
		STATE_CONNECTED,					// 연결됨 
		STATE_DISCONNECTED,					// 연결않됨 
		STATE_FILE_DOWNLOADING				// 파일 받는중 
	}; 
 
	static enum { 
		ERROR_NONE, 
		ERROR_USER, 
		ERROR_SYSTEM 
	}; 
 
	CHAR	m_szIPADDR[MAX_PATH];				// IP주소 
	WORD	m_wPORT;							// 포트번호 
	CHAR	m_szID[MAX_PATH];					// 계정명 
	CHAR	m_szPASS [MAX_PATH];				// 비밀번호 
	string	m_strDownDirectory;					// 로컬의 다운로드할 디렉토리 
	CHAR	m_szDefaultDirectory[MAX_PATH];		// FTP서버의 버전파일이 존재하는 디렉토리 
	CHAR	m_szVersionFileName[MAX_PATH];		// 버전파일의 이름을 저장 (버전파일은 루트에 있는것을 원칙으로 한다.) 
	BYTE	m_bSKIP_UPDATE;						// 이 값이 TRUE일 경우 이번의 업데이트를 헌턴 건너뛰라는 플래그 (FALSE로 바뀌어야 한다.) 
 
	// Thread  
	HANDLE			m_hUpdateThread; 
	unsigned int	m_nUpdateThreadID; 
 
	UPDATE_FOLDERINFO_STRUCT m_strtUpdateFolderInfo[MAX_FOLDERINFO];	// 업데이트 폴더의 목록을 저장할 구조체 
	UINT m_uUpdateFolderCount;											// 업데이트 서버에 있는 폴더의 개수 
 
	INT				m_iLastErrorMode;												// 마지막 에러가 사용자 에러인지 시스템 에러인지? 
	INT				m_iLastErrorCode;												// 마지막 에러 번호 저장 
	CHAR			m_szLastErrorMsg [MAX_PATH];									// 마지막 에러 메시지 저장 
 
	// 연결관련 변수들 (범용) 
	HINTERNET		m_hINET_SESSION;					// 인터넷 세션 핸들 
	INT				m_iCurrentState;					// 현재 객체의 진행상태를 저장 (받는 중에 Connect() 하거나 연결않되었는데 Disconnect() 하지 않도록) 
	BOOL			m_bCancelUpdate;					// 업데이트중 작업을 취소하는 경우에 대한 인자 
	BOOL			m_bUseEventProc;					// EventProceure를 활성화 시킬건지에 대한 인자 
 
	UPDATING_FILEINFO_STRUCT	m_strtUpdatingFileInfo;							// 콜백 함수에 현재의 업데이트 진행상황을 넘길 구조체 
	LPFUNC_RECEIVEEVENT			m_lpFN_EventCallbackProc;						// 사용자 정의 콜백함수 포인터 
 
public : 
	CWZUpdateModule (); 
	CWZUpdateModule (LPSTR lpszIPADDR, WORD wPORT, LPSTR lpszID, LPSTR lpszPASS); 
	virtual ~CWZUpdateModule (); 
 
	virtual BOOL 		Connect ()  = 0; 
	virtual BOOL 		Connect (LPSTR lpszIPADDR, WORD wPORT, LPSTR lpszID, LPSTR lpszPASS) = 0; 
	virtual VOID 		Disconnect () = 0; 
 
	virtual BOOL		StartUpdate(); 
	eRESULT_CONSTANT	CheckUpdateFiles(); 
	// 현재 연결이 성립했다는 가정하에 해당 경로의 파일을 다운로드 받는다. (물론 연결않된 상태라면 에러발생) 
	virtual BOOL		DownloadFile (LPSTR lpszRemoteFilePath,	LPSTR lpszLocalFilePath = "") = 0; 
 
	virtual VOID		SetLastErrorMsg (INT iERROR_MODE, INT iERROR_CODE, LPSTR szUSER_MSG = NULL);	 
	virtual const LPSTR	GetLastErrorMsg (); 
	virtual const INT	GetLastErrorCode (); 
 
	virtual BOOL		UpdateStop();									// 업데이트를 중지한다. 
	virtual BOOL		SetEventCallBackProc (LPFUNC_RECEIVEEVENT lpFN_EventCallbackFunc ); 
			void		SetUseEventProc(BOOL bUse = TRUE);				// 이벤트  
			void		SetVersionFile( const char* pszFile ); 
		   char*		GetVersionFile(){return m_szVersionFileName;} 
			void		SetDownloadDirectory(const char* pszDirectory); 
	const string		GetDownloadDirectory(); 
			void		GetLocalDownAbsoluteDir(string strRelativeDir, string& strAbsolteDir); 
			void		GetModuleDirectory( char* szDirPath, int nSize ); 
			BOOL		Init(); 
			char*		GetIP(){ return m_szIPADDR; } 
			WORD		GetPort(){ return m_wPORT; } 
			void		SetDefaultDirectory( char* pszDir ); 
			char*		GetDefaultDirectory(){ return m_szDefaultDirectory; } 
			void		GetVersionFilePath( TCHAR* pszVerFilePath, int nSize );		//버전파일의 경로를 얻어온다. 
			void		GetFileListFilePath( char* pszFileName, int nSize, const char* pszVerFolder );		//File List 파일의 경로를 얻어온다. 
			void		GetVersionFolderPath( char* pszFileName, int nSize, const char* pszVerFolder );	// 
			void		GetServerName( char* pszServerName, int nSize );			//현재 다운받는 해당 서버의 이름을 구한다. 
 
protected : 
			BOOL 		LoadLocalVerInfo ();	// 환경설정 파일에서 현재 버전정보 등을 불러온다. 
			BOOL 		SaveLocalVerInfo ();	// 환경설정 파일에 현재 버전정보 등을 저장한다. 
 
	virtual BOOL 		CheckVersion (LPSTR lpszFilePath);				// 해당 경로의 버전파일에서 현재 버전을 읽고 (없으면 버전 0), 연결된 루트의 특정 버전파일과 비교한다. 
	virtual BOOL 		UpdateRun();									// 해당 경로의 목록파일에서 받을 파일의 목록을 작성한다. 
	virtual BOOL 		AddEvent(int nEvent); 
			void		GetVersionFromString( char* pszVersion, byte& byVer1, byte& byVer2, byte& byVer3   ); 
	unsigned int 		GetTotalVersion( const byte byVer1, const byte byVer2, const byte byVer3 ); 
			void		SetDefaultData(void); 
	virtual	BOOL		DownloadFilesOnFileList( char* pszRemoteDirPath ) = 0; 
			BOOL		CheckLiveProcess(); 
 
 
	static unsigned __stdcall UpdateWorkerThread (LPVOID lpParam); 
 
protected: 
	CRITICAL_SECTION	m_critERROR_MSG;							// 에러 저장하는 작업을 보호하기 위함 
 
public: 
	CUpdateInfo			m_info; 
	int					m_nServerNumber;							// 현재 업데이트 하려는 서버의 내부 번호 
 
	// 로컬에 있는 버전 정보 
	BYTE 				m_btLocalVer1; 
	BYTE 				m_btLocalVer2; 
	BYTE 				m_btLocalVer3; 
	UINT 				m_uLocalTotalVersion; 
	BOOL				m_bLoadUpdateInfo; 
 
 
	BOOL				IsUpdateInfoLoad(){ return m_bLoadUpdateInfo; } 
}; 
 
 
 
/////////////////////////////////////////////////////////////////////////////////////// 
/* 
	FTP 업데이트 클래스 : 기본 클래스에서 상속 받음 
*/ 
 
 
class WZ_DECLAREDLL CWZFtpUpdateModule : public CWZUpdateModule 
{ 
	// 연결관련 변수들 (FTP) 
	HINTERNET	m_hFTP_CONNECTION; 
	CHAR		m_FTP_cBUFFER[MAX_BUFFER]; 
 
	HANDLE			m_hCheckThread; 
	unsigned int	m_nCheckThreadID; 
	BOOL			m_bConnected; 
 
 
public : 
	CWZFtpUpdateModule (); 
	CWZFtpUpdateModule (LPSTR lpszIPADDR, WORD wPORT, LPSTR lpszID, LPSTR lpszPASS); 
	virtual ~CWZFtpUpdateModule (); 
 
	virtual BOOL Connect (); 
	virtual BOOL Connect (LPSTR lpszIPADDR,  WORD wPORT= PORT_FTP, LPSTR lpszID=NULL, LPSTR lpszPASS=NULL); 
	virtual VOID Disconnect (); 
 
	virtual BOOL DownloadFile (LPSTR lpszRemoteFilePath, LPSTR lpszLocalPath = ""); 
			BOOL SetEventProc (LPFUNC_RECEIVEEVENT lpFN_EventCallbackFunc ); 
			BOOL IsConnected(){return m_bConnected;} 
 
private: 
	BOOL				GetSplitFilePath(const char* pszFilePath, char* pszDir, char* pszFile); 
	BOOL				GetLocalFileName(const char* pszLocalPath, const char* pszRemoteFileName, char* pszLocalFilePath); 
	HANDLE				MakeLocalFile(const char* pszLocalFilePath); 
	BOOL				SetRemoteDirectory(const char* pszRemoteDir); 
	BOOL				GetRemoteFileInfo(const char* pszRemoteFile, DWORD& dwFileLength, FILETIME& fileTime); 
	HINTERNET			OpenRemoteFile(const char* pszRemoteFile); 
	eRESULT_CONSTANT	GetRemoteFile(const HINTERNET hRemoteFile, const HANDLE hLocalFile, DWORD dwFileLength); 
	BOOL				DownloadFilesOnFileList(char* pszRemoteDirPath ); 
	BOOL				CanDownloadFile(string strLocalFilePath, string strRemoteFileName); 
 
public: 
	static unsigned __stdcall CheckConnectThread (LPVOID lpParam); 
}; 
 
 
#endif