net/ipv4/netfilter/ipt_owner.c

   1 /* Kernel module to match various things tied to sockets associated with
   2    locally generated outgoing packets. */
   3
   4 /* (C) 2000 Marc Boucher <marc@mbsi.ca>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 as
   8  * published by the Free Software Foundation.
   9  */
  10
  11 #include <linux/module.h>
  12 #include <linux/skbuff.h>
  13 #include <linux/file.h>
  14 #include <linux/rcupdate.h>
  15 #include <net/sock.h>
  16
  17 #include <linux/netfilter_ipv4/ipt_owner.h>
  18 #include <linux/netfilter_ipv4/ip_tables.h>
  19
  20 MODULE_LICENSE("GPL");
  21 MODULE_AUTHOR("Marc Boucher <marc@mbsi.ca>");
  22 MODULE_DESCRIPTION("iptables owner match");
  23
  24 static int
  25 match(const struct sk_buff *skb,
  26       const struct net_device *in,
  27       const struct net_device *out,
  28       const struct xt_match *match,
  29       const void *matchinfo,
  30       int offset,
  31       unsigned int protoff,
  32       int *hotdrop)
  33 {
  34         const struct ipt_owner_info *info = matchinfo;
  35
  36         if (!skb->sk || !skb->sk->sk_socket || !skb->sk->sk_socket->file)
  37                 return 0;
  38
  39         if(info->match & IPT_OWNER_UID) {
  40                 if ((skb->sk->sk_socket->file->f_uid != info->uid) ^
  41                     !!(info->invert & IPT_OWNER_UID))
  42                         return 0;
  43         }
  44
  45         if(info->match & IPT_OWNER_GID) {
  46                 if ((skb->sk->sk_socket->file->f_gid != info->gid) ^
  47                     !!(info->invert & IPT_OWNER_GID))
  48                         return 0;
  49         }
  50
  51         return 1;
  52 }
  53
  54 static int
  55 checkentry(const char *tablename,
  56            const void *ip,
  57            const struct xt_match *match,
  58            void *matchinfo,
  59            unsigned int matchsize,
  60            unsigned int hook_mask)
  61 {
  62         const struct ipt_owner_info *info = matchinfo;
  63
  64         if (info->match & (IPT_OWNER_PID|IPT_OWNER_SID|IPT_OWNER_COMM)) {
  65                 printk("ipt_owner: pid, sid and command matching "
  66                        "not supported anymore\n");
  67                 return 0;
  68         }
  69         return 1;
  70 }
  71
  72 static struct ipt_match owner_match = {
  73         .name           = "owner",
  74         .match          = match,
  75         .matchsize      = sizeof(struct ipt_owner_info),
  76         .hooks          = (1 << NF_IP_LOCAL_OUT) | (1 << NF_IP_POST_ROUTING),
  77         .checkentry     = checkentry,
  78         .me             = THIS_MODULE,
  79 };
  80
  81 static int __init init(void)
  82 {
  83         return ipt_register_match(&owner_match);
  84 }
  85
  86 static void __exit fini(void)
  87 {
  88         ipt_unregister_match(&owner_match);
  89 }
  90
  91 module_init(init);
  92 module_exit(fini);