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

server: Add a helper to call async_request_complete() while allocating new memory. 

Signed-off-by: Zebediah Figura <zfigura@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2021-09-01 17:28:39 -05:00 committed by Alexandre Julliard
parent 96593370cd
commit 559a8cea57
5 changed files with 20 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
server/async.c
View File

@ -361,6 +361,21 @@ void async_request_complete( struct async *async, unsigned int status, data_size
async_terminate( async, status ); async_terminate( async, status );
} }
/* complete a request-based async */
void async_request_complete_alloc( struct async *async, unsigned int status, data_size_t result,
data_size_t out_size, const void *out_data )
{
void *out_data_copy = NULL;
if (out_size && !(out_data_copy = memdup( out_data, out_size )))
{
async_terminate( async, STATUS_NO_MEMORY );
return;
}
async_request_complete( async, status, result, out_size, out_data_copy );
}
/* set the timeout of an async operation */ /* set the timeout of an async operation */
void async_set_timeout( struct async *async, timeout_t timeout, unsigned int status ) void async_set_timeout( struct async *async, timeout_t timeout, unsigned int status )
{ {

7
server/console.c
View File

@ -1500,12 +1500,7 @@ DECL_HANDLER(get_next_console_request)
{ {
data_size_t out_size = min( iosb->out_size, get_req_data_size() ); data_size_t out_size = min( iosb->out_size, get_req_data_size() );
data_size_t result = ioctl->code == IOCTL_CONDRV_WRITE_FILE ? iosb->in_size : out_size; data_size_t result = ioctl->code == IOCTL_CONDRV_WRITE_FILE ? iosb->in_size : out_size;
void *out_data; async_request_complete_alloc( async, status, result, out_size, get_req_data() );
if (!out_size || (out_data = memdup( get_req_data(), out_size )))
async_request_complete( async, status, result, out_size, out_data );
else
async_terminate( async, STATUS_NO_MEMORY );
} }
release_object( async ); release_object( async );

6
server/device.c
View File

@ -385,12 +385,8 @@ static void set_irp_result( struct irp_call *irp, unsigned int status,
irp->file = NULL; irp->file = NULL;
if (irp->async) if (irp->async)
{ {
void *out_data;
out_size = min( irp->iosb->out_size, out_size ); out_size = min( irp->iosb->out_size, out_size );
if (out_size && !(out_data = memdup( out_data, out_size ))) async_request_complete_alloc( irp->async, status, result, out_size, out_data );
out_size = 0;
async_request_complete( irp->async, status, result, out_size, out_data );
release_object( irp->async ); release_object( irp->async );
irp->async = NULL; irp->async = NULL;
} }

2
server/file.h
View File

@ -229,6 +229,8 @@ extern int async_waiting( struct async_queue *queue );
extern void async_terminate( struct async *async, unsigned int status ); extern void async_terminate( struct async *async, unsigned int status );
extern void async_request_complete( struct async *async, unsigned int status, data_size_t result, extern void async_request_complete( struct async *async, unsigned int status, data_size_t result,
data_size_t out_size, void *out_data ); data_size_t out_size, void *out_data );
extern void async_request_complete_alloc( struct async *async, unsigned int status, data_size_t result,
data_size_t out_size, const void *out_data );
extern void async_wake_up( struct async_queue *queue, unsigned int status ); extern void async_wake_up( struct async_queue *queue, unsigned int status );
extern struct completion *fd_get_completion( struct fd *fd, apc_param_t *p_key ); extern struct completion *fd_get_completion( struct fd *fd, apc_param_t *p_key );
extern void fd_copy_completion( struct fd *src, struct fd *dst ); extern void fd_copy_completion( struct fd *src, struct fd *dst );

6
server/sock.c
View File

@ -710,7 +710,6 @@ static void complete_async_accept( struct sock *sock, struct accept_req *req )
else else
{ {
obj_handle_t handle; obj_handle_t handle;
void *out_data;
if (!(acceptsock = accept_socket( sock ))) if (!(acceptsock = accept_socket( sock )))
{ {
@ -727,10 +726,7 @@ static void complete_async_accept( struct sock *sock, struct accept_req *req )
return; return;
} }
if (!(out_data = malloc( sizeof(handle) ))) return; async_request_complete_alloc( req->async, STATUS_SUCCESS, 0, sizeof(handle), &handle );
memcpy( out_data, &handle, sizeof(handle) );
async_request_complete( req->async, STATUS_SUCCESS, 0, sizeof(handle), out_data );
} }
} }