GopherProxy

	dial.c - 9base - revived minimalist port of Plan 9 userland to Unix
	git clone git://git.suckless.org/9base
	Log
	Files
	Refs
	README
	LICENSE
	---
	dial.c (2943B)
	---
	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", ne…
	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", loca…
	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,…
	93 && n == SOCK_STREAM){
	94 n = 1;
	95 setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&…
	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, siz…
	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