Eveldee/DearFTP
1
1
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity

Add APPE and STOU command

Browse Source
This commit is contained in:
Eveldee
2019-07-18 13:18:15 +02:00
parent a2f867480a
commit 8c2b32d6b1
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
DearFTP/Connection/Commands/StoreCommand.cs
View File

@@ -13,7 +13,9 @@ namespace DearFTP.Connection.Commands
public string[] Aliases { get; } = new string[] public string[] Aliases { get; } = new string[]
{ {
"STOR" "APPE",
"STOR",
"STOU"
}; };
public void Execute(Session session, FtpStream stream, string alias, string argument) public void Execute(Session session, FtpStream stream, string alias, string argument)
@@ -26,7 +28,9 @@ namespace DearFTP.Connection.Commands
return; return;
} }
var (navigablePath, realPath) = session.NavigablePath.GetNewSystemFilePath(argument); string path = alias.ToUpper() == "STOU" ? Path.GetRandomFileName() : argument;
var (navigablePath, realPath) = session.NavigablePath.GetNewSystemFilePath(path);
if (realPath == null) if (realPath == null)
{ {
@@ -48,16 +52,16 @@ namespace DearFTP.Connection.Commands
stream.Send(ResponseCode.FileStatusOK, "Waiting file."); stream.Send(ResponseCode.FileStatusOK, "Waiting file.");
ReceiveFile(dataConnection.Stream, realPath); ReceiveFile(dataConnection.Stream, realPath, alias.ToUpper() == "APPE");
dataConnection.Close(); dataConnection.Close();
stream.Send(ResponseCode.CloseDataConnection, "File received."); stream.Send(ResponseCode.CloseDataConnection, "File received.");
} }
private void ReceiveFile(NetworkStream stream, string path) private void ReceiveFile(NetworkStream stream, string path, bool append)
{ {
using (var file = File.Open(path, FileMode.Create)) using (var file = File.Open(path, append ? FileMode.Append : FileMode.Create))
{ {
Span<byte> buffer = stackalloc byte[BufferSize]; Span<byte> buffer = stackalloc byte[BufferSize];
int readBytes; int readBytes;