Commit | Line | Data |
---|---|---|
897b8e23 DG |
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 | ||
897b8e23 DG |
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 | ||
10a8a223 | 27 | #include <bin/lttng-sessiond/trace-kernel.h> |
990570ed | 28 | #include <common/defaults.h> |
10a8a223 | 29 | |
23aaa19e | 30 | #include <tap/tap.h> |
897b8e23 | 31 | |
98612240 MD |
32 | #define RANDOM_STRING_LEN 11 |
33 | ||
23aaa19e | 34 | /* Number of TAP tests in this file */ |
1c0733db | 35 | #define NUM_TESTS 11 |
23aaa19e | 36 | |
ad7c9c18 | 37 | /* For error.h */ |
97e19046 DG |
38 | int lttng_opt_quiet = 1; |
39 | int lttng_opt_verbose; | |
c7e35b03 | 40 | int lttng_opt_mi; |
897b8e23 | 41 | |
7972aab2 DG |
42 | int ust_consumerd32_fd; |
43 | int ust_consumerd64_fd; | |
44 | ||
897b8e23 DG |
45 | static const char alphanum[] = |
46 | "0123456789" | |
47 | "ABCDEFGHIJKLMNOPQRSTUVWXYZ" | |
48 | "abcdefghijklmnopqrstuvwxyz"; | |
49 | ||
50 | static struct ltt_kernel_session *kern; | |
98612240 | 51 | static char random_string[RANDOM_STRING_LEN]; |
897b8e23 DG |
52 | |
53 | /* | |
54 | * Return random string of 10 characters. | |
98612240 | 55 | * Not thread-safe. |
897b8e23 DG |
56 | */ |
57 | static char *get_random_string(void) | |
58 | { | |
59 | int i; | |
897b8e23 | 60 | |
98612240 MD |
61 | for (i = 0; i < RANDOM_STRING_LEN - 1; i++) { |
62 | random_string[i] = alphanum[rand() % (sizeof(alphanum) - 1)]; | |
897b8e23 DG |
63 | } |
64 | ||
98612240 | 65 | random_string[RANDOM_STRING_LEN - 1] = '\0'; |
897b8e23 | 66 | |
98612240 | 67 | return random_string; |
897b8e23 DG |
68 | } |
69 | ||
23aaa19e | 70 | static void test_create_one_kernel_session(void) |
897b8e23 | 71 | { |
dec56f6c | 72 | kern = trace_kernel_create_session(); |
23aaa19e | 73 | ok(kern != NULL, "Create kernel session"); |
897b8e23 | 74 | |
23aaa19e CB |
75 | ok(kern->fd == -1 && |
76 | kern->metadata_stream_fd == -1 && | |
77 | kern->consumer_fds_sent == 0 && | |
78 | kern->channel_count == 0 && | |
79 | kern->stream_count_global == 0 && | |
80 | kern->metadata == NULL, | |
81 | "Validate kernel session"); | |
897b8e23 DG |
82 | } |
83 | ||
23aaa19e | 84 | static void test_create_kernel_metadata(void) |
897b8e23 DG |
85 | { |
86 | assert(kern != NULL); | |
87 | ||
a4b92340 | 88 | kern->metadata = trace_kernel_create_metadata(); |
23aaa19e CB |
89 | ok(kern->metadata != NULL, "Create kernel metadata"); |
90 | ||
91 | ok(kern->metadata->fd == -1 && | |
92 | kern->metadata->conf != NULL && | |
93 | kern->metadata->conf->attr.overwrite | |
94 | == DEFAULT_CHANNEL_OVERWRITE && | |
95 | kern->metadata->conf->attr.subbuf_size | |
96 | == default_get_metadata_subbuf_size() && | |
97 | kern->metadata->conf->attr.num_subbuf | |
98 | == DEFAULT_METADATA_SUBBUF_NUM && | |
99 | kern->metadata->conf->attr.switch_timer_interval | |
5d2e1e66 | 100 | == DEFAULT_KERNEL_CHANNEL_SWITCH_TIMER && |
23aaa19e | 101 | kern->metadata->conf->attr.read_timer_interval |
5d2e1e66 | 102 | == DEFAULT_KERNEL_CHANNEL_READ_TIMER && |
23aaa19e CB |
103 | kern->metadata->conf->attr.output |
104 | == DEFAULT_KERNEL_CHANNEL_OUTPUT, | |
105 | "Validate kernel session metadata"); | |
897b8e23 | 106 | |
3f43a221 | 107 | trace_kernel_destroy_metadata(kern->metadata); |
897b8e23 DG |
108 | } |
109 | ||
23aaa19e | 110 | static void test_create_kernel_channel(void) |
897b8e23 DG |
111 | { |
112 | struct ltt_kernel_channel *chan; | |
113 | struct lttng_channel attr; | |
114 | ||
441c16a7 MD |
115 | memset(&attr, 0, sizeof(attr)); |
116 | ||
fdd9eb17 | 117 | chan = trace_kernel_create_channel(&attr); |
23aaa19e | 118 | ok(chan != NULL, "Create kernel channel"); |
897b8e23 | 119 | |
23aaa19e CB |
120 | ok(chan->fd == -1 && |
121 | chan->enabled == 1 && | |
122 | chan->stream_count == 0 && | |
23aaa19e CB |
123 | chan->channel->attr.overwrite == attr.attr.overwrite, |
124 | "Validate kernel channel"); | |
897b8e23 DG |
125 | |
126 | /* Init list in order to avoid sefaults from cds_list_del */ | |
127 | CDS_INIT_LIST_HEAD(&chan->list); | |
3f43a221 | 128 | trace_kernel_destroy_channel(chan); |
897b8e23 DG |
129 | } |
130 | ||
23aaa19e | 131 | static void test_create_kernel_event(void) |
897b8e23 DG |
132 | { |
133 | struct ltt_kernel_event *event; | |
134 | struct lttng_event ev; | |
135 | ||
441c16a7 | 136 | memset(&ev, 0, sizeof(ev)); |
72265d8a | 137 | ok(!lttng_strncpy(ev.name, get_random_string(), |
1c0733db MD |
138 | LTTNG_KERNEL_SYM_NAME_LEN), |
139 | "Validate string length"); | |
897b8e23 | 140 | ev.type = LTTNG_EVENT_TRACEPOINT; |
441c16a7 | 141 | ev.loglevel_type = LTTNG_EVENT_LOGLEVEL_ALL; |
897b8e23 | 142 | |
00a62084 | 143 | event = trace_kernel_create_event(&ev, NULL, NULL); |
23aaa19e | 144 | ok(event != NULL, "Create kernel event"); |
897b8e23 | 145 | |
23aaa19e CB |
146 | ok(event->fd == -1 && |
147 | event->enabled == 1 && | |
148 | event->event->instrumentation == LTTNG_KERNEL_TRACEPOINT && | |
149 | strlen(event->event->name), | |
150 | "Validate kernel event"); | |
897b8e23 DG |
151 | |
152 | /* Init list in order to avoid sefaults from cds_list_del */ | |
153 | CDS_INIT_LIST_HEAD(&event->list); | |
3f43a221 | 154 | trace_kernel_destroy_event(event); |
897b8e23 DG |
155 | } |
156 | ||
23aaa19e | 157 | static void test_create_kernel_stream(void) |
897b8e23 DG |
158 | { |
159 | struct ltt_kernel_stream *stream; | |
160 | ||
00e2e675 | 161 | stream = trace_kernel_create_stream("stream1", 0); |
23aaa19e | 162 | ok(stream != NULL, "Create kernel stream"); |
897b8e23 | 163 | |
23aaa19e CB |
164 | ok(stream->fd == -1 && |
165 | stream->state == 0, | |
166 | "Validate kernel stream"); | |
897b8e23 DG |
167 | |
168 | /* Init list in order to avoid sefaults from cds_list_del */ | |
169 | CDS_INIT_LIST_HEAD(&stream->list); | |
3f43a221 | 170 | trace_kernel_destroy_stream(stream); |
897b8e23 DG |
171 | } |
172 | ||
173 | int main(int argc, char **argv) | |
174 | { | |
23aaa19e | 175 | plan_tests(NUM_TESTS); |
897b8e23 | 176 | |
e3bef725 CB |
177 | diag("Kernel data structure unit test"); |
178 | ||
23aaa19e CB |
179 | test_create_one_kernel_session(); |
180 | test_create_kernel_metadata(); | |
181 | test_create_kernel_channel(); | |
182 | test_create_kernel_event(); | |
183 | test_create_kernel_stream(); | |
897b8e23 DG |
184 | |
185 | /* Success */ | |
186 | return 0; | |
187 | } |