From 9320e1645420fb7868c96cfb2f4a2f90ccc6bc6e Mon Sep 17 00:00:00 2001
From: jessejiao <jessejiao@tencent.com>
Date: Tue, 18 May 2021 14:29:05 +0800
Subject: [PATCH] fix: afc open download truncation

---
 afc.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/afc.go b/afc.go
index fee5a5a..62bb68a 100644
--- a/afc.go
+++ b/afc.go
@@ -58,6 +58,9 @@ func (c *afc) ReadDir(dirname string) (names []string, err error) {
 	if respMsg, err = c.client.Receive(); err != nil {
 		return nil, fmt.Errorf("afc receive 'ReadDir': %w", err)
 	}
+	if err = respMsg.Err(); err != nil {
+		return nil, fmt.Errorf("afc 'ReadDir': %w", err)
+	}
 
 	names = respMsg.Strings()
 	return
@@ -450,6 +453,8 @@ func (f *AfcFile) Read(b []byte) (n int, err error) {
 
 	if f.reader == nil {
 		f.reader = bytes.NewReader(respMsg.Payload)
+	} else {
+		f.reader.Reset(respMsg.Payload)
 	}
 
 	return f.reader.Read(b)