user.go 51 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520
  1. package controller
  2. import (
  3. "bytes"
  4. "encoding/base64"
  5. "encoding/json"
  6. "errors"
  7. "fmt"
  8. "github.com/pquerna/otp"
  9. "golang.org/x/crypto/bcrypt"
  10. "image/png"
  11. "net/http"
  12. "reflect"
  13. "time"
  14. "github.com/google/uuid"
  15. "github.com/gorilla/mux"
  16. "github.com/gorilla/websocket"
  17. "github.com/gravitl/netmaker/auth"
  18. "github.com/gravitl/netmaker/logger"
  19. "github.com/gravitl/netmaker/logic"
  20. "github.com/gravitl/netmaker/models"
  21. "github.com/gravitl/netmaker/mq"
  22. "github.com/gravitl/netmaker/schema"
  23. "github.com/gravitl/netmaker/servercfg"
  24. "github.com/pquerna/otp/totp"
  25. "golang.org/x/exp/slog"
  26. )
  27. var (
  28. upgrader = websocket.Upgrader{}
  29. )
  30. var ListRoles = listRoles
  31. func userHandlers(r *mux.Router) {
  32. r.HandleFunc("/api/users/adm/hassuperadmin", hasSuperAdmin).Methods(http.MethodGet)
  33. r.HandleFunc("/api/users/adm/createsuperadmin", createSuperAdmin).Methods(http.MethodPost)
  34. r.HandleFunc("/api/users/adm/transfersuperadmin/{username}", logic.SecurityCheck(true, http.HandlerFunc(transferSuperAdmin))).
  35. Methods(http.MethodPost)
  36. r.HandleFunc("/api/users/adm/authenticate", authenticateUser).Methods(http.MethodPost)
  37. r.HandleFunc("/api/users/{username}/validate-identity", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(validateUserIdentity)))).Methods(http.MethodPost)
  38. r.HandleFunc("/api/users/{username}/auth/init-totp", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(initiateTOTPSetup)))).Methods(http.MethodPost)
  39. r.HandleFunc("/api/users/{username}/auth/complete-totp", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(completeTOTPSetup)))).Methods(http.MethodPost)
  40. r.HandleFunc("/api/users/{username}/auth/verify-totp", logic.PreAuthCheck(logic.ContinueIfUserMatch(http.HandlerFunc(verifyTOTP)))).Methods(http.MethodPost)
  41. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, http.HandlerFunc(updateUser))).Methods(http.MethodPut)
  42. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, checkFreeTierLimits(limitChoiceUsers, http.HandlerFunc(createUser)))).Methods(http.MethodPost)
  43. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(true, http.HandlerFunc(deleteUser))).Methods(http.MethodDelete)
  44. r.HandleFunc("/api/users/{username}", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUser)))).Methods(http.MethodGet)
  45. r.HandleFunc("/api/users/{username}/enable", logic.SecurityCheck(true, http.HandlerFunc(enableUserAccount))).Methods(http.MethodPost)
  46. r.HandleFunc("/api/users/{username}/disable", logic.SecurityCheck(true, http.HandlerFunc(disableUserAccount))).Methods(http.MethodPost)
  47. r.HandleFunc("/api/users/{username}/settings", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUserSettings)))).Methods(http.MethodGet)
  48. r.HandleFunc("/api/users/{username}/settings", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(updateUserSettings)))).Methods(http.MethodPut)
  49. r.HandleFunc("/api/v1/users", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(getUserV1)))).Methods(http.MethodGet)
  50. r.HandleFunc("/api/users", logic.SecurityCheck(true, http.HandlerFunc(getUsers))).Methods(http.MethodGet)
  51. r.HandleFunc("/api/v1/users/roles", logic.SecurityCheck(true, http.HandlerFunc(ListRoles))).Methods(http.MethodGet)
  52. r.HandleFunc("/api/v1/users/access_token", logic.SecurityCheck(true, http.HandlerFunc(createUserAccessToken))).Methods(http.MethodPost)
  53. r.HandleFunc("/api/v1/users/access_token", logic.SecurityCheck(true, http.HandlerFunc(getUserAccessTokens))).Methods(http.MethodGet)
  54. r.HandleFunc("/api/v1/users/access_token", logic.SecurityCheck(true, http.HandlerFunc(deleteUserAccessTokens))).Methods(http.MethodDelete)
  55. r.HandleFunc("/api/v1/users/logout", logic.SecurityCheck(false, logic.ContinueIfUserMatch(http.HandlerFunc(logout)))).Methods(http.MethodPost)
  56. }
  57. // @Summary Authenticate a user to retrieve an authorization token
  58. // @Router /api/v1/users/{username}/access_token [post]
  59. // @Tags Auth
  60. // @Accept json
  61. // @Param body body models.UserAuthParams true "Authentication parameters"
  62. // @Success 200 {object} models.SuccessResponse
  63. // @Failure 400 {object} models.ErrorResponse
  64. // @Failure 401 {object} models.ErrorResponse
  65. // @Failure 500 {object} models.ErrorResponse
  66. func createUserAccessToken(w http.ResponseWriter, r *http.Request) {
  67. // Auth request consists of Mac Address and Password (from node that is authorizing
  68. // in case of Master, auth is ignored and mac is set to "mastermac"
  69. var req schema.UserAccessToken
  70. err := json.NewDecoder(r.Body).Decode(&req)
  71. if err != nil {
  72. logger.Log(0, "error decoding request body: ",
  73. err.Error())
  74. logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
  75. return
  76. }
  77. if req.Name == "" {
  78. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("name is required"), logic.BadReq))
  79. return
  80. }
  81. if req.UserName == "" {
  82. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username is required"), logic.BadReq))
  83. return
  84. }
  85. caller, err := logic.GetUser(r.Header.Get("user"))
  86. if err != nil {
  87. logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.UnAuthorized))
  88. return
  89. }
  90. user, err := logic.GetUser(req.UserName)
  91. if err != nil {
  92. logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.UnAuthorized))
  93. return
  94. }
  95. if caller.UserName != user.UserName && caller.PlatformRoleID != models.SuperAdminRole {
  96. if caller.PlatformRoleID == models.AdminRole {
  97. if user.PlatformRoleID == models.SuperAdminRole || user.PlatformRoleID == models.AdminRole {
  98. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("not enough permissions to create token for user "+user.UserName), logic.Forbidden_Msg))
  99. return
  100. }
  101. } else {
  102. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("not enough permissions to create token for user "+user.UserName), logic.Forbidden_Msg))
  103. return
  104. }
  105. }
  106. req.ID = uuid.New().String()
  107. req.CreatedBy = r.Header.Get("user")
  108. req.CreatedAt = time.Now()
  109. jwt, err := logic.CreateUserAccessJwtToken(user.UserName, user.PlatformRoleID, req.ExpiresAt, req.ID)
  110. if jwt == "" {
  111. // very unlikely that err is !nil and no jwt returned, but handle it anyways.
  112. logic.ReturnErrorResponse(
  113. w,
  114. r,
  115. logic.FormatError(errors.New("error creating access token "+err.Error()), logic.Internal),
  116. )
  117. return
  118. }
  119. err = req.Create(r.Context())
  120. if err != nil {
  121. logic.ReturnErrorResponse(
  122. w,
  123. r,
  124. logic.FormatError(errors.New("error creating access token "+err.Error()), logic.Internal),
  125. )
  126. return
  127. }
  128. logic.LogEvent(&models.Event{
  129. Action: models.Create,
  130. Source: models.Subject{
  131. ID: caller.UserName,
  132. Name: caller.UserName,
  133. Type: models.UserSub,
  134. },
  135. TriggeredBy: caller.UserName,
  136. Target: models.Subject{
  137. ID: req.ID,
  138. Name: req.Name,
  139. Type: models.UserAccessTokenSub,
  140. Info: req,
  141. },
  142. Origin: models.Dashboard,
  143. })
  144. logic.ReturnSuccessResponseWithJson(w, r, models.SuccessfulUserLoginResponse{
  145. AuthToken: jwt,
  146. UserName: req.UserName,
  147. }, "api access token has generated for user "+req.UserName)
  148. }
  149. // @Summary Authenticate a user to retrieve an authorization token
  150. // @Router /api/v1/users/{username}/access_token [post]
  151. // @Tags Auth
  152. // @Accept json
  153. // @Param body body models.UserAuthParams true "Authentication parameters"
  154. // @Success 200 {object} models.SuccessResponse
  155. // @Failure 400 {object} models.ErrorResponse
  156. // @Failure 401 {object} models.ErrorResponse
  157. // @Failure 500 {object} models.ErrorResponse
  158. func getUserAccessTokens(w http.ResponseWriter, r *http.Request) {
  159. username := r.URL.Query().Get("username")
  160. if username == "" {
  161. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username is required"), "badrequest"))
  162. return
  163. }
  164. logic.ReturnSuccessResponseWithJson(w, r, (&schema.UserAccessToken{UserName: username}).ListByUser(r.Context()), "fetched api access tokens for user "+username)
  165. }
  166. // @Summary Authenticate a user to retrieve an authorization token
  167. // @Router /api/v1/users/{username}/access_token [post]
  168. // @Tags Auth
  169. // @Accept json
  170. // @Param body body models.UserAuthParams true "Authentication parameters"
  171. // @Success 200 {object} models.SuccessResponse
  172. // @Failure 400 {object} models.ErrorResponse
  173. // @Failure 401 {object} models.ErrorResponse
  174. // @Failure 500 {object} models.ErrorResponse
  175. func deleteUserAccessTokens(w http.ResponseWriter, r *http.Request) {
  176. id := r.URL.Query().Get("id")
  177. if id == "" {
  178. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("id is required"), "badrequest"))
  179. return
  180. }
  181. a := schema.UserAccessToken{
  182. ID: id,
  183. }
  184. err := a.Get(r.Context())
  185. if err != nil {
  186. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("token does not exist"), "badrequest"))
  187. return
  188. }
  189. caller, err := logic.GetUser(r.Header.Get("user"))
  190. if err != nil {
  191. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "unauthorized"))
  192. return
  193. }
  194. user, err := logic.GetUser(a.UserName)
  195. if err != nil {
  196. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "unauthorized"))
  197. return
  198. }
  199. if caller.UserName != user.UserName && caller.PlatformRoleID != models.SuperAdminRole {
  200. if caller.PlatformRoleID == models.AdminRole {
  201. if user.PlatformRoleID == models.SuperAdminRole || user.PlatformRoleID == models.AdminRole {
  202. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("not enough permissions to delete token of user "+user.UserName), logic.Forbidden_Msg))
  203. return
  204. }
  205. } else {
  206. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("not enough permissions to delete token of user "+user.UserName), logic.Forbidden_Msg))
  207. return
  208. }
  209. }
  210. err = (&schema.UserAccessToken{ID: id}).Delete(r.Context())
  211. if err != nil {
  212. logic.ReturnErrorResponse(
  213. w,
  214. r,
  215. logic.FormatError(errors.New("error deleting access token "+err.Error()), "internal"),
  216. )
  217. return
  218. }
  219. logic.LogEvent(&models.Event{
  220. Action: models.Delete,
  221. Source: models.Subject{
  222. ID: caller.UserName,
  223. Name: caller.UserName,
  224. Type: models.UserSub,
  225. },
  226. TriggeredBy: caller.UserName,
  227. Target: models.Subject{
  228. ID: a.ID,
  229. Name: a.Name,
  230. Type: models.UserAccessTokenSub,
  231. Info: a,
  232. },
  233. Origin: models.Dashboard,
  234. })
  235. logic.ReturnSuccessResponseWithJson(w, r, nil, "revoked access token")
  236. }
  237. // @Summary Authenticate a user to retrieve an authorization token
  238. // @Router /api/users/adm/authenticate [post]
  239. // @Tags Auth
  240. // @Accept json
  241. // @Param body body models.UserAuthParams true "Authentication parameters"
  242. // @Success 200 {object} models.SuccessResponse
  243. // @Failure 400 {object} models.ErrorResponse
  244. // @Failure 401 {object} models.ErrorResponse
  245. // @Failure 500 {object} models.ErrorResponse
  246. func authenticateUser(response http.ResponseWriter, request *http.Request) {
  247. appName := request.Header.Get("X-Application-Name")
  248. if appName == "" {
  249. appName = logic.NetmakerDesktopApp
  250. }
  251. // Auth request consists of Mac Address and Password (from node that is authorizing
  252. // in case of Master, auth is ignored and mac is set to "mastermac"
  253. var authRequest models.UserAuthParams
  254. var errorResponse = models.ErrorResponse{
  255. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  256. }
  257. decoder := json.NewDecoder(request.Body)
  258. decoderErr := decoder.Decode(&authRequest)
  259. defer request.Body.Close()
  260. if decoderErr != nil {
  261. logger.Log(0, "error decoding request body: ",
  262. decoderErr.Error())
  263. logic.ReturnErrorResponse(response, request, errorResponse)
  264. return
  265. }
  266. user, err := logic.GetUser(authRequest.UserName)
  267. if err != nil {
  268. logger.Log(0, authRequest.UserName, "user validation failed: ",
  269. err.Error())
  270. logic.ReturnErrorResponse(response, request, logic.FormatError(err, "unauthorized"))
  271. return
  272. }
  273. if logic.IsOauthUser(user) == nil {
  274. logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("user is registered via SSO"), "badrequest"))
  275. return
  276. }
  277. if user.AccountDisabled {
  278. err = errors.New("user account disabled")
  279. logic.ReturnErrorResponse(response, request, logic.FormatError(err, "unauthorized"))
  280. return
  281. }
  282. if !user.IsSuperAdmin && !logic.IsBasicAuthEnabled() {
  283. logic.ReturnErrorResponse(
  284. response,
  285. request,
  286. logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"),
  287. )
  288. return
  289. }
  290. if val := request.Header.Get("From-Ui"); val == "true" {
  291. // request came from UI, if normal user block Login
  292. role, err := logic.GetRole(user.PlatformRoleID)
  293. if err != nil {
  294. logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("access denied to dashboard"), "unauthorized"))
  295. return
  296. }
  297. if role.DenyDashboardAccess {
  298. logic.ReturnErrorResponse(response, request, logic.FormatError(errors.New("access denied to dashboard"), "unauthorized"))
  299. return
  300. }
  301. }
  302. username := authRequest.UserName
  303. jwt, err := logic.VerifyAuthRequest(authRequest, appName)
  304. if err != nil {
  305. logger.Log(0, username, "user validation failed: ",
  306. err.Error())
  307. logic.ReturnErrorResponse(response, request, logic.FormatError(err, "badrequest"))
  308. return
  309. }
  310. if jwt == "" {
  311. // very unlikely that err is !nil and no jwt returned, but handle it anyways.
  312. logger.Log(0, username, "jwt token is empty")
  313. logic.ReturnErrorResponse(
  314. response,
  315. request,
  316. logic.FormatError(errors.New("no token returned"), "internal"),
  317. )
  318. return
  319. }
  320. var successResponse models.SuccessResponse
  321. if user.IsMFAEnabled {
  322. successResponse = models.SuccessResponse{
  323. Code: http.StatusOK,
  324. Message: "W1R3: TOTP required",
  325. Response: models.PartialUserLoginResponse{
  326. UserName: username,
  327. PreAuthToken: jwt,
  328. },
  329. }
  330. } else {
  331. successResponse = models.SuccessResponse{
  332. Code: http.StatusOK,
  333. Message: "W1R3: Device " + username + " Authorized",
  334. Response: models.SuccessfulUserLoginResponse{
  335. UserName: username,
  336. AuthToken: jwt,
  337. },
  338. }
  339. }
  340. // Send back the JWT
  341. successJSONResponse, jsonError := json.Marshal(successResponse)
  342. if jsonError != nil {
  343. logger.Log(0, username,
  344. "error marshalling resp: ", jsonError.Error())
  345. logic.ReturnErrorResponse(response, request, errorResponse)
  346. return
  347. }
  348. logger.Log(2, username, "was authenticated")
  349. // log user activity
  350. if !user.IsMFAEnabled {
  351. if val := request.Header.Get("From-Ui"); val == "true" {
  352. logic.LogEvent(&models.Event{
  353. Action: models.Login,
  354. Source: models.Subject{
  355. ID: user.UserName,
  356. Name: user.UserName,
  357. Type: models.UserSub,
  358. },
  359. TriggeredBy: user.UserName,
  360. Target: models.Subject{
  361. ID: models.DashboardSub.String(),
  362. Name: models.DashboardSub.String(),
  363. Type: models.DashboardSub,
  364. },
  365. Origin: models.Dashboard,
  366. })
  367. } else {
  368. logic.LogEvent(&models.Event{
  369. Action: models.Login,
  370. Source: models.Subject{
  371. ID: user.UserName,
  372. Name: user.UserName,
  373. Type: models.UserSub,
  374. },
  375. TriggeredBy: user.UserName,
  376. Target: models.Subject{
  377. ID: models.ClientAppSub.String(),
  378. Name: models.ClientAppSub.String(),
  379. Type: models.ClientAppSub,
  380. },
  381. Origin: models.ClientApp,
  382. })
  383. }
  384. }
  385. response.Header().Set("Content-Type", "application/json")
  386. response.Write(successJSONResponse)
  387. go func() {
  388. if servercfg.IsPro {
  389. // enable all associeated clients for the user
  390. clients, err := logic.GetAllExtClients()
  391. if err != nil {
  392. slog.Error("error getting clients: ", "error", err)
  393. return
  394. }
  395. for _, client := range clients {
  396. if client.OwnerID == username && !client.Enabled {
  397. slog.Info(
  398. fmt.Sprintf(
  399. "enabling ext client %s for user %s due to RAC autodisabling feature",
  400. client.ClientID,
  401. client.OwnerID,
  402. ),
  403. )
  404. if newClient, err := logic.ToggleExtClientConnectivity(&client, true); err != nil {
  405. slog.Error(
  406. "error enabling ext client in RAC autodisable hook",
  407. "error",
  408. err,
  409. )
  410. continue // dont return but try for other clients
  411. } else {
  412. // publish peer update to ingress gateway
  413. if ingressNode, err := logic.GetNodeByID(newClient.IngressGatewayID); err == nil {
  414. if err = mq.PublishPeerUpdate(false); err != nil {
  415. slog.Error("error updating ext clients on", "ingress", ingressNode.ID.String(), "err", err.Error())
  416. }
  417. }
  418. }
  419. }
  420. }
  421. }
  422. }()
  423. }
  424. // @Summary Validates a user's identity against it's token. This is used by UI before a user performing a critical operation to validate the user's identity.
  425. // @Router /api/users/{username}/validate-identity [post]
  426. // @Tags Auth
  427. // @Accept json
  428. // @Param body body models.UserIdentityValidationRequest true "User Identity Validation Request"
  429. // @Success 200 {object} models.SuccessResponse
  430. // @Failure 400 {object} models.ErrorResponse
  431. func validateUserIdentity(w http.ResponseWriter, r *http.Request) {
  432. username := r.Header.Get("user")
  433. var req models.UserIdentityValidationRequest
  434. err := json.NewDecoder(r.Body).Decode(&req)
  435. if err != nil {
  436. logger.Log(0, "failed to decode request body: ", err.Error())
  437. err = fmt.Errorf("invalid request body: %v", err)
  438. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  439. return
  440. }
  441. user, err := logic.GetUser(username)
  442. if err != nil {
  443. logger.Log(0, "failed to get user: ", err.Error())
  444. err = fmt.Errorf("user not found: %v", err)
  445. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  446. return
  447. }
  448. var resp models.UserIdentityValidationResponse
  449. err = bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(req.Password))
  450. if err != nil {
  451. logic.ReturnSuccessResponseWithJson(w, r, resp, "user identity validation failed")
  452. } else {
  453. resp.IdentityValidated = true
  454. logic.ReturnSuccessResponseWithJson(w, r, resp, "user identity validated")
  455. }
  456. }
  457. // @Summary Initiate setting up TOTP 2FA for a user.
  458. // @Router /api/users/auth/init-totp [post]
  459. // @Tags Auth
  460. // @Success 200 {object} models.SuccessResponse
  461. // @Failure 400 {object} models.ErrorResponse
  462. // @Failure 500 {object} models.ErrorResponse
  463. func initiateTOTPSetup(w http.ResponseWriter, r *http.Request) {
  464. username := r.Header.Get("user")
  465. user, err := logic.GetUser(username)
  466. if err != nil {
  467. logger.Log(0, "failed to get user: ", err.Error())
  468. err = fmt.Errorf("user not found: %v", err)
  469. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  470. return
  471. }
  472. if user.AuthType == models.OAuth {
  473. err = fmt.Errorf("auth type is %s, cannot process totp setup", user.AuthType)
  474. logger.Log(0, err.Error())
  475. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  476. return
  477. }
  478. key, err := totp.Generate(totp.GenerateOpts{
  479. Issuer: "Netmaker",
  480. AccountName: username,
  481. })
  482. if err != nil {
  483. err = fmt.Errorf("failed to generate totp key: %v", err)
  484. logger.Log(0, err.Error())
  485. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  486. return
  487. }
  488. qrCodeImg, err := key.Image(200, 200)
  489. if err != nil {
  490. err = fmt.Errorf("failed to generate totp key: %v", err)
  491. logger.Log(0, err.Error())
  492. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  493. return
  494. }
  495. var qrCodePng bytes.Buffer
  496. err = png.Encode(&qrCodePng, qrCodeImg)
  497. if err != nil {
  498. err = fmt.Errorf("failed to generate totp key: %v", err)
  499. logger.Log(0, err.Error())
  500. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  501. return
  502. }
  503. qrCode := "data:image/png;base64," + base64.StdEncoding.EncodeToString(qrCodePng.Bytes())
  504. logic.ReturnSuccessResponseWithJson(w, r, models.TOTPInitiateResponse{
  505. OTPAuthURL: key.URL(),
  506. OTPAuthURLSignature: logic.GenerateOTPAuthURLSignature(key.URL()),
  507. QRCode: qrCode,
  508. }, "totp setup initiated")
  509. }
  510. // @Summary Verify and complete setting up TOTP 2FA for a user.
  511. // @Router /api/users/auth/complete-totp [post]
  512. // @Tags Auth
  513. // @Param body body models.UserTOTPVerificationParams true "TOTP verification parameters"
  514. // @Success 200 {object} models.SuccessResponse
  515. // @Failure 400 {object} models.ErrorResponse
  516. // @Failure 500 {object} models.ErrorResponse
  517. func completeTOTPSetup(w http.ResponseWriter, r *http.Request) {
  518. username := r.Header.Get("user")
  519. var req models.UserTOTPVerificationParams
  520. err := json.NewDecoder(r.Body).Decode(&req)
  521. if err != nil {
  522. logger.Log(0, "failed to decode request body: ", err.Error())
  523. err = fmt.Errorf("invalid request body: %v", err)
  524. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  525. return
  526. }
  527. if !logic.VerifyOTPAuthURL(req.OTPAuthURL, req.OTPAuthURLSignature) {
  528. err = fmt.Errorf("otp auth url signature mismatch")
  529. logger.Log(0, err.Error())
  530. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  531. return
  532. }
  533. user, err := logic.GetUser(username)
  534. if err != nil {
  535. logger.Log(0, "failed to get user: ", err.Error())
  536. err = fmt.Errorf("user not found: %v", err)
  537. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  538. return
  539. }
  540. if user.AuthType == models.OAuth {
  541. err = fmt.Errorf("auth type is %s, cannot process totp setup", user.AuthType)
  542. logger.Log(0, err.Error())
  543. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  544. return
  545. }
  546. otpAuthURL, err := otp.NewKeyFromURL(req.OTPAuthURL)
  547. if err != nil {
  548. err = fmt.Errorf("error parsing otp auth url: %v", err)
  549. logger.Log(0, err.Error())
  550. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  551. return
  552. }
  553. totpSecret := otpAuthURL.Secret()
  554. if totp.Validate(req.TOTP, totpSecret) {
  555. user.IsMFAEnabled = true
  556. user.TOTPSecret = totpSecret
  557. err = logic.UpsertUser(*user)
  558. if err != nil {
  559. err = fmt.Errorf("error upserting user: %v", err)
  560. logger.Log(0, err.Error())
  561. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  562. return
  563. }
  564. logic.LogEvent(&models.Event{
  565. Action: models.EnableMFA,
  566. Source: models.Subject{
  567. ID: user.UserName,
  568. Name: user.UserName,
  569. Type: models.UserSub,
  570. },
  571. TriggeredBy: user.UserName,
  572. Target: models.Subject{
  573. ID: user.UserName,
  574. Name: user.UserName,
  575. Type: models.UserSub,
  576. },
  577. Origin: models.Dashboard,
  578. })
  579. logic.ReturnSuccessResponse(w, r, fmt.Sprintf("totp setup complete for user %s", username))
  580. } else {
  581. err = fmt.Errorf("cannot setup totp for user %s: invalid otp", username)
  582. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  583. }
  584. }
  585. // @Summary Verify a user's TOTP token.
  586. // @Router /api/users/auth/verify-totp [post]
  587. // @Tags Auth
  588. // @Accept json
  589. // @Param body body models.UserTOTPVerificationParams true "TOTP verification parameters"
  590. // @Success 200 {object} models.SuccessResponse
  591. // @Failure 400 {object} models.ErrorResponse
  592. // @Failure 401 {object} models.ErrorResponse
  593. // @Failure 500 {object} models.ErrorResponse
  594. func verifyTOTP(w http.ResponseWriter, r *http.Request) {
  595. username := r.Header.Get("user")
  596. appName := r.Header.Get("X-Application-Name")
  597. if appName == "" {
  598. appName = logic.NetmakerDesktopApp
  599. }
  600. var req models.UserTOTPVerificationParams
  601. err := json.NewDecoder(r.Body).Decode(&req)
  602. if err != nil {
  603. logger.Log(0, "failed to decode request body: ", err.Error())
  604. err = fmt.Errorf("invalid request body: %v", err)
  605. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  606. return
  607. }
  608. user, err := logic.GetUser(username)
  609. if err != nil {
  610. logger.Log(0, "failed to get user: ", err.Error())
  611. err = fmt.Errorf("user not found: %v", err)
  612. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  613. return
  614. }
  615. if !user.IsMFAEnabled {
  616. err = fmt.Errorf("mfa is disabled for user(%s), cannot process totp verification", username)
  617. logger.Log(0, err.Error())
  618. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  619. return
  620. }
  621. if totp.Validate(req.TOTP, user.TOTPSecret) {
  622. jwt, err := logic.CreateUserJWT(user.UserName, user.PlatformRoleID, appName)
  623. if err != nil {
  624. err = fmt.Errorf("error creating token: %v", err)
  625. logger.Log(0, err.Error())
  626. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  627. return
  628. }
  629. // update last login time
  630. user.LastLoginTime = time.Now().UTC()
  631. err = logic.UpsertUser(*user)
  632. if err != nil {
  633. err = fmt.Errorf("error upserting user: %v", err)
  634. logger.Log(0, err.Error())
  635. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  636. return
  637. }
  638. logic.LogEvent(&models.Event{
  639. Action: models.Login,
  640. Source: models.Subject{
  641. ID: user.UserName,
  642. Name: user.UserName,
  643. Type: models.UserSub,
  644. },
  645. TriggeredBy: user.UserName,
  646. Target: models.Subject{
  647. ID: models.DashboardSub.String(),
  648. Name: models.DashboardSub.String(),
  649. Type: models.DashboardSub,
  650. },
  651. Origin: models.Dashboard,
  652. })
  653. logic.ReturnSuccessResponseWithJson(w, r, models.SuccessfulUserLoginResponse{
  654. UserName: username,
  655. AuthToken: jwt,
  656. }, "W1R3: User "+username+" Authorized")
  657. } else {
  658. err = fmt.Errorf("invalid otp")
  659. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "unauthorized"))
  660. }
  661. }
  662. // @Summary Check if the server has a super admin
  663. // @Router /api/users/adm/hassuperadmin [get]
  664. // @Tags Users
  665. // @Success 200 {object} bool
  666. // @Failure 500 {object} models.ErrorResponse
  667. func hasSuperAdmin(w http.ResponseWriter, r *http.Request) {
  668. w.Header().Set("Content-Type", "application/json")
  669. hasSuperAdmin, err := logic.HasSuperAdmin()
  670. if err != nil {
  671. logger.Log(0, "failed to check for admin: ", err.Error())
  672. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  673. return
  674. }
  675. json.NewEncoder(w).Encode(hasSuperAdmin)
  676. }
  677. // @Summary Get an individual user
  678. // @Router /api/users/{username} [get]
  679. // @Tags Users
  680. // @Param username path string true "Username of the user to fetch"
  681. // @Success 200 {object} models.User
  682. // @Failure 500 {object} models.ErrorResponse
  683. func getUser(w http.ResponseWriter, r *http.Request) {
  684. // set header.
  685. w.Header().Set("Content-Type", "application/json")
  686. var params = mux.Vars(r)
  687. usernameFetched := params["username"]
  688. user, err := logic.GetReturnUser(usernameFetched)
  689. if err != nil {
  690. logger.Log(0, usernameFetched, "failed to fetch user: ", err.Error())
  691. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  692. return
  693. }
  694. logger.Log(2, r.Header.Get("user"), "fetched user", usernameFetched)
  695. json.NewEncoder(w).Encode(user)
  696. }
  697. // @Summary Enable a user's account
  698. // @Router /api/users/{username}/enable [post]
  699. // @Tags Users
  700. // @Param username path string true "Username of the user to enable"
  701. // @Success 200 {object} models.SuccessResponse
  702. // @Failure 400 {object} models.ErrorResponse
  703. // @Failure 500 {object} models.ErrorResponse
  704. func enableUserAccount(w http.ResponseWriter, r *http.Request) {
  705. username := mux.Vars(r)["username"]
  706. user, err := logic.GetUser(username)
  707. if err != nil {
  708. logger.Log(0, "failed to fetch user: ", err.Error())
  709. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  710. return
  711. }
  712. user.AccountDisabled = false
  713. err = logic.UpsertUser(*user)
  714. if err != nil {
  715. logger.Log(0, "failed to enable user account: ", err.Error())
  716. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  717. }
  718. logic.ReturnSuccessResponse(w, r, "user account enabled")
  719. }
  720. // @Summary Disable a user's account
  721. // @Router /api/users/{username}/disable [post]
  722. // @Tags Users
  723. // @Param username path string true "Username of the user to disable"
  724. // @Success 200 {object} models.SuccessResponse
  725. // @Failure 400 {object} models.ErrorResponse
  726. // @Failure 500 {object} models.ErrorResponse
  727. func disableUserAccount(w http.ResponseWriter, r *http.Request) {
  728. username := mux.Vars(r)["username"]
  729. user, err := logic.GetUser(username)
  730. if err != nil {
  731. logger.Log(0, "failed to fetch user: ", err.Error())
  732. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  733. return
  734. }
  735. if user.PlatformRoleID == models.SuperAdminRole {
  736. err = errors.New("cannot disable super-admin user account")
  737. logger.Log(0, err.Error())
  738. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  739. return
  740. }
  741. user.AccountDisabled = true
  742. err = logic.UpsertUser(*user)
  743. if err != nil {
  744. logger.Log(0, "failed to disable user account: ", err.Error())
  745. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  746. }
  747. logic.ReturnSuccessResponse(w, r, "user account disabled")
  748. }
  749. // @Summary Get a user's preferences and settings
  750. // @Router /api/users/{username}/settings [get]
  751. // @Tags Users
  752. // @Param username path string true "Username of the user"
  753. // @Success 200 {object} models.SuccessResponse
  754. func getUserSettings(w http.ResponseWriter, r *http.Request) {
  755. userID := r.Header.Get("user")
  756. userSettings := logic.GetUserSettings(userID)
  757. logic.ReturnSuccessResponseWithJson(w, r, userSettings, "fetched user settings")
  758. }
  759. // @Summary Update a user's preferences and settings
  760. // @Router /api/users/{username}/settings [put]
  761. // @Tags Users
  762. // @Param username path string true "Username of the user"
  763. // @Success 200 {object} models.SuccessResponse
  764. // @Failure 400 {object} models.ErrorResponse
  765. // @Failure 500 {object} models.ErrorResponse
  766. func updateUserSettings(w http.ResponseWriter, r *http.Request) {
  767. userID := r.Header.Get("user")
  768. var req models.UserSettings
  769. err := json.NewDecoder(r.Body).Decode(&req)
  770. if err != nil {
  771. logger.Log(0, "failed to decode request body: ", err.Error())
  772. err = fmt.Errorf("invalid request body: %v", err)
  773. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  774. return
  775. }
  776. err = logic.UpsertUserSettings(userID, req)
  777. if err != nil {
  778. err = fmt.Errorf("failed to update user settings: %v", err.Error())
  779. logger.Log(0, err.Error())
  780. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  781. return
  782. }
  783. logic.ReturnSuccessResponseWithJson(w, r, req, "updated user settings")
  784. }
  785. // swagger:route GET /api/v1/users user getUserV1
  786. //
  787. // Get an individual user with role info.
  788. //
  789. // Schemes: https
  790. //
  791. // Security:
  792. // oauth
  793. //
  794. // Responses:
  795. // 200: ReturnUserWithRolesAndGroups
  796. func getUserV1(w http.ResponseWriter, r *http.Request) {
  797. // set header.
  798. w.Header().Set("Content-Type", "application/json")
  799. usernameFetched := r.URL.Query().Get("username")
  800. if usernameFetched == "" {
  801. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username is required"), "badrequest"))
  802. return
  803. }
  804. user, err := logic.GetReturnUser(usernameFetched)
  805. if err != nil {
  806. logger.Log(0, usernameFetched, "failed to fetch user: ", err.Error())
  807. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  808. return
  809. }
  810. userRoleTemplate, err := logic.GetRole(user.PlatformRoleID)
  811. if err != nil {
  812. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  813. return
  814. }
  815. resp := models.ReturnUserWithRolesAndGroups{
  816. ReturnUser: user,
  817. PlatformRole: userRoleTemplate,
  818. UserGroups: map[models.UserGroupID]models.UserGroup{},
  819. }
  820. for gId := range user.UserGroups {
  821. grp, err := logic.GetUserGroup(gId)
  822. if err != nil {
  823. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  824. return
  825. }
  826. resp.UserGroups[gId] = grp
  827. }
  828. logger.Log(2, r.Header.Get("user"), "fetched user", usernameFetched)
  829. logic.ReturnSuccessResponseWithJson(w, r, resp, "fetched user with role info")
  830. }
  831. // swagger:route GET /api/users user getUsers
  832. //
  833. // Get all users.
  834. //
  835. // Schemes: https
  836. //
  837. // Security:
  838. // oauth
  839. //
  840. // Responses:
  841. // 200: userBodyResponse
  842. func getUsers(w http.ResponseWriter, r *http.Request) {
  843. // set header.
  844. w.Header().Set("Content-Type", "application/json")
  845. users, err := logic.GetUsers()
  846. if err != nil {
  847. logger.Log(0, "failed to fetch users: ", err.Error())
  848. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  849. return
  850. }
  851. logic.SortUsers(users[:])
  852. logger.Log(2, r.Header.Get("user"), "fetched users")
  853. json.NewEncoder(w).Encode(users)
  854. }
  855. // @Summary Create a super admin
  856. // @Router /api/users/adm/createsuperadmin [post]
  857. // @Tags Users
  858. // @Param body body models.User true "User details"
  859. // @Success 200 {object} models.User
  860. // @Failure 400 {object} models.ErrorResponse
  861. // @Failure 500 {object} models.ErrorResponse
  862. func createSuperAdmin(w http.ResponseWriter, r *http.Request) {
  863. w.Header().Set("Content-Type", "application/json")
  864. var u models.User
  865. err := json.NewDecoder(r.Body).Decode(&u)
  866. if err != nil {
  867. slog.Error("error decoding request body", "error", err.Error())
  868. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  869. return
  870. }
  871. if !logic.IsBasicAuthEnabled() {
  872. logic.ReturnErrorResponse(
  873. w,
  874. r,
  875. logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"),
  876. )
  877. return
  878. }
  879. err = logic.CreateSuperAdmin(&u)
  880. if err != nil {
  881. slog.Error("failed to create admin", "error", err.Error())
  882. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  883. return
  884. }
  885. logger.Log(1, u.UserName, "was made a super admin")
  886. json.NewEncoder(w).Encode(logic.ToReturnUser(u))
  887. }
  888. // @Summary Transfer super admin role to another admin user
  889. // @Router /api/users/adm/transfersuperadmin/{username} [post]
  890. // @Tags Users
  891. // @Param username path string true "Username of the user to transfer super admin role"
  892. // @Success 200 {object} models.User
  893. // @Failure 403 {object} models.ErrorResponse
  894. // @Failure 500 {object} models.ErrorResponse
  895. func transferSuperAdmin(w http.ResponseWriter, r *http.Request) {
  896. w.Header().Set("Content-Type", "application/json")
  897. caller, err := logic.GetUser(r.Header.Get("user"))
  898. if err != nil {
  899. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  900. }
  901. if caller.PlatformRoleID != models.SuperAdminRole {
  902. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can assign the superadmin role to another user"), "forbidden"))
  903. return
  904. }
  905. var params = mux.Vars(r)
  906. username := params["username"]
  907. u, err := logic.GetUser(username)
  908. if err != nil {
  909. slog.Error("error getting user", "user", u.UserName, "error", err.Error())
  910. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  911. return
  912. }
  913. if u.PlatformRoleID != models.AdminRole {
  914. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only admins can be promoted to superadmin role"), "forbidden"))
  915. return
  916. }
  917. if !logic.IsBasicAuthEnabled() {
  918. logic.ReturnErrorResponse(
  919. w,
  920. r,
  921. logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"),
  922. )
  923. return
  924. }
  925. u.PlatformRoleID = models.SuperAdminRole
  926. err = logic.UpsertUser(*u)
  927. if err != nil {
  928. slog.Error("error updating user to superadmin: ", "user", u.UserName, "error", err.Error())
  929. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  930. return
  931. }
  932. caller.PlatformRoleID = models.AdminRole
  933. err = logic.UpsertUser(*caller)
  934. if err != nil {
  935. slog.Error("error demoting user to admin: ", "user", caller.UserName, "error", err.Error())
  936. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  937. return
  938. }
  939. slog.Info("user was made a super admin", "user", u.UserName)
  940. json.NewEncoder(w).Encode(logic.ToReturnUser(*u))
  941. }
  942. // @Summary Create a user
  943. // @Router /api/users/{username} [post]
  944. // @Tags Users
  945. // @Param username path string true "Username of the user to create"
  946. // @Param body body models.User true "User details"
  947. // @Success 200 {object} models.User
  948. // @Failure 400 {object} models.ErrorResponse
  949. // @Failure 403 {object} models.ErrorResponse
  950. // @Failure 500 {object} models.ErrorResponse
  951. func createUser(w http.ResponseWriter, r *http.Request) {
  952. w.Header().Set("Content-Type", "application/json")
  953. caller, err := logic.GetUser(r.Header.Get("user"))
  954. if err != nil {
  955. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  956. return
  957. }
  958. var user models.User
  959. err = json.NewDecoder(r.Body).Decode(&user)
  960. if err != nil {
  961. logger.Log(0, user.UserName, "error decoding request body: ",
  962. err.Error())
  963. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  964. return
  965. }
  966. if !servercfg.IsPro {
  967. user.PlatformRoleID = models.AdminRole
  968. }
  969. if user.UserName == logic.MasterUser {
  970. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username not allowed"), "badrequest"))
  971. return
  972. }
  973. if user.PlatformRoleID == "" {
  974. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("platform role is missing"), "badrequest"))
  975. return
  976. }
  977. userRole, err := logic.GetRole(user.PlatformRoleID)
  978. if err != nil {
  979. err = errors.New("error fetching role " + user.PlatformRoleID.String() + " " + err.Error())
  980. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  981. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  982. return
  983. }
  984. if userRole.ID == models.SuperAdminRole {
  985. err = errors.New("additional superadmins cannot be created")
  986. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  987. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  988. return
  989. }
  990. if caller.PlatformRoleID != models.SuperAdminRole && user.PlatformRoleID == models.AdminRole {
  991. err = errors.New("only superadmin can create admin users")
  992. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  993. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  994. return
  995. }
  996. if !servercfg.IsPro && user.PlatformRoleID != models.AdminRole {
  997. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("non-admins users can only be created on Pro version"), "forbidden"))
  998. return
  999. }
  1000. err = logic.CreateUser(&user)
  1001. if err != nil {
  1002. slog.Error("error creating new user: ", "user", user.UserName, "error", err.Error())
  1003. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1004. return
  1005. }
  1006. logic.LogEvent(&models.Event{
  1007. Action: models.Create,
  1008. Source: models.Subject{
  1009. ID: caller.UserName,
  1010. Name: caller.UserName,
  1011. Type: models.UserSub,
  1012. },
  1013. TriggeredBy: caller.UserName,
  1014. Target: models.Subject{
  1015. ID: user.UserName,
  1016. Name: user.UserName,
  1017. Type: models.UserSub,
  1018. Info: user,
  1019. },
  1020. Origin: models.Dashboard,
  1021. })
  1022. logic.DeleteUserInvite(user.UserName)
  1023. logic.DeletePendingUser(user.UserName)
  1024. go mq.PublishPeerUpdate(false)
  1025. slog.Info("user was created", "username", user.UserName)
  1026. json.NewEncoder(w).Encode(logic.ToReturnUser(user))
  1027. }
  1028. // @Summary Update a user
  1029. // @Router /api/users/{username} [put]
  1030. // @Tags Users
  1031. // @Param username path string true "Username of the user to update"
  1032. // @Param body body models.User true "User details"
  1033. // @Success 200 {object} models.User
  1034. // @Failure 400 {object} models.ErrorResponse
  1035. // @Failure 403 {object} models.ErrorResponse
  1036. // @Failure 500 {object} models.ErrorResponse
  1037. func updateUser(w http.ResponseWriter, r *http.Request) {
  1038. w.Header().Set("Content-Type", "application/json")
  1039. var params = mux.Vars(r)
  1040. // start here
  1041. var caller *models.User
  1042. var err error
  1043. var ismaster bool
  1044. if r.Header.Get("user") == logic.MasterUser {
  1045. ismaster = true
  1046. } else {
  1047. caller, err = logic.GetUser(r.Header.Get("user"))
  1048. if err != nil {
  1049. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1050. }
  1051. }
  1052. username := params["username"]
  1053. user, err := logic.GetUser(username)
  1054. if err != nil {
  1055. logger.Log(0, username,
  1056. "failed to update user info: ", err.Error())
  1057. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1058. return
  1059. }
  1060. var userchange models.User
  1061. // we decode our body request params
  1062. err = json.NewDecoder(r.Body).Decode(&userchange)
  1063. if err != nil {
  1064. slog.Error("failed to decode body", "error ", err.Error())
  1065. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1066. return
  1067. }
  1068. if user.UserName != userchange.UserName {
  1069. logic.ReturnErrorResponse(
  1070. w,
  1071. r,
  1072. logic.FormatError(
  1073. errors.New("user in param and request body not matching"),
  1074. "badrequest",
  1075. ),
  1076. )
  1077. return
  1078. }
  1079. selfUpdate := false
  1080. if !ismaster && caller.UserName == user.UserName {
  1081. selfUpdate = true
  1082. }
  1083. if !ismaster && !selfUpdate {
  1084. if caller.PlatformRoleID == models.AdminRole && user.PlatformRoleID == models.SuperAdminRole {
  1085. slog.Error("non-superadmin user", "caller", caller.UserName, "attempted to update superadmin user", username)
  1086. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
  1087. return
  1088. }
  1089. if caller.PlatformRoleID != models.AdminRole && caller.PlatformRoleID != models.SuperAdminRole {
  1090. slog.Error("operation not allowed", "caller", caller.UserName, "attempted to update user", username)
  1091. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
  1092. return
  1093. }
  1094. if caller.PlatformRoleID == models.AdminRole && user.PlatformRoleID == models.AdminRole {
  1095. slog.Error("an admin user does not have permissions to update another admin user", "caller", caller.UserName, "attempted to update admin user", username)
  1096. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("an admin user does not have permissions to update another admin user"), "forbidden"))
  1097. return
  1098. }
  1099. if caller.PlatformRoleID == models.AdminRole && userchange.PlatformRoleID == models.AdminRole {
  1100. err = errors.New("an admin user does not have permissions to assign the admin role to another user")
  1101. slog.Error(
  1102. "failed to update user",
  1103. "caller",
  1104. caller.UserName,
  1105. "attempted to update user",
  1106. username,
  1107. "error",
  1108. err,
  1109. )
  1110. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1111. return
  1112. }
  1113. }
  1114. if !ismaster && selfUpdate {
  1115. if user.PlatformRoleID != userchange.PlatformRoleID {
  1116. slog.Error("user cannot change his own role", "caller", caller.UserName, "attempted to update user role", username)
  1117. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user not allowed to self assign role"), "forbidden"))
  1118. return
  1119. }
  1120. if logic.IsMFAEnforced() && user.IsMFAEnabled && !userchange.IsMFAEnabled {
  1121. err = errors.New("mfa is enforced, user cannot unset their own mfa")
  1122. slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
  1123. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1124. return
  1125. }
  1126. if servercfg.IsPro {
  1127. // user cannot update his own roles and groups
  1128. if len(user.NetworkRoles) != len(userchange.NetworkRoles) || !reflect.DeepEqual(user.NetworkRoles, userchange.NetworkRoles) {
  1129. err = errors.New("user cannot update self update their network roles")
  1130. slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
  1131. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1132. return
  1133. }
  1134. // user cannot update his own roles and groups
  1135. if len(user.UserGroups) != len(userchange.UserGroups) || !reflect.DeepEqual(user.UserGroups, userchange.UserGroups) {
  1136. err = errors.New("user cannot update self update their groups")
  1137. slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
  1138. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1139. return
  1140. }
  1141. }
  1142. }
  1143. if ismaster {
  1144. if user.PlatformRoleID != models.SuperAdminRole && userchange.PlatformRoleID == models.SuperAdminRole {
  1145. slog.Error("operation not allowed", "caller", logic.MasterUser, "attempted to update user role to superadmin", username)
  1146. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("attempted to update user role to superadmin"), "forbidden"))
  1147. return
  1148. }
  1149. }
  1150. if logic.IsOauthUser(user) == nil && userchange.Password != "" {
  1151. err := fmt.Errorf("cannot update user's password for an oauth user %s", username)
  1152. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1153. return
  1154. }
  1155. logic.AddGlobalNetRolesToAdmins(&userchange)
  1156. if userchange.PlatformRoleID != user.PlatformRoleID || !logic.CompareMaps(user.UserGroups, userchange.UserGroups) {
  1157. (&schema.UserAccessToken{UserName: user.UserName}).DeleteAllUserTokens(r.Context())
  1158. }
  1159. oldUser := *user
  1160. if ismaster {
  1161. caller = &models.User{
  1162. UserName: logic.MasterUser,
  1163. }
  1164. }
  1165. action := models.Update
  1166. // TODO: here we are relying on the dashboard to only
  1167. // make singular updates, but it's possible that the
  1168. // API can be called to make multiple changes to the
  1169. // user. We should update it to log multiple events
  1170. // or create singular update APIs.
  1171. if userchange.IsMFAEnabled != user.IsMFAEnabled {
  1172. if userchange.IsMFAEnabled {
  1173. // the update API won't be used to enable MFA.
  1174. action = models.EnableMFA
  1175. } else {
  1176. action = models.DisableMFA
  1177. }
  1178. }
  1179. e := models.Event{
  1180. Action: action,
  1181. Source: models.Subject{
  1182. ID: caller.UserName,
  1183. Name: caller.UserName,
  1184. Type: models.UserSub,
  1185. },
  1186. TriggeredBy: caller.UserName,
  1187. Target: models.Subject{
  1188. ID: user.UserName,
  1189. Name: user.UserName,
  1190. Type: models.UserSub,
  1191. },
  1192. Diff: models.Diff{
  1193. Old: oldUser,
  1194. New: userchange,
  1195. },
  1196. Origin: models.Dashboard,
  1197. }
  1198. user, err = logic.UpdateUser(&userchange, user)
  1199. if err != nil {
  1200. logger.Log(0, username,
  1201. "failed to update user info: ", err.Error())
  1202. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1203. return
  1204. }
  1205. logic.LogEvent(&e)
  1206. go mq.PublishPeerUpdate(false)
  1207. logger.Log(1, username, "was updated")
  1208. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  1209. }
  1210. // @Summary Delete a user
  1211. // @Router /api/users/{username} [delete]
  1212. // @Tags Users
  1213. // @Param username path string true "Username of the user to delete"
  1214. // @Success 200 {string} string
  1215. // @Failure 500 {object} models.ErrorResponse
  1216. func deleteUser(w http.ResponseWriter, r *http.Request) {
  1217. // Set header
  1218. w.Header().Set("Content-Type", "application/json")
  1219. // get params
  1220. var params = mux.Vars(r)
  1221. caller, err := logic.GetUser(r.Header.Get("user"))
  1222. if err != nil {
  1223. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1224. }
  1225. callerUserRole, err := logic.GetRole(caller.PlatformRoleID)
  1226. if err != nil {
  1227. slog.Error("failed to get role ", "role", callerUserRole.ID, "error", err)
  1228. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1229. return
  1230. }
  1231. username := params["username"]
  1232. user, err := logic.GetUser(username)
  1233. if err != nil {
  1234. logger.Log(0, username,
  1235. "failed to update user info: ", err.Error())
  1236. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1237. return
  1238. }
  1239. userRole, err := logic.GetRole(user.PlatformRoleID)
  1240. if err != nil {
  1241. slog.Error("failed to get role ", "role", userRole.ID, "error", err)
  1242. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1243. return
  1244. }
  1245. if userRole.ID == models.SuperAdminRole {
  1246. slog.Error(
  1247. "failed to delete user: ", "user", username, "error", "superadmin cannot be deleted")
  1248. logic.ReturnErrorResponse(
  1249. w,
  1250. r,
  1251. logic.FormatError(fmt.Errorf("superadmin cannot be deleted"), "internal"),
  1252. )
  1253. return
  1254. }
  1255. if callerUserRole.ID != models.SuperAdminRole {
  1256. if callerUserRole.ID == models.AdminRole && userRole.ID == models.AdminRole {
  1257. slog.Error(
  1258. "failed to delete user: ",
  1259. "user",
  1260. username,
  1261. "error",
  1262. "admin cannot delete another admin user, including oneself",
  1263. )
  1264. logic.ReturnErrorResponse(
  1265. w,
  1266. r,
  1267. logic.FormatError(
  1268. fmt.Errorf("admin cannot delete another admin user, including oneself"),
  1269. "internal",
  1270. ),
  1271. )
  1272. return
  1273. }
  1274. }
  1275. if user.AuthType == models.OAuth || user.ExternalIdentityProviderID != "" {
  1276. err = fmt.Errorf("cannot delete idp user %s", username)
  1277. logger.Log(0, username, "failed to delete user: ", err.Error())
  1278. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1279. return
  1280. }
  1281. err = logic.DeleteUser(username)
  1282. if err != nil {
  1283. logger.Log(0, username,
  1284. "failed to delete user: ", err.Error())
  1285. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1286. return
  1287. }
  1288. logic.LogEvent(&models.Event{
  1289. Action: models.Delete,
  1290. Source: models.Subject{
  1291. ID: caller.UserName,
  1292. Name: caller.UserName,
  1293. Type: models.UserSub,
  1294. },
  1295. TriggeredBy: caller.UserName,
  1296. Target: models.Subject{
  1297. ID: user.UserName,
  1298. Name: user.UserName,
  1299. Type: models.UserSub,
  1300. },
  1301. Origin: models.Dashboard,
  1302. })
  1303. // check and delete extclient with this ownerID
  1304. go func() {
  1305. extclients, err := logic.GetAllExtClients()
  1306. if err != nil {
  1307. slog.Error("failed to get extclients", "error", err)
  1308. return
  1309. }
  1310. for _, extclient := range extclients {
  1311. if extclient.OwnerID == user.UserName {
  1312. err = logic.DeleteExtClientAndCleanup(extclient)
  1313. if err != nil {
  1314. slog.Error("failed to delete extclient",
  1315. "id", extclient.ClientID, "owner", username, "error", err)
  1316. } else {
  1317. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1318. slog.Error("error setting ext peers: " + err.Error())
  1319. }
  1320. }
  1321. }
  1322. }
  1323. _ = logic.DeleteUserInvite(user.UserName)
  1324. mq.PublishPeerUpdate(false)
  1325. if servercfg.IsDNSMode() {
  1326. logic.SetDNS()
  1327. }
  1328. }()
  1329. logger.Log(1, username, "was deleted")
  1330. json.NewEncoder(w).Encode(params["username"] + " deleted.")
  1331. }
  1332. // Called when vpn client dials in to start the auth flow and first stage is to get register URL itself
  1333. func socketHandler(w http.ResponseWriter, r *http.Request) {
  1334. // Upgrade our raw HTTP connection to a websocket based one
  1335. conn, err := upgrader.Upgrade(w, r, nil)
  1336. if err != nil {
  1337. logger.Log(0, "error during connection upgrade for node sign-in:", err.Error())
  1338. return
  1339. }
  1340. if conn == nil {
  1341. logger.Log(0, "failed to establish web-socket connection during node sign-in")
  1342. return
  1343. }
  1344. // Start handling the session
  1345. go auth.SessionHandler(conn)
  1346. }
  1347. // @Summary lists all user roles.
  1348. // @Router /api/v1/user/roles [get]
  1349. // @Tags Users
  1350. // @Param role_id query string true "roleid required to get the role details"
  1351. // @Success 200 {object} []models.UserRolePermissionTemplate
  1352. // @Failure 500 {object} models.ErrorResponse
  1353. func listRoles(w http.ResponseWriter, r *http.Request) {
  1354. var roles []models.UserRolePermissionTemplate
  1355. var err error
  1356. roles, err = logic.ListPlatformRoles()
  1357. if err != nil {
  1358. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  1359. Code: http.StatusInternalServerError,
  1360. Message: err.Error(),
  1361. })
  1362. return
  1363. }
  1364. logic.ReturnSuccessResponseWithJson(w, r, roles, "successfully fetched user roles permission templates")
  1365. }
  1366. // swagger:route POST /api/v1/user/logout user logout
  1367. //
  1368. // LogOut user.
  1369. //
  1370. // Schemes: https
  1371. //
  1372. // Security:
  1373. // oauth
  1374. //
  1375. // Responses:
  1376. // 200: userBodyResponse
  1377. func logout(w http.ResponseWriter, r *http.Request) {
  1378. // set header.
  1379. w.Header().Set("Content-Type", "application/json")
  1380. userName := r.URL.Query().Get("username")
  1381. user, err := logic.GetUser(userName)
  1382. if err != nil {
  1383. logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
  1384. return
  1385. }
  1386. var target models.SubjectType
  1387. if val := r.Header.Get("From-Ui"); val == "true" {
  1388. target = models.DashboardSub
  1389. } else {
  1390. target = models.ClientAppSub
  1391. }
  1392. if target != "" {
  1393. logic.LogEvent(&models.Event{
  1394. Action: models.LogOut,
  1395. Source: models.Subject{
  1396. ID: user.UserName,
  1397. Name: user.UserName,
  1398. Type: models.UserSub,
  1399. },
  1400. TriggeredBy: user.UserName,
  1401. Target: models.Subject{
  1402. ID: target.String(),
  1403. Name: target.String(),
  1404. Type: target,
  1405. },
  1406. Origin: models.Origin(target),
  1407. })
  1408. }
  1409. logic.ReturnSuccessResponse(w, r, "user logged out")
  1410. }