sys/sys/sdt.h

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause
   3  *
   4  * Copyright 2006-2008 John Birrell <jb@FreeBSD.org>
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18  * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
  19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25  * SUCH DAMAGE.
  26  *
  27  * Statically Defined Tracing (SDT) definitions.
  28  *
  29  */
  30
  31 #ifndef _SYS_SDT_H
  32 #define _SYS_SDT_H
  33
  34 #ifndef _KERNEL
  35
  36 #define _DTRACE_VERSION 1
  37
  38 #define DTRACE_PROBE(prov, name) {                              \
  39         extern void __dtrace_##prov##___##name(void);           \
  40         __dtrace_##prov##___##name();                           \
  41 }
  42
  43 #define DTRACE_PROBE1(prov, name, arg1) {                       \
  44         extern void __dtrace_##prov##___##name(unsigned long);  \
  45         __dtrace_##prov##___##name((unsigned long)arg1);        \
  46 }
  47
  48 #define DTRACE_PROBE2(prov, name, arg1, arg2) {                 \
  49         extern void __dtrace_##prov##___##name(unsigned long,   \
  50             unsigned long);                                     \
  51         __dtrace_##prov##___##name((unsigned long)arg1,         \
  52             (unsigned long)arg2);                               \
  53 }
  54
  55 #define DTRACE_PROBE3(prov, name, arg1, arg2, arg3) {           \
  56         extern void __dtrace_##prov##___##name(unsigned long,   \
  57             unsigned long, unsigned long);                      \
  58         __dtrace_##prov##___##name((unsigned long)arg1,         \
  59             (unsigned long)arg2, (unsigned long)arg3);          \
  60 }
  61
  62 #define DTRACE_PROBE4(prov, name, arg1, arg2, arg3, arg4) {     \
  63         extern void __dtrace_##prov##___##name(unsigned long,   \
  64             unsigned long, unsigned long, unsigned long);       \
  65         __dtrace_##prov##___##name((unsigned long)arg1,         \
  66             (unsigned long)arg2, (unsigned long)arg3,           \
  67             (unsigned long)arg4);                               \
  68 }
  69
  70 #define DTRACE_PROBE5(prov, name, arg1, arg2, arg3, arg4, arg5) {       \
  71         extern void __dtrace_##prov##___##name(unsigned long,           \
  72             unsigned long, unsigned long, unsigned long, unsigned long);\
  73         __dtrace_##prov##___##name((unsigned long)arg1,                 \
  74             (unsigned long)arg2, (unsigned long)arg3,                   \
  75             (unsigned long)arg4, (unsigned long)arg5);                  \
  76 }
  77
  78 #else /* _KERNEL */
  79
  80 #include <sys/cdefs.h>
  81 #include <sys/linker_set.h>
  82
  83 extern volatile bool sdt_probes_enabled;
  84
  85 #ifndef KDTRACE_HOOKS
  86
  87 #define __sdt_used      __unused
  88
  89 #define SDT_PROVIDER_DEFINE(prov)
  90 #define SDT_PROVIDER_DECLARE(prov)
  91 #define SDT_PROBE_DEFINE(prov, mod, func, name)
  92 #define SDT_PROBE_DECLARE(prov, mod, func, name)
  93 #define SDT_PROBES_ENABLED()    0
  94 #define SDT_PROBE(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4)
  95 #define SDT_PROBE_ARGTYPE(prov, mod, func, name, num, type, xtype)
  96
  97 #define SDT_PROBE_DEFINE0(prov, mod, func, name)
  98 #define SDT_PROBE_DEFINE1(prov, mod, func, name, arg0)
  99 #define SDT_PROBE_DEFINE2(prov, mod, func, name, arg0, arg1)
 100 #define SDT_PROBE_DEFINE3(prov, mod, func, name, arg0, arg1, arg2)
 101 #define SDT_PROBE_DEFINE4(prov, mod, func, name, arg0, arg1, arg2, arg3)
 102 #define SDT_PROBE_DEFINE5(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4)
 103 #define SDT_PROBE_DEFINE6(prov, mod, func, name, arg0, arg1, arg2,      \
 104     arg3, arg4, arg5)
 105 #define SDT_PROBE_DEFINE7(prov, mod, func, name, arg0, arg1, arg2,      \
 106     arg3, arg4, arg5, arg6)
 107
 108 #define SDT_PROBE0(prov, mod, func, name)
 109 #define SDT_PROBE1(prov, mod, func, name, arg0)
 110 #define SDT_PROBE2(prov, mod, func, name, arg0, arg1)
 111 #define SDT_PROBE3(prov, mod, func, name, arg0, arg1, arg2)
 112 #define SDT_PROBE4(prov, mod, func, name, arg0, arg1, arg2, arg3)
 113 #define SDT_PROBE5(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4)
 114 #define SDT_PROBE6(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4, arg5)
 115 #define SDT_PROBE7(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4, arg5,  \
 116     arg6)
 117
 118 #define SDT_PROBE_DEFINE0_XLATE(prov, mod, func, name)
 119 #define SDT_PROBE_DEFINE1_XLATE(prov, mod, func, name, arg0, xarg0)
 120 #define SDT_PROBE_DEFINE2_XLATE(prov, mod, func, name, arg0, xarg0,     \
 121     arg1, xarg1)
 122 #define SDT_PROBE_DEFINE3_XLATE(prov, mod, func, name, arg0, xarg0,     \
 123     arg1, xarg1, arg2, xarg2)
 124 #define SDT_PROBE_DEFINE4_XLATE(prov, mod, func, name, arg0, xarg0,     \
 125     arg1, xarg1, arg2, xarg2, arg3, xarg3)
 126 #define SDT_PROBE_DEFINE5_XLATE(prov, mod, func, name, arg0, xarg0,     \
 127     arg1, xarg1, arg2, xarg2, arg3, xarg3, arg4, xarg4)
 128 #define SDT_PROBE_DEFINE6_XLATE(prov, mod, func, name, arg0, xarg0,     \
 129     arg1,  xarg1, arg2, xarg2, arg3, xarg3, arg4, xarg4, arg5, xarg5)
 130 #define SDT_PROBE_DEFINE7_XLATE(prov, mod, func, name, arg0, xarg0,     \
 131     arg1, xarg1, arg2, xarg2, arg3, xarg3, arg4, xarg4, arg5, xarg5, arg6,     \
 132     xarg6)
 133
 134 #define DTRACE_PROBE(name)
 135 #define DTRACE_PROBE1(name, type0, arg0)
 136 #define DTRACE_PROBE2(name, type0, arg0, type1, arg1)
 137 #define DTRACE_PROBE3(name, type0, arg0, type1, arg1, type2, arg2)
 138 #define DTRACE_PROBE4(name, type0, arg0, type1, arg1, type2, arg2, type3, arg3)
 139 #define DTRACE_PROBE5(name, type0, arg0, type1, arg1, type2, arg2, type3, arg3,\
 140     type4, arg4)
 141
 142 #else
 143
 144 #define __sdt_used
 145
 146 SET_DECLARE(sdt_providers_set, struct sdt_provider);
 147 SET_DECLARE(sdt_probes_set, struct sdt_probe);
 148 SET_DECLARE(sdt_argtypes_set, struct sdt_argtype);
 149
 150 #define SDT_PROVIDER_DEFINE(_prov)                                      \
 151         struct sdt_provider sdt_provider_##_prov[1] = {                 \
 152                 [0] = { .name = #_prov },                               \
 153         };                                                              \
 154         DATA_SET(sdt_providers_set, sdt_provider_##_prov);
 155
 156 #define SDT_PROVIDER_DECLARE(prov)                                              \
 157         extern struct sdt_provider sdt_provider_##prov[1]
 158
 159 #define SDT_PROBE_DEFINE(_prov, _mod, _func, _name)                     \
 160         struct sdt_probe sdt_##_prov##_##_mod##_##_func##_##_name[1] = {\
 161                 [0] = {                                                 \
 162                     .version = sizeof(struct sdt_probe),                \
 163                     .prov = sdt_provider_##_prov,                       \
 164                     .mod = #_mod,                                       \
 165                     .func = #_func,                                     \
 166                     .name = #_name,                                     \
 167                 },                                                      \
 168         };                                                              \
 169         DATA_SET(sdt_probes_set, sdt_##_prov##_##_mod##_##_func##_##_name)
 170
 171 #define SDT_PROBE_DECLARE(prov, mod, func, name)                                \
 172         extern struct sdt_probe sdt_##prov##_##mod##_##func##_##name[1]
 173
 174 #define SDT_PROBES_ENABLED()    __predict_false(sdt_probes_enabled)
 175
 176 #define SDT_PROBE(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4)  do {    \
 177         if (SDT_PROBES_ENABLED()) {                                             \
 178                 if (__predict_false(sdt_##prov##_##mod##_##func##_##name->id))  \
 179                 (*sdt_probe_func)(sdt_##prov##_##mod##_##func##_##name->id,     \
 180                     (uintptr_t) arg0, (uintptr_t) arg1, (uintptr_t) arg2,       \
 181                     (uintptr_t) arg3, (uintptr_t) arg4);                        \
 182         } \
 183 } while (0)
 184
 185 #define SDT_PROBE_ARGTYPE(_prov, _mod, _func, _name, _num, _type, _xtype) \
 186         static struct sdt_argtype                                       \
 187             sdta_##_prov##_##_mod##_##_func##_##_name##_num[1] = {      \
 188                 [0] = {                                                 \
 189                     .ndx = _num,                                        \
 190                     .type = _type,                                      \
 191                     .xtype = _xtype,                                    \
 192                     .probe = sdt_##_prov##_##_mod##_##_func##_##_name,  \
 193                 },                                                      \
 194         };                                                              \
 195         DATA_SET(sdt_argtypes_set,                                      \
 196             sdta_##_prov##_##_mod##_##_func##_##_name##_num);
 197
 198 #define SDT_PROBE_DEFINE0(prov, mod, func, name)                        \
 199         SDT_PROBE_DEFINE(prov, mod, func, name)
 200
 201 #define SDT_PROBE_DEFINE1(prov, mod, func, name, arg0)                  \
 202         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 203         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL)
 204
 205 #define SDT_PROBE_DEFINE2(prov, mod, func, name, arg0, arg1)            \
 206         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 207         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL);        \
 208         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, NULL)
 209
 210 #define SDT_PROBE_DEFINE3(prov, mod, func, name, arg0, arg1, arg2)\
 211         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 212         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL);        \
 213         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, NULL);        \
 214         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, NULL)
 215
 216 #define SDT_PROBE_DEFINE4(prov, mod, func, name, arg0, arg1, arg2, arg3) \
 217         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 218         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL);        \
 219         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, NULL);        \
 220         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, NULL);        \
 221         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, NULL)
 222
 223 #define SDT_PROBE_DEFINE5(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4) \
 224         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 225         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL);        \
 226         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, NULL);        \
 227         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, NULL);        \
 228         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, NULL);        \
 229         SDT_PROBE_ARGTYPE(prov, mod, func, name, 4, arg4, NULL)
 230
 231 #define SDT_PROBE_DEFINE6(prov, mod, func, name, arg0, arg1, arg2, arg3,\
 232     arg4, arg5) \
 233         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 234         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL);        \
 235         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, NULL);        \
 236         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, NULL);        \
 237         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, NULL);        \
 238         SDT_PROBE_ARGTYPE(prov, mod, func, name, 4, arg4, NULL);        \
 239         SDT_PROBE_ARGTYPE(prov, mod, func, name, 5, arg5, NULL)
 240
 241 #define SDT_PROBE_DEFINE7(prov, mod, func, name, arg0, arg1, arg2, arg3,\
 242     arg4, arg5, arg6) \
 243         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 244         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, NULL);        \
 245         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, NULL);        \
 246         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, NULL);        \
 247         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, NULL);        \
 248         SDT_PROBE_ARGTYPE(prov, mod, func, name, 4, arg4, NULL);        \
 249         SDT_PROBE_ARGTYPE(prov, mod, func, name, 5, arg5, NULL);        \
 250         SDT_PROBE_ARGTYPE(prov, mod, func, name, 6, arg6, NULL)
 251
 252 #define SDT_PROBE_DEFINE0_XLATE(prov, mod, func, name)          \
 253         SDT_PROBE_DEFINE(prov, mod, func, name)
 254
 255 #define SDT_PROBE_DEFINE1_XLATE(prov, mod, func, name, arg0, xarg0) \
 256         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 257         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0)
 258
 259 #define SDT_PROBE_DEFINE2_XLATE(prov, mod, func, name, arg0, xarg0, \
 260     arg1,  xarg1)                                                       \
 261         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 262         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0);       \
 263         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, xarg1)
 264
 265 #define SDT_PROBE_DEFINE3_XLATE(prov, mod, func, name, arg0, xarg0, \
 266     arg1, xarg1, arg2, xarg2)                                           \
 267         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 268         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0);       \
 269         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, xarg1);       \
 270         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, xarg2)
 271
 272 #define SDT_PROBE_DEFINE4_XLATE(prov, mod, func, name, arg0, xarg0, \
 273     arg1, xarg1, arg2, xarg2, arg3, xarg3)                              \
 274         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 275         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0);       \
 276         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, xarg1);       \
 277         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, xarg2);       \
 278         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, xarg3)
 279
 280 #define SDT_PROBE_DEFINE5_XLATE(prov, mod, func, name, arg0, xarg0, \
 281     arg1, xarg1, arg2, xarg2, arg3, xarg3, arg4, xarg4)                 \
 282         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 283         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0);       \
 284         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, xarg1);       \
 285         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, xarg2);       \
 286         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, xarg3);       \
 287         SDT_PROBE_ARGTYPE(prov, mod, func, name, 4, arg4, xarg4)
 288
 289 #define SDT_PROBE_DEFINE6_XLATE(prov, mod, func, name, arg0, xarg0, \
 290     arg1, xarg1, arg2, xarg2, arg3, xarg3, arg4, xarg4, arg5, xarg5)    \
 291         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 292         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0);       \
 293         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, xarg1);       \
 294         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, xarg2);       \
 295         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, xarg3);       \
 296         SDT_PROBE_ARGTYPE(prov, mod, func, name, 4, arg4, xarg4);       \
 297         SDT_PROBE_ARGTYPE(prov, mod, func, name, 5, arg5, xarg5)
 298
 299 #define SDT_PROBE_DEFINE7_XLATE(prov, mod, func, name, arg0, xarg0, \
 300     arg1, xarg1, arg2, xarg2, arg3, xarg3, arg4, xarg4, arg5, xarg5, arg6, \
 301     xarg6)                                                              \
 302         SDT_PROBE_DEFINE(prov, mod, func, name);                        \
 303         SDT_PROBE_ARGTYPE(prov, mod, func, name, 0, arg0, xarg0);       \
 304         SDT_PROBE_ARGTYPE(prov, mod, func, name, 1, arg1, xarg1);       \
 305         SDT_PROBE_ARGTYPE(prov, mod, func, name, 2, arg2, xarg2);       \
 306         SDT_PROBE_ARGTYPE(prov, mod, func, name, 3, arg3, xarg3);       \
 307         SDT_PROBE_ARGTYPE(prov, mod, func, name, 4, arg4, xarg4);       \
 308         SDT_PROBE_ARGTYPE(prov, mod, func, name, 5, arg5, xarg5);       \
 309         SDT_PROBE_ARGTYPE(prov, mod, func, name, 6, arg6, xarg6)
 310
 311 #define SDT_PROBE0(prov, mod, func, name)                               \
 312         SDT_PROBE(prov, mod, func, name, 0, 0, 0, 0, 0)
 313 #define SDT_PROBE1(prov, mod, func, name, arg0)                         \
 314         SDT_PROBE(prov, mod, func, name, arg0, 0, 0, 0, 0)
 315 #define SDT_PROBE2(prov, mod, func, name, arg0, arg1)                   \
 316         SDT_PROBE(prov, mod, func, name, arg0, arg1, 0, 0, 0)
 317 #define SDT_PROBE3(prov, mod, func, name, arg0, arg1, arg2)             \
 318         SDT_PROBE(prov, mod, func, name, arg0, arg1, arg2,  0, 0)
 319 #define SDT_PROBE4(prov, mod, func, name, arg0, arg1, arg2, arg3)       \
 320         SDT_PROBE(prov, mod, func, name, arg0, arg1, arg2, arg3, 0)
 321 #define SDT_PROBE5(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4) \
 322         SDT_PROBE(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4)
 323 #define SDT_PROBE6(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4, arg5)  \
 324         do {                                                                   \
 325                 if (sdt_##prov##_##mod##_##func##_##name->id)                  \
 326                         (*(void (*)(uint32_t, uintptr_t, uintptr_t, uintptr_t, \
 327                             uintptr_t, uintptr_t, uintptr_t))sdt_probe_func)(  \
 328                             sdt_##prov##_##mod##_##func##_##name->id,          \
 329                             (uintptr_t)arg0, (uintptr_t)arg1, (uintptr_t)arg2, \
 330                             (uintptr_t)arg3, (uintptr_t)arg4, (uintptr_t)arg5);\
 331         } while (0)
 332 #define SDT_PROBE7(prov, mod, func, name, arg0, arg1, arg2, arg3, arg4, arg5,  \
 333     arg6)                                                                      \
 334         do {                                                                   \
 335                 if (sdt_##prov##_##mod##_##func##_##name->id)                  \
 336                         (*(void (*)(uint32_t, uintptr_t, uintptr_t, uintptr_t, \
 337                             uintptr_t, uintptr_t, uintptr_t, uintptr_t))       \
 338                             sdt_probe_func)(                                   \
 339                             sdt_##prov##_##mod##_##func##_##name->id,          \
 340                             (uintptr_t)arg0, (uintptr_t)arg1, (uintptr_t)arg2, \
 341                             (uintptr_t)arg3, (uintptr_t)arg4, (uintptr_t)arg5, \
 342                             (uintptr_t)arg6);                                  \
 343         } while (0)
 344
 345 #define DTRACE_PROBE_IMPL_START(name, arg0, arg1, arg2, arg3, arg4)     do { \
 346         static SDT_PROBE_DEFINE(sdt, , , name);                              \
 347         SDT_PROBE(sdt, , , name, arg0, arg1, arg2, arg3, arg4);
 348 #define DTRACE_PROBE_IMPL_END   } while (0)
 349
 350 #define DTRACE_PROBE(name)                                              \
 351         DTRACE_PROBE_IMPL_START(name, 0, 0, 0, 0, 0)                    \
 352         DTRACE_PROBE_IMPL_END
 353
 354 #define DTRACE_PROBE1(name, type0, arg0)                                \
 355         DTRACE_PROBE_IMPL_START(name, arg0, 0, 0, 0, 0)                 \
 356         SDT_PROBE_ARGTYPE(sdt, , , name, 0, #type0, NULL);              \
 357         DTRACE_PROBE_IMPL_END
 358
 359 #define DTRACE_PROBE2(name, type0, arg0, type1, arg1)                   \
 360         DTRACE_PROBE_IMPL_START(name, arg0, arg1, 0, 0, 0)              \
 361         SDT_PROBE_ARGTYPE(sdt, , , name, 0, #type0, NULL);              \
 362         SDT_PROBE_ARGTYPE(sdt, , , name, 1, #type1, NULL);              \
 363         DTRACE_PROBE_IMPL_END
 364
 365 #define DTRACE_PROBE3(name, type0, arg0, type1, arg1, type2, arg2)      \
 366         DTRACE_PROBE_IMPL_START(name, arg0, arg1, arg2, 0, 0)           \
 367         SDT_PROBE_ARGTYPE(sdt, , , name, 0, #type0, NULL);              \
 368         SDT_PROBE_ARGTYPE(sdt, , , name, 1, #type1, NULL);              \
 369         SDT_PROBE_ARGTYPE(sdt, , , name, 2, #type2, NULL);              \
 370         DTRACE_PROBE_IMPL_END
 371
 372 #define DTRACE_PROBE4(name, type0, arg0, type1, arg1, type2, arg2, type3, arg3) \
 373         DTRACE_PROBE_IMPL_START(name, arg0, arg1, arg2, arg3, 0)        \
 374         SDT_PROBE_ARGTYPE(sdt, , , name, 0, #type0, NULL);              \
 375         SDT_PROBE_ARGTYPE(sdt, , , name, 1, #type1, NULL);              \
 376         SDT_PROBE_ARGTYPE(sdt, , , name, 2, #type2, NULL);              \
 377         SDT_PROBE_ARGTYPE(sdt, , , name, 3, #type3, NULL);              \
 378         DTRACE_PROBE_IMPL_END
 379
 380 #define DTRACE_PROBE5(name, type0, arg0, type1, arg1, type2, arg2, type3, arg3, \
 381     type4, arg4)                                                                \
 382         DTRACE_PROBE_IMPL_START(name, arg0, arg1, arg2, arg3, arg4)     \
 383         SDT_PROBE_ARGTYPE(sdt, , , name, 0, #type0, NULL);              \
 384         SDT_PROBE_ARGTYPE(sdt, , , name, 1, #type1, NULL);              \
 385         SDT_PROBE_ARGTYPE(sdt, , , name, 2, #type2, NULL);              \
 386         SDT_PROBE_ARGTYPE(sdt, , , name, 3, #type3, NULL);              \
 387         SDT_PROBE_ARGTYPE(sdt, , , name, 4, #type4, NULL);              \
 388         DTRACE_PROBE_IMPL_END
 389
 390 #endif /* KDTRACE_HOOKS */
 391
 392 /*
 393  * This type definition must match that of dtrace_probe. It is defined this
 394  * way to avoid having to rely on CDDL code.
 395  */
 396 typedef void (*sdt_probe_func_t)(uint32_t, uintptr_t arg0, uintptr_t arg1,
 397     uintptr_t arg2, uintptr_t arg3, uintptr_t arg4);
 398
 399 /*
 400  * The 'sdt' provider will set it to dtrace_probe when it loads.
 401  */
 402 extern sdt_probe_func_t sdt_probe_func;
 403
 404 struct sdt_probe;
 405 struct sdt_provider;
 406 struct linker_file;
 407
 408 struct sdt_argtype {
 409         int             ndx;            /* Argument index. */
 410         const char      *type;          /* Argument type string. */
 411         const char      *xtype;         /* Translated argument type. */
 412         TAILQ_ENTRY(sdt_argtype)
 413                         argtype_entry;  /* Argument type list entry. */
 414         struct sdt_probe *probe;        /* Ptr to the probe structure. */
 415 };
 416
 417 struct sdt_probe {
 418         int             version;        /* Set to sizeof(struct sdt_probe). */
 419         struct sdt_provider *prov;      /* Ptr to the provider structure. */
 420         TAILQ_ENTRY(sdt_probe)
 421                         probe_entry;    /* SDT probe list entry. */
 422         TAILQ_HEAD(, sdt_argtype) argtype_list;
 423         const char      *mod;
 424         const char      *func;
 425         const char      *name;
 426         id_t            id;             /* DTrace probe ID. */
 427         int             n_args;         /* Number of arguments. */
 428         struct linker_file *sdtp_lf;    /* Module in which we're defined. */
 429 };
 430
 431 struct sdt_provider {
 432         char *name;                     /* Provider name. */
 433         TAILQ_ENTRY(sdt_provider)
 434                         prov_entry;     /* SDT provider list entry. */
 435         uintptr_t       id;             /* DTrace provider ID. */
 436         int             sdt_refs;       /* Number of module references. */
 437 };
 438
 439 void sdt_probe_stub(uint32_t, uintptr_t, uintptr_t, uintptr_t, uintptr_t,
 440     uintptr_t);
 441
 442 SDT_PROVIDER_DECLARE(sdt);
 443
 444 #endif /* _KERNEL */
 445
 446 #endif /* _SYS_SDT_H */