#include #include #undef accept #undef announce #undef dial #undef setnetmtpt #undef hangup #undef listen #undef netmkaddr #undef reject #include #include #include #include #include #undef unix #define unix xunix int p9dial(char *addr, char *local, char *dummy2, int *dummy3) { char *buf; char *net, *unix; u32int host; int port; int proto; socklen_t sn; int n; struct sockaddr_in sa, sal; struct sockaddr_un su; int s; if(dummy2 || dummy3){ werrstr("cannot handle extra arguments in dial"); return -1; } buf = strdup(addr); if(buf == nil) return -1; if(p9dialparse(buf, &net, &unix, &host, &port) < 0){ free(buf); return -1; } if(strcmp(net, "unix") != 0 && host == 0){ werrstr("invalid dial address 0.0.0.0 (aka *)"); free(buf); return -1; } if(strcmp(net, "tcp") == 0) proto = SOCK_STREAM; else if(strcmp(net, "udp") == 0) proto = SOCK_DGRAM; else if(strcmp(net, "unix") == 0) goto Unix; else{ werrstr("can only handle tcp, udp, and unix: not %s", net); free(buf); return -1; } free(buf); if((s = socket(AF_INET, proto, 0)) < 0) return -1; if(local){ buf = strdup(local); if(buf == nil){ close(s); return -1; } if(p9dialparse(buf, &net, &unix, &host, &port) < 0){ badlocal: free(buf); close(s); return -1; } if(unix){ werrstr("bad local address %s for dial %s", local, addr); goto badlocal; } memset(&sal, 0, sizeof sal); memmove(&sal.sin_addr, &local, 4); sal.sin_family = AF_INET; sal.sin_port = htons(port); sn = sizeof n; if(port && getsockopt(s, SOL_SOCKET, SO_TYPE, (void*)&n, &sn) >= 0 && n == SOCK_STREAM){ n = 1; setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&n, sizeof n); } if(bind(s, (struct sockaddr*)&sal, sizeof sal) < 0) goto badlocal; free(buf); } n = 1; setsockopt(s, SOL_SOCKET, SO_BROADCAST, &n, sizeof n); if(host != 0){ memset(&sa, 0, sizeof sa); memmove(&sa.sin_addr, &host, 4); sa.sin_family = AF_INET; sa.sin_port = htons(port); if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){ close(s); return -1; } } if(proto == SOCK_STREAM){ int one = 1; setsockopt(s, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one); } return s; Unix: if(local){ werrstr("local address not supported on unix network"); free(buf); return -1; } /* Allow regular files in addition to Unix sockets. */ if((s = open(unix, ORDWR)) >= 0) return s; memset(&su, 0, sizeof su); su.sun_family = AF_UNIX; if(strlen(unix)+1 > sizeof su.sun_path){ werrstr("unix socket name too long"); free(buf); return -1; } strcpy(su.sun_path, unix); free(buf); if((s = socket(AF_UNIX, SOCK_STREAM, 0)) < 0){ werrstr("socket: %r"); return -1; } if(connect(s, (struct sockaddr*)&su, sizeof su) < 0){ werrstr("connect %s: %r", su.sun_path); close(s); return -1; } return s; }