VBoxTpG.cpp revision 81614fc60e096e714022d10d38b70a36b9b21d48
828c58761d90445b8b9d20a82d85dc1479317f71Tripp/* $Id$ */
a89ad754cce3cfc8aee71760e10217b54020360dTripp/** @file
a89ad754cce3cfc8aee71760e10217b54020360dTripp * VBox Build Tool - VBox Tracepoint Generator.
828c58761d90445b8b9d20a82d85dc1479317f71Tripp */
828c58761d90445b8b9d20a82d85dc1479317f71Tripp
828c58761d90445b8b9d20a82d85dc1479317f71Tripp/*
828c58761d90445b8b9d20a82d85dc1479317f71Tripp * Copyright (C) 2012 Oracle Corporation
828c58761d90445b8b9d20a82d85dc1479317f71Tripp *
828c58761d90445b8b9d20a82d85dc1479317f71Tripp * This file is part of VirtualBox Open Source Edition (OSE), as
828c58761d90445b8b9d20a82d85dc1479317f71Tripp * available from http://www.virtualbox.org. This file is free software;
828c58761d90445b8b9d20a82d85dc1479317f71Tripp * you can redistribute it and/or modify it under the terms of the GNU
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * General Public License (GPL) as published by the Free Software
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Foundation, in version 2 as it comes in the "COPYING" file of the
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/*******************************************************************************
828c58761d90445b8b9d20a82d85dc1479317f71Tripp* Header Files *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp*******************************************************************************/
828c58761d90445b8b9d20a82d85dc1479317f71Tripp#include <VBox/VBoxTpG.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
828c58761d90445b8b9d20a82d85dc1479317f71Tripp#include <iprt/alloca.h>
828c58761d90445b8b9d20a82d85dc1479317f71Tripp#include <iprt/assert.h>
828c58761d90445b8b9d20a82d85dc1479317f71Tripp#include <iprt/ctype.h>
828c58761d90445b8b9d20a82d85dc1479317f71Tripp#include <iprt/env.h>
828c58761d90445b8b9d20a82d85dc1479317f71Tripp#include <iprt/err.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/file.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/getopt.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/initterm.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/list.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/mem.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/message.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/path.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/process.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/stream.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include <iprt/string.h>
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#include "scmstream.h"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/*******************************************************************************
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp* Structures and Typedefs *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp*******************************************************************************/
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef struct VTGATTRS
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp kVTGStability enmCode;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp kVTGStability enmData;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp kVTGClass enmDataDep;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp} VTGATTRS;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef VTGATTRS *PVTGATTRS;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef struct VTGARG
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTLISTNODE ListEntry;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp char *pszName;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp const char *pszType;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t fType;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp} VTGARG;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef VTGARG *PVTGARG;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef struct VTGPROBE
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTLISTNODE ListEntry;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp char *pszMangledName;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp const char *pszUnmangledName;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTLISTANCHOR ArgHead;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t cArgs;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp bool fHaveLargeArgs;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t offArgList;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t iProbe;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp} VTGPROBE;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef VTGPROBE *PVTGPROBE;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef struct VTGPROVIDER
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTLISTNODE ListEntry;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp const char *pszName;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint16_t iFirstProbe;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint16_t cProbes;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp VTGATTRS AttrSelf;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp VTGATTRS AttrModules;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp VTGATTRS AttrFunctions;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp VTGATTRS AttrName;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp VTGATTRS AttrArguments;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTLISTANCHOR ProbeHead;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp} VTGPROVIDER;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef VTGPROVIDER *PVTGPROVIDER;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * A string table string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef struct VTGSTRING
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /** The string space core. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTSTRSPACECORE Core;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /** The string table offset. */
9eaaa502227248d304ac9170902697d02158c1d9Tripp uint32_t offStrTab;
9eaaa502227248d304ac9170902697d02158c1d9Tripp /** The actual string. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp char szString[1];
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp} VTGSTRING;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripptypedef VTGSTRING *PVTGSTRING;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/*******************************************************************************
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp* Global Variables *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp*******************************************************************************/
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/** The string space organizing the string table strings. Each node is a VTGSTRING. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic RTSTRSPACE g_StrSpace = NULL;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/** Used by the string table enumerator to set VTGSTRING::offStrTab. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic uint32_t g_offStrTab;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/** List of providers created by the parser. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic RTLISTANCHOR g_ProviderHead;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/** The number of type errors. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic uint32_t g_cTypeErrors = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/** @name Options
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @{ */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic enum
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp kVBoxTpGAction_Nothing,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp kVBoxTpGAction_GenerateHeader,
9eaaa502227248d304ac9170902697d02158c1d9Tripp kVBoxTpGAction_GenerateObject
9eaaa502227248d304ac9170902697d02158c1d9Tripp} g_enmAction = kVBoxTpGAction_Nothing;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic uint32_t g_cBits = ARCH_BITS;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic bool g_fApplyCpp = false;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic uint32_t g_cVerbosity = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszOutput = NULL;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszScript = NULL;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszTempAsm = NULL;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#ifdef RT_OS_DARWIN
9eaaa502227248d304ac9170902697d02158c1d9Trippstatic const char *g_pszAssembler = "yasm";
9eaaa502227248d304ac9170902697d02158c1d9Trippstatic const char *g_pszAssemblerFmtOpt = "-f";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal32[] = "macho32";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal64[] = "macho64";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_DARWIN";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#elif defined(RT_OS_OS2)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *pszAssembler = "nasm.exe";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *pszAssemblerFmtOpt = "-f";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal32[] = "obj";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal64[] = "elf64";
9eaaa502227248d304ac9170902697d02158c1d9Trippstatic const char g_szAssemblerOsDef[] = "RT_OS_OS2";
9eaaa502227248d304ac9170902697d02158c1d9Tripp#elif defined(RT_OS_WINDOWS)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssembler = "yasm.exe";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerFmtOpt = "-f";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal32[] = "win32";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal64[] = "win64";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_WINDOWS";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#else
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssembler = "yasm";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerFmtOpt = "-f";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal32[] = "elf32";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerFmtVal64[] = "elf64";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# ifdef RT_OS_FREEBSD
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_FREEBSD";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# elif defined(RT_OS_NETBSD)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_NETBSD";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# elif defined(RT_OS_OPENBSD)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_OPENBSD";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# elif defined(RT_OS_LINUX)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_LINUX";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# elif defined(RT_OS_SOLARIS)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char g_szAssemblerOsDef[] = "RT_OS_SOLARIS";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# else
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# error "Port me!"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp# endif
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp#endif
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerFmtVal = RT_CONCAT(g_szAssemblerFmtVal, ARCH_BITS);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerDefOpt = "-D";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerIncOpt = "-I";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic char g_szAssemblerIncVal[RTPATH_MAX];
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerIncVal = __FILE__ "/../../../include/";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszAssemblerOutputOpt = "-o";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic unsigned g_cAssemblerOptions = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_apszAssemblerOptions[32];
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *g_pszProbeFnName = "SUPR0TracerFireProbe";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic bool g_fProbeFnImported = true;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic bool g_fPic = false;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/** @} */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Inserts a string into the string table, reusing any matching existing string
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * if possible.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns Read only string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pch The string to insert (need not be terminated).
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param cch The length of the string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic const char *strtabInsertN(const char *pch, size_t cch)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGSTRING pStr = (PVTGSTRING)RTStrSpaceGetN(&g_StrSpace, pch, cch);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (pStr)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return pStr->szString;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Create a new entry.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pStr = (PVTGSTRING)RTMemAlloc(RT_OFFSETOF(VTGSTRING, szString[cch + 1]));
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (!pStr)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return NULL;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pStr->Core.pszString = pStr->szString;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp memcpy(pStr->szString, pch, cch);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pStr->szString[cch] = '\0';
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pStr->offStrTab = UINT32_MAX;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp bool fRc = RTStrSpaceInsert(&g_StrSpace, &pStr->Core);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp Assert(fRc); NOREF(fRc);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return pStr->szString;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Retrieves the string table offset of the given string table string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns String table offset.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszStrTabString The string table string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic uint32_t strtabGetOff(const char *pszStrTabString)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGSTRING pStr = RT_FROM_MEMBER(pszStrTabString, VTGSTRING, szString[0]);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp Assert(pStr->Core.pszString == pszStrTabString);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return pStr->offStrTab;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Invokes the assembler.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns Exit code.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszOutput The output file.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszTempAsm The source file.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic RTEXITCODE generateInvokeAssembler(const char *pszOutput, const char *pszTempAsm)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp const char *apszArgs[64];
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp unsigned iArg = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssembler;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerFmtOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerFmtVal;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerDefOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (!strcmp(g_pszAssemblerFmtVal, "macho32") || !strcmp(g_pszAssemblerFmtVal, "macho64"))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "ASM_FORMAT_MACHO";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else if (!strcmp(g_pszAssemblerFmtVal, "obj") || !strcmp(g_pszAssemblerFmtVal, "omf"))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "ASM_FORMAT_OMF";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else if ( !strcmp(g_pszAssemblerFmtVal, "win32")
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp || !strcmp(g_pszAssemblerFmtVal, "win64")
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp || !strcmp(g_pszAssemblerFmtVal, "pe32")
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp || !strcmp(g_pszAssemblerFmtVal, "pe64")
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp || !strcmp(g_pszAssemblerFmtVal, "pe") )
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "ASM_FORMAT_PE";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else if ( !strcmp(g_pszAssemblerFmtVal, "elf32")
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp || !strcmp(g_pszAssemblerFmtVal, "elf64")
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp || !strcmp(g_pszAssemblerFmtVal, "elf"))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "ASM_FORMAT_ELF";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "Unknown assembler format '%s'", g_pszAssemblerFmtVal);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerDefOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cBits == 32)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "ARCH_BITS=32";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "ARCH_BITS=64";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerDefOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cBits == 32)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "RT_ARCH_X86";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = "RT_ARCH_AMD64";
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_szAssemblerOsDef[0])
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerDefOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_szAssemblerOsDef;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerIncOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerIncVal;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_pszAssemblerOutputOpt;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = pszOutput;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp for (unsigned i = 0; i < g_cAssemblerOptions; i++)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = g_apszAssemblerOptions[i];
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg++] = pszTempAsm;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp apszArgs[iArg] = NULL;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cVerbosity > 1)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTMsgInfo("Starting assmbler '%s' with arguments:\n", g_pszAssembler);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp for (unsigned i = 0; i < iArg; i++)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTMsgInfo(" #%02u: '%s'\n", i, apszArgs[i]);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTPROCESS hProc;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp int rc = RTProcCreate(apszArgs[0], apszArgs, RTENV_DEFAULT, RTPROC_FLAGS_SEARCH_PATH, &hProc);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (RT_FAILURE(rc))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to start '%s' (assembler): %Rrc", apszArgs[0], rc);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTPROCSTATUS Status;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp rc = RTProcWait(hProc, RTPROCWAIT_FLAGS_BLOCK, &Status);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (RT_FAILURE(rc))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTProcTerminate(hProc);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTProcWait failed: %Rrc", rc);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (Status.enmReason == RTPROCEXITREASON_SIGNAL)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "The assembler failed: signal %d", Status.iStatus);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (Status.enmReason != RTPROCEXITREASON_NORMAL)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "The assembler failed: abend");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (Status.iStatus != 0)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit((RTEXITCODE)Status.iStatus, "The assembler failed: exit code %d", Status.iStatus);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTEXITCODE_SUCCESS;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Worker that does the boring bits when generating a file.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns Exit code.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszOutput The name of the output file.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszWhat What kind of file it is.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pfnGenerator The callback function that provides the contents
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * of the file.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic RTEXITCODE generateFile(const char *pszOutput, const char *pszWhat,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTEXITCODE (*pfnGenerator)(PSCMSTREAM))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp SCMSTREAM Strm;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp int rc = ScmStreamInitForWriting(&Strm, NULL);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (RT_FAILURE(rc))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "ScmStreamInitForWriting returned %Rrc when generating the %s file",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp rc, pszWhat);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTEXITCODE rcExit = pfnGenerator(&Strm);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (RT_FAILURE(ScmStreamGetStatus(&Strm)))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Stream error %Rrc generating the %s file",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamGetStatus(&Strm), pszWhat);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (rcExit == RTEXITCODE_SUCCESS)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp rc = ScmStreamWriteToFile(&Strm, "%s", pszOutput);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (RT_FAILURE(rc))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "ScmStreamWriteToFile returned %Rrc when writing '%s' (%s)",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp rc, pszOutput, pszWhat);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (rcExit == RTEXITCODE_SUCCESS)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cVerbosity > 0)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTMsgInfo("Successfully generated '%s'.", pszOutput);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cVerbosity > 1)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTMsgInfo("================ %s - start ================", pszWhat);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamRewindForReading(&Strm);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp const char *pszLine;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp size_t cchLine;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp SCMEOL enmEol;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp while ((pszLine = ScmStreamGetLine(&Strm, &cchLine, &enmEol)) != NULL)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTPrintf("%.*s\n", cchLine, pszLine);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTMsgInfo("================ %s - end ================", pszWhat);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamDelete(&Strm);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return rcExit;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Formats a string and writes it to the SCM stream.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns The number of bytes written (>= 0). Negative value are IPRT error
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * status codes.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pStream The stream to write to.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszFormat The format string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param va The arguments to format.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic ssize_t ScmStreamPrintfV(PSCMSTREAM pStream, const char *pszFormat, va_list va)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp char *psz;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ssize_t cch = RTStrAPrintfV(&psz, pszFormat, va);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (cch)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp int rc = ScmStreamWrite(pStream, psz, cch);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTStrFree(psz);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (RT_FAILURE(rc))
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp cch = rc;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return cch;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Formats a string and writes it to the SCM stream.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns The number of bytes written (>= 0). Negative value are IPRT error
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * status codes.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pStream The stream to write to.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pszFormat The format string.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param ... The arguments to format.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic ssize_t ScmStreamPrintf(PSCMSTREAM pStream, const char *pszFormat, ...)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp va_list va;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp va_start(va, pszFormat);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ssize_t cch = ScmStreamPrintfV(pStream, pszFormat, va);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp va_end(va);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return cch;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @callback_method_impl{FNRTSTRSPACECALLBACK, Writes the string table strings.}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic DECLCALLBACK(int) generateAssemblyStrTabCallback(PRTSTRSPACECORE pStr, void *pvUser)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGSTRING pVtgStr = (PVTGSTRING)pStr;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PSCMSTREAM pStrm = (PSCMSTREAM)pvUser;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pVtgStr->offStrTab = g_offStrTab;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp g_offStrTab += (uint32_t)pVtgStr->Core.cchString + 1;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db '%s', 0 ; off=%u len=%zu\n",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pVtgStr->szString, pVtgStr->offStrTab, pVtgStr->Core.cchString);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return VINF_SUCCESS;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp}
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp/**
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Generate assembly source that can be turned into an object file.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * (This is a generateFile callback.)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp *
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @returns Exit code.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * @param pStrm The output stream.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTrippstatic RTEXITCODE generateAssembly(PSCMSTREAM pStrm)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp{
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGPROVIDER pProvider;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGPROBE pProbe;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGARG pArg;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cVerbosity > 0)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTMsgInfo("Generating assembly code...");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Write the file header.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; $Id$ \n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";; @file\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; Automatically generated from %s. Do NOT edit!\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%include \"iprt/asmdefs.mac\"\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; We put all the data in a dedicated section / segment.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; In order to find the probe location specifiers, we do the necessary\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; trickery here, ASSUMING that this object comes in first in the link\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; editing process.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%ifdef ASM_FORMAT_OMF\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%macro VTG_GLOBAL 2\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " global NAME(%%1)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " NAME(%%1):\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%endmacro\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " segment VTG.Obj public CLASS=DATA align=4096 use32\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%elifdef ASM_FORMAT_MACHO\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%macro VTG_GLOBAL 2\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " global NAME(%%1)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " NAME(%%1):\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%endmacro\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section __VTG __VTGObj align=64]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%elifdef ASM_FORMAT_PE\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%macro VTG_GLOBAL 2\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " global NAME(%%1)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " NAME(%%1):\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%endmacro\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section VTGPrLc.Begin data align=64]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*" times 16 db 0xcc\n"*/
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGPrLc, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section VTGPrLc.Data data align=4]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section VTGPrLc.End data align=4]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGPrLc_End, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*" times 16 db 0xcc\n"*/
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section VTGObj data align=32]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%elifdef ASM_FORMAT_ELF\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%macro VTG_GLOBAL 2\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " global NAME(%%1):%%2 hidden\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " NAME(%%1):\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%endmacro\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section .VTGPrLc.Begin progbits alloc noexec write align=4096]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGPrLc, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section .VTGPrLc progbits alloc noexec write align=1]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section .VTGPrLc.End progbits alloc noexec write align=1]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGPrLc_End, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " [section .VTGData progbits alloc noexec write align=4096]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%else\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " %%error \"ASM_FORMAT_XXX is not defined\"\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%endif\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_VTGObjHeader, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " ;0 1 2 3\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " ;012345678901234567890123456789012\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db 'VTG Object Header v1.4', 0, 0\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd %u\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd 0\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGProviders)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGProviders_End) - NAME(g_aVTGProviders)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGProbes)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGProbes_End) - NAME(g_aVTGProbes)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_afVTGProbeEnabled)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_afVTGProbeEnabled_End) - NAME(g_afVTGProbeEnabled)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_achVTGStringTable)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_achVTGStringTable_End) - NAME(g_achVTGStringTable)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGArgLists)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGArgLists_End) - NAME(g_aVTGArgLists)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%ifdef ASM_FORMAT_MACHO ; Apple has a real decent linker!\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "extern section$start$__VTG$__VTGPrLc\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF section$start$__VTG$__VTGPrLc\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "extern section$end$__VTG$__VTGPrLc\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF section$end$__VTG$__VTGPrLc\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%else\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGPrLc)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF NAME(g_aVTGPrLc_End) ; cross section/segment size not possible\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "%%endif\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF 0\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF 0\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF 0\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " RTCCPTR_DEF 0\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp g_pszScript, g_cBits);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Declare the probe enable flags.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; Probe enabled flags. Since these will be accessed all the time\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; they are placed together and early in the section to get some more\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; cache and TLB hits when the probes are disabled.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_afVTGProbeEnabled, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp );
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t cProbes = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&g_ProviderHead, pProvider, VTGPROVIDER, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProvider->ProbeHead, pProbe, VTGPROBE, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_fVTGProbeEnabled_%s_%s, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db 0\n",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->pszName, pProbe->pszMangledName);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp cProbes++;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm, "VTG_GLOBAL g_afVTGProbeEnabled_End, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (cProbes >= _32K)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp return RTMsgErrorExit(RTEXITCODE_FAILURE, "Too many probes: %u (max %u)", cProbes, _32K - 1);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Dump the string table before we start using the strings.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; The string table.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_achVTGStringTable, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp g_offStrTab = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTStrSpaceEnumerate(&g_StrSpace, generateAssemblyStrTabCallback, pStrm);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_achVTGStringTable_End, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Write out the argument lists before we use them.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; The argument lists.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGArgLists, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t off = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&g_ProviderHead, pProvider, VTGPROVIDER, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProvider->ProbeHead, pProbe, VTGPROBE, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (pProbe->offArgList != UINT32_MAX)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp continue;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /* Write it. */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProbe->offArgList = off;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " ; off=%u\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %2u ; Argument count\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %u ; fHaveLargeArgs\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db 0, 0 ; Reserved\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp , off, pProbe->cArgs, (int)pProbe->fHaveLargeArgs);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp off += 4;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd %8u ; type '%s' (name '%s')\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd 0%08xh ; type flags\n",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp strtabGetOff(pArg->pszType), pArg->pszType, pArg->pszName,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pArg->fType);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp off += 8;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /* Look for matching argument lists (lazy bird walks the whole list). */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGPROVIDER pProv2;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&g_ProviderHead, pProv2, VTGPROVIDER, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGPROBE pProbe2;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProvider->ProbeHead, pProbe2, VTGPROBE, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (pProbe2->offArgList != UINT32_MAX)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp continue;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (pProbe2->cArgs != pProbe->cArgs)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp continue;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp PVTGARG pArg2;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pArg = RTListNodeGetNext(&pProbe->ArgHead, VTGARG, ListEntry);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pArg2 = RTListNodeGetNext(&pProbe2->ArgHead, VTGARG, ListEntry);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp int32_t cArgs = pProbe->cArgs;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp while ( cArgs-- > 0
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp && pArg2->pszType == pArg->pszType
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp && pArg2->fType == pArg->fType)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pArg = RTListNodeGetNext(&pArg->ListEntry, VTGARG, ListEntry);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pArg2 = RTListNodeGetNext(&pArg2->ListEntry, VTGARG, ListEntry);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (cArgs >= 0)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp continue;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProbe2->offArgList = pProbe->offArgList;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGArgLists_End, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Probe definitions.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; Prob definitions.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGProbes, data\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t iProvider = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp uint32_t iProbe = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&g_ProviderHead, pProvider, VTGPROVIDER, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->iFirstProbe = iProbe;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProvider->ProbeHead, pProbe, VTGPROBE, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_VTGProbeData_%s_%s, data ; idx=#%4u\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd %6u ; name\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd %6u ; Argument list offset\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dw NAME(g_fVTGProbeEnabled_%s_%s) - NAME(g_afVTGProbeEnabled)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dw %6u ; provider index\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd NAME(g_VTGObjHeader) - NAME(g_VTGProbeData_%s_%s) ; offset to the object header\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd 0 ; for the application\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd 0 ; for the application\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->pszName, pProbe->pszMangledName, iProbe,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp strtabGetOff(pProbe->pszUnmangledName),
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProbe->offArgList,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->pszName, pProbe->pszMangledName,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp iProvider,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->pszName, pProbe->pszMangledName
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp );
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProbe->iProbe = iProbe;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp iProbe++;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->cProbes = iProbe - pProvider->iFirstProbe;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp iProvider++;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm, "VTG_GLOBAL g_aVTGProbes_End, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * The providers data.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; Provider data.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL g_aVTGProviders, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp iProvider = 0;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&g_ProviderHead, pProvider, VTGPROVIDER, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " ; idx=#%4u - %s\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dd %6u ; name\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dw %6u ; index of first probe\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dw %6u ; count of probes\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %d, %d, %d ; AttrSelf\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %d, %d, %d ; AttrModules\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %d, %d, %d ; AttrFunctions\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %d, %d, %d ; AttrName\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db %d, %d, %d ; AttrArguments\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " db 0 ; reserved\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp iProvider, pProvider->pszName,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp strtabGetOff(pProvider->pszName),
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->iFirstProbe,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->cProbes,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->AttrSelf.enmCode, pProvider->AttrSelf.enmData, pProvider->AttrSelf.enmDataDep,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->AttrModules.enmCode, pProvider->AttrModules.enmData, pProvider->AttrModules.enmDataDep,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->AttrFunctions.enmCode, pProvider->AttrFunctions.enmData, pProvider->AttrFunctions.enmDataDep,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->AttrName.enmCode, pProvider->AttrName.enmData, pProvider->AttrName.enmDataDep,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->AttrArguments.enmCode, pProvider->AttrArguments.enmData, pProvider->AttrArguments.enmDataDep);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp iProvider++;
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm, "VTG_GLOBAL g_aVTGProviders_End, data\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Emit code for the stub functions.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp bool const fWin64 = g_cBits == 64 && (!strcmp(g_pszAssemblerFmtVal, "win64") || !strcmp(g_pszAssemblerFmtVal, "pe64"));
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp bool const fMachO64 = g_cBits == 64 && !strcmp(g_pszAssemblerFmtVal, "macho64");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp bool const fMachO32 = g_cBits == 32 && !strcmp(g_pszAssemblerFmtVal, "macho32");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "; Prob stubs.\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ";\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "BEGINCODE\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "extern %sNAME(%s)\n",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp g_fProbeFnImported ? "IMP" : "",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp g_pszProbeFnName);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (fMachO64 && g_fProbeFnImported && !g_fPic)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "g_pfnVtgProbeFn:\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " dq NAME(%s)\n",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp g_pszProbeFnName);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&g_ProviderHead, pProvider, VTGPROVIDER, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProvider->ProbeHead, pProbe, VTGPROBE, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp "VTG_GLOBAL VTGProbeStub_%s_%s, function; (VBOXTPGPROBELOC pVTGProbeLoc",
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp pProvider->pszName, pProbe->pszMangledName);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp {
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm, ", %s %s", pArg->pszType, pArg->pszName);
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp }
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ");\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Check if the probe in question is enabled.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cBits == 32)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " mov eax, [esp + 4]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " test byte [eax+3], 0x80 ; fEnabled == true?\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jz .return ; jump on false\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else if (fWin64)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " test byte [rcx+3], 0x80 ; fEnabled == true?\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jz .return ; jump on false\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp else
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm,
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " test byte [rdi+3], 0x80 ; fEnabled == true?\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jz .return ; jump on false\n");
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp /*
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp * Jump to the fire-probe function.
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp */
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp if (g_cBits == 32)
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ScmStreamPrintf(pStrm, g_fPic ?
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " call .mov_ecx_eip_plus_5\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ".got_eip:\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " add ecx, _GLOBAL_OFFSET_TABLE + ($$ - .got_eip) wrt ..gotpc\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " mov ecx, [%s@GOT + ecx]\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jmp ecx\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp ".mov_ecx_eip_plus_5:\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " pop ecx\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jmp ecx\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp : g_fProbeFnImported ?
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " mov ecx, IMP2(%s)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jmp ecx\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp :
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp " jmp NAME(%s)\n"
09688ec5ffb8b9cf9883a770e2f9ebd60b28888dTripp , g_pszProbeFnName);
else if (fWin64)
ScmStreamPrintf(pStrm, g_fProbeFnImported ?
" mov rax, IMP2(%s)\n"
" jmp rax\n"
:
" jmp NAME(%s)\n"
, g_pszProbeFnName);
else if (fMachO64 && g_fProbeFnImported)
ScmStreamPrintf(pStrm,
" jmp [g_pfnVtgProbeFn wrt rip]\n");
else
ScmStreamPrintf(pStrm, g_fPic ?
" jmp [rel %s wrt ..got]\n"
: g_fProbeFnImported ?
" lea rax, [IMP2(%s)]\n"
" jmp rax\n"
:
" jmp NAME(%s)\n"
, g_pszProbeFnName);
ScmStreamPrintf(pStrm,
".return:\n"
" ret ; The probe was disabled, return\n"
"\n");
}
}
return RTEXITCODE_SUCCESS;
}
static RTEXITCODE generateObject(const char *pszOutput, const char *pszTempAsm)
{
if (!pszTempAsm)
{
size_t cch = strlen(pszOutput);
char *psz = (char *)alloca(cch + sizeof(".asm"));
memcpy(psz, pszOutput, cch);
memcpy(psz + cch, ".asm", sizeof(".asm"));
pszTempAsm = psz;
}
RTEXITCODE rcExit = generateFile(pszTempAsm, "assembly", generateAssembly);
if (rcExit == RTEXITCODE_SUCCESS)
rcExit = generateInvokeAssembler(pszOutput, pszTempAsm);
RTFileDelete(pszTempAsm);
return rcExit;
}
static RTEXITCODE generateProbeDefineName(char *pszBuf, size_t cbBuf, const char *pszProvider, const char *pszProbe)
{
size_t cbMax = strlen(pszProvider) + 1 + strlen(pszProbe) + 1;
if (cbMax > cbBuf || cbMax > 80)
return RTMsgErrorExit(RTEXITCODE_FAILURE, "Probe '%s' in provider '%s' ends up with a too long defined\n", pszProbe, pszProvider);
while (*pszProvider)
*pszBuf++ = RT_C_TO_UPPER(*pszProvider++);
*pszBuf++ = '_';
while (*pszProbe)
{
if (pszProbe[0] == '_' && pszProbe[1] == '_')
pszProbe++;
*pszBuf++ = RT_C_TO_UPPER(*pszProbe++);
}
*pszBuf = '\0';
return RTEXITCODE_SUCCESS;
}
static RTEXITCODE generateHeaderInner(PSCMSTREAM pStrm)
{
/*
* Calc the double inclusion blocker define and then write the file header.
*/
char szTmp[4096];
const char *pszName = RTPathFilename(g_pszScript);
size_t cchName = strlen(pszName);
if (cchName >= sizeof(szTmp) - 64)
return RTMsgErrorExit(RTEXITCODE_FAILURE, "File name is too long '%s'", pszName);
szTmp[0] = '_';
szTmp[1] = '_';
szTmp[2] = '_';
memcpy(&szTmp[3], pszName, cchName);
szTmp[3 + cchName + 0] = '_';
szTmp[3 + cchName + 1] = '_';
szTmp[3 + cchName + 2] = '_';
szTmp[3 + cchName + 3] = '\0';
char *psz = &szTmp[3];
while (*psz)
{
if (!RT_C_IS_ALNUM(*psz) && *psz != '_')
*psz = '_';
psz++;
}
ScmStreamPrintf(pStrm,
"/* $Id$ */\n"
"/** @file\n"
" * Automatically generated from %s. Do NOT edit!\n"
" */\n"
"\n"
"#ifndef %s\n"
"#define %s\n"
"\n"
"#include <VBox/VBoxTpG.h>\n"
"\n"
"RT_C_DECLS_BEGIN\n"
"\n"
"#ifdef VBOX_WITH_DTRACE\n"
"\n"
"# ifdef _MSC_VER\n"
"# pragma data_seg(VTG_LOC_SECT)\n"
"# pragma data_seg()\n"
"# endif\n"
"\n"
,
g_pszScript,
szTmp,
szTmp);
/*
* Declare data, code and macros for each probe.
*/
PVTGPROVIDER pProv;
PVTGPROBE pProbe;
PVTGARG pArg;
RTListForEach(&g_ProviderHead, pProv, VTGPROVIDER, ListEntry)
{
RTListForEach(&pProv->ProbeHead, pProbe, VTGPROBE, ListEntry)
{
ScmStreamPrintf(pStrm,
"extern bool g_fVTGProbeEnabled_%s_%s;\n"
"extern uint8_t g_VTGProbeData_%s_%s;\n"
"DECLASM(void) VTGProbeStub_%s_%s(PVTGPROBELOC",
pProv->pszName, pProbe->pszMangledName,
pProv->pszName, pProbe->pszMangledName,
pProv->pszName, pProbe->pszMangledName);
RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
{
ScmStreamPrintf(pStrm, ", %s", pArg->pszType);
}
generateProbeDefineName(szTmp, sizeof(szTmp), pProv->pszName, pProbe->pszMangledName);
ScmStreamPrintf(pStrm,
");\n"
"# define %s_ENABLED() \\\n"
" (RT_UNLIKELY(g_fVTGProbeEnabled_%s_%s)) \n"
"# define %s("
, szTmp,
pProv->pszName, pProbe->pszMangledName,
szTmp);
RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
{
if (RTListNodeIsFirst(&pProbe->ArgHead, &pArg->ListEntry))
ScmStreamPrintf(pStrm, "%s", pArg->pszName);
else
ScmStreamPrintf(pStrm, ", %s", pArg->pszName);
}
ScmStreamPrintf(pStrm,
") \\\n"
" do { \\\n"
" if (RT_UNLIKELY(g_fVTGProbeEnabled_%s_%s)) \\\n"
" { \\\n"
" VTG_DECL_VTGPROBELOC(s_VTGProbeLoc) = \\\n"
" { __LINE__, 0, UINT32_MAX, __FUNCTION__, &g_VTGProbeData_%s_%s }; \\\n"
" VTGProbeStub_%s_%s(&s_VTGProbeLoc",
pProv->pszName, pProbe->pszMangledName,
pProv->pszName, pProbe->pszMangledName,
pProv->pszName, pProbe->pszMangledName);
RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
{
ScmStreamPrintf(pStrm, ", %s", pArg->pszName);
}
ScmStreamPrintf(pStrm,
"); \\\n"
" } \\\n"
" { \\\n" );
uint32_t iArg = 0;
RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
{
if (pArg->fType & VTG_TYPE_FIXED_SIZED)
ScmStreamPrintf(pStrm,
" AssertCompile(sizeof(%s) == %u); \\\n"
" AssertCompile(sizeof(%s) <= %u); \\\n",
pArg->pszType, pArg->fType & VTG_TYPE_SIZE_MASK,
pArg->pszName, pArg->fType & VTG_TYPE_SIZE_MASK);
else if (pArg->fType & (VTG_TYPE_POINTER | VTG_TYPE_HC_ARCH_SIZED))
ScmStreamPrintf(pStrm,
" AssertCompile(sizeof(%s) <= sizeof(uintptr_t)); \\\n"
" AssertCompile(sizeof(%s) <= sizeof(uintptr_t)); \\\n",
pArg->pszName,
pArg->pszType);
iArg++;
}
ScmStreamPrintf(pStrm,
" } \\\n"
" } while (0)\n"
"\n");
}
}
ScmStreamPrintf(pStrm,
"\n"
"#else\n"
"\n");
RTListForEach(&g_ProviderHead, pProv, VTGPROVIDER, ListEntry)
{
RTListForEach(&pProv->ProbeHead, pProbe, VTGPROBE, ListEntry)
{
generateProbeDefineName(szTmp, sizeof(szTmp), pProv->pszName, pProbe->pszMangledName);
ScmStreamPrintf(pStrm,
"# define %s_ENABLED() (false)\n"
"# define %s("
, szTmp, szTmp);
RTListForEach(&pProbe->ArgHead, pArg, VTGARG, ListEntry)
{
if (RTListNodeIsFirst(&pProbe->ArgHead, &pArg->ListEntry))
ScmStreamPrintf(pStrm, "%s", pArg->pszName);
else
ScmStreamPrintf(pStrm, ", %s", pArg->pszName);
}
ScmStreamPrintf(pStrm,
") do { } while (0)\n");
}
}
ScmStreamWrite(pStrm, RT_STR_TUPLE("\n"
"#endif\n"
"\n"
"RT_C_DECLS_END\n"
"#endif\n"));
return RTEXITCODE_SUCCESS;
}
static RTEXITCODE generateHeader(const char *pszHeader)
{
return generateFile(pszHeader, "header", generateHeaderInner);
}
/**
* If the given C word is at off - 1, return @c true and skip beyond it,
* otherwise return @c false.
*
* @retval true if the given C-word is at the current position minus one char.
* The stream position changes.
* @retval false if not. The stream position is unchanged.
*
* @param pStream The stream.
* @param cchWord The length of the word.
* @param pszWord The word.
*/
bool ScmStreamCMatchingWordM1(PSCMSTREAM pStream, const char *pszWord, size_t cchWord)
{
/* Check stream state. */
AssertReturn(!pStream->fWriteOrRead, false);
AssertReturn(RT_SUCCESS(pStream->rc), false);
AssertReturn(pStream->fFullyLineated, false);
/* Sufficient chars left on the line? */
size_t const iLine = pStream->iLine;
AssertReturn(pStream->off > pStream->paLines[iLine].off, false);
size_t const cchLeft = pStream->paLines[iLine].cch + pStream->paLines[iLine].off - (pStream->off - 1);
if (cchWord > cchLeft)
return false;
/* Do they match? */
const char *psz = &pStream->pch[pStream->off - 1];
if (memcmp(psz, pszWord, cchWord))
return false;
/* Is it the end of a C word? */
if (cchWord < cchLeft)
{
psz += cchWord;
if (RT_C_IS_ALNUM(*psz) || *psz == '_')
return false;
}
/* Skip ahead. */
pStream->off += cchWord - 1;
return true;
}
/**
* Get's the C word starting at the current position.
*
* @returns Pointer to the word on success and the stream position advanced to
* the end of it.
* NULL on failure, stream position normally unchanged.
* @param pStream The stream to get the C word from.
* @param pcchWord Where to return the word length.
*/
const char *ScmStreamCGetWord(PSCMSTREAM pStream, size_t *pcchWord)
{
/* Check stream state. */
AssertReturn(!pStream->fWriteOrRead, NULL);
AssertReturn(RT_SUCCESS(pStream->rc), NULL);
AssertReturn(pStream->fFullyLineated, NULL);
/* Get the number of chars left on the line and locate the current char. */
size_t const iLine = pStream->iLine;
size_t const cchLeft = pStream->paLines[iLine].cch + pStream->paLines[iLine].off - pStream->off;
const char *psz = &pStream->pch[pStream->off];
/* Is it a leading C character. */
if (!RT_C_IS_ALPHA(*psz) && *psz == '_')
return NULL;
/* Find the end of the word. */
char ch;
size_t off = 1;
while ( off < cchLeft
&& ( (ch = psz[off]) == '_'
|| RT_C_IS_ALNUM(ch)))
off++;
pStream->off += off;
*pcchWord = off;
return psz;
}
/**
* Get's the C word starting at the current position minus one.
*
* @returns Pointer to the word on success and the stream position advanced to
* the end of it.
* NULL on failure, stream position normally unchanged.
* @param pStream The stream to get the C word from.
* @param pcchWord Where to return the word length.
*/
const char *ScmStreamCGetWordM1(PSCMSTREAM pStream, size_t *pcchWord)
{
/* Check stream state. */
AssertReturn(!pStream->fWriteOrRead, NULL);
AssertReturn(RT_SUCCESS(pStream->rc), NULL);
AssertReturn(pStream->fFullyLineated, NULL);
/* Get the number of chars left on the line and locate the current char. */
size_t const iLine = pStream->iLine;
size_t const cchLeft = pStream->paLines[iLine].cch + pStream->paLines[iLine].off - (pStream->off - 1);
const char *psz = &pStream->pch[pStream->off - 1];
/* Is it a leading C character. */
if (!RT_C_IS_ALPHA(*psz) && *psz == '_')
return NULL;
/* Find the end of the word. */
char ch;
size_t off = 1;
while ( off < cchLeft
&& ( (ch = psz[off]) == '_'
|| RT_C_IS_ALNUM(ch)))
off++;
pStream->off += off - 1;
*pcchWord = off;
return psz;
}
/**
* Parser error with line and position.
*
* @returns RTEXITCODE_FAILURE.
* @param pStrm The stream.
* @param cb The offset from the current position to the
* point of failure.
* @param pszMsg The message to display.
*/
static RTEXITCODE parseError(PSCMSTREAM pStrm, size_t cb, const char *pszMsg)
{
if (cb)
ScmStreamSeekRelative(pStrm, -(ssize_t)cb);
size_t const off = ScmStreamTell(pStrm);
size_t const iLine = ScmStreamTellLine(pStrm);
ScmStreamSeekByLine(pStrm, iLine);
size_t const offLine = ScmStreamTell(pStrm);
RTPrintf("%s:%d:%zd: error: %s.\n", g_pszScript, iLine + 1, off - offLine + 1, pszMsg);
size_t cchLine;
SCMEOL enmEof;
const char *pszLine = ScmStreamGetLineByNo(pStrm, iLine, &cchLine, &enmEof);
if (pszLine)
RTPrintf(" %.*s\n"
" %*s^\n",
cchLine, pszLine, off - offLine, "");
return RTEXITCODE_FAILURE;
}
/**
* Parser error with line and position.
*
* @returns RTEXITCODE_FAILURE.
* @param pStrm The stream.
* @param cb The offset from the current position to the
* point of failure.
* @param pszMsg The message to display.
*/
static RTEXITCODE parseErrorAbs(PSCMSTREAM pStrm, size_t off, const char *pszMsg)
{
ScmStreamSeekAbsolute(pStrm, off);
return parseError(pStrm, 0, pszMsg);
}
/**
* Handles a C++ one line comment.
*
* @returns Exit code.
* @param pStrm The stream.
*/
static RTEXITCODE parseOneLineComment(PSCMSTREAM pStrm)
{
ScmStreamSeekByLine(pStrm, ScmStreamTellLine(pStrm) + 1);
return RTEXITCODE_SUCCESS;
}
/**
* Handles a multi-line C/C++ comment.
*
* @returns Exit code.
* @param pStrm The stream.
*/
static RTEXITCODE parseMultiLineComment(PSCMSTREAM pStrm)
{
unsigned ch;
while ((ch = ScmStreamGetCh(pStrm)) != ~(unsigned)0)
{
if (ch == '*')
{
do
ch = ScmStreamGetCh(pStrm);
while (ch == '*');
if (ch == '/')
return RTEXITCODE_SUCCESS;
}
}
parseError(pStrm, 1, "Expected end of comment, got end of file");
return RTEXITCODE_FAILURE;
}
/**
* Skips spaces and comments.
*
* @returns RTEXITCODE_SUCCESS or RTEXITCODE_FAILURE.
* @param pStrm The stream..
*/
static RTEXITCODE parseSkipSpacesAndComments(PSCMSTREAM pStrm)
{
unsigned ch;
while ((ch = ScmStreamPeekCh(pStrm)) != ~(unsigned)0)
{
if (!RT_C_IS_SPACE(ch) && ch != '/')
return RTEXITCODE_SUCCESS;
unsigned ch2 = ScmStreamGetCh(pStrm); AssertBreak(ch == ch2); NOREF(ch2);
if (ch == '/')
{
ch = ScmStreamGetCh(pStrm);
RTEXITCODE rcExit;
if (ch == '*')
rcExit = parseMultiLineComment(pStrm);
else if (ch == '/')
rcExit = parseOneLineComment(pStrm);
else
rcExit = parseError(pStrm, 2, "Unexpected character");
if (rcExit != RTEXITCODE_SUCCESS)
return rcExit;
}
}
return parseError(pStrm, 0, "Unexpected end of file");
}
/**
* Skips spaces and comments, returning the next character.
*
* @returns Next non-space-non-comment character. ~(unsigned)0 on EOF or
* failure.
* @param pStrm The stream.
*/
static unsigned parseGetNextNonSpaceNonCommentCh(PSCMSTREAM pStrm)
{
unsigned ch;
while ((ch = ScmStreamGetCh(pStrm)) != ~(unsigned)0)
{
if (!RT_C_IS_SPACE(ch) && ch != '/')
return ch;
if (ch == '/')
{
ch = ScmStreamGetCh(pStrm);
RTEXITCODE rcExit;
if (ch == '*')
rcExit = parseMultiLineComment(pStrm);
else if (ch == '/')
rcExit = parseOneLineComment(pStrm);
else
rcExit = parseError(pStrm, 2, "Unexpected character");
if (rcExit != RTEXITCODE_SUCCESS)
return ~(unsigned)0;
}
}
parseError(pStrm, 0, "Unexpected end of file");
return ~(unsigned)0;
}
/**
* Get the next non-space-non-comment character on a preprocessor line.
*
* @returns The next character. On error message and ~(unsigned)0.
* @param pStrm The stream.
*/
static unsigned parseGetNextNonSpaceNonCommentChOnPpLine(PSCMSTREAM pStrm)
{
size_t off = ScmStreamTell(pStrm) - 1;
unsigned ch;
while ((ch = ScmStreamGetCh(pStrm)) != ~(unsigned)0)
{
if (RT_C_IS_SPACE(ch))
{
if (ch == '\n' || ch == '\r')
{
parseErrorAbs(pStrm, off, "Invalid preprocessor statement");
break;
}
}
else if (ch == '\\')
{
size_t off2 = ScmStreamTell(pStrm) - 1;
ch = ScmStreamGetCh(pStrm);
if (ch == '\r')
ch = ScmStreamGetCh(pStrm);
if (ch != '\n')
{
parseErrorAbs(pStrm, off2, "Expected new line");
break;
}
}
else
return ch;
}
return ~(unsigned)0;
}
/**
* Skips spaces and comments.
*
* @returns Same as ScmStreamCGetWord
* @param pStrm The stream..
* @param pcchWord Where to return the length.
*/
static const char *parseGetNextCWord(PSCMSTREAM pStrm, size_t *pcchWord)
{
if (parseSkipSpacesAndComments(pStrm) != RTEXITCODE_SUCCESS)
return NULL;
return ScmStreamCGetWord(pStrm, pcchWord);
}
/**
* Parses interface stability.
*
* @returns Interface stability if parsed correctly, otherwise error message and
* kVTGStability_Invalid.
* @param pStrm The stream.
* @param ch The first character in the stability spec.
*/
static kVTGStability parseStability(PSCMSTREAM pStrm, unsigned ch)
{
switch (ch)
{
case 'E':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("External")))
return kVTGStability_External;
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Evolving")))
return kVTGStability_Evolving;
break;
case 'I':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Internal")))
return kVTGStability_Internal;
break;
case 'O':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Obsolete")))
return kVTGStability_Obsolete;
break;
case 'P':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Private")))
return kVTGStability_Private;
break;
case 'S':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Stable")))
return kVTGStability_Stable;
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Standard")))
return kVTGStability_Standard;
break;
case 'U':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Unstable")))
return kVTGStability_Unstable;
break;
}
parseError(pStrm, 1, "Unknown stability specifier");
return kVTGStability_Invalid;
}
/**
* Parses data depndency class.
*
* @returns Data dependency class if parsed correctly, otherwise error message
* and kVTGClass_Invalid.
* @param pStrm The stream.
* @param ch The first character in the stability spec.
*/
static kVTGClass parseDataDepClass(PSCMSTREAM pStrm, unsigned ch)
{
switch (ch)
{
case 'C':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Common")))
return kVTGClass_Common;
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Cpu")))
return kVTGClass_Cpu;
break;
case 'G':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Group")))
return kVTGClass_Group;
break;
case 'I':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Isa")))
return kVTGClass_Isa;
break;
case 'P':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Platform")))
return kVTGClass_Platform;
break;
case 'U':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("Unknown")))
return kVTGClass_Unknown;
break;
}
parseError(pStrm, 1, "Unknown data dependency class specifier");
return kVTGClass_Invalid;
}
/**
* Parses a pragma D attributes statement.
*
* @returns Suitable exit code, errors message already written on failure.
* @param pStrm The stream.
*/
static RTEXITCODE parsePragmaDAttributes(PSCMSTREAM pStrm)
{
/*
* "CodeStability/DataStability/DataDepClass" - no spaces allowed.
*/
unsigned ch = parseGetNextNonSpaceNonCommentChOnPpLine(pStrm);
if (ch == ~(unsigned)0)
return RTEXITCODE_FAILURE;
kVTGStability enmCode = parseStability(pStrm, ch);
if (enmCode == kVTGStability_Invalid)
return RTEXITCODE_FAILURE;
ch = ScmStreamGetCh(pStrm);
if (ch != '/')
return parseError(pStrm, 1, "Expected '/' following the code stability specifier");
kVTGStability enmData = parseStability(pStrm, ScmStreamGetCh(pStrm));
if (enmData == kVTGStability_Invalid)
return RTEXITCODE_FAILURE;
ch = ScmStreamGetCh(pStrm);
if (ch != '/')
return parseError(pStrm, 1, "Expected '/' following the data stability specifier");
kVTGClass enmDataDep = parseDataDepClass(pStrm, ScmStreamGetCh(pStrm));
if (enmDataDep == kVTGClass_Invalid)
return RTEXITCODE_FAILURE;
/*
* Expecting 'provider' followed by the name of an provider defined earlier.
*/
ch = parseGetNextNonSpaceNonCommentChOnPpLine(pStrm);
if (ch == ~(unsigned)0)
return RTEXITCODE_FAILURE;
if (ch != 'p' || !ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("provider")))
return parseError(pStrm, 1, "Expected 'provider'");
size_t cchName;
const char *pszName = parseGetNextCWord(pStrm, &cchName);
if (!pszName)
return parseError(pStrm, 1, "Expected provider name");
PVTGPROVIDER pProv;
RTListForEach(&g_ProviderHead, pProv, VTGPROVIDER, ListEntry)
{
if ( !strncmp(pProv->pszName, pszName, cchName)
&& pProv->pszName[cchName] == '\0')
break;
}
if (RTListNodeIsDummy(&g_ProviderHead, pProv, VTGPROVIDER, ListEntry))
return parseError(pStrm, cchName, "Provider not found");
/*
* Which aspect of the provider?
*/
size_t cchAspect;
const char *pszAspect = parseGetNextCWord(pStrm, &cchAspect);
if (!pszAspect)
return parseError(pStrm, 1, "Expected provider aspect");
PVTGATTRS pAttrs;
if (cchAspect == 8 && !memcmp(pszAspect, "provider", 8))
pAttrs = &pProv->AttrSelf;
else if (cchAspect == 8 && !memcmp(pszAspect, "function", 8))
pAttrs = &pProv->AttrFunctions;
else if (cchAspect == 6 && !memcmp(pszAspect, "module", 6))
pAttrs = &pProv->AttrModules;
else if (cchAspect == 4 && !memcmp(pszAspect, "name", 4))
pAttrs = &pProv->AttrName;
else if (cchAspect == 4 && !memcmp(pszAspect, "args", 4))
pAttrs = &pProv->AttrArguments;
else
return parseError(pStrm, cchAspect, "Unknown aspect");
if (pAttrs->enmCode != kVTGStability_Invalid)
return parseError(pStrm, cchAspect, "You have already specified these attributes");
pAttrs->enmCode = enmCode;
pAttrs->enmData = enmData;
pAttrs->enmDataDep = enmDataDep;
return RTEXITCODE_SUCCESS;
}
/**
* Parses a D pragma statement.
*
* @returns Suitable exit code, errors message already written on failure.
* @param pStrm The stream.
*/
static RTEXITCODE parsePragma(PSCMSTREAM pStrm)
{
RTEXITCODE rcExit;
unsigned ch = parseGetNextNonSpaceNonCommentChOnPpLine(pStrm);
if (ch == ~(unsigned)0)
rcExit = RTEXITCODE_FAILURE;
else if (ch == 'D' && ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("D")))
{
ch = parseGetNextNonSpaceNonCommentChOnPpLine(pStrm);
if (ch == ~(unsigned)0)
rcExit = RTEXITCODE_FAILURE;
else if (ch == 'a' && ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("attributes")))
rcExit = parsePragmaDAttributes(pStrm);
else
rcExit = parseError(pStrm, 1, "Unknown pragma D");
}
else
rcExit = parseError(pStrm, 1, "Unknown pragma");
return rcExit;
}
/**
* Classifies the given type expression.
*
* @return Type flags.
* @param pszType The type expression.
*/
static uint32_t parseTypeExpression(const char *pszType)
{
size_t cchType = strlen(pszType);
#define MY_STRMATCH(a_sz) (cchType == sizeof(a_sz) - 1 && !memcmp(a_sz, pszType, sizeof(a_sz) - 1))
/*
* Try detect pointers.
*/
if (pszType[cchType - 1] == '*') return VTG_TYPE_POINTER;
if (pszType[cchType - 1] == '&')
{
RTMsgWarning("Please avoid using references like '%s' for probe arguments!", pszType);
return VTG_TYPE_POINTER;
}
/*
* Standard integer types and IPRT variants.
* It's important that we catch all types larger than 32-bit here or we'll
* screw up the probe argument handling.
*/
if (MY_STRMATCH("int")) return VTG_TYPE_FIXED_SIZED | sizeof(int) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("uintptr_t")) return VTG_TYPE_HC_ARCH_SIZED | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("intptr_t")) return VTG_TYPE_HC_ARCH_SIZED | VTG_TYPE_SIGNED;
//if (MY_STRMATCH("uint128_t")) return VTG_TYPE_FIXED_SIZED | sizeof(uint128_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("uint64_t")) return VTG_TYPE_FIXED_SIZED | sizeof(uint64_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("uint32_t")) return VTG_TYPE_FIXED_SIZED | sizeof(uint32_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("uint16_t")) return VTG_TYPE_FIXED_SIZED | sizeof(uint16_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("uint8_t")) return VTG_TYPE_FIXED_SIZED | sizeof(uint8_t) | VTG_TYPE_UNSIGNED;
//if (MY_STRMATCH("int128_t")) return VTG_TYPE_FIXED_SIZED | sizeof(int128_t) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("int64_t")) return VTG_TYPE_FIXED_SIZED | sizeof(int64_t) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("int32_t")) return VTG_TYPE_FIXED_SIZED | sizeof(int32_t) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("int16_t")) return VTG_TYPE_FIXED_SIZED | sizeof(int16_t) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("int8_t")) return VTG_TYPE_FIXED_SIZED | sizeof(int8_t) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTUINT64U")) return VTG_TYPE_FIXED_SIZED | sizeof(uint64_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTUINT32U")) return VTG_TYPE_FIXED_SIZED | sizeof(uint32_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTUINT16U")) return VTG_TYPE_FIXED_SIZED | sizeof(uint16_t) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTMSINTERVAL")) return VTG_TYPE_FIXED_SIZED | sizeof(RTMSINTERVAL) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTTIMESPEC")) return VTG_TYPE_FIXED_SIZED | sizeof(RTTIMESPEC) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTPROCESS")) return VTG_TYPE_FIXED_SIZED | sizeof(RTPROCESS) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTHCPHYS")) return VTG_TYPE_FIXED_SIZED | sizeof(RTHCPHYS) | VTG_TYPE_UNSIGNED | VTG_TYPE_PHYS;
if (MY_STRMATCH("RTR3PTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3;
if (MY_STRMATCH("RTR0PTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R0;
if (MY_STRMATCH("RTRCPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_RC;
if (MY_STRMATCH("RTHCPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_CTX_R0;
if (MY_STRMATCH("RTR3UINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTR0UINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R0 | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTRCUINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_RC | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTHCUINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_CTX_R0 | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTR3INTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTR0INTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R0 | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTRCINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_RC | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTHCINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_CTX_R0 | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTUINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_CTX_R0 | VTG_TYPE_CTX_RC | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTINTPTR")) return VTG_TYPE_CTX_POINTER | VTG_TYPE_CTX_R3 | VTG_TYPE_CTX_R0 | VTG_TYPE_CTX_RC | VTG_TYPE_SIGNED;
if (MY_STRMATCH("RTHCUINTREG")) return VTG_TYPE_HC_ARCH_SIZED | VTG_TYPE_CTX_R3 | VTG_TYPE_CTX_R0 | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTR3UINTREG")) return VTG_TYPE_HC_ARCH_SIZED | VTG_TYPE_CTX_R3 | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTR0UINTREG")) return VTG_TYPE_HC_ARCH_SIZED | VTG_TYPE_CTX_R3 | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("RTGCUINTREG")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCUINTREG) | VTG_TYPE_UNSIGNED | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCPTR")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCPTR) | VTG_TYPE_UNSIGNED | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCINTPTR")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCUINTPTR) | VTG_TYPE_SIGNED | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCPTR32")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCPTR32) | VTG_TYPE_SIGNED | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCPTR64")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCPTR64) | VTG_TYPE_SIGNED | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCPHYS")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCPHYS) | VTG_TYPE_UNSIGNED | VTG_TYPE_PHYS | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCPHYS32")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCPHYS32) | VTG_TYPE_UNSIGNED | VTG_TYPE_PHYS | VTG_TYPE_CTX_GST;
if (MY_STRMATCH("RTGCPHYS64")) return VTG_TYPE_FIXED_SIZED | sizeof(RTGCPHYS64) | VTG_TYPE_UNSIGNED | VTG_TYPE_PHYS | VTG_TYPE_CTX_GST;
/*
* The special VBox types.
*/
if (MY_STRMATCH("PVM")) return VTG_TYPE_CTX_POINTER;
if (MY_STRMATCH("PVMCPU")) return VTG_TYPE_CTX_POINTER;
/*
* Preaching time.
*/
if ( MY_STRMATCH("unsigned long")
|| MY_STRMATCH("unsigned long long")
|| MY_STRMATCH("signed long")
|| MY_STRMATCH("signed long long")
|| MY_STRMATCH("long")
|| MY_STRMATCH("long long")
|| MY_STRMATCH("char")
|| MY_STRMATCH("signed char")
|| MY_STRMATCH("unsigned char")
|| MY_STRMATCH("double")
|| MY_STRMATCH("long double")
|| MY_STRMATCH("float")
)
{
RTMsgError("Please do NOT use the type '%s' for probe arguments!", pszType);
g_cTypeErrors++;
return 0;
}
if ( MY_STRMATCH("unsigned")
|| MY_STRMATCH("signed")
|| MY_STRMATCH("signed int")
|| MY_STRMATCH("unsigned int")
|| MY_STRMATCH("short")
|| MY_STRMATCH("signed short")
|| MY_STRMATCH("unsigned short")
)
RTMsgWarning("Please avoid using the type '%s' for probe arguments!", pszType);
if (MY_STRMATCH("unsigned")) return VTG_TYPE_FIXED_SIZED | sizeof(int) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("unsigned int")) return VTG_TYPE_FIXED_SIZED | sizeof(int) | VTG_TYPE_UNSIGNED;
if (MY_STRMATCH("signed")) return VTG_TYPE_FIXED_SIZED | sizeof(int) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("signed int")) return VTG_TYPE_FIXED_SIZED | sizeof(int) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("short")) return VTG_TYPE_FIXED_SIZED | sizeof(short) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("signed short")) return VTG_TYPE_FIXED_SIZED | sizeof(short) | VTG_TYPE_SIGNED;
if (MY_STRMATCH("unsigned short")) return VTG_TYPE_FIXED_SIZED | sizeof(short) | VTG_TYPE_UNSIGNED;
/*
* What we haven't caught by now is either unknown to us or wrong.
*/
if (pszType[0] == 'P')
{
RTMsgError("Type '%s' looks like a pointer typedef, please do NOT use those "
"but rather the non-pointer typedef or struct with '*'",
pszType);
g_cTypeErrors++;
return VTG_TYPE_POINTER;
}
RTMsgError("Don't know '%s' - please change or fix VBoxTpG", pszType);
g_cTypeErrors++;
#undef MY_STRCMP
return 0;
}
/**
* Unmangles the probe name.
*
* This involves translating double underscore to dash.
*
* @returns Pointer to the unmangled name in the string table.
* @param pszMangled The mangled name.
*/
static const char *parseUnmangleProbeName(const char *pszMangled)
{
size_t cchMangled = strlen(pszMangled);
char *pszTmp = (char *)alloca(cchMangled + 2);
const char *pszSrc = pszMangled;
char *pszDst = pszTmp;
while (*pszSrc)
{
if (pszSrc[0] == '_' && pszSrc[1] == '_' && pszSrc[2] != '_')
{
*pszDst++ = '-';
pszSrc += 2;
}
else
*pszDst++ = *pszSrc++;
}
*pszDst = '\0';
return strtabInsertN(pszTmp, pszDst - pszTmp);
}
/**
* Parses a D probe statement.
*
* @returns Suitable exit code, errors message already written on failure.
* @param pStrm The stream.
* @param pProv The provider being parsed.
*/
static RTEXITCODE parseProbe(PSCMSTREAM pStrm, PVTGPROVIDER pProv)
{
/*
* Next up is a name followed by an opening parenthesis.
*/
size_t cchProbe;
const char *pszProbe = parseGetNextCWord(pStrm, &cchProbe);
if (!pszProbe)
return parseError(pStrm, 1, "Expected a probe name starting with an alphabetical character");
unsigned ch = parseGetNextNonSpaceNonCommentCh(pStrm);
if (ch != '(')
return parseError(pStrm, 1, "Expected '(' after the probe name");
/*
* Create a probe instance.
*/
PVTGPROBE pProbe = (PVTGPROBE)RTMemAllocZ(sizeof(*pProbe));
if (!pProbe)
return parseError(pStrm, 0, "Out of memory");
RTListInit(&pProbe->ArgHead);
RTListAppend(&pProv->ProbeHead, &pProbe->ListEntry);
pProbe->offArgList = UINT32_MAX;
pProbe->pszMangledName = RTStrDupN(pszProbe, cchProbe);
if (!pProbe->pszMangledName)
return parseError(pStrm, 0, "Out of memory");
pProbe->pszUnmangledName = parseUnmangleProbeName(pProbe->pszMangledName);
if (!pProbe->pszUnmangledName)
return parseError(pStrm, 0, "Out of memory");
/*
* Parse loop for the argument.
*/
PVTGARG pArg = NULL;
size_t cchName = 0;
size_t cchArg = 0;
char szArg[4096];
for (;;)
{
ch = parseGetNextNonSpaceNonCommentCh(pStrm);
switch (ch)
{
case ')':
case ',':
{
/* commit the argument */
if (pArg)
{
if (!cchName)
return parseError(pStrm, 1, "Argument has no name");
if (cchArg - cchName - 1 >= 128)
return parseError(pStrm, 1, "Argument type too long");
pArg->pszType = strtabInsertN(szArg, cchArg - cchName - 1);
pArg->pszName = RTStrDupN(&szArg[cchArg - cchName], cchName);
if (!pArg->pszType || !pArg->pszName)
return parseError(pStrm, 1, "Out of memory");
pArg->fType = parseTypeExpression(pArg->pszType);
if (VTG_TYPE_IS_LARGE(pArg->fType))
pProbe->fHaveLargeArgs = true;
pArg = NULL;
cchName = cchArg = 0;
}
if (ch == ')')
{
size_t off = ScmStreamTell(pStrm);
ch = parseGetNextNonSpaceNonCommentCh(pStrm);
if (ch != ';')
return parseErrorAbs(pStrm, off, "Expected ';'");
return RTEXITCODE_SUCCESS;
}
break;
}
default:
{
size_t cchWord;
const char *pszWord = ScmStreamCGetWordM1(pStrm, &cchWord);
if (!pszWord)
return parseError(pStrm, 0, "Expected argument");
if (!pArg)
{
pArg = (PVTGARG)RTMemAllocZ(sizeof(*pArg));
if (!pArg)
return parseError(pStrm, 1, "Out of memory");
RTListAppend(&pProbe->ArgHead, &pArg->ListEntry);
pProbe->cArgs++;
if (cchWord + 1 > sizeof(szArg))
return parseError(pStrm, 1, "Too long parameter declaration");
memcpy(szArg, pszWord, cchWord);
szArg[cchWord] = '\0';
cchArg = cchWord;
cchName = 0;
}
else
{
if (cchArg + 1 + cchWord + 1 > sizeof(szArg))
return parseError(pStrm, 1, "Too long parameter declaration");
szArg[cchArg++] = ' ';
memcpy(&szArg[cchArg], pszWord, cchWord);
cchArg += cchWord;
szArg[cchArg] = '\0';
cchName = cchWord;
}
break;
}
case '*':
{
if (!pArg)
return parseError(pStrm, 1, "A parameter type does not start with an asterix");
if (cchArg + sizeof(" *") >= sizeof(szArg))
return parseError(pStrm, 1, "Too long parameter declaration");
szArg[cchArg++] = ' ';
szArg[cchArg++] = '*';
szArg[cchArg ] = '\0';
cchName = 0;
break;
}
case ~(unsigned)0:
return parseError(pStrm, 0, "Missing closing ')' on probe");
}
}
}
/**
* Parses a D provider statement.
*
* @returns Suitable exit code, errors message already written on failure.
* @param pStrm The stream.
*/
static RTEXITCODE parseProvider(PSCMSTREAM pStrm)
{
/*
* Next up is a name followed by a curly bracket. Ignore comments.
*/
RTEXITCODE rcExit = parseSkipSpacesAndComments(pStrm);
if (rcExit != RTEXITCODE_SUCCESS)
return parseError(pStrm, 1, "Expected a provider name starting with an alphabetical character");
size_t cchName;
const char *pszName = ScmStreamCGetWord(pStrm, &cchName);
if (!pszName)
return parseError(pStrm, 0, "Bad provider name");
if (RT_C_IS_DIGIT(pszName[cchName - 1]))
return parseError(pStrm, 1, "A provider name cannot end with digit");
unsigned ch = parseGetNextNonSpaceNonCommentCh(pStrm);
if (ch != '{')
return parseError(pStrm, 1, "Expected '{' after the provider name");
/*
* Create a provider instance.
*/
PVTGPROVIDER pProv = (PVTGPROVIDER)RTMemAllocZ(sizeof(*pProv));
if (!pProv)
return parseError(pStrm, 0, "Out of memory");
RTListInit(&pProv->ProbeHead);
RTListAppend(&g_ProviderHead, &pProv->ListEntry);
pProv->pszName = strtabInsertN(pszName, cchName);
if (!pProv->pszName)
return parseError(pStrm, 0, "Out of memory");
/*
* Parse loop.
*/
for (;;)
{
ch = parseGetNextNonSpaceNonCommentCh(pStrm);
switch (ch)
{
case 'p':
if (ScmStreamCMatchingWordM1(pStrm, RT_STR_TUPLE("probe")))
rcExit = parseProbe(pStrm, pProv);
else
rcExit = parseError(pStrm, 1, "Unexpected character");
break;
case '}':
{
size_t off = ScmStreamTell(pStrm);
ch = parseGetNextNonSpaceNonCommentCh(pStrm);
if (ch == ';')
return RTEXITCODE_SUCCESS;
rcExit = parseErrorAbs(pStrm, off, "Expected ';'");
break;
}
case ~(unsigned)0:
rcExit = parseError(pStrm, 0, "Missing closing '}' on provider");
break;
default:
rcExit = parseError(pStrm, 1, "Unexpected character");
break;
}
if (rcExit != RTEXITCODE_SUCCESS)
return rcExit;
}
}
static RTEXITCODE parseScript(const char *pszScript)
{
SCMSTREAM Strm;
int rc = ScmStreamInitForReading(&Strm, pszScript);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to open & read '%s' into memory: %Rrc", pszScript, rc);
if (g_cVerbosity > 0)
RTMsgInfo("Parsing '%s'...", pszScript);
RTEXITCODE rcExit = RTEXITCODE_SUCCESS;
unsigned ch;
while ((ch = ScmStreamGetCh(&Strm)) != ~(unsigned)0)
{
if (RT_C_IS_SPACE(ch))
continue;
switch (ch)
{
case '/':
ch = ScmStreamGetCh(&Strm);
if (ch == '*')
rcExit = parseMultiLineComment(&Strm);
else if (ch == '/')
rcExit = parseOneLineComment(&Strm);
else
rcExit = parseError(&Strm, 2, "Unexpected character");
break;
case 'p':
if (ScmStreamCMatchingWordM1(&Strm, RT_STR_TUPLE("provider")))
rcExit = parseProvider(&Strm);
else
rcExit = parseError(&Strm, 1, "Unexpected character");
break;
case '#':
{
ch = parseGetNextNonSpaceNonCommentChOnPpLine(&Strm);
if (ch == ~(unsigned)0)
rcExit = RTEXITCODE_FAILURE;
else if (ch == 'p' && ScmStreamCMatchingWordM1(&Strm, RT_STR_TUPLE("pragma")))
rcExit = parsePragma(&Strm);
else
rcExit = parseError(&Strm, 1, "Unsupported preprocessor directive");
break;
}
default:
rcExit = parseError(&Strm, 1, "Unexpected character");
break;
}
if (rcExit != RTEXITCODE_SUCCESS)
return rcExit;
}
ScmStreamDelete(&Strm);
if (g_cVerbosity > 0 && rcExit == RTEXITCODE_SUCCESS)
RTMsgInfo("Successfully parsed '%s'.", pszScript);
return rcExit;
}
/**
* Parses the arguments.
*/
static RTEXITCODE parseArguments(int argc, char **argv)
{
/*
* Set / Adjust defaults.
*/
int rc = RTPathAbs(g_pszAssemblerIncVal, g_szAssemblerIncVal, sizeof(g_szAssemblerIncVal) - 1);
if (RT_FAILURE(rc))
return RTMsgErrorExit(RTEXITCODE_FAILURE, "RTPathAbs failed: %Rrc", rc);
strcat(g_szAssemblerIncVal, "/");
g_pszAssemblerIncVal = g_szAssemblerIncVal;
/*
* Option config.
*/
enum
{
kVBoxTpGOpt_32Bit = 1000,
kVBoxTpGOpt_64Bit,
kVBoxTpGOpt_Assembler,
kVBoxTpGOpt_AssemblerFmtOpt,
kVBoxTpGOpt_AssemblerFmtVal,
kVBoxTpGOpt_AssemblerOutputOpt,
kVBoxTpGOpt_AssemblerOption,
kVBoxTpGOpt_Pic,
kVBoxTpGOpt_ProbeFnName,
kVBoxTpGOpt_ProbeFnImported,
kVBoxTpGOpt_ProbeFnNotImported,
kVBoxTpGOpt_End
};
static RTGETOPTDEF const s_aOpts[] =
{
/* dtrace w/ long options */
{ "-32", kVBoxTpGOpt_32Bit, RTGETOPT_REQ_NOTHING },
{ "-64", kVBoxTpGOpt_64Bit, RTGETOPT_REQ_NOTHING },
{ "--apply-cpp", 'C', RTGETOPT_REQ_NOTHING },
{ "--generate-obj", 'G', RTGETOPT_REQ_NOTHING },
{ "--generate-header", 'h', RTGETOPT_REQ_NOTHING },
{ "--output", 'o', RTGETOPT_REQ_STRING },
{ "--script", 's', RTGETOPT_REQ_STRING },
{ "--verbose", 'v', RTGETOPT_REQ_NOTHING },
/* out stuff */
{ "--assembler", kVBoxTpGOpt_Assembler, RTGETOPT_REQ_STRING },
{ "--assembler-fmt-opt", kVBoxTpGOpt_AssemblerFmtOpt, RTGETOPT_REQ_STRING },
{ "--assembler-fmt-val", kVBoxTpGOpt_AssemblerFmtVal, RTGETOPT_REQ_STRING },
{ "--assembler-output-opt", kVBoxTpGOpt_AssemblerOutputOpt, RTGETOPT_REQ_STRING },
{ "--assembler-option", kVBoxTpGOpt_AssemblerOption, RTGETOPT_REQ_STRING },
{ "--pic", kVBoxTpGOpt_Pic, RTGETOPT_REQ_NOTHING },
{ "--probe-fn-name", kVBoxTpGOpt_ProbeFnName, RTGETOPT_REQ_STRING },
{ "--probe-fn-imported", kVBoxTpGOpt_ProbeFnImported, RTGETOPT_REQ_NOTHING },
{ "--probe-fn-not-imported", kVBoxTpGOpt_ProbeFnNotImported, RTGETOPT_REQ_NOTHING },
/** @todo We're missing a bunch of assembler options! */
};
RTGETOPTUNION ValueUnion;
RTGETOPTSTATE GetOptState;
rc = RTGetOptInit(&GetOptState, argc, argv, &s_aOpts[0], RT_ELEMENTS(s_aOpts), 1, RTGETOPTINIT_FLAGS_OPTS_FIRST);
AssertReleaseRCReturn(rc, RTEXITCODE_FAILURE);
/*
* Process \the options.
*/
while ((rc = RTGetOpt(&GetOptState, &ValueUnion)) != 0)
{
switch (rc)
{
/*
* DTrace compatible options.
*/
case kVBoxTpGOpt_32Bit:
g_cBits = 32;
g_pszAssemblerFmtVal = g_szAssemblerFmtVal32;
break;
case kVBoxTpGOpt_64Bit:
g_cBits = 64;
g_pszAssemblerFmtVal = g_szAssemblerFmtVal64;
break;
case 'C':
g_fApplyCpp = true;
RTMsgWarning("Ignoring the -C option - no preprocessing of the D script will be performed");
break;
case 'G':
if ( g_enmAction != kVBoxTpGAction_Nothing
&& g_enmAction != kVBoxTpGAction_GenerateObject)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "-G and -h does not mix");
g_enmAction = kVBoxTpGAction_GenerateObject;
break;
case 'h':
if (!strcmp(GetOptState.pDef->pszLong, "--generate-header"))
{
if ( g_enmAction != kVBoxTpGAction_Nothing
&& g_enmAction != kVBoxTpGAction_GenerateHeader)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "-h and -G does not mix");
g_enmAction = kVBoxTpGAction_GenerateHeader;
}
else
{
/* --help or similar */
RTPrintf("VirtualBox Tracepoint Generator\n"
"\n"
"Usage: %s [options]\n"
"\n"
"Options:\n", RTProcShortName());
for (size_t i = 0; i < RT_ELEMENTS(s_aOpts); i++)
if ((unsigned)s_aOpts[i].iShort < 128)
RTPrintf(" -%c,%s\n", s_aOpts[i].iShort, s_aOpts[i].pszLong);
else
RTPrintf(" %s\n", s_aOpts[i].pszLong);
return RTEXITCODE_SUCCESS;
}
break;
case 'o':
if (g_pszOutput)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Output file is already set to '%s'", g_pszOutput);
g_pszOutput = ValueUnion.psz;
break;
case 's':
if (g_pszScript)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Script file is already set to '%s'", g_pszScript);
g_pszScript = ValueUnion.psz;
break;
case 'v':
g_cVerbosity++;
break;
case 'V':
{
/* The following is assuming that svn does it's job here. */
static const char s_szRev[] = "$Revision$";
const char *psz = RTStrStripL(strchr(s_szRev, ' '));
RTPrintf("r%.*s\n", strchr(psz, ' ') - psz, psz);
return RTEXITCODE_SUCCESS;
}
case VINF_GETOPT_NOT_OPTION:
if (g_enmAction == kVBoxTpGAction_GenerateObject)
break; /* object files, ignore them. */
return RTGetOptPrintError(rc, &ValueUnion);
/*
* Our options.
*/
case kVBoxTpGOpt_Assembler:
g_pszAssembler = ValueUnion.psz;
break;
case kVBoxTpGOpt_AssemblerFmtOpt:
g_pszAssemblerFmtOpt = ValueUnion.psz;
break;
case kVBoxTpGOpt_AssemblerFmtVal:
g_pszAssemblerFmtVal = ValueUnion.psz;
break;
case kVBoxTpGOpt_AssemblerOutputOpt:
g_pszAssemblerOutputOpt = ValueUnion.psz;
break;
case kVBoxTpGOpt_AssemblerOption:
if (g_cAssemblerOptions >= RT_ELEMENTS(g_apszAssemblerOptions))
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "Too many assembly options (max %u)", RT_ELEMENTS(g_apszAssemblerOptions));
g_apszAssemblerOptions[g_cAssemblerOptions] = ValueUnion.psz;
g_cAssemblerOptions++;
break;
case kVBoxTpGOpt_Pic:
g_fPic = true;
break;
case kVBoxTpGOpt_ProbeFnName:
g_pszProbeFnName = ValueUnion.psz;
break;
case kVBoxTpGOpt_ProbeFnImported:
g_fProbeFnImported = true;
break;
case kVBoxTpGOpt_ProbeFnNotImported:
g_fProbeFnImported = false;
break;
/*
* Errors and bugs.
*/
default:
return RTGetOptPrintError(rc, &ValueUnion);
}
}
/*
* Check that we've got all we need.
*/
if (g_enmAction == kVBoxTpGAction_Nothing)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "No action specified (-h or -G)");
if (!g_pszScript)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "No script file specified (-s)");
if (!g_pszOutput)
return RTMsgErrorExit(RTEXITCODE_SYNTAX, "No output file specified (-o)");
return RTEXITCODE_SUCCESS;
}
int main(int argc, char **argv)
{
int rc = RTR3InitExe(argc, &argv, 0);
if (RT_FAILURE(rc))
return 1;
RTEXITCODE rcExit = parseArguments(argc, argv);
if (rcExit == RTEXITCODE_SUCCESS)
{
/*
* Parse the script.
*/
RTListInit(&g_ProviderHead);
rcExit = parseScript(g_pszScript);
if (rcExit == RTEXITCODE_SUCCESS)
{
/*
* Take action.
*/
if (g_enmAction == kVBoxTpGAction_GenerateHeader)
rcExit = generateHeader(g_pszOutput);
else
rcExit = generateObject(g_pszOutput, g_pszTempAsm);
}
}
if (rcExit == RTEXITCODE_SUCCESS && g_cTypeErrors > 0)
rcExit = RTEXITCODE_FAILURE;
return rcExit;
}