net/netfilter/nf_log.c

   1 #include <linux/kernel.h>
   2 #include <linux/init.h>
   3 #include <linux/module.h>
   4 #include <linux/proc_fs.h>
   5 #include <linux/skbuff.h>
   6 #include <linux/netfilter.h>
   7 #include <linux/seq_file.h>
   8 #include <net/protocol.h>
   9 #include <net/netfilter/nf_log.h>
  10
  11 #include "nf_internals.h"
  12
  13 /* Internal logging interface, which relies on the real
  14    LOG target modules */
  15
  16 #define NF_LOG_PREFIXLEN                128
  17 #define NFLOGGER_NAME_LEN               64
  18
  19 static const struct nf_logger *nf_loggers[NFPROTO_NUMPROTO] __read_mostly;
  20 static struct list_head nf_loggers_l[NFPROTO_NUMPROTO] __read_mostly;
  21 static DEFINE_MUTEX(nf_log_mutex);
  22
  23 static struct nf_logger *__find_logger(int pf, const char *str_logger)
  24 {
  25         struct nf_logger *t;
  26
  27         list_for_each_entry(t, &nf_loggers_l[pf], list[pf]) {
  28                 if (!strnicmp(str_logger, t->name, strlen(t->name)))
  29                         return t;
  30         }
  31
  32         return NULL;
  33 }
  34
  35 /* return EEXIST if the same logger is registred, 0 on success. */
  36 int nf_log_register(u_int8_t pf, struct nf_logger *logger)
  37 {
  38         int i;
  39
  40         if (pf >= ARRAY_SIZE(nf_loggers))
  41                 return -EINVAL;
  42
  43         for (i = 0; i < ARRAY_SIZE(logger->list); i++)
  44                 INIT_LIST_HEAD(&logger->list[i]);
  45
  46         mutex_lock(&nf_log_mutex);
  47
  48         if (pf == NFPROTO_UNSPEC) {
  49                 for (i = NFPROTO_UNSPEC; i < NFPROTO_NUMPROTO; i++)
  50                         list_add_tail(&(logger->list[i]), &(nf_loggers_l[i]));
  51         } else {
  52                 /* register at end of list to honor first register win */
  53                 list_add_tail(&logger->list[pf], &nf_loggers_l[pf]);
  54                 if (nf_loggers[pf] == NULL)
  55                         rcu_assign_pointer(nf_loggers[pf], logger);
  56         }
  57
  58         mutex_unlock(&nf_log_mutex);
  59
  60         return 0;
  61 }
  62 EXPORT_SYMBOL(nf_log_register);
  63
  64 void nf_log_unregister(struct nf_logger *logger)
  65 {
  66         int i;
  67
  68         mutex_lock(&nf_log_mutex);
  69         for (i = 0; i < ARRAY_SIZE(nf_loggers); i++) {
  70                 if (nf_loggers[i] == logger)
  71                         rcu_assign_pointer(nf_loggers[i], NULL);
  72                 list_del(&logger->list[i]);
  73         }
  74         mutex_unlock(&nf_log_mutex);
  75
  76         synchronize_rcu();
  77 }
  78 EXPORT_SYMBOL(nf_log_unregister);
  79
  80 int nf_log_bind_pf(u_int8_t pf, const struct nf_logger *logger)
  81 {
  82         mutex_lock(&nf_log_mutex);
  83         if (__find_logger(pf, logger->name) == NULL) {
  84                 mutex_unlock(&nf_log_mutex);
  85                 return -ENOENT;
  86         }
  87         rcu_assign_pointer(nf_loggers[pf], logger);
  88         mutex_unlock(&nf_log_mutex);
  89         return 0;
  90 }
  91 EXPORT_SYMBOL(nf_log_bind_pf);
  92
  93 void nf_log_unbind_pf(u_int8_t pf)
  94 {
  95         mutex_lock(&nf_log_mutex);
  96         rcu_assign_pointer(nf_loggers[pf], NULL);
  97         mutex_unlock(&nf_log_mutex);
  98 }
  99 EXPORT_SYMBOL(nf_log_unbind_pf);
 100
 101 void nf_log_packet(u_int8_t pf,
 102                    unsigned int hooknum,
 103                    const struct sk_buff *skb,
 104                    const struct net_device *in,
 105                    const struct net_device *out,
 106                    const struct nf_loginfo *loginfo,
 107                    const char *fmt, ...)
 108 {
 109         va_list args;
 110         char prefix[NF_LOG_PREFIXLEN];
 111         const struct nf_logger *logger;
 112
 113         rcu_read_lock();
 114         logger = rcu_dereference(nf_loggers[pf]);
 115         if (logger) {
 116                 va_start(args, fmt);
 117                 vsnprintf(prefix, sizeof(prefix), fmt, args);
 118                 va_end(args);
 119                 logger->logfn(pf, hooknum, skb, in, out, loginfo, prefix);
 120         }
 121         rcu_read_unlock();
 122 }
 123 EXPORT_SYMBOL(nf_log_packet);
 124
 125 #ifdef CONFIG_PROC_FS
 126 static void *seq_start(struct seq_file *seq, loff_t *pos)
 127 {
 128         mutex_lock(&nf_log_mutex);
 129
 130         if (*pos >= ARRAY_SIZE(nf_loggers))
 131                 return NULL;
 132
 133         return pos;
 134 }
 135
 136 static void *seq_next(struct seq_file *s, void *v, loff_t *pos)
 137 {
 138         (*pos)++;
 139
 140         if (*pos >= ARRAY_SIZE(nf_loggers))
 141                 return NULL;
 142
 143         return pos;
 144 }
 145
 146 static void seq_stop(struct seq_file *s, void *v)
 147 {
 148         mutex_unlock(&nf_log_mutex);
 149 }
 150
 151 static int seq_show(struct seq_file *s, void *v)
 152 {
 153         loff_t *pos = v;
 154         const struct nf_logger *logger;
 155         struct nf_logger *t;
 156         int ret;
 157
 158         logger = nf_loggers[*pos];
 159
 160         if (!logger)
 161                 ret = seq_printf(s, "%2lld NONE (", *pos);
 162         else
 163                 ret = seq_printf(s, "%2lld %s (", *pos, logger->name);
 164
 165         if (ret < 0)
 166                 return ret;
 167
 168         list_for_each_entry(t, &nf_loggers_l[*pos], list[*pos]) {
 169                 ret = seq_printf(s, "%s", t->name);
 170                 if (ret < 0)
 171                         return ret;
 172                 if (&t->list[*pos] != nf_loggers_l[*pos].prev) {
 173                         ret = seq_printf(s, ",");
 174                         if (ret < 0)
 175                                 return ret;
 176                 }
 177         }
 178
 179         return seq_printf(s, ")\n");
 180 }
 181
 182 static const struct seq_operations nflog_seq_ops = {
 183         .start  = seq_start,
 184         .next   = seq_next,
 185         .stop   = seq_stop,
 186         .show   = seq_show,
 187 };
 188
 189 static int nflog_open(struct inode *inode, struct file *file)
 190 {
 191         return seq_open(file, &nflog_seq_ops);
 192 }
 193
 194 static const struct file_operations nflog_file_ops = {
 195         .owner   = THIS_MODULE,
 196         .open    = nflog_open,
 197         .read    = seq_read,
 198         .llseek  = seq_lseek,
 199         .release = seq_release,
 200 };
 201
 202
 203 #endif /* PROC_FS */
 204
 205 #ifdef CONFIG_SYSCTL
 206 static struct ctl_path nf_log_sysctl_path[] = {
 207         { .procname = "net", },
 208         { .procname = "netfilter", },
 209         { .procname = "nf_log", },
 210         { }
 211 };
 212
 213 static char nf_log_sysctl_fnames[NFPROTO_NUMPROTO-NFPROTO_UNSPEC][3];
 214 static struct ctl_table nf_log_sysctl_table[NFPROTO_NUMPROTO+1];
 215 static struct ctl_table_header *nf_log_dir_header;
 216
 217 static int nf_log_proc_dostring(ctl_table *table, int write,
 218                          void __user *buffer, size_t *lenp, loff_t *ppos)
 219 {
 220         const struct nf_logger *logger;
 221         char buf[NFLOGGER_NAME_LEN];
 222         size_t size = *lenp;
 223         int r = 0;
 224         int tindex = (unsigned long)table->extra1;
 225
 226         if (write) {
 227                 if (size > sizeof(buf))
 228                         size = sizeof(buf);
 229                 if (copy_from_user(buf, buffer, size))
 230                         return -EFAULT;
 231
 232                 if (!strcmp(buf, "NONE")) {
 233                         nf_log_unbind_pf(tindex);
 234                         return 0;
 235                 }
 236                 mutex_lock(&nf_log_mutex);
 237                 logger = __find_logger(tindex, buf);
 238                 if (logger == NULL) {
 239                         mutex_unlock(&nf_log_mutex);
 240                         return -ENOENT;
 241                 }
 242                 rcu_assign_pointer(nf_loggers[tindex], logger);
 243                 mutex_unlock(&nf_log_mutex);
 244         } else {
 245                 mutex_lock(&nf_log_mutex);
 246                 logger = nf_loggers[tindex];
 247                 if (!logger)
 248                         table->data = "NONE";
 249                 else
 250                         table->data = logger->name;
 251                 r = proc_dostring(table, write, buffer, lenp, ppos);
 252                 mutex_unlock(&nf_log_mutex);
 253         }
 254
 255         return r;
 256 }
 257
 258 static __init int netfilter_log_sysctl_init(void)
 259 {
 260         int i;
 261
 262         for (i = NFPROTO_UNSPEC; i < NFPROTO_NUMPROTO; i++) {
 263                 snprintf(nf_log_sysctl_fnames[i-NFPROTO_UNSPEC], 3, "%d", i);
 264                 nf_log_sysctl_table[i].procname =
 265                         nf_log_sysctl_fnames[i-NFPROTO_UNSPEC];
 266                 nf_log_sysctl_table[i].data = NULL;
 267                 nf_log_sysctl_table[i].maxlen =
 268                         NFLOGGER_NAME_LEN * sizeof(char);
 269                 nf_log_sysctl_table[i].mode = 0644;
 270                 nf_log_sysctl_table[i].proc_handler = nf_log_proc_dostring;
 271                 nf_log_sysctl_table[i].extra1 = (void *)(unsigned long) i;
 272         }
 273
 274         nf_log_dir_header = register_sysctl_paths(nf_log_sysctl_path,
 275                                        nf_log_sysctl_table);
 276         if (!nf_log_dir_header)
 277                 return -ENOMEM;
 278
 279         return 0;
 280 }
 281 #else
 282 static __init int netfilter_log_sysctl_init(void)
 283 {
 284         return 0;
 285 }
 286 #endif /* CONFIG_SYSCTL */
 287
 288 int __init netfilter_log_init(void)
 289 {
 290         int i, r;
 291 #ifdef CONFIG_PROC_FS
 292         if (!proc_create("nf_log", S_IRUGO,
 293                          proc_net_netfilter, &nflog_file_ops))
 294                 return -1;
 295 #endif
 296
 297         /* Errors will trigger panic, unroll on error is unnecessary. */
 298         r = netfilter_log_sysctl_init();
 299         if (r < 0)
 300                 return r;
 301
 302         for (i = NFPROTO_UNSPEC; i < NFPROTO_NUMPROTO; i++)
 303                 INIT_LIST_HEAD(&(nf_loggers_l[i]));
 304
 305         return 0;
 306 }