Timerix/DTLib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fsp bug fix

Browse Source
This commit is contained in:
Timerix22 2024-01-06 03:47:04 +06:00
parent 0275ef1bb1
commit 96feda06e1
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
DTLib.Network/FSP.cs
View File

@ -37,7 +37,7 @@ public class FSP
{
filePath_client.ThrowIfEscapes();
using System.IO.Stream fileStream = File.OpenWrite(filePath_client);
Download_SharedCode(fileStream);
DownloadToStream(fileStream);
}
public byte[] DownloadFileToMemory(IOPath filePath_server)
@ -54,11 +54,11 @@ public class FSP
public byte[] DownloadFileToMemory()
{
using var fileStream = new System.IO.MemoryStream();
Download_SharedCode(fileStream);
DownloadToStream(fileStream);
return fileStream.ToArray();
}
private void Download_SharedCode(System.IO.Stream fileStream)
public void DownloadToStream(System.IO.Stream fileStream)
{
lock (MainSocket)
{
@ -67,13 +67,13 @@ public class FSP
if (Filesize < 0)
throw new Exception("FileSize < 0");
MainSocket.SendPackage("ready");
int recievedCount;
do
while (BytesDownloaded < Filesize)
{
recievedCount = MainSocket.Receive(buffer);
int recievedCount = MainSocket.Receive(buffer);
fileStream.Write(buffer, 0, recievedCount);
BytesDownloaded += recievedCount;
} while (recievedCount == buffer.Length);
}
if (BytesDownloaded != Filesize)
throw new Exception($"expected {Filesize} bytes, but downloaded {BytesDownloaded} bytes");
@ -91,13 +91,13 @@ public class FSP
{
MainSocket.SendPackage(BitConverter.GetBytes(Filesize));
MainSocket.GetAnswer("ready");
int readCount;
do
while (BytesUploaded < Filesize)
{
readCount = fileStream.Read(buffer, 0, buffer.Length);
int readCount = fileStream.Read(buffer, 0, buffer.Length);
MainSocket.Send(buffer, 0, readCount, SocketFlags.None);
BytesUploaded += readCount;
} while (readCount == buffer.Length);
}
if (BytesUploaded != Filesize)
throw new Exception($"expected {Filesize} bytes, but uploaded {BytesDownloaded} bytes");