utils.go
a27b4b8c
 package docker
 
193a7e1d
 import (
551092f9
 	"fmt"
9ae4bcaa
 	"github.com/dotcloud/docker/archive"
a009d4ae
 	"github.com/dotcloud/docker/pkg/namesgenerator"
1cbdaeba
 	"github.com/dotcloud/docker/utils"
e40f5c7c
 	"io/ioutil"
f1127b93
 	"strconv"
193a7e1d
 	"strings"
 )
 
9ae4bcaa
 type Change struct {
 	archive.Change
 }
 
a46fc3a5
 // Compare two Config struct. Do not compare the "Image" nor "Hostname" fields
 // If OpenStdin is set, then it differs
 func CompareConfig(a, b *Config) bool {
 	if a == nil || b == nil ||
 		a.OpenStdin || b.OpenStdin {
 		return false
 	}
 	if a.AttachStdout != b.AttachStdout ||
 		a.AttachStderr != b.AttachStderr ||
 		a.User != b.User ||
 		a.Memory != b.Memory ||
 		a.MemorySwap != b.MemorySwap ||
efd9becb
 		a.CpuShares != b.CpuShares ||
a46fc3a5
 		a.OpenStdin != b.OpenStdin ||
7c002012
 		a.Tty != b.Tty ||
 		a.VolumesFrom != b.VolumesFrom {
a46fc3a5
 		return false
 	}
 	if len(a.Cmd) != len(b.Cmd) ||
 		len(a.Dns) != len(b.Dns) ||
 		len(a.Env) != len(b.Env) ||
71d2ff49
 		len(a.PortSpecs) != len(b.PortSpecs) ||
1cbdaeba
 		len(a.ExposedPorts) != len(b.ExposedPorts) ||
7c002012
 		len(a.Entrypoint) != len(b.Entrypoint) ||
 		len(a.Volumes) != len(b.Volumes) {
a46fc3a5
 		return false
 	}
 
 	for i := 0; i < len(a.Cmd); i++ {
 		if a.Cmd[i] != b.Cmd[i] {
 			return false
 		}
 	}
 	for i := 0; i < len(a.Dns); i++ {
 		if a.Dns[i] != b.Dns[i] {
 			return false
 		}
 	}
 	for i := 0; i < len(a.Env); i++ {
 		if a.Env[i] != b.Env[i] {
 			return false
 		}
 	}
 	for i := 0; i < len(a.PortSpecs); i++ {
 		if a.PortSpecs[i] != b.PortSpecs[i] {
 			return false
 		}
 	}
1cbdaeba
 	for k := range a.ExposedPorts {
 		if _, exists := b.ExposedPorts[k]; !exists {
 			return false
 		}
 	}
b16ff9f8
 	for i := 0; i < len(a.Entrypoint); i++ {
 		if a.Entrypoint[i] != b.Entrypoint[i] {
 			return false
 		}
 	}
7c002012
 	for key := range a.Volumes {
 		if _, exists := b.Volumes[key]; !exists {
 			return false
 		}
 	}
a46fc3a5
 	return true
 }
0f312113
 
