| 1 | /* $Id: d32init.c,v 1.4 1999-11-10 01:45:30 bird Exp $
|
|---|
| 2 | *
|
|---|
| 3 | * d32init.c - 32-bits init routines.
|
|---|
| 4 | *
|
|---|
| 5 | * Copyright (c) 1998-1999 knut st. osmundsen
|
|---|
| 6 | *
|
|---|
| 7 | * Project Odin Software License can be found in LICENSE.TXT
|
|---|
| 8 | *
|
|---|
| 9 | */
|
|---|
| 10 |
|
|---|
| 11 | /*******************************************************************************
|
|---|
| 12 | * Defined Constants *
|
|---|
| 13 | *******************************************************************************/
|
|---|
| 14 | #define MAXSIZE_PROLOG 0x10 /* Note that this must be synced with */
|
|---|
| 15 | /* the one used in calltab.asm. */
|
|---|
| 16 | #define static
|
|---|
| 17 |
|
|---|
| 18 | #define INCL_DOSERRORS
|
|---|
| 19 | #define INCL_NOPMAPI
|
|---|
| 20 | #define LDR_INCL_INITONLY
|
|---|
| 21 |
|
|---|
| 22 | /*******************************************************************************
|
|---|
| 23 | * Header Files *
|
|---|
| 24 | *******************************************************************************/
|
|---|
| 25 | #include <os2.h>
|
|---|
| 26 |
|
|---|
| 27 | #include <string.h>
|
|---|
| 28 |
|
|---|
| 29 | #include "OS2Krnl.h"
|
|---|
| 30 | #include "options.h"
|
|---|
| 31 | #include "dev1632.h"
|
|---|
| 32 | #include "dev32.h"
|
|---|
| 33 | #include "probkrnl.h"
|
|---|
| 34 | #include "log.h"
|
|---|
| 35 | #include "asmutils.h"
|
|---|
| 36 | #include "malloc.h"
|
|---|
| 37 | #include "ldr.h"
|
|---|
| 38 | #include "ldrCalls.h"
|
|---|
| 39 |
|
|---|
| 40 |
|
|---|
| 41 | /*******************************************************************************
|
|---|
| 42 | * Internal Functions *
|
|---|
| 43 | *******************************************************************************/
|
|---|
| 44 | static ULONG readnum(const char *pszNum);
|
|---|
| 45 | static int interpretFunctionProlog(char *p);
|
|---|
| 46 | static int procInit(void);
|
|---|
| 47 |
|
|---|
| 48 |
|
|---|
| 49 | /* externs located in 16-bit data segement */
|
|---|
| 50 | extern ULONG _TKSSBase16;
|
|---|
| 51 | extern USHORT _R0FlatCS16;
|
|---|
| 52 | extern USHORT _R0FlatDS16;
|
|---|
| 53 |
|
|---|
| 54 |
|
|---|
| 55 | /* extern(s) located in calltab.asm */
|
|---|
| 56 | extern char callTab[NUMBER_OF_PROCS][MAXSIZE_PROLOG];
|
|---|
| 57 |
|
|---|
| 58 |
|
|---|
| 59 | /**
|
|---|
| 60 | * Ring-0, 32-bit, init function.
|
|---|
| 61 | * @returns Status word.
|
|---|
| 62 | * @param pRpInit Pointer init request packet.
|
|---|
| 63 | * @sketch Set TKSSBase32.
|
|---|
| 64 | * Set default parameters.
|
|---|
| 65 | * Parse command line options.
|
|---|
| 66 | * Show (kprint) configuration.
|
|---|
| 67 | * Init heap.
|
|---|
| 68 | * Init ldr.
|
|---|
| 69 | * Init procs. (overloaded ldr procedures)
|
|---|
| 70 | * @status completely implemented.
|
|---|
| 71 | * @author knut st. osmundsen
|
|---|
| 72 | */
|
|---|
| 73 | USHORT _loadds _Far32 _Pascal R0Init32(RP32INIT *pRpInit)
|
|---|
| 74 | {
|
|---|
| 75 | char *pszTmp2;
|
|---|
| 76 | char *pszTmp;
|
|---|
| 77 | ULONG ul;
|
|---|
| 78 |
|
|---|
| 79 | pulTKSSBase32 = (PULONG)_TKSSBase16;
|
|---|
| 80 |
|
|---|
| 81 | SET_OPTIONS_TO_DEFAULT(options);
|
|---|
| 82 |
|
|---|
| 83 | /*---------------------*/
|
|---|
| 84 | /* commandline options */
|
|---|
| 85 | /*---------------------*/
|
|---|
| 86 | kprintf(("Options start\n"));
|
|---|
| 87 | pszTmp = strpbrk(pRpInit->InitArgs, "-/");
|
|---|
| 88 | while (pszTmp != NULL)
|
|---|
| 89 | {
|
|---|
| 90 | char cch;
|
|---|
| 91 | pszTmp++; //skip [-/]
|
|---|
| 92 | cch = strlen(pszTmp);
|
|---|
| 93 | switch (*pszTmp)
|
|---|
| 94 | {
|
|---|
| 95 | case 'c':
|
|---|
| 96 | case 'C': /* -C[1|2] - com-port no, def:-C2 */
|
|---|
| 97 | switch (pszTmp[1])
|
|---|
| 98 | {
|
|---|
| 99 | case '1':
|
|---|
| 100 | options.usCom = OUTPUT_COM1;
|
|---|
| 101 | break;
|
|---|
| 102 |
|
|---|
| 103 | case '2':
|
|---|
| 104 | default:
|
|---|
| 105 | options.usCom = OUTPUT_COM2;
|
|---|
| 106 | }
|
|---|
| 107 | break;
|
|---|
| 108 |
|
|---|
| 109 | case 'e':
|
|---|
| 110 | case 'E':/* ELF */
|
|---|
| 111 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 112 | if (pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 113 | && (pszTmp2[1] == 'N' ||pszTmp2[1] == 'n' || pszTmp2[1] == 'D' || pszTmp2[1] == 'd')
|
|---|
| 114 | )
|
|---|
| 115 | options.fElf = FALSE;
|
|---|
| 116 | else
|
|---|
| 117 | options.fElf = TRUE;
|
|---|
| 118 | break;
|
|---|
| 119 |
|
|---|
| 120 | case 'h':
|
|---|
| 121 | case 'H': /* Heap options */
|
|---|
| 122 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 123 | if (pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 124 | && (*pszTmp2 == ':' || *pszTmp2 == '='))
|
|---|
| 125 | {
|
|---|
| 126 | ul = readnum(pszTmp2 + 1);
|
|---|
| 127 | if (ul > 0x1000UL && ul < 0x2000000UL) /* 4KB < ul < 32MB */
|
|---|
| 128 | {
|
|---|
| 129 | if (strnicmp(pszTmp, "heapm", 5) == 0)
|
|---|
| 130 | options.cbHeapMax = ul;
|
|---|
| 131 | else
|
|---|
| 132 | options.cbHeap = ul;
|
|---|
| 133 | }
|
|---|
| 134 | }
|
|---|
| 135 | break;
|
|---|
| 136 |
|
|---|
| 137 | case 'l':
|
|---|
| 138 | case 'L': /* -L[..]<:|=| >[<Y..|E..| > | <N..|D..>] */
|
|---|
| 139 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 140 | if (pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 141 | && (pszTmp2[1] == 'N' ||pszTmp2[1] == 'n' || pszTmp2[1] == 'D' || pszTmp2[1] == 'd')
|
|---|
| 142 | )
|
|---|
| 143 | options.fLogging = FALSE;
|
|---|
| 144 | else
|
|---|
| 145 | options.fLogging = TRUE;
|
|---|
| 146 | break;
|
|---|
| 147 |
|
|---|
| 148 | case 'n':
|
|---|
| 149 | case 'N': /* NoLoader */
|
|---|
| 150 | options.fNoLoader = TRUE;
|
|---|
| 151 | break;
|
|---|
| 152 |
|
|---|
| 153 | case 'p':
|
|---|
| 154 | case 'P': /* PE */
|
|---|
| 155 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 156 | if (pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 157 | && (*pszTmp2 == ':' || *pszTmp2 == '='))
|
|---|
| 158 | {
|
|---|
| 159 | pszTmp++;
|
|---|
| 160 | if (strnicmp(pszTmp, "pe2lx", 5) == 0)
|
|---|
| 161 | options.fPE = FLAGS_PE_PE2LX;
|
|---|
| 162 | else if (strnicmp(pszTmp, "pe", 2) == 0)
|
|---|
| 163 | options.fPE = FLAGS_PE_PE;
|
|---|
| 164 | else if (strnicmp(pszTmp, "mixed", 2) == 0)
|
|---|
| 165 | options.fPE = FLAGS_PE_MIXED;
|
|---|
| 166 | else if (strnicmp(pszTmp, "not", 2) == 0)
|
|---|
| 167 | options.fPE = FLAGS_PE_NOT;
|
|---|
| 168 | else
|
|---|
| 169 | kprintf(("R0Init32: invalid parameter -PE:...\n"));
|
|---|
| 170 | }
|
|---|
| 171 | else
|
|---|
| 172 | kprintf(("R0Init32: invalid parameter -PE...\n"));
|
|---|
| 173 | break;
|
|---|
| 174 |
|
|---|
| 175 | case 'q':
|
|---|
| 176 | case 'Q': /* quiet initialization */
|
|---|
| 177 | options.fQuiet = TRUE;
|
|---|
| 178 | break;
|
|---|
| 179 |
|
|---|
| 180 | case 'r':
|
|---|
| 181 | case 'R': /* ResHeap options */
|
|---|
| 182 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 183 | if (pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 184 | && (*pszTmp2 == ':' || *pszTmp2 == '='))
|
|---|
| 185 | {
|
|---|
| 186 | ul = readnum(pszTmp2 + 1);
|
|---|
| 187 | if (ul > 0x1000UL && ul < 0x700000UL) /* 4KB < ul < 7MB */
|
|---|
| 188 | {
|
|---|
| 189 | if (strnicmp(pszTmp, "resheapm", 8) == 0)
|
|---|
| 190 | options.cbHeapMaxResident = ul;
|
|---|
| 191 | else
|
|---|
| 192 | options.cbHeapResident = ul;
|
|---|
| 193 | }
|
|---|
| 194 | }
|
|---|
| 195 | break;
|
|---|
| 196 |
|
|---|
| 197 | case 's':
|
|---|
| 198 | case 'S': /* Sym:<filename> or Script:<Yes|No> or Smp */
|
|---|
| 199 | /* SMP kernel */
|
|---|
| 200 | if (pszTmp[1] == 'm' || pszTmp[1] == 'M')
|
|---|
| 201 | options.fKernel = KF_SMP;
|
|---|
| 202 | else
|
|---|
| 203 | {
|
|---|
| 204 | if (pszTmp[1] == 'c' || pszTmp[1] == 'C')
|
|---|
| 205 | {
|
|---|
| 206 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 207 | options.fScript = pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 208 | && (*pszTmp2 == ':' || *pszTmp2 == '=')
|
|---|
| 209 | && (pszTmp2[1] == 'Y' || pszTmp2[1] == 'y');
|
|---|
| 210 | }
|
|---|
| 211 | }
|
|---|
| 212 | break;
|
|---|
| 213 |
|
|---|
| 214 | case 'u':
|
|---|
| 215 | case 'U': /* UNI kernel */
|
|---|
| 216 | options.fKernel = KF_UNI;
|
|---|
| 217 | break;
|
|---|
| 218 |
|
|---|
| 219 | case 'v':
|
|---|
| 220 | case 'V': /* verbose initialization */
|
|---|
| 221 | options.fQuiet = FALSE;
|
|---|
| 222 | break;
|
|---|
| 223 |
|
|---|
| 224 | case 'w':
|
|---|
| 225 | case 'W':
|
|---|
| 226 | if (pszTmp[1] >= '0' && pszTmp[1] <= '4')
|
|---|
| 227 | options.ulInfoLevel = pszTmp[1] - '0';
|
|---|
| 228 | else
|
|---|
| 229 | {
|
|---|
| 230 | pszTmp2 = strpbrk(pszTmp, ":=/- ");
|
|---|
| 231 | if (pszTmp2 != NULL && (int)(pszTmp2-pszTmp) < cch-1
|
|---|
| 232 | && (*pszTmp2 == ':' || *pszTmp2 == '=')
|
|---|
| 233 | && pszTmp2[1] >= '0' && pszTmp2[1] <= '4'
|
|---|
| 234 | )
|
|---|
| 235 | options.ulInfoLevel = pszTmp2[1] - '0';
|
|---|
| 236 | }
|
|---|
| 237 | break;
|
|---|
| 238 |
|
|---|
| 239 | }
|
|---|
| 240 | pszTmp = strpbrk(pszTmp, "-/");
|
|---|
| 241 | }
|
|---|
| 242 |
|
|---|
| 243 | /* heap min/max corrections */
|
|---|
| 244 | if (options.cbHeap > options.cbHeapMax)
|
|---|
| 245 | options.cbHeapMax = options.cbHeap;
|
|---|
| 246 | if (options.cbHeapResident > options.cbHeapMaxResident)
|
|---|
| 247 | options.cbHeapMaxResident = options.cbHeapResident;
|
|---|
| 248 |
|
|---|
| 249 | /* Transfer version and build number from 16-bit probkrnl.c */
|
|---|
| 250 | options.ulBuild = _ulBuild;
|
|---|
| 251 | options.usVerMajor = _usVerMajor;
|
|---|
| 252 | options.usVerMinor = _usVerMinor;
|
|---|
| 253 |
|
|---|
| 254 | /* log option summary - FIXME */
|
|---|
| 255 | kprintf(("Options - Summary\n"));
|
|---|
| 256 | kprintf(("\tKernel: ver%d.%d build %d type %s\n",
|
|---|
| 257 | options.usVerMajor,
|
|---|
| 258 | options.usVerMinor,
|
|---|
| 259 | options.ulBuild,
|
|---|
| 260 | (options.fKernel & KF_SMP) ? "SMP" : "UNI"
|
|---|
| 261 | ));
|
|---|
| 262 | kprintf(("\tCom port no.%d\n", options.usCom));
|
|---|
| 263 | if (options.fQuiet)
|
|---|
| 264 | kprintf(("\tQuiet init\n"));
|
|---|
| 265 | else
|
|---|
| 266 | kprintf(("\tVerbose init\n"));
|
|---|
| 267 | if (options.fLogging)
|
|---|
| 268 | kprintf(("\tlogging enabled\n"));
|
|---|
| 269 | else
|
|---|
| 270 | kprintf(("\tlogging disabled\n"));
|
|---|
| 271 | /* end option summary */
|
|---|
| 272 |
|
|---|
| 273 |
|
|---|
| 274 | /*
|
|---|
| 275 | * init sub-parts
|
|---|
| 276 | */
|
|---|
| 277 | /* heap */
|
|---|
| 278 | if (heapInit(options.cbHeap) != NO_ERROR)
|
|---|
| 279 | return STATUS_DONE | STERR | ERROR_I24_QUIET_INIT_FAIL;
|
|---|
| 280 |
|
|---|
| 281 | /* loader */
|
|---|
| 282 | if (!options.fNoLoader)
|
|---|
| 283 | if (ldrInit() != NO_ERROR)
|
|---|
| 284 | return STATUS_DONE | STERR | ERROR_I24_QUIET_INIT_FAIL;
|
|---|
| 285 |
|
|---|
| 286 | /* functionoverrides */
|
|---|
| 287 | if (!options.fNoLoader)
|
|---|
| 288 | if (procInit() != NO_ERROR)
|
|---|
| 289 | return STATUS_DONE | STERR | ERROR_I24_QUIET_INIT_FAIL;
|
|---|
| 290 |
|
|---|
| 291 | return STATUS_DONE;
|
|---|
| 292 | }
|
|---|
| 293 |
|
|---|
| 294 |
|
|---|
| 295 | /**
|
|---|
| 296 | * Reads a number (unsigned long integer) for a string.
|
|---|
| 297 | * @returns number read, ~0UL on error / no number read.
|
|---|
| 298 | * @param pszNum Pointer to the string containing the number.
|
|---|
| 299 | * @status competely implemented.
|
|---|
| 300 | * @author knut st. osmundsen
|
|---|
| 301 | */
|
|---|
| 302 | static ULONG readnum(const char *pszNum)
|
|---|
| 303 | {
|
|---|
| 304 | ULONG ulRet = 0;
|
|---|
| 305 | ULONG ulBase = 10;
|
|---|
| 306 | int i = 0;
|
|---|
| 307 |
|
|---|
| 308 | if (*pszNum == '0')
|
|---|
| 309 | if (pszNum[1] == 'x' || pszNum[1] == 'X')
|
|---|
| 310 | {
|
|---|
| 311 | ulBase == 16;
|
|---|
| 312 | pszNum += 2;
|
|---|
| 313 | }
|
|---|
| 314 | else
|
|---|
| 315 | {
|
|---|
| 316 | ulBase == 8;
|
|---|
| 317 | i = 1;
|
|---|
| 318 | }
|
|---|
| 319 |
|
|---|
| 320 | /* read digits */
|
|---|
| 321 | while (ulBase == 16 ? (pszNum[i] >= '0' && pszNum[i] <= '9') || (pszNum[i] >= 'a' && pszNum[i] <= 'f') || (pszNum[i] >= 'A' && pszNum[i] <= 'F')
|
|---|
| 322 | : (pszNum[i] >= '0' && pszNum[i] <= (ulBase == 10 ? '9' : '7'))
|
|---|
| 323 | )
|
|---|
| 324 | {
|
|---|
| 325 | ulRet *= ulBase;
|
|---|
| 326 | if (ulBase <= 10)
|
|---|
| 327 | ulRet += pszNum[i] - '0';
|
|---|
| 328 | else
|
|---|
| 329 | ulRet += pszNum[i] - (pszNum[i] >= 'A' ? 'A' - 10 : (pszNum[i] >= 'a' ? 'a' + 9 : '0'));
|
|---|
| 330 |
|
|---|
| 331 | i++;
|
|---|
| 332 | }
|
|---|
| 333 |
|
|---|
| 334 | return i > 0 ? ulRet : ~0UL;
|
|---|
| 335 | }
|
|---|
| 336 |
|
|---|
| 337 |
|
|---|
| 338 | /**
|
|---|
| 339 | * Verifies the aProcTab.
|
|---|
| 340 | * @returns 0 if ok. !0 if not ok.
|
|---|
| 341 | * @remark Called from IOCtl.
|
|---|
| 342 | */
|
|---|
| 343 | USHORT _loadds _Far32 _Pascal VerifyProcTab32(void)
|
|---|
| 344 | {
|
|---|
| 345 | int i;
|
|---|
| 346 | int cb;
|
|---|
| 347 |
|
|---|
| 348 | /* verify */
|
|---|
| 349 | for (i = 0; i < NUMBER_OF_PROCS; i++)
|
|---|
| 350 | {
|
|---|
| 351 | /* verify that it is found */
|
|---|
| 352 | if (!_aProcTab[i].fFound)
|
|---|
| 353 | {
|
|---|
| 354 | kprintf(("VerifyProcTab32: procedure no.%d was not found!\n", i));
|
|---|
| 355 | return STATUS_DONE | STERR | 1;
|
|---|
| 356 | }
|
|---|
| 357 |
|
|---|
| 358 | /* verify read/writeable. - FIXME */
|
|---|
| 359 | if (_aProcTab[i].ulAddress < 0xffe00000UL)
|
|---|
| 360 | {
|
|---|
| 361 | kprintf(("VerifyProcTab32: procedure no.%d has an invlalid address, %#08x!\n",
|
|---|
| 362 | i, _aProcTab[i].ulAddress));
|
|---|
| 363 | return STATUS_DONE | STERR | 2;
|
|---|
| 364 | }
|
|---|
| 365 |
|
|---|
| 366 | /* verify known function prolog. (only EPT_PROC) */
|
|---|
| 367 | if (_aProcTab[i].fType == EPT_PROC)
|
|---|
| 368 | {
|
|---|
| 369 | if ((cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress)) <= 0 && cb + 5 >= MAXSIZE_PROLOG)
|
|---|
| 370 | {
|
|---|
| 371 | kprintf(("VerifyProcTab32: verify failed for procedure no.%d\n",i));
|
|---|
| 372 | return STATUS_DONE | STERR | 3;
|
|---|
| 373 | }
|
|---|
| 374 | }
|
|---|
| 375 | else
|
|---|
| 376 | {
|
|---|
| 377 | kprintf(("VerifyProcTab32: only EPT_PROC is implemented\n",i));
|
|---|
| 378 | return STATUS_DONE | STERR | 4;
|
|---|
| 379 | }
|
|---|
| 380 | }
|
|---|
| 381 |
|
|---|
| 382 | return STATUS_DONE;
|
|---|
| 383 | }
|
|---|
| 384 |
|
|---|
| 385 |
|
|---|
| 386 | /**
|
|---|
| 387 | * Get kernel OTEs
|
|---|
| 388 | * @returns Strategy return code:
|
|---|
| 389 | * STATUS_DONE on success.
|
|---|
| 390 | * STATUS_DONE | STERR | errorcode on failure.
|
|---|
| 391 | * @param pOTEBuf Pointer to output buffer.
|
|---|
| 392 | * @status completely implemented and tested.
|
|---|
| 393 | * @author knut st. osmundsen
|
|---|
| 394 | * @remark Called from IOCtl.
|
|---|
| 395 | */
|
|---|
| 396 | USHORT _loadds _Far32 _Pascal GetOTEs32(PKRNLOBJTABLE pOTEBuf)
|
|---|
| 397 | {
|
|---|
| 398 | PMTE pMTE;
|
|---|
| 399 | PSMTE pSMTE;
|
|---|
| 400 | POTE pOTE;
|
|---|
| 401 | int i;
|
|---|
| 402 | USHORT usRc;
|
|---|
| 403 |
|
|---|
| 404 | pMTE = GetOS2KrnlMTE();
|
|---|
| 405 | if (pMTE != NULL)
|
|---|
| 406 | {
|
|---|
| 407 | pSMTE = pMTE->mte_swapmte;
|
|---|
| 408 | if (pSMTE != NULL)
|
|---|
| 409 | {
|
|---|
| 410 | pOTEBuf->cObjects = pSMTE->smte_objcnt;
|
|---|
| 411 | if (pSMTE->smte_objcnt <= MAXKRNLOBJECTS)
|
|---|
| 412 | {
|
|---|
| 413 | pOTE = pSMTE->smte_objtab;
|
|---|
| 414 | if (pOTE != NULL)
|
|---|
| 415 | {
|
|---|
| 416 | for (i = 0; i < pOTEBuf->cObjects; i++)
|
|---|
| 417 | memcpy((void*)&pOTEBuf->aObjects[i], &pOTE[i], sizeof(OTE));
|
|---|
| 418 | usRc = 0;
|
|---|
| 419 | }
|
|---|
| 420 | else
|
|---|
| 421 | usRc = 4;
|
|---|
| 422 | }
|
|---|
| 423 | else
|
|---|
| 424 | usRc = 3;
|
|---|
| 425 | }
|
|---|
| 426 | else
|
|---|
| 427 | usRc = 2;
|
|---|
| 428 | }
|
|---|
| 429 | else
|
|---|
| 430 | usRc = 1;
|
|---|
| 431 |
|
|---|
| 432 | if (usRc != 0)
|
|---|
| 433 | kprintf(("GetOTEs32: failed. usRc = %d\n", usRc));
|
|---|
| 434 |
|
|---|
| 435 | return usRc | (usRc != 0 ? STATUS_DONE | STERR : STATUS_DONE);
|
|---|
| 436 | }
|
|---|
| 437 |
|
|---|
| 438 |
|
|---|
| 439 | /**
|
|---|
| 440 | * Interpret function prolog to find where to jmp back.
|
|---|
| 441 | * @returns Length of prolog need to be copied - which is also the offset of
|
|---|
| 442 | * where the jmp instr should be placed.
|
|---|
| 443 | * On error it returns 0.
|
|---|
| 444 | * @param p Pointer to prolog.
|
|---|
| 445 | */
|
|---|
| 446 | static int interpretFunctionProlog(char *p)
|
|---|
| 447 | {
|
|---|
| 448 | int length;
|
|---|
| 449 | int rc;
|
|---|
| 450 |
|
|---|
| 451 | /*
|
|---|
| 452 | * check for the well known prolog (the only that is supported now)
|
|---|
| 453 | * which is:
|
|---|
| 454 | * push ebp
|
|---|
| 455 | * mov ebp,esp
|
|---|
| 456 | * or
|
|---|
| 457 | * push ebp
|
|---|
| 458 | * mov ecx, dword ptr [xxxxxxxx]
|
|---|
| 459 | */
|
|---|
| 460 |
|
|---|
| 461 | if (p[0] == 0x55 && p[1] == 0x8b)
|
|---|
| 462 | {
|
|---|
| 463 | if (p[2] == 0xec)
|
|---|
| 464 | rc = 3;
|
|---|
| 465 | else
|
|---|
| 466 | rc = 1;
|
|---|
| 467 | while (rc < 5)
|
|---|
| 468 | {
|
|---|
| 469 | /*
|
|---|
| 470 | * This is not at all very stable or correct - but it works
|
|---|
| 471 | * for the current functions.
|
|---|
| 472 | * There will never be any doubt when something goes wrong!
|
|---|
| 473 | */
|
|---|
| 474 | switch(p[rc])
|
|---|
| 475 | {
|
|---|
| 476 | case 0x33: /* xor (ldrClose, ldrOpen) */
|
|---|
| 477 | rc +=2;
|
|---|
| 478 | break;
|
|---|
| 479 | case 0x8b:
|
|---|
| 480 | if (p[rc+1] == 0x0d)
|
|---|
| 481 | rc += 6;
|
|---|
| 482 | else
|
|---|
| 483 | rc += 2; /*????!*/
|
|---|
| 484 | break;
|
|---|
| 485 | case 0x8d: /* lea (ldrRead) */
|
|---|
| 486 | rc += 3;
|
|---|
| 487 | break;
|
|---|
| 488 | case 0x83: /* sub (LDRQAppType) */
|
|---|
| 489 | rc += 3;
|
|---|
| 490 | break;
|
|---|
| 491 | default:
|
|---|
| 492 | kprintf(("interpretFunctionProlog: unknown instruction 0x%x \n", p[rc]));
|
|---|
| 493 | return 0;
|
|---|
| 494 | }
|
|---|
| 495 | }
|
|---|
| 496 | }
|
|---|
| 497 | else
|
|---|
| 498 | rc = 0;
|
|---|
| 499 |
|
|---|
| 500 | return rc;
|
|---|
| 501 | }
|
|---|
| 502 |
|
|---|
| 503 |
|
|---|
| 504 |
|
|---|
| 505 | /**
|
|---|
| 506 | * Initiates the overrided functions.
|
|---|
| 507 | * @returns NO_ERROR on success. !0 on error.
|
|---|
| 508 | */
|
|---|
| 509 | static int procInit(void)
|
|---|
| 510 | {
|
|---|
| 511 | int i;
|
|---|
| 512 | int cb;
|
|---|
| 513 |
|
|---|
| 514 | /* verify */
|
|---|
| 515 | for (i = 0; i < NUMBER_OF_PROCS; i++)
|
|---|
| 516 | {
|
|---|
| 517 | cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress);
|
|---|
| 518 | if (cb <= 0 || cb + 5 >= MAXSIZE_PROLOG)
|
|---|
| 519 | {
|
|---|
| 520 | kprintf(("rehookFunctions: verify failed for procedure no.%d, cb=%d\n", i, cb));
|
|---|
| 521 | return 1;
|
|---|
| 522 | }
|
|---|
| 523 | }
|
|---|
| 524 |
|
|---|
| 525 | /* rehook */
|
|---|
| 526 | for (i = 0; i < NUMBER_OF_PROCS; i++)
|
|---|
| 527 | {
|
|---|
| 528 | cb = interpretFunctionProlog((char*)_aProcTab[i].ulAddress);
|
|---|
| 529 | if (cb > 0 && cb + 5 < MAXSIZE_PROLOG)
|
|---|
| 530 | {
|
|---|
| 531 | char *pMy;
|
|---|
| 532 | switch (i)
|
|---|
| 533 | {
|
|---|
| 534 | case iLDRREAD: pMy = (char*)myldrRead; break;
|
|---|
| 535 | case iLDROPEN: pMy = (char*)myldrOpen; break;
|
|---|
| 536 | case iLDRCLOSE: pMy = (char*)myldrClose; break;
|
|---|
| 537 | case iLDRQAPPTYPE: pMy = (char*)myLDRQAppType; break;
|
|---|
| 538 |
|
|---|
| 539 | default:
|
|---|
| 540 | kprintf(("rehookFunctions: Someone has added function without updating the switch! i=%d\n", i));
|
|---|
| 541 | Int3();
|
|---|
| 542 | return 2;
|
|---|
| 543 | }
|
|---|
| 544 |
|
|---|
| 545 | /* copy function prolog */
|
|---|
| 546 | memcpy(callTab[i], (void*)_aProcTab[i].ulAddress, (size_t)cb);
|
|---|
| 547 |
|
|---|
| 548 | /* jump from calltab to original function */
|
|---|
| 549 | callTab[i][cb] = 0xE9; /* jmp */
|
|---|
| 550 | *(unsigned*)&callTab[i][cb+1] = _aProcTab[i].ulAddress + cb - (unsigned)&callTab[i][cb+5];
|
|---|
| 551 |
|
|---|
| 552 | /* jump from original function to my function - an cli could be needed here */
|
|---|
| 553 | *(char*)_aProcTab[i].ulAddress = 0xE9; /* jmp */
|
|---|
| 554 | *(unsigned*)(_aProcTab[i].ulAddress + 1) = (unsigned)pMy - (_aProcTab[i].ulAddress + 5);
|
|---|
| 555 | }
|
|---|
| 556 | else
|
|---|
| 557 | {
|
|---|
| 558 | /* !fatal! - this could never happen really... */
|
|---|
| 559 | kprintf(("rehookFunctions: FATAL verify failed for procedure no.%d when rehooking it!\n",i));
|
|---|
| 560 | Int3(); /* ipe - later! */
|
|---|
| 561 | return 1;
|
|---|
| 562 | }
|
|---|
| 563 | }
|
|---|
| 564 |
|
|---|
| 565 | return NO_ERROR;
|
|---|
| 566 | }
|
|---|
| 567 |
|
|---|