1 fd04aace 2003-11-23 devnull #include <u.h>
  2 fd04aace 2003-11-23 devnull #include <libc.h>
  3 fd04aace 2003-11-23 devnull 
  4 fd04aace 2003-11-23 devnull #undef	accept
  5 fd04aace 2003-11-23 devnull #undef	announce
  6 fd04aace 2003-11-23 devnull #undef	dial
  7 fd04aace 2003-11-23 devnull #undef	setnetmtpt
  8 fd04aace 2003-11-23 devnull #undef	hangup
  9 fd04aace 2003-11-23 devnull #undef	listen
 10 fd04aace 2003-11-23 devnull #undef	netmkaddr
 11 fd04aace 2003-11-23 devnull #undef	reject
 12 fd04aace 2003-11-23 devnull 
 13 fd04aace 2003-11-23 devnull #include <sys/socket.h>
 14 fd04aace 2003-11-23 devnull #include <netinet/in.h>
 15 d51419bf 2004-02-09 devnull #include <netinet/tcp.h>
 16 fd04aace 2003-11-23 devnull #include <sys/un.h>
 17 fd04aace 2003-11-23 devnull #include <netdb.h>
 18 fd04aace 2003-11-23 devnull 
 19 fd04aace 2003-11-23 devnull #undef unix
 20 835c1e47 2005-07-13 devnull #define unix xunix
 21 fd04aace 2003-11-23 devnull 
 22 3409bc9a 2012-06-02 0intro static int
 23 3409bc9a 2012-06-02 0intro isany(struct sockaddr_storage *ss)
 24 3409bc9a 2012-06-02 0intro {
 25 3409bc9a 2012-06-02 0intro 	switch(ss->ss_family){
 26 3409bc9a 2012-06-02 0intro 	case AF_INET:
 27 3409bc9a 2012-06-02 0intro 		return (((struct sockaddr_in*)ss)->sin_addr.s_addr == INADDR_ANY);
 28 3409bc9a 2012-06-02 0intro 	case AF_INET6:
 29 3409bc9a 2012-06-02 0intro 		return (memcmp(((struct sockaddr_in6*)ss)->sin6_addr.s6_addr,
 30 3409bc9a 2012-06-02 0intro 			in6addr_any.s6_addr, sizeof (struct in6_addr)) == 0);
 31 3409bc9a 2012-06-02 0intro 	}
 32 3409bc9a 2012-06-02 0intro 	return 0;
 33 3409bc9a 2012-06-02 0intro }
 34 3409bc9a 2012-06-02 0intro 
 35 3dade5fe 2012-09-04 rsc static int
 36 3dade5fe 2012-09-04 rsc addrlen(struct sockaddr_storage *ss)
 37 3dade5fe 2012-09-04 rsc {
 38 3dade5fe 2012-09-04 rsc 	switch(ss->ss_family){
 39 3dade5fe 2012-09-04 rsc 	case AF_INET:
 40 3dade5fe 2012-09-04 rsc 		return sizeof(struct sockaddr_in);
 41 3dade5fe 2012-09-04 rsc 	case AF_INET6:
 42 3dade5fe 2012-09-04 rsc 		return sizeof(struct sockaddr_in6);
 43 3dade5fe 2012-09-04 rsc 	case AF_UNIX:
 44 3dade5fe 2012-09-04 rsc 		return sizeof(struct sockaddr_un);
 45 3dade5fe 2012-09-04 rsc 	}
 46 3dade5fe 2012-09-04 rsc 	return 0;
 47 3dade5fe 2012-09-04 rsc }
 48 3dade5fe 2012-09-04 rsc 
 49 fd04aace 2003-11-23 devnull int
 50 e5d68728 2005-02-13 devnull p9dial(char *addr, char *local, char *dummy2, int *dummy3)
 51 fd04aace 2003-11-23 devnull {
 52 fd04aace 2003-11-23 devnull 	char *buf;
 53 fd04aace 2003-11-23 devnull 	char *net, *unix;
 54 fd04aace 2003-11-23 devnull 	int port;
 55 fd04aace 2003-11-23 devnull 	int proto;
 56 e5d68728 2005-02-13 devnull 	socklen_t sn;
 57 e5d68728 2005-02-13 devnull 	int n;
 58 3409bc9a 2012-06-02 0intro 	struct sockaddr_storage ss, ssl;
 59 fd04aace 2003-11-23 devnull 	int s;
 60 fd04aace 2003-11-23 devnull 
 61 e5d68728 2005-02-13 devnull 	if(dummy2 || dummy3){
 62 fd04aace 2003-11-23 devnull 		werrstr("cannot handle extra arguments in dial");
 63 fd04aace 2003-11-23 devnull 		return -1;
 64 fd04aace 2003-11-23 devnull 	}
 65 fd04aace 2003-11-23 devnull 
 66 fd04aace 2003-11-23 devnull 	buf = strdup(addr);
 67 fd04aace 2003-11-23 devnull 	if(buf == nil)
 68 fd04aace 2003-11-23 devnull 		return -1;
 69 fd04aace 2003-11-23 devnull 
 70 3409bc9a 2012-06-02 0intro 	if(p9dialparse(buf, &net, &unix, &ss, &port) < 0){
 71 fd04aace 2003-11-23 devnull 		free(buf);
 72 fd04aace 2003-11-23 devnull 		return -1;
 73 fd04aace 2003-11-23 devnull 	}
 74 3409bc9a 2012-06-02 0intro 	if(strcmp(net, "unix") != 0 && isany(&ss)){
 75 6d558e12 2009-08-15 rsc 		werrstr("invalid dial address 0.0.0.0 (aka *)");
 76 6d558e12 2009-08-15 rsc 		free(buf);
 77 6d558e12 2009-08-15 rsc 		return -1;
 78 6d558e12 2009-08-15 rsc 	}
 79 fd04aace 2003-11-23 devnull 
 80 fd04aace 2003-11-23 devnull 	if(strcmp(net, "tcp") == 0)
 81 fd04aace 2003-11-23 devnull 		proto = SOCK_STREAM;
 82 fd04aace 2003-11-23 devnull 	else if(strcmp(net, "udp") == 0)
 83 fd04aace 2003-11-23 devnull 		proto = SOCK_DGRAM;
 84 fd04aace 2003-11-23 devnull 	else if(strcmp(net, "unix") == 0)
 85 fd04aace 2003-11-23 devnull 		goto Unix;
 86 fd04aace 2003-11-23 devnull 	else{
 87 fd04aace 2003-11-23 devnull 		werrstr("can only handle tcp, udp, and unix: not %s", net);
 88 fd04aace 2003-11-23 devnull 		free(buf);
 89 fd04aace 2003-11-23 devnull 		return -1;
 90 fd04aace 2003-11-23 devnull 	}
 91 fd04aace 2003-11-23 devnull 	free(buf);
 92 fd04aace 2003-11-23 devnull 
 93 3409bc9a 2012-06-02 0intro 	if((s = socket(ss.ss_family, proto, 0)) < 0)
 94 fd04aace 2003-11-23 devnull 		return -1;
 95 fa325e9b 2020-01-10 cross 
 96 e5d68728 2005-02-13 devnull 	if(local){
 97 e5d68728 2005-02-13 devnull 		buf = strdup(local);
 98 e5d68728 2005-02-13 devnull 		if(buf == nil){
 99 e5d68728 2005-02-13 devnull 			close(s);
100 e5d68728 2005-02-13 devnull 			return -1;
101 e5d68728 2005-02-13 devnull 		}
102 3409bc9a 2012-06-02 0intro 		if(p9dialparse(buf, &net, &unix, &ss, &port) < 0){
103 e5d68728 2005-02-13 devnull 		badlocal:
104 e5d68728 2005-02-13 devnull 			free(buf);
105 e5d68728 2005-02-13 devnull 			close(s);
106 e5d68728 2005-02-13 devnull 			return -1;
107 e5d68728 2005-02-13 devnull 		}
108 e5d68728 2005-02-13 devnull 		if(unix){
109 e5d68728 2005-02-13 devnull 			werrstr("bad local address %s for dial %s", local, addr);
110 e5d68728 2005-02-13 devnull 			goto badlocal;
111 e5d68728 2005-02-13 devnull 		}
112 e5d68728 2005-02-13 devnull 		sn = sizeof n;
113 e5d68728 2005-02-13 devnull 		if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0
114 e5d68728 2005-02-13 devnull 		&& n == SOCK_STREAM){
115 e5d68728 2005-02-13 devnull 			n = 1;
116 e5d68728 2005-02-13 devnull 			setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
117 e5d68728 2005-02-13 devnull 		}
118 3dade5fe 2012-09-04 rsc 		if(bind(s, (struct sockaddr*)&ssl, addrlen(&ssl)) < 0)
119 e5d68728 2005-02-13 devnull 			goto badlocal;
120 e5d68728 2005-02-13 devnull 		free(buf);
121 e5d68728 2005-02-13 devnull 	}
122 e5d68728 2005-02-13 devnull 
123 74f9b0c6 2005-12-26 devnull 	n = 1;
124 74f9b0c6 2005-12-26 devnull 	setsockopt(s, SOL_SOCKET, SO_BROADCAST, &n, sizeof n);
125 3409bc9a 2012-06-02 0intro 	if(!isany(&ss)){
126 3dade5fe 2012-09-04 rsc 		if(connect(s, (struct sockaddr*)&ss, addrlen(&ss)) < 0){
127 3d484b0d 2005-12-29 devnull 			close(s);
128 3d484b0d 2005-12-29 devnull 			return -1;
129 3d484b0d 2005-12-29 devnull 		}
130 fd04aace 2003-11-23 devnull 	}
131 d51419bf 2004-02-09 devnull 	if(proto == SOCK_STREAM){
132 d51419bf 2004-02-09 devnull 		int one = 1;
133 d51419bf 2004-02-09 devnull 		setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
134 d51419bf 2004-02-09 devnull 	}
135 fd04aace 2003-11-23 devnull 	return s;
136 fd04aace 2003-11-23 devnull 
137 fd04aace 2003-11-23 devnull Unix:
138 e5d68728 2005-02-13 devnull 	if(local){
139 e5d68728 2005-02-13 devnull 		werrstr("local address not supported on unix network");
140 e5d68728 2005-02-13 devnull 		free(buf);
141 e5d68728 2005-02-13 devnull 		return -1;
142 e5d68728 2005-02-13 devnull 	}
143 6215fd56 2006-07-23 devnull 	/* Allow regular files in addition to Unix sockets. */
144 d96e9e5d 2020-01-13 rsc 	if((s = open(unix, ORDWR)) >= 0){
145 d96e9e5d 2020-01-13 rsc 		free(buf);
146 6215fd56 2006-07-23 devnull 		return s;
147 d96e9e5d 2020-01-13 rsc 	}
148 fd04aace 2003-11-23 devnull 	free(buf);
149 3409bc9a 2012-06-02 0intro 	if((s = socket(ss.ss_family, SOCK_STREAM, 0)) < 0){
150 835c1e47 2005-07-13 devnull 		werrstr("socket: %r");
151 fd04aace 2003-11-23 devnull 		return -1;
152 835c1e47 2005-07-13 devnull 	}
153 3dade5fe 2012-09-04 rsc 	if(connect(s, (struct sockaddr*)&ss, addrlen(&ss)) < 0){
154 3409bc9a 2012-06-02 0intro 		werrstr("connect %s: %r", ((struct sockaddr_un*)&ss)->sun_path);
155 fd04aace 2003-11-23 devnull 		close(s);
156 fd04aace 2003-11-23 devnull 		return -1;
157 fd04aace 2003-11-23 devnull 	}
158 fd04aace 2003-11-23 devnull 	return s;
159 fd04aace 2003-11-23 devnull }