[40518] | 1 | /* $Id: SUPDrv-dtrace.cpp 98103 2023-01-17 14:15:46Z vboxsync $ */
|
---|
| 2 | /** @file
|
---|
| 3 | * VBoxDrv - The VirtualBox Support Driver - DTrace Provider.
|
---|
| 4 | */
|
---|
| 5 |
|
---|
| 6 | /*
|
---|
[98103] | 7 | * Copyright (C) 2012-2023 Oracle and/or its affiliates.
|
---|
[40518] | 8 | *
|
---|
[96407] | 9 | * This file is part of VirtualBox base platform packages, as
|
---|
| 10 | * available from https://www.virtualbox.org.
|
---|
[40518] | 11 | *
|
---|
[96407] | 12 | * This program is free software; you can redistribute it and/or
|
---|
| 13 | * modify it under the terms of the GNU General Public License
|
---|
| 14 | * as published by the Free Software Foundation, in version 3 of the
|
---|
| 15 | * License.
|
---|
| 16 | *
|
---|
| 17 | * This program is distributed in the hope that it will be useful, but
|
---|
| 18 | * WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
| 19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
---|
| 20 | * General Public License for more details.
|
---|
| 21 | *
|
---|
| 22 | * You should have received a copy of the GNU General Public License
|
---|
| 23 | * along with this program; if not, see <https://www.gnu.org/licenses>.
|
---|
| 24 | *
|
---|
[40518] | 25 | * The contents of this file may alternatively be used under the terms
|
---|
| 26 | * of the Common Development and Distribution License Version 1.0
|
---|
[96407] | 27 | * (CDDL), a copy of it is provided in the "COPYING.CDDL" file included
|
---|
| 28 | * in the VirtualBox distribution, in which case the provisions of the
|
---|
[40518] | 29 | * CDDL are applicable instead of those of the GPL.
|
---|
| 30 | *
|
---|
| 31 | * You may elect to license modified versions of this file under the
|
---|
| 32 | * terms and conditions of either the GPL or the CDDL or both.
|
---|
[96407] | 33 | *
|
---|
| 34 | * SPDX-License-Identifier: GPL-3.0-only OR CDDL-1.0
|
---|
[40518] | 35 | */
|
---|
| 36 |
|
---|
| 37 |
|
---|
[57358] | 38 | /*********************************************************************************************************************************
|
---|
| 39 | * Header Files *
|
---|
| 40 | *********************************************************************************************************************************/
|
---|
[40518] | 41 | #define LOG_GROUP LOG_GROUP_SUP_DRV
|
---|
| 42 | #include "SUPDrvInternal.h"
|
---|
| 43 |
|
---|
| 44 | #include <VBox/err.h>
|
---|
[40605] | 45 | #include <VBox/log.h>
|
---|
[40600] | 46 | #include <VBox/VBoxTpG.h>
|
---|
[40602] | 47 |
|
---|
[40518] | 48 | #include <iprt/assert.h>
|
---|
[40602] | 49 | #include <iprt/ctype.h>
|
---|
| 50 | #include <iprt/mem.h>
|
---|
[40857] | 51 | #include <iprt/errno.h>
|
---|
| 52 | #ifdef RT_OS_DARWIN
|
---|
| 53 | # include <iprt/dbg.h>
|
---|
| 54 | #endif
|
---|
[40518] | 55 |
|
---|
[40857] | 56 | #ifdef RT_OS_DARWIN
|
---|
| 57 | # include VBOX_PATH_MACOSX_DTRACE_H
|
---|
[41354] | 58 | #elif defined(RT_OS_LINUX)
|
---|
[49395] | 59 | /* Avoid type and define conflicts. */
|
---|
| 60 | # undef UINT8_MAX
|
---|
| 61 | # undef UINT16_MAX
|
---|
| 62 | # undef UINT32_MAX
|
---|
| 63 | # undef UINT64_MAX
|
---|
| 64 | # undef INT64_MAX
|
---|
| 65 | # undef INT64_MIN
|
---|
| 66 | # define intptr_t dtrace_intptr_t
|
---|
| 67 |
|
---|
| 68 | # if 0
|
---|
| 69 | /* DTrace experiments with the Unbreakable Enterprise Kernel (UEK2)
|
---|
[41354] | 70 | (Oracle Linux).
|
---|
[48952] | 71 | 1. The dtrace.h here is from the dtrace module source, not
|
---|
[41354] | 72 | /usr/include/sys/dtrace.h nor /usr/include/dtrace.h.
|
---|
[48952] | 73 | 2. To generate the missing entries for the dtrace module in Module.symvers
|
---|
[41354] | 74 | of UEK:
|
---|
| 75 | nm /lib/modules/....../kernel/drivers/dtrace/dtrace.ko \
|
---|
| 76 | | grep _crc_ \
|
---|
| 77 | | sed -e 's/^......../0x/' -e 's/ A __crc_/\t/' \
|
---|
| 78 | -e 's/$/\tdrivers\/dtrace\/dtrace\tEXPORT_SYMBOL/' \
|
---|
| 79 | >> Module.symvers
|
---|
[49395] | 80 | Update: Althernative workaround (active), resolve symbols dynamically.
|
---|
[48952] | 81 | 3. No tracepoints in vboxdrv, vboxnet* or vboxpci yet. This requires yasm
|
---|
| 82 | and VBoxTpG and build time. */
|
---|
[49395] | 83 | # include "dtrace.h"
|
---|
| 84 | # else
|
---|
| 85 | /* DTrace experiments with the Unbreakable Enterprise Kernel (UEKR3)
|
---|
| 86 | (Oracle Linux).
|
---|
| 87 | 1. To generate the missing entries for the dtrace module in Module.symvers
|
---|
| 88 | of UEK:
|
---|
| 89 | nm /lib/modules/....../kernel/drivers/dtrace/dtrace.ko \
|
---|
| 90 | | grep _crc_ \
|
---|
| 91 | | sed -e 's/^......../0x/' -e 's/ A __crc_/\t/' \
|
---|
| 92 | -e 's/$/\tdrivers\/dtrace\/dtrace\tEXPORT_SYMBOL/' \
|
---|
| 93 | >> Module.symvers
|
---|
| 94 | Update: Althernative workaround (active), resolve symbols dynamically.
|
---|
| 95 | 2. No tracepoints in vboxdrv, vboxnet* or vboxpci yet. This requires yasm
|
---|
| 96 | and VBoxTpG and build time. */
|
---|
| 97 | # include <dtrace/provider.h>
|
---|
| 98 | # include <dtrace/enabling.h> /* Missing from provider.h. */
|
---|
| 99 | # include <dtrace/arg.h> /* Missing from provider.h. */
|
---|
| 100 | # endif
|
---|
[85771] | 101 | # include <linux/module.h>
|
---|
[49395] | 102 | /** Status code fixer (UEK uses linux convension unlike the others). */
|
---|
[41357] | 103 | # define FIX_UEK_RC(a_rc) (-(a_rc))
|
---|
[40602] | 104 | #else
|
---|
| 105 | # include <sys/dtrace.h>
|
---|
[97897] | 106 | # undef u
|
---|
[40602] | 107 | #endif
|
---|
[40518] | 108 |
|
---|
| 109 |
|
---|
[48952] | 110 | /**
|
---|
| 111 | * The UEK DTrace port is trying to be smart and seems to have turned all
|
---|
[41357] | 112 | * errno return codes negative. While this conforms to the linux kernel way of
|
---|
[48952] | 113 | * doing things, it breaks with the way the interfaces work on Solaris and
|
---|
[41357] | 114 | * Mac OS X.
|
---|
| 115 | */
|
---|
| 116 | #ifndef FIX_UEK_RC
|
---|
[48952] | 117 | # define FIX_UEK_RC(a_rc) (a_rc)
|
---|
[41357] | 118 | #endif
|
---|
| 119 |
|
---|
| 120 |
|
---|
[64584] | 121 | /** @name Macros for preserving EFLAGS.AC (despair / paranoid)
|
---|
| 122 | * @remarks We have to restore it unconditionally on darwin.
|
---|
| 123 | * @{ */
|
---|
| 124 | #if defined(RT_OS_DARWIN) \
|
---|
| 125 | || ( defined(RT_OS_LINUX) \
|
---|
| 126 | && (defined(CONFIG_X86_SMAP) || defined(RT_STRICT) || defined(IPRT_WITH_EFLAGS_AC_PRESERVING) ) )
|
---|
| 127 | # include <iprt/asm-amd64-x86.h>
|
---|
| 128 | # include <iprt/x86.h>
|
---|
| 129 | # define SUPDRV_SAVE_EFL_AC() RTCCUINTREG const fSavedEfl = ASMGetFlags();
|
---|
| 130 | # define SUPDRV_RESTORE_EFL_AC() ASMSetFlags(fSavedEfl)
|
---|
| 131 | # define SUPDRV_RESTORE_EFL_ONLY_AC() ASMChangeFlags(~X86_EFL_AC, fSavedEfl & X86_EFL_AC)
|
---|
| 132 | #else
|
---|
| 133 | # define SUPDRV_SAVE_EFL_AC() do { } while (0)
|
---|
| 134 | # define SUPDRV_RESTORE_EFL_AC() do { } while (0)
|
---|
| 135 | # define SUPDRV_RESTORE_EFL_ONLY_AC() do { } while (0)
|
---|
| 136 | #endif
|
---|
| 137 | /** @} */
|
---|
| 138 |
|
---|
| 139 |
|
---|
[57358] | 140 | /*********************************************************************************************************************************
|
---|
| 141 | * Structures and Typedefs *
|
---|
| 142 | *********************************************************************************************************************************/
|
---|
[40602] | 143 | /* Seems there is some return code difference here. Keep the return code and
|
---|
| 144 | case it to whatever the host desires. */
|
---|
[40893] | 145 | #ifdef RT_OS_DARWIN
|
---|
| 146 | # if MAC_OS_X_VERSION_MIN_REQUIRED < 1070
|
---|
[40602] | 147 | typedef void FNPOPS_ENABLE(void *, dtrace_id_t, void *);
|
---|
[40893] | 148 | # else
|
---|
| 149 | typedef int FNPOPS_ENABLE(void *, dtrace_id_t, void *);
|
---|
| 150 | # endif
|
---|
[40602] | 151 | #else
|
---|
[97897] | 152 | # if !defined(RT_OS_SOLARIS) || defined(DOF_SEC_ISLOADABLE) /* DOF_SEC_ISLOADABLE was added in the next commit after dtps_enable change signature, but it's the simplest check we can do. */
|
---|
[40602] | 153 | typedef int FNPOPS_ENABLE(void *, dtrace_id_t, void *);
|
---|
[97897] | 154 | # else
|
---|
| 155 | typedef void FNPOPS_ENABLE(void *, dtrace_id_t, void *);
|
---|
| 156 | # endif
|
---|
[40602] | 157 | #endif
|
---|
[40600] | 158 |
|
---|
[41130] | 159 | /** Caller indicator. */
|
---|
| 160 | typedef enum VBOXDTCALLER
|
---|
| 161 | {
|
---|
| 162 | kVBoxDtCaller_Invalid = 0,
|
---|
| 163 | kVBoxDtCaller_Generic,
|
---|
| 164 | kVBoxDtCaller_ProbeFireUser,
|
---|
| 165 | kVBoxDtCaller_ProbeFireKernel
|
---|
| 166 | } VBOXDTCALLER;
|
---|
[40602] | 167 |
|
---|
[41130] | 168 |
|
---|
[40851] | 169 | /**
|
---|
| 170 | * Stack data planted before calling dtrace_probe so that we can easily find the
|
---|
| 171 | * stack argument later.
|
---|
| 172 | */
|
---|
[41130] | 173 | typedef struct VBDTSTACKDATA
|
---|
[40851] | 174 | {
|
---|
| 175 | /** Eyecatcher no. 1 (SUPDRVDT_STACK_DATA_MAGIC2). */
|
---|
| 176 | uint32_t u32Magic1;
|
---|
| 177 | /** Eyecatcher no. 2 (SUPDRVDT_STACK_DATA_MAGIC2). */
|
---|
| 178 | uint32_t u32Magic2;
|
---|
[41130] | 179 | /** The format of the caller specific data. */
|
---|
| 180 | VBOXDTCALLER enmCaller;
|
---|
| 181 | /** Caller specific data. */
|
---|
| 182 | union
|
---|
| 183 | {
|
---|
| 184 | /** kVBoxDtCaller_ProbeFireKernel. */
|
---|
| 185 | struct
|
---|
| 186 | {
|
---|
| 187 | /** Pointer to the stack arguments of a probe function call. */
|
---|
| 188 | uintptr_t *pauStackArgs;
|
---|
| 189 | } ProbeFireKernel;
|
---|
| 190 | /** kVBoxDtCaller_ProbeFireUser. */
|
---|
| 191 | struct
|
---|
| 192 | {
|
---|
| 193 | /** The user context. */
|
---|
| 194 | PCSUPDRVTRACERUSRCTX pCtx;
|
---|
| 195 | /** The argument displacement caused by 64-bit arguments passed directly to
|
---|
| 196 | * dtrace_probe. */
|
---|
| 197 | int offArg;
|
---|
| 198 | } ProbeFireUser;
|
---|
| 199 | } u;
|
---|
[40851] | 200 | /** Pointer to this structure.
|
---|
| 201 | * This is the final bit of integrity checking. */
|
---|
[41130] | 202 | struct VBDTSTACKDATA *pSelf;
|
---|
| 203 | } VBDTSTACKDATA;
|
---|
[40851] | 204 | /** Pointer to the on-stack thread specific data. */
|
---|
[41130] | 205 | typedef VBDTSTACKDATA *PVBDTSTACKDATA;
|
---|
[40704] | 206 |
|
---|
[40865] | 207 |
|
---|
[57358] | 208 | /*********************************************************************************************************************************
|
---|
| 209 | * Defined Constants And Macros *
|
---|
| 210 | *********************************************************************************************************************************/
|
---|
[40851] | 211 | /** The first magic value. */
|
---|
| 212 | #define SUPDRVDT_STACK_DATA_MAGIC1 RT_MAKE_U32_FROM_U8('S', 'U', 'P', 'D')
|
---|
| 213 | /** The second magic value. */
|
---|
| 214 | #define SUPDRVDT_STACK_DATA_MAGIC2 RT_MAKE_U32_FROM_U8('D', 'T', 'r', 'c')
|
---|
[40518] | 215 |
|
---|
[40851] | 216 | /** The alignment of the stack data.
|
---|
| 217 | * The data doesn't require more than sizeof(uintptr_t) alignment, but the
|
---|
| 218 | * greater alignment the quicker lookup. */
|
---|
| 219 | #define SUPDRVDT_STACK_DATA_ALIGN 32
|
---|
[40518] | 220 |
|
---|
[40851] | 221 | /** Plants the stack data. */
|
---|
[41130] | 222 | #define VBDT_SETUP_STACK_DATA(a_enmCaller) \
|
---|
| 223 | uint8_t abBlob[sizeof(VBDTSTACKDATA) + SUPDRVDT_STACK_DATA_ALIGN - 1]; \
|
---|
| 224 | PVBDTSTACKDATA pStackData = (PVBDTSTACKDATA)( (uintptr_t)&abBlob[SUPDRVDT_STACK_DATA_ALIGN - 1] \
|
---|
| 225 | & ~(uintptr_t)(SUPDRVDT_STACK_DATA_ALIGN - 1)); \
|
---|
[40851] | 226 | pStackData->u32Magic1 = SUPDRVDT_STACK_DATA_MAGIC1; \
|
---|
| 227 | pStackData->u32Magic2 = SUPDRVDT_STACK_DATA_MAGIC2; \
|
---|
[41130] | 228 | pStackData->enmCaller = a_enmCaller; \
|
---|
[40851] | 229 | pStackData->pSelf = pStackData
|
---|
[40518] | 230 |
|
---|
[40851] | 231 | /** Passifies the stack data and frees up resource held within it. */
|
---|
[41130] | 232 | #define VBDT_CLEAR_STACK_DATA() \
|
---|
[40851] | 233 | do \
|
---|
| 234 | { \
|
---|
| 235 | pStackData->u32Magic1 = 0; \
|
---|
| 236 | pStackData->u32Magic2 = 0; \
|
---|
| 237 | pStackData->pSelf = NULL; \
|
---|
| 238 | } while (0)
|
---|
| 239 |
|
---|
[40867] | 240 | /** Simple SUPR0Printf-style logging. */
|
---|
| 241 | #if 0 /*def DEBUG_bird*/
|
---|
[40865] | 242 | # define LOG_DTRACE(a) SUPR0Printf a
|
---|
| 243 | #else
|
---|
| 244 | # define LOG_DTRACE(a) do { } while (0)
|
---|
| 245 | #endif
|
---|
[40851] | 246 |
|
---|
[40865] | 247 |
|
---|
[57358] | 248 | /*********************************************************************************************************************************
|
---|
| 249 | * Global Variables *
|
---|
| 250 | *********************************************************************************************************************************/
|
---|
[49395] | 251 | #if defined(RT_OS_DARWIN) || defined(RT_OS_LINUX)
|
---|
| 252 | /** @name DTrace kernel interface used on Darwin and Linux.
|
---|
[40857] | 253 | * @{ */
|
---|
[85172] | 254 | static DECLCALLBACKPTR_EX(void, RT_NOTHING, g_pfnDTraceProbeFire,(dtrace_id_t, uint64_t, uint64_t, uint64_t, uint64_t,
|
---|
| 255 | uint64_t));
|
---|
| 256 | static DECLCALLBACKPTR_EX(dtrace_id_t, RT_NOTHING, g_pfnDTraceProbeCreate,(dtrace_provider_id_t, const char *, const char *,
|
---|
| 257 | const char *, int, void *));
|
---|
| 258 | static DECLCALLBACKPTR_EX(dtrace_id_t, RT_NOTHING, g_pfnDTraceProbeLookup,(dtrace_provider_id_t, const char *, const char *,
|
---|
| 259 | const char *));
|
---|
| 260 | static DECLCALLBACKPTR_EX(int, RT_NOTHING, g_pfnDTraceProviderRegister,(const char *, const dtrace_pattr_t *, uint32_t,
|
---|
| 261 | /*cred_t*/ void *, const dtrace_pops_t *,
|
---|
| 262 | void *, dtrace_provider_id_t *));
|
---|
| 263 | static DECLCALLBACKPTR_EX(void, RT_NOTHING, g_pfnDTraceProviderInvalidate,(dtrace_provider_id_t));
|
---|
| 264 | static DECLCALLBACKPTR_EX(int, RT_NOTHING, g_pfnDTraceProviderUnregister,(dtrace_provider_id_t));
|
---|
[40857] | 265 |
|
---|
| 266 | #define dtrace_probe g_pfnDTraceProbeFire
|
---|
| 267 | #define dtrace_probe_create g_pfnDTraceProbeCreate
|
---|
| 268 | #define dtrace_probe_lookup g_pfnDTraceProbeLookup
|
---|
| 269 | #define dtrace_register g_pfnDTraceProviderRegister
|
---|
| 270 | #define dtrace_invalidate g_pfnDTraceProviderInvalidate
|
---|
| 271 | #define dtrace_unregister g_pfnDTraceProviderUnregister
|
---|
| 272 |
|
---|
[85766] | 273 | /** For dynamical resolving and releasing. */
|
---|
| 274 | static const struct
|
---|
| 275 | {
|
---|
| 276 | const char *pszName;
|
---|
| 277 | uintptr_t *ppfn; /**< @note Clang 11 nothrow weirdness forced this from PFNRT * to uintptr_t *. */
|
---|
| 278 | } g_aDTraceFunctions[] =
|
---|
| 279 | {
|
---|
| 280 | { "dtrace_probe", (uintptr_t *)&dtrace_probe },
|
---|
| 281 | { "dtrace_probe_create", (uintptr_t *)&dtrace_probe_create },
|
---|
| 282 | { "dtrace_probe_lookup", (uintptr_t *)&dtrace_probe_lookup },
|
---|
| 283 | { "dtrace_register", (uintptr_t *)&dtrace_register },
|
---|
| 284 | { "dtrace_invalidate", (uintptr_t *)&dtrace_invalidate },
|
---|
| 285 | { "dtrace_unregister", (uintptr_t *)&dtrace_unregister },
|
---|
| 286 | };
|
---|
| 287 |
|
---|
[40851] | 288 | /** @} */
|
---|
[40704] | 289 | #endif
|
---|
[40518] | 290 |
|
---|
| 291 |
|
---|
[41130] | 292 | /**
|
---|
| 293 | * Gets the stack data.
|
---|
| 294 | *
|
---|
| 295 | * @returns Pointer to the stack data. Never NULL.
|
---|
| 296 | */
|
---|
| 297 | static PVBDTSTACKDATA vboxDtGetStackData(void)
|
---|
| 298 | {
|
---|
| 299 | /*
|
---|
| 300 | * Locate the caller of probe_dtrace.
|
---|
| 301 | */
|
---|
| 302 | int volatile iDummy = 1; /* use this to get the stack address. */
|
---|
| 303 | PVBDTSTACKDATA pData = (PVBDTSTACKDATA)( ((uintptr_t)&iDummy + SUPDRVDT_STACK_DATA_ALIGN - 1)
|
---|
| 304 | & ~(uintptr_t)(SUPDRVDT_STACK_DATA_ALIGN - 1));
|
---|
| 305 | for (;;)
|
---|
| 306 | {
|
---|
| 307 | if ( pData->u32Magic1 == SUPDRVDT_STACK_DATA_MAGIC1
|
---|
| 308 | && pData->u32Magic2 == SUPDRVDT_STACK_DATA_MAGIC2
|
---|
| 309 | && pData->pSelf == pData)
|
---|
| 310 | return pData;
|
---|
| 311 | pData = (PVBDTSTACKDATA)((uintptr_t)pData + SUPDRVDT_STACK_DATA_ALIGN);
|
---|
| 312 | }
|
---|
| 313 | }
|
---|
| 314 |
|
---|
| 315 |
|
---|
[40851] | 316 | /*
|
---|
[40602] | 317 | *
|
---|
[40851] | 318 | * Helpers for handling VTG structures.
|
---|
| 319 | * Helpers for handling VTG structures.
|
---|
| 320 | * Helpers for handling VTG structures.
|
---|
| 321 | *
|
---|
[40600] | 322 | */
|
---|
| 323 |
|
---|
[40601] | 324 |
|
---|
[40600] | 325 |
|
---|
[40518] | 326 | /**
|
---|
[40602] | 327 | * Converts an attribute from VTG description speak to DTrace.
|
---|
| 328 | *
|
---|
| 329 | * @param pDtAttr The DTrace attribute (dst).
|
---|
| 330 | * @param pVtgAttr The VTG attribute descriptor (src).
|
---|
| 331 | */
|
---|
[40851] | 332 | static void vboxDtVtgConvAttr(dtrace_attribute_t *pDtAttr, PCVTGDESCATTR pVtgAttr)
|
---|
[40602] | 333 | {
|
---|
| 334 | pDtAttr->dtat_name = pVtgAttr->u8Code - 1;
|
---|
| 335 | pDtAttr->dtat_data = pVtgAttr->u8Data - 1;
|
---|
| 336 | pDtAttr->dtat_class = pVtgAttr->u8DataDep - 1;
|
---|
| 337 | }
|
---|
| 338 |
|
---|
| 339 | /**
|
---|
| 340 | * Gets a string from the string table.
|
---|
| 341 | *
|
---|
| 342 | * @returns Pointer to the string.
|
---|
| 343 | * @param pVtgHdr The VTG object header.
|
---|
| 344 | * @param offStrTab The string table offset.
|
---|
| 345 | */
|
---|
[41130] | 346 | static const char *vboxDtVtgGetString(PVTGOBJHDR pVtgHdr, uint32_t offStrTab)
|
---|
[40602] | 347 | {
|
---|
| 348 | Assert(offStrTab < pVtgHdr->cbStrTab);
|
---|
[41130] | 349 | return (const char *)pVtgHdr + pVtgHdr->offStrTab + offStrTab;
|
---|
[40602] | 350 | }
|
---|
| 351 |
|
---|
| 352 |
|
---|
[40675] | 353 |
|
---|
[40851] | 354 | /*
|
---|
[40756] | 355 | *
|
---|
[40851] | 356 | * DTrace Provider Interface.
|
---|
| 357 | * DTrace Provider Interface.
|
---|
| 358 | * DTrace Provider Interface.
|
---|
[40756] | 359 | *
|
---|
[40704] | 360 | */
|
---|
| 361 |
|
---|
| 362 |
|
---|
| 363 | /**
|
---|
[40602] | 364 | * @callback_method_impl{dtrace_pops_t,dtps_provide}
|
---|
[40518] | 365 | */
|
---|
[41130] | 366 | static void vboxDtPOps_Provide(void *pvProv, const dtrace_probedesc_t *pDtProbeDesc)
|
---|
[40518] | 367 | {
|
---|
[41130] | 368 | PSUPDRVVDTPROVIDERCORE pProv = (PSUPDRVVDTPROVIDERCORE)pvProv;
|
---|
[40865] | 369 | AssertPtrReturnVoid(pProv);
|
---|
[41130] | 370 | LOG_DTRACE(("%s: %p / %p pDtProbeDesc=%p\n", __FUNCTION__, pProv, pProv->TracerData.DTrace.idProvider, pDtProbeDesc));
|
---|
[40602] | 371 |
|
---|
| 372 | if (pDtProbeDesc)
|
---|
| 373 | return; /* We don't generate probes, so never mind these requests. */
|
---|
| 374 |
|
---|
[40851] | 375 | if (pProv->TracerData.DTrace.fZombie)
|
---|
[40675] | 376 | return;
|
---|
| 377 |
|
---|
[41130] | 378 | dtrace_provider_id_t const idProvider = pProv->TracerData.DTrace.idProvider;
|
---|
[40898] | 379 | AssertPtrReturnVoid(idProvider);
|
---|
[40975] | 380 |
|
---|
[40898] | 381 | AssertPtrReturnVoid(pProv->pHdr);
|
---|
[41130] | 382 | AssertReturnVoid(pProv->pHdr->offProbeLocs != 0);
|
---|
| 383 | uint32_t const cProbeLocs = pProv->pHdr->cbProbeLocs / sizeof(VTGPROBELOC);
|
---|
[40617] | 384 |
|
---|
[41130] | 385 | /* Need a buffer for extracting the function names and mangling them in
|
---|
| 386 | case of collision. */
|
---|
| 387 | size_t const cbFnNmBuf = _4K + _1K;
|
---|
| 388 | char *pszFnNmBuf = (char *)RTMemAlloc(cbFnNmBuf);
|
---|
| 389 | if (!pszFnNmBuf)
|
---|
[40602] | 390 | return;
|
---|
| 391 |
|
---|
[41130] | 392 | /*
|
---|
| 393 | * Itereate the probe location list and register all probes related to
|
---|
| 394 | * this provider.
|
---|
| 395 | */
|
---|
| 396 | uint16_t const idxProv = (uint16_t)((PVTGDESCPROVIDER)((uintptr_t)pProv->pHdr + pProv->pHdr->offProviders) - pProv->pDesc);
|
---|
[41354] | 397 | uint32_t idxProbeLoc;
|
---|
| 398 | for (idxProbeLoc = 0; idxProbeLoc < cProbeLocs; idxProbeLoc++)
|
---|
[41130] | 399 | {
|
---|
| 400 | /* Skip probe location belonging to other providers or once that
|
---|
| 401 | we've already reported. */
|
---|
| 402 | PCVTGPROBELOC pProbeLocRO = &pProv->paProbeLocsRO[idxProbeLoc];
|
---|
| 403 | PVTGDESCPROBE pProbeDesc = pProbeLocRO->pProbe;
|
---|
| 404 | if (pProbeDesc->idxProvider != idxProv)
|
---|
| 405 | continue;
|
---|
| 406 |
|
---|
| 407 | uint32_t *pidProbe;
|
---|
| 408 | if (!pProv->fUmod)
|
---|
| 409 | pidProbe = (uint32_t *)&pProbeLocRO->idProbe;
|
---|
| 410 | else
|
---|
| 411 | pidProbe = &pProv->paR0ProbeLocs[idxProbeLoc].idProbe;
|
---|
| 412 | if (*pidProbe != 0)
|
---|
| 413 | continue;
|
---|
| 414 |
|
---|
| 415 | /* The function name may need to be stripped since we're using C++
|
---|
| 416 | compilers for most of the code. ASSUMES nobody are brave/stupid
|
---|
| 417 | enough to use function pointer returns without typedef'ing
|
---|
| 418 | properly them (e.g. signal). */
|
---|
| 419 | const char *pszPrbName = vboxDtVtgGetString(pProv->pHdr, pProbeDesc->offName);
|
---|
| 420 | const char *pszFunc = pProbeLocRO->pszFunction;
|
---|
| 421 | const char *psz = strchr(pProbeLocRO->pszFunction, '(');
|
---|
| 422 | size_t cch;
|
---|
| 423 | if (psz)
|
---|
[40602] | 424 | {
|
---|
[41130] | 425 | /* skip blanks preceeding the parameter parenthesis. */
|
---|
| 426 | while ( (uintptr_t)psz > (uintptr_t)pProbeLocRO->pszFunction
|
---|
| 427 | && RT_C_IS_BLANK(psz[-1]))
|
---|
| 428 | psz--;
|
---|
| 429 |
|
---|
| 430 | /* Find the start of the function name. */
|
---|
| 431 | pszFunc = psz - 1;
|
---|
| 432 | while ((uintptr_t)pszFunc > (uintptr_t)pProbeLocRO->pszFunction)
|
---|
[40602] | 433 | {
|
---|
[41130] | 434 | char ch = pszFunc[-1];
|
---|
| 435 | if (!RT_C_IS_ALNUM(ch) && ch != '_' && ch != ':')
|
---|
| 436 | break;
|
---|
| 437 | pszFunc--;
|
---|
[40602] | 438 | }
|
---|
[41130] | 439 | cch = psz - pszFunc;
|
---|
| 440 | }
|
---|
| 441 | else
|
---|
| 442 | cch = strlen(pszFunc);
|
---|
| 443 | RTStrCopyEx(pszFnNmBuf, cbFnNmBuf, pszFunc, cch);
|
---|
[40602] | 444 |
|
---|
[41130] | 445 | /* Look up the probe, if we have one in the same function, mangle
|
---|
| 446 | the function name a little to avoid having to deal with having
|
---|
| 447 | multiple location entries with the same probe ID. (lazy bird) */
|
---|
| 448 | Assert(!*pidProbe);
|
---|
| 449 | if (dtrace_probe_lookup(idProvider, pProv->pszModName, pszFnNmBuf, pszPrbName) != DTRACE_IDNONE)
|
---|
| 450 | {
|
---|
| 451 | RTStrPrintf(pszFnNmBuf+cch, cbFnNmBuf - cch, "-%u", pProbeLocRO->uLine);
|
---|
[40851] | 452 | if (dtrace_probe_lookup(idProvider, pProv->pszModName, pszFnNmBuf, pszPrbName) != DTRACE_IDNONE)
|
---|
[40605] | 453 | {
|
---|
[41130] | 454 | unsigned iOrd = 2;
|
---|
| 455 | while (iOrd < 128)
|
---|
[40605] | 456 | {
|
---|
[41130] | 457 | RTStrPrintf(pszFnNmBuf+cch, cbFnNmBuf - cch, "-%u-%u", pProbeLocRO->uLine, iOrd);
|
---|
| 458 | if (dtrace_probe_lookup(idProvider, pProv->pszModName, pszFnNmBuf, pszPrbName) == DTRACE_IDNONE)
|
---|
| 459 | break;
|
---|
| 460 | iOrd++;
|
---|
[40605] | 461 | }
|
---|
[41130] | 462 | if (iOrd >= 128)
|
---|
| 463 | {
|
---|
| 464 | LogRel(("VBoxDrv: More than 128 duplicate probe location instances %s at line %u in function %s [%s], probe %s\n",
|
---|
| 465 | pProbeLocRO->uLine, pProbeLocRO->pszFunction, pszFnNmBuf, pszPrbName));
|
---|
| 466 | continue;
|
---|
| 467 | }
|
---|
[40605] | 468 | }
|
---|
[40602] | 469 | }
|
---|
| 470 |
|
---|
[41130] | 471 | /* Create the probe. */
|
---|
| 472 | AssertCompile(sizeof(*pidProbe) == sizeof(dtrace_id_t));
|
---|
| 473 | *pidProbe = dtrace_probe_create(idProvider, pProv->pszModName, pszFnNmBuf, pszPrbName,
|
---|
| 474 | 1 /*aframes*/, (void *)(uintptr_t)idxProbeLoc);
|
---|
| 475 | pProv->TracerData.DTrace.cProvidedProbes++;
|
---|
[40602] | 476 | }
|
---|
| 477 |
|
---|
| 478 | RTMemFree(pszFnNmBuf);
|
---|
[41130] | 479 | LOG_DTRACE(("%s: returns\n", __FUNCTION__));
|
---|
[40518] | 480 | }
|
---|
| 481 |
|
---|
| 482 |
|
---|
| 483 | /**
|
---|
| 484 | * @callback_method_impl{dtrace_pops_t,dtps_enable}
|
---|
| 485 | */
|
---|
[41130] | 486 | static int vboxDtPOps_Enable(void *pvProv, dtrace_id_t idProbe, void *pvProbe)
|
---|
[40518] | 487 | {
|
---|
[41130] | 488 | PSUPDRVVDTPROVIDERCORE pProv = (PSUPDRVVDTPROVIDERCORE)pvProv;
|
---|
[63510] | 489 | RT_NOREF(idProbe);
|
---|
[41130] | 490 | LOG_DTRACE(("%s: %p / %p - %#x / %p\n", __FUNCTION__, pProv, pProv->TracerData.DTrace.idProvider, idProbe, pvProbe));
|
---|
[40865] | 491 | AssertPtrReturn(pProv->TracerData.DTrace.idProvider, EINVAL);
|
---|
| 492 |
|
---|
[40851] | 493 | if (!pProv->TracerData.DTrace.fZombie)
|
---|
[40675] | 494 | {
|
---|
[41130] | 495 | uint32_t idxProbeLoc = (uint32_t)(uintptr_t)pvProbe;
|
---|
| 496 | PVTGPROBELOC32 pProbeLocEn = (PVTGPROBELOC32)( (uintptr_t)pProv->pvProbeLocsEn + idxProbeLoc * pProv->cbProbeLocsEn);
|
---|
| 497 | PCVTGPROBELOC pProbeLocRO = (PVTGPROBELOC)&pProv->paProbeLocsRO[idxProbeLoc];
|
---|
| 498 | PCVTGDESCPROBE pProbeDesc = pProbeLocRO->pProbe;
|
---|
| 499 | uint32_t const idxProbe = pProbeDesc->idxEnabled;
|
---|
[40602] | 500 |
|
---|
[41130] | 501 | if (!pProv->fUmod)
|
---|
[40675] | 502 | {
|
---|
[41130] | 503 | if (!pProbeLocEn->fEnabled)
|
---|
| 504 | {
|
---|
| 505 | pProbeLocEn->fEnabled = 1;
|
---|
| 506 | ASMAtomicIncU32(&pProv->pacProbeEnabled[idxProbe]);
|
---|
[58920] | 507 | ASMAtomicIncU32(&pProv->pDesc->cProbesEnabled);
|
---|
[58976] | 508 | ASMAtomicIncU32(&pProv->pDesc->uSettingsSerialNo);
|
---|
[41130] | 509 | }
|
---|
[40675] | 510 | }
|
---|
[41130] | 511 | else
|
---|
| 512 | {
|
---|
| 513 | /* Update kernel mode structure */
|
---|
| 514 | if (!pProv->paR0ProbeLocs[idxProbeLoc].fEnabled)
|
---|
| 515 | {
|
---|
| 516 | pProv->paR0ProbeLocs[idxProbeLoc].fEnabled = 1;
|
---|
| 517 | ASMAtomicIncU32(&pProv->paR0Probes[idxProbe].cEnabled);
|
---|
[58920] | 518 | ASMAtomicIncU32(&pProv->pDesc->cProbesEnabled);
|
---|
[58976] | 519 | ASMAtomicIncU32(&pProv->pDesc->uSettingsSerialNo);
|
---|
[41130] | 520 | }
|
---|
| 521 |
|
---|
| 522 | /* Update user mode structure. */
|
---|
| 523 | pProbeLocEn->fEnabled = 1;
|
---|
| 524 | pProv->pacProbeEnabled[idxProbe] = pProv->paR0Probes[idxProbe].cEnabled;
|
---|
| 525 | }
|
---|
[40602] | 526 | }
|
---|
| 527 |
|
---|
| 528 | return 0;
|
---|
[40518] | 529 | }
|
---|
| 530 |
|
---|
| 531 |
|
---|
| 532 | /**
|
---|
| 533 | * @callback_method_impl{dtrace_pops_t,dtps_disable}
|
---|
| 534 | */
|
---|
[41130] | 535 | static void vboxDtPOps_Disable(void *pvProv, dtrace_id_t idProbe, void *pvProbe)
|
---|
[40518] | 536 | {
|
---|
[40851] | 537 | PSUPDRVVDTPROVIDERCORE pProv = (PSUPDRVVDTPROVIDERCORE)pvProv;
|
---|
[40865] | 538 | AssertPtrReturnVoid(pProv);
|
---|
[63510] | 539 | RT_NOREF(idProbe);
|
---|
[41130] | 540 | LOG_DTRACE(("%s: %p / %p - %#x / %p\n", __FUNCTION__, pProv, pProv->TracerData.DTrace.idProvider, idProbe, pvProbe));
|
---|
[40865] | 541 | AssertPtrReturnVoid(pProv->TracerData.DTrace.idProvider);
|
---|
| 542 |
|
---|
[40851] | 543 | if (!pProv->TracerData.DTrace.fZombie)
|
---|
[40675] | 544 | {
|
---|
[41130] | 545 | uint32_t idxProbeLoc = (uint32_t)(uintptr_t)pvProbe;
|
---|
| 546 | PVTGPROBELOC32 pProbeLocEn = (PVTGPROBELOC32)( (uintptr_t)pProv->pvProbeLocsEn + idxProbeLoc * pProv->cbProbeLocsEn);
|
---|
| 547 | PCVTGPROBELOC pProbeLocRO = (PVTGPROBELOC)&pProv->paProbeLocsRO[idxProbeLoc];
|
---|
| 548 | PCVTGDESCPROBE pProbeDesc = pProbeLocRO->pProbe;
|
---|
| 549 | uint32_t const idxProbe = pProbeDesc->idxEnabled;
|
---|
[40602] | 550 |
|
---|
[41130] | 551 | if (!pProv->fUmod)
|
---|
[40675] | 552 | {
|
---|
[41130] | 553 | if (pProbeLocEn->fEnabled)
|
---|
| 554 | {
|
---|
| 555 | pProbeLocEn->fEnabled = 0;
|
---|
| 556 | ASMAtomicDecU32(&pProv->pacProbeEnabled[idxProbe]);
|
---|
[58920] | 557 | ASMAtomicIncU32(&pProv->pDesc->cProbesEnabled);
|
---|
[58976] | 558 | ASMAtomicIncU32(&pProv->pDesc->uSettingsSerialNo);
|
---|
[41130] | 559 | }
|
---|
[40675] | 560 | }
|
---|
[41130] | 561 | else
|
---|
| 562 | {
|
---|
| 563 | /* Update kernel mode structure */
|
---|
| 564 | if (pProv->paR0ProbeLocs[idxProbeLoc].fEnabled)
|
---|
| 565 | {
|
---|
| 566 | pProv->paR0ProbeLocs[idxProbeLoc].fEnabled = 0;
|
---|
| 567 | ASMAtomicDecU32(&pProv->paR0Probes[idxProbe].cEnabled);
|
---|
[58920] | 568 | ASMAtomicDecU32(&pProv->pDesc->cProbesEnabled);
|
---|
[58976] | 569 | ASMAtomicIncU32(&pProv->pDesc->uSettingsSerialNo);
|
---|
[41130] | 570 | }
|
---|
| 571 |
|
---|
| 572 | /* Update user mode structure. */
|
---|
| 573 | pProbeLocEn->fEnabled = 0;
|
---|
| 574 | pProv->pacProbeEnabled[idxProbe] = pProv->paR0Probes[idxProbe].cEnabled;
|
---|
| 575 | }
|
---|
[40602] | 576 | }
|
---|
[40518] | 577 | }
|
---|
| 578 |
|
---|
| 579 |
|
---|
| 580 | /**
|
---|
| 581 | * @callback_method_impl{dtrace_pops_t,dtps_getargdesc}
|
---|
| 582 | */
|
---|
[41130] | 583 | static void vboxDtPOps_GetArgDesc(void *pvProv, dtrace_id_t idProbe, void *pvProbe,
|
---|
| 584 | dtrace_argdesc_t *pArgDesc)
|
---|
[40518] | 585 | {
|
---|
[40851] | 586 | PSUPDRVVDTPROVIDERCORE pProv = (PSUPDRVVDTPROVIDERCORE)pvProv;
|
---|
| 587 | unsigned uArg = pArgDesc->dtargd_ndx;
|
---|
[63510] | 588 | RT_NOREF(idProbe);
|
---|
[40602] | 589 |
|
---|
[40865] | 590 | pArgDesc->dtargd_ndx = DTRACE_ARGNONE;
|
---|
| 591 | AssertPtrReturnVoid(pProv);
|
---|
[41130] | 592 | LOG_DTRACE(("%s: %p / %p - %#x / %p uArg=%d\n", __FUNCTION__, pProv, pProv->TracerData.DTrace.idProvider, idProbe, pvProbe, uArg));
|
---|
[40865] | 593 | AssertPtrReturnVoid(pProv->TracerData.DTrace.idProvider);
|
---|
| 594 |
|
---|
[40851] | 595 | if (!pProv->TracerData.DTrace.fZombie)
|
---|
[40602] | 596 | {
|
---|
[41130] | 597 | uint32_t idxProbeLoc = (uint32_t)(uintptr_t)pvProbe;
|
---|
| 598 | PCVTGPROBELOC pProbeLocRO = (PVTGPROBELOC)&pProv->paProbeLocsRO[idxProbeLoc];
|
---|
| 599 | PCVTGDESCPROBE pProbeDesc = pProbeLocRO->pProbe;
|
---|
| 600 | PCVTGDESCARGLIST pArgList = (PCVTGDESCARGLIST)( (uintptr_t)pProv->pHdr
|
---|
| 601 | + pProv->pHdr->offArgLists
|
---|
| 602 | + pProbeDesc->offArgList);
|
---|
| 603 | AssertReturnVoid(pProbeDesc->offArgList < pProv->pHdr->cbArgLists);
|
---|
[40675] | 604 |
|
---|
[40885] | 605 | if (uArg < pArgList->cArgs)
|
---|
[40617] | 606 | {
|
---|
[40851] | 607 | const char *pszType = vboxDtVtgGetString(pProv->pHdr, pArgList->aArgs[uArg].offType);
|
---|
[40675] | 608 | size_t cchType = strlen(pszType);
|
---|
| 609 | if (cchType < sizeof(pArgDesc->dtargd_native))
|
---|
| 610 | {
|
---|
| 611 | memcpy(pArgDesc->dtargd_native, pszType, cchType + 1);
|
---|
[40878] | 612 | /** @todo mapping? */
|
---|
[40865] | 613 | pArgDesc->dtargd_ndx = uArg;
|
---|
[41130] | 614 | LOG_DTRACE(("%s: returns dtargd_native = %s\n", __FUNCTION__, pArgDesc->dtargd_native));
|
---|
[40675] | 615 | return;
|
---|
| 616 | }
|
---|
[40617] | 617 | }
|
---|
[40602] | 618 | }
|
---|
[40518] | 619 | }
|
---|
| 620 |
|
---|
| 621 |
|
---|
| 622 | /**
|
---|
[40857] | 623 | * @callback_method_impl{dtrace_pops_t,dtps_getargval}
|
---|
| 624 | *
|
---|
| 625 | *
|
---|
| 626 | * We just cook our own stuff here, using a stack marker for finding the
|
---|
| 627 | * required information. That's more reliable than subjecting oneself to the
|
---|
| 628 | * solaris bugs and 32-bit apple peculiarities.
|
---|
| 629 | *
|
---|
| 630 | *
|
---|
[40851] | 631 | * @remarks Solaris Bug
|
---|
[40857] | 632 | *
|
---|
| 633 | * dtrace_getarg on AMD64 has a different opinion about how to use the cFrames
|
---|
| 634 | * argument than dtrace_caller() and/or dtrace_getpcstack(), at least when the
|
---|
| 635 | * probe is fired by dtrace_probe() the way we do.
|
---|
| 636 | *
|
---|
[40851] | 637 | * Setting aframes to 1 when calling dtrace_probe_create gives me the right
|
---|
| 638 | * arguments, but the wrong 'caller'. Since I cannot do anything about
|
---|
| 639 | * 'caller', the only solution is this hack.
|
---|
[40857] | 640 | *
|
---|
[40851] | 641 | * Not sure why the Solaris guys hasn't seen this issue before, but maybe there
|
---|
[40857] | 642 | * isn't anyone using the default argument getter path for ring-0 dtrace_probe()
|
---|
| 643 | * calls, SDT surely isn't.
|
---|
| 644 | *
|
---|
[40851] | 645 | * @todo File a solaris bug on dtrace_probe() + dtrace_getarg().
|
---|
[40857] | 646 | *
|
---|
| 647 | *
|
---|
| 648 | * @remarks 32-bit XNU (Apple)
|
---|
| 649 | *
|
---|
| 650 | * The dtrace_probe arguments are 64-bit unsigned integers instead of uintptr_t,
|
---|
| 651 | * so we need to make an extra call.
|
---|
| 652 | *
|
---|
[40518] | 653 | */
|
---|
[41130] | 654 | static uint64_t vboxDtPOps_GetArgVal(void *pvProv, dtrace_id_t idProbe, void *pvProbe,
|
---|
| 655 | int iArg, int cFrames)
|
---|
[40518] | 656 | {
|
---|
[41130] | 657 | PSUPDRVVDTPROVIDERCORE pProv = (PSUPDRVVDTPROVIDERCORE)pvProv;
|
---|
[40865] | 658 | AssertPtrReturn(pProv, UINT64_MAX);
|
---|
[63510] | 659 | RT_NOREF(idProbe, cFrames);
|
---|
[41130] | 660 | LOG_DTRACE(("%s: %p / %p - %#x / %p iArg=%d cFrames=%u\n", __FUNCTION__, pProv, pProv->TracerData.DTrace.idProvider, idProbe, pvProbe, iArg, cFrames));
|
---|
[40851] | 661 | AssertReturn(iArg >= 5, UINT64_MAX);
|
---|
[41130] | 662 | if (pProv->TracerData.DTrace.fZombie)
|
---|
| 663 | return UINT64_MAX;
|
---|
[40756] | 664 |
|
---|
[41130] | 665 | uint32_t idxProbeLoc = (uint32_t)(uintptr_t)pvProbe;
|
---|
| 666 | PCVTGPROBELOC pProbeLocRO = (PVTGPROBELOC)&pProv->paProbeLocsRO[idxProbeLoc];
|
---|
| 667 | PCVTGDESCPROBE pProbeDesc = pProbeLocRO->pProbe;
|
---|
| 668 | PCVTGDESCARGLIST pArgList = (PCVTGDESCARGLIST)( (uintptr_t)pProv->pHdr
|
---|
| 669 | + pProv->pHdr->offArgLists
|
---|
| 670 | + pProbeDesc->offArgList);
|
---|
| 671 | AssertReturn(pProbeDesc->offArgList < pProv->pHdr->cbArgLists, UINT64_MAX);
|
---|
[40756] | 672 |
|
---|
[41130] | 673 | PVBDTSTACKDATA pData = vboxDtGetStackData();
|
---|
| 674 |
|
---|
[40892] | 675 | /*
|
---|
| 676 | * Get the stack data. This is a wee bit complicated on 32-bit systems
|
---|
| 677 | * since we want to support 64-bit integer arguments.
|
---|
| 678 | */
|
---|
[41130] | 679 | uint64_t u64Ret;
|
---|
| 680 | if (iArg >= 20)
|
---|
| 681 | u64Ret = UINT64_MAX;
|
---|
| 682 | else if (pData->enmCaller == kVBoxDtCaller_ProbeFireKernel)
|
---|
| 683 | {
|
---|
[40878] | 684 | #if ARCH_BITS == 64
|
---|
[41130] | 685 | u64Ret = pData->u.ProbeFireKernel.pauStackArgs[iArg - 5];
|
---|
[40878] | 686 | #else
|
---|
[41130] | 687 | if ( !pArgList->fHaveLargeArgs
|
---|
| 688 | || iArg >= pArgList->cArgs)
|
---|
| 689 | u64Ret = pData->u.ProbeFireKernel.pauStackArgs[iArg - 5];
|
---|
| 690 | else
|
---|
| 691 | {
|
---|
| 692 | /* Similar to what we did for mac in when calling dtrace_probe(). */
|
---|
| 693 | uint32_t offArg = 0;
|
---|
| 694 | for (int i = 5; i < iArg; i++)
|
---|
| 695 | if (VTG_TYPE_IS_LARGE(pArgList->aArgs[iArg].fType))
|
---|
| 696 | offArg++;
|
---|
| 697 | u64Ret = pData->u.ProbeFireKernel.pauStackArgs[iArg - 5 + offArg];
|
---|
[40892] | 698 | if (VTG_TYPE_IS_LARGE(pArgList->aArgs[iArg].fType))
|
---|
[41130] | 699 | u64Ret |= (uint64_t)pData->u.ProbeFireKernel.pauStackArgs[iArg - 5 + offArg + 1] << 32;
|
---|
| 700 | }
|
---|
| 701 | #endif
|
---|
[40878] | 702 | }
|
---|
[41130] | 703 | else if (pData->enmCaller == kVBoxDtCaller_ProbeFireUser)
|
---|
| 704 | {
|
---|
| 705 | int offArg = pData->u.ProbeFireUser.offArg;
|
---|
| 706 | PCSUPDRVTRACERUSRCTX pCtx = pData->u.ProbeFireUser.pCtx;
|
---|
| 707 | AssertPtrReturn(pCtx, UINT64_MAX);
|
---|
[40878] | 708 |
|
---|
[41130] | 709 | if (pCtx->cBits == 32)
|
---|
| 710 | {
|
---|
| 711 | if ( !pArgList->fHaveLargeArgs
|
---|
| 712 | || iArg >= pArgList->cArgs)
|
---|
| 713 | {
|
---|
| 714 | if (iArg + offArg < (int)RT_ELEMENTS(pCtx->u.X86.aArgs))
|
---|
[41164] | 715 | u64Ret = pCtx->u.X86.aArgs[iArg + offArg];
|
---|
[41130] | 716 | else
|
---|
| 717 | u64Ret = UINT64_MAX;
|
---|
| 718 | }
|
---|
| 719 | else
|
---|
| 720 | {
|
---|
[41354] | 721 | int i;
|
---|
| 722 | for (i = 5; i < iArg; i++)
|
---|
[41130] | 723 | if (VTG_TYPE_IS_LARGE(pArgList->aArgs[iArg].fType))
|
---|
| 724 | offArg++;
|
---|
| 725 | if (offArg + iArg < (int)RT_ELEMENTS(pCtx->u.X86.aArgs))
|
---|
| 726 | {
|
---|
| 727 | u64Ret = pCtx->u.X86.aArgs[iArg + offArg];
|
---|
| 728 | if ( VTG_TYPE_IS_LARGE(pArgList->aArgs[iArg].fType)
|
---|
| 729 | && offArg + iArg + 1 < (int)RT_ELEMENTS(pCtx->u.X86.aArgs))
|
---|
| 730 | u64Ret |= (uint64_t)pCtx->u.X86.aArgs[iArg + offArg + 1] << 32;
|
---|
| 731 | }
|
---|
| 732 | else
|
---|
| 733 | u64Ret = UINT64_MAX;
|
---|
| 734 | }
|
---|
| 735 | }
|
---|
| 736 | else
|
---|
| 737 | {
|
---|
| 738 | if (iArg + offArg < (int)RT_ELEMENTS(pCtx->u.Amd64.aArgs))
|
---|
| 739 | u64Ret = pCtx->u.Amd64.aArgs[iArg + offArg];
|
---|
| 740 | else
|
---|
| 741 | u64Ret = UINT64_MAX;
|
---|
| 742 | }
|
---|
| 743 | }
|
---|
| 744 | else
|
---|
| 745 | AssertFailedReturn(UINT64_MAX);
|
---|
| 746 |
|
---|
| 747 | LOG_DTRACE(("%s: returns %#llx\n", __FUNCTION__, u64Ret));
|
---|
[40878] | 748 | return u64Ret;
|
---|
[40518] | 749 | }
|
---|
| 750 |
|
---|
| 751 |
|
---|
| 752 | /**
|
---|
| 753 | * @callback_method_impl{dtrace_pops_t,dtps_destroy}
|
---|
| 754 | */
|
---|
[41130] | 755 | static void vboxDtPOps_Destroy(void *pvProv, dtrace_id_t idProbe, void *pvProbe)
|
---|
[40518] | 756 | {
|
---|
[40851] | 757 | PSUPDRVVDTPROVIDERCORE pProv = (PSUPDRVVDTPROVIDERCORE)pvProv;
|
---|
[40865] | 758 | AssertPtrReturnVoid(pProv);
|
---|
[41130] | 759 | LOG_DTRACE(("%s: %p / %p - %#x / %p\n", __FUNCTION__, pProv, pProv->TracerData.DTrace.idProvider, idProbe, pvProbe));
|
---|
[40865] | 760 | AssertReturnVoid(pProv->TracerData.DTrace.cProvidedProbes > 0);
|
---|
| 761 | AssertPtrReturnVoid(pProv->TracerData.DTrace.idProvider);
|
---|
| 762 |
|
---|
[40851] | 763 | if (!pProv->TracerData.DTrace.fZombie)
|
---|
[40675] | 764 | {
|
---|
[41130] | 765 | uint32_t idxProbeLoc = (uint32_t)(uintptr_t)pvProbe;
|
---|
| 766 | PCVTGPROBELOC pProbeLocRO = (PVTGPROBELOC)&pProv->paProbeLocsRO[idxProbeLoc];
|
---|
| 767 | uint32_t *pidProbe;
|
---|
| 768 | if (!pProv->fUmod)
|
---|
| 769 | {
|
---|
| 770 | pidProbe = (uint32_t *)&pProbeLocRO->idProbe;
|
---|
| 771 | Assert(!pProbeLocRO->fEnabled);
|
---|
| 772 | Assert(*pidProbe == idProbe);
|
---|
| 773 | }
|
---|
| 774 | else
|
---|
| 775 | {
|
---|
| 776 | pidProbe = &pProv->paR0ProbeLocs[idxProbeLoc].idProbe;
|
---|
| 777 | Assert(!pProv->paR0ProbeLocs[idxProbeLoc].fEnabled);
|
---|
| 778 | Assert(*pidProbe == idProbe); NOREF(idProbe);
|
---|
| 779 | }
|
---|
| 780 | *pidProbe = 0;
|
---|
[40675] | 781 | }
|
---|
[40851] | 782 | pProv->TracerData.DTrace.cProvidedProbes--;
|
---|
[40518] | 783 | }
|
---|
| 784 |
|
---|
[40851] | 785 |
|
---|
| 786 |
|
---|
| 787 | /**
|
---|
| 788 | * DTrace provider method table.
|
---|
| 789 | */
|
---|
| 790 | static const dtrace_pops_t g_vboxDtVtgProvOps =
|
---|
| 791 | {
|
---|
[41130] | 792 | /* .dtps_provide = */ vboxDtPOps_Provide,
|
---|
[40851] | 793 | /* .dtps_provide_module = */ NULL,
|
---|
[41130] | 794 | /* .dtps_enable = */ (FNPOPS_ENABLE *)vboxDtPOps_Enable,
|
---|
| 795 | /* .dtps_disable = */ vboxDtPOps_Disable,
|
---|
[40851] | 796 | /* .dtps_suspend = */ NULL,
|
---|
| 797 | /* .dtps_resume = */ NULL,
|
---|
[41130] | 798 | /* .dtps_getargdesc = */ vboxDtPOps_GetArgDesc,
|
---|
| 799 | /* .dtps_getargval = */ vboxDtPOps_GetArgVal,
|
---|
[40851] | 800 | /* .dtps_usermode = */ NULL,
|
---|
[41130] | 801 | /* .dtps_destroy = */ vboxDtPOps_Destroy
|
---|
[40851] | 802 | };
|
---|
| 803 |
|
---|
| 804 |
|
---|
| 805 |
|
---|
| 806 |
|
---|
| 807 | /*
|
---|
| 808 | *
|
---|
| 809 | * Support Driver Tracer Interface.
|
---|
| 810 | * Support Driver Tracer Interface.
|
---|
| 811 | * Support Driver Tracer Interface.
|
---|
| 812 | *
|
---|
| 813 | */
|
---|
| 814 |
|
---|
| 815 |
|
---|
| 816 |
|
---|
| 817 | /**
|
---|
[41130] | 818 | * interface_method_impl{SUPDRVTRACERREG,pfnProbeFireKernel}
|
---|
[40851] | 819 | */
|
---|
[41130] | 820 | static DECLCALLBACK(void) vboxDtTOps_ProbeFireKernel(struct VTGPROBELOC *pVtgProbeLoc, uintptr_t uArg0, uintptr_t uArg1, uintptr_t uArg2,
|
---|
| 821 | uintptr_t uArg3, uintptr_t uArg4)
|
---|
[40851] | 822 | {
|
---|
[40865] | 823 | AssertPtrReturnVoid(pVtgProbeLoc);
|
---|
[41130] | 824 | LOG_DTRACE(("%s: %p / %p\n", __FUNCTION__, pVtgProbeLoc, pVtgProbeLoc->idProbe));
|
---|
| 825 | AssertPtrReturnVoid(pVtgProbeLoc->pProbe);
|
---|
[40865] | 826 | AssertPtrReturnVoid(pVtgProbeLoc->pszFunction);
|
---|
| 827 |
|
---|
[64584] | 828 | SUPDRV_SAVE_EFL_AC();
|
---|
[41130] | 829 | VBDT_SETUP_STACK_DATA(kVBoxDtCaller_ProbeFireKernel);
|
---|
[40851] | 830 |
|
---|
[41130] | 831 | pStackData->u.ProbeFireKernel.pauStackArgs = &uArg4 + 1;
|
---|
[40892] | 832 |
|
---|
| 833 | #if defined(RT_OS_DARWIN) && ARCH_BITS == 32
|
---|
| 834 | /*
|
---|
| 835 | * Convert arguments from uintptr_t to uint64_t.
|
---|
| 836 | */
|
---|
[41144] | 837 | PVTGDESCPROBE pProbe = (PVTGDESCPROBE)((PVTGPROBELOC)pVtgProbeLoc)->pProbe;
|
---|
[40892] | 838 | AssertPtrReturnVoid(pProbe);
|
---|
| 839 | PVTGOBJHDR pVtgHdr = (PVTGOBJHDR)((uintptr_t)pProbe + pProbe->offObjHdr);
|
---|
| 840 | AssertPtrReturnVoid(pVtgHdr);
|
---|
[41130] | 841 | PVTGDESCARGLIST pArgList = (PVTGDESCARGLIST)((uintptr_t)pVtgHdr + pVtgHdr->offArgLists + pProbe->offArgList);
|
---|
[40892] | 842 | AssertPtrReturnVoid(pArgList);
|
---|
| 843 | if (!pArgList->fHaveLargeArgs)
|
---|
| 844 | dtrace_probe(pVtgProbeLoc->idProbe, uArg0, uArg1, uArg2, uArg3, uArg4);
|
---|
| 845 | else
|
---|
| 846 | {
|
---|
| 847 | uintptr_t *auSrcArgs = &uArg0;
|
---|
| 848 | uint32_t iSrcArg = 0;
|
---|
| 849 | uint32_t iDstArg = 0;
|
---|
| 850 | uint64_t au64DstArgs[5];
|
---|
| 851 |
|
---|
| 852 | while ( iDstArg < RT_ELEMENTS(au64DstArgs)
|
---|
| 853 | && iSrcArg < pArgList->cArgs)
|
---|
| 854 | {
|
---|
| 855 | au64DstArgs[iDstArg] = auSrcArgs[iSrcArg];
|
---|
| 856 | if (VTG_TYPE_IS_LARGE(pArgList->aArgs[iDstArg].fType))
|
---|
| 857 | au64DstArgs[iDstArg] |= (uint64_t)auSrcArgs[++iSrcArg] << 32;
|
---|
| 858 | iSrcArg++;
|
---|
| 859 | iDstArg++;
|
---|
| 860 | }
|
---|
| 861 | while (iDstArg < RT_ELEMENTS(au64DstArgs))
|
---|
| 862 | au64DstArgs[iDstArg++] = auSrcArgs[iSrcArg++];
|
---|
| 863 |
|
---|
[41144] | 864 | pStackData->u.ProbeFireKernel.pauStackArgs = &auSrcArgs[iSrcArg];
|
---|
[40892] | 865 | dtrace_probe(pVtgProbeLoc->idProbe, au64DstArgs[0], au64DstArgs[1], au64DstArgs[2], au64DstArgs[3], au64DstArgs[4]);
|
---|
| 866 | }
|
---|
| 867 | #else
|
---|
[40851] | 868 | dtrace_probe(pVtgProbeLoc->idProbe, uArg0, uArg1, uArg2, uArg3, uArg4);
|
---|
[40892] | 869 | #endif
|
---|
[40851] | 870 |
|
---|
[41130] | 871 | VBDT_CLEAR_STACK_DATA();
|
---|
[64584] | 872 | SUPDRV_RESTORE_EFL_AC();
|
---|
[41130] | 873 | LOG_DTRACE(("%s: returns\n", __FUNCTION__));
|
---|
[40851] | 874 | }
|
---|
| 875 |
|
---|
| 876 |
|
---|
| 877 | /**
|
---|
| 878 | * interface_method_impl{SUPDRVTRACERREG,pfnProbeFireUser}
|
---|
| 879 | */
|
---|
[41130] | 880 | static DECLCALLBACK(void) vboxDtTOps_ProbeFireUser(PCSUPDRVTRACERREG pThis, PSUPDRVSESSION pSession, PCSUPDRVTRACERUSRCTX pCtx,
|
---|
| 881 | PCVTGOBJHDR pVtgHdr, PCVTGPROBELOC pProbeLocRO)
|
---|
[40851] | 882 | {
|
---|
[63510] | 883 | RT_NOREF(pThis, pSession);
|
---|
[41130] | 884 | LOG_DTRACE(("%s: %p / %p\n", __FUNCTION__, pCtx, pCtx->idProbe));
|
---|
| 885 | AssertPtrReturnVoid(pProbeLocRO);
|
---|
| 886 | AssertPtrReturnVoid(pVtgHdr);
|
---|
| 887 |
|
---|
[64584] | 888 | SUPDRV_SAVE_EFL_AC();
|
---|
[41130] | 889 | VBDT_SETUP_STACK_DATA(kVBoxDtCaller_ProbeFireUser);
|
---|
| 890 |
|
---|
| 891 | if (pCtx->cBits == 32)
|
---|
| 892 | {
|
---|
| 893 | pStackData->u.ProbeFireUser.pCtx = pCtx;
|
---|
| 894 | pStackData->u.ProbeFireUser.offArg = 0;
|
---|
| 895 |
|
---|
| 896 | #if ARCH_BITS == 64 || defined(RT_OS_DARWIN)
|
---|
| 897 | /*
|
---|
| 898 | * Combine two 32-bit arguments into one 64-bit argument where needed.
|
---|
| 899 | */
|
---|
| 900 | PVTGDESCPROBE pProbeDesc = pProbeLocRO->pProbe;
|
---|
| 901 | AssertPtrReturnVoid(pProbeDesc);
|
---|
| 902 | PVTGDESCARGLIST pArgList = (PVTGDESCARGLIST)((uintptr_t)pVtgHdr + pVtgHdr->offArgLists + pProbeDesc->offArgList);
|
---|
| 903 | AssertPtrReturnVoid(pArgList);
|
---|
| 904 |
|
---|
| 905 | if (!pArgList->fHaveLargeArgs)
|
---|
| 906 | dtrace_probe(pCtx->idProbe,
|
---|
| 907 | pCtx->u.X86.aArgs[0],
|
---|
| 908 | pCtx->u.X86.aArgs[1],
|
---|
| 909 | pCtx->u.X86.aArgs[2],
|
---|
| 910 | pCtx->u.X86.aArgs[3],
|
---|
| 911 | pCtx->u.X86.aArgs[4]);
|
---|
| 912 | else
|
---|
| 913 | {
|
---|
| 914 | uint32_t const *auSrcArgs = &pCtx->u.X86.aArgs[0];
|
---|
| 915 | uint32_t iSrcArg = 0;
|
---|
| 916 | uint32_t iDstArg = 0;
|
---|
| 917 | uint64_t au64DstArgs[5];
|
---|
| 918 |
|
---|
| 919 | while ( iDstArg < RT_ELEMENTS(au64DstArgs)
|
---|
| 920 | && iSrcArg < pArgList->cArgs)
|
---|
| 921 | {
|
---|
| 922 | au64DstArgs[iDstArg] = auSrcArgs[iSrcArg];
|
---|
| 923 | if (VTG_TYPE_IS_LARGE(pArgList->aArgs[iDstArg].fType))
|
---|
| 924 | au64DstArgs[iDstArg] |= (uint64_t)auSrcArgs[++iSrcArg] << 32;
|
---|
| 925 | iSrcArg++;
|
---|
| 926 | iDstArg++;
|
---|
| 927 | }
|
---|
| 928 | while (iDstArg < RT_ELEMENTS(au64DstArgs))
|
---|
| 929 | au64DstArgs[iDstArg++] = auSrcArgs[iSrcArg++];
|
---|
| 930 |
|
---|
| 931 | pStackData->u.ProbeFireUser.offArg = iSrcArg - RT_ELEMENTS(au64DstArgs);
|
---|
| 932 | dtrace_probe(pCtx->idProbe, au64DstArgs[0], au64DstArgs[1], au64DstArgs[2], au64DstArgs[3], au64DstArgs[4]);
|
---|
| 933 | }
|
---|
| 934 | #else
|
---|
| 935 | dtrace_probe(pCtx->idProbe,
|
---|
| 936 | pCtx->u.X86.aArgs[0],
|
---|
| 937 | pCtx->u.X86.aArgs[1],
|
---|
| 938 | pCtx->u.X86.aArgs[2],
|
---|
| 939 | pCtx->u.X86.aArgs[3],
|
---|
| 940 | pCtx->u.X86.aArgs[4]);
|
---|
| 941 | #endif
|
---|
| 942 | }
|
---|
| 943 | else if (pCtx->cBits == 64)
|
---|
| 944 | {
|
---|
| 945 | pStackData->u.ProbeFireUser.pCtx = pCtx;
|
---|
| 946 | pStackData->u.ProbeFireUser.offArg = 0;
|
---|
| 947 | dtrace_probe(pCtx->idProbe,
|
---|
| 948 | pCtx->u.Amd64.aArgs[0],
|
---|
| 949 | pCtx->u.Amd64.aArgs[1],
|
---|
| 950 | pCtx->u.Amd64.aArgs[2],
|
---|
| 951 | pCtx->u.Amd64.aArgs[3],
|
---|
| 952 | pCtx->u.Amd64.aArgs[4]);
|
---|
| 953 | }
|
---|
| 954 | else
|
---|
| 955 | AssertFailed();
|
---|
| 956 |
|
---|
| 957 | VBDT_CLEAR_STACK_DATA();
|
---|
[64584] | 958 | SUPDRV_RESTORE_EFL_AC();
|
---|
[41130] | 959 | LOG_DTRACE(("%s: returns\n", __FUNCTION__));
|
---|
[40851] | 960 | }
|
---|
| 961 |
|
---|
| 962 |
|
---|
| 963 | /**
|
---|
| 964 | * interface_method_impl{SUPDRVTRACERREG,pfnTracerOpen}
|
---|
| 965 | */
|
---|
[41130] | 966 | static DECLCALLBACK(int) vboxDtTOps_TracerOpen(PCSUPDRVTRACERREG pThis, PSUPDRVSESSION pSession, uint32_t uCookie,
|
---|
| 967 | uintptr_t uArg, uintptr_t *puSessionData)
|
---|
[40851] | 968 | {
|
---|
| 969 | NOREF(pThis); NOREF(pSession); NOREF(uCookie); NOREF(uArg);
|
---|
| 970 | *puSessionData = 0;
|
---|
| 971 | return VERR_NOT_SUPPORTED;
|
---|
| 972 | }
|
---|
| 973 |
|
---|
| 974 |
|
---|
| 975 | /**
|
---|
| 976 | * interface_method_impl{SUPDRVTRACERREG,pfnTracerClose}
|
---|
| 977 | */
|
---|
[41130] | 978 | static DECLCALLBACK(int) vboxDtTOps_TracerIoCtl(PCSUPDRVTRACERREG pThis, PSUPDRVSESSION pSession, uintptr_t uSessionData,
|
---|
| 979 | uintptr_t uCmd, uintptr_t uArg, int32_t *piRetVal)
|
---|
[40851] | 980 | {
|
---|
| 981 | NOREF(pThis); NOREF(pSession); NOREF(uSessionData);
|
---|
| 982 | NOREF(uCmd); NOREF(uArg); NOREF(piRetVal);
|
---|
| 983 | return VERR_NOT_SUPPORTED;
|
---|
| 984 | }
|
---|
| 985 |
|
---|
| 986 |
|
---|
| 987 | /**
|
---|
| 988 | * interface_method_impl{SUPDRVTRACERREG,pfnTracerClose}
|
---|
| 989 | */
|
---|
[41130] | 990 | static DECLCALLBACK(void) vboxDtTOps_TracerClose(PCSUPDRVTRACERREG pThis, PSUPDRVSESSION pSession, uintptr_t uSessionData)
|
---|
[40851] | 991 | {
|
---|
| 992 | NOREF(pThis); NOREF(pSession); NOREF(uSessionData);
|
---|
| 993 | return;
|
---|
| 994 | }
|
---|
| 995 |
|
---|
| 996 |
|
---|
| 997 | /**
|
---|
| 998 | * interface_method_impl{SUPDRVTRACERREG,pfnProviderRegister}
|
---|
| 999 | */
|
---|
[41130] | 1000 | static DECLCALLBACK(int) vboxDtTOps_ProviderRegister(PCSUPDRVTRACERREG pThis, PSUPDRVVDTPROVIDERCORE pCore)
|
---|
[40851] | 1001 | {
|
---|
[63510] | 1002 | RT_NOREF(pThis);
|
---|
[41130] | 1003 | LOG_DTRACE(("%s: %p %s/%s\n", __FUNCTION__, pThis, pCore->pszModName, pCore->pszName));
|
---|
| 1004 | AssertReturn(pCore->TracerData.DTrace.idProvider == 0, VERR_INTERNAL_ERROR_3);
|
---|
[40851] | 1005 |
|
---|
| 1006 | PVTGDESCPROVIDER pDesc = pCore->pDesc;
|
---|
| 1007 | dtrace_pattr_t DtAttrs;
|
---|
| 1008 | vboxDtVtgConvAttr(&DtAttrs.dtpa_provider, &pDesc->AttrSelf);
|
---|
| 1009 | vboxDtVtgConvAttr(&DtAttrs.dtpa_mod, &pDesc->AttrModules);
|
---|
| 1010 | vboxDtVtgConvAttr(&DtAttrs.dtpa_func, &pDesc->AttrFunctions);
|
---|
| 1011 | vboxDtVtgConvAttr(&DtAttrs.dtpa_name, &pDesc->AttrNames);
|
---|
| 1012 | vboxDtVtgConvAttr(&DtAttrs.dtpa_args, &pDesc->AttrArguments);
|
---|
| 1013 |
|
---|
[40865] | 1014 | /* Note! DTrace may call us back before dtrace_register returns, so we
|
---|
| 1015 | have to point it to pCore->TracerData.DTrace.idProvider. */
|
---|
[41130] | 1016 | AssertCompile(sizeof(dtrace_provider_id_t) == sizeof(pCore->TracerData.DTrace.idProvider));
|
---|
[64584] | 1017 | SUPDRV_SAVE_EFL_AC();
|
---|
[40851] | 1018 | int rc = dtrace_register(pCore->pszName,
|
---|
| 1019 | &DtAttrs,
|
---|
| 1020 | DTRACE_PRIV_KERNEL,
|
---|
| 1021 | NULL /* cred */,
|
---|
| 1022 | &g_vboxDtVtgProvOps,
|
---|
| 1023 | pCore,
|
---|
[40865] | 1024 | &pCore->TracerData.DTrace.idProvider);
|
---|
[64584] | 1025 | SUPDRV_RESTORE_EFL_AC();
|
---|
[40851] | 1026 | if (!rc)
|
---|
| 1027 | {
|
---|
[41130] | 1028 | LOG_DTRACE(("%s: idProvider=%p\n", __FUNCTION__, pCore->TracerData.DTrace.idProvider));
|
---|
[40865] | 1029 | AssertPtr(pCore->TracerData.DTrace.idProvider);
|
---|
[40851] | 1030 | rc = VINF_SUCCESS;
|
---|
| 1031 | }
|
---|
| 1032 | else
|
---|
[40865] | 1033 | {
|
---|
[41130] | 1034 | pCore->TracerData.DTrace.idProvider = 0;
|
---|
[41357] | 1035 | rc = RTErrConvertFromErrno(FIX_UEK_RC(rc));
|
---|
[40865] | 1036 | }
|
---|
[40851] | 1037 |
|
---|
[41130] | 1038 | LOG_DTRACE(("%s: returns %Rrc\n", __FUNCTION__, rc));
|
---|
[40851] | 1039 | return rc;
|
---|
| 1040 | }
|
---|
| 1041 |
|
---|
| 1042 |
|
---|
| 1043 | /**
|
---|
| 1044 | * interface_method_impl{SUPDRVTRACERREG,pfnProviderDeregister}
|
---|
| 1045 | */
|
---|
[41130] | 1046 | static DECLCALLBACK(int) vboxDtTOps_ProviderDeregister(PCSUPDRVTRACERREG pThis, PSUPDRVVDTPROVIDERCORE pCore)
|
---|
[40851] | 1047 | {
|
---|
| 1048 | uintptr_t idProvider = pCore->TracerData.DTrace.idProvider;
|
---|
[63510] | 1049 | RT_NOREF(pThis);
|
---|
[41130] | 1050 | LOG_DTRACE(("%s: %p / %p\n", __FUNCTION__, pThis, idProvider));
|
---|
[40865] | 1051 | AssertPtrReturn(idProvider, VERR_INTERNAL_ERROR_3);
|
---|
[40851] | 1052 |
|
---|
[64584] | 1053 | SUPDRV_SAVE_EFL_AC();
|
---|
[40851] | 1054 | dtrace_invalidate(idProvider);
|
---|
| 1055 | int rc = dtrace_unregister(idProvider);
|
---|
[64584] | 1056 | SUPDRV_RESTORE_EFL_AC();
|
---|
[40851] | 1057 | if (!rc)
|
---|
| 1058 | {
|
---|
[41130] | 1059 | pCore->TracerData.DTrace.idProvider = 0;
|
---|
[40851] | 1060 | rc = VINF_SUCCESS;
|
---|
| 1061 | }
|
---|
| 1062 | else
|
---|
| 1063 | {
|
---|
[41357] | 1064 | AssertMsg(FIX_UEK_RC(rc) == EBUSY, ("%d\n", rc));
|
---|
[40851] | 1065 | pCore->TracerData.DTrace.fZombie = true;
|
---|
| 1066 | rc = VERR_TRY_AGAIN;
|
---|
| 1067 | }
|
---|
| 1068 |
|
---|
[41130] | 1069 | LOG_DTRACE(("%s: returns %Rrc\n", __FUNCTION__, rc));
|
---|
[40851] | 1070 | return rc;
|
---|
| 1071 | }
|
---|
| 1072 |
|
---|
| 1073 |
|
---|
| 1074 | /**
|
---|
| 1075 | * interface_method_impl{SUPDRVTRACERREG,pfnProviderDeregisterZombie}
|
---|
| 1076 | */
|
---|
[41130] | 1077 | static DECLCALLBACK(int) vboxDtTOps_ProviderDeregisterZombie(PCSUPDRVTRACERREG pThis, PSUPDRVVDTPROVIDERCORE pCore)
|
---|
[40851] | 1078 | {
|
---|
| 1079 | uintptr_t idProvider = pCore->TracerData.DTrace.idProvider;
|
---|
[63510] | 1080 | RT_NOREF(pThis);
|
---|
[41130] | 1081 | LOG_DTRACE(("%s: %p / %p\n", __FUNCTION__, pThis, idProvider));
|
---|
[40865] | 1082 | AssertPtrReturn(idProvider, VERR_INTERNAL_ERROR_3);
|
---|
[40851] | 1083 | Assert(pCore->TracerData.DTrace.fZombie);
|
---|
| 1084 |
|
---|
[64584] | 1085 | SUPDRV_SAVE_EFL_AC();
|
---|
[40851] | 1086 | int rc = dtrace_unregister(idProvider);
|
---|
[64584] | 1087 | SUPDRV_RESTORE_EFL_AC();
|
---|
[40851] | 1088 | if (!rc)
|
---|
| 1089 | {
|
---|
[41130] | 1090 | pCore->TracerData.DTrace.idProvider = 0;
|
---|
[40851] | 1091 | rc = VINF_SUCCESS;
|
---|
| 1092 | }
|
---|
| 1093 | else
|
---|
| 1094 | {
|
---|
[41357] | 1095 | AssertMsg(FIX_UEK_RC(rc) == EBUSY, ("%d\n", rc));
|
---|
[40851] | 1096 | rc = VERR_TRY_AGAIN;
|
---|
| 1097 | }
|
---|
| 1098 |
|
---|
[41130] | 1099 | LOG_DTRACE(("%s: returns %Rrc\n", __FUNCTION__, rc));
|
---|
[40851] | 1100 | return rc;
|
---|
| 1101 | }
|
---|
| 1102 |
|
---|
| 1103 |
|
---|
| 1104 |
|
---|
| 1105 | /**
|
---|
| 1106 | * The tracer registration record of the VBox DTrace implementation
|
---|
| 1107 | */
|
---|
[41130] | 1108 | static SUPDRVTRACERREG g_VBoxDTraceReg =
|
---|
[40851] | 1109 | {
|
---|
| 1110 | SUPDRVTRACERREG_MAGIC,
|
---|
| 1111 | SUPDRVTRACERREG_VERSION,
|
---|
[41130] | 1112 | vboxDtTOps_ProbeFireKernel,
|
---|
| 1113 | vboxDtTOps_ProbeFireUser,
|
---|
| 1114 | vboxDtTOps_TracerOpen,
|
---|
| 1115 | vboxDtTOps_TracerIoCtl,
|
---|
| 1116 | vboxDtTOps_TracerClose,
|
---|
| 1117 | vboxDtTOps_ProviderRegister,
|
---|
| 1118 | vboxDtTOps_ProviderDeregister,
|
---|
| 1119 | vboxDtTOps_ProviderDeregisterZombie,
|
---|
[40851] | 1120 | SUPDRVTRACERREG_MAGIC
|
---|
| 1121 | };
|
---|
| 1122 |
|
---|
| 1123 |
|
---|
| 1124 |
|
---|
| 1125 | /**
|
---|
| 1126 | * Module initialization code.
|
---|
| 1127 | */
|
---|
| 1128 | const SUPDRVTRACERREG * VBOXCALL supdrvDTraceInit(void)
|
---|
| 1129 | {
|
---|
[49395] | 1130 | #if defined(RT_OS_DARWIN) || defined(RT_OS_LINUX)
|
---|
[40851] | 1131 | /*
|
---|
| 1132 | * Resolve the kernel symbols we need.
|
---|
| 1133 | */
|
---|
[49395] | 1134 | # ifndef RT_OS_LINUX
|
---|
[40851] | 1135 | RTDBGKRNLINFO hKrnlInfo;
|
---|
| 1136 | int rc = RTR0DbgKrnlInfoOpen(&hKrnlInfo, 0);
|
---|
| 1137 | if (RT_FAILURE(rc))
|
---|
| 1138 | {
|
---|
| 1139 | SUPR0Printf("supdrvDTraceInit: RTR0DbgKrnlInfoOpen failed with rc=%d.\n", rc);
|
---|
| 1140 | return NULL;
|
---|
| 1141 | }
|
---|
[49395] | 1142 | # endif
|
---|
[40851] | 1143 |
|
---|
[49395] | 1144 | unsigned i;
|
---|
[85766] | 1145 | for (i = 0; i < RT_ELEMENTS(g_aDTraceFunctions); i++)
|
---|
[40851] | 1146 | {
|
---|
[49395] | 1147 | # ifndef RT_OS_LINUX
|
---|
[85766] | 1148 | rc = RTR0DbgKrnlInfoQuerySymbol(hKrnlInfo, NULL, g_aDTraceFunctions[i].pszName,
|
---|
| 1149 | (void **)g_aDTraceFunctions[i].ppfn);
|
---|
[40851] | 1150 | if (RT_FAILURE(rc))
|
---|
| 1151 | {
|
---|
[85766] | 1152 | SUPR0Printf("supdrvDTraceInit: Failed to resolved '%s' (rc=%Rrc, i=%u).\n", g_aDTraceFunctions[i].pszName, rc, i);
|
---|
[40851] | 1153 | break;
|
---|
| 1154 | }
|
---|
[85766] | 1155 | # else /* RT_OS_LINUX */
|
---|
| 1156 | uintptr_t ulAddr = (uintptr_t)__symbol_get(g_aDTraceFunctions[i].pszName);
|
---|
[49395] | 1157 | if (!ulAddr)
|
---|
| 1158 | {
|
---|
[85766] | 1159 | SUPR0Printf("supdrvDTraceInit: Failed to resolved '%s' (i=%u).\n", g_aDTraceFunctions[i].pszName, i);
|
---|
| 1160 | while (i-- > 0)
|
---|
| 1161 | {
|
---|
| 1162 | __symbol_put(g_aDTraceFunctions[i].pszName);
|
---|
| 1163 | *g_aDTraceFunctions[i].ppfn = NULL;
|
---|
| 1164 | }
|
---|
[49396] | 1165 | return NULL;
|
---|
[49395] | 1166 | }
|
---|
[85766] | 1167 | *g_aDTraceFunctions[i].ppfn = (PFNRT)ulAddr;
|
---|
| 1168 | # endif /* RT_OS_LINUX */
|
---|
[40851] | 1169 | }
|
---|
| 1170 |
|
---|
[49395] | 1171 | # ifndef RT_OS_LINUX
|
---|
[40851] | 1172 | RTR0DbgKrnlInfoRelease(hKrnlInfo);
|
---|
| 1173 | if (RT_FAILURE(rc))
|
---|
| 1174 | return NULL;
|
---|
[49395] | 1175 | # endif
|
---|
[40851] | 1176 | #endif
|
---|
| 1177 |
|
---|
[41130] | 1178 | return &g_VBoxDTraceReg;
|
---|
[40851] | 1179 | }
|
---|
| 1180 |
|
---|
[85766] | 1181 | /**
|
---|
| 1182 | * Module teardown code.
|
---|
| 1183 | */
|
---|
| 1184 | void VBOXCALL supdrvDTraceFini(void)
|
---|
[85767] | 1185 | {
|
---|
[85766] | 1186 | #ifdef RT_OS_LINUX
|
---|
| 1187 | /* Release the references. */
|
---|
| 1188 | unsigned i;
|
---|
| 1189 | for (i = 0; i < RT_ELEMENTS(g_aDTraceFunctions); i++)
|
---|
| 1190 | if (*g_aDTraceFunctions[i].ppfn)
|
---|
| 1191 | {
|
---|
| 1192 | __symbol_put(g_aDTraceFunctions[i].pszName);
|
---|
| 1193 | *g_aDTraceFunctions[i].ppfn = NULL;
|
---|
| 1194 | }
|
---|
| 1195 | #endif
|
---|
| 1196 | }
|
---|
| 1197 |
|
---|
[40975] | 1198 | #ifndef VBOX_WITH_NATIVE_DTRACE
|
---|
| 1199 | # error "VBOX_WITH_NATIVE_DTRACE is not defined as it should"
|
---|
[40851] | 1200 | #endif
|
---|
| 1201 |
|
---|