get: fix bug + improvements

up until now there has been a very annoying bug with get, we would
get halting behavior. I'm not 100% sure this commit fixes it,
but it should. It certainly fixes others found in the process of
digging into the get / tar extractor code. (wish we could repro
the bug reliably enough to make a test case).

This is a much cleaner tar writer. the ad-hoc, error-prone synch
for the tar reader is gone (with i believe was incorrect). it is
replaced with a simple pipe and bufio. The tar logic is now in
tar.Writer, which writes unixfs dag nodes into a tar archive (no
need for synch here). And get's reader is constructed with DagArchive
which sets up the pipe + bufio.

NOTE: this commit also changes this behavior of `get`:
When retrieving a single file, if the file exists, get would fail.
this emulated the behavior of wget by default, which (without opts)
does not overwrite if the file is there. This change makes get
fail if the file is available locally. This seems more intuitive to
me as expected from a unix tool-- though perhaps it should be
discussed more before adopting.

Everything seems to work fine, and i have not been able to reproduce
the get halt bug.

License: MIT
Signed-off-by: Juan Batiz-Benet <[email protected]>

Author: jbenet

core/commands/get.go

@@ -98,52 +98,85 @@ may also specify the level of compression by specifying '-l=<1-9>'.
 			return
 		}
 
-		if archive, _, _ := req.Option("archive").Bool(); archive || cmplvl != gzip.NoCompression {
-			if archive && !strings.HasSuffix(outPath, ".tar") {
-				outPath += ".tar"
-			}
-			if cmplvl != gzip.NoCompression {
-				outPath += ".gz"
-			}
-			fmt.Printf("Saving archive to %s\n", outPath)
-
-			file, err := os.Create(outPath)
-			if err != nil {
-				res.SetError(err, cmds.ErrNormal)
-				return
-			}
-			defer file.Close()
-
-			bar := pb.New(0).SetUnits(pb.U_BYTES)
-			bar.Output = os.Stderr
-			pbReader := bar.NewProxyReader(outReader)
-			bar.Start()
-			defer bar.Finish()
-
-			if _, err := io.Copy(file, pbReader); err != nil {
-				res.SetError(err, cmds.ErrNormal)
-				return
-			}
+		archive, _, _ := req.Option("archive").Bool()
 
+		gw := getWriter{
+			Out:         os.Stdout,
+			Err:         os.Stderr,
+			Archive:     archive,
+			Compression: cmplvl,
+		}
+
+		if err := gw.Write(outReader, outPath); err != nil {
+			res.SetError(err, cmds.ErrNormal)
 			return
 		}
+	},
+}
 
-		fmt.Printf("Saving file(s) to %s\n", outPath)
+func progressBarForReader(out io.Writer, r io.Reader) (*pb.ProgressBar, *pb.Reader) {
+	// setup bar reader
+	// TODO: get total length of files
+	bar := pb.New(0).SetUnits(pb.U_BYTES)
+	bar.Output = out
+	barR := bar.NewProxyReader(r)
+	return bar, barR
+}
 
-		// TODO: get total length of files
-		bar := pb.New(0).SetUnits(pb.U_BYTES)
-		bar.Output = os.Stderr
+type getWriter struct {
+	Out io.Writer // for output to user
+	Err io.Writer // for progress bar output
 
-		// wrap the reader with the progress bar proxy reader
-		reader := bar.NewProxyReader(outReader)
+	Archive     bool
+	Compression int
+}
 
