blob: 910e011fd6a23c8f5ab630e734b871cf62688214 [file] [log] [blame]
/* Shared library add-on to iptables to add OWNER matching support. */
#include <stdio.h>
#include <netdb.h>
#include <string.h>
#include <stdlib.h>
#include <getopt.h>
#include <pwd.h>
#include <grp.h>
#include <iptables.h>
#include <linux/netfilter_ipv4/ipt_owner.h>
/* Function which prints out usage message. */
static void owner_help(void)
{
#ifdef IPT_OWNER_COMM
printf(
"OWNER match v%s options:\n"
"[!] --uid-owner userid Match local uid\n"
"[!] --gid-owner groupid Match local gid\n"
"[!] --pid-owner processid Match local pid\n"
"[!] --sid-owner sessionid Match local sid\n"
"[!] --cmd-owner name Match local command name\n"
"NOTE: pid, sid and command matching are broken on SMP\n"
"\n",
IPTABLES_VERSION);
#else
printf(
"OWNER match v%s options:\n"
"[!] --uid-owner userid Match local uid\n"
"[!] --gid-owner groupid Match local gid\n"
"[!] --pid-owner processid Match local pid\n"
"[!] --sid-owner sessionid Match local sid\n"
"NOTE: pid and sid matching are broken on SMP\n"
"\n",
IPTABLES_VERSION);
#endif /* IPT_OWNER_COMM */
}
static const struct option owner_opts[] = {
{ "uid-owner", 1, NULL, '1' },
{ "gid-owner", 1, NULL, '2' },
{ "pid-owner", 1, NULL, '3' },
{ "sid-owner", 1, NULL, '4' },
#ifdef IPT_OWNER_COMM
{ "cmd-owner", 1, NULL, '5' },
#endif
{ }
};
/* Function which parses command options; returns true if it
ate an option */
static int owner_parse(int c, char **argv, int invert, unsigned int *flags,
const void *entry, struct xt_entry_match **match)
{
struct ipt_owner_info *ownerinfo = (struct ipt_owner_info *)(*match)->data;
switch (c) {
char *end;
struct passwd *pwd;
struct group *grp;
case '1':
check_inverse(optarg, &invert, &optind, 0);
if ((pwd = getpwnam(optarg)))
ownerinfo->uid = pwd->pw_uid;
else {
ownerinfo->uid = strtoul(optarg, &end, 0);
if (*end != '\0' || end == optarg)
exit_error(PARAMETER_PROBLEM, "Bad OWNER UID value `%s'", optarg);
}
if (invert)
ownerinfo->invert |= IPT_OWNER_UID;
ownerinfo->match |= IPT_OWNER_UID;
*flags = 1;
break;
case '2':
check_inverse(optarg, &invert, &optind, 0);
if ((grp = getgrnam(optarg)))
ownerinfo->gid = grp->gr_gid;
else {
ownerinfo->gid = strtoul(optarg, &end, 0);
if (*end != '\0' || end == optarg)
exit_error(PARAMETER_PROBLEM, "Bad OWNER GID value `%s'", optarg);
}
if (invert)
ownerinfo->invert |= IPT_OWNER_GID;
ownerinfo->match |= IPT_OWNER_GID;
*flags = 1;
break;
case '3':
check_inverse(optarg, &invert, &optind, 0);
ownerinfo->pid = strtoul(optarg, &end, 0);
if (*end != '\0' || end == optarg)
exit_error(PARAMETER_PROBLEM, "Bad OWNER PID value `%s'", optarg);
if (invert)
ownerinfo->invert |= IPT_OWNER_PID;
ownerinfo->match |= IPT_OWNER_PID;
*flags = 1;
break;
case '4':
check_inverse(optarg, &invert, &optind, 0);
ownerinfo->sid = strtoul(optarg, &end, 0);
if (*end != '\0' || end == optarg)
exit_error(PARAMETER_PROBLEM, "Bad OWNER SID value `%s'", optarg);
if (invert)
ownerinfo->invert |= IPT_OWNER_SID;
ownerinfo->match |= IPT_OWNER_SID;
*flags = 1;
break;
#ifdef IPT_OWNER_COMM
case '5':
check_inverse(optarg, &invert, &optind, 0);
if(strlen(optarg) > sizeof(ownerinfo->comm))
exit_error(PARAMETER_PROBLEM, "OWNER CMD `%s' too long, max %u characters", optarg, (unsigned int)sizeof(ownerinfo->comm));
strncpy(ownerinfo->comm, optarg, sizeof(ownerinfo->comm));
ownerinfo->comm[sizeof(ownerinfo->comm)-1] = '\0';
if (invert)
ownerinfo->invert |= IPT_OWNER_COMM;
ownerinfo->match |= IPT_OWNER_COMM;
*flags = 1;
break;
#endif
default:
return 0;
}
return 1;
}
static void
print_item(struct ipt_owner_info *info, u_int8_t flag, int numeric, char *label)
{
if(info->match & flag) {
if (info->invert & flag)
printf("! ");
printf(label);
switch(info->match & flag) {
case IPT_OWNER_UID:
if(!numeric) {
struct passwd *pwd = getpwuid(info->uid);
if(pwd && pwd->pw_name) {
printf("%s ", pwd->pw_name);
break;
}
/* FALLTHROUGH */
}
printf("%u ", info->uid);
break;
case IPT_OWNER_GID:
if(!numeric) {
struct group *grp = getgrgid(info->gid);
if(grp && grp->gr_name) {
printf("%s ", grp->gr_name);
break;
}
/* FALLTHROUGH */
}
printf("%u ", info->gid);
break;
case IPT_OWNER_PID:
printf("%u ", info->pid);
break;
case IPT_OWNER_SID:
printf("%u ", info->sid);
break;
#ifdef IPT_OWNER_COMM
case IPT_OWNER_COMM:
printf("%.*s ", (int)sizeof(info->comm), info->comm);
break;
#endif
default:
break;
}
}
}
/* Final check; must have specified --own. */
static void owner_check(unsigned int flags)
{
if (!flags)
exit_error(PARAMETER_PROBLEM,
"OWNER match: You must specify one or more options");
}
/* Prints out the matchinfo. */
static void owner_print(const void *ip, const struct xt_entry_match *match,
int numeric)
{
struct ipt_owner_info *info = (struct ipt_owner_info *)match->data;
print_item(info, IPT_OWNER_UID, numeric, "OWNER UID match ");
print_item(info, IPT_OWNER_GID, numeric, "OWNER GID match ");
print_item(info, IPT_OWNER_PID, numeric, "OWNER PID match ");
print_item(info, IPT_OWNER_SID, numeric, "OWNER SID match ");
#ifdef IPT_OWNER_COMM
print_item(info, IPT_OWNER_COMM, numeric, "OWNER CMD match ");
#endif
}
/* Saves the union ipt_matchinfo in parsable form to stdout. */
static void owner_save(const void *ip, const struct xt_entry_match *match)
{
struct ipt_owner_info *info = (struct ipt_owner_info *)match->data;
print_item(info, IPT_OWNER_UID, 0, "--uid-owner ");
print_item(info, IPT_OWNER_GID, 0, "--gid-owner ");
print_item(info, IPT_OWNER_PID, 0, "--pid-owner ");
print_item(info, IPT_OWNER_SID, 0, "--sid-owner ");
#ifdef IPT_OWNER_COMM
print_item(info, IPT_OWNER_COMM, 0, "--cmd-owner ");
#endif
}
static struct iptables_match owner_match = {
.name = "owner",
.version = IPTABLES_VERSION,
.size = IPT_ALIGN(sizeof(struct ipt_owner_info)),
.userspacesize = IPT_ALIGN(sizeof(struct ipt_owner_info)),
.help = owner_help,
.parse = owner_parse,
.final_check = owner_check,
.print = owner_print,
.save = owner_save,
.extra_opts = owner_opts,
};
void _init(void)
{
register_match(&owner_match);
}