actually writing some unit tests for the new 64 bit capable AtomicCounter found a...

diff --git a/pdns/misc.hh b/pdns/misc.hh
index a239f57d6be20110f464c173a5ba190fef2c665b..574ce40ab7a1a1163abdd293adbeba09e555e91b 100644 (file)
--- a/pdns/misc.hh
+++ b/pdns/misc.hh
@@ -369,7 +369,7 @@ public:
       return atomic_exchange_and_add( &value_, +1 );
     }
 
-    native_t operator+=(int val)
+    native_t operator+=(native_t val)
     {
       return atomic_exchange_and_add( &value_, val );
     }
@@ -394,13 +394,13 @@ private:
     
     // the below is necessary because __sync_fetch_and_add is not universally available on i386.. I 3> RHEL5. 
 #if defined( __GNUC__ ) && ( defined( __i386__ ) || defined( __x86_64__ ) )
-    static int atomic_exchange_and_add( native_t * pw, int dv )
+    static native_t atomic_exchange_and_add( native_t * pw, native_t dv )
     {
         // int r = *pw;
         // *pw += dv;
         // return r;
 
-        int r;
+        native_t r;
 
         __asm__ __volatile__
         (
@@ -414,7 +414,7 @@ private:
         return r;
     }
     #else 
-    static int atomic_exchange_and_add( native_t * pw, int dv )
+    static native_t atomic_exchange_and_add( native_t * pw, native_t dv )
     {
       return __sync_fetch_and_add(pw, dv);
     }

diff --git a/pdns/test-statbag_cc.cc b/pdns/test-statbag_cc.cc
index fd4c35ca0c3bda44d79537f1d6456659a019b259..906df9ee306bb8bb693af94c373fee0eaede3042 100644 (file)
--- a/pdns/test-statbag_cc.cc
+++ b/pdns/test-statbag_cc.cc
@@ -19,6 +19,16 @@ static void *threadMangler(void* a)
   return 0;
 }
 
+static void *threadMangler2(void* a)
+{
+  StatBag* S = (StatBag*)a;
+  for(unsigned int n=0; n < 10000000; ++n)
+    S->inc("c");
+  return 0;
+}
+
+
+
 BOOST_AUTO_TEST_SUITE(misc_hh)
 
 BOOST_AUTO_TEST_CASE(test_StatBagBasic) {
@@ -50,6 +60,40 @@ BOOST_AUTO_TEST_CASE(test_StatBagBasic) {
     pthread_join(tid[i], &res);
 
   BOOST_CHECK_EQUAL(s.read("c"), 40000000U);
+ 
+  s.set("c", 0);
+
+  for(int i=0; i < 4; ++i) 
+    pthread_create(&tid[i], 0, threadMangler2, (void*)&s);
+
+  for(int i=0; i < 4 ; ++i)
+    pthread_join(tid[i], &res);
+
+  BOOST_CHECK_EQUAL(s.read("c"), 40000000U);
+
+
+  s.set("c", 1ULL<<31);
+  BOOST_CHECK_EQUAL(s.read("c"), (1ULL<<31) );
+  s.inc("c");
+  BOOST_CHECK_EQUAL(s.read("c"), (1ULL<<31) +1 );
+  
+#if UINTPTR_MAX > 0xffffffffULL
+    s.set("c", 1ULL<<33);
+    BOOST_CHECK_EQUAL(s.read("c"), (1ULL<<33) );
+    s.inc("c");
+    BOOST_CHECK_EQUAL(s.read("c"), (1ULL<<33) +1 );
+
+    s.set("c", ~0ULL);
+    BOOST_CHECK_EQUAL(s.read("c"), 0xffffffffffffffffULL );
+    s.inc("c");
+    BOOST_CHECK_EQUAL(s.read("c"), 0 );
+#else
+    s.set("c", ~0UL);
+    BOOST_CHECK_EQUAL(s.read("c"), 0xffffffffUL );
+    s.inc("c");
+    BOOST_CHECK_EQUAL(s.read("c"), 0 );
+#endif
+
 }