forked from gitea/gitea
1
0
Fork 0
gitea/modules/auth/auth.go

277 lines
6.6 KiB
Go
Raw Normal View History

2014-03-06 15:21:44 +08:00
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package auth
import (
"reflect"
2014-07-26 12:24:27 +08:00
"strings"
2015-08-19 06:22:33 +08:00
"time"
2014-03-06 15:21:44 +08:00
"github.com/Unknwon/com"
2015-10-16 09:28:12 +08:00
"github.com/go-macaron/binding"
"github.com/go-macaron/session"
gouuid "github.com/satori/go.uuid"
2015-10-16 09:28:12 +08:00
"gopkg.in/macaron.v1"
2014-03-06 15:21:44 +08:00
2014-07-26 12:24:27 +08:00
"github.com/gogits/gogs/models"
2014-11-10 18:30:07 +08:00
"github.com/gogits/gogs/modules/base"
2014-07-26 12:24:27 +08:00
"github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/setting"
2014-03-06 15:21:44 +08:00
)
func IsAPIPath(url string) bool {
return strings.HasPrefix(url, "/api/")
}
// SignedInID returns the id of signed in user.
func SignedInID(ctx *macaron.Context, sess session.Store) int64 {
2014-07-26 12:24:27 +08:00
if !models.HasEngine {
return 0
}
// Check access token.
if IsAPIPath(ctx.Req.URL.Path) {
tokenSHA := ctx.Query("token")
if len(tokenSHA) == 0 {
// Well, check with header again.
auHead := ctx.Req.Header.Get("Authorization")
if len(auHead) > 0 {
auths := strings.Fields(auHead)
if len(auths) == 2 && auths[0] == "token" {
tokenSHA = auths[1]
}
}
}
// Let's see if token is valid.
if len(tokenSHA) > 0 {
t, err := models.GetAccessTokenBySHA(tokenSHA)
if err != nil {
if models.IsErrAccessTokenNotExist(err) {
log.Error(4, "GetAccessTokenBySHA: %v", err)
}
return 0
2014-11-12 19:48:50 +08:00
}
t.Updated = time.Now()
2016-01-07 03:41:42 +08:00
if err = models.UpdateAccessToken(t); err != nil {
log.Error(4, "UpdateAccessToken: %v", err)
}
return t.UID
2014-11-12 19:48:50 +08:00
}
}
uid := sess.Get("uid")
if uid == nil {
return 0
}
if id, ok := uid.(int64); ok {
2015-08-08 22:43:14 +08:00
if _, err := models.GetUserByID(id); err != nil {
2015-08-05 11:14:17 +08:00
if !models.IsErrUserNotExist(err) {
log.Error(4, "GetUserById: %v", err)
}
return 0
}
return id
}
2014-07-26 12:24:27 +08:00
return 0
2014-07-03 04:42:16 +08:00
}
2014-07-26 12:24:27 +08:00
// SignedInUser returns the user object of signed user.
2014-12-06 06:54:57 +08:00
// It returns a bool value to indicate whether user uses basic auth or not.
func SignedInUser(ctx *macaron.Context, sess session.Store) (*models.User, bool) {
2014-11-10 18:30:07 +08:00
if !models.HasEngine {
2014-11-19 00:07:16 +08:00
return nil, false
2014-11-10 18:30:07 +08:00
}
uid := SignedInID(ctx, sess)
2014-11-10 18:30:07 +08:00
2014-07-26 12:24:27 +08:00
if uid <= 0 {
2014-11-10 18:30:07 +08:00
if setting.Service.EnableReverseProxyAuth {
webAuthUser := ctx.Req.Header.Get(setting.ReverseProxyAuthUser)
2014-11-10 18:30:07 +08:00
if len(webAuthUser) > 0 {
u, err := models.GetUserByName(webAuthUser)
if err != nil {
2015-08-05 11:14:17 +08:00
if !models.IsErrUserNotExist(err) {
2014-11-10 18:30:07 +08:00
log.Error(4, "GetUserByName: %v", err)
2014-12-06 06:54:57 +08:00
return nil, false
}
2014-12-07 09:22:48 +08:00
// Check if enabled auto-registration.
2014-12-06 06:54:57 +08:00
if setting.Service.EnableReverseProxyAutoRegister {
u := &models.User{
Name: webAuthUser,
Email: gouuid.NewV4().String() + "@localhost",
2014-12-06 06:54:57 +08:00
Passwd: webAuthUser,
IsActive: true,
}
if err = models.CreateUser(u); err != nil {
// FIXME: should I create a system notice?
log.Error(4, "CreateUser: %v", err)
return nil, false
} else {
return u, false
}
2014-11-10 18:30:07 +08:00
}
}
2014-11-19 00:07:16 +08:00
return u, false
2014-11-10 18:30:07 +08:00
}
}
// Check with basic auth.
baHead := ctx.Req.Header.Get("Authorization")
2014-11-10 18:30:07 +08:00
if len(baHead) > 0 {
auths := strings.Fields(baHead)
if len(auths) == 2 && auths[0] == "Basic" {
uname, passwd, _ := base.BasicAuthDecode(auths[1])
2015-02-27 20:42:03 +08:00
u, err := models.UserSignIn(uname, passwd)
2014-11-10 18:30:07 +08:00
if err != nil {
2015-08-05 11:14:17 +08:00
if !models.IsErrUserNotExist(err) {
2015-02-27 20:42:03 +08:00
log.Error(4, "UserSignIn: %v", err)
2014-11-10 18:30:07 +08:00
}
2014-11-19 00:07:16 +08:00
return nil, false
2014-11-10 18:30:07 +08:00
}
2015-02-27 20:42:03 +08:00
return u, true
2014-11-10 18:30:07 +08:00
}
}
2014-11-19 00:07:16 +08:00
return nil, false
2014-07-26 12:24:27 +08:00
}
2015-08-08 22:43:14 +08:00
u, err := models.GetUserByID(uid)
2014-07-26 12:24:27 +08:00
if err != nil {
log.Error(4, "GetUserById: %v", err)
2014-11-19 00:07:16 +08:00
return nil, false
2014-07-26 12:24:27 +08:00
}
2014-11-19 00:07:16 +08:00
return u, false
2014-03-29 06:40:31 +08:00
}
2014-10-15 23:19:20 +08:00
type Form interface {
binding.Validator
}
func init() {
binding.SetNameMapper(com.ToSnakeCase)
}
2014-07-26 12:24:27 +08:00
// AssignForm assign form values back to the template data.
func AssignForm(form interface{}, data map[string]interface{}) {
typ := reflect.TypeOf(form)
val := reflect.ValueOf(form)
if typ.Kind() == reflect.Ptr {
typ = typ.Elem()
val = val.Elem()
}
for i := 0; i < typ.NumField(); i++ {
field := typ.Field(i)
fieldName := field.Tag.Get("form")
// Allow ignored fields in the struct
if fieldName == "-" {
continue
} else if len(fieldName) == 0 {
fieldName = com.ToSnakeCase(field.Name)
2014-07-26 12:24:27 +08:00
}
data[fieldName] = val.Field(i).Interface()
}
}
func getRuleBody(field reflect.StructField, prefix string) string {
2014-07-26 12:24:27 +08:00
for _, rule := range strings.Split(field.Tag.Get("binding"), ";") {
2014-08-01 18:12:14 +08:00
if strings.HasPrefix(rule, prefix) {
return rule[len(prefix) : len(rule)-1]
2014-07-26 12:24:27 +08:00
}
}
return ""
}
func GetSize(field reflect.StructField) string {
return getRuleBody(field, "Size(")
}
2014-08-01 18:12:14 +08:00
func GetMinSize(field reflect.StructField) string {
return getRuleBody(field, "MinSize(")
2014-08-01 18:12:14 +08:00
}
func GetMaxSize(field reflect.StructField) string {
return getRuleBody(field, "MaxSize(")
}
func GetInclude(field reflect.StructField) string {
return getRuleBody(field, "Include(")
2014-08-01 18:12:14 +08:00
}
2015-08-28 18:33:09 +08:00
// FIXME: struct contains a struct
func validateStruct(obj interface{}) binding.Errors {
return nil
}
2014-10-15 23:19:20 +08:00
func validate(errs binding.Errors, data map[string]interface{}, f Form, l macaron.Locale) binding.Errors {
if errs.Len() == 0 {
return errs
2014-07-26 12:24:27 +08:00
}
data["HasError"] = true
AssignForm(f, data)
typ := reflect.TypeOf(f)
val := reflect.ValueOf(f)
if typ.Kind() == reflect.Ptr {
typ = typ.Elem()
val = val.Elem()
}
for i := 0; i < typ.NumField(); i++ {
field := typ.Field(i)
fieldName := field.Tag.Get("form")
// Allow ignored fields in the struct
if fieldName == "-" {
continue
}
2014-10-15 23:19:20 +08:00
if errs[0].FieldNames[0] == field.Name {
2014-07-26 12:24:27 +08:00
data["Err_"+field.Name] = true
2015-07-08 19:47:56 +08:00
trName := field.Tag.Get("locale")
if len(trName) == 0 {
trName = l.Tr("form." + field.Name)
} else {
trName = l.Tr(trName)
}
2014-10-15 23:19:20 +08:00
switch errs[0].Classification {
2014-12-15 14:49:59 +08:00
case binding.ERR_REQUIRED:
2014-07-26 12:24:27 +08:00
data["ErrorMsg"] = trName + l.Tr("form.require_error")
2014-12-15 14:49:59 +08:00
case binding.ERR_ALPHA_DASH:
2014-07-26 12:24:27 +08:00
data["ErrorMsg"] = trName + l.Tr("form.alpha_dash_error")
2014-12-15 14:49:59 +08:00
case binding.ERR_ALPHA_DASH_DOT:
2014-07-26 12:24:27 +08:00
data["ErrorMsg"] = trName + l.Tr("form.alpha_dash_dot_error")
case binding.ERR_SIZE:
data["ErrorMsg"] = trName + l.Tr("form.size_error", GetSize(field))
2014-12-15 14:49:59 +08:00
case binding.ERR_MIN_SIZE:
2014-08-01 18:12:14 +08:00
data["ErrorMsg"] = trName + l.Tr("form.min_size_error", GetMinSize(field))
2014-12-15 14:49:59 +08:00
case binding.ERR_MAX_SIZE:
2014-08-01 18:12:14 +08:00
data["ErrorMsg"] = trName + l.Tr("form.max_size_error", GetMaxSize(field))
2014-12-15 14:49:59 +08:00
case binding.ERR_EMAIL:
2014-07-26 12:24:27 +08:00
data["ErrorMsg"] = trName + l.Tr("form.email_error")
2014-12-15 14:49:59 +08:00
case binding.ERR_URL:
2014-07-26 12:24:27 +08:00
data["ErrorMsg"] = trName + l.Tr("form.url_error")
case binding.ERR_INCLUDE:
data["ErrorMsg"] = trName + l.Tr("form.include_error", GetInclude(field))
2014-07-26 12:24:27 +08:00
default:
2014-10-15 23:19:20 +08:00
data["ErrorMsg"] = l.Tr("form.unknown_error") + " " + errs[0].Classification
2014-07-26 12:24:27 +08:00
}
2014-10-15 23:19:20 +08:00
return errs
2014-07-26 12:24:27 +08:00
}
}
2014-10-15 23:19:20 +08:00
return errs
2014-03-29 06:40:31 +08:00
}