[PKTGEN]: Fix module load/unload races.

Adrian Bunk:
Backported to 2.6.16.

Signed-off-by: David S. Miller <davem@davemloft.net>
Signed-off-by: Adrian Bunk <bunk@stusta.de>
This commit is contained in:
Robert Olsson
2007-01-04 00:57:17 +01:00
committed by Adrian Bunk
parent 51b73a0309
commit e79366b556
+18 -2
View File
@@ -139,6 +139,7 @@
#include <linux/proc_fs.h>
#include <linux/seq_file.h>
#include <linux/wait.h>
#include <linux/completion.h>
#include <linux/etherdevice.h>
#include <net/checksum.h>
#include <net/ipv6.h>
@@ -194,6 +195,11 @@ static struct proc_dir_entry *pg_proc_dir = NULL;
#define MAX_CFLOWS 65536
struct pktgen_thread_info {
struct pktgen_thread *t;
struct completion *c;
};
struct flow_state
{
__u32 cur_daddr;
@@ -2656,10 +2662,11 @@ retry_now:
* Main loop of the thread goes here
*/
static void pktgen_thread_worker(struct pktgen_thread *t)
static void pktgen_thread_worker(struct pktgen_thread_info *info)
{
DEFINE_WAIT(wait);
struct pktgen_dev *pkt_dev = NULL;
struct pktgen_thread *t = info->t;
int cpu = t->cpu;
sigset_t tmpsig;
u32 max_before_softirq;
@@ -2700,6 +2707,8 @@ static void pktgen_thread_worker(struct pktgen_thread *t)
__set_current_state(TASK_INTERRUPTIBLE);
mb();
complete(info->c);
while (1) {
__set_current_state(TASK_RUNNING);
@@ -2894,6 +2903,8 @@ static struct pktgen_thread * __init pktgen_find_thread(const char* name)
static int __init pktgen_create_thread(const char* name, int cpu)
{
struct pktgen_thread_info info;
struct completion started;
struct pktgen_thread *t = NULL;
struct proc_dir_entry *pe;
@@ -2931,10 +2942,15 @@ static int __init pktgen_create_thread(const char* name, int cpu)
t->next = pktgen_threads;
pktgen_threads = t;
if (kernel_thread((void *) pktgen_thread_worker, (void *) t,
init_completion(&started);
info.t = t;
info.c = &started;
if (kernel_thread((void *) pktgen_thread_worker, (void *)&info,
CLONE_FS | CLONE_FILES | CLONE_SIGHAND) < 0)
printk("pktgen: kernel_thread() failed for cpu %d\n", t->cpu);
wait_for_completion(&started);
return 0;
}