/deps/gen_smtp/src/gen_smtp_client.erl

https://code.google.com/p/zotonic/ · Erlang · 1019 lines · 903 code · 38 blank · 78 comment · 7 complexity · 31b1c68e5a429b5d7ca3496a09e9eb7b MD5 · raw file 

    

  1. %%% Copyright 2009 Andrew Thompson <andrew@hijacked.us>. All rights reserved.
    2. 

  2. %%%
    3. 

  3. %%% Redistribution and use in source and binary forms, with or without
    4. 

  4. %%% modification, are permitted provided that the following conditions are met:
    5. 

  5. %%%
    6. 

  6. %%%   1. Redistributions of source code must retain the above copyright notice,
    7. 

  7. %%%      this list of conditions and the following disclaimer.
    8. 

  8. %%%   2. Redistributions in binary form must reproduce the above copyright
    9. 

  9. %%%      notice, this list of conditions and the following disclaimer in the
    10. 

  10. %%%      documentation and/or other materials provided with the distribution.
    11. 

  11. %%%
    12. 

  12. %%% THIS SOFTWARE IS PROVIDED BY THE FREEBSD PROJECT ``AS IS'' AND ANY EXPRESS OR
    13. 

  13. %%% IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
    14. 

  14. %%% MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
    15. 

  15. %%% EVENT SHALL THE FREEBSD PROJECT OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
    16. 

  16. %%% INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
    17. 

  17. %%% (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
    18. 

  18. %%% LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
    19. 

  19. %%% ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
    20. 

  20. %%% (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
    21. 

  21. %%% SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
    22. 

  22. 

  23. %% @doc A simple SMTP client used for sending mail - assumes relaying via a
    24. 

  24. %% smarthost.
    25. 

  25. 

  26. -module(gen_smtp_client).
    27. 

  27. 

  28. -define(DEFAULT_OPTIONS, [
    29. 

  29. 		{ssl, false}, % whether to connect on 465 in ssl mode
    30. 

  30. 		{tls, if_available}, % always, never, if_available
    31. 

  31. 		{auth, if_available},
    32. 

  32. 		{hostname, smtp_util:guess_FQDN()},
    33. 

  33. 		{retries, 1} % how many retries per smtp host on temporary failure
    34. 

  34. 	]).
    35. 

  35. 

  36. -define(AUTH_PREFERENCE, [
    37. 

  37. 		"CRAM-MD5",
    38. 

  38. 		"LOGIN",
    39. 

  39. 		"PLAIN"
    40. 

  40. 	]).
    41. 

  41. 

  42. -define(TIMEOUT, 1200000).
    43. 

  43. 

  44. -ifdef(TEST).
    45. 

  45. -include_lib("eunit/include/eunit.hrl").
    46. 

  46. -compile(export_all).
    47. 

  47. -else.
    48. 

  48. -export([send/2, send/3, send_blocking/2]).
    49. 

  49. -endif.
    50. 

  50. 

  51. -type email() :: {string() | binary(), [string() | binary(), ...], string() | binary() | function()}. 
    52. 

  52. 

  53. -spec send(Email :: {string() | binary(), [string() | binary(), ...], string() | binary() | function()}, Options :: list()) -> {'ok', pid()} | {'error', any()}.
    54. 

  54. %% @doc Send an email in a non-blocking fashion via a spawned_linked process.
    55. 

  55. %% The process will exit abnormally on a send failure.
    56. 

  56. send(Email, Options) ->
    57. 

  57. 	send(Email, Options, undefined).
    58. 

  58. 

  59. %% @doc Send an email nonblocking and invoke a callback with the result of the send.
    60. 

  60. %% The callback will receive either `{ok, Receipt}' where Receipt is the SMTP server's receipt
    61. 

  61. %% identifier,  `{error, Type, Message}' or `{exit, ExitReason}', as the single argument.
    62. 

  62. -spec send(Email :: {string() | binary(), [string() | binary(), ...], string() | binary() | function()}, Options :: list(), Callback :: function() | 'undefined') -> {'ok', pid()} | {'error', any()}.
    63. 

  63. send(Email, Options, Callback) ->
    64. 

  64. 	NewOptions = lists:ukeymerge(1, lists:sort(Options),
    65. 

  65. 		lists:sort(?DEFAULT_OPTIONS)),
    66. 

  66. 	case check_options(NewOptions) of
    67. 

  67. 		ok when is_function(Callback) ->
    68. 

  68. 			spawn(fun() ->
    69. 

  69. 						process_flag(trap_exit, true),
    70. 

  70. 						Pid = spawn_link(fun() ->
    71. 

  71. 									send_it_nonblock(Email, NewOptions, Callback)
    72. 

  72. 							end
    73. 

  73. 						),
    74. 

  74. 						receive
    75. 

  75. 							{'EXIT', Pid, Reason} ->
    76. 

  76. 								case Reason of
    77. 

  77. 									X when X == normal; X == shutdown ->
    78. 

  78. 										ok;
    79. 

  79. 									Error ->
    80. 

  80. 										Callback({exit, Error})
    81. 

  81. 								end
    82. 

  82. 						end
    83. 

  83. 				end);
    84. 

  84. 		ok ->
    85. 

  85. 			Pid = spawn_link(fun () ->
    86. 

  86. 						send_it_nonblock(Email, NewOptions, Callback)
    87. 

  87. 				end
    88. 

  88. 			),
    89. 

  89. 			{ok, Pid};
    90. 

  90. 		{error, Reason} ->
    91. 

  91. 			{error, Reason}
    92. 

  92. 	end.
    93. 

  93. 

  94. -spec send_blocking(Email :: {string() | binary(), [string() | binary(), ...], string() | binary() | function()}, Options :: list()) -> binary() | {'error', atom(), any()} | {'error', any()}.
    95. 

  95. %% @doc Send an email and block waiting for the reply. Returns either a binary that contains
    96. 

  96. %% the SMTP server's receipt or `{error, Type, Message}' or `{error, Reason}'.
    97. 

  97. send_blocking(Email, Options) ->
    98. 

  98. 	NewOptions = lists:ukeymerge(1, lists:sort(Options),
    99. 

  99. 		lists:sort(?DEFAULT_OPTIONS)),
    100. 

  100. 	case check_options(NewOptions) of
    101. 

  101. 		ok ->
    102. 

  102. 			send_it(Email, NewOptions);
    103. 

  103. 		{error, Reason} ->
    104. 

  104. 			{error, Reason}
    105. 

  105. 	end.
    106. 

    107. 

  107. -spec send_it_nonblock(Email :: email(), Options :: list(), Callback :: function() | 'undefined') ->{'ok', binary()} | {'error', any(), any()}.
    108. 

  108. send_it_nonblock(Email, Options, Callback) ->
    109. 

  109. 	case send_it(Email, Options) of
    110. 

  110. 		{error, Type, Message} when is_function(Callback) ->
    111. 

  111. 			Callback({error, Type, Message}),
    112. 

  112. 			{error, Type, Message};
    113. 

  113. 		{error, Type, Message} ->
    114. 

  114. 			erlang:exit({error, Type, Message});
    115. 

  115. 		Receipt when is_function(Callback) ->
    116. 

  116. 			Callback({ok, Receipt}),
    117. 

  117. 			{ok, Receipt};
    118. 

  118. 		Receipt ->
    119. 

  119. 			{ok, Receipt}
    120. 

  120. 	end.
    121. 

    122. 

  122. -spec send_it(Email :: {string() | binary(), [string() | binary(), ...], string() | binary() | function()}, Options :: list()) -> binary() | {'error', any(), any()}.
    123. 

  123. send_it(Email, Options) ->
    124. 

  124. 	RelayDomain = proplists:get_value(relay, Options),
    125. 

  125. 	MXRecords = case proplists:get_value(no_mx_lookups, Options) of
    126. 

  126. 		true ->
    127. 

  127. 			[];
    128. 

  128. 		_ ->
    129. 

  129. 			smtp_util:mxlookup(RelayDomain)
    130. 

  130. 	end,
    131. 

  131. 	%io:format("MX records for ~s are ~p~n", [RelayDomain, MXRecords]),
    132. 

  132. 	Hosts = case MXRecords of
    133. 

  133. 		[] ->
    134. 

  134. 			[{0, RelayDomain}]; % maybe we're supposed to relay to a host directly
    135. 

  135. 		_ ->
    136. 

  136. 			MXRecords
    137. 

  137. 	end,
    138. 

  138. 	try_smtp_sessions(Hosts, Email, Options, []).
    139. 

    140. 

  140. -spec try_smtp_sessions(Hosts :: [{non_neg_integer(), string()}, ...], Email :: email(), Options :: list(), RetryList :: list()) -> binary() | {'error', any(), any()}.
    141. 

  141. try_smtp_sessions([{Distance, Host} | Tail], Email, Options, RetryList) ->
    142. 

  142. 	Retries = proplists:get_value(retries, Options),
    143. 

  143. 	try do_smtp_session(Host, Email, Options) of
    144. 

  144. 		Res -> Res
    145. 

  145. 	catch
    146. 

  146. 		throw:{permanent_failure, Message} ->
    147. 

  147. 			% permanent failure means no retries, and don't even continue with other hosts
    148. 

  148. 			{error, no_more_hosts, {permanent_failure, Host, Message}};
    149. 

  149. 		throw:{FailureType, Message} ->
    150. 

  150. 			case proplists:get_value(Host, RetryList) of
    151. 

  151. 				RetryCount when is_integer(RetryCount), RetryCount >= Retries ->
    152. 

  152. 					% out of chances
    153. 

  153. 					%io:format("retries for ~s exceeded (~p of ~p)~n", [Host, RetryCount, Retries]),
    154. 

  154. 					NewHosts = Tail,
    155. 

  155. 					NewRetryList = lists:keydelete(Host, 1, RetryList);
    156. 

  156. 				RetryCount when is_integer(RetryCount) ->
    157. 

  157. 					%io:format("scheduling ~s for retry (~p of ~p)~n", [Host, RetryCount, Retries]),
    158. 

  158. 					NewHosts = Tail ++ [{Distance, Host}],
    159. 

  159. 					NewRetryList = lists:keydelete(Host, 1, RetryList) ++ [{Host, RetryCount + 1}];
    160. 

  160. 				_ when Retries == 0 ->
    161. 

  161. 					% done retrying completely
    162. 

  162. 					NewHosts = Tail,
    163. 

  163. 					NewRetryList = lists:keydelete(Host, 1, RetryList);
    164. 

  164. 				_ ->
    165. 

  165. 					% otherwise...
    166. 

  166. 					%io:format("scheduling ~s for retry (~p of ~p)~n", [Host, 1, Retries]),
    167. 

  167. 					NewHosts = Tail ++ [{Distance, Host}],
    168. 

  168. 					NewRetryList = lists:keydelete(Host, 1, RetryList) ++ [{Host, 1}]
    169. 

  169. 			end,
    170. 

  170. 			case NewHosts of
    171. 

  171. 				[] ->
    172. 

  172. 					{error, retries_exceeded, {FailureType, Host, Message}};
    173. 

  173. 				_ ->
    174. 

  174. 					try_smtp_sessions(NewHosts, Email, Options, NewRetryList)
    175. 

  175. 			end
    176. 

  176. 	end.
    177. 

    178. 

  178. -spec do_smtp_session(Host :: string(), Email :: email(), Options :: list()) -> binary().
    179. 

  179. do_smtp_session(Host, Email, Options) ->
    180. 

  180. 	{ok, Socket, _Host, _Banner} = connect(Host, Options),
    181. 

  181. 	%io:format("connected to ~s; banner was ~s~n", [Host, Banner]),
    182. 

  182. 	{ok, Extensions} = try_EHLO(Socket, Options),
    183. 

  183. 	%io:format("Extensions are ~p~n", [Extensions]),
    184. 

  184. 	{Socket2, Extensions2} = try_STARTTLS(Socket, Options, Extensions),
    185. 

  185. 	%io:format("Extensions are ~p~n", [Extensions2]),
    186. 

  186. 	_Authed = try_AUTH(Socket2, Options, proplists:get_value(<<"AUTH">>, Extensions2)),
    187. 

  187. 	%io:format("Authentication status is ~p~n", [Authed]),
    188. 

  188. 	Receipt = try_sending_it(Email, Socket2, Extensions2),
    189. 

  189. 	%io:format("Mail sending successful~n"),
    190. 

  190. 	quit(Socket2),
    191. 

  191. 	Receipt.
    192. 

    193. 

  193. -spec try_sending_it(Email :: email(), Socket :: socket:socket(), Extensions :: list()) -> binary().
    194. 

  194. try_sending_it({From, To, Body}, Socket, Extensions) ->
    195. 

  195. 	try_MAIL_FROM(From, Socket, Extensions),
    196. 

  196. 	try_RCPT_TO(To, Socket, Extensions),
    197. 

  197. 	try_DATA(Body, Socket, Extensions).
    198. 

    199. 

  199. -spec try_MAIL_FROM(From :: string() | binary(), Socket :: socket:socket(), Extensions :: list()) -> true.
    200. 

  200. try_MAIL_FROM(From, Socket, Extensions) when is_binary(From) ->
    201. 

  201. 	try_MAIL_FROM(binary_to_list(From), Socket, Extensions);
    202. 

  202. try_MAIL_FROM("<" ++ _ = From, Socket, _Extensions) ->
    203. 

  203. 	% TODO do we need to bother with SIZE?
    204. 

  204. 	socket:send(Socket, ["MAIL FROM: ", From, "\r\n"]),
    205. 

  205. 	case read_possible_multiline_reply(Socket) of
    206. 

  206. 		{ok, <<"250", _Rest/binary>>} ->
    207. 

  207. 			true;
    208. 

  208. 		{ok, <<"4", _Rest/binary>> = Msg} ->
    209. 

  209. 			quit(Socket),
    210. 

  210. 			throw({temporary_failure, Msg});
    211. 

  211. 		{ok, Msg} ->
    212. 

  212. 			%io:format("Mail FROM rejected: ~p~n", [Msg]),
    213. 

  213. 			quit(Socket),
    214. 

  214. 			throw({permanent_failure, Msg})
    215. 

  215. 	end;
    216. 

  216. try_MAIL_FROM(From, Socket, Extensions) ->
    217. 

  217. 	% someone was bad and didn't put in the angle brackets
    218. 

  218. 	try_MAIL_FROM("<"++From++">", Socket, Extensions).
    219. 

    220. 

  220. -spec try_RCPT_TO(Tos :: [binary() | string()], Socket :: socket:socket(), Extensions :: list()) -> true.
    221. 

  221. try_RCPT_TO([], _Socket, _Extensions) ->
    222. 

  222. 	true;
    223. 

  223. try_RCPT_TO([To | Tail], Socket, Extensions) when is_binary(To) ->
    224. 

  224. 	try_RCPT_TO([binary_to_list(To) | Tail], Socket, Extensions);
    225. 

  225. try_RCPT_TO(["<" ++ _ = To | Tail], Socket, Extensions) ->
    226. 

  226. 	socket:send(Socket, ["RCPT TO: ",To,"\r\n"]),
    227. 

  227. 	case read_possible_multiline_reply(Socket) of
    228. 

  228. 		{ok, <<"250", _Rest/binary>>} ->
    229. 

  229. 			try_RCPT_TO(Tail, Socket, Extensions);
    230. 

  230. 		{ok, <<"251", _Rest/binary>>} ->
    231. 

  231. 			try_RCPT_TO(Tail, Socket, Extensions);
    232. 

  232. 		{ok, <<"4", _Rest/binary>> = Msg} ->
    233. 

  233. 			quit(Socket),
    234. 

  234. 			throw({temporary_failure, Msg});
    235. 

  235. 		{ok, Msg} ->
    236. 

  236. 			quit(Socket),
    237. 

  237. 			throw({permanent_failure, Msg})
    238. 

  238. 	end;
    239. 

  239. try_RCPT_TO([To | Tail], Socket, Extensions) ->
    240. 

  240. 	% someone was bad and didn't put in the angle brackets
    241. 

  241. 	try_RCPT_TO(["<"++To++">" | Tail], Socket, Extensions).
    242. 

    243. 

  243. -spec try_DATA(Body :: binary() | function(), Socket :: socket:socket(), Extensions :: list()) -> binary().
    244. 

  244. try_DATA(Body, Socket, Extensions) when is_function(Body) ->
    245. 

  245.     try_DATA(Body(), Socket, Extensions);
    246. 

  246. try_DATA(Body, Socket, _Extensions) ->
    247. 

  247. 	socket:send(Socket, "DATA\r\n"),
    248. 

  248. 	case read_possible_multiline_reply(Socket) of
    249. 

  249. 		{ok, <<"354", _Rest/binary>>} ->
    250. 

  250. 			socket:send(Socket, [Body, "\r\n.\r\n"]),
    251. 

  251. 			case read_possible_multiline_reply(Socket) of
    252. 

  252. 				{ok, <<"250 ", Receipt/binary>>} ->
    253. 

  253. 					Receipt;
    254. 

  254. 				{ok, <<"4", _Rest2/binary>> = Msg} ->
    255. 

  255. 					quit(Socket),
    256. 

  256. 					throw({temporary_failure, Msg});
    257. 

  257. 				{ok, Msg} ->
    258. 

  258. 					quit(Socket),
    259. 

  259. 					throw({permanent_failure, Msg})
    260. 

  260. 			end;
    261. 

  261. 		{ok, <<"4", _Rest/binary>> = Msg} ->
    262. 

  262. 			quit(Socket),
    263. 

  263. 			throw({temporary_failure, Msg});
    264. 

  264. 		{ok, Msg} ->
    265. 

  265. 			quit(Socket),
    266. 

  266. 			throw({permanent_failure, Msg})
    267. 

  267. 	end.
    268. 

    269. 

  269. -spec try_AUTH(Socket :: socket:socket(), Options :: list(), AuthTypes :: [string()]) -> boolean().
    270. 

  270. try_AUTH(Socket, Options, []) ->
    271. 

  271. 	case proplists:get_value(auth, Options) of
    272. 

  272. 		always ->
    273. 

  273. 			quit(Socket),
    274. 

  274. 			erlang:throw({missing_requirement, auth});
    275. 

  275. 		_ ->
    276. 

  276. 			false
    277. 

  277. 	end;
    278. 

  278. try_AUTH(Socket, Options, undefined) ->
    279. 

  279. 	case proplists:get_value(auth, Options) of
    280. 

  280. 		always ->
    281. 

  281. 			quit(Socket),
    282. 

  282. 			erlang:throw({missing_requirement, auth});
    283. 

  283. 		_ ->
    284. 

  284. 			false
    285. 

  285. 	end;
    286. 

  286. try_AUTH(Socket, Options, AuthTypes) ->
    287. 

  287. 	case proplists:is_defined(username, Options) and
    288. 

  288. 		proplists:is_defined(password, Options) and
    289. 

  289. 		(proplists:get_value(auth, Options) =/= never) of
    290. 

  290. 		false ->
    291. 

  291. 			case proplists:get_value(auth, Options) of
    292. 

  292. 				always ->
    293. 

  293. 					quit(Socket),
    294. 

  294. 					erlang:throw({missing_requirement, auth});
    295. 

  295. 				_ ->
    296. 

  296. 					false
    297. 

  297. 			end;
    298. 

  298. 		true ->
    299. 

  299. 			Username = proplists:get_value(username, Options),
    300. 

  300. 			Password = proplists:get_value(password, Options),
    301. 

  301. 			%io:format("Auth types: ~p~n", [AuthTypes]),
    302. 

  302. 			Types = re:split(AuthTypes, " ", [{return, list}, trim]),
    303. 

  303. 			case do_AUTH(Socket, Username, Password, Types) of
    304. 

  304. 				false ->
    305. 

  305. 					case proplists:get_value(auth, Options) of
    306. 

  306. 						always ->
    307. 

  307. 							quit(Socket),
    308. 

  308. 							erlang:throw({permanent_failure, auth_failed});
    309. 

  309. 						_ ->
    310. 

  310. 							false
    311. 

  311. 					end;
    312. 

  312. 				true ->
    313. 

  313. 					true
    314. 

  314. 			end
    315. 

  315. 	end.
    316. 

    317. 

  317. -spec do_AUTH(Socket :: socket:socket(), Username :: string(), Password :: string(), Types :: [string()]) -> boolean().
    318. 

  318. do_AUTH(Socket, Username, Password, Types) ->
    319. 

  319. 	FixedTypes = [string:to_upper(X) || X <- Types],
    320. 

  320. 	%io:format("Fixed types: ~p~n", [FixedTypes]),
    321. 

  321. 	AllowedTypes = [X  || X <- ?AUTH_PREFERENCE, lists:member(X, FixedTypes)],
    322. 

  322. 	%io:format("available authentication types, in order of preference: ~p~n",
    323. 

  323. 	%	[AllowedTypes]),
    324. 

  324. 	do_AUTH_each(Socket, Username, Password, AllowedTypes).
    325. 

    326. 

  326. -spec do_AUTH_each(Socket :: socket:socket(), Username :: string() | binary(), Password :: string() | binary(), AuthTypes :: [string()]) -> boolean().
    327. 

  327. do_AUTH_each(_Socket, _Username, _Password, []) ->
    328. 

  328. 	false;
    329. 

  329. do_AUTH_each(Socket, Username, Password, ["CRAM-MD5" | Tail]) ->
    330. 

  330. 	socket:send(Socket, "AUTH CRAM-MD5\r\n"),
    331. 

  331. 	case read_possible_multiline_reply(Socket) of
    332. 

  332. 		{ok, <<"334 ", Rest/binary>>} ->
    333. 

  333. 			Seed64 = binstr:strip(binstr:strip(Rest, right, $\n), right, $\r),
    334. 

  334. 			Seed = base64:decode_to_string(Seed64),
    335. 

  335. 			Digest = smtp_util:compute_cram_digest(Password, Seed),
    336. 

  336. 			String = base64:encode(list_to_binary([Username, " ", Digest])),
    337. 

  337. 			socket:send(Socket, [String, "\r\n"]),
    338. 

  338. 			case read_possible_multiline_reply(Socket) of
    339. 

  339. 				{ok, <<"235", _Rest/binary>>} ->
    340. 

  340. 					%io:format("authentication accepted~n"),
    341. 

  341. 					true;
    342. 

  342. 				{ok, _Msg} ->
    343. 

  343. 					%io:format("authentication rejected: ~s~n", [Msg]),
    344. 

  344. 					do_AUTH_each(Socket, Username, Password, Tail)
    345. 

  345. 			end;
    346. 

  346. 		{ok, _Something} ->
    347. 

  347. 			%io:format("got ~s~n", [Something]),
    348. 

  348. 			do_AUTH_each(Socket, Username, Password, Tail)
    349. 

  349. 	end;
    350. 

  350. do_AUTH_each(Socket, Username, Password, ["LOGIN" | Tail]) ->
    351. 

  351. 	socket:send(Socket, "AUTH LOGIN\r\n"),
    352. 

  352. 	case read_possible_multiline_reply(Socket) of
    353. 

  353. 		{ok, <<"334 VXNlcm5hbWU6\r\n">>} ->
    354. 

  354. 			%io:format("username prompt~n"),
    355. 

  355. 			U = base64:encode(Username),
    356. 

  356. 			socket:send(Socket, [U,"\r\n"]),
    357. 

  357. 			case read_possible_multiline_reply(Socket) of
    358. 

  358. 				{ok, <<"334 UGFzc3dvcmQ6\r\n">>} ->
    359. 

  359. 					%io:format("password prompt~n"),
    360. 

  360. 					P = base64:encode(Password),
    361. 

  361. 					socket:send(Socket, [P,"\r\n"]),
    362. 

  362. 					case read_possible_multiline_reply(Socket) of
    363. 

  363. 						{ok, <<"235 ", _Rest/binary>>} ->
    364. 

  364. 							%io:format("authentication accepted~n"),
    365. 

  365. 							true;
    366. 

  366. 						{ok, _Msg} ->
    367. 

  367. 							%io:format("password rejected: ~s", [Msg]),
    368. 

  368. 							do_AUTH_each(Socket, Username, Password, Tail)
    369. 

  369. 					end;
    370. 

  370. 				{ok, _Msg2} ->
    371. 

  371. 					%io:format("username rejected: ~s", [Msg2]),
    372. 

  372. 					do_AUTH_each(Socket, Username, Password, Tail)
    373. 

  373. 			end;
    374. 

  374. 		{ok, _Something} ->
    375. 

  375. 			%io:format("got ~s~n", [Something]),
    376. 

  376. 			do_AUTH_each(Socket, Username, Password, Tail)
    377. 

  377. 	end;
    378. 

  378. do_AUTH_each(Socket, Username, Password, ["PLAIN" | Tail]) ->
    379. 

  379. 	AuthString = base64:encode("\0"++Username++"\0"++Password),
    380. 

  380. 	socket:send(Socket, ["AUTH PLAIN ", AuthString, "\r\n"]),
    381. 

  381. 	case read_possible_multiline_reply(Socket) of
    382. 

  382. 		{ok, <<"235", _Rest/binary>>} ->
    383. 

  383. 			%io:format("authentication accepted~n"),
    384. 

  384. 			true;
    385. 

  385. 		_Else ->
    386. 

  386. 			% TODO do we need to bother trying the multi-step PLAIN?
    387. 

  387. 			%io:format("authentication rejected~n"),
    388. 

  388. 			%io:format("~p~n", [Else]),
    389. 

  389. 			do_AUTH_each(Socket, Username, Password, Tail)
    390. 

  390. 	end;
    391. 

  391. do_AUTH_each(Socket, Username, Password, [_Type | Tail]) ->
    392. 

  392. 	%io:format("unsupported AUTH type ~s~n", [Type]),
    393. 

  393. 	do_AUTH_each(Socket, Username, Password, Tail).
    394. 

    395. 

  395. -spec try_EHLO(Socket :: socket:socket(), Options :: list()) -> {ok, list()}.
    396. 

  396. try_EHLO(Socket, Options) ->
    397. 

  397. 	ok = socket:send(Socket, ["EHLO ", proplists:get_value(hostname, Options, smtp_util:guess_FQDN()), "\r\n"]),
    398. 

  398. 	case read_possible_multiline_reply(Socket) of
    399. 

  399. 		{ok, <<"500", _Rest/binary>>} ->
    400. 

  400. 			% Unrecognized command, fall back to HELO
    401. 

  401. 			try_HELO(Socket, Options);
    402. 

  402. 		{ok, <<"4", _Rest/binary>> = Msg} ->
    403. 

  403. 			quit(Socket),
    404. 

  404. 			throw({temporary_failure, Msg});
    405. 

  405. 		{ok, Reply} ->
    406. 

  406. 			{ok, parse_extensions(Reply)}
    407. 

  407. 	end.
    408. 

    409. 

  409. -spec try_HELO(Socket :: socket:socket(), Options :: list()) -> {ok, list()}.
    410. 

  410. try_HELO(Socket, Options) ->
    411. 

  411. 	ok = socket:send(Socket, ["HELO ", proplists:get_value(hostname, Options, smtp_util:guess_FQDN()), "\r\n"]),
    412. 

  412. 	case read_possible_multiline_reply(Socket) of
    413. 

  413. 		{ok, <<"250", _Rest/binary>>} ->
    414. 

  414. 			{ok, []};
    415. 

  415. 		{ok, <<"4", _Rest/binary>> = Msg} ->
    416. 

  416. 			quit(Socket),
    417. 

  417. 			throw({temporary_failure, Msg});
    418. 

  418. 		{ok, Msg} ->
    419. 

  419. 			quit(Socket),
    420. 

  420. 			throw({permanent_failure, Msg})
    421. 

  421. 	end.
    422. 

    423. 

  423. % check if we should try to do TLS
    424. 

  424. -spec try_STARTTLS(Socket :: socket:socket(), Options :: list(), Extensions :: list()) -> {socket:socket(), list()}.
    425. 

  425. try_STARTTLS(Socket, Options, Extensions) ->
    426. 

  426. 		case {proplists:get_value(tls, Options),
    427. 

  427. 				proplists:get_value(<<"STARTTLS">>, Extensions)} of
    428. 

  428. 			{Atom, true} when Atom =:= always; Atom =:= if_available ->
    429. 

  429. 			%io:format("Starting TLS~n"),
    430. 

  430. 			case {do_STARTTLS(Socket, Options), Atom} of
    431. 

  431. 				{false, always} ->
    432. 

  432. 					%io:format("TLS failed~n"),
    433. 

  433. 					quit(Socket),
    434. 

  434. 					erlang:throw({temporary_failure, tls_failed});
    435. 

  435. 				{false, if_available} ->
    436. 

  436. 					%io:format("TLS failed~n"),
    437. 

  437. 					{Socket, Extensions};
    438. 

  438. 				{{S, E}, _} ->
    439. 

  439. 					%io:format("TLS started~n"),
    440. 

  440. 					{S, E}
    441. 

  441. 			end;
    442. 

  442. 		{always, _} ->
    443. 

  443. 			quit(Socket),
    444. 

  444. 			erlang:throw({missing_requirement, tls});
    445. 

  445. 		_ ->
    446. 

  446. 			{Socket, Extensions}
    447. 

  447. 	end.
    448. 

    449. 

  449. %% attempt to upgrade socket to TLS
    450. 

  450. -spec do_STARTTLS(Socket :: socket:socket(), Options :: list()) -> {socket:socket(), list()} | false.
    451. 

  451. do_STARTTLS(Socket, Options) ->
    452. 

  452. 	socket:send(Socket, "STARTTLS\r\n"),
    453. 

  453. 	case read_possible_multiline_reply(Socket) of
    454. 

  454. 		{ok, <<"220", _Rest/binary>>} ->
    455. 

  455. 			application:start(crypto),
    456. 

  456. 			application:start(public_key),
    457. 

  457. 			application:start(ssl),
    458. 

  458. 			case socket:to_ssl_client(Socket, [], 5000) of
    459. 

  459. 				{ok, NewSocket} ->
    460. 

  460. 					%NewSocket;
    461. 

  461. 					{ok, Extensions} = try_EHLO(NewSocket, Options),
    462. 

  462. 					{NewSocket, Extensions};
    463. 

  463. 				_Else ->
    464. 

  464. 					%io:format("~p~n", [Else]),
    465. 

  465. 					false
    466. 

  466. 			end;
    467. 

  467. 		{ok, <<"4", _Rest/binary>> = Msg} ->
    468. 

  468. 			quit(Socket),
    469. 

  469. 			throw({temporary_failure, Msg});
    470. 

  470. 		{ok, Msg} ->
    471. 

  471. 			quit(Socket),
    472. 

  472. 			throw({permanent_failure, Msg})
    473. 

  473. 	end.
    474. 

    475. 

  475. %% try connecting to a host
    476. 

  476. connect(Host, Options) when is_binary(Host) ->
    477. 

  477. 	connect(binary_to_list(Host), Options);
    478. 

  478. connect(Host, Options) ->
    479. 

  479. 	SockOpts = [binary, {packet, line}, {keepalive, true}, {active, false}],
    480. 

  480. 	Proto = case proplists:get_value(ssl, Options) of
    481. 

  481. 		true ->
    482. 

  482. 			application:start(crypto),
    483. 

  483. 			application:start(public_key),
    484. 

  484. 			application:start(ssl),
    485. 

  485. 			ssl;
    486. 

  486. 		_ ->
    487. 

  487. 			tcp
    488. 

  488. 	end,
    489. 

  489. 	Port = case proplists:get_value(port, Options) of
    490. 

  490. 		undefined when Proto =:= ssl ->
    491. 

  491. 			465;
    492. 

  492. 		OPort when is_integer(OPort) ->
    493. 

  493. 			OPort;
    494. 

  494. 		_ ->
    495. 

  495. 			25
    496. 

  496. 	end,
    497. 

  497. 	case socket:connect(Proto, Host, Port, SockOpts, 5000) of
    498. 

  498. 		{ok, Socket} ->
    499. 

  499. 			case read_possible_multiline_reply(Socket) of
    500. 

  500. 				{ok, <<"220", Banner/binary>>} ->
    501. 

  501. 					{ok, Socket, Host, Banner};
    502. 

  502. 				{ok, <<"4", _Rest/binary>> = Msg} ->
    503. 

  503. 					quit(Socket),
    504. 

  504. 					throw({temporary_failure, Msg});
    505. 

  505. 				{ok, Msg} ->
    506. 

  506. 					quit(Socket),
    507. 

  507. 					throw({permanent_failure, Msg})
    508. 

  508. 			end;
    509. 

  509. 		{error, Reason} ->
    510. 

  510. 			throw({network_failure, {error, Reason}})
    511. 

  511. 	end.
    512. 

    513. 

  513. %% read a multiline reply (eg. EHLO reply)
    514. 

  514. -spec read_possible_multiline_reply(Socket :: socket:socket()) -> {ok, binary()}.
    515. 

  515. read_possible_multiline_reply(Socket) ->
    516. 

  516. 	case socket:recv(Socket, 0, ?TIMEOUT) of
    517. 

  517. 		{ok, Packet} ->
    518. 

  518. 			case binstr:substr(Packet, 4, 1) of
    519. 

  519. 				<<"-">> ->
    520. 

  520. 					Code = binstr:substr(Packet, 1, 3),
    521. 

  521. 					read_multiline_reply(Socket, Code, [Packet]);
    522. 

  522. 				<<" ">> ->
    523. 

  523. 					{ok, Packet}
    524. 

  524. 			end;
    525. 

  525. 		Error ->
    526. 

  526. 			throw({network_failure, Error})
    527. 

  527. 	end.
    528. 

    529. 

  529. -spec read_multiline_reply(Socket :: socket:socket(), Code :: binary(), Acc :: [binary()]) -> {ok, binary()}.
    530. 

  530. read_multiline_reply(Socket, Code, Acc) ->
    531. 

  531. 	case socket:recv(Socket, 0, ?TIMEOUT) of
    532. 

  532. 		{ok, Packet} ->
    533. 

  533. 			case {binstr:substr(Packet, 1, 3), binstr:substr(Packet, 4, 1)} of
    534. 

  534. 				{Code, <<" ">>} ->
    535. 

  535. 					{ok, list_to_binary(lists:reverse([Packet | Acc]))};
    536. 

  536. 				{Code, <<"-">>} ->
    537. 

  537. 					read_multiline_reply(Socket, Code, [Packet | Acc]);
    538. 

  538. 				_ ->
    539. 

  539. 					quit(Socket),
    540. 

  540. 					throw({unexpected_response, lists:reverse([Packet | Acc])})
    541. 

  541. 			end;
    542. 

  542. 		Error ->
    543. 

  543. 			throw({network_failure, Error})
    544. 

  544. 	end.
    545. 

    546. 

  546. quit(Socket) ->
    547. 

  547. 	socket:send(Socket, "QUIT\r\n"),
    548. 

  548. 	socket:close(Socket),
    549. 

  549. 	ok.
    550. 

    551. 

  551. % TODO - more checking
    552. 

  552. check_options(Options) ->
    553. 

  553. 	case proplists:get_value(relay, Options) of
    554. 

  554. 		undefined ->
    555. 

  555. 			{error, no_relay};
    556. 

  556. 		_ ->
    557. 

  557. 			case proplists:get_value(auth, Options) of
    558. 

  558. 				Atom when Atom =:= always ->
    559. 

  559. 					case proplists:is_defined(username, Options) and
    560. 

  560. 						proplists:is_defined(password, Options) of
    561. 

  561. 						false ->
    562. 

  562. 							{error, no_credentials};
    563. 

  563. 						true ->
    564. 

  564. 							ok
    565. 

  565. 					end;
    566. 

  566. 				_ ->
    567. 

  567. 					ok
    568. 

  568. 			end
    569. 

  569. 	end.
    570. 

    571. 

  571. -spec parse_extensions(Reply :: binary()) -> [{binary(), binary()}].
    572. 

  572. parse_extensions(Reply) ->
    573. 

  573. 	[_ | Reply2] = re:split(Reply, "\r\n", [{return, binary}, trim]),
    574. 

  574. 	[
    575. 

  575. 		begin
    576. 

  576. 				Body = binstr:substr(Entry, 5),
    577. 

  577. 				case re:split(Body, " ",  [{return, binary}, trim, {parts, 2}]) of
    578. 

  578. 					[Verb, Parameters] ->
    579. 

  579. 						{binstr:to_upper(Verb), Parameters};
    580. 

  580. 					[Body] ->
    581. 

  581. 						case binstr:strchr(Body, $=) of
    582. 

  582. 							0 ->
    583. 

  583. 								{binstr:to_upper(Body), true};
    584. 

  584. 							_ ->
    585. 

  585. 								%io:format("discarding option ~p~n", [Body]),
    586. 

  586. 								[]
    587. 

  587. 						end
    588. 

  588. 				end
    589. 

  589. 		end  || Entry <- Reply2].
    590. 

    591. 

  591. -ifdef(TEST).
    592. 

    593. 

  593. session_start_test_() ->
    594. 

  594. 	{foreach,
    595. 

  595. 		local,
    596. 

  596. 		fun() ->
    597. 

  597. 				{ok, ListenSock} = socket:listen(tcp, 9876),
    598. 

  598. 				{ListenSock}
    599. 

  599. 		end,
    600. 

  600. 		fun({ListenSock}) ->
    601. 

  601. 				socket:close(ListenSock)
    602. 

  602. 		end,
    603. 

  603. 		[fun({ListenSock}) ->
    604. 

  604. 					{"simple session initiation",
    605. 

  605. 						fun() ->
    606. 

  606. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    607. 

  607. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    608. 

  608. 								{ok, X} = socket:accept(ListenSock, 1000),
    609. 

  609. 								socket:send(X, "220 Some banner\r\n"),
    610. 

  610. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    611. 

  611. 								ok
    612. 

  612. 						end
    613. 

  613. 					}
    614. 

  614. 			end,
    615. 

  615. 			fun({ListenSock}) ->
    616. 

  616. 					{"retry on crashed EHLO twice if requested",
    617. 

  617. 						fun() ->
    618. 

  618. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}, {retries, 2}],
    619. 

  619. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    620. 

  620. 								{ok, X} = socket:accept(ListenSock, 1000),
    621. 

  621. 								socket:send(X, "220 Some banner\r\n"),
    622. 

  622. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    623. 

  623. 								socket:close(X),
    624. 

  624. 								{ok, Y} = socket:accept(ListenSock, 1000),
    625. 

  625. 								socket:send(Y, "220 Some banner\r\n"),
    626. 

  626. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Y, 0, 1000)),
    627. 

  627. 								socket:close(Y),
    628. 

  628. 								{ok, Z} = socket:accept(ListenSock, 1000),
    629. 

  629. 								socket:send(Z, "220 Some banner\r\n"),
    630. 

  630. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Z, 0, 1000)),
    631. 

  631. 								ok
    632. 

  632. 						end
    633. 

  633. 					}
    634. 

  634. 			end,
    635. 

  635. 			fun({ListenSock}) ->
    636. 

  636. 					{"retry on crashed EHLO",
    637. 

  637. 						fun() ->
    638. 

  638. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    639. 

  639. 								{ok, Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    640. 

  640. 								unlink(Pid),
    641. 

  641. 								Monitor = erlang:monitor(process, Pid),
    642. 

  642. 								{ok, X} = socket:accept(ListenSock, 1000),
    643. 

  643. 								socket:send(X, "220 Some banner\r\n"),
    644. 

  644. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    645. 

  645. 								socket:close(X),
    646. 

  646. 								{ok, Y} = socket:accept(ListenSock, 1000),
    647. 

  647. 								socket:send(Y, "220 Some banner\r\n"),
    648. 

  648. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Y, 0, 1000)),
    649. 

  649. 								socket:close(Y),
    650. 

  650. 								?assertEqual({error, timeout}, socket:accept(ListenSock, 1000)),
    651. 

  651. 								receive {'DOWN', Monitor, _, _, Error} -> ?assertMatch({error, retries_exceeded, _}, Error) end,
    652. 

  652. 								ok
    653. 

  653. 						end
    654. 

  654. 					}
    655. 

  655. 			end,
    656. 

  656. 			fun({ListenSock}) ->
    657. 

  657. 					{"abort on 554 greeting",
    658. 

  658. 						fun() ->
    659. 

  659. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    660. 

  660. 								{ok, Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    661. 

  661. 								unlink(Pid),
    662. 

  662. 								Monitor = erlang:monitor(process, Pid),
    663. 

  663. 								{ok, X} = socket:accept(ListenSock, 1000),
    664. 

  664. 								socket:send(X, "554 get lost, kid\r\n"),
    665. 

  665. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    666. 

  666. 								receive {'DOWN', Monitor, _, _, Error} -> ?assertMatch({error, no_more_hosts, _}, Error) end,
    667. 

  667. 								ok
    668. 

  668. 						end
    669. 

  669. 					}
    670. 

  670. 			end,
    671. 

  671. 			fun({ListenSock}) ->
    672. 

  672. 					{"retry on 421 greeting",
    673. 

  673. 						fun() ->
    674. 

  674. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    675. 

  675. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    676. 

  676. 								{ok, X} = socket:accept(ListenSock, 1000),
    677. 

  677. 								socket:send(X, "421 can't you see I'm busy?\r\n"),
    678. 

  678. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    679. 

  679. 								{ok, Y} = socket:accept(ListenSock, 1000),
    680. 

  680. 								socket:send(Y, "220 Some banner\r\n"),
    681. 

  681. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Y, 0, 1000)),
    682. 

  682. 								ok
    683. 

  683. 						end
    684. 

  684. 					}
    685. 

  685. 			end,
    686. 

  686. 			fun({ListenSock}) ->
    687. 

  687. 					{"retry on messed up EHLO response",
    688. 

  688. 						fun() ->
    689. 

  689. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    690. 

  690. 								{ok, Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    691. 

  691. 								unlink(Pid),
    692. 

  692. 								Monitor = erlang:monitor(process, Pid),
    693. 

  693. 								{ok, X} = socket:accept(ListenSock, 1000),
    694. 

  694. 								socket:send(X, "220 Some banner\r\n"),
    695. 

  695. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    696. 

  696. 								socket:send(X, "250-server.example.com EHLO\r\n250-AUTH LOGIN PLAIN\r\n421 too busy\r\n"),
    697. 

  697. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    698. 

  698. 								
    699. 

  699. 								{ok, Y} = socket:accept(ListenSock, 1000),
    700. 

  700. 								socket:send(Y, "220 Some banner\r\n"),
    701. 

  701. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Y, 0, 1000)),
    702. 

  702. 								socket:send(Y, "250-server.example.com EHLO\r\n250-AUTH LOGIN PLAIN\r\n421 too busy\r\n"),
    703. 

  703. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(Y, 0, 1000)),
    704. 

  704. 								receive {'DOWN', Monitor, _, _, Error} -> ?assertMatch({error, retries_exceeded, _}, Error) end,
    705. 

  705. 								ok
    706. 

  706. 						end
    707. 

  707. 					}
    708. 

  708. 			end,
    709. 

  709. 			fun({ListenSock}) ->
    710. 

  710. 					{"retry with HELO when EHLO not accepted",
    711. 

  711. 						fun() ->
    712. 

  712. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    713. 

  713. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    714. 

  714. 								{ok, X} = socket:accept(ListenSock, 1000),
    715. 

  715. 								socket:send(X, "220 \r\n"),
    716. 

  716. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    717. 

  717. 								socket:send(X, "500 5.3.3 Unrecognized command\r\n"),
    718. 

  718. 								?assertMatch({ok, "HELO testing\r\n"}, socket:recv(X, 0, 1000)),
    719. 

  719. 								socket:send(X, "250 Some banner\r\n"),
    720. 

  720. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    721. 

  721. 								socket:send(X, "250 ok\r\n"),
    722. 

  722. 								?assertMatch({ok, "RCPT TO: <foo@bar.com>\r\n"}, socket:recv(X, 0, 1000)),
    723. 

  723. 								socket:send(X, "250 ok\r\n"),
    724. 

  724. 								?assertMatch({ok, "DATA\r\n"}, socket:recv(X, 0, 1000)),
    725. 

  725. 								socket:send(X, "354 ok\r\n"),
    726. 

  726. 								?assertMatch({ok, "hello world\r\n"}, socket:recv(X, 0, 1000)),
    727. 

  727. 								?assertMatch({ok, ".\r\n"}, socket:recv(X, 0, 1000)),
    728. 

  728. 								socket:send(X, "250 ok\r\n"),
    729. 

  729. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    730. 

  730. 								ok
    731. 

  731. 						end
    732. 

  732. 					}
    733. 

  733. 			end,
    734. 

  734. 			fun({ListenSock}) ->
    735. 

  735. 					{"a valid complete transaction without TLS advertised should succeed",
    736. 

  736. 						fun() ->
    737. 

  737. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    738. 

  738. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    739. 

  739. 								{ok, X} = socket:accept(ListenSock, 1000),
    740. 

  740. 								socket:send(X, "220 Some banner\r\n"),
    741. 

  741. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    742. 

  742. 								socket:send(X, "250 hostname\r\n"),
    743. 

  743. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    744. 

  744. 								socket:send(X, "250 ok\r\n"),
    745. 

  745. 								?assertMatch({ok, "RCPT TO: <foo@bar.com>\r\n"}, socket:recv(X, 0, 1000)),
    746. 

  746. 								socket:send(X, "250 ok\r\n"),
    747. 

  747. 								?assertMatch({ok, "DATA\r\n"}, socket:recv(X, 0, 1000)),
    748. 

  748. 								socket:send(X, "354 ok\r\n"),
    749. 

  749. 								?assertMatch({ok, "hello world\r\n"}, socket:recv(X, 0, 1000)),
    750. 

  750. 								?assertMatch({ok, ".\r\n"}, socket:recv(X, 0, 1000)),
    751. 

  751. 								socket:send(X, "250 ok\r\n"),
    752. 

  752. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    753. 

  753. 								ok
    754. 

  754. 						end
    755. 

  755. 					}
    756. 

  756. 			end,
    757. 

  757. 			fun({ListenSock}) ->
    758. 

  758. 					{"a valid complete transaction with binary arguments shoyld succeed",
    759. 

  759. 						fun() ->
    760. 

  760. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}],
    761. 

  761. 								{ok, _Pid} = send({<<"test@foo.com">>, [<<"foo@bar.com">>], <<"hello world">>}, Options),
    762. 

  762. 								{ok, X} = socket:accept(ListenSock, 1000),
    763. 

  763. 								socket:send(X, "220 Some banner\r\n"),
    764. 

  764. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    765. 

  765. 								socket:send(X, "250 hostname\r\n"),
    766. 

  766. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    767. 

  767. 								socket:send(X, "250 ok\r\n"),
    768. 

  768. 								?assertMatch({ok, "RCPT TO: <foo@bar.com>\r\n"}, socket:recv(X, 0, 1000)),
    769. 

  769. 								socket:send(X, "250 ok\r\n"),
    770. 

  770. 								?assertMatch({ok, "DATA\r\n"}, socket:recv(X, 0, 1000)),
    771. 

  771. 								socket:send(X, "354 ok\r\n"),
    772. 

  772. 								?assertMatch({ok, "hello world\r\n"}, socket:recv(X, 0, 1000)),
    773. 

  773. 								?assertMatch({ok, ".\r\n"}, socket:recv(X, 0, 1000)),
    774. 

  774. 								socket:send(X, "250 ok\r\n"),
    775. 

  775. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    776. 

  776. 								ok
    777. 

  777. 						end
    778. 

  778. 					}
    779. 

  779. 			end,
    780. 

  780. 			fun({ListenSock}) ->
    781. 

  781. 					{"a valid complete transaction with TLS advertised should succeed",
    782. 

  782. 						fun() ->
    783. 

  783. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, <<"testing">>}],
    784. 

  784. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    785. 

  785. 								{ok, X} = socket:accept(ListenSock, 1000),
    786. 

  786. 								socket:send(X, "220 Some banner\r\n"),
    787. 

  787. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    788. 

  788. 								socket:send(X, "250-hostname\r\n250 STARTTLS\r\n"),
    789. 

  789. 								?assertMatch({ok, "STARTTLS\r\n"}, socket:recv(X, 0, 1000)),
    790. 

  790. 								application:start(crypto),
    791. 

  791. 								application:start(public_key),
    792. 

  792. 								application:start(ssl),
    793. 

  793. 								socket:send(X, "220 ok\r\n"),
    794. 

  794. 								{ok, Y} = socket:to_ssl_server(X, [{certfile, "../testdata/server.crt"}, {keyfile, "../testdata/server.key"}], 5000),
    795. 

  795. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Y, 0, 1000)),
    796. 

  796. 								socket:send(Y, "250-hostname\r\n250 STARTTLS\r\n"),
    797. 

  797. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(Y, 0, 1000)),
    798. 

  798. 								socket:send(Y, "250 ok\r\n"),
    799. 

  799. 								?assertMatch({ok, "RCPT TO: <foo@bar.com>\r\n"}, socket:recv(Y, 0, 1000)),
    800. 

  800. 								socket:send(Y, "250 ok\r\n"),
    801. 

  801. 								?assertMatch({ok, "DATA\r\n"}, socket:recv(Y, 0, 1000)),
    802. 

  802. 								socket:send(Y, "354 ok\r\n"),
    803. 

  803. 								?assertMatch({ok, "hello world\r\n"}, socket:recv(Y, 0, 1000)),
    804. 

  804. 								?assertMatch({ok, ".\r\n"}, socket:recv(Y, 0, 1000)),
    805. 

  805. 								socket:send(Y, "250 ok\r\n"),
    806. 

  806. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(Y, 0, 1000)),
    807. 

  807. 								ok
    808. 

  808. 						end
    809. 

  809. 					}
    810. 

  810. 			end,
    811. 

  811. 			fun({ListenSock}) ->
    812. 

  812. 					{"a valid complete transaction with TLS advertised and binary arguments should succeed",
    813. 

  813. 						fun() ->
    814. 

  814. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, <<"testing">>}],
    815. 

  815. 								{ok, _Pid} = send({<<"test@foo.com">>, [<<"foo@bar.com">>], <<"hello world">>}, Options),
    816. 

  816. 								{ok, X} = socket:accept(ListenSock, 1000),
    817. 

  817. 								socket:send(X, "220 Some banner\r\n"),
    818. 

  818. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    819. 

  819. 								socket:send(X, "250-hostname\r\n250 STARTTLS\r\n"),
    820. 

  820. 								?assertMatch({ok, "STARTTLS\r\n"}, socket:recv(X, 0, 1000)),
    821. 

  821. 								application:start(crypto),
    822. 

  822. 								application:start(public_key),
    823. 

  823. 								application:start(ssl),
    824. 

  824. 								socket:send(X, "220 ok\r\n"),
    825. 

  825. 								{ok, Y} = socket:to_ssl_server(X, [{certfile, "../testdata/server.crt"}, {keyfile, "../testdata/server.key"}], 5000),
    826. 

  826. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(Y, 0, 1000)),
    827. 

  827. 								socket:send(Y, "250-hostname\r\n250 STARTTLS\r\n"),
    828. 

  828. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(Y, 0, 1000)),
    829. 

  829. 								socket:send(Y, "250 ok\r\n"),
    830. 

  830. 								?assertMatch({ok, "RCPT TO: <foo@bar.com>\r\n"}, socket:recv(Y, 0, 1000)),
    831. 

  831. 								socket:send(Y, "250 ok\r\n"),
    832. 

  832. 								?assertMatch({ok, "DATA\r\n"}, socket:recv(Y, 0, 1000)),
    833. 

  833. 								socket:send(Y, "354 ok\r\n"),
    834. 

  834. 								?assertMatch({ok, "hello world\r\n"}, socket:recv(Y, 0, 1000)),
    835. 

  835. 								?assertMatch({ok, ".\r\n"}, socket:recv(Y, 0, 1000)),
    836. 

  836. 								socket:send(Y, "250 ok\r\n"),
    837. 

  837. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(Y, 0, 1000)),
    838. 

  838. 								ok
    839. 

  839. 						end
    840. 

  840. 					}
    841. 

  841. 			end,
    842. 

    843. 

  843. 			fun({ListenSock}) ->
    844. 

  844. 					{"AUTH PLAIN should work",
    845. 

  845. 						fun() ->
    846. 

  846. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}, {username, "user"}, {password, "pass"}],
    847. 

  847. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    848. 

  848. 								{ok, X} = socket:accept(ListenSock, 1000),
    849. 

  849. 								socket:send(X, "220 Some banner\r\n"),
    850. 

  850. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    851. 

  851. 								socket:send(X, "250-hostname\r\n250 AUTH PLAIN\r\n"),
    852. 

  852. 								AuthString = binary_to_list(base64:encode("\0user\0pass")),
    853. 

  853. 								AuthPacket = "AUTH PLAIN "++AuthString++"\r\n",
    854. 

  854. 								?assertEqual({ok, AuthPacket}, socket:recv(X, 0, 1000)),
    855. 

  855. 								socket:send(X, "235 ok\r\n"),
    856. 

  856. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    857. 

  857. 								ok
    858. 

  858. 						end
    859. 

  859. 					}
    860. 

  860. 			end,
    861. 

  861. 			fun({ListenSock}) ->
    862. 

  862. 					{"AUTH LOGIN should work",
    863. 

  863. 						fun() ->
    864. 

  864. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}, {username, "user"}, {password, "pass"}],
    865. 

  865. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    866. 

  866. 								{ok, X} = socket:accept(ListenSock, 1000),
    867. 

  867. 								socket:send(X, "220 Some banner\r\n"),
    868. 

  868. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    869. 

  869. 								socket:send(X, "250-hostname\r\n250 AUTH LOGIN\r\n"),
    870. 

  870. 								?assertEqual({ok, "AUTH LOGIN\r\n"}, socket:recv(X, 0, 1000)),
    871. 

  871. 								socket:send(X, "334 VXNlcm5hbWU6\r\n"),
    872. 

  872. 								UserString = binary_to_list(base64:encode("user")),
    873. 

  873. 								?assertEqual({ok, UserString++"\r\n"}, socket:recv(X, 0, 1000)),
    874. 

  874. 								socket:send(X, "334 UGFzc3dvcmQ6\r\n"),
    875. 

  875. 								PassString = binary_to_list(base64:encode("pass")),
    876. 

  876. 								?assertEqual({ok, PassString++"\r\n"}, socket:recv(X, 0, 1000)),
    877. 

  877. 								socket:send(X, "235 ok\r\n"),
    878. 

  878. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    879. 

  879. 								ok
    880. 

  880. 						end
    881. 

  881. 					}
    882. 

  882. 			end,
    883. 

  883. 			fun({ListenSock}) ->
    884. 

  884. 					{"AUTH CRAM-MD5 should work",
    885. 

  885. 						fun() ->
    886. 

  886. 								Options = [{relay, "localhost"}, {port, 9876}, {hostname, "testing"}, {username, "user"}, {password, "pass"}],
    887. 

  887. 								{ok, _Pid} = send({"test@foo.com", ["foo@bar.com"], "hello world"}, Options),
    888. 

  888. 								{ok, X} = socket:accept(ListenSock, 1000),
    889. 

  889. 								socket:send(X, "220 Some banner\r\n"),
    890. 

  890. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    891. 

  891. 								socket:send(X, "250-hostname\r\n250 AUTH CRAM-MD5\r\n"),
    892. 

  892. 								?assertEqual({ok, "AUTH CRAM-MD5\r\n"}, socket:recv(X, 0, 1000)),
    893. 

  893. 								Seed = smtp_util:get_cram_string(smtp_util:guess_FQDN()),
    894. 

  894. 								DecodedSeed = base64:decode_to_string(Seed),
    895. 

  895. 								Digest = smtp_util:compute_cram_digest("pass", DecodedSeed),
    896. 

  896. 								String = binary_to_list(base64:encode(list_to_binary(["user ", Digest]))),
    897. 

  897. 								socket:send(X, "334 "++Seed++"\r\n"),
    898. 

  898. 								{ok, Packet} = socket:recv(X, 0, 1000),
    899. 

  899. 								CramDigest = smtp_util:trim_crlf(Packet),
    900. 

  900. 								?assertEqual(String, CramDigest),
    901. 

  901. 								socket:send(X, "235 ok\r\n"),
    902. 

  902. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    903. 

  903. 								ok
    904. 

  904. 						end
    905. 

  905. 					}
    906. 

  906. 			end,
    907. 

  907. 			fun({ListenSock}) ->
    908. 

  908. 					{"AUTH CRAM-MD5 should work",
    909. 

  909. 						fun() ->
    910. 

  910. 								Options = [{relay, <<"localhost">>}, {port, 9876}, {hostname, <<"testing">>}, {username, <<"user">>}, {password, <<"pass">>}],
    911. 

  911. 								{ok, _Pid} = send({<<"test@foo.com">>, [<<"foo@bar.com">>, <<"baz@bar.com">>], <<"hello world">>}, Options),
    912. 

  912. 								{ok, X} = socket:accept(ListenSock, 1000),
    913. 

  913. 								socket:send(X, "220 Some banner\r\n"),
    914. 

  914. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    915. 

  915. 								socket:send(X, "250-hostname\r\n250 AUTH CRAM-MD5\r\n"),
    916. 

  916. 								?assertEqual({ok, "AUTH CRAM-MD5\r\n"}, socket:recv(X, 0, 1000)),
    917. 

  917. 								Seed = smtp_util:get_cram_string(smtp_util:guess_FQDN()),
    918. 

  918. 								DecodedSeed = base64:decode_to_string(Seed),
    919. 

  919. 								Digest = smtp_util:compute_cram_digest("pass", DecodedSeed),
    920. 

  920. 								String = binary_to_list(base64:encode(list_to_binary(["user ", Digest]))),
    921. 

  921. 								socket:send(X, "334 "++Seed++"\r\n"),
    922. 

  922. 								{ok, Packet} = socket:recv(X, 0, 1000),
    923. 

  923. 								CramDigest = smtp_util:trim_crlf(Packet),
    924. 

  924. 								?assertEqual(String, CramDigest),
    925. 

  925. 								socket:send(X, "235 ok\r\n"),
    926. 

  926. 								?assertMatch({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    927. 

  927. 								ok
    928. 

  928. 						end
    929. 

  929. 					}
    930. 

  930. 			end,
    931. 

  931. 			fun({ListenSock}) ->
    932. 

  932. 					{"should bail when AUTH is required but not provided",
    933. 

  933. 						fun() ->
    934. 

  934. 								Options = [{relay, <<"localhost">>}, {port, 9876}, {hostname, <<"testing">>}, {auth, always}, {username, <<"user">>}, {retries, 0}, {password, <<"pass">>}],
    935. 

  935. 								{ok, Pid} = send({<<"test@foo.com">>, [<<"foo@bar.com">>, <<"baz@bar.com">>], <<"hello world">>}, Options),
    936. 

  936. 								unlink(Pid),
    937. 

  937. 								Monitor = erlang:monitor(process, Pid),
    938. 

  938. 								{ok, X} = socket:accept(ListenSock, 1000),
    939. 

  939. 								socket:send(X, "220 Some banner\r\n"),
    940. 

  940. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    941. 

  941. 								socket:send(X, "250-hostname\r\n250 8BITMIME\r\n"),
    942. 

  942. 								?assertEqual({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    943. 

  943. 								receive {'DOWN', Monitor, _, _, Error} -> ?assertMatch({error, retries_exceeded, {missing_requirement, _, auth}}, Error) end,
    944. 

  944. 								ok
    945. 

  945. 						end
    946. 

  946. 					}
    947. 

  947. 			end,
    948. 

  948. 			fun({ListenSock}) ->
    949. 

  949. 					{"should bail when AUTH is required but of an unsupported type",
    950. 

  950. 						fun() ->
    951. 

  951. 								Options = [{relay, <<"localhost">>}, {port, 9876}, {hostname, <<"testing">>}, {auth, always}, {username, <<"user">>}, {retries, 0}, {password, <<"pass">>}],
    952. 

  952. 								{ok, Pid} = send({<<"test@foo.com">>, [<<"foo@bar.com">>, <<"baz@bar.com">>], <<"hello world">>}, Options),
    953. 

  953. 								unlink(Pid),
    954. 

  954. 								Monitor = erlang:monitor(process, Pid),
    955. 

  955. 								{ok, X} = socket:accept(ListenSock, 1000),
    956. 

  956. 								socket:send(X, "220 Some banner\r\n"),
    957. 

  957. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    958. 

  958. 								socket:send(X, "250-hostname\r\n250-AUTH GSSAPI\r\n250 8BITMIME\r\n"),
    959. 

  959. 								?assertEqual({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    960. 

  960. 								receive {'DOWN', Monitor, _, _, Error} -> ?assertMatch({error, no_more_hosts, {permanent_failure, _, auth_failed}}, Error) end,
    961. 

  961. 								ok
    962. 

  962. 						end
    963. 

  963. 					}
    964. 

  964. 			end,
    965. 

  965. 			fun({_ListenSock}) ->
    966. 

  966. 					{"Connecting to a SSL socket directly should work",
    967. 

  967. 						fun() ->
    968. 

  968. 								application:start(crypto),
    969. 

  969. 								application:start(public_key),
    970. 

  970. 								application:start(ssl),
    971. 

  971. 								{ok, ListenSock} = socket:listen(ssl, 9877, [{certfile, "../testdata/server.crt"}, {keyfile, "../testdata/server.key"}]),
    972. 

  972. 								Options = [{relay, <<"localhost">>}, {port, 9877}, {hostname, <<"testing">>}, {ssl, true}],
    973. 

  973. 								{ok, _Pid} = send({<<"test@foo.com">>, [<<"<foo@bar.com>">>, <<"baz@bar.com">>], <<"hello world">>}, Options),
    974. 

  974. 								{ok, X} = socket:accept(ListenSock, 1000),
    975. 

  975. 								socket:send(X, "220 Some banner\r\n"),
    976. 

  976. 								?assertMatch({ok, "EHLO testing\r\n"}, socket:recv(X, 0, 1000)),
    977. 

  977. 								socket:send(X, "250-hostname\r\n250 AUTH CRAM-MD5\r\n"),
    978. 

  978. 								?assertEqual({ok, "MAIL FROM: <test@foo.com>\r\n"}, socket:recv(X, 0, 1000)),
    979. 

  979. 								socket:send(X, "250 ok\r\n"),
    980. 

  980. 								?assertMatch({ok, "RCPT TO: <foo@bar.com>\r\n"}, socket:recv(X, 0, 1000)),
    981. 

  981. 								socket:send(X, "250 ok\r\n"),
    982. 

  982. 								?assertMatch({ok, "RCPT TO: <baz@bar.com>\r\n"}, socket:recv(X, 0, 1000)),
    983. 

  983. 								socket:send(X, "250 ok\r\n"),
    984. 

  984. 								?assertMatch({ok, "DATA\r\n"}, socket:recv(X, 0, 1000)),
    985. 

  985. 								socket:send(X, "354 ok\r\n"),
    986. 

  986. 								?assertMatch({ok, "hello world\r\n"}, socket:recv(X, 0, 1000)),
    987. 

  987. 								?assertMatch({ok, ".\r\n"}, socket:recv(X, 0, 1000)),
    988. 

  988. 								socket:send(X, "250 ok\r\n"),
    989. 

  989. 								?assertMatch({ok, "QUIT\r\n"}, socket:recv(X, 0, 1000)),
    990. 

  990. 								socket:close(ListenSock),
    991. 

  991. 								ok
    992. 

  992. 						end
    993. 

  993. 					}
    994. 

  994. 			end
    995. 

    996. 

  996. 		]
    997. 

  997. 	}.
    998. 

    999. 

  999. extension_parse_test_() ->
    1000. 

  1000. 	[
    1001. 

  1001. 		{"parse extensions",
    1002. 

  1002. 			fun() ->
    1003. 

  1003. 					Res = parse_extensions(<<"250-smtp.example.com\r\n250-PIPELINING\r\n250-SIZE 20971520\r\n250-VRFY\r\n250-ETRN\r\n250-STARTTLS\r\n250-AUTH CRAM-MD5 PLAIN DIGEST-MD5 LOGIN\r\n250-AUTH=CRAM-MD5 PLAIN DIGEST-MD5 LOGIN\r\n250-ENHANCEDSTATUSCODES\r\n250-8BITMIME\r\n250 DSN">>),
    1004. 

  1004. 					?assertEqual(true, proplists:get_value(<<"PIPELINING">>, Res)),
    1005. 

  1005. 					?assertEqual(<<"20971520">>, proplists:get_value(<<"SIZE">>, Res)),
    1006. 

  1006. 					?assertEqual(true, proplists:get_value(<<"VRFY">>, Res)),
    1007. 

  1007. 					?assertEqual(true, proplists:get_value(<<"ETRN">>, Res)),
    1008. 

  1008. 					?assertEqual(true, proplists:get_value(<<"STARTTLS">>, Res)),
    1009. 

  1009. 					?assertEqual(<<"CRAM-MD5 PLAIN DIGEST-MD5 LOGIN">>, proplists:get_value(<<"AUTH">>, Res)),
    1010. 

  1010. 					?assertEqual(true, proplists:get_value(<<"ENHANCEDSTATUSCODES">>, Res)),
    1011. 

  1011. 					?assertEqual(true, proplists:get_value(<<"8BITMIME">>, Res)),
    1012. 

  1012. 					?assertEqual(true, proplists:get_value(<<"DSN">>, Res)),
    1013. 

  1013. 					?assertEqual(10, length(Res)),
    1014. 

  1014. 					ok
    1015. 

  1015. 			end
    1016. 

  1016. 		}
    1017. 

  1017. 	].
    1018. 

    1019. 

  1019. -endif.
    1020.