106 lines · 3.3 KB · Go
RawDownload
// Copyright 2014 The Gogs Authors. All rights reserved.// Copyright 2019 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package auth import (	"fmt"	"net/http"	"regexp"	"strings" 	user_model "forgejo.org/models/user"	"forgejo.org/modules/auth/webauthn"	"forgejo.org/modules/log"	"forgejo.org/modules/optional"	"forgejo.org/modules/session"	"forgejo.org/modules/setting"	"forgejo.org/modules/web/middleware"	gitea_context "forgejo.org/services/context"	user_service "forgejo.org/services/user") // Init should be called exactly once when the application starts to allow plugins// to allocate necessary resourcesfunc Init() {	webauthn.Init()} // isAttachmentDownload check if request is a file download (GET) with URL to an attachmentfunc isAttachmentDownload(req *http.Request) bool {	return strings.HasPrefix(req.URL.Path, "/attachments/") && req.Method == "GET"} // isContainerPath checks if the request targets the container endpointfunc isContainerPath(req *http.Request) bool {	return strings.HasPrefix(req.URL.Path, "/v2/")} var (	gitRawOrAttachPathRe = regexp.MustCompile(`^/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+/(?:(?:git-(?:(?:upload)|(?:receive))-pack$)|(?:info/refs$)|(?:HEAD$)|(?:objects/)|(?:raw/)|(?:releases/download/)|(?:attachments/))`)	lfsPathRe            = regexp.MustCompile(`^/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+/info/lfs/`)	archivePathRe        = regexp.MustCompile(`^/[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+/archive/`)) func isGitRawOrAttachPath(req *http.Request) bool {	return gitRawOrAttachPathRe.MatchString(req.URL.Path)} func isGitRawOrAttachOrLFSPath(req *http.Request) bool {	if isGitRawOrAttachPath(req) {		return true	}	if setting.LFS.StartServer {		return lfsPathRe.MatchString(req.URL.Path)	}	return false} func isArchivePath(req *http.Request) bool {	return archivePathRe.MatchString(req.URL.Path)} // handleSignIn clears existing session variables and stores new ones for the specified user objectfunc handleSignIn(resp http.ResponseWriter, req *http.Request, sess SessionStore, user *user_model.User) {	// We need to regenerate the session...	newSess, err := session.RegenerateSession(resp, req)	if err != nil {		log.Error(fmt.Sprintf("Error regenerating session: %v", err))	} else {		sess = newSess	} 	_ = sess.Delete("openid_verified_uri")	_ = sess.Delete("openid_signin_remember")	_ = sess.Delete("openid_determined_email")	_ = sess.Delete("openid_determined_username")	_ = sess.Delete("twofaUid")	_ = sess.Delete("twofaRemember")	_ = sess.Delete("webauthnAssertion")	_ = sess.Delete("linkAccount")	err = sess.Set("uid", user.ID)	if err != nil {		log.Error(fmt.Sprintf("Error setting session: %v", err))	} 	// Language setting of the user overwrites the one previously set	// If the user does not have a locale set, we save the current one.	if len(user.Language) == 0 {		lc := middleware.Locale(resp, req)		opts := &user_service.UpdateOptions{			Language: optional.Some(lc.Language()),		}		if err := user_service.UpdateUser(req.Context(), user, opts); err != nil {			log.Error(fmt.Sprintf("Error updating user language [user: %d, locale: %s]", user.ID, user.Language))			return		}	} 	middleware.SetLocaleCookie(resp, user.Language, 0) 	// Clear whatever CSRF has right now, force to generate a new one	if ctx := gitea_context.GetWebContext(req); ctx != nil {		ctx.Csrf.DeleteCookie(ctx)	}}