#include <sys/stat.h>
#include <sys/wait.h>

#include <err.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>

#define UV_SHOULD_SUCCEED(A, B) do {                                    \
        if (A) {                                                        \
                err(1, "%s:%d - %s", __FILE__, __LINE__, B);            \
        }                                                               \
} while (0)

#define UV_SHOULD_ENOENT(A, B) do {                                     \
        if (A) {                                                        \
                if (do_uv && errno != ENOENT)                           \
                        err(1, "%s:%d - %s", __FILE__, __LINE__, B);    \
        } else {                                                        \
                if (do_uv)                                              \
                        errx(1, "%s:%d - %s worked when it should not " \
                            "have",  __FILE__, __LINE__, B);            \
        }                                                               \
} while(0)

#define UV_SHOULD_EACCES(A, B) do {                                     \
        if (A) {                                                        \
                if (do_uv && errno != EACCES)                           \
                        err(1, "%s:%d - %s", __FILE__, __LINE__, B);    \
        } else {                                                        \
                if (do_uv)                                              \
                        errx(1, "%s:%d - %s worked when it should not " \
                            "have",  __FILE__, __LINE__, B);            \
        }                                                               \
} while(0)

#define UV_SHOULD_EPERM(A, B) do {                                      \
        if (A) {                                                        \
                if (do_uv && errno != EPERM)                            \
                        err(1, "%s:%d - %s", __FILE__, __LINE__, B);    \
        } else {                                                        \
                if (do_uv)                                              \
                        errx(1, "%s:%d - %s worked when it should not " \
                            "have",  __FILE__, __LINE__, B);            \
        }                                                               \
} while(0)

static char uv_dir1[] = "/tmp/uvdir1.XXXXXX"; /* unveiled */
static char uv_dir2[] = "/tmp/uvdir2.XXXXXX"; /* not unveiled */
static char uv_file1[] = "/tmp/uvfile1.XXXXXX"; /* unveiled */
static char uv_file2[] = "/tmp/uvfile2.XXXXXX"; /* not unveiled */

static int
runcompare_internal(int (*func)(int), int fail_ok)
{
        int unveil = 0, nonunveil = 0;
        int status;
        pid_t pid = fork();
        if (pid == 0)
                exit(func(0));
        status = 0;
        waitpid(pid, &status, 0);
        if (WIFEXITED(status))
                nonunveil = WEXITSTATUS(status);
        if (WIFSIGNALED(status)) {
                printf("[FAIL] nonunveil exited with signal %d\n",
                    WTERMSIG(status));
                goto fail;
        }
        pid = fork();
        if (pid == 0)
                exit(func(1));
        status = 0;
        waitpid(pid, &status, 0);
        if (WIFEXITED(status))
                unveil = WEXITSTATUS(status);
        if (WIFSIGNALED(status)) {
                printf("[FAIL] nonunveil exited with signal %d\n",
                    WTERMSIG(status));
                goto fail;
        }
        if (!fail_ok && (unveil || nonunveil)) {
                printf("[FAIL] unveil = %d, nonunveil = %d\n", unveil,
                    nonunveil);
                goto fail;
        }
        if (unveil == nonunveil)
                return 0;
        printf("[FAIL] unveil = %d, nonunveil = %d\n", unveil, nonunveil);
 fail:
        return 1;
}

static int
runcompare(int (*func)(int))
{
        return runcompare_internal(func, 1);
}

static void
test_setup(void)
{
        int fd1, fd2;
        char filename[256];

        UV_SHOULD_SUCCEED((mkdtemp(uv_dir1) == NULL), "mkdtmp");
        UV_SHOULD_SUCCEED((mkdtemp(uv_dir2) == NULL), "mkdtmp");
        UV_SHOULD_SUCCEED(((fd1 = mkstemp(uv_file1)) == -1), "mkstemp");
        close(fd1);
        UV_SHOULD_SUCCEED((chmod(uv_file1, S_IRWXU) == -1), "chmod");
        UV_SHOULD_SUCCEED(((fd2 = mkstemp(uv_file2)) == -1), "mkstemp");
        (void)snprintf(filename, sizeof(filename), "/%s/subdir", uv_dir1);
        UV_SHOULD_SUCCEED((mkdir(filename, 0777) == -1), "mkdir");
        (void)snprintf(filename, sizeof(filename), "/%s/subdir", uv_dir2);
        UV_SHOULD_SUCCEED((mkdir(filename, 0777) == -1), "mkdir");
        close(fd2);
}