-
Notifications
You must be signed in to change notification settings - Fork 4.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Initial hyperkit driver implementation.
- Loading branch information
Showing
9 changed files
with
494 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,10 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "github.com/docker/machine/libmachine/drivers/plugin" | ||
| "k8s.io/minikube/pkg/minikube/drivers/hyperkit" | ||
| ) | ||
|
|
||
| func main() { | ||
| plugin.RegisterDriver(hyperkit.NewDriver("", "")) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| package hyperkit | ||
|
|
||
| import ( | ||
| "io/ioutil" | ||
| "os" | ||
| "path/filepath" | ||
| "syscall" | ||
|
|
||
| "github.com/cloudflare/cfssl/log" | ||
| "github.com/docker/machine/libmachine/mcnutils" | ||
| ) | ||
|
|
||
| func createDiskImage(sshKeyPath, diskPath string, diskSize int) error { | ||
| tarBuf, err := mcnutils.MakeDiskImage(sshKeyPath) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| file, err := os.OpenFile(diskPath, os.O_CREATE|os.O_EXCL|os.O_WRONLY, 0644) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| defer file.Close() | ||
| file.Seek(0, os.SEEK_SET) | ||
|
|
||
| if _, err := file.Write(tarBuf.Bytes()); err != nil { | ||
| return err | ||
| } | ||
| file.Close() | ||
|
|
||
| if err := os.Truncate(diskPath, int64(diskSize*1048576)); err != nil { | ||
| return err | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func fixPermissions(path string) error { | ||
| os.Chown(path, syscall.Getuid(), syscall.Getegid()) | ||
| files, _ := ioutil.ReadDir(path) | ||
| for _, f := range files { | ||
| fp := filepath.Join(path, f.Name()) | ||
| log.Debugf(fp) | ||
| if err := os.Chown(fp, syscall.Getuid(), syscall.Getegid()); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,247 @@ | ||
| package hyperkit | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
| "path/filepath" | ||
| "syscall" | ||
| "time" | ||
|
|
||
| "github.com/docker/machine/libmachine/drivers" | ||
| "github.com/docker/machine/libmachine/log" | ||
| "github.com/docker/machine/libmachine/mcnflag" | ||
| "github.com/docker/machine/libmachine/mcnutils" | ||
| "github.com/docker/machine/libmachine/ssh" | ||
| "github.com/docker/machine/libmachine/state" | ||
| hyperkit "github.com/moby/hyperkit/go" | ||
| "github.com/pborman/uuid" | ||
| vmnet "github.com/zchee/go-vmnet" | ||
| commonutil "k8s.io/minikube/pkg/util" | ||
| ) | ||
|
|
||
| const ( | ||
| isoFilename = "boot2docker.iso" | ||
| ) | ||
|
|
||
| type Driver struct { | ||
| *drivers.BaseDriver | ||
| Boot2DockerURL string | ||
| DiskSize int | ||
| Pid int | ||
| CPU int | ||
| Memory int | ||
| Cmdline string | ||
| } | ||
|
|
||
| func NewDriver(hostName, storePath string) *Driver { | ||
| return &Driver{ | ||
| BaseDriver: &drivers.BaseDriver{ | ||
| SSHUser: "docker", | ||
| }, | ||
| } | ||
| } | ||
|
|
||
| func (d *Driver) Create() error { | ||
| b2dutils := mcnutils.NewB2dUtils(d.StorePath) | ||
|
|
||
| if err := ssh.GenerateSSHKey(d.GetSSHKeyPath()); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := b2dutils.CopyIsoToMachineDir(d.Boot2DockerURL, d.MachineName); err != nil { | ||
| return err | ||
| } | ||
| isoPath := d.ResolveStorePath(isoFilename) | ||
| if err := d.extractKernel(isoPath); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return d.Start() | ||
| } | ||
|
|
||
| // DriverName returns the name of the driver | ||
| func (d *Driver) DriverName() string { | ||
| return "hyperkit" | ||
| } | ||
|
|
||
| // GetCreateFlags returns the mcnflag.Flag slice representing the flags | ||
| // that can be set, their descriptions and defaults. | ||
| func (d *Driver) GetCreateFlags() []mcnflag.Flag { | ||
| return nil | ||
| } | ||
|
|
||
| // GetSSHHostname returns hostname for use with ssh | ||
| func (d *Driver) GetSSHHostname() (string, error) { | ||
| return d.IPAddress, nil | ||
| } | ||
|
|
||
| // GetURL returns a Docker compatible host URL for connecting to this host | ||
| // e.g. tcp://1.2.3.4:2376 | ||
| func (d *Driver) GetURL() (string, error) { | ||
| ip, err := d.GetIP() | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| return fmt.Sprintf("tcp://%s:2376", ip), nil | ||
| } | ||
|
|
||
| // GetState returns the state that the host is in (running, stopped, etc) | ||
| func (d *Driver) GetState() (state.State, error) { | ||
| if d.Pid == 0 { | ||
| return state.Stopped, nil | ||
| } | ||
| p, err := os.FindProcess(d.Pid) | ||
| if err != nil { | ||
| return state.Error, err | ||
| } | ||
| if p == nil { | ||
| return state.Stopped, nil | ||
| } | ||
| return state.Running, nil | ||
| } | ||
|
|
||
| // Kill stops a host forcefully | ||
| func (d *Driver) Kill() error { | ||
| return d.sendSignal(syscall.SIGKILL) | ||
| } | ||
|
|
||
| // PreCreateCheck allows for pre-create operations to make sure a driver is ready for creation | ||
| func (d *Driver) PreCreateCheck() error { | ||
| return nil | ||
| } | ||
|
|
||
| // Remove a host | ||
| func (d *Driver) Remove() error { | ||
| s, err := d.GetState() | ||
| if err != nil || s == state.Error { | ||
| log.Infof("Error checking machine status: %s, assuming it has been removed already", err) | ||
| return nil | ||
| } | ||
| if s == state.Running { | ||
| if err := d.Stop(); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // Restart a host. This may just call Stop(); Start() if the provider does not | ||
| // have any special restart behaviour. | ||
| func (d *Driver) Restart() error { | ||
| for _, f := range []func() error{d.Stop, d.Start} { | ||
| if err := f(); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // SetConfigFromFlags configures the driver with the object that was returned | ||
| // by RegisterCreateFlags | ||
| func (d *Driver) SetConfigFromFlags(opts drivers.DriverOptions) error { | ||
| return nil | ||
| } | ||
|
|
||
| // Start a host | ||
| func (d *Driver) Start() error { | ||
|
|
||
| // TODO: handle the disk already existing. | ||
| // TODO: handle different disk types. | ||
| diskPath := filepath.Join(d.ResolveStorePath("."), d.MachineName+".rawdisk") | ||
| if err := createDiskImage(d.publicSSHKeyPath(), diskPath, d.DiskSize); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := fixPermissions(d.ResolveStorePath(".")); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| h, err := hyperkit.New("", "", filepath.Join(d.StorePath, "machines", d.MachineName)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| // TODO: handle the rest of our settings. | ||
| h.Kernel = d.ResolveStorePath("bzimage") | ||
| h.Initrd = d.ResolveStorePath("initrd") | ||
| h.VMNet = true | ||
| h.ISOImage = d.ResolveStorePath(isoFilename) | ||
| h.Console = hyperkit.ConsoleFile | ||
| h.CPUs = d.CPU | ||
| h.Memory = d.Memory | ||
|
|
||
| // Set UUID | ||
| h.UUID = uuid.NewUUID().String() | ||
| log.Infof("Generated UUID %s", h.UUID) | ||
| mac, err := vmnet.GetMACAddressFromUUID(h.UUID) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| // Need to strip 0's | ||
| mac = trimMacAddress(mac) | ||
| log.Infof("Generated MAC %s", mac) | ||
|
|
||
| h.Disks = []hyperkit.DiskConfig{ | ||
| { | ||
| Path: diskPath, | ||
| Size: d.DiskSize, | ||
| Driver: "virtio-blk", | ||
| }, | ||
| } | ||
| log.Infof("Starting with cmdline: %s", d.Cmdline) | ||
| if err := h.Start(d.Cmdline); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| d.Pid = h.Pid | ||
|
|
||
| getIP := func() error { | ||
| var err error | ||
| d.IPAddress, err = GetIPAddressByMACAddress(mac) | ||
| if err != nil { | ||
| return &commonutil.RetriableError{Err: err} | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| if err := commonutil.RetryAfter(30, getIP, 2*time.Second); err != nil { | ||
| return fmt.Errorf("IP address never found in dhcp leases file %v", err) | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // Stop a host gracefully | ||
| func (d *Driver) Stop() error { | ||
| return d.sendSignal(syscall.SIGTERM) | ||
| } | ||
|
|
||
| func (d *Driver) extractKernel(isoPath string) error { | ||
| for _, f := range []struct { | ||
| pathInIso string | ||
| destPath string | ||
| }{ | ||
| {"/boot/bzimage", "bzimage"}, | ||
| {"/boot/initrd", "initrd"}, | ||
| {"/isolinux/isolinux.cfg", "isolinux.cfg"}, | ||
| } { | ||
| fullDestPath := d.ResolveStorePath(f.destPath) | ||
| if err := ExtractFile(isoPath, f.pathInIso, fullDestPath); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func (d *Driver) publicSSHKeyPath() string { | ||
| return d.GetSSHKeyPath() + ".pub" | ||
| } | ||
|
|
||
| func (d *Driver) sendSignal(s os.Signal) error { | ||
| proc, err := os.FindProcess(d.Pid) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return proc.Signal(s) | ||
| } |
Oops, something went wrong.