mirror of
https://github.com/livingcomputermuseum/UniBone.git
synced 2026-01-29 05:01:09 +00:00
98 lines
1.9 KiB
C
98 lines
1.9 KiB
C
#include "11.h"
|
|
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <netinet/tcp.h>
|
|
#include <arpa/inet.h>
|
|
#include <netdb.h>
|
|
|
|
int
|
|
hasinput(int fd)
|
|
{
|
|
fd_set fds;
|
|
struct timeval timeout;
|
|
|
|
timeout.tv_sec = 0;
|
|
timeout.tv_usec = 0;
|
|
FD_ZERO(&fds);
|
|
FD_SET(fd, &fds);
|
|
return select(fd+1, &fds, NULL, NULL, &timeout) > 0;
|
|
}
|
|
|
|
int
|
|
dial(char *host, int port)
|
|
{
|
|
char portstr[32];
|
|
int sockfd;
|
|
struct addrinfo *result, *rp, hints;
|
|
|
|
memset(&hints, 0, sizeof(hints));
|
|
hints.ai_family = AF_UNSPEC;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
|
|
snprintf(portstr, 32, "%d", port);
|
|
if(getaddrinfo(host, portstr, &hints, &result)){
|
|
perror("error: getaddrinfo");
|
|
return -1;
|
|
}
|
|
|
|
for(rp = result; rp; rp = rp->ai_next){
|
|
sockfd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
|
|
if(sockfd < 0)
|
|
continue;
|
|
if(connect(sockfd, rp->ai_addr, rp->ai_addrlen) >= 0)
|
|
goto win;
|
|
close(sockfd);
|
|
}
|
|
freeaddrinfo(result);
|
|
perror("error");
|
|
return -1;
|
|
|
|
win:
|
|
freeaddrinfo(result);
|
|
return sockfd;
|
|
}
|
|
|
|
void
|
|
serve(int port, void (*handlecon)(int, void*), void *arg)
|
|
{
|
|
int sockfd, confd;
|
|
socklen_t len;
|
|
struct sockaddr_in server, client;
|
|
int x;
|
|
|
|
sockfd = socket(AF_INET, SOCK_STREAM, 0);
|
|
if(sockfd < 0){
|
|
perror("error: socket");
|
|
return;
|
|
}
|
|
|
|
x = 1;
|
|
setsockopt (sockfd, SOL_SOCKET, SO_REUSEADDR, (void *)&x, sizeof x);
|
|
|
|
memset(&server, 0, sizeof(server));
|
|
server.sin_family = AF_INET;
|
|
server.sin_addr.s_addr = INADDR_ANY;
|
|
server.sin_port = htons(port);
|
|
if(bind(sockfd, (struct sockaddr*)&server, sizeof(server)) < 0){
|
|
perror("error: bind");
|
|
return;
|
|
}
|
|
listen(sockfd, 5);
|
|
len = sizeof(client);
|
|
while(confd = accept(sockfd, (struct sockaddr*)&client, &len),
|
|
confd >= 0)
|
|
handlecon(confd, arg);
|
|
perror("error: accept");
|
|
return;
|
|
}
|
|
|
|
void
|
|
nodelay(int fd)
|
|
{
|
|
int flag = 1;
|
|
setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &flag, sizeof(flag));
|
|
}
|