DeforaNetworks/zzuf
3
0
Fork 0
Code Releases Activity

* Divert connect() so that -p/--ports now works with outbound connections.

Browse Source
This commit is contained in:
Sam Hocevar
2007-11-02 23:23:25 +00:00
committed by sam
parent 82db9143ca
commit 92fbf5821f
1 changed files with 46 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lib-fd.c
+46
View File
@@ -89,6 +89,10 @@ static int (*ORIG(accept)) (int sockfd, struct sockaddr *addr,
static int (*ORIG(bind)) (int sockfd, const struct sockaddr *my_addr,
SOCKLEN_T addrlen);
#endif
#if defined HAVE_CONNECT
static int (*ORIG(connect)) (int sockfd, const struct sockaddr *serv_addr,
SOCKLEN_T addrlen);
#endif
#if defined HAVE_SOCKET
static int (*ORIG(socket)) (int domain, int type, int protocol);
#endif
@@ -230,6 +234,48 @@ int NEW(bind)(int sockfd, const struct sockaddr *my_addr, SOCKLEN_T addrlen)
}
#endif
#if defined HAVE_CONNECT
int NEW(connect)(int sockfd, const struct sockaddr *serv_addr,
SOCKLEN_T addrlen)
{
int ret;
LOADSYM(connect);
ret = ORIG(connect)(sockfd, serv_addr, addrlen);
if(!_zz_ready || _zz_islocked(-1) || !_zz_network)
return ret;
if(ret >= 0)
{
const struct sockaddr_in* in = (const struct sockaddr_in *)serv_addr;
long int port;
switch(serv_addr->sa_family)
{
case AF_INET:
#if defined AF_INET6
case AF_INET6:
#endif
case AF_UNSPEC:
port = ntohs(in->sin_port);
if(!_zz_portwatched(port))
{
_zz_unregister(sockfd);
return ret;
}
break;
default:
break;
}
debug("%s(%i, %p, %i) = %i", __func__,
sockfd, serv_addr, (int)addrlen, ret);
}
return ret;
}
#endif
#if defined HAVE_SOCKET
int NEW(socket)(int domain, int type, int protocol)
{