Authentication hook

This commit is contained in:
Andreas Neue 2016-07-25 18:48:41 +02:00
parent ad4276152e
commit 9061079361
2 changed files with 25 additions and 9 deletions

View File

@ -16,6 +16,7 @@ import (
type Client interface {
Name() string
Password() string
Register(bool)
Send(*irc.Message)
Receive(*irc.Message)
@ -74,6 +75,10 @@ func (cl *RemoteClient) Name() string {
return cl.name
}
func (cl *RemoteClient) Password() string {
return cl.password
}
func (cl *RemoteClient) Register(success bool) {
cl.registered <- success
}

View File

@ -48,6 +48,8 @@ type Server struct {
connectionsCurrent float64
connectionsCount float64
queueLen float64
authCallback func(name, pass string) bool
}
// Create a new server instance.
@ -82,6 +84,10 @@ func NewServer(configPath, software, version string) *Server {
return sv
}
func (sv *Server) SetAuthCallback(authCB func(name, pass string) bool) {
sv.authCallback = authCB
}
// Open the listening port and start the main server loop.
func (sv *Server) Run() {
xlog.Info("%s/%s", sv.software, sv.version)
@ -142,8 +148,14 @@ func (sv *Server) dispatcher() {
clid := strings.ToLower(cl.Name())
if _, exists := sv.clients[clid]; exists {
cl.Register(false)
xlog.Info("Client registration failed: '%s'", clid)
} else {
xlog.Info("Client registration failed: '%s' (client exists)", clid)
continue
}
if !sv.authCallback(cl.Name(), cl.Password()) {
cl.Register(false)
xlog.Info("Client registration failed: '%s' (wrong password)", clid)
continue
}
sv.clients[clid] = cl
sv.sendLogon(cl.Name())
sv.connectionsCurrent = float64(len(sv.clients))
@ -151,7 +163,6 @@ func (sv *Server) dispatcher() {
xlog.Info("Client registered: '%s'", clid)
xlog.Info("Server has %d client(s)", len(sv.clients))
xlog.Debug("Goroutines running: %d", runtime.NumGoroutine())
}
case cl := <-sv.delq:
clid := strings.ToLower(cl.Name())
cl.Destroy()