int main(int argc, char **argv) { struct katcp_dispatch *d; int status; if(argc <= 1){ fprintf(stderr, "usage: %s [bind-ip:]listen-port\n", argv[0]); return 1; } d = startup_katcp(); if(d == NULL){ fprintf(stderr, "%s: unable to allocate state\n", argv[0]); return 1; } version_katcp(d, "echo-test", 0, 1); build_katcp(d, BUILD); if(register_katcp(d, "?echo", "echo returns its parameters", &echo_cmd)){ fprintf(stderr, "server: unable to enroll command\n"); return 1; } if(run_server_katcp(d, argv[1], 0) < 0){ fprintf(stderr, "server: run failed\n"); return 1; } status = exited_katcp(d); shutdown_katcp(d); return status; }
int run_client(struct list_loop *ls, struct task_loop *tl, int mask) { struct mul_client *ci; int update, idle, result; #if 0 struct task_loop *tx; #endif struct link_loop *lk; struct mul_msg *mm; ci = user_loop(ls, tl); if(ci == NULL){ stop_loop(ls, tl); return 0; } ci->c_task = tl; idle = LO_STOP_MASK | LO_WAIT_MASK; #ifdef DEBUG fprintf(stderr, "client[%p]: mask 0x%02x\n", tl, mask); #endif update = idle; if(mask & LO_STOP_MASK){ /* WARNING: may have to remove fd to prevent almost harmless but ugly double close */ shutdown_client(ci); return 0; } /* handle read case */ update |= LO_READ_MASK; if(mask & LO_READ_MASK){ result = read_katcp(ci->c_dispatch); #ifdef DEBUG fprintf(stderr, "client: read code is %d\n", result); #endif if(result < 0){ stop_loop(ls, tl); return 0; } if(result > 0){ update &= ~(LO_READ_MASK); /* on EOF stop reading */ } } /* handle work function */ result = lookup_katcp(ci->c_dispatch); if(result < 0){ stop_loop(ls, tl); return 0; } if(result > 0){ /* WARNING: statement triggers dispatch if no request pending (new) or if something has arrived (with pending request) */ if((ci->c_waiting == 0) || (mask & LO_WAIT_MASK)){ #ifdef DEBUG fprintf(stderr, "client[%p]: calling dispatch (queue=%d)\n", tl, ci->c_queue); #endif result = call_katcp(ci->c_dispatch); if(result == KATCP_RESULT_RESUME){ ci->c_waiting = (ci->c_queue > 0) ? 1 : 0; } else { ci->c_waiting = 0; if(ci->c_queue > 0){ /* WARNING: this should actually be an abort, as the dispatch routines have broken the queue */ fprintf(stderr, "client[%p]: warning: finished with %d outstanding requests\n", tl, ci->c_queue); log_message_katcp(ci->c_dispatch, KATCP_LEVEL_ERROR, NULL, "client function has %d outstand requests", ci->c_queue); } ci->c_queue = 0; } } } if(mask & LO_WAIT_MASK){ if(!(ci->c_waiting)){ /* no dispatch function busy, so we handle the informs and ditch everything else */ #ifdef DEBUG fprintf(stderr, "client[%p]: idle, handling queue\n", tl); #endif while((lk = receive_link_loop(ls, tl)) != NULL){ #ifdef DEBUG fprintf(stderr, "client[%p]: received event type %d\n", tl, lk->k_type); #endif if((lk->k_type == ci->c_overall->o_type_msg) && (lk->k_data != NULL)){ mm = lk->k_data; if(arg_inform_msg(mm)){ /* inform messages get sent out */ dispatch_from_msg(ci->c_dispatch, mm); } } discard_link_loop(ls, lk); } } } /* handle write case */ if(mask & LO_WRITE_MASK){ if(write_katcp(ci->c_dispatch) < 0){ stop_loop(ls, tl); return 0; } } if(flushing_katcp(ci->c_dispatch)){ /* WARNING: could disable reads and running if flush buffer too large */ update |= LO_WRITE_MASK; } if(update == idle){ /* if we are only interested in stopping then stop */ stop_loop(ls, tl); } set_mask_loop(ls, tl, update, 0); if(exited_katcp(ci->c_dispatch) != KATCP_EXIT_NOTYET){ /* TODO: make it stop */ } return 0; }
int main(int argc, char **argv) { struct katcp_dispatch *d; #if 0 struct cached_sensor_state local_data; struct fifo_sensor_state *fss; #endif int status, result; if(argc <= 1){ fprintf(stderr, "usage: %s [bind-ip:]listen-port\n", argv[0]); return 1; } /* create a state handle */ d = startup_katcp(); if(d == NULL){ fprintf(stderr, "%s: unable to allocate state\n", argv[0]); return 1; } /* load up build and version information */ add_version_katcp(d, "mylabel", 0, "myversion", "mybuildtime"); /* example sensor */ if(register_integer_sensor_katcp(d, 0, "check.integer.simple", "unix time in decaseconds", "Ds", &simple_integer_check_sensor, NULL, NULL, 0, INT_MAX)){ fprintf(stderr, "server: unable to register sensors\n"); return 1; } /* register example commands */ result = 0; result += register_katcp(d, "?check-own", "return self generated code", &own_check_cmd); result += register_katcp(d, "?check-ok", "return ok", &ok_check_cmd); result += register_katcp(d, "?check-fail", "return fail", &fail_check_cmd); result += register_katcp(d, "?check-pause", "pauses", &pause_check_cmd); result += register_katcp(d, "?check-subprocess", "runs sleep 10 as a subprocess and waits for completion", &subprocess_check_cmd); if(result < 0){ fprintf(stderr, "server: unable to register commands\n"); return 1; } #if 1 /* alternative - run with more than one client */ #define CLIENT_COUNT 3 if(run_multi_server_katcp(d, CLIENT_COUNT, argv[1], 0) < 0){ fprintf(stderr, "server: run failed\n"); } #else if(run_server_katcp(d, argv[1], 0) < 0){ fprintf(stderr, "server: run failed\n"); } #endif status = exited_katcp(d); shutdown_katcp(d); #if 0 fifo_boolean_destroy_sensor(fss); #endif return status; }