package headscale import ( "encoding/json" "fmt" "log" "time" "github.com/jinzhu/gorm/dialects/postgres" "github.com/tailscale/wireguard-go/wgcfg" "tailscale.com/tailcfg" ) type Machine struct { ID uint64 `gorm:"primary_key"` MachineKey string `gorm:"type:varchar(64);unique_index"` NodeKey string 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 } addrs := []wgcfg.CIDR{} allowedIPs := []wgcfg.CIDR{} ip, err := wgcfg.ParseCIDR(fmt.Sprintf("%s/32", m.IPAddress)) 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{ ID: tailcfg.NodeID(m.ID), // this is the actual ID Name: "", User: 1, Key: tailcfg.NodeKey(nKey), KeyExpiry: *m.Expiry, Machine: tailcfg.MachineKey(mKey), 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 <> ?", m.MachineKey).Find(&machines).Error; err != nil { 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) } return &peers, nil }