int proto_common_recv(int sock, unsigned char *data, size_t size, int *fdp) { ssize_t done; PJDLOG_ASSERT(sock >= 0); if (data == NULL) { /* The caller is just trying to decide about direction. */ PJDLOG_ASSERT(size == 0); if (shutdown(sock, SHUT_WR) == -1) return (errno); return (0); } PJDLOG_ASSERT(data != NULL); PJDLOG_ASSERT(size > 0); do { done = recv(sock, data, size, MSG_WAITALL); } while (done == -1 && errno == EINTR); if (done == 0) { return (ENOTCONN); } else if (done == -1) { /* * If this is blocking socket and we got EAGAIN, this * means the request timed out. Translate errno to * ETIMEDOUT, to give administrator a hint to * eventually increase timeout. */ if (errno == EAGAIN && blocking_socket(sock)) errno = ETIMEDOUT; return (errno); } if (fdp == NULL) return (0); return (proto_descriptor_recv(sock, fdp)); }
int proto_common_send(int sock, const unsigned char *data, size_t size, int fd) { ssize_t done; size_t sendsize; int errcount = 0; PJDLOG_ASSERT(sock >= 0); if (data == NULL) { /* The caller is just trying to decide about direction. */ PJDLOG_ASSERT(size == 0); if (shutdown(sock, SHUT_RD) == -1) return (errno); return (0); } PJDLOG_ASSERT(data != NULL); PJDLOG_ASSERT(size > 0); do { sendsize = size < MAX_SEND_SIZE ? size : MAX_SEND_SIZE; done = send(sock, data, sendsize, MSG_NOSIGNAL); if (done == 0) { return (ENOTCONN); } else if (done == -1) { if (errno == EINTR) continue; if (errno == ENOBUFS) { /* * If there are no buffers we retry. * After each try we increase delay before the * next one and we give up after fifteen times. * This gives 11s of total wait time. */ if (errcount == 15) { pjdlog_warning("Getting ENOBUFS errors for 11s on send(), giving up."); } else { if (errcount == 0) pjdlog_warning("Got ENOBUFS error on send(), retrying for a bit."); errcount++; usleep(100000 * errcount); continue; } } /* * If this is blocking socket and we got EAGAIN, this * means the request timed out. Translate errno to * ETIMEDOUT, to give administrator a hint to * eventually increase timeout. */ if (errno == EAGAIN && blocking_socket(sock)) errno = ETIMEDOUT; return (errno); } data += done; size -= done; } while (size > 0); if (errcount > 0) { pjdlog_info("Data sent successfully after %d ENOBUFS error%s.", errcount, errcount == 1 ? "" : "s"); } if (fd == -1) return (0); return (proto_descriptor_send(sock, fd)); }
/*==========================================================================*/ void Socket::enableBlocking() { blocking_socket( this->id() ); }