1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <utf.h>
#include <fmt.h>
#include "dat.h"
#include "fns.h"
extern int debug;
void
sysfatal(char *fmt, ...)
{
va_list arg;
char buf[2048];
va_start(arg, fmt);
vseprint(buf, buf+sizeof(buf), fmt, arg);
va_end(arg);
fprint(2, "%s\n", buf);
exit(1);
}
void *
emalloc(ulong n)
{
void *p;
p = malloc(n);
if(p == nil)
sysfatal("malloc: %r");
memset(p, 0, n);
return p;
}
sockaddr_in*
mkinetsa(char *ipaddr, int port)
{
sockaddr_in *sa;
sa = emalloc(sizeof(sockaddr_in));
sa->sin_family = AF_INET;
if(inet_pton(AF_INET, ipaddr, &sa->sin_addr.s_addr) <= 0)
sysfatal("inet_pton: %r");
sa->sin_port = htons(port);
return sa;
}
int
listentcp(int port)
{
sockaddr_in *addr;
int fd;
addr = mkinetsa("0.0.0.0", port);
if((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
return -1;
if(bind(fd, (sockaddr*)addr, sizeof(sockaddr_in)) < 0){
close(fd);
return -1;
}
if(listen(fd, 128) < 0){
close(fd);
return -1;
}
if(debug)
fprint(2, "listening on tcp!%s!%d\n", inet_ntoa(addr->sin_addr), port);
free(addr);
return fd;
}
int
bindudp(int port)
{
sockaddr_in *addr;
int fd;
addr = mkinetsa("0.0.0.0", port);
if((fd = socket(PF_INET, SOCK_DGRAM, 0)) < 0)
return -1;
if(bind(fd, (sockaddr*)addr, sizeof(sockaddr_in)) < 0){
close(fd);
return -1;
}
if(debug)
fprint(2, "listening on udp!%s!%d\n", inet_ntoa(addr->sin_addr), port);
free(addr);
return fd;
}
int
acceptcall(int lfd, char *caddr, int caddrlen)
{
sockaddr_in addr;
uint addrlen;
int fd, port;
char *cs;
memset(&addr, 0, sizeof(sockaddr_in));
addrlen = sizeof(sockaddr_in);
if((fd = accept(lfd, (sockaddr*)&addr, &addrlen)) < 0)
return -1;
cs = inet_ntoa(addr.sin_addr);
port = ntohs(addr.sin_port);
snprint(caddr, caddrlen, "tcp!%s!%d", cs, port);
return fd;
}
|
