diff options
Diffstat (limited to 'lib/extra_opts.c')
| -rw-r--r-- | lib/extra_opts.c | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/lib/extra_opts.c b/lib/extra_opts.c new file mode 100644 index 00000000..3a0ce045 --- /dev/null +++ b/lib/extra_opts.c | |||
| @@ -0,0 +1,155 @@ | |||
| 1 | /***************************************************************************** | ||
| 2 | * | ||
| 3 | * Nagios-plugins extra_opts library | ||
| 4 | * | ||
| 5 | * License: GPL | ||
| 6 | * Copyright (c) 2007 Nagios Plugins Development Team | ||
| 7 | * | ||
| 8 | * Last Modified: $Date: 2008-03-15 18:42:01 -0400 (Sat, 15 Mar 2008) $ | ||
| 9 | * | ||
| 10 | * | ||
| 11 | * This program is free software: you can redistribute it and/or modify | ||
| 12 | * it under the terms of the GNU General Public License as published by | ||
| 13 | * the Free Software Foundation, either version 3 of the License, or | ||
| 14 | * (at your option) any later version. | ||
| 15 | * | ||
| 16 | * This program is distributed in the hope that it will be useful, | ||
| 17 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 18 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 19 | * GNU General Public License for more details. | ||
| 20 | * | ||
| 21 | * You should have received a copy of the GNU General Public License | ||
| 22 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
| 23 | * | ||
| 24 | * $Id: parse_ini.c 1950 2008-03-15 22:42:01Z dermoth $ | ||
| 25 | * | ||
| 26 | *****************************************************************************/ | ||
| 27 | |||
| 28 | #include "common.h" | ||
| 29 | #include "extra_opts.h" | ||
| 30 | #include "parse_ini.h" | ||
| 31 | #include "utils_base.h" | ||
| 32 | #include <ctype.h> | ||
| 33 | |||
| 34 | /* FIXME: copied from utils.h; we should move a bunch of libs! */ | ||
| 35 | int | ||
| 36 | is_option (char *str) | ||
| 37 | { | ||
| 38 | if (!str) | ||
| 39 | return 0; | ||
| 40 | else if (strspn (str, "-") == 1 || strspn (str, "-") == 2) | ||
| 41 | return 1; | ||
| 42 | else | ||
| 43 | return 0; | ||
| 44 | } | ||
| 45 | |||
| 46 | /* this is the externally visible function used by plugins */ | ||
| 47 | /* Shouldn't se modify directly **argv (passed as a char ***) and argc | ||
| 48 | * (as int *) ? | ||
| 49 | */ | ||
| 50 | char **np_extra_opts(int argc, char **argv, const char *plugin_name, int *argc_new){ | ||
| 51 | np_arg_list *extra_args=NULL, *ea_tmp1=NULL, *ea_tmp2=NULL; | ||
| 52 | char **argv_new=NULL; | ||
| 53 | char *argptr=NULL; | ||
| 54 | int i, j, optfound, ea_num=argc; | ||
| 55 | |||
| 56 | if(argc<2) { | ||
| 57 | /* No arguments provided */ | ||
| 58 | *argc_new=argc; | ||
| 59 | argv_new=argv; | ||
| 60 | return argv_new; | ||
| 61 | } | ||
| 62 | |||
| 63 | for(i=1; i<argc; i++){ | ||
| 64 | argptr=NULL; | ||
| 65 | optfound=0; | ||
| 66 | |||
| 67 | /* Do we have an extra-opts parameter? */ | ||
| 68 | if(strncmp(argv[i], "--extra-opts=", 13)==0){ | ||
| 69 | /* It is a single argument with value */ | ||
| 70 | argptr=argv[i]+13; | ||
| 71 | /* Delete the extra opts argument */ | ||
| 72 | for(j=i;j<argc;j++) argv[j]=argv[j+1]; | ||
| 73 | i--; | ||
| 74 | argc--; | ||
| 75 | }else if(strcmp(argv[i], "--extra-opts")==0){ | ||
| 76 | if(!is_option(argv[i+1])){ | ||
| 77 | /* It is a argument with separate value */ | ||
| 78 | argptr=argv[i+1]; | ||
| 79 | /* Delete the extra-opts argument/value */ | ||
| 80 | for(j=i;j<argc-1;j++) argv[j]=argv[j+2]; | ||
| 81 | i-=2; | ||
| 82 | argc-=2; | ||
| 83 | ea_num--; | ||
| 84 | }else{ | ||
| 85 | /* It has no value */ | ||
| 86 | optfound=1; | ||
| 87 | /* Delete the extra opts argument */ | ||
| 88 | for(j=i;j<argc;j++) argv[j]=argv[j+1]; | ||
| 89 | i--; | ||
| 90 | argc--; | ||
| 91 | } | ||
| 92 | } | ||
| 93 | |||
| 94 | if(argptr||optfound){ | ||
| 95 | /* Process ini section, returning a linked list of arguments */ | ||
| 96 | ea_tmp1=np_get_defaults(argptr, plugin_name); | ||
| 97 | if(ea_tmp1==NULL) { | ||
| 98 | /* no extra args? */ | ||
| 99 | ea_num--; | ||
| 100 | continue; | ||
| 101 | } | ||
| 102 | |||
| 103 | /* append the list to extra_args */ | ||
| 104 | if(extra_args==NULL){ | ||
| 105 | extra_args=ea_tmp2=ea_tmp1; | ||
| 106 | while(ea_tmp2->next) { | ||
| 107 | ea_tmp2=ea_tmp2->next; | ||
| 108 | ea_num++; | ||
| 109 | } | ||
| 110 | }else{ | ||
| 111 | ea_tmp2=extra_args; | ||
| 112 | while(ea_tmp2->next) { | ||
| 113 | ea_tmp2=ea_tmp2->next; | ||
| 114 | ea_num++; | ||
| 115 | } | ||
| 116 | ea_tmp2->next=ea_tmp1; | ||
| 117 | } | ||
| 118 | ea_tmp1=ea_tmp2=NULL; | ||
| 119 | } | ||
| 120 | /* lather, rince, repeat */ | ||
| 121 | } | ||
| 122 | |||
| 123 | if(ea_num==argc && extra_args==NULL){ | ||
| 124 | /* No extra-opts */ | ||
| 125 | *argc_new=argc; | ||
| 126 | argv_new=argv; | ||
| 127 | return argv_new; | ||
| 128 | } | ||
| 129 | |||
| 130 | /* done processing arguments. now create a new argc/argv set... */ | ||
| 131 | argv_new=(char**)malloc((ea_num+1)*sizeof(char**)); | ||
| 132 | if(argv_new==NULL) die(STATE_UNKNOWN, _("malloc() failed!\n")); | ||
| 133 | |||
| 134 | /* starting with program name (Should we strdup or just use the poiter?) */ | ||
| 135 | argv_new[0]=strdup(argv[0]); | ||
| 136 | *argc_new=1; | ||
| 137 | /* then parsed ini opts (frying them up in the same run) */ | ||
| 138 | while(extra_args){ | ||
| 139 | argv_new[*argc_new]=strdup(extra_args->arg); | ||
| 140 | *argc_new+=1; | ||
| 141 | ea_tmp1=extra_args; | ||
| 142 | extra_args=extra_args->next; | ||
| 143 | free(ea_tmp1); | ||
| 144 | } | ||
| 145 | /* finally the rest of the argv array (Should we strdup or just use the poiter?) */ | ||
| 146 | for (i=1; i<argc; i++){ | ||
| 147 | argv_new[*argc_new]=strdup(argv[i]); | ||
| 148 | *argc_new+=1; | ||
| 149 | } | ||
| 150 | /* and terminate. */ | ||
| 151 | argv_new[*argc_new]=NULL; | ||
| 152 | |||
| 153 | return argv_new; | ||
| 154 | } | ||
| 155 | |||
