1 | /* $Id: myldrCheckInternalName.cpp,v 1.5 2001-07-07 04:39:57 bird Exp $
|
---|
2 | *
|
---|
3 | * ldrCheckInternalName - ldrCheckInternalName replacement with support for
|
---|
4 | * long DLL names and no .DLL-extention dependency.
|
---|
5 | *
|
---|
6 | * Copyright (c) 1999-2000 knut st. osmundsen (knut.stange.osmundsen@mynd.no)
|
---|
7 | *
|
---|
8 | * Project Odin Software License can be found in LICENSE.TXT
|
---|
9 | *
|
---|
10 | */
|
---|
11 |
|
---|
12 |
|
---|
13 | /*******************************************************************************
|
---|
14 | * Defined Constants And Macros *
|
---|
15 | *******************************************************************************/
|
---|
16 | #define INCL_DOSERRORS
|
---|
17 | #define INCL_NOPMAPI
|
---|
18 | #define INCL_OS2KRNL_LDR
|
---|
19 |
|
---|
20 |
|
---|
21 | /*******************************************************************************
|
---|
22 | * Header Files *
|
---|
23 | *******************************************************************************/
|
---|
24 | #include <os2.h>
|
---|
25 |
|
---|
26 | #include <memory.h>
|
---|
27 | #include <stdlib.h>
|
---|
28 | #include <string.h>
|
---|
29 |
|
---|
30 | #include "devSegDf.h" /* Win32k segment definitions. */
|
---|
31 | #include "log.h"
|
---|
32 | #include "avl.h"
|
---|
33 | #include <peexe.h>
|
---|
34 | #include <exe386.h>
|
---|
35 | #include "OS2Krnl.h"
|
---|
36 | #include "dev32.h"
|
---|
37 | #include "ldr.h"
|
---|
38 | #include "options.h"
|
---|
39 |
|
---|
40 |
|
---|
41 | /**
|
---|
42 | * Checks if the internal name (first name in the resident nametable) matches
|
---|
43 | * the filename.
|
---|
44 | *
|
---|
45 | * This replacement will support DLLs with non DLL extention
|
---|
46 | * and names that are longer than 8 chars.
|
---|
47 | *
|
---|
48 | * @returns NO_ERROR on success (the name matched).
|
---|
49 | * ERROR_INVALID_NAME if mismatch.
|
---|
50 | * @param pMTE Pointer to the MTE of the module to check.<br>
|
---|
51 | * Assumes! that the filename for this module is present in ldrpFileNameBuf.
|
---|
52 | * @sketch Check if library module - this check only applies to library modules (ie. DLLs).
|
---|
53 | * Uppercase filename.
|
---|
54 | * Parse filename - get the length of the name including any extention different from .DLL.
|
---|
55 | * Compare internal name and filename returning NO_ERROR if matches and errorcode if mismatch.
|
---|
56 | * If this module is not in the GLOBAL CLASS we don't compare the extention.
|
---|
57 | *
|
---|
58 | * @remark This function will have to change slightly when we're starting to
|
---|
59 | * support ELF shared libraries (ie. .so-files).
|
---|
60 | */
|
---|
61 | ULONG LDRCALL myldrCheckInternalName(PMTE pMTE)
|
---|
62 | {
|
---|
63 | /* Check if this feature is enabled */
|
---|
64 | if (isDllFixesDisabled())
|
---|
65 | {
|
---|
66 | #ifdef DEBUG
|
---|
67 | APIRET rc = ldrCheckInternalName(pMTE);
|
---|
68 | kprintf(("myldrCheckInternalName: pMTE=0x%08x intname=%.*s path=%s rc=%d\n", /* (original)\",*/
|
---|
69 | pMTE, *(PCHAR)pMTE->mte_swapmte->smte_restab, (PCHAR)pMTE->mte_swapmte->smte_restab + 1, ldrpFileNameBuf, rc));
|
---|
70 | return rc;
|
---|
71 | #else
|
---|
72 | return ldrCheckInternalName(pMTE);
|
---|
73 | #endif
|
---|
74 | }
|
---|
75 |
|
---|
76 | /* Local Variables */
|
---|
77 | PCHAR pachName; /* Pointer to the name part of pachFilename. */
|
---|
78 | int cchName; /* Length of the name part of pachFilename.
|
---|
79 | * Includes extention if extention is not .DLL.
|
---|
80 | * this is the length relative to pachName used to match the internal name. */
|
---|
81 | PCHAR pachExt; /* Pointer to the extention part of pachFilename. (not dot!) */
|
---|
82 | int cchExt; /* Length of the extention part of pachFilename. (not dot!) */
|
---|
83 | PCHAR pachResName; /* Pointer to the internal name - resname.0 */
|
---|
84 | APIRET rc; /* Return code. */
|
---|
85 |
|
---|
86 |
|
---|
87 | /*
|
---|
88 | * Return successfully if not library module.
|
---|
89 | */
|
---|
90 | if (!(pMTE->mte_flags1 & LIBRARYMOD))
|
---|
91 | return NO_ERROR;
|
---|
92 |
|
---|
93 |
|
---|
94 | /*
|
---|
95 | * Uppercase and parse filename in ldrpFileNameBuf
|
---|
96 | */
|
---|
97 | cchName = (int)ldrGetFileName2(ldrpFileNameBuf, (PCHAR*)SSToDS(&pachName), (PCHAR*)SSToDS(&pachExt));
|
---|
98 | cchExt = (pachExt) ? strlen(pachExt) : 0;
|
---|
99 | ldrUCaseString(pachName, cchName + cchExt + 1);
|
---|
100 | pachResName = (PCHAR)pMTE->mte_swapmte->smte_restab;
|
---|
101 |
|
---|
102 | /*
|
---|
103 | * Do the compare - DllFix case or standard case.
|
---|
104 | */
|
---|
105 | if ( (cchName > 8 && *pachResName > 8)
|
---|
106 | || ( (pMTE->mte_flags1 & CLASS_MASK) == CLASS_GLOBAL
|
---|
107 | && (cchExt != 3 || memcmp(pachExt, "DLL", 3)) /* Extention != DLL. */
|
---|
108 | )
|
---|
109 | )
|
---|
110 | { /*
|
---|
111 | * Rules for long DLL names or GLOBAL dlls with extention <> DLL:
|
---|
112 | * 1. If DLL extention, the internal name don't need to have an extention,
|
---|
113 | * but it could have.
|
---|
114 | * 2. If not DLL extention, then internal name must have an extention.
|
---|
115 | * 3. If no extention the internal name should end with a '.'.
|
---|
116 | */
|
---|
117 | if (pachExt != NULL && cchExt == 3 && !memcmp(pachExt, "DLL", 3)) /* DLL extention. */
|
---|
118 | { /* (1) */
|
---|
119 | rc =( ( *pachResName == cchName
|
---|
120 | || *pachResName == cchName + cchExt + 1)
|
---|
121 | && !memcmp(pachResName + 1, pachName, *pachResName)
|
---|
122 | );
|
---|
123 | }
|
---|
124 | else if (cchExt > 0) /* Extention. */
|
---|
125 | { /* (2) */
|
---|
126 | rc =( *pachResName == cchName + cchExt + 1
|
---|
127 | && !memcmp(pachResName + 1, pachName, *pachResName)
|
---|
128 | );
|
---|
129 | } /* No extetion. */
|
---|
130 | else
|
---|
131 | { /* (3) */
|
---|
132 | rc =( *pachResName == cchName + 1
|
---|
133 | && pachResName[cchName + 1] == '.'
|
---|
134 | && !memcmp(pachResName + 1, pachName, cchName)
|
---|
135 | );
|
---|
136 | }
|
---|
137 | rc = (rc) ? NO_ERROR : ERROR_INVALID_NAME;
|
---|
138 | }
|
---|
139 | else
|
---|
140 | { /*
|
---|
141 | * Rules for short DLL names. ( < 8 chars):
|
---|
142 | * 1. The internal name must match the DLL name.
|
---|
143 | * 2b. If the DLL name is 8 chars the internal name could have extra chars (but we don't check).
|
---|
144 | * (This is a feature/bug.)
|
---|
145 | * 2a. If the DLL name is less than 8 chars the internal name should match exactly.
|
---|
146 | */
|
---|
147 | #if 0
|
---|
148 | /* This was the way it should be implemented, but code is buggy.
|
---|
149 | * Current code works like this:
|
---|
150 | * rc =( memcmp(pachName, pMTE->mte_modname, cchName)
|
---|
151 | * && ( cchName == 8
|
---|
152 | * || pMTE->mte_modname[cchName] == '\0'
|
---|
153 | * )
|
---|
154 | * ) ? ERROR_INVALID_NAME : NO_ERROR;
|
---|
155 | *
|
---|
156 | * This is so old that it has become an persistant bug in some ways.
|
---|
157 | * The correct check will break Lotus Freelance for example.
|
---|
158 | * But, the applications which are broken all seems to include the
|
---|
159 | * .DLL extention in the internal name (and have length which is
|
---|
160 | * shorter than 8 chars).
|
---|
161 | * So, a correction will simply be to remove any .DLL extention
|
---|
162 | * of the internal name before setting it ldrCreateMte. This fix
|
---|
163 | * could always be done here too.
|
---|
164 | *
|
---|
165 | * BTW. I managed to exploit this bug to replace doscall1.dll.
|
---|
166 | * Which is very nasty!
|
---|
167 | */
|
---|
168 | rc =( !memcmp(pachName, pMTE->mte_modname, cchName) /* (1) */
|
---|
169 | && ( cchName == 8 /* (2a) */
|
---|
170 | || pMTE->mte_modname[cchName] == '\0' /* (2b) */
|
---|
171 | )
|
---|
172 | ) ? NO_ERROR : ERROR_INVALID_NAME;
|
---|
173 | #else
|
---|
174 | /* For the issue of compatibly with the bug we'll call the real function. */
|
---|
175 | rc = ldrCheckInternalName(pMTE);
|
---|
176 | #endif
|
---|
177 | }
|
---|
178 |
|
---|
179 |
|
---|
180 | /*
|
---|
181 | * Log answer and return it.
|
---|
182 | */
|
---|
183 | kprintf(("myldrCheckInternalName: pMTE=0x%08x intname=%.*s path=%s rc=%d\n",
|
---|
184 | pMTE, *(PCHAR)pMTE->mte_swapmte->smte_restab, (PCHAR)pMTE->mte_swapmte->smte_restab + 1, ldrpFileNameBuf, rc));
|
---|
185 |
|
---|
186 | return rc;
|
---|
187 | }
|
---|
188 |
|
---|