516 lines
17 KiB
Go
516 lines
17 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/coreos/go-oidc/v3/oidc"
|
|
"github.com/gin-gonic/gin"
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/oauth2"
|
|
|
|
"github.com/authorizerdev/authorizer/server/constants"
|
|
"github.com/authorizerdev/authorizer/server/cookie"
|
|
"github.com/authorizerdev/authorizer/server/db"
|
|
"github.com/authorizerdev/authorizer/server/db/models"
|
|
"github.com/authorizerdev/authorizer/server/memorystore"
|
|
"github.com/authorizerdev/authorizer/server/oauth"
|
|
"github.com/authorizerdev/authorizer/server/token"
|
|
"github.com/authorizerdev/authorizer/server/utils"
|
|
)
|
|
|
|
// OAuthCallbackHandler handles the OAuth callback for various oauth providers
|
|
func OAuthCallbackHandler() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
provider := c.Param("oauth_provider")
|
|
state := c.Request.FormValue("state")
|
|
|
|
sessionState, err := memorystore.Provider.GetState(state)
|
|
if sessionState == "" || err != nil {
|
|
log.Debug("Invalid oauth state: ", state)
|
|
c.JSON(400, gin.H{"error": "invalid oauth state"})
|
|
}
|
|
// contains random token, redirect url, role
|
|
sessionSplit := strings.Split(state, "___")
|
|
|
|
if len(sessionSplit) < 3 {
|
|
log.Debug("Unable to get redirect url from state: ", state)
|
|
c.JSON(400, gin.H{"error": "invalid redirect url"})
|
|
return
|
|
}
|
|
|
|
// remove state from store
|
|
go memorystore.Provider.RemoveState(state)
|
|
|
|
stateValue := sessionSplit[0]
|
|
redirectURL := sessionSplit[1]
|
|
inputRoles := strings.Split(sessionSplit[2], ",")
|
|
scopes := strings.Split(sessionSplit[3], ",")
|
|
|
|
user := models.User{}
|
|
code := c.Request.FormValue("code")
|
|
switch provider {
|
|
case constants.SignupMethodGoogle:
|
|
user, err = processGoogleUserInfo(code)
|
|
case constants.SignupMethodGithub:
|
|
user, err = processGithubUserInfo(code)
|
|
case constants.SignupMethodFacebook:
|
|
user, err = processFacebookUserInfo(code)
|
|
case constants.SignupMethodLinkedIn:
|
|
user, err = processLinkedInUserInfo(code)
|
|
case constants.SignupMethodApple:
|
|
user, err = processAppleUserInfo(code)
|
|
default:
|
|
log.Info("Invalid oauth provider")
|
|
err = fmt.Errorf(`invalid oauth provider`)
|
|
}
|
|
|
|
if err != nil {
|
|
log.Debug("Failed to process user info: ", err)
|
|
c.JSON(400, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
existingUser, err := db.Provider.GetUserByEmail(user.Email)
|
|
log := log.WithField("user", user.Email)
|
|
|
|
if err != nil {
|
|
isSignupDisabled, err := memorystore.Provider.GetBoolStoreEnvVariable(constants.EnvKeyDisableSignUp)
|
|
if err != nil {
|
|
log.Debug("Failed to get signup disabled env variable: ", err)
|
|
c.JSON(400, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
if isSignupDisabled {
|
|
log.Debug("Failed to signup as disabled")
|
|
c.JSON(400, gin.H{"error": "signup is disabled for this instance"})
|
|
return
|
|
}
|
|
// user not registered, register user and generate session token
|
|
user.SignupMethods = provider
|
|
// make sure inputRoles don't include protected roles
|
|
hasProtectedRole := false
|
|
for _, ir := range inputRoles {
|
|
protectedRolesString, err := memorystore.Provider.GetStringStoreEnvVariable(constants.EnvKeyProtectedRoles)
|
|
protectedRoles := []string{}
|
|
if err != nil {
|
|
log.Debug("Failed to get protected roles: ", err)
|
|
protectedRolesString = ""
|
|
} else {
|
|
protectedRoles = strings.Split(protectedRolesString, ",")
|
|
}
|
|
if utils.StringSliceContains(protectedRoles, ir) {
|
|
hasProtectedRole = true
|
|
}
|
|
}
|
|
|
|
if hasProtectedRole {
|
|
log.Debug("Signup is not allowed with protected roles:", inputRoles)
|
|
c.JSON(400, gin.H{"error": "invalid role"})
|
|
return
|
|
}
|
|
|
|
user.Roles = strings.Join(inputRoles, ",")
|
|
now := time.Now().Unix()
|
|
user.EmailVerifiedAt = &now
|
|
user, _ = db.Provider.AddUser(user)
|
|
} else {
|
|
user = existingUser
|
|
if user.RevokedTimestamp != nil {
|
|
log.Debug("User access revoked at: ", user.RevokedTimestamp)
|
|
c.JSON(400, gin.H{"error": "user access has been revoked"})
|
|
return
|
|
}
|
|
|
|
// user exists in db, check if method was google
|
|
// if not append google to existing signup method and save it
|
|
signupMethod := existingUser.SignupMethods
|
|
if !strings.Contains(signupMethod, provider) {
|
|
signupMethod = signupMethod + "," + provider
|
|
}
|
|
user.SignupMethods = signupMethod
|
|
|
|
if user.EmailVerifiedAt == nil {
|
|
now := time.Now().Unix()
|
|
user.EmailVerifiedAt = &now
|
|
}
|
|
|
|
// There multiple scenarios with roles here in social login
|
|
// 1. user has access to protected roles + roles and trying to login
|
|
// 2. user has not signed up for one of the available role but trying to signup.
|
|
// Need to modify roles in this case
|
|
|
|
// find the unassigned roles
|
|
existingRoles := strings.Split(existingUser.Roles, ",")
|
|
unasignedRoles := []string{}
|
|
for _, ir := range inputRoles {
|
|
if !utils.StringSliceContains(existingRoles, ir) {
|
|
unasignedRoles = append(unasignedRoles, ir)
|
|
}
|
|
}
|
|
|
|
if len(unasignedRoles) > 0 {
|
|
// check if it contains protected unassigned role
|
|
hasProtectedRole := false
|
|
for _, ur := range unasignedRoles {
|
|
protectedRolesString, err := memorystore.Provider.GetStringStoreEnvVariable(constants.EnvKeyProtectedRoles)
|
|
protectedRoles := []string{}
|
|
if err != nil {
|
|
log.Debug("Failed to get protected roles: ", err)
|
|
protectedRolesString = ""
|
|
} else {
|
|
protectedRoles = strings.Split(protectedRolesString, ",")
|
|
}
|
|
if utils.StringSliceContains(protectedRoles, ur) {
|
|
hasProtectedRole = true
|
|
}
|
|
}
|
|
|
|
if hasProtectedRole {
|
|
log.Debug("Invalid role. User is using protected unassigned role")
|
|
c.JSON(400, gin.H{"error": "invalid role"})
|
|
return
|
|
} else {
|
|
user.Roles = existingUser.Roles + "," + strings.Join(unasignedRoles, ",")
|
|
}
|
|
} else {
|
|
user.Roles = existingUser.Roles
|
|
}
|
|
|
|
user, err = db.Provider.UpdateUser(user)
|
|
if err != nil {
|
|
log.Debug("Failed to update user: ", err)
|
|
c.JSON(500, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
}
|
|
|
|
authToken, err := token.CreateAuthToken(c, user, inputRoles, scopes)
|
|
if err != nil {
|
|
log.Debug("Failed to create auth token: ", err)
|
|
c.JSON(500, gin.H{"error": err.Error()})
|
|
}
|
|
|
|
expiresIn := authToken.AccessToken.ExpiresAt - time.Now().Unix()
|
|
if expiresIn <= 0 {
|
|
expiresIn = 1
|
|
}
|
|
|
|
params := "access_token=" + authToken.AccessToken.Token + "&token_type=bearer&expires_in=" + strconv.FormatInt(expiresIn, 10) + "&state=" + stateValue + "&id_token=" + authToken.IDToken.Token
|
|
|
|
cookie.SetSession(c, authToken.FingerPrintHash)
|
|
memorystore.Provider.SetUserSession(user.ID, constants.TokenTypeSessionToken+"_"+authToken.FingerPrint, authToken.FingerPrintHash)
|
|
memorystore.Provider.SetUserSession(user.ID, constants.TokenTypeAccessToken+"_"+authToken.FingerPrint, authToken.AccessToken.Token)
|
|
|
|
if authToken.RefreshToken != nil {
|
|
params = params + `&refresh_token=` + authToken.RefreshToken.Token
|
|
memorystore.Provider.SetUserSession(user.ID, constants.TokenTypeRefreshToken+"_"+authToken.FingerPrint, authToken.RefreshToken.Token)
|
|
}
|
|
|
|
go db.Provider.AddSession(models.Session{
|
|
UserID: user.ID,
|
|
UserAgent: utils.GetUserAgent(c.Request),
|
|
IP: utils.GetIP(c.Request),
|
|
})
|
|
if strings.Contains(redirectURL, "?") {
|
|
redirectURL = redirectURL + "&" + params
|
|
} else {
|
|
redirectURL = redirectURL + "?" + strings.TrimPrefix(params, "&")
|
|
}
|
|
|
|
c.Redirect(http.StatusFound, redirectURL)
|
|
}
|
|
}
|
|
|
|
func processGoogleUserInfo(code string) (models.User, error) {
|
|
user := models.User{}
|
|
ctx := context.Background()
|
|
oauth2Token, err := oauth.OAuthProviders.GoogleConfig.Exchange(ctx, code)
|
|
if err != nil {
|
|
log.Debug("Failed to exchange code for token: ", err)
|
|
return user, fmt.Errorf("invalid google exchange code: %s", err.Error())
|
|
}
|
|
|
|
verifier := oauth.OIDCProviders.GoogleOIDC.Verifier(&oidc.Config{ClientID: oauth.OAuthProviders.GoogleConfig.ClientID})
|
|
|
|
// Extract the ID Token from OAuth2 token.
|
|
rawIDToken, ok := oauth2Token.Extra("id_token").(string)
|
|
if !ok {
|
|
log.Debug("Failed to extract ID Token from OAuth2 token")
|
|
return user, fmt.Errorf("unable to extract id_token")
|
|
}
|
|
|
|
// Parse and verify ID Token payload.
|
|
idToken, err := verifier.Verify(ctx, rawIDToken)
|
|
if err != nil {
|
|
log.Debug("Failed to verify ID Token: ", err)
|
|
return user, fmt.Errorf("unable to verify id_token: %s", err.Error())
|
|
}
|
|
|
|
if err := idToken.Claims(&user); err != nil {
|
|
log.Debug("Failed to parse ID Token claims: ", err)
|
|
return user, fmt.Errorf("unable to extract claims")
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func processGithubUserInfo(code string) (models.User, error) {
|
|
user := models.User{}
|
|
oauth2Token, err := oauth.OAuthProviders.GithubConfig.Exchange(oauth2.NoContext, code)
|
|
if err != nil {
|
|
log.Debug("Failed to exchange code for token: ", err)
|
|
return user, fmt.Errorf("invalid github exchange code: %s", err.Error())
|
|
}
|
|
client := http.Client{}
|
|
req, err := http.NewRequest("GET", constants.GithubUserInfoURL, nil)
|
|
if err != nil {
|
|
log.Debug("Failed to create github user info request: ", err)
|
|
return user, fmt.Errorf("error creating github user info request: %s", err.Error())
|
|
}
|
|
req.Header = http.Header{
|
|
"Authorization": []string{fmt.Sprintf("token %s", oauth2Token.AccessToken)},
|
|
}
|
|
|
|
response, err := client.Do(req)
|
|
if err != nil {
|
|
log.Debug("Failed to request github user info: ", err)
|
|
return user, err
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
body, err := ioutil.ReadAll(response.Body)
|
|
if err != nil {
|
|
log.Debug("Failed to read github user info response body: ", err)
|
|
return user, fmt.Errorf("failed to read github response body: %s", err.Error())
|
|
}
|
|
if response.StatusCode >= 400 {
|
|
log.Debug("Failed to request github user info: ", string(body))
|
|
return user, fmt.Errorf("failed to request github user info: %s", string(body))
|
|
}
|
|
|
|
userRawData := make(map[string]string)
|
|
json.Unmarshal(body, &userRawData)
|
|
|
|
name := strings.Split(userRawData["name"], " ")
|
|
firstName := ""
|
|
lastName := ""
|
|
if len(name) >= 1 && strings.TrimSpace(name[0]) != "" {
|
|
firstName = name[0]
|
|
}
|
|
if len(name) > 1 && strings.TrimSpace(name[1]) != "" {
|
|
lastName = name[0]
|
|
}
|
|
|
|
picture := userRawData["avatar_url"]
|
|
|
|
user = models.User{
|
|
GivenName: &firstName,
|
|
FamilyName: &lastName,
|
|
Picture: &picture,
|
|
Email: userRawData["email"],
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func processFacebookUserInfo(code string) (models.User, error) {
|
|
user := models.User{}
|
|
oauth2Token, err := oauth.OAuthProviders.FacebookConfig.Exchange(oauth2.NoContext, code)
|
|
if err != nil {
|
|
log.Debug("Invalid facebook exchange code: ", err)
|
|
return user, fmt.Errorf("invalid facebook exchange code: %s", err.Error())
|
|
}
|
|
client := http.Client{}
|
|
req, err := http.NewRequest("GET", constants.FacebookUserInfoURL+oauth2Token.AccessToken, nil)
|
|
if err != nil {
|
|
log.Debug("Error creating facebook user info request: ", err)
|
|
return user, fmt.Errorf("error creating facebook user info request: %s", err.Error())
|
|
}
|
|
|
|
response, err := client.Do(req)
|
|
if err != nil {
|
|
log.Debug("Failed to process facebook user: ", err)
|
|
return user, err
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
body, err := ioutil.ReadAll(response.Body)
|
|
if err != nil {
|
|
log.Debug("Failed to read facebook response: ", err)
|
|
return user, fmt.Errorf("failed to read facebook response body: %s", err.Error())
|
|
}
|
|
if response.StatusCode >= 400 {
|
|
log.Debug("Failed to request facebook user info: ", string(body))
|
|
return user, fmt.Errorf("failed to request facebook user info: %s", string(body))
|
|
}
|
|
userRawData := make(map[string]interface{})
|
|
json.Unmarshal(body, &userRawData)
|
|
|
|
email := fmt.Sprintf("%v", userRawData["sub"])
|
|
|
|
picObject := userRawData["picture"].(map[string]interface{})["data"]
|
|
picDataObject := picObject.(map[string]interface{})
|
|
firstName := fmt.Sprintf("%v", userRawData["first_name"])
|
|
lastName := fmt.Sprintf("%v", userRawData["last_name"])
|
|
picture := fmt.Sprintf("%v", picDataObject["url"])
|
|
|
|
user = models.User{
|
|
GivenName: &firstName,
|
|
FamilyName: &lastName,
|
|
Picture: &picture,
|
|
Email: email,
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func processLinkedInUserInfo(code string) (models.User, error) {
|
|
user := models.User{}
|
|
oauth2Token, err := oauth.OAuthProviders.LinkedInConfig.Exchange(oauth2.NoContext, code)
|
|
if err != nil {
|
|
log.Debug("Failed to exchange code for token: ", err)
|
|
return user, fmt.Errorf("invalid linkedin exchange code: %s", err.Error())
|
|
}
|
|
|
|
client := http.Client{}
|
|
req, err := http.NewRequest("GET", constants.LinkedInUserInfoURL, nil)
|
|
if err != nil {
|
|
log.Debug("Failed to create linkedin user info request: ", err)
|
|
return user, fmt.Errorf("error creating linkedin user info request: %s", err.Error())
|
|
}
|
|
req.Header = http.Header{
|
|
"Authorization": []string{fmt.Sprintf("Bearer %s", oauth2Token.AccessToken)},
|
|
}
|
|
|
|
response, err := client.Do(req)
|
|
if err != nil {
|
|
log.Debug("Failed to request linkedin user info: ", err)
|
|
return user, err
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
body, err := ioutil.ReadAll(response.Body)
|
|
if err != nil {
|
|
log.Debug("Failed to read linkedin user info response body: ", err)
|
|
return user, fmt.Errorf("failed to read linkedin response body: %s", err.Error())
|
|
}
|
|
|
|
if response.StatusCode >= 400 {
|
|
log.Debug("Failed to request linkedin user info: ", string(body))
|
|
return user, fmt.Errorf("failed to request linkedin user info: %s", string(body))
|
|
}
|
|
|
|
userRawData := make(map[string]interface{})
|
|
json.Unmarshal(body, &userRawData)
|
|
|
|
req, err = http.NewRequest("GET", constants.LinkedInEmailURL, nil)
|
|
if err != nil {
|
|
log.Debug("Failed to create linkedin email info request: ", err)
|
|
return user, fmt.Errorf("error creating linkedin user info request: %s", err.Error())
|
|
}
|
|
req.Header = http.Header{
|
|
"Authorization": []string{fmt.Sprintf("Bearer %s", oauth2Token.AccessToken)},
|
|
}
|
|
|
|
response, err = client.Do(req)
|
|
if err != nil {
|
|
log.Debug("Failed to request linkedin email info: ", err)
|
|
return user, err
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
body, err = ioutil.ReadAll(response.Body)
|
|
if err != nil {
|
|
log.Debug("Failed to read linkedin email info response body: ", err)
|
|
return user, fmt.Errorf("failed to read linkedin email response body: %s", err.Error())
|
|
}
|
|
if response.StatusCode >= 400 {
|
|
log.Debug("Failed to request linkedin user info: ", string(body))
|
|
return user, fmt.Errorf("failed to request linkedin user info: %s", string(body))
|
|
}
|
|
emailRawData := make(map[string]interface{})
|
|
json.Unmarshal(body, &emailRawData)
|
|
|
|
firstName := userRawData["localizedFirstName"].(string)
|
|
lastName := userRawData["localizedLastName"].(string)
|
|
profilePicture := userRawData["profilePicture"].(map[string]interface{})["displayImage~"].(map[string]interface{})["elements"].([]interface{})[0].(map[string]interface{})["identifiers"].([]interface{})[0].(map[string]interface{})["identifier"].(string)
|
|
emailAddress := emailRawData["elements"].([]interface{})[0].(map[string]interface{})["handle~"].(map[string]interface{})["emailAddress"].(string)
|
|
|
|
user = models.User{
|
|
GivenName: &firstName,
|
|
FamilyName: &lastName,
|
|
Picture: &profilePicture,
|
|
Email: emailAddress,
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func processAppleUserInfo(code string) (models.User, error) {
|
|
user := models.User{}
|
|
fmt.Println("=> code:", code)
|
|
oauth2Token, err := oauth.OAuthProviders.AppleConfig.Exchange(oauth2.NoContext, code)
|
|
if err != nil {
|
|
log.Debug("Failed to exchange code for token: ", err)
|
|
return user, fmt.Errorf("invalid apple exchange code: %s", err.Error())
|
|
}
|
|
|
|
fmt.Println("=> oauth2Token:", oauth2Token)
|
|
|
|
// Extract the ID Token from OAuth2 token.
|
|
rawIDToken, ok := oauth2Token.Extra("id_token").(string)
|
|
if !ok {
|
|
log.Debug("Failed to extract ID Token from OAuth2 token")
|
|
return user, fmt.Errorf("unable to extract id_token")
|
|
}
|
|
|
|
tokenSplit := strings.Split(rawIDToken, ".")
|
|
claimsData := tokenSplit[1]
|
|
decodedClaimsData, err := base64.RawURLEncoding.DecodeString(claimsData)
|
|
if err != nil {
|
|
log.Debugf("Failed to decrypt claims %s: %s", claimsData, err.Error())
|
|
return user, fmt.Errorf("failed to decrypt claims data: %s", err.Error())
|
|
}
|
|
|
|
fmt.Println("=> decodedClaimsData:", string(decodedClaimsData))
|
|
|
|
claims := make(map[string]interface{})
|
|
err = json.Unmarshal(decodedClaimsData, &claims)
|
|
if err != nil {
|
|
log.Debug("Failed to unmarshal claims data: ", err)
|
|
return user, fmt.Errorf("failed to unmarshal claims data: %s", err.Error())
|
|
}
|
|
|
|
fmt.Println("=> claims:", claims)
|
|
|
|
if val, ok := claims["email"]; !ok {
|
|
log.Debug("Failed to extract email from claims.")
|
|
return user, fmt.Errorf("unable to extract email, please check the scopes enabled for your app. It needs `email`, `name` scopes")
|
|
} else {
|
|
user.Email = val.(string)
|
|
}
|
|
|
|
if val, ok := claims["name"]; ok {
|
|
nameData := val.(map[string]interface{})
|
|
if nameVal, ok := nameData["firstName"]; ok {
|
|
givenName := nameVal.(string)
|
|
user.GivenName = &givenName
|
|
}
|
|
|
|
if nameVal, ok := nameData["lastName"]; ok {
|
|
familyName := nameVal.(string)
|
|
user.FamilyName = &familyName
|
|
}
|
|
}
|
|
|
|
return user, err
|
|
}
|