test-istream-binary-converter.c revision 2c26b19c543706fff723c3f52b401817507a0ee0
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen/* Copyright (c) 2007-2012 Dovecot authors, see the included COPYING file */
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "lib.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "str.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "sha1.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "istream.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "istream-crlf.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "istream-binary-converter.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include "test-common.h"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#include <stdio.h>
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#define BINARY_TEXT_LONG "we have\ra lot \nof \0binary stuff in here\n" \
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"b adjig sadjg jasidgjiaehga3wht8a3w8ghxjc dsgad hasdghsd gasd ds" \
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"jdsoga sjdga0w3tjhawjgsertniq3n5oqerjqw2r89q23h awhrqh835r8a"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#define BINARY_TEXT_LONG_BASE64 \
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"d2UgaGF2ZQ1hIGxvdCAKb2YgAGJpbmFyeSBzdHVmZiBpbiBoZXJlCmIgYWRqaWcgc2FkamcgamFz\r\n" \
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"aWRnamlhZWhnYTN3aHQ4YTN3OGdoeGpjIGRzZ2FkIGhhc2RnaHNkIGdhc2QgZHNqZHNvZ2Egc2pk\r\n" \
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Z2EwdzN0amhhd2pnc2VydG5pcTNuNW9xZXJqcXcycjg5cTIzaCBhd2hycWg4MzVyOGE="
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#define BINARY_TEXT_SHORT "eh"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen#define BINARY_TEXT_SHORT_BASE64 "ZWg="
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainenstatic const char mail_input[] =
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"MIME-Version: 1.0\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Type: multipart/alternative;\r\n boundary=\"bound\"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"mime header\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Transfer-Encoding: binary\r\n"
2c26b19c543706fff723c3f52b401817507a0ee0Timo Sirainen"Content-Type: text/plain\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo SirainenBINARY_TEXT_LONG
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Type: text/plain\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Transfer-Encoding: binary\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo SirainenBINARY_TEXT_SHORT
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Type: text/plain\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"hello world\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound--\r\n";
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainenstatic const char mail_output[] =
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"MIME-Version: 1.0\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Type: multipart/alternative;\r\n boundary=\"bound\"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"mime header\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Transfer-Encoding: base64\r\n"
2c26b19c543706fff723c3f52b401817507a0ee0Timo Sirainen"Content-Type: text/plain\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo SirainenBINARY_TEXT_LONG_BASE64
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Type: text/plain\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Transfer-Encoding: base64\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo SirainenBINARY_TEXT_SHORT_BASE64
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"Content-Type: text/plain\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"hello world\r\n"
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen"\r\n--bound--\r\n";
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainenstatic void test_istream_binary_converter(void)
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen{
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen struct istream *datainput, *input;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen const unsigned char *data;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen size_t i, size;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen int ret;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_begin("istream binary converter");
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen datainput = test_istream_create_data(mail_input, sizeof(mail_input));
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_istream_set_allow_eof(datainput, FALSE);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen input = i_stream_create_binary_converter(datainput);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen for (i = 1; i <= sizeof(mail_input); i++) {
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_istream_set_size(datainput, i);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen while ((ret = i_stream_read(input)) > 0) ;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_assert(ret == 0);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen }
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_istream_set_allow_eof(datainput, TRUE);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen while ((ret = i_stream_read(input)) > 0) ;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_assert(ret == -1);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen data = i_stream_get_data(input, &size);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_assert(size == sizeof(mail_output) &&
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen memcmp(data, mail_output, size) == 0);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_unref(&input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_unref(&datainput);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_end();
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen}
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainenstatic int test_input_file(const char *path)
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen{
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen struct istream *file_input, *input, *input2;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen const unsigned char *data;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen size_t size;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen struct sha1_ctxt hash;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen unsigned char hash_file[SHA1_RESULTLEN], hash_converter[SHA1_RESULTLEN];
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen int ret = 0;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen lib_init();
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen file_input = i_stream_create_file(path, 64);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen /* get hash when directly reading input */
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen input = i_stream_create_crlf(file_input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen sha1_init(&hash);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen while (i_stream_read_data(input, &data, &size, 0) > 0) {
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen sha1_loop(&hash, data, size);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_skip(input, size);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen }
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen sha1_result(&hash, hash_file);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_unref(&input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen /* get hash when going through converter */
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_seek(file_input, 0);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen input = i_stream_create_crlf(file_input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen input2 = i_stream_create_binary_converter(input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen sha1_init(&hash);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen while (i_stream_read_data(input2, &data, &size, 0) > 0) {
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen sha1_loop(&hash, data, size);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_skip(input2, size);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen }
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen sha1_result(&hash, hash_converter);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_unref(&input2);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_unref(&input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen if (memcmp(hash_file, hash_converter, SHA1_RESULTLEN) != 0) {
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen fprintf(stderr, "istream-binary-converter: mismatch on file %s\n",
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen path);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen ret = 1;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen }
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen i_stream_unref(&file_input);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen lib_deinit();
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen return ret;
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen}
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainenint main(int argc, char *argv[])
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen{
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen static void (*test_functions[])(void) = {
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen test_istream_binary_converter,
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen NULL
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen };
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen if (argc > 1)
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen return test_input_file(argv[1]);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen else
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen return test_run(test_functions);
d51bddedf8c02326124935e1ffbee5da2275f8d4Timo Sirainen}