150 lines
3.5 KiB
C++
150 lines
3.5 KiB
C++
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <fcntl.h>
|
|
|
|
bool socketError = false;
|
|
bool socketBlocking = false;
|
|
|
|
int DDV_OpenUnix(const char adres[], bool nonblock = false){
|
|
int s = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
int on = 1;
|
|
setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
|
|
sockaddr_un addr;
|
|
addr.sun_family = AF_UNIX;
|
|
strcpy(addr.sun_path, adres);
|
|
int r = connect(s, (sockaddr*)&addr, sizeof(addr));
|
|
if (r == 0){
|
|
if (nonblock){
|
|
int flags = fcntl(s, F_GETFL, 0);
|
|
flags |= O_NONBLOCK;
|
|
fcntl(s, F_SETFL, flags);
|
|
}
|
|
return s;
|
|
}else{
|
|
close(s);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
int DDV_Listen(int port){
|
|
int s = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
|
struct sockaddr_in addr;
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = htons(port);//port 8888
|
|
inet_pton(AF_INET, "0.0.0.0", &addr.sin_addr);//listen on all interfaces
|
|
int ret = bind(s, (sockaddr*)&addr, sizeof(addr));//bind to all interfaces, chosen port
|
|
if (ret == 0){
|
|
ret = listen(s, 100);//start listening, backlog of 100 allowed
|
|
if (ret == 0){
|
|
return s;
|
|
}else{
|
|
printf("Listen failed! Error: %s\n", strerror(errno));
|
|
close(s);
|
|
return 0;
|
|
}
|
|
}else{
|
|
printf("Binding failed! Error: %s\n", strerror(errno));
|
|
close(s);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
int DDV_Accept(int sock, bool nonblock = false){
|
|
int r = accept(sock, 0, 0);
|
|
if ((r > 0) && nonblock){
|
|
int flags = fcntl(r, F_GETFL, 0);
|
|
flags |= O_NONBLOCK;
|
|
fcntl(r, F_SETFL, flags);
|
|
}
|
|
return r;
|
|
}
|
|
|
|
bool DDV_write(void * buffer, int todo, int sock){
|
|
int sofar = 0;
|
|
socketBlocking = false;
|
|
while (sofar != todo){
|
|
int r = send(sock, (char*)buffer + sofar, todo-sofar, 0);
|
|
if (r <= 0){
|
|
switch (errno){
|
|
case EWOULDBLOCK: socketBlocking = true; break;
|
|
default:
|
|
socketError = true;
|
|
printf("Could not write! %s\n", strerror(errno));
|
|
return false;
|
|
break;
|
|
}
|
|
}
|
|
sofar += r;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool DDV_ready(int sock){
|
|
char tmp;
|
|
int r = recv(sock, &tmp, 1, MSG_PEEK);
|
|
return (r == 1);
|
|
}
|
|
|
|
bool DDV_read(void * buffer, int todo, int sock){
|
|
int sofar = 0;
|
|
socketBlocking = false;
|
|
while (sofar != todo){
|
|
int r = recv(sock, (char*)buffer + sofar, todo-sofar, 0);
|
|
if (r <= 0){
|
|
switch (errno){
|
|
case EWOULDBLOCK: socketBlocking = true; break;
|
|
default:
|
|
socketError = true;
|
|
printf("Could not read! %s\n", strerror(errno));
|
|
return false;
|
|
break;
|
|
}
|
|
}
|
|
sofar += r;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
bool DDV_read(void * buffer, int width, int count, int sock){return DDV_read(buffer, width*count, sock);}
|
|
bool DDV_write(void * buffer, int width, int count, int sock){return DDV_write(buffer, width*count, sock);}
|
|
|
|
|
|
int DDV_iwrite(void * buffer, int todo, int sock){
|
|
int r = send(sock, buffer, todo, 0);
|
|
if (r < 0){
|
|
switch (errno){
|
|
case EWOULDBLOCK: break;
|
|
default:
|
|
socketError = true;
|
|
printf("Could not write! %s\n", strerror(errno));
|
|
return false;
|
|
break;
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
int DDV_iread(void * buffer, int todo, int sock){
|
|
int r = recv(sock, buffer, todo, 0);
|
|
if (r < 0){
|
|
switch (errno){
|
|
case EWOULDBLOCK: break;
|
|
default:
|
|
socketError = true;
|
|
printf("Could not read! %s\n", strerror(errno));
|
|
return false;
|
|
break;
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
|
|
|