minix3/tests/ipc/lib/tst_tmpdir.c

391 lines
11 KiB
C

/*
* 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., 59
* Temple Place - Suite 330, Boston MA 02111-1307, 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().
*
* Setting the env variable "TDIRECTORY" will override the creation
* of a new temp dir. The directory specified by TDIRECTORY will
* be used as the temporary directory, and no removal will be done
* in tst_rmdir().
*
* RETURN VALUE
* Neither tst_tmpdir() or tst_rmdir() has a return value.
*
*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#*#**/
#include <stdio.h>
#include <errno.h>
#include <stdlib.h> /* for getenv() */
#include <string.h> /* for string functions */
#include <unistd.h> /* for sysconf(), getcwd(), rmdir() */
#include <sys/types.h> /* for mkdir() */
#include <sys/stat.h> /* for mkdir() */
#include "test.h"
#include "rmobj.h"
/*
* Define some useful macros.
*/
#define PREFIX_SIZE 4
#define STRING_SIZE 256
#define DIR_MODE 0777 /* mode of tmp dir that will be created */
#ifndef PATH_MAX
#ifdef MAXPATHLEN
#define PATH_MAX MAXPATHLEN
#else
#define PATH_MAX 1024
#endif
#endif
/*
* Define function prototypes.
*/
static void tmpdir_cleanup(void);
/*
* Define global variables.
*/
extern char *TCID; /* defined/initialized in main() */
extern int TST_TOTAL; /* defined/initialized in main() */
extern char *TESTDIR; /* the directory created; defined in */
/* tst_res.c */
/*
* tst_tmpdir() - Create a unique temporary directory and chdir() to it.
* It expects the caller to have defined/initialized the
* TCID/TST_TOTAL global variables. The TESTDIR global
* variable will be set to the directory that gets used
* as the testing directory.
*
* NOTE: This function must be called BEFORE any activity
* that would require CLEANUP. If tst_tmpdir() fails, it
* cleans up afer itself and calls tst_exit() (i.e. does
* not return).
*/
#undef FN_NAME
#define FN_NAME "tst_tmpdir()"
void
tst_tmpdir()
{
char template[PATH_MAX]; /* template for mkstemp, mkdtemp */
int no_cleanup = 0; /* !0 means TDIRECTORY env var was set */
char *env_tmpdir; /* temporary storage for TMPDIR env var */
/* This is an AWEFUL hack to figure out if mkdtemp() is available */
#if defined(__GLIBC_PREREQ)
# if __GLIBC_PREREQ(2,2)
# define HAVE_MKDTEMP 1
# else
# define HAVE_MKDTEMP 0
int tfd;
# endif
#else
# define HAVE_MKDTEMP 0
int tfd;
#endif
/*
* If the TDIRECTORY env variable is not set, a temp dir will be
* created.
*/
if ((TESTDIR = getenv(TDIRECTORY))) {
/*
* The TDIRECTORY env. variable is set, so no temp dir is created.
* Also, no clean up will be done via tst_rmdir().
*/
no_cleanup++;
#if UNIT_TEST
printf("TDIRECTORY env var is set\n");
#endif
} else {
/*
* Create a template for the temporary directory. Use the
* environment variable TMPDIR if it is available, otherwise
* use our default TEMPDIR.
*/
if ((env_tmpdir = getenv("TMPDIR"))) {
snprintf(template, PATH_MAX, "%s/%.3sXXXXXX", env_tmpdir, TCID);
} else {
snprintf(template, PATH_MAX, "%s/%.3sXXXXXX", TEMPDIR, TCID);
}
#if HAVE_MKDTEMP
/*
* Make the temporary directory in one shot using mkdtemp()
*/
if (mkdtemp(template) == NULL)
tst_brkm(TBROK, tmpdir_cleanup,
"%s: mkdtemp(%s) failed; errno = %d: %s",
FN_NAME, template, errno, strerror(errno));
TESTDIR = strdup(template);
#else
/*
* Make the template name, then the directory
*/
if ((tfd = mkstemp(template)) == -1)
tst_brkm(TBROK, tmpdir_cleanup,
"%s: mkstemp(%s) failed; errno = %d: %s",
FN_NAME, template, errno, strerror(errno));
close(tfd);
unlink(template);
TESTDIR = strdup(template);
if (mkdir(TESTDIR, DIR_MODE)) {
/* If we start failing with EEXIST, wrap this section in
* a loop so we can try again.
*/
tst_brkm(TBROK, tmpdir_cleanup,
"%s: mkdir(%s, %#o) failed; errno = %d: %s",
FN_NAME, TESTDIR, DIR_MODE, errno,
strerror(errno));
}
#endif
}
/*
* Change the group on this temporary directory to be that of the
* gid of the person running the tests and permissions to 777.
*/
if ( chown(TESTDIR, -1, getgid()) == -1 )
tst_brkm(TBROK, tmpdir_cleanup,
"chown(%s, -1, %d) failed; errno = %d: %s",
TESTDIR, getgid(), errno, strerror(errno));
if ( chmod(TESTDIR,S_IRWXU | S_IRWXG | S_IRWXO) == -1 )
tst_brkm(TBROK, tmpdir_cleanup,
"chmod(%s,777) failed; errno %d: %s",
TESTDIR, errno, strerror(errno));
#if UNIT_TEST
printf("TESTDIR = %s\n", TESTDIR);
#endif
/*
* 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_brkm(TBROK, NULL, "%s: chdir(%s) failed; errno = %d: %s",
FN_NAME, TESTDIR, errno, strerror(errno) );
/* Try to remove the directory */
if ( !no_cleanup && rmdir(TESTDIR) == -1 )
tst_resm(TWARN, "%s: rmdir(%s) failed; errno = %d: %s",
FN_NAME, TESTDIR, errno, strerror(errno) );
tmpdir_cleanup();
}
#if UNIT_TEST
printf("CWD is %s\n", getcwd((char *)NULL, PATH_MAX));
#endif
/*
* If we made through all this stuff, return.
*/
return;
} /* tst_tmpdir() */
/*
*
* tst_rmdir() - Recursively remove the temporary directory created by
* tst_tmpdir(). This function is intended ONLY as a
* companion to tst_tmpdir(). If the TDIRECTORY
* environment variable is set, no cleanup will be
* attempted.
*/
#undef FN_NAME
#define FN_NAME "tst_rmdir()"
void
tst_rmdir()
{
char *errmsg;
char *tdirectory;
char current_dir[PATH_MAX]; /* current working directory */
char parent_dir[PATH_MAX]; /* directory above TESTDIR */
char *basename; /* basename of the TESTDIR */
/*
* If the TDIRECTORY env variable is set, this indicates that no
* temp dir was created by tst_tmpdir(). Thus no cleanup will be
* necessary.
*/
if ( (tdirectory = getenv(TDIRECTORY)) != NULL ) {
#if UNIT_TEST
printf("\"TDIRECORY\" env variable is set; no cleanup was performed\n");
#endif
return;
}
/*
* Check that TESTDIR is not NULL.
*/
if ( TESTDIR == NULL ) {
tst_resm(TWARN, "%s: TESTDIR was NULL; no removal attempted",
FN_NAME);
return;
}
/*
* Check that the value of TESTDIR is not "*" or "/". These could
* have disastrous effects in a test run by root.
*/
if ( strcmp(TESTDIR, "/") == 0 ) {
tst_resm(TWARN,
"%s: Recursive remove of root directory not attempted",
FN_NAME);
return;
}
if ( strchr(TESTDIR, '*') != NULL ) {
tst_resm(TWARN, "%s: Recursive remove of '*' not attempted",
FN_NAME);
return;
}
/*
* Get the directory name of TESTDIR. If TESTDIR is a relative path,
* get full path.
*/
if ( TESTDIR[0] != '/' ) {
if ( getcwd(current_dir,PATH_MAX) == NULL )
strcpy(parent_dir, TESTDIR);
else
sprintf(parent_dir, "%s/%s", current_dir, TESTDIR);
} else {
strcpy(parent_dir, TESTDIR);
}
if ( (basename = strrchr(parent_dir, '/')) != NULL ) {
*basename='\0'; /* terminate at end of parent_dir */
}
/*
* Change directory to parent_dir (The dir above TESTDIR).
*/
if ( chdir(parent_dir) != 0 )
tst_resm(TWARN,
"%s: chdir(%s) failed; errno = %d: %s\nAttempting to remove temp dir anyway",
FN_NAME, parent_dir, errno, strerror(errno));
/*
* Attempt to remove the "TESTDIR" directory, using rmobj().
*/
if ( rmobj(TESTDIR, &errmsg) == -1 )
tst_resm(TWARN, "%s: rmobj(%s) failed: %s",
FN_NAME, TESTDIR, errmsg);
return;
} /* tst_rmdir() */
/*
* tmpdir_cleanup() - This function is used when tst_tmpdir()
* encounters an error, and must cleanup and exit.
* It prints a warning message via tst_resm(), and
* then calls tst_exit().
*/
#undef FN_NAME
#define FN_NAME "tst_tmpdir()"
static void
tmpdir_cleanup()
{
/*
* Print a warning message and call tst_exit() to exit the test.
*/
tst_resm(TWARN, "%s: No user cleanup function called before exiting",
FN_NAME);
tst_exit();
} /* tmpdir_cleanup() */
#ifdef UNIT_TEST
/****************************************************************************
* Unit test code: Takes input from stdin and can make the following
* calls: tst_tmpdir(), tst_rmdir().
****************************************************************************/
int TST_TOTAL = 10;
char *TCID = "TESTTCID";
main()
{
int option;
char *chrptr;
printf("UNIT TEST of tst_tmpdir.c. Options to try:\n\
-1 : call tst_exit()\n\
0 : call tst_tmpdir()\n\
1 : call tst_rmdir()\n\n");
while ( 1 ) {
printf("Enter options (-1, 0, 1): ");
(void) scanf("%d%c", &option, &chrptr);
switch ( option ) {
case -1:
tst_exit();
break;
case 0:
tst_tmpdir();
break;
case 1:
tst_rmdir();
break;
} /* switch() */
} /* while() */
}
#endif /* UNIT_TEST */