/*      $OpenBSD: lockf.c,v 1.1 2018/11/06 18:11:11 anton Exp $ */

/*
 * Copyright (c) 2018 Anton Lindqvist <anton@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <err.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "util.h"

int verbose = 0;

/*
 * Test 1 - F_LOCK positive length
 */
static int
test1(int fd)
{
        struct flock fl;
        pid_t pid;
        int res;

        fl.l_start = 0;
        fl.l_len = 16;
        fl.l_type = F_WRLCK;
        fl.l_whence = SEEK_CUR;
        res = fcntl(fd, F_SETLK, &fl);
        FAIL(res != 0);

        pid = fork();
        if (pid == -1)
                err(1, "fork");
        if (pid == 0) {
                res = fcntl(fd, F_SETLK, &fl);
                FAIL(res != -1);
                FAIL(fl.l_type != F_WRLCK);
                _exit(0);
        }
        res = safe_waitpid(pid);
        FAIL(res != 0);

        SUCCEED;
}

/*
 * Test 2 - F_LOCK negative length
 */
static int
test2(int fd)
{
        struct flock fl;
        pid_t pid;
        off_t len = 16;
        int res;

        if (lseek(fd, len, SEEK_SET) == -1)
                err(1, "lseek");

        fl.l_start = 0;
        fl.l_len = -len;
        fl.l_type = F_WRLCK;
        fl.l_whence = SEEK_CUR;
        res = fcntl(fd, F_SETLK, &fl);
        FAIL(res != 0);

        pid = fork();
        if (pid == -1)
                err(1, "fork");
        if (pid == 0) {
                if (lseek(fd, 0, SEEK_SET) == -1)
                        err(1, "lseek");

                fl.l_start = 0;
                fl.l_len = len;
                fl.l_whence = SEEK_SET;
                res = fcntl(fd, F_GETLK, &fl);
                FAIL(res != 0);
                FAIL(fl.l_type != F_WRLCK);
                _exit(0);
        }
        res = safe_waitpid(pid);
        FAIL(res != 0);

        SUCCEED;
}

static struct test tests[] = {
        {       test1,          0       },
        {       test2,          0       },
};

static int test_count = sizeof(tests) / sizeof(tests[0]);

int
main(int argc, char *argv[])
{
        const char *errstr;
        int c, error, fd, i;
        int testnum = 0;

        while ((c = getopt(argc, argv, "v")) != -1)
                switch (c) {
                case 'v':
                        verbose = 1;
                        break;
                default:
                        usage();
                }
        argc -= optind;
        argv += optind;
        if (argc > 1)
                usage();
        if (argc == 1) {
                testnum = strtonum(argv[0], 1, test_count, &errstr);
                if (testnum == 0)
                        errx(1, "test number %s", errstr);
        }

        fd = make_file(1024);

        error = 0;
        for (i = 0; i < test_count; i++) {
                if (testnum == 0 || i + 1 == testnum)
                        error |= tests[i].testfn(fd);
        }

        return (error ? 1 : 0);
}