func main() { flag.Parse() maxProcs := runtime.NumCPU() prevMaxProcs := runtime.GOMAXPROCS(maxProcs) log.Println("set GOMAXPROCS to", maxProcs, "was", prevMaxProcs) var backend backend.Backend switch *backendName { case "linux": if *rootPath == "" { log.Fatalln("must specify -root with linux backend") } if *depotPath == "" { log.Fatalln("must specify -depot with linux backend") } if *rootFSPath == "" { log.Fatalln("must specify -rootfs with linux backend") } uidPool := uid_pool.New(10000, 256) _, ipNet, err := net.ParseCIDR("10.254.0.0/22") if err != nil { log.Fatalln("error parsing CIDR:", err) } networkPool := network_pool.New(ipNet) // TODO: base on ephemeral port range portPool := port_pool.New(61000, 6501) var runner command_runner.CommandRunner runner = command_runner.New(*debug) quotaManager, err := quota_manager.New(*depotPath, *rootPath, runner) if err != nil { log.Fatalln("error creating quota manager:", err) } if *disableQuotas { quotaManager.Disable() } pool := container_pool.New( path.Join(*rootPath, "linux"), *depotPath, *rootFSPath, uidPool, networkPool, portPool, runner, quotaManager, ) backend = linux_backend.New(pool, *snapshotsPath) case "fake": backend = fake_backend.New() } log.Println("setting up backend") err := backend.Setup() if err != nil { log.Fatalln("failed to set up backend:", err) } log.Println("starting server; listening with", *listenNetwork, "on", *listenAddr) graceTime := time.Duration(*containerGraceTime) * time.Second wardenServer := server.New(*listenNetwork, *listenAddr, graceTime, backend) err = wardenServer.Start() if err != nil { log.Fatalln("failed to start:", err) } signals := make(chan os.Signal, 1) go func() { <-signals log.Println("stopping...") wardenServer.Stop() os.Exit(0) }() signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP) select {} }
BeforeEach(func() { _, ipNet, err := net.ParseCIDR("1.2.0.0/20") Expect(err).ToNot(HaveOccurred()) fakeUIDPool = fake_uid_pool.New(10000) fakeNetworkPool = fake_network_pool.New(ipNet) fakeRunner = fake_command_runner.New() fakeQuotaManager = fake_quota_manager.New() fakePortPool = fake_port_pool.New(1000) pool = container_pool.New( "/root/path", "/depot/path", "/rootfs/path", fakeUIDPool, fakeNetworkPool, fakePortPool, fakeRunner, fakeQuotaManager, ) }) Describe("setup", func() { It("executes setup.sh with the correct environment", func() { fakeQuotaManager.MountPointResult = "/depot/mount/point" err := pool.Setup() Expect(err).ToNot(HaveOccurred()) Expect(fakeRunner).To(HaveExecutedSerially( fake_command_runner.CommandSpec{