/*-
* 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);
}