From d48659dd4583859135f671a12160f35d3307d243 Mon Sep 17 00:00:00 2001 From: hantianfeng Date: Thu, 15 Aug 2024 20:12:58 +0800 Subject: [PATCH] fix tests[7] --- include/swoole_socket.h | 12 +++++------- src/network/socket.cc | 11 +++++++++++ src/server/master.cc | 2 +- 3 files changed, 17 insertions(+), 8 deletions(-) diff --git a/include/swoole_socket.h b/include/swoole_socket.h index c43e27996e6..2bba34c7cc6 100644 --- a/include/swoole_socket.h +++ b/include/swoole_socket.h @@ -492,13 +492,11 @@ struct Socket { return ::read(fd, __buf, __len); } - ssize_t read_sync(void *__buf, size_t __len, int timeout_ms = -1) { - if (wait_event(timeout_ms, SW_EVENT_READ) == SW_OK) { - return read(__buf, __len); - } else { - return -1; - } - } + /** + * Read data from the socket synchronously without setting non-blocking or blocking IO, + * and allow interruptions by signals. + */ + ssize_t read_sync(void *__buf, size_t __len, int timeout_ms = -1); int shutdown(int __how) { return ::shutdown(fd, __how); diff --git a/src/network/socket.cc b/src/network/socket.cc index eac5107f3e9..c56ef63936c 100644 --- a/src/network/socket.cc +++ b/src/network/socket.cc @@ -773,6 +773,17 @@ ssize_t Socket::send_async(const void *__buf, size_t __n) { } } +ssize_t Socket::read_sync(void *__buf, size_t __len, int timeout_ms) { + struct pollfd event; + event.fd = fd; + event.events = POLLIN; + if (poll(&event, 1, timeout_ms) == 1) { + return read(__buf, __len); + } else { + return -1; + } +} + ssize_t Socket::readv(IOVector *io_vector) { ssize_t retval; diff --git a/src/server/master.cc b/src/server/master.cc index 88bb944ebf4..fa0e35c3cee 100644 --- a/src/server/master.cc +++ b/src/server/master.cc @@ -1841,7 +1841,7 @@ ListenPort *Server::add_port(SocketType type, const char *host, int port) { } void Server::master_signal_handler(int signo) { - swoole_trace_log(SW_TRACE_SERVER, "signal[%d] %s triggered in %d", sig, swoole_signal_to_str(sig), getpid()); + swoole_trace_log(SW_TRACE_SERVER, "signal[%d] %s triggered in %d", signo, swoole_signal_to_str(signo), getpid()); Server *serv = sw_server(); if (!SwooleG.running || !serv || !serv->is_running()) {