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 }