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;
50 fd04aace 2003-11-23 devnull if(strcmp(net, "tcp") == 0)
51 fd04aace 2003-11-23 devnull proto = SOCK_STREAM;
52 fd04aace 2003-11-23 devnull else if(strcmp(net, "udp") == 0)
53 fd04aace 2003-11-23 devnull proto = SOCK_DGRAM;
54 fd04aace 2003-11-23 devnull else if(strcmp(net, "unix") == 0)
55 fd04aace 2003-11-23 devnull goto Unix;
57 fd04aace 2003-11-23 devnull werrstr("can only handle tcp, udp, and unix: not %s", net);
58 fd04aace 2003-11-23 devnull free(buf);
59 fd04aace 2003-11-23 devnull return -1;
61 fd04aace 2003-11-23 devnull free(buf);
63 fd04aace 2003-11-23 devnull if((s = socket(AF_INET, proto, 0)) < 0)
64 fd04aace 2003-11-23 devnull return -1;
66 e5d68728 2005-02-13 devnull if(local){
67 e5d68728 2005-02-13 devnull buf = strdup(local);
68 e5d68728 2005-02-13 devnull if(buf == nil){
69 e5d68728 2005-02-13 devnull close(s);
70 e5d68728 2005-02-13 devnull return -1;
72 e5d68728 2005-02-13 devnull if(p9dialparse(buf, &net, &unix, &host, &port) < 0){
73 e5d68728 2005-02-13 devnull badlocal:
74 e5d68728 2005-02-13 devnull free(buf);
75 e5d68728 2005-02-13 devnull close(s);
76 e5d68728 2005-02-13 devnull return -1;
78 e5d68728 2005-02-13 devnull if(unix){
79 e5d68728 2005-02-13 devnull werrstr("bad local address %s for dial %s", local, addr);
80 e5d68728 2005-02-13 devnull goto badlocal;
82 e5d68728 2005-02-13 devnull memset(&sal, 0, sizeof sal);
83 e5d68728 2005-02-13 devnull memmove(&sal.sin_addr, &local, 4);
84 e5d68728 2005-02-13 devnull sal.sin_family = AF_INET;
85 e5d68728 2005-02-13 devnull sal.sin_port = htons(port);
86 e5d68728 2005-02-13 devnull sn = sizeof n;
87 e5d68728 2005-02-13 devnull if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0
88 e5d68728 2005-02-13 devnull && n == SOCK_STREAM){
90 e5d68728 2005-02-13 devnull setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
92 e5d68728 2005-02-13 devnull if(bind(s, (struct sockaddr*)&sal, sizeof sal) < 0)
93 e5d68728 2005-02-13 devnull goto badlocal;
94 e5d68728 2005-02-13 devnull free(buf);
98 74f9b0c6 2005-12-26 devnull setsockopt(s, SOL_SOCKET, SO_BROADCAST, &n, sizeof n);
99 3d484b0d 2005-12-29 devnull if(host != 0){
100 3d484b0d 2005-12-29 devnull memset(&sa, 0, sizeof sa);
101 3d484b0d 2005-12-29 devnull memmove(&sa.sin_addr, &host, 4);
102 3d484b0d 2005-12-29 devnull sa.sin_family = AF_INET;
103 3d484b0d 2005-12-29 devnull sa.sin_port = htons(port);
104 3d484b0d 2005-12-29 devnull if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){
105 3d484b0d 2005-12-29 devnull close(s);
106 3d484b0d 2005-12-29 devnull return -1;
109 d51419bf 2004-02-09 devnull if(proto == SOCK_STREAM){
110 d51419bf 2004-02-09 devnull int one = 1;
111 d51419bf 2004-02-09 devnull setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
113 fd04aace 2003-11-23 devnull return s;
116 e5d68728 2005-02-13 devnull if(local){
117 e5d68728 2005-02-13 devnull werrstr("local address not supported on unix network");
118 e5d68728 2005-02-13 devnull free(buf);
119 e5d68728 2005-02-13 devnull return -1;
121 6215fd56 2006-07-23 devnull /* Allow regular files in addition to Unix sockets. */
122 6215fd56 2006-07-23 devnull if((s = open(unix, ORDWR)) >= 0)
123 6215fd56 2006-07-23 devnull return s;
124 fd04aace 2003-11-23 devnull memset(&su, 0, sizeof su);
125 fd04aace 2003-11-23 devnull su.sun_family = AF_UNIX;
126 fd04aace 2003-11-23 devnull if(strlen(unix)+1 > sizeof su.sun_path){
127 fd04aace 2003-11-23 devnull werrstr("unix socket name too long");
128 fd04aace 2003-11-23 devnull free(buf);
129 fd04aace 2003-11-23 devnull return -1;
131 fd04aace 2003-11-23 devnull strcpy(su.sun_path, unix);
132 fd04aace 2003-11-23 devnull free(buf);
133 835c1e47 2005-07-13 devnull if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0){
134 835c1e47 2005-07-13 devnull werrstr("socket: %r");
135 fd04aace 2003-11-23 devnull return -1;
137 fd04aace 2003-11-23 devnull if(connect(s, (struct sockaddr*)&su, sizeof su) < 0){
138 835c1e47 2005-07-13 devnull werrstr("connect %s: %r", su.sun_path);
139 fd04aace 2003-11-23 devnull close(s);
140 fd04aace 2003-11-23 devnull return -1;
142 fd04aace 2003-11-23 devnull return s;
