static int print_input(int sd, int events, void *wp_)
{
int ret, pkt = 0;
simple_worker *wp = (simple_worker *)wp_;
struct kvvec kvv = KVVEC_INITIALIZER;
char *buf;
unsigned long tot_bytes = 0, size;
/*
* if some command filled the buffer, we grow it and read some
* more until we hit the limit
* @todo Define a limit :p
*/
size = iocache_size(wp->ioc);
if (!iocache_capacity(wp->ioc)) {
if (iocache_size(wp->ioc) < MAX_IOCACHE_SIZE) {
/* double the size */
iocache_grow(wp->ioc, iocache_size(wp->ioc));
printf("Growing iocache for worker %d. sizes old/new %lu/%lu\n",
wp->pid, size, iocache_size(wp->ioc));
} else {
printf("iocache_size() for worker %d is already at max\n", wp->pid);
}
}
ret = iocache_read(wp->ioc, sd);
if (!ret) {
printf("Worker with pid %d seems to have crashed. Exiting\n", wp->pid);
exit(1);
}
if (ret < 0) {
printf("iocache_read() from worker %d returned %d: %m\n", wp->pid, ret);
return 0;
}
printf("read %d bytes from worker with pid %d::\n", ret, wp->pid);
while ((buf = worker_ioc2msg(wp->ioc, &size, 0))) {
int i, ret;
tot_bytes += size;
ret = worker_buf2kvvec_prealloc(&kvv, buf, (unsigned int)size, KVVEC_ASSIGN);
if (!ret < 0) {
printf("main: Failed to parse buffer of size %lu to key/value vector\n", size);
continue;
}
for (i = 0; i < kvv.kv_pairs; i++) {
struct key_value *kv = &kvv.kv[i];
if (!i && memcmp(kv->key, buf, kv->key_len)) {
printf("### kv[0]->key doesn't match buf. error in kvvec?\n");
}
printf("main: %2d.%02d: %s=%s\n", pkt, i, kv->key, kv->value);
}
pkt++;
}
printf("iocache: available: %lu; size: %lu; capacity: %lu\n",
iocache_available(wp->ioc), iocache_size(wp->ioc), iocache_capacity(wp->ioc));
printf("Got %d packets in %ld bytes (ret: %d)\n", pkt, tot_bytes, ret);
return 0;
}
/*
* Three cases to handle:
* - buf has data, iocache doesn't.
* - iocache has data, buf doesn't.
* - both buf and iocache has data.
*/
int iocache_sendto(iocache *ioc, int fd, char *buf, unsigned int len, int flags, const struct sockaddr *dest_addr, socklen_t addrlen)
{
int sent;
errno = 0;
if (!ioc)
return -1;
if (!ioc->ioc_buflen && !len)
return 0;
if (ioc->ioc_buf && iocache_available(ioc)) {
if (buf && len) {
/* copy buf and len to iocache buffer to use just one write */
if (iocache_capacity(ioc) < len) {
if (iocache_grow(ioc, iocache_size(ioc)) < 0)
return -1;
}
if (iocache_add(ioc, buf, len) < 0)
return -1;
}
buf = ioc->ioc_buf;
len = iocache_available(ioc);
}
sent = sendto(fd, buf, len, flags, dest_addr, addrlen);
if (sent < 1)
return -errno;
if (iocache_available(ioc))
iocache_use_size(ioc, sent);
return sent;
}
int iocache_grow(iocache *ioc, unsigned long increment)
{
return iocache_resize(ioc, iocache_size(ioc) + increment);
}
