镜像自地址
https://github.com/tuna/tunasync.git
已同步 2025-12-06 14:36:47 +00:00
@@ -3,7 +3,7 @@ package worker
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"io"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
@@ -12,32 +12,32 @@ import (
|
||||
|
||||
"golang.org/x/sys/unix"
|
||||
|
||||
"github.com/moby/moby/pkg/reexec"
|
||||
cgv1 "github.com/containerd/cgroups"
|
||||
cgv2 "github.com/containerd/cgroups/v2"
|
||||
"github.com/moby/moby/pkg/reexec"
|
||||
contspecs "github.com/opencontainers/runtime-spec/specs-go"
|
||||
)
|
||||
|
||||
type cgroupHook struct {
|
||||
emptyHook
|
||||
cgCfg cgroupConfig
|
||||
memLimit MemBytes
|
||||
cgMgrV1 cgv1.Cgroup
|
||||
cgMgrV2 *cgv2.Manager
|
||||
cgCfg cgroupConfig
|
||||
memLimit MemBytes
|
||||
cgMgrV1 cgv1.Cgroup
|
||||
cgMgrV2 *cgv2.Manager
|
||||
}
|
||||
|
||||
type execCmd string
|
||||
|
||||
const (
|
||||
cmdCont execCmd = "cont"
|
||||
cmdAbrt execCmd = "abrt"
|
||||
cmdCont execCmd = "cont"
|
||||
cmdAbrt execCmd = "abrt"
|
||||
)
|
||||
|
||||
func init () {
|
||||
func init() {
|
||||
reexec.Register("tunasync-exec", waitExec)
|
||||
}
|
||||
|
||||
func waitExec () {
|
||||
func waitExec() {
|
||||
binary, err := exec.LookPath(os.Args[1])
|
||||
if err != nil {
|
||||
panic(err)
|
||||
@@ -46,7 +46,7 @@ func waitExec () {
|
||||
pipe := os.NewFile(3, "pipe")
|
||||
if pipe != nil {
|
||||
if _, err := pipe.Stat(); err == nil {
|
||||
cmdBytes, err := ioutil.ReadAll(pipe)
|
||||
cmdBytes, err := io.ReadAll(pipe)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
@@ -54,11 +54,11 @@ func waitExec () {
|
||||
}
|
||||
cmd := execCmd(string(cmdBytes))
|
||||
switch cmd {
|
||||
case cmdAbrt:
|
||||
fallthrough
|
||||
default:
|
||||
panic("Exited on request")
|
||||
case cmdCont:
|
||||
case cmdAbrt:
|
||||
fallthrough
|
||||
default:
|
||||
panic("Exited on request")
|
||||
case cmdCont:
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -71,7 +71,7 @@ func waitExec () {
|
||||
panic("Exec failed.")
|
||||
}
|
||||
|
||||
func initCgroup(cfg *cgroupConfig) (error) {
|
||||
func initCgroup(cfg *cgroupConfig) error {
|
||||
|
||||
logger.Debugf("Initializing cgroup")
|
||||
baseGroup := cfg.Group
|
||||
@@ -103,7 +103,7 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
}
|
||||
if baseGroup == "" {
|
||||
logger.Debugf("Creating a sub group and move all processes into it")
|
||||
wkrMgr, err := cfg.cgMgrV2.NewChild("__worker", nil);
|
||||
wkrMgr, err := cfg.cgMgrV2.NewChild("__worker", nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -117,8 +117,8 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
if len(procs) == 0 {
|
||||
break
|
||||
}
|
||||
for _, p := range(procs) {
|
||||
if err := wkrMgr.AddProc(p); err != nil{
|
||||
for _, p := range procs {
|
||||
if err := wkrMgr.AddProc(p); err != nil {
|
||||
if errors.Is(err, syscall.ESRCH) {
|
||||
logger.Debugf("Write pid %d to sub group failed: process vanished, ignoring")
|
||||
} else {
|
||||
@@ -129,7 +129,7 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
}
|
||||
} else {
|
||||
logger.Debugf("Trying to create a sub group in that group")
|
||||
testMgr, err := cfg.cgMgrV2.NewChild("__test", nil);
|
||||
testMgr, err := cfg.cgMgrV2.NewChild("__test", nil)
|
||||
if err != nil {
|
||||
logger.Errorf("Cannot create a sub group in the cgroup")
|
||||
return err
|
||||
@@ -152,9 +152,9 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
if baseGroup != "" {
|
||||
pather = cgv1.StaticPath(baseGroup)
|
||||
} else {
|
||||
pather = (func(p cgv1.Path) (cgv1.Path){
|
||||
return func(subsys cgv1.Name) (string, error){
|
||||
path, err := p(subsys);
|
||||
pather = (func(p cgv1.Path) cgv1.Path {
|
||||
return func(subsys cgv1.Name) (string, error) {
|
||||
path, err := p(subsys)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
@@ -167,14 +167,14 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
}
|
||||
logger.Infof("Loading cgroup")
|
||||
var err error
|
||||
if cfg.cgMgrV1, err = cgv1.Load(cgv1.V1, pather, func(cfg *cgv1.InitConfig) error{
|
||||
if cfg.cgMgrV1, err = cgv1.Load(cgv1.V1, pather, func(cfg *cgv1.InitConfig) error {
|
||||
cfg.InitCheck = cgv1.AllowAny
|
||||
return nil
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
logger.Debugf("Available subsystems:")
|
||||
for _, subsys := range(cfg.cgMgrV1.Subsystems()) {
|
||||
for _, subsys := range cfg.cgMgrV1.Subsystems() {
|
||||
p, err := pather(subsys.Name())
|
||||
if err != nil {
|
||||
return err
|
||||
@@ -183,11 +183,11 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
}
|
||||
if baseGroup == "" {
|
||||
logger.Debugf("Creating a sub group and move all processes into it")
|
||||
wkrMgr, err := cfg.cgMgrV1.New("__worker", &contspecs.LinuxResources{});
|
||||
wkrMgr, err := cfg.cgMgrV1.New("__worker", &contspecs.LinuxResources{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, subsys := range(cfg.cgMgrV1.Subsystems()) {
|
||||
for _, subsys := range cfg.cgMgrV1.Subsystems() {
|
||||
logger.Debugf("Reading pids for subsystem %s", subsys.Name())
|
||||
for {
|
||||
procs, err := cfg.cgMgrV1.Processes(subsys.Name(), false)
|
||||
@@ -202,7 +202,7 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
if len(procs) == 0 {
|
||||
break
|
||||
}
|
||||
for _, proc := range(procs) {
|
||||
for _, proc := range procs {
|
||||
if err := wkrMgr.Add(proc); err != nil {
|
||||
if errors.Is(err, syscall.ESRCH) {
|
||||
logger.Debugf("Write pid %d to sub group failed: process vanished, ignoring")
|
||||
@@ -215,7 +215,7 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
}
|
||||
} else {
|
||||
logger.Debugf("Trying to create a sub group in that group")
|
||||
testMgr, err := cfg.cgMgrV1.New("__test", &contspecs.LinuxResources{});
|
||||
testMgr, err := cfg.cgMgrV1.New("__test", &contspecs.LinuxResources{})
|
||||
if err != nil {
|
||||
logger.Errorf("Cannot create a sub group in the cgroup")
|
||||
return err
|
||||
@@ -223,7 +223,7 @@ func initCgroup(cfg *cgroupConfig) (error) {
|
||||
if err := testMgr.Delete(); err != nil {
|
||||
return err
|
||||
}
|
||||
for _, subsys := range(cfg.cgMgrV1.Subsystems()) {
|
||||
for _, subsys := range cfg.cgMgrV1.Subsystems() {
|
||||
logger.Debugf("Reading pids for subsystem %s", subsys.Name())
|
||||
procs, err := cfg.cgMgrV1.Processes(subsys.Name(), false)
|
||||
if err != nil {
|
||||
@@ -253,7 +253,7 @@ func newCgroupHook(p mirrorProvider, cfg cgroupConfig, memLimit MemBytes) *cgrou
|
||||
emptyHook: emptyHook{
|
||||
provider: p,
|
||||
},
|
||||
cgCfg: cfg,
|
||||
cgCfg: cfg,
|
||||
memLimit: memLimit,
|
||||
}
|
||||
}
|
||||
@@ -263,7 +263,7 @@ func (c *cgroupHook) preExec() error {
|
||||
logger.Debugf("Creating v2 cgroup for task %s", c.provider.Name())
|
||||
var resSet *cgv2.Resources
|
||||
if c.memLimit != 0 {
|
||||
resSet = &cgv2.Resources {
|
||||
resSet = &cgv2.Resources{
|
||||
Memory: &cgv2.Memory{
|
||||
Max: func(i int64) *int64 { return &i }(c.memLimit.Value()),
|
||||
},
|
||||
@@ -279,7 +279,7 @@ func (c *cgroupHook) preExec() error {
|
||||
logger.Debugf("Creating v1 cgroup for task %s", c.provider.Name())
|
||||
var resSet contspecs.LinuxResources
|
||||
if c.memLimit != 0 {
|
||||
resSet = contspecs.LinuxResources {
|
||||
resSet = contspecs.LinuxResources{
|
||||
Memory: &contspecs.LinuxMemory{
|
||||
Limit: func(i int64) *int64 { return &i }(c.memLimit.Value()),
|
||||
},
|
||||
@@ -334,7 +334,7 @@ func (c *cgroupHook) killAll() error {
|
||||
taskList := []int{}
|
||||
if c.cgCfg.isUnified {
|
||||
procs, err := c.cgMgrV2.Procs(false)
|
||||
if (err != nil) {
|
||||
if err != nil {
|
||||
return []int{}, err
|
||||
}
|
||||
for _, proc := range procs {
|
||||
@@ -342,16 +342,16 @@ func (c *cgroupHook) killAll() error {
|
||||
}
|
||||
} else {
|
||||
taskSet := make(map[int]struct{})
|
||||
for _, subsys := range(c.cgMgrV1.Subsystems()) {
|
||||
for _, subsys := range c.cgMgrV1.Subsystems() {
|
||||
procs, err := c.cgMgrV1.Processes(subsys.Name(), false)
|
||||
if err != nil {
|
||||
return []int{}, err
|
||||
}
|
||||
for _, proc := range(procs) {
|
||||
for _, proc := range procs {
|
||||
taskSet[proc.Pid] = struct{}{}
|
||||
}
|
||||
}
|
||||
for proc := range(taskSet) {
|
||||
for proc := range taskSet {
|
||||
taskList = append(taskList, proc)
|
||||
}
|
||||
}
|
||||
|
||||
在新工单中引用
屏蔽一个用户