Преглед изворни кода

Simplify transparent + add support for BSD PF/IPF

z3APA3A пре 8 година
родитељ
комит
b99fa7aaa4
5 измењених фајлова са 35 додато и 29 уклоњено
  1. 1 1
      Makefile.FreeBSD
  2. 1 1
      Makefile.Linux
  3. 1 1
      Makefile.Solaris
  4. 1 1
      Makefile.unix
  5. 31 25
      src/plugins/TransparentPlugin/transparent_plugin.c

+ 1 - 1
Makefile.FreeBSD

@@ -31,7 +31,7 @@ REMOVECOMMAND = rm -f
 TYPECOMMAND = cat
 COMPATLIBS =
 MAKEFILE = Makefile.unix
-PLUGINS = StringsPlugin TrafficPlugin PCREPlugin PamAuth
+PLUGINS = StringsPlugin TrafficPlugin PCREPlugin PamAuth TransparentPlugin
 
 include Makefile.inc
 

+ 1 - 1
Makefile.Linux

@@ -10,7 +10,7 @@
 BUILDDIR =
 CC = gcc
 
-CFLAGS = -g -O2 -c -pthread -DWITHSPLICE -DGETHOSTBYNAME_R -D_THREAD_SAFE -D_REENTRANT -DNOODBC -DWITH_STD_MALLOC -DFD_SETSIZE=4096 -DWITH_POLL
+CFLAGS = -g -O2 -c -pthread -DWITHSPLICE -DGETHOSTBYNAME_R -D_THREAD_SAFE -D_REENTRANT -DNOODBC -DWITH_STD_MALLOC -DFD_SETSIZE=4096 -DWITH_POLL -DWITH_NETFILTER
 COUT = -o 
 LN = gcc
 DCFLAGS = -fpic

+ 1 - 1
Makefile.Solaris

@@ -28,7 +28,7 @@ REMOVECOMMAND = rm -f
 TYPECOMMAND = cat
 COMPATLIBS =
 MAKEFILE = Makefile.Solaris
-PLUGINS = StringsPlugin TrafficPlugin PCREPlugin
+PLUGINS = StringsPlugin TrafficPlugin PCREPlugin TransparentPlugin
 
 include Makefile.inc
 

+ 1 - 1
Makefile.unix

@@ -32,7 +32,7 @@ REMOVECOMMAND = rm -f
 TYPECOMMAND = cat
 COMPATLIBS =
 MAKEFILE = Makefile.unix
-PLUGINS = StringsPlugin TrafficPlugin PCREPlugin PamAuth
+PLUGINS = StringsPlugin TrafficPlugin PCREPlugin PamAuth TransparentPlugin
 
 include Makefile.inc
 

+ 31 - 25
src/plugins/TransparentPlugin/transparent_plugin.c

@@ -6,14 +6,18 @@
 
 */
 
+
+#ifdef WITH_NETFILTER
 #include <sys/utsname.h>
+#endif
 #include "../../structures.h"
 #include "../../proxy.h"
+#ifdef WITH_NETFILTER
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <limits.h>
 #include <linux/netfilter_ipv4.h>
-
+#endif
 
 #ifdef  __cplusplus
 extern "C" {
@@ -38,35 +42,37 @@ static void* transparent_filter_open(void * idata, struct srvparam * param){
 
 static FILTER_ACTION transparent_filter_client(void *fo, struct clientparam * param, void** fc){
 
-#ifdef SO_ORIGINAL_DST
-	struct sockaddr_in addr;
 	socklen_t len;
-	unsigned u;
-	unsigned short p;
-	char addrbuf[24];
-
-		len = sizeof(addr);
-		if(getsockopt(param->clisock, SOL_IP, SO_ORIGINAL_DST,(struct sockaddr *) &addr, &len) || !addr.sin_addr.s_addr){
-			return PASS;
-		}
-		u = ntohl(addr.sin_addr.s_addr);
-		p = ntohs(addr.sin_port);
-		sprintf(addrbuf, "%u.%u.%u.%u:%hu", 
-			((u&0xFF000000)>>24), 
-			((u&0x00FF0000)>>16),
-			((u&0x0000FF00)>>8),
-			((u&0x000000FF)),
-			p);
-
-        	pl->parsehostname(addrbuf, param, 0);
+	char addrbuf[60];
+	int i=0;
+
+	len = sizeof(param->req);
+
+#ifdef WITH_NETFILTER
+#ifdef SO_ORIGINAL_DST
+	if(getsockopt(param->clisock, SOL_IP, SO_ORIGINAL_DST,(struct sockaddr *) &param->req, &len)){
 		return PASS;
+	}
 #else
 #error No SO_ORIGINAL_DST defined
-        	param->srv->logfunc(param, (unsigned char *)"transparent_plugin: No SO_ORIGINAL_DST defined");
-		return REJECT;
+       	param->srv->logfunc(param, (unsigned char *)"transparent_plugin: No SO_ORIGINAL_DST defined");
+	return REJECT;
 #endif
-
-
+#else
+	if(!memcmp(&param->req, &param->sincl,sizeof(param->req))){
+		param->req = param->sincl;
+		param->sincl = param->srv->intsa;
+	}
+#endif
+	addrbuf[i++] = '[';
+	i += pl->myinet_ntop(*SAFAMILY(&param->req), SAADDR(&param->req), (char *)addrbuf + i, sizeof(addrbuf));
+	sprintf((char *)addrbuf+i, "]:%hu", ntohs(*SAPORT(&param->req)));
+#ifdef mystrdup
+#undef mystrdup
+#undef myfree
+#endif
+	if(param->hostname) pl->myfree(param->hostname);
+	param->hostname = pl->mystrdup(addrbuf);
 	return PASS;
 }