// SONIC ROBO BLAST 2 //----------------------------------------------------------------------------- // Copyright (C) 1998-2000 by DooM Legacy Team. // Copyright (C) 1999-2020 by Sonic Team Junior. // // This program is free software distributed under the // terms of the GNU General Public License, version 2. // See the 'LICENSE' file for more details. //----------------------------------------------------------------------------- /// \file d_netfil.c /// \brief Transfer a file using HSendPacket. #include #include #include #if defined (_WIN32) || defined (__DJGPP__) #include #include #else #include #include #include #endif #ifdef __GNUC__ #include #include #elif defined (_WIN32) #include #endif #ifdef __DJGPP__ #include #include #endif #include "doomdef.h" #include "doomstat.h" #include "d_main.h" #include "g_game.h" #include "i_net.h" #include "i_system.h" #include "m_argv.h" #include "d_net.h" #include "w_wad.h" #include "d_netfil.h" #include "z_zone.h" #include "byteptr.h" #include "p_setup.h" #include "m_misc.h" #include "m_menu.h" #include "md5.h" #include "filesrch.h" #include // Prototypes static boolean AddFileToSendQueue(INT32 node, const char *filename, UINT8 fileid); // Sender structure typedef struct filetx_s { INT32 ram; union { char *filename; // Name of the file char *ram; // Pointer to the data in RAM } id; UINT32 size; // Size of the file UINT8 fileid; INT32 node; // Destination struct filetx_s *next; // Next file in the list } filetx_t; // Current transfers (one for each node) typedef struct filetran_s { filetx_t *txlist; // Linked list of all files for the node UINT8 iteration; UINT8 ackediteration; UINT32 position; // The current position in the file boolean *ackedfragments; UINT32 ackedsize; FILE *currentfile; // The file currently being sent/received tic_t dontsenduntil; } filetran_t; static filetran_t transfer[MAXNETNODES]; // Read time of file: stat _stmtime // Write time of file: utime // Receiver structure INT32 fileneedednum; // Number of files needed to join the server fileneeded_t fileneeded[MAX_WADFILES]; // List of needed files static tic_t lasttimeackpacketsent = 0; char downloaddir[512] = "DOWNLOAD"; // For resuming failed downloads typedef struct { char filename[MAX_WADPATH]; UINT8 md5sum[16]; boolean *receivedfragments; UINT32 fragmentsize; UINT32 currentsize; } pauseddownload_t; static pauseddownload_t *pauseddownload = NULL; #ifndef NONET // for cl loading screen INT32 lastfilenum = -1; #endif luafiletransfer_t *luafiletransfers = NULL; boolean waitingforluafiletransfer = false; boolean waitingforluafilecommand = false; char luafiledir[256 + 16] = "luafiles"; /** Fills a serverinfo packet with information about wad files loaded. * * \todo Give this function a better name since it is in global scope. * Used to have size limiting built in - now handled via W_LoadWadFile in w_wad.c * */ UINT8 *PutFileNeeded(void) { size_t i, count = 0; UINT8 *p = netbuffer->u.serverinfo.fileneeded; char wadfilename[MAX_WADPATH] = ""; UINT8 filestatus; for (i = 0; i < numwadfiles; i++) { // If it has only music/sound lumps, don't put it in the list if (!wadfiles[i]->important) continue; filestatus = 1; // Importance - not really used any more, holds 1 by default for backwards compat with MS // Store in the upper four bits if (!cv_downloading.value) filestatus += (2 << 4); // Won't send else if ((wadfiles[i]->filesize <= (UINT32)cv_maxsend.value * 1024)) filestatus += (1 << 4); // Will send if requested // else // filestatus += (0 << 4); -- Won't send, too big WRITEUINT8(p, filestatus); count++; WRITEUINT32(p, wadfiles[i]->filesize); nameonly(strcpy(wadfilename, wadfiles[i]->filename)); WRITESTRINGN(p, wadfilename, MAX_WADPATH); WRITEMEM(p, wadfiles[i]->md5sum, 16); } netbuffer->u.serverinfo.fileneedednum = (UINT8)count; return p; } /** Parses the serverinfo packet and fills the fileneeded table on client * * \param fileneedednum_parm The number of files needed to join the server * \param fileneededstr The memory block containing the list of needed files * */ void D_ParseFileneeded(INT32 fileneedednum_parm, UINT8 *fileneededstr) { INT32 i; UINT8 *p; UINT8 filestatus; fileneedednum = fileneedednum_parm; p = (UINT8 *)fileneededstr; for (i = 0; i < fileneedednum; i++) { fileneeded[i].status = FS_NOTFOUND; // We haven't even started looking for the file yet fileneeded[i].justdownloaded = false; filestatus = READUINT8(p); // The first byte is the file status fileneeded[i].willsend = (UINT8)(filestatus >> 4); fileneeded[i].totalsize = READUINT32(p); // The four next bytes are the file size fileneeded[i].file = NULL; // The file isn't open yet READSTRINGN(p, fileneeded[i].filename, MAX_WADPATH); // The next bytes are the file name READMEM(p, fileneeded[i].md5sum, 16); // The last 16 bytes are the file checksum } } void CL_PrepareDownloadSaveGame(const char *tmpsave) { #ifndef NONET lastfilenum = -1; #endif fileneedednum = 1; fileneeded[0].status = FS_REQUESTED; fileneeded[0].justdownloaded = false; fileneeded[0].totalsize = UINT32_MAX; fileneeded[0].file = NULL; memset(fileneeded[0].md5sum, 0, 16); strcpy(fileneeded[0].filename, tmpsave); } /** Checks the server to see if we CAN download all the files, * before starting to create them and requesting. * * \return True if we can download all the files * */ boolean CL_CheckDownloadable(void) { UINT8 i,dlstatus = 0; for (i = 0; i < fileneedednum; i++) if (fileneeded[i].status != FS_FOUND && fileneeded[i].status != FS_OPEN) { if (fileneeded[i].willsend == 1) continue; if (fileneeded[i].willsend == 0) dlstatus = 1; else //if (fileneeded[i].willsend == 2) dlstatus = 2; } // Downloading locally disabled if (!dlstatus && M_CheckParm("-nodownload")) dlstatus = 3; if (!dlstatus) return true; // not downloadable, put reason in console CONS_Alert(CONS_NOTICE, M_GetText("You need additional files to connect to this server:\n")); for (i = 0; i < fileneedednum; i++) if (fileneeded[i].status != FS_FOUND && fileneeded[i].status != FS_OPEN) { CONS_Printf(" * \"%s\" (%dK)", fileneeded[i].filename, fileneeded[i].totalsize >> 10); if (fileneeded[i].status == FS_NOTFOUND) CONS_Printf(M_GetText(" not found, md5: ")); else if (fileneeded[i].status == FS_MD5SUMBAD) CONS_Printf(M_GetText(" wrong version, md5: ")); { INT32 j; char md5tmp[33]; for (j = 0; j < 16; j++) sprintf(&md5tmp[j*2], "%02x", fileneeded[i].md5sum[j]); CONS_Printf("%s", md5tmp); } CONS_Printf("\n"); } switch (dlstatus) { case 1: CONS_Printf(M_GetText("Some files are larger than the server is willing to send.\n")); break; case 2: CONS_Printf(M_GetText("The server is not allowing download requests.\n")); break; case 3: CONS_Printf(M_GetText("All files downloadable, but you have chosen to disable downloading locally.\n")); break; } return false; } /** Returns true if a needed file transfer can be resumed * * \param file The needed file to resume the transfer for * \return True if the transfer can be resumed * */ static boolean CL_CanResumeDownload(fileneeded_t *file) { return pauseddownload && !strcmp(pauseddownload->filename, file->filename) // Same name && !memcmp(pauseddownload->md5sum, file->md5sum, 16) // Same checksum && pauseddownload->fragmentsize == file->fragmentsize; // Same fragment size } void CL_AbortDownloadResume(void) { if (!pauseddownload) return; free(pauseddownload->receivedfragments); remove(pauseddownload->filename); free(pauseddownload); pauseddownload = NULL; } /** Sends requests for files in the ::fileneeded table with a status of * ::FS_NOTFOUND. * * \return True if the packet was successfully sent * \note Sends a PT_REQUESTFILE packet * */ boolean CL_SendFileRequest(void) { char *p; INT32 i; INT64 totalfreespaceneeded = 0, availablefreespace; #ifdef PARANOIA if (M_CheckParm("-nodownload")) I_Error("Attempted to download files in -nodownload mode"); for (i = 0; i < fileneedednum; i++) if (fileneeded[i].status != FS_FOUND && fileneeded[i].status != FS_OPEN && (fileneeded[i].willsend == 0 || fileneeded[i].willsend == 2)) { I_Error("Attempted to download files that were not sendable"); } #endif netbuffer->packettype = PT_REQUESTFILE; p = (char *)netbuffer->u.textcmd; for (i = 0; i < fileneedednum; i++) if ((fileneeded[i].status == FS_NOTFOUND || fileneeded[i].status == FS_MD5SUMBAD)) { totalfreespaceneeded += fileneeded[i].totalsize; nameonly(fileneeded[i].filename); WRITEUINT8(p, i); // fileid WRITESTRINGN(p, fileneeded[i].filename, MAX_WADPATH); // put it in download dir strcatbf(fileneeded[i].filename, downloaddir, "/"); fileneeded[i].status = FS_REQUESTED; } WRITEUINT8(p, 0xFF); I_GetDiskFreeSpace(&availablefreespace); if (totalfreespaceneeded > availablefreespace) I_Error("To play on this server you must download %s KB,\n" "but you have only %s KB free space on this drive\n", sizeu1((size_t)(totalfreespaceneeded>>10)), sizeu2((size_t)(availablefreespace>>10))); // prepare to download I_mkdir(downloaddir, 0755); return HSendPacket(servernode, true, 0, p - (char *)netbuffer->u.textcmd); } // get request filepak and put it on the send queue // returns false if a requested file was not found or cannot be sent boolean PT_RequestFile(INT32 node) { char wad[MAX_WADPATH+1]; UINT8 *p = netbuffer->u.textcmd; UINT8 id; while (p < netbuffer->u.textcmd + MAXTEXTCMD-1) // Don't allow hacked client to overflow { id = READUINT8(p); if (id == 0xFF) break; READSTRINGN(p, wad, MAX_WADPATH); if (!AddFileToSendQueue(node, wad, id)) { SV_AbortSendFiles(node); return false; // don't read the rest of the files } } return true; // no problems with any files } /** Checks if the files needed aren't already loaded or on the disk * * \return 0 if some files are missing * 1 if all files exist * 2 if some already loaded files are not requested or are in a different order * */ INT32 CL_CheckFiles(void) { INT32 i, j; char wadfilename[MAX_WADPATH]; INT32 ret = 1; size_t packetsize = 0; size_t filestoget = 0; // if (M_CheckParm("-nofiles")) // return 1; // the first is the iwad (the main wad file) // we don't care if it's called srb2.pk3 or not. // Never download the IWAD, just assume it's there and identical fileneeded[0].status = FS_OPEN; // Modified game handling -- check for an identical file list // must be identical in files loaded AND in order // Return 2 on failure -- disconnect from server if (modifiedgame) { CONS_Debug(DBG_NETPLAY, "game is modified; only doing basic checks\n"); for (i = 1, j = 1; i < fileneedednum || j < numwadfiles;) { if (j < numwadfiles && !wadfiles[j]->important) { // Unimportant on our side. ++j; continue; } // If this test is true, we've reached the end of one file list // and the other still has a file that's important if (i >= fileneedednum || j >= numwadfiles) return 2; // For the sake of speed, only bother with a md5 check if (memcmp(wadfiles[j]->md5sum, fileneeded[i].md5sum, 16)) return 2; // It's accounted for! let's keep going. CONS_Debug(DBG_NETPLAY, "'%s' accounted for\n", fileneeded[i].filename); fileneeded[i].status = FS_OPEN; ++i; ++j; } return 1; } // See W_LoadWadFile in w_wad.c packetsize = packetsizetally; for (i = 1; i < fileneedednum; i++) { CONS_Debug(DBG_NETPLAY, "searching for '%s' ", fileneeded[i].filename); // Check in already loaded files for (j = 1; wadfiles[j]; j++) { nameonly(strcpy(wadfilename, wadfiles[j]->filename)); if (!stricmp(wadfilename, fileneeded[i].filename) && !memcmp(wadfiles[j]->md5sum, fileneeded[i].md5sum, 16)) { CONS_Debug(DBG_NETPLAY, "already loaded\n"); fileneeded[i].status = FS_OPEN; break; } } if (fileneeded[i].status != FS_NOTFOUND) continue; packetsize += nameonlylength(fileneeded[i].filename) + 22; if ((numwadfiles+filestoget >= MAX_WADFILES) || (packetsize > MAXFILENEEDED*sizeof(UINT8))) return 3; filestoget++; fileneeded[i].status = findfile(fileneeded[i].filename, fileneeded[i].md5sum, true); CONS_Debug(DBG_NETPLAY, "found %d\n", fileneeded[i].status); if (fileneeded[i].status != FS_FOUND) ret = 0; } return ret; } // Load it now void CL_LoadServerFiles(void) { INT32 i; // if (M_CheckParm("-nofiles")) // return; for (i = 1; i < fileneedednum; i++) { if (fileneeded[i].status == FS_OPEN) continue; // Already loaded else if (fileneeded[i].status == FS_FOUND) { P_AddWadFile(fileneeded[i].filename); G_SetGameModified(true); fileneeded[i].status = FS_OPEN; } else if (fileneeded[i].status == FS_MD5SUMBAD) I_Error("Wrong version of file %s", fileneeded[i].filename); else { const char *s; switch(fileneeded[i].status) { case FS_NOTFOUND: s = "FS_NOTFOUND"; break; case FS_REQUESTED: s = "FS_REQUESTED"; break; case FS_DOWNLOADING: s = "FS_DOWNLOADING"; break; default: s = "unknown"; break; } I_Error("Try to load file \"%s\" with status of %d (%s)\n", fileneeded[i].filename, fileneeded[i].status, s); } } } void AddLuaFileTransfer(const char *filename, const char *mode) { luafiletransfer_t **prevnext; // A pointer to the "next" field of the last transfer in the list luafiletransfer_t *filetransfer; static INT32 id; // Find the last transfer in the list and set a pointer to its "next" field prevnext = &luafiletransfers; while (*prevnext) prevnext = &((*prevnext)->next); // Allocate file transfer information and append it to the transfer list filetransfer = malloc(sizeof(luafiletransfer_t)); if (!filetransfer) I_Error("AddLuaFileTransfer: Out of memory\n"); *prevnext = filetransfer; filetransfer->next = NULL; // Allocate the file name filetransfer->filename = strdup(filename); if (!filetransfer->filename) I_Error("AddLuaFileTransfer: Out of memory\n"); // Create and allocate the real file name if (server) filetransfer->realfilename = strdup(va("%s" PATHSEP "%s", luafiledir, filename)); else filetransfer->realfilename = strdup(va("%s" PATHSEP "client" PATHSEP "$$$%d%d.tmp", luafiledir, rand(), rand())); if (!filetransfer->realfilename) I_Error("AddLuaFileTransfer: Out of memory\n"); strlcpy(filetransfer->mode, mode, sizeof(filetransfer->mode)); // Only if there is no transfer already going on if (server && filetransfer == luafiletransfers) SV_PrepareSendLuaFile(); else filetransfer->ongoing = false; // Store the callback so it can be called once everyone has the file filetransfer->id = id; StoreLuaFileCallback(id); id++; if (waitingforluafiletransfer) { waitingforluafiletransfer = false; CL_PrepareDownloadLuaFile(); } } static void SV_PrepareSendLuaFileToNextNode(void) { INT32 i; UINT8 success = 1; // Find a client to send the file to for (i = 1; i < MAXNETNODES; i++) if (nodeingame[i] && luafiletransfers->nodestatus[i] == LFTNS_WAITING) // Node waiting { // Tell the client we're about to send them the file netbuffer->packettype = PT_SENDINGLUAFILE; if (!HSendPacket(i, true, 0, 0)) I_Error("Failed to send a PT_SENDINGLUAFILE packet\n"); // !!! Todo: Handle failure a bit better lol luafiletransfers->nodestatus[i] = LFTNS_ASKED; return; } // No client found, everyone has the file // Send a net command with 1 as its first byte to indicate the file could be opened SendNetXCmd(XD_LUAFILE, &success, 1); } void SV_PrepareSendLuaFile(void) { char *binfilename; INT32 i; luafiletransfers->ongoing = true; // Set status to "waiting" for everyone for (i = 0; i < MAXNETNODES; i++) luafiletransfers->nodestatus[i] = LFTNS_WAITING; if (FIL_ReadFileOK(luafiletransfers->realfilename)) { // If opening in text mode, convert all newlines to LF if (!strchr(luafiletransfers->mode, 'b')) { binfilename = strdup(va("%s" PATHSEP "$$$%d%d.tmp", luafiledir, rand(), rand())); if (!binfilename) I_Error("SV_PrepareSendLuaFile: Out of memory\n"); if (!FIL_ConvertTextFileToBinary(luafiletransfers->realfilename, binfilename)) I_Error("SV_PrepareSendLuaFile: Failed to convert file newlines\n"); // Use the temporary file instead free(luafiletransfers->realfilename); luafiletransfers->realfilename = binfilename; } SV_PrepareSendLuaFileToNextNode(); } else { // Send a net command with 0 as its first byte to indicate the file couldn't be opened UINT8 success = 0; SendNetXCmd(XD_LUAFILE, &success, 1); } } void SV_HandleLuaFileSent(UINT8 node) { luafiletransfers->nodestatus[node] = LFTNS_SENT; SV_PrepareSendLuaFileToNextNode(); } void RemoveLuaFileTransfer(void) { luafiletransfer_t *filetransfer = luafiletransfers; // If it was a temporary file, delete it if (server && !strchr(filetransfer->mode, 'b')) remove(filetransfer->realfilename); RemoveLuaFileCallback(filetransfer->id); luafiletransfers = filetransfer->next; free(filetransfer->filename); free(filetransfer->realfilename); free(filetransfer); } void RemoveAllLuaFileTransfers(void) { while (luafiletransfers) RemoveLuaFileTransfer(); } void SV_AbortLuaFileTransfer(INT32 node) { if (luafiletransfers && (luafiletransfers->nodestatus[node] == LFTNS_ASKED || luafiletransfers->nodestatus[node] == LFTNS_SENDING)) { luafiletransfers->nodestatus[node] = LFTNS_WAITING; SV_PrepareSendLuaFileToNextNode(); } } void CL_PrepareDownloadLuaFile(void) { // If there is no transfer in the list, this normally means the server // called io.open before us, so we have to wait until we call it too if (!luafiletransfers) { waitingforluafiletransfer = true; return; } if (luafiletransfers->ongoing) { waitingforluafilecommand = true; return; } // Tell the server we are ready to receive the file netbuffer->packettype = PT_ASKLUAFILE; HSendPacket(servernode, true, 0, 0); fileneedednum = 1; fileneeded[0].status = FS_REQUESTED; fileneeded[0].justdownloaded = false; fileneeded[0].totalsize = UINT32_MAX; fileneeded[0].file = NULL; memset(fileneeded[0].md5sum, 0, 16); strcpy(fileneeded[0].filename, luafiletransfers->realfilename); // Make sure all directories in the file path exist MakePathDirs(fileneeded[0].filename); luafiletransfers->ongoing = true; } // Number of files to send // Little optimization to quickly test if there is a file in the queue static INT32 filestosend = 0; /** Adds a file to the file list for a node * * \param node The node to send the file to * \param filename The file to send * \param fileid The index of the file in the list of added files * \sa AddRamToSendQueue * \sa AddLuaFileToSendQueue * */ static boolean AddFileToSendQueue(INT32 node, const char *filename, UINT8 fileid) { filetx_t **q; // A pointer to the "next" field of the last file in the list filetx_t *p; // The new file request INT32 i; char wadfilename[MAX_WADPATH]; if (cv_noticedownload.value) CONS_Printf("Sending file \"%s\" to node %d (%s)\n", filename, node, I_GetNodeAddress(node)); // Find the last file in the list and set a pointer to its "next" field q = &transfer[node].txlist; while (*q) q = &((*q)->next); // Allocate a file request and append it to the file list p = *q = (filetx_t *)malloc(sizeof (filetx_t)); if (!p) I_Error("AddFileToSendQueue: No more memory\n"); // Initialise with zeros memset(p, 0, sizeof (filetx_t)); // Allocate the file name p->id.filename = (char *)malloc(MAX_WADPATH); if (!p->id.filename) I_Error("AddFileToSendQueue: No more memory\n"); // Set the file name and get rid of the path strlcpy(p->id.filename, filename, MAX_WADPATH); nameonly(p->id.filename); // Look for the requested file through all loaded files for (i = 0; wadfiles[i]; i++) { strlcpy(wadfilename, wadfiles[i]->filename, MAX_WADPATH); nameonly(wadfilename); if (!stricmp(wadfilename, p->id.filename)) { // Copy file name with full path strlcpy(p->id.filename, wadfiles[i]->filename, MAX_WADPATH); break; } } // Handle non-loaded file requests if (!wadfiles[i]) { DEBFILE(va("%s not found in wadfiles\n", filename)); // This formerly checked if (!findfile(p->id.filename, NULL, true)) // Not found // Don't inform client (probably someone who thought they could leak 2.2 ACZ) DEBFILE(va("Client %d request %s: not found\n", node, filename)); free(p->id.filename); free(p); *q = NULL; return false; // cancel the rest of the requests } // Handle huge file requests (i.e. bigger than cv_maxsend.value KB) if (wadfiles[i]->filesize > (UINT32)cv_maxsend.value * 1024) { // Too big // Don't inform client (client sucks, man) DEBFILE(va("Client %d request %s: file too big, not sending\n", node, filename)); free(p->id.filename); free(p); *q = NULL; return false; // cancel the rest of the requests } DEBFILE(va("Sending file %s (id=%d) to %d\n", filename, fileid, node)); p->ram = SF_FILE; // It's a file, we need to close it and free its name once we're done sending it p->fileid = fileid; p->next = NULL; // End of list filestosend++; return true; } /** Adds a memory block to the file list for a node * * \param node The node to send the memory block to * \param data The memory block to send * \param size The size of the block in bytes * \param freemethod How to free the block after it has been sent * \param fileid The index of the file in the list of added files * \sa AddFileToSendQueue * \sa AddLuaFileToSendQueue * */ void AddRamToSendQueue(INT32 node, void *data, size_t size, freemethod_t freemethod, UINT8 fileid) { filetx_t **q; // A pointer to the "next" field of the last file in the list filetx_t *p; // The new file request // Find the last file in the list and set a pointer to its "next" field q = &transfer[node].txlist; while (*q) q = &((*q)->next); // Allocate a file request and append it to the file list p = *q = (filetx_t *)malloc(sizeof (filetx_t)); if (!p) I_Error("AddRamToSendQueue: No more memory\n"); // Initialise with zeros memset(p, 0, sizeof (filetx_t)); p->ram = freemethod; // Remember how to free the memory block for when we're done sending it p->id.ram = data; p->size = (UINT32)size; p->fileid = fileid; p->next = NULL; // End of list DEBFILE(va("Sending ram %p(size:%u) to %d (id=%u)\n",p->id.ram,p->size,node,fileid)); filestosend++; } /** Adds a file requested by Lua to the file list for a node * * \param node The node to send the file to * \param filename The file to send * \sa AddFileToSendQueue * \sa AddRamToSendQueue * */ boolean AddLuaFileToSendQueue(INT32 node, const char *filename) { filetx_t **q; // A pointer to the "next" field of the last file in the list filetx_t *p; // The new file request //INT32 i; //char wadfilename[MAX_WADPATH]; luafiletransfers->nodestatus[node] = LFTNS_SENDING; // Find the last file in the list and set a pointer to its "next" field q = &transfer[node].txlist; while (*q) q = &((*q)->next); // Allocate a file request and append it to the file list p = *q = (filetx_t *)malloc(sizeof (filetx_t)); if (!p) I_Error("AddLuaFileToSendQueue: No more memory\n"); // Initialise with zeros memset(p, 0, sizeof (filetx_t)); // Allocate the file name p->id.filename = (char *)malloc(MAX_WADPATH); // !!! if (!p->id.filename) I_Error("AddLuaFileToSendQueue: No more memory\n"); // Set the file name and get rid of the path strlcpy(p->id.filename, filename, MAX_WADPATH); // !!! //nameonly(p->id.filename); DEBFILE(va("Sending Lua file %s to %d\n", filename, node)); p->ram = SF_FILE; // It's a file, we need to close it and free its name once we're done sending it p->next = NULL; // End of list filestosend++; return true; } /** Stops sending a file for a node, and removes the file request from the list, * either because the file has been fully sent or because the node was disconnected * * \param node The destination * */ static void SV_EndFileSend(INT32 node) { filetx_t *p = transfer[node].txlist; // Free the file request according to the freemethod // parameter used with AddFileToSendQueue/AddRamToSendQueue switch (p->ram) { case SF_FILE: // It's a file, close it and free its filename if (cv_noticedownload.value) CONS_Printf("Ending file transfer for node %d\n", node); if (transfer[node].currentfile) fclose(transfer[node].currentfile); free(p->id.filename); break; case SF_Z_RAM: // It's a memory block allocated with Z_Alloc or the likes, use Z_Free Z_Free(p->id.ram); break; case SF_RAM: // It's a memory block allocated with malloc, use free free(p->id.ram); case SF_NOFREERAM: // Nothing to free break; } // Remove the file request from the list transfer[node].txlist = p->next; free(p); // Indicate that the transmission is over transfer[node].currentfile = NULL; if (transfer[node].ackedfragments) free(transfer[node].ackedfragments); transfer[node].ackedfragments = NULL; filestosend--; } #define PACKETPERTIC net_bandwidth/(TICRATE*software_MAXPACKETLENGTH) #define FILEFRAGMENTSIZE (software_MAXPACKETLENGTH - (FILETXHEADER + BASEPACKETSIZE)) /** Handles file transmission * */ void FileSendTicker(void) { static INT32 currentnode = 0; filetx_pak *p; size_t fragmentsize; filetx_t *f; INT32 packetsent, ram, i, j; if (!filestosend) // No file to send return; if (cv_downloadspeed.value) // New behavior packetsent = cv_downloadspeed.value; else // Old behavior { packetsent = PACKETPERTIC; if (!packetsent) packetsent = 1; } netbuffer->packettype = PT_FILEFRAGMENT; // (((sendbytes-nowsentbyte)*TICRATE)/(I_GetTime()-starttime)<(UINT32)net_bandwidth) while (packetsent-- && filestosend != 0) { for (i = currentnode, j = 0; j < MAXNETNODES; i = (i+1) % MAXNETNODES, j++) { if (transfer[i].txlist) break; } // no transfer to do if (j >= MAXNETNODES) I_Error("filestosend=%d but no file to send found\n", filestosend); currentnode = (i+1) % MAXNETNODES; f = transfer[i].txlist; ram = f->ram; // Open the file if it isn't open yet, or if (!transfer[i].currentfile) { if (!ram) // Sending a file { long filesize; transfer[i].currentfile = fopen(f->id.filename, "rb"); if (!transfer[i].currentfile) I_Error("File %s does not exist", f->id.filename); fseek(transfer[i].currentfile, 0, SEEK_END); filesize = ftell(transfer[i].currentfile); // Nobody wants to transfer a file bigger // than 4GB! if (filesize >= LONG_MAX) I_Error("filesize of %s is too large", f->id.filename); if (filesize == -1) I_Error("Error getting filesize of %s", f->id.filename); f->size = (UINT32)filesize; fseek(transfer[i].currentfile, 0, SEEK_SET); } else // Sending RAM transfer[i].currentfile = (FILE *)1; // Set currentfile to a non-null value to indicate that it is open transfer[i].iteration = 1; transfer[i].ackediteration = 0; transfer[i].position = 0; transfer[i].ackedsize = 0; transfer[i].ackedfragments = calloc(f->size / FILEFRAGMENTSIZE + 1, sizeof(*transfer[i].ackedfragments)); if (!transfer[i].ackedfragments) I_Error("FileSendTicker: No more memory\n"); transfer[i].dontsenduntil = 0; } // If the client hasn't acknowledged any fragment from the previous iteration, // it is most likely because their acks haven't had enough time to reach the server // yet, due to latency. In that case, we wait a little to avoid useless resend. if (I_GetTime() < transfer[i].dontsenduntil) continue; // Find the first non-acknowledged fragment while (transfer[i].ackedfragments[transfer[i].position / FILEFRAGMENTSIZE]) { transfer[i].position += FILEFRAGMENTSIZE; if (transfer[i].position >= f->size) { if (transfer[i].ackediteration < transfer[i].iteration) transfer[i].dontsenduntil = I_GetTime() + TICRATE / 2; transfer[i].position = 0; transfer[i].iteration++; } } // Build a packet containing a file fragment p = &netbuffer->u.filetxpak; fragmentsize = FILEFRAGMENTSIZE; if (f->size-transfer[i].position < fragmentsize) fragmentsize = f->size-transfer[i].position; if (ram) M_Memcpy(p->data, &f->id.ram[transfer[i].position], fragmentsize); else { fseek(transfer[i].currentfile, transfer[i].position, SEEK_SET); if (fread(p->data, 1, fragmentsize, transfer[i].currentfile) != fragmentsize) I_Error("FileSendTicker: can't read %s byte on %s at %d because %s", sizeu1(fragmentsize), f->id.filename, transfer[i].position, M_FileError(transfer[i].currentfile)); } p->iteration = transfer[i].iteration; p->position = LONG(transfer[i].position); p->fileid = f->fileid; p->filesize = LONG(f->size); p->size = SHORT((UINT16)FILEFRAGMENTSIZE); // Send the packet if (HSendPacket(i, false, 0, FILETXHEADER + fragmentsize)) // Don't use the default acknowledgement system { // Success transfer[i].position = (UINT32)(transfer[i].position + fragmentsize); if (transfer[i].position >= f->size) { if (transfer[i].ackediteration < transfer[i].iteration) transfer[i].dontsenduntil = I_GetTime() + TICRATE / 2; transfer[i].position = 0; transfer[i].iteration++; } } else { // Not sent for some odd reason, retry at next call // Exit the while (can't send this one so why should i send the next?) break; } } } void PT_FileAck(void) { fileack_pak *packet = &netbuffer->u.fileack; INT32 node = doomcom->remotenode; filetran_t *trans = &transfer[node]; INT32 i, j; // Wrong file id? Ignore it, it's probably a late packet if (!(trans->txlist && packet->fileid == trans->txlist->fileid)) return; if (packet->numsegments * sizeof(*packet->segments) != doomcom->datalength - BASEPACKETSIZE - sizeof(*packet)) { Net_CloseConnection(node); return; } if (packet->iteration > trans->ackediteration) { trans->ackediteration = packet->iteration; if (trans->ackediteration >= trans->iteration - 1) trans->dontsenduntil = 0; } for (i = 0; i < packet->numsegments; i++) { fileacksegment_t *segment = &packet->segments[i]; for (j = 0; j < 32; j++) if (LONG(segment->acks) & (1 << j)) { if (LONG(segment->start) * FILEFRAGMENTSIZE >= trans->txlist->size) { Net_CloseConnection(node); return; } if (!trans->ackedfragments[LONG(segment->start) + j]) { trans->ackedfragments[LONG(segment->start) + j] = true; trans->ackedsize += FILEFRAGMENTSIZE; // If the last missing fragment was acked, finish! if (trans->ackedsize == trans->txlist->size) { SV_EndFileSend(node); return; } } } } } void PT_FileReceived(void) { filetx_t *trans = transfer[doomcom->remotenode].txlist; if (trans && netbuffer->u.filereceived == trans->fileid) SV_EndFileSend(doomcom->remotenode); } static void SendAckPacket(fileack_pak *packet, UINT8 fileid) { size_t packetsize; INT32 i; packetsize = sizeof(*packet) + packet->numsegments * sizeof(*packet->segments); // Finalise the packet packet->fileid = fileid; for (i = 0; i < packet->numsegments; i++) { packet->segments[i].start = LONG(packet->segments[i].start); packet->segments[i].acks = LONG(packet->segments[i].acks); } // Send the packet netbuffer->packettype = PT_FILEACK; M_Memcpy(&netbuffer->u.fileack, packet, packetsize); HSendPacket(servernode, false, 0, packetsize); // Clear the packet memset(packet, 0, sizeof(*packet) + 512); } static void AddFragmentToAckPacket(fileack_pak *packet, UINT8 iteration, UINT32 fragmentpos, UINT8 fileid) { fileacksegment_t *segment = &packet->segments[packet->numsegments - 1]; packet->iteration = max(packet->iteration, iteration); if (packet->numsegments == 0 || fragmentpos < segment->start || fragmentpos - segment->start >= 32) { // If the packet becomes too big, send it if ((packet->numsegments + 1) * sizeof(*segment) > 512) SendAckPacket(packet, fileid); packet->numsegments++; segment = &packet->segments[packet->numsegments - 1]; segment->start = fragmentpos; } // Set the bit that represents the fragment segment->acks |= 1 << (fragmentpos - segment->start); } void FileReceiveTicker(void) { INT32 i; for (i = 0; i < fileneedednum; i++) { fileneeded_t *file = &fileneeded[i]; if (file->status == FS_DOWNLOADING) { if (lasttimeackpacketsent - I_GetTime() > TICRATE / 2) SendAckPacket(file->ackpacket, i); // When resuming a tranfer, start with telling // the server what parts we already received if (file->ackresendposition != UINT32_MAX && file->status == FS_DOWNLOADING) { // Acknowledge ~70 MB/s, whichs means the client sends ~18 KB/s INT32 j; for (j = 0; j < 2048; j++) { if (file->receivedfragments[file->ackresendposition]) AddFragmentToAckPacket(file->ackpacket, file->iteration, file->ackresendposition, i); file->ackresendposition++; if (file->ackresendposition * file->fragmentsize >= file->totalsize) { file->ackresendposition = UINT32_MAX; break; } } } } } } void PT_FileFragment(void) { INT32 filenum = netbuffer->u.filetxpak.fileid; fileneeded_t *file = &fileneeded[filenum]; UINT32 fragmentpos = LONG(netbuffer->u.filetxpak.position); UINT16 fragmentsize = SHORT(netbuffer->u.filetxpak.size); UINT16 boundedfragmentsize = doomcom->datalength - BASEPACKETSIZE - sizeof(netbuffer->u.filetxpak); char *filename; filename = va("%s", file->filename); nameonly(filename); if (!(strcmp(filename, "srb2.pk3") && strcmp(filename, "zones.pk3") && strcmp(filename, "player.dta") && strcmp(filename, "patch.pk3") && strcmp(filename, "music.dta") )) I_Error("Tried to download \"%s\"", filename); filename = file->filename; if (filenum >= fileneedednum) { DEBFILE(va("fileframent not needed %d>%d\n", filenum, fileneedednum)); //I_Error("Received an unneeded file fragment (file id received: %d, file id needed: %d)\n", filenum, fileneedednum); return; } if (file->status == FS_REQUESTED) { if (file->file) I_Error("PT_FileFragment: already open file\n"); file->status = FS_DOWNLOADING; file->fragmentsize = fragmentsize; file->iteration = 0; file->ackpacket = calloc(1, sizeof(*file->ackpacket) + 512); if (!file->ackpacket) I_Error("FileSendTicker: No more memory\n"); if (CL_CanResumeDownload(file)) { file->file = fopen(filename, "r+b"); if (!file->file) I_Error("Can't reopen file %s: %s", filename, strerror(errno)); CONS_Printf("\r%s...\n", filename); CONS_Printf("Resuming download...\n"); file->currentsize = pauseddownload->currentsize; file->receivedfragments = pauseddownload->receivedfragments; file->ackresendposition = 0; free(pauseddownload); pauseddownload = NULL; } else { CL_AbortDownloadResume(); file->file = fopen(filename, "wb"); if (!file->file) I_Error("Can't create file %s: %s", filename, strerror(errno)); CONS_Printf("\r%s...\n",filename); file->currentsize = 0; file->totalsize = LONG(netbuffer->u.filetxpak.filesize); file->ackresendposition = UINT32_MAX; // Only used for resumed downloads file->receivedfragments = calloc(file->totalsize / fragmentsize + 1, sizeof(*file->receivedfragments)); if (!file->receivedfragments) I_Error("FileSendTicker: No more memory\n"); } lasttimeackpacketsent = I_GetTime(); } if (file->status == FS_DOWNLOADING) { if (fragmentpos >= file->totalsize) I_Error("Invalid file fragment\n"); file->iteration = max(file->iteration, netbuffer->u.filetxpak.iteration); if (!file->receivedfragments[fragmentpos / fragmentsize]) // Not received yet { file->receivedfragments[fragmentpos / fragmentsize] = true; // We can receive packets in the wrong order, anyway all OSes support gaped files fseek(file->file, fragmentpos, SEEK_SET); if (fragmentsize && fwrite(netbuffer->u.filetxpak.data, boundedfragmentsize, 1, file->file) != 1) I_Error("Can't write to %s: %s\n",filename, M_FileError(file->file)); file->currentsize += boundedfragmentsize; AddFragmentToAckPacket(file->ackpacket, file->iteration, fragmentpos / fragmentsize, filenum); // Finished? if (file->currentsize == file->totalsize) { fclose(file->file); file->file = NULL; free(file->receivedfragments); free(file->ackpacket); file->status = FS_FOUND; file->justdownloaded = true; CONS_Printf(M_GetText("Downloading %s...(done)\n"), filename); // Tell the server we have received the file netbuffer->packettype = PT_FILERECEIVED; netbuffer->u.filereceived = filenum; HSendPacket(servernode, true, 0, 1); if (luafiletransfers) { // Tell the server we have received the file netbuffer->packettype = PT_HASLUAFILE; HSendPacket(servernode, true, 0, 0); } } } else // Already received { // If they are sending us the fragment again, it's probably because // they missed our previous ack, so we must re-acknowledge it AddFragmentToAckPacket(file->ackpacket, file->iteration, fragmentpos / fragmentsize, filenum); } } else if (!file->justdownloaded) { const char *s; switch(file->status) { case FS_NOTFOUND: s = "FS_NOTFOUND"; break; case FS_FOUND: s = "FS_FOUND"; break; case FS_OPEN: s = "FS_OPEN"; break; case FS_MD5SUMBAD: s = "FS_MD5SUMBAD"; break; default: s = "unknown"; break; } I_Error("Received a file not requested (file id: %d, file status: %s)\n", filenum, s); } #ifndef NONET lastfilenum = filenum; #endif } /** \brief Checks if a node is downloading a file * * \param node The node to check for * \return True if the node is downloading a file * */ boolean SendingFile(INT32 node) { return transfer[node].txlist != NULL; } /** Cancels all file requests for a node * * \param node The destination * \sa SV_EndFileSend * */ void SV_AbortSendFiles(INT32 node) { while (transfer[node].txlist) SV_EndFileSend(node); } void CloseNetFile(void) { INT32 i; // Is sending? for (i = 0; i < MAXNETNODES; i++) SV_AbortSendFiles(i); // Receiving a file? for (i = 0; i < MAX_WADFILES; i++) if (fileneeded[i].status == FS_DOWNLOADING && fileneeded[i].file) { fclose(fileneeded[i].file); free(fileneeded[i].ackpacket); if (!pauseddownload && i != 0) // 0 is either srb2.srb or the gamestate... { // Don't remove the file, save it for later in case we resume the download pauseddownload = malloc(sizeof(*pauseddownload)); if (!pauseddownload) I_Error("CloseNetFile: No more memory\n"); strcpy(pauseddownload->filename, fileneeded[i].filename); memcpy(pauseddownload->md5sum, fileneeded[i].md5sum, 16); pauseddownload->currentsize = fileneeded[i].currentsize; pauseddownload->receivedfragments = fileneeded[i].receivedfragments; pauseddownload->fragmentsize = fileneeded[i].fragmentsize; } else { free(fileneeded[i].receivedfragments); // File is not complete delete it remove(fileneeded[i].filename); } } } void Command_Downloads_f(void) { INT32 node; for (node = 0; node < MAXNETNODES; node++) if (transfer[node].txlist && transfer[node].txlist->ram == SF_FILE) // Node is downloading a file? { const char *name = transfer[node].txlist->id.filename; UINT32 position = transfer[node].ackedsize; UINT32 size = transfer[node].txlist->size; char ratecolor; // Avoid division by zero errors if (!size) size = 1; name = &name[strlen(name) - nameonlylength(name)]; switch (4 * (position - 1) / size) { case 0: ratecolor = '\x85'; break; case 1: ratecolor = '\x87'; break; case 2: ratecolor = '\x82'; break; case 3: ratecolor = '\x83'; break; default: ratecolor = '\x80'; } CONS_Printf("%2d %c%s ", node, ratecolor, name); // Node and file name CONS_Printf("\x80%uK\x84/\x80%uK ", position / 1024, size / 1024); // Progress in kB CONS_Printf("\x80(%c%u%%\x80) ", ratecolor, (UINT32)(100.0 * position / size)); // Progress in % CONS_Printf("%s\n", I_GetNodeAddress(node)); // Address and newline } } // Functions cut and pasted from Doomatic :) void nameonly(char *s) { size_t j, len; void *ns; for (j = strlen(s); j != (size_t)-1; j--) if ((s[j] == '\\') || (s[j] == ':') || (s[j] == '/')) { ns = &(s[j+1]); len = strlen(ns); #if 0 M_Memcpy(s, ns, len+1); #else memmove(s, ns, len+1); #endif return; } } // Returns the length in characters of the last element of a path. size_t nameonlylength(const char *s) { size_t j, len = strlen(s); for (j = len; j != (size_t)-1; j--) if ((s[j] == '\\') || (s[j] == ':') || (s[j] == '/')) return len - j - 1; return len; } #ifndef O_BINARY #define O_BINARY 0 #endif filestatus_t checkfilemd5(char *filename, const UINT8 *wantedmd5sum) { #if defined (NOMD5) (void)wantedmd5sum; (void)filename; #else FILE *fhandle; UINT8 md5sum[16]; if (!wantedmd5sum) return FS_FOUND; fhandle = fopen(filename, "rb"); if (fhandle) { md5_stream(fhandle,md5sum); fclose(fhandle); if (!memcmp(wantedmd5sum, md5sum, 16)) return FS_FOUND; return FS_MD5SUMBAD; } I_Error("Couldn't open %s for md5 check", filename); #endif return FS_FOUND; // will never happen, but makes the compiler shut up } // Rewritten by Monster Iestyn to be less stupid // Note: if completepath is true, "filename" is modified, but only if FS_FOUND is going to be returned // (Don't worry about WinCE's version of filesearch, nobody cares about that OS anymore) filestatus_t findfile(char *filename, const UINT8 *wantedmd5sum, boolean completepath) { filestatus_t homecheck; // store result of last file search boolean badmd5 = false; // store whether md5 was bad from either of the first two searches (if nothing was found in the third) // first, check SRB2's "home" directory homecheck = filesearch(filename, srb2home, wantedmd5sum, completepath, 10); if (homecheck == FS_FOUND) // we found the file, so return that we have :) return FS_FOUND; else if (homecheck == FS_MD5SUMBAD) // file has a bad md5; move on and look for a file with the right md5 badmd5 = true; // if not found at all, just move on without doing anything // next, check SRB2's "path" directory homecheck = filesearch(filename, srb2path, wantedmd5sum, completepath, 10); if (homecheck == FS_FOUND) // we found the file, so return that we have :) return FS_FOUND; else if (homecheck == FS_MD5SUMBAD) // file has a bad md5; move on and look for a file with the right md5 badmd5 = true; // if not found at all, just move on without doing anything // finally check "." directory homecheck = filesearch(filename, ".", wantedmd5sum, completepath, 10); if (homecheck != FS_NOTFOUND) // if not found this time, fall back on the below return statement return homecheck; // otherwise return the result we got return (badmd5 ? FS_MD5SUMBAD : FS_NOTFOUND); // md5 sum bad or file not found }