mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1590 lines
50 KiB
1590 lines
50 KiB
//--------------------------------------------------------------
|
|
//
|
|
// File: loadfiles
|
|
//
|
|
// Contents: Load files.
|
|
//
|
|
//---------------------------------------------------------------
|
|
|
|
#include "loadhead.cxx"
|
|
#pragma hdrstop
|
|
|
|
#include <common.hxx>
|
|
#include <loadstate.hxx>
|
|
#include <bothchar.hxx>
|
|
|
|
#include <section.hxx>
|
|
#include <special.hxx>
|
|
|
|
#include <fileutil.cxx>
|
|
|
|
#define CREATE_DIRECTORY_MAX_PATH 248
|
|
//---------------------------------------------------------------
|
|
|
|
CSection *g_pcsSectionList = NULL;
|
|
|
|
CSpecialDirectory g_csdCurrent;
|
|
CSpecialDirectory g_csdOld;
|
|
|
|
CRuleList g_crlExcludeWildcards;
|
|
CRuleList g_crlIncludeWildcards;
|
|
|
|
|
|
//---------------------------------------------------------------
|
|
DWORD ComputeTemp()
|
|
{
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
|
|
//---------------------------------------------------------------
|
|
void EraseTemp()
|
|
{
|
|
}
|
|
|
|
DWORD FixShellShortcut(const TCHAR *ptsFile)
|
|
{
|
|
HRESULT hr;
|
|
IShellLink *pisl = NULL;
|
|
IPersistFile *pipf = NULL;
|
|
TCHAR *ptsDest = NULL;
|
|
WIN32_FIND_DATA wfd;
|
|
|
|
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile,
|
|
"Called FixShellShortcut for %s\r\n",
|
|
ptsFile);
|
|
|
|
hr = CoCreateInstance(CLSID_ShellLink,
|
|
NULL,
|
|
CLSCTX_INPROC_SERVER,
|
|
IID_IShellLink,
|
|
(void **)&pisl);
|
|
if (FAILED(hr))
|
|
goto cleanup;
|
|
|
|
hr = pisl->QueryInterface(IID_IPersistFile,
|
|
(void **)&pipf);
|
|
if (FAILED(hr))
|
|
goto cleanup;
|
|
|
|
hr = pipf->Load(ptsFile,
|
|
0);
|
|
if (FAILED(hr))
|
|
goto cleanup;
|
|
|
|
TCHAR ptsPath[MAX_PATH + 1];
|
|
|
|
hr = pisl->GetPath(ptsPath,
|
|
MAX_PATH + 1,
|
|
&wfd,
|
|
SLGP_RAWPATH);
|
|
if (FAILED(hr))
|
|
goto cleanup;
|
|
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile,
|
|
"Retrieved path %s from shell link %s\r\n",
|
|
ptsPath,
|
|
ptsFile);
|
|
|
|
|
|
hr = WhereIsThisFile(ptsPath, &ptsDest);
|
|
if (hr == ERROR_SUCCESS)
|
|
{
|
|
//Change the shortcut
|
|
|
|
//Expand any environment strings in the original data with
|
|
//the values for the user we're loading for. If the final
|
|
//paths match, we'll retain the version of the data with the
|
|
//unexpanded environment variables in it. Otherwise, we take
|
|
//the full path.
|
|
|
|
|
|
//We could be much smarter here if we wanted to.
|
|
//For instance:
|
|
// 1) Check the old path against the old special directories
|
|
// list, and try to remap things that match.
|
|
// 2) Double check to make sure the shortcut is actually
|
|
// broken before we go ahead and fix it.
|
|
TCHAR tsExpand[MAX_PATH + 1];
|
|
TCHAR tsTemp[MAX_PATH + 1];
|
|
TCHAR *ptsFinalDest;
|
|
|
|
if (_tcslen(ptsPath) > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: ptsPath too long %s\r\n", ptsPath);
|
|
goto cleanup;
|
|
}
|
|
_tcscpy(tsExpand, ptsPath);
|
|
hr = ExpandEnvStringForUser(tsExpand,
|
|
tsTemp,
|
|
&ptsFinalDest);
|
|
if (hr)
|
|
goto cleanup;
|
|
|
|
if (_tcsicmp(ptsDest, ptsFinalDest) == 0)
|
|
{
|
|
//They're the same, use the string with the environment
|
|
//variables in it.
|
|
ptsFinalDest = ptsPath;
|
|
}
|
|
else
|
|
{
|
|
ptsFinalDest = ptsDest;
|
|
}
|
|
|
|
hr = pisl->SetPath(ptsFinalDest);
|
|
if (FAILED(hr))
|
|
goto cleanup;
|
|
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile,
|
|
"FixShellShortcut fixed %s from %s to %s\r\n",
|
|
ptsFile,
|
|
ptsPath,
|
|
ptsFinalDest);
|
|
}
|
|
|
|
//If this function failed, we leave the shortcut alone. A possible
|
|
//change is to delete the shortcut for this case.
|
|
|
|
cleanup:
|
|
if (ptsDest != NULL)
|
|
free(ptsDest);
|
|
|
|
if (pipf != NULL)
|
|
pipf->Release();
|
|
|
|
if (pisl != NULL)
|
|
pisl->Release();
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
DWORD ExpandEnvStringForUser(TCHAR *ptsString,
|
|
TCHAR *ptsTemp,
|
|
TCHAR **pptsFinal)
|
|
{
|
|
return g_csdCurrent.ExpandMacro(ptsString,
|
|
ptsTemp,
|
|
pptsFinal,
|
|
FALSE);
|
|
}
|
|
|
|
|
|
//---------------------------------------------------------------
|
|
DWORD WhereIsThisFile( const TCHAR *ptsFile, TCHAR **pptsNewFile )
|
|
{
|
|
CSection *pcsCurrent = g_pcsSectionList;
|
|
TCHAR tsExpName[MAX_PATH + 1];
|
|
TCHAR tsTemp[MAX_PATH + 1];
|
|
TCHAR *ptsFullName;
|
|
DWORD dwErr;
|
|
|
|
if (_tcslen(ptsFile) > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: ptsFile too long %s\r\n", ptsFile);
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
|
|
_tcscpy(tsExpName, ptsFile);
|
|
dwErr = g_csdOld.ExpandMacro(tsExpName, tsTemp, &ptsFullName, FALSE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
while (pcsCurrent != NULL)
|
|
{
|
|
TCHAR *ptsFileOnly;
|
|
ptsFileOnly = _tcsrchr(ptsFullName, '\\');
|
|
|
|
if ((NULL == ptsFileOnly) ||
|
|
(_tcsnicmp(ptsFullName,
|
|
pcsCurrent->GetSectionPath(),
|
|
pcsCurrent->GetSectionPathLength()) == 0))
|
|
{
|
|
//Section matches, search for file
|
|
if (NULL == ptsFileOnly)
|
|
{
|
|
ptsFileOnly = ptsFullName;
|
|
}
|
|
else
|
|
{
|
|
ptsFileOnly = ptsFullName + pcsCurrent->GetSectionPathLength() + 1;
|
|
}
|
|
|
|
for (ULONG i = 0; i < pcsCurrent->GetNameCount(); i++)
|
|
{
|
|
const TCHAR *ptsCurrent = pcsCurrent->GetFullFileName(i);
|
|
|
|
if (_tcsicmp(ptsFileOnly,
|
|
ptsCurrent) == 0)
|
|
{
|
|
TCHAR tsDest[MAX_PATH + 1];
|
|
const TCHAR *ptsDest = pcsCurrent->GetDestination(i);
|
|
|
|
if (ptsDest == NULL)
|
|
{
|
|
DWORD dwDestLen = pcsCurrent->GetSectionDestLength();
|
|
|
|
if (dwDestLen + _tcslen(ptsCurrent) + 1 > MAX_PATH)
|
|
{
|
|
if (Verbose)
|
|
{
|
|
Win32Printf(LogFile,
|
|
"Error: destination too long %s\\%s\r\n",
|
|
pcsCurrent->GetSectionDest(),
|
|
ptsCurrent);
|
|
}
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
_tcscpy(tsDest, pcsCurrent->GetSectionDest());
|
|
tsDest[dwDestLen] = TEXT('\\');
|
|
_tcscpy(tsDest + dwDestLen + 1, ptsCurrent);
|
|
ptsDest = tsDest;
|
|
}
|
|
|
|
//Bingo, direct hit.
|
|
*pptsNewFile = (TCHAR *) malloc(
|
|
(_tcslen(ptsDest) + 1) * sizeof(TCHAR));
|
|
|
|
if (*pptsNewFile == NULL)
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
_tcscpy( *pptsNewFile, ptsDest);
|
|
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile,
|
|
"WhereIsThisFile(%ws) found %ws\r\n",
|
|
ptsFile,
|
|
*pptsNewFile);
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
}
|
|
}
|
|
pcsCurrent = pcsCurrent->GetNextSection();
|
|
}
|
|
|
|
return ERROR_NOT_FOUND;
|
|
}
|
|
|
|
|
|
DWORD AddInfSectionToRuleList(INFCONTEXT *pic,
|
|
CRuleList *pfl,
|
|
BOOL fAllowRename)
|
|
{
|
|
TCHAR buf[MAX_PATH + 1];
|
|
TCHAR bufMacro[MAX_PATH + 1];
|
|
TCHAR *ptsFinalName;
|
|
TCHAR bufTag[MAX_PATH + 1];
|
|
DWORD dwErr;
|
|
CRuleList *prl;
|
|
|
|
do
|
|
{
|
|
DWORD dwIndex;
|
|
DWORD cFields;
|
|
BOOL fDirectoryTag = FALSE;
|
|
cFields = SetupGetFieldCount(pic);
|
|
|
|
if (((cFields != 1) && !fAllowRename) ||
|
|
((cFields > 2) && fAllowRename))
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"Line contains more than one file name\r\n");
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if (!SetupGetStringField(pic,
|
|
1,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField returned %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
if (SetupGetStringField(pic,
|
|
0,
|
|
bufTag,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
if (_tcsicmp(bufTag, buf))
|
|
{
|
|
//Someone put a field identifier on there. The only
|
|
//one we recognize is 'dir'
|
|
if (_tcsicmp(bufTag, TEXT("dir")))
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"Unknown tag %s\r\n",
|
|
bufTag);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
fDirectoryTag = TRUE;
|
|
}
|
|
}
|
|
|
|
dwErr = g_csdOld.ExpandMacro(buf, bufMacro, &ptsFinalName, TRUE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
if (fDirectoryTag)
|
|
{
|
|
//Append a backslash
|
|
|
|
if (_tcslen(ptsFinalName) >= MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
{
|
|
Win32Printf(LogFile, "Error: ptsFinalName too long %s\r\n", ptsFinalName);
|
|
}
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
|
|
_tcscat(ptsFinalName, TEXT("\\"));
|
|
}
|
|
|
|
dwErr = pfl->SetName(ptsFinalName, &prl);
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
|
|
if (cFields == 2)
|
|
{
|
|
if (!SetupGetStringField(pic,
|
|
2,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField returned %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
dwErr = prl->SetDestination(buf);
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
}
|
|
}
|
|
while (SetupFindNextLine(pic, pic));
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
DWORD ProcessCopyFiles(HINF hi, const TCHAR *ptsName)
|
|
{
|
|
DWORD dwErr;
|
|
INFCONTEXT ic;
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
ptsName,
|
|
NULL,
|
|
&ic))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_SECTION_NAME_NOT_FOUND, ptsName);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindFirstLine failed on section %s with %lu\r\n",
|
|
ptsName,
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
dwErr = AddInfSectionToRuleList(&ic, &g_crlIncludeWildcards, TRUE);
|
|
|
|
return dwErr;
|
|
}
|
|
|
|
DWORD ProcessDelFiles(HINF hi, const TCHAR *ptsName)
|
|
{
|
|
DWORD dwErr;
|
|
INFCONTEXT ic;
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
ptsName,
|
|
NULL,
|
|
&ic))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_SECTION_NAME_NOT_FOUND, ptsName);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindFirstLine failed on section %s with %lu\r\n",
|
|
ptsName,
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
dwErr = AddInfSectionToRuleList(&ic, &g_crlExcludeWildcards, FALSE);
|
|
|
|
return dwErr;
|
|
}
|
|
|
|
DWORD ProcessRules(HINF hi)
|
|
{
|
|
DWORD dwErr;
|
|
INFCONTEXT ic;
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
EXTENSION_SECTION,
|
|
NULL,
|
|
&ic))
|
|
{
|
|
//Ignore - this section is optional
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
do
|
|
{
|
|
DWORD cFields;
|
|
cFields = SetupGetFieldCount(&ic);
|
|
TCHAR buf[MAX_PATH + 1];
|
|
|
|
if (!SetupGetStringField(&ic,
|
|
0,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
if (_tcsicmp(buf, COPYFILES_LABEL) == 0)
|
|
{
|
|
//Add files in all sections to the include list
|
|
for (DWORD j = 1; j < cFields + 1; j++)
|
|
{
|
|
if (!SetupGetStringField(&ic,
|
|
j,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
dwErr = ProcessCopyFiles(hi, buf);
|
|
if (dwErr != 0)
|
|
return dwErr;
|
|
}
|
|
}
|
|
else if (_tcsicmp(buf, DELFILES_LABEL) == 0)
|
|
{
|
|
//Add files in all sections to the include list
|
|
for (DWORD j = 1; j < cFields + 1; j++)
|
|
{
|
|
if (!SetupGetStringField(&ic,
|
|
j,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
dwErr = ProcessDelFiles(hi, buf);
|
|
if (dwErr != 0)
|
|
return dwErr;
|
|
}
|
|
}
|
|
}
|
|
while (SetupFindNextLine(&ic, &ic));
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
DWORD ProcessSpecialDirs(HINF hi)
|
|
{
|
|
DWORD dwErr;
|
|
INFCONTEXT ic;
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
SPECIALDIRS_SECTION,
|
|
NULL,
|
|
&ic))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile,
|
|
IDS_SECTION_NAME_NOT_FOUND,
|
|
SPECIALDIRS_SECTION);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindFirstLine failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
do
|
|
{
|
|
TCHAR bufName[MAX_PATH + 1];
|
|
TCHAR bufPath[MAX_PATH + 2];
|
|
DWORD cFields;
|
|
cFields = SetupGetFieldCount(&ic);
|
|
|
|
if (cFields != 1)
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"INF line contains too many fields in "
|
|
"section %s\r\n",
|
|
SPECIALDIRS_SECTION);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if (!SetupGetStringField(&ic,
|
|
0,
|
|
bufName,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField returned %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
if (!SetupGetStringField(&ic,
|
|
1,
|
|
bufPath,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField returned %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
|
|
dwErr = g_csdOld.InitFromInf(bufName, bufPath);
|
|
if (dwErr)
|
|
return dwErr;
|
|
}
|
|
while (SetupFindNextLine(&ic, &ic));
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
DWORD AddLoadFileSection(HINF hi, TCHAR *ptsName, CSection **ppcs)
|
|
{
|
|
DWORD dwErr;
|
|
INFCONTEXT ic;
|
|
CSection *pcsSection;
|
|
TCHAR tsSection[MAX_PATH + 1];
|
|
TCHAR buf[MAX_PATH + 1];
|
|
TCHAR bufDest[MAX_PATH + 1];
|
|
TCHAR bufMacro[MAX_PATH + 1];
|
|
TCHAR *ptsFinalName;
|
|
|
|
pcsSection = new CSection;
|
|
if (pcsSection == NULL)
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_NOT_ENOUGH_MEMORY);
|
|
return ERROR_OUTOFMEMORY;
|
|
}
|
|
|
|
if (ppcs)
|
|
*ppcs = pcsSection;
|
|
|
|
if (g_pcsSectionList == NULL)
|
|
{
|
|
g_pcsSectionList = pcsSection;
|
|
}
|
|
else
|
|
{
|
|
g_pcsSectionList->AddToList(pcsSection);
|
|
}
|
|
|
|
dwErr = pcsSection->SetSectionTitle(ptsName);
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
|
|
if (_tcslen(ptsName) > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: ptsName too long %s\r\n", ptsName);
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
_tcscpy(buf, ptsName);
|
|
dwErr = g_csdOld.ExpandMacro(buf, bufMacro, &ptsFinalName, TRUE);
|
|
if (dwErr)
|
|
{
|
|
//Try with the current list
|
|
dwErr = g_csdCurrent.ExpandMacro(buf, bufMacro, &ptsFinalName, FALSE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
}
|
|
|
|
dwErr = pcsSection->SetSectionPath(ptsFinalName);
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
|
|
const TCHAR *ptsSectionPath = pcsSection->GetSectionPath();
|
|
|
|
if (_tcslen(ptsSectionPath) + 1 > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: ptsSectionPath too long %s\r\n", ptsSectionPath);
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
_tcscpy(tsSection, ptsSectionPath);
|
|
_tcscat(tsSection, TEXT("\\"));
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
ptsName,
|
|
NULL,
|
|
&ic))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_SECTION_NAME_NOT_FOUND, ptsName);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindFirstLine failed on section %s with %lu\r\n",
|
|
ptsName,
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
do
|
|
{
|
|
DWORD cFields;
|
|
cFields = SetupGetFieldCount(&ic);
|
|
|
|
if ((cFields != 1) && (cFields != 2))
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"INF line contains too many fields in "
|
|
"section %s\r\n",
|
|
ptsName);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if (!SetupGetStringField(&ic,
|
|
1,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField returned %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
dwErr = g_csdCurrent.ExpandMacro(buf,
|
|
bufMacro,
|
|
&ptsFinalName,
|
|
TRUE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
//Check if we're supposed to exclude this file by a rule
|
|
CRuleList *prl;
|
|
if (!g_crlExcludeWildcards.MatchAgainstRuleList(tsSection,
|
|
buf,
|
|
&prl,
|
|
NULL))
|
|
{
|
|
DWORD i;
|
|
dwErr = pcsSection->SetName(ptsFinalName, &i, FALSE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
if (cFields == 2)
|
|
{
|
|
if (!SetupGetStringField(&ic,
|
|
2,
|
|
bufDest,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField returned %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
dwErr = g_csdCurrent.ExpandMacro(bufDest,
|
|
bufMacro,
|
|
&ptsFinalName,
|
|
TRUE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
dwErr = pcsSection->SetDestination(ptsFinalName, i);
|
|
if (dwErr)
|
|
return dwErr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (Verbose)
|
|
Win32Printf(LogFile,
|
|
"Excluding %s by rule %s\r\n",
|
|
buf,
|
|
prl->GetFullName());
|
|
}
|
|
}
|
|
while (SetupFindNextLine(&ic, &ic));
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
|
|
DWORD CopyAllFiles(void)
|
|
{
|
|
DWORD dwErr;
|
|
CSection *pcs = g_pcsSectionList;
|
|
|
|
DWORD ccMigPath;
|
|
TCHAR tsFinalSource[MAX_PATH + 1];
|
|
TCHAR tsSectionDest[MAX_PATH + 1];
|
|
TCHAR tsFinalDest[MAX_PATH + 1];
|
|
TCHAR *ptsJustFile;
|
|
DWORD ccPredictedLength;
|
|
DWORD dwReturnUp = ERROR_SUCCESS;
|
|
|
|
#ifdef UNICODE
|
|
if (_tcslen(wcsMigrationPath) > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: wcsMigrationPath too long %s\r\n", wcsMigrationPath);
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
wcscpy(tsFinalSource, wcsMigrationPath);
|
|
ccMigPath = wcslen(wcsMigrationPath);
|
|
#else
|
|
if (_tcslen(MigrationPath) > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: MigrationPath too long %s\r\n", MigrationPath);
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
strcpy(tsFinalSource, MigrationPath);
|
|
ccMigPath = strlen(MigrationPath);
|
|
#endif
|
|
|
|
while (pcs != NULL)
|
|
{
|
|
ULONG ulNameCount = pcs->GetNameCount();
|
|
|
|
//Add section to source path
|
|
const TCHAR *ptsSection = pcs->GetSectionPath();
|
|
|
|
DWORD ccSection = pcs->GetSectionPathLength() + ccMigPath;
|
|
if ( ccSection + 1 > MAX_PATH )
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
ptsSection);
|
|
if (Verbose)
|
|
{
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Source Filename: %s\\%s\r\n",
|
|
tsFinalSource,
|
|
ptsSection);
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
pcs = pcs->GetNextSection();
|
|
continue;
|
|
}
|
|
tsFinalSource[ccMigPath] = TEXT('\\');
|
|
tsFinalSource[ccMigPath + 1] = ptsSection[0];
|
|
_tcscpy(tsFinalSource + ccMigPath + 2, ptsSection + 2);
|
|
|
|
tsFinalSource[ccSection++] = TEXT('\\');
|
|
tsFinalSource[ccSection] = 0;
|
|
|
|
// Build Destination Path
|
|
INT_PTR ccDest = pcs->GetSectionDestLength();
|
|
if ( ccDest + 1 > MAX_PATH )
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
pcs->GetSectionDest);
|
|
if (Verbose)
|
|
{
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Destination Filename: %s\r\n",
|
|
pcs->GetSectionDest);
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
pcs = pcs->GetNextSection();
|
|
continue;
|
|
}
|
|
|
|
_tcscpy(tsSectionDest, pcs->GetSectionDest());
|
|
if (tsSectionDest[ccDest - 1] != TEXT('\\'))
|
|
{
|
|
tsSectionDest[ccDest++] = TEXT('\\');
|
|
tsSectionDest[ccDest] = 0;
|
|
}
|
|
|
|
for (ULONG i = 0; i < ulNameCount; i++)
|
|
{
|
|
TCHAR *ptsDestFinal;
|
|
const TCHAR *ptsName = pcs->GetFullFileName(i);
|
|
const TCHAR *ptsDest = pcs->GetDestination(i);
|
|
|
|
if (ptsDest != NULL)
|
|
{
|
|
TCHAR *ptsTopDir;
|
|
|
|
// File is explicitly being migrated
|
|
// Build the destination filename with these pieces:
|
|
// - Destination dir specified in the file rule
|
|
// - Last piece of the path in the section heading
|
|
// - Filename
|
|
|
|
DWORD ccFinalDest = _tcslen(ptsDest);
|
|
if (ccFinalDest > MAX_PATH)
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_FILE_COPYERROR, ptsName);
|
|
if (Verbose)
|
|
Win32Printf(STDERR, "Skipping Too Long Destination Filename: %s\r\n",
|
|
ptsDest);
|
|
continue;
|
|
}
|
|
_tcscpy(tsFinalDest, ptsDest);
|
|
|
|
if (tsFinalDest[ccFinalDest] != TEXT('\\') && ccFinalDest < MAX_PATH)
|
|
{
|
|
tsFinalDest[ccFinalDest++] = TEXT('\\');
|
|
}
|
|
|
|
ptsTopDir = _tcsrchr( ptsSection, TEXT('\\'));
|
|
if (ptsTopDir != NULL)
|
|
{
|
|
ptsTopDir++; // move past the '\'
|
|
}
|
|
|
|
// Skip this if we're going to create a filename that is too long
|
|
ccPredictedLength = ccFinalDest + _tcslen(ptsName);
|
|
if (ptsTopDir != NULL)
|
|
{
|
|
ccPredictedLength += _tcslen(ptsTopDir) + 1;
|
|
}
|
|
if ( ccPredictedLength > MAX_PATH )
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
ptsName);
|
|
if (Verbose)
|
|
{
|
|
tsFinalDest[ccFinalDest] = 0; // Null terminate for printing
|
|
if (ptsTopDir == NULL)
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Destination Filename: %s%s\r\n",
|
|
tsFinalDest,
|
|
ptsName);
|
|
else
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Destination Filename: %s%s\\%s\r\n",
|
|
tsFinalDest,
|
|
ptsTopDir,
|
|
ptsName);
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (ptsTopDir != NULL)
|
|
{
|
|
_tcscpy(tsFinalDest + ccFinalDest, ptsTopDir);
|
|
ccFinalDest += _tcslen(ptsTopDir);
|
|
if (tsFinalDest[ccFinalDest] != TEXT('\\') )
|
|
{
|
|
tsFinalDest[ccFinalDest++] = TEXT('\\');
|
|
}
|
|
}
|
|
_tcscpy(tsFinalDest + ccFinalDest, ptsName);
|
|
ptsDestFinal = tsFinalDest;
|
|
}
|
|
else
|
|
{
|
|
if ( ccDest + _tcslen(ptsName) > MAX_PATH )
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
ptsName);
|
|
if (Verbose)
|
|
{
|
|
tsFinalDest[ccDest] = 0; // Null terminate for printing
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Destination Filename: %s%s\r\n",
|
|
tsFinalDest,
|
|
ptsName);
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
continue;
|
|
}
|
|
//Use section destination
|
|
_tcscpy(tsSectionDest + ccDest, ptsName);
|
|
ptsDestFinal = tsSectionDest;
|
|
}
|
|
|
|
DWORD_PTR ccDestDir;
|
|
// If the directory is more than 248 characters, then CreateDirectory will fail
|
|
// There is no system define for 248, unfortunately.. It's only mentioned in the
|
|
// documentation. How odd.
|
|
ptsJustFile = _tcsrchr(ptsDestFinal, '\\');
|
|
if ( ptsJustFile == NULL )
|
|
{
|
|
ccDestDir = _tcslen(ptsDestFinal);
|
|
}
|
|
else
|
|
{
|
|
ccDestDir = ptsJustFile - ptsDestFinal;
|
|
ptsJustFile++; // Move past the '\'
|
|
}
|
|
if ( ccDestDir > CREATE_DIRECTORY_MAX_PATH )
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
ptsName);
|
|
if (Verbose)
|
|
{
|
|
ptsDestFinal[ccDestDir] = 0; // Null terminate for printing
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Destination Directory: %s for %s\r\n",
|
|
ptsDestFinal,
|
|
ptsJustFile ? ptsJustFile : TEXT("file list") );
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
// Windows appears to enforce that an existing file cannot exceed MAX_PATH,
|
|
// but we'll check just to make sure.
|
|
if ( (ccSection + _tcslen(ptsName)) > MAX_PATH )
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
ptsName);
|
|
if (Verbose)
|
|
{
|
|
tsFinalSource[ccSection] = 0; // Null terminate for printing
|
|
Win32Printf(STDERR,
|
|
"Skipping Too Long Source Filename: %s\\%s\r\n",
|
|
tsFinalSource,
|
|
ptsName);
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
continue;
|
|
}
|
|
_tcscpy(tsFinalSource + ccSection, ptsName);
|
|
|
|
// Store the full destination in the file list
|
|
pcs->SetDestination(ptsDestFinal, i);
|
|
|
|
//Finally we have the filenames constructed, now try
|
|
//the CopyFile operation
|
|
|
|
BOOL fPath = FALSE;
|
|
ULONG ulVersion = 1;
|
|
TCHAR tsCollision[MAX_PATH + 1];
|
|
INT_PTR ccExt;
|
|
TCHAR *ptsDestOriginal = ptsDestFinal;
|
|
|
|
if (DebugOutput)
|
|
{
|
|
Win32Printf(LogFile, "Copying %s to %s\r\n", tsFinalSource, ptsDestFinal);
|
|
}
|
|
|
|
while (!CopyFile(tsFinalSource,
|
|
ptsDestFinal,
|
|
TRUE))
|
|
{
|
|
dwErr = GetLastError();
|
|
|
|
if (dwErr == ERROR_PATH_NOT_FOUND)
|
|
{
|
|
if (fPath)
|
|
{
|
|
//We already tried to create the path, so something
|
|
//else must be wrong. Punt-arooney.
|
|
break;
|
|
}
|
|
|
|
dwErr = ERROR_SUCCESS;
|
|
|
|
TCHAR *ptsPos;
|
|
DWORD dwPos;
|
|
//Try to create all the necessary directories
|
|
TCHAR ptsDirectory[MAX_PATH + 1];
|
|
|
|
// ptsDestFinal was built inside this function and verified to
|
|
// be less than MAX_PATH in length
|
|
_tcscpy(ptsDirectory, ptsDestFinal);
|
|
dwPos = 0;
|
|
|
|
// Skip any leading drive specifier.
|
|
if (ptsDirectory[0] == TEXT('\\'))
|
|
dwPos = 1;
|
|
else if (ptsDirectory[0] != 0 &&
|
|
ptsDirectory[1] == TEXT(':'))
|
|
if (ptsDirectory[2] == TEXT('\\'))
|
|
dwPos = 3;
|
|
else
|
|
dwPos = 2;
|
|
|
|
//Create every directory along this path
|
|
while (ptsPos = _tcschr(ptsDirectory + dwPos, TEXT('\\')))
|
|
{
|
|
*ptsPos = 0;
|
|
//Create the directory
|
|
|
|
if (!CreateDirectory(ptsDirectory,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
if (dwErr != ERROR_ALREADY_EXISTS)
|
|
{
|
|
break;
|
|
}
|
|
dwErr = ERROR_SUCCESS;
|
|
}
|
|
|
|
//Put the backslash back in
|
|
*ptsPos = TEXT('\\');
|
|
//Update dwLen
|
|
dwPos = ptsPos - ptsDirectory + 1;
|
|
}
|
|
if (dwErr)
|
|
break;
|
|
fPath = TRUE;
|
|
}
|
|
else if (dwErr == ERROR_FILE_EXISTS)
|
|
{
|
|
TCHAR tsSquigs[MAX_PATH];
|
|
TCHAR *ptsDestSquig;
|
|
INT_PTR ccSquig;
|
|
|
|
//Add squiggles until we get an OK name.
|
|
if (ptsDestFinal != tsCollision)
|
|
{
|
|
TCHAR *ptsDestExt;
|
|
|
|
//First time
|
|
ptsDestExt = _tcsrchr(ptsDestOriginal, TEXT('.'));
|
|
if (ptsDestExt == NULL)
|
|
{
|
|
//No extension, just tack onto the end.
|
|
ccExt = _tcslen(ptsDestOriginal);
|
|
}
|
|
else
|
|
{
|
|
ccExt = ptsDestExt - ptsDestOriginal;
|
|
}
|
|
|
|
// ptsDestOriginal was built inside this function and verified to
|
|
// be less than MAX_PATH in length
|
|
_tcscpy(tsCollision, ptsDestOriginal);
|
|
ptsDestFinal = tsCollision;
|
|
|
|
// temporarily terminate the original to find the squig.
|
|
if( ptsDestExt != NULL )
|
|
*ptsDestExt = TEXT('\0');
|
|
|
|
ptsDestSquig = _tcsrchr(ptsDestOriginal, TEXT('('));
|
|
if( ptsDestSquig == NULL )
|
|
{
|
|
ccSquig = ccExt;
|
|
}
|
|
else
|
|
{
|
|
ccSquig = ptsDestSquig - ptsDestOriginal;
|
|
}
|
|
|
|
// put the period back where we took it off.
|
|
if( ptsDestExt != NULL )
|
|
*ptsDestExt = TEXT('.');
|
|
|
|
}
|
|
wsprintf(tsSquigs,
|
|
TEXT("(%lu)"),
|
|
ulVersion++);
|
|
if (_tcslen(ptsDestOriginal) + _tcslen(tsSquigs) > MAX_PATH)
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
tsFinalSource);
|
|
if (Verbose)
|
|
{
|
|
Win32Printf(STDERR,
|
|
"Could Not Copy To Too Long Destination Filename %s\r\n",
|
|
ptsDestOriginal);
|
|
}
|
|
if (dwReturnUp == ERROR_SUCCESS)
|
|
{
|
|
dwReturnUp = ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
continue;
|
|
}
|
|
wsprintf(tsCollision + ccSquig,
|
|
TEXT("%s%s"),
|
|
tsSquigs,
|
|
ptsDestOriginal + ccExt);
|
|
|
|
//Go back around and try again.
|
|
}
|
|
else
|
|
{
|
|
Win32PrintfResource(LogFile,
|
|
IDS_FILE_COPYERROR,
|
|
tsFinalSource);
|
|
if (Verbose)
|
|
{
|
|
Win32Printf(STDERR,
|
|
"Error %lu trying to copy %s to %s\r\n",
|
|
dwErr,
|
|
tsFinalSource,
|
|
ptsDestFinal);
|
|
}
|
|
dwReturnUp = dwErr;
|
|
break;
|
|
}
|
|
}
|
|
if (ptsDestFinal == tsCollision)
|
|
{
|
|
dwErr = pcs->SetDestination(tsCollision, i);
|
|
Win32Printf(LogFile,
|
|
"Filename collision on %s, file renamed to %s\r\n",
|
|
ptsDestOriginal,
|
|
ptsDestFinal);
|
|
}
|
|
|
|
//Check if the file has a .lnk extension.
|
|
TCHAR *ptsLastDot;
|
|
ptsLastDot = _tcsrchr(ptsDestFinal, TEXT('.'));
|
|
if (ptsLastDot != NULL)
|
|
{
|
|
if (_tcsicmp(ptsLastDot + 1, TEXT("lnk")) == 0)
|
|
{
|
|
//It's a link, try to fix it. Ignore errors.
|
|
FixShellShortcut(ptsDestFinal);
|
|
}
|
|
}
|
|
}
|
|
pcs = pcs->GetNextSection();
|
|
}
|
|
|
|
return dwReturnUp;
|
|
}
|
|
|
|
|
|
DWORD ParseInputFile(HINF hi)
|
|
{
|
|
DWORD dwErr;
|
|
CSection *pcs;
|
|
BOOL fMapping = FALSE;
|
|
INFCONTEXT ic;
|
|
INFCONTEXT icDestinationDirs;
|
|
INFCONTEXT icDirectoryMapping;
|
|
TCHAR tsFirstDestKey[MAX_PATH + 1];
|
|
TCHAR tsFirstMapping[10];
|
|
|
|
dwErr = ProcessSpecialDirs(hi);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
dwErr = ProcessRules(hi);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
COPYFILE_SECTION,
|
|
NULL,
|
|
&ic))
|
|
{
|
|
dwErr = GetLastError();
|
|
|
|
if (dwErr == ERROR_LINE_NOT_FOUND)
|
|
{
|
|
if (Verbose)
|
|
Win32Printf(LogFile,
|
|
"Warning: No [Copy These Files] section found.\r\n");
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
Win32PrintfResource(LogFile,
|
|
IDS_SECTION_NAME_NOT_FOUND,
|
|
COPYFILE_SECTION);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindFirstLine failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
|
|
if (!SetupFindFirstLine(hi,
|
|
DESTINATIONDIRS_SECTION,
|
|
NULL,
|
|
&icDestinationDirs))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile,
|
|
IDS_SECTION_NAME_NOT_FOUND,
|
|
DESTINATIONDIRS_SECTION);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindFirstLine failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
//Get the key for the first line, we'll need it later.
|
|
if (!SetupGetStringField(&icDestinationDirs,
|
|
0,
|
|
tsFirstDestKey,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
//Error
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField couldn't get "
|
|
"first line in %s\r\n",
|
|
DESTINATIONDIRS_SECTION);
|
|
return dwErr;
|
|
}
|
|
|
|
|
|
if (SetupFindFirstLine(hi,
|
|
DIRECTORYMAPPING_SECTION,
|
|
NULL,
|
|
&icDirectoryMapping))
|
|
{
|
|
fMapping = TRUE;
|
|
//Get the first key, we'll need it later.
|
|
|
|
if (!SetupGetStringField(&icDirectoryMapping,
|
|
0,
|
|
tsFirstMapping,
|
|
10,
|
|
NULL))
|
|
{
|
|
//Error
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField couldn't get "
|
|
"first line in %s, error %lX\r\n",
|
|
DIRECTORYMAPPING_SECTION,
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
//Ignore errors here, this section is optional
|
|
}
|
|
|
|
do
|
|
{
|
|
DWORD cFields;
|
|
cFields = SetupGetFieldCount(&ic);
|
|
TCHAR buf[MAX_PATH + 1];
|
|
|
|
if (cFields != 1)
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
if (!SetupGetStringField(&ic,
|
|
0,
|
|
buf,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField failed with %lu\r\n",
|
|
dwErr);
|
|
return dwErr;
|
|
}
|
|
|
|
dwErr = AddLoadFileSection(hi, buf, &pcs);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
|
|
if (fMapping)
|
|
{
|
|
INFCONTEXT icMap;
|
|
TCHAR tsMapping[MAX_PATH + 1];
|
|
|
|
//Find the destination path for this directory.
|
|
|
|
//First we have to check the first line, because
|
|
// of SetupFindNextMatchLine
|
|
|
|
if (_tcscmp(tsFirstDestKey, pcs->GetSectionTitle()) == 0)
|
|
{
|
|
//It's the first line
|
|
icMap = icDestinationDirs;
|
|
}
|
|
else if (!SetupFindNextMatchLine(&icDestinationDirs,
|
|
pcs->GetSectionTitle(),
|
|
&icMap))
|
|
{
|
|
//This is an error - we should have output this line
|
|
//ourselves in scanstate.
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupFindNextMatch couldn't find "
|
|
"destination dir for %s\r\n",
|
|
pcs->GetSectionTitle());
|
|
return dwErr;
|
|
}
|
|
|
|
if (!SetupGetStringField(&icMap,
|
|
1,
|
|
tsMapping,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
//Error, malformed INF
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetIntField couldn't get "
|
|
"destination dir for %s\r\n",
|
|
pcs->GetSectionTitle());
|
|
return dwErr;
|
|
}
|
|
|
|
//Now look this up in the DirectoryMapping section
|
|
if (_tcscmp(tsFirstMapping, tsMapping) == 0)
|
|
icMap = icDirectoryMapping;
|
|
if ((_tcscmp(tsFirstMapping, tsMapping) == 0) ||
|
|
(SetupFindNextMatchLine(&icDirectoryMapping,
|
|
tsMapping,
|
|
&icMap)))
|
|
{
|
|
TCHAR bufDest[MAX_PATH + 1];
|
|
if (!SetupGetStringField(&icMap,
|
|
1,
|
|
bufDest,
|
|
MAX_PATH + 1,
|
|
NULL))
|
|
{
|
|
//Error
|
|
dwErr = GetLastError();
|
|
Win32PrintfResource(LogFile, IDS_INF_ERROR);
|
|
if (Verbose)
|
|
Win32Printf(STDERR,
|
|
"SetupGetStringField couldn't get "
|
|
"directory mapping for %s\r\n",
|
|
tsMapping);
|
|
return dwErr;
|
|
}
|
|
dwErr = pcs->SetSectionDest(bufDest);
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//Fall through
|
|
}
|
|
|
|
}
|
|
|
|
|
|
if ((pcs->GetSectionDest())[0] == 0)
|
|
{
|
|
//Check for implicit relocation, for special directories
|
|
TCHAR bufMacro[MAX_PATH + 1];
|
|
TCHAR *ptsFinal;
|
|
const TCHAR *ptsSectionTitle = pcs->GetSectionTitle();
|
|
|
|
if (_tcslen(ptsSectionTitle) > MAX_PATH)
|
|
{
|
|
if (DebugOutput)
|
|
Win32Printf(LogFile, "Error: ptsSectionTitle too long %s\r\n", ptsSectionTitle);
|
|
return ERROR_FILENAME_EXCED_RANGE;
|
|
}
|
|
_tcscpy(buf, ptsSectionTitle);
|
|
dwErr = g_csdCurrent.ExpandMacro(buf, bufMacro, &ptsFinal, TRUE);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
//Compare to section path - if they're different, set the
|
|
//section destination to the new section
|
|
if (_tcsicmp(ptsFinal, pcs->GetSectionPath()) != 0)
|
|
{
|
|
dwErr = pcs->SetSectionDest(ptsFinal);
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ((pcs->GetSectionDest())[0] == 0)
|
|
{
|
|
dwErr = pcs->SetSectionDest(pcs->GetSectionPath());
|
|
if (dwErr)
|
|
{
|
|
return dwErr;
|
|
}
|
|
}
|
|
}
|
|
while (SetupFindNextLine(&ic, &ic));
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
|
|
//---------------------------------------------------------------
|
|
DWORD LoadFiles()
|
|
{
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
if (UserPath == NULL)
|
|
{
|
|
UserPath = (TCHAR *) malloc(MAX_PATH + 1);
|
|
if (UserPath == NULL)
|
|
{
|
|
Win32PrintfResource(LogFile, IDS_NOT_ENOUGH_MEMORY);
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
return dwErr;
|
|
}
|
|
|
|
//We need this but it hasn't been set in LoadUser, so get the
|
|
//USERPROFILE variable for the current user and put it in there.
|
|
dwErr = GetEnvironmentVariable(TEXT("USERPROFILE"),
|
|
UserPath,
|
|
MAX_PATH + 1);
|
|
if (dwErr == 0)
|
|
{
|
|
//Fatal error.
|
|
Win32PrintfResource(LogFile, IDS_NOT_ENOUGH_MEMORY);
|
|
dwErr = ERROR_OUTOFMEMORY;
|
|
return dwErr;
|
|
}
|
|
}
|
|
|
|
dwErr = g_csdCurrent.InitForUser(CurrentUser);
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
if (DebugOutput)
|
|
{
|
|
for (ULONG i = 0; i < g_csdCurrent.GetDirectoryCount(); i++)
|
|
{
|
|
if (g_csdCurrent.GetDirectoryPath(i) != NULL)
|
|
{
|
|
Win32Printf(LogFile,
|
|
"%s=%s\r\n",
|
|
g_csdCurrent.GetDirectoryName(i),
|
|
g_csdCurrent.GetDirectoryPath(i));
|
|
}
|
|
}
|
|
Win32Printf(LogFile, "\r\n");
|
|
}
|
|
|
|
dwErr = ParseInputFile(InputInf);
|
|
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
//If CopyFiles is FALSE, do nothing
|
|
if (!CopyFiles)
|
|
return ERROR_SUCCESS;
|
|
|
|
dwErr = CopyAllFiles();
|
|
if (dwErr)
|
|
return dwErr;
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|