TestUpdateDLLDlg.cpp
上传用户:dzyhzl
上传日期:2019-04-29
资源大小:56270k
文件大小:7k
源码类别:

模拟服务器

开发平台:

C/C++

  1. // TestUpdateDLLDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "TestUpdateDLL.h"
  5. #include "TestUpdateDLLDlg.h"
  6. #include "UpdateExport.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  12. /////////////////////////////////////////////////////////////////////////////
  13. // CAboutDlg dialog used for App About
  14. class CAboutDlg : public CDialog
  15. {
  16. public:
  17. CAboutDlg();
  18. // Dialog Data
  19. //{{AFX_DATA(CAboutDlg)
  20. enum { IDD = IDD_ABOUTBOX };
  21. //}}AFX_DATA
  22. // ClassWizard generated virtual function overrides
  23. //{{AFX_VIRTUAL(CAboutDlg)
  24. protected:
  25. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  26. //}}AFX_VIRTUAL
  27. // Implementation
  28. protected:
  29. //{{AFX_MSG(CAboutDlg)
  30. //}}AFX_MSG
  31. DECLARE_MESSAGE_MAP()
  32. };
  33. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  34. {
  35. //{{AFX_DATA_INIT(CAboutDlg)
  36. //}}AFX_DATA_INIT
  37. }
  38. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  39. {
  40. CDialog::DoDataExchange(pDX);
  41. //{{AFX_DATA_MAP(CAboutDlg)
  42. //}}AFX_DATA_MAP
  43. }
  44. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  45. //{{AFX_MSG_MAP(CAboutDlg)
  46. // No message handlers
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  49. /////////////////////////////////////////////////////////////////////////////
  50. // CTestUpdateDLLDlg dialog
  51. CTestUpdateDLLDlg::CTestUpdateDLLDlg(CWnd* pParent /*=NULL*/)
  52. : CDialog(CTestUpdateDLLDlg::IDD, pParent)
  53. {
  54. //{{AFX_DATA_INIT(CTestUpdateDLLDlg)
  55. // NOTE: the ClassWizard will add member initialization here
  56. //}}AFX_DATA_INIT
  57. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  58. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  59. }
  60. void CTestUpdateDLLDlg::DoDataExchange(CDataExchange* pDX)
  61. {
  62. CDialog::DoDataExchange(pDX);
  63. //{{AFX_DATA_MAP(CTestUpdateDLLDlg)
  64. // NOTE: the ClassWizard will add DDX and DDV calls here
  65. //}}AFX_DATA_MAP
  66. }
  67. BEGIN_MESSAGE_MAP(CTestUpdateDLLDlg, CDialog)
  68. //{{AFX_MSG_MAP(CTestUpdateDLLDlg)
  69. ON_WM_SYSCOMMAND()
  70. ON_WM_PAINT()
  71. ON_WM_QUERYDRAGICON()
  72. ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
  73. //}}AFX_MSG_MAP
  74. END_MESSAGE_MAP()
  75. /////////////////////////////////////////////////////////////////////////////
  76. // CTestUpdateDLLDlg message handlers
  77. BOOL CTestUpdateDLLDlg::OnInitDialog()
  78. {
  79. CDialog::OnInitDialog();
  80. // Add "About..." menu item to system menu.
  81. // IDM_ABOUTBOX must be in the system command range.
  82. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  83. ASSERT(IDM_ABOUTBOX < 0xF000);
  84. CMenu* pSysMenu = GetSystemMenu(FALSE);
  85. if (pSysMenu != NULL)
  86. {
  87. CString strAboutMenu;
  88. strAboutMenu.LoadString(IDS_ABOUTBOX);
  89. if (!strAboutMenu.IsEmpty())
  90. {
  91. pSysMenu->AppendMenu(MF_SEPARATOR);
  92. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  93. }
  94. }
  95. // Set the icon for this dialog.  The framework does this automatically
  96. //  when the application's main window is not a dialog
  97. SetIcon(m_hIcon, TRUE); // Set big icon
  98. SetIcon(m_hIcon, FALSE); // Set small icon
  99. // TODO: Add extra initialization here
  100. return TRUE;  // return TRUE  unless you set the focus to a control
  101. }
  102. void CTestUpdateDLLDlg::OnSysCommand(UINT nID, LPARAM lParam)
  103. {
  104. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  105. {
  106. CAboutDlg dlgAbout;
  107. dlgAbout.DoModal();
  108. }
  109. else
  110. {
  111. CDialog::OnSysCommand(nID, lParam);
  112. }
  113. }
  114. // If you add a minimize button to your dialog, you will need the code below
  115. //  to draw the icon.  For MFC applications using the document/view model,
  116. //  this is automatically done for you by the framework.
  117. void CTestUpdateDLLDlg::OnPaint() 
  118. {
  119. if (IsIconic())
  120. {
  121. CPaintDC dc(this); // device context for painting
  122. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  123. // Center icon in client rectangle
  124. int cxIcon = GetSystemMetrics(SM_CXICON);
  125. int cyIcon = GetSystemMetrics(SM_CYICON);
  126. CRect rect;
  127. GetClientRect(&rect);
  128. int x = (rect.Width() - cxIcon + 1) / 2;
  129. int y = (rect.Height() - cyIcon + 1) / 2;
  130. // Draw the icon
  131. dc.DrawIcon(x, y, m_hIcon);
  132. }
  133. else
  134. {
  135. CDialog::OnPaint();
  136. }
  137. }
  138. // The system calls this to obtain the cursor to display while the user drags
  139. //  the minimized window.
  140. HCURSOR CTestUpdateDLLDlg::OnQueryDragIcon()
  141. {
  142. return (HCURSOR) m_hIcon;
  143. }
  144. static int TestDLL()
  145. {
  146.     typedef int (__stdcall *UPDATEA_INIT)(KUPDATE_SETTING UpdateSetting);
  147.     typedef int (__stdcall *UPDATE_UNINIT)();
  148.     typedef int (__stdcall *UPDATE_START)();
  149.     typedef int (__stdcall *UPDATE_VERIFY)();
  150.     typedef int (__stdcall *UPDATE_CHECKNEEDUPDATE)();
  151.     typedef int (__stdcall *UPDATE_DOWNLOAD)();
  152.     typedef int (__stdcall *UPDATE_UPDATE)();
  153.     UPDATEA_INIT Update_Init;
  154.     UPDATE_UNINIT Update_UnInit;
  155.     UPDATE_START Update_Start;
  156.     HMODULE hModule;
  157.     KUPDATE_SETTING UpdateSet;
  158.     int nRetCode = 0;
  159.     CString sText;
  160.     UpdateSet.bAutoTryNextHost = false;
  161.     UpdateSet.bUseVerify = false;
  162.     UpdateSet.bUseFastestHost = false;
  163.     UpdateSet.pfnCallBackProc = NULL;
  164.     UpdateSet.ProxySetting.bUpdateAuth = true; 
  165.     UpdateSet.ProxySetting.nHostPort = 0;
  166.     UpdateSet.ProxySetting.nProxyMethod = PROXY_METHOD_DIRECT;
  167.     strcpy(UpdateSet.szVerifyInfo, "102400-010999-106075-054738");
  168.     UpdateSet.ProxySetting.szHostAddr[0] = '';
  169.     UpdateSet.ProxySetting.szPassword[0] = '';
  170.     UpdateSet.ProxySetting.szUserName[0] = '';
  171. char szModulePath[MAX_PATH + 1];
  172. ::GetModuleFileName(NULL, szModulePath, (MAX_PATH + 1));
  173. char *pszOffset = NULL;
  174. pszOffset = strrchr(szModulePath, '\');
  175.     ASSERT(pszOffset);
  176.     pszOffset[1] = '';
  177.     strcpy(UpdateSet.szDownloadPath, szModulePath);
  178. strcat(UpdateSet.szDownloadPath, "UpdateDownload\");
  179.     strcpy(UpdateSet.szUpdatePath, szModulePath);
  180. UpdateSet.nUpdateMode = 0;
  181.     strcpy(UpdateSet.szUpdateSite, "ftp://linuxhome/incoming/UpdateDownLoad/");
  182. CString strVersion = szModulePath;
  183. UpdateSet.nVersion = ::GetPrivateProfileInt(
  184. "Version",
  185. "Version",
  186. 0,
  187. strVersion + "Version.cfg");
  188.     hModule = LoadLibrary("UpdateDLL.dll");
  189.     if (!hModule)
  190.         goto Exit0;
  191.     Update_Init = (UPDATEA_INIT)GetProcAddress(hModule, "Update_Init");
  192.     if (!Update_Init)
  193.         goto Exit0;
  194.     Update_UnInit = (UPDATE_UNINIT)GetProcAddress(hModule, "Update_UnInit");
  195.     if (!Update_UnInit)
  196.         goto Exit0;
  197.     Update_Start = (UPDATE_START)GetProcAddress(hModule, "Update_Start");
  198.     if (!Update_Start)
  199.         goto Exit0;
  200.     nRetCode = Update_Init(UpdateSet);
  201.     if (nRetCode == defUPDATE_RESULT_INIT_FAILED)
  202.         goto Exit0;
  203.     nRetCode = Update_Start();
  204.     Update_UnInit();
  205. Exit0:
  206.     if (hModule)
  207.         FreeLibrary(hModule);
  208.     
  209.     
  210.     sText.Format("%d", nRetCode);
  211.     MessageBox(NULL, sText, "Test", MB_OK);
  212.     return 0;
  213. }
  214. ULONG WINAPI TestThread(LPVOID /*lpVoid*/)
  215. {
  216.     return TestDLL();
  217. }
  218. void CTestUpdateDLLDlg::OnButton1() 
  219. {
  220.     DWORD dwThreadID = 0;
  221.     HANDLE hThread;
  222.     
  223.     hThread = CreateThread(NULL, 0, TestThread, 0, 0, &dwThreadID);
  224.     if (hThread)
  225.         CloseHandle(hThread);
  226. }