/*-
 * Copyright (c) 2009,2010 Michihiro NAKAJIMA
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#include "test.h"

/*
 * Check that an ISO 9660 image is correctly created.
 */
static void
add_entry(struct archive *a, const char *fname, const char *sym)
{
	struct archive_entry *ae;

	assert((ae = archive_entry_new()) != NULL);
	archive_entry_set_birthtime(ae, 2, 20);
	archive_entry_set_atime(ae, 3, 30);
	archive_entry_set_ctime(ae, 4, 40);
	archive_entry_set_mtime(ae, 5, 50);
	archive_entry_copy_pathname(ae, fname);
	if (sym != NULL)
		archive_entry_set_symlink(ae, sym);
	archive_entry_set_mode(ae, S_IFREG | 0555);
	archive_entry_set_size(ae, 0);
	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
	archive_entry_free(ae);
}

struct fns {
	size_t	maxlen;
	size_t	longest_len;
	size_t	maxflen;
	size_t	maxelen;
	size_t	alloc;
	int	cnt;
	char	**names;
	int	opt;
#define	UPPER_CASE_ONLY	0x00001
#define	ONE_DOT		0x00002
#define	ALLOW_LDOT	0x00004
};

enum vtype {
	ROCKRIDGE,
	JOLIET,
	ISO9660
};

/*
 * Verify file
 */
static void
verify_file(struct archive *a, enum vtype type, struct fns *fns)
{
	struct archive_entry *ae;
	int i;

	assertEqualIntA(a, 0, archive_read_next_header(a, &ae));
	if (type == ROCKRIDGE) {
		assertEqualInt(2, archive_entry_birthtime(ae));
		assertEqualInt(3, archive_entry_atime(ae));
		assertEqualInt(4, archive_entry_ctime(ae));
	} else {
		assertEqualInt(0, archive_entry_birthtime_is_set(ae));
		assertEqualInt(5, archive_entry_atime(ae));
		assertEqualInt(5, archive_entry_ctime(ae));
	}
	assertEqualInt(5, archive_entry_mtime(ae));
	if (type == ROCKRIDGE)
		assert((S_IFREG | 0555) == archive_entry_mode(ae));
	else
		assert((S_IFREG | 0400) == archive_entry_mode(ae));
	assertEqualInt(0, archive_entry_size(ae));

	/*
	 * Check if the same filename does not appear.
	 */
	for (i = 0; i < fns->cnt; i++) {
		const char *p;
		const char *pathname = archive_entry_pathname(ae);
		const char *symlinkname = archive_entry_symlink(ae);
		size_t length;

		if (symlinkname != NULL) {
			length = strlen(symlinkname);
			assert(length == 1 || length == 128 || length == 255);
			assertEqualInt(symlinkname[length-1], 'x');
		}
		failure("Found duplicate for %s", pathname);
		assert(strcmp(fns->names[i], pathname) != 0);
		assert((length = strlen(pathname)) <= fns->maxlen);
		if (length > fns->longest_len)
			fns->longest_len = length;
		p = strrchr(pathname, '.');
		if (p != NULL) {
			/* Check a length of file name. */
			assert((size_t)(p - pathname) <= fns->maxflen);
			/* Check a length of file extension. */
			assert(strlen(p+1) <= fns->maxelen);
			if (fns->opt & ONE_DOT) {
				/* Do not have multi dot. */
				assert(strchr(pathname, '.') == p);
			}
		}
		for (p = pathname; *p; p++) {
			if (fns->opt & UPPER_CASE_ONLY) {
				/* Do not have any lower-case character. */
				assert(*p < 'a' || *p > 'z');
			} else
				break;
		}
		if ((fns->opt & ALLOW_LDOT) == 0)
			/* Do not have a dot at the first position. */
			assert(*pathname != '.');
	}
	/* Save the filename which is appeared to use above next time. */
	fns->names[fns->cnt++] = strdup(archive_entry_pathname(ae));
}