5bd0437e
 func MergeConfig(userConf, imageConf *Config) error {
83bc5b74
 	if userConf.User == "" {
0f312113
 		userConf.User = imageConf.User
 	}
 	if userConf.Memory == 0 {
 		userConf.Memory = imageConf.Memory
 	}
 	if userConf.MemorySwap == 0 {
 		userConf.MemorySwap = imageConf.MemorySwap
 	}
 	if userConf.CpuShares == 0 {
 		userConf.CpuShares = imageConf.CpuShares
 	}
1cbdaeba
 	if userConf.ExposedPorts == nil || len(userConf.ExposedPorts) == 0 {
 		userConf.ExposedPorts = imageConf.ExposedPorts
49c4231f
 	} else if imageConf.ExposedPorts != nil {
 		if userConf.ExposedPorts == nil {
 			userConf.ExposedPorts = make(map[Port]struct{})
 		}
 		for port := range imageConf.ExposedPorts {
 			if _, exists := userConf.ExposedPorts[port]; !exists {
 				userConf.ExposedPorts[port] = struct{}{}
 			}
 		}
1cbdaeba
 	}
 
 	if userConf.PortSpecs != nil && len(userConf.PortSpecs) > 0 {
 		if userConf.ExposedPorts == nil {
 			userConf.ExposedPorts = make(map[Port]struct{})
 		}
 		ports, _, err := parsePortSpecs(userConf.PortSpecs)
 		if err != nil {
 			return err
 		}
 		for port := range ports {
 			if _, exists := userConf.ExposedPorts[port]; !exists {
 				userConf.ExposedPorts[port] = struct{}{}
193a7e1d
 			}
1cbdaeba
 		}
 		userConf.PortSpecs = nil
 	}
 	if imageConf.PortSpecs != nil && len(imageConf.PortSpecs) > 0 {
 		utils.Debugf("Migrating image port specs to containter: %s", strings.Join(imageConf.PortSpecs, ", "))
 		if userConf.ExposedPorts == nil {
 			userConf.ExposedPorts = make(map[Port]struct{})
 		}
 
 		ports, _, err := parsePortSpecs(imageConf.PortSpecs)
 		if err != nil {
 			return err
 		}
 		for port := range ports {
 			if _, exists := userConf.ExposedPorts[port]; !exists {
 				userConf.ExposedPorts[port] = struct{}{}
193a7e1d
 			}
 		}
0f312113
 	}
 	if !userConf.Tty {
 		userConf.Tty = imageConf.Tty
 	}
 	if !userConf.OpenStdin {
 		userConf.OpenStdin = imageConf.OpenStdin
 	}
 	if !userConf.StdinOnce {
 		userConf.StdinOnce = imageConf.StdinOnce
 	}
 	if userConf.Env == nil || len(userConf.Env) == 0 {
 		userConf.Env = imageConf.Env
193a7e1d
 	} else {
 		for _, imageEnv := range imageConf.Env {
 			found := false
 			imageEnvKey := strings.Split(imageEnv, "=")[0]
 			for _, userEnv := range userConf.Env {
 				userEnvKey := strings.Split(userEnv, "=")[0]
 				if imageEnvKey == userEnvKey {
 					found = true
 				}
 			}
 			if !found {
 				userConf.Env = append(userConf.Env, imageEnv)
 			}
 		}
0f312113
 	}
 	if userConf.Cmd == nil || len(userConf.Cmd) == 0 {
 		userConf.Cmd = imageConf.Cmd
 	}
 	if userConf.Dns == nil || len(userConf.Dns) == 0 {
 		userConf.Dns = imageConf.Dns
193a7e1d
 	} else {
 		//duplicates aren't an issue here
 		userConf.Dns = append(userConf.Dns, imageConf.Dns...)
0f312113
 	}
b16ff9f8
 	if userConf.Entrypoint == nil || len(userConf.Entrypoint) == 0 {
 		userConf.Entrypoint = imageConf.Entrypoint
 	}
31998833
 	if userConf.WorkingDir == "" {
 		userConf.WorkingDir = imageConf.WorkingDir
 	}
1a226f0e
 	if userConf.VolumesFrom == "" {
 		userConf.VolumesFrom = imageConf.VolumesFrom
 	}
eb9fef2c
 	if userConf.Volumes == nil || len(userConf.Volumes) == 0 {
 		userConf.Volumes = imageConf.Volumes
193a7e1d
 	} else {
 		for k, v := range imageConf.Volumes {
 			userConf.Volumes[k] = v
 		}
eb9fef2c
 	}
5bd0437e
 	return nil
0f312113
 }
551092f9
 
1ba11384
 func parseLxcConfOpts(opts ListOpts) ([]KeyValuePair, error) {
 	out := make([]KeyValuePair, opts.Len())
 	for i, o := range opts.GetAll() {
551092f9
 		k, v, err := parseLxcOpt(o)
 		if err != nil {
 			return nil, err
 		}
 		out[i] = KeyValuePair{Key: k, Value: v}
 	}
 	return out, nil
 }
 
 func parseLxcOpt(opt string) (string, string, error) {
 	parts := strings.SplitN(opt, "=", 2)
 	if len(parts) != 2 {
 		return "", "", fmt.Errorf("Unable to parse lxc conf option: %s", opt)
 	}
 	return strings.TrimSpace(parts[0]), strings.TrimSpace(parts[1]), nil
 }
1cbdaeba
 
1f9223a7
 // FIXME: network related stuff (including parsing) should be grouped in network file
 const (
 	PortSpecTemplate       = "ip:hostPort:containerPort"
 	PortSpecTemplateFormat = "ip:hostPort:containerPort | ip::containerPort | hostPort:containerPort"
 )
 
