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 memset(&sa, 0, sizeof sa);
64 fd04aace 2003-11-23 devnull memmove(&sa.sin_addr, &host, 4);
65 fd04aace 2003-11-23 devnull sa.sin_family = AF_INET;
66 fd04aace 2003-11-23 devnull sa.sin_port = htons(port);
67 fd04aace 2003-11-23 devnull if((s = socket(AF_INET, proto, 0)) < 0)
68 fd04aace 2003-11-23 devnull return -1;
70 e5d68728 2005-02-13 devnull if(local){
71 e5d68728 2005-02-13 devnull buf = strdup(local);
72 e5d68728 2005-02-13 devnull if(buf == nil){
73 e5d68728 2005-02-13 devnull close(s);
74 e5d68728 2005-02-13 devnull return -1;
76 e5d68728 2005-02-13 devnull if(p9dialparse(buf, &net, &unix, &host, &port) < 0){
77 e5d68728 2005-02-13 devnull badlocal:
78 e5d68728 2005-02-13 devnull free(buf);
79 e5d68728 2005-02-13 devnull close(s);
80 e5d68728 2005-02-13 devnull return -1;
82 e5d68728 2005-02-13 devnull if(unix){
83 e5d68728 2005-02-13 devnull werrstr("bad local address %s for dial %s", local, addr);
84 e5d68728 2005-02-13 devnull goto badlocal;
86 e5d68728 2005-02-13 devnull memset(&sal, 0, sizeof sal);
87 e5d68728 2005-02-13 devnull memmove(&sal.sin_addr, &local, 4);
88 e5d68728 2005-02-13 devnull sal.sin_family = AF_INET;
89 e5d68728 2005-02-13 devnull sal.sin_port = htons(port);
90 e5d68728 2005-02-13 devnull sn = sizeof n;
91 e5d68728 2005-02-13 devnull if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0
92 e5d68728 2005-02-13 devnull && n == SOCK_STREAM){
94 e5d68728 2005-02-13 devnull setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
96 e5d68728 2005-02-13 devnull if(bind(s, (struct sockaddr*)&sal, sizeof sal) < 0)
97 e5d68728 2005-02-13 devnull goto badlocal;
98 e5d68728 2005-02-13 devnull free(buf);
101 fd04aace 2003-11-23 devnull if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){
102 fd04aace 2003-11-23 devnull close(s);
103 fd04aace 2003-11-23 devnull return -1;
105 d51419bf 2004-02-09 devnull if(proto == SOCK_STREAM){
106 d51419bf 2004-02-09 devnull int one = 1;
107 d51419bf 2004-02-09 devnull setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
109 fd04aace 2003-11-23 devnull return s;
112 e5d68728 2005-02-13 devnull if(local){
113 e5d68728 2005-02-13 devnull werrstr("local address not supported on unix network");
114 e5d68728 2005-02-13 devnull free(buf);
115 e5d68728 2005-02-13 devnull return -1;
117 fd04aace 2003-11-23 devnull memset(&su, 0, sizeof su);
118 fd04aace 2003-11-23 devnull su.sun_family = AF_UNIX;
119 fd04aace 2003-11-23 devnull if(strlen(unix)+1 > sizeof su.sun_path){
120 fd04aace 2003-11-23 devnull werrstr("unix socket name too long");
121 fd04aace 2003-11-23 devnull free(buf);
122 fd04aace 2003-11-23 devnull return -1;
124 fd04aace 2003-11-23 devnull strcpy(su.sun_path, unix);
125 fd04aace 2003-11-23 devnull free(buf);
126 835c1e47 2005-07-13 devnull if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0){
127 835c1e47 2005-07-13 devnull werrstr("socket: %r");
128 fd04aace 2003-11-23 devnull return -1;
130 fd04aace 2003-11-23 devnull if(connect(s, (struct sockaddr*)&su, sizeof su) < 0){
131 835c1e47 2005-07-13 devnull werrstr("connect %s: %r", su.sun_path);
132 fd04aace 2003-11-23 devnull close(s);
133 fd04aace 2003-11-23 devnull return -1;
135 fd04aace 2003-11-23 devnull return s;
