/*
 * Copyright (c) 2020 Sine Nomine Associates. 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 `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 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.
 */

/*!
 * Common file-related functions for test programs
 */

#include <afsconfig.h>
#include <afs/param.h>
#include <roken.h>

#include <afs/opr.h>
#include <afs/afsutil.h>

#include <tests/tap/basic.h>

#include "common.h"

char *
afstest_mkdtemp(void)
{
    char *template;
    char *tmp;

    tmp = afstest_obj_path("tests/tmp");

    /* Try to make sure 'tmp' exists. */
    (void)mkdir(tmp, 0700);

    template = afstest_asprintf("%s/afs_XXXXXX", tmp);

    free(tmp);

#if defined(HAVE_MKDTEMP)
    return mkdtemp(template);
#else
    /*
     * Note that using the following is not a robust replacement
     * for mkdtemp as there is a possible race condition between
     * creating the name and creating the directory itself.  The
     * use of this routine is limited to running tests.
     */
    if (mktemp(template) == NULL)
	return NULL;
    if (mkdir(template, 0700))
	return NULL;
    return template;
#endif
}

void
afstest_rmdtemp(char *path)
{
    int code;
    struct stat st;

    /* Sanity check, only zap directories that look like ours */
    opr_Assert(strstr(path, "afs_") != NULL);
    if (getenv("MAKECHECK") == NULL) {
	/* Don't delete tmp dirs if we're not running under 'make check'. */
	return;
    }
    code = lstat(path, &st);
    if (code != 0) {
	/* Given path doesn't exist (or we can't access it?) */
	return;
    }
    if (!S_ISDIR(st.st_mode)) {
	/* Path isn't a dir; that's weird. Bail out to be on the safe side. */
	return;
    }
    afstest_systemlp("rm", "-rf",
#if defined(AFS_LINUX_ENV)
		     "--one-file-system",
#endif
		     path, (char*)NULL);
}

static char *
path_from_tdir(char *env_var, char *filename)
{
    char *tdir;
    char *path;
    char *top_rel, *top_abs;

    /* C_TAP_SOURCE/C_TAP_BUILD in the env points to 'tests/' in the
     * srcdir/objdir. */

    tdir = getenv(env_var);
    if (tdir == NULL) {
	/*
	 * If C_TAP_SOURCE/C_TAP_BUILD isn't set, we assume we're running from
	 * the same cwd as one of the test programs (e.g. 'tests/foo/'). So to
	 * get to 'tests/', just go up one level.
	 */
	tdir = "..";
    }

    /* 'tdir' points to the 'tests' dir, so go up one level to get to the top
     * srcdir/objdir. */
    top_rel = afstest_asprintf("%s/..", tdir);
    top_abs = bcalloc(1, PATH_MAX);
    top_abs = realpath(top_rel, top_abs);
    if (top_abs == NULL) {
	sysbail("realpath");
    }

    free(top_rel);

    /*
     * The given 'filename' is relative to the top srcdir/objdir, so to get the
     * full path, append 'filename' to the top srcdir/objdir. Note that the
     * given path may not exist yet, so we cannot run the full path through
     * realpath().
     */
    path = afstest_asprintf("%s/%s", top_abs, filename);
    free(top_abs);

    return path;
}

char *
afstest_src_path(char *path)
{
    return path_from_tdir("C_TAP_SOURCE", path);
}

char *
afstest_obj_path(char *path)
{
    return path_from_tdir("C_TAP_BUILD", path);
}

/**
 * Check if the given file contains a string. To keep things simple, the string
 * to look for must appear on a single line, and must appear in the first 128
 * bytes of that line.
 *
 * @param[in] path  Path to the file to check.
 * @param[in] target	The string to look for in 'path'.
 *
 * @retval 1 The file contains the given string.
 * @retval 0 The file does not contain the given string (or we encountered an
 *           error).
 */
int
afstest_file_contains(char *path, char *target)
{
    FILE *fh;
    char line[128];
    int found = 0;

    fh = fopen(path, "r");
    if (fh == NULL) {
	diag("%s: Failed to open %s", __func__, path);
	goto done;
    }

    while (fgets(line, sizeof(line), fh) != NULL) {
	if (strstr(line, target) != NULL) {
	    found = 1;
	    goto done;
	}
    }

 done:
    if (fh != NULL) {
	fclose(fh);
    }
    return found;
}