static int counter_get_sem(pr_fh_t *fh, const char *path) {
int semid;
key_t key;
/* Obtain a key for this path. */
key = counter_get_key(path);
if (key == (key_t) -1) {
(void) pr_log_writefile(counter_logfd, MOD_COUNTER_VERSION,
"unable to get key for '%s': %s", path, strerror(errno));
return -1;
}
/* Try first using IPC_CREAT|IPC_EXCL, to check if there is an existing
* semaphore set for this key. If there is, try again, using a flag of
* zero.
*/
semid = semget(key, COUNTER_NSEMS, IPC_CREAT|IPC_EXCL|0666);
if (semid < 0) {
if (errno == EEXIST) {
semid = semget(key, 0, 0);
} else {
return -1;
}
} else {
/* Set the values of the newly created semaphore to the configured
* CounterMaxReaders and CounterMaxWriters.
*/
if (counter_set_readers(semid) < 0) {
(void) pr_log_writefile(counter_logfd, MOD_COUNTER_VERSION,
"error setting readers (semaphore ID %d): %s", semid, strerror(errno));
}
if (counter_set_writers(semid) < 0) {
(void) pr_log_writefile(counter_logfd, MOD_COUNTER_VERSION,
"error setting writers (semaphore ID %d): %s", semid, strerror(errno));
}
if (counter_set_procs(semid) < 0) {
(void) pr_log_writefile(counter_logfd, MOD_COUNTER_VERSION,
"error setting procs (semaphore ID %d): %s", semid, strerror(errno));
}
/* Record the ID of the created semaphore in the CounterFile. */
if (counter_file_add_id(fh, semid) < 0) {
(void) pr_log_writefile(counter_logfd, MOD_COUNTER_VERSION,
"error recording semaphore (semaphore ID %d) in CounterFile '%s': %s",
semid, fh->fh_path, strerror(errno));
}
}
return semid;
}
static int format_sample_response(struct context *ctx, struct evbuffer *evb)
{
int i;
char counter_name[MAX_COUNTER_KEY_LENGTH+1];
evbuffer_add_printf(evb, "{\"status\":\"ok\",\"sample_time\":%lld,\"sample\":{",
ctx->sample->sample_time);
for (i = 0; i < ctx->cl->cl_count; i++)
{
counter_get_key(ctx->cl->cl_ctr[i],counter_name,MAX_COUNTER_KEY_LENGTH+1);
if (i > 0)
evbuffer_add_printf(evb, ",");
evbuffer_add_printf(evb,"\"%s\":%lld", counter_name, stats_sample_get_value(ctx->sample,i));
}
evbuffer_add_printf(evb, "}}");
return 0;
}
int main(int argc, char **argv)
{
struct context ctx = {0};
struct event *signal_int;
struct evhttp_bound_socket *handle;
char listen_addr[256];
if (argc != 2)
{
printf("usage: statsrv STATS\n");
return -1;
}
if (stats_cl_create(&ctx.cl) != S_OK)
{
printf("Failed to allocate stats counter list\n");
return ERROR_FAIL;
}
if (stats_sample_create(&ctx.sample) != S_OK)
{
printf("Failed to allocate stats sample\n");
return ERROR_FAIL;
}
if (stats_sample_create(&ctx.prev_sample) != S_OK)
{
printf("Failed to allocate stats sample\n");
return ERROR_FAIL;
}
ctx.stats = open_stats(argv[1]);
if (!ctx.stats)
{
printf("Failed to open stats %s\n", argv[1]);
return ERROR_FAIL;
}
ctx.base = event_base_new();
if (!ctx.base)
{
printf("Could not allocate event base\n");
return 1;
}
/* add a handler for SIGINT */
signal_int = evsignal_new(ctx.base, SIGINT, sigint_cb, event_self_cbarg());
evsignal_add(signal_int,0);
/* Create a new evhttp object to handle requests. */
ctx.http = evhttp_new(ctx.base);
if (!ctx.http)
{
printf("could not create evhttp.\n");
return ERROR_FAIL;
}
evhttp_set_gencb(ctx.http, http_request_cb, &ctx);
/* Now we tell the evhttp what port to listen on */
handle = evhttp_bind_socket_with_handle(ctx.http, "0.0.0.0", 8080);
if (!handle)
{
printf("couldn't bind to http port %d.\n", (int)8080);
return ERROR_FAIL;
}
if (http_get_address(handle, listen_addr, sizeof(listen_addr)) == S_OK)
printf("http: listening at %s\n", listen_addr);
event_base_dispatch(ctx.base);
event_free(signal_int);
#if 0
start_time = current_time();
while (!signal_received)
{
err = stats_get_sample(stats,cl,sample);
if (err != S_OK)
{
printf("Error %08x (%s) getting sample\n",err,error_message(err));
}
clear();
sample_time = TIME_DELTA_TO_NANOS(start_time, sample->sample_time);
mvprintw(0,0,"SAMPLE @ %6lld.%03llds SEQ:%d\n", sample_time / 1000000000ll, (sample->sample_time % 1000000000ll) / 1000000ll, sample->sample_seq_no);
n = 1;
maxy = getmaxy(stdscr);
col = 0;
for (j = 0; j < cl->cl_count; j++)
{
counter_get_key(cl->cl_ctr[j],counter_name,MAX_COUNTER_KEY_LENGTH+1);
mvprintw(n,col+0,"%s", counter_name);
mvprintw(n,col+29,"%15lld", stats_sample_get_value(sample,j));
mvprintw(n,col+46,"%15lld", stats_sample_get_delta(sample,prev_sample,j));
if (++n == maxy)
{
col += 66;
n = 1;
}
}
refresh();
tmp = prev_sample;
prev_sample = sample;
sample = tmp;
FD_ZERO(&fds);
FD_SET(0,&fds);
now = current_time();
tv.tv_sec = 0;
tv.tv_usec = 1000000 - (now % 1000000000) / 1000;
ret = select(1, &fds, NULL, NULL, &tv);
if (ret == 1)
{
ch = getch();
if (ch == 'c' || ch == 'C')
{
stats_reset_counters(stats);
}
}
}
close_screen();
#endif
if (ctx.base)
event_base_free(ctx.base);
if (ctx.http)
evhttp_free(ctx.http);
if (ctx.stats)
{
stats_close(ctx.stats);
stats_free(ctx.stats);
}
if (ctx.cl)
stats_cl_free(ctx.cl);
if (ctx.sample)
stats_sample_free(ctx.sample);
if (ctx.prev_sample)
stats_sample_free(ctx.prev_sample);
return 0;
}
