03e6b91611ad0732fa82faf0c386c252cf0ca4c5
[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 <common/sessiond-comm/sessiond-comm.h>
36 #include <common/common.h>
37
38 #define SESSION1 "test1"
39
40 #define MAX_SESSIONS 10000
41 #define RANDOM_STRING_LEN 11
42
43 /* Number of TAP tests in this file */
44 #define NUM_TESTS 11
45
46 struct health_app *health_sessiond;
47 static struct ltt_session_list *session_list;
48 static pthread_t ht_cleanup_thread;
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 cds_list_for_each_entry_safe(iter, tmp, &session_list->head, list) {
116 session_destroy(iter);
117 }
118
119 /* Session list must be 0 */
120 assert(!session_list_count());
121 }
122
123 /*
124 * Test creation of 1 session
125 */
126 static int create_one_session(char *name)
127 {
128 int ret;
129
130 ret = session_create(name, geteuid(), getegid());
131 if (ret == LTTNG_OK) {
132 /* Validate */
133 ret = find_session_name(name);
134 if (ret < 0) {
135 /* Session not found by name */
136 printf("session not found after creation\n");
137 return -1;
138 } else {
139 /* Success */
140 return 0;
141 }
142 } else {
143 if (ret == LTTNG_ERR_EXIST_SESS) {
144 printf("(session already exists) ");
145 }
146 return -1;
147 }
148
149 return 0;
150 }
151
152 /*
153 * Test deletion of 1 session
154 */
155 static int destroy_one_session(struct ltt_session *session)
156 {
157 int ret;
158 char session_name[NAME_MAX];
159
160 strncpy(session_name, session->name, sizeof(session->name));
161 session_name[sizeof(session_name) - 1] = '\0';
162
163 ret = session_destroy(session);
164 if (ret == LTTNG_OK) {
165 ret = find_session_name(session_name);
166 if (ret < 0) {
167 /* Success, -1 means that the sesion is NOT found */
168 return 0;
169 } else {
170 /* Fail */
171 return -1;
172 }
173 }
174
175 return 0;
176 }
177
178 /*
179 * This test is supposed to fail at the second create call. If so, return 0 for
180 * test success, else -1.
181 */
182 static int two_session_same_name(void)
183 {
184 int ret;
185 struct ltt_session *sess;
186
187 ret = create_one_session(SESSION1);
188 if (ret < 0) {
189 /* Fail */
190 return -1;
191 }
192
193 sess = session_find_by_name(SESSION1);
194 if (sess) {
195 /* Success */
196 return 0;
197 }
198
199 /* Fail */
200 return -1;
201 }
202
203 void test_session_list(void)
204 {
205 session_list = session_get_list();
206 ok(session_list != NULL, "Session list: not NULL");
207 }
208
209 void test_create_one_session(void)
210 {
211 ok(create_one_session(SESSION1) == 0,
212 "Create session: %s",
213 SESSION1);
214 }
215
216 void test_validate_session(void)
217 {
218 struct ltt_session *tmp;
219
220 tmp = session_find_by_name(SESSION1);
221
222 ok(tmp != NULL,
223 "Validating session: session found");
224
225 ok(tmp->kernel_session == NULL &&
226 strlen(tmp->name),
227 "Validating session: basic sanity check");
228
229 session_lock(tmp);
230 session_unlock(tmp);
231 }
232
233 void test_destroy_session(void)
234 {
235 struct ltt_session *tmp;
236
237 tmp = session_find_by_name(SESSION1);
238
239 ok(tmp != NULL,
240 "Destroying session: session found");
241
242 ok(destroy_one_session(tmp) == 0,
243 "Destroying session: %s destroyed",
244 SESSION1);
245 }
246
247 void test_duplicate_session(void)
248 {
249 ok(two_session_same_name() == 0,
250 "Duplicate session creation");
251 }
252
253 void test_bogus_session_param(void)
254 {
255 ok(create_one_session(NULL) < 0,
256 "Create session with bogus param: NULL should fail");
257
258 ok(session_list_count() == 0,
259 "Create session with bogus param: session list empty");
260 }
261
262 void test_large_session_number(void)
263 {
264 int ret, i, failed = 0;
265 struct ltt_session *iter, *tmp;
266
267 for (i = 0; i < MAX_SESSIONS; i++) {
268 char *tmp_name = get_random_string();
269 ret = create_one_session(tmp_name);
270 if (ret < 0) {
271 diag("session %d (name: %s) creation failed", i, tmp_name);
272 ++failed;
273 }
274 }
275
276 ok(failed == 0,
277 "Large sessions number: created %u sessions",
278 MAX_SESSIONS);
279
280 failed = 0;
281
282 for (i = 0; i < MAX_SESSIONS; i++) {
283 cds_list_for_each_entry_safe(iter, tmp, &session_list->head, list) {
284 ret = destroy_one_session(iter);
285 if (ret < 0) {
286 diag("session %d destroy failed", i);
287 ++failed;
288 }
289 }
290 }
291
292 ok(failed == 0 && session_list_count() == 0,
293 "Large sessions number: destroyed %u sessions",
294 MAX_SESSIONS);
295 }
296
297 int main(int argc, char **argv)
298 {
299 plan_tests(NUM_TESTS);
300
301 health_sessiond = health_app_create(NR_HEALTH_SESSIOND_TYPES);
302 assert(!init_ht_cleanup_thread(&ht_cleanup_thread));
303
304 diag("Sessions unit tests");
305
306 rcu_register_thread();
307
308 test_session_list();
309
310 test_create_one_session();
311
312 test_validate_session();
313
314 test_destroy_session();
315
316 test_duplicate_session();
317
318 empty_session_list();
319
320 test_bogus_session_param();
321
322 test_large_session_number();
323
324 rcu_unregister_thread();
325 assert(!fini_ht_cleanup_thread(&ht_cleanup_thread));
326
327 return exit_status();
328 }
This page took 0.053673 seconds and 4 git commands to generate.