/**
* Initialize configuration based on command-line arguments
*
* @param config
* LRMA-specific configuration state
* @param argc
* Count of command-line arguments to the job manager.
* @param argv
* Array of command-line arguments to the job manager.
*
* @retval GLOBUS_SUCCESS
* Success
* @retval GLOBUS_GRAM_PROTOCOL_ERROR_GATEKEEPER_MISCONFIGURED
* Command-line includes -help
*/
int
globus_gram_job_manager_config_init(
globus_gram_job_manager_config_t * config,
int argc,
char ** argv)
{
int i;
int rc = 0;
char * tmp;
char hostname[MAXHOSTNAMELEN];
struct utsname utsname;
char * conf_path = NULL;
char * dot;
char * gatekeeper_contact;
memset(config, 0, sizeof(globus_gram_job_manager_config_t));
/* if -conf is passed then get the arguments from the file
* specified
*/
if (argc > 2 && !strcmp(argv[1],"-conf"))
{
char ** newargv;
char * newbuf;
int newargc = 52;
int length;
FILE *fp;
newargv = (char**) malloc(newargc * sizeof(char *)); /* not freed */
newargv[0] = argv[0];
/* get file length via fseek & ftell */
if ((fp = fopen(argv[2], "r")) == NULL)
{
globus_gram_job_manager_log(
NULL,
GLOBUS_GRAM_JOB_MANAGER_LOG_FATAL,
"event=gram.config.end level=FATAL path=\"%s\" "
"status=-1 msg=\"%s\" errno=%d reason=\"%s\"\n",
argv[2],
"Error opening configuration file",
errno,
strerror(errno));
exit(1);
}
conf_path = argv[2];
fseek(fp, 0, SEEK_END);
length = ftell(fp);
if (length <=0)
{
globus_gram_job_manager_log(
NULL,
GLOBUS_GRAM_JOB_MANAGER_LOG_FATAL,
"event=gram.config.end level=FATAL path=\"%s\" "
"status=-1 msg=\"%s\" errno=%d reason=\"%s\"\n",
conf_path,
"Error determining config file length",
errno,
strerror(errno));
exit(1);
}
rewind(fp);
newbuf = (char *) malloc(length+1); /* dont free */
i = fread(newbuf, 1, length, fp);
if (i < 0)
{
globus_gram_job_manager_log(
NULL,
GLOBUS_GRAM_JOB_MANAGER_LOG_FATAL,
"event=gram.config.end level=FATAL path=\"%s\" "
"status=-1 msg=\"%s\" errno=%d reason=\"%s\"\n",
conf_path,
"Error reading configuration file",
errno,
strerror(errno));
exit(1);
}
newbuf[i] = '\0';
fclose(fp);
newargv[0] = argv[0];
newargc--;
globus_l_gram_tokenize(newbuf, &newargv[1], &newargc);
for (i=3; i<argc; i++)
newargv[++newargc] = strdup(argv[i]);
argv = newargv;
argc = newargc + 1;
}
/* Default log level if nothing specified on command-line or config file */
config->log_levels = -1;
/* Default to using GLOBUS_USAGE_TARGETS environment variable.
* If not set, use the Globus usage stats service
* Eitehr can be overridden by using -disable-usagestats or setting
* -usagestats-targets in the configuration file
*/
if ((tmp = getenv("GLOBUS_USAGE_TARGETS")) != NULL)
{
config->usage_targets = strdup(tmp);
}
else
{
config->usage_targets = strdup("usage-stats.globus.org:4810");
}
/*
* Parse the command line arguments
*/
for (i = 1; i < argc; i++)
{
if (strcmp(argv[i], "-k") == 0)
{
config->kerberos = GLOBUS_TRUE;
}
else if ((strcmp(argv[i], "-home") == 0) && (i + 1 < argc))
{
config->globus_location = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-target-globus-location") == 0)
&& (i + 1 < argc))
{
config->target_globus_location = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-type") == 0) && (i + 1 < argc))
{
config->jobmanager_type = strdup(argv[++i]);
}
else if((strcmp(argv[i], "-history") == 0) && (i + 1 < argc))
{
config->job_history_dir = strdup(argv[++i]);
}
else if (strcmp(argv[i], "-cache-location") == 0)
{
config->cache_location = strdup(argv[++i]);
}
else if (strcmp(argv[i], "-scratch-dir-base") == 0)
{
config->scratch_dir_base = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-condor-arch") == 0) && (i + 1 < argc))
{
config->condor_arch = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-condor-os") == 0) && (i + 1 < argc))
{
config->condor_os = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-gatekeeper-host") == 0)
&& (i + 1 < argc))
{
config->globus_gatekeeper_host = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-gatekeeper-port") == 0)
&& (i + 1 < argc))
{
config->globus_gatekeeper_port = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-gatekeeper-subject") == 0)
&& (i + 1 < argc))
{
config->globus_gatekeeper_subject = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-host-manufacturer") == 0)
&& (i + 1 < argc))
{
config->globus_host_manufacturer = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-host-cputype") == 0)
&& (i + 1 < argc))
{
config->globus_host_cputype = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-host-osname") == 0)
&& (i + 1 < argc))
{
config->globus_host_osname = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-host-osversion") == 0)
&& (i + 1 < argc))
{
config->globus_host_osversion = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-tcp-port-range") == 0)
&& (i + 1 < argc))
{
config->tcp_port_range = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-tcp-source-range") == 0)
&& (i + 1 < argc))
{
config->tcp_source_range = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-state-file-dir") == 0)
&& (i + 1 < argc))
{
config->job_state_file_dir = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-x509-cert-dir") == 0)
&& (i + 1 < argc))
{
config->x509_cert_dir = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-extra-envvars") == 0)
&& (i + 1 < argc))
{
char * extra_envvars = strdup(argv[++i]);
char *p, *q;
p = extra_envvars;
while (p && *p)
{
q = strchr(p, ',');
if (q)
{
*q = 0;
}
globus_list_insert(
&config->extra_envvars,
strdup(p));
if (q)
{
p = q+1;
}
else
{
p = q;
}
}
free(extra_envvars);
}
else if ((strcasecmp(argv[i], "-seg-module" ) == 0)
&& (i + 1 < argc))
{
config->seg_module = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-audit-directory") == 0)
&& (i+1 < argc))
{
globus_eval_path(argv[++i], &config->auditing_dir);
}
else if ((strcmp(argv[i], "-globus-toolkit-version") == 0)
&& (i+1 < argc))
{
config->globus_version = strdup(argv[++i]);
}
else if (strcmp(argv[i], "-disable-streaming") == 0)
{
/* Ignore this request, as we don't do streaming any more */
config->streaming_disabled = GLOBUS_FALSE;
}
else if (strcmp(argv[i], "-service-tag") == 0
&& (i+1 < argc))
{
config->service_tag = strdup(argv[++i]);
}
else if (strcmp(argv[i], "-enable-syslog") == 0)
{
config->syslog_enabled = GLOBUS_TRUE;
}
else if (strcmp(argv[i], "-stdio-log") == 0
&& (i+1 < argc))
{
/* Backward-compatible definition of -stdio-log based on
* -log-pattern implementation
*/
config->log_pattern = globus_common_create_string(
"%s/gram_$(DATE).log",
argv[++i]);
}
else if (strcmp(argv[i], "-log-pattern") == 0
&& (i+1 < argc))
{
config->log_pattern = strdup(argv[++i]);
}
else if ((strcmp(argv[i], "-globus-job-dir") == 0)
&& (i + 1 < argc))
{
config->job_dir_home =
globus_common_create_string(
"%s/%s",
argv[++i],
strdup(getenv("USER")));
if (config->job_dir_home == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
globus_gram_job_manager_log(
NULL,
GLOBUS_GRAM_JOB_MANAGER_LOG_TRACE,
"event=gram.config.info "
"level=TRACE "
"option=\"-globus-job-dir\" "
"path=\"%s\" "
"\n",
config->job_dir_home);
}
else if (strcmp(argv[i], "-log-levels") == 0
&& (i+1 < argc))
{
rc = globus_i_gram_parse_log_levels(
argv[++i],
&config->log_levels,
NULL);
}
else if (strcmp(argv[i], "-disable-usagestats") == 0)
{
config->usage_disabled = GLOBUS_TRUE;
}
else if (strcmp(argv[i], "-usagestats-targets") == 0
&& (i+1 < argc))
{
if (config->usage_targets)
{
free(config->usage_targets);
config->usage_targets = NULL;
}
config->usage_targets = strdup(argv[++i]);
}
else if (strcmp(argv[i], "-enable-callout") == 0)
{
config->enable_callout = GLOBUS_TRUE;
}
else if ((strcasecmp(argv[i], "-help" ) == 0) ||
(strcasecmp(argv[i], "--help") == 0))
{
fprintf(stderr,
"Usage: globus-gram-jobmanager\n"
"\n"
"Required Arguments:\n"
"\t-type jobmanager type, i.e. fork, lsf ...\n"
"\t-globus-host-manufacturer manufacturer\n"
"\t-globus-host-cputype cputype\n"
"\t-globus-host-osname osname\n"
"\t-globus-host-osversion osversion\n"
"\t-globus-gatekeeper-host host\n"
"\t-globus-gatekeeper-port port\n"
"\t-globus-gatekeeper-subject subject\n"
"\n"
"Options:\n"
"\t-home globus_location\n"
"\t-target-globus-location globus_location\n"
"\t-condor-arch arch, i.e. SUN4x\n"
"\t-condor-os os, i.e. SOLARIS26\n"
"\t-history job-history-directory\n"
"\t-scratch-dir-base scratch-directory\n"
"\t-enable-syslog\n"
"\t-stdio-log DIRECTORY\n"
"\t-log-levels TRACE|INFO|DEBUG|WARN|ERROR|FATAL\n"
"\t-state-file-dir state-directory\n"
"\t-globus-tcp-port-range <min port #>,<max port #>\n"
"\t-globus-tcp-source-range <min port #>,<max port #>\n"
"\t-x509-cert-dir DIRECTORY\n"
"\t-cache-location PATH\n"
"\t-k\n"
"\t-extra-envvars VAR1,VAR2,...\n"
"\t-seg-module SEG-MODULE\n"
"\t-audit-directory DIRECTORY\n"
"\t-globus-toolkit-version VERSION\n"
"\t-usagestats-targets <host:port>[!<default | all>],...\n"
"\t-enable-callout\n"
"\t-globus-job-dir DIRECTORY\n"
"\n"
"Note: if type=condor then\n"
" -condor-os & -condor-arch are required.\n"
"\n");
rc = GLOBUS_GRAM_PROTOCOL_ERROR_GATEKEEPER_MISCONFIGURED;
goto out;
}
else
{
globus_gram_job_manager_log(
NULL,
GLOBUS_GRAM_JOB_MANAGER_LOG_ERROR,
"event=gram.config level=ERROR path=\"%s\" "
"argument=%s reason=\"Invalid command-line option\"\n",
conf_path ? conf_path : "ARGV",
argv[i] ? argv[i] : "");
}
}
/* If log levels were not specified on the command-line or configuration, set the
* service default
*/
if (config->log_levels == -1)
{
config->log_levels = 0;
}
/* Always have these at a minimum */
config->log_levels |= GLOBUS_GRAM_JOB_MANAGER_LOG_FATAL
| GLOBUS_GRAM_JOB_MANAGER_LOG_ERROR;
/* Verify that required values are present */
if(config->jobmanager_type == NULL)
{
globus_gram_job_manager_log(
NULL,
GLOBUS_GRAM_JOB_MANAGER_LOG_ERROR,
"event=gram.config level=ERROR path=\"%s\" argument=\"-type\" reason=\"Missing -type command-line option\"\n",
conf_path ? conf_path : "ARGV");
rc = GLOBUS_GRAM_PROTOCOL_ERROR_GATEKEEPER_MISCONFIGURED;
goto out;
}
if(config->home == NULL)
{
config->home = strdup(getenv("HOME"));
if (config->home == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
}
if (config->service_tag == NULL)
{
config->service_tag = strdup("untagged");
}
if (config->tcp_port_range == NULL)
{
char * ev = getenv("GLOBUS_TCP_PORT_RANGE");
if (ev != NULL)
{
config->tcp_port_range = strdup(ev);
}
}
if (config->tcp_source_range == NULL)
{
char * ev = getenv("GLOBUS_TCP_SOURCE_RANGE");
if (ev != NULL)
{
config->tcp_source_range = strdup(ev);
}
}
if (! globus_list_search_pred(
config->extra_envvars, globus_l_env_present, "PATH"))
{
char * path = strdup("PATH");
if (!path)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
globus_list_insert(&config->extra_envvars, path);
}
/* Now initialize values from our environment */
config->logname = strdup(getenv("LOGNAME"));
if (config->logname == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
if (config->globus_location == NULL)
{
rc = globus_location(&config->globus_location);
if (rc != GLOBUS_SUCCESS)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
}
if (config->target_globus_location == NULL)
{
config->target_globus_location = strdup(config->globus_location);
if (config->target_globus_location == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
}
if (config->job_state_file_dir == NULL)
{
rc = globus_eval_path("${localstatedir}/lib/globus/gram_job_state",
&config->job_state_file_dir);
if (rc != 0 || config->job_state_file_dir == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
}
if (config->job_dir_home == NULL)
{
config->job_dir_home =
globus_common_create_string("%s/%s",
config->job_state_file_dir,
config->logname);
if (config->job_dir_home == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
}
if (config->scratch_dir_base == NULL)
{
config->scratch_dir_base = strdup(
config->home);
if (config->scratch_dir_base == NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
}
rc = globus_libc_gethostname(hostname, sizeof(hostname));
if (rc != GLOBUS_SUCCESS)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
config->hostname = strdup(hostname);
if (config->hostname == GLOBUS_NULL)
{
rc = GLOBUS_GRAM_PROTOCOL_ERROR_MALLOC_FAILED;
goto out;
}
config->short_hostname = strdup(hostname);
dot = strchr(config->short_hostname, '.');
if (dot != NULL)
{
*dot = 0;
}
rc = uname(&utsname);
if (rc >= 0)
{
if (config->globus_host_osname == NULL)
{
config->globus_host_osname = strdup(utsname.sysname);
}
if (config->globus_host_osversion == NULL)
{
if (strcmp(utsname.sysname, "AIX") == 0)
{
config->globus_host_osversion = globus_common_create_string(
"%s.%s",
utsname.version,
utsname.release);
}
else
{
config->globus_host_osversion = globus_common_create_string(
"%s",
utsname.release);
}
}
}
gatekeeper_contact = getenv("GLOBUS_GATEKEEPER_CONTACT_STRING");
if (gatekeeper_contact)
{
char *colon;
char *save = strdup(gatekeeper_contact);
gatekeeper_contact = save;
if (gatekeeper_contact)
{
colon = strchr(gatekeeper_contact, ':');
if (colon)
{
if (!config->globus_gatekeeper_host)
{
*colon = '\0';
config->globus_gatekeeper_host = strdup(gatekeeper_contact);
}
gatekeeper_contact = colon + 1;
colon = strchr(gatekeeper_contact, ':');
if (colon)
{
if (!config->globus_gatekeeper_port)
{
*colon = '\0';
config->globus_gatekeeper_port =
strdup(gatekeeper_contact);
}
gatekeeper_contact = colon + 1;
if (!config->globus_gatekeeper_subject)
{
config->globus_gatekeeper_subject =
strdup(gatekeeper_contact);
}
}
}
}
free(save);
}
rc = globus_module_activate(GLOBUS_GSI_GSSAPI_MODULE);
if (rc != GLOBUS_SUCCESS)
{
goto out;
}
rc = globus_gram_job_manager_gsi_get_subject(&config->subject);
if (rc != GLOBUS_SUCCESS)
{
goto out;
}
config->proxy_timeout = 10*60;
out:
return rc;
}
globus_result_t
globus_usage_stats_handle_init(
globus_usage_stats_handle_t * handle,
uint16_t code,
uint16_t version,
const char * targets)
{
globus_result_t result = GLOBUS_SUCCESS;
#ifndef TARGET_ARCH_ARM
globus_i_usage_stats_handle_t * new_handle;
char * targets_env;
globus_list_t * targets_list;
char * contact;
globus_sockaddr_t addr;
int host[16];
int count;
char hostname[255];
int rc = 0;
int i;
new_handle = globus_calloc(1, sizeof(globus_i_usage_stats_handle_t));
if(!new_handle)
{
return globus_error_put(
globus_error_construct_error(
GLOBUS_USAGE_MODULE,
NULL,
GLOBUS_USAGE_STATS_ERROR_TYPE_OOM,
__FILE__,
_globus_func_name,
__LINE__,
"Out of memory"));
}
new_handle->optout = getenv("GLOBUS_USAGE_OPTOUT");
if(new_handle->optout)
{
*handle = new_handle;
return GLOBUS_SUCCESS;
}
globus_mutex_init(&new_handle->mutex, NULL);
new_handle->inuse = GLOBUS_FALSE;
new_handle->code = htons(code);
new_handle->version = htons(version);
memset(new_handle->data, 0, PACKET_SIZE);
memcpy(new_handle->data + new_handle->data_length,
(void *)&new_handle->code, 2);
new_handle->data_length += 2;
memcpy(new_handle->data + new_handle->data_length,
(void *)&new_handle->version, 2);
new_handle->data_length += 2;
rc = globus_libc_gethostaddr(&addr);
if(rc != 0)
{
return globus_error_put(
globus_error_construct_error(
GLOBUS_USAGE_MODULE,
NULL,
GLOBUS_USAGE_STATS_ERROR_TYPE_UNKNOWN_HOSTNAME,
__FILE__,
_globus_func_name,
__LINE__,
"Unable to get hostaddr."));
}
result = globus_libc_addr_to_contact_string(
&addr, GLOBUS_LIBC_ADDR_NUMERIC, &contact);
if(result != GLOBUS_SUCCESS)
{
return result;
}
result = globus_libc_contact_string_to_ints(
contact, host, &count, NULL);
if(result != GLOBUS_SUCCESS)
{
return result;
}
globus_libc_free(contact);
if(count == 4)
{
memset(new_handle->data + new_handle->data_length, 0, 12);
new_handle->data_length += 12;
}
for (i = 0; i < count; i++)
{
new_handle->data[new_handle->data_length++] = (unsigned char) host[i];
}
/* timestamp will go here */
new_handle->data_length += 4;
if(globus_libc_gethostname(hostname, 255) == 0)
{
new_handle->data_length +=
sprintf((char *) new_handle->data + new_handle->data_length,
"HOSTNAME=%s", hostname);
}
new_handle->header_length = new_handle->data_length;
if(targets)
{
globus_l_usage_stats_split_targets(targets, &new_handle->targets);
}
else if((targets_env = getenv("GLOBUS_USAGE_TARGETS"))
!= NULL)
{
globus_l_usage_stats_split_targets(
targets_env, &new_handle->targets);
}
else
{
globus_l_usage_stats_split_targets(
GLOBUS_L_USAGE_STATS_DEFAULT_TARGETS,
&new_handle->targets);
}
result = globus_xio_handle_create(
&new_handle->xio_handle,
globus_l_usage_stats_stack);
if(result != GLOBUS_SUCCESS)
{
return result;
}
result = globus_xio_open(
new_handle->xio_handle,
NULL, NULL);
if(result != GLOBUS_SUCCESS)
{
return result;
}
targets_list = new_handle->targets;
while(targets_list)
{
globus_xio_data_descriptor_t * dd;
dd = (globus_xio_data_descriptor_t *) globus_malloc(
sizeof(globus_xio_data_descriptor_t));
result = globus_xio_data_descriptor_init(
dd,
new_handle->xio_handle);
if(result != GLOBUS_SUCCESS)
{
return result;
}
result = globus_xio_data_descriptor_cntl(
*dd,
globus_l_usage_stats_udp_driver,
GLOBUS_XIO_UDP_SET_CONTACT,
(char *)globus_list_first(targets_list));
if(result != GLOBUS_SUCCESS)
{
goto exit;
}
globus_list_insert(&new_handle->xio_desc_list, dd);
targets_list = globus_list_rest(targets_list);
}
*handle = new_handle;
return GLOBUS_SUCCESS;
exit:
#endif
return result;
}
