check_nodestore.c 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301
  1. /* This Source Code Form is subject to the terms of the Mozilla Public
  2. * License, v. 2.0. If a copy of the MPL was not distributed with this
  3. * file, You can obtain one at http://mozilla.org/MPL/2.0/.*/
  4. #include <stdio.h>
  5. #include <stdlib.h>
  6. #include <time.h>
  7. #include "ua_types.h"
  8. #include "server/ua_nodestore.h"
  9. #include "server/ua_server_internal.h"
  10. #include "ua_util.h"
  11. #include "check.h"
  12. #ifdef UA_ENABLE_MULTITHREADING
  13. #include <pthread.h>
  14. #include <urcu.h>
  15. #endif
  16. UA_NodeStore *ns;
  17. static void setup(void) {
  18. ns = UA_NodeStore_new();
  19. UA_RCU_LOCK();
  20. }
  21. static void teardown(void) {
  22. UA_NodeStore_delete(ns);
  23. UA_RCU_UNLOCK();
  24. }
  25. int zeroCnt = 0;
  26. int visitCnt = 0;
  27. static void checkZeroVisitor(const UA_Node* node) {
  28. visitCnt++;
  29. if (node == NULL) zeroCnt++;
  30. }
  31. static void printVisitor(const UA_Node* node) {
  32. printf("%d\n", node->nodeId.identifier.numeric);
  33. }
  34. static UA_Node* createNode(UA_Int16 nsid, UA_Int32 id) {
  35. UA_Node *p = (UA_Node *)UA_NodeStore_newVariableNode();
  36. p->nodeId.identifierType = UA_NODEIDTYPE_NUMERIC;
  37. p->nodeId.namespaceIndex = nsid;
  38. p->nodeId.identifier.numeric = id;
  39. p->nodeClass = UA_NODECLASS_VARIABLE;
  40. return p;
  41. }
  42. START_TEST(replaceExistingNode) {
  43. UA_Node* n1 = createNode(0,2253);
  44. UA_NodeStore_insert(ns, n1);
  45. UA_NodeId in1 = UA_NODEID_NUMERIC(0, 2253);
  46. UA_Node* n2 = UA_NodeStore_getCopy(ns, &in1);
  47. UA_StatusCode retval = UA_NodeStore_replace(ns, n2);
  48. ck_assert_int_eq(retval, UA_STATUSCODE_GOOD);
  49. }
  50. END_TEST
  51. START_TEST(replaceOldNode) {
  52. UA_Node* n1 = createNode(0,2253);
  53. UA_NodeStore_insert(ns, n1);
  54. UA_NodeId in1 = UA_NODEID_NUMERIC(0,2253);
  55. UA_Node* n2 = UA_NodeStore_getCopy(ns, &in1);
  56. UA_Node* n3 = UA_NodeStore_getCopy(ns, &in1);
  57. /* shall succeed */
  58. UA_StatusCode retval = UA_NodeStore_replace(ns, n2);
  59. ck_assert_int_eq(retval, UA_STATUSCODE_GOOD);
  60. /* shall fail */
  61. retval = UA_NodeStore_replace(ns, n3);
  62. ck_assert_int_ne(retval, UA_STATUSCODE_GOOD);
  63. }
  64. END_TEST
  65. START_TEST(findNodeInUA_NodeStoreWithSingleEntry) {
  66. UA_Node* n1 = createNode(0,2253);
  67. UA_NodeStore_insert(ns, n1);
  68. UA_NodeId in1 = UA_NODEID_NUMERIC(0,2253);
  69. const UA_Node* nr = UA_NodeStore_get(ns, &in1);
  70. ck_assert_int_eq((uintptr_t)n1, (uintptr_t)nr);
  71. }
  72. END_TEST
  73. START_TEST(failToFindNodeInOtherUA_NodeStore) {
  74. UA_Node* n1 = createNode(0,2255);
  75. UA_NodeStore_insert(ns, n1);
  76. UA_NodeId in1 = UA_NODEID_NUMERIC(1, 2255);
  77. const UA_Node* nr = UA_NodeStore_get(ns, &in1);
  78. ck_assert_int_eq((uintptr_t)nr, 0);
  79. }
  80. END_TEST
  81. START_TEST(findNodeInUA_NodeStoreWithSeveralEntries) {
  82. UA_Node* n1 = createNode(0,2253);
  83. UA_NodeStore_insert(ns, n1);
  84. UA_Node* n2 = createNode(0,2255);
  85. UA_NodeStore_insert(ns, n2);
  86. UA_Node* n3 = createNode(0,2257);
  87. UA_NodeStore_insert(ns, n3);
  88. UA_Node* n4 = createNode(0,2200);
  89. UA_NodeStore_insert(ns, n4);
  90. UA_Node* n5 = createNode(0,1);
  91. UA_NodeStore_insert(ns, n5);
  92. UA_Node* n6 = createNode(0,12);
  93. UA_NodeStore_insert(ns, n6);
  94. UA_NodeId in3 = UA_NODEID_NUMERIC(0, 2257);
  95. const UA_Node* nr = UA_NodeStore_get(ns, &in3);
  96. ck_assert_int_eq((uintptr_t)nr, (uintptr_t)n3);
  97. }
  98. END_TEST
  99. START_TEST(iterateOverUA_NodeStoreShallNotVisitEmptyNodes) {
  100. UA_Node* n1 = createNode(0,2253);
  101. UA_NodeStore_insert(ns, n1);
  102. UA_Node* n2 = createNode(0,2255);
  103. UA_NodeStore_insert(ns, n2);
  104. UA_Node* n3 = createNode(0,2257);
  105. UA_NodeStore_insert(ns, n3);
  106. UA_Node* n4 = createNode(0,2200);
  107. UA_NodeStore_insert(ns, n4);
  108. UA_Node* n5 = createNode(0,1);
  109. UA_NodeStore_insert(ns, n5);
  110. UA_Node* n6 = createNode(0,12);
  111. UA_NodeStore_insert(ns, n6);
  112. zeroCnt = 0;
  113. visitCnt = 0;
  114. UA_NodeStore_iterate(ns,checkZeroVisitor);
  115. ck_assert_int_eq(zeroCnt, 0);
  116. ck_assert_int_eq(visitCnt, 6);
  117. }
  118. END_TEST
  119. START_TEST(findNodeInExpandedNamespace) {
  120. for(UA_UInt32 i = 0; i < 200; i++) {
  121. UA_Node* n = createNode(0,i);
  122. UA_NodeStore_insert(ns, n);
  123. }
  124. // when
  125. UA_Node *n2 = createNode(0,25);
  126. const UA_Node* nr = UA_NodeStore_get(ns,&n2->nodeId);
  127. ck_assert_int_eq(nr->nodeId.identifier.numeric,n2->nodeId.identifier.numeric);
  128. UA_NodeStore_deleteNode(n2);
  129. }
  130. END_TEST
  131. START_TEST(iterateOverExpandedNamespaceShallNotVisitEmptyNodes) {
  132. for(UA_UInt32 i = 0; i < 200; i++) {
  133. UA_Node* n = createNode(0,i);
  134. UA_NodeStore_insert(ns, n);
  135. }
  136. // when
  137. zeroCnt = 0;
  138. visitCnt = 0;
  139. UA_NodeStore_iterate(ns,checkZeroVisitor);
  140. // then
  141. ck_assert_int_eq(zeroCnt, 0);
  142. ck_assert_int_eq(visitCnt, 200);
  143. }
  144. END_TEST
  145. START_TEST(failToFindNonExistantNodeInUA_NodeStoreWithSeveralEntries) {
  146. UA_Node* n1 = createNode(0,2253);
  147. UA_NodeStore_insert(ns, n1);
  148. UA_Node* n2 = createNode(0,2255);
  149. UA_NodeStore_insert(ns, n2);
  150. UA_Node* n3 = createNode(0,2257);
  151. UA_NodeStore_insert(ns, n3);
  152. UA_Node* n4 = createNode(0,2200);
  153. UA_NodeStore_insert(ns, n4);
  154. UA_Node* n5 = createNode(0,1);
  155. UA_NodeStore_insert(ns, n5);
  156. UA_NodeId id = UA_NODEID_NUMERIC(0, 12);
  157. const UA_Node* nr = UA_NodeStore_get(ns, &id);
  158. ck_assert_int_eq((uintptr_t)nr, 0);
  159. }
  160. END_TEST
  161. /************************************/
  162. /* Performance Profiling Test Cases */
  163. /************************************/
  164. #ifdef UA_ENABLE_MULTITHREADING
  165. struct UA_NodeStoreProfileTest {
  166. UA_Int32 min_val;
  167. UA_Int32 max_val;
  168. UA_Int32 rounds;
  169. };
  170. static void *profileGetThread(void *arg) {
  171. rcu_register_thread();
  172. UA_RCU_LOCK();
  173. struct UA_NodeStoreProfileTest *test = (struct UA_NodeStoreProfileTest*) arg;
  174. UA_NodeId id;
  175. UA_NodeId_init(&id);
  176. UA_Int32 max_val = test->max_val;
  177. for(UA_Int32 x = 0; x<test->rounds; x++) {
  178. for(UA_Int32 i=test->min_val; i<max_val; i++) {
  179. id.identifier.numeric = i;
  180. UA_NodeStore_get(ns,&id);
  181. }
  182. }
  183. UA_RCU_UNLOCK();
  184. rcu_unregister_thread();
  185. return NULL;
  186. }
  187. #endif
  188. START_TEST(profileGetDelete) {
  189. #define N 1000000
  190. for(UA_UInt32 i = 0; i < N; i++) {
  191. UA_Node *n = createNode(0,i);
  192. UA_NodeStore_insert(ns, n);
  193. }
  194. clock_t begin, end;
  195. begin = clock();
  196. #ifdef UA_ENABLE_MULTITHREADING
  197. #define THREADS 4
  198. pthread_t t[THREADS];
  199. struct UA_NodeStoreProfileTest p[THREADS];
  200. for (int i = 0; i < THREADS; i++) {
  201. p[i] = (struct UA_NodeStoreProfileTest){i*(N/THREADS), (i+1)*(N/THREADS), 50};
  202. pthread_create(&t[i], NULL, profileGetThread, &p[i]);
  203. }
  204. for (int i = 0; i < THREADS; i++)
  205. pthread_join(t[i], NULL);
  206. end = clock();
  207. printf("Time for %d create/get/delete on %d threads in a namespace: %fs.\n", N, THREADS, (double)(end - begin) / CLOCKS_PER_SEC);
  208. #else
  209. UA_NodeId id;
  210. UA_NodeId_init(&id);
  211. for(UA_Int32 x = 0; x<50; x++) {
  212. for(int i=0; i<N; i++) {
  213. id.identifier.numeric = i;
  214. UA_NodeStore_get(ns,&id);
  215. }
  216. }
  217. end = clock();
  218. printf("Time for single-threaded %d create/get/delete in a namespace: %fs.\n", N, (double)(end - begin) / CLOCKS_PER_SEC);
  219. #endif
  220. }
  221. END_TEST
  222. static Suite * namespace_suite (void) {
  223. Suite *s = suite_create ("UA_NodeStore");
  224. TCase* tc_find = tcase_create ("Find");
  225. tcase_add_checked_fixture(tc_find, setup, teardown);
  226. tcase_add_test (tc_find, findNodeInUA_NodeStoreWithSingleEntry);
  227. tcase_add_test (tc_find, findNodeInUA_NodeStoreWithSeveralEntries);
  228. tcase_add_test (tc_find, findNodeInExpandedNamespace);
  229. tcase_add_test (tc_find, failToFindNonExistantNodeInUA_NodeStoreWithSeveralEntries);
  230. tcase_add_test (tc_find, failToFindNodeInOtherUA_NodeStore);
  231. suite_add_tcase (s, tc_find);
  232. TCase *tc_replace = tcase_create("Replace");
  233. tcase_add_checked_fixture(tc_replace, setup, teardown);
  234. tcase_add_test (tc_replace, replaceExistingNode);
  235. tcase_add_test (tc_replace, replaceOldNode);
  236. suite_add_tcase (s, tc_replace);
  237. TCase* tc_iterate = tcase_create ("Iterate");
  238. tcase_add_checked_fixture(tc_iterate, setup, teardown);
  239. tcase_add_test (tc_iterate, iterateOverUA_NodeStoreShallNotVisitEmptyNodes);
  240. tcase_add_test (tc_iterate, iterateOverExpandedNamespaceShallNotVisitEmptyNodes);
  241. suite_add_tcase (s, tc_iterate);
  242. /* TCase* tc_profile = tcase_create ("Profile"); */
  243. /* tcase_add_checked_fixture(tc_profile, setup, teardown); */
  244. /* tcase_add_test (tc_profile, profileGetDelete); */
  245. /* suite_add_tcase (s, tc_profile); */
  246. return s;
  247. }
  248. int main (void) {
  249. #ifdef UA_ENABLE_MULTITHREADING
  250. rcu_init();
  251. rcu_register_thread();
  252. #endif
  253. int number_failed = 0;
  254. Suite *s = namespace_suite();
  255. SRunner *sr = srunner_create(s);
  256. srunner_set_fork_status(sr,CK_NOFORK);
  257. srunner_run_all(sr, CK_NORMAL);
  258. number_failed += srunner_ntests_failed (sr);
  259. srunner_free(sr);
  260. #ifdef UA_ENABLE_MULTITHREADING
  261. rcu_barrier();
  262. rcu_unregister_thread();
  263. #endif
  264. return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
  265. }