54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
|
package search
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/getsentry/sentry-go"
|
||
|
|
"github.com/google/uuid"
|
||
|
|
"github.com/nmcclain/ldap"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
"goauthentik.io/internal/utils"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Request struct {
|
||
|
|
ldap.SearchRequest
|
||
|
|
BindDN string
|
||
|
|
log *log.Entry
|
||
|
|
|
||
|
|
id string
|
||
|
|
conn net.Conn
|
||
|
|
ctx context.Context
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRequest(bindDN string, searchReq ldap.SearchRequest, conn net.Conn) (*Request, *sentry.Span) {
|
||
|
|
rid := uuid.New().String()
|
||
|
|
bindDN = strings.ToLower(bindDN)
|
||
|
|
span := sentry.StartSpan(context.TODO(), "authentik.providers.ldap.search", sentry.TransactionName("authentik.providers.ldap.search"))
|
||
|
|
span.SetTag("request_uid", rid)
|
||
|
|
span.SetTag("user.username", bindDN)
|
||
|
|
span.SetTag("ak_filter", searchReq.Filter)
|
||
|
|
span.SetTag("ak_base_dn", searchReq.BaseDN)
|
||
|
|
return &Request{
|
||
|
|
SearchRequest: searchReq,
|
||
|
|
BindDN: bindDN,
|
||
|
|
conn: conn,
|
||
|
|
log: log.WithField("bindDN", bindDN).WithField("requestId", rid).WithField("scope", ldap.ScopeMap[searchReq.Scope]).WithField("client", utils.GetIP(conn.RemoteAddr())).WithField("filter", searchReq.Filter).WithField("baseDN", searchReq.BaseDN),
|
||
|
|
id: rid,
|
||
|
|
ctx: span.Context(),
|
||
|
|
}, span
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Request) Context() context.Context {
|
||
|
|
return r.ctx
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Request) Log() *log.Entry {
|
||
|
|
return r.log
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Request) RemoteAddr() string {
|
||
|
|
return utils.GetIP(r.conn.RemoteAddr())
|
||
|
|
}
|