diff options
author | Denys Vlasenko | 2017-02-16 16:27:39 +0100 |
---|---|---|
committer | Denys Vlasenko | 2017-02-16 16:51:18 +0100 |
commit | 0ec4d08ea357e680506d9f7df90fa1e53a180509 (patch) | |
tree | 15f4db9dd6b089adfe3e3efdf5831c8ea25aa0d9 | |
parent | c39ee047051fb36f4db907c037e22db586490f42 (diff) | |
download | busybox-0ec4d08ea357e680506d9f7df90fa1e53a180509.zip busybox-0ec4d08ea357e680506d9f7df90fa1e53a180509.tar.gz |
tls: covert i/o loop from using select() to poll()
function old new delta
tls_run_copy_loop 377 282 -95
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
-rw-r--r-- | networking/tls.c | 25 |
1 files changed, 11 insertions, 14 deletions
diff --git a/networking/tls.c b/networking/tls.c index 30afd9e..cb6b094 100644 --- a/networking/tls.c +++ b/networking/tls.c @@ -1740,26 +1740,23 @@ static void tls_xwrite(tls_state_t *tls, int len) void FAST_FUNC tls_run_copy_loop(tls_state_t *tls) { - fd_set readfds; int inbuf_size; const int INBUF_STEP = 4 * 1024; + struct pollfd pfds[2]; -//TODO: convert to poll - /* Select loop copying stdin to ofd, and ifd to stdout */ - FD_ZERO(&readfds); - FD_SET(tls->ifd, &readfds); - FD_SET(STDIN_FILENO, &readfds); + pfds[0].fd = STDIN_FILENO; + pfds[0].events = POLLIN; + pfds[1].fd = tls->ifd; + pfds[1].events = POLLIN; inbuf_size = INBUF_STEP; for (;;) { - fd_set testfds; int nread; - testfds = readfds; - if (select(tls->ifd + 1, &testfds, NULL, NULL, NULL) < 0) - bb_perror_msg_and_die("select"); + if (safe_poll(pfds, 2, -1) < 0) + bb_perror_msg_and_die("poll"); - if (FD_ISSET(STDIN_FILENO, &testfds)) { + if (pfds[0].revents) { void *buf; dbg("STDIN HAS DATA\n"); @@ -1774,7 +1771,7 @@ void FAST_FUNC tls_run_copy_loop(tls_state_t *tls) /* But TLS has no way to encode this, * doubt it's ok to do it "raw" */ - FD_CLR(STDIN_FILENO, &readfds); + pfds[0].fd = -1; tls_free_outbuf(tls); /* mem usage optimization */ } else { if (nread == inbuf_size) { @@ -1788,7 +1785,7 @@ void FAST_FUNC tls_run_copy_loop(tls_state_t *tls) tls_xwrite(tls, nread); } } - if (FD_ISSET(tls->ifd, &testfds)) { + if (pfds[1].revents) { dbg("NETWORK HAS DATA\n"); read_record: nread = tls_xread_record(tls); @@ -1796,7 +1793,7 @@ void FAST_FUNC tls_run_copy_loop(tls_state_t *tls) /* TLS protocol has no real concept of one-sided shutdowns: * if we get "TLS EOF" from the peer, writes will fail too */ - //FD_CLR(tls->ifd, &readfds); + //pfds[1].fd = -1; //close(STDOUT_FILENO); //tls_free_inbuf(tls); /* mem usage optimization */ //continue; |