Factor out some code
This commit is contained in:
parent
bbdc18ed1b
commit
2ed92fadf9
53
src/meshname/config.go
Normal file
53
src/meshname/config.go
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
package meshname
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net"
|
||||||
|
|
||||||
|
"github.com/miekg/dns"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
func GenConf(target, zone string) (string, error) {
|
||||||
|
ip := net.ParseIP(target)
|
||||||
|
if ip == nil {
|
||||||
|
return "", errors.New("Invalid IP address")
|
||||||
|
}
|
||||||
|
subDomain := DomainFromIP(&ip)
|
||||||
|
selfRecord := fmt.Sprintf("\t\t\"%s.%s AAAA %s\"\n", subDomain, zone, target)
|
||||||
|
confString := fmt.Sprintf("{\n\t\"%s\":[\n%s\t]\n}", subDomain, selfRecord)
|
||||||
|
|
||||||
|
return confString, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load zoneConfig from a JSON file
|
||||||
|
func ParseConfigFile(configPath string) (map[string][]dns.RR, error) {
|
||||||
|
conf, err := ioutil.ReadFile(configPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var dat map[string][]string
|
||||||
|
if err := json.Unmarshal(conf, &dat); err == nil {
|
||||||
|
return ParseZoneConfigMap(dat)
|
||||||
|
} else {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseZoneConfigMap(zoneConfigMap map[string][]string) (map[string][]dns.RR, error) {
|
||||||
|
var zoneConfig = make(map[string][]dns.RR)
|
||||||
|
for subDomain, records := range zoneConfigMap {
|
||||||
|
for _, r := range records {
|
||||||
|
if rr, err := dns.NewRR(r); err == nil {
|
||||||
|
zoneConfig[subDomain] = append(zoneConfig[subDomain], rr)
|
||||||
|
} else {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return zoneConfig, nil
|
||||||
|
}
|
||||||
|
|
28
src/meshname/domain.go
Normal file
28
src/meshname/domain.go
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
package meshname
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/base32"
|
||||||
|
"errors"
|
||||||
|
"net"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func DomainFromIP(target *net.IP) string {
|
||||||
|
return strings.ToLower(base32.StdEncoding.EncodeToString(*target)[0:26])
|
||||||
|
}
|
||||||
|
|
||||||
|
func IPFromDomain(domain *string) (net.IP, error) {
|
||||||
|
name := strings.ToUpper(*domain) + "======"
|
||||||
|
data, err := base32.StdEncoding.DecodeString(name)
|
||||||
|
if err != nil {
|
||||||
|
return net.IP{}, err
|
||||||
|
}
|
||||||
|
if len(data) != 16 {
|
||||||
|
return net.IP{}, errors.New("Invalid subdomain")
|
||||||
|
}
|
||||||
|
ipAddr := net.IP(data)
|
||||||
|
if ipAddr == nil {
|
||||||
|
return net.IP{}, errors.New("Invalid IP address")
|
||||||
|
}
|
||||||
|
return ipAddr, nil
|
||||||
|
}
|
@ -1,11 +1,7 @@
|
|||||||
package meshname
|
package meshname
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/base32"
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
@ -14,66 +10,6 @@ import (
|
|||||||
"github.com/miekg/dns"
|
"github.com/miekg/dns"
|
||||||
)
|
)
|
||||||
|
|
||||||
func DomainFromIP(target *net.IP) string {
|
|
||||||
return strings.ToLower(base32.StdEncoding.EncodeToString(*target)[0:26])
|
|
||||||
}
|
|
||||||
|
|
||||||
func IPFromDomain(domain *string) (net.IP, error) {
|
|
||||||
name := strings.ToUpper(*domain) + "======"
|
|
||||||
data, err := base32.StdEncoding.DecodeString(name)
|
|
||||||
if err != nil {
|
|
||||||
return net.IP{}, err
|
|
||||||
}
|
|
||||||
if len(data) != 16 {
|
|
||||||
return net.IP{}, errors.New("Invalid subdomain")
|
|
||||||
}
|
|
||||||
ipAddr := net.IP(data)
|
|
||||||
if ipAddr == nil {
|
|
||||||
return net.IP{}, errors.New("Invalid IP address")
|
|
||||||
}
|
|
||||||
return ipAddr, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenConf(target, zone string) (string, error) {
|
|
||||||
ip := net.ParseIP(target)
|
|
||||||
if ip == nil {
|
|
||||||
return "", errors.New("Invalid IP address")
|
|
||||||
}
|
|
||||||
subDomain := DomainFromIP(&ip)
|
|
||||||
selfRecord := fmt.Sprintf("\t\t\"%s.%s AAAA %s\"\n", subDomain, zone, target)
|
|
||||||
confString := fmt.Sprintf("{\n\t\"%s\":[\n%s\t]\n}", subDomain, selfRecord)
|
|
||||||
|
|
||||||
return confString, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Load zoneConfig from a JSON file
|
|
||||||
func ParseConfigFile(configPath string) (map[string][]dns.RR, error) {
|
|
||||||
conf, err := ioutil.ReadFile(configPath)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
var dat map[string][]string
|
|
||||||
if err := json.Unmarshal(conf, &dat); err == nil {
|
|
||||||
return ParseZoneConfigMap(dat)
|
|
||||||
} else {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func ParseZoneConfigMap(zoneConfigMap map[string][]string) (map[string][]dns.RR, error) {
|
|
||||||
var zoneConfig = make(map[string][]dns.RR)
|
|
||||||
for subDomain, records := range zoneConfigMap {
|
|
||||||
for _, r := range records {
|
|
||||||
if rr, err := dns.NewRR(r); err == nil {
|
|
||||||
zoneConfig[subDomain] = append(zoneConfig[subDomain], rr)
|
|
||||||
} else {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return zoneConfig, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type MeshnameServer struct {
|
type MeshnameServer struct {
|
||||||
log *log.Logger
|
log *log.Logger
|
||||||
listenAddr string
|
listenAddr string
|
||||||
|
Loading…
Reference in New Issue
Block a user