-
Notifications
You must be signed in to change notification settings - Fork 4
/
ipnets.go
110 lines (101 loc) · 2.41 KB
/
ipnets.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
// Copyright © 2019 Platina Systems, Inc. All rights reserved.
// Use of this source code is governed by the GPL-2 license described in the
// LICENSE file.
package xeth
import (
"net"
"sync"
"unsafe"
)
type DevAddIPNet struct {
Xid
*net.IPNet
}
type DevDelIPNet struct {
Xid
Prefix string
}
var poolIPNet = sync.Pool{
New: func() interface{} {
return &net.IPNet{
IP: make([]byte, net.IPv6len, net.IPv6len),
Mask: make([]byte, net.IPv6len, net.IPv6len),
}
},
}
func (xid Xid) RxIP4Add(addr, mask uint32) *DevAddIPNet {
l := LinkOf(xid)
ip := net.IP(make([]byte, net.IPv4len, net.IPv4len))
*(*uint32)(unsafe.Pointer(&ip[0])) = addr
nets := l.IPNets()
for _, entry := range nets {
if ip.Equal(entry.IP) {
return &DevAddIPNet{xid, entry}
}
}
clone := poolIPNet.Get().(*net.IPNet)
*(*uint32)(unsafe.Pointer(&clone.IP[0])) = addr
*(*uint32)(unsafe.Pointer(&clone.Mask[0])) = mask
clone.IP = clone.IP[:net.IPv4len]
clone.Mask = clone.Mask[:net.IPv4len]
l.IPNets(append(nets, clone))
return &DevAddIPNet{xid, clone}
}
func (xid Xid) RxIP4Del(addr, mask uint32) *DevDelIPNet {
l := LinkOf(xid)
if l == nil {
return nil
}
ip := net.IP(make([]byte, net.IPv4len, net.IPv4len))
*(*uint32)(unsafe.Pointer(&ip[0])) = addr
nets := l.IPNets()
for i, entry := range nets {
if ip.Equal(entry.IP) {
prefix := entry.String()
n := len(nets) - 1
copy(nets[i:], nets[i+1:])
l.IPNets(nets[:n])
entry.IP = entry.IP[:cap(entry.IP)]
entry.Mask = entry.Mask[:cap(entry.Mask)]
poolIPNet.Put(entry)
return &DevDelIPNet{xid, prefix}
}
}
return nil
}
func (xid Xid) RxIP6Add(addr []byte, len int) *DevAddIPNet {
l := LinkOf(xid)
ip := net.IP(addr)
nets := l.IPNets()
for _, entry := range nets {
if ip.Equal(entry.IP) {
return &DevAddIPNet{xid, entry}
}
}
clone := poolIPNet.Get().(*net.IPNet)
copy(clone.IP, ip)
copy(clone.Mask, net.CIDRMask(len, net.IPv6len*8))
l.IPNets(append(nets, clone))
return &DevAddIPNet{xid, clone}
}
func (xid Xid) RxIP6Del(addr []byte) *DevDelIPNet {
l := LinkOf(xid)
if l == nil {
return nil
}
ip := net.IP(addr)
nets := l.IPNets()
for i, entry := range nets {
if ip.Equal(entry.IP) {
prefix := entry.String()
n := len(nets) - 1
copy(nets[i:], nets[i+1:])
l.IPNets(nets[:n])
entry.IP = entry.IP[:cap(entry.IP)]
entry.Mask = entry.Mask[:cap(entry.Mask)]
poolIPNet.Put(entry)
return &DevDelIPNet{xid, prefix}
}
}
return nil
}