summaryrefslogtreecommitdiff
path: root/pkg/varlinkapi/transfers.go
diff options
context:
space:
mode:
authorMatthew Heon <mheon@redhat.com>2019-05-16 14:35:49 -0400
committerMatthew Heon <mheon@redhat.com>2019-05-16 14:36:11 -0400
commit601fc2ce4fa888cdd0cc33c41f6eee7ca39f585c (patch)
treedc80e1320d2aa8eeb95915c327435b7f324d7623 /pkg/varlinkapi/transfers.go
parent8161802f7df857e0850f842261079c83290f9891 (diff)
downloadpodman-601fc2ce4fa888cdd0cc33c41f6eee7ca39f585c.tar.gz
podman-601fc2ce4fa888cdd0cc33c41f6eee7ca39f585c.tar.bz2
podman-601fc2ce4fa888cdd0cc33c41f6eee7ca39f585c.zip
Revert "Add VarlinkCall.RequiresUpgrade() type and method"
This reverts commit bd3154fcf6a48b37cfde5d9b1226900cd863c0d9. Commit in question may be breaking upstream CI. Signed-off-by: Matthew Heon <mheon@redhat.com>
Diffstat (limited to 'pkg/varlinkapi/transfers.go')
-rw-r--r--pkg/varlinkapi/transfers.go35
1 files changed, 16 insertions, 19 deletions
diff --git a/pkg/varlinkapi/transfers.go b/pkg/varlinkapi/transfers.go
index 7b38ec5e6..96f76bcdc 100644
--- a/pkg/varlinkapi/transfers.go
+++ b/pkg/varlinkapi/transfers.go
@@ -15,61 +15,58 @@ import (
// SendFile allows a client to send a file to the varlink server
func (i *LibpodAPI) SendFile(call iopodman.VarlinkCall, ftype string, length int64) error {
- varlink := VarlinkCall{&call}
- if err := varlink.RequiresUpgrade(); err != nil {
- return varlink.ReplyErrorOccurred(err.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 varlink.ReplyErrorOccurred(err.Error())
+ return call.ReplyErrorOccurred(err.Error())
}
defer outputFile.Close()
- if err = varlink.ReplySendFile(outputFile.Name()); err != nil {
- return varlink.ReplyErrorOccurred(err.Error())
+ if err = call.ReplySendFile(outputFile.Name()); err != nil {
+ return call.ReplyErrorOccurred(err.Error())
}
writer := bufio.NewWriter(outputFile)
defer writer.Flush()
- reader := varlink.Call.Reader
+ reader := call.Call.Reader
if _, err := io.CopyN(writer, reader, length); err != nil {
return err
}
logrus.Debugf("successfully received %s", outputFile.Name())
// Send an ACK to the client
- varlink.Call.Writer.WriteString(fmt.Sprintf("%s:", outputFile.Name()))
- varlink.Call.Writer.Flush()
+ 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 {
- varlink := VarlinkCall{&call}
- if err := varlink.RequiresUpgrade(); err != nil {
- return varlink.ReplyErrorOccurred(err.Error())
+ if !call.WantsUpgrade() {
+ return call.ReplyErrorOccurred("client must use upgraded connection to send files")
}
-
fs, err := os.Open(filepath)
if err != nil {
- return varlink.ReplyErrorOccurred(err.Error())
+ return call.ReplyErrorOccurred(err.Error())
}
fileInfo, err := fs.Stat()
if err != nil {
- return varlink.ReplyErrorOccurred(err.Error())
+ return call.ReplyErrorOccurred(err.Error())
}
// Send the file length down to client
// Varlink connection upraded
- if err = varlink.ReplyReceiveFile(fileInfo.Size()); err != nil {
- return varlink.ReplyErrorOccurred(err.Error())
+ if err = call.ReplyReceiveFile(fileInfo.Size()); err != nil {
+ return call.ReplyErrorOccurred(err.Error())
}
reader := bufio.NewReader(fs)
- _, err = reader.WriteTo(varlink.Writer)
+ _, err = reader.WriteTo(call.Writer)
if err != nil {
return err
}
@@ -78,5 +75,5 @@ func (i *LibpodAPI) ReceiveFile(call iopodman.VarlinkCall, filepath string, dele
return err
}
}
- return varlink.Writer.Flush()
+ return call.Writer.Flush()
}