user.go 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528
  1. package controller
  2. import (
  3. "bytes"
  4. "encoding/base64"
  5. "encoding/json"
  6. "errors"
  7. "fmt"
  8. "image/png"
  9. "net/http"
  10. "reflect"
  11. "time"
  12. "github.com/pquerna/otp"
  13. "golang.org/x/crypto/bcrypt"
  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. user.NumAccessTokens, _ = (&schema.UserAccessToken{
  811. UserName: user.UserName,
  812. }).CountByUser(r.Context())
  813. userRoleTemplate, err := logic.GetRole(user.PlatformRoleID)
  814. if err != nil {
  815. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  816. return
  817. }
  818. resp := models.ReturnUserWithRolesAndGroups{
  819. ReturnUser: user,
  820. PlatformRole: userRoleTemplate,
  821. UserGroups: map[models.UserGroupID]models.UserGroup{},
  822. }
  823. for gId := range user.UserGroups {
  824. grp, err := logic.GetUserGroup(gId)
  825. if err != nil {
  826. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  827. return
  828. }
  829. resp.UserGroups[gId] = grp
  830. }
  831. logger.Log(2, r.Header.Get("user"), "fetched user", usernameFetched)
  832. logic.ReturnSuccessResponseWithJson(w, r, resp, "fetched user with role info")
  833. }
  834. // swagger:route GET /api/users user getUsers
  835. //
  836. // Get all users.
  837. //
  838. // Schemes: https
  839. //
  840. // Security:
  841. // oauth
  842. //
  843. // Responses:
  844. // 200: userBodyResponse
  845. func getUsers(w http.ResponseWriter, r *http.Request) {
  846. // set header.
  847. w.Header().Set("Content-Type", "application/json")
  848. users, err := logic.GetUsers()
  849. if err != nil {
  850. logger.Log(0, "failed to fetch users: ", err.Error())
  851. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  852. return
  853. }
  854. for i := range users {
  855. users[i].NumAccessTokens, _ = (&schema.UserAccessToken{
  856. UserName: users[i].UserName,
  857. }).CountByUser(r.Context())
  858. }
  859. logic.SortUsers(users[:])
  860. logger.Log(2, r.Header.Get("user"), "fetched users")
  861. json.NewEncoder(w).Encode(users)
  862. }
  863. // @Summary Create a super admin
  864. // @Router /api/users/adm/createsuperadmin [post]
  865. // @Tags Users
  866. // @Param body body models.User true "User details"
  867. // @Success 200 {object} models.User
  868. // @Failure 400 {object} models.ErrorResponse
  869. // @Failure 500 {object} models.ErrorResponse
  870. func createSuperAdmin(w http.ResponseWriter, r *http.Request) {
  871. w.Header().Set("Content-Type", "application/json")
  872. var u models.User
  873. err := json.NewDecoder(r.Body).Decode(&u)
  874. if err != nil {
  875. slog.Error("error decoding request body", "error", err.Error())
  876. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  877. return
  878. }
  879. if !logic.IsBasicAuthEnabled() {
  880. logic.ReturnErrorResponse(
  881. w,
  882. r,
  883. logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"),
  884. )
  885. return
  886. }
  887. err = logic.CreateSuperAdmin(&u)
  888. if err != nil {
  889. slog.Error("failed to create admin", "error", err.Error())
  890. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  891. return
  892. }
  893. logger.Log(1, u.UserName, "was made a super admin")
  894. json.NewEncoder(w).Encode(logic.ToReturnUser(u))
  895. }
  896. // @Summary Transfer super admin role to another admin user
  897. // @Router /api/users/adm/transfersuperadmin/{username} [post]
  898. // @Tags Users
  899. // @Param username path string true "Username of the user to transfer super admin role"
  900. // @Success 200 {object} models.User
  901. // @Failure 403 {object} models.ErrorResponse
  902. // @Failure 500 {object} models.ErrorResponse
  903. func transferSuperAdmin(w http.ResponseWriter, r *http.Request) {
  904. w.Header().Set("Content-Type", "application/json")
  905. caller, err := logic.GetUser(r.Header.Get("user"))
  906. if err != nil {
  907. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  908. }
  909. if caller.PlatformRoleID != models.SuperAdminRole {
  910. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only superadmin can assign the superadmin role to another user"), "forbidden"))
  911. return
  912. }
  913. var params = mux.Vars(r)
  914. username := params["username"]
  915. u, err := logic.GetUser(username)
  916. if err != nil {
  917. slog.Error("error getting user", "user", u.UserName, "error", err.Error())
  918. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  919. return
  920. }
  921. if u.PlatformRoleID != models.AdminRole {
  922. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("only admins can be promoted to superadmin role"), "forbidden"))
  923. return
  924. }
  925. if !logic.IsBasicAuthEnabled() {
  926. logic.ReturnErrorResponse(
  927. w,
  928. r,
  929. logic.FormatError(fmt.Errorf("basic auth is disabled"), "badrequest"),
  930. )
  931. return
  932. }
  933. u.PlatformRoleID = models.SuperAdminRole
  934. err = logic.UpsertUser(*u)
  935. if err != nil {
  936. slog.Error("error updating user to superadmin: ", "user", u.UserName, "error", err.Error())
  937. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  938. return
  939. }
  940. caller.PlatformRoleID = models.AdminRole
  941. err = logic.UpsertUser(*caller)
  942. if err != nil {
  943. slog.Error("error demoting user to admin: ", "user", caller.UserName, "error", err.Error())
  944. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  945. return
  946. }
  947. slog.Info("user was made a super admin", "user", u.UserName)
  948. json.NewEncoder(w).Encode(logic.ToReturnUser(*u))
  949. }
  950. // @Summary Create a user
  951. // @Router /api/users/{username} [post]
  952. // @Tags Users
  953. // @Param username path string true "Username of the user to create"
  954. // @Param body body models.User true "User details"
  955. // @Success 200 {object} models.User
  956. // @Failure 400 {object} models.ErrorResponse
  957. // @Failure 403 {object} models.ErrorResponse
  958. // @Failure 500 {object} models.ErrorResponse
  959. func createUser(w http.ResponseWriter, r *http.Request) {
  960. w.Header().Set("Content-Type", "application/json")
  961. caller, err := logic.GetUser(r.Header.Get("user"))
  962. if err != nil {
  963. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  964. return
  965. }
  966. var user models.User
  967. err = json.NewDecoder(r.Body).Decode(&user)
  968. if err != nil {
  969. logger.Log(0, user.UserName, "error decoding request body: ",
  970. err.Error())
  971. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  972. return
  973. }
  974. if !servercfg.IsPro {
  975. user.PlatformRoleID = models.AdminRole
  976. }
  977. if user.UserName == logic.MasterUser {
  978. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("username not allowed"), "badrequest"))
  979. return
  980. }
  981. if user.PlatformRoleID == "" {
  982. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("platform role is missing"), "badrequest"))
  983. return
  984. }
  985. userRole, err := logic.GetRole(user.PlatformRoleID)
  986. if err != nil {
  987. err = errors.New("error fetching role " + user.PlatformRoleID.String() + " " + err.Error())
  988. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  989. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  990. return
  991. }
  992. if userRole.ID == models.SuperAdminRole {
  993. err = errors.New("additional superadmins cannot be created")
  994. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  995. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  996. return
  997. }
  998. if caller.PlatformRoleID != models.SuperAdminRole && user.PlatformRoleID == models.AdminRole {
  999. err = errors.New("only superadmin can create admin users")
  1000. slog.Error("error creating new user: ", "user", user.UserName, "error", err)
  1001. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1002. return
  1003. }
  1004. if !servercfg.IsPro && user.PlatformRoleID != models.AdminRole {
  1005. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("non-admins users can only be created on Pro version"), "forbidden"))
  1006. return
  1007. }
  1008. err = logic.CreateUser(&user)
  1009. if err != nil {
  1010. slog.Error("error creating new user: ", "user", user.UserName, "error", err.Error())
  1011. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1012. return
  1013. }
  1014. logic.LogEvent(&models.Event{
  1015. Action: models.Create,
  1016. Source: models.Subject{
  1017. ID: caller.UserName,
  1018. Name: caller.UserName,
  1019. Type: models.UserSub,
  1020. },
  1021. TriggeredBy: caller.UserName,
  1022. Target: models.Subject{
  1023. ID: user.UserName,
  1024. Name: user.UserName,
  1025. Type: models.UserSub,
  1026. Info: user,
  1027. },
  1028. Origin: models.Dashboard,
  1029. })
  1030. logic.DeleteUserInvite(user.UserName)
  1031. logic.DeletePendingUser(user.UserName)
  1032. go mq.PublishPeerUpdate(false)
  1033. slog.Info("user was created", "username", user.UserName)
  1034. json.NewEncoder(w).Encode(logic.ToReturnUser(user))
  1035. }
  1036. // @Summary Update a user
  1037. // @Router /api/users/{username} [put]
  1038. // @Tags Users
  1039. // @Param username path string true "Username of the user to update"
  1040. // @Param body body models.User true "User details"
  1041. // @Success 200 {object} models.User
  1042. // @Failure 400 {object} models.ErrorResponse
  1043. // @Failure 403 {object} models.ErrorResponse
  1044. // @Failure 500 {object} models.ErrorResponse
  1045. func updateUser(w http.ResponseWriter, r *http.Request) {
  1046. w.Header().Set("Content-Type", "application/json")
  1047. var params = mux.Vars(r)
  1048. // start here
  1049. var caller *models.User
  1050. var err error
  1051. var ismaster bool
  1052. if r.Header.Get("user") == logic.MasterUser {
  1053. ismaster = true
  1054. } else {
  1055. caller, err = logic.GetUser(r.Header.Get("user"))
  1056. if err != nil {
  1057. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1058. }
  1059. }
  1060. username := params["username"]
  1061. user, err := logic.GetUser(username)
  1062. if err != nil {
  1063. logger.Log(0, username,
  1064. "failed to update user info: ", err.Error())
  1065. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1066. return
  1067. }
  1068. var userchange models.User
  1069. // we decode our body request params
  1070. err = json.NewDecoder(r.Body).Decode(&userchange)
  1071. if err != nil {
  1072. slog.Error("failed to decode body", "error ", err.Error())
  1073. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1074. return
  1075. }
  1076. if user.UserName != userchange.UserName {
  1077. logic.ReturnErrorResponse(
  1078. w,
  1079. r,
  1080. logic.FormatError(
  1081. errors.New("user in param and request body not matching"),
  1082. "badrequest",
  1083. ),
  1084. )
  1085. return
  1086. }
  1087. selfUpdate := false
  1088. if !ismaster && caller.UserName == user.UserName {
  1089. selfUpdate = true
  1090. }
  1091. if !ismaster && !selfUpdate {
  1092. if caller.PlatformRoleID == models.AdminRole && user.PlatformRoleID == models.SuperAdminRole {
  1093. slog.Error("non-superadmin user", "caller", caller.UserName, "attempted to update superadmin user", username)
  1094. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
  1095. return
  1096. }
  1097. if caller.PlatformRoleID != models.AdminRole && caller.PlatformRoleID != models.SuperAdminRole {
  1098. slog.Error("operation not allowed", "caller", caller.UserName, "attempted to update user", username)
  1099. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("cannot update superadmin user"), "forbidden"))
  1100. return
  1101. }
  1102. if caller.PlatformRoleID == models.AdminRole && user.PlatformRoleID == models.AdminRole {
  1103. slog.Error("an admin user does not have permissions to update another admin user", "caller", caller.UserName, "attempted to update admin user", username)
  1104. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("an admin user does not have permissions to update another admin user"), "forbidden"))
  1105. return
  1106. }
  1107. if caller.PlatformRoleID == models.AdminRole && userchange.PlatformRoleID == models.AdminRole {
  1108. err = errors.New("an admin user does not have permissions to assign the admin role to another user")
  1109. slog.Error(
  1110. "failed to update user",
  1111. "caller",
  1112. caller.UserName,
  1113. "attempted to update user",
  1114. username,
  1115. "error",
  1116. err,
  1117. )
  1118. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1119. return
  1120. }
  1121. }
  1122. if !ismaster && selfUpdate {
  1123. if user.PlatformRoleID != userchange.PlatformRoleID {
  1124. slog.Error("user cannot change his own role", "caller", caller.UserName, "attempted to update user role", username)
  1125. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("user not allowed to self assign role"), "forbidden"))
  1126. return
  1127. }
  1128. if logic.IsMFAEnforced() && user.IsMFAEnabled && !userchange.IsMFAEnabled {
  1129. err = errors.New("mfa is enforced, user cannot unset their own mfa")
  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. if servercfg.IsPro {
  1135. // user cannot update his own roles and groups
  1136. if len(user.NetworkRoles) != len(userchange.NetworkRoles) || !reflect.DeepEqual(user.NetworkRoles, userchange.NetworkRoles) {
  1137. err = errors.New("user cannot update self update their network roles")
  1138. slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
  1139. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1140. return
  1141. }
  1142. // user cannot update his own roles and groups
  1143. if len(user.UserGroups) != len(userchange.UserGroups) || !reflect.DeepEqual(user.UserGroups, userchange.UserGroups) {
  1144. err = errors.New("user cannot update self update their groups")
  1145. slog.Error("failed to update user", "caller", caller.UserName, "attempted to update user", username, "error", err)
  1146. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1147. return
  1148. }
  1149. }
  1150. }
  1151. if ismaster {
  1152. if user.PlatformRoleID != models.SuperAdminRole && userchange.PlatformRoleID == models.SuperAdminRole {
  1153. slog.Error("operation not allowed", "caller", logic.MasterUser, "attempted to update user role to superadmin", username)
  1154. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("attempted to update user role to superadmin"), "forbidden"))
  1155. return
  1156. }
  1157. }
  1158. if logic.IsOauthUser(user) == nil && userchange.Password != "" {
  1159. err := fmt.Errorf("cannot update user's password for an oauth user %s", username)
  1160. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "forbidden"))
  1161. return
  1162. }
  1163. logic.AddGlobalNetRolesToAdmins(&userchange)
  1164. if userchange.PlatformRoleID != user.PlatformRoleID || !logic.CompareMaps(user.UserGroups, userchange.UserGroups) {
  1165. (&schema.UserAccessToken{UserName: user.UserName}).DeleteAllUserTokens(r.Context())
  1166. }
  1167. oldUser := *user
  1168. if ismaster {
  1169. caller = &models.User{
  1170. UserName: logic.MasterUser,
  1171. }
  1172. }
  1173. action := models.Update
  1174. // TODO: here we are relying on the dashboard to only
  1175. // make singular updates, but it's possible that the
  1176. // API can be called to make multiple changes to the
  1177. // user. We should update it to log multiple events
  1178. // or create singular update APIs.
  1179. if userchange.IsMFAEnabled != user.IsMFAEnabled {
  1180. if userchange.IsMFAEnabled {
  1181. // the update API won't be used to enable MFA.
  1182. action = models.EnableMFA
  1183. } else {
  1184. action = models.DisableMFA
  1185. }
  1186. }
  1187. e := models.Event{
  1188. Action: action,
  1189. Source: models.Subject{
  1190. ID: caller.UserName,
  1191. Name: caller.UserName,
  1192. Type: models.UserSub,
  1193. },
  1194. TriggeredBy: caller.UserName,
  1195. Target: models.Subject{
  1196. ID: user.UserName,
  1197. Name: user.UserName,
  1198. Type: models.UserSub,
  1199. },
  1200. Diff: models.Diff{
  1201. Old: oldUser,
  1202. New: userchange,
  1203. },
  1204. Origin: models.Dashboard,
  1205. }
  1206. user, err = logic.UpdateUser(&userchange, user)
  1207. if err != nil {
  1208. logger.Log(0, username,
  1209. "failed to update user info: ", err.Error())
  1210. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1211. return
  1212. }
  1213. logic.LogEvent(&e)
  1214. go mq.PublishPeerUpdate(false)
  1215. logger.Log(1, username, "was updated")
  1216. json.NewEncoder(w).Encode(logic.ToReturnUser(*user))
  1217. }
  1218. // @Summary Delete a user
  1219. // @Router /api/users/{username} [delete]
  1220. // @Tags Users
  1221. // @Param username path string true "Username of the user to delete"
  1222. // @Success 200 {string} string
  1223. // @Failure 500 {object} models.ErrorResponse
  1224. func deleteUser(w http.ResponseWriter, r *http.Request) {
  1225. // Set header
  1226. w.Header().Set("Content-Type", "application/json")
  1227. // get params
  1228. var params = mux.Vars(r)
  1229. caller, err := logic.GetUser(r.Header.Get("user"))
  1230. if err != nil {
  1231. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1232. }
  1233. callerUserRole, err := logic.GetRole(caller.PlatformRoleID)
  1234. if err != nil {
  1235. slog.Error("failed to get role ", "role", callerUserRole.ID, "error", err)
  1236. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1237. return
  1238. }
  1239. username := params["username"]
  1240. user, err := logic.GetUser(username)
  1241. if err != nil {
  1242. logger.Log(0, username,
  1243. "failed to update user info: ", err.Error())
  1244. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1245. return
  1246. }
  1247. userRole, err := logic.GetRole(user.PlatformRoleID)
  1248. if err != nil {
  1249. slog.Error("failed to get role ", "role", userRole.ID, "error", err)
  1250. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1251. return
  1252. }
  1253. if userRole.ID == models.SuperAdminRole {
  1254. slog.Error(
  1255. "failed to delete user: ", "user", username, "error", "superadmin cannot be deleted")
  1256. logic.ReturnErrorResponse(
  1257. w,
  1258. r,
  1259. logic.FormatError(fmt.Errorf("superadmin cannot be deleted"), "internal"),
  1260. )
  1261. return
  1262. }
  1263. if callerUserRole.ID != models.SuperAdminRole {
  1264. if callerUserRole.ID == models.AdminRole && userRole.ID == models.AdminRole {
  1265. slog.Error(
  1266. "failed to delete user: ",
  1267. "user",
  1268. username,
  1269. "error",
  1270. "admin cannot delete another admin user, including oneself",
  1271. )
  1272. logic.ReturnErrorResponse(
  1273. w,
  1274. r,
  1275. logic.FormatError(
  1276. fmt.Errorf("admin cannot delete another admin user, including oneself"),
  1277. "internal",
  1278. ),
  1279. )
  1280. return
  1281. }
  1282. }
  1283. if user.AuthType == models.OAuth || user.ExternalIdentityProviderID != "" {
  1284. err = fmt.Errorf("cannot delete idp user %s", username)
  1285. logger.Log(0, username, "failed to delete user: ", err.Error())
  1286. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1287. return
  1288. }
  1289. err = logic.DeleteUser(username)
  1290. if err != nil {
  1291. logger.Log(0, username,
  1292. "failed to delete user: ", err.Error())
  1293. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1294. return
  1295. }
  1296. logic.LogEvent(&models.Event{
  1297. Action: models.Delete,
  1298. Source: models.Subject{
  1299. ID: caller.UserName,
  1300. Name: caller.UserName,
  1301. Type: models.UserSub,
  1302. },
  1303. TriggeredBy: caller.UserName,
  1304. Target: models.Subject{
  1305. ID: user.UserName,
  1306. Name: user.UserName,
  1307. Type: models.UserSub,
  1308. },
  1309. Origin: models.Dashboard,
  1310. })
  1311. // check and delete extclient with this ownerID
  1312. go func() {
  1313. extclients, err := logic.GetAllExtClients()
  1314. if err != nil {
  1315. slog.Error("failed to get extclients", "error", err)
  1316. return
  1317. }
  1318. for _, extclient := range extclients {
  1319. if extclient.OwnerID == user.UserName {
  1320. err = logic.DeleteExtClientAndCleanup(extclient)
  1321. if err != nil {
  1322. slog.Error("failed to delete extclient",
  1323. "id", extclient.ClientID, "owner", username, "error", err)
  1324. } else {
  1325. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1326. slog.Error("error setting ext peers: " + err.Error())
  1327. }
  1328. }
  1329. }
  1330. }
  1331. _ = logic.DeleteUserInvite(user.UserName)
  1332. mq.PublishPeerUpdate(false)
  1333. if servercfg.IsDNSMode() {
  1334. logic.SetDNS()
  1335. }
  1336. }()
  1337. logger.Log(1, username, "was deleted")
  1338. json.NewEncoder(w).Encode(params["username"] + " deleted.")
  1339. }
  1340. // Called when vpn client dials in to start the auth flow and first stage is to get register URL itself
  1341. func socketHandler(w http.ResponseWriter, r *http.Request) {
  1342. // Upgrade our raw HTTP connection to a websocket based one
  1343. conn, err := upgrader.Upgrade(w, r, nil)
  1344. if err != nil {
  1345. logger.Log(0, "error during connection upgrade for node sign-in:", err.Error())
  1346. return
  1347. }
  1348. if conn == nil {
  1349. logger.Log(0, "failed to establish web-socket connection during node sign-in")
  1350. return
  1351. }
  1352. // Start handling the session
  1353. go auth.SessionHandler(conn)
  1354. }
  1355. // @Summary lists all user roles.
  1356. // @Router /api/v1/user/roles [get]
  1357. // @Tags Users
  1358. // @Param role_id query string true "roleid required to get the role details"
  1359. // @Success 200 {object} []models.UserRolePermissionTemplate
  1360. // @Failure 500 {object} models.ErrorResponse
  1361. func listRoles(w http.ResponseWriter, r *http.Request) {
  1362. var roles []models.UserRolePermissionTemplate
  1363. var err error
  1364. roles, err = logic.ListPlatformRoles()
  1365. if err != nil {
  1366. logic.ReturnErrorResponse(w, r, models.ErrorResponse{
  1367. Code: http.StatusInternalServerError,
  1368. Message: err.Error(),
  1369. })
  1370. return
  1371. }
  1372. logic.ReturnSuccessResponseWithJson(w, r, roles, "successfully fetched user roles permission templates")
  1373. }
  1374. // swagger:route POST /api/v1/user/logout user logout
  1375. //
  1376. // LogOut user.
  1377. //
  1378. // Schemes: https
  1379. //
  1380. // Security:
  1381. // oauth
  1382. //
  1383. // Responses:
  1384. // 200: userBodyResponse
  1385. func logout(w http.ResponseWriter, r *http.Request) {
  1386. // set header.
  1387. w.Header().Set("Content-Type", "application/json")
  1388. userName := r.URL.Query().Get("username")
  1389. user, err := logic.GetUser(userName)
  1390. if err != nil {
  1391. logic.ReturnErrorResponse(w, r, logic.FormatError(err, logic.BadReq))
  1392. return
  1393. }
  1394. var target models.SubjectType
  1395. if val := r.Header.Get("From-Ui"); val == "true" {
  1396. target = models.DashboardSub
  1397. } else {
  1398. target = models.ClientAppSub
  1399. }
  1400. if target != "" {
  1401. logic.LogEvent(&models.Event{
  1402. Action: models.LogOut,
  1403. Source: models.Subject{
  1404. ID: user.UserName,
  1405. Name: user.UserName,
  1406. Type: models.UserSub,
  1407. },
  1408. TriggeredBy: user.UserName,
  1409. Target: models.Subject{
  1410. ID: target.String(),
  1411. Name: target.String(),
  1412. Type: target,
  1413. },
  1414. Origin: models.Origin(target),
  1415. })
  1416. }
  1417. logic.ReturnSuccessResponse(w, r, "user logged out")
  1418. }