69 lines
1.4 KiB
C
69 lines
1.4 KiB
C
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include "ucore/fd_utils.h"
|
|
|
|
int set_nonblocking(int fd)
|
|
{
|
|
int flags = fcntl(fd,F_GETFL,NULL);
|
|
if(flags < 0 ) {
|
|
return flags;
|
|
}
|
|
return fcntl(fd,F_SETFL,flags | O_NONBLOCK);
|
|
}
|
|
|
|
int set_blocking(int fd)
|
|
{
|
|
int flags = fcntl(fd,F_GETFL,NULL);
|
|
if(flags < 0 ) {
|
|
return flags;
|
|
}
|
|
|
|
return fcntl(fd,F_SETFL,flags&~O_NONBLOCK);
|
|
}
|
|
|
|
int set_reuseaddr(int sockfd)
|
|
{
|
|
int opt = 1;
|
|
return setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt));
|
|
|
|
}
|
|
|
|
int set_tcp_keepalive(int sockfd)
|
|
{
|
|
int optval = 1;
|
|
return setsockopt(sockfd, SOL_SOCKET, SO_KEEPALIVE, &optval, sizeof(optval));
|
|
}
|
|
|
|
|
|
int set_send_timeout(int sockfd,long timeoutms)
|
|
{
|
|
struct timeval so_sndtimeo = {timeoutms/1000,(timeoutms%1000)*1000};
|
|
|
|
return setsockopt(sockfd, SOL_SOCKET, SO_SNDTIMEO, &so_sndtimeo, sizeof(so_sndtimeo));
|
|
}
|
|
|
|
int set_receive_timeout(int sockfd,long timeoutms)
|
|
{
|
|
struct timeval so_sndtimeo = {timeoutms/1000,(timeoutms%1000)*1000};
|
|
|
|
return setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &so_sndtimeo, sizeof(so_sndtimeo));
|
|
}
|
|
int
|
|
set_ip_dscp(int fd, unsigned int dscp) {
|
|
int val;
|
|
|
|
if (dscp > 63) {
|
|
errno = EINVAL;
|
|
return -1;
|
|
}
|
|
|
|
val = dscp << 2;
|
|
return setsockopt(fd, IPPROTO_IP, IP_TOS, &val, sizeof val);
|
|
}
|
|
|
|
|