glibc/sysdeps/unix/sysv/linux/select.c
YunQiang Su a9acb7b39e Define in_int32_t_range to check if the 64 bit time_t syscall should be used
Currently glibc uses in_time_t_range to detects time_t overflow,
and if it occurs fallbacks to 64 bit syscall version.

The function name is confusing because internally time_t might be
either 32 bits or 64 bits (depending on __TIMESIZE).

This patch refactors the in_time_t_range by replacing it with
in_int32_t_range for the case to check if the 64 bit time_t syscall
should be used.

The in_time_t range is used to detect overflow of the
syscall return value.

Reviewed-by: Adhemerval Zanella  <adhemerval.zanella@linaro.org>
2022-11-17 14:35:13 -03:00

143 lines
4.0 KiB
C

/* Linux select implementation.
Copyright (C) 2017-2022 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library. If not, see
<https://www.gnu.org/licenses/>. */
#include <sys/time.h>
#include <sys/types.h>
#include <sys/select.h>
#include <errno.h>
#include <sysdep-cancel.h>
/* Check the first NFDS descriptors each in READFDS (if not NULL) for read
readiness, in WRITEFDS (if not NULL) for write readiness, and in EXCEPTFDS
(if not NULL) for exceptional conditions. If TIMEOUT is not NULL, time out
after waiting the interval specified therein. Returns the number of ready
descriptors, or -1 for errors. */
int
__select64 (int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds,
struct __timeval64 *timeout)
{
__time64_t s = timeout != NULL ? timeout->tv_sec : 0;
int32_t us = timeout != NULL ? timeout->tv_usec : 0;
int32_t ns;
if (s < 0 || us < 0)
return INLINE_SYSCALL_ERROR_RETURN_VALUE (EINVAL);
/* Normalize the timeout, as legacy Linux __NR_select and __NR__newselect.
Different than syscall, it also handle possible overflow. */
if (us / USEC_PER_SEC > INT64_MAX - s)
{
s = INT64_MAX;
ns = NSEC_PER_SEC - 1;
}
else
{
s += us / USEC_PER_SEC;
us = us % USEC_PER_SEC;
ns = us * NSEC_PER_USEC;
}
struct __timespec64 ts64, *pts64 = NULL;
if (timeout != NULL)
{
ts64.tv_sec = s;
ts64.tv_nsec = ns;
pts64 = &ts64;
}
#ifndef __NR_pselect6_time64
# define __NR_pselect6_time64 __NR_pselect6
#endif
#ifdef __ASSUME_TIME64_SYSCALLS
int r = SYSCALL_CANCEL (pselect6_time64, nfds, readfds, writefds, exceptfds,
pts64, NULL);
if (timeout != NULL)
TIMESPEC_TO_TIMEVAL (timeout, pts64);
return r;
#else
bool need_time64 = timeout != NULL && !in_int32_t_range (timeout->tv_sec);
if (need_time64)
{
int r = SYSCALL_CANCEL (pselect6_time64, nfds, readfds, writefds,
exceptfds, pts64, NULL);
if ((r >= 0 || errno != ENOSYS) && timeout != NULL)
{
TIMESPEC_TO_TIMEVAL (timeout, &ts64);
}
else
__set_errno (EOVERFLOW);
return r;
}
# ifdef __ASSUME_PSELECT
struct timespec ts32, *pts32 = NULL;
if (pts64 != NULL)
{
ts32.tv_sec = pts64->tv_sec;
ts32.tv_nsec = pts64->tv_nsec;
pts32 = &ts32;
}
int r = SYSCALL_CANCEL (pselect6, nfds, readfds, writefds, exceptfds, pts32,
NULL);
if (timeout != NULL)
TIMESPEC_TO_TIMEVAL (timeout, pts32);
return r;
# else
struct timeval tv32, *ptv32 = NULL;
if (pts64 != NULL)
{
tv32 = valid_timespec64_to_timeval (*pts64);
ptv32 = &tv32;
}
int r = SYSCALL_CANCEL (_newselect, nfds, readfds, writefds, exceptfds, ptv32);
if (timeout != NULL)
*timeout = valid_timeval_to_timeval64 (tv32);
return r;
# endif /* __ASSUME_PSELECT */
#endif
}
#if __TIMESIZE != 64
libc_hidden_def (__select64)
int
__select (int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds,
struct timeval *timeout)
{
struct __timeval64 tv64, *ptv64 = NULL;
if (timeout != NULL)
{
tv64 = valid_timeval_to_timeval64 (*timeout);
ptv64 = &tv64;
}
int r = __select64 (nfds, readfds, writefds, exceptfds, ptv64);
if (timeout != NULL)
/* The remanining timeout will be always less the input TIMEOUT. */
*timeout = valid_timeval64_to_timeval (tv64);
return r;
}
#endif
libc_hidden_def (__select)
weak_alias (__select, select)
weak_alias (__select, __libc_select)