brouzouf/ftp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for append (APPE) command

Browse Source
This commit is contained in:
Florian Goetghebeur
2020-03-10 11:43:17 +01:00
parent adfb8ed9f3
commit 827e50c0bd
3 changed files with 54 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
conn_test.go
View File

@@ -136,7 +136,14 @@ func (mock *ftpMock) listen(t *testing.T) {
break
}
mock.proto.Writer.PrintfLine("150 please send")
mock.recvDataConn()
mock.recvDataConn(false)
case "APPE":
if mock.dataConn == nil {
mock.proto.Writer.PrintfLine("425 Unable to build data connection: Connection refused")
break
}
mock.proto.Writer.PrintfLine("150 please send")
mock.recvDataConn(true)
case "LIST":
if mock.dataConn == nil {
mock.proto.Writer.PrintfLine("425 Unable to build data connection: Connection refused")
@@ -269,9 +276,11 @@ func (mock *ftpMock) listenDataConn() (int64, error) {
return p, nil
}
func (mock *ftpMock) recvDataConn() {
func (mock *ftpMock) recvDataConn(append bool) {
mock.dataConn.Wait()
mock.fileCont = new(bytes.Buffer)
if !append {
mock.fileCont = new(bytes.Buffer)
}
io.Copy(mock.fileCont, mock.dataConn.conn)
mock.proto.Writer.PrintfLine("226 Transfer Complete")
mock.closeDataConn()