mirror of
https://github.com/containers/podman.git
synced 2025-06-17 23:20:59 +08:00
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>
This commit is contained in:
@ -45,24 +45,22 @@ func (i *LibpodAPI) Attach(call iopodman.VarlinkCall, name string, detachKeys st
|
|||||||
var finalErr error
|
var finalErr error
|
||||||
resize := make(chan remotecommand.TerminalSize)
|
resize := make(chan remotecommand.TerminalSize)
|
||||||
errChan := make(chan error)
|
errChan := make(chan error)
|
||||||
varlink := VarlinkCall{&call}
|
|
||||||
|
|
||||||
if err := varlink.RequiresUpgrade(); err != nil {
|
if !call.WantsUpgrade() {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred("client must use upgraded connection to attach")
|
||||||
}
|
}
|
||||||
|
|
||||||
ctr, err := i.Runtime.LookupContainer(name)
|
ctr, err := i.Runtime.LookupContainer(name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
state, err := ctr.State()
|
state, err := ctr.State()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
if !start && state != libpod.ContainerStateRunning {
|
if !start && state != libpod.ContainerStateRunning {
|
||||||
return varlink.ReplyErrorOccurred("container must be running to attach")
|
return call.ReplyErrorOccurred("container must be running to attach")
|
||||||
}
|
}
|
||||||
|
call.Reply(nil)
|
||||||
reader, writer, _, pw, streams := setupStreams(call)
|
reader, writer, _, pw, streams := setupStreams(call)
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
@ -83,7 +81,7 @@ func (i *LibpodAPI) Attach(call iopodman.VarlinkCall, name string, detachKeys st
|
|||||||
quitWriter := virtwriter.NewVirtWriteCloser(writer, virtwriter.Quit)
|
quitWriter := virtwriter.NewVirtWriteCloser(writer, virtwriter.Quit)
|
||||||
_, err = quitWriter.Write([]byte("HANG-UP"))
|
_, err = quitWriter.Write([]byte("HANG-UP"))
|
||||||
// TODO error handling is not quite right here yet
|
// TODO error handling is not quite right here yet
|
||||||
return varlink.Writer.Flush()
|
return call.Writer.Flush()
|
||||||
}
|
}
|
||||||
|
|
||||||
func attach(ctr *libpod.Container, streams *libpod.AttachStreams, detachKeys string, resize chan remotecommand.TerminalSize, errChan chan error) error {
|
func attach(ctr *libpod.Container, streams *libpod.AttachStreams, detachKeys string, resize chan remotecommand.TerminalSize, errChan chan error) error {
|
||||||
|
@ -21,7 +21,3 @@ func New(cli *cliconfig.PodmanCommand, runtime *libpod.Runtime) *iopodman.Varlin
|
|||||||
lp := LibpodAPI{Cli: cli.Command, Runtime: runtime}
|
lp := LibpodAPI{Cli: cli.Command, Runtime: runtime}
|
||||||
return iopodman.VarlinkNew(&lp)
|
return iopodman.VarlinkNew(&lp)
|
||||||
}
|
}
|
||||||
|
|
||||||
type VarlinkCall struct {
|
|
||||||
*iopodman.VarlinkCall
|
|
||||||
}
|
|
||||||
|
@ -15,61 +15,58 @@ import (
|
|||||||
|
|
||||||
// SendFile allows a client to send a file to the varlink server
|
// SendFile allows a client to send a file to the varlink server
|
||||||
func (i *LibpodAPI) SendFile(call iopodman.VarlinkCall, ftype string, length int64) error {
|
func (i *LibpodAPI) SendFile(call iopodman.VarlinkCall, ftype string, length int64) error {
|
||||||
varlink := VarlinkCall{&call}
|
if !call.WantsUpgrade() {
|
||||||
if err := varlink.RequiresUpgrade(); err != nil {
|
return call.ReplyErrorOccurred("client must use upgraded connection to send files")
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
outputFile, err := ioutil.TempFile("", "varlink_send")
|
outputFile, err := ioutil.TempFile("", "varlink_send")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
defer outputFile.Close()
|
defer outputFile.Close()
|
||||||
|
|
||||||
if err = varlink.ReplySendFile(outputFile.Name()); err != nil {
|
if err = call.ReplySendFile(outputFile.Name()); err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
writer := bufio.NewWriter(outputFile)
|
writer := bufio.NewWriter(outputFile)
|
||||||
defer writer.Flush()
|
defer writer.Flush()
|
||||||
|
|
||||||
reader := varlink.Call.Reader
|
reader := call.Call.Reader
|
||||||
if _, err := io.CopyN(writer, reader, length); err != nil {
|
if _, err := io.CopyN(writer, reader, length); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
logrus.Debugf("successfully received %s", outputFile.Name())
|
logrus.Debugf("successfully received %s", outputFile.Name())
|
||||||
// Send an ACK to the client
|
// Send an ACK to the client
|
||||||
varlink.Call.Writer.WriteString(fmt.Sprintf("%s:", outputFile.Name()))
|
call.Call.Writer.WriteString(fmt.Sprintf("%s:", outputFile.Name()))
|
||||||
varlink.Call.Writer.Flush()
|
call.Call.Writer.Flush()
|
||||||
return nil
|
return nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReceiveFile allows the varlink server to send a file to a client
|
// ReceiveFile allows the varlink server to send a file to a client
|
||||||
func (i *LibpodAPI) ReceiveFile(call iopodman.VarlinkCall, filepath string, delete bool) error {
|
func (i *LibpodAPI) ReceiveFile(call iopodman.VarlinkCall, filepath string, delete bool) error {
|
||||||
varlink := VarlinkCall{&call}
|
if !call.WantsUpgrade() {
|
||||||
if err := varlink.RequiresUpgrade(); err != nil {
|
return call.ReplyErrorOccurred("client must use upgraded connection to send files")
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fs, err := os.Open(filepath)
|
fs, err := os.Open(filepath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
fileInfo, err := fs.Stat()
|
fileInfo, err := fs.Stat()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
// Send the file length down to client
|
// Send the file length down to client
|
||||||
// Varlink connection upraded
|
// Varlink connection upraded
|
||||||
if err = varlink.ReplyReceiveFile(fileInfo.Size()); err != nil {
|
if err = call.ReplyReceiveFile(fileInfo.Size()); err != nil {
|
||||||
return varlink.ReplyErrorOccurred(err.Error())
|
return call.ReplyErrorOccurred(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
reader := bufio.NewReader(fs)
|
reader := bufio.NewReader(fs)
|
||||||
_, err = reader.WriteTo(varlink.Writer)
|
_, err = reader.WriteTo(call.Writer)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -78,5 +75,5 @@ func (i *LibpodAPI) ReceiveFile(call iopodman.VarlinkCall, filepath string, dele
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return varlink.Writer.Flush()
|
return call.Writer.Flush()
|
||||||
}
|
}
|
||||||
|
@ -13,11 +13,6 @@ import (
|
|||||||
"github.com/containers/libpod/cmd/podman/varlink"
|
"github.com/containers/libpod/cmd/podman/varlink"
|
||||||
"github.com/containers/libpod/libpod"
|
"github.com/containers/libpod/libpod"
|
||||||
"github.com/containers/storage/pkg/archive"
|
"github.com/containers/storage/pkg/archive"
|
||||||
"github.com/pkg/errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
ErrUpgradedConnectionRequired = errors.New("peer must use upgraded connection for operation")
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// getContext returns a non-nil, empty context
|
// getContext returns a non-nil, empty context
|
||||||
@ -200,13 +195,3 @@ func makePsOpts(inOpts iopodman.PsOpts) shared.PsOptions {
|
|||||||
Sync: derefBool(inOpts.Sync),
|
Sync: derefBool(inOpts.Sync),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// RequiresUpgrade tests if varlink connection has been marked for upgrade.
|
|
||||||
func (v *VarlinkCall) RequiresUpgrade() error {
|
|
||||||
if v.WantsUpgrade() {
|
|
||||||
// A nil is sent to the peer as required by the varlink protocol.
|
|
||||||
return v.Reply(nil)
|
|
||||||
} else {
|
|
||||||
return ErrUpgradedConnectionRequired
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
Reference in New Issue
Block a user