-
Notifications
You must be signed in to change notification settings - Fork 135
/
startvm.go
106 lines (87 loc) · 3.47 KB
/
startvm.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
// Copyright (c) Contributors to the Apptainer project, established as
// Apptainer a Series of LF Projects LLC.
// For website terms of use, trademark policy, privacy policy and other
// project policies see https://lfprojects.org/policies
// Copyright (c) 2019-2021, Sylabs Inc. All rights reserved.
// This software is licensed under a 3-clause BSD license. Please consult the
// LICENSE.md file distributed with the sources of this project regarding your
// rights to use or distribute this software.
package cli
import (
"fmt"
"os"
"os/exec"
"path/filepath"
"runtime"
"strings"
"syscall"
"github.com/apptainer/apptainer/internal/pkg/buildcfg"
"github.com/apptainer/apptainer/pkg/sylog"
"github.com/spf13/cobra"
)
func getHypervisorArgs(sifImage, bzImage, initramfs, singAction, cliExtra string) []string {
// Setup some needed variables
appendArgs := fmt.Sprintf("root=/dev/ram0 console=ttyS0 quiet apptainer_action=%s apptainer_arguments=\"%s\"", singAction, cliExtra)
args := []string{"/usr/libexec/qemu-kvm", "-cpu", "host", "-smp", vmCPU, "-enable-kvm", "-device", "virtio-rng-pci", "-display", "none", "-realtime", "mlock=on", "-hda", sifImage, "-serial", "stdio", "-kernel", bzImage, "-initrd", initramfs, "-m", vmRAM, "-append", appendArgs}
return args
}
func execVM(cmd *cobra.Command, image string, args []string) {
// SIF image we are running
sifImage := image
cliExtra := ""
singAction := cmd.Name()
imgPath := strings.Split(sifImage, ":")
isInternal := false
if strings.HasPrefix("internal", filepath.Base(imgPath[0])) {
cliExtra = "syos"
isInternal = true
} else {
// Get our "action" (run, exec, shell) based on the action script being called
singAction = filepath.Base(args[0])
cliExtra = strings.Join(args[1:], " ")
}
if err := startVM(sifImage, singAction, cliExtra, isInternal); err != nil {
sylog.Errorf("VM instance failed: %s", err)
os.Exit(2)
}
}
func startVM(sifImage, singAction, cliExtra string, isInternal bool) error {
bzImage := fmt.Sprintf("%s/%s-%s", buildcfg.LIBEXECDIR, "/apptainer/vm/syos-kernel", runtime.GOARCH)
initramfs := fmt.Sprintf("%s/%s_%s.gz", buildcfg.LIBEXECDIR, "/apptainer/vm/initramfs", runtime.GOARCH)
args := getHypervisorArgs(sifImage, bzImage, initramfs, singAction, cliExtra)
sylog.Debugf("About to launch VM using: %+v", args)
hvExec, lookErr := exec.LookPath(args[0])
if lookErr != nil {
sylog.Fatalf("Failed to find hypervisor executable at %s", args[0])
}
if _, err := os.Stat(sifImage); os.IsNotExist(err) {
sylog.Fatalf("Failed to determine image absolute path for %s: %s", sifImage, err)
}
if _, err := os.Stat(bzImage); os.IsNotExist(err) {
sylog.Fatalf("This functionality is not supported. For more information visit: https://sylabs.io/singularity-desktop-macos")
}
if _, err := os.Stat(initramfs); os.IsNotExist(err) {
sylog.Fatalf("This functionality is not supported. For more information visit: https://sylabs.io/singularity-desktop-macos")
}
args[0] = "Sylabs"
cmd := exec.Command(hvExec)
cmd.Args = args
cmd.Env = os.Environ()
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
if vmErr || debug {
cmd.Stderr = os.Stderr
}
err := cmd.Run()
if err != nil {
sylog.Debugf("Hypervisor exit code: %v\n", err)
if exitErr, ok := err.(*exec.ExitError); ok {
// Program exited with non-zero return code
if status, ok := exitErr.Sys().(syscall.WaitStatus); ok {
sylog.Debugf("Process exited with non-zero return code: %d\n", status.ExitStatus())
return nil
}
}
}
return err
}