server: Don't block when writing to named pipes in PIPE_NOWAIT mode.

Signed-off-by: Jacek Caban <jacek@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Brendan Shanks 2019-10-16 14:50:34 +02:00 committed by Alexandre Julliard
parent add0556991
commit 91c4e9bf9a
2 changed files with 30 additions and 23 deletions

View File

@ -3941,7 +3941,7 @@ static void test_nowait(void)
/* write one byte larger than the buffer size, should fail */ /* write one byte larger than the buffer size, should fail */
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
todo_wine ok(WriteFile(pipewrite, readbuf, 513, &write, &ol), "WriteFile should succeed\n"); ok(WriteFile(pipewrite, readbuf, 513, &write, &ol), "WriteFile should succeed\n");
/* WriteFile only documents that 'write < sizeof(readbuf)' for this case, but Windows /* WriteFile only documents that 'write < sizeof(readbuf)' for this case, but Windows
* doesn't seem to do partial writes ('write == 0' always) * doesn't seem to do partial writes ('write == 0' always)
*/ */
@ -3952,32 +3952,32 @@ static void test_nowait(void)
/* overlapped read of 32768, non-blocking write of 512 */ /* overlapped read of 32768, non-blocking write of 512 */
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
todo_wine ok(ReadFile(piperead, readbuf, sizeof(readbuf), &read, &ol2) == FALSE, "ReadFile should fail\n"); ok(ReadFile(piperead, readbuf, sizeof(readbuf), &read, &ol2) == FALSE, "ReadFile should fail\n");
todo_wine ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError()); ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError());
ok(WriteFile(pipewrite, teststring, sizeof(teststring), &write, &ol), "WriteFile should succeed\n"); ok(WriteFile(pipewrite, teststring, sizeof(teststring), &write, &ol), "WriteFile should succeed\n");
ok(write == sizeof(teststring), "got %d\n", write); ok(write == sizeof(teststring), "got %d\n", write);
ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n"); ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n");
todo_wine ok(read == sizeof(teststring), "got %d\n", read); ok(read == sizeof(teststring), "got %d\n", read);
if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE) if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE)
CancelIo(piperead); CancelIo(piperead);
/* overlapped read of 32768, non-blocking write of 513 */ /* overlapped read of 32768, non-blocking write of 513 */
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
todo_wine ok(ReadFile(piperead, readbuf, sizeof(readbuf), &read, &ol2) == FALSE, "ReadFile should fail\n"); ok(ReadFile(piperead, readbuf, sizeof(readbuf), &read, &ol2) == FALSE, "ReadFile should fail\n");
todo_wine ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError()); ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError());
todo_wine ok(WriteFile(pipewrite, readbuf, 513, &write, &ol), "WriteFile should succeed\n"); ok(WriteFile(pipewrite, readbuf, 513, &write, &ol), "WriteFile should succeed\n");
todo_wine ok(write == 513, "got %d, write should be %d\n", write, 513); ok(write == 513, "got %d, write should be %d\n", write, 513);
ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n"); ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n");
todo_wine ok(read == 513, "got %d, read should be %d\n", read, 513); ok(read == 513, "got %d, read should be %d\n", read, 513);
if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE) if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE)
CancelIo(piperead); CancelIo(piperead);
/* overlapped read of 1 byte, non-blocking write of 513 bytes */ /* overlapped read of 1 byte, non-blocking write of 513 bytes */
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
todo_wine ok(ReadFile(piperead, readbuf, 1, &read, &ol2) == FALSE, "ReadFile should fail\n"); ok(ReadFile(piperead, readbuf, 1, &read, &ol2) == FALSE, "ReadFile should fail\n");
todo_wine ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError()); ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError());
todo_wine ok(WriteFile(pipewrite, readbuf, 513, &write, &ol), "WriteFile should succeed\n"); ok(WriteFile(pipewrite, readbuf, 513, &write, &ol), "WriteFile should succeed\n");
todo_wine ok(write == 513, "got %d, write should be %d\n", write, 513); ok(write == 513, "got %d, write should be %d\n", write, 513);
ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n"); ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n");
ok(read == 1, "got %d, read should be %d\n", read, 1); ok(read == 1, "got %d, read should be %d\n", read, 1);
if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE) if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE)
@ -3985,7 +3985,7 @@ static void test_nowait(void)
/* read the remaining 512 bytes */ /* read the remaining 512 bytes */
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
ok(ReadFile(piperead, readbuf, sizeof(readbuf), &read, &ol2), "ReadFile should succeed\n"); ok(ReadFile(piperead, readbuf, sizeof(readbuf), &read, &ol2), "ReadFile should succeed\n");
todo_wine ok(read == 512, "got %d, write should be %d\n", write, 512); ok(read == 512, "got %d, write should be %d\n", write, 512);
if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE) if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE)
CancelIo(piperead); CancelIo(piperead);
@ -3993,8 +3993,8 @@ static void test_nowait(void)
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
ok(ReadFile(piperead, readbuf, 1, &read, &ol2) == FALSE, "ReadFile should fail\n"); ok(ReadFile(piperead, readbuf, 1, &read, &ol2) == FALSE, "ReadFile should fail\n");
ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError()); ok(GetLastError() == ERROR_IO_PENDING, "got %d should be ERROR_IO_PENDING\n", GetLastError());
todo_wine ok(WriteFile(pipewrite, readbuf, 514, &write, &ol), "WriteFile should succeed\n"); ok(WriteFile(pipewrite, readbuf, 514, &write, &ol), "WriteFile should succeed\n");
todo_wine ok(write == 1, "got %d, write should be %d\n", write, 1); ok(write == 1, "got %d, write should be %d\n", write, 1);
ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n"); ok(GetOverlappedResult(piperead, &ol2, &read, FALSE), "GetOverlappedResult should succeed\n");
ok(read == 1, "got %d, read should be %d\n", read, 1); ok(read == 1, "got %d, read should be %d\n", read, 1);
if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE) if (GetOverlappedResult(piperead, &ol2, &read, FALSE) == FALSE)
@ -4002,8 +4002,8 @@ static void test_nowait(void)
/* write the exact buffer size, should succeed */ /* write the exact buffer size, should succeed */
SetLastError(0xdeadbeef); SetLastError(0xdeadbeef);
todo_wine ok(WriteFile(pipewrite, readbuf, 512, &write, &ol), "WriteFile should succeed\n"); ok(WriteFile(pipewrite, readbuf, 512, &write, &ol), "WriteFile should succeed\n");
todo_wine ok(write == 512, "WriteFile should write the whole buffer\n"); ok(write == 512, "WriteFile should write the whole buffer\n");
if (GetLastError() == ERROR_IO_PENDING) if (GetLastError() == ERROR_IO_PENDING)
CancelIo(piperead); CancelIo(piperead);

