diff options
author | deraadt@openbsd.org <deraadt@openbsd.org> | 2021-11-13 21:14:13 +0000 |
---|---|---|
committer | Damien Miller <djm@mindrot.org> | 2021-11-18 09:09:59 +1100 |
commit | 6582a31c388968f4073af2bd8621880735c3d42b (patch) | |
tree | da5704116e32132963b6fc5e317784db44725a33 /packet.c | |
parent | 7c025c005550c86a40200a2bcdd355d09413d61a (diff) |
upstream: replace select() with ppoll(), including converting
timeval's to timespec's to make things easier. back and forth and ok; djm
OpenBSD-Commit-ID: 89d3b23c60875da919e7820f9de6213286ffbec9
Diffstat (limited to 'packet.c')
-rw-r--r-- | packet.c | 59 |
1 files changed, 20 insertions, 39 deletions
@@ -1,4 +1,4 @@ -/* $OpenBSD: packet.c,v 1.301 2021/07/16 09:00:23 djm Exp $ */ +/* $OpenBSD: packet.c,v 1.302 2021/11/13 21:14:13 deraadt Exp $ */ /* * Author: Tatu Ylonen <ylo@cs.hut.fi> * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland @@ -1325,17 +1325,13 @@ ssh_packet_read_seqnr(struct ssh *ssh, u_char *typep, u_int32_t *seqnr_p) { struct session_state *state = ssh->state; int len, r, ms_remain; - fd_set *setp; + struct pollfd pfd; char buf[8192]; - struct timeval timeout, start, *timeoutp = NULL; + struct timeval start; + struct timespec timespec, *timespecp = NULL; DBG(debug("packet_read()")); - setp = calloc(howmany(state->connection_in + 1, - NFDBITS), sizeof(fd_mask)); - if (setp == NULL) - return SSH_ERR_ALLOC_FAIL; - /* * Since we are blocking, ensure that all written packets have * been sent. @@ -1356,22 +1352,20 @@ ssh_packet_read_seqnr(struct ssh *ssh, u_char *typep, u_int32_t *seqnr_p) * Otherwise, wait for some data to arrive, add it to the * buffer, and try again. */ - memset(setp, 0, howmany(state->connection_in + 1, - NFDBITS) * sizeof(fd_mask)); - FD_SET(state->connection_in, setp); + pfd.fd = state->connection_in; + pfd.events = POLLIN; if (state->packet_timeout_ms > 0) { ms_remain = state->packet_timeout_ms; - timeoutp = &timeout; + timespecp = ×pec; } /* Wait for some data to arrive. */ for (;;) { if (state->packet_timeout_ms > 0) { - ms_to_timeval(&timeout, ms_remain); + ms_to_timespec(×pec, ms_remain); monotime_tv(&start); } - if ((r = select(state->connection_in + 1, setp, - NULL, NULL, timeoutp)) >= 0) + if ((r = ppoll(&pfd, 1, timespecp, NULL)) >= 0) break; if (errno != EAGAIN && errno != EINTR && errno != EWOULDBLOCK) { @@ -1406,7 +1400,6 @@ ssh_packet_read_seqnr(struct ssh *ssh, u_char *typep, u_int32_t *seqnr_p) goto out; } out: - free(setp); return r; } @@ -1986,35 +1979,28 @@ ssh_packet_write_poll(struct ssh *ssh) int ssh_packet_write_wait(struct ssh *ssh) { - fd_set *setp; int ret, r, ms_remain = 0; - struct timeval start, timeout, *timeoutp = NULL; + struct timeval start; + struct timespec timespec, *timespecp = NULL; struct session_state *state = ssh->state; + struct pollfd pfd; - setp = calloc(howmany(state->connection_out + 1, - NFDBITS), sizeof(fd_mask)); - if (setp == NULL) - return SSH_ERR_ALLOC_FAIL; - if ((r = ssh_packet_write_poll(ssh)) != 0) { - free(setp); + if ((r = ssh_packet_write_poll(ssh)) != 0) return r; - } while (ssh_packet_have_data_to_write(ssh)) { - memset(setp, 0, howmany(state->connection_out + 1, - NFDBITS) * sizeof(fd_mask)); - FD_SET(state->connection_out, setp); + pfd.fd = state->connection_out; + pfd.events = POLLOUT; if (state->packet_timeout_ms > 0) { ms_remain = state->packet_timeout_ms; - timeoutp = &timeout; + timespecp = ×pec; } for (;;) { if (state->packet_timeout_ms > 0) { - ms_to_timeval(&timeout, ms_remain); + ms_to_timespec(×pec, ms_remain); monotime_tv(&start); } - if ((ret = select(state->connection_out + 1, - NULL, setp, NULL, timeoutp)) >= 0) + if ((ret = ppoll(&pfd, 1, timespecp, NULL)) >= 0) break; if (errno != EAGAIN && errno != EINTR && errno != EWOULDBLOCK) @@ -2027,16 +2013,11 @@ ssh_packet_write_wait(struct ssh *ssh) break; } } - if (ret == 0) { - free(setp); + if (ret == 0) return SSH_ERR_CONN_TIMEOUT; - } - if ((r = ssh_packet_write_poll(ssh)) != 0) { - free(setp); + if ((r = ssh_packet_write_poll(ssh)) != 0) return r; - } } - free(setp); return 0; } |