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
23 e5d68728 2005-02-13 devnull p9dial(char *addr, char *local, char *dummy2, int *dummy3)
25 fd04aace 2003-11-23 devnull char *buf;
26 fd04aace 2003-11-23 devnull char *net, *unix;
27 fd04aace 2003-11-23 devnull u32int host;
28 fd04aace 2003-11-23 devnull int port;
29 fd04aace 2003-11-23 devnull int proto;
30 e5d68728 2005-02-13 devnull socklen_t sn;
32 e5d68728 2005-02-13 devnull struct sockaddr_in sa, sal;
33 fd04aace 2003-11-23 devnull struct sockaddr_un su;
36 e5d68728 2005-02-13 devnull if(dummy2 || dummy3){
37 fd04aace 2003-11-23 devnull werrstr("cannot handle extra arguments in dial");
38 fd04aace 2003-11-23 devnull return -1;
41 fd04aace 2003-11-23 devnull buf = strdup(addr);
42 fd04aace 2003-11-23 devnull if(buf == nil)
43 fd04aace 2003-11-23 devnull return -1;
45 e42882dc 2004-06-16 devnull if(p9dialparse(buf, &net, &unix, &host, &port) < 0){
46 fd04aace 2003-11-23 devnull free(buf);
47 fd04aace 2003-11-23 devnull return -1;
49 5186b55a 2009-08-17 fshahriar if(strcmp(net, "unix") != 0 && host == 0){
50 6d558e12 2009-08-15 rsc werrstr("invalid dial address 0.0.0.0 (aka *)");
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;
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;
66 fd04aace 2003-11-23 devnull free(buf);
68 fd04aace 2003-11-23 devnull if((s = socket(AF_INET, proto, 0)) < 0)
69 fd04aace 2003-11-23 devnull return -1;
71 e5d68728 2005-02-13 devnull if(local){
72 e5d68728 2005-02-13 devnull buf = strdup(local);
73 e5d68728 2005-02-13 devnull if(buf == nil){
74 e5d68728 2005-02-13 devnull close(s);
75 e5d68728 2005-02-13 devnull return -1;
77 e5d68728 2005-02-13 devnull if(p9dialparse(buf, &net, &unix, &host, &port) < 0){
78 e5d68728 2005-02-13 devnull badlocal:
79 e5d68728 2005-02-13 devnull free(buf);
80 e5d68728 2005-02-13 devnull close(s);
81 e5d68728 2005-02-13 devnull return -1;
83 e5d68728 2005-02-13 devnull if(unix){
84 e5d68728 2005-02-13 devnull werrstr("bad local address %s for dial %s", local, addr);
85 e5d68728 2005-02-13 devnull goto badlocal;
87 e5d68728 2005-02-13 devnull memset(&sal, 0, sizeof sal);
88 e5d68728 2005-02-13 devnull memmove(&sal.sin_addr, &local, 4);
89 e5d68728 2005-02-13 devnull sal.sin_family = AF_INET;
90 e5d68728 2005-02-13 devnull sal.sin_port = htons(port);
91 e5d68728 2005-02-13 devnull sn = sizeof n;
92 e5d68728 2005-02-13 devnull if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0
93 e5d68728 2005-02-13 devnull && n == SOCK_STREAM){
95 e5d68728 2005-02-13 devnull setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
97 e5d68728 2005-02-13 devnull if(bind(s, (struct sockaddr*)&sal, sizeof sal) < 0)
98 e5d68728 2005-02-13 devnull goto badlocal;
99 e5d68728 2005-02-13 devnull free(buf);
103 74f9b0c6 2005-12-26 devnull setsockopt(s, SOL_SOCKET, SO_BROADCAST, &n, sizeof n);
104 3d484b0d 2005-12-29 devnull if(host != 0){
105 3d484b0d 2005-12-29 devnull memset(&sa, 0, sizeof sa);
106 3d484b0d 2005-12-29 devnull memmove(&sa.sin_addr, &host, 4);
107 3d484b0d 2005-12-29 devnull sa.sin_family = AF_INET;
108 3d484b0d 2005-12-29 devnull sa.sin_port = htons(port);
109 3d484b0d 2005-12-29 devnull if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){
110 3d484b0d 2005-12-29 devnull close(s);
111 3d484b0d 2005-12-29 devnull return -1;
114 d51419bf 2004-02-09 devnull if(proto == SOCK_STREAM){
115 d51419bf 2004-02-09 devnull int one = 1;
116 d51419bf 2004-02-09 devnull setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
118 fd04aace 2003-11-23 devnull return s;
121 e5d68728 2005-02-13 devnull if(local){
122 e5d68728 2005-02-13 devnull werrstr("local address not supported on unix network");
123 e5d68728 2005-02-13 devnull free(buf);
124 e5d68728 2005-02-13 devnull return -1;
126 6215fd56 2006-07-23 devnull /* Allow regular files in addition to Unix sockets. */
127 6215fd56 2006-07-23 devnull if((s = open(unix, ORDWR)) >= 0)
128 6215fd56 2006-07-23 devnull return s;
129 fd04aace 2003-11-23 devnull memset(&su, 0, sizeof su);
130 fd04aace 2003-11-23 devnull su.sun_family = AF_UNIX;
131 fd04aace 2003-11-23 devnull if(strlen(unix)+1 > sizeof su.sun_path){
132 fd04aace 2003-11-23 devnull werrstr("unix socket name too long");
133 fd04aace 2003-11-23 devnull free(buf);
134 fd04aace 2003-11-23 devnull return -1;
136 fd04aace 2003-11-23 devnull strcpy(su.sun_path, unix);
137 fd04aace 2003-11-23 devnull free(buf);
138 835c1e47 2005-07-13 devnull if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0){
139 835c1e47 2005-07-13 devnull werrstr("socket: %r");
140 fd04aace 2003-11-23 devnull return -1;
142 fd04aace 2003-11-23 devnull if(connect(s, (struct sockaddr*)&su, sizeof su) < 0){
143 835c1e47 2005-07-13 devnull werrstr("connect %s: %r", su.sun_path);
144 fd04aace 2003-11-23 devnull close(s);
145 fd04aace 2003-11-23 devnull return -1;
147 fd04aace 2003-11-23 devnull return s;
