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 bd2e8020 2004-06-16 devnull #include <netinet/tcp.h>
  8 fd04aace 2003-11-23 devnull #include <sys/un.h>
  9 32f69c36 2003-12-11 devnull #include <errno.h>
 10 fd04aace 2003-11-23 devnull 
 11 1c253ceb 2003-11-23 devnull #undef sun
 12 1c253ceb 2003-11-23 devnull #define sun sockun
 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 8ad51794 2004-03-25 devnull 	int n;
 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;
 47 fd04aace 2003-11-23 devnull 
 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;
 51 fd04aace 2003-11-23 devnull 
 52 e42882dc 2004-06-16 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;
 55 fd04aace 2003-11-23 devnull 	}
 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;
 62 fd04aace 2003-11-23 devnull 	else{
 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;
 66 fd04aace 2003-11-23 devnull 	}
 67 fd04aace 2003-11-23 devnull 	free(buf);
 68 fd04aace 2003-11-23 devnull 
 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){
 78 fd04aace 2003-11-23 devnull 		n = 1;
 79 fd04aace 2003-11-23 devnull 		setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
 80 fd04aace 2003-11-23 devnull 	}
 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;
 84 fd04aace 2003-11-23 devnull 	}
 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);
 88 fd04aace 2003-11-23 devnull 	}
 89 fd04aace 2003-11-23 devnull 	return s;
 90 fd04aace 2003-11-23 devnull 
 91 fd04aace 2003-11-23 devnull Unix:
 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;
109 32f69c36 2003-12-11 devnull 		}
110 fd04aace 2003-11-23 devnull 		close(s);
111 fd04aace 2003-11-23 devnull 		return -1;
112 fd04aace 2003-11-23 devnull 	}
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;
117 fd04aace 2003-11-23 devnull }
118 fd04aace 2003-11-23 devnull 
119 fd04aace 2003-11-23 devnull int
120 fd04aace 2003-11-23 devnull p9listen(char *dir, char *newdir)
121 fd04aace 2003-11-23 devnull {
122 bd2e8020 2004-06-16 devnull 	int fd, one;
123 fd04aace 2003-11-23 devnull 
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;
127 fd04aace 2003-11-23 devnull 	}
128 fd04aace 2003-11-23 devnull 
129 fd04aace 2003-11-23 devnull 	if((fd = accept(fd, nil, nil)) < 0)
130 fd04aace 2003-11-23 devnull 		return -1;
131 fd04aace 2003-11-23 devnull 
132 bd2e8020 2004-06-16 devnull 	one = 1;
133 bd2e8020 2004-06-16 devnull 	setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
134 bd2e8020 2004-06-16 devnull 
135 fd04aace 2003-11-23 devnull 	putfd(newdir, fd);
136 fd04aace 2003-11-23 devnull 	return fd;
137 fd04aace 2003-11-23 devnull }
138 fd04aace 2003-11-23 devnull 
139 fd04aace 2003-11-23 devnull int
140 fd04aace 2003-11-23 devnull p9accept(int cfd, char *dir)
141 fd04aace 2003-11-23 devnull {
142 fd04aace 2003-11-23 devnull 	int fd;
143 fd04aace 2003-11-23 devnull 
144 32f69c36 2003-12-11 devnull 	if((fd = _p9netfd(dir)) < 0){
145 fd04aace 2003-11-23 devnull 		werrstr("bad 'directory' in accept");
146 fd04aace 2003-11-23 devnull 		return -1;
147 fd04aace 2003-11-23 devnull 	}
148 fd04aace 2003-11-23 devnull 	/* need to dup because the listen fd will be closed */
149 fd04aace 2003-11-23 devnull 	return dup(fd);
150 fd04aace 2003-11-23 devnull }
151 fd04aace 2003-11-23 devnull