packer-cn/builder/virtualbox/builder.go

317 lines
7.7 KiB
Go
Raw Normal View History

2013-06-11 18:12:45 -04:00
package virtualbox
import (
2013-06-11 23:00:30 -04:00
"errors"
"fmt"
2013-06-11 18:12:45 -04:00
"github.com/mitchellh/mapstructure"
"github.com/mitchellh/multistep"
2013-06-15 13:51:09 -04:00
"github.com/mitchellh/packer/builder/common"
2013-06-11 18:12:45 -04:00
"github.com/mitchellh/packer/packer"
"log"
2013-06-11 23:00:30 -04:00
"net/url"
"os"
"os/exec"
"path/filepath"
2013-06-11 23:00:30 -04:00
"strings"
2013-06-11 23:56:44 -04:00
"time"
2013-06-11 18:12:45 -04:00
)
const BuilderId = "mitchellh.virtualbox"
type Builder struct {
config config
driver Driver
2013-06-11 18:12:45 -04:00
runner multistep.Runner
}
type config struct {
BootCommand []string `mapstructure:"boot_command"`
BootWait time.Duration ``
DiskSize uint `mapstructure:"disk_size"`
GuestAdditionsPath string `mapstructure:"guest_additions_path"`
GuestOSType string `mapstructure:"guest_os_type"`
HTTPDir string `mapstructure:"http_directory"`
HTTPPortMin uint `mapstructure:"http_port_min"`
HTTPPortMax uint `mapstructure:"http_port_max"`
ISOMD5 string `mapstructure:"iso_md5"`
ISOUrl string `mapstructure:"iso_url"`
OutputDir string `mapstructure:"output_directory"`
ShutdownCommand string `mapstructure:"shutdown_command"`
ShutdownTimeout time.Duration ``
SSHHostPortMin uint `mapstructure:"ssh_host_port_min"`
SSHHostPortMax uint `mapstructure:"ssh_host_port_max"`
SSHPassword string `mapstructure:"ssh_password"`
SSHPort uint `mapstructure:"ssh_port"`
SSHUser string `mapstructure:"ssh_username"`
SSHWaitTimeout time.Duration ``
VBoxVersionFile string `mapstructure:"virtualbox_version_file"`
VBoxManage [][]string `mapstructure:"vboxmanage"`
VMName string `mapstructure:"vm_name"`
2013-06-13 13:24:10 -04:00
2013-06-15 13:51:09 -04:00
PackerDebug bool `mapstructure:"packer_debug"`
2013-06-13 13:24:10 -04:00
RawBootWait string `mapstructure:"boot_wait"`
2013-06-12 21:02:42 -04:00
RawShutdownTimeout string `mapstructure:"shutdown_timeout"`
2013-06-13 13:24:10 -04:00
RawSSHWaitTimeout string `mapstructure:"ssh_wait_timeout"`
2013-06-11 18:12:45 -04:00
}
2013-06-14 15:29:48 -04:00
func (b *Builder) Prepare(raws ...interface{}) error {
var err error
2013-06-14 15:29:48 -04:00
for _, raw := range raws {
err := mapstructure.Decode(raw, &b.config)
if err != nil {
return err
}
2013-06-11 18:12:45 -04:00
}
if b.config.DiskSize == 0 {
b.config.DiskSize = 40000
}
if b.config.GuestAdditionsPath == "" {
b.config.GuestAdditionsPath = "VBoxGuestAdditions.iso"
}
if b.config.GuestOSType == "" {
b.config.GuestOSType = "Other"
}
if b.config.HTTPPortMin == 0 {
b.config.HTTPPortMin = 8000
}
if b.config.HTTPPortMax == 0 {
b.config.HTTPPortMax = 9000
}
if b.config.OutputDir == "" {
b.config.OutputDir = "virtualbox"
}
if b.config.RawBootWait == "" {
b.config.RawBootWait = "10s"
}
if b.config.SSHHostPortMin == 0 {
b.config.SSHHostPortMin = 2222
}
if b.config.SSHHostPortMax == 0 {
b.config.SSHHostPortMax = 4444
}
if b.config.SSHPort == 0 {
b.config.SSHPort = 22
}
if b.config.VBoxManage == nil {
b.config.VBoxManage = make([][]string, 0)
}
if b.config.VBoxVersionFile == "" {
b.config.VBoxVersionFile = ".vbox_version"
}
2013-06-11 19:12:19 -04:00
if b.config.VMName == "" {
b.config.VMName = "packer"
}
errs := make([]error, 0)
if b.config.HTTPPortMin > b.config.HTTPPortMax {
errs = append(errs, errors.New("http_port_min must be less than http_port_max"))
}
2013-06-11 23:00:30 -04:00
if b.config.ISOMD5 == "" {
errs = append(errs, errors.New("Due to large file sizes, an iso_md5 is required"))
} else {
b.config.ISOMD5 = strings.ToLower(b.config.ISOMD5)
}
if b.config.ISOUrl == "" {
errs = append(errs, errors.New("An iso_url must be specified."))
} else {
url, err := url.Parse(b.config.ISOUrl)
if err != nil {
errs = append(errs, fmt.Errorf("iso_url is not a valid URL: %s", err))
} else {
if url.Scheme == "" {
url.Scheme = "file"
}
if url.Scheme == "file" {
if _, err := os.Stat(url.Path); err != nil {
2013-06-11 23:00:30 -04:00
errs = append(errs, fmt.Errorf("iso_url points to bad file: %s", err))
}
} else {
supportedSchemes := []string{"file", "http", "https"}
scheme := strings.ToLower(url.Scheme)
found := false
for _, supported := range supportedSchemes {
if scheme == supported {
found = true
break
}
}
if !found {
errs = append(errs, fmt.Errorf("Unsupported URL scheme in iso_url: %s", scheme))
}
}
}
if len(errs) == 0 {
// Put the URL back together since we may have modified it
b.config.ISOUrl = url.String()
}
}
if _, err := os.Stat(b.config.OutputDir); err == nil {
errs = append(errs, errors.New("Output directory already exists. It must not exist."))
}
b.config.BootWait, err = time.ParseDuration(b.config.RawBootWait)
if err != nil {
errs = append(errs, fmt.Errorf("Failed parsing boot_wait: %s", err))
2013-06-11 23:56:44 -04:00
}
2013-06-12 21:02:42 -04:00
if b.config.RawShutdownTimeout == "" {
b.config.RawShutdownTimeout = "5m"
}
if b.config.RawSSHWaitTimeout == "" {
b.config.RawSSHWaitTimeout = "20m"
}
2013-06-12 21:02:42 -04:00
b.config.ShutdownTimeout, err = time.ParseDuration(b.config.RawShutdownTimeout)
if err != nil {
errs = append(errs, fmt.Errorf("Failed parsing shutdown_timeout: %s", err))
}
if b.config.SSHHostPortMin > b.config.SSHHostPortMax {
errs = append(errs, errors.New("ssh_host_port_min must be less than ssh_host_port_max"))
}
if b.config.SSHUser == "" {
errs = append(errs, errors.New("An ssh_username must be specified."))
}
b.config.SSHWaitTimeout, err = time.ParseDuration(b.config.RawSSHWaitTimeout)
if err != nil {
errs = append(errs, fmt.Errorf("Failed parsing ssh_wait_timeout: %s", err))
}
b.driver, err = b.newDriver()
if err != nil {
errs = append(errs, fmt.Errorf("Failed creating VirtualBox driver: %s", err))
}
if len(errs) > 0 {
return &packer.MultiError{errs}
}
2013-06-11 18:12:45 -04:00
return nil
}
func (b *Builder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packer.Artifact, error) {
steps := []multistep.Step{
new(stepDownloadGuestAdditions),
2013-06-11 23:00:30 -04:00
new(stepDownloadISO),
new(stepPrepareOutputDir),
new(stepHTTPServer),
new(stepSuppressMessages),
2013-06-11 19:12:19 -04:00
new(stepCreateVM),
new(stepCreateDisk),
2013-06-11 23:07:11 -04:00
new(stepAttachISO),
new(stepForwardSSH),
new(stepVBoxManage),
2013-06-11 23:46:32 -04:00
new(stepRun),
2013-06-12 03:26:08 -04:00
new(stepTypeBootCommand),
new(stepWaitForSSH),
new(stepUploadVersion),
new(stepUploadGuestAdditions),
2013-06-12 11:47:36 -04:00
new(stepProvision),
2013-06-12 21:02:42 -04:00
new(stepShutdown),
2013-06-12 21:07:08 -04:00
new(stepExport),
}
// Setup the state bag
state := make(map[string]interface{})
state["cache"] = cache
state["config"] = &b.config
state["driver"] = b.driver
state["hook"] = hook
state["ui"] = ui
// Run
2013-06-15 13:51:09 -04:00
if b.config.PackerDebug {
b.runner = &multistep.DebugRunner{
Steps: steps,
PauseFn: common.MultistepDebugFn(ui),
}
} else {
b.runner = &multistep.BasicRunner{Steps: steps}
}
b.runner.Run(state)
// If there was an error, return that
if rawErr, ok := state["error"]; ok {
return nil, rawErr.(error)
}
// If we were interrupted or cancelled, then just exit.
if _, ok := state[multistep.StateCancelled]; ok {
return nil, errors.New("Build was cancelled.")
}
if _, ok := state[multistep.StateHalted]; ok {
return nil, errors.New("Build was halted.")
}
// Compile the artifact list
files := make([]string, 0, 5)
visit := func(path string, info os.FileInfo, err error) error {
if !info.IsDir() {
files = append(files, path)
}
return err
}
if err := filepath.Walk(b.config.OutputDir, visit); err != nil {
return nil, err
}
artifact := &Artifact{
dir: b.config.OutputDir,
2013-06-20 00:20:54 -04:00
f: files,
}
return artifact, nil
2013-06-11 18:12:45 -04:00
}
func (b *Builder) Cancel() {
if b.runner != nil {
log.Println("Cancelling the step runner...")
b.runner.Cancel()
}
}
func (b *Builder) newDriver() (Driver, error) {
vboxmanagePath, err := exec.LookPath("VBoxManage")
if err != nil {
return nil, err
}
log.Printf("VBoxManage path: %s", vboxmanagePath)
driver := &VBox42Driver{vboxmanagePath}
if err := driver.Verify(); err != nil {
return nil, err
}
return driver, nil
2013-06-11 18:12:45 -04:00
}