Fix parsing invalid EPSV response - Fixes #451

Author: czoIg

client_test.go

@@ -417,3 +417,47 @@ func TestDialWithDialer(t *testing.T) {
 
 	assert.Equal(t, true, dialerCalled)
 }
+
+func runEPSVWithResponse(response string) (int, error) {
+	client, server := net.Pipe()
+	tpServer := textproto.NewConn(server)
+	go func() {
+		defer tpServer.Close()
+		_, _ = tpServer.ReadLine()
+		tpServer.PrintfLine("229 %s", response)
+	}()
+
+	c := &ServerConn{options: &dialOptions{}, conn: textproto.NewConn(client)}
+	defer c.conn.Close()
+	return c.epsv()
+}
+
+func TestEPSV_Parse_Valid(t *testing.T) {
+	port, err := runEPSVWithResponse("Entering Extended Passive Mode (|||4242|)")
+	assert.NoError(t, err)
+	assert.Equal(t, 4242, port)
+}
+
+func TestEPSV_Parse_MissingTrailingPipe_ShouldError(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Fatalf("epsv() panicked, want graceful error: %v", r)
+		}
+	}()
+	_, err := runEPSVWithResponse("Entering Extended Passive Mode (|||4242)")
+	if err == nil {
+		t.Fatalf("expected error for malformed EPSV response, got nil")
+	}
+}
+
+func TestEPSV_Parse_MissingPortBetweenPipes_ShouldError(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Fatalf("epsv() panicked, want graceful error: %v", r)
+		}
+	}()
+	_, err := runEPSVWithResponse("Entering Extended Passive Mode (||||)")
+	if err == nil {
+		t.Fatalf("expected error for malformed EPSV response, got nil")
+	}
+}

ftp.go

@@ -484,7 +484,7 @@ func (c *ServerConn) epsv() (port int, err error) {
 
 	start := strings.Index(line, "|||")
 	end := strings.LastIndex(line, "|")
-	if start == -1 || end == -1 {
+	if start == -1 || start+3 >= end {
 		return 0, errors.New("invalid EPSV response format")
 	}
 	port, err = strconv.Atoi(line[start+3 : end])