|  | @@ -10,30 +10,21 @@ import (
 | 
	
		
			
				|  |  |  	"fmt"
 | 
	
		
			
				|  |  |  	"net/smtp"
 | 
	
		
			
				|  |  |  	"net/textproto"
 | 
	
		
			
				|  |  | -	"os"
 | 
	
		
			
				|  |  | -	"path/filepath"
 | 
	
		
			
				|  |  |  	"strings"
 | 
	
		
			
				|  |  | -	"sync"
 | 
	
		
			
				|  |  | -	"time"
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  	"github.com/go-macaron/binding"
 | 
	
		
			
				|  |  | -	"github.com/json-iterator/go"
 | 
	
		
			
				|  |  |  	"github.com/unknwon/com"
 | 
	
		
			
				|  |  | -	"gopkg.in/ini.v1"
 | 
	
		
			
				|  |  | -	log "unknwon.dev/clog/v2"
 | 
	
		
			
				|  |  | -	"xorm.io/core"
 | 
	
		
			
				|  |  | -	"xorm.io/xorm"
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  	"gogs.io/gogs/internal/auth/github"
 | 
	
		
			
				|  |  |  	"gogs.io/gogs/internal/auth/ldap"
 | 
	
		
			
				|  |  |  	"gogs.io/gogs/internal/auth/pam"
 | 
	
		
			
				|  |  | -	"gogs.io/gogs/internal/conf"
 | 
	
		
			
				|  |  |  	"gogs.io/gogs/internal/db/errors"
 | 
	
		
			
				|  |  |  )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  type LoginType int
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  // Note: new type must append to the end of list to maintain compatibility.
 | 
	
		
			
				|  |  | +// TODO: Move to authutil.
 | 
	
		
			
				|  |  |  const (
 | 
	
		
			
				|  |  |  	LoginNotype LoginType = iota
 | 
	
		
			
				|  |  |  	LoginPlain            // 1
 | 
	
	
		
			
				|  | @@ -52,497 +43,24 @@ var LoginNames = map[LoginType]string{
 | 
	
		
			
				|  |  |  	LoginGitHub: "GitHub",
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -var SecurityProtocolNames = map[ldap.SecurityProtocol]string{
 | 
	
		
			
				|  |  | -	ldap.SECURITY_PROTOCOL_UNENCRYPTED: "Unencrypted",
 | 
	
		
			
				|  |  | -	ldap.SECURITY_PROTOCOL_LDAPS:       "LDAPS",
 | 
	
		
			
				|  |  | -	ldap.SECURITY_PROTOCOL_START_TLS:   "StartTLS",
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// Ensure structs implemented interface.
 | 
	
		
			
				|  |  | -var (
 | 
	
		
			
				|  |  | -	_ core.Conversion = &LDAPConfig{}
 | 
	
		
			
				|  |  | -	_ core.Conversion = &SMTPConfig{}
 | 
	
		
			
				|  |  | -	_ core.Conversion = &PAMConfig{}
 | 
	
		
			
				|  |  | -	_ core.Conversion = &GitHubConfig{}
 | 
	
		
			
				|  |  | -)
 | 
	
		
			
				|  |  | +// ***********************
 | 
	
		
			
				|  |  | +// ----- LDAP config -----
 | 
	
		
			
				|  |  | +// ***********************
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  type LDAPConfig struct {
 | 
	
		
			
				|  |  | -	*ldap.Source `ini:"config"`
 | 
	
		
			
				|  |  | +	ldap.Source `ini:"config"`
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -func (cfg *LDAPConfig) FromDB(bs []byte) error {
 | 
	
		
			
				|  |  | -	return jsoniter.Unmarshal(bs, &cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *LDAPConfig) ToDB() ([]byte, error) {
 | 
	
		
			
				|  |  | -	return jsoniter.Marshal(cfg)
 | 
	
		
			
				|  |  | +var SecurityProtocolNames = map[ldap.SecurityProtocol]string{
 | 
	
		
			
				|  |  | +	ldap.SecurityProtocolUnencrypted: "Unencrypted",
 | 
	
		
			
				|  |  | +	ldap.SecurityProtocolLDAPS:       "LDAPS",
 | 
	
		
			
				|  |  | +	ldap.SecurityProtocolStartTLS:    "StartTLS",
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  func (cfg *LDAPConfig) SecurityProtocolName() string {
 | 
	
		
			
				|  |  |  	return SecurityProtocolNames[cfg.SecurityProtocol]
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -type SMTPConfig struct {
 | 
	
		
			
				|  |  | -	Auth           string
 | 
	
		
			
				|  |  | -	Host           string
 | 
	
		
			
				|  |  | -	Port           int
 | 
	
		
			
				|  |  | -	AllowedDomains string `xorm:"TEXT"`
 | 
	
		
			
				|  |  | -	TLS            bool   `ini:"tls"`
 | 
	
		
			
				|  |  | -	SkipVerify     bool
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *SMTPConfig) FromDB(bs []byte) error {
 | 
	
		
			
				|  |  | -	return jsoniter.Unmarshal(bs, cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *SMTPConfig) ToDB() ([]byte, error) {
 | 
	
		
			
				|  |  | -	return jsoniter.Marshal(cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -type PAMConfig struct {
 | 
	
		
			
				|  |  | -	ServiceName string // PAM service (e.g. system-auth)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *PAMConfig) FromDB(bs []byte) error {
 | 
	
		
			
				|  |  | -	return jsoniter.Unmarshal(bs, &cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *PAMConfig) ToDB() ([]byte, error) {
 | 
	
		
			
				|  |  | -	return jsoniter.Marshal(cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -type GitHubConfig struct {
 | 
	
		
			
				|  |  | -	APIEndpoint string // GitHub service (e.g. https://api.github.com/)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *GitHubConfig) FromDB(bs []byte) error {
 | 
	
		
			
				|  |  | -	return jsoniter.Unmarshal(bs, &cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (cfg *GitHubConfig) ToDB() ([]byte, error) {
 | 
	
		
			
				|  |  | -	return jsoniter.Marshal(cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// AuthSourceFile contains information of an authentication source file.
 | 
	
		
			
				|  |  | -type AuthSourceFile struct {
 | 
	
		
			
				|  |  | -	abspath string
 | 
	
		
			
				|  |  | -	file    *ini.File
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// SetGeneral sets new value to the given key in the general (default) section.
 | 
	
		
			
				|  |  | -func (f *AuthSourceFile) SetGeneral(name, value string) {
 | 
	
		
			
				|  |  | -	f.file.Section("").Key(name).SetValue(value)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// SetConfig sets new values to the "config" section.
 | 
	
		
			
				|  |  | -func (f *AuthSourceFile) SetConfig(cfg core.Conversion) error {
 | 
	
		
			
				|  |  | -	return f.file.Section("config").ReflectFrom(cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// Save writes updates into file system.
 | 
	
		
			
				|  |  | -func (f *AuthSourceFile) Save() error {
 | 
	
		
			
				|  |  | -	return f.file.SaveTo(f.abspath)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// LoginSource represents an external way for authorizing users.
 | 
	
		
			
				|  |  | -type LoginSource struct {
 | 
	
		
			
				|  |  | -	ID        int64
 | 
	
		
			
				|  |  | -	Type      LoginType
 | 
	
		
			
				|  |  | -	Name      string          `xorm:"UNIQUE"`
 | 
	
		
			
				|  |  | -	IsActived bool            `xorm:"NOT NULL DEFAULT false"`
 | 
	
		
			
				|  |  | -	IsDefault bool            `xorm:"DEFAULT false"`
 | 
	
		
			
				|  |  | -	Cfg       core.Conversion `xorm:"TEXT" gorm:"COLUMN:remove-me-when-migrated-to-gorm"`
 | 
	
		
			
				|  |  | -	RawCfg    string          `xorm:"-" gorm:"COLUMN:cfg"` // TODO: Remove me when migrated to GORM.
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	Created     time.Time `xorm:"-" json:"-"`
 | 
	
		
			
				|  |  | -	CreatedUnix int64
 | 
	
		
			
				|  |  | -	Updated     time.Time `xorm:"-" json:"-"`
 | 
	
		
			
				|  |  | -	UpdatedUnix int64
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	LocalFile *AuthSourceFile `xorm:"-" json:"-"`
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) BeforeInsert() {
 | 
	
		
			
				|  |  | -	s.CreatedUnix = time.Now().Unix()
 | 
	
		
			
				|  |  | -	s.UpdatedUnix = s.CreatedUnix
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) BeforeUpdate() {
 | 
	
		
			
				|  |  | -	s.UpdatedUnix = time.Now().Unix()
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// Cell2Int64 converts a xorm.Cell type to int64,
 | 
	
		
			
				|  |  | -// and handles possible irregular cases.
 | 
	
		
			
				|  |  | -func Cell2Int64(val xorm.Cell) int64 {
 | 
	
		
			
				|  |  | -	switch (*val).(type) {
 | 
	
		
			
				|  |  | -	case []uint8:
 | 
	
		
			
				|  |  | -		log.Trace("Cell2Int64 ([]uint8): %v", *val)
 | 
	
		
			
				|  |  | -		return com.StrTo(string((*val).([]uint8))).MustInt64()
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return (*val).(int64)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) BeforeSet(colName string, val xorm.Cell) {
 | 
	
		
			
				|  |  | -	switch colName {
 | 
	
		
			
				|  |  | -	case "type":
 | 
	
		
			
				|  |  | -		switch LoginType(Cell2Int64(val)) {
 | 
	
		
			
				|  |  | -		case LoginLDAP, LoginDLDAP:
 | 
	
		
			
				|  |  | -			s.Cfg = new(LDAPConfig)
 | 
	
		
			
				|  |  | -		case LoginSMTP:
 | 
	
		
			
				|  |  | -			s.Cfg = new(SMTPConfig)
 | 
	
		
			
				|  |  | -		case LoginPAM:
 | 
	
		
			
				|  |  | -			s.Cfg = new(PAMConfig)
 | 
	
		
			
				|  |  | -		case LoginGitHub:
 | 
	
		
			
				|  |  | -			s.Cfg = new(GitHubConfig)
 | 
	
		
			
				|  |  | -		default:
 | 
	
		
			
				|  |  | -			panic("unrecognized login source type: " + com.ToStr(*val))
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) AfterSet(colName string, _ xorm.Cell) {
 | 
	
		
			
				|  |  | -	switch colName {
 | 
	
		
			
				|  |  | -	case "created_unix":
 | 
	
		
			
				|  |  | -		s.Created = time.Unix(s.CreatedUnix, 0).Local()
 | 
	
		
			
				|  |  | -	case "updated_unix":
 | 
	
		
			
				|  |  | -		s.Updated = time.Unix(s.UpdatedUnix, 0).Local()
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// NOTE: This is a GORM query hook.
 | 
	
		
			
				|  |  | -func (s *LoginSource) AfterFind() error {
 | 
	
		
			
				|  |  | -	switch s.Type {
 | 
	
		
			
				|  |  | -	case LoginLDAP, LoginDLDAP:
 | 
	
		
			
				|  |  | -		s.Cfg = new(LDAPConfig)
 | 
	
		
			
				|  |  | -	case LoginSMTP:
 | 
	
		
			
				|  |  | -		s.Cfg = new(SMTPConfig)
 | 
	
		
			
				|  |  | -	case LoginPAM:
 | 
	
		
			
				|  |  | -		s.Cfg = new(PAMConfig)
 | 
	
		
			
				|  |  | -	case LoginGitHub:
 | 
	
		
			
				|  |  | -		s.Cfg = new(GitHubConfig)
 | 
	
		
			
				|  |  | -	default:
 | 
	
		
			
				|  |  | -		return fmt.Errorf("unrecognized login source type: %v", s.Type)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return jsoniter.UnmarshalFromString(s.RawCfg, s.Cfg)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) TypeName() string {
 | 
	
		
			
				|  |  | -	return LoginNames[s.Type]
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) IsLDAP() bool {
 | 
	
		
			
				|  |  | -	return s.Type == LoginLDAP
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) IsDLDAP() bool {
 | 
	
		
			
				|  |  | -	return s.Type == LoginDLDAP
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) IsSMTP() bool {
 | 
	
		
			
				|  |  | -	return s.Type == LoginSMTP
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) IsPAM() bool {
 | 
	
		
			
				|  |  | -	return s.Type == LoginPAM
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) IsGitHub() bool {
 | 
	
		
			
				|  |  | -	return s.Type == LoginGitHub
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) HasTLS() bool {
 | 
	
		
			
				|  |  | -	return ((s.IsLDAP() || s.IsDLDAP()) &&
 | 
	
		
			
				|  |  | -		s.LDAP().SecurityProtocol > ldap.SECURITY_PROTOCOL_UNENCRYPTED) ||
 | 
	
		
			
				|  |  | -		s.IsSMTP()
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) UseTLS() bool {
 | 
	
		
			
				|  |  | -	switch s.Type {
 | 
	
		
			
				|  |  | -	case LoginLDAP, LoginDLDAP:
 | 
	
		
			
				|  |  | -		return s.LDAP().SecurityProtocol != ldap.SECURITY_PROTOCOL_UNENCRYPTED
 | 
	
		
			
				|  |  | -	case LoginSMTP:
 | 
	
		
			
				|  |  | -		return s.SMTP().TLS
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	return false
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) SkipVerify() bool {
 | 
	
		
			
				|  |  | -	switch s.Type {
 | 
	
		
			
				|  |  | -	case LoginLDAP, LoginDLDAP:
 | 
	
		
			
				|  |  | -		return s.LDAP().SkipVerify
 | 
	
		
			
				|  |  | -	case LoginSMTP:
 | 
	
		
			
				|  |  | -		return s.SMTP().SkipVerify
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	return false
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) LDAP() *LDAPConfig {
 | 
	
		
			
				|  |  | -	return s.Cfg.(*LDAPConfig)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) SMTP() *SMTPConfig {
 | 
	
		
			
				|  |  | -	return s.Cfg.(*SMTPConfig)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) PAM() *PAMConfig {
 | 
	
		
			
				|  |  | -	return s.Cfg.(*PAMConfig)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LoginSource) GitHub() *GitHubConfig {
 | 
	
		
			
				|  |  | -	return s.Cfg.(*GitHubConfig)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func CreateLoginSource(source *LoginSource) error {
 | 
	
		
			
				|  |  | -	has, err := x.Get(&LoginSource{Name: source.Name})
 | 
	
		
			
				|  |  | -	if err != nil {
 | 
	
		
			
				|  |  | -		return err
 | 
	
		
			
				|  |  | -	} else if has {
 | 
	
		
			
				|  |  | -		return ErrLoginSourceAlreadyExist{source.Name}
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	_, err = x.Insert(source)
 | 
	
		
			
				|  |  | -	if err != nil {
 | 
	
		
			
				|  |  | -		return err
 | 
	
		
			
				|  |  | -	} else if source.IsDefault {
 | 
	
		
			
				|  |  | -		return ResetNonDefaultLoginSources(source)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return nil
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ListLoginSources returns all login sources defined.
 | 
	
		
			
				|  |  | -func ListLoginSources() ([]*LoginSource, error) {
 | 
	
		
			
				|  |  | -	sources := make([]*LoginSource, 0, 2)
 | 
	
		
			
				|  |  | -	if err := x.Find(&sources); err != nil {
 | 
	
		
			
				|  |  | -		return nil, err
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	return append(sources, localLoginSources.List()...), nil
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ActivatedLoginSources returns login sources that are currently activated.
 | 
	
		
			
				|  |  | -func ActivatedLoginSources() ([]*LoginSource, error) {
 | 
	
		
			
				|  |  | -	sources := make([]*LoginSource, 0, 2)
 | 
	
		
			
				|  |  | -	if err := x.Where("is_actived = ?", true).Find(&sources); err != nil {
 | 
	
		
			
				|  |  | -		return nil, fmt.Errorf("find activated login sources: %v", err)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return append(sources, localLoginSources.ActivatedList()...), nil
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ResetNonDefaultLoginSources clean other default source flag
 | 
	
		
			
				|  |  | -func ResetNonDefaultLoginSources(source *LoginSource) error {
 | 
	
		
			
				|  |  | -	// update changes to DB
 | 
	
		
			
				|  |  | -	if _, err := x.NotIn("id", []int64{source.ID}).Cols("is_default").Update(&LoginSource{IsDefault: false}); err != nil {
 | 
	
		
			
				|  |  | -		return err
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	// write changes to local authentications
 | 
	
		
			
				|  |  | -	for i := range localLoginSources.sources {
 | 
	
		
			
				|  |  | -		if localLoginSources.sources[i].LocalFile != nil && localLoginSources.sources[i].ID != source.ID {
 | 
	
		
			
				|  |  | -			localLoginSources.sources[i].LocalFile.SetGeneral("is_default", "false")
 | 
	
		
			
				|  |  | -			if err := localLoginSources.sources[i].LocalFile.SetConfig(source.Cfg); err != nil {
 | 
	
		
			
				|  |  | -				return fmt.Errorf("LocalFile.SetConfig: %v", err)
 | 
	
		
			
				|  |  | -			} else if err = localLoginSources.sources[i].LocalFile.Save(); err != nil {
 | 
	
		
			
				|  |  | -				return fmt.Errorf("LocalFile.Save: %v", err)
 | 
	
		
			
				|  |  | -			}
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	// flush memory so that web page can show the same behaviors
 | 
	
		
			
				|  |  | -	localLoginSources.UpdateLoginSource(source)
 | 
	
		
			
				|  |  | -	return nil
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// UpdateLoginSource updates information of login source to database or local file.
 | 
	
		
			
				|  |  | -func UpdateLoginSource(source *LoginSource) error {
 | 
	
		
			
				|  |  | -	if source.LocalFile == nil {
 | 
	
		
			
				|  |  | -		if _, err := x.Id(source.ID).AllCols().Update(source); err != nil {
 | 
	
		
			
				|  |  | -			return err
 | 
	
		
			
				|  |  | -		} else {
 | 
	
		
			
				|  |  | -			return ResetNonDefaultLoginSources(source)
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	source.LocalFile.SetGeneral("name", source.Name)
 | 
	
		
			
				|  |  | -	source.LocalFile.SetGeneral("is_activated", com.ToStr(source.IsActived))
 | 
	
		
			
				|  |  | -	source.LocalFile.SetGeneral("is_default", com.ToStr(source.IsDefault))
 | 
	
		
			
				|  |  | -	if err := source.LocalFile.SetConfig(source.Cfg); err != nil {
 | 
	
		
			
				|  |  | -		return fmt.Errorf("LocalFile.SetConfig: %v", err)
 | 
	
		
			
				|  |  | -	} else if err = source.LocalFile.Save(); err != nil {
 | 
	
		
			
				|  |  | -		return fmt.Errorf("LocalFile.Save: %v", err)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return ResetNonDefaultLoginSources(source)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func DeleteSource(source *LoginSource) error {
 | 
	
		
			
				|  |  | -	count, err := x.Count(&User{LoginSource: source.ID})
 | 
	
		
			
				|  |  | -	if err != nil {
 | 
	
		
			
				|  |  | -		return err
 | 
	
		
			
				|  |  | -	} else if count > 0 {
 | 
	
		
			
				|  |  | -		return ErrLoginSourceInUse{source.ID}
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	_, err = x.Id(source.ID).Delete(new(LoginSource))
 | 
	
		
			
				|  |  | -	return err
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// CountLoginSources returns total number of login sources.
 | 
	
		
			
				|  |  | -func CountLoginSources() int64 {
 | 
	
		
			
				|  |  | -	count, _ := x.Count(new(LoginSource))
 | 
	
		
			
				|  |  | -	return count + int64(localLoginSources.Len())
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// LocalLoginSources contains authentication sources configured and loaded from local files.
 | 
	
		
			
				|  |  | -// Calling its methods is thread-safe; otherwise, please maintain the mutex accordingly.
 | 
	
		
			
				|  |  | -type LocalLoginSources struct {
 | 
	
		
			
				|  |  | -	sync.RWMutex
 | 
	
		
			
				|  |  | -	sources []*LoginSource
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -func (s *LocalLoginSources) Len() int {
 | 
	
		
			
				|  |  | -	return len(s.sources)
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// List returns full clone of login sources.
 | 
	
		
			
				|  |  | -func (s *LocalLoginSources) List() []*LoginSource {
 | 
	
		
			
				|  |  | -	s.RLock()
 | 
	
		
			
				|  |  | -	defer s.RUnlock()
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	list := make([]*LoginSource, s.Len())
 | 
	
		
			
				|  |  | -	for i := range s.sources {
 | 
	
		
			
				|  |  | -		list[i] = &LoginSource{}
 | 
	
		
			
				|  |  | -		*list[i] = *s.sources[i]
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return list
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// ActivatedList returns clone of activated login sources.
 | 
	
		
			
				|  |  | -func (s *LocalLoginSources) ActivatedList() []*LoginSource {
 | 
	
		
			
				|  |  | -	s.RLock()
 | 
	
		
			
				|  |  | -	defer s.RUnlock()
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	list := make([]*LoginSource, 0, 2)
 | 
	
		
			
				|  |  | -	for i := range s.sources {
 | 
	
		
			
				|  |  | -		if !s.sources[i].IsActived {
 | 
	
		
			
				|  |  | -			continue
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -		source := &LoginSource{}
 | 
	
		
			
				|  |  | -		*source = *s.sources[i]
 | 
	
		
			
				|  |  | -		list = append(list, source)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -	return list
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// GetLoginSourceByID returns a clone of login source by given ID.
 | 
	
		
			
				|  |  | -func (s *LocalLoginSources) GetLoginSourceByID(id int64) (*LoginSource, error) {
 | 
	
		
			
				|  |  | -	s.RLock()
 | 
	
		
			
				|  |  | -	defer s.RUnlock()
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	for i := range s.sources {
 | 
	
		
			
				|  |  | -		if s.sources[i].ID == id {
 | 
	
		
			
				|  |  | -			source := &LoginSource{}
 | 
	
		
			
				|  |  | -			*source = *s.sources[i]
 | 
	
		
			
				|  |  | -			return source, nil
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	return nil, errors.LoginSourceNotExist{ID: id}
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// UpdateLoginSource updates in-memory copy of the authentication source.
 | 
	
		
			
				|  |  | -func (s *LocalLoginSources) UpdateLoginSource(source *LoginSource) {
 | 
	
		
			
				|  |  | -	s.Lock()
 | 
	
		
			
				|  |  | -	defer s.Unlock()
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	source.Updated = time.Now()
 | 
	
		
			
				|  |  | -	for i := range s.sources {
 | 
	
		
			
				|  |  | -		if s.sources[i].ID == source.ID {
 | 
	
		
			
				|  |  | -			*s.sources[i] = *source
 | 
	
		
			
				|  |  | -		} else if source.IsDefault {
 | 
	
		
			
				|  |  | -			s.sources[i].IsDefault = false
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -var localLoginSources = &LocalLoginSources{}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// LoadAuthSources loads authentication sources from local files
 | 
	
		
			
				|  |  | -// and converts them into login sources.
 | 
	
		
			
				|  |  | -func LoadAuthSources() {
 | 
	
		
			
				|  |  | -	authdPath := filepath.Join(conf.CustomDir(), "conf", "auth.d")
 | 
	
		
			
				|  |  | -	if !com.IsDir(authdPath) {
 | 
	
		
			
				|  |  | -		return
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	paths, err := com.GetFileListBySuffix(authdPath, ".conf")
 | 
	
		
			
				|  |  | -	if err != nil {
 | 
	
		
			
				|  |  | -		log.Fatal("Failed to list authentication sources: %v", err)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	localLoginSources.sources = make([]*LoginSource, 0, len(paths))
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -	for _, fpath := range paths {
 | 
	
		
			
				|  |  | -		authSource, err := ini.Load(fpath)
 | 
	
		
			
				|  |  | -		if err != nil {
 | 
	
		
			
				|  |  | -			log.Fatal("Failed to load authentication source: %v", err)
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -		authSource.NameMapper = ini.TitleUnderscore
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -		// Set general attributes
 | 
	
		
			
				|  |  | -		s := authSource.Section("")
 | 
	
		
			
				|  |  | -		loginSource := &LoginSource{
 | 
	
		
			
				|  |  | -			ID:        s.Key("id").MustInt64(),
 | 
	
		
			
				|  |  | -			Name:      s.Key("name").String(),
 | 
	
		
			
				|  |  | -			IsActived: s.Key("is_activated").MustBool(),
 | 
	
		
			
				|  |  | -			IsDefault: s.Key("is_default").MustBool(),
 | 
	
		
			
				|  |  | -			LocalFile: &AuthSourceFile{
 | 
	
		
			
				|  |  | -				abspath: fpath,
 | 
	
		
			
				|  |  | -				file:    authSource,
 | 
	
		
			
				|  |  | -			},
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -		fi, err := os.Stat(fpath)
 | 
	
		
			
				|  |  | -		if err != nil {
 | 
	
		
			
				|  |  | -			log.Fatal("Failed to load authentication source: %v", err)
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -		loginSource.Updated = fi.ModTime()
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -		// Parse authentication source file
 | 
	
		
			
				|  |  | -		authType := s.Key("type").String()
 | 
	
		
			
				|  |  | -		switch authType {
 | 
	
		
			
				|  |  | -		case "ldap_bind_dn":
 | 
	
		
			
				|  |  | -			loginSource.Type = LoginLDAP
 | 
	
		
			
				|  |  | -			loginSource.Cfg = &LDAPConfig{}
 | 
	
		
			
				|  |  | -		case "ldap_simple_auth":
 | 
	
		
			
				|  |  | -			loginSource.Type = LoginDLDAP
 | 
	
		
			
				|  |  | -			loginSource.Cfg = &LDAPConfig{}
 | 
	
		
			
				|  |  | -		case "smtp":
 | 
	
		
			
				|  |  | -			loginSource.Type = LoginSMTP
 | 
	
		
			
				|  |  | -			loginSource.Cfg = &SMTPConfig{}
 | 
	
		
			
				|  |  | -		case "pam":
 | 
	
		
			
				|  |  | -			loginSource.Type = LoginPAM
 | 
	
		
			
				|  |  | -			loginSource.Cfg = &PAMConfig{}
 | 
	
		
			
				|  |  | -		case "github":
 | 
	
		
			
				|  |  | -			loginSource.Type = LoginGitHub
 | 
	
		
			
				|  |  | -			loginSource.Cfg = &GitHubConfig{}
 | 
	
		
			
				|  |  | -		default:
 | 
	
		
			
				|  |  | -			log.Fatal("Failed to load authentication source: unknown type '%s'", authType)
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -		if err = authSource.Section("config").MapTo(loginSource.Cfg); err != nil {
 | 
	
		
			
				|  |  | -			log.Fatal("Failed to parse authentication source 'config': %v", err)
 | 
	
		
			
				|  |  | -		}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -		localLoginSources.sources = append(localLoginSources.sources, loginSource)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -// .____     ________      _____ __________
 | 
	
		
			
				|  |  | -// |    |    \______ \    /  _  \\______   \
 | 
	
		
			
				|  |  | -// |    |     |    |  \  /  /_\  \|     ___/
 | 
	
		
			
				|  |  | -// |    |___  |    `   \/    |    \    |
 | 
	
		
			
				|  |  | -// |_______ \/_______  /\____|__  /____|
 | 
	
		
			
				|  |  | -//         \/        \/         \/
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  |  func composeFullName(firstname, surname, username string) string {
 | 
	
		
			
				|  |  |  	switch {
 | 
	
		
			
				|  |  |  	case len(firstname) == 0 && len(surname) == 0:
 | 
	
	
		
			
				|  | @@ -559,7 +77,7 @@ func composeFullName(firstname, surname, username string) string {
 | 
	
		
			
				|  |  |  // LoginViaLDAP queries if login/password is valid against the LDAP directory pool,
 | 
	
		
			
				|  |  |  // and create a local user if success when enabled.
 | 
	
		
			
				|  |  |  func LoginViaLDAP(login, password string, source *LoginSource, autoRegister bool) (*User, error) {
 | 
	
		
			
				|  |  | -	username, fn, sn, mail, isAdmin, succeed := source.Cfg.(*LDAPConfig).SearchEntry(login, password, source.Type == LoginDLDAP)
 | 
	
		
			
				|  |  | +	username, fn, sn, mail, isAdmin, succeed := source.Config.(*LDAPConfig).SearchEntry(login, password, source.Type == LoginDLDAP)
 | 
	
		
			
				|  |  |  	if !succeed {
 | 
	
		
			
				|  |  |  		// User not in LDAP, do nothing
 | 
	
		
			
				|  |  |  		return nil, ErrUserNotExist{args: map[string]interface{}{"login": login}}
 | 
	
	
		
			
				|  | @@ -606,12 +124,18 @@ func LoginViaLDAP(login, password string, source *LoginSource, autoRegister bool
 | 
	
		
			
				|  |  |  	return user, CreateUser(user)
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -//   _________   __________________________
 | 
	
		
			
				|  |  | -//  /   _____/  /     \__    ___/\______   \
 | 
	
		
			
				|  |  | -//  \_____  \  /  \ /  \|    |    |     ___/
 | 
	
		
			
				|  |  | -//  /        \/    Y    \    |    |    |
 | 
	
		
			
				|  |  | -// /_______  /\____|__  /____|    |____|
 | 
	
		
			
				|  |  | -//         \/         \/
 | 
	
		
			
				|  |  | +// ***********************
 | 
	
		
			
				|  |  | +// ----- SMTP config -----
 | 
	
		
			
				|  |  | +// ***********************
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +type SMTPConfig struct {
 | 
	
		
			
				|  |  | +	Auth           string
 | 
	
		
			
				|  |  | +	Host           string
 | 
	
		
			
				|  |  | +	Port           int
 | 
	
		
			
				|  |  | +	AllowedDomains string
 | 
	
		
			
				|  |  | +	TLS            bool `ini:"tls"`
 | 
	
		
			
				|  |  | +	SkipVerify     bool
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  type smtpLoginAuth struct {
 | 
	
		
			
				|  |  |  	username, password string
 | 
	
	
		
			
				|  | @@ -634,11 +158,11 @@ func (auth *smtpLoginAuth) Next(fromServer []byte, more bool) ([]byte, error) {
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  const (
 | 
	
		
			
				|  |  | -	SMTP_PLAIN = "PLAIN"
 | 
	
		
			
				|  |  | -	SMTP_LOGIN = "LOGIN"
 | 
	
		
			
				|  |  | +	SMTPPlain = "PLAIN"
 | 
	
		
			
				|  |  | +	SMTPLogin = "LOGIN"
 | 
	
		
			
				|  |  |  )
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -var SMTPAuths = []string{SMTP_PLAIN, SMTP_LOGIN}
 | 
	
		
			
				|  |  | +var SMTPAuths = []string{SMTPPlain, SMTPLogin}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  func SMTPAuth(a smtp.Auth, cfg *SMTPConfig) error {
 | 
	
		
			
				|  |  |  	c, err := smtp.Dial(fmt.Sprintf("%s:%d", cfg.Host, cfg.Port))
 | 
	
	
		
			
				|  | @@ -687,9 +211,9 @@ func LoginViaSMTP(login, password string, sourceID int64, cfg *SMTPConfig, autoR
 | 
	
		
			
				|  |  |  	}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  	var auth smtp.Auth
 | 
	
		
			
				|  |  | -	if cfg.Auth == SMTP_PLAIN {
 | 
	
		
			
				|  |  | +	if cfg.Auth == SMTPPlain {
 | 
	
		
			
				|  |  |  		auth = smtp.PlainAuth("", login, password, cfg.Host)
 | 
	
		
			
				|  |  | -	} else if cfg.Auth == SMTP_LOGIN {
 | 
	
		
			
				|  |  | +	} else if cfg.Auth == SMTPLogin {
 | 
	
		
			
				|  |  |  		auth = &smtpLoginAuth{login, password}
 | 
	
		
			
				|  |  |  	} else {
 | 
	
		
			
				|  |  |  		return nil, errors.New("Unsupported SMTP authentication type")
 | 
	
	
		
			
				|  | @@ -729,12 +253,14 @@ func LoginViaSMTP(login, password string, sourceID int64, cfg *SMTPConfig, autoR
 | 
	
		
			
				|  |  |  	return user, CreateUser(user)
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -// __________  _____      _____
 | 
	
		
			
				|  |  | -// \______   \/  _  \    /     \
 | 
	
		
			
				|  |  | -//  |     ___/  /_\  \  /  \ /  \
 | 
	
		
			
				|  |  | -//  |    |  /    |    \/    Y    \
 | 
	
		
			
				|  |  | -//  |____|  \____|__  /\____|__  /
 | 
	
		
			
				|  |  | -//                  \/         \/
 | 
	
		
			
				|  |  | +// **********************
 | 
	
		
			
				|  |  | +// ----- PAM config -----
 | 
	
		
			
				|  |  | +// **********************
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +type PAMConfig struct {
 | 
	
		
			
				|  |  | +	// The name of the PAM service, e.g. system-auth.
 | 
	
		
			
				|  |  | +	ServiceName string
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  // LoginViaPAM queries if login/password is valid against the PAM,
 | 
	
		
			
				|  |  |  // and create a local user if success when enabled.
 | 
	
	
		
			
				|  | @@ -763,12 +289,14 @@ func LoginViaPAM(login, password string, sourceID int64, cfg *PAMConfig, autoReg
 | 
	
		
			
				|  |  |  	return user, CreateUser(user)
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -// ________.__  __     ___ ___      ___.
 | 
	
		
			
				|  |  | -// /  _____/|__|/  |_  /   |   \ __ _\_ |__
 | 
	
		
			
				|  |  | -// /   \  ___|  \   __\/    ~    \  |  \ __ \
 | 
	
		
			
				|  |  | -// \    \_\  \  ||  |  \    Y    /  |  / \_\ \
 | 
	
		
			
				|  |  | -// \______  /__||__|   \___|_  /|____/|___  /
 | 
	
		
			
				|  |  | -// \/                 \/           \/
 | 
	
		
			
				|  |  | +// *************************
 | 
	
		
			
				|  |  | +// ----- GitHub config -----
 | 
	
		
			
				|  |  | +// *************************
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +type GitHubConfig struct {
 | 
	
		
			
				|  |  | +	// the GitHub service endpoint, e.g. https://api.github.com/.
 | 
	
		
			
				|  |  | +	APIEndpoint string
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  func LoginViaGitHub(login, password string, sourceID int64, cfg *GitHubConfig, autoRegister bool) (*User, error) {
 | 
	
		
			
				|  |  |  	fullname, email, url, location, err := github.Authenticate(cfg.APIEndpoint, login, password)
 | 
	
	
		
			
				|  | @@ -807,11 +335,11 @@ func authenticateViaLoginSource(source *LoginSource, login, password string, aut
 | 
	
		
			
				|  |  |  	case LoginLDAP, LoginDLDAP:
 | 
	
		
			
				|  |  |  		return LoginViaLDAP(login, password, source, autoRegister)
 | 
	
		
			
				|  |  |  	case LoginSMTP:
 | 
	
		
			
				|  |  | -		return LoginViaSMTP(login, password, source.ID, source.Cfg.(*SMTPConfig), autoRegister)
 | 
	
		
			
				|  |  | +		return LoginViaSMTP(login, password, source.ID, source.Config.(*SMTPConfig), autoRegister)
 | 
	
		
			
				|  |  |  	case LoginPAM:
 | 
	
		
			
				|  |  | -		return LoginViaPAM(login, password, source.ID, source.Cfg.(*PAMConfig), autoRegister)
 | 
	
		
			
				|  |  | +		return LoginViaPAM(login, password, source.ID, source.Config.(*PAMConfig), autoRegister)
 | 
	
		
			
				|  |  |  	case LoginGitHub:
 | 
	
		
			
				|  |  | -		return LoginViaGitHub(login, password, source.ID, source.Cfg.(*GitHubConfig), autoRegister)
 | 
	
		
			
				|  |  | +		return LoginViaGitHub(login, password, source.ID, source.Config.(*GitHubConfig), autoRegister)
 | 
	
		
			
				|  |  |  	}
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  	return nil, errors.InvalidLoginSourceType{Type: source.Type}
 |