f2343c9916bd5636a455807616b549cfc1e56811
[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
28 #include <tap/tap.h>
29
30 #include <bin/lttng-sessiond/session.h>
31 #include <bin/lttng-sessiond/ust-app.h>
32 #include <bin/lttng-sessiond/ht-cleanup.h>
33 #include <bin/lttng-sessiond/health-sessiond.h>
34 #include <common/sessiond-comm/sessiond-comm.h>
35 #include <common/common.h>
36
37 #define SESSION1 "test1"
38
39 #define MAX_SESSIONS 10000
40 #define RANDOM_STRING_LEN 11
41
42 /* Number of TAP tests in this file */
43 #define NUM_TESTS 11
44
45 struct health_app *health_sessiond;
46 static struct ltt_session_list *session_list;
47 static pthread_t ht_cleanup_thread;
48
49 /* For error.h */
50 int lttng_opt_quiet = 1;
51 int lttng_opt_verbose = 0;
52 int lttng_opt_mi;
53
54 int ust_consumerd32_fd;
55 int ust_consumerd64_fd;
56
57 static const char alphanum[] =
58 "0123456789"
59 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
60 "abcdefghijklmnopqrstuvwxyz";
61 static char random_string[RANDOM_STRING_LEN];
62
63 /*
64 * Return random string of 10 characters.
65 * Not thread-safe.
66 */
67 static char *get_random_string(void)
68 {
69 int i;
70
71 for (i = 0; i < RANDOM_STRING_LEN - 1; i++) {
72 random_string[i] = alphanum[rand() % (sizeof(alphanum) - 1)];
73 }
74
75 random_string[RANDOM_STRING_LEN - 1] = '\0';
76
77 return random_string;
78 }
79
80 /*
81 * Return 0 if session name is found, else -1
82 */
83 static int find_session_name(char *name)
84 {
85 struct ltt_session *iter;
86
87 cds_list_for_each_entry(iter, &session_list->head, list) {
88 if (strcmp(iter->name, name) == 0) {
89 return 0;
90 }
91 }
92
93 return -1;
94 }
95
96 static int session_list_count(void)
97 {
98 int count = 0;
99 struct ltt_session *iter;
100
101 cds_list_for_each_entry(iter, &session_list->head, list) {
102 count++;
103 }
104 return count;
105 }
106
107 /*
108 * Empty session list manually.
109 */
110 static void empty_session_list(void)
111 {
112 struct ltt_session *iter, *tmp;
113
114 cds_list_for_each_entry_safe(iter, tmp, &session_list->head, list) {
115 session_destroy(iter);
116 }
117
118 /* Session list must be 0 */
119 assert(!session_list_count());
120 }
121
122 /*
123 * Test creation of 1 session
124 */
125 static int create_one_session(char *name)
126 {
127 int ret;
128
129 ret = session_create(name, geteuid(), getegid());
130 if (ret == LTTNG_OK) {
131 /* Validate */
132 ret = find_session_name(name);
133 if (ret < 0) {
134 /* Session not found by name */
135 printf("session not found after creation\n");
136 return -1;
137 } else {
138 /* Success */
139 return 0;
140 }
141 } else {
142 if (ret == LTTNG_ERR_EXIST_SESS) {
143 printf("(session already exists) ");
144 }
145 return -1;
146 }
147
148 return 0;
149 }
150
151 /*
152 * Test deletion of 1 session
153 */
154 static int destroy_one_session(struct ltt_session *session)
155 {
156 int ret;
157 char session_name[NAME_MAX];
158
159 strncpy(session_name, session->name, sizeof(session->name));
160 session_name[sizeof(session_name) - 1] = '\0';
161
162 ret = session_destroy(session);
163 if (ret == LTTNG_OK) {
164 ret = find_session_name(session_name);
165 if (ret < 0) {
166 /* Success, -1 means that the sesion is NOT found */
167 return 0;
168 } else {
169 /* Fail */
170 return -1;
171 }
172 }
173
174 return 0;
175 }
176
177 /*
178 * This test is supposed to fail at the second create call. If so, return 0 for
179 * test success, else -1.
180 */
181 static int two_session_same_name(void)
182 {
183 int ret;
184 struct ltt_session *sess;
185
186 ret = create_one_session(SESSION1);
187 if (ret < 0) {
188 /* Fail */
189 return -1;
190 }
191
192 sess = session_find_by_name(SESSION1);
193 if (sess) {
194 /* Success */
195 return 0;
196 }
197
198 /* Fail */
199 return -1;
200 }
201
202 void test_session_list(void)
203 {
204 session_list = session_get_list();
205 ok(session_list != NULL, "Session list: not NULL");
206 }
207
208 void test_create_one_session(void)
209 {
210 ok(create_one_session(SESSION1) == 0,
211 "Create session: %s",
212 SESSION1);
213 }
214
215 void test_validate_session(void)
216 {
217 struct ltt_session *tmp;
218
219 tmp = session_find_by_name(SESSION1);
220
221 ok(tmp != NULL,
222 "Validating session: session found");
223
224 ok(tmp->kernel_session == NULL &&
225 strlen(tmp->name),
226 "Validating session: basic sanity check");
227
228 session_lock(tmp);
229 session_unlock(tmp);
230 }
231
232 void test_destroy_session(void)
233 {
234 struct ltt_session *tmp;
235
236 tmp = session_find_by_name(SESSION1);
237
238 ok(tmp != NULL,
239 "Destroying session: session found");
240
241 ok(destroy_one_session(tmp) == 0,
242 "Destroying session: %s destroyed",
243 SESSION1);
244 }
245
246 void test_duplicate_session(void)
247 {
248 ok(two_session_same_name() == 0,
249 "Duplicate session creation");
250 }
251
252 void test_bogus_session_param(void)
253 {
254 ok(create_one_session(NULL) < 0,
255 "Create session with bogus param: NULL should fail");
256
257 ok(session_list_count() == 0,
258 "Create session with bogus param: session list empty");
259 }
260
261 void test_large_session_number(void)
262 {
263 int ret, i, failed = 0;
264 struct ltt_session *iter, *tmp;
265
266 for (i = 0; i < MAX_SESSIONS; i++) {
267 char *tmp_name = get_random_string();
268 ret = create_one_session(tmp_name);
269 if (ret < 0) {
270 diag("session %d (name: %s) creation failed", i, tmp_name);
271 ++failed;
272 }
273 }
274
275 ok(failed == 0,
276 "Large sessions number: created %u sessions",
277 MAX_SESSIONS);
278
279 failed = 0;
280
281 for (i = 0; i < MAX_SESSIONS; i++) {
282 cds_list_for_each_entry_safe(iter, tmp, &session_list->head, list) {
283 ret = destroy_one_session(iter);
284 if (ret < 0) {
285 diag("session %d destroy failed", i);
286 ++failed;
287 }
288 }
289 }
290
291 ok(failed == 0 && session_list_count() == 0,
292 "Large sessions number: destroyed %u sessions",
293 MAX_SESSIONS);
294 }
295
296 int main(int argc, char **argv)
297 {
298 plan_tests(NUM_TESTS);
299
300 health_sessiond = health_app_create(NR_HEALTH_SESSIOND_TYPES);
301 assert(!init_ht_cleanup_thread(&ht_cleanup_thread));
302
303 diag("Sessions unit tests");
304
305 test_session_list();
306
307 test_create_one_session();
308
309 test_validate_session();
310
311 test_destroy_session();
312
313 test_duplicate_session();
314
315 empty_session_list();
316
317 test_bogus_session_param();
318
319 test_large_session_number();
320
321 assert(!fini_ht_cleanup_thread(&ht_cleanup_thread));
322
323 return exit_status();
324 }
This page took 0.053472 seconds and 4 git commands to generate.