static void
verify(unsigned char *buff, size_t used, enum vtype type, struct fns *fns)
{
	struct archive *a;
	struct archive_entry *ae;
	size_t i;

	/*
	 * Read ISO image.
	 */
	assert((a = archive_read_new()) != NULL);
	assertEqualIntA(a, 0, archive_read_support_format_all(a));
	assertEqualIntA(a, 0, archive_read_support_filter_all(a));
	if (type >= 1)
		assertA(0 == archive_read_set_option(a, NULL, "rockridge",
		    NULL));
	if (type >= 2)
		assertA(0 == archive_read_set_option(a, NULL, "joliet",
		    NULL));
	assertEqualIntA(a, 0, archive_read_open_memory(a, buff, used));

	/*
	 * Read Root Directory
	 * Root Directory entry must be in ISO image.
	 */
	assertEqualIntA(a, 0, archive_read_next_header(a, &ae));
	assertEqualInt(archive_entry_atime(ae), archive_entry_ctime(ae));
	assertEqualInt(archive_entry_atime(ae), archive_entry_mtime(ae));
	assertEqualString(".", archive_entry_pathname(ae));
	switch (type) {
	case ROCKRIDGE:
		assert((S_IFDIR | 0555) == archive_entry_mode(ae));
		break;
	case JOLIET:
		assert((S_IFDIR | 0700) == archive_entry_mode(ae));
		break;
	case ISO9660:
		assert((S_IFDIR | 0700) == archive_entry_mode(ae));
		break;
	}

	/*
	 * Verify file status.
	 */
	memset(fns->names, 0, sizeof(char *) * fns->alloc);
	fns->cnt = 0;
	for (i = 0; i < fns->alloc; i++)
		verify_file(a, type, fns);
	for (i = 0; i < fns->alloc; i++)
		free(fns->names[i]);
	assertEqualInt((int)fns->longest_len, (int)fns->maxlen);

	/*
	 * Verify the end of the archive.
	 */
	assertEqualIntA(a, ARCHIVE_EOF, archive_read_next_header(a, &ae));
	assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
	assertEqualIntA(a, ARCHIVE_OK, archive_read_free(a));
}

static int
create_iso_image(unsigned char *buff, size_t buffsize, size_t *used,
    const char *opt)
{
	struct archive *a;
	int i, l, fcnt;
	const int lens[] = {
	    0, 1, 3, 5, 7, 8, 9, 29, 30, 31, 32,
		62, 63, 64, 65, 101, 102, 103, 104,
	    191, 192, 193, 194, 204, 205, 206, 207, 208,
		252, 253, 254, 255,
	    -1 };
	char fname1[256];
	char fname2[256];
	char sym1[2];
	char sym128[129];
	char sym255[256];

	/* ISO9660 format: Create a new archive in memory. */
	assert((a = archive_write_new()) != NULL);
	assertA(0 == archive_write_set_format_iso9660(a));
	assertA(0 == archive_write_add_filter_none(a));
	assertA(0 == archive_write_set_option(a, NULL, "pad", NULL));
	if (opt)
		assertA(0 == archive_write_set_options(a, opt));
	assertA(0 == archive_write_set_bytes_per_block(a, 1));
	assertA(0 == archive_write_set_bytes_in_last_block(a, 1));
	assertA(0 == archive_write_open_memory(a, buff, buffsize, used));

	sym1[0] = 'x';
	sym1[1] = '\0';
	for (i = 0; i < (int)sizeof(sym128)-2; i++)
		sym128[i] = 'a';
	sym128[sizeof(sym128)-2] = 'x';
	sym128[sizeof(sym128)-1] = '\0';
	for (i = 0; i < (int)sizeof(sym255)-2; i++)
		sym255[i] = 'a';
	sym255[sizeof(sym255)-2] = 'x';
	sym255[sizeof(sym255)-1] = '\0';

	fcnt = 0;
	for (i = 0; lens[i] >= 0; i++) {
		for (l = 0; l < lens[i]; l++) {
			fname1[l] = 'a';
			fname2[l] = 'A';
		}
		if (l > 0) {
			fname1[l] = '\0';
			fname2[l] = '\0';
			add_entry(a, fname1, NULL);
			add_entry(a, fname2, sym1);
			fcnt += 2;
		}
		if (l < 254) {
			fname1[l] = '.';
			fname1[l+1] = 'c';
			fname1[l+2] = '\0';
			fname2[l] = '.';
			fname2[l+1] = 'C';
			fname2[l+2] = '\0';
			add_entry(a, fname1, sym128);
			add_entry(a, fname2, sym255);
			fcnt += 2;
		}
		if (l < 252) {
			fname1[l] = '.';
			fname1[l+1] = 'p';
			fname1[l+2] = 'n';
			fname1[l+3] = 'g';
			fname1[l+4] = '\0';
			fname2[l] = '.';
			fname2[l+1] = 'P';
			fname2[l+2] = 'N';
			fname2[l+3] = 'G';
			fname2[l+4] = '\0';
			add_entry(a, fname1, NULL);
			add_entry(a, fname2, sym1);
			fcnt += 2;
		}
		if (l < 251) {
			fname1[l] = '.';
			fname1[l+1] = 'j';
			fname1[l+2] = 'p';
			fname1[l+3] = 'e';
			fname1[l+4] = 'g';
			fname1[l+5] = '\0';
			fname2[l] = '.';
			fname2[l+1] = 'J';
			fname2[l+2] = 'P';
			fname2[l+3] = 'E';
			fname2[l+4] = 'G';
			fname2[l+5] = '\0';
			add_entry(a, fname1, sym128);
			add_entry(a, fname2, sym255);
			fcnt += 2;
		}
	}

	/* Close out the archive. */
	assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
	assertEqualIntA(a, ARCHIVE_OK, archive_write_free(a));

	return (fcnt);
}

