1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
v/vlib/net/smtp/smtp.v

259 lines
5.6 KiB
V
Raw Normal View History

module smtp
/*
*
2020-07-13 17:41:23 +03:00
* smtp module
* Created by: nedimf (07/2020)
*/
import net
import net.ssl
import encoding.base64
import strings
2020-07-13 17:41:23 +03:00
import time
import io
2020-07-13 17:41:23 +03:00
const (
recv_size = 128
2020-07-13 17:41:23 +03:00
)
enum ReplyCode {
ready = 220
close = 221
auth_ok = 235
action_ok = 250
2020-07-13 17:41:23 +03:00
mail_start = 354
}
2020-07-13 17:41:23 +03:00
pub enum BodyType {
text
html
}
2020-07-13 17:41:23 +03:00
pub struct Client {
mut:
conn net.TcpConn
ssl_conn &ssl.SSLConn = unsafe { nil }
reader io.BufferedReader
2020-07-13 17:41:23 +03:00
pub:
server string
port int = 25
username string
password string
from string
ssl bool
starttls bool
2020-07-13 17:41:23 +03:00
pub mut:
is_open bool
encrypted bool
}
2020-07-13 17:41:23 +03:00
pub struct Mail {
from string
to string
cc string
bcc string
date time.Time = time.now()
subject string
body_type BodyType
body string
}
// new_client returns a new SMTP client and connects to it
pub fn new_client(config Client) !&Client {
if config.ssl && config.starttls {
return error('Can not use both implicit SSL and STARTTLS')
}
mut c := &Client{
...config
}
c.reconnect()!
2020-07-13 17:41:23 +03:00
return c
}
// reconnect reconnects to the SMTP server if the connection was closed
pub fn (mut c Client) reconnect() ! {
if c.is_open {
return error('Already connected to server')
}
2020-07-13 17:41:23 +03:00
conn := net.dial_tcp('${c.server}:${c.port}') or { return error('Connecting to server failed') }
c.conn = conn
if c.ssl {
c.connect_ssl()!
} else {
c.reader = io.new_buffered_reader(reader: c.conn)
}
2020-07-13 17:41:23 +03:00
c.expect_reply(.ready) or { return error('Received invalid response from server') }
c.send_ehlo() or { return error('Sending EHLO packet failed') }
if c.starttls && !c.encrypted {
c.send_starttls() or { return error('Sending STARTTLS failed') }
}
2020-07-13 17:41:23 +03:00
c.send_auth() or { return error('Authenticating to server failed') }
c.is_open = true
}
// send sends an email
pub fn (mut c Client) send(config Mail) ! {
if !c.is_open {
return error('Disconnected from server')
}
2020-07-13 17:41:23 +03:00
from := if config.from != '' { config.from } else { c.from }
c.send_mailfrom(from) or { return error('Sending mailfrom failed') }
c.send_mailto(config.to) or { return error('Sending mailto failed') }
c.send_data() or { return error('Sending mail data failed') }
2021-01-23 01:24:48 +03:00
c.send_body(Mail{
...config
from: from
}) or { return error('Sending mail body failed') }
2020-07-13 17:41:23 +03:00
}
// quit closes the connection to the server
pub fn (mut c Client) quit() ! {
c.send_str('QUIT\r\n')!
c.expect_reply(.close)!
if c.encrypted {
c.ssl_conn.shutdown()!
} else {
c.conn.close()!
}
2020-07-13 17:41:23 +03:00
c.is_open = false
c.encrypted = false
}
fn (mut c Client) connect_ssl() ! {
c.ssl_conn = ssl.new_ssl_conn()!
c.ssl_conn.connect(mut c.conn, c.server) or {
return error('Connecting to server using OpenSSL failed: ${err}')
}
c.reader = io.new_buffered_reader(reader: c.ssl_conn)
c.encrypted = true
2020-07-13 17:41:23 +03:00
}
// expect_reply checks if the SMTP server replied with the expected reply code
fn (mut c Client) expect_reply(expected ReplyCode) ! {
mut str := ''
for {
str = c.reader.read_line()!
if str.len < 4 {
return error('Invalid SMTP response: ${str}')
}
if str.runes()[3] == `-` {
continue
} else {
break
}
}
2020-07-13 17:41:23 +03:00
$if smtp_debug ? {
eprintln('\n\n[RECV]')
2020-07-13 17:41:23 +03:00
eprint(str)
}
2020-07-13 17:41:23 +03:00
if str.len >= 3 {
2020-07-13 17:41:23 +03:00
status := str[..3].int()
if unsafe { ReplyCode(status) } != expected {
return error('Received unexpected status code ${status}, expecting ${expected}')
}
} else {
return error('Recieved unexpected SMTP data: ${str}')
}
2020-07-13 17:41:23 +03:00
}
[inline]
fn (mut c Client) send_str(s string) ! {
$if smtp_debug ? {
2020-07-13 17:41:23 +03:00
eprintln('\n\n[SEND START]')
eprint(s.trim_space())
eprintln('\n[SEND END]')
}
if c.encrypted {
c.ssl_conn.write(s.bytes())!
} else {
c.conn.write(s.bytes())!
}
}
2020-07-13 17:41:23 +03:00
[inline]
fn (mut c Client) send_ehlo() ! {
c.send_str('EHLO ${c.server}\r\n')!
c.expect_reply(.action_ok)!
2020-07-13 17:41:23 +03:00
}
[inline]
fn (mut c Client) send_starttls() ! {
c.send_str('STARTTLS\r\n')!
c.expect_reply(.ready)!
c.connect_ssl()!
}
2020-07-13 17:41:23 +03:00
[inline]
fn (mut c Client) send_auth() ! {
if c.username.len == 0 {
return
}
mut sb := strings.new_builder(100)
2022-04-15 14:58:56 +03:00
sb.write_u8(0)
sb.write_string(c.username)
2022-04-15 14:58:56 +03:00
sb.write_u8(0)
sb.write_string(c.password)
2020-07-13 17:41:23 +03:00
a := sb.str()
auth := 'AUTH PLAIN ${base64.encode_str(a)}\r\n'
c.send_str(auth)!
c.expect_reply(.auth_ok)!
2020-07-13 17:41:23 +03:00
}
fn (mut c Client) send_mailfrom(from string) ! {
c.send_str('MAIL FROM: <${from}>\r\n')!
c.expect_reply(.action_ok)!
2020-07-13 17:41:23 +03:00
}
fn (mut c Client) send_mailto(to string) ! {
for rcpt in to.split(';') {
c.send_str('RCPT TO: <${rcpt}>\r\n')!
c.expect_reply(.action_ok)!
}
}
fn (mut c Client) send_data() ! {
c.send_str('DATA\r\n')!
c.expect_reply(.mail_start)!
}
fn (mut c Client) send_body(cfg Mail) ! {
2020-07-13 17:41:23 +03:00
is_html := cfg.body_type == .html
date := cfg.date.custom_format('ddd, D MMM YYYY HH:mm ZZ')
nonascii_subject := cfg.subject.bytes().any(it < u8(` `) || it > u8(`~`))
2020-07-13 17:41:23 +03:00
mut sb := strings.new_builder(200)
sb.write_string('From: ${cfg.from}\r\n')
sb.write_string('To: <${cfg.to.split(';').join('>; <')}>\r\n')
sb.write_string('Cc: <${cfg.cc.split(';').join('>; <')}>\r\n')
sb.write_string('Bcc: <${cfg.bcc.split(';').join('>; <')}>\r\n')
sb.write_string('Date: ${date}\r\n')
if nonascii_subject {
// handle UTF-8 subjects according RFC 1342
sb.write_string('Subject: =?utf-8?B?' + base64.encode_str(cfg.subject) + '?=\r\n')
} else {
sb.write_string('Subject: ${cfg.subject}\r\n')
}
2020-07-13 17:41:23 +03:00
if is_html {
sb.write_string('Content-Type: text/html; charset=UTF-8\r\n')
} else {
sb.write_string('Content-Type: text/plain; charset=UTF-8\r\n')
}
sb.write_string('Content-Transfer-Encoding: base64')
sb.write_string('\r\n\r\n')
sb.write_string(base64.encode_str(cfg.body))
sb.write_string('\r\n.\r\n')
c.send_str(sb.str())!
c.expect_reply(.action_ok)!
}