package virtualbox import ( "fmt" "github.com/mitchellh/multistep" "github.com/mitchellh/packer/packer" "io" "io/ioutil" "log" "os" "path/filepath" ) // This step attaches the ISO to the virtual machine. // // Uses: // // Produces: type stepAttachFloppy struct { floppyPath string } func (s *stepAttachFloppy) Run(state map[string]interface{}) multistep.StepAction { // Determine if we even have a floppy disk to attach var floppyPath string if floppyPathRaw, ok := state["floppy_path"]; ok { floppyPath = floppyPathRaw.(string) } else { log.Println("No floppy disk, not attaching.") return multistep.ActionContinue } // VirtualBox is really dumb and can't figure out the format of the file // without an extension, so we need to add the "vfd" extension to the // floppy. floppyPath, err := s.copyFloppy(floppyPath) if err != nil { state["error"] = fmt.Errorf("Error preparing floppy: %s", err) return multistep.ActionHalt } driver := state["driver"].(Driver) ui := state["ui"].(packer.Ui) vmName := state["vmName"].(string) ui.Say("Attaching floppy disk...") // Create the floppy disk controller command := []string{ "storagectl", vmName, "--name", "Floppy Controller", "--add", "floppy", } if err := driver.VBoxManage(command...); err != nil { state["error"] = fmt.Errorf("Error creating floppy controller: %s", err) return multistep.ActionHalt } // Attach the floppy to the controller command = []string{ "storageattach", vmName, "--storagectl", "Floppy Controller", "--port", "0", "--device", "0", "--type", "fdd", "--medium", floppyPath, } if err := driver.VBoxManage(command...); err != nil { state["error"] = fmt.Errorf("Error attaching floppy: %s", err) return multistep.ActionHalt } // Track the path so that we can unregister it from VirtualBox later s.floppyPath = floppyPath return multistep.ActionContinue } func (s *stepAttachFloppy) Cleanup(state map[string]interface{}) { if s.floppyPath == "" { return } // Delete the floppy disk defer os.Remove(s.floppyPath) driver := state["driver"].(Driver) vmName := state["vmName"].(string) command := []string{ "storageattach", vmName, "--storagectl", "Floppy Controller", "--port", "0", "--device", "0", "--medium", "none", } if err := driver.VBoxManage(command...); err != nil { log.Printf("Error unregistering floppy: %s", err) } } func (s *stepAttachFloppy) copyFloppy(path string) (string, error) { tempdir, err := ioutil.TempDir("", "packer") if err != nil { return "", err } floppyPath := filepath.Join(tempdir, "floppy.vfd") f, err := os.Create(floppyPath) if err != nil { return "", err } defer f.Close() sourceF, err := os.Open(path) if err != nil { return "", err } defer sourceF.Close() log.Printf("Copying floppy to temp location: %s", floppyPath) if _, err := io.Copy(f, sourceF); err != nil { return "", err } return floppyPath, nil }