1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00

Network: Do not assume that we are using a class C. Instead, compute the

IP addresses range and network size in order to allocate an IP address.
This commit is contained in:
Andrea Luzzardi 2013-02-21 18:33:23 -08:00
parent 6124c5eb31
commit e0e49b9a22

View file

@ -1,6 +1,8 @@
package docker
import (
"bytes"
"encoding/binary"
"fmt"
"math/rand"
"net"
@ -16,10 +18,61 @@ type NetworkInterface struct {
Gateway net.IP
}
func allocateIPAddress(network *net.IPNet) net.IP {
ip := network.IP.Mask(network.Mask)
ip[3] = byte(rand.Intn(254))
return ip
func networkRange(network *net.IPNet) (net.IP, net.IP) {
netIP := network.IP.To4()
firstIP := netIP.Mask(network.Mask)
lastIP := net.IPv4(0, 0, 0, 0).To4()
for i := 0; i < len(lastIP); i++ {
lastIP[i] = netIP[i] | ^network.Mask[i]
}
return firstIP, lastIP
}
func ipToInt(ip net.IP) (int32, error) {
buf := bytes.NewBuffer(ip.To4())
var n int32
if err := binary.Read(buf, binary.BigEndian, &n); err != nil {
return 0, err
}
return n, nil
}
func intToIp(n int32) (net.IP, error) {
var buf bytes.Buffer
if err := binary.Write(&buf, binary.BigEndian, &n); err != nil {
return net.IP{}, err
}
ip := net.IPv4(0, 0, 0, 0).To4()
for i := 0; i < net.IPv4len; i++ {
ip[i] = buf.Bytes()[i]
}
return ip, nil
}
func networkSize(mask net.IPMask) (int32, error) {
for i := 0; i < net.IPv4len; i++ {
mask[i] = ^mask[i]
}
buf := bytes.NewBuffer(mask)
var n int32
if err := binary.Read(buf, binary.BigEndian, &n); err != nil {
return 0, err
}
return n + 1, nil
}
func allocateIPAddress(network *net.IPNet) (net.IP, error) {
ip, _ := networkRange(network)
netSize, err := networkSize(network.Mask)
if err != nil {
return net.IP{}, err
}
numIp, err := ipToInt(ip)
if err != nil {
return net.IP{}, err
}
numIp += rand.Int31n(netSize)
return intToIp(numIp)
}
func getBridgeAddr(name string) (net.Addr, error) {
@ -54,8 +107,12 @@ func allocateNetwork() (*NetworkInterface, error) {
}
bridge := bridgeAddr.(*net.IPNet)
ipPrefixLen, _ := bridge.Mask.Size()
ip, err := allocateIPAddress(bridge)
if err != nil {
return nil, err
}
iface := &NetworkInterface{
IpAddress: allocateIPAddress(bridge).String(),
IpAddress: ip.String(),
IpPrefixLen: ipPrefixLen,
Gateway: bridge.IP,
}