145 lines
3.5 KiB
C
145 lines
3.5 KiB
C
#include <sys/types.h>
|
||
#include <sys/socket.h>
|
||
#include <unistd.h>
|
||
#include <fcntl.h>
|
||
#include <netdb.h>
|
||
#include <netinet/in.h>
|
||
#include <netinet/tcp.h>
|
||
#include "TCPIP.h"
|
||
#include "version.h"
|
||
|
||
int getAddress(char *host, struct in_addr *iaddr)
|
||
{
|
||
char *p = host;
|
||
int ishost = 0;
|
||
|
||
if (strcmp(host,"*")==0) {
|
||
iaddr->s_addr=htonl(INADDR_ANY);
|
||
return 1;
|
||
}
|
||
while (*p) {
|
||
if (!(isdigit(*p) || ((*p) == '.'))) {
|
||
ishost = 1;
|
||
break;
|
||
}
|
||
p++;
|
||
}
|
||
if (ishost) {
|
||
struct hostent *h;
|
||
h = gethostbyname(host);
|
||
if (!h) { return 0; }
|
||
memcpy((void *) &iaddr->s_addr,(void *) h->h_addr,4);
|
||
return 1;
|
||
} else {
|
||
iaddr->s_addr = inet_addr(host);
|
||
return 1;
|
||
}
|
||
}
|
||
|
||
int TCPbind(char *bindAddress,int bindPort)
|
||
{
|
||
struct in_addr iaddr;
|
||
struct sockaddr_in saddr;
|
||
struct linger lg;
|
||
int sd;
|
||
int nodelay,reuse;
|
||
|
||
if (!getAddress(bindAddress, &iaddr))
|
||
errexit2("Host %s could not be resolved\n",bindAddress);
|
||
sd = socket(PF_INET, SOCK_STREAM, 0);
|
||
if (sd < 0)
|
||
errexit("Couldn't create server socket!\n");
|
||
|
||
saddr.sin_family = AF_INET;
|
||
memcpy(&saddr.sin_addr, &iaddr, sizeof(iaddr));
|
||
saddr.sin_port = htons(bindPort);
|
||
|
||
if (setsockopt(sd,IPPROTO_TCP,TCP_NODELAY,&nodelay,sizeof(nodelay))<0)
|
||
error("TCP_NODELAY failed\n");
|
||
if (setsockopt(sd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
|
||
error("SO_REUSEADDR failed\n");
|
||
lg.l_onoff=1; lg.l_linger=5;
|
||
if (setsockopt(sd,SOL_SOCKET,SO_LINGER,&lg,sizeof(lg))<0)
|
||
error("SO_LINGER failed\n");
|
||
|
||
if (bind(sd, (struct sockaddr *)&saddr, sizeof(saddr)) < 0)
|
||
errexit3("Couldn't bind to address %s port %d\n",
|
||
bindAddress, bindPort);
|
||
if (listen(sd, 5) < 0)
|
||
errexit3("Couldn't listen to address %s port %d\n",
|
||
bindAddress, bindPort);
|
||
return(sd);
|
||
}
|
||
|
||
int TCPaccept0(int sd,int block,unsigned long *ip)
|
||
{
|
||
struct sockaddr_in saddr;
|
||
int ad,len,j;
|
||
|
||
if (block==0) {
|
||
j=0;
|
||
setsockopt(sd, SOL_SOCKET, SO_LINGER, &j, sizeof(j));
|
||
fcntl(sd, F_SETFL, O_NONBLOCK);
|
||
}
|
||
len=sizeof(saddr);
|
||
if ((ad = accept(sd, (struct sockaddr *)&saddr, (int *)&len)) < 0) {
|
||
error("Couldn't accept\n");
|
||
return -1;
|
||
} else if (ip) memcpy(ip,&saddr.sin_addr, sizeof(unsigned long));
|
||
return(ad);
|
||
}
|
||
|
||
int TCPaccept(int sd) { return TCPaccept0(sd,1,0); }
|
||
int TCPacceptIP(int sd,unsigned long *ip) { return TCPaccept0(sd,1,ip); }
|
||
int TCPacceptNB(int sd) { return TCPaccept0(sd,0,0); }
|
||
|
||
int TCPconnect0(char *bindAddress,int bindPort,int block)
|
||
{
|
||
struct in_addr iaddr;
|
||
struct sockaddr_in saddr;
|
||
int sd,j;
|
||
int nodelay;
|
||
|
||
if (!getAddress(bindAddress, &iaddr)) {
|
||
error2("Host %s could not be resolved\n",bindAddress);
|
||
return -1;
|
||
}
|
||
sd = socket(PF_INET, SOCK_STREAM, 0);
|
||
if (sd < 0) {
|
||
error("Couldn't create server socket!\n");
|
||
return -1;
|
||
}
|
||
saddr.sin_family = AF_INET;
|
||
memcpy(&saddr.sin_addr, &iaddr, sizeof(iaddr));
|
||
saddr.sin_port = htons(bindPort);
|
||
|
||
if (setsockopt(sd,IPPROTO_TCP,TCP_NODELAY,&nodelay,sizeof(nodelay))<0)
|
||
error("TCP_NODELAY failed\n");
|
||
|
||
if (block==0) {
|
||
j=0;
|
||
setsockopt(sd, SOL_SOCKET, SO_LINGER, &j, sizeof(j));
|
||
fcntl(sd, F_SETFL, O_NONBLOCK);
|
||
}
|
||
|
||
if (connect(sd, (struct sockaddr *)&saddr, sizeof(saddr)) < 0) {
|
||
error3("<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ӵ<EFBFBD>ַ %s <20>˿<EFBFBD> %d\n",
|
||
bindAddress, bindPort);
|
||
return -1;
|
||
}
|
||
return(sd);
|
||
}
|
||
|
||
int TCPconnect(char *bindAddress,int bindPort)
|
||
{
|
||
TCPconnect0(bindAddress,bindPort,1);
|
||
}
|
||
|
||
int TCPconnectNB(char *bindAddress,int bindPort)
|
||
{
|
||
TCPconnect0(bindAddress,bindPort,0);
|
||
}
|
||
|
||
|
||
|