Merge pull request #2227 from baude/remoteexport

podman-remote import|export

1 files changed, 75 insertions, 0 deletions

diff --git a/pkg/varlinkapi/transfers.go b/pkg/varlinkapi/transfers.go
new file mode 100644
index 000000000..0cb7e5e2e
--- /dev/null
+++ b/pkg/varlinkapi/transfers.go
@@ -0,0 +1,75 @@
+package varlinkapi
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"os"
+
+	"github.com/containers/libpod/cmd/podman/varlink"
+)
+
+// SendFile allows a client to send a file to the varlink server
+func (i *LibpodAPI) SendFile(call iopodman.VarlinkCall, ftype string, length int64) error {
+	if !call.WantsUpgrade() {
+		return call.ReplyErrorOccurred("client must use upgraded connection to send files")
+	}
+
+	outputFile, err := ioutil.TempFile("", "varlink_send")
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	defer outputFile.Close()
+
+	if err = call.ReplySendFile(outputFile.Name()); err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+
+	writer := bufio.NewWriter(outputFile)
+	defer writer.Flush()
+
+	reader := call.Call.Reader
+	if _, err := io.CopyN(writer, reader, length); err != nil {
+		return err
+	}
+
+	// Send an ACK to the client
+	call.Call.Writer.WriteString(fmt.Sprintf("%s:", outputFile.Name()))
+	call.Call.Writer.Flush()
+	return nil
+
+}
+
+// ReceiveFile allows the varlink server to send a file to a client
+func (i *LibpodAPI) ReceiveFile(call iopodman.VarlinkCall, filepath string, delete bool) error {
+	if !call.WantsUpgrade() {
+		return call.ReplyErrorOccurred("client must use upgraded connection to send files")
+	}
+	fs, err := os.Open(filepath)
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	fileInfo, err := fs.Stat()
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+
+	// Send the file length down to client
+	// Varlink connection upraded
+	if err = call.ReplyReceiveFile(fileInfo.Size()); err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+
+	reader := bufio.NewReader(fs)
+	_, err = reader.WriteTo(call.Writer)
+	if err != nil {
+		return err
+	}
+	if delete {
+		if err := os.Remove(filepath); err != nil {
+			return err
+		}
+	}
+	return call.Writer.Flush()
+}