os_path.c

来自「MPI stands for the Message Passing Inter」· C语言 代码 · 共 110 行

/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 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 "opal_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/os_path.h"
#include "opal/constants.h"

static const char *path_sep = OPAL_PATH_SEP;

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

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

    /* 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(3);
        path[0] = '\0';
    	if (relative) {
    	    strcpy(path, ".");
            strcat(path, path_sep);
    	}
    	else {
#ifndef __WINDOWS__
    	    strcpy(path, 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 * strlen(path_sep) + 1;
    if (total_length > MAXPATHLEN) {  /* path length is too long - reject it */
    	return(NULL);
    }

    path = (char *)malloc(total_length);
    if (NULL == path) {
        return(NULL);
    }
    path[0] = 0;

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

    /* Windows does not require to have the initial separator. */
    if( NULL != (element=va_arg(ap1, char*)) ) {
#ifndef __WINDOWS__
    	if (path_sep[0] != element[0]) {
            strcat(path, path_sep);
        }
#endif  /* __WINDOWS__ */
        strcat(path, element);
    }
    while (NULL != (element=va_arg(ap1, char*))) {
    	if (path_sep[0] != element[0]) {
            strcat(path, path_sep);
        }
        strcat(path, element);
    }

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