system/wine
system/wine
1
0
Fork 0
mirror of git://source.winehq.org/git/wine.git synced 2024-09-15 05:57:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

http.sys: Receive data from connected sockets.

Browse source 
Signed-off-by: Zebediah Figura <z.figura12@gmail.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2019-08-22 22:35:28 -05:00 committed by Alexandre Julliard
parent 9fd1b0c228
commit fb6956c7d1
1 changed files with 83 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

83
dlls/http.sys/http.c
View file

@ -49,6 +49,9 @@ struct connection
struct list entry; /* in "connections" below */
int socket;
char *buffer;
unsigned int len, size;
};
static struct list connections = LIST_INIT(connections);
@ -79,6 +82,15 @@ static void accept_connection(int socket)
closesocket(peer);
return;
}
if (!(conn->buffer = heap_alloc(8192)))
{
ERR("Failed to allocate buffer memory.\n");
heap_free(conn);
shutdown(peer, SD_BOTH);
closesocket(peer);
return;
}
conn->size = 8192;
WSAEventSelect(peer, request_event, FD_READ | FD_CLOSE);
ioctlsocket(peer, FIONBIO, &true);
conn->socket = peer;
@ -93,8 +105,74 @@ static void close_connection(struct connection *conn)
heap_free(conn);
}
/* Upon receiving a request, parse it to ensure that it is a valid HTTP request,
* and mark down some information that we will use later. Returns 1 if we parsed
* a complete request, 0 if incomplete, -1 if invalid. */
static int parse_request(struct connection *conn)
{
FIXME("Not implemented.\n");
return -1;
}
static void receive_data(struct connection *conn)
{
int len, ret;
/* We might be waiting for an IRP, but always call recv() anyway, since we
* might have been woken up by the socket closing. */
if ((len = recv(conn->socket, conn->buffer + conn->len, conn->size - conn->len, 0)) <= 0)
{
if (WSAGetLastError() == WSAEWOULDBLOCK)
return; /* nothing to receive */
else if (!len)
TRACE("Connection was shut down by peer.\n");
else
ERR("Got error %u; shutting down connection.\n", WSAGetLastError());
close_connection(conn);
return;
}
conn->len += len;
TRACE("Received %u bytes of data.\n", len);
if (!(ret = parse_request(conn)))
{
ULONG available;
ioctlsocket(conn->socket, FIONREAD, &available);
if (available)
{
TRACE("%u more bytes of data available, trying with larger buffer.\n", available);
if (!(conn->buffer = heap_realloc(conn->buffer, conn->len + available)))
{
ERR("Failed to allocate %u bytes of memory.\n", conn->len + available);
close_connection(conn);
return;
}
conn->size = conn->len + available;
if ((len = recv(conn->socket, conn->buffer + conn->len, conn->size - conn->len, 0)) < 0)
{
ERR("Got error %u; shutting down connection.\n", WSAGetLastError());
close_connection(conn);
return;
}
TRACE("Received %u bytes of data.\n", len);
conn->len += len;
ret = parse_request(conn);
}
}
if (!ret)
TRACE("Request is incomplete, waiting for more data.\n");
else if (ret < 0)
{
WARN("Failed to parse request; shutting down connection.\n");
close_connection(conn);
}
}
static DWORD WINAPI request_thread_proc(void *arg)
{
struct connection *conn, *cursor;
struct request_queue *queue;
TRACE("Starting request thread.\n");
@ -109,6 +187,11 @@ static DWORD WINAPI request_thread_proc(void *arg)
accept_connection(queue->socket);
}
LIST_FOR_EACH_ENTRY_SAFE(conn, cursor, &connections, struct connection, entry)
{
receive_data(conn);
}
LeaveCriticalSection(&http_cs);
}