openmpi/opal/mca/base/mca_base_cmd_line.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 <stdio.h>
#include <string.h>

#include "include/constants.h"
#include "opal/util/cmd_line.h"
#include "opal/util/argv.h"
#include "opal/util/opal_environ.h"
#include "mca/base/base.h"


/* 
 * Private variables
 */
static int mca_param_argc = 0;
static char **mca_param_argv = NULL;
static int mca_value_argc = 0;
static char **mca_value_argv = NULL;


/*
 * Add -mca to the possible command line options list
 */
int mca_base_cmd_line_setup(opal_cmd_line_t *cmd)
{
  return opal_cmd_line_make_opt3(cmd, '\0', "mca", "mca", 2,
                                 "General mechanism to pass MCA parameters");
}


/*
 * Look for and handle any -mca options on the command line
 */
int mca_base_cmd_line_process_args(opal_cmd_line_t *cmd,
                                   char ***env)
{
  int i, num_insts;
  char *name;

  /* First, wipe out any previous results */

  if (mca_param_argc > 0) {
      opal_argv_free(mca_param_argv);
      opal_argv_free(mca_value_argv);
      mca_param_argv = mca_value_argv = NULL;
      mca_param_argc = mca_value_argc = 0;
  }

  /* If no "-mca" parameters were given, just return */

  if (!opal_cmd_line_is_taken(cmd, "mca")) {
      return OMPI_SUCCESS;
  }

  /* Otherwise, assemble them into an argc/argv */

  num_insts = opal_cmd_line_get_ninsts(cmd, "mca");
  for (i = 0; i < num_insts; ++i) {
      mca_base_cmd_line_process_arg(opal_cmd_line_get_param(cmd, "mca", i, 0), 
                                    opal_cmd_line_get_param(cmd, "mca", i, 1));
  }

  /* Now put that argc/argv in the environment */

  if (NULL == mca_param_argv) {
      return OMPI_SUCCESS;
  }

  /* Loop through all the -mca args that we've gotten and make env
     vars of the form OMPI_MCA_*=value.  This is a memory leak, but
     that's how putenv works.  :-( */

  for (i = 0; NULL != mca_param_argv[i]; ++i) {
      name = mca_base_param_environ_variable(mca_param_argv[i], NULL, NULL);
      opal_setenv(name, mca_value_argv[i], true, env);
      free(name);
  }

  return OMPI_SUCCESS;
}


/*
 * Process a single MCA argument.  Done as a separate function so that
 * top-level applications can directly invoke this to effect MCA
 * command line arguments.  
 */
int mca_base_cmd_line_process_arg(const char *param, const char *value)
{
  int i;
  char *new_str;

  /* Look to see if we've already got an -mca argument for the same
     param.  Check against the list of MCA param's that we've already
     saved arguments for. */

  for (i = 0; NULL != mca_param_argv && NULL != mca_param_argv[i]; ++i) {
    if (0 == strcmp(param, mca_param_argv[i])) {
      asprintf(&new_str, "%s,%s", mca_value_argv[i], value);
      free(mca_value_argv[i]);
      mca_value_argv[i] = new_str;

      return OMPI_SUCCESS;
    }
  }

  /* If we didn't already have an value for the same param, save this
     one away */
  
  opal_argv_append(&mca_param_argc, &mca_param_argv, param);
  opal_argv_append(&mca_value_argc, &mca_value_argv, value);

  return OMPI_SUCCESS;
}