mirror of
https://github.com/juanfont/headscale.git
synced 2024-11-10 09:02:30 +08:00
593040b73d
In TS2021 the MachineKey can be obtained from noiseConn.Peer() - contrary to what I thought before, where I assumed MachineKey was dropped in TS2021. By having a ts2021App and hanging from there the TS2021 handlers, we can fetch again the MachineKey.
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
//go:build ts2019
|
|
|
|
package headscale
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/rs/zerolog/log"
|
|
"tailscale.com/tailcfg"
|
|
"tailscale.com/types/key"
|
|
)
|
|
|
|
// RegistrationHandler handles the actual registration process of a machine
|
|
// Endpoint /machine/:mkey.
|
|
func (h *Headscale) RegistrationHandler(
|
|
writer http.ResponseWriter,
|
|
req *http.Request,
|
|
) {
|
|
vars := mux.Vars(req)
|
|
machineKeyStr, ok := vars["mkey"]
|
|
if !ok || machineKeyStr == "" {
|
|
log.Error().
|
|
Str("handler", "RegistrationHandler").
|
|
Msg("No machine ID in request")
|
|
http.Error(writer, "No machine ID in request", http.StatusBadRequest)
|
|
|
|
return
|
|
}
|
|
|
|
body, _ := io.ReadAll(req.Body)
|
|
|
|
var machineKey key.MachinePublic
|
|
err := machineKey.UnmarshalText([]byte(MachinePublicKeyEnsurePrefix(machineKeyStr)))
|
|
if err != nil {
|
|
log.Error().
|
|
Caller().
|
|
Err(err).
|
|
Msg("Cannot parse machine key")
|
|
machineRegistrations.WithLabelValues("unknown", "web", "error", "unknown").Inc()
|
|
http.Error(writer, "Cannot parse machine key", http.StatusBadRequest)
|
|
|
|
return
|
|
}
|
|
registerRequest := tailcfg.RegisterRequest{}
|
|
err = decode(body, ®isterRequest, &machineKey, h.privateKey)
|
|
if err != nil {
|
|
log.Error().
|
|
Caller().
|
|
Err(err).
|
|
Msg("Cannot decode message")
|
|
machineRegistrations.WithLabelValues("unknown", "web", "error", "unknown").Inc()
|
|
http.Error(writer, "Cannot decode message", http.StatusBadRequest)
|
|
|
|
return
|
|
}
|
|
|
|
h.handleRegisterCommon(writer, req, registerRequest, machineKey, false)
|
|
}
|