IPMsg（飞鸽传书）文件发送源码分析

本文分析了飞鸽传输核心传送过程。

2. DWORD WINAPITMainWin::SendFileThread( void *_sendFileObj)
3. {
4. SendFileObj*obj=(SendFileObj*)_sendFileObj;
5. fd_setfds;
6. fd_set*rfds=NULL,*wfds=&fds;
7. timevaltv;
8. int sock_ret;
9. BOOL ret=FALSE,completeWait=FALSE;
10. //这里SendFileFunc根据command类型自动选择两种函数:sendfileorsenddirectory
11. BOOL (TMainWin::*SendFileFunc)(SendFileObj*obj)=
12. obj->command==IPMSG_GETDIRFILES?TMainWin::SendDirFile:TMainWin::SendFile;
14. FD_ZERO(&fds);
15. FD_SET(obj->conInfo->sd,&fds);
16. //这里for条件引入了一个简单的超时机制
17. //正常情况下，只要文件未传送完，循环不会退出
18. for ( int waitCnt=0;waitCnt<180&&obj->hThread!=NULL;waitCnt++)
19. {
20. tv.tv_sec=1,tv.tv_usec=0;
21. //这里select有什么用途呢?对于select功能我还不是完全明白
22. //根据我的分析，这里主要是利用了select函数的等待功能
23. //如果sd描述符没有就绪，则在select中最久等待1秒
24. //如此反复等待最多180次，也就是3分钟，超过三分钟后，for循环结束
25. if ((sock_ret=::select(obj->conInfo->sd+1,rfds,wfds,NULL,&tv))>0)
26. {
27. //套接字可用，清除等待
28. waitCnt=0;
30. //下面的代码是一个有限状态机
31. /*
32. *控制变量completeWait,obj->status
33. *状态机迁移过程(一般状态下):
34. *(1)if((mainWin->*SendFileFunc)(obj)!=TRUE)
35. *(2)if(obj->status==FS_COMPLETE)
36. *(3)if(completeWait)
37. *
38. *1首先被反复执行，直到文件发送完毕。
39. *在没有错误的情况下，1总是等价于if(false)
40. *于是其后的2每次都会被执行，判断是否完成
41. *一旦完成，completeWait被设置为True，
42. *在下一次循环里，进入3
43. *在3的语句体内执行recv函数，等待对方应答
44. *
45. */
46. if (completeWait)
47. {
48. //本分支在文件发送完后执行
49. if (::recv(obj->conInfo->sd,( char *)&ret, sizeof (ret),0)>=0)
50. ret=TRUE;
51. break ;
52. }
53. else if ((mainWin->*SendFileFunc)(obj)!=TRUE)
54. {
55. //本分支仅在发送出错时进行
56. break ;
57. }
58. else if (obj->status==FS_COMPLETE)
59. {
60. //本分支在发送完成后执行
61. completeWait=TRUE,rfds=&fds,wfds=NULL;
62. if (obj->fileSize==0){ret=TRUE; break ;}
63. }
64. }
65. else if (sock_ret==0){
66. //select超时，重置fds
67. FD_ZERO(&fds);
68. FD_SET(obj->conInfo->sd,&fds);
69. }
70. else if (sock_ret==SOCKET_ERROR){
71. //select错误，算了，离去吧~
72. break ;
73. }
74. }
76. //如果发送的是文件夹，还需要擦一下屁股
77. if (obj->isDir)
78. {
79. mainWin->CloseSendFile(obj);
80. while (--obj->dirCnt>=0)
81. ::FindClose(obj->hDir[obj->dirCnt]);
82. }
84. //ret是对方发回的返回值，告知发送方是否完成接收
85. obj->status=ret?FS_COMPLETE:FS_ERROR;
86. //发送TCPEVENT消息，关闭句柄
87. //消息处理流程:EventUser->TcpEvent->EndSendFile
88. mainWin->PostMessage(WM_TCPEVENT,obj->conInfo->sd,FD_CLOSE);
89. //退出发送线程
90. ::ExitThread(0);
91. return 0;
92. }

上面传送数据最重要的一句是：

else if ((mainWin->*SendFileFunc)(obj)!=TRUE)

SendFileFunc的实际内容是什么呢？由函数开始赋值的指针知道：

2. BOOL TMainWin::SendFile(SendFileObj*obj)
3. {
4. if (obj==NULL||obj->hFile==INVALID_HANDLE_VALUE) //判断文件句柄是否合法
5. return FALSE;
7. int size=0;
8. _int64remain=obj->fileSize-obj->offset; //取得还需要传递的总字节数
10. //传数据
11. if (remain>0&&(size=::send(obj->conInfo->sd,obj->mapAddr+(obj->offset%cfg->ViewMax),remain>cfg->TransMax?cfg->TransMax:( int )remain,0))<0)
12. return FALSE;
14. //根据本次成功发送的数据量，调整offset
15. obj->offset+=size;
17. //如果offset等于文件大小了，那么设置obj状态为完成
18. //由于存在传文件夹模式和传文件模式，所以状态分情况设置
19. if (obj->offset==obj->fileSize)
20. obj->status=obj->command==IPMSG_GETDIRFILES?FS_ENDFILE:FS_COMPLETE;
21. else if ((obj->offset%cfg->ViewMax)==0) //没有完成，但是已经传送完成了本部分数据映射，需要调整映射窗口
22. {
23. ::UnmapViewOfFile(obj->mapAddr); //删除旧映射
24. remain=obj->fileSize-obj->offset; //计算新的剩余量
25. //映射下一块，一次8M，如果只剩下最后一点了，则少于8M(remain)
26. obj->mapAddr=( char *)::MapViewOfFile(obj->hMap,FILE_MAP_READ,( int )(obj->offset>>32),( int )obj->offset,( int )(remain>cfg->ViewMax?cfg->ViewMax:remain));
27. }
28. //更新总消耗时间
29. obj->conInfo->lastTick=::GetTickCount();
31. return TRUE;
32. }

很多朋友向我要飞鸽带注释的源码，实在很抱歉，我只注释了这么多，其余的也没有深入地看。如果你对带注释的源码感兴趣，不妨来这里看看：http://code.google.com/p/ipigeon/ 这是我在GoogleCode上开的一个项目，大家一起来注释飞鸽源码吧！