/* Copyright (c) Wipro Technologies Ltd, 2002.  All Rights Reserved.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it would be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */
/*
 * This test case checks whether swapon(2) system call  returns
 *  1. ENOENT when the path does not exist
 *  2. EINVAL when the path exists but is invalid
 *  3. EPERM when user is not a superuser
 *  4. EBUSY when the specified path is already being used as a swap area
 */

#include <unistd.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pwd.h>
#include <string.h>
#include <sys/utsname.h>
#include <signal.h>
#include "test.h"
#include "linux_syscall_numbers.h"
#include "safe_macros.h"
#include "tst_fs_type.h"
#include "libswapon.h"

static void setup(void);
static void cleanup(void);
static void setup01(void);
static void cleanup01(void);

char *TCID = "swapon02";
int TST_TOTAL = 4;

static uid_t nobody_uid;
static int do_swapoff;
static long fs_type;

static struct test_case_t {
	char *err_desc;
	int exp_errno;
	char *exp_errval;
	char *path;
	void (*setup)(void);
	void (*cleanup)(void);
} testcases[] = {
	{"Path does not exist", ENOENT, "ENOENT", "./doesnotexist", NULL, NULL},
	{"Invalid path", EINVAL, "EINVAL", "./notswap", NULL, NULL},
	{"Permission denied", EPERM, "EPERM", "./swapfile01",
	 setup01, cleanup01},
	{"File already used", EBUSY, "EBUSY", "./alreadyused", NULL, NULL},
};

static void verify_swapon(struct test_case_t *test)
{
	if (test->setup)
		test->setup();

	TEST(ltp_syscall(__NR_swapon, test->path, 0));

	if (test->cleanup)
		test->cleanup();

	if (TEST_RETURN == -1 && TEST_ERRNO == test->exp_errno) {
		tst_resm(TPASS, "swapon(2) expected failure;"
			 " Got errno - %s : %s",
			 test->exp_errval, test->err_desc);
		return;
	}

	if (fs_type == TST_BTRFS_MAGIC && errno == EINVAL) {
		tst_resm(TCONF, "Swapfile on BTRFS not implemeted");
			return;
	}

	tst_resm(TFAIL, "swapon(2) failed to produce expected error:"
	         " %d, errno: %s and got %d.", test->exp_errno,
	         test->exp_errval, TEST_ERRNO);
}

int main(int ac, char **av)
{
	int lc, i;
	const char *msg;

	if ((msg = parse_opts(ac, av, NULL, NULL)) != NULL)
		tst_brkm(TBROK, NULL, "OPTION PARSING ERROR - %s", msg);

	setup();

	for (lc = 0; TEST_LOOPING(lc); lc++) {
		tst_count = 0;
		for (i = 0; i < TST_TOTAL; i++)
			verify_swapon(testcases + i);
	}

	cleanup();
	tst_exit();
}

static void setup01(void)
{
	SAFE_SETEUID(cleanup, nobody_uid);
}

static void cleanup01(void)
{
	SAFE_SETEUID(cleanup, 0);
}

static void setup(void)
{
	struct passwd *nobody;

	tst_sig(FORK, DEF_HANDLER, cleanup);

	tst_require_root(NULL);

	nobody = SAFE_GETPWNAM(cleanup, "nobody");
	nobody_uid = nobody->pw_uid;

	tst_tmpdir();

	switch ((fs_type = tst_fs_type(cleanup, "."))) {
	case TST_NFS_MAGIC:
	case TST_TMPFS_MAGIC:
		tst_brkm(TCONF, cleanup,
			 "Cannot do swapon on a file on %s filesystem",
			 tst_fs_type_name(fs_type));
	break;
	}

	SAFE_TOUCH(cleanup, "notswap", 0777, NULL);
	make_swapfile(cleanup, "swapfile01");
	make_swapfile(cleanup, "alreadyused");

	if (ltp_syscall(__NR_swapon, "alreadyused", 0)) {
		if (fs_type != TST_BTRFS_MAGIC || errno != EINVAL)
			tst_resm(TWARN | TERRNO, "swapon(alreadyused) failed");
	} else {
		do_swapoff = 1;
	}

	TEST_PAUSE;
}

void cleanup(void)
{
	if (do_swapoff && ltp_syscall(__NR_swapoff, "alreadyused"))
		tst_resm(TWARN | TERRNO, "swapoff(alreadyused) failed");

	tst_rmdir();
}