lib/util/blocking.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Jeremy Allison 2001-2002
   6    Copyright (C) Simo Sorce 2001
   7    Copyright (C) Jim McDonough (jmcd@us.ibm.com)  2003.
   8    Copyright (C) James J Myers 2003
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "replace.h"
  25 #include "system/network.h"
  26 #include "system/filesys.h"
  27 #include "system/locale.h"
  28 #include "blocking.h"
  29 #undef malloc
  30 #undef strcasecmp
  31 #undef strncasecmp
  32 #undef strdup
  33 #undef realloc
  34
  35 /**
  36  Set a fd into blocking/nonblocking mode. Uses POSIX O_NONBLOCK if available,
  37  else
  38   if SYSV use O_NDELAY
  39   if BSD use FNDELAY
  40 **/
  41
  42 _PUBLIC_ int set_blocking(int fd, bool set)
  43 {
  44         int val;
  45 #ifdef O_NONBLOCK
  46 #define FLAG_TO_SET O_NONBLOCK
  47 #else
  48 #ifdef SYSV
  49 #define FLAG_TO_SET O_NDELAY
  50 #else /* BSD */
  51 #define FLAG_TO_SET FNDELAY
  52 #endif
  53 #endif
  54
  55         if((val = fcntl(fd, F_GETFL, 0)) == -1)
  56                 return -1;
  57         if(set) /* Turn blocking on - ie. clear nonblock flag */
  58                 val &= ~FLAG_TO_SET;
  59         else
  60                 val |= FLAG_TO_SET;
  61         return fcntl( fd, F_SETFL, val);
  62 #undef FLAG_TO_SET
  63 }
  64
  65
  66 _PUBLIC_ bool smb_set_close_on_exec(int fd)
  67 {
  68 #ifdef FD_CLOEXEC
  69         int val;
  70
  71         val = fcntl(fd, F_GETFD, 0);
  72         if (val >= 0) {
  73                 val |= FD_CLOEXEC;
  74                 val = fcntl(fd, F_SETFD, val);
  75                 if (val != -1) {
  76                         return true;
  77                 }
  78         }
  79 #endif
  80         return false;
  81 }