server: Add a helper function to change a queue thread input.

This commit is contained in:
Alexandre Julliard 2010-03-22 17:48:22 +01:00
parent e9be1b4094
commit 5efe996c6a
1 changed files with 29 additions and 25 deletions

View File

@ -241,23 +241,19 @@ static struct thread_input *create_thread_input( struct thread *thread )
return input; return input;
} }
/* release the thread input data of a given thread */
static inline void release_thread_input( struct thread *thread )
{
struct thread_input *input = thread->queue->input;
if (!input) return;
release_object( input );
thread->queue->input = NULL;
}
/* create a message queue object */ /* create a message queue object */
static struct msg_queue *create_msg_queue( struct thread *thread, struct thread_input *input ) static struct msg_queue *create_msg_queue( struct thread *thread, struct thread_input *input )
{ {
struct thread_input *new_input = NULL;
struct msg_queue *queue; struct msg_queue *queue;
int i; int i;
if (!input && !(input = create_thread_input( thread ))) return NULL; if (!input)
{
if (!(new_input = create_thread_input( thread ))) return NULL;
input = new_input;
}
if ((queue = alloc_object( &msg_queue_ops ))) if ((queue = alloc_object( &msg_queue_ops )))
{ {
queue->fd = NULL; queue->fd = NULL;
@ -281,7 +277,7 @@ static struct msg_queue *create_msg_queue( struct thread *thread, struct thread_
thread->queue = queue; thread->queue = queue;
} }
release_object( input ); if (new_input) release_object( new_input );
return queue; return queue;
} }
@ -294,6 +290,20 @@ void free_msg_queue( struct thread *thread )
thread->queue = NULL; thread->queue = NULL;
} }
/* change the thread input data of a given thread */
static int assign_thread_input( struct thread *thread, struct thread_input *new_input )
{
if (!thread->queue)
{
thread->queue = create_msg_queue( thread, new_input );
return thread->queue != NULL;
}
if (thread->queue->input) release_object( thread->queue->input );
thread->queue->input = (struct thread_input *)grab_object( new_input );
return 1;
}
/* get the hook table for a given thread */ /* get the hook table for a given thread */
struct hook_table *get_queue_hooks( struct thread *thread ) struct hook_table *get_queue_hooks( struct thread *thread )
{ {
@ -917,6 +927,7 @@ int attach_thread_input( struct thread *thread_from, struct thread *thread_to )
{ {
struct desktop *desktop; struct desktop *desktop;
struct thread_input *input; struct thread_input *input;
int ret;
if (!thread_to->queue && !(thread_to->queue = create_msg_queue( thread_to, NULL ))) return 0; if (!thread_to->queue && !(thread_to->queue = create_msg_queue( thread_to, NULL ))) return 0;
if (!(desktop = get_thread_desktop( thread_from, 0 ))) return 0; if (!(desktop = get_thread_desktop( thread_from, 0 ))) return 0;
@ -930,17 +941,10 @@ int attach_thread_input( struct thread *thread_from, struct thread *thread_to )
} }
release_object( desktop ); release_object( desktop );
if (thread_from->queue) ret = assign_thread_input( thread_from, input );
{ if (ret) memset( input->keystate, 0, sizeof(input->keystate) );
release_thread_input( thread_from ); release_object( input );
thread_from->queue->input = input; return ret;
}
else
{
if (!(thread_from->queue = create_msg_queue( thread_from, input ))) return 0;
}
memset( input->keystate, 0, sizeof(input->keystate) );
return 1;
} }
/* detach two thread input data structures */ /* detach two thread input data structures */
@ -950,8 +954,8 @@ void detach_thread_input( struct thread *thread_from )
if ((input = create_thread_input( thread_from ))) if ((input = create_thread_input( thread_from )))
{ {
release_thread_input( thread_from ); assign_thread_input( thread_from, input );
thread_from->queue->input = input; release_object( input );
} }
} }