1 #include <u.h>
2 #include <libc.h>
3 
4 #undef	accept
5 #undef	announce
6 #undef	dial
7 #undef	setnetmtpt
8 #undef	hangup
9 #undef	listen
10 #undef	netmkaddr
11 #undef	reject
12 
13 #include <sys/socket.h>
14 #include <netinet/in.h>
15 #include <netinet/tcp.h>
16 #include <sys/un.h>
17 #include <netdb.h>
18 
19 #undef unix
20 #define unix xunix
21 
22 int
23 p9dial(char *addr, char *local, char *dummy2, int *dummy3)
24 {
25 	char *buf;
26 	char *net, *unix;
27 	u32int host;
28 	int port;
29 	int proto;
30 	socklen_t sn;
31 	int n;
32 	struct sockaddr_in sa, sal;	
33 	struct sockaddr_un su;
34 	int s;
35 
36 	if(dummy2 || dummy3){
37 		werrstr("cannot handle extra arguments in dial");
38 		return -1;
39 	}
40 
41 	buf = strdup(addr);
42 	if(buf == nil)
43 		return -1;
44 
45 	if(p9dialparse(buf, &net, &unix, &host, &port) < 0){
46 		free(buf);
47 		return -1;
48 	}
49 	if(strcmp(net, "unix") != 0 && host == 0){
50 		werrstr("invalid dial address 0.0.0.0 (aka *)");
51 		free(buf);
52 		return -1;
53 	}
54 
55 	if(strcmp(net, "tcp") == 0)
56 		proto = SOCK_STREAM;
57 	else if(strcmp(net, "udp") == 0)
58 		proto = SOCK_DGRAM;
59 	else if(strcmp(net, "unix") == 0)
60 		goto Unix;
61 	else{
62 		werrstr("can only handle tcp, udp, and unix: not %s", net);
63 		free(buf);
64 		return -1;
65 	}
66 	free(buf);
67 
68 	if((s = socket(AF_INET, proto, 0)) < 0)
69 		return -1;
70 		
71 	if(local){
72 		buf = strdup(local);
73 		if(buf == nil){
74 			close(s);
75 			return -1;
76 		}
77 		if(p9dialparse(buf, &net, &unix, &host, &port) < 0){
78 		badlocal:
79 			free(buf);
80 			close(s);
81 			return -1;
82 		}
83 		if(unix){
84 			werrstr("bad local address %s for dial %s", local, addr);
85 			goto badlocal;
86 		}
87 		memset(&sal, 0, sizeof sal);
88 		memmove(&sal.sin_addr, &local, 4);
89 		sal.sin_family = AF_INET;
90 		sal.sin_port = htons(port);
91 		sn = sizeof n;
92 		if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0
93 		&& n == SOCK_STREAM){
94 			n = 1;
95 			setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n);
96 		}
97 		if(bind(s, (struct sockaddr*)&sal, sizeof sal) < 0)
98 			goto badlocal;
99 		free(buf);
100 	}
101 
102 	n = 1;
103 	setsockopt(s, SOL_SOCKET, SO_BROADCAST, &n, sizeof n);
104 	if(host != 0){
105 		memset(&sa, 0, sizeof sa);
106 		memmove(&sa.sin_addr, &host, 4);
107 		sa.sin_family = AF_INET;
108 		sa.sin_port = htons(port);
109 		if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){
110 			close(s);
111 			return -1;
112 		}
113 	}
114 	if(proto == SOCK_STREAM){
115 		int one = 1;
116 		setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
117 	}
118 	return s;
119 
120 Unix:
121 	if(local){
122 		werrstr("local address not supported on unix network");
123 		free(buf);
124 		return -1;
125 	}
126 	/* Allow regular files in addition to Unix sockets. */
127 	if((s = open(unix, ORDWR)) >= 0)
128 		return s;
129 	memset(&su, 0, sizeof su);
130 	su.sun_family = AF_UNIX;
131 	if(strlen(unix)+1 > sizeof su.sun_path){
132 		werrstr("unix socket name too long");
133 		free(buf);
134 		return -1;
135 	}
136 	strcpy(su.sun_path, unix);
137 	free(buf);
138 	if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0){
139 		werrstr("socket: %r");
140 		return -1;
141 	}
142 	if(connect(s, (struct sockaddr*)&su, sizeof su) < 0){
143 		werrstr("connect %s: %r", su.sun_path);
144 		close(s);
145 		return -1;
146 	}
147 	return s;
148 }
149