mirror of
https://github.com/foxcpp/maddy.git
synced 2025-04-05 22:17:39 +03:00
305fdddf24
It is useful to define background tasks lifetimes more precisely, especially involving timeouts and other cancellation methods. On top of that, several tracing facilities are context-based (e.g. runtime/trace), so it is possible to use them now.
88 lines
3.1 KiB
Go
88 lines
3.1 KiB
Go
package module
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/emersion/go-message/textproto"
|
|
"github.com/emersion/go-msgauth/authres"
|
|
"github.com/emersion/go-smtp"
|
|
"github.com/foxcpp/maddy/internal/buffer"
|
|
)
|
|
|
|
// Check is the module interface that is meant for read-only (with the
|
|
// exception of the message header modifications) (meta-)data checking.
|
|
type Check interface {
|
|
// CheckStateForMsg initializes the "internal" check state required for
|
|
// processing of the new message.
|
|
//
|
|
// NOTE: Returned CheckState object must be hashable (usable as a map key).
|
|
// This is used to deduplicate Check* calls, the easiest way to achieve
|
|
// this is to have CheckState as a pointer to some struct, all pointers
|
|
// are hashable.
|
|
CheckStateForMsg(ctx context.Context, msgMeta *MsgMetadata) (CheckState, error)
|
|
}
|
|
|
|
// EarlyCheck is an optional module interface that can be implemented
|
|
// by module implementing Check.
|
|
//
|
|
// It is used as an optimization to reject obviously malicious connections
|
|
// before allocating resources for SMTP session.
|
|
//
|
|
// The Status of this check is accept (no error) or reject (error) only, no
|
|
// advanced handling is available (such as 'quarantine' action and headers
|
|
// prepending).
|
|
type EarlyCheck interface {
|
|
CheckConnection(ctx context.Context, state *smtp.ConnectionState) error
|
|
}
|
|
|
|
type CheckState interface {
|
|
// CheckConnection is executed once when client sends a new message.
|
|
//
|
|
// Result may be cached for the whole client connection so this function
|
|
// may not be called sometimes.
|
|
CheckConnection(ctx context.Context) CheckResult
|
|
|
|
// CheckSender is executed once when client sends the message sender
|
|
// information (e.g. on the MAIL FROM command).
|
|
CheckSender(ctx context.Context, mailFrom string) CheckResult
|
|
|
|
// CheckRcpt is executed for each recipient when its address is received
|
|
// from the client (e.g. on the RCPT TO command).
|
|
CheckRcpt(ctx context.Context, rcptTo string) CheckResult
|
|
|
|
// CheckBody is executed once after the message body is received and
|
|
// buffered in memory or on disk.
|
|
//
|
|
// Check code should use passed mutex when working with the message header.
|
|
// Body can be read without locking it since it is read-only.
|
|
CheckBody(ctx context.Context, header textproto.Header, body buffer.Buffer) CheckResult
|
|
|
|
// Close is called after the message processing ends, even if any of the
|
|
// Check* functions return an error.
|
|
Close() error
|
|
}
|
|
|
|
type CheckResult struct {
|
|
// Reason is the error that is reported to the message source
|
|
// if check decided that the message should be rejected.
|
|
Reason error
|
|
|
|
// Reject is the flag that specifies that the message
|
|
// should be rejected.
|
|
Reject bool
|
|
|
|
// Quarantine is the flag that specifies that the message
|
|
// is considered "possibly malicious" and should be
|
|
// put into Junk mailbox.
|
|
//
|
|
// This value is copied into MsgMetadata by the msgpipeline.
|
|
Quarantine bool
|
|
|
|
// AuthResult is the information that is supposed to
|
|
// be included in Authentication-Results header.
|
|
AuthResult []authres.Result
|
|
|
|
// Header is the header fields that should be
|
|
// added to the header after all checks.
|
|
Header textproto.Header
|
|
}
|