added iptable extension codule

pull/1931/head
Sasmita Panda 9 months ago
parent c21969fa41
commit 83f5ee8355

@ -0,0 +1,73 @@
CC?=gcc
CFLAGS ?= -O2 -Wall -Wextra -Wno-sign-compare -Wno-unused-parameter -Wstrict-prototypes
CFLAGS += -shared -fPIC
ifneq ($(RTPENGINE_VERSION),)
CFLAGS += -DRTPENGINE_VERSION="\"$(RTPENGINE_VERSION)\""
else
DPKG_PRSCHNGLG= $(shell which dpkg-parsechangelog 2>/dev/null)
ifneq ($(DPKG_PRSCHNGLG),)
CFLAGS += -DRTPENGINE_VERSION="\"$(shell dpkg-parsechangelog -l../debian/changelog | awk '/^Version: / {print $$2}')\""
else
CFLAGS += -DRTPENGINE_VERSION="\"undefined\""
endif
endif
XTABLES_INSTALLED=$(shell \
if pkg-config --exists xtables; then \
echo 'xtables found'; \
fi)
ifneq ($(XTABLES_INSTALLED),)
XTABLES = 1
CFLAGS += $(shell pkg-config --cflags --libs xtables)
else
XTABLES = $(shell test -e /usr/include/xtables.h && echo 1)
endif
IPTABLES = $(shell test -e /usr/include/iptables.h && echo 1)
IP6TABLES = $(shell test -e /usr/include/ip6tables.h && echo 1)
WORK=0
.PHONY: all module clean install
all: module
ifeq ($(XTABLES),1)
WORK=1
module: libxt_RTPENGINE.so
XTABLES_DIR ?= $(shell pkg-config xtables --variable=xtlibdir || echo /lib/xtables)
libxt_RTPENGINE.so: libxt_RTPENGINE.c
$(CC) $(CFLAGS) -o libxt_RTPENGINE.so libxt_RTPENGINE.c
else
ifeq ($(IPTABLES),1)
WORK=1
module: libipt_RTPENGINE.so
endif
ifeq ($(IP6TABLES),1)
WORK=1
module: libip6t_RTPENGINE.so
endif
libipt_RTPENGINE.so: libxt_RTPENGINE.c
$(CC) $(CFLAGS) -D__ipt -o libipt_RTPENGINE.so libxt_RTPENGINE.c
libip6t_RTPENGINE.so: libxt_RTPENGINE.c
$(CC) $(CFLAGS) -D__ip6t -o libip6t_RTPENGINE.so libxt_RTPENGINE.c
endif
ifeq ($(WORK),0)
module:
@echo No iptables dev packages installed!
endif
clean:
rm -f libxt_RTPENGINE.so libipt_RTPENGINE.so libip6t_RTPENGINE.so
install:
install -D libxt_RTPENGINE.so $(DESTDIR)$(XTABLES_DIR)/libxt_RTPENGINE.so

