mirror of
https://github.com/MirageNetwork/MirageServer.git
synced 2025-09-27 04:45:54 +08:00
72 lines
2.1 KiB
Go
72 lines
2.1 KiB
Go
package controller
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"io"
|
||
"net/http"
|
||
|
||
"github.com/rs/zerolog/log"
|
||
"gorm.io/gorm"
|
||
"tailscale.com/tailcfg"
|
||
)
|
||
|
||
// NoisePollNetMapHandler takes care of /machine/:id/map using the Noise protocol
|
||
//
|
||
// This is the busiest endpoint, as it keeps the HTTP long poll that updates
|
||
// the clients when something in the network changes.
|
||
//
|
||
// The clients POST stuff like HostInfo and their Endpoints here, but
|
||
// only after their first request (marked with the ReadOnly field).
|
||
//
|
||
// At this moment the updates are sent in a quite horrendous way, but they kinda work.
|
||
func (t *noiseServer) NoisePollNetMapHandler(
|
||
writer http.ResponseWriter,
|
||
req *http.Request,
|
||
) {
|
||
log.Trace().
|
||
Str("handler", "NoisePollNetMap").
|
||
Msg("PollNetMapHandler called")
|
||
body, _ := io.ReadAll(req.Body)
|
||
|
||
mapRequest := tailcfg.MapRequest{}
|
||
if err := json.Unmarshal(body, &mapRequest); err != nil {
|
||
log.Error().
|
||
Caller().
|
||
Err(err).
|
||
Msg("Cannot parse MapRequest")
|
||
http.Error(writer, "Internal error", http.StatusInternalServerError)
|
||
|
||
return
|
||
}
|
||
|
||
t.nodeKey = mapRequest.NodeKey
|
||
|
||
//machine, err := t.headscale.GetMachineByAnyKey(t.conn.Peer(), mapRequest.NodeKey, key.NodePublic{})
|
||
//cgao6: 因为MachineKey会用于多个用户，不具备unique特点，故此处应该只判断NodeKey！
|
||
machine, err := t.mirage.GetMachineByNodeKey(mapRequest.NodeKey)
|
||
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
log.Debug().
|
||
Str("handler", "NoisePollNetMap").
|
||
Msgf("Ignoring request, cannot find machine with key %s", mapRequest.NodeKey.String())
|
||
http.Error(writer, "Internal error", http.StatusNotFound)
|
||
|
||
return
|
||
}
|
||
log.Error().
|
||
Str("handler", "NoisePollNetMap").
|
||
Msgf("Failed to fetch machine from the database with node key: %s", mapRequest.NodeKey.String())
|
||
http.Error(writer, "Internal error", http.StatusInternalServerError)
|
||
|
||
return
|
||
}
|
||
log.Debug().
|
||
Str("handler", "NoisePollNetMap").
|
||
Str("machine", machine.Hostname).
|
||
Msg("A machine is entering polling via the Noise protocol")
|
||
|
||
t.mirage.handlePollCommon(writer, req.Context(), machine, mapRequest)
|
||
}
|