NTSTATUS
LwioCheckRemotePathIsDirectory(
IN PCSTR pszPath,
IN OUT PBOOLEAN pbIsDirectory
)
{
NTSTATUS ntStatus = STATUS_SUCCESS;
IO_FILE_HANDLE hFile = NULL;
IO_STATUS_BLOCK ioStatusBlock;
FILE_STANDARD_INFORMATION fileStdInfo;
ntStatus = LwioRemoteOpenFile(
pszPath,
FILE_READ_ATTRIBUTES,
FILE_SHARE_READ,
FILE_OPEN,
0,
&hFile);
BAIL_ON_NT_STATUS(ntStatus);
ntStatus = LwNtQueryInformationFile(
hFile,
NULL,
&ioStatusBlock,
(PVOID*)&fileStdInfo,
sizeof(fileStdInfo),
FileStandardInformation);
BAIL_ON_NT_STATUS(ntStatus);
*pbIsDirectory = fileStdInfo.Directory;
cleanup:
if (hFile)
{
LwNtCloseFile(hFile);
}
return ntStatus;
error:
*pbIsDirectory = FALSE;
goto cleanup;
}
NTSTATUS
LwioCopyDirToRemote(
IN PCSTR pszSourcePath,
IN PCSTR pszTargetPath
)
{
NTSTATUS status = STATUS_SUCCESS;
DIR* pDir = NULL;
struct dirent* pDirEntry = NULL;
struct stat statbuf;
IO_FILE_HANDLE handle = NULL;
PSTR pszLocalPath = NULL;
PSTR pszRemotePath = NULL;
BAIL_ON_NULL_POINTER(pszSourcePath);
BAIL_ON_NULL_POINTER(pszTargetPath);
if ((pDir = opendir(pszSourcePath)) == NULL)
{
status = LwErrnoToNtStatus(errno);
BAIL_ON_NT_STATUS(status);
}
status = LwioRemoteOpenFile(
pszTargetPath,
FILE_LIST_DIRECTORY,
FILE_SHARE_READ |FILE_SHARE_WRITE |FILE_SHARE_DELETE,
FILE_OPEN_IF,
FILE_DIRECTORY_FILE,
&handle);
BAIL_ON_NT_STATUS(status);
while ((pDirEntry = readdir(pDir)) != NULL)
{
RTL_FREE(&pszRemotePath);
RTL_FREE(&pszLocalPath);
if (!strcmp(pDirEntry->d_name, "..") ||
!strcmp(pDirEntry->d_name, "."))
continue;
status = LwRtlCStringAllocatePrintf(
&pszLocalPath,
"%s/%s",
pszSourcePath,
pDirEntry->d_name);
BAIL_ON_NT_STATUS(status);
memset(&statbuf, 0, sizeof(struct stat));
if (stat(pszLocalPath, &statbuf) < 0)
{
status = LwErrnoToNtStatus(errno);
BAIL_ON_NT_STATUS(status);
}
status = LwRtlCStringAllocatePrintf(
&pszRemotePath,
"%s/%s",
pszTargetPath,
pDirEntry->d_name);
BAIL_ON_NT_STATUS(status);
if ((statbuf.st_mode & S_IFMT) == S_IFDIR)
{
status = LwioCopyDirToRemote(
pszLocalPath,
pszRemotePath);
BAIL_ON_NT_STATUS(status);
}
else
{
status = LwioCopyFileToRemote(
pszLocalPath,
pszRemotePath);
BAIL_ON_NT_STATUS(status);
}
}
if(closedir(pDir) < 0)
{
pDir = NULL;
status = LwErrnoToNtStatus(status);
BAIL_ON_NT_STATUS(status);
}
pDir = NULL;
cleanup:
if (handle)
LwNtCloseFile(handle);
if (pDir)
closedir(pDir);
RTL_FREE(&pszLocalPath);
RTL_FREE(&pszRemotePath);
return status;
error:
goto cleanup;
}
NTSTATUS
LwioCopyFileToRemote(
IN PCSTR pszSourcePath,
IN PCSTR pszTargetPath
)
{
NTSTATUS status = STATUS_SUCCESS;
IO_FILE_HANDLE hRemoteFile = NULL;
int hLocalFile = -1;
DWORD dwBytesRead = 0;
CHAR szBuf[BUFF_SIZE];
BAIL_ON_NULL_POINTER(pszSourcePath);
BAIL_ON_NULL_POINTER(pszTargetPath);
status = LwioLocalOpenFile(
(PCSTR)pszSourcePath,
O_RDONLY,
0,
&hLocalFile);
BAIL_ON_NT_STATUS(status);
status = LwioRemoteOpenFile(
pszTargetPath,
FILE_WRITE_DATA,
FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,
FILE_OPEN_IF,
FILE_NON_DIRECTORY_FILE,
&hRemoteFile);
BAIL_ON_NT_STATUS(status);
do
{
DWORD dwWritten = 0;
memset (szBuf,0,BUFF_SIZE);
if ((dwBytesRead = read(hLocalFile, szBuf, sizeof(szBuf))) == -1)
{
status = LwErrnoToNtStatus(errno);
BAIL_ON_NT_STATUS(status);
}
if (dwBytesRead == 0)
{
break;
}
status = LwioRemoteWriteFile(
hRemoteFile,
szBuf,
dwBytesRead,
&dwWritten);
BAIL_ON_NT_STATUS(status);
} while (dwBytesRead != 0);
cleanup:
if (hRemoteFile)
{
LwNtCloseFile(hRemoteFile);
}
if (hLocalFile >= 0)
{
close(hLocalFile);
}
return (status);
error:
goto cleanup;
}
NTSTATUS
LwioCopyDirFromRemote(
IN PCSTR pszSourcePath,
IN PCSTR pszTargetPath
)
{
NTSTATUS status = STATUS_SUCCESS;
BOOL bRestart = TRUE;
IO_FILE_NAME filename = {0};
IO_FILE_HANDLE handle = NULL;
IO_STATUS_BLOCK ioStatus ;
PSTR pszEntryFilename = NULL;
BYTE buffer[MAX_BUFFER];
PFILE_BOTH_DIR_INFORMATION pInfo = NULL;
PSTR pszLocalPath = NULL;
PSTR pszRemotePath = NULL;
BAIL_ON_NULL_POINTER(pszSourcePath);
BAIL_ON_NULL_POINTER(pszTargetPath);
status = LwioRemoteOpenFile(
pszSourcePath,
FILE_LIST_DIRECTORY, /* Desired access mask */
FILE_SHARE_READ, /* Share access */
FILE_OPEN, /* Create disposition */
FILE_DIRECTORY_FILE, /* Create options */
&handle);
BAIL_ON_NT_STATUS(status);
status = LwioLocalCreateDir(
pszTargetPath,
S_IRWXU|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH);
BAIL_ON_NT_STATUS(status);
for (;;)
{
status = LwNtQueryDirectoryFile(
handle, /* File handle */
NULL, /* Async control block */
&ioStatus, /* IO status block */
buffer, /* Info structure */
sizeof(buffer), /* Info structure size */
FileBothDirectoryInformation, /* Info level */
FALSE, /* Do not return single entry */
NULL, /* File spec */
bRestart); /* Restart scan */
switch (status)
{
case STATUS_NO_MORE_MATCHES:
status = STATUS_SUCCESS;
goto cleanup;
default:
BAIL_ON_NT_STATUS(status);
}
bRestart = FALSE;
for (pInfo = (PFILE_BOTH_DIR_INFORMATION) buffer; pInfo;
pInfo = (pInfo->NextEntryOffset)?(PFILE_BOTH_DIR_INFORMATION) (((PBYTE) pInfo) + pInfo->NextEntryOffset):NULL)
{
RTL_FREE(&pszEntryFilename);
RTL_FREE(&pszRemotePath);
RTL_FREE(&pszLocalPath);
status = LwRtlCStringAllocateFromWC16String(
&pszEntryFilename,
pInfo->FileName
);
BAIL_ON_NT_STATUS(status);
if (!strcmp(pszEntryFilename, "..") ||
!strcmp(pszEntryFilename, "."))
continue;
status = LwRtlCStringAllocatePrintf(
&pszRemotePath,
"%s/%s",
pszSourcePath,
pszEntryFilename);
BAIL_ON_NT_STATUS(status);
status = LwRtlCStringAllocatePrintf(
&pszLocalPath,
"%s/%s",
pszTargetPath,
pszEntryFilename);
BAIL_ON_NT_STATUS(status);
if(pInfo->FileAttributes == FILE_ATTRIBUTE_DIRECTORY)
{
status = LwioCopyDirFromRemote(
pszRemotePath,
pszLocalPath);
BAIL_ON_NT_STATUS(status);
}
else
{
status = LwioCopyFileFromRemote(
pszRemotePath,
pszLocalPath);
BAIL_ON_NT_STATUS(status);
}
}
}
cleanup:
if (handle)
{
LwNtCloseFile(handle);
}
RTL_FREE(&pszLocalPath);
RTL_FREE(&pszRemotePath);
RTL_FREE(&pszEntryFilename);
RTL_FREE(&filename.FileName);
return status;
error:
goto cleanup;
}
NTSTATUS
LwioCopyFileFromRemote(
IN PCSTR pszSourcePath,
IN PCSTR pszTargetPath
)
{
NTSTATUS status = STATUS_SUCCESS;
IO_FILE_HANDLE hRemoteFile = NULL;
int hLocalFile = -1;
BAIL_ON_NULL_POINTER(pszSourcePath);
BAIL_ON_NULL_POINTER(pszTargetPath);
status = LwioRemoteOpenFile(
pszSourcePath,
FILE_READ_DATA, /* Desired access mask */
FILE_SHARE_READ, /* Share access */
FILE_OPEN, /* Create disposition */
FILE_NON_DIRECTORY_FILE, /* Create options */
&hRemoteFile);
BAIL_ON_NT_STATUS(status);
status = LwioLocalOpenFile(
(PCSTR)pszTargetPath,
O_WRONLY|O_TRUNC|O_CREAT,
0666,
&hLocalFile);
BAIL_ON_NT_STATUS(status);
do
{
BYTE szBuff[BUFF_SIZE];
DWORD dwRead = 0;
DWORD dwWrote = 0;
status = LwioRemoteReadFile(
hRemoteFile,
szBuff,
sizeof(szBuff),
&dwRead);
BAIL_ON_NT_STATUS(status);
if (!dwRead)
{
break;
}
if ((dwWrote = write(hLocalFile, szBuff, dwRead)) == -1)
{
status = LwErrnoToNtStatus(errno);
BAIL_ON_NT_STATUS(status);
}
} while(1);
cleanup:
if (hRemoteFile)
{
LwNtCloseFile(hRemoteFile);
}
if (hLocalFile >= 0)
{
close(hLocalFile);
}
return (status);
error:
goto cleanup;
}
NTSTATUS
LwioCopyFileFromRemoteToRemote(
IN PCSTR pszSourcePath,
IN PCSTR pszTargetPath
)
{
NTSTATUS status = STATUS_SUCCESS;
IO_FILE_HANDLE hRemSrcFile = NULL;
IO_FILE_HANDLE hRemDstFile = NULL;
BAIL_ON_NULL_POINTER(pszSourcePath);
BAIL_ON_NULL_POINTER(pszTargetPath);
status = LwioRemoteOpenFile(
pszSourcePath,
FILE_READ_DATA, /* Desired access mask */
FILE_SHARE_READ, /* Share access */
FILE_OPEN, /* Create disposition */
FILE_NON_DIRECTORY_FILE, /* Create options */
&hRemSrcFile);
BAIL_ON_NT_STATUS(status);
status = LwioRemoteOpenFile(
pszTargetPath,
FILE_WRITE_DATA,
FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,
FILE_OPEN_IF,
FILE_NON_DIRECTORY_FILE,
&hRemDstFile);
BAIL_ON_NT_STATUS(status);
do
{
BYTE szBuff[BUFF_SIZE];
DWORD dwRead = 0;
DWORD dwWrote = 0;
status = LwioRemoteReadFile(
hRemSrcFile,
szBuff,
sizeof(szBuff),
&dwRead);
BAIL_ON_NT_STATUS(status);
if (!dwRead)
{
break;
}
status = LwioRemoteWriteFile(
hRemDstFile,
szBuff,
dwRead,
&dwWrote);
BAIL_ON_NT_STATUS(status);
} while(1);
cleanup:
if (hRemSrcFile)
{
LwNtCloseFile(hRemSrcFile);
}
if (hRemDstFile)
{
LwNtCloseFile(hRemDstFile);
}
return (status);
error:
goto cleanup;
}
