test/util/orte_session_dir.c

   1 /*
   2  * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
   3  *                         University Research and Technology
   4  *                         Corporation.  All rights reserved.
   5  * Copyright (c) 2004-2005 The University of Tennessee and The University
   6  *                         of Tennessee Research Foundation.  All rights
   7  *                         reserved.
   8  * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
   9  *                         University of Stuttgart.  All rights reserved.
  10  * Copyright (c) 2004-2005 The Regents of the University of California.
  11  *                         All rights reserved.
  12  * $COPYRIGHT$
  13  *
  14  * Additional copyrights may follow
  15  *
  16  * $HEADER$
  17  */
  18
  19 #include "orte_config.h"
  20
  21 #include <stdio.h>
  22 #include <string.h>
  23 #include <stdlib.h>
  24 #ifdef HAVE_UNISTD_H
  25 #include <unistd.h>
  26 #endif
  27 #ifdef HAVE_LIBGEN_H
  28 #include <libgen.h>
  29 #endif
  30 #ifdef HAVE_SYS_PARAM_H
  31 #include <sys/param.h>
  32 #endif
  33 #include <sys/stat.h>
  34
  35 #include "support.h"
  36 #include "orte/orte_constants.h"
  37 #include "orte/mca/ns/ns_types.h"
  38 #include "orte/util/proc_info.h"
  39 #include "orte/util/sys_info.h"
  40 #include "opal/util/os_path.h"
  41 #include "opal/util/os_create_dirpath.h"
  42 #include "orte/util/session_dir.h"
  43 #include "orte/util/proc_info.h"
  44 #include "orte/runtime/runtime.h"
  45
  46
  47 static bool test1(void);   /* given prefix, both one that works and one that fails */
  48 static bool test2(void);   /* no prefix given, ORTE_PREFIX_ENV set, one good and one bad */
  49 static bool test3(void);   /* no prefix given, TMPDIR set, one good and one bad */
  50 static bool test4(void);   /* no prefix given, TMP set, one good and one bad */
  51 static bool test5(void);   /* no prefix given, HOME set, one good and one bad */
  52 static bool test6(void);   /* no prefix given, nothing set, one good and one bad */
  53 static bool test7(void);   /* remove session directory tree */
  54 static bool test8(void);   /* attempt to remove tree when subdirs present */
  55
  56 void clear_proc_info(void);
  57
  58 static FILE *test_out=NULL;
  59
  60 int main(int argc, char* argv[])
  61 {
  62     orte_sys_info(); /* initialize system */
  63     orte_proc_info(); /* initialize proc info structure */
  64     orte_process_info.my_name = (orte_process_name_t*)malloc(sizeof(orte_process_name_t));
  65     orte_process_info.my_name->cellid = 0;
  66     orte_process_info.my_name->jobid = 0;
  67     orte_process_info.my_name->vpid = 0;
  68
  69     test_init("orte_session_dir_t");
  70     test_out = fopen( "test_session_dir_out", "w+" );
  71     if( test_out == NULL ) {
  72       test_failure("test_session_dir couldn't open test file failed");
  73       test_finalize();
  74       exit(1);
  75     }
  76
  77
  78     fprintf(test_out, "running test1\n");
  79     if (test1()) {
  80         test_success();
  81     }
  82     else {
  83       test_failure("orte_session_dir_t test1 failed");
  84     }
  85
  86     fprintf(test_out, "running test2\n");
  87     if (test2()) {
  88         test_success();
  89     }
  90     else {
  91       test_failure("orte_session_dir_t test2 failed");
  92     }
  93
  94     fprintf(test_out, "running test3\n");
  95     if (test3()) {
  96         test_success();
  97     }
  98     else {
  99       test_failure("orte_session_dir_t test3 failed");
 100     }
 101
 102     fprintf(test_out, "running test4\n");
 103     if (test4()) {
 104         test_success();
 105     }
 106     else {
 107       test_failure("orte_session_dir_t test4 failed");
 108     }
 109
 110     fprintf(test_out, "running test5\n");
 111     if (test5()) {
 112         test_success();
 113     }
 114     else {
 115       test_failure("orte_session_dir_t test5 failed");
 116     }
 117
 118     fprintf(test_out, "running test6\n");
 119     if (test6()) {
 120         test_success();
 121     }
 122     else {
 123       test_failure("orte_session_dir_t test6 failed");
 124     }
 125
 126     fprintf(test_out, "running test7\n");
 127     if (test7()) {
 128         test_success();
 129     }
 130     else {
 131       test_failure("orte_session_dir_t test7 failed");
 132     }
 133
 134     fprintf(test_out, "running test8\n");
 135     if (test8()) {
 136         test_success();
 137     }
 138     else {
 139       test_failure("orte_session_dir_t test8 failed");
 140     }
 141
 142     fprintf(test_out, "completed all tests\n");
 143
 144     fclose(test_out);
 145
 146     /* clean up */
 147     orte_sys_info_finalize();
 148     orte_proc_info_finalize();
 149
 150     test_finalize();
 151     return 0;
 152 }
 153
 154
 155 static bool test1(void)
 156 {
 157     /* Test proper action when given a prefix */
 158
 159     char *prefix;
 160
 161     /* see if we can create a specified path */
 162
 163     clear_proc_info();
 164
 165     prefix = opal_os_path(false, "tmp", NULL);
 166     if (ORTE_ERROR == orte_session_dir(true, prefix, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 167         fprintf(test_out, "test1 - couldn't create specified path\n");
 168         free(prefix);
 169         return(false);
 170     }
 171     /* see if it can access an existing path */
 172
 173     if (ORTE_ERROR == orte_session_dir(false, prefix, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 174         fprintf(test_out, "test1 - couldn't access existing path\n");
 175         free(prefix);
 176         return(false);
 177     }
 178
 179     orte_session_dir_finalize(orte_process_info.my_name);
 180     free(orte_process_info.universe_session_dir);
 181     free(prefix);
 182
 183     return true;
 184 }
 185
 186
 187 static bool test2(void)
 188 {
 189     clear_proc_info();
 190
 191     /* use the ORTE_PREFIX_ENV variable */
 192
 193     setenv("OMPI_PREFIX_ENV", "/tmp/trythis", 1);
 194
 195     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 196         unsetenv("OMPI_PREFIX_ENV");
 197         return(false);
 198     }
 199
 200     orte_session_dir_finalize(orte_process_info.my_name);
 201
 202     unsetenv("OMPI_PREFIX_ENV");
 203
 204     return(true);
 205
 206 }
 207
 208
 209 static bool test3(void)
 210 {
 211     /* use the TMPDIR enviro variable */
 212     clear_proc_info();
 213
 214     setenv("TMPDIR", "/tmp/trythis", 1);
 215
 216     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 217         unsetenv("TMPDIR");
 218         return(false);
 219     }
 220
 221     orte_session_dir_finalize(orte_process_info.my_name);
 222
 223     unsetenv("TMPDIR");
 224
 225     return(true);
 226 }
 227
 228
 229 static bool test4(void)
 230 {
 231     /* use the TMP enviro variable */
 232
 233     clear_proc_info();
 234
 235     setenv("TMP", "/tmp/trythis", 1);
 236
 237     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 238         unsetenv("TMP");
 239         return(false);
 240     }
 241
 242     orte_session_dir_finalize(orte_process_info.my_name);
 243
 244     unsetenv("TMP");
 245
 246     return(true);
 247 }
 248
 249
 250 static bool test5(void)
 251 {
 252     /* use the HOME enviro variable */
 253
 254     clear_proc_info();
 255
 256     setenv("HOME", "/tmp/trythis", 1);
 257
 258     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 259         unsetenv("HOME");
 260         return(false);
 261     }
 262
 263     orte_session_dir_finalize(orte_process_info.my_name);
 264
 265     unsetenv("HOME");
 266
 267     return(true);
 268 }
 269
 270
 271 static bool test6(void)
 272 {
 273
 274     clear_proc_info();
 275
 276     /* no enviro variables set, no prefix given
 277     * Program should turn to default of /tmp (where "/" is whatever
 278     * top-level directory is appropriate for given system)
 279     */
 280     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, NULL, NULL, "test-universe", NULL, NULL)) {
 281         return(false);
 282     }
 283
 284     orte_session_dir_finalize(orte_process_info.my_name);
 285
 286     return(true);
 287 }
 288
 289 static bool test7(void)
 290 {
 291     char *filenm[5];
 292     FILE *fp;
 293     int i;
 294
 295     clear_proc_info();
 296
 297     /* create test proc session directory tree */
 298     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, "localhost", NULL, "test-universe", "test-job", "test-proc")) {
 299         return(false);
 300     }
 301
 302     fprintf(test_out, "removing directories: %s\n\t%s\n\t%s\n",
 303             orte_process_info.proc_session_dir,
 304             orte_process_info.job_session_dir,
 305             orte_process_info.universe_session_dir);
 306
 307     /* create some files */
 308
 309     filenm[0] = opal_os_path(false, orte_process_info.proc_session_dir, "dum1", NULL);
 310     fp = fopen(filenm[0], "w");
 311     fprintf(fp, "ss");
 312     fclose(fp);
 313
 314     filenm[1] = opal_os_path(false, orte_process_info.job_session_dir, "dum2", NULL);
 315     fp = fopen(filenm[1], "w");
 316     fprintf(fp, "ss");
 317     fclose(fp);
 318
 319     filenm[2] = opal_os_path(false, orte_process_info.universe_session_dir, "dum3", NULL);
 320     fp = fopen(filenm[2], "w");
 321     fprintf(fp, "ss");
 322     fclose(fp);
 323
 324     if (ORTE_ERROR == orte_session_dir_finalize(orte_process_info.my_name)) {
 325         return(false);
 326     }
 327
 328     for (i=0; i < 3; i++) unlink(filenm[i]);
 329     orte_session_dir_finalize(orte_process_info.my_name);
 330
 331     return true;
 332 }
 333
 334 static bool test8(void)
 335 {
 336     char *filenm[5];
 337     FILE *fp;
 338     int i;
 339
 340     clear_proc_info();
 341
 342     /* create test proc session directory tree */
 343     if (ORTE_ERROR == orte_session_dir(true, NULL, orte_system_info.user, "localhost", NULL, "test-universe2", "test-job2", "test-proc2")) {
 344         return(false);
 345     }
 346
 347     fprintf(test_out, "removing directories: %s\n\t%s\n\t%s\n",
 348             orte_process_info.proc_session_dir,
 349             orte_process_info.job_session_dir,
 350             orte_process_info.universe_session_dir);
 351
 352     /* create some files */
 353
 354     filenm[0] = opal_os_path(false, orte_process_info.proc_session_dir, "dum1", NULL);
 355     fp = fopen(filenm[0], "w");
 356     fprintf(fp, "ss");
 357     fclose(fp);
 358
 359     filenm[1] = opal_os_path(false, orte_process_info.job_session_dir, "dum2", NULL);
 360     fp = fopen(filenm[1], "w");
 361     fprintf(fp, "ss");
 362     fclose(fp);
 363
 364     filenm[2] = opal_os_path(false, orte_process_info.universe_session_dir, "dum3", NULL);
 365     fp = fopen(filenm[2], "w");
 366     fprintf(fp, "ss");
 367     fclose(fp);
 368
 369
 370     if (ORTE_ERROR == orte_session_dir_finalize(orte_process_info.my_name)) {
 371            return(false);
 372     }
 373
 374     for (i=0; i < 3; i++) unlink(filenm[i]);
 375     orte_session_dir_finalize(orte_process_info.my_name);
 376
 377     return true;
 378 }
 379
 380 void clear_proc_info(void)
 381 {
 382     orte_process_info.tmpdir_base = NULL;
 383     orte_process_info.top_session_dir = NULL;
 384     orte_process_info.universe_session_dir = NULL;
 385     orte_process_info.job_session_dir = NULL;
 386     orte_process_info.proc_session_dir = NULL;
 387
 388 }