void rr_enqueue_internal(rr_dev device, rr_prio priority, void *cbdata, const char *block, size_t nbytes, long long line) { blocknode *node = malloc(sizeof(blocknode)); node->next = NULL; node->cbdata = cbdata; node->block = malloc(nbytes); memcpy(node->block, block, nbytes); node->blocksize = nbytes; node->line = line; if(!device->sendhead[priority]) { device->sendhead[priority] = node; device->sendtail[priority] = node; device->want_writable(device, device->ww_data, 1); } else { device->sendtail[priority]->next = node; device->sendtail[priority] = node; } }
int rr_handle_writable(rr_dev device) { ssize_t result; if(device->sendbuf_fill == 0) { /* Last block is gone; prepare to send a new block */ int prio; blocknode *node = NULL; for(prio = RR_PRIO_COUNT - 1; prio >= 0; --prio) { node = device->sendhead[prio]; if(node) { /* We have a block to send! Get it ready. */ device->bytes_sent = 0; result = fmtblock(device, node); if(result < 0) { /* FIXME: This will confuse code expecting errno to be set */ return result; } device->sendbuf_fill = result; device->sending_prio = prio; break; } } if(!node) { /* No data to write */ device->want_writable(device, device->ww_data, 0); return 0; } } /* Perform write */ do { result = write(device->fd, device->sendbuf + device->bytes_sent, device->sendbuf_fill - device->bytes_sent); } while(result < 0 && errno == EINTR); if(result < 0) { return result; } device->bytes_sent += result; if(device->bytes_sent == device->sendbuf_fill) { /* We've sent the complete block. */ blocknode *node = device->sendhead[device->sending_prio]; if(device->onsend) { device->onsend(device, device->onsend_data, node->cbdata, device->sendbuf, device->sendbuf_fill); } device->sendhead[device->sending_prio] = node->next; node->line = device->lineno; ++(device->lineno); /* Update sent cache */ if(device->sentcache[device->sentcachesize - 1]) { blocknode_free(device->sentcache[device->sentcachesize - 1]); } ssize_t i; for(i = device->sentcachesize - 1; i > 0 ; --i) { device->sentcache[i] = device->sentcache[i-1]; } device->sentcache[0] = node; /* Indicate that we're ready for the next. */ device->sendbuf_fill = 0; } return result; }