#include <windows.h>
#include <stdio.h>
#include <string>
#include <set>
#include <iostream>
#include <algorithm>
#include <dbghelp.h>
#define F_USE_USH_NAME 0x000001
#define F_OUT_DIFF 0x000002
#define F_OUT_USH 0x000004
#define F_OUT_WCH 0x000008
#define F_OUT_IDEN 0x000010
#define F_OUT_DNAME 0x000020
#define F_OUT_ALL 0x00001e
using namespace std;
int flags = 0;
enum Ext { none, lib, dll};
char tmp[512];
Ext FindExt(const char *Fname)
{
Fname = strrchr(Fname, '.');
if ( Fname == NULL)
return none;
else if ( !stricmp(Fname, ".Lib"))
return lib;
else if ( !stricmp(Fname, ".Dll"))
return dll;
else
return none;
}
char *filter(char* ch, Ext fext)
{
char *str = strchr(ch, '?');
return str == NULL ? ch: str;
}
const char* setflags(int argc, char *argv[])
{
const char *Fname = NULL;
for (int i = 1; i < argc; i++)
{
if (argv[i][0] == '-' || argv[i][0] == '/')
{
for ( int j = 1; argv[i][j] != '\0'; j++)
{
switch (argv[i][j])
{
case 'u':
case 'U':
flags |= F_OUT_USH;
break;
case 'w':
case 'W':
flags |= F_OUT_WCH;
break;
case 'd':
case 'D':
flags |= F_OUT_DNAME;
break;
case 'i':
case 'I':
flags |= F_OUT_IDEN;
break;
case 'x':
case 'X':
flags |= F_USE_USH_NAME;
break;
default:
return NULL;
}
}
}
else
{
if (!Fname)
Fname = argv[i];
else
return NULL;
}
}
if (!(flags & F_OUT_USH) && !(flags & F_OUT_WCH) && !(flags & F_OUT_IDEN))
flags |= F_OUT_ALL;
return Fname;
}
void print(string st)
{
if ( flags & F_OUT_DNAME)
{
if (UnDecorateSymbolName( st.c_str(), tmp, 511, 0))
cout << tmp << '\n';
else
cout << st << '\n';
} else {
cout << st << '\n';
}
}
int funccmp(const char *func1, const char *func2)
{
while (*func1 != 0 && *func2 != 0)
{
if ( *func1 == *func2)
func1++, func2++;
else if ( func1 == strstr(func1, "_W") &&
func2 == strstr(func2, "G"))
func1 += sizeof("_W"), func2 += sizeof("G");
else if ( func2 == strstr(func2, "_W") &&
func1 == strstr(func1, "G"))
func2 += sizeof("_W"), func1 += sizeof("G");
else
break;
}
if (*func1 < *func2)
return -1;
else if (*func1 > *func2)
return 1;
else
return 0;
}
int Funccmp(const char *func1, const char *func2)
{
static char tmp1[512];
UnDecorateSymbolName(func1, tmp, 511, 0);
UnDecorateSymbolName(func2, tmp1, 511, 0);
func1 = tmp;
func2 = tmp1;
if (strstr(tmp, "wchar_t") || strstr(tmp, "unsigned short"))
return 11;
else if (strstr(tmp1, "wchar_t") || strstr(tmp1, "unsigned short"))
return 22;
while (*func1 != 0 && *func2 != 0)
{
if ( *func1 == *func2)
func1++, func2++;
else if ( func1 == strstr(func1, "wchar_t") &&
func2 == strstr(func2, "unsigned short"))
func1 += sizeof("wchar_t"), func2 += sizeof("unsigned short");
else if ( func2 == strstr(func2, "wchar_t") &&
func1 == strstr(func1, "unsigned short"))
func2 += sizeof("wchar_t"), func1 += sizeof("unsigned short");
else
break;
}
if (*func1 < *func2)
return -1;
else if (*func1 > *func2)
return 1;
else
return 0;
}
int main(int argc, char *argv[])
{
FILE *phDumpbin;
Ext Fext;
const char *Fname;
int StripLine, i, result;
char ExeName[256];
char TmpStr[512];
char *tch;
set<string> Sdiff, Wdiff, Iden;
if (!(Fname = setflags(argc, argv)))
{
cout << "Usage : wsfdiff -[options] filename" << endl;
cout << "-x -X: Use unsigned char as data type in identical functions" << endl;
cout << "-u -U: Print diff with unsigned short version" << endl;
cout << "-w -W: Print diff with wchar_t version" << endl;
cout << "-i -I: Print Identical function" << endl;
cout << "-d -D: UnDname the function names" << endl;
cout << "filename can only have be .lib or .dll" << endl;
return -1;
}
strcpy(ExeName, "dumpbin ");
switch (Fext = FindExt(Fname))
{
case dll:
strcat(ExeName, "-exports ");
break;
case lib:
strcat(ExeName, "-linkermember ");
break;
case none:
return -1;
}
strcat(ExeName, Fname);
phDumpbin = _popen(ExeName, "r");
//Strip header
if (Fext == dll)
StripLine = 19;
else
StripLine = 18;
for ( i = 0; i < StripLine; i++)
{
fgets(TmpStr, 511, phDumpbin);
cerr << TmpStr;
}
set<string>::iterator spos, wpos;
while (fgets(TmpStr, 511, phDumpbin))
{
tch = filter(TmpStr, Fext);
tch[strlen(tch) -1] = '\0';
UnDecorateSymbolName(tch, tmp, 511, 0);
if (strstr (tmp, "wchar_t"))
{
spos = Sdiff.begin();
result = -1;
while(spos != Sdiff.end())
{
switch(result = funccmp(tch, (*spos).c_str()))
{
case 1:
++spos;
break;
case 0:
if (flags & F_USE_USH_NAME)
Iden.insert(*spos);
else
Iden.insert(string(tch));
Sdiff.erase(spos);
break;
}
if ( result != 1)
{
break;
}
}
if (result != 0)
Wdiff.insert(string(tch));
} else if ( strstr(tmp, "unsigned short"))
{
spos = Wdiff.begin();
result = -1;
while(spos != Wdiff.end())
{
switch(result = funccmp(tch, (*spos).c_str()))
{
case 1:
++spos;
break;
case 0:
if (flags & F_USE_USH_NAME)
Iden.insert(string(tch));
else
Iden.insert(*spos);
Wdiff.erase(spos);
break;
}
if ( result != 1)
break;
}
if (result != 0)
Sdiff.insert(string(tch));
}
}
// final checks for diffs and validity
for ( wpos = Wdiff.begin(), spos = Sdiff.begin();
spos != Sdiff.end() && wpos != Wdiff.end();)
{
switch(Funccmp((*wpos).c_str(), (*spos).c_str()))
{
case 0:
if (flags & F_USE_USH_NAME)
Iden.insert(*spos);
else
Iden.insert(*wpos);
spos = Sdiff.erase(spos);
wpos = Wdiff.erase(wpos);
break;
case 1:
++spos;
break;
case -1:
++wpos;
break;
case 11:
wpos = Wdiff.erase(wpos);
break;
case 22:
spos = Sdiff.erase(spos);
}
}
if ( flags & F_OUT_USH)
{
cout << "Only Unsigned Short version " << Sdiff.size() << '\n';
for_each (Sdiff.begin(), Sdiff.end(), print);
cout <<"-----------------------------------------------------------\n";
}
if ( flags & F_OUT_WCH)
{
cout << "Only wchar_t versions " << Wdiff.size() << '\n';
for_each (Wdiff.begin(), Wdiff.end(), print);
cout <<"-----------------------------------------------------------\n";
}
if ( flags & F_OUT_IDEN)
{
cout <<"Both the versions are present " << Iden.size() << '\n';
for_each (Iden.begin(), Iden.end(), print );
cout <<"-----------------------------------------------------------\n";
}
return 1;
}