1 fd04aace 2003-11-23 devnull #include <u.h>
 2 fd04aace 2003-11-23 devnull #include <libc.h>
 3 fd04aace 2003-11-23 devnull 
 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
12 fd04aace 2003-11-23 devnull 
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>
18 fd04aace 2003-11-23 devnull 
19 fd04aace 2003-11-23 devnull 
20 fd04aace 2003-11-23 devnull extern int _p9dialparse(char*, char**, char**, u32int*, int*);
21 fd04aace 2003-11-23 devnull #undef unix
22 fd04aace 2003-11-23 devnull 
23 fd04aace 2003-11-23 devnull int
24 fd04aace 2003-11-23 devnull p9dial(char *addr, char *dummy1, char *dummy2, int *dummy3)
25 fd04aace 2003-11-23 devnull {
26 fd04aace 2003-11-23 devnull 	char *buf;
27 fd04aace 2003-11-23 devnull 	char *net, *unix;
28 fd04aace 2003-11-23 devnull 	u32int host;
29 fd04aace 2003-11-23 devnull 	int port;
30 fd04aace 2003-11-23 devnull 	int proto;
31 fd04aace 2003-11-23 devnull 	struct sockaddr_in sa;	
32 fd04aace 2003-11-23 devnull 	struct sockaddr_un su;
33 fd04aace 2003-11-23 devnull 	int s;
34 fd04aace 2003-11-23 devnull 
35 fd04aace 2003-11-23 devnull 	if(dummy1 || dummy2 || dummy3){
36 fd04aace 2003-11-23 devnull 		werrstr("cannot handle extra arguments in dial");
37 fd04aace 2003-11-23 devnull 		return -1;
38 fd04aace 2003-11-23 devnull 	}
39 fd04aace 2003-11-23 devnull 
40 fd04aace 2003-11-23 devnull 	buf = strdup(addr);
41 fd04aace 2003-11-23 devnull 	if(buf == nil)
42 fd04aace 2003-11-23 devnull 		return -1;
43 fd04aace 2003-11-23 devnull 
44 fd04aace 2003-11-23 devnull 	if(_p9dialparse(buf, &net, &unix, &host, &port) < 0){
45 fd04aace 2003-11-23 devnull 		free(buf);
46 fd04aace 2003-11-23 devnull 		return -1;
47 fd04aace 2003-11-23 devnull 	}
48 fd04aace 2003-11-23 devnull 
49 fd04aace 2003-11-23 devnull 	if(strcmp(net, "tcp") == 0)
50 fd04aace 2003-11-23 devnull 		proto = SOCK_STREAM;
51 fd04aace 2003-11-23 devnull 	else if(strcmp(net, "udp") == 0)
52 fd04aace 2003-11-23 devnull 		proto = SOCK_DGRAM;
53 fd04aace 2003-11-23 devnull 	else if(strcmp(net, "unix") == 0)
54 fd04aace 2003-11-23 devnull 		goto Unix;
55 fd04aace 2003-11-23 devnull 	else{
56 fd04aace 2003-11-23 devnull 		werrstr("can only handle tcp, udp, and unix: not %s", net);
57 fd04aace 2003-11-23 devnull 		free(buf);
58 fd04aace 2003-11-23 devnull 		return -1;
59 fd04aace 2003-11-23 devnull 	}
60 fd04aace 2003-11-23 devnull 	free(buf);
61 fd04aace 2003-11-23 devnull 
62 fd04aace 2003-11-23 devnull 	memset(&sa, 0, sizeof sa);
63 fd04aace 2003-11-23 devnull 	memmove(&sa.sin_addr, &host, 4);
64 fd04aace 2003-11-23 devnull 	sa.sin_family = AF_INET;
65 fd04aace 2003-11-23 devnull 	sa.sin_port = htons(port);
66 fd04aace 2003-11-23 devnull 	if((s = socket(AF_INET, proto, 0)) < 0)
67 fd04aace 2003-11-23 devnull 		return -1;
68 fd04aace 2003-11-23 devnull 	if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){
69 fd04aace 2003-11-23 devnull 		close(s);
70 fd04aace 2003-11-23 devnull 		return -1;
71 fd04aace 2003-11-23 devnull 	}
72 d51419bf 2004-02-09 devnull 	if(proto == SOCK_STREAM){
73 d51419bf 2004-02-09 devnull 		int one = 1;
74 d51419bf 2004-02-09 devnull 		setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one);
75 d51419bf 2004-02-09 devnull 	}
76 fd04aace 2003-11-23 devnull 	return s;
77 fd04aace 2003-11-23 devnull 
78 fd04aace 2003-11-23 devnull Unix:
79 fd04aace 2003-11-23 devnull 	memset(&su, 0, sizeof su);
80 fd04aace 2003-11-23 devnull 	su.sun_family = AF_UNIX;
81 fd04aace 2003-11-23 devnull 	if(strlen(unix)+1 > sizeof su.sun_path){
82 fd04aace 2003-11-23 devnull 		werrstr("unix socket name too long");
83 fd04aace 2003-11-23 devnull 		free(buf);
84 fd04aace 2003-11-23 devnull 		return -1;
85 fd04aace 2003-11-23 devnull 	}
86 fd04aace 2003-11-23 devnull 	strcpy(su.sun_path, unix);
87 fd04aace 2003-11-23 devnull 	free(buf);
88 fd04aace 2003-11-23 devnull 	if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
89 fd04aace 2003-11-23 devnull 		return -1;
90 fd04aace 2003-11-23 devnull 	if(connect(s, (struct sockaddr*)&su, sizeof su) < 0){
91 fd04aace 2003-11-23 devnull 		close(s);
92 fd04aace 2003-11-23 devnull 		return -1;
93 fd04aace 2003-11-23 devnull 	}
94 fd04aace 2003-11-23 devnull 	return s;
95 fd04aace 2003-11-23 devnull }
96 fd04aace 2003-11-23 devnull