forked from Shiloh/githaven
103 lines
3.5 KiB
Go
103 lines
3.5 KiB
Go
// Copyright github.com/juju2013. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// package ldap provide functions & structure to query a LDAP ldap directory
|
|
// For now, it's mainly tested again an MS Active Directory service, see README.md for more information
|
|
package ldap
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gogits/gogs/modules/ldap"
|
|
"github.com/gogits/gogs/modules/log"
|
|
)
|
|
|
|
// Basic LDAP authentication service
|
|
type Ldapsource struct {
|
|
Name string // canonical name (ie. corporate.ad)
|
|
Host string // LDAP host
|
|
Port int // port number
|
|
UseSSL bool // Use SSL
|
|
BaseDN string // Base DN
|
|
AttributeUsername string // Username attribute
|
|
AttributeName string // First name attribute
|
|
AttributeSurname string // Surname attribute
|
|
AttributeMail string // E-mail attribute
|
|
Filter string // Query filter to validate entry
|
|
MsAdSAFormat string // in the case of MS AD Simple Authen, the format to use (see: http://msdn.microsoft.com/en-us/library/cc223499.aspx)
|
|
Enabled bool // if this source is disabled
|
|
}
|
|
|
|
//Global LDAP directory pool
|
|
var (
|
|
Authensource []Ldapsource
|
|
)
|
|
|
|
// Add a new source (LDAP directory) to the global pool
|
|
func AddSource(name string, host string, port int, usessl bool, basedn string, attribcn string, attribname string, attribsn string, attribmail string, filter string, msadsaformat string) {
|
|
ldaphost := Ldapsource{name, host, port, usessl, basedn, attribcn, attribname, attribsn, attribmail, filter, msadsaformat, true}
|
|
Authensource = append(Authensource, ldaphost)
|
|
}
|
|
|
|
//LoginUser : try to login an user to LDAP sources, return requested (attribute,true) if ok, ("",false) other wise
|
|
//First match wins
|
|
//Returns first attribute if exists
|
|
func LoginUser(name, passwd string) (cn, fn, sn, mail string, r bool) {
|
|
r = false
|
|
for _, ls := range Authensource {
|
|
cn, fn, sn, mail, r = ls.SearchEntry(name, passwd)
|
|
if r {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// searchEntry : search an LDAP source if an entry (name, passwd) is valide and in the specific filter
|
|
func (ls Ldapsource) SearchEntry(name, passwd string) (string, string, string, string, bool) {
|
|
l, err := ldapDial(ls)
|
|
if err != nil {
|
|
log.Error(4, "LDAP Connect error, %s:%v", ls.Host, err)
|
|
ls.Enabled = false
|
|
return "", "", "", "", false
|
|
}
|
|
defer l.Close()
|
|
|
|
nx := fmt.Sprintf(ls.MsAdSAFormat, name)
|
|
err = l.Bind(nx, passwd)
|
|
if err != nil {
|
|
log.Debug("LDAP Authan failed for %s, reason: %s", nx, err.Error())
|
|
return "", "", "", "", false
|
|
}
|
|
|
|
search := ldap.NewSearchRequest(
|
|
ls.BaseDN,
|
|
ldap.ScopeWholeSubtree, ldap.NeverDerefAliases, 0, 0, false,
|
|
fmt.Sprintf(ls.Filter, name),
|
|
[]string{ls.AttributeUsername, ls.AttributeName, ls.AttributeSurname, ls.AttributeMail},
|
|
nil)
|
|
sr, err := l.Search(search)
|
|
if err != nil {
|
|
log.Debug("LDAP Authen OK but not in filter %s", name)
|
|
return "", "", "", "", false
|
|
}
|
|
log.Debug("LDAP Authen OK: %s", name)
|
|
if len(sr.Entries) > 0 {
|
|
cn := sr.Entries[0].GetAttributeValue(ls.AttributeUsername)
|
|
name := sr.Entries[0].GetAttributeValue(ls.AttributeName)
|
|
sn := sr.Entries[0].GetAttributeValue(ls.AttributeSurname)
|
|
mail := sr.Entries[0].GetAttributeValue(ls.AttributeMail)
|
|
return cn, name, sn, mail, true
|
|
}
|
|
return "", "", "", "", true
|
|
}
|
|
|
|
func ldapDial(ls Ldapsource) (*ldap.Conn, error) {
|
|
if ls.UseSSL {
|
|
return ldap.DialTLS("tcp", fmt.Sprintf("%s:%d", ls.Host, ls.Port), nil)
|
|
} else {
|
|
return ldap.Dial("tcp", fmt.Sprintf("%s:%d", ls.Host, ls.Port))
|
|
}
|
|
}
|