initial import
[vuplus_webkit] / Source / WebCore / platform / win / ImportedFunctionsEnumerator.cpp
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 "ImportedFunctionsEnumerator.h"
28
29 // See <http://msdn.microsoft.com/en-us/magazine/cc301808.aspx> and
30 // <http://msdn.microsoft.com/en-us/windows/hardware/gg463119> for more information about the PE
31 // image format.
32
33 namespace WebCore {
34
35 ImportedFunctionsEnumerator::ImportedFunctionsEnumerator(const PEImage& image, const IMAGE_THUNK_DATA* importNameTable, const IMAGE_THUNK_DATA* importAddressTable)
36     : m_image(image)
37     , m_nameTableEntry(importNameTable)
38     , m_addressTableEntry(importAddressTable)
39 {
40     ASSERT(!importNameTable == !importAddressTable);
41 }
42
43 bool ImportedFunctionsEnumerator::isAtEnd() const
44 {
45     ASSERT(!m_nameTableEntry || !m_nameTableEntry->u1.AddressOfData == !m_addressTableEntry->u1.Function);
46     return !m_nameTableEntry || !m_nameTableEntry->u1.AddressOfData;
47 }
48
49 void ImportedFunctionsEnumerator::next()
50 {
51     ASSERT(!isAtEnd());
52     ++m_nameTableEntry;
53     ++m_addressTableEntry;
54 }
55
56 const char* ImportedFunctionsEnumerator::currentFunctionName() const
57 {
58     ASSERT(m_nameTableEntry);
59
60     // Ordinal imports have no name.
61     if (IMAGE_SNAP_BY_ORDINAL(m_nameTableEntry->u1.Ordinal))
62         return 0;
63
64     const IMAGE_IMPORT_BY_NAME* importByName = static_cast<const IMAGE_IMPORT_BY_NAME*>(m_image.convertRVAToAddress(m_nameTableEntry->u1.AddressOfData));
65     return reinterpret_cast<const char*>(importByName->Name);
66 }
67
68 const void* const* ImportedFunctionsEnumerator::addressOfCurrentFunctionPointer() const
69 {
70     ASSERT(m_addressTableEntry);
71     COMPILE_ASSERT(sizeof(void*) == sizeof(m_addressTableEntry->u1.Function), FunctionAddressSizeMatchesPointerSize);
72     return reinterpret_cast<const void* const*>(&m_addressTableEntry->u1.Function);
73 }
74
75 } // namespace WebCore