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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
package namespace
import (
"fmt"
"os"
"path/filepath"
"strconv"
"syscall"
"git.theodohertyfamily.com/tools/wg-wrap/internal/paths"
)
// GetProfileNamespacePath returns the path to the pinned namespace file for a profile.
func GetProfileNamespacePath(pm *paths.PathManager, profile string) string {
return pm.ProfileNamespacePath(profile)
}
// GetPidsDirPath returns the path to the directory where process PIDs are tracked for a profile.
func GetPidsDirPath(pm *paths.PathManager, profile string) string {
return pm.ProfilePidsDir(profile)
}
// GetControllerPidPath returns the path to the file storing the PID of the tunnel controller.
func GetControllerPidPath(pm *paths.PathManager, profile string) string {
return filepath.Join(pm.RuntimeBaseDir(), "profiles", profile, "controller.pid")
}
// RegisterProcess marks the current process as using the specified profile.
func RegisterProcess(pm *paths.PathManager, profile string) error {
pidsDir := GetPidsDirPath(pm, profile)
if err := os.MkdirAll(pidsDir, 0755); err != nil {
return fmt.Errorf("failed to create pids directory: %v", err)
}
pid := os.Getpid()
pidFile := filepath.Join(pidsDir, strconv.Itoa(pid))
if err := os.WriteFile(pidFile, []byte(""), 0644); err != nil {
return fmt.Errorf("failed to register process pid %d: %v", pid, err)
}
return nil
}
// UnregisterProcess removes the current process from the profile's tracking.
func UnregisterProcess(pm *paths.PathManager, profile string) error {
pid := os.Getpid()
pidFile := filepath.Join(GetPidsDirPath(pm, profile), strconv.Itoa(pid))
if err := os.Remove(pidFile); err != nil && !os.IsNotExist(err) {
return fmt.Errorf("failed to unregister process pid %d: %v", pid, err)
}
return nil
}
// PruneStalePids removes PID files that no longer correspond to active processes.
func PruneStalePids(pm *paths.PathManager, profile string) error {
pidsDir := GetPidsDirPath(pm, profile)
files, err := os.ReadDir(pidsDir)
if err != nil {
if os.IsNotExist(err) {
return nil
}
return fmt.Errorf("failed to read pids directory: %v", err)
}
for _, file := range files {
if file.Name() == "controller.pid" {
continue
}
pid, err := strconv.Atoi(file.Name())
if err != nil {
continue // Ignore non-numeric files
}
process, err := os.FindProcess(pid)
if err != nil {
if err := os.Remove(filepath.Join(pidsDir, file.Name())); err != nil {
fmt.Printf("failed to remove stale pid file %s: %v\n", file.Name(), err)
}
continue
}
err = process.Signal(syscall.Signal(0))
if err != nil {
if err := os.Remove(filepath.Join(pidsDir, file.Name())); err != nil {
fmt.Printf("failed to remove stale pid file %s: %v\n", file.Name(), err)
}
}
}
return nil
}
// IsLastProcess checks if the current process is the only active user of the profile.
func IsLastProcess(pm *paths.PathManager, profile string) (bool, error) {
pidsDir := GetPidsDirPath(pm, profile)
files, err := os.ReadDir(pidsDir)
if err != nil {
if os.IsNotExist(err) {
return true, nil
}
return false, fmt.Errorf("failed to read pids directory: %v", err)
}
activeCount := 0
for _, file := range files {
pid, err := strconv.Atoi(file.Name())
if err != nil {
continue
}
process, err := os.FindProcess(pid)
if err != nil {
continue
}
if process.Signal(syscall.Signal(0)) == nil {
activeCount++
}
}
return activeCount <= 1, nil
}
// SetControllerPid records the current process as the owner of the namespace.
func SetControllerPid(pm *paths.PathManager, profile string) error {
path := GetControllerPidPath(pm, profile)
if err := os.WriteFile(path, []byte(strconv.Itoa(os.Getpid())), 0644); err != nil {
return fmt.Errorf("failed to write controller pid: %w", err)
}
return nil
}
// GetControllerPid retrieves the PID of the process responsible for cleaning up the namespace.
func GetControllerPid(pm *paths.PathManager, profile string) (int, error) {
path := GetControllerPidPath(pm, profile)
data, err := os.ReadFile(path)
if err != nil {
return 0, err
}
return strconv.Atoi(string(data))
}
|
