/*-

 * Copyright (c) 2003-2007 Tim Kientzle

 * 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"

__FBSDID("$FreeBSD: src/usr.bin/cpio/test/test_basic.c,v 1.4 2008/08/25 06:39:29 kientzle Exp $");

static void

verify_files(const char *msg)

{

	/*

	 * Verify unpacked files.

	 */

	/* Regular file with 2 links. */

	failure(msg);

	assertIsReg("file", 0644);

	failure(msg);

	assertFileSize("file", 10);

	failure(msg);

	assertFileNLinks("file", 2);

	/* Another name for the same file. */

	failure(msg);

	assertIsHardlink("linkfile", "file");

	/* Symlink */

	if (canSymlink())

		assertIsSymlink("symlink", "file");

	/* Another file with 1 link and different permissions. */

	failure(msg);

	assertIsReg("file2", 0777);

	failure(msg);

	assertFileSize("file2", 10);

	failure(msg);

	assertFileNLinks("file2", 1);

	/* dir */

	assertIsDir("dir", 0775);

}

static void

basic_cpio(const char *target,

    const char *pack_options,

    const char *unpack_options,

    const char *se, const char *se2)

{

	int r;

	if (!assertMakeDir(target, 0775))

	    return;

	/* Use the cpio program to create an archive. */

	r = systemf("%s -R 1000:1000 -o %s < filelist >%s/archive 2>%s/pack.err",

	    testprog, pack_options, target, target);

	failure("Error invoking %s -o %s", testprog, pack_options);

	assertEqualInt(r, 0);

	assertChdir(target);

	/* Verify stderr. */

	failure("Expected: %s, options=%s", se, pack_options);

	assertTextFileContents(se, "pack.err");

	/*

	 * Use cpio to unpack the archive into another directory.

	 */

	r = systemf("%s -i %s< archive >unpack.out 2>unpack.err",

	    testprog, unpack_options);

	failure("Error invoking %s -i %s", testprog, unpack_options);

	assertEqualInt(r, 0);

	/* Verify stderr. */

	failure("Error invoking %s -i %s in dir %s", testprog, unpack_options, target);

	assertTextFileContents(se2, "unpack.err");

	verify_files(pack_options);

	assertChdir("..");

}

static void

passthrough(const char *target)

{

	int r;

	if (!assertMakeDir(target, 0775))

		return;

	/*

	 * Use cpio passthrough mode to copy files to another directory.

	 */

	r = systemf("%s -p %s <filelist >%s/stdout 2>%s/stderr",

	    testprog, target, target, target);

	failure("Error invoking %s -p", testprog);

	assertEqualInt(r, 0);

	assertChdir(target);

	/* Verify stderr. */

	failure("Error invoking %s -p in dir %s",

	    testprog, target);

	assertTextFileContents("1 block\n", "stderr");

	verify_files("passthrough");

	assertChdir("..");

}

DEFINE_TEST(test_basic)

{

	FILE *filelist;

	const char *msg;

	char result[1024];

	assertUmask(0);

	/*

	 * Create an assortment of files on disk.

	 */

	filelist = fopen("filelist", "w");

	memset(result, 0, sizeof(result));

	/* File with 10 bytes content. */

	assertMakeFile("file", 0644, "1234567890");

	fprintf(filelist, "file\n");

	if (is_LargeInode("file"))

		strncat(result,

		    "bsdcpio: file: large inode number truncated: "

		    "Numerical result out of range\n",

		    sizeof(result) - strlen(result) -1);

	/* hardlink to above file. */

	assertMakeHardlink("linkfile", "file");

	fprintf(filelist, "linkfile\n");

	if (is_LargeInode("linkfile"))

		strncat(result,

		    "bsdcpio: linkfile: large inode number truncated: "

		    "Numerical result out of range\n",

		    sizeof(result) - strlen(result) -1);

	/* Symlink to above file. */

	if (canSymlink()) {

		assertMakeSymlink("symlink", "file");

		fprintf(filelist, "symlink\n");

		if (is_LargeInode("symlink"))

			strncat(result,

			    "bsdcpio: symlink: large inode number truncated: "

				"Numerical result out of range\n",

			    sizeof(result) - strlen(result) -1);

	}

	/* Another file with different permissions. */

	assertMakeFile("file2", 0777, "1234567890");

	fprintf(filelist, "file2\n");

	if (is_LargeInode("file2"))

		strncat(result,

		    "bsdcpio: file2: large inode number truncated: "

		    "Numerical result out of range\n",

		    sizeof(result) - strlen(result) -1);

	/* Directory. */

	assertMakeDir("dir", 0775);

	fprintf(filelist, "dir\n");

	if (is_LargeInode("dir"))

		strncat(result,

		    "bsdcpio: dir: large inode number truncated: "

		    "Numerical result out of range\n",

		    sizeof(result) - strlen(result) -1);

	strncat(result, "2 blocks\n", sizeof(result) - strlen(result) -1);

	/* All done. */

	fclose(filelist);

	assertUmask(022);

	/* Archive/dearchive with a variety of options. */

	msg = canSymlink() ? "2 blocks\n" : "1 block\n";

	basic_cpio("copy", "", "", msg, msg);

	basic_cpio("copy_odc", "--format=odc", "", msg, msg);

	basic_cpio("copy_newc", "-H newc", "", result, "2 blocks\n");

	basic_cpio("copy_cpio", "-H odc", "", msg, msg);

	msg = canSymlink() ? "9 blocks\n" : "8 blocks\n";

	basic_cpio("copy_ustar", "-H ustar", "", msg, msg);

	/* Copy in one step using -p */

	passthrough("passthrough");

}