forked from apptainer/singularity
-
Notifications
You must be signed in to change notification settings - Fork 0
/
stage.go
56 lines (49 loc) · 1.44 KB
/
stage.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
// Copyright (c) 2018, 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 starter
import (
"net"
"os"
"github.com/sylabs/singularity/internal/pkg/runtime/engines"
sarterConfig "github.com/sylabs/singularity/internal/pkg/runtime/engines/config/starter"
"github.com/sylabs/singularity/internal/pkg/sylog"
)
// Stage performs container startup.
func Stage(stage, masterSocket int, sconfig *sarterConfig.Config, jsonBytes []byte) {
var conn net.Conn
var err error
if masterSocket != -1 {
comm := os.NewFile(uintptr(masterSocket), "master-socket")
conn, err = net.FileConn(comm)
if err != nil {
sylog.Fatalf("failed to copy master unix socket descriptor: %s", err)
return
}
if stage == 2 {
comm.Close()
}
} else {
conn = nil
}
engine, err := engines.NewEngine(jsonBytes)
if err != nil {
sylog.Fatalf("failed to initialize runtime engine: %s\n", err)
}
if stage == 1 {
sylog.Debugf("Entering scontainer stage 1\n")
if err := engine.PrepareConfig(conn, sconfig); err != nil {
sylog.Fatalf("%s\n", err)
}
if err := sconfig.WritePayload(conn, engine.Common); err != nil {
sylog.Fatalf("%s", err)
}
conn.Close()
os.Exit(0)
} else {
if err := engine.StartProcess(conn); err != nil {
sylog.Fatalf("%s\n", err)
}
}
}