37 lines
995 B
Go
37 lines
995 B
Go
package pop3
|
|
|
|
import (
|
|
"net"
|
|
"regexp"
|
|
|
|
"github.com/zmap/zgrab2"
|
|
)
|
|
|
|
// This is the regex used in zgrab.
|
|
var pop3EndRegex = regexp.MustCompile(`(?:\r\n\.\r\n$)|(?:\r\n$)`)
|
|
|
|
const readBufferSize int = 0x10000
|
|
|
|
// Connection wraps the state and access to the SMTP connection.
|
|
type Connection struct {
|
|
Conn net.Conn
|
|
}
|
|
|
|
// ReadResponse reads from the connection until it matches the pop3EndRegex. Copied from the original zgrab.
|
|
// TODO: Catch corner cases, parse out success/error character.
|
|
func (conn *Connection) ReadResponse() (string, error) {
|
|
ret := make([]byte, readBufferSize)
|
|
n, err := zgrab2.ReadUntilRegex(conn.Conn, ret, pop3EndRegex)
|
|
if err != nil {
|
|
return "", nil
|
|
}
|
|
return string(ret[0:n]), nil
|
|
}
|
|
|
|
// SendCommand sends a command, followed by a CRLF, then wait for / read the server's response.
|
|
func (conn *Connection) SendCommand(cmd string) (string, error) {
|
|
if _, err := conn.Conn.Write([]byte(cmd + "\r\n")); err != nil {
|
|
return "", err
|
|
}
|
|
return conn.ReadResponse()
|
|
} |