View File

@ -363,7 +363,7 @@ static struct pipe_message *queue_message( struct pipe_end *pipe_end, struct ios
return message; return message;
} }
static void wake_message( struct pipe_message *message ) static void wake_message( struct pipe_message *message, data_size_t result )
{ {
struct async *async = message->async; struct async *async = message->async;
@ -371,7 +371,7 @@ static void wake_message( struct pipe_message *message )
if (!async) return; if (!async) return;
message->iosb->status = STATUS_SUCCESS; message->iosb->status = STATUS_SUCCESS;
message->iosb->result = message->iosb->in_size; message->iosb->result = result;
async_terminate( async, message->iosb->result ? STATUS_ALERTED : STATUS_SUCCESS ); async_terminate( async, message->iosb->result ? STATUS_ALERTED : STATUS_SUCCESS );
release_object( async ); release_object( async );
} }
@ -749,7 +749,7 @@ static void message_queue_read( struct pipe_end *pipe_end, struct iosb *iosb )
{ {
iosb->out_data = message->iosb->in_data; iosb->out_data = message->iosb->in_data;
message->iosb->in_data = NULL; message->iosb->in_data = NULL;
wake_message( message ); wake_message( message, message->iosb->in_size );
free_message( message ); free_message( message );
} }
else else
@ -773,7 +773,7 @@ static void message_queue_read( struct pipe_end *pipe_end, struct iosb *iosb )
message->read_pos += writing; message->read_pos += writing;
if (message->read_pos == message->iosb->in_size) if (message->read_pos == message->iosb->in_size)
{ {
wake_message(message); wake_message(message, message->iosb->in_size);
free_message(message); free_message(message);
} }
} while (write_pos < iosb->out_size); } while (write_pos < iosb->out_size);
@ -835,7 +835,14 @@ static void reselect_write_queue( struct pipe_end *pipe_end )
{ {
avail += message->iosb->in_size - message->read_pos; avail += message->iosb->in_size - message->read_pos;
if (message->async && (avail <= reader->buffer_size || !message->iosb->in_size)) if (message->async && (avail <= reader->buffer_size || !message->iosb->in_size))
wake_message( message ); {
wake_message( message, message->iosb->in_size );
}
else if (message->async && (pipe_end->flags & NAMED_PIPE_NONBLOCKING_MODE))
{
wake_message( message, message->read_pos );
free_message( message );
}
} }
} }