michael@0: // Copyright (c) 2010, Google Inc.
michael@0: // All rights reserved.
michael@0: //
michael@0: // Redistribution and use in source and binary forms, with or without
michael@0: // modification, are permitted provided that the following conditions are
michael@0: // met:
michael@0: //
michael@0: //     * Redistributions of source code must retain the above copyright
michael@0: // notice, this list of conditions and the following disclaimer.
michael@0: //     * Redistributions in binary form must reproduce the above
michael@0: // copyright notice, this list of conditions and the following disclaimer
michael@0: // in the documentation and/or other materials provided with the
michael@0: // distribution.
michael@0: //     * Neither the name of Google Inc. nor the names of its
michael@0: // contributors may be used to endorse or promote products derived from
michael@0: // this software without specific prior written permission.
michael@0: //
michael@0: // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
michael@0: // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
michael@0: // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
michael@0: // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
michael@0: // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
michael@0: // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
michael@0: // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
michael@0: // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
michael@0: // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
michael@0: // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
michael@0: // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
michael@0: 
michael@0: #ifndef CLIENT_WINDOWS_UNITTESTS_DUMP_ANALYSIS_H_
michael@0: #define CLIENT_WINDOWS_UNITTESTS_DUMP_ANALYSIS_H_
michael@0: 
michael@0: #include "client/windows/crash_generation/minidump_generator.h"
michael@0: 
michael@0: // Convenience to get to the PEB pointer in a TEB.
michael@0: struct FakeTEB {
michael@0:   char dummy[0x30];
michael@0:   void* peb;
michael@0: };
michael@0: 
michael@0: class DumpAnalysis {
michael@0:  public:
michael@0:   explicit DumpAnalysis(const std::wstring& file_path)
michael@0:       : dump_file_(file_path), dump_file_view_(NULL), dump_file_mapping_(NULL),
michael@0:         dump_file_handle_(NULL) {
michael@0:     EnsureDumpMapped();
michael@0:   }
michael@0:   ~DumpAnalysis();
michael@0: 
michael@0:   bool HasStream(ULONG stream_number) const;
michael@0: 
michael@0:   // This is template to keep type safety in the front, but we end up casting
michael@0:   // to void** inside the implementation to pass the pointer to Win32. So
michael@0:   // casting here is considered safe.
michael@0:   template <class StreamType>
michael@0:   size_t GetStream(ULONG stream_number, StreamType** stream) const {
michael@0:     return GetStreamImpl(stream_number, reinterpret_cast<void**>(stream));
michael@0:   }
michael@0: 
michael@0:   bool HasTebs() const;
michael@0:   bool HasPeb() const;
michael@0:   bool HasMemory(ULONG64 address) const {
michael@0:     return HasMemory<BYTE>(address, NULL);
michael@0:   }
michael@0: 
michael@0:   bool HasMemory(const void* address) const {
michael@0:     return HasMemory<BYTE>(address, NULL);
michael@0:   }
michael@0: 
michael@0:   template <class StructureType>
michael@0:   bool HasMemory(ULONG64 address, StructureType** structure = NULL) const {
michael@0:     // We can't cope with 64 bit addresses for now.
michael@0:     if (address > 0xFFFFFFFFUL)
michael@0:       return false;
michael@0: 
michael@0:     return HasMemory(reinterpret_cast<void*>(address), structure);
michael@0:   }
michael@0: 
michael@0:   template <class StructureType>
michael@0:   bool HasMemory(const void* addr_in, StructureType** structure = NULL) const {
michael@0:     return HasMemoryImpl(addr_in, sizeof(StructureType),
michael@0:                              reinterpret_cast<void**>(structure));
michael@0:   }
michael@0: 
michael@0:  protected:
michael@0:   void EnsureDumpMapped();
michael@0: 
michael@0:   HANDLE dump_file_mapping_;
michael@0:   HANDLE dump_file_handle_;
michael@0:   void* dump_file_view_;
michael@0:   std::wstring dump_file_;
michael@0: 
michael@0:  private:
michael@0:   // This is the implementation of GetStream<>.
michael@0:   size_t GetStreamImpl(ULONG stream_number, void** stream) const;
michael@0: 
michael@0:   // This is the implementation of HasMemory<>.
michael@0:   bool HasMemoryImpl(const void* addr_in, size_t pointersize,
michael@0:                      void** structure) const;
michael@0: };
michael@0: 
michael@0: #endif  // CLIENT_WINDOWS_UNITTESTS_DUMP_ANALYSIS_H_