conterm/kern/devip-win.c

209 lines
3.5 KiB
C

#include <windows.h>
#include "winduhz.h"
#include "u.h"
#include "lib.h"
#include "dat.h"
#include "fns.h"
#include "error.h"
#include "devip.h"
#pragma comment(lib, "wsock32.lib")
#undef listen
#undef accept
#undef bind
void
osipinit(void)
{
WSADATA wasdat;
char buf[1024];
if(WSAStartup(MAKEWORD(1, 1), &wasdat) != 0)
panic("no winsock.dll");
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)
error(sys_errlist[errno]);
one = 1;
if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof(one)) > 0)
print("setsockopt: %s", sys_errlist[errno]);
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)
error(sys_errlist[errno]);
}
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)
error(sys_errlist[errno]);
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)
error(sys_errlist[errno]);
}
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)
error(sys_errlist[errno]);
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)
print("setsockopt: %s", sys_errlist[errno]);
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;
}
error(sys_errlist[errno]);
}
memset(&sin, 0, sizeof(sin));
sin.sin_family = AF_INET;
hnputs(&sin.sin_port, port);
if(bind(fd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
error(sys_errlist[errno]);
}
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;
HOSTENT *he;
he = gethostbyname(host);
if(he != 0 && he->h_addr_list[0]) {
p = 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);
}