1 fd04aace 2003-11-23 devnull #include <u.h>
2 fd04aace 2003-11-23 devnull #include <libc.h>
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
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>
19 fd04aace 2003-11-23 devnull #undef unix
20 835c1e47 2005-07-13 devnull #define unix xunix
22 3409bc9a 2012-06-02 0intro static int
23 3409bc9a 2012-06-02 0intro isany(struct sockaddr_storage *ss)
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);
36 3dade5fe 2012-09-04 rsc addrlen(struct sockaddr_storage *ss)
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);
50 e5d68728 2005-02-13 devnull p9dial(char *addr, char *local, char *dummy2, int *dummy3)
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;
58 3409bc9a 2012-06-02 0intro struct sockaddr_storage ss, ssl;
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;
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;
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;
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 *)");
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;
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;
91 fd04aace 2003-11-23 devnull free(buf);
93 3409bc9a 2012-06-02 0intro if((s = socket(ss.ss_family, proto, 0)) < 0)
94 fd04aace 2003-11-23 devnull return -1;
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;
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;
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;
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){
116 e5d68728 2005-02-13 devnull setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
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);
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;
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);
135 fd04aace 2003-11-23 devnull return s;
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;
143 6215fd56 2006-07-23 devnull /* Allow regular files in addition to Unix sockets. */
144 6215fd56 2006-07-23 devnull if((s = open(unix, ORDWR)) >= 0)
145 6215fd56 2006-07-23 devnull return s;
146 fd04aace 2003-11-23 devnull free(buf);
147 3409bc9a 2012-06-02 0intro if((s = socket(ss.ss_family, SOCK_STREAM, 0)) < 0){
148 835c1e47 2005-07-13 devnull werrstr("socket: %r");
149 fd04aace 2003-11-23 devnull return -1;
151 3dade5fe 2012-09-04 rsc if(connect(s, (struct sockaddr*)&ss, addrlen(&ss)) < 0){
152 3409bc9a 2012-06-02 0intro werrstr("connect %s: %r", ((struct sockaddr_un*)&ss)->sun_path);
153 fd04aace 2003-11-23 devnull close(s);
154 fd04aace 2003-11-23 devnull return -1;
156 fd04aace 2003-11-23 devnull return s;