package headscale import ( "encoding/json" "errors" "log" "gorm.io/datatypes" "inet.af/netaddr" ) // GetNodeRoutes returns the subnet routes advertised by a node (identified by // namespace and node name) func (h *Headscale) GetNodeRoutes(namespace string, nodeName string) (*[]netaddr.IPPrefix, error) { m, err := h.GetMachine(namespace, nodeName) if err != nil { return nil, err } hi, err := m.GetHostInfo() if err != nil { return nil, err } return &hi.RoutableIPs, nil } // EnableNodeRoute enables a subnet route advertised by a node (identified by // namespace and node name) func (h *Headscale) EnableNodeRoute(namespace string, nodeName string, routeStr string) (*netaddr.IPPrefix, error) { m, err := h.GetMachine(namespace, nodeName) if err != nil { return nil, err } hi, err := m.GetHostInfo() if err != nil { return nil, err } route, err := netaddr.ParseIPPrefix(routeStr) if err != nil { return nil, err } for _, rIP := range hi.RoutableIPs { if rIP == route { db, err := h.db() if err != nil { log.Printf("Cannot open DB: %s", err) return nil, err } routes, _ := json.Marshal([]string{routeStr}) // TODO: only one for the time being, so overwriting the rest m.EnabledRoutes = datatypes.JSON(routes) db.Save(&m) db.Close() // THIS IS COMPLETELY USELESS. // The peers map is stored in memory in the server process. // Definetely not accessible from the CLI tool. // We need RPC to the server - or some kind of 'needsUpdate' field in the DB peers, _ := h.getPeers(*m) h.pollMu.Lock() for _, p := range *peers { if pUp, ok := h.clientsPolling[uint64(p.ID)]; ok { pUp <- []byte{} } } h.pollMu.Unlock() return &rIP, nil } } return nil, errors.New("could not find routable range") }