|
|
|
@@ -15,7 +15,9 @@ import (
|
|
|
|
|
"github.com/authorizerdev/authorizer/server/cookie"
|
|
|
|
|
"github.com/authorizerdev/authorizer/server/db"
|
|
|
|
|
"github.com/authorizerdev/authorizer/server/memorystore"
|
|
|
|
|
"github.com/authorizerdev/authorizer/server/parsers"
|
|
|
|
|
"github.com/authorizerdev/authorizer/server/token"
|
|
|
|
|
"github.com/authorizerdev/authorizer/server/utils"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// AuthorizeHandler is the handler for the /authorize route
|
|
|
|
@@ -42,6 +44,7 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
scopeString := strings.TrimSpace(gc.Query("scope"))
|
|
|
|
|
clientID := strings.TrimSpace(gc.Query("client_id"))
|
|
|
|
|
responseMode := strings.TrimSpace(gc.Query("response_mode"))
|
|
|
|
|
nonce := strings.TrimSpace(gc.Query("nonce"))
|
|
|
|
|
|
|
|
|
|
var scope []string
|
|
|
|
|
if scopeString == "" {
|
|
|
|
@@ -64,10 +67,15 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
|
|
|
|
|
if err := validateAuthorizeRequest(responseType, responseMode, clientID, state, codeChallenge); err != nil {
|
|
|
|
|
log.Debug("invalid authorization request: ", err)
|
|
|
|
|
gc.JSON(http.StatusBadRequest, gin.H{"error": err})
|
|
|
|
|
gc.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
code := uuid.New().String()
|
|
|
|
|
if nonce == "" {
|
|
|
|
|
nonce = uuid.New().String()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
log := log.WithFields(log.Fields{
|
|
|
|
|
"response_mode": responseMode,
|
|
|
|
|
"response_type": responseType,
|
|
|
|
@@ -75,23 +83,48 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
"code_challenge": codeChallenge,
|
|
|
|
|
"scope": scope,
|
|
|
|
|
"redirect_uri": redirectURI,
|
|
|
|
|
"nonce": nonce,
|
|
|
|
|
"code": code,
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
memorystore.Provider.SetState(codeChallenge, code)
|
|
|
|
|
|
|
|
|
|
// used for response mode query or fragment
|
|
|
|
|
loginState := "state=" + state + "&scope=" + strings.Join(scope, " ") + "&redirect_uri=" + redirectURI
|
|
|
|
|
loginState := "state=" + state + "&scope=" + strings.Join(scope, " ") + "&redirect_uri=" + redirectURI + "&code=" + code + "&nonce=" + nonce
|
|
|
|
|
loginURL := "/app?" + loginState
|
|
|
|
|
|
|
|
|
|
if responseMode == constants.ResponseModeFragment {
|
|
|
|
|
loginURL = "/app#" + loginState
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if state == "" {
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, map[string]interface{}{
|
|
|
|
|
"type": "authorization_response",
|
|
|
|
|
"response": map[string]interface{}{
|
|
|
|
|
"error": "state_required",
|
|
|
|
|
"error_description": "state is required",
|
|
|
|
|
},
|
|
|
|
|
}, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if responseType == constants.ResponseTypeCode && codeChallenge == "" {
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, map[string]interface{}{
|
|
|
|
|
"type": "authorization_response",
|
|
|
|
|
"response": map[string]interface{}{
|
|
|
|
|
"error": "code_challenge_required",
|
|
|
|
|
"error_description": "code challenge is required",
|
|
|
|
|
},
|
|
|
|
|
}, http.StatusOK)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
loginError := map[string]interface{}{
|
|
|
|
|
"type": "authorization_response",
|
|
|
|
|
"response": map[string]string{
|
|
|
|
|
"response": map[string]interface{}{
|
|
|
|
|
"error": "login_required",
|
|
|
|
|
"error_description": "Login is required",
|
|
|
|
|
},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
sessionToken, err := cookie.GetSession(gc)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Debug("GetSession failed: ", err)
|
|
|
|
@@ -113,7 +146,7 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
log.Debug("GetUserByID failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, map[string]interface{}{
|
|
|
|
|
"type": "authorization_response",
|
|
|
|
|
"response": map[string]string{
|
|
|
|
|
"response": map[string]interface{}{
|
|
|
|
|
"error": "signup_required",
|
|
|
|
|
"error_description": "Sign up required",
|
|
|
|
|
},
|
|
|
|
@@ -129,7 +162,6 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
// rollover the session for security
|
|
|
|
|
go memorystore.Provider.DeleteUserSession(sessionKey, claims.Nonce)
|
|
|
|
|
if responseType == constants.ResponseTypeCode {
|
|
|
|
|
nonce := uuid.New().String()
|
|
|
|
|
newSessionTokenData, newSessionToken, err := token.CreateSessionToken(user, nonce, claims.Roles, scope, claims.LoginMethod)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Debug("CreateSessionToken failed: ", err)
|
|
|
|
@@ -137,6 +169,12 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := memorystore.Provider.SetState(codeChallenge, code+"@"+newSessionToken); err != nil {
|
|
|
|
|
log.Debug("SetState failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeSessionToken+"_"+newSessionTokenData.Nonce, newSessionToken); err != nil {
|
|
|
|
|
log.Debug("SetUserSession failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
@@ -144,12 +182,6 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
cookie.SetSession(gc, newSessionToken)
|
|
|
|
|
code := uuid.New().String()
|
|
|
|
|
if err := memorystore.Provider.SetState(codeChallenge, code+"@"+newSessionToken); err != nil {
|
|
|
|
|
log.Debug("SetState failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// in case, response type is code and user is already logged in send the code and state
|
|
|
|
|
// and cookie session will already be rolled over and set
|
|
|
|
@@ -164,7 +196,7 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
// },
|
|
|
|
|
// })
|
|
|
|
|
|
|
|
|
|
params := "code=" + code + "&state=" + state
|
|
|
|
|
params := "code=" + code + "&state=" + state + "&nonce=" + nonce
|
|
|
|
|
if responseMode == constants.ResponseModeQuery {
|
|
|
|
|
if strings.Contains(redirectURI, "?") {
|
|
|
|
|
redirectURI = redirectURI + "&" + params
|
|
|
|
@@ -181,7 +213,7 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, map[string]interface{}{
|
|
|
|
|
"type": "authorization_response",
|
|
|
|
|
"response": map[string]string{
|
|
|
|
|
"response": map[string]interface{}{
|
|
|
|
|
"code": code,
|
|
|
|
|
"state": state,
|
|
|
|
|
},
|
|
|
|
@@ -191,49 +223,78 @@ func AuthorizeHandler() gin.HandlerFunc {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if responseType == constants.ResponseTypeToken || responseType == constants.ResponseTypeIDToken {
|
|
|
|
|
// rollover the session for security
|
|
|
|
|
authToken, err := token.CreateAuthToken(gc, user, claims.Roles, scope, claims.LoginMethod)
|
|
|
|
|
hostname := parsers.GetHost(gc)
|
|
|
|
|
nonce := uuid.New().String()
|
|
|
|
|
_, fingerPrintHash, err := token.CreateSessionToken(user, nonce, claims.Roles, scope, claims.LoginMethod)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Debug("CreateAuthToken failed: ", err)
|
|
|
|
|
log.Debug("CreateSessionToken failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
accessToken, accessTokenExpiresAt, err := token.CreateAccessToken(user, claims.Roles, scope, hostname, nonce, claims.LoginMethod)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Debug("CreateAccessToken failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeSessionToken+"_"+authToken.FingerPrint, authToken.FingerPrintHash); err != nil {
|
|
|
|
|
idToken, _, err := token.CreateIDToken(user, claims.Roles, hostname, nonce, claims.LoginMethod)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Debug("CreateIDToken failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
// rollover the session for security
|
|
|
|
|
// authToken, err := token.CreateAuthToken(gc, user, claims.Roles, scope, claims.LoginMethod)
|
|
|
|
|
// if err != nil {
|
|
|
|
|
// log.Debug("CreateAuthToken failed: ", err)
|
|
|
|
|
// handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
// return
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
if err := memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeSessionToken+"_"+nonce, fingerPrintHash); err != nil {
|
|
|
|
|
log.Debug("SetUserSession failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeAccessToken+"_"+authToken.FingerPrint, authToken.AccessToken.Token); err != nil {
|
|
|
|
|
if err := memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeAccessToken+"_"+nonce, accessToken); err != nil {
|
|
|
|
|
log.Debug("SetUserSession failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
cookie.SetSession(gc, authToken.FingerPrintHash)
|
|
|
|
|
cookie.SetSession(gc, fingerPrintHash)
|
|
|
|
|
|
|
|
|
|
expiresIn := authToken.AccessToken.ExpiresAt - time.Now().Unix()
|
|
|
|
|
expiresIn := accessTokenExpiresAt - time.Now().Unix()
|
|
|
|
|
if expiresIn <= 0 {
|
|
|
|
|
expiresIn = 1
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// used of query mode
|
|
|
|
|
params := "access_token=" + authToken.AccessToken.Token + "&token_type=bearer&expires_in=" + strconv.FormatInt(expiresIn, 10) + "&state=" + state + "&id_token=" + authToken.IDToken.Token
|
|
|
|
|
params := "access_token=" + accessToken + "&token_type=bearer&expires_in=" + strconv.FormatInt(expiresIn, 10) + "&state=" + state + "&id_token=" + idToken + "&code=" + code + "&nonce=" + nonce
|
|
|
|
|
|
|
|
|
|
res := map[string]interface{}{
|
|
|
|
|
"access_token": authToken.AccessToken.Token,
|
|
|
|
|
"id_token": authToken.IDToken.Token,
|
|
|
|
|
"access_token": accessToken,
|
|
|
|
|
"id_token": idToken,
|
|
|
|
|
"state": state,
|
|
|
|
|
"scope": scope,
|
|
|
|
|
"token_type": "Bearer",
|
|
|
|
|
"expires_in": expiresIn,
|
|
|
|
|
"code": code,
|
|
|
|
|
"nonce": nonce,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if authToken.RefreshToken != nil {
|
|
|
|
|
res["refresh_token"] = authToken.RefreshToken.Token
|
|
|
|
|
params += "&refresh_token=" + authToken.RefreshToken.Token
|
|
|
|
|
memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeRefreshToken+"_"+authToken.FingerPrint, authToken.RefreshToken.Token)
|
|
|
|
|
if utils.StringSliceContains(scope, "offline_access") {
|
|
|
|
|
refreshToken, _, err := token.CreateRefreshToken(user, claims.Roles, scope, hostname, nonce, claims.LoginMethod)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Debug("SetUserSession failed: ", err)
|
|
|
|
|
handleResponse(gc, responseMode, loginURL, redirectURI, loginError, http.StatusOK)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
res["refresh_token"] = refreshToken
|
|
|
|
|
params += "&refresh_token=" + refreshToken
|
|
|
|
|
memorystore.Provider.SetUserSession(sessionKey, constants.TokenTypeRefreshToken+"_"+nonce, refreshToken)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if responseMode == constants.ResponseModeQuery {
|
|
|
|
@@ -270,24 +331,16 @@ func validateAuthorizeRequest(responseType, responseMode, clientID, state, codeC
|
|
|
|
|
return fmt.Errorf("invalid response mode %s. 'query', 'fragment', 'form_post' and 'web_message' are valid response_mode", responseMode)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if responseType == constants.ResponseTypeCode && strings.TrimSpace(codeChallenge) == "" {
|
|
|
|
|
return fmt.Errorf("code_challenge is required for %s '%s'", responseType, constants.ResponseTypeCode)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if client, err := memorystore.Provider.GetStringStoreEnvVariable(constants.EnvKeyClientID); client != clientID || err != nil {
|
|
|
|
|
return fmt.Errorf("invalid client_id %s", clientID)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if strings.TrimSpace(state) == "" {
|
|
|
|
|
return fmt.Errorf("state is required")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func handleResponse(gc *gin.Context, responseMode, loginURI, redirectURI string, data map[string]interface{}, httpStatusCode int) {
|
|
|
|
|
isAuthenticationRequired := false
|
|
|
|
|
if _, ok := data["response"].(map[string]string)["error"]; ok {
|
|
|
|
|
if _, ok := data["response"].(map[string]interface{})["error"]; ok {
|
|
|
|
|
isAuthenticationRequired = true
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@@ -308,7 +361,7 @@ func handleResponse(gc *gin.Context, responseMode, loginURI, redirectURI string,
|
|
|
|
|
case constants.ResponseModeFormPost:
|
|
|
|
|
gc.HTML(httpStatusCode, authorizeFormPostTemplate, gin.H{
|
|
|
|
|
"target_origin": redirectURI,
|
|
|
|
|
"authorization_response": data,
|
|
|
|
|
"authorization_response": data["response"],
|
|
|
|
|
})
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|