/src/sprio/opts.c
C | 514 lines | 398 code | 44 blank | 72 comment | 90 complexity | d1efb75c8ded4ecc15cc830efe821aad MD5 | raw file
Possible License(s): GPL-2.0, AGPL-1.0
- /****************************************************************************\
- * opts.c - sprio command line option parsing
- *****************************************************************************
- * Copyright (C) 2009 Lawrence Livermore National Security.
- * Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER).
- * Written by Don Lipari <lipari1@llnl.gov>
- * CODE-OCEC-09-009. All rights reserved.
- *
- * This file is part of SLURM, a resource management program.
- * For details, see <http://www.schedmd.com/slurmdocs/>.
- * Please also read the included file: DISCLAIMER.
- *
- * SLURM is free software; you can redistribute it and/or modify it under
- * the terms of the GNU General Public License as published by the Free
- * Software Foundation; either version 2 of the License, or (at your option)
- * any later version.
- *
- * In addition, as a special exception, the copyright holders give permission
- * to link the code of portions of this program with the OpenSSL library under
- * certain conditions as described in each individual source file, and
- * distribute linked combinations including the two. You must obey the GNU
- * General Public License in all respects for all of the code used other than
- * OpenSSL. If you modify file(s) with this exception, you may extend this
- * exception to your version of the file(s), but you are not obligated to do
- * so. If you do not wish to do so, delete this exception statement from your
- * version. If you delete this exception statement from all source files in
- * the program, then also delete it here.
- *
- * SLURM is distributed in the hope that it will be useful, but WITHOUT ANY
- * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
- * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
- * details.
- *
- * You should have received a copy of the GNU General Public License along
- * with SLURM; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- \*****************************************************************************/
- #ifndef _GNU_SOURCE
- # define _GNU_SOURCE
- #endif
- #ifdef HAVE_CONFIG_H
- # include "config.h"
- #endif
- #if HAVE_GETOPT_H
- # include <getopt.h>
- #else
- # include "src/common/getopt.h"
- #endif
- #include <pwd.h>
- #include <stdlib.h>
- #include <string.h>
- #include <sys/types.h>
- #include <unistd.h>
- #include "src/common/proc_args.h"
- #include "src/common/read_config.h"
- #include "src/common/uid.h"
- #include "src/common/xstring.h"
- #include "src/sprio/sprio.h"
- /* getopt_long options, integers but not characters */
- #define OPT_LONG_HELP 0x100
- #define OPT_LONG_USAGE 0x101
- /* FUNCTIONS */
- static List _build_job_list( char* str );
- static List _build_user_list( char* str );
- static char *_get_prefix(char *token);
- static void _help( void );
- static void _parse_token( char *token, char *field, int *field_size,
- bool *right_justify, char **suffix);
- static void _print_options( void );
- static void _usage( void );
- static void _opt_env(void)
- {
- char *env_val;
- if ((env_val = getenv("SLURM_CLUSTERS"))) {
- if (!(params.clusters = slurmdb_get_info_cluster(env_val))) {
- error("'%s' can't be reached now, "
- "or it is an invalid entry for "
- "SLURM_CLUSTERS. Use 'sacctmgr --list "
- "cluster' to see available clusters.",
- env_val);
- exit(1);
- }
- }
- }
- /*
- * parse_command_line
- */
- extern void
- parse_command_line( int argc, char* argv[] )
- {
- int opt_char;
- int option_index;
- static struct option long_options[] = {
- {"noheader", no_argument, 0, 'h'},
- {"jobs", optional_argument, 0, 'j'},
- {"long", no_argument, 0, 'l'},
- {"cluster", required_argument, 0, 'M'},
- {"clusters", required_argument, 0, 'M'},
- {"norm", no_argument, 0, 'n'},
- {"format", required_argument, 0, 'o'},
- {"user", required_argument, 0, 'u'},
- {"users", required_argument, 0, 'u'},
- {"verbose", no_argument, 0, 'v'},
- {"version", no_argument, 0, 'V'},
- {"weights", no_argument, 0, 'w'},
- {"help", no_argument, 0, OPT_LONG_HELP},
- {"usage", no_argument, 0, OPT_LONG_USAGE},
- {NULL, 0, 0, 0}
- };
- /* get defaults from environment */
- _opt_env();
- while((opt_char = getopt_long(argc, argv, "hj::lM:no:u:vVw",
- long_options, &option_index)) != -1) {
- switch (opt_char) {
- case (int)'?':
- fprintf(stderr, "Try \"sprio --help\" "
- "for more information\n");
- exit(1);
- case (int)'h':
- params.no_header = true;
- break;
- case (int) 'j':
- if (optarg) {
- params.jobs = xstrdup(optarg);
- params.job_list = _build_job_list(params.jobs);
- }
- params.job_flag = true;
- break;
- case (int) 'l':
- params.long_list = true;
- break;
- case (int) 'M':
- if(params.clusters)
- list_destroy(params.clusters);
- if(!(params.clusters =
- slurmdb_get_info_cluster(optarg))) {
- error("'%s' can't be reached now, "
- "or it is an invalid entry for "
- "--cluster. Use 'sacctmgr --list "
- "cluster' to see available clusters.",
- optarg);
- exit(1);
- }
- break;
- case (int) 'n':
- params.normalized = true;
- break;
- case (int) 'o':
- xfree(params.format);
- params.format = xstrdup(optarg);
- break;
- case (int) 'u':
- xfree(params.users);
- params.users = xstrdup(optarg);
- params.user_list = _build_user_list(params.users);
- break;
- case (int) 'v':
- params.verbose++;
- break;
- case (int) 'V':
- print_slurm_version ();
- exit(0);
- case (int) 'w':
- params.weights = true;
- break;
- case OPT_LONG_HELP:
- _help();
- exit(0);
- case OPT_LONG_USAGE:
- _usage();
- exit(0);
- }
- }
- if (optind < argc) {
- if (params.job_flag) {
- params.jobs = xstrdup(argv[optind++]);
- params.job_list = _build_job_list(params.jobs);
- }
- if (optind < argc) {
- error("Unrecognized option: %s",argv[optind]);
- _usage();
- exit(1);
- }
- }
- if (params.verbose)
- _print_options();
- if (params.clusters) {
- if (list_count(params.clusters) > 1) {
- fatal("Only one cluster can be used at a time with "
- "sprio");
- }
- working_cluster_rec = list_peek(params.clusters);
- }
- }
- /*
- * parse_format - Take the user's format specification and use it to build
- * build the format specifications (internalize it to print.c data
- * structures)
- * IN format - user's format specification
- * RET zero or error code
- */
- extern int parse_format( char* format )
- {
- int field_size;
- bool right_justify;
- char *prefix = NULL, *suffix = NULL, *token = NULL;
- char *tmp_char = NULL, *tmp_format = NULL;
- char field[1];
- if (format == NULL) {
- error ("Format option lacks specification.");
- exit( 1 );
- }
- params.format_list = list_create( NULL );
- if ((prefix = _get_prefix(format))) {
- job_format_add_prefix( params.format_list, 0, 0, prefix);
- }
- field_size = strlen( format );
- tmp_format = xmalloc( field_size + 1 );
- strcpy( tmp_format, format );
- token = strtok_r( tmp_format, "%", &tmp_char);
- if (token && (format[0] != '%')) /* toss header */
- token = strtok_r( NULL, "%", &tmp_char );
- while (token) {
- _parse_token( token, field, &field_size, &right_justify,
- &suffix);
- if (field[0] == 'a')
- job_format_add_age_priority_normalized(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'A')
- job_format_add_age_priority_weighted(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'f')
- job_format_add_fs_priority_normalized(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'F')
- job_format_add_fs_priority_weighted(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'i')
- job_format_add_job_id(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'j')
- job_format_add_js_priority_normalized(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'J')
- job_format_add_js_priority_weighted(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'N')
- job_format_add_job_nice(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'p')
- job_format_add_part_priority_normalized(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'P')
- job_format_add_part_priority_weighted(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'q')
- job_format_add_qos_priority_normalized(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'Q')
- job_format_add_qos_priority_weighted(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'u')
- job_format_add_user_name(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'y')
- job_format_add_job_priority_normalized(params.format_list,
- field_size,
- right_justify,
- suffix );
- else if (field[0] == 'Y')
- job_format_add_job_priority_weighted(params.format_list,
- field_size,
- right_justify,
- suffix );
- else
- error( "Invalid job format specification: %c",
- field[0] );
- token = strtok_r( NULL, "%", &tmp_char);
- }
- xfree( tmp_format );
- return SLURM_SUCCESS;
- }
- /* Take a format specification and copy out it's prefix
- * IN/OUT token - input specification, everything before "%" is removed
- * RET - everything before "%" in the token
- */
- static char *
- _get_prefix( char *token )
- {
- char *pos, *prefix;
- if (token == NULL)
- return NULL;
- pos = strchr(token, (int) '%');
- if (pos == NULL) /* everything is prefix */
- return xstrdup(token);
- if (pos == token) /* no prefix */
- return NULL;
- pos[0] = '\0'; /* some prefix */
- prefix = xstrdup(token);
- pos[0] = '%';
- memmove(token, pos, (strlen(pos)+1));
- return prefix;
- }
- /* Take a format specification and break it into its components
- * IN token - input specification without leading "%", eg. ".5u"
- * OUT field - the letter code for the data type
- * OUT field_size - byte count
- * OUT right_justify - true of field to be right justified
- * OUT suffix - string containing everthing after the field specification
- */
- static void
- _parse_token( char *token, char *field, int *field_size, bool *right_justify,
- char **suffix)
- {
- int i = 0;
- assert (token != NULL);
- if (token[i] == '.') {
- *right_justify = true;
- i++;
- } else
- *right_justify = false;
- *field_size = 0;
- while ((token[i] >= '0') && (token[i] <= '9'))
- *field_size = (*field_size * 10) + (token[i++] - '0');
- field[0] = token[i++];
- *suffix = xstrdup(&token[i]);
- }
- /* print the parameters specified */
- static void
- _print_options()
- {
- ListIterator iterator;
- int i;
- uint32_t *job_id;
- uint32_t *user;
- printf( "-----------------------------\n" );
- printf( "format = %s\n", params.format );
- printf( "job_flag = %d\n", params.job_flag );
- printf( "jobs = %s\n", params.jobs );
- printf( "users = %s\n", params.users );
- printf( "verbose = %d\n", params.verbose );
- if ((params.verbose > 1) && params.job_list) {
- i = 0;
- iterator = list_iterator_create( params.job_list );
- while ( (job_id = list_next( iterator )) ) {
- printf( "job_list[%d] = %u\n", i++, *job_id);
- }
- list_iterator_destroy( iterator );
- }
- if ((params.verbose > 1) && params.user_list) {
- i = 0;
- iterator = list_iterator_create( params.user_list );
- while ( (user = list_next( iterator )) ) {
- printf( "user_list[%d] = %u\n", i++, *user);
- }
- list_iterator_destroy( iterator );
- }
- printf( "-----------------------------\n\n\n" );
- } ;
- /*
- * _build_job_list- build a list of job_ids
- * IN str - comma separated list of job_ids
- * RET List of job_ids (uint32_t)
- */
- static List
- _build_job_list( char* str )
- {
- List my_list;
- char *job = NULL, *tmp_char = NULL, *my_job_list = NULL;
- int i;
- uint32_t *job_id = NULL;
- if ( str == NULL)
- return NULL;
- my_list = list_create( NULL );
- my_job_list = xstrdup( str );
- job = strtok_r( my_job_list, ",", &tmp_char );
- while (job)
- {
- i = strtol( job, (char **) NULL, 10 );
- if (i <= 0) {
- error( "Invalid job id: %s", job );
- exit( 1 );
- }
- job_id = xmalloc( sizeof( uint32_t ) );
- *job_id = (uint32_t) i;
- list_append( my_list, job_id );
- job = strtok_r (NULL, ",", &tmp_char);
- }
- return my_list;
- }
- /*
- * _build_user_list- build a list of UIDs
- * IN str - comma separated list of user names
- * RET List of UIDs (uint32_t)
- */
- static List
- _build_user_list( char* str )
- {
- List my_list;
- char *user = NULL;
- char *tmp_char = NULL, *my_user_list = NULL;
- uid_t uid = (uid_t) 0;
- if ( str == NULL)
- return NULL;
- my_list = list_create( NULL );
- my_user_list = xstrdup( str );
- user = strtok_r( my_user_list, ",", &tmp_char );
- while (user) {
- if (uid_from_string (user, &uid) < 0) {
- error( "Invalid user: %s\n", user);
- } else {
- uint32_t *u_tmp = xmalloc( sizeof( uint32_t ));
- *u_tmp = (uint32_t) uid;
- list_append( my_list, u_tmp );
- }
- user = strtok_r (NULL, ",", &tmp_char);
- }
- return my_list;
- }
- static void _usage(void)
- {
- printf("Usage: sprio [-j jid[s]] [-u user_name[s]] [-o format] [--usage] [-hlnvVw]\n");
- }
- static void _help(void)
- {
- printf("\
- Usage: sprio [OPTIONS]\n\
- -h, --noheader no headers on output\n\
- -j, --jobs comma separated list of jobs\n\
- to view, default is all\n\
- -l, --long long report\n\
- -M, --cluster=cluster_name cluster to issue commands to. Default is\n\
- current cluster. cluster with no name will\n\
- reset to default.\n\
- -n, --norm display normalized values\n\
- -o, --format=format format specification\n\
- -u, --user=user_name comma separated list of users to view\n\
- -v, --verbose verbosity level\n\
- -V, --version output version information and exit\n\
- -w, --weights show the weights for each priority factor\n\
- \nHelp options:\n\
- --help show this help message\n\
- --usage display a brief summary of sprio options\n");
- }