1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
package milter
// Response represents a response structure returned by callback
// handlers to indicate how the milter server should proceed
type Response interface {
Response() *Message
Continue() bool
}
// SimpleResponse type to define list of pre-defined responses
type SimpleResponse byte
// Response returns a Message object reference
func (r SimpleResponse) Response() *Message {
return &Message{byte(r), nil}
}
// Continue to process milter messages only if current code is Continue
func (r SimpleResponse) Continue() bool {
return ActionCode(r) == ActContinue
}
// Define standard responses with no data
const (
RespAccept = SimpleResponse(ActAccept)
RespContinue = SimpleResponse(ActContinue)
RespDiscard = SimpleResponse(ActDiscard)
RespReject = SimpleResponse(ActReject)
RespTempFail = SimpleResponse(ActTempFail)
)
// CustomResponse is a response instance used by callback handlers to indicate
// how the milter should continue processing of current message
type CustomResponse struct {
code byte
data []byte
}
// Response returns message instance with data
func (c *CustomResponse) Response() *Message {
return &Message{c.code, c.data}
}
// Continue returns false if milter chain should be stopped, true otherwise
func (c *CustomResponse) Continue() bool {
for _, q := range []ActionCode{ActAccept, ActDiscard, ActReject, ActTempFail} {
if c.code == byte(q) {
return false
}
}
return true
}
// NewResponse generates a new CustomResponse suitable for WritePacket
func NewResponse(code byte, data []byte) *CustomResponse {
return &CustomResponse{code, data}
}
// NewResponseStr generates a new CustomResponse with string payload
func NewResponseStr(code byte, data string) *CustomResponse {
return NewResponse(code, []byte(data+null))
}
|
