From aeeb51e25cdf55e93a0e99c1cfcfaa384aa541cd Mon Sep 17 00:00:00 2001
From: Doug MacEachern <dougm@hyperic.com>
Date: Tue, 22 Jun 2004 06:12:51 +0000
Subject: [PATCH] proc_modules impl

---
 src/os/win32/sigar_os.h    |  1 +
 src/os/win32/win32_sigar.c | 85 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 86 insertions(+)

diff --git a/src/os/win32/sigar_os.h b/src/os/win32/sigar_os.h
index 57e61103..c89318b3 100644
--- a/src/os/win32/sigar_os.h
+++ b/src/os/win32/sigar_os.h
@@ -12,6 +12,7 @@
 #include <malloc.h>
 #include <stdio.h>
 #include <errno.h>
+#include <tlhelp32.h>
 
 #define INT64_C(val) val##i64
 
diff --git a/src/os/win32/win32_sigar.c b/src/os/win32/win32_sigar.c
index 8604dea3..c4612b03 100644
--- a/src/os/win32/win32_sigar.c
+++ b/src/os/win32/win32_sigar.c
@@ -1133,6 +1133,91 @@ SIGAR_DECLARE(int) sigar_proc_exe_get(sigar_t *sigar, sigar_pid_t pid,
     return status;
 }
 
+typedef HANDLE (CALLBACK *LPCREATESNAPSHOT)(DWORD, DWORD);
+typedef BOOL (CALLBACK *LPMODULEITER)(HANDLE, LPMODULEENTRY32);
+
+/* not available on NT */
+static int sigar_proc_modules_get_toolhelp(sigar_t *sigar,
+                                           sigar_pid_t pid,
+                                           sigar_proc_modules_t *procmods)
+{
+    HINSTANCE k32_handle;
+    HANDLE snap_shot;
+    MODULEENTRY32 module;
+    LPCREATESNAPSHOT create_snapshot;
+    LPMODULEITER module_first, module_next;
+
+    /* XXX: cache this stuff within sigar_t */
+    k32_handle = LoadLibrary("kernel32.dll");
+    if (!k32_handle) {
+        return GetLastError();
+    }
+
+    create_snapshot =
+        (LPCREATESNAPSHOT)GetProcAddress(k32_handle,
+                                         "CreateToolhelp32Snapshot");
+
+    if (!create_snapshot) {
+        FreeLibrary(k32_handle);
+        return GetLastError();
+    }
+
+    module_first =
+        (LPMODULEITER)GetProcAddress(k32_handle, "Module32First");
+
+    if (!module_first) {
+        FreeLibrary(k32_handle);
+        return GetLastError();
+    }
+
+    module_next =
+        (LPMODULEITER)GetProcAddress(k32_handle, "Module32Next");
+
+    if (!module_next) {
+        FreeLibrary(k32_handle);
+        return GetLastError();
+    }
+
+    snap_shot = create_snapshot(TH32CS_SNAPMODULE, (DWORD)pid);
+
+    if (snap_shot == INVALID_HANDLE_VALUE) {
+        FreeLibrary(k32_handle);
+        return GetLastError();
+    }
+
+    module.dwSize = sizeof(MODULEENTRY32);
+    if (!module_first(snap_shot, &module)) {
+        return SIGAR_OK;
+    }
+
+    do {
+        int status =
+            procmods->module_getter(procmods->data,
+                                    module.szExePath,
+                                    strlen(module.szExePath));
+
+        if (status != SIGAR_OK) {
+            /* not an error; just stop iterating */
+            break;
+        }
+
+        module.dwSize = sizeof(MODULEENTRY32);
+    } while (module_next(snap_shot, &module));
+
+    FreeLibrary(k32_handle);
+
+    return SIGAR_OK;
+}
+
+SIGAR_DECLARE(int) sigar_proc_modules_get(sigar_t *sigar, sigar_pid_t pid,
+                                          sigar_proc_modules_t *procmods)
+{
+    /* XXX need to use psapi.dll for NT */
+    return sigar_proc_modules_get_toolhelp(sigar,
+                                           pid,
+                                           procmods);
+}
+
 int sigar_os_fs_type_get(sigar_file_system_t *fsp)
 {
     return fsp->type;