sys/sys/sdt.h

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