headscale/machine.go

137 lines
3 KiB
Go
Raw Normal View History

2020-06-21 04:32:08 -06:00
package headscale
import (
"encoding/json"
"fmt"
"log"
"sort"
2021-02-22 15:27:33 -07:00
"strconv"
2020-06-21 04:32:08 -06:00
"time"
"github.com/jinzhu/gorm/dialects/postgres"
2021-02-20 14:43:07 -07:00
"inet.af/netaddr"
2020-06-21 04:32:08 -06:00
"tailscale.com/tailcfg"
2021-02-20 14:43:07 -07:00
"tailscale.com/wgengine/wgcfg"
2020-06-21 04:32:08 -06:00
)
type Machine struct {
ID uint64 `gorm:"primary_key"`
MachineKey string `gorm:"type:varchar(64);unique_index"`
NodeKey string
DiscoKey string
2020-06-21 04:32:08 -06:00
IPAddress string
Registered bool // temp
LastSeen *time.Time
Expiry *time.Time
HostInfo postgres.Jsonb
Endpoints postgres.Jsonb
CreatedAt time.Time
UpdatedAt time.Time
DeletedAt *time.Time
}
// For the time being this method is rather naive
func (m Machine) isAlreadyRegistered() bool {
return m.Registered
}
func (m Machine) toNode() (*tailcfg.Node, error) {
nKey, err := wgcfg.ParseHexKey(m.NodeKey)
if err != nil {
return nil, err
}
mKey, err := wgcfg.ParseHexKey(m.MachineKey)
if err != nil {
return nil, err
}
dKey, err := wgcfg.ParseHexKey(m.DiscoKey)
if err != nil {
return nil, err
}
2021-02-20 14:43:07 -07:00
addrs := []netaddr.IPPrefix{}
allowedIPs := []netaddr.IPPrefix{}
2020-06-21 04:32:08 -06:00
2021-02-20 14:43:07 -07:00
ip, err := netaddr.ParseIPPrefix(fmt.Sprintf("%s/32", m.IPAddress))
2020-06-21 04:32:08 -06:00
if err != nil {
return nil, err
}
addrs = append(addrs, ip)
allowedIPs = append(allowedIPs, ip) // looks like the client expect this
endpoints := []string{}
be, err := m.Endpoints.MarshalJSON()
if err != nil {
return nil, err
}
err = json.Unmarshal(be, &endpoints)
if err != nil {
return nil, err
}
hostinfo := tailcfg.Hostinfo{}
hi, err := m.HostInfo.MarshalJSON()
if err != nil {
return nil, err
}
err = json.Unmarshal(hi, &hostinfo)
if err != nil {
return nil, err
}
n := tailcfg.Node{
2021-02-22 15:27:33 -07:00
ID: tailcfg.NodeID(m.ID), // this is the actual ID
StableID: tailcfg.StableNodeID(strconv.FormatUint(m.ID, 10)), // in headscale, unlike tailcontrol server, IDs are permantent
2020-06-21 04:32:08 -06:00
Name: "",
User: 1,
Key: tailcfg.NodeKey(nKey),
KeyExpiry: *m.Expiry,
Machine: tailcfg.MachineKey(mKey),
DiscoKey: tailcfg.DiscoKey(dKey),
2020-06-21 04:32:08 -06:00
Addresses: addrs,
AllowedIPs: allowedIPs,
Endpoints: endpoints,
// DERP: "127.3.3.40:4", // wtf?
Hostinfo: hostinfo,
Created: m.CreatedAt,
LastSeen: m.LastSeen,
KeepAlive: false,
MachineAuthorized: m.Registered,
}
// n.Key.MarshalText()
return &n, nil
}
func (h *Headscale) getPeers(m Machine) (*[]*tailcfg.Node, error) {
db, err := h.db()
if err != nil {
log.Printf("Cannot open DB: %s", err)
return nil, err
}
defer db.Close()
// Add user management here?
machines := []Machine{}
if err = db.Where("machine_key <> ? AND registered", m.MachineKey).Find(&machines).Error; err != nil {
2020-06-21 04:32:08 -06:00
log.Printf("Error accessing db: %s", err)
return nil, err
}
log.Printf("Found %d peers of %s", len(machines), m.MachineKey)
peers := []*tailcfg.Node{}
for _, mn := range machines {
peer, err := mn.toNode()
if err != nil {
return nil, err
}
peers = append(peers, peer)
}
sort.Slice(peers, func(i, j int) bool { return peers[i].ID < peers[j].ID })
2020-06-21 04:32:08 -06:00
return &peers, nil
}