/* Copyright (c) 2017-2018 Dovecot authors, see the included COPYING file */

#include "test-lib.h"
#include "array.h"
#include "buffer.h"
#include "str.h"
#include "str-sanitize.h"
#include "istream.h"
#include "ostream.h"
#include "test-common.h"
#include "smtp-command-parser.h"

#include <time.h>

/*
 * Valid command tests
 */

struct smtp_command_parse_valid_test {
	const char *command;

	struct smtp_command_limits limits;

	const char *cmd_name;
	const char *cmd_params;
};

static const struct smtp_command_parse_valid_test
valid_command_parse_tests[] = {
	{
		.command = "RSET\r\n",
		.cmd_name = "RSET",
		.cmd_params = "",
	},
	{
		.command = "RSET    \r\n",
		.cmd_name = "RSET",
		.cmd_params = "",
	},
	{
		.command = "EHLO example.com\r\n",
		.cmd_name = "EHLO",
		.cmd_params = "example.com",
	},
	{
		.command = "EHLO example.com     \r\n",
		.cmd_name = "EHLO",
		.cmd_params = "example.com",
	},
	{
		.command = "MAIL FROM:<sender@example.com> ENVID=frop\r\n",
		.cmd_name = "MAIL",
		.cmd_params = "FROM:<sender@example.com> ENVID=frop",
	},
	{
		.command = "VRFY \"Sherlock Holmes\"\r\n",
		.cmd_name = "VRFY",
		.cmd_params = "\"Sherlock Holmes\"",
	},
	{
		.command = "RCPT TO:<recipient@example.com> NOTIFY=NEVER\r\n",
		.limits = { .max_parameters_size = 39 },
		.cmd_name = "RCPT",
		.cmd_params = "TO:<recipient@example.com> NOTIFY=NEVER",
	},
	{
		.command = "MAIL FROM:<f\xc3\xb6\xc3\xa4@\xc3\xb6\xc3\xa4>\r\n",
		.cmd_name = "MAIL",
		.cmd_params = "FROM:<f\xc3\xb6\xc3\xa4@\xc3\xb6\xc3\xa4>",
	},
};

unsigned int valid_command_parse_test_count =
	N_ELEMENTS(valid_command_parse_tests);

static void
test_smtp_command_parse_valid_check(
	const struct smtp_command_parse_valid_test *test,
	const char *cmd_name,  const char *cmd_params)
{
	test_out(t_strdup_printf("command name = `%s'", test->cmd_name),
		 null_strcmp(cmd_name, test->cmd_name) == 0);
	test_out(t_strdup_printf("command params = `%s'",
				 str_sanitize(test->cmd_params, 24)),
		 null_strcmp(cmd_params, test->cmd_params) == 0);
}

static void test_smtp_command_parse_valid(void)
{
	unsigned int i;

	for (i = 0; i < valid_command_parse_test_count; i++) T_BEGIN {
		struct istream *input;
		const struct smtp_command_parse_valid_test *test;
		struct smtp_command_parser *parser;
		const char *command_text, *cmd_name, *cmd_params, *error;
		enum smtp_command_parse_error error_code;
		unsigned int pos, command_text_len;
		int ret;

		test_begin(t_strdup_printf("smtp command valid [%d]", i));

		cmd_name = cmd_params = error = NULL;

		test = &valid_command_parse_tests[i];
		command_text = test->command;
		command_text_len = strlen(command_text);

		/* Fully buffered input */
		input = i_stream_create_from_data(command_text,
						  command_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		while ((ret = smtp_command_parse_next(
			parser, &cmd_name, &cmd_params,
			&error_code, &error)) > 0);

		test_out_reason("parse success [buffer]", ret == -2,
				(ret == -2 ? NULL : error));
		if (ret == 0) {
			/* Verify last command only */
			test_smtp_command_parse_valid_check(
				test, cmd_name, cmd_params);
		}

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		error = NULL;
		error_code = SMTP_COMMAND_PARSE_ERROR_NONE;
		ret = 0;

		/* Trickle stream */
		input = test_istream_create_data(command_text,
						 command_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		for (pos = 0; pos <= command_text_len && ret == 0; pos++) {
			test_istream_set_size(input, pos);
			ret = smtp_command_parse_next(
				parser, &cmd_name, &cmd_params,
				&error_code, &error);
		}
		test_istream_set_size(input, command_text_len);
		if (ret >= 0) {
			while ((ret = smtp_command_parse_next(
				parser, &cmd_name, &cmd_params,
				&error_code, &error)) > 0);
		}

		test_out_reason("parse success [stream]", ret == -2,
				(ret == -2 ? NULL : error));
		if (ret == 0) {
			/* Verify last command only */
			test_smtp_command_parse_valid_check(
				test, cmd_name, cmd_params);
		}

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		test_end();
	} T_END;
}

/*
 * Invalid command tests
 */

struct smtp_command_parse_invalid_test {
	const char *command;

	struct smtp_command_limits limits;

	enum smtp_command_parse_error error_code;
};

static const struct smtp_command_parse_invalid_test
	invalid_command_parse_tests[] = {
	{
		.command = "B52\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "BELL\x08\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "EHLO  example.com\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "NOOP \"\x01\x02\x03\"\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "RSET\rQUIT\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "INSANELYREDICULOUSLYLONGCOMMANDNAME\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "RCPT TO:<recipient@example.com> NOTIFY=NEVER\r\n",
		.limits = { .max_parameters_size = 38 },
		.error_code = SMTP_COMMAND_PARSE_ERROR_LINE_TOO_LONG,
	},
	{
		.command = "MAIL FROM:<f\xc3\xb6\xc3\xa4@\xc3\xb6\xc3>\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "MAIL FROM:f\xc3\xb6\xc3\xa4@\xc3\xb6\xc3\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.command = "MAIL FROM:f\xc3\xb6\xc3\xa4@\xc3\xb6\xc3",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BROKEN_COMMAND,
	},
	{
		.command = "FROP \xF1",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BROKEN_COMMAND,
	},
	{
		.command = "FROP \xF1\x80",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BROKEN_COMMAND,
	},
	{
		.command = "FROP \xF1\x80\x80",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BROKEN_COMMAND,
	},
	{
		.command = "FROP \xF1\x80\x80\x80",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BROKEN_COMMAND,
	},
};

unsigned int invalid_command_parse_test_count =
	N_ELEMENTS(invalid_command_parse_tests);

static void test_smtp_command_parse_invalid(void)
{
	unsigned int i;

	for (i = 0; i < invalid_command_parse_test_count; i++) T_BEGIN {
		const struct smtp_command_parse_invalid_test *test;
		struct istream *input;
		struct smtp_command_parser *parser;
		const char *command_text, *cmd_name, *cmd_params, *error;
		enum smtp_command_parse_error error_code;
		unsigned int pos, command_text_len;
		int ret;

		test_begin(t_strdup_printf("smtp command invalid [%d]", i));

		test = &invalid_command_parse_tests[i];
		command_text = test->command;
		command_text_len = strlen(command_text);

		/* Fully buffered input */
		input = i_stream_create_from_data(command_text,
						  command_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		while ((ret = smtp_command_parse_next(
			parser, &cmd_name, &cmd_params,
			&error_code, &error)) > 0);

		test_out_reason(t_strdup_printf("parse(\"%s\") [buffer]",
						str_sanitize(command_text, 28)),
				ret == -1, error);
		test_out_quiet("error code", error_code == test->error_code);

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		error = NULL;
		error_code = SMTP_COMMAND_PARSE_ERROR_NONE;
		ret = 0;

		/* Trickle stream */
		input = test_istream_create_data(command_text,
						 command_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		for (pos = 0; pos <= command_text_len && ret == 0; pos++) {
			test_istream_set_size(input, pos);
			ret = smtp_command_parse_next(
				parser, &cmd_name, &cmd_params,
				&error_code, &error);
		}
		test_istream_set_size(input, command_text_len);
		if (ret >= 0) {
			while ((ret = smtp_command_parse_next(
				parser, &cmd_name, &cmd_params,
				&error_code, &error)) > 0);
		}

		test_out_reason(t_strdup_printf("parse(\"%s\") [stream]",
						str_sanitize(command_text, 28)),
				ret == -1, error);
		test_out_quiet("error code", error_code == test->error_code);

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		test_end();
	} T_END;
}

/*
 * Valid auth response tests
 */

struct smtp_auth_response_parse_valid_test {
	const char *auth_response;

	struct smtp_command_limits limits;

	const char *line;
};

static const struct smtp_auth_response_parse_valid_test
valid_auth_response_parse_tests[] = {
	{
		.auth_response = "U3R1cGlkIEJhc2U2NCB0ZXN0\r\n",
		.line = "U3R1cGlkIEJhc2U2NCB0ZXN0",
	},
	{
		.auth_response = "U3R1cGlkIEJhc2U2NCB0ZXN0    \r\n",
		.line = "U3R1cGlkIEJhc2U2NCB0ZXN0",
	},
	{
		.auth_response =
			"U3R1cGlkIHZlcnkgdmVyeSB2ZXJ5IHZlcnkgdmVyeS"
			"B2ZXJ5IHZlcnkgdmVyeSBsb25nIEJhc2U2NCB0ZXN0\r\n",
		.limits = { .max_auth_size = 84 },
		.line = "U3R1cGlkIHZlcnkgdmVyeSB2ZXJ5IHZlcnkgdmVyeS"
			"B2ZXJ5IHZlcnkgdmVyeSBsb25nIEJhc2U2NCB0ZXN0",
	},
	{
		.auth_response =
			"dXNlcj10ZXN0dXNlcjEBYXV0aD1CZWFyZXIgZXlKaG"
			"JHY2lPaUpTVXpJMU5pSXNJblI1Y0NJZ09pQWlTbGRV"
			"SWl3aWEybGtJaUE2SUNKdVRIRlVlRnBXWVhKSlgwWn"
			"dSa0Z3Umt3MloyUnhiak4xV1VSS2R6WnNWVjlMYVZo"
			"a2JWazJialpSSW4wLmV5SmxlSEFpT2pFMk16UTJNem"
			"MyTlRFc0ltbGhkQ0k2TVRZek5EWXpOek0xTVN3aWFu"
			"UnBJam9pT1RFM1lUYzFaalF0WTJZME9DMDBOVEEyTF"
			"RnNVpXSXRNRE13WldaaU5tSTVOMlZrSWl3aWFYTnpJ"
			"am9pYUhSMGNEb3ZMekU1TWk0eE5qZ3VNUzR5TVRveE"
			"9EQTRNQzloZFhSb0wzSmxZV3h0Y3k5eVpXeDBaWE4w"
			"SWl3aVlYVmtJam9pWVdOamIzVnVkQ0lzSW5OMVlpST"
			"ZJamhsWVRRME1UWTNMVGN6TTJVdE5EVTBZeTFpT0dJ"
			"MUxXTmpabVl3WkRnek1URTVaQ0lzSW5SNWNDSTZJa0"
			"psWVhKbGNpSXNJbUY2Y0NJNkltUnZkbVZqYjNRaUxD"
			"SnpaWE56YVc5dVgzTjBZWFJsSWpvaU1tTTNPVEUzWl"
			"dJdE16QTFOUzAwTkRZeExXSXdZell0WTJVeFlUbGlN"
			"VEZoTWpReklpd2lZV055SWpvaU1TSXNJbkpsWVd4dF"
			"gyRmpZMlZ6Y3lJNmV5SnliMnhsY3lJNld5SnZabVpz"
			"YVc1bFgyRmpZMlZ6Y3lJc0luVnRZVjloZFhSb2IzSn"
			"BlbUYwYVc5dUlsMTlMQ0p5WlhOdmRYSmpaVjloWTJO"
			"bGMzTWlPbnNpWVdOamIzVnVkQ0k2ZXlKeWIyeGxjeU"
			"k2V3lKdFlXNWhaMlV0WVdOamIzVnVkQ0lzSW0xaGJt"
			"Rm5aUzFoWTJOdmRXNTBMV3hwYm10eklpd2lkbWxsZH"
			"kxd2NtOW1hV3hsSWwxOWZTd2ljMk52Y0dVaU9pSndj"
			"bTltYVd4bElHVnRZV2xzSWl3aVpXMWhhV3hmZG1WeW"
			"FXWnBaV1FpT21aaGJITmxMQ0p1WVcxbElqb2lkR1Z6"
			"ZEhWelpYSXhJRUYxZEc5SFpXNWxjbUYwWldRaUxDSn"
			"djbVZtWlhKeVpXUmZkWE5sY201aGJXVWlPaUowWlhO"
			"MGRYTmxjakVpTENKbmFYWmxibDl1WVcxbElqb2lkR1"
			"Z6ZEhWelpYSXhJaXdpWm1GdGFXeDVYMjVoYldVaU9p"
			"SkJkWFJ2UjJWdVpYSmhkR1ZrSWl3aVpXMWhhV3dpT2"
			"lKMFpYTjBkWE5sY2pGQWJYbGtiMjFoYVc0dWIzZ2lm"
			"US5ta2JGSURpT0FhbENCcVMwODRhVHJURjBIdDk1c1"
			"Z4cGlSbTFqZnhJd0JiN1hMM2gzWUJkdXVrVXlZdDJq"
			"X1pqUFlhMDhDcVVYNWFrLVBOSjdSVWRTUXNmUlgwM1"
			"ZicXA4MHFZZjNGYzJpcDR0YmhHLXFEV0R6NzdhZDhW"
			"cEFNei16YWlSamZCclZ2R3hBT3ZsZnFDVWhaZTJDR3"
			"ZqWjZ1Q3RKTlFaS0dyazZHOXoxX2pqekZkTjBXWjUx"
			"bEZsUS1JdE5LREpoTjNIekJ5SW93M19qQU9kWEI0R0"
			"w4R3JHM1hqU09rSFVRam5GTEQwQUF1QXY4SkxmTXY1"
			"NGc1a2tKaklxRFgxZlgyWVo0Y2JQOWV3TUp6UV84ZW"
			"dLeW5TVV9XSk8xRU9Qa1NVZjlMX19RX3FwY0dNbzFt"
			"TkxuTURKUlU2dmZFY3JrM2k0cVNzMXRPdHdLaHcBAQ"
			"==\r\n",
		.line =
			"dXNlcj10ZXN0dXNlcjEBYXV0aD1CZWFyZXIgZXlKaG"
			"JHY2lPaUpTVXpJMU5pSXNJblI1Y0NJZ09pQWlTbGRV"
			"SWl3aWEybGtJaUE2SUNKdVRIRlVlRnBXWVhKSlgwWn"
			"dSa0Z3Umt3MloyUnhiak4xV1VSS2R6WnNWVjlMYVZo"
			"a2JWazJialpSSW4wLmV5SmxlSEFpT2pFMk16UTJNem"
			"MyTlRFc0ltbGhkQ0k2TVRZek5EWXpOek0xTVN3aWFu"
			"UnBJam9pT1RFM1lUYzFaalF0WTJZME9DMDBOVEEyTF"
			"RnNVpXSXRNRE13WldaaU5tSTVOMlZrSWl3aWFYTnpJ"
			"am9pYUhSMGNEb3ZMekU1TWk0eE5qZ3VNUzR5TVRveE"
			"9EQTRNQzloZFhSb0wzSmxZV3h0Y3k5eVpXeDBaWE4w"
			"SWl3aVlYVmtJam9pWVdOamIzVnVkQ0lzSW5OMVlpST"
			"ZJamhsWVRRME1UWTNMVGN6TTJVdE5EVTBZeTFpT0dJ"
			"MUxXTmpabVl3WkRnek1URTVaQ0lzSW5SNWNDSTZJa0"
			"psWVhKbGNpSXNJbUY2Y0NJNkltUnZkbVZqYjNRaUxD"
			"SnpaWE56YVc5dVgzTjBZWFJsSWpvaU1tTTNPVEUzWl"
			"dJdE16QTFOUzAwTkRZeExXSXdZell0WTJVeFlUbGlN"
			"VEZoTWpReklpd2lZV055SWpvaU1TSXNJbkpsWVd4dF"
			"gyRmpZMlZ6Y3lJNmV5SnliMnhsY3lJNld5SnZabVpz"
			"YVc1bFgyRmpZMlZ6Y3lJc0luVnRZVjloZFhSb2IzSn"
			"BlbUYwYVc5dUlsMTlMQ0p5WlhOdmRYSmpaVjloWTJO"
			"bGMzTWlPbnNpWVdOamIzVnVkQ0k2ZXlKeWIyeGxjeU"
			"k2V3lKdFlXNWhaMlV0WVdOamIzVnVkQ0lzSW0xaGJt"
			"Rm5aUzFoWTJOdmRXNTBMV3hwYm10eklpd2lkbWxsZH"
			"kxd2NtOW1hV3hsSWwxOWZTd2ljMk52Y0dVaU9pSndj"
			"bTltYVd4bElHVnRZV2xzSWl3aVpXMWhhV3hmZG1WeW"
			"FXWnBaV1FpT21aaGJITmxMQ0p1WVcxbElqb2lkR1Z6"
			"ZEhWelpYSXhJRUYxZEc5SFpXNWxjbUYwWldRaUxDSn"
			"djbVZtWlhKeVpXUmZkWE5sY201aGJXVWlPaUowWlhO"
			"MGRYTmxjakVpTENKbmFYWmxibDl1WVcxbElqb2lkR1"
			"Z6ZEhWelpYSXhJaXdpWm1GdGFXeDVYMjVoYldVaU9p"
			"SkJkWFJ2UjJWdVpYSmhkR1ZrSWl3aVpXMWhhV3dpT2"
			"lKMFpYTjBkWE5sY2pGQWJYbGtiMjFoYVc0dWIzZ2lm"
			"US5ta2JGSURpT0FhbENCcVMwODRhVHJURjBIdDk1c1"
			"Z4cGlSbTFqZnhJd0JiN1hMM2gzWUJkdXVrVXlZdDJq"
			"X1pqUFlhMDhDcVVYNWFrLVBOSjdSVWRTUXNmUlgwM1"
			"ZicXA4MHFZZjNGYzJpcDR0YmhHLXFEV0R6NzdhZDhW"
			"cEFNei16YWlSamZCclZ2R3hBT3ZsZnFDVWhaZTJDR3"
			"ZqWjZ1Q3RKTlFaS0dyazZHOXoxX2pqekZkTjBXWjUx"
			"bEZsUS1JdE5LREpoTjNIekJ5SW93M19qQU9kWEI0R0"
			"w4R3JHM1hqU09rSFVRam5GTEQwQUF1QXY4SkxmTXY1"
			"NGc1a2tKaklxRFgxZlgyWVo0Y2JQOWV3TUp6UV84ZW"
			"dLeW5TVV9XSk8xRU9Qa1NVZjlMX19RX3FwY0dNbzFt"
			"TkxuTURKUlU2dmZFY3JrM2k0cVNzMXRPdHdLaHcBAQ",
	},
};

unsigned int valid_auth_response_parse_test_count =
	N_ELEMENTS(valid_auth_response_parse_tests);

static void
test_smtp_auth_response_parse_valid_check(
	const struct smtp_auth_response_parse_valid_test *test,
	const char *line)
{
	test_out(t_strdup_printf("line = `%s'",
			 str_sanitize(test->line, 24)),
		 null_strcmp(line, test->line) == 0);
}

static void test_smtp_auth_response_parse_valid(void)
{
	unsigned int i;

	for (i = 0; i < valid_auth_response_parse_test_count; i++) T_BEGIN {
		struct istream *input;
		const struct smtp_auth_response_parse_valid_test *test;
		struct smtp_command_parser *parser;
		const char *response_text, *line, *error;
		enum smtp_command_parse_error error_code;
		unsigned int pos, response_text_len;
		int ret;

		test_begin(t_strdup_printf("smtp auth_response valid [%d]", i));

		line = error = NULL;

		test = &valid_auth_response_parse_tests[i];
		response_text = test->auth_response;
		response_text_len = strlen(response_text);

		/* Fully buffered input */
		input = i_stream_create_from_data(response_text,
						  response_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		while ((ret = smtp_command_parse_auth_response(
			parser, &line, &error_code, &error)) > 0);

		test_out_reason("parse success [buffer]", ret == -2,
				(ret == -2 ? NULL : error));
		if (ret == 0) {
			/* Verify last reponse only */
			test_smtp_auth_response_parse_valid_check(test, line);
		}

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		error = NULL;
		error_code = SMTP_COMMAND_PARSE_ERROR_NONE;
		ret = 0;

		/* Trickle stream */
		input = test_istream_create_data(response_text,
						 response_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		for (pos = 0; pos <= response_text_len && ret == 0; pos++) {
			test_istream_set_size(input, pos);
			ret = smtp_command_parse_auth_response(
				parser, &line, &error_code, &error);
		}
		test_istream_set_size(input, response_text_len);
		if (ret >= 0) {
			while ((ret = smtp_command_parse_auth_response(
				parser, &line, &error_code, &error)) > 0);
		}

		test_out_reason("parse success [stream]", ret == -2,
				(ret == -2 ? NULL : error));
		if (ret == 0) {
			/* Verify last reponse only */
			test_smtp_auth_response_parse_valid_check(test, line);
		}

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		test_end();
	} T_END;
}

/*
 * Invalid auth response tests
 */

struct smtp_auth_response_parse_invalid_test {
	const char *auth_response;

	struct smtp_command_limits limits;

	enum smtp_command_parse_error error_code;
};

static const struct smtp_auth_response_parse_invalid_test
	invalid_auth_response_parse_tests[] = {
	{
		.auth_response = "\x01\x02\x03\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.auth_response = "U3R1cGlkIEJhc\r2U2NCB0ZXN0\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
	{
		.auth_response =
			"U3R1cGlkIHZlcnkgdmVyeSB2ZXJ5IHZlcnkgdmVyeS"
			"B2ZXJ5IHZlcnkgdmVyeSBsb25nIEJhc2U2NCB0ZXN0\r\n",
		.limits = { .max_auth_size = 83 },
		.error_code = SMTP_COMMAND_PARSE_ERROR_LINE_TOO_LONG,
	},
	{
		.auth_response = "\xc3\xb6\xc3\xa4\xc3\xb6\xc3\xa4\r\n",
		.error_code = SMTP_COMMAND_PARSE_ERROR_BAD_COMMAND,
	},
};

unsigned int invalid_auth_response_parse_test_count =
	N_ELEMENTS(invalid_auth_response_parse_tests);

static void test_smtp_auth_response_parse_invalid(void)
{
	unsigned int i;

	for (i = 0; i < invalid_auth_response_parse_test_count; i++) T_BEGIN {
		const struct smtp_auth_response_parse_invalid_test *test;
		struct istream *input;
		struct smtp_command_parser *parser;
		const char *response_text, *line, *error;
		enum smtp_command_parse_error error_code;
		unsigned int pos, response_text_len;
		int ret;

		test_begin(
			t_strdup_printf("smtp auth response invalid [%d]", i));

		test = &invalid_auth_response_parse_tests[i];
		response_text = test->auth_response;
		response_text_len = strlen(response_text);

		/* Fully buffered input */
		input = i_stream_create_from_data(response_text,
						  strlen(response_text));
		parser = smtp_command_parser_init(input, &test->limits);

		while ((ret = smtp_command_parse_auth_response(
			parser, &line, &error_code, &error)) > 0);

		test_out_reason(t_strdup_printf("parse(\"%s\") [buffer]",
						str_sanitize(response_text,
							     28)),
				ret == -1, error);
		test_out_quiet("error code", error_code == test->error_code);

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		error = NULL;
		error_code = SMTP_COMMAND_PARSE_ERROR_NONE;
		ret = 0;

		/* Trickle stream */
		input = test_istream_create_data(response_text,
						 response_text_len);
		parser = smtp_command_parser_init(input, &test->limits);

		for (pos = 0; pos <= response_text_len && ret == 0; pos++) {
			test_istream_set_size(input, pos);
			ret = smtp_command_parse_auth_response(
				parser, &line, &error_code, &error);
		}
		test_istream_set_size(input, response_text_len);
		if (ret >= 0) {
			while ((ret = smtp_command_parse_auth_response(
				parser, &line, &error_code, &error)) > 0);
		}

		test_out_reason(t_strdup_printf("parse(\"%s\") [stream]",
						str_sanitize(response_text,
							     28)),
				ret == -1, error);
		test_out_quiet("error code", error_code == test->error_code);

		smtp_command_parser_deinit(&parser);
		i_stream_unref(&input);

		test_end();
	} T_END;
}

/*
 * Tests
 */

int main(void)
{
	static void (*test_functions[])(void) = {
		test_smtp_command_parse_valid,
		test_smtp_command_parse_invalid,
		test_smtp_auth_response_parse_valid,
		test_smtp_auth_response_parse_invalid,
		NULL
	};
	return test_run(test_functions);
}