Fix tests: NULL pointer dereference in ltt_session unit tests
[lttng-tools.git] / tests / unit / test_session.c
1 /*
2 * Copyright (c) 2011 David Goulet <david.goulet@polymtl.ca>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * as published by the Free Software Foundation; only version 2
7 * of the License.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
17 */
18
19 #include <assert.h>
20 #include <errno.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <unistd.h>
25 #include <time.h>
26 #include <sys/types.h>
27 #include <urcu.h>
28
29 #include <tap/tap.h>
30
31 #include <bin/lttng-sessiond/session.h>
32 #include <bin/lttng-sessiond/ust-app.h>
33 #include <bin/lttng-sessiond/ht-cleanup.h>
34 #include <bin/lttng-sessiond/health-sessiond.h>
35 #include <bin/lttng-sessiond/thread.h>
36 #include <common/sessiond-comm/sessiond-comm.h>
37 #include <common/common.h>
38
39 #define SESSION1 "test1"
40
41 #define MAX_SESSIONS 10000
42 #define RANDOM_STRING_LEN 11
43
44 /* Number of TAP tests in this file */
45 #define NUM_TESTS 11
46
47 struct health_app *health_sessiond;
48 static struct ltt_session_list *session_list;
49
50 /* For error.h */
51 int lttng_opt_quiet = 1;
52 int lttng_opt_verbose = 0;
53 int lttng_opt_mi;
54
55 int ust_consumerd32_fd;
56 int ust_consumerd64_fd;
57
58 static const char alphanum[] =
59 "0123456789"
60 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
61 "abcdefghijklmnopqrstuvwxyz";
62 static char random_string[RANDOM_STRING_LEN];
63
64 /*
65 * Return random string of 10 characters.
66 * Not thread-safe.
67 */
68 static char *get_random_string(void)
69 {
70 int i;
71
72 for (i = 0; i < RANDOM_STRING_LEN - 1; i++) {
73 random_string[i] = alphanum[rand() % (sizeof(alphanum) - 1)];
74 }
75
76 random_string[RANDOM_STRING_LEN - 1] = '\0';
77
78 return random_string;
79 }
80
81 /*
82 * Return 0 if session name is found, else -1
83 */
84 static int find_session_name(char *name)
85 {
86 struct ltt_session *iter;
87
88 cds_list_for_each_entry(iter, &session_list->head, list) {
89 if (strcmp(iter->name, name) == 0) {
90 return 0;
91 }
92 }
93
94 return -1;
95 }
96
97 static int session_list_count(void)
98 {
99 int count = 0;
100 struct ltt_session *iter;
101
102 cds_list_for_each_entry(iter, &session_list->head, list) {
103 count++;
104 }
105 return count;
106 }
107
108 /*
109 * Empty session list manually.
110 */
111 static void empty_session_list(void)
112 {
113 struct ltt_session *iter, *tmp;
114
115 session_lock_list();
116 cds_list_for_each_entry_safe(iter, tmp, &session_list->head, list) {
117 session_destroy(iter);
118 }
119 session_unlock_list();
120
121 /* Session list must be 0 */
122 assert(!session_list_count());
123 }
124
125 /*
126 * Test creation of 1 session
127 */
128 static int create_one_session(char *name)
129 {
130 int ret;
131 enum lttng_error_code ret_code;
132 struct ltt_session *session = NULL;
133
134 session_lock_list();
135 ret_code = session_create(name, geteuid(), getegid(), &session);
136 session_put(session);
137 if (ret_code == LTTNG_OK) {
138 /* Validate */
139 ret = find_session_name(name);
140 if (ret < 0) {
141 /* Session not found by name */
142 printf("session not found after creation\n");
143 ret = -1;
144 goto end;
145 } else {
146 /* Success */
147 ret = 0;
148 goto end;
149 }
150 } else {
151 if (ret_code == LTTNG_ERR_EXIST_SESS) {
152 printf("(session already exists) ");
153 }
154 ret = -1;
155 goto end;
156 }
157 ret = 0;
158 end:
159 session_unlock_list();
160 return ret;
161 }
162
163 /*
164 * Test deletion of 1 session
165 */
166 static int destroy_one_session(struct ltt_session *session)
167 {
168 int ret;
169 char session_name[NAME_MAX];
170
171 strncpy(session_name, session->name, sizeof(session_name));
172 session_name[sizeof(session_name) - 1] = '\0';
173
174 session_destroy(session);
175 session_put(session);
176
177 ret = find_session_name(session_name);
178 if (ret < 0) {
179 /* Success, -1 means that the sesion is NOT found */
180 ret = 0;
181 } else {
182 /* Fail */
183 ret = -1;
184 }
185 return ret;
186 }
187
188 /*
189 * This test is supposed to fail at the second create call. If so, return 0 for
190 * test success, else -1.
191 */
192 static int two_session_same_name(void)
193 {
194 int ret;
195 struct ltt_session *sess;
196
197 ret = create_one_session(SESSION1);
198 if (ret < 0) {
199 /* Fail */
200 ret = -1;
201 goto end;
202 }
203
204 session_lock_list();
205 sess = session_find_by_name(SESSION1);
206 if (sess) {
207 /* Success */
208 session_put(sess);
209 session_unlock_list();
210 ret = 0;
211 goto end_unlock;
212 } else {
213 /* Fail */
214 ret = -1;
215 goto end_unlock;
216 }
217 end_unlock:
218 session_unlock_list();
219 end:
220 return ret;
221 }
222
223 void test_session_list(void)
224 {
225 session_list = session_get_list();
226 ok(session_list != NULL, "Session list: not NULL");
227 }
228
229 void test_create_one_session(void)
230 {
231 ok(create_one_session(SESSION1) == 0,
232 "Create session: %s",
233 SESSION1);
234 }
235
236 void test_validate_session(void)
237 {
238 struct ltt_session *tmp;
239
240 session_lock_list();
241 tmp = session_find_by_name(SESSION1);
242
243 ok(tmp != NULL,
244 "Validating session: session found");
245
246 if (tmp) {
247 ok(tmp->kernel_session == NULL &&
248 strlen(tmp->name),
249 "Validating session: basic sanity check");
250 } else {
251 skip(1, "Skipping session validation check as session was not found");
252 goto end;
253 }
254
255 session_lock(tmp);
256 session_unlock(tmp);
257 session_put(tmp);
258 end:
259 session_unlock_list();
260 }
261
262 void test_destroy_session(void)
263 {
264 struct ltt_session *tmp;
265
266 session_lock_list();
267 tmp = session_find_by_name(SESSION1);
268
269 ok(tmp != NULL,
270 "Destroying session: session found");
271
272 if (tmp) {
273 ok(destroy_one_session(tmp) == 0,
274 "Destroying session: %s destroyed",
275 SESSION1);
276 } else {
277 skip(1, "Skipping session destruction as it was not found");
278 }
279 session_unlock_list();
280 }
281
282 void test_duplicate_session(void)
283 {
284 ok(two_session_same_name() == 0,
285 "Duplicate session creation");
286 }
287
288 void test_session_name_generation(void)
289 {
290 struct ltt_session *session = NULL;
291 enum lttng_error_code ret_code;
292 const char *expected_session_name_prefix = DEFAULT_SESSION_NAME;
293
294 session_lock_list();
295 ret_code = session_create(NULL, geteuid(), getegid(), &session);
296 ok(ret_code == LTTNG_OK,
297 "Create session with a NULL name (auto-generate a name)");
298 if (!session) {
299 skip(1, "Skipping session name generation tests as session_create() failed.");
300 goto end;
301 }
302 diag("Automatically-generated session name: %s", *session->name ?
303 session->name : "ERROR");
304 ok(*session->name && !strncmp(expected_session_name_prefix, session->name,
305 sizeof(DEFAULT_SESSION_NAME) - 1),
306 "Auto-generated session name starts with %s",
307 DEFAULT_SESSION_NAME);
308 end:
309 session_put(session);
310 session_unlock_list();
311 }
312
313 void test_large_session_number(void)
314 {
315 int ret, i, failed = 0;
316 struct ltt_session *iter, *tmp;
317
318 for (i = 0; i < MAX_SESSIONS; i++) {
319 char *tmp_name = get_random_string();
320 ret = create_one_session(tmp_name);
321 if (ret < 0) {
322 diag("session %d (name: %s) creation failed", i, tmp_name);
323 ++failed;
324 }
325 }
326
327 ok(failed == 0,
328 "Large sessions number: created %u sessions",
329 MAX_SESSIONS);
330
331 failed = 0;
332
333 session_lock_list();
334 for (i = 0; i < MAX_SESSIONS; i++) {
335 cds_list_for_each_entry_safe(iter, tmp, &session_list->head, list) {
336 assert(session_get(iter));
337 ret = destroy_one_session(iter);
338 if (ret < 0) {
339 diag("session %d destroy failed", i);
340 ++failed;
341 }
342 }
343 }
344 session_unlock_list();
345
346 ok(failed == 0 && session_list_count() == 0,
347 "Large sessions number: destroyed %u sessions",
348 MAX_SESSIONS);
349 }
350
351 int main(int argc, char **argv)
352 {
353 struct lttng_thread *ht_cleanup_thread;
354
355 plan_tests(NUM_TESTS);
356
357 health_sessiond = health_app_create(NR_HEALTH_SESSIOND_TYPES);
358 ht_cleanup_thread = launch_ht_cleanup_thread();
359 assert(ht_cleanup_thread);
360 lttng_thread_put(ht_cleanup_thread);
361
362 diag("Sessions unit tests");
363
364 rcu_register_thread();
365
366 test_session_list();
367
368 test_create_one_session();
369
370 test_validate_session();
371
372 test_destroy_session();
373
374 test_duplicate_session();
375
376 empty_session_list();
377
378 test_session_name_generation();
379
380 test_large_session_number();
381
382 rcu_unregister_thread();
383 lttng_thread_list_shutdown_orphans();
384
385 return exit_status();
386 }
This page took 0.056406 seconds and 4 git commands to generate.