From b8888a69bd9066e672d0f2218db1be035bf18102 Mon Sep 17 00:00:00 2001
From: kichik <kichik@212acab6-be3b-0410-9dea-997c60f758d6>
Date: Tue, 24 Apr 2007 14:06:50 +0000
Subject: [PATCH] added GetDLLVersion

git-svn-id: https://svn.code.sf.net/p/nsis/code/NSIS/trunk@5116 212acab6-be3b-0410-9dea-997c60f758d6
---
 Source/util.cpp | 106 ++++++++++++++++++++++++++++++++++++++++++++++++
 Source/util.h   |   2 +
 2 files changed, 108 insertions(+)

diff --git a/Source/util.cpp b/Source/util.cpp
index 93f957d4..6636eb01 100644
--- a/Source/util.cpp
+++ b/Source/util.cpp
@@ -729,3 +729,109 @@ int sane_system(const char *command) {
 
 #endif
 }
+
+static bool GetDLLVersionUsingRE(const string& filepath, DWORD& high, DWORD & low)
+{
+  bool found = false;
+
+  FILE *fdll = FOPEN(filepath.c_str(), "rb");
+  if (!fdll)
+    return 0;
+
+  fseek(fdll, 0, SEEK_END);
+  unsigned int len = ftell(fdll);
+  fseek(fdll, 0, SEEK_SET);
+
+  LPBYTE dll = (LPBYTE) malloc(len);
+
+  if (!dll)
+  {
+    fclose(fdll);
+    return 0;
+  }
+
+  if (fread(dll, 1, len, fdll) != len)
+  {
+    fclose(fdll);
+    free(dll);
+    return 0;
+  }
+
+  try
+  {
+    CResourceEditor *dllre = new CResourceEditor(dll, len);
+    LPBYTE ver = dllre->GetResourceA(VS_FILE_INFO, MAKEINTRESOURCE(VS_VERSION_INFO), 0);
+    int versize = dllre->GetResourceSizeA(VS_FILE_INFO, MAKEINTRESOURCE(VS_VERSION_INFO), 0);
+
+    if (ver)
+    {
+      if ((size_t) versize > sizeof(WORD) * 3)
+      {
+        // get VS_FIXEDFILEINFO from VS_VERSIONINFO
+        WCHAR *szKey = (WCHAR *)(ver + sizeof(WORD) * 3);
+        int len = (winchar_strlen(szKey) + 1) * sizeof(WCHAR) + sizeof(WORD) * 3;
+        len = (len + 3) & ~3; // align on DWORD boundry
+        VS_FIXEDFILEINFO *verinfo = (VS_FIXEDFILEINFO *)(ver + len);
+        if (versize > len && verinfo->dwSignature == VS_FFI_SIGNATURE)
+        {
+          low = verinfo->dwFileVersionLS;
+          high = verinfo->dwFileVersionMS;
+          found = true;
+        }
+      }
+      dllre->FreeResource(ver);
+    }
+
+    delete dllre;
+  }
+  catch (exception&)
+  {
+  }
+
+  return found;
+}
+
+static bool GetDLLVersionUsingAPI(const string& filepath, DWORD& high, DWORD& low)
+{
+  bool found = false;
+
+#ifdef _WIN32
+  char path[1024];
+  char *name;
+  path[0] = 0;
+
+  GetFullPathName(filepath.c_str(), 1024, path, &name);
+
+  DWORD d;
+  DWORD verSize = GetFileVersionInfoSize(path, &d);
+  if (verSize)
+  {
+    void *buf = (void *) GlobalAlloc(GPTR, verSize);
+    if (buf)
+    {
+      UINT uLen;
+      VS_FIXEDFILEINFO *pvsf;
+      if (GetFileVersionInfo(path, 0, verSize, buf) && VerQueryValue(buf, "\\", (void**) &pvsf, &uLen))
+      {
+        low = pvsf->dwFileVersionLS;
+        high = pvsf->dwFileVersionMS;
+        found = true;
+      }
+      GlobalFree(buf);
+    }
+  }
+#endif
+
+  return found;
+}
+
+bool GetDLLVersion(const string& filepath, DWORD& high, DWORD& low)
+{
+  if (GetDLLVersionUsingAPI(filepath, high, low))
+    return true;
+
+  if (GetDLLVersionUsingRE(filepath, high, low))
+    return true;
+
+  return false;
+}
diff --git a/Source/util.h b/Source/util.h
index 82b5aeb9..9c206c66 100644
--- a/Source/util.h
+++ b/Source/util.h
@@ -48,6 +48,8 @@ int generate_unicons_offsets(unsigned char* exeHeader, size_t exeHeaderSize, uns
 
 size_t my_strftime(char *s, size_t max, const char  *fmt, const struct tm *tm);
 
+bool GetDLLVersion(const std::string& filepath, DWORD& high, DWORD& low);
+
 std::string get_full_path(const std::string& path);
 std::string get_dir_name(const std::string& path);
 std::string get_file_name(const std::string& path);