mirror of
git://git.code.sf.net/p/xtables-addons/xtables-addons
synced 2025-09-07 21:25:12 +02:00
geoip: use struct list_head instead of self-cooked list
This commit is contained in:
@@ -10,6 +10,7 @@
|
|||||||
*/
|
*/
|
||||||
#include <linux/ip.h>
|
#include <linux/ip.h>
|
||||||
#include <linux/kernel.h>
|
#include <linux/kernel.h>
|
||||||
|
#include <linux/list.h>
|
||||||
#include <linux/module.h>
|
#include <linux/module.h>
|
||||||
#include <linux/netdevice.h>
|
#include <linux/netdevice.h>
|
||||||
#include <linux/skbuff.h>
|
#include <linux/skbuff.h>
|
||||||
@@ -28,15 +29,14 @@ MODULE_DESCRIPTION("xtables module for geoip match");
|
|||||||
MODULE_ALIAS("ipt_geoip");
|
MODULE_ALIAS("ipt_geoip");
|
||||||
|
|
||||||
struct geoip_country_kernel {
|
struct geoip_country_kernel {
|
||||||
|
struct list_head list;
|
||||||
struct geoip_subnet *subnets;
|
struct geoip_subnet *subnets;
|
||||||
u_int32_t count;
|
u_int32_t count;
|
||||||
u_int32_t ref;
|
u_int32_t ref;
|
||||||
u_int16_t cc;
|
u_int16_t cc;
|
||||||
struct geoip_country_kernel *next;
|
|
||||||
struct geoip_country_kernel *prev;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
struct geoip_country_kernel *head = NULL;
|
static LIST_HEAD(geoip_head);
|
||||||
static spinlock_t geoip_lock = SPIN_LOCK_UNLOCKED;
|
static spinlock_t geoip_lock = SPIN_LOCK_UNLOCKED;
|
||||||
|
|
||||||
static struct geoip_country_kernel *
|
static struct geoip_country_kernel *
|
||||||
@@ -67,11 +67,8 @@ geoip_add_node(const struct geoip_country_user __user *umem_ptr)
|
|||||||
|
|
||||||
p->subnets = s;
|
p->subnets = s;
|
||||||
p->ref = 1;
|
p->ref = 1;
|
||||||
p->next = head;
|
INIT_LIST_HEAD(&p->list);
|
||||||
p->prev = NULL;
|
list_add_tail(&p->list, &geoip_head);
|
||||||
if (p->next)
|
|
||||||
p->next->prev = p;
|
|
||||||
head = p;
|
|
||||||
|
|
||||||
spin_unlock_bh(&geoip_lock);
|
spin_unlock_bh(&geoip_lock);
|
||||||
return p;
|
return p;
|
||||||
@@ -90,17 +87,7 @@ static void geoip_try_remove_node(struct geoip_country_kernel *p)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (p->next) { /* Am I following a node ? */
|
list_del(&p->list);
|
||||||
p->next->prev = p->prev;
|
|
||||||
if (p->prev) p->prev->next = p->next; /* Is there a node behind me ? */
|
|
||||||
else head = p->next; /* No? Then I was the head */
|
|
||||||
}
|
|
||||||
|
|
||||||
else
|
|
||||||
if (p->prev) /* Is there a node behind me ? */
|
|
||||||
p->prev->next = NULL;
|
|
||||||
else
|
|
||||||
head = NULL; /* No, we're alone */
|
|
||||||
|
|
||||||
/* So now am unlinked or the only one alive, right ?
|
/* So now am unlinked or the only one alive, right ?
|
||||||
* What are you waiting ? Free up some memory!
|
* What are you waiting ? Free up some memory!
|
||||||
@@ -113,17 +100,16 @@ static void geoip_try_remove_node(struct geoip_country_kernel *p)
|
|||||||
|
|
||||||
static struct geoip_country_kernel *find_node(u_int16_t cc)
|
static struct geoip_country_kernel *find_node(u_int16_t cc)
|
||||||
{
|
{
|
||||||
struct geoip_country_kernel *p = head;
|
struct geoip_country_kernel *p;
|
||||||
spin_lock_bh(&geoip_lock);
|
spin_lock_bh(&geoip_lock);
|
||||||
|
|
||||||
while (p) {
|
list_for_each_entry(p, &geoip_head, list)
|
||||||
if (p->cc == cc) {
|
if (p->cc == cc) {
|
||||||
atomic_inc((atomic_t *)&p->ref);
|
atomic_inc((atomic_t *)&p->ref);
|
||||||
spin_unlock_bh(&geoip_lock);
|
spin_unlock_bh(&geoip_lock);
|
||||||
return p;
|
return p;
|
||||||
}
|
}
|
||||||
p = p->next;
|
|
||||||
}
|
|
||||||
spin_unlock_bh(&geoip_lock);
|
spin_unlock_bh(&geoip_lock);
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user