开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp客户端 > 查看源码
main.cpp
资源名称：update.zip　[点击查看]
上传用户：wangying89
上传日期：2007-01-07
资源大小：61k
文件大小：12k
源码类别：

Ftp客户端

开发平台：

Visual C++

main.cpp：源码内容
  1. /* sample of file wzjf.ver:
  2. 134.104.9.165  /caijia/cai_ji/datas/data06/success
  3. N2.H01
  4. N4.H01
  5. NH.H01
  6. */
  8. #include <windows.h>
  9. #include <stdio.h>
  10. #include <stdlib.h>
  11. #include <time.h>
  12. #include <ras.h>
  14. #include "resource.h"
  15. #include "tcp.h"
  16. #include "ftp.h"
  18. char hostname[20];
  19. #define UPDATE_PORT 7788
  20. #define P_VERSION_FILE "wzjf.ver"
  22. char app_name[100], host[20], path[128], version[20], new_version[20];
  23. char init_file[128];
  25. HANDLE hThread_app =NULL, hProcess_app =NULL;
  26. HWND g_hDlg =NULL;
  27. int g_fconnected =false;
  28. HRASCONN g_hRasConn =NULL;
  30. char new_file_name[100][128];
  31. int TransNewFiles(HWND hDlg);
  32. int CheckNewFiles(HWND hDlg);
  33. int sd, new_file_count =0;
  34. void log_msg(char *msg);
  35. int show_msg =0;
  36. int run_app();
  37. int term_app();
  38. int day_is_changed();
  39. int run_dial();
  40. void run_hangup();
  41. void SetState(char *msg);
  42. void WINAPI RasDialFunc(UINT msg, RASCONNSTATE state, DWORD dwError);
  44. time_t start_t =0;
  45. int f_auto_hide =0;
  46. int f_auto_hangup =0;
  48. BOOL CALLBACK MainDlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
  49. {
  50. int x, y;
  51. RECT rc;
  52. char temp[100];
  53. static int f_first =1;
  55. switch(msg)
  56. {
  57. case WM_INITDIALOG:
  58. g_hDlg =hDlg;
  59. GetWindowRect(hDlg, &rc);
  60. x =GetSystemMetrics(SM_CXSCREEN);
  61. y =GetSystemMetrics(SM_CYSCREEN);
  62. x =x/2-(rc.right-rc.left)/2;
  63. y =y/2-(rc.bottom-rc.top)/2;
  64. if(GetPrivateProfileString("update", "app", NULL, app_name, sizeof(app_name), init_file) ==0)
  65. app_name[0] =0;    // 不执行计费程序
  66. if(GetPrivateProfileInt("update", "autohide", 0, init_file) !=0)
  67. {
  68. f_auto_hide =1;    // 做完后自动隐藏
  69. CheckDlgButton(hDlg, IDK_AUTOHIDE, BST_CHECKED);
  70. }
  71. if(GetPrivateProfileInt("update", "autohangup", 0, init_file) !=0)
  72. {
  73. f_auto_hangup =1;    // 退出时自动断开
  74. CheckDlgButton(hDlg, IDK_AUTOHANGUP, BST_CHECKED);
  75. }
  76. SetWindowPos(hDlg, HWND_TOPMOST, x, y, 0, 0, SWP_NOSIZE);
  77. SetTimer(hDlg, 1, 0, NULL);
  78. return TRUE;
  79. case WM_KEYDOWN:
  80. switch(wParam)
  81. {
  82. case VK_ESCAPE:
  83. ShowWindow(hDlg, SW_MINIMIZE);
  84. break;
  85. case VK_F3:
  86. run_hangup();
  87. break;
  88. case VK_F1:
  89. WinExec("notepad update.log", SW_SHOWNORMAL);
  90. break;
  91. }
  92. break;
  93. case WM_COMMAND:
  94. switch(LOWORD(wParam))
  95. {
  96. case IDK_AUTOHIDE:
  97. if(HIWORD(wParam) ==BN_CLICKED)
  98. {
  99. if(SendMessage((HWND)lParam, BM_GETCHECK, 0, 0L) ==BST_CHECKED)
  100. f_auto_hide =1;
  101. else f_auto_hide =0;
  102. }
  103. break;
  104. case IDK_AUTOHANGUP:
  105. if(HIWORD(wParam) ==BN_CLICKED)
  106. {
  107. if(SendMessage((HWND)lParam, BM_GETCHECK, 0, 0L) ==BST_CHECKED)
  108. f_auto_hangup =1;
  109. else f_auto_hangup =0;
  110. }
  111. break;
  112. case ID_HIDE:
  113. ShowWindow(hDlg, SW_HIDE);
  114. break;
  115. case IDOK:
  116. ShowWindow(hDlg, SW_MINIMIZE);
  117. break;
  118. case ID_CLOSE:
  119. case IDCANCEL:      // press ALT+F4
  120. if(MessageBox(hDlg, "您是否确定退出?", "update", MB_YESNO|MB_DEFBUTTON2)
  121. ==IDNO)
  122. break;
  123. term_app();
  124. if(!f_auto_hangup) run_hangup();
  125. EndDialog(hDlg, 0);
  126. break;
  127. }
  128. break;
  129. case WM_TIMER:
  130. if(f_first)
  131. {
  132. f_first =0;
  133. KillTimer(hDlg, 1);
  134. if(run_dial() <0)              // 自动拨号
  135. {
  136. run_hangup();
  137. log_msg("拨号失败");
  138. }
  139. else ShowWindow(hDlg, SW_MINIMIZE);
  140. SetTimer(hDlg, 1, 10000, NULL);     // 10秒钟检查日期
  141. tcp_init();
  142. /*if(run_app() <0)
  143. {
  144. sprintf(temp, "运行程序 %s 失败", app_name);
  145. ::MessageBox(hDlg, temp, "update", MB_OK);
  146. //EndDialog(hDlg, 0);
  147. break;
  148. }*/
  149. //break;
  150. }
  151. // 检查程序是否退出
  152.     if(hProcess_app && WaitForSingleObject(hProcess_app, 0) ==WAIT_OBJECT_0)
  153. {
  154. log_msg("应用程序已经退出");
  155.     hProcess_app =NULL;
  156. }
  158. if(f_auto_hide)
  159. ShowWindow(hDlg, SW_HIDE);
  160. if(day_is_changed() ==0)
  161. break;  // 晚上24点过了？
  162. if(f_auto_hide)
  163. ShowWindow(hDlg, SW_SHOW);
  164. if(CheckNewFiles(hDlg) <=0)      // 检查是否有新版本
  165. {
  166. if(f_auto_hide) ShowWindow(hDlg, SW_HIDE);
  167. break;
  168. }
  169. if(term_app() <0)           // 中断计费程序
  170. {
  171. sprintf(temp, "term_app %s failed", app_name);
  172. log_msg(temp);
  173. if(f_auto_hide) ShowWindow(hDlg, SW_HIDE);
  174. break;
  175. }
  176. TransNewFiles(hDlg);      // 传输新版本文件
  177. if(run_app() <0)          // 运行计费程序
  178. {
  179. sprintf(temp, "run_app %s failed", app_name);
  180. log_msg(temp);
  181. }
  182. if(f_auto_hide) ShowWindow(hDlg, SW_HIDE);
  183. break;
  184. case WM_QUERYENDSESSION:      // Windows退出
  185. if(!f_auto_hangup) run_hangup();
  186. return TRUE;
  187. }
  188. return FALSE;
  189. }
  191. int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hInstPre, LPSTR lpCmdLine, int nCmdShow)
  192. {
  193. char temp[120];
  194. /*if(FindWindow(NULL, "正在检查是否更新系统...")) return 0;*/
  196. GetCurrentDirectory(sizeof(temp), temp);
  197. wsprintf(init_file, "%s\update.ini", temp);
  198. DialogBox(hInst, MAKEINTRESOURCE(IDD_MAIN), NULL, MainDlgProc);
  199. tcp_exit();
  201. return 0;
  202. }
  203. // 对SUNOS系统，由于它的ftp server不支持针对文件的SIZE和DATE命令，所以不能正常使用
  204. int CheckNewFiles(HWND hDlg)
  205. {
  206. FILE *fp =NULL;
  207. char temp[256];
  208. int ret =0;
  210. GetPrivateProfileString("update", "version", "1.0", version, sizeof(version), init_file);
  211. GetPrivateProfileString("update", "host", "134.104.69.168", host, sizeof(host), init_file);
  212. GetPrivateProfileString("update", "path", "./", path, sizeof(path), init_file);
  213. SetState("登录主机 ...");
  214. if((sd =ftp_login(host, "caijia", "caijia")) <0)
  215. {
  216. sprintf(temp, "ftp 登录失败: host=%s", host);
  217. if(show_msg) ::MessageBox(hDlg, temp, "update", MB_OK);
  218. else log_msg(temp);
  219. return -1;
  220. }
  221. if(ftp_chdir(sd, path) <0)
  222. {
  223. ftp_quit(sd);
  224. sprintf(temp, "ftp 目录失败: path=%s", path);
  225. if(show_msg) ::MessageBox(hDlg, temp, "update", MB_OK);
  226. else log_msg(temp);
  227. return -1;
  228. }
  229. SetState("读取升级文件 ...");
  230. if(ftp_get_file(sd, P_VERSION_FILE, P_VERSION_FILE, 0, 30) <=0)
  231. {
  232. ftp_quit(sd);
  233. sprintf(temp, "ftp 取升级信息文件失败: host =%s", host);
  234. if(show_msg) ::MessageBox(hDlg, temp, "update", MB_OK);
  235. else log_msg(temp);
  236. return -1;
  237. }
  238. if((fp =fopen(P_VERSION_FILE, "r")) ==NULL)
  239. {
  240. sprintf(temp, "fopen version file for read failed");
  241. if(show_msg) ::MessageBox(hDlg, temp, "update", MB_OK);
  242. else log_msg(temp);
  243. ftp_quit(sd);
  244. return -8;
  245. }
  246. new_file_count =0;
  247. if(fgets(temp, sizeof(temp), fp) ==NULL)
  248. {
  249. sprintf(temp, "read new.ver failed");
  250. if(show_msg) ::MessageBox(hDlg, temp, "ftp_ctl", MB_OK);
  251. else log_msg(temp);
  252. fclose(fp);
  253. ftp_quit(sd);
  254. return -9;
  255. }
  256. app_name[0] =0;
  257. if(sscanf(temp, "%s %s %s %s", new_version, host, path, app_name) <3)
  258. {
  259. fclose(fp);
  260. ftp_quit(sd);
  261. sprintf(temp, "读升级信息文件失败: %s, format error", P_VERSION_FILE);
  262. if(show_msg) ::MessageBox(hDlg, temp, "update", MB_OK);
  263. else log_msg(temp);
  264. return -1;
  265. }
  266. if(strcmp(new_version, version) >0)
  267. {
  268. while(!feof(fp) && fgets(new_file_name[new_file_count], 128, fp))
  269. {
  270. if(new_file_name[new_file_count][0] =='r' || new_file_name[new_file_count][0] =='n') continue;
  271. strtok(new_file_name[new_file_count], "r");
  272. strtok(new_file_name[new_file_count], "n");
  273. new_file_count++;
  274. }
  275. }
  276. fclose(fp);
  277. if(new_file_count ==0)
  278. ftp_quit(sd);
  280. return new_file_count;
  281. }
  283. int TransNewFiles(HWND hDlg)
  284. {
  285. int i, ret =0;
  286. char temp[200];
  288. ShowWindow(hDlg, SW_RESTORE);
  289. SetWindowPos(hDlg, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE|SWP_NOMOVE);
  290. for(i =0; i<new_file_count; i++)
  291. {
  292. sprintf(temp, "更新系统：正在取文件 %s...", new_file_name[i]);
  293. SetDlgItemText(hDlg, IDC_MSG, temp);
  294. if(ftp_get_file(sd, new_file_name[i], new_file_name[i], 0, 300) <=0)
  295. {
  296. sprintf(temp, "ftp 取文件失败: file=%s", new_file_name[i]);
  297. if(show_msg) ::MessageBox(hDlg, temp, "update", MB_OK);
  298. else log_msg(temp);
  299. ret =-1;
  300. continue;
  301. }
  302. }
  303. ftp_quit(sd);
  304. if(ret ==0)
  305. {
  306. WritePrivateProfileString("update", "version", new_version, init_file);
  307. WritePrivateProfileString("update", "host", host, init_file);
  308. WritePrivateProfileString("update", "path", path, init_file);
  309. WritePrivateProfileString("update", "app", app_name, init_file);
  310. sprintf(temp, "更新系统成功. 新版本=%s", new_version);
  311. }
  312. else
  313. sprintf(temp, "更新系统失败. 系统需要更新为%s", new_version);
  314. log_msg(temp);
  315. return ret;
  316. }
  318. void log_msg(char *msg)
  319. {
  320. FILE *fp;
  321. time_t t;
  322. struct tm *ptm;
  323. char temp[300];
  325. time(&t);
  326. ptm =localtime(&t);
  328. SetDlgItemText(g_hDlg, IDC_MSG, msg);
  330. if((fp =fopen("update.log", "a")) !=NULL)
  331. {
  332. fprintf(fp, "%02d.%02d %02d.%02d errno:%d, %sn",
  333. ptm->tm_mon+1, ptm->tm_mday, ptm->tm_hour,
  334. ptm->tm_min, GetLastError(), msg);
  335. fclose(fp);
  336. if(strlen(msg) >200) *(msg+200) =0;
  337. wsprintf(temp, "%02d.%02d %02d.%02d errno:%d, %srn",
  338. ptm->tm_mon+1, ptm->tm_mday, ptm->tm_hour,
  339. ptm->tm_min, GetLastError(), msg);
  340. SendDlgItemMessage(g_hDlg, IDE_MSG, EM_REPLACESEL, 0, (LPARAM)(LPSTR)temp);
  341. }
  342. }
  344. int day_is_changed()
  345. {
  346. time_t t;
  347. struct tm *ptm;
  348. static int cur_day =-1;
  349. char temp[30];
  350. int tt;
  352. time(&t);
  353. if(start_t >0)
  354. {
  355. tt =t-start_t;
  356. wsprintf(temp, "%d:%02d:%02d:", tt/3600, tt%3600/60, tt%60);
  357. SetDlgItemText(g_hDlg, IDC_TIME, temp);
  358. }
  359. ptm =localtime(&t);
  360. if(cur_day ==ptm->tm_mday) return 0;  /* first time is changed */ 
  361. cur_day =ptm->tm_mday;
  363. return 1;
  364. }
  366. int term_app()
  367. {
  368. if(f_auto_hide) WritePrivateProfileString("update", "autohide", "1", init_file);
  369. else  WritePrivateProfileString("update", "autohide", "0", init_file);
  370. if(f_auto_hangup) WritePrivateProfileString("update", "autohangup", "1", init_file);
  371. else  WritePrivateProfileString("update", "autohangup", "0", init_file);
  372. if(hProcess_app ==NULL) return 0;
  373. TerminateProcess(hProcess_app, 1);
  374. CloseHandle(hThread_app);
  375. CloseHandle(hProcess_app);
  376. return 0;
  377. }
  379. int run_app()
  380. {
  381. STARTUPINFO si;
  382. PROCESS_INFORMATION pi;
  383. char temp[100];
  385. if(app_name[0] ==0) return 0;
  386. memset(&si, 0, sizeof(si));
  387. if(CreateProcess(app_name, app_name, NULL, NULL, false, 0, NULL, NULL, &si, &pi) ==0)
  388. {
  389. sprintf(temp, "CreateProcess failed! errno:%d", GetLastError());
  390. log_msg(temp);
  391. return -1;
  392. }
  393. hThread_app =pi.hThread;
  394. hProcess_app =pi.hProcess;
  395. return 0;
  396. }
  398. int run_dial()
  399. {
  400. // rundll rnaui.dll,RnaDial ???
  401. char dial_name[100], dial_user[50], dial_pass[20], temp[200], err_buf[100];
  402. RASDIALPARAMS rdParams;
  403. DWORD dwRet;
  404. BOOL f_get_password;
  405.  
  406. if(GetPrivateProfileString("update", "dialname", NULL, dial_name,
  407. sizeof(dial_name), init_file) ==0) return 0;
  408. dial_user[0] =dial_pass[0] =0;
  409. GetPrivateProfileString("update", "dialuser", NULL, dial_user,
  410. sizeof(dial_user), init_file);
  411. GetPrivateProfileString("update", "dialpass", NULL, dial_pass,
  412. sizeof(dial_pass), init_file);
  414. memset(&rdParams, 0, sizeof(rdParams));
  415. rdParams.dwSize=sizeof(RASDIALPARAMS);
  416. strcpy(rdParams.szEntryName,dial_name);
  417. //strcpy(rdParams.szPhoneNumber,"169");
  418. //strcpy(rdParams.szCallbackNumber,"*");
  419. if(dial_user[0] ==0 || dial_pass[0] ==0)
  420. {
  421. if(RasGetEntryDialParams(NULL, &rdParams, &f_get_password) !=0)
  422. {
  423. wsprintf(temp, "读取拨号网络%s的用户名和密码失败", dial_name);
  424. log_msg(temp);
  425. return -1;
  426. }
  427. if(f_get_password ==false)
  428. {
  429. wsprintf(temp, "读取拨号网络%s的密码失败，请先使用保存密码成功拨号一次", dial_name);
  430. log_msg(temp);
  431. return -1;
  432. }
  433. }
  434. else
  435. {
  436. strcpy(rdParams.szUserName, dial_user);
  437. strcpy(rdParams.szPassword, dial_pass);
  438. }
  439. //strcpy(rdParams.szDomain,"*");
  440. g_hRasConn=NULL;
  441. //if((dwRet=RasDial(NULL,NULL,&rdParams,0L, (RASDIALFUNC)RasDialFunc,&g_hRasConn)) !=0)
  442. wsprintf(temp, "正在拨号... name=%s", dial_name);
  443. log_msg(temp);
  444. if((dwRet=RasDial(NULL,NULL,&rdParams,0L, NULL, &g_hRasConn)) !=0)
  445. {
  446. RasGetErrorString(dwRet, err_buf, sizeof(err_buf));
  447. wsprintf(temp, "拨号失败, name=%s, ret=%d:%s", dial_name, dwRet, err_buf);
  448. log_msg(temp);
  449. return -1;
  450. }
  451. time(&start_t);   // 开始计时
  452. wsprintf(temp, "拨号成功, name=%s", dial_name);
  453. log_msg(temp);
  454. return 0;
  455. }
  457. void run_hangup()
  458. {
  459. if(g_hRasConn) RasHangUp(g_hRasConn);
  460. g_hRasConn =NULL;
  461. }
  463. void WINAPI RasDialFunc(UINT msg, RASCONNSTATE state, DWORD dwError)
  464. {
  465. switch(state)
  466. {
  467. case RASCS_Connected:
  468. g_fconnected =true;
  469. break;
  470. case RASCS_Disconnected:
  471. g_fconnected =false;
  472. break;
  473. }
  474. }
  476. void SetState(char *msg)
  477. {
  478. SetDlgItemText(g_hDlg, IDC_MSG, msg);
  479. }