mirrors/maddy
1
0
Fork 0
mirror of https://github.com/foxcpp/maddy.git synced 2025-04-05 14:07:38 +03:00
Code Issues Projects Releases Packages Wiki Activity

targets/remote: Implement experimental connection caching

Browse source 
Closes #235.
This commit is contained in:
fox.cpp 2020-06-13 21:18:20 +03:00
parent aa9e3c6a40
commit 7f7393e480
No known key found for this signature in database
GPG key ID: 5B991F6215D2FCC0
7 changed files with 349 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

12
internal/testutils/smtp_server.go
View file

@ -28,6 +28,8 @@ type SMTPMessage struct {
type SMTPBackend struct {
Messages []*SMTPMessage
MailFromCounter int
SessionCounter int
SourceEndpoints map[string]struct{}
AuthErr error
MailErr error
@ -40,6 +42,11 @@ func (be *SMTPBackend) Login(state *smtp.ConnectionState, username, password str
if be.AuthErr != nil {
return nil, be.AuthErr
}
be.SessionCounter++
if be.SourceEndpoints == nil {
be.SourceEndpoints = make(map[string]struct{})
}
be.SourceEndpoints[state.RemoteAddr.String()] = struct{}{}
return &session{
backend: be,
user: username,
@ -49,6 +56,11 @@ func (be *SMTPBackend) Login(state *smtp.ConnectionState, username, password str
}
func (be *SMTPBackend) AnonymousLogin(state *smtp.ConnectionState) (smtp.Session, error) {
be.SessionCounter++
if be.SourceEndpoints == nil {
be.SourceEndpoints = make(map[string]struct{})
}
be.SourceEndpoints[state.RemoteAddr.String()] = struct{}{}
return &session{backend: be, state: state}, nil
}