1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#include <stdio.h>
#include <sys/socket.h>
#include <dlfcn.h>
#include <netinet/tcp.h>
#include <netinet/in.h>
int socket(int domain, int type, int protocol){
static int (*real_socket) (int, int, int) = NULL;
if(!real_socket){
real_socket = dlsym(RTLD_NEXT, "socket");
}
int fd = real_socket(domain, type, protocol);
int enable = 1;
if(domain == AF_INET && type == SOCK_STREAM){
setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &enable, sizeof(enable));
}
return fd;
}
ssize_t send(int sockfd, const void* buf, size_t len, int flags){
static ssize_t (*real_send) (int, void*, size_t, int) = NULL;
if(!real_send){
real_send = dlsym(RTLD_NEXT, "send");
}
ssize_t rv = 0;
for(size_t u = 0; u < len; u++){
ssize_t sent = real_send(sockfd, buf + u, 1, flags);
if(sent < 0){
return -1;
}
rv += sent;
}
return rv;
}
|
