Skip to content

Commit

Permalink
Wait for podman stop to complete
Browse files Browse the repository at this point in the history
if users run podman machine stop && podman machine ls, the status of the
machine in the subsequent ls command would running.  now we wait for
everything to complete for stop so that scripting is more accurate.

Fixes: containers#12815

[NO NEW TESTS NEEDED]

Signed-off-by: Brent Baude <[email protected]>
  • Loading branch information
baude committed Jan 12, 2022
1 parent 4db1aff commit b7fe25d
Show file tree
Hide file tree
Showing 2 changed files with 119 additions and 12 deletions.
9 changes: 9 additions & 0 deletions pkg/machine/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,15 @@ type InitOptions struct {
ReExec bool
}

type QemuMachineStatus = string

const (
// Running indicates the qemu vm is running
Running QemuMachineStatus = "running"
// Stopped indicates the vm has stopped
Stopped QemuMachineStatus = "stopped"
)

type Provider interface {
NewMachine(opts InitOptions) (VM, error)
LoadVMByName(name string) (VM, error)
Expand Down
122 changes: 110 additions & 12 deletions pkg/machine/qemu/machine.go
Original file line number Diff line number Diff line change
Expand Up @@ -386,8 +386,16 @@ func (v *MachineVM) Start(name string, _ machine.StartOptions) error {
}

if len(v.Mounts) > 0 {
for !v.isRunning() || !v.isListening() {
running, err := v.isRunning()
if err != nil {
return err
}
for running || !v.isListening() {
time.Sleep(100 * time.Millisecond)
running, err = v.isRunning()
if err != nil {
return err
}
}
}
for _, mount := range v.Mounts {
Expand Down Expand Up @@ -416,8 +424,48 @@ func (v *MachineVM) Start(name string, _ machine.StartOptions) error {
return nil
}

func (v *MachineVM) checkStatus(monitor *qmp.SocketMonitor) (machine.QemuMachineStatus, error) {
// this is the format returned from the monitor
// {"return": {"status": "running", "singlestep": false, "running": true}}

type statusDetails struct {
Status string `json:"status"`
Step bool `json:"singlestep"`
Running bool `json:"running"`
}
type statusResponse struct {
Response statusDetails `json:"return"`
}
var response statusResponse

checkCommand := struct {
Execute string `json:"execute"`
}{
Execute: "query-status",
}
input, err := json.Marshal(checkCommand)
if err != nil {
return "", err
}
b, err := monitor.Run(input)
if err != nil {
if errors.Cause(err) == os.ErrNotExist {
return machine.Stopped, nil
}
return "", err
}
if err := json.Unmarshal(b, &response); err != nil {
return "", err
}
if response.Response.Status == machine.Running {
return machine.Running, nil
}
return machine.Stopped, nil
}

// Stop uses the qmp monitor to call a system_powerdown
func (v *MachineVM) Stop(name string, _ machine.StopOptions) error {
var disconnected bool
// check if the qmp socket is there. if not, qemu instance is gone
if _, err := os.Stat(v.QMPMonitor.Address); os.IsNotExist(err) {
// Right now it is NOT an error to stop a stopped machine
Expand All @@ -442,19 +490,22 @@ func (v *MachineVM) Stop(name string, _ machine.StopOptions) error {
return err
}
defer func() {
if err := qmpMonitor.Disconnect(); err != nil {
logrus.Error(err)
if !disconnected {
if err := qmpMonitor.Disconnect(); err != nil {
logrus.Error(err)
}
}
}()

if _, err = qmpMonitor.Run(input); err != nil {
return err
}

qemuSocketFile, pidFile, err := v.getSocketandPid()
if err != nil {
return err
}
if _, err := os.Stat(pidFile); os.IsNotExist(err) {
logrus.Info(err)
return nil
}
pidString, err := ioutil.ReadFile(pidFile)
Expand Down Expand Up @@ -483,6 +534,24 @@ func (v *MachineVM) Stop(name string, _ machine.StopOptions) error {
return err
}

if err := qmpMonitor.Disconnect(); err != nil {
return nil
}

disconnected = true
waitInternal := 250 * time.Millisecond
for i := 0; i < 5; i++ {
running, err := v.isRunning()
if err != nil {
return err
}
if !running {
break
}
time.Sleep(waitInternal)
waitInternal = waitInternal * 2
}

return nil
}

Expand Down Expand Up @@ -519,7 +588,11 @@ func (v *MachineVM) Remove(name string, opts machine.RemoveOptions) (string, fun
)

// cannot remove a running vm
if v.isRunning() {
running, err := v.isRunning()
if err != nil {
return "", nil, err
}
if running {
return "", nil, errors.Errorf("running vm %q cannot be destroyed", v.Name)
}

Expand Down Expand Up @@ -578,16 +651,33 @@ func (v *MachineVM) Remove(name string, opts machine.RemoveOptions) (string, fun
}, nil
}

func (v *MachineVM) isRunning() bool {
func (v *MachineVM) isRunning() (bool, error) {
// Check if qmp socket path exists
if _, err := os.Stat(v.QMPMonitor.Address); os.IsNotExist(err) {
return false
return false, nil
}
// Check if we can dial it
if _, err := qmp.NewSocketMonitor(v.QMPMonitor.Network, v.QMPMonitor.Address, v.QMPMonitor.Timeout); err != nil {
return false
monitor, err := qmp.NewSocketMonitor(v.QMPMonitor.Network, v.QMPMonitor.Address, v.QMPMonitor.Timeout)
if err != nil {
return false, nil
}
return true
if err := monitor.Connect(); err != nil {
return false, err
}
defer func() {
if err := monitor.Disconnect(); err != nil {
logrus.Error(err)
}
}()
// If there is a monitor, lets see if we can query state
state, err := v.checkStatus(monitor)
if err != nil {
return false, err
}
if state == machine.Running {
return true, nil
}
return false, nil
}

func (v *MachineVM) isListening() bool {
Expand All @@ -603,7 +693,11 @@ func (v *MachineVM) isListening() bool {
// SSH opens an interactive SSH session to the vm specified.
// Added ssh function to VM interface: pkg/machine/config/go : line 58
func (v *MachineVM) SSH(name string, opts machine.SSHOptions) error {
if !v.isRunning() {
running, err := v.isRunning()
if err != nil {
return err
}
if !running {
return errors.Errorf("vm %q is not running.", v.Name)
}

Expand Down Expand Up @@ -707,7 +801,11 @@ func GetVMInfos() ([]*machine.ListResponse, error) {
return err
}
listEntry.LastUp = fi.ModTime()
if vm.isRunning() {
running, err := vm.isRunning()
if err != nil {
return err
}
if running {
listEntry.Running = true
}

Expand Down

0 comments on commit b7fe25d

Please sign in to comment.