_, 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, []string{"1.1.0.0/16", "2.2.0.0/16"}, []string{"1.1.1.1/32", "2.2.2.2/32"}, fakeRunner, fakeQuotaManager, ) }) Describe("MaxContainer", func() { Context("when constrained by network pool size", func() { BeforeEach(func() { fakeNetworkPool.InitialPoolSize = 5 fakeUIDPool.InitialPoolSize = 3000 }) It("returns the network pool size", func() {
func main() { flag.Parse() maxProcs := runtime.NumCPU() prevMaxProcs := runtime.GOMAXPROCS(maxProcs) log.Println("set GOMAXPROCS to", maxProcs, "was", prevMaxProcs) if *binPath == "" { log.Fatalln("must specify -bin 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(uint32(*uidPoolStart), uint32(*uidPoolSize)) _, ipNet, err := net.ParseCIDR(*networkPool) if err != nil { log.Fatalln("error parsing CIDR:", err) } networkPool := network_pool.New(ipNet) // TODO: use /proc/sys/net/ipv4/ip_local_port_range by default (end + 1) portPool := port_pool.New(uint32(*portPoolStart), uint32(*portPoolSize)) runner := linux_command_runner.New(*debug) quotaManager, err := quota_manager.New(*depotPath, *binPath, runner) if err != nil { log.Fatalln("error creating quota manager:", err) } if *disableQuotas { quotaManager.Disable() } pool := container_pool.New( *binPath, *depotPath, *rootFSPath, uidPool, networkPool, portPool, strings.Split(*denyNetworks, ","), strings.Split(*allowNetworks, ","), runner, quotaManager, ) systemInfo := system_info.NewProvider(*depotPath) backend := linux_backend.New(pool, systemInfo, *snapshotsPath) 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 := *containerGraceTime 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 {} }