67 lines
1014 B
C
67 lines
1014 B
C
#include "core/socket.h"
|
|
|
|
#include <errno.h>
|
|
#include <sys/socket.h>
|
|
|
|
int cws_socket_read(int sockfd, string_s *str) {
|
|
char tmp[4096] = {0};
|
|
|
|
for (;;) {
|
|
ssize_t n = recv(sockfd, tmp, sizeof tmp, 0);
|
|
|
|
/* We have some data */
|
|
if (n > 0) {
|
|
tmp[n] = '\0';
|
|
string_append(str, tmp);
|
|
return n;
|
|
}
|
|
|
|
/* Client closed */
|
|
if (n == 0) {
|
|
return 0;
|
|
}
|
|
|
|
if (errno == EINTR) {
|
|
continue;
|
|
}
|
|
|
|
/* No data now */
|
|
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
|
return -2;
|
|
}
|
|
|
|
/* Something happened */
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int cws_socket_send(int sockfd, const char *buffer, size_t len, int flags) {
|
|
size_t total_sent = 0;
|
|
|
|
while (total_sent < len) {
|
|
ssize_t n = send(sockfd, buffer + total_sent, len - total_sent, flags);
|
|
|
|
if (n > 0) {
|
|
total_sent += (size_t)n;
|
|
continue;
|
|
}
|
|
|
|
if (n == 0) {
|
|
break;
|
|
}
|
|
|
|
if (errno == EINTR) {
|
|
continue;
|
|
}
|
|
|
|
/* Partial write */
|
|
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
|
break;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
return (ssize_t)total_sent;
|
|
}
|