260 строки
6.4 KiB
C
260 строки
6.4 KiB
C
/*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <libgen.h>
|
|
#include <sys/param.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include "ompi_config.h"
|
|
#include "include/constants.h"
|
|
#include "util/sys_info.h"
|
|
#include "util/os_path.h"
|
|
#include "util/session_dir.h"
|
|
#include "support.h"
|
|
|
|
struct ompi_proc_info_t {
|
|
bool init; /**< Certifies that values have been filled.
|
|
* Certifies that the ompi_sys_info() function has been
|
|
* called at least once so fields have valid values
|
|
*/
|
|
char *universe_session_dir; /**< Location of universe temp dir.
|
|
* The session directory has the form
|
|
* <prefix><openmpi-sessions-user><universe>, where the prefix
|
|
* can either be provided by the user via the
|
|
* --tmpdir command-line flag, the use of one of several
|
|
* environmental variables, or else a default location.
|
|
*/
|
|
|
|
char *job_session_dir; /**< Session directory for job */
|
|
|
|
char *proc_session_dir; /**< Session directory for the process */
|
|
};
|
|
typedef struct ompi_proc_info_t ompi_proc_info_t;
|
|
|
|
ompi_proc_info_t ompi_process_info;
|
|
|
|
static bool test1(void); /* given prefix, both one that works and one that fails */
|
|
static bool test2(void); /* no prefix given, OMPI_PREFIX_ENV set, one good and one bad */
|
|
static bool test3(void); /* no prefix given, TMPDIR set, one good and one bad */
|
|
static bool test4(void); /* no prefix given, TMP set, one good and one bad */
|
|
static bool test5(void); /* no prefix given, HOME set, one good and one bad */
|
|
static bool test6(void); /* no prefix given, nothing set, one good and one bad */
|
|
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
ompi_sys_info(); /* initialize system */
|
|
|
|
test_init("ompi_session_dir_t");
|
|
|
|
|
|
if (test1()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("ompi_session_dir_t test1 failed");
|
|
}
|
|
|
|
if (test2()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("ompi_session_dir_t test2 failed");
|
|
}
|
|
|
|
if (test3()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("ompi_session_dir_t test3 failed");
|
|
}
|
|
|
|
if (test4()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("ompi_session_dir_t test4 failed");
|
|
}
|
|
|
|
if (test5()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("ompi_session_dir_t test5 failed");
|
|
}
|
|
|
|
if (test6()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("ompi_session_dir_t test6 failed");
|
|
}
|
|
|
|
test_finalize();
|
|
return 0;
|
|
}
|
|
|
|
|
|
static bool test1(void)
|
|
{
|
|
/* Test proper action when given a prefix */
|
|
|
|
char *prefix, *tmp, *tmp2;
|
|
|
|
/* see if we can create a specified path */
|
|
|
|
prefix = ompi_os_path(false, "tmp", NULL);
|
|
if (OMPI_ERROR == ompi_session_dir(true, prefix, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
printf("test1 - couldn't create specified path\n");
|
|
free(prefix);
|
|
return(false);
|
|
}
|
|
/* see if it can access an existing path */
|
|
|
|
if (OMPI_ERROR == ompi_session_dir(false, prefix, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
printf("test1 - couldn't access existing path\n");
|
|
free(prefix);
|
|
return(false);
|
|
}
|
|
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
free(ompi_process_info.universe_session_dir);
|
|
free(prefix);
|
|
free(tmp);
|
|
|
|
/* check what happens when given prefix that won't allow access */
|
|
tmp2 = ompi_os_path(false, "test", NULL); /* assume we don't have root privileges */
|
|
if (OMPI_SUCCESS == ompi_session_dir(true, tmp2, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
printf("created temp directory in %s - shouldn't have been able to do so\n", tmp2);
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
free(tmp);
|
|
free(tmp2);
|
|
return(false);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
static bool test2(void)
|
|
{
|
|
char *tmp;
|
|
|
|
/* use the OMPI_PREFIX_ENV variable */
|
|
|
|
setenv("OMPI_PREFIX_ENV", "/tmp/trythis", 1);
|
|
|
|
if (OMPI_ERROR == ompi_session_dir(true, NULL, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("OMPI_PREFIX_ENV");
|
|
return(false);
|
|
}
|
|
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
free(tmp);
|
|
|
|
unsetenv("OMPI_PREFIX_ENV");
|
|
|
|
return(true);
|
|
|
|
}
|
|
|
|
|
|
static bool test3(void)
|
|
{
|
|
/* use the TMPDIR enviro variable */
|
|
char *tmp;
|
|
|
|
setenv("TMPDIR", "/tmp/trythis", 1);
|
|
|
|
if (OMPI_ERROR == ompi_session_dir(true, NULL, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("TMPDIR");
|
|
return(false);
|
|
}
|
|
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
free(tmp);
|
|
|
|
unsetenv("TMPDIR");
|
|
|
|
return(true);
|
|
}
|
|
|
|
|
|
static bool test4(void)
|
|
{
|
|
/* use the TMP enviro variable */
|
|
char *tmp;
|
|
|
|
setenv("TMP", "/tmp/trythis", 1);
|
|
|
|
if (OMPI_ERROR == ompi_session_dir(true, NULL, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("TMP");
|
|
return(false);
|
|
}
|
|
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
free(tmp);
|
|
|
|
unsetenv("TMP");
|
|
|
|
return(true);
|
|
}
|
|
|
|
|
|
static bool test5(void)
|
|
{
|
|
/* use the HOME enviro variable */
|
|
char *tmp;
|
|
|
|
setenv("HOME", "/tmp/trythis", 1);
|
|
|
|
if (OMPI_ERROR == ompi_session_dir(true, NULL, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("HOME");
|
|
return(false);
|
|
}
|
|
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
free(tmp);
|
|
|
|
unsetenv("HOME");
|
|
|
|
return(true);
|
|
}
|
|
|
|
|
|
static bool test6(void)
|
|
{
|
|
char *tmp;
|
|
|
|
/* no enviro variables set, no prefix given
|
|
* Program should turn to default of /tmp (where "/" is whatever
|
|
* top-level directory is appropriate for given system)
|
|
*/
|
|
if (OMPI_ERROR == ompi_session_dir(true, NULL, ompi_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
return(false);
|
|
}
|
|
|
|
rmdir(ompi_process_info.universe_session_dir);
|
|
tmp = strdup(dirname(ompi_process_info.universe_session_dir));
|
|
rmdir(tmp);
|
|
|
|
return(true);
|
|
}
|