include cleanup: Update gfp.h and slab.h includes to prepare for breaking implicit...

[mv-sheeva.git] / net / sched / sch_dsmark.c

diff --git a/net/sched/sch_dsmark.c b/net/sched/sch_dsmark.c
index 3f9427a4b757ec8cce0090ecac4d364cb1f36a53..63d41f86679c0653951090c81f189665c5deaa00 100644 (file)
--- a/net/sched/sch_dsmark.c
+++ b/net/sched/sch_dsmark.c
@@ -5,6 +5,7 @@
 
 #include <linux/module.h>
 #include <linux/init.h>
+#include <linux/slab.h>
 #include <linux/types.h>
 #include <linux/string.h>
 #include <linux/errno.h>
@@ -68,7 +69,8 @@ static int dsmark_graft(struct Qdisc *sch, unsigned long arg,
        }
 
        sch_tree_lock(sch);
-       *old = xchg(&p->q, new);
+       *old = p->q;
+       p->q = new;
        qdisc_tree_decrease_qlen(*old, (*old)->q.qlen);
        qdisc_reset(*old);
        sch_tree_unlock(sch);