xcopy: Add support for /EXCLUDELIST:file1+file2 etc.

oldstable
Jason Edmeades 2007-03-30 19:20:17 +01:00 committed by Alexandre Julliard
parent f44cb11a73
commit 54b1d19c5b
1 changed files with 159 additions and 1 deletions

View File

@ -57,6 +57,7 @@
#define OPT_SRCPROMPT 0x00004000
#define OPT_ARCHIVEONLY 0x00008000
#define OPT_REMOVEARCH 0x00010000
#define OPT_EXCLUDELIST 0x00020000
#define MAXSTRING 8192
@ -70,9 +71,20 @@ static int XCOPY_DoCopy(WCHAR *srcstem, WCHAR *srcspec,
WCHAR *deststem, WCHAR *destspec,
DWORD flags);
static BOOL XCOPY_CreateDirectory(const WCHAR* path);
static BOOL XCOPY_ProcessExcludeList(WCHAR* parms);
static BOOL XCOPY_ProcessExcludeFile(WCHAR* filename, WCHAR* endOfName);
/* Typedefs */
typedef struct _EXCLUDELIST
{
struct _EXCLUDELIST *next;
WCHAR *name;
} EXCLUDELIST;
/* Global variables */
static ULONG filesCopied = 0; /* Number of files copied */
static EXCLUDELIST *excludeList = NULL; /* Excluded strings list */
static const WCHAR wchr_slash[] = {'\\', 0};
static const WCHAR wchr_star[] = {'*', 0};
static const WCHAR wchr_dot[] = {'.', 0};
@ -107,6 +119,7 @@ int main (int argc, char *argv[])
const WCHAR PROMPTSTR1[] = {'/', 'Y', 0};
const WCHAR PROMPTSTR2[] = {'/', 'y', 0};
const WCHAR COPYCMD[] = {'C', 'O', 'P', 'Y', 'C', 'M', 'D', 0};
const WCHAR EXCLUDE[] = {'E', 'X', 'C', 'L', 'U', 'D', 'E', ':', 0};
/*
* Parse the command line
@ -155,7 +168,6 @@ int main (int argc, char *argv[])
switch (toupper(argvW[0][1])) {
case 'I': flags |= OPT_ASSUMEDIR; break;
case 'S': flags |= OPT_RECURSIVE; break;
case 'E': flags |= OPT_EMPTYDIR; break;
case 'Q': flags |= OPT_QUIET; break;
case 'F': flags |= OPT_FULL; break;
case 'L': flags |= OPT_SIMULATE; break;
@ -171,6 +183,32 @@ int main (int argc, char *argv[])
case 'A': flags |= OPT_ARCHIVEONLY; break;
case 'M': flags |= OPT_ARCHIVEONLY |
OPT_REMOVEARCH; break;
/* E can be /E or /EXCLUDE */
case 'E': if (CompareString (LOCALE_USER_DEFAULT,
NORM_IGNORECASE | SORT_STRINGSORT,
&argvW[0][1], 8,
EXCLUDE, -1) == 2) {
if (XCOPY_ProcessExcludeList(&argvW[0][9])) {
LPWSTR lpMsgBuf;
int status;
status = FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM,
NULL, ERROR_INVALID_PARAMETER, 0,
(LPTSTR) &lpMsgBuf, 0, NULL);
if (!status) {
WINE_FIXME("FIXME: Cannot display message for error %d, status %d\n",
ERROR_INVALID_PARAMETER, GetLastError());
} else {
printf("%S\n", lpMsgBuf);
LocalFree ((HLOCAL)lpMsgBuf);
}
return RC_INITERROR;
} else flags |= OPT_EXCLUDELIST;
} else flags |= OPT_EMPTYDIR;
break;
case '-': if (toupper(argvW[0][2])=='Y')
flags &= ~OPT_NOPROMPT; break;
default:
@ -218,6 +256,14 @@ int main (int argc, char *argv[])
destinationstem, destinationspec,
flags);
/* Clear up exclude list allocated memory */
while (excludeList) {
EXCLUDELIST *pos = excludeList;
excludeList = excludeList -> next;
HeapFree(GetProcessHeap(), 0, pos->name);
HeapFree(GetProcessHeap(), 0, pos);
}
/* Finished - print trailer and exit */
if (flags & OPT_SIMULATE) {
printf("%d file(s) would be copied\n", filesCopied);
@ -523,6 +569,30 @@ static int XCOPY_DoCopy(WCHAR *srcstem, WCHAR *srcspec,
skipFile = TRUE;
}
/* See if exclude list provided. Note since filenames are case
insensitive, need to uppercase the filename before doing
strstr */
if (!skipFile && (flags & OPT_EXCLUDELIST)) {
EXCLUDELIST *pos = excludeList;
WCHAR copyFromUpper[MAX_PATH];
/* Uppercase source filename */
lstrcpyW(copyFromUpper, copyFrom);
CharUpperBuff(copyFromUpper, lstrlenW(copyFromUpper));
/* Loop through testing each exclude line */
while (pos) {
if (wcsstr(copyFromUpper, pos->name) != NULL) {
WINE_TRACE("Skipping file as matches exclude '%s'\n",
wine_dbgstr_w(pos->name));
skipFile = TRUE;
pos = NULL;
} else {
pos = pos->next;
}
}
}
/* Output a status message */
if (!skipFile) {
if (flags & OPT_QUIET) {
@ -680,3 +750,91 @@ static BOOL XCOPY_CreateDirectory(const WCHAR* path)
HeapFree(GetProcessHeap(),0,new_path);
return ret;
}
/* =========================================================================
* Process the /EXCLUDE: file list, building up a list of substrings to
* avoid copying
* Returns TRUE on any failure
* ========================================================================= */
static BOOL XCOPY_ProcessExcludeList(WCHAR* parms) {
WCHAR *filenameStart = parms;
WINE_TRACE("/EXCLUDE parms: '%s'\n", wine_dbgstr_w(parms));
excludeList = NULL;
while (*parms && *parms != ' ' && *parms != '/') {
/* If found '+' then process the file found so far */
if (*parms == '+') {
if (XCOPY_ProcessExcludeFile(filenameStart, parms)) {
return TRUE;
}
filenameStart = parms+1;
}
parms++;
}
if (filenameStart != parms) {
if (XCOPY_ProcessExcludeFile(filenameStart, parms)) {
return TRUE;
}
}
return FALSE;
}
/* =========================================================================
* Process a single file from the /EXCLUDE: file list, building up a list
* of substrings to avoid copying
* Returns TRUE on any failure
* ========================================================================= */
static BOOL XCOPY_ProcessExcludeFile(WCHAR* filename, WCHAR* endOfName) {
WCHAR endChar = *endOfName;
WCHAR buffer[MAXSTRING];
FILE *inFile = NULL;
const WCHAR readTextMode[] = {'r', 't', 0};
/* Null terminate the filename (temporarily updates the filename hence
parms not const) */
*endOfName = 0x00;
/* Open the file */
inFile = _wfopen(filename, readTextMode);
if (inFile == NULL) {
printf("Failed to open '%S'\n", filename);
*endOfName = endChar;
return TRUE;
}
/* Process line by line */
while (fgetws(buffer, sizeof(buffer), inFile) != NULL) {
EXCLUDELIST *thisEntry;
int length = lstrlenW(buffer);
/* Strip CRLF */
buffer[length-1] = 0x00;
thisEntry = HeapAlloc(GetProcessHeap(), 0, sizeof(EXCLUDELIST));
thisEntry->next = excludeList;
excludeList = thisEntry;
thisEntry->name = HeapAlloc(GetProcessHeap(), 0,
(length * sizeof(WCHAR))+1);
lstrcpyW(thisEntry->name, buffer);
CharUpperBuff(thisEntry->name, length);
WINE_TRACE("Read line : '%s'\n", wine_dbgstr_w(thisEntry->name));
}
/* See if EOF or error occurred */
if (!feof(inFile)) {
printf("Failed during reading of '%S'\n", filename);
*endOfName = endChar;
return TRUE;
}
/* Revert the input string to original form, and cleanup + return */
*endOfName = endChar;
fclose(inFile);
return FALSE;
}