server: Add a separate function to set the timeout of an async I/O operation.

This commit is contained in:
Alexandre Julliard 2007-04-02 20:41:59 +02:00
parent 2a86f347fa
commit 0aae1ca8bf
8 changed files with 82 additions and 48 deletions

View File

@ -166,8 +166,7 @@ struct async_queue *create_async_queue( struct fd *fd )
}
/* create an async on a given queue of a fd */
struct async *create_async( struct thread *thread, const struct timeval *timeout,
struct async_queue *queue, const async_data_t *data )
struct async *create_async( struct thread *thread, struct async_queue *queue, const async_data_t *data )
{
struct event *event = NULL;
struct async *async;
@ -184,16 +183,23 @@ struct async *create_async( struct thread *thread, const struct timeval *timeout
async->thread = (struct thread *)grab_object( thread );
async->event = event;
async->data = *data;
async->timeout = NULL;
list_add_tail( &queue->queue, &async->queue_entry );
if (timeout) async->timeout = add_timeout_user( timeout, async_timeout, async );
else async->timeout = NULL;
grab_object( async );
if (event) reset_event( event );
return async;
}
/* set the timeout of an async operation */
void async_set_timeout( struct async *async, const struct timeval *timeout )
{
if (async->timeout) remove_timeout_user( async->timeout );
if (timeout) async->timeout = add_timeout_user( timeout, async_timeout, async );
else async->timeout = NULL;
}
/* store the result of the client-side async callback */
void async_set_result( struct object *obj, unsigned int status )
{

View File

@ -1058,6 +1058,7 @@ DECL_HANDLER(read_directory_changes)
{
struct event *event = NULL;
struct dir *dir;
struct async *async;
if (!req->filter)
{
@ -1079,7 +1080,7 @@ DECL_HANDLER(read_directory_changes)
dir->event = event;
/* requests don't timeout */
if (!fd_queue_async_timeout( dir->fd, &req->async, ASYNC_TYPE_WAIT, 0, NULL )) goto end;
if (!(async = fd_queue_async( dir->fd, &req->async, ASYNC_TYPE_WAIT, 0 ))) goto end;
/* assign it once */
if (!dir->filter)
@ -1103,6 +1104,7 @@ DECL_HANDLER(read_directory_changes)
if (!inotify_adjust_changes( dir ))
dnotify_adjust_changes( dir );
release_object( async );
set_error(STATUS_PENDING);
end:

View File

@ -1708,38 +1708,37 @@ void default_poll_event( struct fd *fd, int event )
wake_up( fd->user, 0 );
}
int fd_queue_async_timeout( struct fd *fd, const async_data_t *data, int type, int count,
const struct timeval *timeout )
struct async *fd_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
{
struct async_queue *queue;
struct async *async;
switch (type)
{
case ASYNC_TYPE_READ:
if (!fd->read_q && !(fd->read_q = create_async_queue( fd ))) return 0;
if (!fd->read_q && !(fd->read_q = create_async_queue( fd ))) return NULL;
queue = fd->read_q;
break;
case ASYNC_TYPE_WRITE:
if (!fd->write_q && !(fd->write_q = create_async_queue( fd ))) return 0;
if (!fd->write_q && !(fd->write_q = create_async_queue( fd ))) return NULL;
queue = fd->write_q;
break;
case ASYNC_TYPE_WAIT:
if (!fd->wait_q && !(fd->wait_q = create_async_queue( fd ))) return 0;
if (!fd->wait_q && !(fd->wait_q = create_async_queue( fd ))) return NULL;
queue = fd->wait_q;
break;
default:
assert(0);
}
if (!create_async( current, timeout, queue, data )) return 0;
set_error( STATUS_PENDING );
if (!fd->inode)
set_fd_events( fd, fd->fd_ops->get_poll_events( fd ) );
else /* regular files are always ready for read and write */
if (type != ASYNC_TYPE_WAIT) async_wake_up( queue, STATUS_ALERTED );
return 1;
if ((async = create_async( current, queue, data )))
{
if (!fd->inode)
set_fd_events( fd, fd->fd_ops->get_poll_events( fd ) );
else /* regular files are always ready for read and write */
if (type != ASYNC_TYPE_WAIT) async_wake_up( queue, STATUS_ALERTED );
}
return async;
}
void fd_async_wake_up( struct fd *fd, int type, unsigned int status )
@ -1763,6 +1762,7 @@ void fd_async_wake_up( struct fd *fd, int type, unsigned int status )
void default_fd_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
{
int flags;
struct async *async;
fd->fd_ops->get_file_info( fd, &flags );
if (!(flags & (FD_FLAG_OVERLAPPED|FD_FLAG_TIMEOUT)))
@ -1770,7 +1770,11 @@ void default_fd_queue_async( struct fd *fd, const async_data_t *data, int type,
set_error( STATUS_INVALID_HANDLE );
return;
}
fd_queue_async_timeout( fd, data, type, count, NULL );
if ((async = fd_queue_async( fd, data, type, count )))
{
release_object( async );
set_error( STATUS_PENDING );
}
}
void default_fd_cancel_async( struct fd *fd )

View File

@ -69,8 +69,7 @@ extern void default_fd_remove_queue( struct object *obj, struct wait_queue_entry
extern int default_fd_signaled( struct object *obj, struct thread *thread );
extern int default_fd_get_poll_events( struct fd *fd );
extern void default_poll_event( struct fd *fd, int event );
extern int fd_queue_async_timeout( struct fd *fd, const async_data_t *data, int type,
int count, const struct timeval *timeout );
extern struct async *fd_queue_async( struct fd *fd, const async_data_t *data, int type, int count );
extern void fd_async_wake_up( struct fd *fd, int type, unsigned int status );
extern void default_fd_queue_async( struct fd *fd, const async_data_t *data, int type, int count );
extern void default_fd_cancel_async( struct fd *fd );
@ -125,8 +124,9 @@ extern struct object *create_serial( struct fd *fd, unsigned int options );
/* async I/O functions */
extern struct async_queue *create_async_queue( struct fd *fd );
extern struct async *create_async( struct thread *thread, const struct timeval *timeout,
struct async_queue *queue, const async_data_t *data );
extern struct async *create_async( struct thread *thread, struct async_queue *queue,
const async_data_t *data );
extern void async_set_timeout( struct async *async, const struct timeval *timeout );
extern void async_set_result( struct object *obj, unsigned int status );
extern int async_waiting( struct async_queue *queue );
extern void async_wake_up( struct async_queue *queue, unsigned int status );

View File

@ -282,6 +282,7 @@ static struct object *mailslot_open_file( struct object *obj, unsigned int acces
static void mailslot_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
{
struct mailslot *mailslot = get_fd_user( fd );
struct async *async;
assert(mailslot->obj.ops == &mailslot_ops);
@ -292,13 +293,17 @@ static void mailslot_queue_async( struct fd *fd, const async_data_t *data, int t
return;
}
if (mailslot->read_timeout != -1)
if ((async = fd_queue_async( fd, data, type, count )))
{
struct timeval when = current_time;
add_timeout( &when, max(1,mailslot->read_timeout) );
fd_queue_async_timeout( fd, data, type, count, &when );
if (mailslot->read_timeout != -1)
{
struct timeval when = current_time;
add_timeout( &when, max(1,mailslot->read_timeout) );
async_set_timeout( async, &when );
}
release_object( async );
set_error( STATUS_PENDING );
}
else fd_queue_async_timeout( fd, data, type, count, NULL );
}
static void mailslot_device_dump( struct object *obj, int verbose )

View File

@ -793,6 +793,7 @@ DECL_HANDLER(create_named_pipe)
DECL_HANDLER(connect_named_pipe)
{
struct pipe_server *server;
struct async *async;
server = get_pipe_server_obj(current->process, req->handle, 0);
if (!server)
@ -804,9 +805,12 @@ DECL_HANDLER(connect_named_pipe)
case ps_wait_connect:
assert( !server->fd );
server->state = ps_wait_open;
create_async( current, NULL, server->wait_q, &req->async );
if (server->pipe->waiters) async_wake_up( server->pipe->waiters, STATUS_SUCCESS );
set_error( STATUS_PENDING );
if ((async = create_async( current, server->wait_q, &req->async )))
{
if (server->pipe->waiters) async_wake_up( server->pipe->waiters, STATUS_SUCCESS );
release_object( async );
set_error( STATUS_PENDING );
}
break;
case ps_connected_server:
assert( server->fd );
@ -848,23 +852,25 @@ DECL_HANDLER(wait_named_pipe)
server = find_available_server( pipe );
if (!server)
{
struct async *async;
if (!pipe->waiters && !(pipe->waiters = create_async_queue( NULL )))
{
release_object( pipe );
return;
}
if (req->timeout == NMPWAIT_WAIT_FOREVER)
if ((async = create_async( current, pipe->waiters, &req->async )))
{
if (create_async( current, NULL, pipe->waiters, &req->async ))
set_error( STATUS_PENDING );
}
else
{
struct timeval when = current_time;
if (req->timeout == NMPWAIT_USE_DEFAULT_WAIT) add_timeout( &when, pipe->timeout );
else add_timeout( &when, req->timeout );
if (create_async( current, &when, pipe->waiters, &req->async ))
set_error( STATUS_PENDING );
if (req->timeout != NMPWAIT_WAIT_FOREVER)
{
struct timeval when = current_time;
if (req->timeout == NMPWAIT_USE_DEFAULT_WAIT) add_timeout( &when, pipe->timeout );
else add_timeout( &when, req->timeout );
async_set_timeout( async, &when );
}
release_object( async );
set_error( STATUS_PENDING );
}
}
else release_object( server );

View File

@ -198,8 +198,8 @@ static enum server_fd_type serial_get_info( struct fd *fd, int *flags )
static void serial_queue_async( struct fd *fd, const async_data_t *data, int type, int count )
{
struct serial *serial = get_fd_user( fd );
struct timeval when = current_time;
int timeout = 0;
struct async *async;
assert(serial->obj.ops == &serial_ops);
@ -213,8 +213,17 @@ static void serial_queue_async( struct fd *fd, const async_data_t *data, int typ
break;
}
add_timeout( &when, timeout );
fd_queue_async_timeout( fd, data, type, count, timeout ? &when : NULL );
if ((async = fd_queue_async( fd, data, type, count )))
{
if (timeout)
{
struct timeval when = current_time;
add_timeout( &when, timeout );
async_set_timeout( async, &when );
}
release_object( async );
set_error( STATUS_PENDING );
}
}
static void serial_flush( struct fd *fd, struct event **event )

View File

@ -546,7 +546,9 @@ static void sock_queue_async( struct fd *fd, const async_data_t *data, int type,
}
else
{
if (!create_async( current, NULL, queue, data )) return;
struct async *async;
if (!(async = create_async( current, queue, data ))) return;
release_object( async );
set_error( STATUS_PENDING );
}