void CFileConn::_HandleClientFilePullFileReq(CImPdu *pPdu) { if (!_IsAuth()) { return; } IM::File::IMFilePullDataReq 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 mode = msg.trans_mode(); uint32_t offset = msg.offset(); uint32_t datasize = msg.data_size(); IM::File::IMFilePullDataRsp msg2; msg2.set_result_code(1); msg2.set_task_id(task_id); msg2.set_user_id(user_id); msg2.set_offset(offset); msg2.set_data(""); CImPdu pdu; pdu.SetPBMsg(&msg2); pdu.SetServiceId(SID_FILE); pdu.SetCommandId(CID_FILE_PULL_DATA_RSP); pdu.SetSeqNum(pPdu->GetSeqNum()); // since the task had been created when the recver logged-in // we can find task in g_file_task_map here 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); log("invalid task id %s ", task_id.c_str()); SendPdu(&pdu); 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 /*for the realtime recver*/ && t->to_user_id != user_id /*for the offline download*/) { // invalid user log("illieage user %d for task %s", user_id, task_id.c_str()); SendPdu(&pdu); t->unlock(__LINE__); return; } switch (mode) { case IM::BaseDefine::FILE_TYPE_ONLINE: // transfer request to sender { CFileConn* pConn = (CFileConn*)t->GetOpponentConn(user_id); pConn->SendPdu(pPdu); break; } case IM::BaseDefine::FILE_TYPE_OFFLINE: // for the offline download { // find file use task id // send header info to user // send data // save path manager not used // save transfered info into task // like FILE* // transfered size // haven't been opened size_t size = 0; if (NULL == t->fp) { char save_path[BUFSIZ] = {0}; snprintf(save_path, BUFSIZ, "%s/%d", g_current_save_path, user_id); // those who can only get files under their user_id-dir int ret = mkdir(save_path, 0755); if ( (ret != 0) && (errno != EEXIST) ) { log("mkdir failed for path: %s", save_path); SendPdu(&pdu); t->unlock(__LINE__); return; } strncat(save_path, "/", BUFSIZ); strncat(save_path, task_id.c_str(), BUFSIZ); // use task_id as file name, in case of same-name file // open at first time t->fp = fopen(save_path, "rb"); // save fp if (!t->fp) { log("can not open file"); SendPdu(&pdu); t->unlock(__LINE__); return; } // read head at open if (NULL == t->file_head) { t->file_head = new file_header_t; if (NULL == t->file_head) { // close to ensure next time will new file-header again log("read file head failed."); fclose(t->fp); SendPdu(&pdu); t->unlock(__LINE__); return; } } size = fread(t->file_head, 1, sizeof(file_header_t), t->fp); // read header if (sizeof(file_header_t) > size) { // close to ensure next time will read again log("read file head failed."); fclose(t->fp); // error to get header SendPdu(&pdu); t->unlock(__LINE__); return; } // the header won't be sent to recver, because the msg svr had already notified it. // if the recver needs to check it, it could be helpful // or sometime later, the recver needs it in some way. } // read data and send based on offset and datasize. char* tmpbuf = new char[datasize]; if (NULL == tmpbuf) { // alloc mem failed log("alloc mem failed."); SendPdu(&pdu); t->unlock(__LINE__); return; } memset(tmpbuf, 0, datasize); // offset file_header_t int iret = fseek(t->fp, sizeof(file_header_t) + offset, SEEK_SET); // read after file_header_t if (0 != iret) { log("seek offset failed."); SendPdu(&pdu); delete[] tmpbuf; t->unlock(__LINE__); return; // offset failed } size = fread(tmpbuf, 1, datasize, t->fp); msg2.set_data(tmpbuf, size); CImPdu pdu2; pdu2.SetPBMsg(&msg2); pdu2.SetServiceId(SID_FILE); pdu2.SetCommandId(CID_FILE_PULL_DATA_RSP); pdu2.SetSeqNum(pPdu->GetSeqNum()); pdu2.SetSeqNum(pPdu->GetSeqNum()); SendPdu(&pdu2); delete[] tmpbuf; t->transfered_size += size; // record transfered size for next time offset if (0 == size) { fclose(t->fp); t->fp = NULL; _StatesNotify(CLIENT_FILE_DONE, task_id.c_str(), user_id, this); Close(); t->self_destroy = true; t->unlock(__LINE__); return; } 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); }