2013-01-19 19:07:19 -05:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2013-03-13 03:29:40 -04:00
|
|
|
"flag"
|
2013-04-10 06:24:15 -04:00
|
|
|
"fmt"
|
2013-03-13 03:29:40 -04:00
|
|
|
"github.com/dotcloud/docker"
|
2013-05-14 18:37:35 -04:00
|
|
|
"github.com/dotcloud/docker/utils"
|
2013-04-22 18:57:31 -04:00
|
|
|
"io/ioutil"
|
2013-01-19 19:07:19 -05:00
|
|
|
"log"
|
|
|
|
"os"
|
2013-04-10 06:24:15 -04:00
|
|
|
"os/signal"
|
2013-04-22 18:57:31 -04:00
|
|
|
"strconv"
|
2013-05-23 12:09:28 -04:00
|
|
|
"strings"
|
2013-04-10 06:24:15 -04:00
|
|
|
"syscall"
|
2013-01-24 02:14:46 -05:00
|
|
|
)
|
|
|
|
|
2013-04-11 15:58:23 -04:00
|
|
|
var (
|
2013-06-04 14:00:22 -04:00
|
|
|
GITCOMMIT string
|
2013-04-11 15:58:23 -04:00
|
|
|
)
|
2013-04-01 14:12:56 -04:00
|
|
|
|
2013-01-19 19:07:19 -05:00
|
|
|
func main() {
|
2013-05-14 18:37:35 -04:00
|
|
|
if utils.SelfPath() == "/sbin/init" {
|
2013-03-13 03:29:40 -04:00
|
|
|
// Running in init mode
|
|
|
|
docker.SysInit()
|
|
|
|
return
|
2013-01-25 14:26:18 -05:00
|
|
|
}
|
2013-05-24 14:31:19 -04:00
|
|
|
host := "127.0.0.1"
|
|
|
|
port := 4243
|
2013-03-22 14:44:12 -04:00
|
|
|
// FIXME: Switch d and D ? (to be more sshd like)
|
2013-03-28 20:12:23 -04:00
|
|
|
flDaemon := flag.Bool("d", false, "Daemon mode")
|
|
|
|
flDebug := flag.Bool("D", false, "Debug mode")
|
2013-04-24 22:01:23 -04:00
|
|
|
flAutoRestart := flag.Bool("r", false, "Restart previously running containers")
|
2013-04-05 17:16:04 -04:00
|
|
|
bridgeName := flag.String("b", "", "Attach containers to a pre-existing network bridge")
|
2013-04-10 06:24:15 -04:00
|
|
|
pidfile := flag.String("p", "/var/run/docker.pid", "File containing process PID")
|
2013-05-24 12:49:18 -04:00
|
|
|
flHost := flag.String("H", fmt.Sprintf("%s:%d", host, port), "Host:port to bind/connect to")
|
2013-06-03 21:39:00 -04:00
|
|
|
flEnableCors := flag.Bool("api-enable-cors", false, "Enable CORS requests in the remote api.")
|
2013-03-13 03:29:40 -04:00
|
|
|
flag.Parse()
|
2013-04-03 17:53:09 -04:00
|
|
|
if *bridgeName != "" {
|
|
|
|
docker.NetworkBridgeIface = *bridgeName
|
|
|
|
} else {
|
2013-04-04 08:33:28 -04:00
|
|
|
docker.NetworkBridgeIface = docker.DefaultNetworkBridge
|
2013-04-03 17:53:09 -04:00
|
|
|
}
|
2013-05-23 12:09:28 -04:00
|
|
|
|
2013-05-24 14:31:36 -04:00
|
|
|
if strings.Contains(*flHost, ":") {
|
2013-05-23 12:09:28 -04:00
|
|
|
hostParts := strings.Split(*flHost, ":")
|
2013-05-24 14:31:36 -04:00
|
|
|
if len(hostParts) != 2 {
|
|
|
|
log.Fatal("Invalid bind address format.")
|
|
|
|
os.Exit(-1)
|
|
|
|
}
|
2013-05-23 12:09:28 -04:00
|
|
|
if hostParts[0] != "" {
|
|
|
|
host = hostParts[0]
|
|
|
|
}
|
|
|
|
if p, err := strconv.Atoi(hostParts[1]); err == nil {
|
|
|
|
port = p
|
|
|
|
}
|
2013-05-24 14:31:36 -04:00
|
|
|
} else {
|
2013-05-23 12:09:28 -04:00
|
|
|
host = *flHost
|
|
|
|
}
|
|
|
|
|
2013-04-02 13:58:16 -04:00
|
|
|
if *flDebug {
|
|
|
|
os.Setenv("DEBUG", "1")
|
|
|
|
}
|
2013-06-04 14:00:22 -04:00
|
|
|
docker.GITCOMMIT = GITCOMMIT
|
2013-03-28 20:12:23 -04:00
|
|
|
if *flDaemon {
|
2013-03-13 03:29:40 -04:00
|
|
|
if flag.NArg() != 0 {
|
|
|
|
flag.Usage()
|
|
|
|
return
|
|
|
|
}
|
2013-06-03 21:39:00 -04:00
|
|
|
if err := daemon(*pidfile, host, port, *flAutoRestart, *flEnableCors); err != nil {
|
2013-03-13 03:29:40 -04:00
|
|
|
log.Fatal(err)
|
2013-05-06 05:31:22 -04:00
|
|
|
os.Exit(-1)
|
2013-03-13 03:29:40 -04:00
|
|
|
}
|
|
|
|
} else {
|
2013-05-23 12:09:28 -04:00
|
|
|
if err := docker.ParseCommands(host, port, flag.Args()...); err != nil {
|
2013-03-13 03:29:40 -04:00
|
|
|
log.Fatal(err)
|
2013-05-06 05:31:22 -04:00
|
|
|
os.Exit(-1)
|
2013-03-13 03:29:40 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-04-10 06:24:15 -04:00
|
|
|
func createPidFile(pidfile string) error {
|
2013-04-22 18:57:31 -04:00
|
|
|
if pidString, err := ioutil.ReadFile(pidfile); err == nil {
|
|
|
|
pid, err := strconv.Atoi(string(pidString))
|
|
|
|
if err == nil {
|
|
|
|
if _, err := os.Stat(fmt.Sprintf("/proc/%d/", pid)); err == nil {
|
|
|
|
return fmt.Errorf("pid file found, ensure docker is not running or delete %s", pidfile)
|
|
|
|
}
|
|
|
|
}
|
2013-04-10 06:24:15 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
file, err := os.Create(pidfile)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
defer file.Close()
|
|
|
|
|
|
|
|
_, err = fmt.Fprintf(file, "%d", os.Getpid())
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func removePidFile(pidfile string) {
|
|
|
|
if err := os.Remove(pidfile); err != nil {
|
|
|
|
log.Printf("Error removing %s: %s", pidfile, err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-03 21:39:00 -04:00
|
|
|
func daemon(pidfile, addr string, port int, autoRestart, enableCors bool) error {
|
2013-05-23 12:15:36 -04:00
|
|
|
if addr != "127.0.0.1" {
|
|
|
|
log.Println("/!\\ DON'T BIND ON ANOTHER IP ADDRESS THAN 127.0.0.1 IF YOU DON'T KNOW WHAT YOU'RE DOING /!\\")
|
|
|
|
}
|
2013-04-12 10:37:24 -04:00
|
|
|
if err := createPidFile(pidfile); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
defer removePidFile(pidfile)
|
2013-04-10 06:24:15 -04:00
|
|
|
|
2013-04-12 10:37:24 -04:00
|
|
|
c := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(c, os.Interrupt, os.Kill, os.Signal(syscall.SIGTERM))
|
|
|
|
go func() {
|
|
|
|
sig := <-c
|
|
|
|
log.Printf("Received signal '%v', exiting\n", sig)
|
|
|
|
removePidFile(pidfile)
|
|
|
|
os.Exit(0)
|
|
|
|
}()
|
2013-04-10 06:24:15 -04:00
|
|
|
|
2013-06-03 21:39:00 -04:00
|
|
|
server, err := docker.NewServer(autoRestart, enableCors)
|
2013-04-17 21:13:43 -04:00
|
|
|
if err != nil {
|
2013-04-10 22:48:21 -04:00
|
|
|
return err
|
|
|
|
}
|
2013-04-17 21:13:43 -04:00
|
|
|
|
2013-05-23 12:09:28 -04:00
|
|
|
return docker.ListenAndServe(fmt.Sprintf("%s:%d", addr, port), server, true)
|
2013-01-19 19:07:19 -05:00
|
|
|
}
|