mirror of
https://github.com/aljazceru/kata-containers.git
synced 2026-01-21 23:34:22 +01:00
bdd5c66fc5
Add vm factory support per design in the VM Factory plugin section. The vm factory controls how a new vm is created: 1. direct: vm is created directly 2. template: vm is created via vm template. A template vm is pre-created and saved. Later vm is just a clone of the template vm so that they readonly share a portion of initial memory (including kernel, initramfs and the kata agent). CPU and memory are hot plugged when necessary. 3. cache: vm is created via vm caches. A set of cached vm are pre-created and maintained alive. New vms are created by just picking a cached vm. CPU and memory are hot plugged when necessary. Fixes: #303 Signed-off-by: Peng Tao <bergwolf@gmail.com>
144 lines
3.2 KiB
Go
144 lines
3.2 KiB
Go
// Copyright (c) 2018 HyperHQ Inc.
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// template implements base vm factory with vm templating.
|
|
|
|
package template
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"syscall"
|
|
"time"
|
|
|
|
vc "github.com/kata-containers/runtime/virtcontainers"
|
|
"github.com/kata-containers/runtime/virtcontainers/factory/base"
|
|
"github.com/kata-containers/runtime/virtcontainers/factory/direct"
|
|
)
|
|
|
|
type template struct {
|
|
statePath string
|
|
config vc.VMConfig
|
|
}
|
|
|
|
// Fetch finds and returns a pre-built template factory.
|
|
// TODO: save template metadata and fetch from storage.
|
|
func Fetch(config vc.VMConfig) (base.FactoryBase, error) {
|
|
statePath := vc.RunVMStoragePath + "/template"
|
|
t := &template{statePath, config}
|
|
|
|
err := t.checkTemplateVM()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return t, nil
|
|
}
|
|
|
|
// New creates a new VM template factory.
|
|
func New(config vc.VMConfig) base.FactoryBase {
|
|
statePath := vc.RunVMStoragePath + "/template"
|
|
t := &template{statePath, config}
|
|
|
|
err := t.prepareTemplateFiles()
|
|
if err != nil {
|
|
// fallback to direct factory if template is not supported.
|
|
return direct.New(config)
|
|
}
|
|
|
|
err = t.createTemplateVM()
|
|
if err != nil {
|
|
// fallback to direct factory if template is not supported.
|
|
return direct.New(config)
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// Config returns template factory's configuration.
|
|
func (t *template) Config() vc.VMConfig {
|
|
return t.config
|
|
}
|
|
|
|
// GetBaseVM creates a new paused VM from the template VM.
|
|
func (t *template) GetBaseVM() (*vc.VM, error) {
|
|
return t.createFromTemplateVM()
|
|
}
|
|
|
|
// CloseFactory cleans up the template VM.
|
|
func (t *template) CloseFactory() {
|
|
syscall.Unmount(t.statePath, 0)
|
|
os.RemoveAll(t.statePath)
|
|
}
|
|
|
|
func (t *template) prepareTemplateFiles() error {
|
|
// create and mount tmpfs for the shared memory file
|
|
err := os.MkdirAll(t.statePath, 0700)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
flags := uintptr(syscall.MS_NOSUID | syscall.MS_NODEV)
|
|
opts := fmt.Sprintf("size=%dM", t.config.HypervisorConfig.DefaultMemSz+8)
|
|
if err = syscall.Mount("tmpfs", t.statePath, "tmpfs", flags, opts); err != nil {
|
|
return err
|
|
}
|
|
f, err := os.Create(t.statePath + "/memory")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f.Close()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (t *template) createTemplateVM() error {
|
|
// create the template vm
|
|
config := t.config
|
|
config.HypervisorConfig.BootToBeTemplate = true
|
|
config.HypervisorConfig.BootFromTemplate = false
|
|
config.HypervisorConfig.MemoryPath = t.statePath + "/memory"
|
|
config.HypervisorConfig.DevicesStatePath = t.statePath + "/state"
|
|
|
|
vm, err := vc.NewVM(config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer vm.Stop()
|
|
|
|
err = vm.Pause()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = vm.Save()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// qemu QMP does not wait for migration to finish...
|
|
time.Sleep(1 * time.Second)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (t *template) createFromTemplateVM() (*vc.VM, error) {
|
|
config := t.config
|
|
config.HypervisorConfig.BootToBeTemplate = false
|
|
config.HypervisorConfig.BootFromTemplate = true
|
|
config.HypervisorConfig.MemoryPath = t.statePath + "/memory"
|
|
config.HypervisorConfig.DevicesStatePath = t.statePath + "/state"
|
|
|
|
return vc.NewVM(config)
|
|
}
|
|
|
|
func (t *template) checkTemplateVM() error {
|
|
_, err := os.Stat(t.statePath + "/memory")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = os.Stat(t.statePath + "/state")
|
|
return err
|
|
}
|