1
1
openmpi/opal/util/os_path.c
2007-06-05 14:50:15 +00:00

111 строки
2.9 KiB
C

/*
* 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 /* HAVE_UNISTD_H */
#ifdef HAVE_SYS_PARAM_H
#include <sys/param.h>
#endif /* HAVE_SYS_PARAM_H */
#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(relative) {
total_length++;
}
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);
}