drawterm/kern/devip-posix.c
2005-11-07 17:13:41 +00:00

210 lines
3.4 KiB
C

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <netdb.h>
#include "u.h"
#include "lib.h"
#include "dat.h"
#include "fns.h"
#include "error.h"
#include "devip.h"
#undef listen
#undef accept
#undef bind
void
osipinit(void)
{
char buf[1024];
gethostname(buf, sizeof(buf));
kstrdup(&sysname, buf);
}
int
so_socket(int type)
{
int fd, one;
switch(type) {
default:
error("bad protocol type");
case S_TCP:
type = SOCK_STREAM;
break;
case S_UDP:
type = SOCK_DGRAM;
break;
}
fd = socket(AF_INET, type, 0);
if(fd < 0)
oserror();
one = 1;
if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof(one)) > 0){
oserrstr();
print("setsockopt: %r");
}
return fd;
}
void
so_connect(int fd, unsigned long raddr, unsigned short rport)
{
struct sockaddr_in sin;
memset(&sin, 0, sizeof(sin));
sin.sin_family = AF_INET;
hnputs(&sin.sin_port, rport);
hnputl(&sin.sin_addr.s_addr, raddr);
if(connect(fd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
oserror();
}
void
so_getsockname(int fd, unsigned long *laddr, unsigned short *lport)
{
int len;
struct sockaddr_in sin;
len = sizeof(sin);
if(getsockname(fd, (struct sockaddr*)&sin, &len) < 0)
oserror();
if(sin.sin_family != AF_INET || len != sizeof(sin))
error("not AF_INET");
*laddr = nhgetl(&sin.sin_addr.s_addr);
*lport = nhgets(&sin.sin_port);
}
void
so_listen(int fd)
{
if(listen(fd, 5) < 0)
oserror();
}
int
so_accept(int fd, unsigned long *raddr, unsigned short *rport)
{
int nfd, len;
struct sockaddr_in sin;
len = sizeof(sin);
nfd = accept(fd, (struct sockaddr*)&sin, &len);
if(nfd < 0)
oserror();
if(sin.sin_family != AF_INET || len != sizeof(sin))
error("not AF_INET");
*raddr = nhgetl(&sin.sin_addr.s_addr);
*rport = nhgets(&sin.sin_port);
return nfd;
}
void
so_bind(int fd, int su, unsigned short port)
{
int i, one;
struct sockaddr_in sin;
one = 1;
if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char*)&one, sizeof(one)) < 0){
oserrstr();
print("setsockopt: %r");
}
if(su) {
for(i = 600; i < 1024; i++) {
memset(&sin, 0, sizeof(sin));
sin.sin_family = AF_INET;
sin.sin_port = i;
if(bind(fd, (struct sockaddr*)&sin, sizeof(sin)) >= 0)
return;
}
oserror();
}
memset(&sin, 0, sizeof(sin));
sin.sin_family = AF_INET;
hnputs(&sin.sin_port, port);
if(bind(fd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
oserror();
}
int
so_gethostbyname(char *host, char**hostv, int n)
{
int i;
char buf[32];
unsigned char *p;
struct hostent *hp;
hp = gethostbyname(host);
if(hp == 0)
return 0;
for(i = 0; hp->h_addr_list[i] && i < n; i++) {
p = (unsigned char*)hp->h_addr_list[i];
sprint(buf, "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
hostv[i] = strdup(buf);
if(hostv[i] == 0)
break;
}
return i;
}
char*
hostlookup(char *host)
{
char buf[100];
uchar *p;
struct hostent *he;
he = gethostbyname(host);
if(he != 0 && he->h_addr_list[0]) {
p = (uchar*)he->h_addr_list[0];
sprint(buf, "%ud.%ud.%ud.%ud", p[0], p[1], p[2], p[3]);
} else
strcpy(buf, host);
return strdup(buf);
}
int
so_getservbyname(char *service, char *net, char *port)
{
struct servent *s;
s = getservbyname(service, net);
if(s == 0)
return -1;
sprint(port, "%d", nhgets(&s->s_port));
return 0;
}
int
so_send(int fd, void *d, int n, int f)
{
return send(fd, d, n, f);
}
int
so_recv(int fd, void *d, int n, int f)
{
return recv(fd, d, n, f);
}