-		bar.Start()
-		defer bar.Finish()
-		extractor := &tar.Extractor{outPath}
-		if err := extractor.Extract(reader); err != nil {
-			res.SetError(err, cmds.ErrNormal)
+func (gw *getWriter) Write(r io.Reader, fpath string) error {
+	if gw.Archive || gw.Compression != gzip.NoCompression {
+		return gw.writeArchive(r, fpath)
+	}
+	return gw.writeExtracted(r, fpath)
+}
+
+func (gw *getWriter) writeArchive(r io.Reader, fpath string) error {
+	// adjust file name if tar
+	if gw.Archive {
+		if !strings.HasSuffix(fpath, ".tar") && !strings.HasSuffix(fpath, ".tar.gz") {
+			fpath += ".tar"
 		}
-	},
+	}
+
+	// adjust file name if gz
+	if gw.Compression != gzip.NoCompression {
+		if !strings.HasSuffix(fpath, ".gz") {
+			fpath += ".gz"
+		}
+	}
+
+	// create file
+	file, err := os.Create(fpath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	fmt.Fprintf(gw.Out, "Saving archive to %s\n", fpath)
+	bar, barR := progressBarForReader(gw.Err, r)
+	bar.Start()
+	defer bar.Finish()
+
+	_, err = io.Copy(file, barR)
+	return err
+}
+
+func (gw *getWriter) writeExtracted(r io.Reader, fpath string) error {
+	fmt.Fprintf(gw.Out, "Saving file(s) to %s\n", fpath)
+	bar, barR := progressBarForReader(gw.Err, r)
+	bar.Start()
+	defer bar.Finish()
+
+	extractor := &tar.Extractor{fpath}
+	return extractor.Extract(barR)
 }
 
 func getCompressOptions(req cmds.Request) (int, error) {
@@ -161,12 +194,12 @@ func getCompressOptions(req cmds.Request) (int, error) {
 }
 
 func get(ctx context.Context, node *core.IpfsNode, p path.Path, compression int) (io.Reader, error) {
-	dagnode, err := core.Resolve(ctx, node, p)
+	dn, err := core.Resolve(ctx, node, p)
 	if err != nil {
 		return nil, err
 	}
 
-	return utar.NewReader(ctx, p, node.DAG, dagnode, compression)
+	return utar.DagArchive(ctx, dn, p.String(), node.DAG, compression)
 }
 
 // getZip is equivalent to `ipfs getdag $hash | gzip`

test/sharness/t0090-get.sh

@@ -37,9 +37,8 @@ test_get_cmd() {
 		test_cmp "$HASH" data
 	'
 
-	# this started failing after this change. fixed in later commit
-	test_expect_failure "ipfs get errors when trying to overwrite a file" '
-		test_must_fail ipfs get "$HASH" >actual &&
+	test_expect_success "ipfs get DOES NOT error when trying to overwrite a file" '
+		ipfs get "$HASH" >actual &&
 		rm "$HASH"
 	'
 

thirdparty/tar/extractor.go

@@ -4,6 +4,7 @@ import (
 	"archive/tar"
 	"io"
 	"os"
+	gopath "path"
 	fp "path/filepath"
 	"strings"
 )
@@ -39,7 +40,7 @@ func (te *Extractor) Extract(reader io.Reader) error {
 		}
 
 		if header.Typeflag == tar.TypeDir {
-			if err := te.extractDir(header, i, rootExists); err != nil {
+			if err := te.extractDir(header, i); err != nil {
 				return err
 			}
 			continue
@@ -52,13 +53,19 @@ func (te *Extractor) Extract(reader io.Reader) error {
 	return nil
 }
 
-func (te *Extractor) extractDir(h *tar.Header, depth int, rootExists bool) error {
-	pathElements := strings.Split(h.Name, "/")
-	if !rootExists {
-		pathElements = pathElements[1:]
-	}
-	path := fp.Join(pathElements...)
-	path = fp.Join(te.Path, path)
+// outputPath returns the path at whicht o place tarPath
+func (te *Extractor) outputPath(tarPath string) string {
+	elems := strings.Split(tarPath, "/") // break into elems
+	elems = elems[1:]                    // remove original root
+
+	path := fp.Join(elems...)     // join elems
+	path = fp.Join(te.Path, path) // rebase on extractor root
+	return path
+}
+
+func (te *Extractor) extractDir(h *tar.Header, depth int) error {
+	path := te.outputPath(h.Name)
+
 	if depth == 0 {
 		// if this is the root root directory, use it as the output path for remaining files
 		te.Path = path
@@ -73,18 +80,18 @@ func (te *Extractor) extractDir(h *tar.Header, depth int, rootExists bool) error
 }
 
 func (te *Extractor) extractFile(h *tar.Header, r *tar.Reader, depth int, rootExists bool, rootIsDir bool) error {
-	path := te.Path
-	if depth == 0 && rootExists {
-		// if depth is 0, this is the only file (we aren't 'ipfs get'ing a directory)
-		if rootIsDir { // putting file inside of a root dir.
-			path = fp.Join(te.Path, h.Name)
-		}
-		// else if the file exists, just overwrite it.
-	} else {
-		// we are outputting a directory, this file is inside of it
-		pathElements := strings.Split(h.Name, "/")[1:]
-		path = fp.Join(pathElements...)
-		path = fp.Join(te.Path, path)
+	path := te.outputPath(h.Name)
+
+	if depth == 0 { // if depth is 0, this is the only file (we aren't 'ipfs get'ing a directory)
+		if rootExists && rootIsDir {
+			// putting file inside of a root dir.
+			fnameo := gopath.Base(h.Name)
+			fnamen := fp.Base(path)
+			// add back original name if lost.
+			if fnameo != fnamen {
+				path = fp.Join(path, fnameo)
+			}
+		} // else if old file exists, just overwrite it.
 	}
 
 	file, err := os.Create(path)