@ -0,0 +1,183 @@
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#ifndef _init
#define _init __attribute__((constructor)) _INIT
#endif
#if defined(__ipt)
#include <iptables.h>
#elif defined(__ip6t)
#include <ip6tables.h>
#else
#include <xtables.h>
#endif
#include <linux/netfilter.h>
#if defined(__ipt)
#include <linux/netfilter_ipv4/ip_tables.h>
#elif defined(__ip6t)
#include <linux/netfilter_ipv6/ip6_tables.h>
#else
#include <linux/netfilter/x_tables.h>
#endif
#include "../kernel-module/xt_RTPENGINE.h"
static void help(void) {
printf(
"RTPENGINE (version %s) target options:\n"
" --id <id>\n"
" Unique ID for this instance\n",
RTPENGINE_VERSION
);
}
#if defined(__ipt)
static int parse(int c,
char **argv,
int invert,
unsigned int *flags,
const struct ipt_entry *entry,
struct ipt_entry_target **target) {
#elif defined(__ip6t)
static int parse(int c,
char **argv,
int invert,
unsigned int *flags,
const struct ip6t_entry *entry,
struct ip6t_entry_target **target) {
#else
static int parse(int c,
char **argv,
int invert,
unsigned int *flags,
const void *entry,
struct xt_entry_target **target) {
#endif
struct xt_rtpengine_info *info = (void *) (*target)->data;
if (c == '1') {
info->id = atoi(optarg);
if (flags)
*flags = 1;
}
else
return 0;
return 1;
}
static void final_check(unsigned int flags) {
#if defined(__ipt) || defined(__ip6t)
if (!flags)
exit_error(PARAMETER_PROBLEM, "You must specify --id");
#else
if (!flags)
xtables_error(PARAMETER_PROBLEM, "You must specify --id");
#endif
}
#if defined(__ipt)
static void print(const struct ipt_ip *ip, const struct xt_entry_target *target, int numeric) {
#elif defined(__ip6t)
static void print(const struct ip6t_ip6 *ip, const struct xt_entry_target *target, int numeric) {
#else
static void print(const void *ip, const struct xt_entry_target *target, int numeric) {
#endif
struct xt_rtpengine_info *info = (void *) target->data;
printf(" RTPENGINE id:%u", info->id);
}
#if defined(__ipt)
static void save(const struct ipt_ip *ip, const struct xt_entry_target *target) {
#elif defined(__ip6t)
static void save(const struct ip6t_ip6 *ip, const struct xt_entry_target *target) {
#else
static void save(const void *ip, const struct xt_entry_target *target) {
#endif
struct xt_rtpengine_info *info = (void *) target->data;
printf(" --id %u", info->id);
}
static struct option opts[] = {
{ "id", 1, NULL, '1' },
{ NULL, },
};
#if defined(__ipt)
static struct iptables_target rtpengine4 = {
.name = "RTPENGINE",
.version = "1.3.6",
.size = IPT_ALIGN(sizeof(struct xt_rtpengine_info)),
.userspacesize = IPT_ALIGN(sizeof(struct xt_rtpengine_info)),
.help = help,
.parse = parse,
.final_check = final_check,
.print = print,
.save = save,
.extra_opts = opts,
};
#elif defined(__ip6t)
static struct ip6tables_target rtpengine6 = {
.name = "RTPENGINE",
.version = "1.3.6",
.size = IP6T_ALIGN(sizeof(struct xt_rtpengine_info)),
.userspacesize = IP6T_ALIGN(sizeof(struct xt_rtpengine_info)),
.help = help,
.parse = parse,
.final_check = final_check,
.print = print,
.save = save,
.extra_opts = opts,
};
#else
static struct xtables_target rtpengine4 = {
.name = "RTPENGINE",
.family = NFPROTO_IPV4,
.version = XTABLES_VERSION,
.size = XT_ALIGN(sizeof(struct xt_rtpengine_info)),
.userspacesize = XT_ALIGN(sizeof(struct xt_rtpengine_info)),
.help = help,
.parse = parse,
.final_check = final_check,
.print = print,
.save = save,
.extra_opts = opts,
};
static struct xtables_target rtpengine6 = {
.name = "RTPENGINE",
.family = NFPROTO_IPV6,
.version = XTABLES_VERSION,
.size = XT_ALIGN(sizeof(struct xt_rtpengine_info)),
.userspacesize = XT_ALIGN(sizeof(struct xt_rtpengine_info)),
.help = help,
.parse = parse,
.final_check = final_check,
.print = print,
.save = save,
.extra_opts = opts,
};
#endif
void _init(void) {
#if defined(__ipt)
register_target(&rtpengine4);
#elif defined(__ip6t)
register_target6(&rtpengine6);
#else
xtables_register_target(&rtpengine4);
xtables_register_target(&rtpengine6);
#endif
}
Loading…
Cancel
Save