openmpi/opal/util/os_path.c

/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University.
 *                         All rights reserved.
 * Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
 *                         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 "ompi_config.h"

#include <string.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_LIBGEN_H
#include <libgen.h>
#endif
#ifdef HAVE_SYS_PARAM_H
#include <sys/param.h>
#endif
#include <stdlib.h>
#include <stdarg.h>

#include "opal/util/sys_info.h"
#include "opal/util/os_path.h"
#include "opal/util/sys_info.h"
#include "ompi/include/constants.h"

char *opal_os_path(bool relative, ...)
{
    va_list ap, ap1;
    char *element, *path;
    int num_elements, total_length;

    va_start(ap, relative);
    va_start(ap1, relative);

    /* make sure system info is filled and separator is non-NULL */
    orte_sys_info();
    if (NULL == orte_system_info.path_sep) {
	   return(NULL);
    }

    /* no way to protect ourselves from reading too far, so have to trust caller
       that they ended the list with the NULL */

    num_elements = 0;
    total_length = 0;
    while (NULL != (element=va_arg(ap, char*))) {
	   num_elements++;
	   total_length = total_length + strlen(element);
    }

    if (0 == num_elements) { /* must be looking for a simple answer */
    	path = (char *)malloc(2);
            path[0] = 0;
    	if (relative) {
    	    strcpy(path, ".");
            strcat(path, orte_system_info.path_sep);
    	}
    	else {
#ifndef WIN32
    	    strcpy(path, orte_system_info.path_sep);
#endif
    	}
    	return(path);
    }

    /* setup path with enough room for the string terminator, the elements, and
       the separator between each of the elements */
    total_length = total_length + num_elements + 1;
    if (total_length > MAXPATHLEN) {  /* path length is too long - reject it */
    	return(NULL);
    }

    path = (char *)malloc(2 + total_length + num_elements-1);
    if (NULL == path) {
	   return(NULL);
    }
    path[0] = 0;

    if (relative) {
	   strcpy(path, ".");
    }

    while (NULL != (element=va_arg(ap1, char*))) {
    	if (orte_system_info.path_sep[0] != element[0]) {
            strcat(path, orte_system_info.path_sep);
        }
        strcat(path, element);
    }

    va_end(ap);
    va_end(ap1);
    return(path);
}