mirror of
https://github.com/containers/podman
synced 2024-10-22 18:23:39 +00:00
fe3faa517e
due to a lack of "locking" on cni operations, we could get ourselves in trouble when doing rapid creation or removal of networks. added a simple file lock to deal with the collision and because it is not considered a performent path, use of the file lock should be ok. if proven otherwise in the future, some generic shared memory lock should be implemented for libpod and also used here. moved pkog/network to libpod/network because libpod is now being pulled into the package and it has therefore lost its generic nature. this will make it easier to absorb into libpod as we try to make the network closer to core operations. Fixes: #7807 Signed-off-by: baude <bbaude@redhat.com>
79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package network
|
|
|
|
/*
|
|
The code in this was kindly contributed by Dan Williams(dcbw@redhat.com). Many thanks
|
|
for his contributions.
|
|
*/
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
)
|
|
|
|
func incByte(subnet *net.IPNet, idx int, shift uint) error {
|
|
if idx < 0 {
|
|
return fmt.Errorf("no more subnets left")
|
|
}
|
|
if subnet.IP[idx] == 255 {
|
|
subnet.IP[idx] = 0
|
|
return incByte(subnet, idx-1, 0)
|
|
}
|
|
subnet.IP[idx] += 1 << shift
|
|
return nil
|
|
}
|
|
|
|
// NextSubnet returns subnet incremented by 1
|
|
func NextSubnet(subnet *net.IPNet) (*net.IPNet, error) {
|
|
newSubnet := &net.IPNet{
|
|
IP: subnet.IP,
|
|
Mask: subnet.Mask,
|
|
}
|
|
ones, bits := newSubnet.Mask.Size()
|
|
if ones == 0 {
|
|
return nil, fmt.Errorf("%s has only one subnet", subnet.String())
|
|
}
|
|
zeroes := uint(bits - ones)
|
|
shift := zeroes % 8
|
|
idx := ones/8 - 1
|
|
if idx < 0 {
|
|
idx = 0
|
|
}
|
|
if err := incByte(newSubnet, idx, shift); err != nil {
|
|
return nil, err
|
|
}
|
|
return newSubnet, nil
|
|
}
|
|
|
|
// LastIPInSubnet gets the last IP in a subnet
|
|
func LastIPInSubnet(addr *net.IPNet) (net.IP, error) { //nolint:interfacer
|
|
// re-parse to ensure clean network address
|
|
_, cidr, err := net.ParseCIDR(addr.String())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ones, bits := cidr.Mask.Size()
|
|
if ones == bits {
|
|
return FirstIPInSubnet(cidr)
|
|
}
|
|
hostStart := ones / 8
|
|
// Handle the first host byte
|
|
cidr.IP[hostStart] |= 0xff & cidr.Mask[hostStart]
|
|
// Fill the rest with ones
|
|
for i := hostStart; i < len(cidr.IP); i++ {
|
|
cidr.IP[i] = 0xff
|
|
}
|
|
return cidr.IP, nil
|
|
}
|
|
|
|
// FirstIPInSubnet gets the first IP in a subnet
|
|
func FirstIPInSubnet(addr *net.IPNet) (net.IP, error) { //nolint:interfacer
|
|
// re-parse to ensure clean network address
|
|
_, cidr, err := net.ParseCIDR(addr.String())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cidr.IP[len(cidr.IP)-1]++
|
|
return cidr.IP, nil
|
|
}
|