/* $Id: RTTraceLogTool.cpp $ */ /** @file * IPRT - Utility for reading/receiving and dissecting trace logs. */ /* * Copyright (C) 2018-2020 Oracle Corporation * * This file is part of VirtualBox Open Source Edition (OSE), as * available from http://www.virtualbox.org. This file is free software; * you can redistribute it and/or modify it under the terms of the GNU * General Public License (GPL) as published by the Free Software * Foundation, in version 2 as it comes in the "COPYING" file of the * VirtualBox OSE distribution. VirtualBox OSE is distributed in the * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. * * The contents of this file may alternatively be used under the terms * of the Common Development and Distribution License Version 1.0 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the * VirtualBox OSE distribution, in which case the provisions of the * CDDL are applicable instead of those of the GPL. * * You may elect to license modified versions of this file under the * terms and conditions of either the GPL or the CDDL or both. */ /********************************************************************************************************************************* * Header Files * *********************************************************************************************************************************/ #include #include #include #include #include #include #include #include #include #include #include /** * The tracelog tool TCP server/client state. */ typedef struct RTTRACELOGTOOLTCP { /** Flag whether this is a server. */ bool fIsServer; /** The TCP socket handle for the connection. */ RTSOCKET hSock; /** The TCP server. */ PRTTCPSERVER pTcpSrv; } RTTRACELOGTOOLTCP; /** Pointer to the TCP server/client state. */ typedef RTTRACELOGTOOLTCP *PRTTRACELOGTOOLTCP; static void rtTraceLogTcpDestroy(PRTTRACELOGTOOLTCP pTrcLogTcp) { if (pTrcLogTcp->fIsServer) RTTcpServerDestroy(pTrcLogTcp->pTcpSrv); if (pTrcLogTcp->hSock != NIL_RTSOCKET) { if (pTrcLogTcp->fIsServer) RTTcpServerDisconnectClient2(pTrcLogTcp->hSock); else RTTcpClientClose(pTrcLogTcp->hSock); } RTMemFree(pTrcLogTcp); } static DECLCALLBACK(int) rtTraceLogToolTcpInput(void *pvUser, void *pvBuf, size_t cbBuf, size_t *pcbRead, RTMSINTERVAL cMsTimeout) { PRTTRACELOGTOOLTCP pTrcLogTcp = (PRTTRACELOGTOOLTCP)pvUser; if ( pTrcLogTcp->fIsServer && pTrcLogTcp->hSock == NIL_RTSOCKET) { int rc = RTTcpServerListen2(pTrcLogTcp->pTcpSrv, &pTrcLogTcp->hSock); if (RT_FAILURE(rc)) return rc; } int rc = RTTcpSelectOne(pTrcLogTcp->hSock, cMsTimeout); if (RT_SUCCESS(rc)) rc = RTTcpReadNB(pTrcLogTcp->hSock, pvBuf, cbBuf, pcbRead); return rc; } static DECLCALLBACK(int) rtTraceLogToolTcpClose(void *pvUser) { PRTTRACELOGTOOLTCP pTrcLogTcp = (PRTTRACELOGTOOLTCP)pvUser; rtTraceLogTcpDestroy(pTrcLogTcp); return VINF_SUCCESS; } /** * Tries to create a new trace log reader using the given input. * * @returns IPRT status code. * @param phTraceLogRdr Where to store the handle to the trace log reader instance on success. * @param pszInput The input path. * @param pszSave The optional path to save */ static int rtTraceLogToolReaderCreate(PRTTRACELOGRDR phTraceLogRdr, const char *pszInput, const char *pszSave) { RT_NOREF(pszSave); /* Try treating the input as a file first. */ int rc = RTTraceLogRdrCreateFromFile(phTraceLogRdr, pszInput); if (RT_FAILURE(rc)) { /* * Check whether the input looks like a port number or an address:port pair. * The former will create a server listening on the port while the latter tries * to connect to the given address:port combination. */ uint32_t uPort = 0; bool fIsServer = false; PRTTCPSERVER pTcpSrv = NULL; RTSOCKET hSock = NIL_RTSOCKET; rc = RTStrToUInt32Full(pszInput, 10, &uPort); if (rc == VINF_SUCCESS) { fIsServer = true; rc = RTTcpServerCreateEx(NULL, uPort, &pTcpSrv); } else { /* Try treating the input as an address:port pair. */ } if (RT_SUCCESS(rc)) { /* Initialize structure and reader. */ PRTTRACELOGTOOLTCP pTrcLogTcp = (PRTTRACELOGTOOLTCP)RTMemAllocZ(sizeof(*pTrcLogTcp)); if (pTrcLogTcp) { pTrcLogTcp->fIsServer = fIsServer; pTrcLogTcp->hSock = hSock; pTrcLogTcp->pTcpSrv = pTcpSrv; rc = RTTraceLogRdrCreate(phTraceLogRdr, rtTraceLogToolTcpInput, rtTraceLogToolTcpClose, pTrcLogTcp); if (RT_FAILURE(rc)) rtTraceLogTcpDestroy(pTrcLogTcp); } else { if (fIsServer) RTTcpServerDestroy(pTcpSrv); else RTSocketClose(hSock); } } } return rc; } int main(int argc, char **argv) { int rc = RTR3InitExe(argc, &argv, 0); if (RT_FAILURE(rc)) return RTMsgInitFailure(rc); /* * Parse arguments. */ static const RTGETOPTDEF s_aOptions[] = { { "--input", 'i', RTGETOPT_REQ_STRING }, { "--save", 's', RTGETOPT_REQ_STRING }, { "--help", 'h', RTGETOPT_REQ_NOTHING }, { "--version", 'V', RTGETOPT_REQ_NOTHING }, }; RTEXITCODE rcExit = RTEXITCODE_SUCCESS; const char *pszInput = NULL; const char *pszSave = NULL; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, argc, argv, s_aOptions, RT_ELEMENTS(s_aOptions), 1, RTGETOPTINIT_FLAGS_OPTS_FIRST); while ((rc = RTGetOpt(&GetState, &ValueUnion))) { switch (rc) { case 'h': RTPrintf("Usage: %s [options]\n" "\n" "Options:\n" " -i,--input=\n" " Input path, can be a file a port to start listening on for incoming connections or an address:port to connect to\n" " -s,--save=file\n" " Save the input to a file for later use\n" " -h, -?, --help\n" " Display this help text and exit successfully.\n" " -V, --version\n" " Display the revision and exit successfully.\n" , RTPathFilename(argv[0])); return RTEXITCODE_SUCCESS; case 'V': RTPrintf("$Revision: 135976 $\n"); return RTEXITCODE_SUCCESS; case 'i': pszInput = ValueUnion.psz; break; case 's': pszSave = ValueUnion.psz; break; default: return RTGetOptPrintError(rc, &ValueUnion); } } if (!pszInput) { RTPrintf("An input path must be given\n"); return RTEXITCODE_FAILURE; } /* * Create trace log reader instance. */ RTTRACELOGRDR hTraceLogRdr = NIL_RTTRACELOGRDR; rc = rtTraceLogToolReaderCreate(&hTraceLogRdr, pszInput, pszSave); if (RT_SUCCESS(rc)) { do { RTTRACELOGRDRPOLLEVT enmEvt = RTTRACELOGRDRPOLLEVT_INVALID; rc = RTTraceLogRdrEvtPoll(hTraceLogRdr, &enmEvt, RT_INDEFINITE_WAIT); if (RT_SUCCESS(rc)) { switch (enmEvt) { case RTTRACELOGRDRPOLLEVT_HDR_RECVD: RTMsgInfo("A valid header was received\n"); break; case RTTRACELOGRDRPOLLEVT_TRACE_EVENT_RECVD: { RTTRACELOGRDREVT hTraceLogEvt; rc = RTTraceLogRdrQueryLastEvt(hTraceLogRdr, &hTraceLogEvt); if (RT_SUCCESS(rc)) { PCRTTRACELOGEVTDESC pEvtDesc = RTTraceLogRdrEvtGetDesc(hTraceLogEvt); RTMsgInfo("%llu %llu %s\n", RTTraceLogRdrEvtGetSeqNo(hTraceLogEvt), RTTraceLogRdrEvtGetTs(hTraceLogEvt), pEvtDesc->pszId); for (unsigned i = 0; i < pEvtDesc->cEvtItems; i++) { RTTRACELOGEVTVAL Val; unsigned cVals = 0; rc = RTTraceLogRdrEvtFillVals(hTraceLogEvt, i, &Val, 1, &cVals); if (RT_SUCCESS(rc)) { switch (Val.pItemDesc->enmType) { case RTTRACELOGTYPE_BOOL: RTMsgInfo(" %s: %s\n", Val.pItemDesc->pszName, Val.u.f ? "true" : "false"); break; case RTTRACELOGTYPE_UINT8: RTMsgInfo(" %s: %u\n", Val.pItemDesc->pszName, Val.u.u8); break; case RTTRACELOGTYPE_INT8: RTMsgInfo(" %s: %d\n", Val.pItemDesc->pszName, Val.u.i8); break; case RTTRACELOGTYPE_UINT16: RTMsgInfo(" %s: %u\n", Val.pItemDesc->pszName, Val.u.u16); break; case RTTRACELOGTYPE_INT16: RTMsgInfo(" %s: %d\n", Val.pItemDesc->pszName, Val.u.i16); break; case RTTRACELOGTYPE_UINT32: RTMsgInfo(" %s: %u\n", Val.pItemDesc->pszName, Val.u.u32); break; case RTTRACELOGTYPE_INT32: RTMsgInfo(" %s: %d\n", Val.pItemDesc->pszName, Val.u.i32); break; case RTTRACELOGTYPE_UINT64: RTMsgInfo(" %s: %llu\n", Val.pItemDesc->pszName, Val.u.u64); break; case RTTRACELOGTYPE_INT64: RTMsgInfo(" %s: %lld\n", Val.pItemDesc->pszName, Val.u.i64); break; case RTTRACELOGTYPE_FLOAT32: case RTTRACELOGTYPE_FLOAT64: case RTTRACELOGTYPE_RAWDATA: RTMsgInfo(" %s: Float32, Float64 and raw data not supported yet\n", Val.pItemDesc->pszName); break; case RTTRACELOGTYPE_POINTER: RTMsgInfo(" %s: %#llx\n", Val.pItemDesc->pszName, Val.u.uPtr); break; case RTTRACELOGTYPE_SIZE: RTMsgInfo(" %s: %llu\n", Val.pItemDesc->pszName, Val.u.sz); break; default: RTMsgError(" %s: Invalid type given %d\n", Val.pItemDesc->pszName, Val.pItemDesc->enmType); } } else RTMsgInfo(" Failed to retrieve event data with %Rrc\n", rc); } } break; } default: RTMsgInfo("Invalid event received: %d\n", enmEvt); } } else rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Polling for an event failed with %Rrc\n", rc); } while (RT_SUCCESS(rc)); RTTraceLogRdrDestroy(hTraceLogRdr); } else rcExit = RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to create trace log reader with %Rrc\n", rc); return rcExit; }