-/*
- * SPDX-License-Identifier: LGPL-2.1-only
- *
- * Copyright (C) 2009 Pierre-Marc Fournier
- * Copyright (C) 2011 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
- */
-
-#include <fcntl.h>
-#include <signal.h>
-#include <stdarg.h>
-#include <stdint.h>
-#include <stdio.h>
-#include <string.h>
-#include <sys/mman.h>
-#include <sys/stat.h>
-#include <sys/types.h>
-#include <unistd.h>
-/*
- * Work-around inet.h missing struct mmsghdr forward declaration, with
- * triggers a warning when system files warnings are enabled.
- */
-struct mmsghdr;
-#include <arpa/inet.h>
-#include <stddef.h>
-#include <stdlib.h>
-#include <stdbool.h>
-
-#define LTTNG_UST_TRACEPOINT_DEFINE
-#include "ust_tests_hello.h"
-
-#include <lttng/ust-events.h>
-#include <lttng/ust-ringbuffer-context.h>
-/* Internal header. */
-#include <common/ust-context-provider.h>
-
-static __thread unsigned int test_count;
-
-static
-void test_inc_count(void)
-{
- test_count++;
-}
-
-static
-size_t test_get_size(void *priv __attribute__((unused)),
- struct lttng_ust_probe_ctx *probe_ctx __attribute__((unused)),
- size_t offset)
-{
- int sel = test_count % _NR_LTTNG_UST_DYNAMIC_TYPES;
- size_t size = 0;
-
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(char));
- size += sizeof(char); /* tag */
- switch (sel) {
- case LTTNG_UST_DYNAMIC_TYPE_NONE:
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S8:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(int8_t));
- size += sizeof(int8_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S16:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(int16_t));
- size += sizeof(int16_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S32:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(int32_t));
- size += sizeof(int32_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S64:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(int64_t));
- size += sizeof(int64_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U8:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(uint8_t));
- size += sizeof(uint8_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U16:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(uint16_t));
- size += sizeof(uint16_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U32:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(uint32_t));
- size += sizeof(uint32_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U64:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(uint64_t));
- size += sizeof(uint64_t); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_FLOAT:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(float));
- size += sizeof(float); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_DOUBLE:
- size += lttng_ust_ring_buffer_align(offset, lttng_ust_rb_alignof(double));
- size += sizeof(double); /* variant */
- break;
- case LTTNG_UST_DYNAMIC_TYPE_STRING:
- size += strlen("teststr") + 1;
- break;
- default:
- abort();
- }
-
- return size;
-}
-
-static
-void test_record(void *priv __attribute__((unused)),
- struct lttng_ust_probe_ctx *probe_ctx __attribute__((unused)),
- struct lttng_ust_ring_buffer_ctx *ctx,
- struct lttng_ust_channel_buffer *lttng_chan_buf)
-{
- int sel = test_count % _NR_LTTNG_UST_DYNAMIC_TYPES;
- char sel_char = (char) sel;
-
- lttng_chan_buf->ops->event_write(ctx, &sel_char, sizeof(sel_char), lttng_ust_rb_alignof(char));
- switch (sel) {
- case LTTNG_UST_DYNAMIC_TYPE_NONE:
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S8:
- {
- int8_t v = -8;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_S16:
- {
- int16_t v = -16;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_S32:
- {
- int32_t v = -32;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_S64:
- {
- int64_t v = -64;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_U8:
- {
- uint8_t v = 8;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_U16:
- {
- uint16_t v = 16;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_U32:
- {
- uint32_t v = 32;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_U64:
- {
- uint64_t v = 64;
-
- lttng_chan_buf->ops->event_write(ctx, &v, sizeof(v), lttng_ust_rb_alignof(v));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_FLOAT:
- {
- float f = 22322.0;
-
- lttng_chan_buf->ops->event_write(ctx, &f, sizeof(f), lttng_ust_rb_alignof(f));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_DOUBLE:
- {
- double d = 2.0;
-
- lttng_chan_buf->ops->event_write(ctx, &d, sizeof(d), lttng_ust_rb_alignof(d));
- break;
- }
- case LTTNG_UST_DYNAMIC_TYPE_STRING:
- {
- const char *str = "teststr";
- lttng_chan_buf->ops->event_write(ctx, str, strlen(str) + 1, 1);
- break;
- }
- default:
- abort();
- }
-}
-
-static
-void test_get_value(void *priv __attribute__((unused)),
- struct lttng_ust_probe_ctx *probe_ctx __attribute__((unused)),
- struct lttng_ust_ctx_value *value)
-{
- int sel = test_count % _NR_LTTNG_UST_DYNAMIC_TYPES;
-
- value->sel = sel;
- switch (sel) {
- case LTTNG_UST_DYNAMIC_TYPE_NONE:
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S8:
- value->u.s64 = -8;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S16:
- value->u.s64 = -16;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S32:
- value->u.s64 = -32;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_S64:
- value->u.s64 = -64;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U8:
- value->u.u64 = 8;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U16:
- value->u.u64 = 16;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U32:
- value->u.u64 = 32;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_U64:
- value->u.u64 = 64;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_FLOAT:
- value->u.d = 22322.0;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_DOUBLE:
- value->u.d = 2.0;
- break;
- case LTTNG_UST_DYNAMIC_TYPE_STRING:
- value->u.str = "teststr";
- break;
- default:
- abort();
- }
-}
-
-static char myprovider_name[] = "$app.myprovider";
-struct lttng_ust_context_provider myprovider = {
- .struct_size = sizeof(struct lttng_ust_context_provider),
- .name = myprovider_name,
- .get_size = test_get_size,
- .record = test_record,
- .get_value = test_get_value,
-};
-
-static
-void inthandler(int sig __attribute__((unused)))
-{
- printf("in SIGUSR1 handler\n");
- lttng_ust_tracepoint(ust_tests_hello, tptest_sighandler);
-}
-
-static
-int init_int_handler(void)
-{
- int result;
- struct sigaction act;
-
- memset(&act, 0, sizeof(act));
- result = sigemptyset(&act.sa_mask);
- if (result == -1) {
- perror("sigemptyset");
- return -1;
- }
-
- act.sa_handler = inthandler;
- act.sa_flags = SA_RESTART;
-
- /* Only defer ourselves. Also, try to restart interrupted
- * syscalls to disturb the traced program as little as possible.
- */
- result = sigaction(SIGUSR1, &act, NULL);
- if (result == -1) {
- perror("sigaction");
- return -1;
- }
-
- return 0;
-}
-
-int main(int argc, char **argv)
-{
- struct lttng_ust_registered_context_provider *reg_provider;
- int i, netint;
- long values[] = { 1, 2, 3 };
- char text[10] = "test";
- double dbl = 2.0;
- float flt = 2222.0;
- int delay = 0;
- bool mybool = 123; /* should print "1" */
-
- init_int_handler();
-
- if (argc == 2)
- delay = atoi(argv[1]);
-
- reg_provider = lttng_ust_context_provider_register(&myprovider);
- if (!reg_provider)
- abort();
-
- fprintf(stderr, "Hello, World!\n");
-
- sleep(delay);
-
- fprintf(stderr, "Tracing... ");
- for (i = 0; i < 1000000; i++) {
- netint = htonl(i);
- lttng_ust_tracepoint(ust_tests_hello, tptest, i, netint, values,
- text, strlen(text), dbl, flt, mybool);
- test_inc_count();
- //usleep(100000);
- }
- lttng_ust_context_provider_unregister(reg_provider);
- fprintf(stderr, " done.\n");
- return 0;
-}