#include <string.h>
#include <iostream>
#include <io.h>
#include <stdio.h>
#include <windows.h>
#include <new.h>
#include "List.h"
#include "File.h"
#include "File32.h"
#include "File64.h"
#include "Object.h"
#include "depend.h"
List* pQueue;
List* pDependQueue;
List* pSearchPath;
List* pMissingFiles;
bool bNoisy;
bool bListDependencies;
DWORD dwERROR;
//
// Global variables used to get formatted message for this program.
//
HMODULE ThisModule = NULL;
WCHAR Message[4096];
// Define a function to be called if new fails to allocate memory.
//
int __cdecl MyNewHandler( size_t size )
{
_putws( GetFormattedMessage( ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_MEM_ALLOC_FAILED) );
// Exit program
//
ExitProcess(errOUT_OF_MEMORY);
}
/*
usage: depend [/s] [/l] /f:filespec;filespec;... [/d:directory;directory;..]
If directories are not specififed the Windows search path will be used to look for dependencies
/s Specifies silent mode.
filespec - file path and name. Can include wildcards.
*/
DWORD _cdecl wmain(int argc,wchar_t *argv[]) {
// Set the failure handler for new operator.
//
_set_new_handler( MyNewHandler );
TCHAR *pszFileName = new TCHAR[256];
File* pTempFile,*pCurrentFile;
StringNode* pCurFile;
char buf[256];
dwERROR = 0;
pSearchPath = 0;
bNoisy = true;
bListDependencies = false;
pQueue = new List();
pDependQueue = new List();
pMissingFiles = new List();
ThisModule = GetModuleHandle(NULL);
//Load the initial files into the queue and load the search path
if (!ParseCommandLine(argc,argv)) goto CLEANUP;
pCurFile = (StringNode*)pQueue->tail;
//while the queue isn't empty
while (pCurFile!=0) {
WideCharToMultiByte(CP_ACP,0,pCurFile->Data(),-1,buf,256,0,0);
//get a file pointer for the current file
if (!(pCurrentFile = CreateFile(pCurFile->Data()))) {
if (bListDependencies) {
StringNode* s;
if (s = (StringNode*)pDependQueue->Find(pCurFile->Data())) {
pDependQueue->Remove(pCurFile->Data());
}
}
//if there was an error and we are running in silent mode, quit
if (!bNoisy) goto CLEANUP;
}
else { //if we got a file pointer, proceed
if (bListDependencies) {
StringNode* s;
if (s = (StringNode*)pDependQueue->Find(pCurFile->Data())) {
pDependQueue->Remove(pCurFile->Data());
pDependQueue->Add(pCurrentFile);
}
}
//Check this files dependencies
pCurrentFile->CheckDependencies();
if ((dwERROR)&&(!bNoisy)) goto CLEANUP;
//Close the file
pCurrentFile->CloseFile();
}
//next file
pCurFile = (StringNode*)pCurFile->prev;
}
StringNode* s;
//if list dependencies is set, print out all dependencies
if (bListDependencies) {
pCurrentFile = (File*)pDependQueue->head;
//while the queue isn't empty
while (pCurrentFile!=0) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_DEPENDENCY_HEAD,
pCurrentFile->Data()) );
s = (StringNode*)pCurrentFile->dependencies->head;
while(s) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_FILE_NAME,
s->Data()) );
s = (StringNode*)s->next;
}
pCurrentFile = (File*)pCurrentFile->next;
}
}
//print out list of broken files
pTempFile = (File*)pMissingFiles->head;
while (pTempFile) {
if(bNoisy){
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_LIST_OF_BROKEN_FILES,
pTempFile->Data()) );
}
s = (StringNode*)pTempFile->owners->head;
while(s) {
if(bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_FILE_NAME,
s->Data()) );
}
s = (StringNode*)s->next;
}
pTempFile = (File*)pTempFile->next;
}
//Done. Clean up and go home.
if(bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_COMPLETED) );
}
CLEANUP:
delete [] pszFileName;
delete pQueue;
pszFileName = 0;pQueue = 0;
return dwERROR;
}
//Given path and filename in 'pathname', fill 'path' with just the path
void GetPath(TCHAR * pathname,TCHAR* path) {
TCHAR* end,t;
path[0] = '\0';
//find last \ in the filename
end = wcsrchr(pathname,'\\');
if (!end) return;
//copy just the path
t = end[1];
end[1] = '\0';
wcscpy(path,pathname);
end[1] = t;
return;
}
/*Fill queue with the files in the command line and fill searchpath with directories in command line
usage: depend [/s] /f:filespec;filespec;... [/d:directory;directory;..]
If directories are not specififed the Windows search path will be used to look for dependencies
/s Specifies silent mode.
filespec - file path and name. Can include wildcards.
*/
bool ParseCommandLine(int argc,wchar_t* argv[]){
HANDLE handle;
int nArg,nFile = 0;
TCHAR *pszDirectory = new TCHAR[256],*pszFileName = new TCHAR[256],*ptr;
WIN32_FIND_DATA fileData;
bool bReturn;
if (argc==1) {
//if there are no arguments, display some kind of help
if(bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_PGM_USAGE) );
}
bReturn = false;
dwERROR = errBAD_ARGUMENTS;
goto CLEANUP;
}
for (nArg=1;nArg<argc;nArg++) {
if (argv[nArg][0] == '/') {
//if this is the files argument
if (argv[nArg][1] == 'f') {
do {
ptr = wcschr(argv[nArg]+3,';');
if (ptr) {
*ptr = '\0';
wcscpy(pszFileName,ptr+1);
} else wcscpy(pszFileName,argv[nArg]+3);
//get the first file, put the path in pszDirectory
handle = FindFirstFile(pszFileName,&fileData);
GetPath(pszFileName,pszDirectory);
if (*pszDirectory=='\0') {
GetCurrentDirectory(256,pszDirectory);
pszDirectory[wcslen(pszDirectory)+1] = '\0';
pszDirectory[wcslen(pszDirectory)] = '\\';
}
//if the file wasn't found, error and quit
if (handle == INVALID_HANDLE_VALUE) {
if(bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_ARG_NOT_FOUND,
argv[nArg]) );
}
dwERROR = errFILE_NOT_FOUND;
bReturn = false;
goto CLEANUP;
}
//put each file into the queue
nFile = 1;
while (nFile) {
//standardize the name: full path, all lowercase
wcscpy(pszFileName,pszDirectory);
wcscat(pszFileName,fileData.cFileName);
_wcslwr(pszFileName);
//add the file to the queue if it's not already there
if (!pQueue->Find(pszFileName)) pQueue->Add(new StringNode(pszFileName));
//if list all dependencies is set, add to dependQueue
if (bListDependencies)
if (!pDependQueue->Find(pszFileName))
pDependQueue->Add(new StringNode(pszFileName));
nFile = FindNextFile(handle,&fileData);
}//end while files
} while (ptr);
} else if (argv[nArg][1] == 'd') {
//load directories
pSearchPath = new List();
do {
ptr = wcschr(argv[nArg]+3,';');
if (ptr) {
*ptr = '\0';
wcscpy(pszDirectory,ptr+1);
} else wcscpy(pszDirectory,argv[nArg]+3);
if (pszDirectory[wcslen(pszDirectory)-1]!='\\') {
pszDirectory[wcslen(pszDirectory)+1] = '\0';
pszDirectory[wcslen(pszDirectory)] = '\\';
}
pSearchPath->Add(new StringNode(pszDirectory));
} while (ptr);
} else
//if silent mode, turn off noisy flag
if (argv[nArg][1] == 's') bNoisy = false;
else
//if you want to list the dependencies of all files
//turn on this flag
if (argv[nArg][1] == 'l') bListDependencies = true;
else
{
//unrecognized flag
if(bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_BAD_ARGUMENT,
argv[nArg]) );
}
dwERROR = errBAD_ARGUMENTS;
bReturn = false;
goto CLEANUP;
}
} else {
//didn't start with a /
if(bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_BAD_ARGUMENT,
argv[nArg]) );
}
dwERROR = errBAD_ARGUMENTS;
bReturn = false;
goto CLEANUP;
}
}//end for arguments
bReturn = true;
CLEANUP:
delete[] pszFileName;
delete[] pszDirectory;
pszFileName = pszDirectory = 0;
return bReturn;
}
//Search for the given file in the given path
//Arguments:
//pszFileName - file to look for
//pszPathName - path to look for it in
bool SearchPath(TCHAR* pszFileName,TCHAR* pszPathName) {
StringNode* s;
WIN32_FIND_DATA buf;
if (!pSearchPath) return false;
s = (StringNode*)pSearchPath->head;
while (s) {
wcscpy(pszPathName,s->Data());
wcscat(pszPathName,pszFileName);
if (FindFirstFile(pszPathName,&buf)!=INVALID_HANDLE_VALUE) return true;
s = (StringNode*)s->next;
}
pszPathName = 0;
return false;
}
//Determine what type of file was passed in (16 bit,32,64) and create the appropriate file ptr.
//pszFileName - file to be loaded
File* CreateFile(TCHAR* pszFileName) {
try {
return new File32(pszFileName);
} catch(int x) {
if (x == errFILE_LOCKED) return 0;
try {
return new File64(pszFileName);
} catch(int x) {
if (x == errFILE_LOCKED) return 0;
if (bNoisy) {
_putws( GetFormattedMessage(ThisModule,
FALSE,
Message,
sizeof(Message)/sizeof(Message[0]),
MSG_ERROR_UNRECOGNIZED_FILE_TYPE,
pszFileName) );
}
return 0;
}
}
}