1cbdaeba
 // We will receive port specs in the format of ip:public:private/proto and these need to be
 // parsed in the internal types
 func parsePortSpecs(ports []string) (map[Port]struct{}, map[Port][]PortBinding, error) {
47655945
 	var (
 		exposedPorts = make(map[Port]struct{}, len(ports))
 		bindings     = make(map[Port][]PortBinding)
 	)
1cbdaeba
 
 	for _, rawPort := range ports {
 		proto := "tcp"
47655945
 
1cbdaeba
 		if i := strings.LastIndex(rawPort, "/"); i != -1 {
 			proto = rawPort[i+1:]
 			rawPort = rawPort[:i]
 		}
 		if !strings.Contains(rawPort, ":") {
 			rawPort = fmt.Sprintf("::%s", rawPort)
 		} else if len(strings.Split(rawPort, ":")) == 2 {
 			rawPort = fmt.Sprintf(":%s", rawPort)
 		}
 
1f9223a7
 		parts, err := utils.PartParser(PortSpecTemplate, rawPort)
1cbdaeba
 		if err != nil {
 			return nil, nil, err
 		}
47655945
 
 		var (
 			containerPort = parts["containerPort"]
 			rawIp         = parts["ip"]
 			hostPort      = parts["hostPort"]
 		)
1cbdaeba
 
 		if containerPort == "" {
 			return nil, nil, fmt.Errorf("No port specified: %s<empty>", rawPort)
 		}
7f07ae9e
 		if _, err := strconv.ParseUint(containerPort, 10, 16); err != nil {
 			return nil, nil, fmt.Errorf("Invalid containerPort: %s", containerPort)
 		}
 		if _, err := strconv.ParseUint(hostPort, 10, 16); hostPort != "" && err != nil {
 			return nil, nil, fmt.Errorf("Invalid hostPort: %s", hostPort)
 		}
1cbdaeba
 
 		port := NewPort(proto, containerPort)
 		if _, exists := exposedPorts[port]; !exists {
 			exposedPorts[port] = struct{}{}
 		}
 
 		binding := PortBinding{
 			HostIp:   rawIp,
 			HostPort: hostPort,
 		}
 		bslice, exists := bindings[port]
 		if !exists {
 			bslice = []PortBinding{}
 		}
 		bindings[port] = append(bslice, binding)
 	}
 	return exposedPorts, bindings, nil
 }
 
 // Splits a port in the format of port/proto
 func splitProtoPort(rawPort string) (string, string) {
 	parts := strings.Split(rawPort, "/")
 	l := len(parts)
 	if l == 0 {
 		return "", ""
 	}
 	if l == 1 {
 		return "tcp", rawPort
 	}
 	return parts[0], parts[1]
 }
 
 func parsePort(rawPort string) (int, error) {
 	port, err := strconv.ParseUint(rawPort, 10, 16)
 	if err != nil {
 		return 0, err
 	}
 	return int(port), nil
 }
 
746ba127
 func migratePortMappings(config *Config, hostConfig *HostConfig) error {
1cbdaeba
 	if config.PortSpecs != nil {
746ba127
 		ports, bindings, err := parsePortSpecs(config.PortSpecs)
1cbdaeba
 		if err != nil {
 			return err
 		}
 		config.PortSpecs = nil
746ba127
 		if len(bindings) > 0 {
 			if hostConfig == nil {
 				hostConfig = &HostConfig{}
 			}
 			hostConfig.PortBindings = bindings
 		}
1cbdaeba
 
 		if config.ExposedPorts == nil {
 			config.ExposedPorts = make(map[Port]struct{}, len(ports))
 		}
 		for k, v := range ports {
 			config.ExposedPorts[k] = v
 		}
 	}
 	return nil
 }
 
 // Links come in the format of
 // name:alias
 func parseLink(rawLink string) (map[string]string, error) {
 	return utils.PartParser("name:alias", rawLink)
 }
0d292440
 
157d99a7
 func RootIsShared() bool {
 	if data, err := ioutil.ReadFile("/proc/self/mountinfo"); err == nil {
 		for _, line := range strings.Split(string(data), "\n") {
 			cols := strings.Split(line, " ")
 			if len(cols) >= 6 && cols[4] == "/" {
 				return strings.HasPrefix(cols[6], "shared")
 			}
 		}
 	}
 
 	// No idea, probably safe to assume so
 	return true
 }
 
0d292440
 type checker struct {
 	runtime *Runtime
 }
 
 func (c *checker) Exists(name string) bool {
 	return c.runtime.containerGraph.Exists("/" + name)
 }
 
 // Generate a random and unique name
8227ea35
 func generateRandomName(runtime *Runtime) (string, error) {
 	return namesgenerator.GenerateRandomName(&checker{runtime})
0d292440
 }