check_discovery.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615
  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 <open62541/client.h>
  5. #include <open62541/client_config_default.h>
  6. #include <open62541/server_config_default.h>
  7. #include "server/ua_server_internal.h"
  8. #include "ua_network_tcp.h"
  9. #include <fcntl.h>
  10. #include "testing_clock.h"
  11. #include "thread_wrapper.h"
  12. #ifndef WIN32
  13. #include <sys/stat.h>
  14. #endif
  15. #include <check.h>
  16. #ifndef _WIN32
  17. #include <sys/stat.h>
  18. #endif
  19. // set register timeout to 1 second so we are able to test it.
  20. #define registerTimeout 1
  21. // cleanup is only triggered every 10 seconds, thus wait a bit longer to check
  22. #define checkWait registerTimeout + 11
  23. UA_Server *server_lds;
  24. UA_ServerConfig *config_lds;
  25. UA_Boolean *running_lds;
  26. THREAD_HANDLE server_thread_lds;
  27. UA_Client *clientRegisterRepeated;
  28. THREAD_CALLBACK(serverloop_lds) {
  29. while(*running_lds)
  30. UA_Server_run_iterate(server_lds, true);
  31. return 0;
  32. }
  33. static void setup_lds(void) {
  34. // start LDS server
  35. running_lds = UA_Boolean_new();
  36. *running_lds = true;
  37. config_lds = UA_ServerConfig_new_default();
  38. config_lds->applicationDescription.applicationType = UA_APPLICATIONTYPE_DISCOVERYSERVER;
  39. UA_String_deleteMembers(&config_lds->applicationDescription.applicationUri);
  40. config_lds->applicationDescription.applicationUri =
  41. UA_STRING_ALLOC("urn:open62541.test.local_discovery_server");
  42. UA_LocalizedText_deleteMembers(&config_lds->applicationDescription.applicationName);
  43. config_lds->applicationDescription.applicationName
  44. = UA_LOCALIZEDTEXT_ALLOC("en", "LDS Server");
  45. config_lds->mdnsServerName = UA_String_fromChars("LDS_test");
  46. config_lds->serverCapabilitiesSize = 1;
  47. UA_String *caps = UA_String_new();
  48. *caps = UA_String_fromChars("LDS");
  49. config_lds->serverCapabilities = caps;
  50. config_lds->discoveryCleanupTimeout = registerTimeout;
  51. server_lds = UA_Server_new(config_lds);
  52. UA_Server_run_startup(server_lds);
  53. THREAD_CREATE(server_thread_lds, serverloop_lds);
  54. // wait until LDS started
  55. UA_fakeSleep(1000);
  56. UA_realSleep(1000);
  57. }
  58. static void teardown_lds(void) {
  59. *running_lds = false;
  60. THREAD_JOIN(server_thread_lds);
  61. UA_Server_run_shutdown(server_lds);
  62. UA_Boolean_delete(running_lds);
  63. UA_Server_delete(server_lds);
  64. UA_ServerConfig_delete(config_lds);
  65. }
  66. UA_Server *server_register;
  67. UA_ServerConfig *config_register;
  68. UA_Boolean *running_register;
  69. THREAD_HANDLE server_thread_register;
  70. UA_UInt64 periodicRegisterCallbackId;
  71. THREAD_CALLBACK(serverloop_register) {
  72. while(*running_register)
  73. UA_Server_run_iterate(server_register, true);
  74. return 0;
  75. }
  76. static void setup_register(void) {
  77. // start register server
  78. running_register = UA_Boolean_new();
  79. *running_register = true;
  80. config_register = UA_ServerConfig_new_minimal(16664, NULL);
  81. UA_String_deleteMembers(&config_register->applicationDescription.applicationUri);
  82. config_register->applicationDescription.applicationUri =
  83. UA_String_fromChars("urn:open62541.test.server_register");
  84. UA_LocalizedText_deleteMembers(&config_register->applicationDescription.applicationName);
  85. config_register->applicationDescription.applicationName =
  86. UA_LOCALIZEDTEXT_ALLOC("de", "Anmeldungsserver");
  87. config_register->mdnsServerName = UA_String_fromChars("Register_test");
  88. server_register = UA_Server_new(config_register);
  89. UA_Server_run_startup(server_register);
  90. THREAD_CREATE(server_thread_register, serverloop_register);
  91. }
  92. static void teardown_register(void) {
  93. *running_register = false;
  94. THREAD_JOIN(server_thread_register);
  95. UA_Server_run_shutdown(server_register);
  96. UA_Boolean_delete(running_register);
  97. UA_Server_delete(server_register);
  98. UA_ServerConfig_delete(config_register);
  99. }
  100. START_TEST(Server_register) {
  101. UA_Client *clientRegister = UA_Client_new();
  102. UA_ClientConfig_setDefault(UA_Client_getConfig(clientRegister));
  103. UA_StatusCode retval = UA_Client_connect_noSession(clientRegister, "opc.tcp://localhost:4840");
  104. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  105. retval = UA_Server_register_discovery(server_register, clientRegister , NULL);
  106. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  107. UA_Client_disconnect(clientRegister);
  108. UA_Client_delete(clientRegister);
  109. }
  110. END_TEST
  111. START_TEST(Server_unregister) {
  112. UA_Client *clientRegister = UA_Client_new();
  113. UA_ClientConfig_setDefault(UA_Client_getConfig(clientRegister));
  114. UA_StatusCode retval = UA_Client_connect_noSession(clientRegister, "opc.tcp://localhost:4840");
  115. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  116. retval = UA_Server_unregister_discovery(server_register, clientRegister);
  117. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  118. UA_Client_disconnect(clientRegister);
  119. UA_Client_delete(clientRegister);
  120. }
  121. END_TEST
  122. #ifndef WIN32
  123. #define SEMAPHORE_PATH "/tmp/open62541-unit-test-semaphore"
  124. #else
  125. #define SEMAPHORE_PATH ".\\open62541-unit-test-semaphore"
  126. #endif
  127. START_TEST(Server_register_semaphore) {
  128. // create the semaphore
  129. #ifndef WIN32
  130. int fd = open(SEMAPHORE_PATH, O_RDWR|O_CREAT, S_IRWXU | S_IRWXG | S_IRWXO);
  131. ck_assert_int_ne(fd, -1);
  132. close(fd);
  133. #else
  134. FILE *fp;
  135. fopen_s(&fp, SEMAPHORE_PATH, "ab+");
  136. ck_assert_ptr_ne(fp, NULL);
  137. fclose(fp);
  138. #endif
  139. UA_Client *clientRegister = UA_Client_new();
  140. UA_ClientConfig_setDefault(UA_Client_getConfig(clientRegister));
  141. UA_StatusCode retval = UA_Client_connect_noSession(clientRegister, "opc.tcp://localhost:4840");
  142. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  143. retval = UA_Server_register_discovery(server_register, clientRegister, SEMAPHORE_PATH);
  144. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  145. UA_Client_disconnect(clientRegister);
  146. UA_Client_delete(clientRegister);
  147. }
  148. END_TEST
  149. START_TEST(Server_unregister_semaphore) {
  150. // delete the semaphore, this should remove the registration automatically on next check
  151. ck_assert_int_eq(remove(SEMAPHORE_PATH), 0);
  152. }
  153. END_TEST
  154. START_TEST(Server_register_periodic) {
  155. ck_assert(clientRegisterRepeated == NULL);
  156. clientRegisterRepeated = UA_Client_new();
  157. UA_ClientConfig_setDefault(UA_Client_getConfig(clientRegisterRepeated));
  158. ck_assert(clientRegisterRepeated != NULL);
  159. // periodic register every minute, first register immediately
  160. UA_StatusCode retval = UA_Server_addPeriodicServerRegisterCallback(server_register, clientRegisterRepeated, "opc.tcp://localhost:4840",
  161. 60*1000, 100, &periodicRegisterCallbackId);
  162. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  163. }
  164. END_TEST
  165. START_TEST(Server_unregister_periodic) {
  166. // wait for first register delay
  167. UA_fakeSleep(1000);
  168. UA_realSleep(1000);
  169. UA_Server_removeRepeatedCallback(server_register, periodicRegisterCallbackId);
  170. UA_StatusCode retval = UA_Server_unregister_discovery(server_register, clientRegisterRepeated);
  171. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  172. UA_Client_disconnect(clientRegisterRepeated);
  173. UA_Client_delete(clientRegisterRepeated);
  174. clientRegisterRepeated=NULL;
  175. }
  176. END_TEST
  177. static void
  178. FindAndCheck(const UA_String expectedUris[], size_t expectedUrisSize,
  179. const UA_String expectedLocales[],
  180. const UA_String expectedNames[],
  181. const char *filterUri,
  182. const char *filterLocale) {
  183. UA_Client *client = UA_Client_new();
  184. UA_ClientConfig_setDefault(UA_Client_getConfig(client));
  185. UA_ApplicationDescription* applicationDescriptionArray = NULL;
  186. size_t applicationDescriptionArraySize = 0;
  187. size_t serverUrisSize = 0;
  188. UA_String *serverUris = NULL;
  189. if(filterUri) {
  190. serverUrisSize = 1;
  191. serverUris = UA_String_new();
  192. serverUris[0] = UA_String_fromChars(filterUri);
  193. }
  194. size_t localeIdsSize = 0;
  195. UA_String *localeIds = NULL;
  196. if(filterLocale) {
  197. localeIdsSize = 1;
  198. localeIds = UA_String_new();
  199. localeIds[0] = UA_String_fromChars(filterLocale);
  200. }
  201. UA_StatusCode retval =
  202. UA_Client_findServers(client, "opc.tcp://localhost:4840",
  203. serverUrisSize, serverUris, localeIdsSize, localeIds,
  204. &applicationDescriptionArraySize, &applicationDescriptionArray);
  205. if(filterUri)
  206. UA_Array_delete(serverUris, serverUrisSize, &UA_TYPES[UA_TYPES_STRING]);
  207. if(filterLocale)
  208. UA_Array_delete(localeIds, localeIdsSize, &UA_TYPES[UA_TYPES_STRING]);
  209. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  210. // only the discovery server is expected
  211. ck_assert_uint_eq(applicationDescriptionArraySize, expectedUrisSize);
  212. ck_assert(applicationDescriptionArray != NULL);
  213. for(size_t i = 0; i < expectedUrisSize; ++i) {
  214. ck_assert(UA_String_equal(&applicationDescriptionArray[i].applicationUri,
  215. &expectedUris[i]));
  216. if(expectedNames)
  217. ck_assert(UA_String_equal(&applicationDescriptionArray[i].applicationName.text,
  218. &expectedNames[i]));
  219. if (expectedLocales)
  220. ck_assert(UA_String_equal(&applicationDescriptionArray[i].applicationName.locale,
  221. &expectedLocales[i]));
  222. }
  223. UA_Array_delete(applicationDescriptionArray, applicationDescriptionArraySize,
  224. &UA_TYPES[UA_TYPES_APPLICATIONDESCRIPTION]);
  225. UA_Client_disconnect(client);
  226. UA_Client_delete(client);
  227. }
  228. #ifdef UA_ENABLE_DISCOVERY_MULTICAST
  229. static void
  230. FindOnNetworkAndCheck(UA_String expectedServerNames[], size_t expectedServerNamesSize,
  231. const char *filterUri, const char *filterLocale,
  232. const char** filterCapabilities, size_t filterCapabilitiesSize) {
  233. UA_Client *client = UA_Client_new();
  234. UA_ClientConfig_setDefault(UA_Client_getConfig(client));
  235. UA_ServerOnNetwork* serverOnNetwork = NULL;
  236. size_t serverOnNetworkSize = 0;
  237. size_t serverCapabilityFilterSize = 0;
  238. UA_String *serverCapabilityFilter = NULL;
  239. if(filterCapabilitiesSize) {
  240. serverCapabilityFilterSize = filterCapabilitiesSize;
  241. serverCapabilityFilter = (UA_String*)UA_malloc(sizeof(UA_String) * filterCapabilitiesSize);
  242. for(size_t i = 0; i < filterCapabilitiesSize; i++)
  243. serverCapabilityFilter[i] = UA_String_fromChars(filterCapabilities[i]);
  244. }
  245. UA_StatusCode retval =
  246. UA_Client_findServersOnNetwork(client, "opc.tcp://localhost:4840", 0, 0,
  247. serverCapabilityFilterSize, serverCapabilityFilter,
  248. &serverOnNetworkSize, &serverOnNetwork);
  249. if(serverCapabilityFilterSize)
  250. UA_Array_delete(serverCapabilityFilter, serverCapabilityFilterSize,
  251. &UA_TYPES[UA_TYPES_STRING]);
  252. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  253. // only the discovery server is expected
  254. ck_assert_uint_eq(serverOnNetworkSize , expectedServerNamesSize);
  255. if(expectedServerNamesSize > 0)
  256. ck_assert_ptr_ne(serverOnNetwork, NULL);
  257. if(serverOnNetwork != NULL) {
  258. for(size_t i = 0; i < expectedServerNamesSize; i++) {
  259. UA_Boolean expectedServerNameInServerOnNetwork = false;
  260. for(size_t j = 0; j < expectedServerNamesSize && !expectedServerNameInServerOnNetwork; j++) {
  261. expectedServerNameInServerOnNetwork = UA_String_equal(&serverOnNetwork[j].serverName,
  262. &expectedServerNames[i]);
  263. }
  264. ck_assert_msg(expectedServerNameInServerOnNetwork, "Expected %.*s in serverOnNetwork list, but not found",
  265. expectedServerNames[i].length, expectedServerNames[i].data);
  266. }
  267. }
  268. UA_Array_delete(serverOnNetwork, serverOnNetworkSize, &UA_TYPES[UA_TYPES_SERVERONNETWORK]);
  269. UA_Client_disconnect(client);
  270. UA_Client_delete(client);
  271. }
  272. static UA_StatusCode
  273. GetEndpoints(UA_Client *client, const UA_String* endpointUrl,
  274. size_t* endpointDescriptionsSize,
  275. UA_EndpointDescription** endpointDescriptions,
  276. const char* filterTransportProfileUri) {
  277. UA_GetEndpointsRequest request;
  278. UA_GetEndpointsRequest_init(&request);
  279. //request.requestHeader.authenticationToken = client->authenticationToken;
  280. request.requestHeader.timestamp = UA_DateTime_now();
  281. request.requestHeader.timeoutHint = 10000;
  282. request.endpointUrl = *endpointUrl; // assume the endpointurl outlives the service call
  283. if (filterTransportProfileUri) {
  284. request.profileUrisSize = 1;
  285. request.profileUris = (UA_String*)UA_malloc(sizeof(UA_String));
  286. request.profileUris[0] = UA_String_fromChars(filterTransportProfileUri);
  287. }
  288. UA_GetEndpointsResponse response;
  289. UA_GetEndpointsResponse_init(&response);
  290. __UA_Client_Service(client, &request, &UA_TYPES[UA_TYPES_GETENDPOINTSREQUEST],
  291. &response, &UA_TYPES[UA_TYPES_GETENDPOINTSRESPONSE]);
  292. if (filterTransportProfileUri) {
  293. UA_Array_delete(request.profileUris, request.profileUrisSize, &UA_TYPES[UA_TYPES_STRING]);
  294. }
  295. ck_assert_uint_eq(response.responseHeader.serviceResult, UA_STATUSCODE_GOOD);
  296. *endpointDescriptionsSize = response.endpointsSize;
  297. *endpointDescriptions =
  298. (UA_EndpointDescription*)UA_Array_new(response.endpointsSize,
  299. &UA_TYPES[UA_TYPES_ENDPOINTDESCRIPTION]);
  300. for(size_t i=0;i<response.endpointsSize;i++) {
  301. UA_EndpointDescription_init(&(*endpointDescriptions)[i]);
  302. UA_EndpointDescription_copy(&response.endpoints[i], &(*endpointDescriptions)[i]);
  303. }
  304. UA_GetEndpointsResponse_deleteMembers(&response);
  305. return UA_STATUSCODE_GOOD;
  306. }
  307. static void
  308. GetEndpointsAndCheck(const char* discoveryUrl, const char* filterTransportProfileUri,
  309. const UA_String expectedEndpointUrls[], size_t expectedEndpointUrlsSize) {
  310. UA_Client *client = UA_Client_new();
  311. UA_ClientConfig_setDefault(UA_Client_getConfig(client));
  312. ck_assert_uint_eq(UA_Client_connect(client, discoveryUrl), UA_STATUSCODE_GOOD);
  313. UA_EndpointDescription* endpointArray = NULL;
  314. size_t endpointArraySize = 0;
  315. UA_String discoveryUrlUA = UA_String_fromChars(discoveryUrl);
  316. UA_StatusCode retval = GetEndpoints(client, &discoveryUrlUA, &endpointArraySize,
  317. &endpointArray, filterTransportProfileUri);
  318. ck_assert_uint_eq(retval, UA_STATUSCODE_GOOD);
  319. UA_String_deleteMembers(&discoveryUrlUA);
  320. ck_assert_uint_eq(endpointArraySize , expectedEndpointUrlsSize);
  321. for(size_t j = 0; j < endpointArraySize && j < expectedEndpointUrlsSize; j++) {
  322. UA_EndpointDescription* endpoint = &endpointArray[j];
  323. ck_assert(UA_String_equal(&endpoint->endpointUrl, &expectedEndpointUrls[j]));
  324. }
  325. UA_Array_delete(endpointArray, endpointArraySize, &UA_TYPES[UA_TYPES_ENDPOINTDESCRIPTION]);
  326. UA_Client_delete(client);
  327. }
  328. // Test if discovery server lists himself as registered server if it is filtered by his uri
  329. START_TEST(Client_filter_discovery) {
  330. UA_String expectedUris[1];
  331. expectedUris[0] = UA_STRING("urn:open62541.test.local_discovery_server");
  332. FindAndCheck(expectedUris, 1, NULL, NULL, "urn:open62541.test.local_discovery_server", NULL);
  333. }
  334. END_TEST
  335. // Test if server filters locale
  336. START_TEST(Client_filter_locale) {
  337. UA_String expectedUris[2];
  338. expectedUris[0] = UA_STRING("urn:open62541.test.local_discovery_server"),
  339. expectedUris[1] = UA_STRING("urn:open62541.test.server_register");
  340. UA_String expectedNames[2];
  341. expectedNames[0]= UA_STRING("LDS Server");
  342. expectedNames[1]= UA_STRING("Anmeldungsserver");
  343. UA_String expectedLocales[2];
  344. expectedLocales[0] = UA_STRING("en");
  345. expectedLocales[1] = UA_STRING("de");
  346. // even if we request en-US, the server will return de-DE because it only has that name.
  347. FindAndCheck(expectedUris, 2, expectedLocales, expectedNames, NULL, "en");
  348. }
  349. END_TEST
  350. // Test if registered server is returned from LDS using FindServersOnNetwork
  351. START_TEST(Client_find_on_network_registered) {
  352. char urls[2][64];
  353. UA_String expectedUris[2];
  354. char hostname[256];
  355. ck_assert_uint_eq(gethostname(hostname, 255), 0);
  356. //DNS limits name to max 63 chars (+ \0)
  357. //We need this ugly casting, otherwise gcc >7.2 will complain about format-truncation, but we want it here
  358. void *hostnameVoid = (void*)hostname;
  359. snprintf(urls[0], 64, "LDS_test-%s", (char*)hostnameVoid);
  360. snprintf(urls[1], 64, "Register_test-%s", (char*)hostnameVoid);
  361. expectedUris[0] = UA_STRING(urls[0]);
  362. expectedUris[1] = UA_STRING(urls[1]);
  363. FindOnNetworkAndCheck(expectedUris, 2, NULL, NULL, NULL, 0);
  364. // filter by Capabilities
  365. const char* capsLDS[] = {"LDS"};
  366. const char* capsNA[] = {"NA"};
  367. const char* capsMultiple[] = {"LDS", "NA"};
  368. // only LDS expected
  369. FindOnNetworkAndCheck(expectedUris, 1, NULL, NULL, capsLDS, 1);
  370. // only register server expected
  371. FindOnNetworkAndCheck(&expectedUris[1], 1, NULL, NULL, capsNA, 1);
  372. // no server expected
  373. FindOnNetworkAndCheck(NULL, 0, NULL, NULL, capsMultiple, 2);
  374. }
  375. END_TEST
  376. // Test if filtering with uris works
  377. START_TEST(Client_find_filter) {
  378. UA_String expectedUris[1];
  379. expectedUris[0] = UA_STRING("urn:open62541.test.server_register");
  380. FindAndCheck(expectedUris, 1, NULL, NULL, "urn:open62541.test.server_register", NULL);
  381. }
  382. END_TEST
  383. START_TEST(Client_get_endpoints) {
  384. UA_String expectedEndpoints[1];
  385. expectedEndpoints[0] = UA_STRING("opc.tcp://localhost:4840");
  386. // general check if expected endpoints are returned
  387. GetEndpointsAndCheck("opc.tcp://localhost:4840", NULL,expectedEndpoints, 1);
  388. // check if filtering transport profile still returns the endpoint
  389. GetEndpointsAndCheck("opc.tcp://localhost:4840",
  390. "http://opcfoundation.org/UA-Profile/Transport/uatcp-uasc-uabinary",
  391. expectedEndpoints, 1);
  392. // filter transport profily by HTTPS, which should return no endpoint
  393. GetEndpointsAndCheck("opc.tcp://localhost:4840",
  394. "http://opcfoundation.org/UA-Profile/Transport/https-uabinary", NULL, 0);
  395. }
  396. END_TEST
  397. #endif
  398. // Test if discovery server lists himself as registered server, before any other registration.
  399. START_TEST(Client_find_discovery) {
  400. UA_String expectedUris[1];
  401. expectedUris[0] = UA_STRING("urn:open62541.test.local_discovery_server");
  402. FindAndCheck(expectedUris, 1, NULL, NULL, NULL, NULL);
  403. }
  404. END_TEST
  405. // Test if registered server is returned from LDS
  406. START_TEST(Client_find_registered) {
  407. UA_String expectedUris[2];
  408. expectedUris[0] = UA_STRING("urn:open62541.test.local_discovery_server");
  409. expectedUris[1] = UA_STRING("urn:open62541.test.server_register");
  410. FindAndCheck(expectedUris, 2, NULL, NULL, NULL, NULL);
  411. }
  412. END_TEST
  413. START_TEST(Util_start_lds) {
  414. setup_lds();
  415. }
  416. END_TEST
  417. START_TEST(Util_stop_lds) {
  418. teardown_lds();
  419. }
  420. END_TEST
  421. START_TEST(Util_wait_timeout) {
  422. // wait until server is removed by timeout. Additionally wait a few seconds more to be sure.
  423. UA_fakeSleep(100000 * checkWait);
  424. UA_realSleep(1000);
  425. }
  426. END_TEST
  427. #ifdef UA_ENABLE_DISCOVERY_MULTICAST
  428. START_TEST(Util_wait_mdns) {
  429. UA_fakeSleep(5000);
  430. UA_realSleep(5000);
  431. }
  432. END_TEST
  433. #endif
  434. START_TEST(Util_wait_startup) {
  435. UA_fakeSleep(1000);
  436. UA_realSleep(1000);
  437. }
  438. END_TEST
  439. START_TEST(Util_wait_retry) {
  440. // first retry is after 2 seconds, then 4, so it should be enough to wait 3 seconds
  441. UA_fakeSleep(3000);
  442. UA_realSleep(3000);
  443. }
  444. END_TEST
  445. static Suite* testSuite_Client(void) {
  446. Suite *s = suite_create("Register Server and Client");
  447. TCase *tc_register = tcase_create("RegisterServer");
  448. tcase_add_unchecked_fixture(tc_register, setup_lds, teardown_lds);
  449. tcase_add_unchecked_fixture(tc_register, setup_register, teardown_register);
  450. tcase_add_test(tc_register, Server_register);
  451. // register two times, just for fun
  452. tcase_add_test(tc_register, Server_register);
  453. tcase_add_test(tc_register, Server_unregister);
  454. tcase_add_test(tc_register, Server_register_periodic);
  455. tcase_add_test(tc_register, Server_unregister_periodic);
  456. suite_add_tcase(s,tc_register);
  457. TCase *tc_register_retry = tcase_create("RegisterServer Retry");
  458. //tcase_add_unchecked_fixture(tc_register, setup_lds, teardown_lds);
  459. tcase_add_unchecked_fixture(tc_register_retry, setup_register, teardown_register);
  460. tcase_add_test(tc_register_retry, Server_register_periodic);
  461. tcase_add_test(tc_register_retry, Util_wait_startup); // wait a bit to let first try run through
  462. // now start LDS
  463. tcase_add_test(tc_register_retry, Util_start_lds);
  464. tcase_add_test(tc_register_retry, Util_wait_retry);
  465. // check if there
  466. tcase_add_test(tc_register_retry, Client_find_registered);
  467. tcase_add_test(tc_register_retry, Server_unregister_periodic);
  468. tcase_add_test(tc_register_retry, Client_find_discovery);
  469. tcase_add_test(tc_register_retry, Util_stop_lds);
  470. suite_add_tcase(s,tc_register_retry);
  471. #ifdef UA_ENABLE_DISCOVERY_MULTICAST
  472. TCase *tc_register_find = tcase_create("RegisterServer and FindServers");
  473. tcase_add_unchecked_fixture(tc_register_find, setup_lds, teardown_lds);
  474. tcase_add_unchecked_fixture(tc_register_find, setup_register, teardown_register);
  475. tcase_add_test(tc_register_find, Client_find_discovery);
  476. tcase_add_test(tc_register_find, Server_register);
  477. tcase_add_test(tc_register_find, Client_find_registered);
  478. tcase_add_test(tc_register_find, Util_wait_mdns);
  479. tcase_add_test(tc_register_find, Client_find_on_network_registered);
  480. tcase_add_test(tc_register_find, Client_find_filter);
  481. tcase_add_test(tc_register_find, Client_get_endpoints);
  482. tcase_add_test(tc_register_find, Client_filter_locale);
  483. tcase_add_test(tc_register_find, Server_unregister);
  484. tcase_add_test(tc_register_find, Client_find_discovery);
  485. tcase_add_test(tc_register_find, Client_filter_discovery);
  486. suite_add_tcase(s,tc_register_find);
  487. #endif
  488. // register server again, then wait for timeout and auto unregister
  489. TCase *tc_register_timeout = tcase_create("RegisterServer timeout");
  490. tcase_add_unchecked_fixture(tc_register_timeout, setup_lds, teardown_lds);
  491. tcase_add_unchecked_fixture(tc_register_timeout, setup_register, teardown_register);
  492. tcase_set_timeout(tc_register_timeout, checkWait+2);
  493. tcase_add_test(tc_register_timeout, Server_register);
  494. tcase_add_test(tc_register_timeout, Client_find_registered);
  495. tcase_add_test(tc_register_timeout, Util_wait_timeout);
  496. tcase_add_test(tc_register_timeout, Client_find_discovery);
  497. #ifdef UA_ENABLE_DISCOVERY_SEMAPHORE
  498. // now check if semaphore file works
  499. tcase_add_test(tc_register_timeout, Server_register_semaphore);
  500. tcase_add_test(tc_register_timeout, Client_find_registered);
  501. tcase_add_test(tc_register_timeout, Server_unregister_semaphore);
  502. tcase_add_test(tc_register_timeout, Util_wait_timeout);
  503. tcase_add_test(tc_register_timeout, Client_find_discovery);
  504. #endif
  505. suite_add_tcase(s,tc_register_timeout);
  506. return s;
  507. }
  508. int main(void) {
  509. Suite *s = testSuite_Client();
  510. SRunner *sr = srunner_create(s);
  511. srunner_set_fork_status(sr, CK_NOFORK);
  512. srunner_run_all(sr,CK_NORMAL);
  513. int number_failed = srunner_ntests_failed(sr);
  514. srunner_free(sr);
  515. return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
  516. }