VirtualBox

source: vbox/trunk/src/VBox/Runtime/r3/win/ldrNative-win.cpp@ 57671

Last change on this file since 57671 was 57358, checked in by vboxsync, 9 years ago

*: scm cleanup run.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id Revision
File size: 6.4 KB
Line 
1/* $Id: ldrNative-win.cpp 57358 2015-08-14 15:16:38Z vboxsync $ */
2/** @file
3 * IPRT - Binary Image Loader, Win32 native.
4 */
5
6/*
7 * Copyright (C) 2006-2015 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 *
17 * The contents of this file may alternatively be used under the terms
18 * of the Common Development and Distribution License Version 1.0
19 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
20 * VirtualBox OSE distribution, in which case the provisions of the
21 * CDDL are applicable instead of those of the GPL.
22 *
23 * You may elect to license modified versions of this file under the
24 * terms and conditions of either the GPL or the CDDL or both.
25 */
26
27
28/*********************************************************************************************************************************
29* Header Files *
30*********************************************************************************************************************************/
31#define LOG_GROUP RTLOGGROUP_LDR
32#include <Windows.h>
33
34#include <iprt/ldr.h>
35#include "internal/iprt.h"
36
37#include <iprt/alloca.h>
38#include <iprt/assert.h>
39#include <iprt/err.h>
40#include <iprt/file.h>
41#include <iprt/log.h>
42#include <iprt/path.h>
43#include <iprt/string.h>
44
45#include <iprt/once.h>
46#include <iprt/string.h>
47#include "internal/ldr.h"
48
49#ifndef LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR
50# define LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR UINT32_C(0x100)
51# define LOAD_LIBRARY_SEARCH_APPLICATION_DIR UINT32_C(0x200)
52# define LOAD_LIBRARY_SEARCH_SYSTEM32 UINT32_C(0x800)
53#endif
54
55
56int rtldrNativeLoad(const char *pszFilename, uintptr_t *phHandle, uint32_t fFlags, PRTERRINFO pErrInfo)
57{
58 Assert(sizeof(*phHandle) >= sizeof(HMODULE));
59 AssertReturn(!(fFlags & RTLDRLOAD_FLAGS_GLOBAL), VERR_INVALID_FLAGS);
60 AssertLogRelMsgReturn(RTPathStartsWithRoot(pszFilename), /* Relative names will still be applied to the search path. */
61 ("pszFilename='%s'\n", pszFilename),
62 VERR_INTERNAL_ERROR_2);
63
64 /*
65 * Convert to UTF-16 and make sure it got a .DLL suffix.
66 */
67 int rc;
68 RTUTF16 *pwszNative = NULL;
69 if (RTPathHasSuffix(pszFilename))
70 rc = RTStrToUtf16(pszFilename, &pwszNative);
71 else
72 {
73 size_t cwcAlloc;
74 rc = RTStrCalcUtf16LenEx(pszFilename, RTSTR_MAX, &cwcAlloc);
75 if (RT_SUCCESS(rc))
76 {
77 cwcAlloc += sizeof(".DLL");
78 pwszNative = RTUtf16Alloc(cwcAlloc * sizeof(RTUTF16));
79 if (pwszNative)
80 {
81 size_t cwcNative;
82 rc = RTStrToUtf16Ex(pszFilename, RTSTR_MAX, &pwszNative, cwcAlloc, &cwcNative);
83 if (RT_SUCCESS(rc))
84 rc = RTUtf16CopyAscii(&pwszNative[cwcNative], cwcAlloc - cwcNative, ".DLL");
85 }
86 else
87 rc = VERR_NO_UTF16_MEMORY;
88 }
89 }
90 if (RT_SUCCESS(rc))
91 {
92 /*
93 * Attempt load.
94 */
95 HMODULE hmod;
96 static int s_iSearchDllLoadDirSupported = 0;
97 if ( !(fFlags & RTLDRLOAD_FLAGS_NT_SEARCH_DLL_LOAD_DIR)
98 || s_iSearchDllLoadDirSupported < 0)
99 hmod = LoadLibraryExW(pwszNative, NULL, 0);
100 else
101 {
102 hmod = LoadLibraryExW(pwszNative, NULL, LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR | LOAD_LIBRARY_SEARCH_SYSTEM32
103 | LOAD_LIBRARY_SEARCH_APPLICATION_DIR);
104 if (s_iSearchDllLoadDirSupported == 0)
105 {
106 if (hmod != NULL || GetLastError() != ERROR_INVALID_PARAMETER)
107 s_iSearchDllLoadDirSupported = 1;
108 else
109 {
110 s_iSearchDllLoadDirSupported = -1;
111 hmod = LoadLibraryExW(pwszNative, NULL, 0);
112 }
113 }
114 }
115 if (hmod)
116 {
117 *phHandle = (uintptr_t)hmod;
118 RTUtf16Free(pwszNative);
119 return VINF_SUCCESS;
120 }
121
122 /*
123 * Try figure why it failed to load.
124 */
125 DWORD dwErr = GetLastError();
126 rc = RTErrConvertFromWin32(dwErr);
127 rc = RTErrInfoSetF(pErrInfo, rc, "GetLastError=%u", dwErr);
128 }
129 else
130 rc = RTErrInfoSetF(pErrInfo, rc, "Error converting UTF-8 to UTF-16 string.");
131 RTUtf16Free(pwszNative);
132 return rc;
133}
134
135
136DECLCALLBACK(int) rtldrNativeGetSymbol(PRTLDRMODINTERNAL pMod, const char *pszSymbol, void **ppvValue)
137{
138 PRTLDRMODNATIVE pModNative = (PRTLDRMODNATIVE)pMod;
139 FARPROC pfn = GetProcAddress((HMODULE)pModNative->hNative, pszSymbol);
140 if (pfn)
141 {
142 *ppvValue = (void *)pfn;
143 return VINF_SUCCESS;
144 }
145 *ppvValue = NULL;
146 return RTErrConvertFromWin32(GetLastError());
147}
148
149
150DECLCALLBACK(int) rtldrNativeClose(PRTLDRMODINTERNAL pMod)
151{
152 PRTLDRMODNATIVE pModNative = (PRTLDRMODNATIVE)pMod;
153 if ( (pModNative->fFlags & RTLDRLOAD_FLAGS_NO_UNLOAD)
154 || FreeLibrary((HMODULE)pModNative->hNative))
155 {
156 pModNative->hNative = (uintptr_t)INVALID_HANDLE_VALUE;
157 return VINF_SUCCESS;
158 }
159 return RTErrConvertFromWin32(GetLastError());
160}
161
162
163int rtldrNativeLoadSystem(const char *pszFilename, const char *pszExt, uint32_t fFlags, PRTLDRMOD phLdrMod)
164{
165 /*
166 * We only try the System32 directory.
167 */
168 WCHAR wszSysDir[MAX_PATH];
169 UINT cwcSysDir = GetSystemDirectoryW(wszSysDir, MAX_PATH);
170 if (cwcSysDir >= MAX_PATH)
171 return VERR_FILENAME_TOO_LONG;
172
173 char szPath[RTPATH_MAX];
174 char *pszPath = szPath;
175 int rc = RTUtf16ToUtf8Ex(wszSysDir, RTSTR_MAX, &pszPath, sizeof(szPath), NULL);
176 if (RT_SUCCESS(rc))
177 {
178 rc = RTPathAppend(szPath, sizeof(szPath), pszFilename);
179 if (pszExt && RT_SUCCESS(rc))
180 rc = RTStrCat(szPath, sizeof(szPath), pszExt);
181 if (RT_SUCCESS(rc))
182 {
183 if (RTFileExists(szPath))
184 rc = RTLdrLoadEx(szPath, phLdrMod, fFlags, NULL);
185 else
186 rc = VERR_MODULE_NOT_FOUND;
187 }
188 }
189
190 return rc;
191}
192
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette