3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch#include "lib.h"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch#include "istream.h"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch#include "str.h"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch#include "message-parser.h"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch#include "imap-bodystructure.h"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch#include "test-common.h"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Boschstruct parse_test {

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch const char *message;

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch const char *body;

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch const char *bodystructure;

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch};

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Boschstruct parse_test parse_tests[] = {

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch {

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .message =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "From: user@domain.org\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Mime-Version: 1.0\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Type: text/plain; charset=us-ascii\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "body\n",

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .bodystructure =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 6 1 NIL NIL NIL NIL",

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .body =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 6 1"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch },{

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .message =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "From: user@domain.org\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Mime-Version: 1.0\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Type: text/plain; charset=utf-8\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Transfer-Encoding: 8bit\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n"

47fee1a942e4797548b1232354f6676b8ff809f4Stephan Bosch "body\n"

47fee1a942e4797548b1232354f6676b8ff809f4Stephan Bosch "\n",

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .bodystructure =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\"text\" \"plain\" (\"charset\" \"utf-8\") NIL NIL \"8bit\" 8 2 NIL NIL NIL NIL",

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .body =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\"text\" \"plain\" (\"charset\" \"utf-8\") NIL NIL \"8bit\" 8 2"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch },{

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .message =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "From: user@domain.org\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Date: Sat, 24 Mar 2007 23:00:00 +0200\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Mime-Version: 1.0\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Type: multipart/mixed; boundary=\"foo\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch " bar\"\n"

f9d2a1f21ad65262bc630f0834d7eead06a1bac3Timo Sirainen "\n"

f9d2a1f21ad65262bc630f0834d7eead06a1bac3Timo Sirainen "--foo bar\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Type: text/x-myown; charset=us-ascii\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "hello\n"

15d19d6e4daf460d8d2c82b981e23996dbdf7ba5Timo Sirainen "\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "--foo bar--\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n",

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .bodystructure =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "(\"text\" \"x-myown\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 7 1 NIL NIL NIL NIL) \"mixed\" (\"boundary\" \"foo bar\") NIL NIL NIL",

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .body =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "(\"text\" \"x-myown\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 7 1) \"mixed\""

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch },{

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch .message =

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "From: user@domain.org\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Mime-Version: 1.0\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Type: multipart/mixed; boundary=\"foo bar\"\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "--foo bar\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "Content-Type: text/plain; charset=us-ascii\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "See attached...\n"

3fcb3d2d1f3583025ff62bae95ec706920f398b1Stephan Bosch "\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "--foo bar\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "Content-Type: message/rfc822\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "From: user@domain.org\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "Mime-Version: 1.0\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "Content-Type: text/plain; charset=us-ascii\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "body\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "--foo bar--\n"

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "\n",

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch .bodystructure =

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "(\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 17 1 NIL NIL NIL NIL)(\"message\" \"rfc822\" NIL NIL NIL \"7bit\" 133 (\"Sat, 24 Mar 2017 23:00:00 +0200\" NIL ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) NIL NIL NIL NIL NIL) (\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 6 1 NIL NIL NIL NIL) 6 NIL NIL NIL NIL) \"mixed\" (\"boundary\" \"foo bar\") NIL NIL NIL",

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch .body =

e1a4ea6ad3e799ef8df7395e765c0ae9218e6c5dStephan Bosch "(\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 17 1)(\"message\" \"rfc822\" NIL NIL NIL \"7bit\" 133 (\"Sat, 24 Mar 2017 23:00:00 +0200\" NIL ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) NIL NIL NIL NIL NIL) (\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 6 1) 6) \"mixed\""

},{

.message =

"From: user@domain.org\n"

"Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

"Mime-Version: 1.0\n"

"Content-Type: multipart/mixed; boundary=\"foo bar\"\n"

"\n"

"--foo bar\n"

"Content-Type: text/plain; charset=us-ascii\n"

"Content-ID: <A.frop.example.com>\n"

"Content-Description: Container message\n"

"\n"

"See attached...\n"

"\n"

"--foo bar\n"

"Content-Type: message/rfc822\n"

"Content-ID: <B.frop.example.com>\n"

"Content-Description: Forwarded\n"

"\n"

"From: user@domain.org\n"

"Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

"Mime-Version: 1.0\n"

"Content-Type: text/plain; charset=us-ascii\n"

"\n"

"body\n"

"\n"

"--foo bar--\n",

.bodystructure =

"(\"text\" \"plain\" (\"charset\" \"us-ascii\") \"<A.frop.example.com>\" \"Container message\" \"7bit\" 17 1 NIL NIL NIL NIL)(\"message\" \"rfc822\" NIL \"<B.frop.example.com>\" \"Forwarded\" \"7bit\" 133 (\"Sat, 24 Mar 2017 23:00:00 +0200\" NIL ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) NIL NIL NIL NIL NIL) (\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 6 1 NIL NIL NIL NIL) 6 NIL NIL NIL NIL) \"mixed\" (\"boundary\" \"foo bar\") NIL NIL NIL",

.body =

"(\"text\" \"plain\" (\"charset\" \"us-ascii\") \"<A.frop.example.com>\" \"Container message\" \"7bit\" 17 1)(\"message\" \"rfc822\" NIL \"<B.frop.example.com>\" \"Forwarded\" \"7bit\" 133 (\"Sat, 24 Mar 2017 23:00:00 +0200\" NIL ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) ((NIL NIL \"user\" \"domain.org\")) NIL NIL NIL NIL NIL) (\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 6 1) 6) \"mixed\""

},{

.message =

"From: user@domain.org\n"

"Date: Sat, 24 Mar 2017 23:00:00 +0200\n"

"Mime-Version: 1.0\n"

"Content-Type: multipart/mixed; boundary=\"foo bar\"\n"

"\n"

"--foo bar\n"

"Content-Type: text/plain; charset=us-ascii; format=\"flowed\";\n"

" delsp=\"no\"\n"

"Content-Language: la\n"

"\n"

"Lorem ipsum dolor sit amet, consectetuer adipiscing elit. Aenean commodo\n"

"ligula eget dolor. Aenean massa. Cum sociis natoque penatibus et magnis dis\n"

"parturient montes, nascetur ridiculus mus. Donec quam felis, ultricies nec,\n"

"pellentesque eu, pretium quis, sem. Nulla consequat massa quis enim. Donec\n"

"pede justo, fringilla vel, aliquet nec, vulputate eget, arcu. In enim justo,\n"

"rhoncus ut, imperdiet a, venenatis vitae, justo. Nullam dictum felis eu pede\n"

"mollis pretium. Integer tincidunt. Cras dapibus. Vivamus elementum semper\n"

"nisi. Aenean vulputate eleifend tellus. Aenean leo ligula, porttitor eu,\n"

"consequat vitae, eleifend ac, enim. Aliquam lorem ante, dapibus in, viverra\n"

"quis, feugiat a, tellus. Phasellus viverra nulla ut metus varius laoreet.\n"

"Quisque rutrum. Aenean imperdiet. Etiam ultricies nisi vel augue. Curabitur\n"

"ullamcorper ultricies nisi. Nam eget dui.\n"

"\n"

"--foo bar\n"

"Content-Type: image/png\n"

"Content-Transfer-Encoding: base64\n"

"Content-Disposition: attachment; filename=\"pigeon.png\"\n"

"\n"

"iVBORw0KGgoAAAANSUhEUgAAAB8AAAAfCAYAAAAfrhY5AAAABHNCSVQICAgIfAhkiAAAAAlwSFlz\n"

"AAAGJwAABicBTVTYxwAAABl0RVh0U29mdHdhcmUAd3d3Lmlua3NjYXBlLm9yZ5vuPBoAAAN2SURB\n"

"VEiJ7ZfdK7tvHMffZuWpKTlR1my3bTEPU0hJcsABh0QrKYUTJ5KyHCzlyPwBkpSyg1mUpB3ILYna\n"

"lEYKcbADm4cDD7NpM2bv75HF1/az8l3fX/1+77pO7vu6rtf1/jx0X3caSeIvSfK3wP/DPykcDsNs\n"

"NqOqqgpdXV3Y2NhIGVz6+4PJyUlcXFxAFEV4vV709fVBrVZDpVL9eTo/aGVlhXl5efT5fDw/P2c4\n"

"HKbdbmd3dzdTIZDk6+srh4eHqdPpqNfrSZJGo5H7+/sMBoNUKpUpgUsAQCqVQq1WIxgMfgp/dXU1\n"

"srKykJ+fD6/X+8ejHiu4wcFB2Gy2uJPq6uqwt7eXOjgAaLXahHCn05laeCKlyvmXVosnrVaLQCCA\n"

"4uJiyOVyCIIQGyqVCoIgoKCgIDXwtLQ0HBwc4O3tDR6PB263G263G8fHx1hbW4Pb7cbNzQ1yc3Nj\n"

"hxEEAdXV1WhoaEi88cfSf3h4iLXaR01MTFChUFChULCjo4M7OztxW8fn89HlcnF5eZlTU1Nsb29n\n"

"W1sb/X5/3PlJwd/19vbG7e1tlpWVcXV19ftGJjk0NESbzRb33aeCy8zMhM/nSxgliUSCxsZGiKKI\n"

"ra2tb9N1e3uLSCSC8fFxGAwGDAwMfC7c309TWFjI09PTpFwlis76+jo7OzspCAJNJhN3d3fpcDho\n"

"tVqpUCgYCoVIkl8Krr6+Hna7HSUlJd86+yiPx4P5+XlYLBZUVFSgv78fVqsV6enpAIC7uzt4vV5E\n"

"IhFIJJL4zp1OJ2traxmJRL51+fLywuXlZba2trK0tJRms5mXl5c8PT3l4uIix8bG2NbWRqVSyfLy\n"

"cvb09PDw8DC2Po38eocbGRlBIBDA9PQ0pNKv3Xh2doa5uTksLCwgJycHNTU1yM3NxdHREa6vr6HR\n"

"aKDX62NDp9MhIyPjyz5x4SQxOjoKURTR29sLmUyG7Oxs+P1+WCwW7O7uQiKRQKvVQq/Xo7KyMgaS\n"

"y+VJpyou/F0nJydYWlqCy+XC1dUV7u/vYTQa8fT0BIfDgaWlpaRBcZVsFc/OznJsbIwkaTKZODMz\n"

"k+zShEr6Arm5uYnm5mYAgCiKaGlp+ZnrZJ1Ho1EWFRXx+fmZj4+P1Gg0P3ZNJqj2dxkMhth3PBAI\n"

"QCaTIRqNIhQKoamp6cc5/0d4qvXv+mn4z8B/AV1UVu6zi+zUAAAAAElFTkSuQmCC\n"

"--foo bar--\n",

.bodystructure =

"(\"text\" \"plain\" (\"charset\" \"us-ascii\" \"format\" \"flowed\" \"delsp\" \"no\") NIL NIL \"7bit\" 881 12 NIL NIL (\"la\") NIL)(\"image\" \"png\" NIL NIL NIL \"base64\" 1390 NIL (\"attachment\" (\"filename\" \"pigeon.png\")) NIL NIL) \"mixed\" (\"boundary\" \"foo bar\") NIL NIL NIL",

.body =

"(\"text\" \"plain\" (\"charset\" \"us-ascii\" \"format\" \"flowed\" \"delsp\" \"no\") NIL NIL \"7bit\" 881 12)(\"image\" \"png\" NIL NIL NIL \"base64\" 1390) \"mixed\""

},{

.message =

"From: user@domain.org\n"

"Date: Sat, 24 Mar 2007 23:00:00 +0200\n"

"Mime-Version: 1.0\n"

"Content-Type: multipart/mixed; boundary=\"foo\n"

" bar\"\n"

"\n"

"Root MIME prologue\n"

"\n"

"--foo bar\n"

"Content-Type: text/x-myown; charset=us-ascii; foo=\"quoted\\\"string\"\n"

"Content-ID: <foo@example.com>\n"

"Content-MD5: Q2hlY2sgSW50ZWdyaXR5IQ==\n"

"Content-Disposition: inline; foo=bar\n"

"Content-Description: hellodescription\n"

"Content-Language: en, fi, se\n"

"Content-Location: http://example.com/test.txt\n"

"\n"

"hello\n"

"\n"

"--foo bar\n"

"Content-Type: message/rfc822\n"

"\n"

"From: sub@domain.org\n"

"To: sub-to1@domain.org, sub-to2@domain.org\n"

"Date: Sun, 12 Aug 2012 12:34:56 +0300\n"

"Subject: submsg\n"

"Content-Type: multipart/alternative; boundary=\"sub1\"\n"

"\n"

"Sub MIME prologue\n"

"--sub1\n"

"Content-Type: text/html\n"

"Content-Transfer-Encoding: 8bit\n"

"\n"

"<p>Hello world</p>\n"

"\n"

"--sub1\n"

"Content-Type: text/plain\n"

"Content-Transfer-Encoding: ?invalid\n"

"\n"

"Hello another world\n"

"\n"

"--sub1--\n"

"Sub MIME epilogue\n"

"\n"

"--foo bar--\n"

"Root MIME epilogue\n",

.bodystructure =

"(\"text\" \"x-myown\" (\"charset\" \"us-ascii\" \"foo\" \"quoted\\\"string\") \"<foo@example.com>\" \"hellodescription\" \"7bit\" 7 1 \"Q2hlY2sgSW50ZWdyaXR5IQ==\" (\"inline\" (\"foo\" \"bar\")) (\"en\" \"fi\" \"se\") \"http://example.com/test.txt\")(\"message\" \"rfc822\" NIL NIL NIL \"7bit\" 412 (\"Sun, 12 Aug 2012 12:34:56 +0300\" \"submsg\" ((NIL NIL \"sub\" \"domain.org\")) ((NIL NIL \"sub\" \"domain.org\")) ((NIL NIL \"sub\" \"domain.org\")) ((NIL NIL \"sub-to1\" \"domain.org\")(NIL NIL \"sub-to2\" \"domain.org\")) NIL NIL NIL NIL) ((\"text\" \"html\" (\"charset\" \"us-ascii\") NIL NIL \"8bit\" 20 1 NIL NIL NIL NIL)(\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 21 1 NIL NIL NIL NIL) \"alternative\" (\"boundary\" \"sub1\") NIL NIL NIL) 21 NIL NIL NIL NIL) \"mixed\" (\"boundary\" \"foo bar\") NIL NIL NIL",

.body =

"(\"text\" \"x-myown\" (\"charset\" \"us-ascii\" \"foo\" \"quoted\\\"string\") \"<foo@example.com>\" \"hellodescription\" \"7bit\" 7 1)(\"message\" \"rfc822\" NIL NIL NIL \"7bit\" 412 (\"Sun, 12 Aug 2012 12:34:56 +0300\" \"submsg\" ((NIL NIL \"sub\" \"domain.org\")) ((NIL NIL \"sub\" \"domain.org\")) ((NIL NIL \"sub\" \"domain.org\")) ((NIL NIL \"sub-to1\" \"domain.org\")(NIL NIL \"sub-to2\" \"domain.org\")) NIL NIL NIL NIL) ((\"text\" \"html\" (\"charset\" \"us-ascii\") NIL NIL \"8bit\" 20 1)(\"text\" \"plain\" (\"charset\" \"us-ascii\") NIL NIL \"7bit\" 21 1) \"alternative\") 21) \"mixed\""

}

};

static const unsigned int parse_tests_count = N_ELEMENTS(parse_tests);

static struct message_part *

msg_parse(pool_t pool, const char *message, bool parse_bodystructure)

{

struct message_parser_ctx *parser;

struct istream *input;

struct message_block block;

struct message_part *parts;

int ret;

input = i_stream_create_from_data(message, strlen(message));

parser = message_parser_init(pool, input,

MESSAGE_HEADER_PARSER_FLAG_SKIP_INITIAL_LWSP |

MESSAGE_HEADER_PARSER_FLAG_DROP_CR,

MESSAGE_PARSER_FLAG_SKIP_BODY_BLOCK);

while ((ret = message_parser_parse_next_block(parser, &block)) > 0) {

if (parse_bodystructure) {

imap_bodystructure_parse_header(pool, block.part,

block.hdr);

}

}

test_assert(ret < 0);

message_parser_deinit(&parser, &parts);

i_stream_unref(&input);

return parts;

}

static void test_imap_bodystructure_write(void)

{

struct message_part *parts;

unsigned int i;

for (i = 0; i < parse_tests_count; i++) T_BEGIN {

struct parse_test *test = &parse_tests[i];

string_t *str = t_str_new(128);

pool_t pool = pool_alloconly_create("imap bodystructure write", 1024);

test_begin(t_strdup_printf("imap bodystructure write [%u]", i));

parts = msg_parse(pool, test->message, TRUE);

imap_bodystructure_write(parts, str, TRUE);

test_assert(strcmp(str_c(str), test->bodystructure) == 0);

str_truncate(str, 0);

imap_bodystructure_write(parts, str, FALSE);

test_assert(strcmp(str_c(str), test->body) == 0);

pool_unref(&pool);

test_end();

} T_END;

}

static void test_imap_bodystructure_parse(void)

{

struct message_part *parts;

const char *error;

unsigned int i;

int ret;

for (i = 0; i < parse_tests_count; i++) T_BEGIN {

struct parse_test *test = &parse_tests[i];

string_t *str = t_str_new(128);

pool_t pool = pool_alloconly_create("imap bodystructure parse", 1024);

test_begin(t_strdup_printf("imap bodystructure parser [%u]", i));

parts = msg_parse(pool, test->message, FALSE);

test_assert(imap_body_parse_from_bodystructure(test->bodystructure,

str, &error) == 0);

test_assert(strcmp(str_c(str), test->body) == 0);

ret = imap_bodystructure_parse(test->bodystructure,

pool, parts, &error);

test_assert(ret == 0);

if (ret == 0) {

str_truncate(str, 0);

imap_bodystructure_write(parts, str, TRUE);

test_assert(strcmp(str_c(str), test->bodystructure) == 0);

} else {

i_error("Invalid BODYSTRUCTURE: %s", error);

}

pool_unref(&pool);

test_end();

} T_END;

}

int main(void)

{

static void (*const test_functions[])(void) = {

test_imap_bodystructure_write,

test_imap_bodystructure_parse,

NULL

};

return test_run(test_functions);

}