int main (int argc, char **argv, char **env) {
argc = argc; argv = argv; env = env;
char *result, *error;
char cmd[120];
int x;
/* set hostname */
gethostname(hostname, GM_BUFFERSIZE-1);
/* create options structure and set debug level */
mod_gm_opt = malloc(sizeof(mod_gm_opt_t));
set_default_options(mod_gm_opt);
mod_gm_opt->debug_level = 4;
#ifdef EMBEDDEDPERL
char p1[150];
snprintf(p1, 150, "--p1_file=worker/mod_gearman2_p1.pl");
parse_args_line(mod_gm_opt, p1, 0);
init_embedded_perl(env);
#endif
gm_job_t * exec_job;
exec_job = ( gm_job_t * )malloc( sizeof *exec_job );
set_default_job(exec_job, mod_gm_opt);
strcpy(cmd, "BLAH=BLUB /bin/hostname");
printf("this should be popen\n");
run_check(cmd, &result, &error);
free(result);
free(error);
strcpy(cmd, "/bin/hostname");
printf("this should be execvp\n");
run_check(cmd, &result, &error);
free(result);
free(error);
mod_gm_opt->debug_level = 0;
for(x=0;x<100;x++) {
run_check(cmd, &result, &error);
free(result);
free(error);
}
free_job(exec_job);
mod_gm_free_opt(mod_gm_opt);
#ifdef EMBEDDEDPERL
deinit_embedded_perl(0);
#endif
exit(0);
}
int main(void) {
int rc, rrc;
char *result, *error;
char cmd[120];
char hostname[GM_BUFFERSIZE];
plan(54);
/* set hostname */
gethostname(hostname, GM_BUFFERSIZE-1);
/* create options structure and set debug level */
mod_gm_opt = malloc(sizeof(mod_gm_opt_t));
set_default_options(mod_gm_opt);
mod_gm_opt->debug_level = 0;
/*****************************************
* arg parsing test 1
*/
char *argv[MAX_CMD_ARGS];
strcpy(cmd, "/bin/true");
parse_command_line(cmd, argv);
like(argv[0], cmd, "parsing args cmd 1");
/*****************************************
* arg parsing test 2
*/
strcpy(cmd, "/bin/cmd blah blub foo");
parse_command_line(cmd,argv);
like(argv[0], "/bin/cmd", "parsing args cmd 2");
like(argv[1], "blah", "parsing args cmd 2");
like(argv[2], "blub", "parsing args cmd 2");
like(argv[3], "foo", "parsing args cmd 2");
/*****************************************
* send_gearman
*/
strcpy(cmd, "./send_gearman --server=blah --key=testtest --host=test --service=test --message=test --returncode=0");
rrc = real_exit_code(run_check(cmd, &result, &error));
diag(result);
cmp_ok(rrc, "==", 3, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* send_gearman
*/
//mod_gm_opt->debug_level = 4;
strcpy(cmd, "./send_multi --server=blah --host=blah < t/data/send_multi.txt");
rrc = real_exit_code(run_check(cmd, &result, &error));
diag(result);
cmp_ok(rrc, "==", 3, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 1
*/
strcpy(cmd, "/bin/true");
rc = run_check(cmd, &result, &error);
cmp_ok(rc, "==", 0, "pclose for cmd '%s' returned rc %d", cmd, rc);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 0, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 2
*/
strcpy(cmd, "/bin/true 2>&1");
rc = run_check(cmd, &result, &error);
cmp_ok(rc, "==", 0, "pclose for cmd '%s' returned rc %d", cmd, rc);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 0, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 3
*/
strcpy(cmd, "/usr/lib/nagios/plugins/check_icmp -H 127.0.0.1");
rc = run_check(cmd, &result, &error);
cmp_ok(rc, "==", 0, "pclose for cmd '%s' returned rc %d", cmd, rc);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 0, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 4
*/
strcpy(cmd, "echo -n 'test'; exit 2");
rc = run_check(cmd, &result, &error);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 2, "cmd '%s' returned rc %d", cmd, rrc);
like(result, "test", "returned result string");
free(result);
free(error);
gm_job_t * exec_job;
exec_job = ( gm_job_t * )malloc( sizeof *exec_job );
set_default_job(exec_job, mod_gm_opt);
/*****************************************
* non existing command 1
*/
exec_job->command_line = strdup("/bin/doesntexist");
exec_job->type = strdup("service");
exec_job->timeout = 10;
int fork_on_exec = 0;
execute_safe_command(exec_job, fork_on_exec, hostname);
cmp_ok(exec_job->return_code, "==", 2, "cmd '%s' returns rc 2", exec_job->command_line);
like(exec_job->output, "CRITICAL: Return code of 127 is out of bounds. Make sure the plugin you're trying to run actually exists. \\(worker:", "returned result string");
free(exec_job->output);
free(exec_job->error);
fork_on_exec = 1;
lives_ok({execute_safe_command(exec_job, fork_on_exec, hostname);}, "executing command using fork on exec");
/* get a job */
void *get_job( gearman_job_st *job, void *context, size_t *result_size, gearman_return_t *ret_ptr ) {
sigset_t block_mask;
int wsize, valid_lines;
char workload[GM_BUFFERSIZE];
char * decrypted_data;
char * decrypted_data_c;
char * decrypted_orig;
char *ptr;
/* reset timeout for now, will be set befor execution again */
alarm(0);
signal(SIGALRM, SIG_IGN);
jobs_done++;
/* send start signal to parent */
set_state(GM_JOB_START);
gm_log( GM_LOG_TRACE, "get_job()\n" );
/* contect is unused */
context = context;
/* set size of result */
*result_size = 0;
/* reset sleep time */
sleep_time_after_error = 1;
/* ignore sigterms while running job */
sigemptyset(&block_mask);
sigaddset(&block_mask, SIGTERM);
sigprocmask(SIG_BLOCK, &block_mask, NULL);
/* get the data */
current_gearman_job = job;
wsize = gearman_job_workload_size(job);
strncpy(workload, (const char*)gearman_job_workload(job), wsize);
workload[wsize] = '\0';
gm_log( GM_LOG_TRACE, "got new job %s\n", gearman_job_handle( job ) );
gm_log( GM_LOG_TRACE, "%d +++>\n%s\n<+++\n", strlen(workload), workload );
/* decrypt data */
decrypted_data = malloc(GM_BUFFERSIZE);
decrypted_data_c = decrypted_data;
mod_gm_decrypt(&decrypted_data, workload, mod_gm_opt->transportmode);
decrypted_orig = strdup(decrypted_data);
if(decrypted_data == NULL) {
*ret_ptr = GEARMAN_WORK_FAIL;
free(decrypted_orig);
return NULL;
}
gm_log( GM_LOG_TRACE, "%d --->\n%s\n<---\n", strlen(decrypted_data), decrypted_data );
/* set result pointer to success */
*ret_ptr= GEARMAN_SUCCESS;
exec_job = ( gm_job_t * )malloc( sizeof *exec_job );
set_default_job(exec_job, mod_gm_opt);
valid_lines = 0;
while ( (ptr = strsep(&decrypted_data, "\n" )) != NULL ) {
char *key = strsep( &ptr, "=" );
char *value = strsep( &ptr, "\x0" );
if ( key == NULL )
continue;
if ( value == NULL || !strcmp( value, "") )
break;
if ( !strcmp( key, "host_name" ) ) {
exec_job->host_name = strdup(value);
valid_lines++;
} else if ( !strcmp( key, "service_description" ) ) {
exec_job->service_description = strdup(value);
valid_lines++;
} else if ( !strcmp( key, "type" ) ) {
exec_job->type = strdup(value);
valid_lines++;
} else if ( !strcmp( key, "result_queue" ) ) {
exec_job->result_queue = strdup(value);
valid_lines++;
} else if ( !strcmp( key, "check_options" ) ) {
exec_job->check_options = atoi(value);
valid_lines++;
} else if ( !strcmp( key, "scheduled_check" ) ) {
exec_job->scheduled_check = atoi(value);
valid_lines++;
} else if ( !strcmp( key, "reschedule_check" ) ) {
exec_job->reschedule_check = atoi(value);
valid_lines++;
} else if ( !strcmp( key, "latency" ) ) {
exec_job->latency = atof(value);
valid_lines++;
} else if ( !strcmp( key, "next_check" ) ) {
string2timeval(value, &exec_job->next_check);
valid_lines++;
} else if ( !strcmp( key, "start_time" ) ) {
/* for compatibility reasons... (used by older mod-gearman neb modules) */
string2timeval(value, &exec_job->next_check);
string2timeval(value, &exec_job->core_time);
valid_lines++;
} else if ( !strcmp( key, "core_time" ) ) {
string2timeval(value, &exec_job->core_time);
valid_lines++;
} else if ( !strcmp( key, "timeout" ) ) {
exec_job->timeout = atoi(value);
valid_lines++;
} else if ( !strcmp( key, "command_line" ) ) {
exec_job->command_line = strdup(value);
valid_lines++;
}
}
#ifdef GM_DEBUG
if(exec_job->next_check.tv_sec < 10000)
write_debug_file(&decrypted_orig);
#endif
if(valid_lines == 0) {
gm_log( GM_LOG_ERROR, "discarded invalid job (%s), check your encryption settings\n", gearman_job_handle( job ) );
} else {
do_exec_job();
}
current_gearman_job = NULL;
/* start listening to SIGTERMs */
sigprocmask(SIG_UNBLOCK, &block_mask, NULL);
free(decrypted_orig);
free(decrypted_data_c);
free_job(exec_job);
/* send finish signal to parent */
set_state(GM_JOB_END);
if(mod_gm_opt->max_jobs > 0 && jobs_done >= mod_gm_opt->max_jobs) {
gm_log( GM_LOG_TRACE, "jobs done: %i -> exiting...\n", jobs_done );
clean_worker_exit(0);
_exit( EXIT_SUCCESS );
}
return NULL;
}
int main (int argc, char **argv, char **env) {
argc = argc; argv = argv; env = env;
int rc, rrc;
char *result, *error;
char cmd[120];
char hostname[GM_BUFFERSIZE];
plan(56);
/* set hostname */
gethostname(hostname, GM_BUFFERSIZE-1);
/* create options structure and set debug level */
mod_gm_opt = malloc(sizeof(mod_gm_opt_t));
set_default_options(mod_gm_opt);
mod_gm_opt->debug_level = 0;
#ifdef EMBEDDEDPERL
char p1[150];
snprintf(p1, 150, "--p1_file=worker/mod_gearman_p1.pl");
parse_args_line(mod_gm_opt, p1, 0);
init_embedded_perl(env);
#endif
/*****************************************
* arg parsing test 1
*/
char *args[MAX_CMD_ARGS];
strcpy(cmd, "/bin/hostname");
parse_command_line(cmd, args);
like(args[0], cmd, "parsing args cmd 1");
/*****************************************
* arg parsing test 2
*/
strcpy(cmd, "/bin/cmd blah blub foo");
parse_command_line(cmd,args);
like(args[0], "/bin/cmd", "parsing args cmd 2");
like(args[1], "blah", "parsing args cmd 2");
like(args[2], "blub", "parsing args cmd 2");
like(args[3], "foo", "parsing args cmd 2");
/*****************************************
* send_gearman
*/
strcpy(cmd, "./send_gearman --server=blah --key=testtest --host=test --service=test --message=test --returncode=0");
rrc = real_exit_code(run_check(cmd, &result, &error));
//diag(result);
cmp_ok(rrc, "==", 3, "cmd '%s' returned rc %d", cmd, rrc);
if(atof(gearman_version()) >= 0.31) {
like(result, "send_gearman UNKNOWN:", "result");
} else {
like(result, "sending job to gearmand failed:", "result");
}
free(result);
free(error);
/*****************************************
* send_gearman
*/
//mod_gm_opt->debug_level = 4;
strcpy(cmd, "./send_multi --server=blah --host=blah < t/data/send_multi.txt");
rrc = real_exit_code(run_check(cmd, &result, &error));
//diag(result);
cmp_ok(rrc, "==", 3, "cmd '%s' returned rc %d", cmd, rrc);
if(atof(gearman_version()) >= 0.31) {
like(result, "send_multi UNKNOWN:", "result");
} else {
like(result, "sending job to gearmand failed:", "result");
}
free(result);
free(error);
/*****************************************
* simple test command 1
*/
strcpy(cmd, "/bin/hostname");
rc = run_check(cmd, &result, &error);
cmp_ok(rc, "==", 0, "pclose for cmd '%s' returned rc %d", cmd, rc);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 0, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 2
*/
strcpy(cmd, "/bin/hostname 2>&1");
rc = run_check(cmd, &result, &error);
cmp_ok(rc, "==", 0, "pclose for cmd '%s' returned rc %d", cmd, rc);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 0, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 3
*/
strcpy(cmd, "/usr/lib/nagios/plugins/check_icmp -H 127.0.0.1");
rc = run_check(cmd, &result, &error);
cmp_ok(rc, "==", 0, "pclose for cmd '%s' returned rc %d", cmd, rc);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 0, "cmd '%s' returned rc %d", cmd, rrc);
free(result);
free(error);
/*****************************************
* simple test command 4
*/
strcpy(cmd, "echo -n 'test'; exit 2");
rc = run_check(cmd, &result, &error);
rrc = real_exit_code(rc);
cmp_ok(rrc, "==", 2, "cmd '%s' returned rc %d", cmd, rrc);
like(result, "test", "returned result string");
free(result);
free(error);
gm_job_t * exec_job;
exec_job = ( gm_job_t * )malloc( sizeof *exec_job );
set_default_job(exec_job, mod_gm_opt);
/*****************************************
* non existing command 1
*/
exec_job->command_line = strdup("/bin/doesntexist");
exec_job->type = strdup("service");
exec_job->timeout = 10;
int fork_on_exec = 0;
execute_safe_command(exec_job, fork_on_exec, hostname);
cmp_ok(exec_job->return_code, "==", 2, "cmd '%s' returns rc 2", exec_job->command_line);
like(exec_job->output, "CRITICAL: Return code of 127 is out of bounds. Make sure the plugin you're trying to run actually exists. \\(worker:", "returned result string");
free(exec_job->output);
free(exec_job->error);
fork_on_exec = 1;
lives_ok({execute_safe_command(exec_job, fork_on_exec, hostname);}, "executing command using fork on exec");
