You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
3.6 KiB
139 lines
3.6 KiB
4 months ago
|
/*
|
||
|
|
* libxt_LED.c - shared library add-on to iptables to add customized LED
|
||
|
|
* trigger support.
|
||
|
|
*
|
||
|
|
* (C) 2008 Adam Nielsen <a.nielsen@shikadi.net>
|
||
|
|
*
|
||
|
|
* This program is free software; you can redistribute it and/or modify
|
||
|
|
* it under the terms of the GNU General Public License version 2 as
|
||
|
|
* published by the Free Software Foundation.
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <xtables.h>
|
||
|
|
#include <linux/netfilter/xt_LED.h>
|
||
|
|
|
||
|
|
enum {
|
||
|
|
O_LED_TRIGGER_ID = 0,
|
||
|
|
O_LED_DELAY,
|
||
|
|
O_LED_ALWAYS_BLINK,
|
||
|
|
};
|
||
|
|
|
||
|
|
#define s struct xt_led_info
|
||
|
|
static const struct xt_option_entry LED_opts[] = {
|
||
|
|
{.name = "led-trigger-id", .id = O_LED_TRIGGER_ID,
|
||
|
|
.flags = XTOPT_MAND, .type = XTTYPE_STRING, .min = 0,
|
||
|
|
.max = sizeof(((struct xt_led_info *)NULL)->id) -
|
||
|
|
sizeof("netfilter-")},
|
||
|
|
{.name = "led-delay", .id = O_LED_DELAY, .type = XTTYPE_STRING},
|
||
|
|
{.name = "led-always-blink", .id = O_LED_ALWAYS_BLINK,
|
||
|
|
.type = XTTYPE_NONE},
|
||
|
|
XTOPT_TABLEEND,
|
||
|
|
};
|
||
|
|
#undef s
|
||
|
|
|
||
|
|
static void LED_help(void)
|
||
|
|
{
|
||
|
|
printf(
|
||
|
|
"LED target options:\n"
|
||
|
|
"--led-trigger-id name suffix for led trigger name\n"
|
||
|
|
"--led-delay ms leave the LED on for this number of\n"
|
||
|
|
" milliseconds after triggering.\n"
|
||
|
|
"--led-always-blink blink on arriving packets, even if\n"
|
||
|
|
" the LED is already on.\n"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void LED_parse(struct xt_option_call *cb)
|
||
|
|
{
|
||
|
|
struct xt_led_info *led = cb->data;
|
||
|
|
|
||
|
|
xtables_option_parse(cb);
|
||
|
|
switch (cb->entry->id) {
|
||
|
|
case O_LED_TRIGGER_ID:
|
||
|
|
snprintf(led->id, sizeof(led->id), "netfilter-%s", cb->arg);
|
||
|
|
break;
|
||
|
|
case O_LED_DELAY:
|
||
|
|
if (strncasecmp(cb->arg, "inf", 3) == 0)
|
||
|
|
led->delay = -1;
|
||
|
|
else if (!xtables_strtoui(cb->arg, NULL, &led->delay, 0, UINT32_MAX))
|
||
|
|
xtables_error(PARAMETER_PROBLEM,
|
||
|
|
"Delay value must be within range 0..%u",
|
||
|
|
UINT32_MAX);
|
||
|
|
break;
|
||
|
|
case O_LED_ALWAYS_BLINK:
|
||
|
|
led->always_blink = 1;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void LED_print(const void *ip, const struct xt_entry_target *target,
|
||
|
|
int numeric)
|
||
|
|
{
|
||
|
|
const struct xt_led_info *led = (void *)target->data;
|
||
|
|
const char *id = led->id + strlen("netfilter-"); /* trim off prefix */
|
||
|
|
|
||
|
|
printf(" led-trigger-id:\"");
|
||
|
|
/* Escape double quotes and backslashes in the ID */
|
||
|
|
while (*id != '\0') {
|
||
|
|
if (*id == '"' || *id == '\\')
|
||
|
|
printf("\\");
|
||
|
|
printf("%c", *id++);
|
||
|
|
}
|
||
|
|
printf("\"");
|
||
|
|
|
||
|
|
if (led->delay == -1)
|
||
|
|
printf(" led-delay:inf");
|
||
|
|
else
|
||
|
|
printf(" led-delay:%dms", led->delay);
|
||
|
|
|
||
|
|
if (led->always_blink)
|
||
|
|
printf(" led-always-blink");
|
||
|
|
}
|
||
|
|
|
||
|
|
static void LED_save(const void *ip, const struct xt_entry_target *target)
|
||
|
|
{
|
||
|
|
const struct xt_led_info *led = (void *)target->data;
|
||
|
|
const char *id = led->id + strlen("netfilter-"); /* trim off prefix */
|
||
|
|
|
||
|
|
printf(" --led-trigger-id \"");
|
||
|
|
/* Escape double quotes and backslashes in the ID */
|
||
|
|
while (*id != '\0') {
|
||
|
|
if (*id == '"' || *id == '\\')
|
||
|
|
printf("\\");
|
||
|
|
printf("%c", *id++);
|
||
|
|
}
|
||
|
|
printf("\"");
|
||
|
|
|
||
|
|
/* Only print the delay if it's not zero (the default) */
|
||
|
|
if (led->delay > 0)
|
||
|
|
printf(" --led-delay %d", led->delay);
|
||
|
|
else if (led->delay == -1)
|
||
|
|
printf(" --led-delay inf");
|
||
|
|
|
||
|
|
/* Only print always_blink if it's not set to the default */
|
||
|
|
if (led->always_blink)
|
||
|
|
printf(" --led-always-blink");
|
||
|
|
}
|
||
|
|
|
||
|
|
static struct xtables_target led_tg_reg = {
|
||
|
|
.version = XTABLES_VERSION,
|
||
|
|
.name = "LED",
|
||
|
|
.family = PF_UNSPEC,
|
||
|
|
.revision = 0,
|
||
|
|
.size = XT_ALIGN(sizeof(struct xt_led_info)),
|
||
|
|
.userspacesize = offsetof(struct xt_led_info, internal_data),
|
||
|
|
.help = LED_help,
|
||
|
|
.print = LED_print,
|
||
|
|
.save = LED_save,
|
||
|
|
.x6_parse = LED_parse,
|
||
|
|
.x6_options = LED_opts,
|
||
|
|
};
|
||
|
|
|
||
|
|
void _init(void)
|
||
|
|
{
|
||
|
|
xtables_register_target(&led_tg_reg);
|
||
|
|
}
|