haiwen/libsearpc
1
0
Fork 0
mirror of https://github.com/haiwen/libsearpc.git synced 2025-09-24 11:47:22 +00:00
Code Issues Releases Wiki Activity

Support write with non-blocking

Browse Source
This commit is contained in:
杨赫然
2024-09-23 16:53:00 +08:00
parent ed12cfba81
commit 986ddf5e4f
1 changed files with 33 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
lib/searpc-named-pipe-transport.c
View File

@@ -240,6 +240,37 @@ epoll_read_n(int fd, void *vptr, size_t n)
return(n - nleft); /* return >= 0 */
}
// This function will keep writing until the following conditions are met:
// 1. the socket has beed closed.
// 2. the requested size has been writed;
// 3. an unrecoverable error has been encountered;
// The first two cases are not errors.
gssize
epoll_write_n(int fd, const void *vptr, size_t n)
{
size_t nleft;
gssize nwritten;
const char *ptr;
ptr = vptr;
nleft = n;
while (nleft > 0) {
if ( (nwritten = write(fd, ptr, nleft)) <= 0)
{
if (errno == EAGAIN)
nwritten = 0; /* and call write() again */
else if (nwritten < 0 && errno == EINTR)
nwritten = 0; /* and call write() again */
else
return(-1); /* error */
}
nleft -= nwritten;
ptr += nwritten;
}
return(n);
}
static void epoll_handler(void *data)
{
ServerHandlerData *handler_data = data;
@@ -262,12 +293,12 @@ static void epoll_handler(void *data)
g_free (body);
len = (guint32)ret_len;
if (pipe_write_n(connfd, &len, sizeof(guint32)) < 0) {
if (epoll_write_n(connfd, &len, sizeof(guint32)) < 0) {
ret = -1;
goto out;
}
if (pipe_write_n(connfd, ret_str, ret_len) < 0) {
if (epoll_write_n(connfd, ret_str, ret_len) < 0) {
ret = -1;
goto out;
}