Minor code reorg
This commit is contained in:
Juan Font Alonso
parent
977962736a
commit
541d676b98
2 changed files with 121 additions and 114 deletions
114
app.go
114
app.go
|
|
@ -1,15 +1,11 @@
|
||||||
package headscale
|
package headscale
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
|
||||||
"os"
|
"os"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/jinzhu/gorm/dialects/postgres"
|
|
||||||
"inet.af/netaddr"
|
|
||||||
"tailscale.com/tailcfg"
|
"tailscale.com/tailcfg"
|
||||||
"tailscale.com/wgengine/wgcfg"
|
"tailscale.com/wgengine/wgcfg"
|
||||||
)
|
)
|
||||||
|
|
@ -75,113 +71,3 @@ func (h *Headscale) Serve() error {
|
||||||
err := r.Run(h.cfg.Addr)
|
err := r.Run(h.cfg.Addr)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// RegisterMachine is executed from the CLI to register a new Machine using its MachineKey
|
|
||||||
func (h *Headscale) RegisterMachine(key string, namespace string) error {
|
|
||||||
ns, err := h.GetNamespace(namespace)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
mKey, err := wgcfg.ParseHexKey(key)
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Cannot parse client key: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
db, err := h.db()
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Cannot open DB: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer db.Close()
|
|
||||||
m := Machine{}
|
|
||||||
if db.First(&m, "machine_key = ?", mKey.HexString()).RecordNotFound() {
|
|
||||||
log.Printf("Cannot find machine with machine key: %s", mKey.Base64())
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if m.isAlreadyRegistered() {
|
|
||||||
fmt.Println("This machine already registered")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
ip, err := h.getAvailableIP()
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
m.IPAddress = ip.String()
|
|
||||||
m.NamespaceID = ns.ID
|
|
||||||
m.Registered = true
|
|
||||||
db.Save(&m)
|
|
||||||
fmt.Println("Machine registered 🎉")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *Headscale) ListNodeRoutes(namespace string, nodeName string) error {
|
|
||||||
m, err := h.GetMachine(namespace, nodeName)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
hi, err := m.GetHostInfo()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
fmt.Println(hi.RoutableIPs)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *Headscale) EnableNodeRoute(namespace string, nodeName string, routeStr string) error {
|
|
||||||
m, err := h.GetMachine(namespace, nodeName)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
hi, err := m.GetHostInfo()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
route, err := netaddr.ParseIPPrefix(routeStr)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, rIP := range hi.RoutableIPs {
|
|
||||||
if rIP == route {
|
|
||||||
db, err := h.db()
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Cannot open DB: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer db.Close()
|
|
||||||
routes, _ := json.Marshal([]string{routeStr}) // TODO: only one for the time being, so overwriting the rest
|
|
||||||
m.EnabledRoutes = postgres.Jsonb{RawMessage: json.RawMessage(routes)}
|
|
||||||
db.Save(&m)
|
|
||||||
db.Close()
|
|
||||||
|
|
||||||
peers, _ := h.getPeers(*m)
|
|
||||||
h.pollMu.Lock()
|
|
||||||
for _, p := range *peers {
|
|
||||||
if pUp, ok := h.clientsPolling[uint64(p.ID)]; ok {
|
|
||||||
pUp <- []byte{}
|
|
||||||
} else {
|
|
||||||
}
|
|
||||||
}
|
|
||||||
h.pollMu.Unlock()
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return fmt.Errorf("Could not find routable range")
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func eqCIDRs(a, b []netaddr.IPPrefix) bool {
|
|
||||||
if len(a) != len(b) || ((a == nil) != (b == nil)) {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
for i, v := range a {
|
|
||||||
if v != b[i] {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
|
||||||
121
cli.go
Normal file
121
cli.go
Normal file
|
|
@ -0,0 +1,121 @@
|
||||||
|
package headscale
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
|
||||||
|
"github.com/jinzhu/gorm/dialects/postgres"
|
||||||
|
"inet.af/netaddr"
|
||||||
|
"tailscale.com/wgengine/wgcfg"
|
||||||
|
)
|
||||||
|
|
||||||
|
// RegisterMachine is executed from the CLI to register a new Machine using its MachineKey
|
||||||
|
func (h *Headscale) RegisterMachine(key string, namespace string) error {
|
||||||
|
ns, err := h.GetNamespace(namespace)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
mKey, err := wgcfg.ParseHexKey(key)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Cannot parse client key: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
db, err := h.db()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Cannot open DB: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
m := Machine{}
|
||||||
|
if db.First(&m, "machine_key = ?", mKey.HexString()).RecordNotFound() {
|
||||||
|
log.Printf("Cannot find machine with machine key: %s", mKey.Base64())
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if m.isAlreadyRegistered() {
|
||||||
|
fmt.Println("This machine already registered")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
ip, err := h.getAvailableIP()
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
m.IPAddress = ip.String()
|
||||||
|
m.NamespaceID = ns.ID
|
||||||
|
m.Registered = true
|
||||||
|
db.Save(&m)
|
||||||
|
fmt.Println("Machine registered 🎉")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Headscale) ListNodeRoutes(namespace string, nodeName string) error {
|
||||||
|
m, err := h.GetMachine(namespace, nodeName)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
hi, err := m.GetHostInfo()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
fmt.Println(hi.RoutableIPs)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Headscale) EnableNodeRoute(namespace string, nodeName string, routeStr string) error {
|
||||||
|
m, err := h.GetMachine(namespace, nodeName)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
hi, err := m.GetHostInfo()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
route, err := netaddr.ParseIPPrefix(routeStr)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, rIP := range hi.RoutableIPs {
|
||||||
|
if rIP == route {
|
||||||
|
db, err := h.db()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Cannot open DB: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
routes, _ := json.Marshal([]string{routeStr}) // TODO: only one for the time being, so overwriting the rest
|
||||||
|
m.EnabledRoutes = postgres.Jsonb{RawMessage: json.RawMessage(routes)}
|
||||||
|
db.Save(&m)
|
||||||
|
db.Close()
|
||||||
|
|
||||||
|
peers, _ := h.getPeers(*m)
|
||||||
|
h.pollMu.Lock()
|
||||||
|
for _, p := range *peers {
|
||||||
|
if pUp, ok := h.clientsPolling[uint64(p.ID)]; ok {
|
||||||
|
pUp <- []byte{}
|
||||||
|
} else {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
h.pollMu.Unlock()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return fmt.Errorf("Could not find routable range")
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func eqCIDRs(a, b []netaddr.IPPrefix) bool {
|
||||||
|
if len(a) != len(b) || ((a == nil) != (b == nil)) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for i, v := range a {
|
||||||
|
if v != b[i] {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue