From 4f9e805d4412da3dda11078bb23d051bc4a4a13c Mon Sep 17 00:00:00 2001
From: Nick Craig-Wood <nick@craig-wood.com>
Date: Fri, 5 Aug 2016 23:42:49 +0100
Subject: [PATCH] acd: Work around 408 REQUEST_TIMEOUT and 504 GATEWAY_TIMEOUT
 errors

Amazon Drive sometimes returns errors at the end of large uploads

  * 408 REQUEST_TIMEOUT
  * 504 GATEWAY_TIMEOUT
  * 500 Internal error

The file may have been uploaded correctly though, so, on error, wait
for up to 2 minutes for it to appear if it was fully
uploaded (configure timeout with --acd-upload-wait-time).

Issues: #601 #605 #606
---
 amazonclouddrive/amazonclouddrive.go | 69 ++++++++++++++++++++++++++++
 fstest/fstests/fstests.go            |  2 +-
 2 files changed, 70 insertions(+), 1 deletion(-)

diff --git a/amazonclouddrive/amazonclouddrive.go b/amazonclouddrive/amazonclouddrive.go
index 290a425a9..ecfc1331c 100644
--- a/amazonclouddrive/amazonclouddrive.go
+++ b/amazonclouddrive/amazonclouddrive.go
@@ -47,6 +47,7 @@ const (
 var (
 	// Flags
 	tempLinkThreshold = fs.SizeSuffix(9 << 30) // Download files bigger than this via the tempLink
+	uploadWaitTime    = pflag.DurationP("acd-upload-wait-time", "", 2*60*time.Second, "Time to wait after a failed complete upload to see if it appears.")
 	// Description of how to auth for this app
 	acdConfig = &oauth2.Config{
 		Scopes: []string{"clouddrive:read_all", "clouddrive:write"},
@@ -416,6 +417,64 @@ func (f *Fs) List(out fs.ListOpts, dir string) {
 	f.dirCache.List(f, out, dir)
 }
 
+// checkUpload checks to see if an error occurred after the file was
+// completely uploaded.
+//
+// If it was then it waits for a while to see if the file really
+// exists and is the right size and returns an updated info.
+//
+// If the file wasn't found or was the wrong size then it returns the
+// original error.
+//
+// This is a workaround for Amazon sometimes returning
+//
+//  * 408 REQUEST_TIMEOUT
+//  * 504 GATEWAY_TIMEOUT
+//  * 500 Internal server error
+//
+// At the end of large uploads.  The speculation is that the timeout
+// is waiting for the sha1 hashing to complete and the file may well
+// be properly uploaded.
+func (f *Fs) checkUpload(in io.Reader, src fs.ObjectInfo, inInfo *acd.File, inErr error) (fixedError bool, info *acd.File, err error) {
+	// Return if no error - all is well
+	if inErr == nil {
+		return false, inInfo, inErr
+	}
+	const sleepTime = 5 * time.Second           // sleep between tries
+	retries := int(*uploadWaitTime / sleepTime) // number of retries
+	if retries <= 0 {
+		retries = 1
+	}
+	buf := make([]byte, 1)
+	n, err := in.Read(buf)
+	if !(n == 0 && err == io.EOF) {
+		fs.Debug(src, "Upload error detected but didn't finish upload (n=%d, err=%v): %v", n, err, inErr)
+		return false, inInfo, inErr
+	}
+	fs.Debug(src, "Error detected after finished upload - waiting to see if object was uploaded correctly: %v", inErr)
+	remote := src.Remote()
+	for i := 1; i <= retries; i++ {
+		o, err := f.NewObject(remote)
+		if err == fs.ErrorObjectNotFound {
+			fs.Debug(src, "Object not found - waiting (%d/%d)", i, retries)
+		} else if err != nil {
+			fs.Debug(src, "Object returned error - waiting (%d/%d): %v", i, retries, err)
+		} else {
+			if src.Size() == o.Size() {
+				fs.Debug(src, "Object found with correct size - returning with no error")
+				info = &acd.File{
+					Node: o.(*Object).info,
+				}
+				return true, info, nil
+			}
+			fs.Debug(src, "Object found but wrong size %d vs %d - waiting (%d/%d)", src.Size(), o.Size(), i, retries)
+		}
+		time.Sleep(sleepTime)
+	}
+	fs.Debug(src, "Finished waiting for object - returning original error: %v", inErr)
+	return false, inInfo, inErr
+}
+
 // Put the object into the container
 //
 // Copy the reader in to the new object which is returned
@@ -456,6 +515,11 @@ func (f *Fs) Put(in io.Reader, src fs.ObjectInfo) (fs.Object, error) {
 		} else {
 			info, resp, err = folder.PutSized(in, size, leaf)
 		}
+		var ok bool
+		ok, info, err = f.checkUpload(in, src, info, err)
+		if ok {
+			return false, nil
+		}
 		return f.shouldRetry(resp, err)
 	})
 	if err != nil {
@@ -706,6 +770,11 @@ func (o *Object) Update(in io.Reader, src fs.ObjectInfo) error {
 		} else {
 			info, resp, err = file.Overwrite(in)
 		}
+		var ok bool
+		ok, info, err = o.fs.checkUpload(in, src, info, err)
+		if ok {
+			return false, nil
+		}
 		return o.fs.shouldRetry(resp, err)
 	})
 	if err != nil {
diff --git a/fstest/fstests/fstests.go b/fstest/fstests/fstests.go
index 2c8d4126b..c9e9fb084 100644
--- a/fstest/fstests/fstests.go
+++ b/fstest/fstests/fstests.go
@@ -532,7 +532,7 @@ func TestObjectUpdate(t *testing.T) {
 
 	file1.Size = int64(buf.Len())
 	obj := findObject(t, file1.Path)
-	obji := fs.NewStaticObjectInfo("", file1.ModTime, file1.Size, true, nil, obj.Fs())
+	obji := fs.NewStaticObjectInfo(file1.Path, file1.ModTime, file1.Size, true, nil, obj.Fs())
 	err := obj.Update(in, obji)
 	require.NoError(t, err)
 	file1.Hashes = hash.Sums()