72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Node struct {
|
|
Real string `yaml:"real"`
|
|
Virt string `yaml:"virt"`
|
|
}
|
|
|
|
type Config struct {
|
|
Network struct {
|
|
Self Node `yaml:"self"`
|
|
Nodes []Node `yaml:"nodes"`
|
|
} `yaml:"network"`
|
|
|
|
CurrentNodeKey uint32
|
|
IPv4Nodes map[uint32]string
|
|
lock sync.Mutex
|
|
}
|
|
|
|
var config = NewConfig()
|
|
|
|
func NewConfig() *Config {
|
|
cnf := &Config{IPv4Nodes: make(map[uint32]string)}
|
|
f, err := os.Open("config.yaml")
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
dec := yaml.NewDecoder(f)
|
|
err = dec.Decode(cnf)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
for _, node := range cnf.Network.Nodes {
|
|
values := strings.Split(node.Virt, "/")
|
|
if len(values) != 2 {
|
|
log.Panic("virt must have '/'. eg 1.1.1.1/24")
|
|
}
|
|
inet, net, err := net.ParseCIDR(node.Virt)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
cnf.IPv4Nodes[binary.BigEndian.Uint32([]byte(inet.To4()))] = node.Real
|
|
log.Printf("real %v virt %v %v", node.Real, inet, net)
|
|
}
|
|
|
|
inet, _, err := net.ParseCIDR(cnf.Network.Self.Virt)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
cnf.CurrentNodeKey = binary.BigEndian.Uint32([]byte(inet.To4()))
|
|
|
|
return cnf
|
|
}
|
|
|
|
func (cnf *Config) Lock(do func()) {
|
|
cnf.lock.Lock()
|
|
defer cnf.lock.Unlock()
|
|
do()
|
|
}
|