cirno/plan9port
1
0
Fork 0
mirror of https://github.com/9fans/plan9port.git synced 2025-01-12 11:10:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add local address to dial

Browse source
This commit is contained in:
rsc 2005-02-13 18:33:26 +00:00
parent fb7a39bc56
commit e5d68728eb
1 changed files with 42 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
src/lib9/dial.c
View file

@ -19,18 +19,20 @@
#undef unix #undef unix
int int
p9dial(char *addr, char *dummy1, char *dummy2, int *dummy3) p9dial(char *addr, char *local, char *dummy2, int *dummy3)
{ {
char *buf; char *buf;
char *net, *unix; char *net, *unix;
u32int host; u32int host;
int port; int port;
int proto; int proto;
struct sockaddr_in sa; socklen_t sn;
int n;
struct sockaddr_in sa, sal;
struct sockaddr_un su; struct sockaddr_un su;
int s; int s;
if(dummy1 || dummy2 || dummy3){ if(dummy2 || dummy3){
werrstr("cannot handle extra arguments in dial"); werrstr("cannot handle extra arguments in dial");
return -1; return -1;
} }
@ -63,6 +65,38 @@ p9dial(char *addr, char *dummy1, char *dummy2, int *dummy3)
sa.sin_port = htons(port); sa.sin_port = htons(port);
if((s = socket(AF_INET, proto, 0)) < 0) if((s = socket(AF_INET, proto, 0)) < 0)
return -1; 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);
}
if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){ if(connect(s, (struct sockaddr*)&sa, sizeof sa) < 0){
close(s); close(s);
return -1; return -1;
@ -74,6 +108,11 @@ p9dial(char *addr, char *dummy1, char *dummy2, int *dummy3)
return s; return s;
Unix: Unix:
if(local){
werrstr("local address not supported on unix network");
free(buf);
return -1;
}
memset(&su, 0, sizeof su); memset(&su, 0, sizeof su);
su.sun_family = AF_UNIX; su.sun_family = AF_UNIX;
if(strlen(unix)+1 > sizeof su.sun_path){ if(strlen(unix)+1 > sizeof su.sun_path){