ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

winhttp: Pass a socket pointer to send_bytes() and receive_bytes(). 

Signed-off-by: Hans Leidekker <hans@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Hans Leidekker 2021-09-22 12:44:12 +02:00 committed by Alexandre Julliard
parent 09895076c6
commit 54742a9001
1 changed files with 25 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
dlls/winhttp/request.c
View File

@ -3135,11 +3135,11 @@ HINTERNET WINAPI WinHttpWebSocketCompleteUpgrade( HINTERNET hrequest, DWORD_PTR
return hsocket; return hsocket;
} }
static DWORD send_bytes( struct netconn *netconn, char *bytes, int len ) static DWORD send_bytes( struct socket *socket, char *bytes, int len )
{ {
int count; int count;
DWORD err; DWORD err;
if ((err = netconn_send( netconn, bytes, len, &count ))) return err; if ((err = netconn_send( socket->request->netconn, bytes, len, &count ))) return err;
return (count == len) ? ERROR_SUCCESS : ERROR_INTERNAL_ERROR; return (count == len) ? ERROR_SUCCESS : ERROR_INTERNAL_ERROR;
} }
@ -3148,7 +3148,7 @@ static DWORD send_bytes( struct netconn *netconn, char *bytes, int len )
#define RESERVED_BIT (7 << 4) #define RESERVED_BIT (7 << 4)
#define CONTROL_BIT (1 << 3) #define CONTROL_BIT (1 << 3)
static DWORD send_frame( struct netconn *netconn, enum socket_opcode opcode, USHORT status, const char *buf, static DWORD send_frame( struct socket *socket, enum socket_opcode opcode, USHORT status, const char *buf,
DWORD buflen, BOOL final ) DWORD buflen, BOOL final )
{ {
DWORD i = 0, j, ret, offset = 2, len = buflen; DWORD i = 0, j, ret, offset = 2, len = buflen;
@ -3180,27 +3180,27 @@ static DWORD send_frame( struct netconn *netconn, enum socket_opcode opcode, USH
offset += 8; offset += 8;
} }
if ((ret = send_bytes( netconn, hdr, offset ))) return ret; if ((ret = send_bytes( socket, hdr, offset ))) return ret;
if (len) if (len)
{ {
mask = &hdr[offset]; mask = &hdr[offset];
RtlGenRandom( mask, 4 ); RtlGenRandom( mask, 4 );
if ((ret = send_bytes( netconn, mask, 4 ))) return ret; if ((ret = send_bytes( socket, mask, 4 ))) return ret;
} }
if (opcode == SOCKET_OPCODE_CLOSE) /* prepend status code */ if (opcode == SOCKET_OPCODE_CLOSE) /* prepend status code */
{ {
byte = (status >> 8) ^ mask[i++ % 4]; byte = (status >> 8) ^ mask[i++ % 4];
if ((ret = send_bytes( netconn, &byte, 1 ))) return ret; if ((ret = send_bytes( socket, &byte, 1 ))) return ret;
byte = (status & 0xff) ^ mask[i++ % 4]; byte = (status & 0xff) ^ mask[i++ % 4];
if ((ret = send_bytes( netconn, &byte, 1 ))) return ret; if ((ret = send_bytes( socket, &byte, 1 ))) return ret;
} }
for (j = 0; j < buflen; j++) for (j = 0; j < buflen; j++)
{ {
byte = buf[j] ^ mask[i++ % 4]; byte = buf[j] ^ mask[i++ % 4];
if ((ret = send_bytes( netconn, &byte, 1 ))) return ret; if ((ret = send_bytes( socket, &byte, 1 ))) return ret;
} }
return ERROR_SUCCESS; return ERROR_SUCCESS;
@ -3225,7 +3225,7 @@ static DWORD socket_send( struct socket *socket, WINHTTP_WEB_SOCKET_BUFFER_TYPE
enum socket_opcode opcode = map_buffer_type( type ); enum socket_opcode opcode = map_buffer_type( type );
DWORD ret; DWORD ret;
ret = send_frame( socket->request->netconn, opcode, 0, buf, len, TRUE ); ret = send_frame( socket, opcode, 0, buf, len, TRUE );
if (async) if (async)
{ {
if (!ret) if (!ret)
@ -3307,10 +3307,10 @@ DWORD WINAPI WinHttpWebSocketSend( HINTERNET hsocket, WINHTTP_WEB_SOCKET_BUFFER_
return ret; return ret;
} }
static DWORD receive_bytes( struct netconn *netconn, char *buf, DWORD len, DWORD *ret_len ) static DWORD receive_bytes( struct socket *socket, char *buf, DWORD len, DWORD *ret_len )
{ {
DWORD err; DWORD err;
if ((err = netconn_recv( netconn, buf, len, 0, (int *)ret_len ))) return err; if ((err = netconn_recv( socket->request->netconn, buf, len, 0, (int *)ret_len ))) return err;
if (*ret_len != len) return ERROR_WINHTTP_INVALID_SERVER_RESPONSE; if (*ret_len != len) return ERROR_WINHTTP_INVALID_SERVER_RESPONSE;
return ERROR_SUCCESS; return ERROR_SUCCESS;
} }
@ -3331,12 +3331,12 @@ static BOOL is_supported_opcode( enum socket_opcode opcode )
} }
} }
static DWORD receive_frame( struct netconn *netconn, DWORD *ret_len, enum socket_opcode *opcode ) static DWORD receive_frame( struct socket *socket, DWORD *ret_len, enum socket_opcode *opcode )
{ {
DWORD ret, len, count; DWORD ret, len, count;
char hdr[2]; char hdr[2];
if ((ret = receive_bytes( netconn, hdr, sizeof(hdr), &count ))) return ret; if ((ret = receive_bytes( socket, hdr, sizeof(hdr), &count ))) return ret;
if ((hdr[0] & RESERVED_BIT) || (hdr[1] & MASK_BIT) || !is_supported_opcode( hdr[0] & 0xf )) if ((hdr[0] & RESERVED_BIT) || (hdr[1] & MASK_BIT) || !is_supported_opcode( hdr[0] & 0xf ))
{ {
return ERROR_WINHTTP_INVALID_SERVER_RESPONSE; return ERROR_WINHTTP_INVALID_SERVER_RESPONSE;
@ -3348,13 +3348,13 @@ static DWORD receive_frame( struct netconn *netconn, DWORD *ret_len, enum socket
if (len == 126) if (len == 126)
{ {
USHORT len16; USHORT len16;
if ((ret = receive_bytes( netconn, (char *)&len16, sizeof(len16), &count ))) return ret; if ((ret = receive_bytes( socket, (char *)&len16, sizeof(len16), &count ))) return ret;
len = RtlUshortByteSwap( len16 ); len = RtlUshortByteSwap( len16 );
} }
else if (len == 127) else if (len == 127)
{ {
ULONGLONG len64; ULONGLONG len64;
if ((ret = receive_bytes( netconn, (char *)&len64, sizeof(len64), &count ))) return ret; if ((ret = receive_bytes( socket, (char *)&len64, sizeof(len64), &count ))) return ret;
if ((len64 = RtlUlonglongByteSwap( len64 )) > ~0u) return ERROR_NOT_SUPPORTED; if ((len64 = RtlUlonglongByteSwap( len64 )) > ~0u) return ERROR_NOT_SUPPORTED;
len = len64; len = len64;
} }
@ -3368,7 +3368,7 @@ static void CALLBACK task_socket_send_pong( TP_CALLBACK_INSTANCE *instance, void
struct socket_send *s = ctx; struct socket_send *s = ctx;
TRACE("running %p\n", work); TRACE("running %p\n", work);
send_frame( s->socket->request->netconn, SOCKET_OPCODE_PONG, 0, NULL, 0, TRUE ); send_frame( s->socket, SOCKET_OPCODE_PONG, 0, NULL, 0, TRUE );
release_object( &s->socket->hdr ); release_object( &s->socket->hdr );
free( s ); free( s );
@ -3392,18 +3392,17 @@ static DWORD socket_send_pong( struct socket *socket )
} }
return ret; return ret;
} }
return send_frame( socket->request->netconn, SOCKET_OPCODE_PONG, 0, NULL, 0, TRUE ); return send_frame( socket, SOCKET_OPCODE_PONG, 0, NULL, 0, TRUE );
} }
static DWORD socket_drain( struct socket *socket ) static DWORD socket_drain( struct socket *socket )
{ {
struct netconn *netconn = socket->request->netconn;
DWORD ret, count; DWORD ret, count;
while (socket->read_size) while (socket->read_size)
{ {
char buf[1024]; char buf[1024];
if ((ret = receive_bytes( netconn, buf, min(socket->read_size, sizeof(buf)), &count ))) return ret; if ((ret = receive_bytes( socket, buf, min(socket->read_size, sizeof(buf)), &count ))) return ret;
socket->read_size -= count; socket->read_size -= count;
} }
return ERROR_SUCCESS; return ERROR_SUCCESS;
@ -3451,14 +3450,13 @@ static WINHTTP_WEB_SOCKET_BUFFER_TYPE map_opcode( enum socket_opcode opcode, BOO
static DWORD socket_receive( struct socket *socket, void *buf, DWORD len, DWORD *ret_len, static DWORD socket_receive( struct socket *socket, void *buf, DWORD len, DWORD *ret_len,
WINHTTP_WEB_SOCKET_BUFFER_TYPE *ret_type, BOOL async ) WINHTTP_WEB_SOCKET_BUFFER_TYPE *ret_type, BOOL async )
{ {
struct netconn *netconn = socket->request->netconn;
DWORD count, ret = ERROR_SUCCESS; DWORD count, ret = ERROR_SUCCESS;
if (!socket->read_size) if (!socket->read_size)
{ {
for (;;) for (;;)
{ {
if (!(ret = receive_frame( netconn, &socket->read_size, &socket->opcode ))) if (!(ret = receive_frame( socket, &socket->read_size, &socket->opcode )))
{ {
if (!(socket->opcode & CONTROL_BIT) || (ret = handle_control_frame( socket ))) break; if (!(socket->opcode & CONTROL_BIT) || (ret = handle_control_frame( socket ))) break;
} }
@ -3466,7 +3464,7 @@ static DWORD socket_receive( struct socket *socket, void *buf, DWORD len, DWORD
if (ret) break; if (ret) break;
} }
} }
if (!ret) ret = receive_bytes( netconn, buf, min(len, socket->read_size), &count ); if (!ret) ret = receive_bytes( socket, buf, min(len, socket->read_size), &count );
if (!ret) if (!ret)
{ {
socket->read_size -= count; socket->read_size -= count;
@ -3554,11 +3552,10 @@ DWORD WINAPI WinHttpWebSocketReceive( HINTERNET hsocket, void *buf, DWORD len, D
static DWORD socket_shutdown( struct socket *socket, USHORT status, const void *reason, DWORD len, BOOL async ) static DWORD socket_shutdown( struct socket *socket, USHORT status, const void *reason, DWORD len, BOOL async )
{ {
struct netconn *netconn = socket->request->netconn;
DWORD ret; DWORD ret;
stop_queue( &socket->send_q ); stop_queue( &socket->send_q );
if (!(ret = send_frame( netconn, SOCKET_OPCODE_CLOSE, status, reason, len, TRUE ))) if (!(ret = send_frame( socket, SOCKET_OPCODE_CLOSE, status, reason, len, TRUE )))
{ {
socket->state = SOCKET_STATE_SHUTDOWN; socket->state = SOCKET_STATE_SHUTDOWN;
} }
@ -3634,7 +3631,6 @@ DWORD WINAPI WinHttpWebSocketShutdown( HINTERNET hsocket, USHORT status, void *r
static DWORD socket_close( struct socket *socket, USHORT status, const void *reason, DWORD len, BOOL async ) static DWORD socket_close( struct socket *socket, USHORT status, const void *reason, DWORD len, BOOL async )
{ {
struct netconn *netconn = socket->request->netconn;
DWORD ret, count; DWORD ret, count;
if ((ret = socket_drain( socket ))) goto done; if ((ret = socket_drain( socket ))) goto done;
@ -3642,11 +3638,11 @@ static DWORD socket_close( struct socket *socket, USHORT status, const void *rea
if (socket->state < SOCKET_STATE_SHUTDOWN) if (socket->state < SOCKET_STATE_SHUTDOWN)
{ {
stop_queue( &socket->send_q ); stop_queue( &socket->send_q );
if ((ret = send_frame( netconn, SOCKET_OPCODE_CLOSE, status, reason, len, TRUE ))) goto done; if ((ret = send_frame( socket, SOCKET_OPCODE_CLOSE, status, reason, len, TRUE ))) goto done;
socket->state = SOCKET_STATE_SHUTDOWN; socket->state = SOCKET_STATE_SHUTDOWN;
} }
if ((ret = receive_frame( netconn, &count, &socket->opcode ))) goto done; if ((ret = receive_frame( socket, &count, &socket->opcode ))) goto done;
if (socket->opcode != SOCKET_OPCODE_CLOSE || if (socket->opcode != SOCKET_OPCODE_CLOSE ||
(count && (count < sizeof(socket->status) || count > sizeof(socket->status) + sizeof(socket->reason)))) (count && (count < sizeof(socket->status) || count > sizeof(socket->status) + sizeof(socket->reason))))
{ {
@ -3657,9 +3653,9 @@ static DWORD socket_close( struct socket *socket, USHORT status, const void *rea
if (count) if (count)
{ {
DWORD reason_len = count - sizeof(socket->status); DWORD reason_len = count - sizeof(socket->status);
if ((ret = receive_bytes( netconn, (char *)&socket->status, sizeof(socket->status), &count ))) goto done; if ((ret = receive_bytes( socket, (char *)&socket->status, sizeof(socket->status), &count ))) goto done;
socket->status = RtlUshortByteSwap( socket->status ); socket->status = RtlUshortByteSwap( socket->status );
if ((ret = receive_bytes( netconn, socket->reason, reason_len, &socket->reason_len ))) goto done; if ((ret = receive_bytes( socket, socket->reason, reason_len, &socket->reason_len ))) goto done;
} }
socket->state = SOCKET_STATE_CLOSED; socket->state = SOCKET_STATE_CLOSED;