From: Anoop Saldanha <poonaatsoc@gmail.com>
Date: Fri, 22 Jun 2012 16:12:57 +0000 (+0530)
Subject: add unittest for atomic operation with void *
X-Git-Tag: suricata-1.3rc1~30
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=e5edcfaca8105e71f647daf9cb952e63831a02c3;p=thirdparty%2Fsuricata.git

add unittest for atomic operation with void *
---

diff --git a/src/Makefile.am b/src/Makefile.am
index 5a4d434c41..3508af10cc 100644
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -3,7 +3,7 @@ noinst_HEADERS = action-globals.h \
     debug.h \
 	flow-private.h queue.h source-nfq-prototypes.h \
 	suricata-common.h threadvars.h util-binsearch.h \
-    util-atomic.h util-validate.h
+        util-validate.h
 bin_PROGRAMS = suricata
 
 AM_CPPFLAGS = -DCONFIG_DIR=\"$(e_sysconfdir)\"
@@ -180,7 +180,7 @@ decode-events.c decode-events.h \
 detect-replace.c detect-replace.h \
 util-magic.c util-magic.h \
 util-misc.c util-misc.h \
-util-atomic.h \
+util-atomic.c util-atomic.h \
 util-print.c util-print.h \
 util-fmemopen.c util-fmemopen.h \
 util-cpu.c util-cpu.h \
diff --git a/src/detect-engine.c b/src/detect-engine.c
index 669294ea61..06ae579323 100644
--- a/src/detect-engine.c
+++ b/src/detect-engine.c
@@ -212,8 +212,8 @@ static void *DetectEngineLiveRuleSwap(void *arg)
             }
 
 
-            SC_ATOMIC_CAS(&slots->slot_data, SC_ATOMIC_GET(slots->slot_data),
-                          det_ctx);
+            BUG_ON(SC_ATOMIC_CAS(&slots->slot_data, SC_ATOMIC_GET(slots->slot_data),
+                                 det_ctx) == 0);;
             SCLogDebug("swapping new det_ctx with older one");
 
             slots = slots->slot_next;
diff --git a/src/suricata.c b/src/suricata.c
index f29161e36a..5a9ed6654a 100644
--- a/src/suricata.c
+++ b/src/suricata.c
@@ -1570,6 +1570,7 @@ int main(int argc, char **argv)
         DetectAddressTests();
         DetectProtoTests();
         DetectPortTests();
+        SCAtomicRegisterTests();
         if (list_unittests) {
             UtListTests(regex_arg);
         }
diff --git a/src/util-atomic.c b/src/util-atomic.c
new file mode 100644
index 0000000000..de85c3162a
--- /dev/null
+++ b/src/util-atomic.c
@@ -0,0 +1,67 @@
+/* Copyright (C) 2007-2012 Open Information Security Foundation
+ *
+ * You can copy, redistribute or modify this Program under the terms of
+ * the GNU General Public License version 2 as published by the Free
+ * Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * version 2 along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+ * 02110-1301, USA.
+ */
+
+/**
+ * \file
+ *
+ * \author Anoop Saldanha <anoopsaldanha@gmail.com>
+ */
+
+#include "suricata-common.h"
+#include "suricata.h"
+#include "util-atomic.h"
+#include "util-unittest.h"
+
+static int SCAtomicTest01(void)
+{
+    int result = 0;
+    int a = 10;
+    int b = 20;
+    int *temp_int = NULL;
+
+    SC_ATOMIC_DECL_AND_INIT(void *, temp);
+
+    temp_int = SC_ATOMIC_GET(temp);
+    if (temp_int != NULL)
+        goto end;
+
+    SC_ATOMIC_CAS(&temp, SC_ATOMIC_GET(temp), &a);
+    temp_int = SC_ATOMIC_GET(temp);
+    if (temp_int == NULL)
+        goto end;
+    if (*temp_int != a)
+        goto end;
+
+    SC_ATOMIC_CAS(&temp, SC_ATOMIC_GET(temp), &b);
+    temp_int = SC_ATOMIC_GET(temp);
+    if (temp_int == NULL)
+        goto end;
+    if (*temp_int != b)
+        goto end;
+
+    result = 1;
+
+ end:
+    return result;
+}
+
+void SCAtomicRegisterTests(void)
+{
+    UtRegisterTest("SCAtomicTest01", SCAtomicTest01, 1);
+
+    return;
+}
diff --git a/src/util-atomic.h b/src/util-atomic.h
index 4f2868e26a..650bf7deda 100644
--- a/src/util-atomic.h
+++ b/src/util-atomic.h
@@ -466,5 +466,8 @@
         })
 
 #endif /* !no atomic operations */
+
+void SCAtomicRegisterTests(void);
+
 #endif /* __UTIL_ATOMIC_H__ */