1133 lines
40 KiB
C
1133 lines
40 KiB
C
/*
|
|
* XCOPY - Wine-compatible xcopy program
|
|
*
|
|
* Copyright (C) 2007 J. Edmeades
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
/*
|
|
* FIXME:
|
|
* This should now support all options listed in the xcopy help from
|
|
* windows XP except:
|
|
* /Z - Copy from network drives in restartable mode
|
|
* /X - Copy file audit settings (sets /O)
|
|
* /O - Copy file ownership + ACL info
|
|
* /G - Copy encrypted files to unencrypted destination
|
|
* /V - Verifies files
|
|
*/
|
|
|
|
/*
|
|
* Notes:
|
|
* Apparently, valid return codes are:
|
|
* 0 - OK
|
|
* 1 - No files found to copy
|
|
* 2 - CTRL+C during copy
|
|
* 4 - Initialization error, or invalid source specification
|
|
* 5 - Disk write error
|
|
*/
|
|
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <windows.h>
|
|
#include <wine/debug.h>
|
|
#include <wine/unicode.h>
|
|
#include "xcopy.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(xcopy);
|
|
|
|
|
|
/* 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 FILETIME dateRange; /* Date range to copy after*/
|
|
static const WCHAR wchr_slash[] = {'\\', 0};
|
|
static const WCHAR wchr_star[] = {'*', 0};
|
|
static const WCHAR wchr_dot[] = {'.', 0};
|
|
static const WCHAR wchr_dotdot[] = {'.', '.', 0};
|
|
|
|
|
|
/* To minimize stack usage during recursion, some temporary variables
|
|
made global */
|
|
static WCHAR copyFrom[MAX_PATH];
|
|
static WCHAR copyTo[MAX_PATH];
|
|
|
|
|
|
/* =========================================================================
|
|
* Load a string from the resource file, handling any error
|
|
* Returns string retrieved from resource file
|
|
* ========================================================================= */
|
|
static WCHAR *XCOPY_LoadMessage(UINT id) {
|
|
static WCHAR msg[MAXSTRING];
|
|
const WCHAR failedMsg[] = {'F', 'a', 'i', 'l', 'e', 'd', '!', 0};
|
|
|
|
if (!LoadStringW(GetModuleHandleW(NULL), id, msg, sizeof(msg)/sizeof(WCHAR))) {
|
|
WINE_FIXME("LoadString failed with %d\n", GetLastError());
|
|
lstrcpyW(msg, failedMsg);
|
|
}
|
|
return msg;
|
|
}
|
|
|
|
/* =========================================================================
|
|
* Output a formatted unicode string. Ideally this will go to the console
|
|
* and hence required WriteConsoleW to output it, however if file i/o is
|
|
* redirected, it needs to be WriteFile'd using OEM (not ANSI) format
|
|
* ========================================================================= */
|
|
static int __cdecl XCOPY_wprintf(const WCHAR *format, ...) {
|
|
|
|
static WCHAR *output_bufW = NULL;
|
|
static char *output_bufA = NULL;
|
|
static BOOL toConsole = TRUE;
|
|
static BOOL traceOutput = FALSE;
|
|
#define MAX_WRITECONSOLE_SIZE 65535
|
|
|
|
__ms_va_list parms;
|
|
DWORD nOut;
|
|
int len;
|
|
DWORD res = 0;
|
|
|
|
/*
|
|
* Allocate buffer to use when writing to console
|
|
* Note: Not freed - memory will be allocated once and released when
|
|
* xcopy ends
|
|
*/
|
|
|
|
if (!output_bufW) output_bufW = HeapAlloc(GetProcessHeap(), 0,
|
|
MAX_WRITECONSOLE_SIZE*sizeof(WCHAR));
|
|
if (!output_bufW) {
|
|
WINE_FIXME("Out of memory - could not allocate 2 x 64K buffers\n");
|
|
return 0;
|
|
}
|
|
|
|
__ms_va_start(parms, format);
|
|
SetLastError(NO_ERROR);
|
|
len = FormatMessageW(FORMAT_MESSAGE_FROM_STRING, format, 0, 0, output_bufW,
|
|
MAX_WRITECONSOLE_SIZE/sizeof(*output_bufW), &parms);
|
|
__ms_va_end(parms);
|
|
if (len == 0 && GetLastError() != NO_ERROR) {
|
|
WINE_FIXME("Could not format string: le=%u, fmt=%s\n", GetLastError(), wine_dbgstr_w(format));
|
|
return 0;
|
|
}
|
|
|
|
/* Try to write as unicode whenever we think it's a console */
|
|
if (toConsole) {
|
|
res = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE),
|
|
output_bufW, len, &nOut, NULL);
|
|
}
|
|
|
|
/* If writing to console has failed (ever) we assume it's file
|
|
i/o so convert to OEM codepage and output */
|
|
if (!res) {
|
|
BOOL usedDefaultChar = FALSE;
|
|
DWORD convertedChars;
|
|
|
|
toConsole = FALSE;
|
|
|
|
/*
|
|
* Allocate buffer to use when writing to file. Not freed, as above
|
|
*/
|
|
if (!output_bufA) output_bufA = HeapAlloc(GetProcessHeap(), 0,
|
|
MAX_WRITECONSOLE_SIZE);
|
|
if (!output_bufA) {
|
|
WINE_FIXME("Out of memory - could not allocate 2 x 64K buffers\n");
|
|
return 0;
|
|
}
|
|
|
|
/* Convert to OEM, then output */
|
|
convertedChars = WideCharToMultiByte(GetConsoleOutputCP(), 0, output_bufW,
|
|
len, output_bufA, MAX_WRITECONSOLE_SIZE,
|
|
"?", &usedDefaultChar);
|
|
WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), output_bufA, convertedChars,
|
|
&nOut, FALSE);
|
|
}
|
|
|
|
/* Trace whether screen or console */
|
|
if (!traceOutput) {
|
|
WINE_TRACE("Writing to console? (%d)\n", toConsole);
|
|
traceOutput = TRUE;
|
|
}
|
|
return nOut;
|
|
}
|
|
|
|
/* =========================================================================
|
|
* Load a string for a system error and writes it to the screen
|
|
* Returns string retrieved from resource file
|
|
* ========================================================================= */
|
|
static void XCOPY_FailMessage(DWORD err) {
|
|
LPWSTR lpMsgBuf;
|
|
int status;
|
|
|
|
status = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
|
|
FORMAT_MESSAGE_FROM_SYSTEM,
|
|
NULL, err, 0,
|
|
(LPWSTR) &lpMsgBuf, 0, NULL);
|
|
if (!status) {
|
|
WINE_FIXME("FIXME: Cannot display message for error %d, status %d\n",
|
|
err, GetLastError());
|
|
} else {
|
|
const WCHAR infostr[] = {'%', '1', '\n', 0};
|
|
XCOPY_wprintf(infostr, lpMsgBuf);
|
|
LocalFree ((HLOCAL)lpMsgBuf);
|
|
}
|
|
}
|
|
|
|
|
|
/* =========================================================================
|
|
* Routine copied from cmd.exe md command -
|
|
* This works recursively. so creating dir1\dir2\dir3 will create dir1 and
|
|
* dir2 if they do not already exist.
|
|
* ========================================================================= */
|
|
static BOOL XCOPY_CreateDirectory(const WCHAR* path)
|
|
{
|
|
int len;
|
|
WCHAR *new_path;
|
|
BOOL ret = TRUE;
|
|
|
|
new_path = HeapAlloc(GetProcessHeap(),0, sizeof(WCHAR) * (lstrlenW(path)+1));
|
|
lstrcpyW(new_path,path);
|
|
|
|
while ((len = lstrlenW(new_path)) && new_path[len - 1] == '\\')
|
|
new_path[len - 1] = 0;
|
|
|
|
while (!CreateDirectoryW(new_path,NULL))
|
|
{
|
|
WCHAR *slash;
|
|
DWORD last_error = GetLastError();
|
|
if (last_error == ERROR_ALREADY_EXISTS)
|
|
break;
|
|
|
|
if (last_error != ERROR_PATH_NOT_FOUND)
|
|
{
|
|
ret = FALSE;
|
|
break;
|
|
}
|
|
|
|
if (!(slash = wcsrchr(new_path,'\\')) && ! (slash = wcsrchr(new_path,'/')))
|
|
{
|
|
ret = FALSE;
|
|
break;
|
|
}
|
|
|
|
len = slash - new_path;
|
|
new_path[len] = 0;
|
|
if (!XCOPY_CreateDirectory(new_path))
|
|
{
|
|
ret = FALSE;
|
|
break;
|
|
}
|
|
new_path[len] = '\\';
|
|
}
|
|
HeapFree(GetProcessHeap(),0,new_path);
|
|
return ret;
|
|
}
|
|
|
|
/* =========================================================================
|
|
* 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) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_OPENFAIL), filename);
|
|
*endOfName = endChar;
|
|
return TRUE;
|
|
}
|
|
|
|
/* Process line by line */
|
|
while (fgetws(buffer, sizeof(buffer)/sizeof(WCHAR), inFile) != NULL) {
|
|
EXCLUDELIST *thisEntry;
|
|
int length = lstrlenW(buffer);
|
|
|
|
/* Strip CRLF */
|
|
buffer[length-1] = 0x00;
|
|
|
|
/* If more than CRLF */
|
|
if (length > 1) {
|
|
thisEntry = HeapAlloc(GetProcessHeap(), 0, sizeof(EXCLUDELIST));
|
|
thisEntry->next = excludeList;
|
|
excludeList = thisEntry;
|
|
thisEntry->name = HeapAlloc(GetProcessHeap(), 0,
|
|
(length * sizeof(WCHAR))+1);
|
|
lstrcpyW(thisEntry->name, buffer);
|
|
CharUpperBuffW(thisEntry->name, length);
|
|
WINE_TRACE("Read line : '%s'\n", wine_dbgstr_w(thisEntry->name));
|
|
}
|
|
}
|
|
|
|
/* See if EOF or error occurred */
|
|
if (!feof(inFile)) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_READFAIL), filename);
|
|
*endOfName = endChar;
|
|
fclose(inFile);
|
|
return TRUE;
|
|
}
|
|
|
|
/* Revert the input string to original form, and cleanup + return */
|
|
*endOfName = endChar;
|
|
fclose(inFile);
|
|
return FALSE;
|
|
}
|
|
|
|
/* =========================================================================
|
|
* 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;
|
|
}
|
|
|
|
/* =========================================================================
|
|
XCOPY_DoCopy - Recursive function to copy files based on input parms
|
|
of a stem and a spec
|
|
|
|
This works by using FindFirstFile supplying the source stem and spec.
|
|
If results are found, any non-directory ones are processed
|
|
Then, if /S or /E is supplied, another search is made just for
|
|
directories, and this function is called again for that directory
|
|
|
|
========================================================================= */
|
|
static int XCOPY_DoCopy(WCHAR *srcstem, WCHAR *srcspec,
|
|
WCHAR *deststem, WCHAR *destspec,
|
|
DWORD flags)
|
|
{
|
|
WIN32_FIND_DATAW *finddata;
|
|
HANDLE h;
|
|
BOOL findres = TRUE;
|
|
WCHAR *inputpath, *outputpath;
|
|
BOOL copiedFile = FALSE;
|
|
DWORD destAttribs, srcAttribs;
|
|
BOOL skipFile;
|
|
int ret = 0;
|
|
|
|
/* Allocate some working memory on heap to minimize footprint */
|
|
finddata = HeapAlloc(GetProcessHeap(), 0, sizeof(WIN32_FIND_DATAW));
|
|
inputpath = HeapAlloc(GetProcessHeap(), 0, MAX_PATH * sizeof(WCHAR));
|
|
outputpath = HeapAlloc(GetProcessHeap(), 0, MAX_PATH * sizeof(WCHAR));
|
|
|
|
/* Build the search info into a single parm */
|
|
lstrcpyW(inputpath, srcstem);
|
|
lstrcatW(inputpath, srcspec);
|
|
|
|
/* Search 1 - Look for matching files */
|
|
h = FindFirstFileW(inputpath, finddata);
|
|
while (h != INVALID_HANDLE_VALUE && findres) {
|
|
|
|
skipFile = FALSE;
|
|
|
|
/* Ignore . and .. */
|
|
if (lstrcmpW(finddata->cFileName, wchr_dot)==0 ||
|
|
lstrcmpW(finddata->cFileName, wchr_dotdot)==0 ||
|
|
finddata->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
|
|
|
|
WINE_TRACE("Skipping directory, . or .. (%s)\n", wine_dbgstr_w(finddata->cFileName));
|
|
} else {
|
|
|
|
/* Get the filename information */
|
|
lstrcpyW(copyFrom, srcstem);
|
|
if (flags & OPT_SHORTNAME) {
|
|
lstrcatW(copyFrom, finddata->cAlternateFileName);
|
|
} else {
|
|
lstrcatW(copyFrom, finddata->cFileName);
|
|
}
|
|
|
|
lstrcpyW(copyTo, deststem);
|
|
if (*destspec == 0x00) {
|
|
if (flags & OPT_SHORTNAME) {
|
|
lstrcatW(copyTo, finddata->cAlternateFileName);
|
|
} else {
|
|
lstrcatW(copyTo, finddata->cFileName);
|
|
}
|
|
} else {
|
|
lstrcatW(copyTo, destspec);
|
|
}
|
|
|
|
/* Do the copy */
|
|
WINE_TRACE("ACTION: Copy '%s' -> '%s'\n", wine_dbgstr_w(copyFrom),
|
|
wine_dbgstr_w(copyTo));
|
|
if (!copiedFile && !(flags & OPT_SIMULATE)) XCOPY_CreateDirectory(deststem);
|
|
|
|
/* See if allowed to copy it */
|
|
srcAttribs = GetFileAttributesW(copyFrom);
|
|
WINE_TRACE("Source attribs: %d\n", srcAttribs);
|
|
|
|
if ((srcAttribs & FILE_ATTRIBUTE_HIDDEN) ||
|
|
(srcAttribs & FILE_ATTRIBUTE_SYSTEM)) {
|
|
|
|
if (!(flags & OPT_COPYHIDSYS)) {
|
|
skipFile = TRUE;
|
|
}
|
|
}
|
|
|
|
if (!(srcAttribs & FILE_ATTRIBUTE_ARCHIVE) &&
|
|
(flags & OPT_ARCHIVEONLY)) {
|
|
skipFile = TRUE;
|
|
}
|
|
|
|
/* See if file exists */
|
|
destAttribs = GetFileAttributesW(copyTo);
|
|
WINE_TRACE("Dest attribs: %d\n", srcAttribs);
|
|
|
|
/* Check date ranges if a destination file already exists */
|
|
if (!skipFile && (flags & OPT_DATERANGE) &&
|
|
(CompareFileTime(&finddata->ftLastWriteTime, &dateRange) < 0)) {
|
|
WINE_TRACE("Skipping file as modified date too old\n");
|
|
skipFile = TRUE;
|
|
}
|
|
|
|
/* If just /D supplied, only overwrite if src newer than dest */
|
|
if (!skipFile && (flags & OPT_DATENEWER) &&
|
|
(destAttribs != INVALID_FILE_ATTRIBUTES)) {
|
|
HANDLE h = CreateFileW(copyTo, GENERIC_READ, FILE_SHARE_READ,
|
|
NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL,
|
|
NULL);
|
|
if (h != INVALID_HANDLE_VALUE) {
|
|
FILETIME writeTime;
|
|
GetFileTime(h, NULL, NULL, &writeTime);
|
|
|
|
if (CompareFileTime(&finddata->ftLastWriteTime, &writeTime) <= 0) {
|
|
WINE_TRACE("Skipping file as dest newer or same date\n");
|
|
skipFile = TRUE;
|
|
}
|
|
CloseHandle(h);
|
|
}
|
|
}
|
|
|
|
/* 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);
|
|
CharUpperBuffW(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;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Prompt each file if necessary */
|
|
if (!skipFile && (flags & OPT_SRCPROMPT)) {
|
|
DWORD count;
|
|
char answer[10];
|
|
BOOL answered = FALSE;
|
|
WCHAR yesChar[2];
|
|
WCHAR noChar[2];
|
|
|
|
/* Read the Y and N characters from the resource file */
|
|
wcscpy(yesChar, XCOPY_LoadMessage(STRING_YES_CHAR));
|
|
wcscpy(noChar, XCOPY_LoadMessage(STRING_NO_CHAR));
|
|
|
|
while (!answered) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_SRCPROMPT), copyFrom);
|
|
ReadFile (GetStdHandle(STD_INPUT_HANDLE), answer, sizeof(answer),
|
|
&count, NULL);
|
|
|
|
answered = TRUE;
|
|
if (toupper(answer[0]) == noChar[0])
|
|
skipFile = TRUE;
|
|
else if (toupper(answer[0]) != yesChar[0])
|
|
answered = FALSE;
|
|
}
|
|
}
|
|
|
|
if (!skipFile &&
|
|
destAttribs != INVALID_FILE_ATTRIBUTES && !(flags & OPT_NOPROMPT)) {
|
|
DWORD count;
|
|
char answer[10];
|
|
BOOL answered = FALSE;
|
|
WCHAR yesChar[2];
|
|
WCHAR allChar[2];
|
|
WCHAR noChar[2];
|
|
|
|
/* Read the A,Y and N characters from the resource file */
|
|
wcscpy(yesChar, XCOPY_LoadMessage(STRING_YES_CHAR));
|
|
wcscpy(allChar, XCOPY_LoadMessage(STRING_ALL_CHAR));
|
|
wcscpy(noChar, XCOPY_LoadMessage(STRING_NO_CHAR));
|
|
|
|
while (!answered) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_OVERWRITE), copyTo);
|
|
ReadFile (GetStdHandle(STD_INPUT_HANDLE), answer, sizeof(answer),
|
|
&count, NULL);
|
|
|
|
answered = TRUE;
|
|
if (toupper(answer[0]) == allChar[0])
|
|
flags |= OPT_NOPROMPT;
|
|
else if (toupper(answer[0]) == noChar[0])
|
|
skipFile = TRUE;
|
|
else if (toupper(answer[0]) != yesChar[0])
|
|
answered = FALSE;
|
|
}
|
|
}
|
|
|
|
/* See if it has to exist! */
|
|
if (destAttribs == INVALID_FILE_ATTRIBUTES && (flags & OPT_MUSTEXIST)) {
|
|
skipFile = TRUE;
|
|
}
|
|
|
|
/* Output a status message */
|
|
if (!skipFile) {
|
|
if (flags & OPT_QUIET) {
|
|
/* Skip message */
|
|
} else if (flags & OPT_FULL) {
|
|
const WCHAR infostr[] = {'%', '1', ' ', '-', '>', ' ',
|
|
'%', '2', '\n', 0};
|
|
|
|
XCOPY_wprintf(infostr, copyFrom, copyTo);
|
|
} else {
|
|
const WCHAR infostr[] = {'%', '1', '\n', 0};
|
|
XCOPY_wprintf(infostr, copyFrom);
|
|
}
|
|
|
|
/* If allowing overwriting of read only files, remove any
|
|
write protection */
|
|
if ((destAttribs & FILE_ATTRIBUTE_READONLY) &&
|
|
(flags & OPT_REPLACEREAD)) {
|
|
SetFileAttributesW(copyTo, destAttribs & ~FILE_ATTRIBUTE_READONLY);
|
|
}
|
|
|
|
copiedFile = TRUE;
|
|
if (flags & OPT_SIMULATE || flags & OPT_NOCOPY) {
|
|
/* Skip copy */
|
|
} else if (CopyFileW(copyFrom, copyTo, FALSE) == 0) {
|
|
|
|
DWORD error = GetLastError();
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_COPYFAIL),
|
|
copyFrom, copyTo, error);
|
|
XCOPY_FailMessage(error);
|
|
|
|
if (flags & OPT_IGNOREERRORS) {
|
|
skipFile = TRUE;
|
|
} else {
|
|
ret = RC_WRITEERROR;
|
|
goto cleanup;
|
|
}
|
|
}
|
|
|
|
/* If /M supplied, remove the archive bit after successful copy */
|
|
if (!skipFile) {
|
|
if ((srcAttribs & FILE_ATTRIBUTE_ARCHIVE) &&
|
|
(flags & OPT_REMOVEARCH)) {
|
|
SetFileAttributesW(copyFrom, (srcAttribs & ~FILE_ATTRIBUTE_ARCHIVE));
|
|
}
|
|
filesCopied++;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Find next file */
|
|
findres = FindNextFileW(h, finddata);
|
|
}
|
|
FindClose(h);
|
|
|
|
/* Search 2 - do subdirs */
|
|
if (flags & OPT_RECURSIVE) {
|
|
lstrcpyW(inputpath, srcstem);
|
|
lstrcatW(inputpath, wchr_star);
|
|
findres = TRUE;
|
|
WINE_TRACE("Processing subdirs with spec: %s\n", wine_dbgstr_w(inputpath));
|
|
|
|
h = FindFirstFileW(inputpath, finddata);
|
|
while (h != INVALID_HANDLE_VALUE && findres) {
|
|
|
|
/* Only looking for dirs */
|
|
if ((finddata->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
|
|
(lstrcmpW(finddata->cFileName, wchr_dot) != 0) &&
|
|
(lstrcmpW(finddata->cFileName, wchr_dotdot) != 0)) {
|
|
|
|
WINE_TRACE("Handling subdir: %s\n", wine_dbgstr_w(finddata->cFileName));
|
|
|
|
/* Make up recursive information */
|
|
lstrcpyW(inputpath, srcstem);
|
|
lstrcatW(inputpath, finddata->cFileName);
|
|
lstrcatW(inputpath, wchr_slash);
|
|
|
|
lstrcpyW(outputpath, deststem);
|
|
if (*destspec == 0x00) {
|
|
lstrcatW(outputpath, finddata->cFileName);
|
|
|
|
/* If /E is supplied, create the directory now */
|
|
if ((flags & OPT_EMPTYDIR) &&
|
|
!(flags & OPT_SIMULATE))
|
|
XCOPY_CreateDirectory(outputpath);
|
|
|
|
lstrcatW(outputpath, wchr_slash);
|
|
}
|
|
|
|
XCOPY_DoCopy(inputpath, srcspec, outputpath, destspec, flags);
|
|
}
|
|
|
|
/* Find next one */
|
|
findres = FindNextFileW(h, finddata);
|
|
}
|
|
FindClose(h);
|
|
}
|
|
|
|
cleanup:
|
|
|
|
/* free up memory */
|
|
HeapFree(GetProcessHeap(), 0, finddata);
|
|
HeapFree(GetProcessHeap(), 0, inputpath);
|
|
HeapFree(GetProcessHeap(), 0, outputpath);
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
/* =========================================================================
|
|
XCOPY_ParseCommandLine - Parses the command line
|
|
========================================================================= */
|
|
static BOOL is_whitespace(WCHAR c)
|
|
{
|
|
return c == ' ' || c == '\t';
|
|
}
|
|
|
|
static WCHAR *skip_whitespace(WCHAR *p)
|
|
{
|
|
for (; *p && is_whitespace(*p); p++);
|
|
return p;
|
|
}
|
|
|
|
/* Windows XCOPY uses a simplified command line parsing algorithm
|
|
that lacks the escaped-quote logic of build_argv(), because
|
|
literal double quotes are illegal in any of its arguments.
|
|
Example: 'XCOPY "c:\DIR A" "c:DIR B\"' is OK. */
|
|
static int find_end_of_word(const WCHAR *word, WCHAR **end)
|
|
{
|
|
BOOL in_quotes = 0;
|
|
const WCHAR *ptr = word;
|
|
for (;;) {
|
|
for (; *ptr != '\0' && *ptr != '"' &&
|
|
(in_quotes || !is_whitespace(*ptr)); ptr++);
|
|
if (*ptr == '"') {
|
|
in_quotes = !in_quotes;
|
|
ptr++;
|
|
}
|
|
/* Odd number of double quotes is illegal for XCOPY */
|
|
if (in_quotes && *ptr == '\0')
|
|
return RC_INITERROR;
|
|
if (*ptr == '\0' || (!in_quotes && is_whitespace(*ptr)))
|
|
break;
|
|
}
|
|
*end = (WCHAR*)ptr;
|
|
return RC_OK;
|
|
}
|
|
|
|
/* Remove all double quotes from a word */
|
|
static void strip_quotes(WCHAR *word, WCHAR **end)
|
|
{
|
|
WCHAR *rp, *wp;
|
|
for (rp = word, wp = word; *rp != '\0'; rp++) {
|
|
if (*rp == '"')
|
|
continue;
|
|
if (wp < rp)
|
|
*wp = *rp;
|
|
wp++;
|
|
}
|
|
*wp = '\0';
|
|
*end = wp;
|
|
}
|
|
|
|
static int XCOPY_ParseCommandLine(WCHAR *suppliedsource,
|
|
WCHAR *supplieddestination, DWORD *pflags)
|
|
{
|
|
const WCHAR EXCLUDE[] = {'E', 'X', 'C', 'L', 'U', 'D', 'E', ':', 0};
|
|
DWORD flags = *pflags;
|
|
WCHAR *cmdline, *word, *end, *next;
|
|
int rc = RC_INITERROR;
|
|
|
|
cmdline = _wcsdup(GetCommandLineW());
|
|
if (cmdline == NULL)
|
|
return rc;
|
|
|
|
/* Skip first arg, which is the program name */
|
|
if ((rc = find_end_of_word(cmdline, &word)) != RC_OK)
|
|
goto out;
|
|
word = skip_whitespace(word);
|
|
|
|
while (*word)
|
|
{
|
|
WCHAR first;
|
|
if ((rc = find_end_of_word(word, &end)) != RC_OK)
|
|
goto out;
|
|
|
|
next = skip_whitespace(end);
|
|
first = word[0];
|
|
*end = '\0';
|
|
strip_quotes(word, &end);
|
|
WINE_TRACE("Processing Arg: '%s'\n", wine_dbgstr_w(word));
|
|
|
|
/* First non-switch parameter is source, second is destination */
|
|
if (first != '/') {
|
|
if (suppliedsource[0] == 0x00) {
|
|
lstrcpyW(suppliedsource, word);
|
|
} else if (supplieddestination[0] == 0x00) {
|
|
lstrcpyW(supplieddestination, word);
|
|
} else {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_INVPARMS));
|
|
goto out;
|
|
}
|
|
} else {
|
|
/* Process all the switch options
|
|
Note: Windows docs say /P prompts when dest is created
|
|
but tests show it is done for each src file
|
|
regardless of the destination */
|
|
switch (toupper(word[1])) {
|
|
case 'I': flags |= OPT_ASSUMEDIR; break;
|
|
case 'S': flags |= OPT_RECURSIVE; break;
|
|
case 'Q': flags |= OPT_QUIET; break;
|
|
case 'F': flags |= OPT_FULL; break;
|
|
case 'L': flags |= OPT_SIMULATE; break;
|
|
case 'W': flags |= OPT_PAUSE; break;
|
|
case 'T': flags |= OPT_NOCOPY | OPT_RECURSIVE; break;
|
|
case 'Y': flags |= OPT_NOPROMPT; break;
|
|
case 'N': flags |= OPT_SHORTNAME; break;
|
|
case 'U': flags |= OPT_MUSTEXIST; break;
|
|
case 'R': flags |= OPT_REPLACEREAD; break;
|
|
case 'H': flags |= OPT_COPYHIDSYS; break;
|
|
case 'C': flags |= OPT_IGNOREERRORS; break;
|
|
case 'P': flags |= OPT_SRCPROMPT; break;
|
|
case 'A': flags |= OPT_ARCHIVEONLY; break;
|
|
case 'M': flags |= OPT_ARCHIVEONLY |
|
|
OPT_REMOVEARCH; break;
|
|
|
|
/* E can be /E or /EXCLUDE */
|
|
case 'E': if (CompareStringW(LOCALE_USER_DEFAULT,
|
|
NORM_IGNORECASE | SORT_STRINGSORT,
|
|
&word[1], 8,
|
|
EXCLUDE, -1) == CSTR_EQUAL) {
|
|
if (XCOPY_ProcessExcludeList(&word[9])) {
|
|
XCOPY_FailMessage(ERROR_INVALID_PARAMETER);
|
|
goto out;
|
|
} else flags |= OPT_EXCLUDELIST;
|
|
} else flags |= OPT_EMPTYDIR | OPT_RECURSIVE;
|
|
break;
|
|
|
|
/* D can be /D or /D: */
|
|
case 'D': if (word[2]==':' && isdigit(word[3])) {
|
|
SYSTEMTIME st;
|
|
WCHAR *pos = &word[3];
|
|
BOOL isError = FALSE;
|
|
memset(&st, 0x00, sizeof(st));
|
|
|
|
/* Parse the arg : Month */
|
|
st.wMonth = _wtol(pos);
|
|
while (*pos && isdigit(*pos)) pos++;
|
|
if (*pos++ != '-') isError = TRUE;
|
|
|
|
/* Parse the arg : Day */
|
|
if (!isError) {
|
|
st.wDay = _wtol(pos);
|
|
while (*pos && isdigit(*pos)) pos++;
|
|
if (*pos++ != '-') isError = TRUE;
|
|
}
|
|
|
|
/* Parse the arg : Year */
|
|
if (!isError) {
|
|
st.wYear = _wtol(pos);
|
|
while (*pos && isdigit(*pos)) pos++;
|
|
if (st.wYear < 100) st.wYear+=2000;
|
|
}
|
|
|
|
if (!isError && SystemTimeToFileTime(&st, &dateRange)) {
|
|
SYSTEMTIME st;
|
|
WCHAR datestring[32], timestring[32];
|
|
|
|
flags |= OPT_DATERANGE;
|
|
|
|
/* Debug info: */
|
|
FileTimeToSystemTime (&dateRange, &st);
|
|
GetDateFormatW(0, DATE_SHORTDATE, &st, NULL, datestring,
|
|
sizeof(datestring)/sizeof(WCHAR));
|
|
GetTimeFormatW(0, TIME_NOSECONDS, &st,
|
|
NULL, timestring, sizeof(timestring)/sizeof(WCHAR));
|
|
|
|
WINE_TRACE("Date being used is: %s %s\n",
|
|
wine_dbgstr_w(datestring), wine_dbgstr_w(timestring));
|
|
} else {
|
|
XCOPY_FailMessage(ERROR_INVALID_PARAMETER);
|
|
goto out;
|
|
}
|
|
} else {
|
|
flags |= OPT_DATENEWER;
|
|
}
|
|
break;
|
|
|
|
case '-': if (toupper(word[2])=='Y')
|
|
flags &= ~OPT_NOPROMPT;
|
|
break;
|
|
case '?': XCOPY_wprintf(XCOPY_LoadMessage(STRING_HELP));
|
|
rc = RC_HELP;
|
|
goto out;
|
|
default:
|
|
WINE_TRACE("Unhandled parameter '%s'\n", wine_dbgstr_w(word));
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_INVPARM), word);
|
|
goto out;
|
|
}
|
|
}
|
|
word = next;
|
|
}
|
|
|
|
/* Default the destination if not supplied */
|
|
if (supplieddestination[0] == 0x00)
|
|
lstrcpyW(supplieddestination, wchr_dot);
|
|
|
|
*pflags = flags;
|
|
rc = RC_OK;
|
|
|
|
out:
|
|
free(cmdline);
|
|
return rc;
|
|
}
|
|
|
|
|
|
/* =========================================================================
|
|
XCOPY_ProcessSourceParm - Takes the supplied source parameter, and
|
|
converts it into a stem and a filespec
|
|
========================================================================= */
|
|
static int XCOPY_ProcessSourceParm(WCHAR *suppliedsource, WCHAR *stem,
|
|
WCHAR *spec, DWORD flags)
|
|
{
|
|
WCHAR actualsource[MAX_PATH];
|
|
WCHAR *starPos;
|
|
WCHAR *questPos;
|
|
DWORD attribs;
|
|
|
|
/*
|
|
* Validate the source, expanding to full path ensuring it exists
|
|
*/
|
|
if (GetFullPathNameW(suppliedsource, MAX_PATH, actualsource, NULL) == 0) {
|
|
WINE_FIXME("Unexpected failure expanding source path (%d)\n", GetLastError());
|
|
return RC_INITERROR;
|
|
}
|
|
|
|
/* If full names required, convert to using the full path */
|
|
if (flags & OPT_FULL) {
|
|
lstrcpyW(suppliedsource, actualsource);
|
|
}
|
|
|
|
/*
|
|
* Work out the stem of the source
|
|
*/
|
|
|
|
/* If a directory is supplied, use that as-is (either fully or
|
|
partially qualified)
|
|
If a filename is supplied + a directory or drive path, use that
|
|
as-is
|
|
Otherwise
|
|
If no directory or path specified, add eg. C:
|
|
stem is Drive/Directory is bit up to last \ (or first :)
|
|
spec is bit after that */
|
|
|
|
starPos = wcschr(suppliedsource, '*');
|
|
questPos = wcschr(suppliedsource, '?');
|
|
if (starPos || questPos) {
|
|
attribs = 0x00; /* Ensures skips invalid or directory check below */
|
|
} else {
|
|
attribs = GetFileAttributesW(actualsource);
|
|
}
|
|
|
|
if (attribs == INVALID_FILE_ATTRIBUTES) {
|
|
XCOPY_FailMessage(GetLastError());
|
|
return RC_INITERROR;
|
|
|
|
/* Directory:
|
|
stem should be exactly as supplied plus a '\', unless it was
|
|
eg. C: in which case no slash required */
|
|
} else if (attribs & FILE_ATTRIBUTE_DIRECTORY) {
|
|
WCHAR lastChar;
|
|
|
|
WINE_TRACE("Directory supplied\n");
|
|
lstrcpyW(stem, suppliedsource);
|
|
lastChar = stem[lstrlenW(stem)-1];
|
|
if (lastChar != '\\' && lastChar != ':') {
|
|
lstrcatW(stem, wchr_slash);
|
|
}
|
|
lstrcpyW(spec, wchr_star);
|
|
|
|
/* File or wildcard search:
|
|
stem should be:
|
|
Up to and including last slash if directory path supplied
|
|
If c:filename supplied, just the c:
|
|
Otherwise stem should be the current drive letter + ':' */
|
|
} else {
|
|
WCHAR *lastDir;
|
|
|
|
WINE_TRACE("Filename supplied\n");
|
|
lastDir = wcsrchr(suppliedsource, '\\');
|
|
|
|
if (lastDir) {
|
|
lstrcpyW(stem, suppliedsource);
|
|
stem[(lastDir-suppliedsource) + 1] = 0x00;
|
|
lstrcpyW(spec, (lastDir+1));
|
|
} else if (suppliedsource[1] == ':') {
|
|
lstrcpyW(stem, suppliedsource);
|
|
stem[2] = 0x00;
|
|
lstrcpyW(spec, suppliedsource+2);
|
|
} else {
|
|
WCHAR curdir[MAXSTRING];
|
|
GetCurrentDirectoryW(sizeof(curdir)/sizeof(WCHAR), curdir);
|
|
stem[0] = curdir[0];
|
|
stem[1] = curdir[1];
|
|
stem[2] = 0x00;
|
|
lstrcpyW(spec, suppliedsource);
|
|
}
|
|
}
|
|
|
|
return RC_OK;
|
|
}
|
|
|
|
/* =========================================================================
|
|
XCOPY_ProcessDestParm - Takes the supplied destination parameter, and
|
|
converts it into a stem
|
|
========================================================================= */
|
|
static int XCOPY_ProcessDestParm(WCHAR *supplieddestination, WCHAR *stem, WCHAR *spec,
|
|
WCHAR *srcspec, DWORD flags)
|
|
{
|
|
WCHAR actualdestination[MAX_PATH];
|
|
DWORD attribs;
|
|
BOOL isDir = FALSE;
|
|
|
|
/*
|
|
* Validate the source, expanding to full path ensuring it exists
|
|
*/
|
|
if (GetFullPathNameW(supplieddestination, MAX_PATH, actualdestination, NULL) == 0) {
|
|
WINE_FIXME("Unexpected failure expanding source path (%d)\n", GetLastError());
|
|
return RC_INITERROR;
|
|
}
|
|
|
|
/* Destination is either a directory or a file */
|
|
attribs = GetFileAttributesW(actualdestination);
|
|
|
|
if (attribs == INVALID_FILE_ATTRIBUTES) {
|
|
|
|
/* If /I supplied and wildcard copy, assume directory */
|
|
/* Also if destination ends with backslash */
|
|
if ((flags & OPT_ASSUMEDIR &&
|
|
(wcschr(srcspec, '?') || wcschr(srcspec, '*'))) ||
|
|
(supplieddestination[lstrlenW(supplieddestination)-1] == '\\')) {
|
|
|
|
isDir = TRUE;
|
|
|
|
} else {
|
|
DWORD count;
|
|
char answer[10] = "";
|
|
WCHAR fileChar[2];
|
|
WCHAR dirChar[2];
|
|
|
|
/* Read the F and D characters from the resource file */
|
|
wcscpy(fileChar, XCOPY_LoadMessage(STRING_FILE_CHAR));
|
|
wcscpy(dirChar, XCOPY_LoadMessage(STRING_DIR_CHAR));
|
|
|
|
while (answer[0] != fileChar[0] && answer[0] != dirChar[0]) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_QISDIR), supplieddestination);
|
|
|
|
ReadFile(GetStdHandle(STD_INPUT_HANDLE), answer, sizeof(answer), &count, NULL);
|
|
WINE_TRACE("User answer %c\n", answer[0]);
|
|
|
|
answer[0] = toupper(answer[0]);
|
|
}
|
|
|
|
if (answer[0] == dirChar[0]) {
|
|
isDir = TRUE;
|
|
} else {
|
|
isDir = FALSE;
|
|
}
|
|
}
|
|
} else {
|
|
isDir = (attribs & FILE_ATTRIBUTE_DIRECTORY);
|
|
}
|
|
|
|
if (isDir) {
|
|
lstrcpyW(stem, actualdestination);
|
|
*spec = 0x00;
|
|
|
|
/* Ensure ends with a '\' */
|
|
if (stem[lstrlenW(stem)-1] != '\\') {
|
|
lstrcatW(stem, wchr_slash);
|
|
}
|
|
|
|
} else {
|
|
WCHAR drive[MAX_PATH];
|
|
WCHAR dir[MAX_PATH];
|
|
WCHAR fname[MAX_PATH];
|
|
WCHAR ext[MAX_PATH];
|
|
_wsplitpath(actualdestination, drive, dir, fname, ext);
|
|
lstrcpyW(stem, drive);
|
|
lstrcatW(stem, dir);
|
|
lstrcpyW(spec, fname);
|
|
lstrcatW(spec, ext);
|
|
}
|
|
return RC_OK;
|
|
}
|
|
|
|
|
|
/* =========================================================================
|
|
main - Main entrypoint for the xcopy command
|
|
|
|
Processes the args, and drives the actual copying
|
|
========================================================================= */
|
|
int wmain (int argc, WCHAR *argvW[])
|
|
{
|
|
int rc = 0;
|
|
WCHAR suppliedsource[MAX_PATH] = {0}; /* As supplied on the cmd line */
|
|
WCHAR supplieddestination[MAX_PATH] = {0};
|
|
WCHAR sourcestem[MAX_PATH] = {0}; /* Stem of source */
|
|
WCHAR sourcespec[MAX_PATH] = {0}; /* Filespec of source */
|
|
WCHAR destinationstem[MAX_PATH] = {0}; /* Stem of destination */
|
|
WCHAR destinationspec[MAX_PATH] = {0}; /* Filespec of destination */
|
|
WCHAR copyCmd[MAXSTRING]; /* COPYCMD env var */
|
|
DWORD flags = 0; /* Option flags */
|
|
const WCHAR PROMPTSTR1[] = {'/', 'Y', 0};
|
|
const WCHAR PROMPTSTR2[] = {'/', 'y', 0};
|
|
const WCHAR COPYCMD[] = {'C', 'O', 'P', 'Y', 'C', 'M', 'D', 0};
|
|
|
|
/* Preinitialize flags based on COPYCMD */
|
|
if (GetEnvironmentVariableW(COPYCMD, copyCmd, MAXSTRING)) {
|
|
if (wcsstr(copyCmd, PROMPTSTR1) != NULL ||
|
|
wcsstr(copyCmd, PROMPTSTR2) != NULL) {
|
|
flags |= OPT_NOPROMPT;
|
|
}
|
|
}
|
|
|
|
/* FIXME: On UNIX, files starting with a '.' are treated as hidden under
|
|
wine, but on windows these can be normal files. At least one installer
|
|
uses files such as .packlist and (validly) expects them to be copied.
|
|
Under wine, if we do not copy hidden files by default then they get
|
|
lose */
|
|
flags |= OPT_COPYHIDSYS;
|
|
|
|
/*
|
|
* Parse the command line
|
|
*/
|
|
if ((rc = XCOPY_ParseCommandLine(suppliedsource, supplieddestination,
|
|
&flags)) != RC_OK) {
|
|
if (rc == RC_HELP)
|
|
return RC_OK;
|
|
else
|
|
return rc;
|
|
}
|
|
|
|
/* Trace out the supplied information */
|
|
WINE_TRACE("Supplied parameters:\n");
|
|
WINE_TRACE("Source : '%s'\n", wine_dbgstr_w(suppliedsource));
|
|
WINE_TRACE("Destination : '%s'\n", wine_dbgstr_w(supplieddestination));
|
|
|
|
/* Extract required information from source specification */
|
|
rc = XCOPY_ProcessSourceParm(suppliedsource, sourcestem, sourcespec, flags);
|
|
if (rc != RC_OK) return rc;
|
|
|
|
/* Extract required information from destination specification */
|
|
rc = XCOPY_ProcessDestParm(supplieddestination, destinationstem,
|
|
destinationspec, sourcespec, flags);
|
|
if (rc != RC_OK) return rc;
|
|
|
|
/* Trace out the resulting information */
|
|
WINE_TRACE("Resolved parameters:\n");
|
|
WINE_TRACE("Source Stem : '%s'\n", wine_dbgstr_w(sourcestem));
|
|
WINE_TRACE("Source Spec : '%s'\n", wine_dbgstr_w(sourcespec));
|
|
WINE_TRACE("Dest Stem : '%s'\n", wine_dbgstr_w(destinationstem));
|
|
WINE_TRACE("Dest Spec : '%s'\n", wine_dbgstr_w(destinationspec));
|
|
|
|
/* Pause if necessary */
|
|
if (flags & OPT_PAUSE) {
|
|
DWORD count;
|
|
char pausestr[10];
|
|
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_PAUSE));
|
|
ReadFile (GetStdHandle(STD_INPUT_HANDLE), pausestr, sizeof(pausestr),
|
|
&count, NULL);
|
|
}
|
|
|
|
/* Now do the hard work... */
|
|
rc = XCOPY_DoCopy(sourcestem, sourcespec,
|
|
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) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_SIMCOPY), filesCopied);
|
|
} else if (!(flags & OPT_NOCOPY)) {
|
|
XCOPY_wprintf(XCOPY_LoadMessage(STRING_COPY), filesCopied);
|
|
}
|
|
if (rc == RC_OK && filesCopied == 0) rc = RC_NOFILES;
|
|
return rc;
|
|
|
|
}
|