git.cworth.org Git - apitrace/blob - trace_local_writer.cpp

   1 /**************************************************************************
   2  *
   3  * Copyright 2007-2011 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26
  27 #include <assert.h>
  28 #include <stdarg.h>
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31 #include <string.h>
  32
  33 #include <zlib.h>
  34
  35 #include "os.hpp"
  36 #include "trace_writer.hpp"
  37 #include "trace_format.hpp"
  38
  39
  40 namespace Trace {
  41
  42
  43 void
  44 LocalWriter::open(void) {
  45
  46     static unsigned dwCounter = 0;
  47
  48     const char *szExtension = "trace";
  49     char szFileName[PATH_MAX];
  50     const char *lpFileName;
  51
  52     lpFileName = getenv("TRACE_FILE");
  53     if (lpFileName) {
  54         strncpy(szFileName, lpFileName, PATH_MAX);
  55     }
  56     else {
  57         char szProcessName[PATH_MAX];
  58         char szCurrentDir[PATH_MAX];
  59         OS::GetProcessName(szProcessName, PATH_MAX);
  60         OS::GetCurrentDir(szCurrentDir, PATH_MAX);
  61
  62         for (;;) {
  63             FILE *file;
  64
  65             if (dwCounter)
  66                 snprintf(szFileName, PATH_MAX, "%s%c%s.%u.%s", szCurrentDir, PATH_SEP, szProcessName, dwCounter, szExtension);
  67             else
  68                 snprintf(szFileName, PATH_MAX, "%s%c%s.%s", szCurrentDir, PATH_SEP, szProcessName, szExtension);
  69
  70             file = fopen(szFileName, "rb");
  71             if (file == NULL)
  72                 break;
  73
  74             fclose(file);
  75
  76             ++dwCounter;
  77         }
  78     }
  79
  80     OS::DebugMessage("apitrace: tracing to %s\n", szFileName);
  81
  82     Writer::open(szFileName);
  83 }
  84
  85 unsigned LocalWriter::beginEnter(const FunctionSig *sig) {
  86     OS::AcquireMutex();
  87
  88     if (!g_gzFile) {
  89         open();
  90     }
  91
  92     return Writer::beginEnter(sig);
  93 }
  94
  95 void LocalWriter::endEnter(void) {
  96     Writer::endEnter();
  97     gzflush(g_gzFile, Z_SYNC_FLUSH);
  98     OS::ReleaseMutex();
  99 }
 100
 101 void LocalWriter::beginLeave(unsigned call) {
 102     OS::AcquireMutex();
 103     Writer::beginLeave(call);
 104 }
 105
 106 void LocalWriter::endLeave(void) {
 107     Writer::endLeave();
 108     gzflush(g_gzFile, Z_SYNC_FLUSH);
 109     OS::ReleaseMutex();
 110 }
 111
 112
 113 } /* namespace Trace */
 114