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); }