cuberite/libevent
1
0
Fork 0
mirror of https://github.com/cuberite/libevent.git synced 2025-09-08 03:44:22 -04:00
Code Issues Packages Projects Releases Wiki Activity

Improve the behavior of le-proxy in a few cases.

Browse Source 
svn:r1458
This commit is contained in:
Nick Mathewson 2009-10-23 17:40:00 +00:00
parent fdd11c00bb
commit c119e4a13f
1 changed files with 47 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
sample/le-proxy.c
View File

@ -26,25 +26,56 @@ static int use_wrapper = 1;
static SSL_CTX *ssl_ctx = NULL; static SSL_CTX *ssl_ctx = NULL;
#define MAX_OUTPUT (512*1024)
static void drained_writecb(struct bufferevent *bev, void *ctx);
static void eventcb(struct bufferevent *bev, short what, void *ctx);
static void static void
readcb(struct bufferevent *bev, void *ctx) readcb(struct bufferevent *bev, void *ctx)
{ {
struct bufferevent *partner = ctx; struct bufferevent *partner = ctx;
struct evbuffer *src, *dst; struct evbuffer *src, *dst;
size_t len;
src = bufferevent_get_input(bev); src = bufferevent_get_input(bev);
len = evbuffer_get_length(src);
if (!partner) { if (!partner) {
evbuffer_drain(src, evbuffer_get_length(src)); evbuffer_drain(src, len);
return; return;
} }
dst = bufferevent_get_output(partner); dst = bufferevent_get_output(partner);
evbuffer_add_buffer(dst, src); evbuffer_add_buffer(dst, src);
if (evbuffer_get_length(dst) >= MAX_OUTPUT) {
/* We're giving the other side data faster than it can
* pass it on. Stop reading here until we have drained the
* other side to MAX_OUTPUT/2 bytes. */
bufferevent_setcb(partner, readcb, drained_writecb,
eventcb, bev);
bufferevent_setwatermark(partner, EV_WRITE, MAX_OUTPUT/2,
MAX_OUTPUT);
bufferevent_disable(bev, EV_READ);
}
}
static void
drained_writecb(struct bufferevent *bev, void *ctx)
{
struct bufferevent *partner = ctx;
/* We were choking the other side until we drained our outbuf a bit.
* Now it seems drained. */
bufferevent_setcb(bev, readcb, NULL, eventcb, partner);
bufferevent_setwatermark(bev, EV_WRITE, 0, 0);
if (partner)
bufferevent_enable(partner, EV_READ);
} }
static void static void
close_on_finished_writecb(struct bufferevent *bev, void *ctx) close_on_finished_writecb(struct bufferevent *bev, void *ctx)
{ {
struct evbuffer *b = bufferevent_get_output(bev); struct evbuffer *b = bufferevent_get_output(bev);
if (evbuffer_get_length(b) == 0) { if (evbuffer_get_length(b) == 0) {
bufferevent_free(bev); bufferevent_free(bev);
} }
@ -63,10 +94,20 @@ eventcb(struct bufferevent *bev, short what, void *ctx)
/* Flush all pending data */ /* Flush all pending data */
readcb(bev, ctx); readcb(bev, ctx);
/* Disconnect this one from the partner. */ if (evbuffer_get_length(
bufferevent_setcb(partner, bufferevent_get_output(partner))) {
NULL, close_on_finished_writecb, /* We still have to flush data from the other
eventcb, NULL); * side, but when that's done, close the other
* side. */
bufferevent_setcb(partner,
NULL, close_on_finished_writecb,
eventcb, NULL);
bufferevent_disable(partner, EV_READ);
} else {
/* We have nothing left to say to the other
* side; close it. */
bufferevent_free(partner);
}
} }
bufferevent_free(bev); bufferevent_free(bev);
} }