本文分析了飛鴿傳輸核心傳送過程。
- DWORD WINAPI TMainWin::SendFileThread(void *_sendFileObj)
- {
- SendFileObj *obj = (SendFileObj *)_sendFileObj;
- fd_set fds;
- fd_set *rfds = NULL, *wfds = &fds;
- timeval tv;
- int sock_ret;
- BOOL ret = FALSE, completeWait = FALSE;
- // 這里SendFileFunc根據command類型自動選擇兩種函數 : send file or send directory
- BOOL (TMainWin::*SendFileFunc)(SendFileObj *obj) =
- obj->command == IPMSG_GETDIRFILES ? TMainWin::SendDirFile : TMainWin::SendFile;
- FD_ZERO(&fds);
- FD_SET(obj->conInfo->sd, &fds);
- // 這里for條件引入了一個簡單的超時機制
- // 正常情況下，只要文件未傳送完，循環不會退出
- for (int waitCnt=0; waitCnt < 180 && obj->hThread != NULL; waitCnt++)
- {
- tv.tv_sec = 1, tv.tv_usec = 0;
- // 這里select有什么用途呢? 對于select功能我還不是完全明白
- // 根據我的分析，這里主要是利用了select函數的等待功能
- // 如果sd描述符沒有就緒，則在select中最久等待1秒
- // 如此反復等待最多180次，也就是3分鐘，超過三分鐘后，for循環結束
- if ((sock_ret = ::select(obj->conInfo->sd + 1, rfds, wfds, NULL, &tv)) > 0)
- {
- // 套接字可用，清除等待
- waitCnt = 0;
- //下面的代碼是一個有限狀態機
- if (completeWait)
- {
- // 本分支在文件發送完后執行
- if (::recv(obj->conInfo->sd, (char *)&ret, sizeof(ret), 0) >= 0)
- ret = TRUE;
- break;
- }
- else if ((mainWin->*SendFileFunc)(obj) != TRUE)
- {
- //本分支僅在發送出錯時進行
- break;
- }
- else if (obj->status == FS_COMPLETE)
- {
- // 本分支在發送完成后執行
- completeWait = TRUE, rfds = &fds, wfds = NULL;
- if (obj->fileSize == 0) { ret = TRUE; break; }
- }
- }
- else if (sock_ret == 0) {
- // select超時，重置fds
- FD_ZERO(&fds);
- FD_SET(obj->conInfo->sd, &fds);
- }
- else if (sock_ret == SOCKET_ERROR) {
- // select錯誤，算了，離去吧~
- break;
- }
- }
- // 如果發送的是文件夾，還需要擦一下屁股
- if (obj->isDir)
- {
- mainWin->CloseSendFile(obj);
- while (--obj->dirCnt >= 0)
- ::FindClose(obj->hDir[obj->dirCnt]);
- }
- // ret是對方發回的返回值，告知發送方是否完成接收
- obj->status = ret ? FS_COMPLETE : FS_ERROR;
- // 發送TCPEVENT消息，關閉句柄
- // 消息處理流程: EventUser->TcpEvent->EndSendFile
- mainWin->PostMessage(WM_TCPEVENT, obj->conInfo->sd, FD_CLOSE);
- // 退出發送線程
- ::ExitThread(0);
- return 0;
- }
上面傳送數據最重要的一句是：
else if ((mainWin->*SendFileFunc)(obj) != TRUE)
SendFileFunc的實際內容是什么呢？由函數開始賦值的指針知道：
- BOOL TMainWin::SendFile(SendFileObj *obj)
- {
- if (obj == NULL || obj->hFile == INVALID_HANDLE_VALUE) //判斷文件句柄是否合法
- return FALSE;
- int size = 0;
- _int64 remain = obj->fileSize - obj->offset; //取得還需要傳遞的總字節數
- //傳數據
- if (remain > 0 && (size = ::send(obj->conInfo->sd, obj->mapAddr + (obj->offset % cfg->ViewMax), remain > cfg->TransMax ? cfg->TransMax : (int)remain, 0)) < 0)
- return FALSE;
- // 根據本次成功發送的數據量，調整offset
- obj->offset += size;
- // 如果offset等于文件大小了，那么設置obj狀態為完成
- // 由于存在傳文件夾模式和傳文件模式，所以狀態分情況設置
- if (obj->offset == obj->fileSize)
- obj->status = obj->command == IPMSG_GETDIRFILES ? FS_ENDFILE : FS_COMPLETE;
- else if ((obj->offset % cfg->ViewMax) == 0)//沒有完成，但是已經傳送完成了本部分數據映射，需要調整映射窗口
- {
- ::UnmapViewOfFile(obj->mapAddr); // 刪除舊映射
- remain = obj->fileSize - obj->offset; // 計算新的剩余量
- // 映射下一塊，一次8M ，如果只剩下最后一點了，則少于8M (remain)
- obj->mapAddr = (char *)::MapViewOfFile(obj->hMap, FILE_MAP_READ, (int)(obj->offset >> 32), (int)obj->offset, (int)(remain > cfg->ViewMax ? cfg->ViewMax : remain));
- }
- // 更新總消耗時間
- obj->conInfo->lastTick = ::GetTickCount();
- return TRUE;
- }
很多朋友向我要飛鴿帶注釋的源碼，實在很抱歉，我只注釋了這么多，其余的也沒有深入地看。如果你對帶注釋的源碼感興趣，不妨來這里看看：
http://code.google.com/p/ipigeon/這是我在GoogleCode上開的一個項目，大家一起來注釋飛鴿源碼吧！
400 186 1886
