1 /****************************************************************************
2 * Copyright (C) 2016 Intel Corporation. All Rights Reserved.
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
25 * @brief Event handler interface. auto-generated file
29 ******************************************************************************/
32 #include "common/os.h"
33 #include "${event_header}"
39 //////////////////////////////////////////////////////////////////////////
40 /// EventHandlerFile - interface for handling events.
41 //////////////////////////////////////////////////////////////////////////
42 class EventHandlerFile : public EventHandler
45 EventHandlerFile(uint32_t id)
49 DWORD pid = GetCurrentProcessId();
50 TCHAR procname[MAX_PATH];
51 GetModuleFileName(NULL, procname, MAX_PATH);
52 const char* pBaseName = strrchr(procname, '\\');
53 std::stringstream outDir;
54 outDir << KNOB_DEBUG_OUTPUT_DIR << pBaseName << "_" << pid << std::ends;
55 CreateDirectory(outDir.str().c_str(), NULL);
58 // There could be multiple threads creating thread pools. We
59 // want to make sure they are uniquly identified by adding in
60 // the creator's thread id into the filename.
61 sprintf(buf, "%s\\ar_event%d_%d.bin", outDir.str().c_str(), GetCurrentThreadId(), id);
62 mFilename = std::string(buf);
65 // There could be multiple threads creating thread pools. We
66 // want to make sure they are uniquly identified by adding in
67 // the creator's thread id into the filename.
68 sprintf(buf, "%s/ar_event%d_%d.bin", "/tmp", GetCurrentThreadId(), id);
69 mFilename = std::string(buf);
73 virtual ~EventHandlerFile()
78 //////////////////////////////////////////////////////////////////////////
79 /// @brief Flush buffer to file.
84 if (mBufOffset == mHeaderBufOffset)
86 // Nothing to flush. Only header has been generated.
91 file.open(mFilename, std::ios::out | std::ios::app | std::ios::binary);
95 SWR_INVALID("ArchRast: Could not open event file!");
99 file.write((char*)mBuffer, mBufOffset);
103 mHeaderBufOffset = 0; // Reset header offset so its no longer considered.
108 //////////////////////////////////////////////////////////////////////////
109 /// @brief Write event and its payload to the memory buffer.
110 void Write(uint32_t eventId, const char* pBlock, uint32_t size)
112 if ((mBufOffset + size + sizeof(eventId)) > mBufferSize)
116 // Don't corrupt what's already in the buffer?
117 /// @todo Maybe add corrupt marker to buffer here in case we can open file in future?
122 memcpy(&mBuffer[mBufOffset], (char*)&eventId, sizeof(eventId));
123 mBufOffset += sizeof(eventId);
124 memcpy(&mBuffer[mBufOffset], pBlock, size);
128 % for name in protos['event_names']:
129 //////////////////////////////////////////////////////////////////////////
130 /// @brief Handle ${name} event
131 virtual void Handle(const ${name}& event)
133 % if protos['events'][name]['num_fields'] == 0:
134 Write(${protos['events'][name]['event_id']}, (char*)&event.data, 0);
136 Write(${protos['events'][name]['event_id']}, (char*)&event.data, sizeof(event.data));
141 //////////////////////////////////////////////////////////////////////////
142 /// @brief Everything written to buffer this point is the header.
143 virtual void MarkHeader()
145 mHeaderBufOffset = mBufOffset;
148 std::string mFilename;
150 static const uint32_t mBufferSize = 1024;
151 uint8_t mBuffer[mBufferSize];
152 uint32_t mBufOffset{0};
153 uint32_t mHeaderBufOffset{0};