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