- Add conf_class.c, conf_class.h

git-svn-id: svn://svn.ircd-hybrid.org/svnroot/ircd-hybrid/trunk@1633 82007160-df01-0410-b94d-b575c5fd34c7

2 files changed, 522 insertions, 0 deletions

diff --git a/include/conf_class.h b/include/conf_class.h
new file mode 100644
index 0000000..7e4f0e2
--- /dev/null
+++ b/include/conf_class.h
@@ -0,0 +1,73 @@
+/*
+ *  ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd).
+ *
+ *  Copyright (C) 2002 by the past and present ircd coders, and others.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
+ *  USA
+ */
+
+/*! \file
+ * \brief Configuration managment for class{} blocks
+ * \version $Id: conf_class.h,v 1.54 2007/10/21 21:57:05 devil Exp $
+ */
+
+
+#ifndef INCLUDED_conf_class_h
+#define INCLUDED_conf_class_h
+
+struct ClassItem
+{
+  char *name;
+  dlink_node node;
+  dlink_list list_ipv4;         /* base of per cidr ipv4 client link list */
+  dlink_list list_ipv6;         /* base of per cidr ipv6 client link list */
+  unsigned int max_sendq;
+  unsigned int max_recvq;
+  int con_freq;
+  int ping_freq;
+  int ping_warning;
+  int max_total;
+  int max_local;
+  int max_global;
+  int max_ident;
+  int max_perip;
+  int ref_count;
+  int cidr_bitlen_ipv4;
+  int cidr_bitlen_ipv6;
+  int number_per_cidr;
+  int active;
+};
+
+/* address of default class conf */
+extern struct ClassItem *class_default;
+
+extern struct ClassItem *class_make(void);
+extern const dlink_list *class_get_list(void);
+extern void class_free(struct ClassItem *);
+extern void class_init(void);
+extern const char *get_client_class(const dlink_list *const);
+extern unsigned int get_client_ping(const dlink_list *const, int *);
+extern unsigned int get_sendq(const dlink_list *const);
+extern unsigned int get_recvq(const dlink_list *const);
+extern struct ClassItem *class_find(const char *, int);
+extern void class_mark_for_deletion(void);
+extern void class_delete_marked(void);
+extern void *check_client(va_list args);
+extern void destroy_cidr_class(struct ClassItem *);
+extern void rebuild_cidr_class(struct ClassItem *, struct ClassItem *);
+extern int cidr_limit_reached(int, struct irc_ssaddr *, struct ClassItem *);
+extern void remove_from_cidr_check(struct irc_ssaddr *, struct ClassItem *);
+#endif
diff --git a/src/conf_class.c b/src/conf_class.c
new file mode 100755
index 0000000..ff1fc63
--- /dev/null
+++ b/src/conf_class.c
@@ -0,0 +1,449 @@
+/*
+ *  ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd).
+ *
+ *  Copyright (C) 2002 by the past and present ircd coders, and others.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
+ *  USA
+ */
+
+/*! \file
+ * \brief Configuration managment for class{} blocks
+ * \version $Id: conf_class.c,v 1.54 2007/10/21 21:57:05 devil Exp $
+ */
+
+#include "stdinc.h"
+#include "list.h"
+#include "ircd_defs.h"
+#include "balloc.h"
+#include "conf.h"
+#include "s_serv.h"
+#include "resv.h"
+#include "channel.h"
+#include "client.h"
+#include "event.h"
+#include "hook.h"
+#include "irc_string.h"
+#include "s_bsd.h"
+#include "ircd.h"
+#include "listener.h"
+#include "hostmask.h"
+#include "modules.h"
+#include "numeric.h"
+#include "fdlist.h"
+#include "log.h"
+#include "send.h"
+#include "s_gline.h"
+#include "memory.h"
+#include "irc_res.h"
+#include "userhost.h"
+#include "s_user.h"
+#include "channel_mode.h"
+#include "parse.h"
+#include "s_misc.h"
+#include "conf_db.h"
+
+
+struct ClassItem *class_default;
+
+static dlink_list class_list = { NULL, NULL, 0 };
+
+
+const dlink_list *
+class_get_list(void)
+{
+  return &class_list;
+}
+
+struct ClassItem *
+class_make(void)
+{
+  struct ClassItem *class = MyMalloc(sizeof(*class));
+
+  dlinkAdd(class, &class->node, &class_list);
+
+  return class;
+}
+
+void
+class_free(struct ClassItem *class)
+{
+  assert(class);
+  assert(class->active    == 0);
+  assert(class->ref_count == 0);
+
+  dlinkDelete(&class->node, &class_list);
+  MyFree(class->name);
+  MyFree(class);
+}
+
+void
+class_init(void)
+{
+  class_default = class_make();
+
+  class_default->active = 1;
+  DupString(class_default->name, "default");
+  class_default->con_freq  = DEFAULT_CONNECTFREQUENCY;
+  class_default->ping_freq = DEFAULT_PINGFREQUENCY;
+  class_default->max_total = MAXIMUM_LINKS_DEFAULT;
+  class_default->max_sendq = DEFAULT_SENDQ;
+  class_default->max_recvq = DEFAULT_RECVQ;
+
+  client_check_cb = register_callback("check_client", check_client);
+}
+
+const char *
+get_client_class(const dlink_list *const list)
+{
+  const dlink_node *ptr = NULL;
+
+  if ((ptr = list->head)) {
+    const struct MaskItem *conf = ptr->data;
+
+    assert(conf->class);
+    assert(conf->type & (CONF_OPERATOR | CONF_CLIENT | CONF_SERVER));
+
+    return conf->class->name;
+  }
+
+  return class_default->name;
+}
+
+unsigned int
+get_client_ping(const dlink_list *const list, int *pingwarn)
+{
+  const dlink_node *ptr = NULL;
+  int ping = 0;
+
+  if ((ptr = list->head)) {
+    const struct MaskItem *conf = ptr->data;
+
+    assert(aconf->class);
+    assert(aconf->type & (CONF_OPERATOR | CONF_CLIENT | CONF_SERVER));
+
+    ping = get_conf_ping(conf, pingwarn);
+    if (ping > 0)
+      return ping;
+  }
+
+  return class_default->ping_freq;
+}
+
+unsigned int
+get_sendq(const dlink_list *const list)
+{
+  const dlink_node *ptr = NULL;
+
+  if ((ptr = list->head)) {
+    const struct MaskItem *conf = ptr->data;
+
+    assert(conf->class);
+    assert(conf->type & (CONF_OPERATOR | CONF_CLIENT | CONF_SERVER));
+
+    return conf->class->max_sendq;
+  }
+
+  return class_default->max_sendq;
+}
+
+unsigned int
+get_recvq(const dlink_list *const list)
+{
+  const dlink_node *ptr = NULL;
+
+  if ((ptr = list->head)) {
+    const struct MaskItem *conf = ptr->data;
+
+    assert(conf->class);
+    assert(conf->type & (CONF_OPERATOR | CONF_CLIENT | CONF_SERVER));
+
+    return conf->class->max_recvq;
+  }
+
+  return class_default->max_recvq;
+}
+
+/*
+ * inputs       - Integer (Number of class)
+ * output       - Pointer to ClassItem struct. Non-NULL expected
+ * side effects - NONE
+ */
+struct ClassItem *
+class_find(const char *name, int active)
+{
+  dlink_node *ptr = NULL;
+
+  DLINK_FOREACH(ptr, class_list.head) {
+    struct ClassItem *class = ptr->data;
+
+    if (!irccmp(class->name, name))
+      return active && !class->active ? NULL : class;
+  }
+
+  return NULL;
+}
+
+/*
+ * We don't delete the class table, rather mark all entries for deletion.
+ * The table is cleaned up by delete_marked_classes. - avalon
+ */
+void
+class_mark_for_deletion(void)
+{
+  dlink_node *ptr = NULL;
+
+  DLINK_FOREACH(ptr, class_list.head)
+  {
+    struct ClassItem *class = ptr->data;
+
+    if (class != class_default)
+      class->active = 0;
+  }
+}
+
+void
+class_delete_marked(void)
+{
+  dlink_node *ptr = NULL, *ptr_next = NULL;
+
+  DLINK_FOREACH_SAFE(ptr, ptr_next, class_list.head) {
+    struct ClassItem *class = ptr->data;
+
+    if (!class->active && !class->ref_count)
+    {
+      destroy_cidr_class(class);
+      class_free(class);
+    }
+  }
+}
+
+/*
+ * cidr_limit_reached
+ *
+ * inputs	- int flag allowing over_rule of limits
+ *		- pointer to the ip to be added
+ *		- pointer to the class
+ * output	- non zero if limit reached
+ *		  0 if limit not reached
+ * side effects	-
+ */
+int
+cidr_limit_reached(int over_rule,
+		   struct irc_ssaddr *ip, struct ClassItem *class)
+{
+  dlink_node *ptr = NULL;
+  struct CidrItem *cidr = NULL;
+
+  if (class->number_per_cidr <= 0)
+    return 0;
+
+  if (ip->ss.ss_family == AF_INET)
+  {
+    if (class->cidr_bitlen_ipv4 <= 0)
+      return 0;
+
+    DLINK_FOREACH(ptr, class->list_ipv4.head)
+    {
+      cidr = ptr->data;
+      if (match_ipv4(ip, &cidr->mask, class->cidr_bitlen_ipv4))
+      {
+        if (!over_rule && (cidr->number_on_this_cidr >= class->number_per_cidr))
+          return -1;
+        cidr->number_on_this_cidr++;
+        return 0;
+      }
+    }
+    cidr = MyMalloc(sizeof(struct CidrItem));
+    cidr->number_on_this_cidr = 1;
+    cidr->mask = *ip;
+    mask_addr(&cidr->mask, class->cidr_bitlen_ipv4);
+    dlinkAdd(cidr, &cidr->node, &class->list_ipv4);
+  }
+#ifdef IPV6
+  else if (class->cidr_bitlen_ipv6 > 0)
+  {
+    DLINK_FOREACH(ptr, class->list_ipv6.head)
+    {
+      cidr = ptr->data;
+      if (match_ipv6(ip, &cidr->mask, class->cidr_bitlen_ipv6))
+      {
+        if (!over_rule && (cidr->number_on_this_cidr >= class->number_per_cidr))
+          return -1;
+        cidr->number_on_this_cidr++;
+        return 0;
+      }
+    }
+    cidr = MyMalloc(sizeof(struct CidrItem));
+    cidr->number_on_this_cidr = 1;
+    cidr->mask = *ip;
+    mask_addr(&cidr->mask, class->cidr_bitlen_ipv6);
+    dlinkAdd(cidr, &cidr->node, &class->list_ipv6);
+  }
+#endif
+  return 0;
+}
+
+/*
+ * remove_from_cidr_check
+ *
+ * inputs	- pointer to the ip to be removed
+ *		- pointer to the class
+ * output	- NONE
+ * side effects	-
+ */
+void
+remove_from_cidr_check(struct irc_ssaddr *ip, struct ClassItem *aclass)
+{
+  dlink_node *ptr = NULL;
+  dlink_node *next_ptr = NULL;
+  struct CidrItem *cidr;
+
+  if (aclass->number_per_cidr == 0)
+    return;
+
+  if (ip->ss.ss_family == AF_INET)
+  {
+    if (aclass->cidr_bitlen_ipv4 <= 0)
+      return;
+
+    DLINK_FOREACH_SAFE(ptr, next_ptr, aclass->list_ipv4.head)
+    {
+      cidr = ptr->data;
+      if (match_ipv4(ip, &cidr->mask, aclass->cidr_bitlen_ipv4))
+      {
+	cidr->number_on_this_cidr--;
+	if (cidr->number_on_this_cidr == 0)
+	{
+	  dlinkDelete(ptr, &aclass->list_ipv4);
+	  MyFree(cidr);
+	  return;
+	}
+      }
+    }
+  }
+#ifdef IPV6
+  else if (aclass->cidr_bitlen_ipv6 > 0)
+  {
+    DLINK_FOREACH_SAFE(ptr, next_ptr, aclass->list_ipv6.head)
+    {
+      cidr = ptr->data;
+      if (match_ipv6(ip, &cidr->mask, aclass->cidr_bitlen_ipv6))
+      {
+	cidr->number_on_this_cidr--;
+	if (cidr->number_on_this_cidr == 0)
+	{
+	  dlinkDelete(ptr, &aclass->list_ipv6);
+	  MyFree(cidr);
+	  return;
+	}
+      }
+    }
+  }
+#endif
+}
+
+static void
+rebuild_cidr_list(int aftype, struct ClassItem *oldcl, struct ClassItem *newcl,
+                  dlink_list *old_list, dlink_list *new_list, int changed)
+{
+  dlink_node *ptr;
+  struct Client *client_p;
+  struct MaskItem *conf;
+
+  if (!changed)
+  {
+    *new_list = *old_list;
+    old_list->head = old_list->tail = NULL;
+    old_list->length = 0;
+    return;
+  }
+
+  DLINK_FOREACH(ptr, local_client_list.head)
+  {
+    client_p = ptr->data;
+    if (client_p->localClient->aftype != aftype)
+      continue;
+    if (dlink_list_length(&client_p->localClient->confs) == 0)
+      continue;
+
+    conf = client_p->localClient->confs.tail->data;
+    if (conf->type == CONF_CLIENT)
+      if (conf->class == oldcl)
+        cidr_limit_reached(1, &client_p->localClient->ip, newcl);
+  }
+}
+
+/*
+ * rebuild_cidr_class
+ *
+ * inputs	- pointer to old conf
+ *		- pointer to new_class
+ * output	- none
+ * side effects	- rebuilds the class link list of cidr blocks
+ */
+void
+rebuild_cidr_class(struct ClassItem *old_class, struct ClassItem *new_class)
+{
+  if (old_class->number_per_cidr > 0 && new_class->number_per_cidr > 0)
+  {
+    if (old_class->cidr_bitlen_ipv4 > 0 && new_class->cidr_bitlen_ipv4 > 0)
+      rebuild_cidr_list(AF_INET, old_class, new_class,
+                        &old_class->list_ipv4, &new_class->list_ipv4,
+                        old_class->cidr_bitlen_ipv4 != new_class->cidr_bitlen_ipv4);
+
+#ifdef IPV6
+    if (old_class->cidr_bitlen_ipv6 > 0 && new_class->cidr_bitlen_ipv6 > 0)
+      rebuild_cidr_list(AF_INET6, old_class, new_class,
+                        &old_class->list_ipv6, &new_class->list_ipv6,
+                        old_class->cidr_bitlen_ipv6 != new_class->cidr_bitlen_ipv6);
+#endif
+  }
+
+  destroy_cidr_class(old_class);
+}
+
+/*
+ * destroy_cidr_list
+ *
+ * inputs	- pointer to class dlink list of cidr blocks
+ * output	- none
+ * side effects	- completely destroys the class link list of cidr blocks
+ */
+static void
+destroy_cidr_list(dlink_list *list)
+{
+  dlink_node *ptr = NULL, *next_ptr = NULL;
+
+  DLINK_FOREACH_SAFE(ptr, next_ptr, list->head)
+  {
+    dlinkDelete(ptr, list);
+    MyFree(ptr->data);
+  }
+}
+
+/*
+ * destroy_cidr_class
+ *
+ * inputs	- pointer to class
+ * output	- none
+ * side effects	- completely destroys the class link list of cidr blocks
+ */
+void
+destroy_cidr_class(struct ClassItem *class)
+{
+  destroy_cidr_list(&class->list_ipv4);
+  destroy_cidr_list(&class->list_ipv6);
+}