You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
333 lines
8.1 KiB
333 lines
8.1 KiB
/**********************************************************
|
|
* Copyright (c) 2000 Silicon Graphics, Inc. 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.
|
|
*
|
|
* Further, this software is distributed without any warranty that it is
|
|
* free of the rightful claim of any third person regarding infringement
|
|
* or the like. Any license provided herein, whether implied or
|
|
* otherwise, applies only to this software file. Patent licenses, if
|
|
* any, provided herein do not apply to combinations of this program with
|
|
* other software, or any other product whatsoever.
|
|
*
|
|
* 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.
|
|
*
|
|
* Contact information: Silicon Graphics, Inc., 1600 Amphitheatre Pkwy,
|
|
* Mountain View, CA 94043, or:
|
|
*
|
|
* http://www.sgi.com
|
|
*
|
|
* For further information regarding this notice, see:
|
|
*
|
|
* http://oss.sgi.com/projects/GenInfo/NoticeExplan/
|
|
*********************************************************/
|
|
|
|
/**********************************************************
|
|
*
|
|
* OS Testing - Silicon Graphics, Inc.
|
|
*
|
|
* FUNCTION NAME : tst_tmpdir, tst_rmdir
|
|
*
|
|
* FUNCTION TITLE : Create/remove a testing temp dir
|
|
*
|
|
* SYNOPSIS:
|
|
* void tst_tmpdir();
|
|
* void tst_rmdir();
|
|
*
|
|
* AUTHOR : Dave Fenner
|
|
*
|
|
* INITIAL RELEASE : UNICOS 8.0
|
|
*
|
|
* DESCRIPTION
|
|
* tst_tmpdir() is used to create a unique, temporary testing
|
|
* directory, and make it the current working directory.
|
|
* tst_rmdir() is used to remove the directory created by
|
|
* tst_tmpdir().
|
|
*
|
|
* RETURN VALUE
|
|
* Neither tst_tmpdir() or tst_rmdir() has a return value.
|
|
*
|
|
*********************************************************/
|
|
#define _GNU_SOURCE
|
|
#include <sys/mman.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <libgen.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
|
|
#include "test.h"
|
|
#include "safe_macros.h"
|
|
#include "ltp_priv.h"
|
|
#include "lapi/futex.h"
|
|
|
|
/*
|
|
* Define some useful macros.
|
|
*/
|
|
#define DIR_MODE (S_IRWXU|S_IRWXG|S_IRWXO)
|
|
|
|
#ifndef PATH_MAX
|
|
#ifdef MAXPATHLEN
|
|
#define PATH_MAX MAXPATHLEN
|
|
#else
|
|
#define PATH_MAX 1024
|
|
#endif
|
|
#endif
|
|
|
|
/*
|
|
* Define global variables.
|
|
*/
|
|
extern char *TCID; /* defined/initialized in main() */
|
|
static char *TESTDIR = NULL; /* the directory created */
|
|
|
|
static char test_start_work_dir[PATH_MAX];
|
|
|
|
/* lib/tst_checkpoint.c */
|
|
extern futex_t *tst_futexes;
|
|
|
|
int tst_tmpdir_created(void)
|
|
{
|
|
return TESTDIR != NULL;
|
|
}
|
|
|
|
char *tst_get_tmpdir(void)
|
|
{
|
|
if (TESTDIR == NULL) {
|
|
tst_brkm(TBROK, NULL, "you must call tst_tmpdir() first");
|
|
return NULL;
|
|
}
|
|
|
|
return strdup(TESTDIR);
|
|
}
|
|
|
|
const char *tst_get_startwd(void)
|
|
{
|
|
return test_start_work_dir;
|
|
}
|
|
|
|
static int rmobj(char *obj, char **errmsg)
|
|
{
|
|
int ret_val = 0;
|
|
DIR *dir;
|
|
struct dirent *dir_ent;
|
|
char dirobj[PATH_MAX];
|
|
struct stat statbuf;
|
|
static char err_msg[1024];
|
|
int fd;
|
|
|
|
fd = open(obj, O_DIRECTORY | O_NOFOLLOW);
|
|
if (fd != -1) {
|
|
close(fd);
|
|
|
|
/* Do NOT perform the request if the directory is "/" */
|
|
if (!strcmp(obj, "/")) {
|
|
if (errmsg != NULL) {
|
|
sprintf(err_msg, "Cannot remove /");
|
|
*errmsg = err_msg;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
/* Open the directory to get access to what is in it */
|
|
if ((dir = opendir(obj)) == NULL) {
|
|
if (rmdir(obj) != 0) {
|
|
if (errmsg != NULL) {
|
|
sprintf(err_msg,
|
|
"rmdir(%s) failed; errno=%d: %s",
|
|
obj, errno, tst_strerrno(errno));
|
|
*errmsg = err_msg;
|
|
}
|
|
return -1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
/* Loop through the entries in the directory, removing each one */
|
|
for (dir_ent = (struct dirent *)readdir(dir);
|
|
dir_ent != NULL; dir_ent = (struct dirent *)readdir(dir)) {
|
|
|
|
/* Don't remove "." or ".." */
|
|
if (!strcmp(dir_ent->d_name, ".")
|
|
|| !strcmp(dir_ent->d_name, ".."))
|
|
continue;
|
|
|
|
/* Recursively call this routine to remove the current entry */
|
|
sprintf(dirobj, "%s/%s", obj, dir_ent->d_name);
|
|
if (rmobj(dirobj, errmsg) != 0)
|
|
ret_val = -1;
|
|
}
|
|
|
|
closedir(dir);
|
|
|
|
/* If there were problems removing an entry, don't attempt to
|
|
remove the directory itself */
|
|
if (ret_val == -1)
|
|
return -1;
|
|
|
|
/* Get the link count, now that all the entries have been removed */
|
|
if (lstat(obj, &statbuf) < 0) {
|
|
if (errmsg != NULL) {
|
|
sprintf(err_msg,
|
|
"lstat(%s) failed; errno=%d: %s", obj,
|
|
errno, tst_strerrno(errno));
|
|
*errmsg = err_msg;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
/* Remove the directory itself */
|
|
if (statbuf.st_nlink >= 3) {
|
|
/* The directory is linked; unlink() must be used */
|
|
if (unlink(obj) < 0) {
|
|
if (errmsg != NULL) {
|
|
sprintf(err_msg,
|
|
"unlink(%s) failed; errno=%d: %s",
|
|
obj, errno, tst_strerrno(errno));
|
|
*errmsg = err_msg;
|
|
}
|
|
return -1;
|
|
}
|
|
} else {
|
|
/* The directory is not linked; remove() can be used */
|
|
if (remove(obj) < 0) {
|
|
if (errmsg != NULL) {
|
|
sprintf(err_msg,
|
|
"remove(%s) failed; errno=%d: %s",
|
|
obj, errno, tst_strerrno(errno));
|
|
*errmsg = err_msg;
|
|
}
|
|
return -1;
|
|
}
|
|
}
|
|
} else {
|
|
if (unlink(obj) < 0) {
|
|
if (errmsg != NULL) {
|
|
sprintf(err_msg,
|
|
"unlink(%s) failed; errno=%d: %s", obj,
|
|
errno, tst_strerrno(errno));
|
|
*errmsg = err_msg;
|
|
}
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void tst_tmpdir(void)
|
|
{
|
|
char template[PATH_MAX];
|
|
char *env_tmpdir;
|
|
char *errmsg, *c;
|
|
|
|
/*
|
|
* Create a template for the temporary directory. Use the
|
|
* environment variable TMPDIR if it is available, otherwise
|
|
* use our default TEMPDIR.
|
|
*/
|
|
env_tmpdir = getenv("TMPDIR");
|
|
if (env_tmpdir) {
|
|
c = strchr(env_tmpdir, '/');
|
|
/*
|
|
* Now we force environment variable TMPDIR to be an absolute
|
|
* pathname, which dose not make much sense, but it will
|
|
* greatly simplify code in tst_rmdir().
|
|
*/
|
|
if (c != env_tmpdir) {
|
|
tst_brkm(TBROK, NULL, "You must specify an absolute "
|
|
"pathname for environment variable TMPDIR");
|
|
return;
|
|
}
|
|
snprintf(template, PATH_MAX, "%s/%.3sXXXXXX", env_tmpdir, TCID);
|
|
} else {
|
|
snprintf(template, PATH_MAX, "%s/%.3sXXXXXX", TEMPDIR, TCID);
|
|
}
|
|
|
|
/* Make the temporary directory in one shot using mkdtemp. */
|
|
if (mkdtemp(template) == NULL) {
|
|
tst_brkm(TBROK | TERRNO, NULL,
|
|
"%s: mkdtemp(%s) failed", __func__, template);
|
|
return;
|
|
}
|
|
|
|
if ((TESTDIR = strdup(template)) == NULL) {
|
|
tst_brkm(TBROK | TERRNO, NULL,
|
|
"%s: strdup(%s) failed", __func__, template);
|
|
return;
|
|
}
|
|
|
|
SAFE_CHOWN(NULL, TESTDIR, -1, getgid());
|
|
|
|
SAFE_CHMOD(NULL, TESTDIR, DIR_MODE);
|
|
|
|
if (getcwd(test_start_work_dir, sizeof(test_start_work_dir)) == NULL) {
|
|
tst_resm(TINFO, "Failed to record test working dir");
|
|
test_start_work_dir[0] = '\0';
|
|
}
|
|
|
|
/*
|
|
* Change to the temporary directory. If the chdir() fails, issue
|
|
* TBROK messages for all test cases, attempt to remove the
|
|
* directory (if it was created), and exit. If the removal also
|
|
* fails, also issue a TWARN message.
|
|
*/
|
|
if (chdir(TESTDIR) == -1) {
|
|
tst_resm(TERRNO, "%s: chdir(%s) failed", __func__, TESTDIR);
|
|
|
|
/* Try to remove the directory */
|
|
if (rmobj(TESTDIR, &errmsg) == -1) {
|
|
tst_resm(TWARN, "%s: rmobj(%s) failed: %s",
|
|
__func__, TESTDIR, errmsg);
|
|
}
|
|
|
|
tst_exit();
|
|
}
|
|
}
|
|
|
|
void tst_rmdir(void)
|
|
{
|
|
char *errmsg;
|
|
|
|
/*
|
|
* Check that TESTDIR is not NULL.
|
|
*/
|
|
if (TESTDIR == NULL) {
|
|
tst_resm(TWARN,
|
|
"%s: TESTDIR was NULL; no removal attempted",
|
|
__func__);
|
|
return;
|
|
}
|
|
|
|
/*
|
|
* Unmap the backend file.
|
|
* This is needed to overcome the NFS "silly rename" feature.
|
|
*/
|
|
if (tst_futexes) {
|
|
msync((void *)tst_futexes, getpagesize(), MS_SYNC);
|
|
munmap((void *)tst_futexes, getpagesize());
|
|
}
|
|
|
|
/*
|
|
* Attempt to remove the "TESTDIR" directory, using rmobj().
|
|
*/
|
|
if (rmobj(TESTDIR, &errmsg) == -1) {
|
|
tst_resm(TWARN, "%s: rmobj(%s) failed: %s",
|
|
__func__, TESTDIR, errmsg);
|
|
}
|
|
}
|