void FileTransferSocket::_filePullDataRspResponse(IN std::string& body)//收 { IM::File::IMFilePullDataRsp imFilePullDataRsp; if (!imFilePullDataRsp.ParseFromString(body)) { LOG__(ERR, _T("parse failed,body:%s"), util::stringToCString(body)); return; } UInt32 nRes = imFilePullDataRsp.result_code(); if (0 != nRes) { LOG__(ERR, _T("PullDataRspResponse: error result:%d"),nRes); return; } std::string taskId = imFilePullDataRsp.task_id(); const std::string& strData = imFilePullDataRsp.file_data();//todo ?????要长度 void* pData = (void*)(strData.data()); UInt32 nBlockSize = strData.size(); UInt32 fileOffset = imFilePullDataRsp.offset(); TransferFileEntity fileEntity; if (!TransferFileEntityManager::getInstance()->getFileInfoByTaskId(taskId, fileEntity)) { LOG__(ERR, _T("can't find the fileInfo")); return; } LOG__(DEBG, _T("receive:taskId=%s,filesize=%d,name=%s,BolckSize=%d") , util::stringToCString(fileEntity.sTaskID) , fileEntity.nFileSize , fileEntity.getRealFileName() , nBlockSize); //存文件... if (!fileEntity.pFileObject->writeBlock(fileOffset, nBlockSize, pData)) { LOG__(DEBG, _T("writeBlock failed ")); return; } fileEntity.nProgress = fileOffset + nBlockSize; if (fileEntity.nProgress < fileEntity.nFileSize) { //更新进度条 TransferFileEntityManager::getInstance()->updateFileInfoBysTaskID(fileEntity);//保存当前进度 module::getFileTransferModule()->asynNotifyObserver(module::KEY_FILESEVER_UPDATA_PROGRESSBAR , fileEntity.sTaskID); //继续发file block req... int mode = fileEntity.nClientMode == IM::BaseDefine::ClientFileRole::CLIENT_OFFLINE_DOWNLOAD ? IM::BaseDefine::TransferFileType::FILE_TYPE_OFFLINE : IM::BaseDefine::TransferFileType::FILE_TYPE_ONLINE; IM::File::IMFilePullDataReq imFilePullDataReq; imFilePullDataReq.set_task_id(taskId); imFilePullDataReq.set_user_id(util::stringToInt32(fileEntity.sToID)); imFilePullDataReq.set_trans_mode(static_cast<IM::BaseDefine::TransferFileType>(mode)); imFilePullDataReq.set_offset(fileEntity.nProgress); UInt32 pullSize = fileEntity.nFileSize - fileEntity.nProgress; pullSize > nBlockSize ? imFilePullDataReq.set_data_size(nBlockSize) : imFilePullDataReq.set_data_size(pullSize); // 发包 sendPacket(IM::BaseDefine::ServiceID::SID_FILE, IM::BaseDefine::FileCmdID::CID_FILE_PULL_DATA_REQ, &imFilePullDataReq); } else//传输完成 { if (fileEntity.pFileObject) { delete fileEntity.pFileObject; fileEntity.pFileObject = nullptr; } //告知对方文件传输完成了。 IM::File::IMFileState imFileState; imFileState.set_state(IM::BaseDefine::ClientFileState::CLIENT_FILE_DONE); imFileState.set_task_id(taskId); imFileState.set_user_id(util::stringToInt32(fileEntity.sToID)); sendPacket(IM::BaseDefine::ServiceID::SID_FILE, IM::BaseDefine::FileCmdID::CID_FILE_STATE, &imFileState); TransferFileEntityManager::getInstance()->updateFileInfoBysTaskID(fileEntity); module::getFileTransferModule()->asynNotifyObserver(module::KEY_FILESEVER_PROGRESSBAR_FINISHED, fileEntity.sTaskID); } }
void CFileConn::_HandleClientFilePullFileRsp(CImPdu *pPdu) { if (!_IsAuth()) { return; } IM::File::IMFilePullDataRsp msg; CHECK_PB_PARSE_MSG(msg.ParseFromArray(pPdu->GetBodyData(), pPdu->GetBodyLength())); uint32_t user_id = msg.user_id(); string task_id = msg.task_id(); uint32_t offset = msg.offset(); uint32_t data_size = msg.data().length(); pthread_rwlock_wrlock(&g_file_task_map_lock); TaskMap_t::iterator iter = g_file_task_map.find(task_id.c_str()); if (g_file_task_map.end() == iter) { // invalid task id pthread_rwlock_unlock(&g_file_task_map_lock); return; } transfer_task_t* t = iter->second; pthread_rwlock_unlock(&g_file_task_map_lock); t->lock(__LINE__); t->create_time = time(NULL); if (t->from_user_id != user_id && t->to_user_id != user_id) { // invalid user t->unlock(__LINE__); return; } switch (t->transfer_mode) { case FILE_TYPE_ONLINE: // transfer request to sender { CFileConn* pConn = (CFileConn*)t->GetOpponentConn(user_id); pConn->SendPdu(pPdu); /// send to recver break; } case FILE_TYPE_OFFLINE: /// this is the response to the server pull-data-req { if (t->upload_packages.size() <= 0) { log("FATAL ERROR"); t->unlock(__LINE__); return; } // check if data size ok std::map<uint32_t, upload_package_t*>::iterator itPacks = t->upload_packages.find(offset); if (t->upload_packages.end() != itPacks) { // offset right // check if data size ok if (data_size != itPacks->second->size) { // the rsp's data size is different from req's // refuse it or dynamic adjust // do not adjust now, maybe later uint32_t offset = itPacks->second->offset; uint32_t size = itPacks->second->size; // resend req IM::File::IMFilePullDataReq msg2; msg2.set_task_id(task_id); msg2.set_user_id(user_id); msg2.set_trans_mode((::IM::BaseDefine::FileType)t->transfer_mode); msg2.set_offset(offset); msg2.set_data_size(size); CImPdu pdu; pdu.SetPBMsg(&msg2); pdu.SetServiceId(SID_FILE); pdu.SetCommandId(CID_FILE_PULL_DATA_REQ); SendPdu(&pdu); log("size not match"); t->unlock(__LINE__); return; } // check if data-ptr OK if (NULL == itPacks->second->data) { itPacks->second->data = new char[itPacks->second->size]; if (NULL == itPacks->second->data) { uint32_t offset = itPacks->second->offset; uint32_t size = itPacks->second->size; log("alloc mem failed"); // resend req IM::File::IMFilePullDataReq msg2; msg2.set_task_id(task_id); msg2.set_user_id(user_id); msg2.set_trans_mode((::IM::BaseDefine::FileType)t->transfer_mode); msg2.set_offset(offset); msg2.set_data_size(size); CImPdu pdu; pdu.SetPBMsg(&msg2); pdu.SetServiceId(SID_FILE); pdu.SetCommandId(CID_FILE_PULL_DATA_REQ); SendPdu(&pdu); t->unlock(__LINE__); return; } } // copy data memset(itPacks->second->data, 0, itPacks->second->size); memcpy(itPacks->second->data, msg.data().c_str(), msg.data().length()); ++t->transfered_size; } // find which segment hasn't got data yet bool bFound = false; std::map<uint32_t, upload_package_t*>::iterator itor = t->upload_packages.begin(); for ( ; itor != t->upload_packages.end(); ++itor) { if (NULL == itor->second->data) { bFound = true; break; } } if (!bFound) { // all packages recved _StatesNotify(IM::BaseDefine::CLIENT_FILE_DONE, task_id.c_str(), user_id, t->GetConn(user_id)); Close(); t->unlock(__LINE__); return; } // prepare to send req for this segment uint32_t next_offset = itor->second->offset; uint32_t next_size = itor->second->size; // send pull-data-req IM::File::IMFilePullDataReq msg2; msg2.set_task_id(task_id); msg2.set_user_id(user_id); msg2.set_trans_mode((::IM::BaseDefine::FileType)t->transfer_mode); msg2.set_offset(next_offset); msg2.set_data_size(next_size); CImPdu pdu; pdu.SetPBMsg(&msg2); pdu.SetServiceId(SID_FILE); pdu.SetCommandId(CID_FILE_PULL_DATA_REQ); SendPdu(&pdu); break; } default: break; } t->unlock(__LINE__); return; }
void FileTransferSocket::_filePullDataReqResponse(IN std::string& body)//发 { IM::File::IMFilePullDataReq imFilePullDataReq; if (!imFilePullDataReq.ParseFromString(body)) { LOG__(ERR, _T("parse failed,body:%s"), util::stringToCString(body)); return; } UInt32 fileSize = imFilePullDataReq.data_size(); UInt32 fileOffset = imFilePullDataReq.offset(); std::string taskId = imFilePullDataReq.task_id(); TransferFileEntity fileEntity; if (!TransferFileEntityManager::getInstance()->getFileInfoByTaskId(taskId, fileEntity)) { LOG__(ERR, _T("PullDataReqResponse: can't find the fileInfo")); return; } LOG__(DEBG, _T("send:taskId=%s,filesize=%d,name=%s,BolckSize=%d") ,util::stringToCString(fileEntity.sTaskID) ,fileEntity.nFileSize ,fileEntity.getRealFileName() ,fileSize); std::string buff; if (nullptr == fileEntity.pFileObject) { LOG__(ERR, _T("PullDataReqResponse: file boject Destoryed!")); return; } fileEntity.pFileObject->readBlock(fileOffset, fileSize, buff);//读取本地文件的数据块 IM::File::IMFilePullDataRsp imFilePullDataRsp;//todo check imFilePullDataRsp.set_result_code(0); imFilePullDataRsp.set_task_id(taskId); imFilePullDataRsp.set_user_id(util::stringToInt32(fileEntity.sFromID)); imFilePullDataRsp.set_offset(fileOffset); imFilePullDataRsp.set_file_data((void*)buff.data(), fileSize); //send packet sendPacket(IM::BaseDefine::ServiceID::SID_FILE, IM::BaseDefine::FileCmdID::CID_FILE_PULL_DATA_RSP , &imFilePullDataRsp); fileEntity.nProgress = fileOffset + fileSize; if (fileEntity.nProgress < fileEntity.nFileSize) { //更新进度条 TransferFileEntityManager::getInstance()->updateFileInfoBysTaskID(fileEntity);//保存当前进度 module::getFileTransferModule()->asynNotifyObserver(module::KEY_FILESEVER_UPDATA_PROGRESSBAR , fileEntity.sTaskID); } else//传输完成 { if (fileEntity.pFileObject) { delete fileEntity.pFileObject; fileEntity.pFileObject = nullptr; } module::getFileTransferModule()->asynNotifyObserver(module::KEY_FILESEVER_PROGRESSBAR_FINISHED , fileEntity.sTaskID); } TransferFileEntityManager::getInstance()->updateFileInfoBysTaskID(fileEntity); }