tyler.durden/Win32-OpenSSH
1
0
Fork 0
mirror of https://github.com/PowerShell/Win32-OpenSSH.git synced 2025-07-23 22:15:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

1-15 C1

Browse Source
This commit is contained in:
Manoj Ampalam 2016-01-15 17:48:52 -08:00
parent ea5c31f5a1
commit b345f4581f
1 changed files with 134 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

145
contrib/win32/w32-posix-prototype/win32posix/win32posix/fileio.c
View File

@ -157,34 +157,142 @@ VOID CALLBACK ReadCompletionRoutine(
pio->read_details.pending = FALSE;
}
int fileio_ReadFileEx(struct w32_io* pio, BOOL *completed) {
#define READ_BUFFER_SIZE 100*1024
int fileio_ReadFileEx(struct w32_io* pio) {
if (pio->read_details.buf == NULL)
{
pio->read_details.buf = malloc(READ_BUFFER_SIZE);
if (!pio->read_details.buf)
{
errno = ENOMEM;
return -1;
}
pio->read_details.buf_size = READ_BUFFER_SIZE;
}
if (ReadFileEx(pio->handle, pio->read_details.buf, pio->read_details.buf_size, &pio->read_overlapped, &ReadCompletionRoutine))
{
pio->read_details.pending = TRUE;
}
else
{
errno = errno_from_Win32Error();
return -1;
}
return 0;
}
int fileio_read(struct w32_io* pio, void *dst, unsigned int max) {
int bytes_copied;
BOOL WINAPI ReadFileEx(
_In_ HANDLE hFile,
_Out_opt_ LPVOID lpBuffer,
_In_ DWORD nNumberOfBytesToRead,
_Inout_ LPOVERLAPPED lpOverlapped,
_In_ LPOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine
);
if (fileio_is_io_available(pio, TRUE) == FALSE)
{
if (-1 == fileio_ReadFileEx(pio))
return -1;
}
//Pick up APC if IO has completed
SleepEx(0, TRUE);
if (w32_io_is_blocking(pio)) {
while (fileio_is_io_available(pio, TRUE) == FALSE) {
if (-1 == wait_for_any_event(NULL, 0, INFINITE))
return -1;
}
}
else {
if (pio->read_details.pending) {
errno = EAGAIN;
return -1;
}
else {
if (pio->read_details.error) {
errno = EOTHER;
return -1;
}
}
}
}
if (pio->read_details.error) {
errno = EOTHER;
return -1;
}
bytes_copied = min(max, pio->read_details.remaining);
memcpy(dst, pio->read_details.buf, bytes_copied);
return bytes_copied;
}
VOID CALLBACK WriteCompletionRoutine(
_In_ DWORD dwErrorCode,
_In_ DWORD dwNumberOfBytesTransfered,
_Inout_ LPOVERLAPPED lpOverlapped
) {
struct w32_io* pio = (struct w32_io*)((char*)lpOverlapped - offsetof(struct w32_io, write_overlapped));
pio->write_details.error = dwErrorCode;
//assert that remaining == dwNumberOfBytesTransfered
pio->write_details.remaining -= dwNumberOfBytesTransfered;
pio->write_details.pending = FALSE;
}
#define WRITE_BUFFER_SIZE 100*1024
int fileio_write(struct w32_io* pio, const void *buf, unsigned int max) {
int bytes_copied;
if (pio->write_details.pending) {
errno = EAGAIN;
return -1;
}
if (pio->write_details.error) {
errno = EOTHER;
return -1;
}
if (pio->write_details.buf == NULL) {
pio->write_details.buf = malloc(WRITE_BUFFER_SIZE);
if (pio->write_details.buf == NULL) {
errno = ENOMEM;
return -1;
}
pio->write_details.buf_size = WRITE_BUFFER_SIZE;
}
bytes_copied = min(max, pio->write_details.buf_size);
memcpy(pio->write_details.buf, buf, bytes_copied);
if (WriteFileEx(pio->handle, pio->write_details.buf, bytes_copied, &pio->write_overlapped, &WriteCompletionRoutine))
{
pio->write_details.pending = TRUE;
SleepEx(0, TRUE);
if (w32_io_is_blocking(pio, FALSE)) {
while (pio->write_details.pending) {
if (wait_for_any_event(NULL, 0, INFINITE) == -1)
return -1;
}
}
else {
if (!pio->write_details.pending && pio->write_details.error){
errno = EOTHER;
return -1;
}
return bytes_copied;
}
}
else
{
errno = EOTHER;
return -1;
}
}
int fileio_fstat(struct w32_io* pio, struct stat *buf) {
@ -239,7 +347,20 @@ FILE* fileio_fdopen(struct w32_io* pio, const char *mode) {
return _fdopen(fd, mode);
}
int fileio_on_select(struct w32_io* pio, BOOL rd);
int fileio_on_select(struct w32_io* pio, BOOL rd) {
if (rd && pio->read_details.pending)
return 0;
if (!rd && pio->write_details.pending)
return 0;
if (rd) {
if (fileio_ReadFileEx(pio) == -1)
return -1;
}
else {
//nothing to do with write
}
}
int fileio_close(struct w32_io* pio) {