#ifdef __KERNEL__
 
+#include <asm/proc/system.h>
+
 #define smp_mb__before_clear_bit()     do { } while (0)
 #define smp_mb__after_clear_bit()      do { } while (0)
 
        *p ^= mask;
 }
 
-extern int test_and_set_bit(int nr, volatile void * addr);
-
 static inline int __test_and_set_bit(int nr, volatile void *addr)
 {
        unsigned long mask = BIT_MASK(nr);
        return (old & mask) != 0;
 }
 
-extern int test_and_clear_bit(int nr, volatile void * addr);
+static inline int test_and_set_bit(int nr, volatile void * addr)
+{
+       unsigned long flags;
+       int out;
+
+       local_irq_save(flags);
+       out = __test_and_set_bit(nr, addr);
+       local_irq_restore(flags);
+
+       return out;
+}
 
 static inline int __test_and_clear_bit(int nr, volatile void *addr)
 {
        return (old & mask) != 0;
 }
 
+static inline int test_and_clear_bit(int nr, volatile void * addr)
+{
+       unsigned long flags;
+       int out;
+
+       local_irq_save(flags);
+       out = __test_and_clear_bit(nr, addr);
+       local_irq_restore(flags);
+
+       return out;
+}
+
 extern int test_and_change_bit(int nr, volatile void * addr);
 
 static inline int __test_and_change_bit(int nr, volatile void *addr)