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
|
package hst
import (
"errors"
"fmt"
"os"
"strings"
"github.com/edwarnicke/exechelper"
)
type AddressCounter = int
type Ip4AddressAllocator struct {
networks map[int]AddressCounter
chosenOctet int
assignedIps []string
}
func (a *Ip4AddressAllocator) AddNetwork(networkNumber int) {
a.networks[networkNumber] = 1
}
func (a *Ip4AddressAllocator) NewIp4InterfaceAddress(inputNetworkNumber ...int) (string, error) {
var networkNumber int = 0
if len(inputNetworkNumber) > 0 {
networkNumber = inputNetworkNumber[0]
}
if _, ok := a.networks[networkNumber]; !ok {
a.AddNetwork(networkNumber)
}
numberOfAddresses := a.networks[networkNumber]
if numberOfAddresses == 254 {
return "", fmt.Errorf("no available IPv4 addresses")
}
address, err := a.createIpAddress(networkNumber, numberOfAddresses)
a.networks[networkNumber] = numberOfAddresses + 1
return address + "/24", err
}
// Creates a file every time an IP is assigned: used to keep track of addresses in use.
// If an address is not in use, 'counter' is then copied to 'chosenOctet' and it is used for the remaining tests.
// Also checks host IP addresses.
func (a *Ip4AddressAllocator) createIpAddress(networkNumber int, numberOfAddresses int) (string, error) {
hostIps, _ := exechelper.CombinedOutput("ip a")
counter := 10
var address string
for {
if a.chosenOctet != 0 {
address = fmt.Sprintf("10.%v.%v.%v", a.chosenOctet, networkNumber, numberOfAddresses)
file, err := os.Create(address)
if err != nil {
return "", errors.New("unable to create file: " + fmt.Sprint(err))
}
file.Close()
break
} else {
address = fmt.Sprintf("10.%v.%v.%v", counter, networkNumber, numberOfAddresses)
_, err := os.Stat(address)
if err == nil || strings.Contains(string(hostIps), address) {
counter++
} else if os.IsNotExist(err) {
file, err := os.Create(address)
if err != nil {
return "", errors.New("unable to create file: " + fmt.Sprint(err))
}
file.Close()
a.chosenOctet = counter
break
} else {
return "", errors.New("an error occurred while checking if a file exists: " + fmt.Sprint(err))
}
}
}
a.assignedIps = append(a.assignedIps, address)
return address, nil
}
func (a *Ip4AddressAllocator) DeleteIpAddresses() {
for ip := range a.assignedIps {
os.Remove(a.assignedIps[ip])
}
}
func NewIp4AddressAllocator() *Ip4AddressAllocator {
var ip4AddrAllocator = new(Ip4AddressAllocator)
ip4AddrAllocator.networks = make(map[int]AddressCounter)
ip4AddrAllocator.AddNetwork(0)
return ip4AddrAllocator
}
|
