| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright (C) 2011 Apple Inc. All rights reserved. | |
| 3 * | |
| 4 * Redistribution and use in source and binary forms, with or without | |
| 5 * modification, are permitted provided that the following conditions | |
| 6 * are met: | |
| 7 * 1. Redistributions of source code must retain the above copyright | |
| 8 * notice, this list of conditions and the following disclaimer. | |
| 9 * 2. Redistributions in binary form must reproduce the above copyright | |
| 10 * notice, this list of conditions and the following disclaimer in the | |
| 11 * documentation and/or other materials provided with the distribution. | |
| 12 * | |
| 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' | |
| 14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, | |
| 15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR | |
| 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS | |
| 17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | |
| 18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | |
| 19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | |
| 20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | |
| 21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | |
| 22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF | |
| 23 * THE POSSIBILITY OF SUCH DAMAGE. | |
| 24 */ | |
| 25 | |
| 26 #include "config.h" | |
| 27 #include "DelayLoadedModulesEnumerator.h" | |
| 28 | |
| 29 #include "ImportedFunctionsEnumerator.h" | |
| 30 #include <delayimp.h> | |
| 31 | |
| 32 // See <http://msdn.microsoft.com/en-us/magazine/cc301808.aspx> and | |
| 33 // <http://msdn.microsoft.com/en-us/windows/hardware/gg463119> for more informat
ion about the PE | |
| 34 // image format. | |
| 35 | |
| 36 namespace WebCore { | |
| 37 | |
| 38 DelayLoadedModulesEnumerator::DelayLoadedModulesEnumerator(const PEImage& image) | |
| 39 : m_image(image) | |
| 40 { | |
| 41 if (m_image.isValid()) | |
| 42 m_descriptor = static_cast<const ImgDelayDescr*>(m_image.dataDirectoryEn
tryAddress(IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT)); | |
| 43 else | |
| 44 m_descriptor = 0; | |
| 45 } | |
| 46 | |
| 47 bool DelayLoadedModulesEnumerator::isAtEnd() const | |
| 48 { | |
| 49 return !m_descriptor || !m_descriptor->rvaHmod; | |
| 50 } | |
| 51 | |
| 52 void DelayLoadedModulesEnumerator::next() | |
| 53 { | |
| 54 ASSERT(!isAtEnd()); | |
| 55 ++m_descriptor; | |
| 56 } | |
| 57 | |
| 58 const char* DelayLoadedModulesEnumerator::currentModuleName() const | |
| 59 { | |
| 60 ASSERT(!isAtEnd()); | |
| 61 return static_cast<const char*>(convertPotentialRVAToAddress(m_descriptor->r
vaDLLName)); | |
| 62 } | |
| 63 | |
| 64 ImportedFunctionsEnumerator DelayLoadedModulesEnumerator::functionsEnumerator()
const | |
| 65 { | |
| 66 ASSERT(!isAtEnd()); | |
| 67 | |
| 68 const IMAGE_THUNK_DATA* importNameTable = static_cast<const IMAGE_THUNK_DATA
*>(convertPotentialRVAToAddress(m_descriptor->rvaINT)); | |
| 69 const IMAGE_THUNK_DATA* importAddressTable = static_cast<const IMAGE_THUNK_D
ATA*>(convertPotentialRVAToAddress(m_descriptor->rvaIAT)); | |
| 70 | |
| 71 return ImportedFunctionsEnumerator(m_image, importNameTable, importAddressTa
ble); | |
| 72 } | |
| 73 | |
| 74 const void* DelayLoadedModulesEnumerator::convertPotentialRVAToAddress(DWORD pot
entialRVA) const | |
| 75 { | |
| 76 ASSERT(!isAtEnd()); | |
| 77 | |
| 78 if (!(m_descriptor->grAttrs & dlattrRva)) { | |
| 79 // m_image is a pre-VC7.0 image, so addresses stored in the ImageDelayDe
scr are non-relative. | |
| 80 return reinterpret_cast<void*>(potentialRVA); | |
| 81 } | |
| 82 | |
| 83 return m_image.convertRVAToAddress(potentialRVA); | |
| 84 } | |
| 85 | |
| 86 } // namespace WebCore | |
| OLD | NEW |