TLA  Line data    Source code

       1                 : /*-------------------------------------------------------------------------
       2                 :  *
       3                 :  * noblock.c
       4                 :  *    set a file descriptor as blocking or non-blocking
       5                 :  *
       6                 :  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
       7                 :  * Portions Copyright (c) 1994, Regents of the University of California
       8                 :  *
       9                 :  * IDENTIFICATION
      10                 :  *    src/port/noblock.c
      11                 :  *
      12                 :  *-------------------------------------------------------------------------
      13                 :  */
      14                 : 
      15                 : #include "c.h"
      16                 : 
      17                 : #include <fcntl.h>
      18                 : 
      19                 : 
      20                 : /*
      21                 :  * Put socket into nonblock mode.
      22                 :  * Returns true on success, false on failure.
      23                 :  */
      24                 : bool
      25 CBC        8773 : pg_set_noblock(pgsocket sock)
      26                 : {
      27                 : #if !defined(WIN32)
      28                 :     int         flags;
      29                 : 
      30            8773 :     flags = fcntl(sock, F_GETFL);
      31            8773 :     if (flags < 0)
      32 UBC           0 :         return false;
      33 CBC        8773 :     if (fcntl(sock, F_SETFL, (flags | O_NONBLOCK)) == -1)
      34 UBC           0 :         return false;
      35 CBC        8773 :     return true;
      36                 : #else
      37                 :     unsigned long ioctlsocket_ret = 1;
      38                 : 
      39                 :     /* Returns non-0 on failure, while fcntl() returns -1 on failure */
      40                 :     return (ioctlsocket(sock, FIONBIO, &ioctlsocket_ret) == 0);
      41                 : #endif
      42                 : }
      43                 : 
      44                 : /*
      45                 :  * Put socket into blocking mode.
      46                 :  * Returns true on success, false on failure.
      47                 :  */
      48                 : bool
      49 UBC           0 : pg_set_block(pgsocket sock)
      50                 : {
      51                 : #if !defined(WIN32)
      52                 :     int         flags;
      53                 : 
      54               0 :     flags = fcntl(sock, F_GETFL);
      55               0 :     if (flags < 0)
      56               0 :         return false;
      57               0 :     if (fcntl(sock, F_SETFL, (flags & ~O_NONBLOCK)) == -1)
      58               0 :         return false;
      59               0 :     return true;
      60                 : #else
      61                 :     unsigned long ioctlsocket_ret = 0;
      62                 : 
      63                 :     /* Returns non-0 on failure, while fcntl() returns -1 on failure */
      64                 :     return (ioctlsocket(sock, FIONBIO, &ioctlsocket_ret) == 0);
      65                 : #endif
      66                 : }