27ed0e502b
git-svn-id: https://svn.code.sf.net/p/nsis/code/NSIS/trunk@6802 212acab6-be3b-0410-9dea-997c60f758d6
160 lines
2.7 KiB
C++
160 lines
2.7 KiB
C++
/*
|
|
|
|
LibraryLocal - used by the Library.nsh macros
|
|
Get the version of local DLL and TLB files
|
|
Written by Joost Verburg
|
|
Unicode support by Jim Park -- 07/27/2007
|
|
|
|
*/
|
|
|
|
#include "../../../Source/Platform.h"
|
|
#include "../../../Source/tstring.h"
|
|
|
|
#include <stdio.h>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
|
|
#include "../../../Source/util.h"
|
|
#include "../../../Source/winchar.h"
|
|
|
|
using namespace std;
|
|
|
|
int g_noconfig=0; // TODO: Not used?
|
|
NSISRT_DEFINEGLOBALS();
|
|
|
|
int GetTLBVersion(tstring& filepath, DWORD& high, DWORD & low)
|
|
{
|
|
#ifdef _WIN32
|
|
|
|
int found = 0;
|
|
|
|
TCHAR fullpath[1024];
|
|
TCHAR *p;
|
|
if (!GetFullPathName(filepath.c_str(), COUNTOF(fullpath), fullpath, &p))
|
|
return 0;
|
|
|
|
ITypeLib* typeLib;
|
|
HRESULT hr;
|
|
|
|
#ifdef _UNICODE
|
|
hr = LoadTypeLib(fullpath, &typeLib);
|
|
#else
|
|
// If built without UNICODE, we still need to convert this string to a Unicode string.
|
|
WCHAR *ole_filename = (WCHAR*) WinWStrDupFromTChar(fullpath);
|
|
if (!ole_filename) return 0;
|
|
hr = LoadTypeLib(ole_filename, &typeLib);
|
|
free(ole_filename);
|
|
#endif //~ _UNICODE
|
|
|
|
if (SUCCEEDED(hr)) {
|
|
|
|
TLIBATTR* typelibAttr;
|
|
|
|
hr = typeLib->GetLibAttr(&typelibAttr);
|
|
|
|
if (SUCCEEDED(hr)) {
|
|
|
|
high = typelibAttr->wMajorVerNum;
|
|
low = typelibAttr->wMinorVerNum;
|
|
|
|
found = 1;
|
|
|
|
}
|
|
|
|
typeLib->Release();
|
|
|
|
}
|
|
|
|
return found;
|
|
|
|
#else
|
|
|
|
return 0;
|
|
|
|
#endif //~ _WIN32
|
|
}
|
|
|
|
NSIS_ENTRYPOINT_TMAIN
|
|
int _tmain(int argc, TCHAR* argv[])
|
|
{
|
|
if (!NSISRT_Initialize()) return 1;
|
|
|
|
// Parse the command line
|
|
|
|
tstring cmdline;
|
|
|
|
tstring mode;
|
|
tstring filename;
|
|
tstring filepath;
|
|
|
|
int filefound = 0;
|
|
|
|
if (argc != 4)
|
|
return 1;
|
|
|
|
// Get the full path of the local file
|
|
|
|
mode = argv[1];
|
|
filename = argv[2];
|
|
|
|
// Validate filename
|
|
|
|
FILE*fIn = FOPEN(filename.c_str(), ("rb"));
|
|
filefound = !!fIn;
|
|
if (fIn)
|
|
fclose(fIn);
|
|
|
|
// Work
|
|
|
|
int versionfound = 0;
|
|
DWORD low = 0, high = 0;
|
|
|
|
if (filefound)
|
|
{
|
|
|
|
// Get version
|
|
|
|
// DLL / EXE
|
|
|
|
if (mode.compare(_T("D")) == 0)
|
|
{
|
|
|
|
versionfound = GetDLLVersion(filename, high, low);
|
|
|
|
}
|
|
|
|
// TLB
|
|
|
|
if (mode.compare(_T("T")) == 0)
|
|
{
|
|
|
|
versionfound = GetTLBVersion(filename, high, low);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Write the version to an NSIS header file
|
|
|
|
FILE*fHdr = FOPEN(argv[3], ("wt"));
|
|
if (!fHdr) return 1;
|
|
|
|
// File content is always ASCII so we don't use TCHAR
|
|
if (!filefound)
|
|
{
|
|
fputs("!define LIBRARY_VERSION_FILENOTFOUND\n", fHdr);
|
|
}
|
|
else if (!versionfound)
|
|
{
|
|
fputs("!define LIBRARY_VERSION_NONE\n", fHdr);
|
|
}
|
|
else
|
|
{
|
|
fprintf(fHdr, "!define LIBRARY_VERSION_HIGH %lu\n", high);
|
|
fprintf(fHdr, "!define LIBRARY_VERSION_LOW %lu\n", low);
|
|
}
|
|
|
|
fclose(fHdr);
|
|
return 0;
|
|
|
|
}
|