github.com/fafucoder/cilium@v1.6.11/pkg/ip/ip.go (about) 1 // Copyright 2017-2019 Authors of Cilium 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package ip 16 17 import ( 18 "bytes" 19 "encoding/binary" 20 "fmt" 21 "math/big" 22 "net" 23 "sort" 24 ) 25 26 const ( 27 ipv4BitLen = 8 * net.IPv4len 28 ipv6BitLen = 8 * net.IPv6len 29 ) 30 31 // CountIPsInCIDR takes a RFC4632/RFC4291-formatted IPv4/IPv6 CIDR and 32 // determines how many IP addresses reside within that CIDR. 33 // Returns 0 if the input CIDR cannot be parsed. 34 func CountIPsInCIDR(ipnet *net.IPNet) int { 35 subnet, size := ipnet.Mask.Size() 36 if subnet == size { 37 return 1 38 } 39 return 1<<uint(size-subnet) - 1 40 } 41 42 var ( 43 // v4Mappedv6Prefix is the RFC2765 IPv4-mapped address prefix. 44 v4Mappedv6Prefix = []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0xff, 0xff} 45 ipv4LeadingZeroes = []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0} 46 defaultIPv4 = []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0xff, 0xff, 0x0, 0x0, 0x0, 0x0} 47 defaultIPv6 = []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0} 48 upperIPv4 = []byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0xff, 0xff, 255, 255, 255, 255} 49 upperIPv6 = []byte{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff} 50 ) 51 52 // NetsByMask is used to sort a list of IP networks by the size of their masks. 53 // Implements sort.Interface. 54 type NetsByMask []*net.IPNet 55 56 func (s NetsByMask) Swap(i, j int) { 57 s[i], s[j] = s[j], s[i] 58 } 59 60 func (s NetsByMask) Less(i, j int) bool { 61 iPrefixSize, _ := s[i].Mask.Size() 62 jPrefixSize, _ := s[j].Mask.Size() 63 if iPrefixSize == jPrefixSize { 64 return bytes.Compare(s[i].IP, s[j].IP) < 0 65 } 66 return iPrefixSize < jPrefixSize 67 } 68 69 func (s NetsByMask) Len() int { 70 return len(s) 71 } 72 73 // Assert that NetsByMask implements sort.Interface. 74 var _ sort.Interface = NetsByMask{} 75 var _ sort.Interface = NetsByRange{} 76 77 // NetsByRange is used to sort a list of ranges, first by their last IPs, then by 78 // their first IPs 79 // Implements sort.Interface. 80 type NetsByRange []*netWithRange 81 82 func (s NetsByRange) Swap(i, j int) { 83 s[i], s[j] = s[j], s[i] 84 } 85 86 func (s NetsByRange) Less(i, j int) bool { 87 // First compare by last IP. 88 lastComparison := bytes.Compare(*s[i].Last, *s[j].Last) 89 if lastComparison < 0 { 90 return true 91 } else if lastComparison > 0 { 92 return false 93 } 94 95 // Then compare by first IP. 96 firstComparison := bytes.Compare(*s[i].First, *s[i].First) 97 if firstComparison < 0 { 98 return true 99 } else if firstComparison > 0 { 100 return false 101 } 102 103 // First and last IPs are the same, so thus are equal, and s[i] 104 // is not less than s[j]. 105 return false 106 } 107 108 func (s NetsByRange) Len() int { 109 return len(s) 110 } 111 112 // RemoveCIDRs removes the specified CIDRs from another set of CIDRs. If a CIDR 113 // to remove is not contained within the CIDR, the CIDR to remove is ignored. A 114 // slice of CIDRs is returned which contains the set of CIDRs provided minus 115 // the set of CIDRs which were removed. Both input slices may be modified by 116 // calling this function. 117 func RemoveCIDRs(allowCIDRs, removeCIDRs []*net.IPNet) ([]*net.IPNet, error) { 118 119 // Ensure that we iterate through the provided CIDRs in order of largest 120 // subnet first. 121 sort.Sort(NetsByMask(removeCIDRs)) 122 123 PreLoop: 124 // Remove CIDRs which are contained within CIDRs that we want to remove; 125 // such CIDRs are redundant. 126 for j, removeCIDR := range removeCIDRs { 127 for i, removeCIDR2 := range removeCIDRs { 128 if i == j { 129 continue 130 } 131 if removeCIDR.Contains(removeCIDR2.IP) { 132 removeCIDRs = append(removeCIDRs[:i], removeCIDRs[i+1:]...) 133 // Re-trigger loop since we have modified the slice we are iterating over. 134 goto PreLoop 135 } 136 } 137 } 138 139 for _, remove := range removeCIDRs { 140 Loop: 141 for i, allowCIDR := range allowCIDRs { 142 143 // Don't allow comparison of different address spaces. 144 if allowCIDR.IP.To4() != nil && remove.IP.To4() == nil || 145 allowCIDR.IP.To4() == nil && remove.IP.To4() != nil { 146 return nil, fmt.Errorf("cannot mix IP addresses of different IP protocol versions") 147 } 148 149 // Only remove CIDR if it is contained in the subnet we are allowing. 150 if allowCIDR.Contains(remove.IP.Mask(remove.Mask)) { 151 nets, err := removeCIDR(allowCIDR, remove) 152 if err != nil { 153 return nil, err 154 } 155 156 // Remove CIDR that we have just processed and append new CIDRs 157 // that we computed from removing the CIDR to remove. 158 allowCIDRs = append(allowCIDRs[:i], allowCIDRs[i+1:]...) 159 allowCIDRs = append(allowCIDRs, nets...) 160 goto Loop 161 } else if remove.Contains(allowCIDR.IP.Mask(allowCIDR.Mask)) { 162 // If a CIDR that we want to remove contains a CIDR in the list 163 // that is allowed, then we can just remove the CIDR to allow. 164 allowCIDRs = append(allowCIDRs[:i], allowCIDRs[i+1:]...) 165 goto Loop 166 } 167 } 168 } 169 170 return allowCIDRs, nil 171 } 172 173 func getNetworkPrefix(ipNet *net.IPNet) *net.IP { 174 var mask net.IP 175 176 if ipNet.IP.To4() == nil { 177 mask = make(net.IP, net.IPv6len) 178 for i := 0; i < len(ipNet.Mask); i++ { 179 mask[net.IPv6len-i-1] = ipNet.IP[net.IPv6len-i-1] & ^ipNet.Mask[i] 180 } 181 } else { 182 mask = make(net.IP, net.IPv4len) 183 for i := 0; i < net.IPv4len; i++ { 184 mask[net.IPv4len-i-1] = ipNet.IP[net.IPv6len-i-1] & ^ipNet.Mask[i] 185 } 186 } 187 188 return &mask 189 } 190 191 func removeCIDR(allowCIDR, removeCIDR *net.IPNet) ([]*net.IPNet, error) { 192 var allows []*net.IPNet 193 var allowIsIpv4, removeIsIpv4 bool 194 var allowBitLen int 195 196 if allowCIDR.IP.To4() != nil { 197 allowIsIpv4 = true 198 allowBitLen = ipv4BitLen 199 } else { 200 allowBitLen = ipv6BitLen 201 } 202 203 if removeCIDR.IP.To4() != nil { 204 removeIsIpv4 = true 205 } 206 207 if removeIsIpv4 != allowIsIpv4 { 208 return nil, fmt.Errorf("cannot mix IP addresses of different IP protocol versions") 209 } 210 211 // Get size of each CIDR mask. 212 allowSize, _ := allowCIDR.Mask.Size() 213 removeSize, _ := removeCIDR.Mask.Size() 214 215 if allowSize >= removeSize { 216 return nil, fmt.Errorf("allow CIDR prefix must be a superset of " + 217 "remove CIDR prefix") 218 } 219 220 allowFirstIPMasked := allowCIDR.IP.Mask(allowCIDR.Mask) 221 removeFirstIPMasked := removeCIDR.IP.Mask(removeCIDR.Mask) 222 223 // Convert to IPv4 in IPv6 addresses if needed. 224 if allowIsIpv4 { 225 allowFirstIPMasked = append(v4Mappedv6Prefix, allowFirstIPMasked...) 226 } 227 228 if removeIsIpv4 { 229 removeFirstIPMasked = append(v4Mappedv6Prefix, removeFirstIPMasked...) 230 } 231 232 allowFirstIP := &allowFirstIPMasked 233 removeFirstIP := &removeFirstIPMasked 234 235 // Create CIDR prefixes with mask size of Y+1, Y+2 ... X where Y is the mask 236 // length of the CIDR prefix B from which we are excluding a CIDR prefix A 237 // with mask length X. 238 for i := (allowBitLen - allowSize - 1); i >= (allowBitLen - removeSize); i-- { 239 // The mask for each CIDR prefix is simply the ith bit flipped, and then 240 // zero'ing out all subsequent bits (the host identifier part of the 241 // prefix). 242 newMaskSize := allowBitLen - i 243 newIP := (*net.IP)(flipNthBit((*[]byte)(removeFirstIP), uint(i))) 244 for k := range *allowFirstIP { 245 (*newIP)[k] = (*allowFirstIP)[k] | (*newIP)[k] 246 } 247 248 newMask := net.CIDRMask(newMaskSize, allowBitLen) 249 newIPMasked := newIP.Mask(newMask) 250 251 newIPNet := net.IPNet{IP: newIPMasked, Mask: newMask} 252 allows = append(allows, &newIPNet) 253 } 254 255 return allows, nil 256 } 257 258 func getByteIndexOfBit(bit uint) uint { 259 return net.IPv6len - (bit / 8) - 1 260 } 261 262 func getNthBit(ip *net.IP, bitNum uint) uint8 { 263 byteNum := getByteIndexOfBit(bitNum) 264 bits := (*ip)[byteNum] 265 b := uint8(bits) 266 return b >> (bitNum % 8) & 1 267 } 268 269 func flipNthBit(ip *[]byte, bitNum uint) *[]byte { 270 ipCopy := make([]byte, len(*ip)) 271 copy(ipCopy, *ip) 272 byteNum := getByteIndexOfBit(bitNum) 273 ipCopy[byteNum] = ipCopy[byteNum] ^ 1<<(bitNum%8) 274 275 return &ipCopy 276 } 277 278 func ipNetToRange(ipNet net.IPNet) netWithRange { 279 firstIP := make(net.IP, len(ipNet.IP)) 280 lastIP := make(net.IP, len(ipNet.IP)) 281 282 copy(firstIP, ipNet.IP) 283 copy(lastIP, ipNet.IP) 284 285 firstIP = firstIP.Mask(ipNet.Mask) 286 lastIP = lastIP.Mask(ipNet.Mask) 287 288 if firstIP.To4() != nil { 289 firstIP = append(v4Mappedv6Prefix, firstIP...) 290 lastIP = append(v4Mappedv6Prefix, lastIP...) 291 } 292 293 lastIPMask := make(net.IPMask, len(ipNet.Mask)) 294 copy(lastIPMask, ipNet.Mask) 295 for i := range lastIPMask { 296 lastIPMask[len(lastIPMask)-i-1] = ^lastIPMask[len(lastIPMask)-i-1] 297 lastIP[net.IPv6len-i-1] = lastIP[net.IPv6len-i-1] | lastIPMask[len(lastIPMask)-i-1] 298 } 299 300 return netWithRange{First: &firstIP, Last: &lastIP, Network: &ipNet} 301 } 302 303 func getPreviousIP(ip net.IP) net.IP { 304 // Cannot go lower than zero! 305 if ip.Equal(net.IP(defaultIPv4)) || ip.Equal(net.IP(defaultIPv6)) { 306 return ip 307 } 308 309 previousIP := make(net.IP, len(ip)) 310 copy(previousIP, ip) 311 312 var overflow bool 313 var lowerByteBound int 314 if ip.To4() != nil { 315 lowerByteBound = net.IPv6len - net.IPv4len 316 } else { 317 lowerByteBound = 0 318 } 319 for i := len(ip) - 1; i >= lowerByteBound; i-- { 320 if overflow || i == len(ip)-1 { 321 previousIP[i]-- 322 } 323 // Track if we have overflowed and thus need to continue subtracting. 324 if ip[i] == 0 && previousIP[i] == 255 { 325 overflow = true 326 } else { 327 overflow = false 328 } 329 } 330 return previousIP 331 } 332 333 // GetNextIP returns the next IP from the given IP address. If the given IP is 334 // the last IP of a v4 or v6 range, the same IP is returned. 335 func GetNextIP(ip net.IP) net.IP { 336 if ip.Equal(upperIPv4) || ip.Equal(upperIPv6) { 337 return ip 338 } 339 340 nextIP := make(net.IP, len(ip)) 341 switch len(ip) { 342 case net.IPv4len: 343 ipU32 := binary.BigEndian.Uint32(ip) 344 ipU32++ 345 binary.BigEndian.PutUint32(nextIP, ipU32) 346 return nextIP 347 case net.IPv6len: 348 ipU64 := binary.BigEndian.Uint64(ip[net.IPv6len/2:]) 349 ipU64++ 350 binary.BigEndian.PutUint64(nextIP[net.IPv6len/2:], ipU64) 351 if ipU64 == 0 { 352 ipU64 = binary.BigEndian.Uint64(ip[:net.IPv6len/2]) 353 ipU64++ 354 binary.BigEndian.PutUint64(nextIP[:net.IPv6len/2], ipU64) 355 } else { 356 copy(nextIP[:net.IPv6len/2], ip[:net.IPv6len/2]) 357 } 358 return nextIP 359 default: 360 return ip 361 } 362 } 363 364 func createSpanningCIDR(r netWithRange) net.IPNet { 365 // Don't want to modify the values of the provided range, so make copies. 366 lowest := *r.First 367 highest := *r.Last 368 369 var isIPv4 bool 370 var spanningMaskSize, bitLen, byteLen int 371 if lowest.To4() != nil { 372 isIPv4 = true 373 bitLen = ipv4BitLen 374 byteLen = net.IPv4len 375 } else { 376 bitLen = ipv6BitLen 377 byteLen = net.IPv6len 378 } 379 380 if isIPv4 { 381 spanningMaskSize = ipv4BitLen 382 } else { 383 spanningMaskSize = ipv6BitLen 384 } 385 386 // Convert to big Int so we can easily do bitshifting on the IP addresses, 387 // since golang only provides up to 64-bit unsigned integers. 388 lowestBig := big.NewInt(0).SetBytes(lowest) 389 highestBig := big.NewInt(0).SetBytes(highest) 390 391 // Starting from largest mask / smallest range possible, apply a mask one bit 392 // larger in each iteration to the upper bound in the range until we have 393 // masked enough to pass the lower bound in the range. This 394 // gives us the size of the prefix for the spanning CIDR to return as 395 // well as the IP for the CIDR prefix of the spanning CIDR. 396 for spanningMaskSize > 0 && lowestBig.Cmp(highestBig) < 0 { 397 spanningMaskSize-- 398 mask := big.NewInt(1) 399 mask = mask.Lsh(mask, uint(bitLen-spanningMaskSize)) 400 mask = mask.Mul(mask, big.NewInt(-1)) 401 highestBig = highestBig.And(highestBig, mask) 402 } 403 404 // If ipv4, need to append 0s because math.Big gets rid of preceding zeroes. 405 if isIPv4 { 406 highest = append(ipv4LeadingZeroes, highestBig.Bytes()...) 407 } else { 408 highest = highestBig.Bytes() 409 } 410 411 // Int does not store leading zeroes. 412 if len(highest) == 0 { 413 highest = make([]byte, byteLen) 414 } 415 416 newNet := net.IPNet{IP: highest, Mask: net.CIDRMask(spanningMaskSize, bitLen)} 417 return newNet 418 } 419 420 type netWithRange struct { 421 First *net.IP 422 Last *net.IP 423 Network *net.IPNet 424 } 425 426 func mergeAdjacentCIDRs(ranges []*netWithRange) []*netWithRange { 427 // Sort the ranges. This sorts first by the last IP, then first IP, then by 428 // the IP network in the list itself 429 sort.Sort(NetsByRange(ranges)) 430 431 // Merge adjacent CIDRs if possible. 432 for i := len(ranges) - 1; i > 0; i-- { 433 first1 := getPreviousIP(*ranges[i].First) 434 435 // Since the networks are sorted, we know that if a network in the list 436 // is adjacent to another one in the list, it will be the network next 437 // to it in the list. If the previous IP of the current network we are 438 // processing overlaps with the last IP of the previous network in the 439 // list, then we can merge the two ranges together. 440 if bytes.Compare(first1, *ranges[i-1].Last) <= 0 { 441 // Pick the minimum of the first two IPs to represent the start 442 // of the new range. 443 var minFirstIP *net.IP 444 if bytes.Compare(*ranges[i-1].First, *ranges[i].First) < 0 { 445 minFirstIP = ranges[i-1].First 446 } else { 447 minFirstIP = ranges[i].First 448 } 449 450 // Always take the last IP of the ith IP. 451 newRangeLast := make(net.IP, len(*ranges[i].Last)) 452 copy(newRangeLast, *ranges[i].Last) 453 454 newRangeFirst := make(net.IP, len(*minFirstIP)) 455 copy(newRangeFirst, *minFirstIP) 456 457 // Can't set the network field because since we are combining a 458 // range of IPs, and we don't yet know what CIDR prefix(es) represent 459 // the new range. 460 ranges[i-1] = &netWithRange{First: &newRangeFirst, Last: &newRangeLast, Network: nil} 461 462 // Since we have combined ranges[i] with the preceding item in the 463 // ranges list, we can delete ranges[i] from the slice. 464 ranges = append(ranges[:i], ranges[i+1:]...) 465 } 466 } 467 return ranges 468 } 469 470 // coalesceRanges converts ranges into an equivalent list of net.IPNets. 471 // All IPs in ranges should be of the same address family (IPv4 or IPv6). 472 func coalesceRanges(ranges []*netWithRange) []*net.IPNet { 473 coalescedCIDRs := []*net.IPNet{} 474 // Create CIDRs from ranges that were combined if needed. 475 for _, netRange := range ranges { 476 // If the Network field of netWithRange wasn't modified, then we can 477 // add it to the list which we will return, as it cannot be joined with 478 // any other CIDR in the list. 479 if netRange.Network != nil { 480 coalescedCIDRs = append(coalescedCIDRs, netRange.Network) 481 } else { 482 // We have joined two ranges together, so we need to find the new CIDRs 483 // that represent this range. 484 rangeCIDRs := rangeToCIDRs(*netRange.First, *netRange.Last) 485 coalescedCIDRs = append(coalescedCIDRs, rangeCIDRs...) 486 } 487 } 488 489 return coalescedCIDRs 490 } 491 492 // CoalesceCIDRs transforms the provided list of CIDRs into the most-minimal 493 // equivalent set of IPv4 and IPv6 CIDRs. 494 // It removes CIDRs that are subnets of other CIDRs in the list, and groups 495 // together CIDRs that have the same mask size into a CIDR of the same mask 496 // size provided that they share the same number of most significant 497 // mask-size bits. 498 // 499 // Note: this algorithm was ported from the Python library netaddr. 500 // https://github.com/drkjam/netaddr . 501 func CoalesceCIDRs(cidrs []*net.IPNet) ([]*net.IPNet, []*net.IPNet) { 502 503 ranges4 := []*netWithRange{} 504 ranges6 := []*netWithRange{} 505 506 for _, network := range cidrs { 507 newNetToRange := ipNetToRange(*network) 508 if network.IP.To4() != nil { 509 ranges4 = append(ranges4, &newNetToRange) 510 } else { 511 ranges6 = append(ranges6, &newNetToRange) 512 } 513 } 514 515 return coalesceRanges(mergeAdjacentCIDRs(ranges4)), coalesceRanges(mergeAdjacentCIDRs(ranges6)) 516 } 517 518 // rangeToCIDRs converts the range of IPs covered by firstIP and lastIP to 519 // a list of CIDRs that contains all of the IPs covered by the range. 520 func rangeToCIDRs(firstIP, lastIP net.IP) []*net.IPNet { 521 522 // First, create a CIDR that spans both IPs. 523 spanningCIDR := createSpanningCIDR(netWithRange{&firstIP, &lastIP, nil}) 524 spanningRange := ipNetToRange(spanningCIDR) 525 firstIPSpanning := spanningRange.First 526 lastIPSpanning := spanningRange.Last 527 528 cidrList := []*net.IPNet{} 529 530 // If the first IP of the spanning CIDR passes the lower bound (firstIP), 531 // we need to split the spanning CIDR and only take the IPs that are 532 // greater than the value which we split on, as we do not want the lesser 533 // values since they are less than the lower-bound (firstIP). 534 if bytes.Compare(*firstIPSpanning, firstIP) < 0 { 535 // Split on the previous IP of the first IP so that the right list of IPs 536 // of the partition includes the firstIP. 537 prevFirstRangeIP := getPreviousIP(firstIP) 538 var bitLen int 539 if prevFirstRangeIP.To4() != nil { 540 bitLen = ipv4BitLen 541 } else { 542 bitLen = ipv6BitLen 543 } 544 _, _, right := partitionCIDR(spanningCIDR, net.IPNet{IP: prevFirstRangeIP, Mask: net.CIDRMask(bitLen, bitLen)}) 545 546 // Append all CIDRs but the first, as this CIDR includes the upper 547 // bound of the spanning CIDR, which we still need to partition on. 548 cidrList = append(cidrList, right...) 549 spanningCIDR = *right[0] 550 cidrList = cidrList[1:] 551 } 552 553 // Conversely, if the last IP of the spanning CIDR passes the upper bound 554 // (lastIP), we need to split the spanning CIDR and only take the IPs that 555 // are greater than the value which we split on, as we do not want the greater 556 // values since they are greater than the upper-bound (lastIP). 557 if bytes.Compare(*lastIPSpanning, lastIP) > 0 { 558 // Split on the next IP of the last IP so that the left list of IPs 559 // of the partition include the lastIP. 560 nextFirstRangeIP := GetNextIP(lastIP) 561 var bitLen int 562 if nextFirstRangeIP.To4() != nil { 563 bitLen = ipv4BitLen 564 } else { 565 bitLen = ipv6BitLen 566 } 567 left, _, _ := partitionCIDR(spanningCIDR, net.IPNet{IP: nextFirstRangeIP, Mask: net.CIDRMask(bitLen, bitLen)}) 568 cidrList = append(cidrList, left...) 569 } else { 570 // Otherwise, there is no need to partition; just use add the spanning 571 // CIDR to the list of networks. 572 cidrList = append(cidrList, &spanningCIDR) 573 } 574 return cidrList 575 } 576 577 // partitionCIDR returns a list of IP Networks partitioned upon excludeCIDR. 578 // The first list contains the networks to the left of the excludeCIDR in the 579 // partition, the second is a list containing the excludeCIDR itself if it is 580 // contained within the targetCIDR (nil otherwise), and the 581 // third is a list containing the networks to the right of the excludeCIDR in 582 // the partition. 583 func partitionCIDR(targetCIDR net.IPNet, excludeCIDR net.IPNet) ([]*net.IPNet, []*net.IPNet, []*net.IPNet) { 584 var targetIsIPv4 bool 585 if targetCIDR.IP.To4() != nil { 586 targetIsIPv4 = true 587 } 588 589 targetIPRange := ipNetToRange(targetCIDR) 590 excludeIPRange := ipNetToRange(excludeCIDR) 591 592 targetFirstIP := *targetIPRange.First 593 targetLastIP := *targetIPRange.Last 594 595 excludeFirstIP := *excludeIPRange.First 596 excludeLastIP := *excludeIPRange.Last 597 598 targetMaskSize, _ := targetCIDR.Mask.Size() 599 excludeMaskSize, _ := excludeCIDR.Mask.Size() 600 601 if bytes.Compare(excludeLastIP, targetFirstIP) < 0 { 602 return nil, nil, []*net.IPNet{&targetCIDR} 603 } else if bytes.Compare(targetLastIP, excludeFirstIP) < 0 { 604 return []*net.IPNet{&targetCIDR}, nil, nil 605 } 606 607 if targetMaskSize >= excludeMaskSize { 608 return nil, []*net.IPNet{&targetCIDR}, nil 609 } 610 611 left := []*net.IPNet{} 612 right := []*net.IPNet{} 613 614 newPrefixLen := targetMaskSize + 1 615 616 targetFirstCopy := make(net.IP, len(targetFirstIP)) 617 copy(targetFirstCopy, targetFirstIP) 618 619 iLowerOld := make(net.IP, len(targetFirstCopy)) 620 copy(iLowerOld, targetFirstCopy) 621 622 // Since golang only supports up to unsigned 64-bit integers, and we need 623 // to perform addition on addresses, use math/big library, which allows 624 // for manipulation of large integers. 625 626 // Used to track the current lower and upper bounds of the ranges to compare 627 // to excludeCIDR. 628 iLower := big.NewInt(0) 629 iUpper := big.NewInt(0) 630 iLower = iLower.SetBytes(targetFirstCopy) 631 632 var bitLen int 633 634 if targetIsIPv4 { 635 bitLen = ipv4BitLen 636 } else { 637 bitLen = ipv6BitLen 638 } 639 shiftAmount := (uint)(bitLen - newPrefixLen) 640 641 targetIPInt := big.NewInt(0) 642 targetIPInt.SetBytes(targetFirstIP.To16()) 643 644 exp := big.NewInt(0) 645 646 // Use left shift for exponentiation 647 exp = exp.Lsh(big.NewInt(1), shiftAmount) 648 iUpper = iUpper.Add(targetIPInt, exp) 649 650 matched := big.NewInt(0) 651 652 for excludeMaskSize >= newPrefixLen { 653 // Append leading zeros to IPv4 addresses, as math.Big.Int does not 654 // append them when the IP address is copied from a byte array to 655 // math.Big.Int. Leading zeroes are required for parsing IPv4 addresses 656 // for use with net.IP / net.IPNet. 657 var iUpperBytes, iLowerBytes []byte 658 if targetIsIPv4 { 659 iUpperBytes = append(ipv4LeadingZeroes, iUpper.Bytes()...) 660 iLowerBytes = append(ipv4LeadingZeroes, iLower.Bytes()...) 661 } else { 662 iUpperBytesLen := len(iUpper.Bytes()) 663 // Make sure that the number of bytes in the array matches what net 664 // package expects, as big package doesn't append leading zeroes. 665 if iUpperBytesLen != net.IPv6len { 666 numZeroesToAppend := net.IPv6len - iUpperBytesLen 667 zeroBytes := make([]byte, numZeroesToAppend) 668 iUpperBytes = append(zeroBytes, iUpper.Bytes()...) 669 } else { 670 iUpperBytes = iUpper.Bytes() 671 672 } 673 674 iLowerBytesLen := len(iLower.Bytes()) 675 if iLowerBytesLen != net.IPv6len { 676 numZeroesToAppend := net.IPv6len - iLowerBytesLen 677 zeroBytes := make([]byte, numZeroesToAppend) 678 iLowerBytes = append(zeroBytes, iLower.Bytes()...) 679 } else { 680 iLowerBytes = iLower.Bytes() 681 682 } 683 } 684 // If the IP we are excluding over is of a higher value than the current 685 // CIDR prefix we are generating, add the CIDR prefix to the set of IPs 686 // to the left of the exclude CIDR 687 if bytes.Compare(excludeFirstIP, iUpperBytes) >= 0 { 688 left = append(left, &net.IPNet{IP: iLowerBytes, Mask: net.CIDRMask(newPrefixLen, bitLen)}) 689 matched = matched.Set(iUpper) 690 } else { 691 // Same as above, but opposite. 692 right = append(right, &net.IPNet{IP: iUpperBytes, Mask: net.CIDRMask(newPrefixLen, bitLen)}) 693 matched = matched.Set(iLower) 694 } 695 696 newPrefixLen++ 697 698 if newPrefixLen > bitLen { 699 break 700 } 701 702 iLower = iLower.Set(matched) 703 iUpper = iUpper.Add(matched, big.NewInt(0).Lsh(big.NewInt(1), uint(bitLen-newPrefixLen))) 704 705 } 706 excludeList := []*net.IPNet{&excludeCIDR} 707 708 return left, excludeList, right 709 } 710 711 // KeepUniqueIPs transforms the provided multiset of IPs into a single set, 712 // lexicographically sorted via a byte-wise comparison of the IP slices (i.e. 713 // IPv4 addresses show up before IPv6). 714 // The slice is manipulated in-place destructively. 715 // 716 // 1- Sort the slice by comparing the IPs as bytes 717 // 2- For every unseen unique IP in the sorted slice, move it to the end of 718 // the return slice. 719 // Note that the slice is always large enough and, because it is sorted, we 720 // will not overwrite a valid element with another. To overwrite an element i 721 // with j, i must have come before j AND we decided it was a duplicate of the 722 // element at i-1. 723 func KeepUniqueIPs(ips []net.IP) []net.IP { 724 sort.Slice(ips, func(i, j int) bool { 725 return bytes.Compare(ips[i], ips[j]) == -1 726 }) 727 728 returnIPs := ips[:0] // len==0 but cap==cap(ips) 729 for readIdx, ip := range ips { 730 if len(returnIPs) == 0 || !returnIPs[len(returnIPs)-1].Equal(ips[readIdx]) { 731 returnIPs = append(returnIPs, ip) 732 } 733 } 734 735 return returnIPs 736 } 737 738 var privateIPBlocks []*net.IPNet 739 740 func initPrivatePrefixes() { 741 // We only care about global scope prefixes here. 742 for _, cidr := range []string{ 743 "10.0.0.0/8", // RFC1918 744 "172.16.0.0/12", // RFC1918 745 "192.168.0.0/16", // RFC1918 746 "fc00::/7", // IPv6 ULA 747 } { 748 _, block, _ := net.ParseCIDR(cidr) 749 privateIPBlocks = append(privateIPBlocks, block) 750 } 751 } 752 753 var excludedIPs []net.IP 754 755 func init() { 756 initPrivatePrefixes() 757 initExcludedIPs() 758 } 759 760 // IsExcluded returns whether a given IP is must be excluded 761 // due to coming from blacklisted device. 762 func IsExcluded(excludeList []net.IP, ip net.IP) bool { 763 for _, e := range excludeList { 764 if e.Equal(ip) { 765 return true 766 } 767 } 768 return false 769 } 770 771 // IsPublicAddr returns whether a given global IP is from 772 // a public range. 773 func IsPublicAddr(ip net.IP) bool { 774 for _, block := range privateIPBlocks { 775 if block.Contains(ip) { 776 return false 777 } 778 } 779 return true 780 } 781 782 // GetExcludedIPs returns a list of IPs from netdevices that Cilium 783 // needs to exclude to operate 784 func GetExcludedIPs() []net.IP { 785 return excludedIPs 786 } 787 788 // GetCIDRPrefixesFromIPs returns all of the ips as a slice of *net.IPNet. 789 func GetCIDRPrefixesFromIPs(ips []net.IP) []*net.IPNet { 790 if len(ips) == 0 { 791 return nil 792 } 793 res := make([]*net.IPNet, 0, len(ips)) 794 for _, ip := range ips { 795 res = append(res, IPToPrefix(ip)) 796 } 797 return res 798 } 799 800 // IPToPrefix returns the corresponding IPNet for the given IP. 801 func IPToPrefix(ip net.IP) *net.IPNet { 802 bits := net.IPv6len * 8 803 if ip.To4() != nil { 804 ip = ip.To4() 805 bits = net.IPv4len * 8 806 } 807 prefix := &net.IPNet{ 808 IP: ip, 809 Mask: net.CIDRMask(bits, bits), 810 } 811 return prefix 812 }