mirror of
https://github.com/rancher/os.git
synced 2025-06-21 12:37:04 +00:00
54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/docker/docker/pkg/reexec"
|
|
"github.com/rancherio/os/cmd/cloudinit"
|
|
"github.com/rancherio/os/cmd/control"
|
|
"github.com/rancherio/os/cmd/power"
|
|
"github.com/rancherio/os/cmd/respawn"
|
|
"github.com/rancherio/os/cmd/sysinit"
|
|
"github.com/rancherio/os/cmd/systemdocker"
|
|
"github.com/rancherio/os/cmd/network"
|
|
osInit "github.com/rancherio/os/init"
|
|
)
|
|
|
|
func registerCmd(cmd string, mainFunc func()) {
|
|
log.Debugf("Registering main %s", cmd)
|
|
reexec.Register(cmd, mainFunc)
|
|
|
|
parts := strings.Split(cmd, "/")
|
|
if len(parts) == 0 {
|
|
return
|
|
}
|
|
|
|
last := parts[len(parts)-1]
|
|
|
|
log.Debugf("Registering main %s", last)
|
|
reexec.Register(last, mainFunc)
|
|
|
|
log.Debugf("Registering main %s", "./"+last)
|
|
reexec.Register("./"+last, mainFunc)
|
|
}
|
|
|
|
func main() {
|
|
registerCmd("/init", osInit.MainInit)
|
|
registerCmd(osInit.SYSINIT, sysinit.Main)
|
|
registerCmd("/usr/bin/system-docker", systemdocker.Main)
|
|
registerCmd("/sbin/poweroff", power.PowerOff)
|
|
registerCmd("/sbin/reboot", power.Reboot)
|
|
registerCmd("/sbin/halt", power.Halt)
|
|
registerCmd("/sbin/shutdown", power.Main)
|
|
registerCmd("/usr/bin/respawn", respawn.Main)
|
|
registerCmd("/usr/sbin/rancherctl", control.Main)
|
|
registerCmd("/usr/bin/cloud-init", cloudinit.Main)
|
|
registerCmd("/usr/sbin/netconf", network.Main)
|
|
|
|
if !reexec.Init() {
|
|
log.Fatalf("Failed to find an entry point for %s", os.Args[0])
|
|
}
|
|
}
|