1 fd04aace 2003-11-23 devnull #include <u.h>
2 fd04aace 2003-11-23 devnull #define NOPLAN9DEFINES
3 fd04aace 2003-11-23 devnull #include <libc.h>
5 fd04aace 2003-11-23 devnull #include <sys/socket.h>
6 fd04aace 2003-11-23 devnull #include <netinet/in.h>
7 fd04aace 2003-11-23 devnull #include <sys/un.h>
8 32f69c36 2003-12-11 devnull #include <errno.h>
10 1c253ceb 2003-11-23 devnull #undef sun
11 1c253ceb 2003-11-23 devnull #define sun sockun
12 fd04aace 2003-11-23 devnull extern int _p9dialparse(char*, char**, char**, u32int*, int*);
15 32f69c36 2003-12-11 devnull _p9netfd(char *dir)
19 fd04aace 2003-11-23 devnull if(strncmp(dir, "/dev/fd/", 8) != 0)
20 fd04aace 2003-11-23 devnull return -1;
21 fd04aace 2003-11-23 devnull fd = strtol(dir+8, &dir, 0);
22 fd04aace 2003-11-23 devnull if(*dir != 0)
23 fd04aace 2003-11-23 devnull return -1;
24 fd04aace 2003-11-23 devnull return fd;
27 fd04aace 2003-11-23 devnull static void
28 fd04aace 2003-11-23 devnull putfd(char *dir, int fd)
30 fd04aace 2003-11-23 devnull snprint(dir, NETPATHLEN, "/dev/fd/%d", fd);
33 fd04aace 2003-11-23 devnull #undef unix
36 fd04aace 2003-11-23 devnull p9announce(char *addr, char *dir)
38 fd04aace 2003-11-23 devnull int proto;
39 fd04aace 2003-11-23 devnull char *buf, *unix;
40 fd04aace 2003-11-23 devnull char *net;
41 fd04aace 2003-11-23 devnull u32int host;
42 fd04aace 2003-11-23 devnull int port, s;
44 8ad51794 2004-03-25 devnull socklen_t sn;
45 fd04aace 2003-11-23 devnull struct sockaddr_in sa;
46 fd04aace 2003-11-23 devnull struct sockaddr_un sun;
48 fd04aace 2003-11-23 devnull buf = strdup(addr);
49 fd04aace 2003-11-23 devnull if(buf == nil)
50 fd04aace 2003-11-23 devnull return -1;
52 fd04aace 2003-11-23 devnull if(_p9dialparse(buf, &net, &unix, &host, &port) < 0){
53 fd04aace 2003-11-23 devnull free(buf);
54 fd04aace 2003-11-23 devnull return -1;
56 fd04aace 2003-11-23 devnull if(strcmp(net, "tcp") == 0)
57 fd04aace 2003-11-23 devnull proto = SOCK_STREAM;
58 fd04aace 2003-11-23 devnull else if(strcmp(net, "udp") == 0)
59 fd04aace 2003-11-23 devnull proto = SOCK_DGRAM;
60 fd04aace 2003-11-23 devnull else if(strcmp(net, "unix") == 0)
61 fd04aace 2003-11-23 devnull goto Unix;
63 fd04aace 2003-11-23 devnull werrstr("can only handle tcp, udp, and unix: not %s", net);
64 fd04aace 2003-11-23 devnull free(buf);
65 fd04aace 2003-11-23 devnull return -1;
67 fd04aace 2003-11-23 devnull free(buf);
69 fd04aace 2003-11-23 devnull memset(&sa, 0, sizeof sa);
70 fd04aace 2003-11-23 devnull memmove(&sa.sin_addr, &host, 4);
71 fd04aace 2003-11-23 devnull sa.sin_family = AF_INET;
72 fd04aace 2003-11-23 devnull sa.sin_port = htons(port);
73 fd04aace 2003-11-23 devnull if((s = socket(AF_INET, proto, 0)) < 0)
74 fd04aace 2003-11-23 devnull return -1;
75 fd04aace 2003-11-23 devnull sn = sizeof n;
76 8ad51794 2004-03-25 devnull if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0
77 fd04aace 2003-11-23 devnull && n == SOCK_STREAM){
79 fd04aace 2003-11-23 devnull setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
81 fd04aace 2003-11-23 devnull if(bind(s, (struct sockaddr*)&sa, sizeof sa) < 0){
82 fd04aace 2003-11-23 devnull close(s);
83 fd04aace 2003-11-23 devnull return -1;
85 fd04aace 2003-11-23 devnull if(proto == SOCK_STREAM){
86 fd04aace 2003-11-23 devnull listen(s, 8);
87 fd04aace 2003-11-23 devnull putfd(dir, s);
89 fd04aace 2003-11-23 devnull return s;
92 fd04aace 2003-11-23 devnull memset(&sun, 0, sizeof sun);
93 fd04aace 2003-11-23 devnull sun.sun_family = AF_UNIX;
94 fd04aace 2003-11-23 devnull strcpy(sun.sun_path, unix);
95 fd04aace 2003-11-23 devnull if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
96 fd04aace 2003-11-23 devnull return -1;
97 fd04aace 2003-11-23 devnull sn = sizeof sun;
98 fd04aace 2003-11-23 devnull if(bind(s, (struct sockaddr*)&sun, sizeof sun) < 0){
99 32f69c36 2003-12-11 devnull if(errno == EADDRINUSE
100 32f69c36 2003-12-11 devnull && connect(s, (struct sockaddr*)&sun, sizeof sun) < 0
101 32f69c36 2003-12-11 devnull && errno == ECONNREFUSED){
102 32f69c36 2003-12-11 devnull /* dead socket, so remove it */
103 32f69c36 2003-12-11 devnull remove(unix);
104 32f69c36 2003-12-11 devnull close(s);
105 32f69c36 2003-12-11 devnull if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
106 32f69c36 2003-12-11 devnull return -1;
107 32f69c36 2003-12-11 devnull if(bind(s, (struct sockaddr*)&sun, sizeof sun) >= 0)
108 32f69c36 2003-12-11 devnull goto Success;
110 fd04aace 2003-11-23 devnull close(s);
111 fd04aace 2003-11-23 devnull return -1;
113 32f69c36 2003-12-11 devnull Success:
114 fd04aace 2003-11-23 devnull listen(s, 8);
115 fd04aace 2003-11-23 devnull putfd(dir, s);
116 fd04aace 2003-11-23 devnull return s;
120 fd04aace 2003-11-23 devnull p9listen(char *dir, char *newdir)
124 32f69c36 2003-12-11 devnull if((fd = _p9netfd(dir)) < 0){
125 fd04aace 2003-11-23 devnull werrstr("bad 'directory' in listen: %s", dir);
126 fd04aace 2003-11-23 devnull return -1;
129 fd04aace 2003-11-23 devnull if((fd = accept(fd, nil, nil)) < 0)
130 fd04aace 2003-11-23 devnull return -1;
132 fd04aace 2003-11-23 devnull putfd(newdir, fd);
133 fd04aace 2003-11-23 devnull return fd;
137 fd04aace 2003-11-23 devnull p9accept(int cfd, char *dir)
141 32f69c36 2003-12-11 devnull if((fd = _p9netfd(dir)) < 0){
142 fd04aace 2003-11-23 devnull werrstr("bad 'directory' in accept");
143 fd04aace 2003-11-23 devnull return -1;
145 fd04aace 2003-11-23 devnull /* need to dup because the listen fd will be closed */
146 fd04aace 2003-11-23 devnull return dup(fd);