bd8b4f7f1e
Roll in the ORTE state machine. Remove last traces of opal_sos. Remove UTK epoch code. Please see the various emails about the state machine change for details. I'll send something out later with more info on the new arch. This commit was SVN r26242.
380 строки
9.2 KiB
C
380 строки
9.2 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2011 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif /* HAVE_UNISTD_H */
|
|
#ifdef HAVE_SYS_PARAM_H
|
|
#include <sys/param.h>
|
|
#endif /* HAVE_SYS_PARAM_H */
|
|
#include <sys/stat.h>
|
|
|
|
#include "support.h"
|
|
#include "orte/constants.h"
|
|
#include "orte/util/proc_info.h"
|
|
#include "opal/util/os_path.h"
|
|
#include "orte/util/session_dir.h"
|
|
#include "orte/util/proc_info.h"
|
|
|
|
|
|
static bool test1(void); /* given prefix, both one that works and one that fails */
|
|
static bool test2(void); /* no prefix given, ORTE_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 */
|
|
static bool test7(void); /* remove session directory tree */
|
|
static bool test8(void); /* attempt to remove tree when subdirs present */
|
|
|
|
void clear_proc_info(void);
|
|
|
|
static FILE *test_out=NULL;
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
orte_proc_info(); /* initialize proc info structure */
|
|
orte_process_info.my_name = (orte_process_name_t*)malloc(sizeof(orte_process_name_t));
|
|
orte_process_info.my_name->cellid = 0;
|
|
orte_process_info.my_name->jobid = 0;
|
|
orte_process_info.my_name->vpid = 0;
|
|
|
|
test_init("orte_session_dir_t");
|
|
test_out = fopen( "test_session_dir_out", "w+" );
|
|
if( test_out == NULL ) {
|
|
test_failure("test_session_dir couldn't open test file failed");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
|
|
fprintf(test_out, "running test1\n");
|
|
if (test1()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test1 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test2\n");
|
|
if (test2()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test2 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test3\n");
|
|
if (test3()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test3 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test4\n");
|
|
if (test4()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test4 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test5\n");
|
|
if (test5()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test5 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test6\n");
|
|
if (test6()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test6 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test7\n");
|
|
if (test7()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test7 failed");
|
|
}
|
|
|
|
fprintf(test_out, "running test8\n");
|
|
if (test8()) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure("orte_session_dir_t test8 failed");
|
|
}
|
|
|
|
fprintf(test_out, "completed all tests\n");
|
|
|
|
fclose(test_out);
|
|
|
|
/* clean up */
|
|
orte_proc_info_finalize();
|
|
|
|
test_finalize();
|
|
return 0;
|
|
}
|
|
|
|
|
|
static bool test1(void)
|
|
{
|
|
/* Test proper action when given a prefix */
|
|
|
|
char *prefix;
|
|
|
|
/* see if we can create a specified path */
|
|
|
|
clear_proc_info();
|
|
|
|
prefix = opal_os_path(false, "tmp", NULL);
|
|
if (ORTE_SUCCESS != orte_session_dir(true, prefix, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
fprintf(test_out, "test1 - couldn't create specified path\n");
|
|
free(prefix);
|
|
return(false);
|
|
}
|
|
/* see if it can access an existing path */
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir(false, prefix, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
fprintf(test_out, "test1 - couldn't access existing path\n");
|
|
free(prefix);
|
|
return(false);
|
|
}
|
|
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
free(orte_process_info.universe_session_dir);
|
|
free(prefix);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
static bool test2(void)
|
|
{
|
|
clear_proc_info();
|
|
|
|
/* use the ORTE_PREFIX_ENV variable */
|
|
|
|
setenv("OMPI_PREFIX_ENV", "/tmp/trythis", 1);
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir(true, NULL, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("OMPI_PREFIX_ENV");
|
|
return(false);
|
|
}
|
|
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
unsetenv("OMPI_PREFIX_ENV");
|
|
|
|
return(true);
|
|
|
|
}
|
|
|
|
|
|
static bool test3(void)
|
|
{
|
|
/* use the TMPDIR enviro variable */
|
|
clear_proc_info();
|
|
|
|
setenv("TMPDIR", "/tmp/trythis", 1);
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir(true, NULL, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("TMPDIR");
|
|
return(false);
|
|
}
|
|
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
unsetenv("TMPDIR");
|
|
|
|
return(true);
|
|
}
|
|
|
|
|
|
static bool test4(void)
|
|
{
|
|
/* use the TMP enviro variable */
|
|
|
|
clear_proc_info();
|
|
|
|
setenv("TMP", "/tmp/trythis", 1);
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir(true, NULL, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("TMP");
|
|
return(false);
|
|
}
|
|
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
unsetenv("TMP");
|
|
|
|
return(true);
|
|
}
|
|
|
|
|
|
static bool test5(void)
|
|
{
|
|
/* use the HOME enviro variable */
|
|
|
|
clear_proc_info();
|
|
|
|
setenv("HOME", "/tmp/trythis", 1);
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir(true, NULL, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
unsetenv("HOME");
|
|
return(false);
|
|
}
|
|
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
unsetenv("HOME");
|
|
|
|
return(true);
|
|
}
|
|
|
|
|
|
static bool test6(void)
|
|
{
|
|
|
|
clear_proc_info();
|
|
|
|
/* 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 (ORTE_SUCCESS != orte_session_dir(true, NULL, NULL, NULL, "test-universe", NULL, NULL)) {
|
|
return(false);
|
|
}
|
|
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
return(true);
|
|
}
|
|
|
|
static bool test7(void)
|
|
{
|
|
char *filenm[5];
|
|
FILE *fp;
|
|
int i;
|
|
|
|
clear_proc_info();
|
|
|
|
/* create test proc session directory tree */
|
|
if (ORTE_SUCCESS != orte_session_dir(true, NULL, "localhost", NULL, "test-universe", "test-job", "test-proc")) {
|
|
return(false);
|
|
}
|
|
|
|
fprintf(test_out, "removing directories: %s\n\t%s\n\t%s\n",
|
|
orte_process_info.proc_session_dir,
|
|
orte_process_info.job_session_dir,
|
|
orte_process_info.universe_session_dir);
|
|
|
|
/* create some files */
|
|
|
|
filenm[0] = opal_os_path(false, orte_process_info.proc_session_dir, "dum1", NULL);
|
|
fp = fopen(filenm[0], "w");
|
|
fprintf(fp, "ss");
|
|
fclose(fp);
|
|
|
|
filenm[1] = opal_os_path(false, orte_process_info.job_session_dir, "dum2", NULL);
|
|
fp = fopen(filenm[1], "w");
|
|
fprintf(fp, "ss");
|
|
fclose(fp);
|
|
|
|
filenm[2] = opal_os_path(false, orte_process_info.universe_session_dir, "dum3", NULL);
|
|
fp = fopen(filenm[2], "w");
|
|
fprintf(fp, "ss");
|
|
fclose(fp);
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir_finalize(orte_process_info.my_name)) {
|
|
return(false);
|
|
}
|
|
|
|
for (i=0; i < 3; i++) unlink(filenm[i]);
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool test8(void)
|
|
{
|
|
char *filenm[5];
|
|
FILE *fp;
|
|
int i;
|
|
|
|
clear_proc_info();
|
|
|
|
/* create test proc session directory tree */
|
|
if (ORTE_SUCCESS != orte_session_dir(true, NULL, "localhost", NULL, "test-universe2", "test-job2", "test-proc2")) {
|
|
return(false);
|
|
}
|
|
|
|
fprintf(test_out, "removing directories: %s\n\t%s\n\t%s\n",
|
|
orte_process_info.proc_session_dir,
|
|
orte_process_info.job_session_dir,
|
|
orte_process_info.universe_session_dir);
|
|
|
|
/* create some files */
|
|
|
|
filenm[0] = opal_os_path(false, orte_process_info.proc_session_dir, "dum1", NULL);
|
|
fp = fopen(filenm[0], "w");
|
|
fprintf(fp, "ss");
|
|
fclose(fp);
|
|
|
|
filenm[1] = opal_os_path(false, orte_process_info.job_session_dir, "dum2", NULL);
|
|
fp = fopen(filenm[1], "w");
|
|
fprintf(fp, "ss");
|
|
fclose(fp);
|
|
|
|
filenm[2] = opal_os_path(false, orte_process_info.universe_session_dir, "dum3", NULL);
|
|
fp = fopen(filenm[2], "w");
|
|
fprintf(fp, "ss");
|
|
fclose(fp);
|
|
|
|
|
|
if (ORTE_SUCCESS != orte_session_dir_finalize(orte_process_info.my_name)) {
|
|
return(false);
|
|
}
|
|
|
|
for (i=0; i < 3; i++) unlink(filenm[i]);
|
|
orte_session_dir_finalize(orte_process_info.my_name);
|
|
|
|
return true;
|
|
}
|
|
|
|
void clear_proc_info(void)
|
|
{
|
|
orte_process_info.tmpdir_base = NULL;
|
|
orte_process_info.top_session_dir = NULL;
|
|
orte_process_info.universe_session_dir = NULL;
|
|
orte_process_info.job_session_dir = NULL;
|
|
orte_process_info.proc_session_dir = NULL;
|
|
|
|
}
|