-/* $OpenBSD: kex.c,v 1.178 2023/03/12 10:40:39 dtucker Exp $ */
+/* $OpenBSD: kex.c,v 1.179 2023/08/18 01:37:41 djm Exp $ */
/*
* Copyright (c) 2000, 2001 Markus Friedl. All rights reserved.
*
for (;;) {
if (timeout_ms > 0) {
r = waitrfd(ssh_packet_get_connection_in(ssh),
- &timeout_ms);
+ &timeout_ms, NULL);
if (r == -1 && errno == ETIMEDOUT) {
send_error(ssh, "Timed out waiting "
"for SSH identification string.");
-/* $OpenBSD: misc.c,v 1.185 2023/08/04 06:32:40 dtucker Exp $ */
+/* $OpenBSD: misc.c,v 1.186 2023/08/18 01:37:41 djm Exp $ */
/*
* Copyright (c) 2000 Markus Friedl. All rights reserved.
* Copyright (c) 2005-2020 Damien Miller. All rights reserved.
* Returns 0 if fd ready or -1 on timeout or error (see errno).
*/
static int
-waitfd(int fd, int *timeoutp, short events)
+waitfd(int fd, int *timeoutp, short events, volatile sig_atomic_t *stop)
{
struct pollfd pfd;
- struct timeval t_start;
- int oerrno, r, have_timeout = (*timeoutp >= 0);
+ struct timespec timeout;
+ int oerrno, r;
+ sigset_t nsigset, osigset;
+ if (timeoutp && *timeoutp == -1)
+ timeoutp = NULL;
pfd.fd = fd;
pfd.events = events;
- for (; !have_timeout || *timeoutp >= 0;) {
- monotime_tv(&t_start);
- r = poll(&pfd, 1, *timeoutp);
+ ptimeout_init(&timeout);
+ if (timeoutp != NULL)
+ ptimeout_deadline_ms(&timeout, *timeoutp);
+ if (stop != NULL)
+ sigfillset(&nsigset);
+ for (; timeoutp == NULL || *timeoutp >= 0;) {
+ if (stop != NULL) {
+ sigprocmask(SIG_BLOCK, &nsigset, &osigset);
+ if (*stop) {
+ sigprocmask(SIG_SETMASK, &osigset, NULL);
+ errno = EINTR;
+ return -1;
+ }
+ }
+ r = ppoll(&pfd, 1, ptimeout_get_tsp(&timeout),
+ stop != NULL ? &osigset : NULL);
oerrno = errno;
- if (have_timeout)
- ms_subtract_diff(&t_start, timeoutp);
+ if (stop != NULL)
+ sigprocmask(SIG_SETMASK, &osigset, NULL);
+ if (timeoutp)
+ *timeoutp = ptimeout_get_ms(&timeout);
errno = oerrno;
if (r > 0)
return 0;
* Returns 0 if fd ready or -1 on timeout or error (see errno).
*/
int
-waitrfd(int fd, int *timeoutp) {
- return waitfd(fd, timeoutp, POLLIN);
+waitrfd(int fd, int *timeoutp, volatile sig_atomic_t *stop) {
+ return waitfd(fd, timeoutp, POLLIN, stop);
}
/*
break;
}
- if (waitfd(sockfd, timeoutp, POLLIN | POLLOUT) == -1)
+ if (waitfd(sockfd, timeoutp, POLLIN | POLLOUT, NULL) == -1)
return -1;
/* Completed or failed */
-/* $OpenBSD: misc.h,v 1.103 2023/07/19 14:02:27 djm Exp $ */
+/* $OpenBSD: misc.h,v 1.104 2023/08/18 01:37:41 djm Exp $ */
/*
* Author: Tatu Ylonen <ylo@cs.hut.fi>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
+#include <signal.h>
/* Data structure for representing a forwarding request. */
struct Forward {
int set_rdomain(int, const char *);
int get_sock_af(int);
void set_sock_tos(int, int);
-int waitrfd(int, int *);
+int waitrfd(int, int *, volatile sig_atomic_t *);
int timeout_connect(int, const struct sockaddr *, socklen_t, int *);
int a2port(const char *);
int a2tun(const char *, int *);