12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160 |
- package controller
- import (
- "encoding/json"
- "errors"
- "fmt"
- "net/http"
- "time"
- "github.com/google/uuid"
- "github.com/gorilla/mux"
- "github.com/gravitl/netmaker/database"
- "github.com/gravitl/netmaker/logger"
- "github.com/gravitl/netmaker/logic"
- "github.com/gravitl/netmaker/models"
- "github.com/gravitl/netmaker/mq"
- "github.com/gravitl/netmaker/servercfg"
- "golang.org/x/crypto/bcrypt"
- "golang.org/x/exp/slog"
- )
- func hostHandlers(r *mux.Router) {
- r.HandleFunc("/api/hosts", logic.SecurityCheck(true, http.HandlerFunc(getHosts))).
- Methods(http.MethodGet)
- r.HandleFunc("/api/hosts/keys", logic.SecurityCheck(true, http.HandlerFunc(updateAllKeys))).
- Methods(http.MethodPut)
- r.HandleFunc("/api/hosts/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHosts))).
- Methods(http.MethodPost)
- r.HandleFunc("/api/hosts/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHosts))).
- Methods(http.MethodPost)
- r.HandleFunc("/api/hosts/{hostid}/keys", logic.SecurityCheck(true, http.HandlerFunc(updateKeys))).
- Methods(http.MethodPut)
- r.HandleFunc("/api/hosts/{hostid}/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHost))).
- Methods(http.MethodPost)
- r.HandleFunc("/api/hosts/{hostid}", logic.SecurityCheck(true, http.HandlerFunc(updateHost))).
- Methods(http.MethodPut)
- r.HandleFunc("/api/hosts/{hostid}", Authorize(true, false, "all", http.HandlerFunc(deleteHost))).
- Methods(http.MethodDelete)
- r.HandleFunc("/api/hosts/{hostid}/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHost))).
- Methods(http.MethodPut)
- r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(addHostToNetwork))).
- Methods(http.MethodPost)
- r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(deleteHostFromNetwork))).
- Methods(http.MethodDelete)
- r.HandleFunc("/api/hosts/adm/authenticate", authenticateHost).Methods(http.MethodPost)
- r.HandleFunc("/api/v1/host", Authorize(true, false, "host", http.HandlerFunc(pull))).
- Methods(http.MethodGet)
- r.HandleFunc("/api/v1/host/{hostid}/signalpeer", Authorize(true, false, "host", http.HandlerFunc(signalPeer))).
- Methods(http.MethodPost)
- r.HandleFunc("/api/v1/fallback/host/{hostid}", Authorize(true, false, "host", http.HandlerFunc(hostUpdateFallback))).
- Methods(http.MethodPut)
- r.HandleFunc("/api/v1/host/{hostid}/peer_info", Authorize(true, false, "host", http.HandlerFunc(getHostPeerInfo))).
- Methods(http.MethodGet)
- r.HandleFunc("/api/emqx/hosts", logic.SecurityCheck(true, http.HandlerFunc(delEmqxHosts))).
- Methods(http.MethodDelete)
- r.HandleFunc("/api/v1/auth-register/host", socketHandler)
- }
- // @Summary Requests all the hosts to upgrade their version
- // @Router /api/hosts/upgrade [post]
- // @Tags Hosts
- // @Security oauth
- // @Param force query bool false "Force upgrade"
- // @Success 200 {string} string "upgrade all hosts request received"
- func upgradeHosts(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- action := models.Upgrade
- if r.URL.Query().Get("force") == "true" {
- action = models.ForceUpgrade
- }
- user := r.Header.Get("user")
- go func() {
- slog.Info("requesting all hosts to upgrade", "user", user)
- hosts, err := logic.GetAllHosts()
- if err != nil {
- slog.Error("failed to retrieve all hosts", "user", user, "error", err)
- return
- }
- for _, host := range hosts {
- go func(host models.Host) {
- hostUpdate := models.HostUpdate{
- Action: action,
- Host: host,
- }
- if err = mq.HostUpdate(&hostUpdate); err != nil {
- slog.Error("failed to request host to upgrade", "user", user, "host", host.ID.String(), "error", err)
- } else {
- slog.Info("host upgrade requested", "user", user, "host", host.ID.String())
- }
- }(host)
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.UpgradeAll,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: "All Hosts",
- Name: "All Hosts",
- Type: models.DeviceSub,
- },
- Origin: models.Dashboard,
- })
- slog.Info("upgrade all hosts request received", "user", user)
- logic.ReturnSuccessResponse(w, r, "upgrade all hosts request received")
- }
- // @Summary Upgrade a host
- // @Router /api/hosts/{hostid}/upgrade [put]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param force query bool false "Force upgrade"
- // @Success 200 {string} string "passed message to upgrade host"
- // @Failure 500 {object} models.ErrorResponse
- // upgrade host is a handler to send upgrade message to a host
- func upgradeHost(w http.ResponseWriter, r *http.Request) {
- host, err := logic.GetHost(mux.Vars(r)["hostid"])
- if err != nil {
- slog.Error("failed to find host", "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
- return
- }
- action := models.Upgrade
- if r.URL.Query().Get("force") == "true" {
- action = models.ForceUpgrade
- }
- if err := mq.HostUpdate(&models.HostUpdate{Action: action, Host: *host}); err != nil {
- slog.Error("failed to upgrade host", "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.ReturnSuccessResponse(w, r, "passed message to upgrade host")
- }
- // @Summary List all hosts
- // @Router /api/hosts [get]
- // @Tags Hosts
- // @Security oauth
- // @Success 200 {array} models.ApiHost
- // @Failure 500 {object} models.ErrorResponse
- func getHosts(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- currentHosts, err := logic.GetAllHosts()
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- apiHosts := logic.GetAllHostsAPI(currentHosts[:])
- logger.Log(2, r.Header.Get("user"), "fetched all hosts")
- logic.SortApiHosts(apiHosts[:])
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(apiHosts)
- }
- // @Summary Used by clients for "pull" command
- // @Router /api/v1/host [get]
- // @Tags Hosts
- // @Security oauth
- // @Success 200 {object} models.HostPull
- // @Failure 500 {object} models.ErrorResponse
- func pull(w http.ResponseWriter, r *http.Request) {
- hostID := r.Header.Get(hostIDHeader) // return JSON/API formatted keys
- if len(hostID) == 0 {
- logger.Log(0, "no host authorized to pull")
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(fmt.Errorf("no host authorized to pull"), "internal"),
- )
- return
- }
- host, err := logic.GetHost(hostID)
- if err != nil {
- logger.Log(0, "no host found during pull", hostID)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- sendPeerUpdate := false
- for _, nodeID := range host.Nodes {
- node, err := logic.GetNodeByID(nodeID)
- if err != nil {
- slog.Error("failed to get node:", "id", node.ID, "error", err)
- continue
- }
- if node.FailedOverBy != uuid.Nil && r.URL.Query().Get("reset_failovered") == "true" {
- logic.ResetFailedOverPeer(&node)
- sendPeerUpdate = true
- }
- }
- if sendPeerUpdate {
- if err := mq.PublishPeerUpdate(false); err != nil {
- logger.Log(0, "fail to publish peer update: ", err.Error())
- }
- }
- allNodes, err := logic.GetAllNodes()
- if err != nil {
- logger.Log(0, "failed to get nodes: ", hostID)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- hPU, err := logic.GetPeerUpdateForHost("", host, allNodes, nil, nil)
- if err != nil {
- logger.Log(0, "could not pull peers for host", hostID, err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- serverConf := logic.GetServerInfo()
- key, keyErr := logic.RetrievePublicTrafficKey()
- if keyErr != nil {
- logger.Log(0, "error retrieving key:", keyErr.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(keyErr, "internal"))
- return
- }
- _ = logic.CheckHostPorts(host)
- serverConf.TrafficKey = key
- response := models.HostPull{
- Host: *host,
- Nodes: logic.GetHostNodes(host),
- ServerConfig: serverConf,
- Peers: hPU.Peers,
- PeerIDs: hPU.PeerIDs,
- HostNetworkInfo: hPU.HostNetworkInfo,
- EgressRoutes: hPU.EgressRoutes,
- FwUpdate: hPU.FwUpdate,
- ChangeDefaultGw: hPU.ChangeDefaultGw,
- DefaultGwIp: hPU.DefaultGwIp,
- IsInternetGw: hPU.IsInternetGw,
- NameServers: hPU.NameServers,
- EgressWithDomains: hPU.EgressWithDomains,
- EndpointDetection: logic.IsEndpointDetectionEnabled(),
- }
- logger.Log(1, hostID, "completed a pull")
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(&response)
- }
- // @Summary Updates a Netclient host on Netmaker server
- // @Router /api/hosts/{hostid} [put]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param body body models.ApiHost true "New host data"
- // @Success 200 {object} models.ApiHost
- // @Failure 500 {object} models.ErrorResponse
- func updateHost(w http.ResponseWriter, r *http.Request) {
- var newHostData models.ApiHost
- err := json.NewDecoder(r.Body).Decode(&newHostData)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- // confirm host exists
- currHost, err := logic.GetHost(newHostData.ID)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- newHost := newHostData.ConvertAPIHostToNMHost(currHost)
- logic.UpdateHost(newHost, currHost) // update the in memory struct values
- if err = logic.UpsertHost(newHost); err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- // publish host update through MQ
- if err := mq.HostUpdate(&models.HostUpdate{
- Action: models.UpdateHost,
- Host: *newHost,
- }); err != nil {
- logger.Log(
- 0,
- r.Header.Get("user"),
- "failed to send host update: ",
- currHost.ID.String(),
- err.Error(),
- )
- }
- go func() {
- if err := mq.PublishPeerUpdate(false); err != nil {
- logger.Log(0, "fail to publish peer update: ", err.Error())
- }
- if newHost.Name != currHost.Name {
- if servercfg.IsDNSMode() {
- logic.SetDNS()
- }
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.Update,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: currHost.ID.String(),
- Name: newHost.Name,
- Type: models.DeviceSub,
- },
- Diff: models.Diff{
- Old: currHost,
- New: newHost,
- },
- Origin: models.Dashboard,
- })
- apiHostData := newHost.ConvertNMHostToAPI()
- logger.Log(2, r.Header.Get("user"), "updated host", newHost.ID.String())
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(apiHostData)
- }
- // @Summary Updates a Netclient host on Netmaker server
- // @Router /api/v1/fallback/host/{hostid} [put]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param body body models.HostUpdate true "Host update data"
- // @Success 200 {string} string "updated host data"
- // @Failure 500 {object} models.ErrorResponse
- func hostUpdateFallback(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- hostid := params["hostid"]
- currentHost, err := logic.GetHost(hostid)
- if err != nil {
- slog.Error("error getting host", "id", hostid, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- var sendPeerUpdate bool
- var replacePeers bool
- var hostUpdate models.HostUpdate
- err = json.NewDecoder(r.Body).Decode(&hostUpdate)
- if err != nil {
- slog.Error("failed to update a host:", "user", r.Header.Get("user"), "error", err.Error(), "host", currentHost.Name)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- slog.Info("recieved host update", "name", hostUpdate.Host.Name, "id", hostUpdate.Host.ID, "action", hostUpdate.Action)
- switch hostUpdate.Action {
- case models.CheckIn:
- sendPeerUpdate = mq.HandleHostCheckin(&hostUpdate.Host, currentHost)
- case models.UpdateHost:
- if hostUpdate.Host.PublicKey != currentHost.PublicKey {
- //remove old peer entry
- replacePeers = true
- }
- sendPeerUpdate = logic.UpdateHostFromClient(&hostUpdate.Host, currentHost)
- err := logic.UpsertHost(currentHost)
- if err != nil {
- slog.Error("failed to update host", "id", currentHost.ID, "error", err)
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.Internal))
- return
- }
- case models.UpdateMetrics:
- mq.UpdateMetricsFallBack(hostUpdate.Node.ID.String(), hostUpdate.NewMetrics)
- case models.EgressUpdate:
- e := schema.Egress{ID: hostUpdate.EgressDomain.ID}
- err = e.Get(db.WithContext(r.Context()))
- if err != nil {
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
- return
- }
- if len(hostUpdate.Node.EgressGatewayRanges) > 0 {
- e.DomainAns = hostUpdate.Node.EgressGatewayRanges
- e.Update(db.WithContext(r.Context()))
- }
- sendPeerUpdate = true
- }
- if sendPeerUpdate {
- err := mq.PublishPeerUpdate(replacePeers)
- if err != nil {
- slog.Error("failed to publish peer update", "error", err)
- }
- }
- logic.ReturnSuccessResponse(w, r, "updated host data")
- }
- // @Summary Deletes a Netclient host from Netmaker server
- // @Router /api/hosts/{hostid} [delete]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param force query bool false "Force delete"
- // @Success 200 {object} models.ApiHost
- // @Failure 500 {object} models.ErrorResponse
- func deleteHost(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- hostid := params["hostid"]
- forceDelete := r.URL.Query().Get("force") == "true"
- // confirm host exists
- currHost, err := logic.GetHost(hostid)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- for _, nodeID := range currHost.Nodes {
- node, err := logic.GetNodeByID(nodeID)
- if err != nil {
- slog.Error("failed to get node", "nodeid", nodeID, "error", err)
- continue
- }
- var gwClients []models.ExtClient
- if node.IsIngressGateway {
- gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
- }
- go mq.PublishMqUpdatesForDeletedNode(node, false, gwClients)
- }
- if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
- // delete EMQX credentials for host
- if err := mq.GetEmqxHandler().DeleteEmqxUser(currHost.ID.String()); err != nil {
- slog.Error(
- "failed to remove host credentials from EMQX",
- "id",
- currHost.ID,
- "error",
- err,
- )
- }
- }
- if err = mq.HostUpdate(&models.HostUpdate{
- Action: models.DeleteHost,
- Host: *currHost,
- }); err != nil {
- logger.Log(
- 0,
- r.Header.Get("user"),
- "failed to send delete host update: ",
- currHost.ID.String(),
- err.Error(),
- )
- }
- if err = logic.RemoveHost(currHost, forceDelete); err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logic.LogEvent(&models.Event{
- Action: models.Delete,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: currHost.ID.String(),
- Name: currHost.Name,
- Type: models.DeviceSub,
- },
- Origin: models.Dashboard,
- })
- apiHostData := currHost.ConvertNMHostToAPI()
- logger.Log(2, r.Header.Get("user"), "removed host", currHost.Name)
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(apiHostData)
- }
- // @Summary To Add Host To Network
- // @Router /api/hosts/{hostid}/networks/{network} [post]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param network path string true "Network name"
- // @Success 200 {string} string "OK"
- // @Failure 500 {object} models.ErrorResponse
- func addHostToNetwork(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- hostid := params["hostid"]
- network := params["network"]
- if hostid == "" || network == "" {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
- )
- return
- }
- // confirm host exists
- currHost, err := logic.GetHost(hostid)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to find host:", hostid, err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- newNode, err := logic.UpdateHostNetwork(currHost, network, true)
- if err != nil {
- logger.Log(
- 0,
- r.Header.Get("user"),
- "failed to add host to network:",
- hostid,
- network,
- err.Error(),
- )
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- logger.Log(1, "added new node", newNode.ID.String(), "to host", currHost.Name)
- if currHost.IsDefault {
- // make host failover
- logic.CreateFailOver(*newNode)
- // make host remote access gateway
- logic.CreateIngressGateway(network, newNode.ID.String(), models.IngressRequest{})
- logic.CreateRelay(models.RelayRequest{
- NodeID: newNode.ID.String(),
- NetID: network,
- })
- }
- go func() {
- mq.HostUpdate(&models.HostUpdate{
- Action: models.JoinHostToNetwork,
- Host: *currHost,
- Node: *newNode,
- })
- mq.PublishPeerUpdate(false)
- if servercfg.IsDNSMode() {
- logic.SetDNS()
- }
- }()
- logger.Log(
- 2,
- r.Header.Get("user"),
- fmt.Sprintf("added host %s to network %s", currHost.Name, network),
- )
- logic.LogEvent(&models.Event{
- Action: models.JoinHostToNet,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: currHost.ID.String(),
- Name: currHost.Name,
- Type: models.DeviceSub,
- },
- NetworkID: models.NetworkID(network),
- Origin: models.Dashboard,
- })
- w.WriteHeader(http.StatusOK)
- }
- // @Summary To Remove Host from Network
- // @Router /api/hosts/{hostid}/networks/{network} [delete]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param network path string true "Network name"
- // @Param force query bool false "Force delete"
- // @Success 200 {string} string "OK"
- // @Failure 500 {object} models.ErrorResponse
- func deleteHostFromNetwork(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- hostid := params["hostid"]
- network := params["network"]
- forceDelete := r.URL.Query().Get("force") == "true"
- if hostid == "" || network == "" {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
- )
- return
- }
- // confirm host exists
- currHost, err := logic.GetHost(hostid)
- if err != nil {
- if database.IsEmptyRecord(err) {
- // check if there is any daemon nodes that needs to be deleted
- node, err := logic.GetNodeByHostRef(hostid, network)
- if err != nil {
- slog.Error(
- "couldn't get node for host",
- "hostid",
- hostid,
- "network",
- network,
- "error",
- err,
- )
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if err = logic.DeleteNodeByID(&node); err != nil {
- slog.Error("failed to force delete daemon node",
- "nodeid", node.ID.String(), "hostid", hostid, "network", network, "error", err)
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
- "internal",
- ),
- )
- return
- }
- logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
- return
- }
- logger.Log(0, r.Header.Get("user"), "failed to find host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- node, err := logic.UpdateHostNetwork(currHost, network, false)
- if err != nil {
- if node == nil && forceDelete {
- // force cleanup the node
- node, err := logic.GetNodeByHostRef(hostid, network)
- if err != nil {
- slog.Error(
- "couldn't get node for host",
- "hostid",
- hostid,
- "network",
- network,
- "error",
- err,
- )
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if err = logic.DeleteNodeByID(&node); err != nil {
- slog.Error("failed to force delete daemon node",
- "nodeid", node.ID.String(), "hostid", hostid, "network", network, "error", err)
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
- "internal",
- ),
- )
- return
- }
- logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
- return
- }
- logger.Log(
- 0,
- r.Header.Get("user"),
- "failed to remove host from network:",
- hostid,
- network,
- err.Error(),
- )
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- var gwClients []models.ExtClient
- if node.IsIngressGateway {
- gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
- }
- logger.Log(1, "deleting node", node.ID.String(), "from host", currHost.Name)
- if err := logic.DeleteNode(node, forceDelete); err != nil {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(fmt.Errorf("failed to delete node"), "internal"),
- )
- return
- }
- go func() {
- mq.PublishMqUpdatesForDeletedNode(*node, true, gwClients)
- if servercfg.IsDNSMode() {
- logic.SetDNS()
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.RemoveHostFromNet,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: currHost.ID.String(),
- Name: currHost.Name,
- Type: models.DeviceSub,
- },
- NetworkID: models.NetworkID(network),
- Origin: models.Dashboard,
- })
- logger.Log(
- 2,
- r.Header.Get("user"),
- fmt.Sprintf("removed host %s from network %s", currHost.Name, network),
- )
- w.WriteHeader(http.StatusOK)
- }
- // @Summary To Fetch Auth Token for a Host
- // @Router /api/hosts/adm/authenticate [post]
- // @Tags Auth
- // @Accept json
- // @Param body body models.AuthParams true "Authentication parameters"
- // @Success 200 {object} models.SuccessResponse
- // @Failure 400 {object} models.ErrorResponse
- // @Failure 401 {object} models.ErrorResponse
- // @Failure 500 {object} models.ErrorResponse
- func authenticateHost(response http.ResponseWriter, request *http.Request) {
- var authRequest models.AuthParams
- var errorResponse = models.ErrorResponse{
- Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
- }
- decoder := json.NewDecoder(request.Body)
- decoderErr := decoder.Decode(&authRequest)
- defer request.Body.Close()
- if decoderErr != nil {
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = decoderErr.Error()
- logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
- decoderErr.Error())
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- errorResponse.Code = http.StatusBadRequest
- if authRequest.ID == "" {
- errorResponse.Message = "W1R3: ID can't be empty"
- logger.Log(0, request.Header.Get("user"), errorResponse.Message)
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- } else if authRequest.Password == "" {
- errorResponse.Message = "W1R3: Password can't be empty"
- logger.Log(0, request.Header.Get("user"), errorResponse.Message)
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- host, err := logic.GetHost(authRequest.ID)
- if err != nil {
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logger.Log(0, request.Header.Get("user"),
- "error retrieving host: ", authRequest.ID, err.Error())
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- err = bcrypt.CompareHashAndPassword([]byte(host.HostPass), []byte(authRequest.Password))
- if err != nil {
- errorResponse.Code = http.StatusUnauthorized
- errorResponse.Message = "unauthorized"
- logger.Log(0, request.Header.Get("user"),
- "error validating user password: ", err.Error())
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, "")
- if tokenString == "" {
- errorResponse.Code = http.StatusUnauthorized
- errorResponse.Message = "unauthorized"
- logger.Log(0, request.Header.Get("user"),
- fmt.Sprintf("%s: %v", errorResponse.Message, err))
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- var successResponse = models.SuccessResponse{
- Code: http.StatusOK,
- Message: "W1R3: Host " + authRequest.ID + " Authorized",
- Response: models.SuccessfulLoginResponse{
- AuthToken: tokenString,
- ID: authRequest.ID,
- },
- }
- successJSONResponse, jsonError := json.Marshal(successResponse)
- if jsonError != nil {
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logger.Log(0, request.Header.Get("user"),
- "error marshalling resp: ", err.Error())
- logic.ReturnErrorResponse(response, request, errorResponse)
- return
- }
- go func() {
- // Create EMQX creds
- if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
- if err := mq.GetEmqxHandler().CreateEmqxUser(host.ID.String(), authRequest.Password); err != nil {
- slog.Error("failed to create host credentials for EMQX: ", err.Error())
- }
- }
- }()
- response.WriteHeader(http.StatusOK)
- response.Header().Set("Content-Type", "application/json")
- response.Write(successJSONResponse)
- }
- // @Summary Send signal to peer
- // @Router /api/v1/host/{hostid}/signalpeer [post]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Param body body models.Signal true "Signal data"
- // @Success 200 {object} models.Signal
- // @Failure 400 {object} models.ErrorResponse
- func signalPeer(w http.ResponseWriter, r *http.Request) {
- var params = mux.Vars(r)
- hostid := params["hostid"]
- // confirm host exists
- _, err := logic.GetHost(hostid)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to get host:", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- var signal models.Signal
- w.Header().Set("Content-Type", "application/json")
- err = json.NewDecoder(r.Body).Decode(&signal)
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
- return
- }
- if signal.ToHostPubKey == "" {
- msg := "insufficient data to signal peer"
- logger.Log(0, r.Header.Get("user"), msg)
- logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New(msg), "badrequest"))
- return
- }
- signal.IsPro = servercfg.IsPro
- peerHost, err := logic.GetHost(signal.ToHostID)
- if err != nil {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(errors.New("failed to signal, peer not found"), "badrequest"),
- )
- return
- }
- err = mq.HostUpdate(&models.HostUpdate{
- Action: models.SignalHost,
- Host: *peerHost,
- Signal: signal,
- })
- if err != nil {
- logic.ReturnErrorResponse(
- w,
- r,
- logic.FormatError(
- errors.New("failed to publish signal to peer: "+err.Error()),
- "badrequest",
- ),
- )
- return
- }
- w.WriteHeader(http.StatusOK)
- json.NewEncoder(w).Encode(signal)
- }
- // @Summary Update keys for all hosts
- // @Router /api/hosts/keys [put]
- // @Tags Hosts
- // @Security oauth
- // @Success 200 {string} string "OK"
- // @Failure 400 {object} models.ErrorResponse
- func updateAllKeys(w http.ResponseWriter, r *http.Request) {
- var errorResponse = models.ErrorResponse{}
- w.Header().Set("Content-Type", "application/json")
- hosts, err := logic.GetAllHosts()
- if err != nil {
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logger.Log(0, r.Header.Get("user"),
- "error retrieving hosts ", err.Error())
- logic.ReturnErrorResponse(w, r, errorResponse)
- return
- }
- go func() {
- hostUpdate := models.HostUpdate{}
- hostUpdate.Action = models.UpdateKeys
- for _, host := range hosts {
- hostUpdate.Host = host
- logger.Log(2, "updating host", host.ID.String(), " for a key update")
- if err = mq.HostUpdate(&hostUpdate); err != nil {
- logger.Log(
- 0,
- "failed to send update to node during a network wide key update",
- host.ID.String(),
- err.Error(),
- )
- }
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.RefreshAllKeys,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: "All Devices",
- Name: "All Devices",
- Type: models.DeviceSub,
- },
- Origin: models.Dashboard,
- })
- logger.Log(2, r.Header.Get("user"), "updated keys for all hosts")
- w.WriteHeader(http.StatusOK)
- }
- // @Summary Update keys for a host
- // @Router /api/hosts/{hostid}/keys [put]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Success 200 {string} string "OK"
- // @Failure 400 {object} models.ErrorResponse
- func updateKeys(w http.ResponseWriter, r *http.Request) {
- var errorResponse = models.ErrorResponse{}
- w.Header().Set("Content-Type", "application/json")
- var params = mux.Vars(r)
- hostid := params["hostid"]
- host, err := logic.GetHost(hostid)
- if err != nil {
- logger.Log(0, "failed to retrieve host", hostid, err.Error())
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logger.Log(0, r.Header.Get("user"),
- "error retrieving hosts ", err.Error())
- logic.ReturnErrorResponse(w, r, errorResponse)
- return
- }
- go func() {
- hostUpdate := models.HostUpdate{
- Action: models.UpdateKeys,
- Host: *host,
- }
- if err = mq.HostUpdate(&hostUpdate); err != nil {
- logger.Log(0, "failed to send host key update", host.ID.String(), err.Error())
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.RefreshKey,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: host.ID.String(),
- Name: host.Name,
- Type: models.DeviceSub,
- },
- Origin: models.Dashboard,
- })
- logger.Log(2, r.Header.Get("user"), "updated key on host", host.Name)
- w.WriteHeader(http.StatusOK)
- }
- // @Summary Requests all the hosts to pull
- // @Router /api/hosts/sync [post]
- // @Tags Hosts
- // @Security oauth
- // @Success 200 {string} string "sync all hosts request received"
- func syncHosts(w http.ResponseWriter, r *http.Request) {
- w.Header().Set("Content-Type", "application/json")
- user := r.Header.Get("user")
- go func() {
- slog.Info("requesting all hosts to sync", "user", user)
- hosts, err := logic.GetAllHosts()
- if err != nil {
- slog.Error("failed to retrieve all hosts", "user", user, "error", err)
- return
- }
- for _, host := range hosts {
- go func(host models.Host) {
- hostUpdate := models.HostUpdate{
- Action: models.RequestPull,
- Host: host,
- }
- if err = mq.HostUpdate(&hostUpdate); err != nil {
- slog.Error("failed to request host to sync", "user", user, "host", host.ID.String(), "error", err)
- } else {
- slog.Info("host sync requested", "user", user, "host", host.ID.String())
- }
- }(host)
- time.Sleep(time.Millisecond * 100)
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.SyncAll,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: "All Devices",
- Name: "All Devices",
- Type: models.DeviceSub,
- },
- Origin: models.Dashboard,
- })
- slog.Info("sync all hosts request received", "user", user)
- logic.ReturnSuccessResponse(w, r, "sync all hosts request received")
- }
- // @Summary Requests a host to pull
- // @Router /api/hosts/{hostid}/sync [post]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Success 200 {string} string "OK"
- // @Failure 400 {object} models.ErrorResponse
- func syncHost(w http.ResponseWriter, r *http.Request) {
- hostId := mux.Vars(r)["hostid"]
- var errorResponse = models.ErrorResponse{}
- w.Header().Set("Content-Type", "application/json")
- host, err := logic.GetHost(hostId)
- if err != nil {
- slog.Error("failed to retrieve host", "user", r.Header.Get("user"), "error", err)
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logic.ReturnErrorResponse(w, r, errorResponse)
- return
- }
- go func() {
- hostUpdate := models.HostUpdate{
- Action: models.RequestPull,
- Host: *host,
- }
- if err = mq.HostUpdate(&hostUpdate); err != nil {
- slog.Error("failed to send host pull request", "host", host.ID.String(), "error", err)
- }
- }()
- logic.LogEvent(&models.Event{
- Action: models.Sync,
- Source: models.Subject{
- ID: r.Header.Get("user"),
- Name: r.Header.Get("user"),
- Type: models.UserSub,
- },
- TriggeredBy: r.Header.Get("user"),
- Target: models.Subject{
- ID: host.ID.String(),
- Name: host.Name,
- Type: models.DeviceSub,
- },
- Origin: models.Dashboard,
- })
- slog.Info("requested host pull", "user", r.Header.Get("user"), "host", host.ID.String())
- w.WriteHeader(http.StatusOK)
- }
- // @Summary Deletes all EMQX hosts
- // @Router /api/emqx/hosts [delete]
- // @Tags Hosts
- // @Security oauth
- // @Success 200 {string} string "deleted hosts data on emqx"
- // @Failure 500 {object} models.ErrorResponse
- func delEmqxHosts(w http.ResponseWriter, r *http.Request) {
- currentHosts, err := logic.GetAllHosts()
- if err != nil {
- logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
- return
- }
- for _, host := range currentHosts {
- // delete EMQX credentials for host
- if err := mq.GetEmqxHandler().DeleteEmqxUser(host.ID.String()); err != nil {
- slog.Error("failed to remove host credentials from EMQX", "id", host.ID, "error", err)
- }
- }
- err = mq.GetEmqxHandler().DeleteEmqxUser(servercfg.GetMqUserName())
- if err != nil {
- slog.Error(
- "failed to remove server credentials from EMQX",
- "user",
- servercfg.GetMqUserName(),
- "error",
- err,
- )
- }
- logic.ReturnSuccessResponse(w, r, "deleted hosts data on emqx")
- }
- // @Summary Fetches host peerinfo
- // @Router /api/host/{hostid}/peer_info [get]
- // @Tags Hosts
- // @Security oauth
- // @Param hostid path string true "Host ID"
- // @Success 200 {object} models.SuccessResponse
- // @Failure 500 {object} models.ErrorResponse
- func getHostPeerInfo(w http.ResponseWriter, r *http.Request) {
- hostId := mux.Vars(r)["hostid"]
- var errorResponse = models.ErrorResponse{}
- host, err := logic.GetHost(hostId)
- if err != nil {
- slog.Error("failed to retrieve host", "error", err)
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logic.ReturnErrorResponse(w, r, errorResponse)
- return
- }
- peerInfo, err := logic.GetHostPeerInfo(host)
- if err != nil {
- slog.Error("failed to retrieve host peerinfo", "error", err)
- errorResponse.Code = http.StatusBadRequest
- errorResponse.Message = err.Error()
- logic.ReturnErrorResponse(w, r, errorResponse)
- return
- }
- logic.ReturnSuccessResponseWithJson(w, r, peerInfo, "fetched host peer info")
- }
|