mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
devmapper: Add DeviceSet device-mapper helper
This is a module that uses the device-mapper create CoW snapshots You instantiate a DeviceSetDM object on a specified root (/var/lib/docker), and it will create a subdirectory there called "loopback". It will contain two sparse files which are loopback mounted into a thin-pool device-mapper device called "docker-pool". We then create a base snapshot in the pool with an empty filesystem which can be used as a base for docker snapshots. It also keeps track of the mapping between docker image ids and the snapshots in the pool. Typical use of is something like (without error checking): devices = NewDeviceSetDM("/var/lib/docker") devices.AddDevice(imageId, "") // "" is the base image id devices.MountDevice(imageId, "/mnt/image") ... extract base image to /mnt/image devices.AddDevice(containerId, imageId) devices.MountDevice(containerId, "/mnt/container") ... start container at /mnt/container
This commit is contained in:
parent
739af0a17f
commit
0b12702c0c
1 changed files with 895 additions and 0 deletions
895
devmapper/deviceset_devmapper.go
Normal file
895
devmapper/deviceset_devmapper.go
Normal file
|
@ -0,0 +1,895 @@
|
|||
package devmapper
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"syscall"
|
||||
)
|
||||
|
||||
const defaultDataLoopbackSize int64 = 100 * 1024 * 1024 * 1024
|
||||
const defaultMetaDataLoopbackSize int64 = 2 * 1024 * 1024 * 1024
|
||||
const defaultBaseFsSize uint64 = 10 * 1024 * 1024 * 1024
|
||||
|
||||
type DevInfo struct {
|
||||
Hash string `json:"-"`
|
||||
DeviceId int `json:"device_id"`
|
||||
Size uint64 `json:"size"`
|
||||
TransactionId uint64 `json:"transaction_id"`
|
||||
Initialized bool `json:"initialized"`
|
||||
}
|
||||
|
||||
type MetaData struct {
|
||||
Devices map[string]*DevInfo `json:devices`
|
||||
}
|
||||
|
||||
type DeviceSetDM struct {
|
||||
initialized bool
|
||||
root string
|
||||
MetaData
|
||||
TransactionId uint64
|
||||
NewTransactionId uint64
|
||||
nextFreeDevice int
|
||||
}
|
||||
|
||||
func getDevName(name string) string {
|
||||
return "/dev/mapper/" + name
|
||||
}
|
||||
|
||||
func (info *DevInfo) Name() string {
|
||||
hash := info.Hash
|
||||
if hash == "" {
|
||||
hash = "base"
|
||||
}
|
||||
return fmt.Sprintf("docker-%s", hash)
|
||||
}
|
||||
|
||||
func (info *DevInfo) DevName() string {
|
||||
return getDevName(info.Name())
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) loopbackDir() string {
|
||||
return path.Join(devices.root, "loopback")
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) jsonFile() string {
|
||||
return path.Join(devices.loopbackDir(), "json")
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) getPoolName() string {
|
||||
return "docker-pool"
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) getPoolDevName() string {
|
||||
return getDevName(devices.getPoolName())
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) createTask(t TaskType, name string) (*Task, error) {
|
||||
task := TaskCreate(t)
|
||||
if task == nil {
|
||||
return nil, fmt.Errorf("Can't create task of type %d", int(t))
|
||||
}
|
||||
err := task.SetName(name)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Can't set task name %s", name)
|
||||
}
|
||||
return task, nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) getInfo(name string) (*Info, error) {
|
||||
task, err := devices.createTask(DeviceInfo, name)
|
||||
if task == nil {
|
||||
return nil, err
|
||||
}
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
info, err := task.GetInfo()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return info, nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) getStatus(name string) (uint64, uint64, string, string, error) {
|
||||
task, err := devices.createTask(DeviceStatus, name)
|
||||
if task == nil {
|
||||
return 0, 0, "", "", err
|
||||
}
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return 0, 0, "", "", err
|
||||
}
|
||||
|
||||
devinfo, err := task.GetInfo()
|
||||
if err != nil {
|
||||
return 0, 0, "", "", err
|
||||
}
|
||||
if devinfo.Exists == 0 {
|
||||
return 0, 0, "", "", fmt.Errorf("Non existing device %s", name)
|
||||
}
|
||||
|
||||
var next uintptr = 0
|
||||
next, start, length, target_type, params := task.GetNextTarget(next)
|
||||
|
||||
return start, length, target_type, params, nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) setTransactionId(oldId uint64, newId uint64) error {
|
||||
task, err := devices.createTask(DeviceTargetMsg, devices.getPoolDevName())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = task.SetSector(0)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set sector")
|
||||
}
|
||||
|
||||
message := fmt.Sprintf("set_transaction_id %d %d", oldId, newId)
|
||||
err = task.SetMessage(message)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set message")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running setTransactionId")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) hasImage(name string) bool {
|
||||
dirname := devices.loopbackDir()
|
||||
filename := path.Join(dirname, name)
|
||||
|
||||
_, err := os.Stat(filename)
|
||||
return err == nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) ensureImage(name string, size int64) (string, error) {
|
||||
dirname := devices.loopbackDir()
|
||||
filename := path.Join(dirname, name)
|
||||
|
||||
if err := os.MkdirAll(dirname, 0700); err != nil && !os.IsExist(err) {
|
||||
return "", err
|
||||
}
|
||||
|
||||
_, err := os.Stat(filename)
|
||||
if err != nil {
|
||||
if !os.IsNotExist(err) {
|
||||
return "", err
|
||||
}
|
||||
log.Printf("Creating loopback file %s for device-manage use", filename)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE, 0600)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
err = file.Truncate(size)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
}
|
||||
return filename, nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) createPool(dataFile *os.File, metadataFile *os.File) error {
|
||||
log.Printf("Activating device-mapper pool %s", devices.getPoolName())
|
||||
task, err := devices.createTask(DeviceCreate, devices.getPoolName())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
size, err := GetBlockDeviceSize(dataFile)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't get data size")
|
||||
}
|
||||
|
||||
params := metadataFile.Name() + " " + dataFile.Name() + " 512 8192"
|
||||
err = task.AddTarget(0, size/512, "thin-pool", params)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't add target")
|
||||
}
|
||||
|
||||
var cookie uint32 = 0
|
||||
err = task.SetCookie(&cookie, 32)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set cookie")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running DeviceCreate")
|
||||
}
|
||||
|
||||
UdevWait(cookie)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) suspendDevice(info *DevInfo) error {
|
||||
task, err := devices.createTask(DeviceSuspend, info.Name())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running DeviceSuspend")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) resumeDevice(info *DevInfo) error {
|
||||
task, err := devices.createTask(DeviceResume, info.Name())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var cookie uint32 = 0
|
||||
err = task.SetCookie(&cookie, 32)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set cookie")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running DeviceSuspend")
|
||||
}
|
||||
|
||||
UdevWait(cookie)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) createDevice(deviceId int) error {
|
||||
task, err := devices.createTask(DeviceTargetMsg, devices.getPoolDevName())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = task.SetSector(0)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set sector")
|
||||
}
|
||||
|
||||
message := fmt.Sprintf("create_thin %d", deviceId)
|
||||
err = task.SetMessage(message)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set message")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running createDevice")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) createSnapDevice(deviceId int, baseInfo *DevInfo) error {
|
||||
doSuspend := false
|
||||
devinfo, _ := devices.getInfo(baseInfo.Name())
|
||||
if devinfo != nil && devinfo.Exists != 0 {
|
||||
doSuspend = true
|
||||
}
|
||||
|
||||
if doSuspend {
|
||||
err := devices.suspendDevice(baseInfo)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
task, err := devices.createTask(DeviceTargetMsg, devices.getPoolDevName())
|
||||
if task == nil {
|
||||
_ = devices.resumeDevice(baseInfo)
|
||||
return err
|
||||
}
|
||||
err = task.SetSector(0)
|
||||
if err != nil {
|
||||
_ = devices.resumeDevice(baseInfo)
|
||||
return fmt.Errorf("Can't set sector")
|
||||
}
|
||||
|
||||
message := fmt.Sprintf("create_snap %d %d", deviceId, baseInfo.DeviceId)
|
||||
err = task.SetMessage(message)
|
||||
if err != nil {
|
||||
_ = devices.resumeDevice(baseInfo)
|
||||
return fmt.Errorf("Can't set message")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
_ = devices.resumeDevice(baseInfo)
|
||||
return fmt.Errorf("Error running DeviceCreate")
|
||||
}
|
||||
|
||||
if doSuspend {
|
||||
err = devices.resumeDevice(baseInfo)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) deleteDevice(deviceId int) error {
|
||||
task, err := devices.createTask(DeviceTargetMsg, devices.getPoolDevName())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = task.SetSector(0)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set sector")
|
||||
}
|
||||
|
||||
message := fmt.Sprintf("delete %d", deviceId)
|
||||
err = task.SetMessage(message)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set message")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running deleteDevice")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) removeDevice(info *DevInfo) error {
|
||||
task, err := devices.createTask(DeviceRemove, info.Name())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running removeDevice")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) activateDevice(info *DevInfo) error {
|
||||
task, err := devices.createTask(DeviceCreate, info.Name())
|
||||
if task == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
params := fmt.Sprintf("%s %d", devices.getPoolDevName(), info.DeviceId)
|
||||
err = task.AddTarget(0, info.Size/512, "thin", params)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't add target")
|
||||
}
|
||||
|
||||
var cookie uint32 = 0
|
||||
err = task.SetCookie(&cookie, 32)
|
||||
if err != nil {
|
||||
return fmt.Errorf("Can't set cookie")
|
||||
}
|
||||
|
||||
err = task.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Error running DeviceCreate")
|
||||
}
|
||||
|
||||
UdevWait(cookie)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) allocateDeviceId() int {
|
||||
// TODO: Add smarter reuse of deleted devices
|
||||
id := devices.nextFreeDevice
|
||||
devices.nextFreeDevice = devices.nextFreeDevice + 1
|
||||
return id
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) allocateTransactionId() uint64 {
|
||||
devices.NewTransactionId = devices.NewTransactionId + 1
|
||||
return devices.NewTransactionId
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) saveMetadata() error {
|
||||
jsonData, err := json.Marshal(devices.MetaData)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tmpFile, err := ioutil.TempFile(filepath.Dir(devices.jsonFile()), ".json")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
n, err := tmpFile.Write(jsonData)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if n < len(jsonData) {
|
||||
err = io.ErrShortWrite
|
||||
}
|
||||
err = tmpFile.Sync()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = tmpFile.Close()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = os.Rename(tmpFile.Name(), devices.jsonFile())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if devices.NewTransactionId != devices.TransactionId {
|
||||
err = devices.setTransactionId(devices.TransactionId, devices.NewTransactionId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
devices.TransactionId = devices.NewTransactionId
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) registerDevice(id int, hash string, size uint64) (*DevInfo, error) {
|
||||
transaction := devices.allocateTransactionId()
|
||||
|
||||
info := &DevInfo{
|
||||
Hash: hash,
|
||||
DeviceId: id,
|
||||
Size: size,
|
||||
TransactionId: transaction,
|
||||
Initialized: false,
|
||||
}
|
||||
|
||||
devices.Devices[hash] = info
|
||||
err := devices.saveMetadata()
|
||||
if err != nil {
|
||||
// Try to remove unused device
|
||||
devices.Devices[hash] = nil
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return info, nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) activateDeviceIfNeeded(hash string) error {
|
||||
info := devices.Devices[hash]
|
||||
if info == nil {
|
||||
return fmt.Errorf("Unknown device %s", hash)
|
||||
}
|
||||
|
||||
name := info.Name()
|
||||
devinfo, _ := devices.getInfo(name)
|
||||
if devinfo != nil && devinfo.Exists != 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
return devices.activateDevice(info)
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) createFilesystem(info *DevInfo) error {
|
||||
devname := info.DevName()
|
||||
|
||||
err := exec.Command("mkfs.ext4", "-E",
|
||||
"discard,lazy_itable_init=0,lazy_journal_init=0", devname).Run()
|
||||
if err != nil {
|
||||
err = exec.Command("mkfs.ext4", "-E",
|
||||
"discard,lazy_itable_init=0", devname).Run()
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) loadMetaData() error {
|
||||
_, _, _, params, err := devices.getStatus(devices.getPoolName())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
var currentTransaction uint64
|
||||
_, err = fmt.Sscanf(params, "%d", ¤tTransaction)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
devices.TransactionId = currentTransaction
|
||||
devices.NewTransactionId = devices.TransactionId
|
||||
|
||||
jsonData, err := ioutil.ReadFile(devices.jsonFile())
|
||||
if err != nil && !os.IsNotExist(err) {
|
||||
return err
|
||||
}
|
||||
|
||||
metadata := &MetaData{
|
||||
Devices: make(map[string]*DevInfo),
|
||||
}
|
||||
if jsonData != nil {
|
||||
if err := json.Unmarshal(jsonData, metadata); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
devices.MetaData = *metadata
|
||||
|
||||
for hash, d := range devices.Devices {
|
||||
d.Hash = hash
|
||||
|
||||
if d.DeviceId >= devices.nextFreeDevice {
|
||||
devices.nextFreeDevice = d.DeviceId + 1
|
||||
}
|
||||
|
||||
// If the transaction id is larger than the actual one we lost the device due to some crash
|
||||
if d.TransactionId > currentTransaction {
|
||||
log.Printf("Removing lost device %s with id %d", hash, d.TransactionId)
|
||||
delete(devices.Devices, hash)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) createBaseLayer(dir string) error {
|
||||
for pth, typ := range map[string]string{
|
||||
"/dev/pts": "dir",
|
||||
"/dev/shm": "dir",
|
||||
"/proc": "dir",
|
||||
"/sys": "dir",
|
||||
"/.dockerinit": "file",
|
||||
"/etc/resolv.conf": "file",
|
||||
"/etc/hosts": "file",
|
||||
"/etc/hostname": "file",
|
||||
// "var/run": "dir",
|
||||
// "var/lock": "dir",
|
||||
} {
|
||||
if _, err := os.Stat(path.Join(dir, pth)); err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
switch typ {
|
||||
case "dir":
|
||||
if err := os.MkdirAll(path.Join(dir, pth), 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
case "file":
|
||||
if err := os.MkdirAll(path.Join(dir, path.Dir(pth)), 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if f, err := os.OpenFile(path.Join(dir, pth), os.O_CREATE, 0755); err != nil {
|
||||
return err
|
||||
} else {
|
||||
f.Close()
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) setupBaseImage() error {
|
||||
oldInfo := devices.Devices[""]
|
||||
if oldInfo != nil && oldInfo.Initialized {
|
||||
return nil
|
||||
}
|
||||
|
||||
if oldInfo != nil && !oldInfo.Initialized {
|
||||
log.Printf("Removing uninitialized base image")
|
||||
if err := devices.RemoveDevice(""); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
log.Printf("Initializing base device-manager snapshot")
|
||||
|
||||
id := devices.allocateDeviceId()
|
||||
|
||||
// Create initial device
|
||||
err := devices.createDevice(id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
info, err := devices.registerDevice(id, "", defaultBaseFsSize)
|
||||
if err != nil {
|
||||
_ = devices.deleteDevice(id)
|
||||
return err
|
||||
}
|
||||
|
||||
log.Printf("Creating filesystem on base device-manager snapshot")
|
||||
|
||||
err = devices.activateDeviceIfNeeded("")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = devices.createFilesystem(info)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tmpDir := path.Join(devices.loopbackDir(), "basefs")
|
||||
if err = os.MkdirAll(tmpDir, 0700); err != nil && !os.IsExist(err) {
|
||||
return err
|
||||
}
|
||||
|
||||
err = devices.MountDevice("", tmpDir)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = devices.createBaseLayer(tmpDir)
|
||||
if err != nil {
|
||||
_ = syscall.Unmount(tmpDir, 0)
|
||||
return err
|
||||
}
|
||||
|
||||
err = syscall.Unmount(tmpDir, 0)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_ = os.Remove(tmpDir)
|
||||
|
||||
info.Initialized = true
|
||||
|
||||
err = devices.saveMetadata()
|
||||
if err != nil {
|
||||
info.Initialized = false
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) initDevmapper() error {
|
||||
info, err := devices.getInfo(devices.getPoolName())
|
||||
if info == nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if info.Exists != 0 {
|
||||
/* Pool exists, assume everything is up */
|
||||
err = devices.loadMetaData()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = devices.setupBaseImage()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
createdLoopback := false
|
||||
if !devices.hasImage("data") || !devices.hasImage("metadata") {
|
||||
/* If we create the loopback mounts we also need to initialize the base fs */
|
||||
createdLoopback = true
|
||||
}
|
||||
|
||||
data, err := devices.ensureImage("data", defaultDataLoopbackSize)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
metadata, err := devices.ensureImage("metadata", defaultMetaDataLoopbackSize)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
dataFile, err := AttachLoopDevice(data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer dataFile.Close()
|
||||
|
||||
metadataFile, err := AttachLoopDevice(metadata)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer metadataFile.Close()
|
||||
|
||||
err = devices.createPool(dataFile, metadataFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if !createdLoopback {
|
||||
err = devices.loadMetaData()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
err = devices.setupBaseImage()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) AddDevice(hash, baseHash string) error {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if devices.Devices[hash] != nil {
|
||||
return fmt.Errorf("hash %s already exists", hash)
|
||||
}
|
||||
|
||||
baseInfo := devices.Devices[baseHash]
|
||||
if baseInfo == nil {
|
||||
return fmt.Errorf("Unknown base hash %s", baseHash)
|
||||
}
|
||||
|
||||
deviceId := devices.allocateDeviceId()
|
||||
|
||||
err := devices.createSnapDevice(deviceId, baseInfo)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = devices.registerDevice(deviceId, hash, baseInfo.Size)
|
||||
if err != nil {
|
||||
_ = devices.deleteDevice(deviceId)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) RemoveDevice(hash string) error {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
info := devices.Devices[hash]
|
||||
if info == nil {
|
||||
return fmt.Errorf("hash %s doesn't exists", hash)
|
||||
}
|
||||
|
||||
devinfo, _ := devices.getInfo(info.Name())
|
||||
if devinfo != nil && devinfo.Exists != 0 {
|
||||
err := devices.removeDevice(info)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if info.Initialized {
|
||||
info.Initialized = false
|
||||
err := devices.saveMetadata()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
err := devices.deleteDevice(info.DeviceId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_ = devices.allocateTransactionId()
|
||||
delete(devices.Devices, info.Hash)
|
||||
|
||||
err = devices.saveMetadata()
|
||||
if err != nil {
|
||||
devices.Devices[info.Hash] = info
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) DeactivateDevice(hash string) error {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
info := devices.Devices[hash]
|
||||
if info == nil {
|
||||
return fmt.Errorf("hash %s doesn't exists", hash)
|
||||
}
|
||||
|
||||
devinfo, err := devices.getInfo(info.Name())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if devinfo.Exists != 0 {
|
||||
err := devices.removeDevice(info)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) MountDevice(hash, path string) error {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err := devices.activateDeviceIfNeeded(hash)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
info := devices.Devices[hash]
|
||||
|
||||
err = syscall.Mount(info.DevName(), path, "ext4", syscall.MS_MGC_VAL, "discard")
|
||||
if err != nil && err == syscall.EINVAL {
|
||||
err = syscall.Mount(info.DevName(), path, "ext4", syscall.MS_MGC_VAL, "")
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) HasDevice(hash string) bool {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
info := devices.Devices[hash]
|
||||
return info != nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) HasInitializedDevice(hash string) bool {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
info := devices.Devices[hash]
|
||||
return info != nil && info.Initialized
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) SetInitialized(hash string) error {
|
||||
if err := devices.ensureInit(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
info := devices.Devices[hash]
|
||||
if info == nil {
|
||||
return fmt.Errorf("Unknown device %s", hash)
|
||||
}
|
||||
|
||||
info.Initialized = true
|
||||
err := devices.saveMetadata()
|
||||
if err != nil {
|
||||
info.Initialized = false
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (devices *DeviceSetDM) ensureInit() error {
|
||||
if (!devices.initialized) {
|
||||
devices.initialized = true
|
||||
err := devices.initDevmapper()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewDeviceSetDM(root string) *DeviceSetDM {
|
||||
SetDevDir("/dev")
|
||||
devices := &DeviceSetDM{
|
||||
initialized: false,
|
||||
root: root,
|
||||
}
|
||||
devices.Devices = make(map[string]*DevInfo)
|
||||
|
||||
return devices
|
||||
}
|
Loading…
Add table
Reference in a new issue