test_write_format_mtree.c [plain text]
#include "test.h"
__FBSDID("$FreeBSD: head/lib/libarchive/test/test_write_format_mtree.c 191183 2009-04-17 01:06:31Z kientzle $");
static char buff[4096];
static struct {
const char *path;
mode_t mode;
time_t mtime;
uid_t uid;
gid_t gid;
} entries[] = {
{ "./Makefile", S_IFREG | 0644, 1233041050, 1001, 1001 },
{ "./NEWS", S_IFREG | 0644, 1231975636, 1001, 1001 },
{ "./PROJECTS", S_IFREG | 0644, 1231975636, 1001, 1001 },
{ "./README", S_IFREG | 0644, 1231975636, 1001, 1001 },
{ "./COPYING", S_IFREG | 0644, 1231975636, 1001, 1001 },
{ "./subdir", S_IFDIR | 0755, 1233504586, 1001, 1001 },
{ "./subdir/README", S_IFREG | 0664, 1231975636, 1002, 1001 },
{ "./subdir/config", S_IFREG | 0664, 1232266273, 1003, 1003 },
{ "./subdir2", S_IFDIR | 0755, 1233504586, 1001, 1001 },
{ "./subdir3", S_IFDIR | 0755, 1233504586, 1001, 1001 },
{ "./subdir3/mtree", S_IFREG | 0664, 1232266273, 1003, 1003 },
{ NULL, 0, 0, 0, 0 }
};
static void
test_write_format_mtree_sub(int use_set, int dironly)
{
struct archive_entry *ae;
struct archive* a;
size_t used;
int i;
assert((a = archive_write_new()) != NULL);
assertA(0 == archive_write_set_format_mtree(a));
if (use_set)
assertA(0 == archive_write_set_options(a, "use-set"));
if (dironly)
assertA(0 == archive_write_set_options(a, "dironly"));
assertA(0 == archive_write_open_memory(a, buff, sizeof(buff)-1, &used));
for (i = 0; entries[i].path != NULL; i++) {
assert((ae = archive_entry_new()) != NULL);
archive_entry_set_mtime(ae, entries[i].mtime, 0);
assert(entries[i].mtime == archive_entry_mtime(ae));
archive_entry_set_mode(ae, entries[i].mode);
assert(entries[i].mode == archive_entry_mode(ae));
archive_entry_set_uid(ae, entries[i].uid);
assert(entries[i].uid == archive_entry_uid(ae));
archive_entry_set_gid(ae, entries[i].gid);
assert(entries[i].gid == archive_entry_gid(ae));
archive_entry_copy_pathname(ae, entries[i].path);
if ((entries[i].mode & AE_IFMT) != S_IFDIR)
archive_entry_set_size(ae, 8);
assertA(0 == archive_write_header(a, ae));
if ((entries[i].mode & AE_IFMT) != S_IFDIR)
assertA(8 == archive_write_data(a, "Hello012", 15));
archive_entry_free(ae);
}
archive_write_close(a);
#if ARCHIVE_VERSION_NUMBER < 2000000
archive_write_finish(a);
#else
assertEqualInt(0, archive_write_finish(a));
#endif
if (use_set) {
const char *p;
buff[used] = '\0';
assert(NULL != (p = strstr(buff, "\n/set ")));
if (p != NULL) {
char *r;
const char *o;
p++;
r = strchr(p, '\n');
if (r != NULL)
*r = '\0';
if (dironly)
o = "/set type=dir uid=1001 gid=1001 mode=755";
else
o = "/set type=file uid=1001 gid=1001 mode=644";
assertEqualString(o, p);
if (r != NULL)
*r = '\n';
}
}
assert((a = archive_read_new()) != NULL);
assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
assertEqualIntA(a, ARCHIVE_OK, archive_read_support_compression_all(a));
assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
for (i = 0; entries[i].path != NULL; i++) {
if (dironly && (entries[i].mode & AE_IFMT) != S_IFDIR)
continue;
assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
assertEqualInt(entries[i].mtime, archive_entry_mtime(ae));
assertEqualInt(entries[i].mode, archive_entry_mode(ae));
assertEqualInt(entries[i].uid, archive_entry_uid(ae));
assertEqualInt(entries[i].gid, archive_entry_gid(ae));
assertEqualString(entries[i].path, archive_entry_pathname(ae));
if ((entries[i].mode & AE_IFMT) != S_IFDIR)
assertEqualInt(8, archive_entry_size(ae));
}
assertEqualIntA(a, 0, archive_read_close(a));
#if ARCHIVE_VERSION_NUMBER < 2000000
archive_read_finish(a);
#else
assertEqualInt(0, archive_read_finish(a));
#endif
}
DEFINE_TEST(test_write_format_mtree)
{
test_write_format_mtree_sub(0, 0);
test_write_format_mtree_sub(0, 1);
test_write_format_mtree_sub(1, 0);
test_write_format_mtree_sub(1, 1);
}