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>
  4 fd04aace 2003-11-23 devnull 
  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>
  9 fd04aace 2003-11-23 devnull 
 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*);
 13 fd04aace 2003-11-23 devnull 
 14 32f69c36 2003-12-11 devnull int
 15 32f69c36 2003-12-11 devnull _p9netfd(char *dir)
 16 fd04aace 2003-11-23 devnull {
 17 fd04aace 2003-11-23 devnull 	int fd;
 18 fd04aace 2003-11-23 devnull 
 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;
 25 fd04aace 2003-11-23 devnull }
 26 fd04aace 2003-11-23 devnull 
 27 fd04aace 2003-11-23 devnull static void
 28 fd04aace 2003-11-23 devnull putfd(char *dir, int fd)
 29 fd04aace 2003-11-23 devnull {
 30 fd04aace 2003-11-23 devnull 	snprint(dir, NETPATHLEN, "/dev/fd/%d", fd);
 31 fd04aace 2003-11-23 devnull }
 32 fd04aace 2003-11-23 devnull 
 33 fd04aace 2003-11-23 devnull #undef unix
 34 fd04aace 2003-11-23 devnull 
 35 fd04aace 2003-11-23 devnull int
 36 fd04aace 2003-11-23 devnull p9announce(char *addr, char *dir)
 37 fd04aace 2003-11-23 devnull {
 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;
 43 fd04aace 2003-11-23 devnull 	int n, sn;
 44 fd04aace 2003-11-23 devnull 	struct sockaddr_in sa;
 45 fd04aace 2003-11-23 devnull 	struct sockaddr_un sun;
 46 fd04aace 2003-11-23 devnull 
 47 fd04aace 2003-11-23 devnull 	buf = strdup(addr);
 48 fd04aace 2003-11-23 devnull 	if(buf == nil)
 49 fd04aace 2003-11-23 devnull 		return -1;
 50 fd04aace 2003-11-23 devnull 
 51 fd04aace 2003-11-23 devnull 	if(_p9dialparse(buf, &net, &unix, &host, &port) < 0){
 52 fd04aace 2003-11-23 devnull 		free(buf);
 53 fd04aace 2003-11-23 devnull 		return -1;
 54 fd04aace 2003-11-23 devnull 	}
 55 fd04aace 2003-11-23 devnull 	if(strcmp(net, "tcp") == 0)
 56 fd04aace 2003-11-23 devnull 		proto = SOCK_STREAM;
 57 fd04aace 2003-11-23 devnull 	else if(strcmp(net, "udp") == 0)
 58 fd04aace 2003-11-23 devnull 		proto = SOCK_DGRAM;
 59 fd04aace 2003-11-23 devnull 	else if(strcmp(net, "unix") == 0)
 60 fd04aace 2003-11-23 devnull 		goto Unix;
 61 fd04aace 2003-11-23 devnull 	else{
 62 fd04aace 2003-11-23 devnull 		werrstr("can only handle tcp, udp, and unix: not %s", net);
 63 fd04aace 2003-11-23 devnull 		free(buf);
 64 fd04aace 2003-11-23 devnull 		return -1;
 65 fd04aace 2003-11-23 devnull 	}
 66 fd04aace 2003-11-23 devnull 	free(buf);
 67 fd04aace 2003-11-23 devnull 
 68 fd04aace 2003-11-23 devnull 	memset(&sa, 0, sizeof sa);
 69 fd04aace 2003-11-23 devnull 	memmove(&sa.sin_addr, &host, 4);
 70 fd04aace 2003-11-23 devnull 	sa.sin_family = AF_INET;
 71 fd04aace 2003-11-23 devnull 	sa.sin_port = htons(port);
 72 fd04aace 2003-11-23 devnull 	if((s = socket(AF_INET, proto, 0)) < 0)
 73 fd04aace 2003-11-23 devnull 		return -1;
 74 fd04aace 2003-11-23 devnull 	sn = sizeof n;
 75 fd04aace 2003-11-23 devnull 	if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (char*)&n, &sn) >= 0
 76 fd04aace 2003-11-23 devnull 	&& n == SOCK_STREAM){
 77 fd04aace 2003-11-23 devnull 		n = 1;
 78 fd04aace 2003-11-23 devnull 		setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
 79 fd04aace 2003-11-23 devnull 	}
 80 fd04aace 2003-11-23 devnull 	if(bind(s, (struct sockaddr*)&sa, sizeof sa) < 0){
 81 fd04aace 2003-11-23 devnull 		close(s);
 82 fd04aace 2003-11-23 devnull 		return -1;
 83 fd04aace 2003-11-23 devnull 	}
 84 fd04aace 2003-11-23 devnull 	if(proto == SOCK_STREAM){
 85 fd04aace 2003-11-23 devnull 		listen(s, 8);
 86 fd04aace 2003-11-23 devnull 		putfd(dir, s);
 87 fd04aace 2003-11-23 devnull 	}
 88 fd04aace 2003-11-23 devnull 	return s;
 89 fd04aace 2003-11-23 devnull 
 90 fd04aace 2003-11-23 devnull Unix:
 91 fd04aace 2003-11-23 devnull 	memset(&sun, 0, sizeof sun);
 92 fd04aace 2003-11-23 devnull 	sun.sun_family = AF_UNIX;
 93 fd04aace 2003-11-23 devnull 	strcpy(sun.sun_path, unix);
 94 fd04aace 2003-11-23 devnull 	if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
 95 fd04aace 2003-11-23 devnull 		return -1;
 96 fd04aace 2003-11-23 devnull 	sn = sizeof sun;
 97 fd04aace 2003-11-23 devnull 	if(bind(s, (struct sockaddr*)&sun, sizeof sun) < 0){
 98 32f69c36 2003-12-11 devnull 		if(errno == EADDRINUSE
 99 32f69c36 2003-12-11 devnull 		&& connect(s, (struct sockaddr*)&sun, sizeof sun) < 0
100 32f69c36 2003-12-11 devnull 		&& errno == ECONNREFUSED){
101 32f69c36 2003-12-11 devnull 			/* dead socket, so remove it */
102 32f69c36 2003-12-11 devnull 			remove(unix);
103 32f69c36 2003-12-11 devnull 			close(s);
104 32f69c36 2003-12-11 devnull 			if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
105 32f69c36 2003-12-11 devnull 				return -1;
106 32f69c36 2003-12-11 devnull 			if(bind(s, (struct sockaddr*)&sun, sizeof sun) >= 0)
107 32f69c36 2003-12-11 devnull 				goto Success;
108 32f69c36 2003-12-11 devnull 		}
109 fd04aace 2003-11-23 devnull 		close(s);
110 fd04aace 2003-11-23 devnull 		return -1;
111 fd04aace 2003-11-23 devnull 	}
112 32f69c36 2003-12-11 devnull Success:
113 fd04aace 2003-11-23 devnull 	listen(s, 8);
114 fd04aace 2003-11-23 devnull 	putfd(dir, s);
115 fd04aace 2003-11-23 devnull 	return s;
116 fd04aace 2003-11-23 devnull }
117 fd04aace 2003-11-23 devnull 
118 fd04aace 2003-11-23 devnull int
119 fd04aace 2003-11-23 devnull p9listen(char *dir, char *newdir)
120 fd04aace 2003-11-23 devnull {
121 fd04aace 2003-11-23 devnull 	int fd;
122 fd04aace 2003-11-23 devnull 
123 32f69c36 2003-12-11 devnull 	if((fd = _p9netfd(dir)) < 0){
124 fd04aace 2003-11-23 devnull 		werrstr("bad 'directory' in listen: %s", dir);
125 fd04aace 2003-11-23 devnull 		return -1;
126 fd04aace 2003-11-23 devnull 	}
127 fd04aace 2003-11-23 devnull 
128 fd04aace 2003-11-23 devnull 	if((fd = accept(fd, nil, nil)) < 0)
129 fd04aace 2003-11-23 devnull 		return -1;
130 fd04aace 2003-11-23 devnull 
131 fd04aace 2003-11-23 devnull 	putfd(newdir, fd);
132 fd04aace 2003-11-23 devnull 	return fd;
133 fd04aace 2003-11-23 devnull }
134 fd04aace 2003-11-23 devnull 
135 fd04aace 2003-11-23 devnull int
136 fd04aace 2003-11-23 devnull p9accept(int cfd, char *dir)
137 fd04aace 2003-11-23 devnull {
138 fd04aace 2003-11-23 devnull 	int fd;
139 fd04aace 2003-11-23 devnull 
140 32f69c36 2003-12-11 devnull 	if((fd = _p9netfd(dir)) < 0){
141 fd04aace 2003-11-23 devnull 		werrstr("bad 'directory' in accept");
142 fd04aace 2003-11-23 devnull 		return -1;
143 fd04aace 2003-11-23 devnull 	}
144 fd04aace 2003-11-23 devnull 	/* need to dup because the listen fd will be closed */
145 fd04aace 2003-11-23 devnull 	return dup(fd);
146 fd04aace 2003-11-23 devnull }
147 fd04aace 2003-11-23 devnull