package lxc import ( "context" "errors" "fmt" "log" "strings" "time" "github.com/hashicorp/packer/helper/multistep" "github.com/hashicorp/packer/packer" ) type StepWaitInit struct { WaitTimeout time.Duration } func (s *StepWaitInit) Run(ctx context.Context, state multistep.StateBag) multistep.StepAction { ui := state.Get("ui").(packer.Ui) var err error cancel := make(chan struct{}) waitDone := make(chan bool, 1) go func() { ui.Say("Waiting for container to finish init...") err = s.waitForInit(state, cancel) waitDone <- true }() log.Printf("Waiting for container to finish init, up to timeout: %s", s.WaitTimeout) timeout := time.After(s.WaitTimeout) WaitLoop: for { select { case <-waitDone: if err != nil { ui.Error(fmt.Sprintf("Error waiting for container to finish init: %s", err)) return multistep.ActionHalt } ui.Say("Container finished init!") break WaitLoop case <-timeout: err := fmt.Errorf("Timeout waiting for container to finish init.") state.Put("error", err) ui.Error(err.Error()) close(cancel) return multistep.ActionHalt case <-time.After(1 * time.Second): if _, ok := state.GetOk(multistep.StateCancelled); ok { close(cancel) log.Println("Interrupt detected, quitting waiting for container to finish init.") return multistep.ActionHalt } } } return multistep.ActionContinue } func (s *StepWaitInit) Cleanup(multistep.StateBag) { } func (s *StepWaitInit) waitForInit(state multistep.StateBag, cancel <-chan struct{}) error { config := state.Get("config").(*Config) mountPath := state.Get("mount_path").(string) wrappedCommand := state.Get("wrappedCommand").(CommandWrapper) for { select { case <-cancel: log.Println("Cancelled. Exiting loop.") return errors.New("Wait cancelled") case <-time.After(1 * time.Second): } comm := &LxcAttachCommunicator{ ContainerName: config.ContainerName, AttachOptions: config.AttachOptions, RootFs: mountPath, CmdWrapper: wrappedCommand, } runlevel, _ := comm.CheckInit() currentRunlevel := "unknown" if arr := strings.Split(runlevel, " "); len(arr) >= 2 { currentRunlevel = arr[1] } log.Printf("Current runlevel in container: '%s'", runlevel) targetRunlevel := fmt.Sprintf("%d", config.TargetRunlevel) if currentRunlevel == targetRunlevel { log.Printf("Container finished init.") break } else if currentRunlevel > targetRunlevel { log.Printf("Expected Runlevel %s, Got Runlevel %s, continuing", targetRunlevel, currentRunlevel) break } } return nil }