mirror of
https://github.com/sheumann/NetDisk.git
synced 2024-11-27 19:49:38 +00:00
Work around Marinetti bug 57.
This could cause Marinetti to return more data than it should, starting with valid data but then including corrupt data. The workaround I'm using is to have Marinetti return a new handle and then copy the data out of it, using the size of the handle as the true size of the data that was read. Because of details of how Marinetti works, that size is correct. This shouldn't have a major performance impact, because Marinetti would internally read data into a new handle and then copy it anyway.
This commit is contained in:
parent
08fe7fe4b6
commit
f524630f1a
18
readtcp.c
18
readtcp.c
@ -2,7 +2,9 @@
|
||||
|
||||
#include "readtcp.h"
|
||||
#include "session.h"
|
||||
#include <string.h>
|
||||
#include <tcpip.h>
|
||||
#include <memory.h>
|
||||
#include <misctool.h>
|
||||
#include <orca.h>
|
||||
|
||||
@ -24,15 +26,25 @@ ReadStatus TryReadTCP(Session *sess) {
|
||||
rrBuff rrBuff;
|
||||
|
||||
TCPIPPoll();
|
||||
sess->tcperr = TCPIPReadTCP(sess->ipid, buffTypePointer, (Ref)sess->readPtr,
|
||||
sess->tcperr = TCPIPReadTCP(sess->ipid, buffTypeNewHandle, NULL,
|
||||
sess->readCount, &rrBuff);
|
||||
sess->toolerr = toolerror();
|
||||
if (sess->tcperr || sess->toolerr) {
|
||||
return rsError;
|
||||
}
|
||||
|
||||
sess->readCount -= rrBuff.rrBuffCount;
|
||||
sess->readPtr += rrBuff.rrBuffCount;
|
||||
if (rrBuff.rrBuffCount != 0) {
|
||||
/* Work around Marinetti bug #57 */
|
||||
rrBuff.rrBuffCount = GetHandleSize(rrBuff.rrBuffHandle);
|
||||
|
||||
HLock(rrBuff.rrBuffHandle);
|
||||
memcpy(sess->readPtr, *rrBuff.rrBuffHandle, rrBuff.rrBuffCount);
|
||||
DisposeHandle(rrBuff.rrBuffHandle);
|
||||
|
||||
sess->readCount -= rrBuff.rrBuffCount;
|
||||
sess->readPtr += rrBuff.rrBuffCount;
|
||||
}
|
||||
|
||||
|
||||
if (sess->readCount == 0) {
|
||||
return rsDone;
|
||||
|
Loading…
Reference in New Issue
Block a user