1 /*
   2  * Copyright (c) 2001, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  *
  23  */
  24 
  25 // Disable too-long symbol warnings
  26 #pragma warning ( disable : 4786 )
  27 
  28 #include "libInfo.hpp"
  29 #include "nt4internals.hpp"
  30 #include "isNT4.hpp"
  31 #include "toolHelp.hpp"
  32 #include <assert.h>
  33 
  34 using namespace std;
  35 
  36 typedef void LibInfoImplFunc(DWORD pid, vector<LibInfo>& info);
  37 
  38 static void libInfoImplNT4(DWORD pid, vector<LibInfo>& info);
  39 static void libInfoImplToolHelp(DWORD pid, vector<LibInfo>& info);
  40 
  41 void
  42 libInfo(DWORD pid, vector<LibInfo>& info) {
  43   static LibInfoImplFunc* impl = NULL;
  44 
  45   if (impl == NULL) {
  46     // See which operating system we're on
  47     impl = (isNT4() ? &libInfoImplNT4 : &libInfoImplToolHelp);
  48   }
  49 
  50   assert(impl != NULL);
  51 
  52   (*impl)(pid, info);
  53 }
  54 
  55 static ULONG
  56 ModuleCount(NT4::PDEBUG_BUFFER db) {
  57   return db->ModuleInformation ? *PULONG(db->ModuleInformation) : 0;
  58 }
  59 
  60 #define MAX2(a, b) (((a) < (b)) ? (b) : (a))
  61 
  62 static void
  63 libInfoImplNT4(DWORD pid, vector<LibInfo>& info) {
  64   static EnumProcessModulesFunc*   enumFunc = NULL;
  65   static GetModuleFileNameExFunc*  fnFunc   = NULL;
  66   static GetModuleInformationFunc* infoFunc = NULL;
  67 
  68   if (enumFunc == NULL) {
  69     HMODULE dll = loadPSAPIDLL();
  70 
  71     enumFunc = (EnumProcessModulesFunc*)   GetProcAddress(dll, "EnumProcessModules");
  72     fnFunc   = (GetModuleFileNameExFunc*)  GetProcAddress(dll, "GetModuleFileNameExA");
  73     infoFunc = (GetModuleInformationFunc*) GetProcAddress(dll, "GetModuleInformation");
  74 
  75     assert(enumFunc != NULL);
  76     assert(fnFunc   != NULL);
  77     assert(infoFunc != NULL);
  78   }
  79 
  80   static HMODULE* mods = new HMODULE[256];
  81   static int      numMods = 256;
  82 
  83   if (mods == NULL) {
  84     mods = new HMODULE[numMods];
  85     if (mods == NULL) {
  86       return;
  87     }
  88   }
  89 
  90   bool done = false;
  91 
  92   HANDLE proc = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);
  93   if (proc == NULL) {
  94     return;
  95   }
  96 
  97   do {
  98     DWORD bufSize = numMods * sizeof(HMODULE);
  99     DWORD neededSize;
 100 
 101     if (!(*enumFunc)(proc, mods, bufSize, &neededSize)) {
 102       // Enum failed
 103       CloseHandle(proc);
 104       return;
 105     }
 106 
 107     int numFetched = neededSize / sizeof(HMODULE);
 108 
 109     if (numMods < numFetched) {
 110       // Grow buffer
 111       numMods = MAX2(numFetched, 2 * numMods);
 112       delete[] mods;
 113       mods = new HMODULE[numMods];
 114       if (mods == NULL) {
 115         CloseHandle(proc);
 116         return;
 117       }
 118     } else {
 119       char filename[MAX_PATH];
 120       MODULEINFO modInfo;
 121 
 122       // Iterate through and fetch each one's info
 123       for (int i = 0; i < numFetched; i++) {
 124         if (!(*fnFunc)(proc, mods[i], filename, MAX_PATH)) {
 125           CloseHandle(proc);
 126           return;
 127         }
 128 
 129         if (!(*infoFunc)(proc, mods[i], &modInfo, sizeof(MODULEINFO))) {
 130           CloseHandle(proc);
 131           return;
 132         }
 133 
 134         info.push_back(LibInfo(string(filename), (void*) modInfo.lpBaseOfDll));
 135       }
 136 
 137       done = true;
 138     }
 139   } while (!done);
 140 
 141   CloseHandle(proc);
 142   return;
 143 }
 144 
 145 void
 146 libInfoImplToolHelp(DWORD pid, vector<LibInfo>& info) {
 147   using namespace ToolHelp;
 148 
 149   static CreateToolhelp32SnapshotFunc* snapshotFunc = NULL;
 150   static Module32FirstFunc*            firstFunc    = NULL;
 151   static Module32NextFunc*             nextFunc     = NULL;
 152 
 153   if (snapshotFunc == NULL) {
 154     HMODULE dll = loadDLL();
 155 
 156     snapshotFunc =
 157       (CreateToolhelp32SnapshotFunc*) GetProcAddress(dll,
 158                                                      "CreateToolhelp32Snapshot");
 159 
 160     firstFunc = (Module32FirstFunc*) GetProcAddress(dll,
 161                                                     "Module32First");
 162 
 163     nextFunc = (Module32NextFunc*) GetProcAddress(dll,
 164                                                   "Module32Next");
 165 
 166     assert(snapshotFunc != NULL);
 167     assert(firstFunc    != NULL);
 168     assert(nextFunc     != NULL);
 169   }
 170 
 171   HANDLE snapshot = (*snapshotFunc)(TH32CS_SNAPMODULE, pid);
 172   if (snapshot == (HANDLE) -1) {
 173     // Error occurred during snapshot
 174     return;
 175   }
 176 
 177   // Iterate
 178   MODULEENTRY32 module;
 179   if ((*firstFunc)(snapshot, &module)) {
 180     do {
 181       info.push_back(LibInfo(string(module.szExePath), (void*) module.modBaseAddr));
 182     } while ((*nextFunc)(snapshot, &module));
 183   }
 184 
 185   CloseHandle(snapshot);
 186 }