DEFINE_TEST(test_write_format_iso9660_filename)
{
	unsigned char *buff;
	size_t buffsize = 120 * 2048;
	size_t used;
	int fcnt;
	struct fns fns;

	buff = malloc(buffsize);
	assert(buff != NULL);
	if (buff == NULL)
		return;
	memset(&fns, 0, sizeof(fns));

	/*
	 * Create ISO image with no option.
	 */
	fcnt = create_iso_image(buff, buffsize, &used, NULL);

	fns.names = (char **)malloc(sizeof(char *) * fcnt);
	assert(fns.names != NULL);
	if (fns.names == NULL) {
		free(buff);
		return;
	}
	fns.alloc = fcnt;

	/* Verify rockridge filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 255;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ROCKRIDGE, &fns);

	/* Verify joliet filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 64;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, JOLIET, &fns);

	/* Verify ISO9660 filenames. */
	fns.longest_len = 0;
	fns.maxlen = 8+3+1;
	fns.maxflen = 8;
	fns.maxelen = 3;
	fns.opt = UPPER_CASE_ONLY | ONE_DOT;
	verify(buff, used, ISO9660, &fns);

	/*
	 * Create ISO image with iso-level=2.
	 */
	assertEqualInt(fcnt, create_iso_image(buff, buffsize, &used,
	    "iso-level=2"));

	/* Verify rockridge filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 255;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ROCKRIDGE, &fns);

	/* Verify joliet filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 64;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, JOLIET, &fns);

	/* Verify ISO9660 filenames. */
	fns.longest_len = 0;
	fns.maxlen = 31;
	fns.maxflen = 30;
	fns.maxelen = 30;
	fns.opt = UPPER_CASE_ONLY | ONE_DOT;
	verify(buff, used, ISO9660, &fns);

	/*
	 * Create ISO image with iso-level=3.
	 */
	assertEqualInt(fcnt, create_iso_image(buff, buffsize, &used,
	    "iso-level=3"));

	/* Verify rockridge filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 255;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ROCKRIDGE, &fns);

	/* Verify joliet filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 64;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, JOLIET, &fns);

	/* Verify ISO9660 filenames. */
	fns.longest_len = 0;
	fns.maxlen = 31;
	fns.maxflen = 30;
	fns.maxelen = 30;
	fns.opt = UPPER_CASE_ONLY | ONE_DOT;
	verify(buff, used, ISO9660, &fns);

	/*
	 * Create ISO image with iso-level=4.
	 */
	assertEqualInt(fcnt, create_iso_image(buff, buffsize, &used,
	    "iso-level=4"));

	/* Verify rockridge filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 255;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ROCKRIDGE, &fns);

	/* Verify joliet filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 64;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, JOLIET, &fns);

	/* Verify ISO9660 filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 193;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ISO9660, &fns);

	/*
	 * Create ISO image with iso-level=4 and !rockridge.
	 */
	assertEqualInt(fcnt, create_iso_image(buff, buffsize, &used,
	    "iso-level=4,!rockridge"));

	/* Verify joliet filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 64;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, JOLIET, &fns);

	/* Verify ISO9660 filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 207;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ISO9660, &fns);

	/*
	 * Create ISO image with joliet=long.
	 */
	assertEqualInt(fcnt, create_iso_image(buff, buffsize, &used,
	    "joliet=long"));

	/* Verify rockridge filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 255;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, ROCKRIDGE, &fns);

	/* Verify joliet filenames. */
	fns.longest_len = 0;
	fns.maxlen = fns.maxflen = fns.maxelen = 103;
	fns.opt = ALLOW_LDOT;
	verify(buff, used, JOLIET, &fns);

	/* Verify ISO9660 filenames. */
	fns.longest_len = 0;
	fns.maxlen = 8+3+1;
	fns.maxflen = 8;
	fns.maxelen = 3;
	fns.opt = UPPER_CASE_ONLY | ONE_DOT;
	verify(buff, used, ISO9660, &fns);

	free(fns.names);
	free(buff);
}