include/share.h

   1 /* Copyright (C) 2009  Pierre-Marc Fournier
   2  *
   3  * This library is free software; you can redistribute it and/or
   4  * modify it under the terms of the GNU Lesser General Public
   5  * License as published by the Free Software Foundation; either
   6  * version 2.1 of the License, or (at your option) any later version.
   7  *
   8  * This library is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * Lesser General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU Lesser General Public
  14  * License along with this library; if not, write to the Free Software
  15  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
  16  */
  17
  18 #ifndef UST_SHARE_H
  19 #define UST_SHARE_H
  20
  21 /* write() */
  22 #include <unistd.h>
  23
  24 /* send() */
  25 #include <sys/types.h>
  26 #include <sys/socket.h>
  27
  28 #include <errno.h>
  29
  30 /* This write is patient because it restarts if it was incomplete.
  31  */
  32
  33 static __inline__ ssize_t patient_write(int fd, const void *buf, size_t count)
  34 {
  35         const char *bufc = (const char *) buf;
  36         int result;
  37
  38         for(;;) {
  39                 result = write(fd, bufc, count);
  40                 if(result == -1 && errno == EINTR) {
  41                         continue;
  42                 }
  43                 if(result <= 0) {
  44                         return result;
  45                 }
  46                 count -= result;
  47                 bufc += result;
  48
  49                 if(count == 0) {
  50                         break;
  51                 }
  52         }
  53
  54         return bufc-(const char *)buf;
  55 }
  56
  57 static __inline__ ssize_t patient_send(int fd, const void *buf, size_t count, int flags)
  58 {
  59         const char *bufc = (const char *) buf;
  60         int result;
  61
  62         for(;;) {
  63                 result = send(fd, bufc, count, flags);
  64                 if(result == -1 && errno == EINTR) {
  65                         continue;
  66                 }
  67                 if(result <= 0) {
  68                         return result;
  69                 }
  70                 count -= result;
  71                 bufc += result;
  72
  73                 if(count == 0) {
  74                         break;
  75                 }
  76         }
  77
  78         return bufc-(const char *)buf;
  79 }
  80
  81 #endif /* UST_SHARE_H */