Sweden-Number/server/completion.c

257 lines
7.9 KiB
C

/*
* Server-side IO completion ports implementation
*
* Copyright (C) 2007 Andrey Turkin
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*
*/
/* FIXMEs:
* - built-in wait queues used which means:
* + threads are awaken FIFO and not LIFO as native does
* + "max concurrent active threads" parameter not used
* + completion handle is waitable, while native isn't
*/
#include "config.h"
#include "wine/port.h"
#include <stdarg.h>
#include <stdio.h>
#include "ntstatus.h"
#define WIN32_NO_STATUS
#include "windef.h"
#include "winternl.h"
#include "wine/unicode.h"
#include "object.h"
#include "file.h"
#include "handle.h"
#include "request.h"
struct completion
{
struct object obj;
struct list queue;
unsigned int depth;
};
static void completion_dump( struct object*, int );
static struct object_type *completion_get_type( struct object *obj );
static int completion_signaled( struct object *obj, struct wait_queue_entry *entry );
static unsigned int completion_map_access( struct object *obj, unsigned int access );
static void completion_destroy( struct object * );
static const struct object_ops completion_ops =
{
sizeof(struct completion), /* size */
completion_dump, /* dump */
completion_get_type, /* get_type */
add_queue, /* add_queue */
remove_queue, /* remove_queue */
completion_signaled, /* signaled */
no_satisfied, /* satisfied */
no_signal, /* signal */
no_get_fd, /* get_fd */
completion_map_access, /* map_access */
default_get_sd, /* get_sd */
default_set_sd, /* set_sd */
no_lookup_name, /* lookup_name */
directory_link_name, /* link_name */
default_unlink_name, /* unlink_name */
no_open_file, /* open_file */
no_close_handle, /* close_handle */
completion_destroy /* destroy */
};
struct comp_msg
{
struct list queue_entry;
apc_param_t ckey;
apc_param_t cvalue;
apc_param_t information;
unsigned int status;
};
static void completion_destroy( struct object *obj)
{
struct completion *completion = (struct completion *) obj;
struct comp_msg *tmp, *next;
LIST_FOR_EACH_ENTRY_SAFE( tmp, next, &completion->queue, struct comp_msg, queue_entry )
{
free( tmp );
}
}
static void completion_dump( struct object *obj, int verbose )
{
struct completion *completion = (struct completion *) obj;
assert( obj->ops == &completion_ops );
fprintf( stderr, "Completion depth=%u", completion->depth );
}
static struct object_type *completion_get_type( struct object *obj )
{
static const WCHAR name[] = {'I','o','C','o','m','p','l','e','t','i','o','n'};
static const struct unicode_str str = { name, sizeof(name) };
return get_object_type( &str );
}
static int completion_signaled( struct object *obj, struct wait_queue_entry *entry )
{
struct completion *completion = (struct completion *)obj;
return !list_empty( &completion->queue );
}
static unsigned int completion_map_access( struct object *obj, unsigned int access )
{
if (access & GENERIC_READ) access |= STANDARD_RIGHTS_READ | SYNCHRONIZE | IO_COMPLETION_QUERY_STATE;
if (access & GENERIC_WRITE) access |= STANDARD_RIGHTS_WRITE;
if (access & GENERIC_EXECUTE) access |= STANDARD_RIGHTS_EXECUTE;
if (access & GENERIC_ALL) access |= STANDARD_RIGHTS_ALL | IO_COMPLETION_ALL_ACCESS;
return access & ~(GENERIC_READ | GENERIC_WRITE | GENERIC_EXECUTE | GENERIC_ALL);
}
static struct completion *create_completion( struct directory *root, const struct unicode_str *name,
unsigned int attr, unsigned int concurrent,
const struct security_descriptor *sd )
{
struct completion *completion;
if ((completion = create_named_object_dir( root, name, attr, &completion_ops )))
{
if (get_error() != STATUS_OBJECT_NAME_EXISTS)
{
list_init( &completion->queue );
completion->depth = 0;
if (sd) default_set_sd( &completion->obj, sd,
OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION |
DACL_SECURITY_INFORMATION | SACL_SECURITY_INFORMATION );
}
}
return completion;
}
struct completion *get_completion_obj( struct process *process, obj_handle_t handle, unsigned int access )
{
return (struct completion *) get_handle_obj( process, handle, access, &completion_ops );
}
void add_completion( struct completion *completion, apc_param_t ckey, apc_param_t cvalue,
unsigned int status, apc_param_t information )
{
struct comp_msg *msg = mem_alloc( sizeof( *msg ) );
if (!msg)
return;
msg->ckey = ckey;
msg->cvalue = cvalue;
msg->status = status;
msg->information = information;
list_add_tail( &completion->queue, &msg->queue_entry );
completion->depth++;
wake_up( &completion->obj, 1 );
}
/* create a completion */
DECL_HANDLER(create_completion)
{
struct completion *completion;
struct unicode_str name;
struct directory *root;
const struct security_descriptor *sd;
const struct object_attributes *objattr = get_req_object_attributes( &sd, &name, &root );
if (!objattr) return;
if ((completion = create_completion( root, &name, objattr->attributes, req->concurrent, sd )))
{
reply->handle = alloc_handle( current->process, completion, req->access, objattr->attributes );
release_object( completion );
}
if (root) release_object( root );
}
/* open a completion */
DECL_HANDLER(open_completion)
{
struct unicode_str name = get_req_unicode_str();
reply->handle = open_object( current->process, req->rootdir, req->access,
&completion_ops, &name, req->attributes );
}
/* add completion to completion port */
DECL_HANDLER(add_completion)
{
struct completion* completion = get_completion_obj( current->process, req->handle, IO_COMPLETION_MODIFY_STATE );
if (!completion) return;
add_completion( completion, req->ckey, req->cvalue, req->status, req->information );
release_object( completion );
}
/* get completion from completion port */
DECL_HANDLER(remove_completion)
{
struct completion* completion = get_completion_obj( current->process, req->handle, IO_COMPLETION_MODIFY_STATE );
struct list *entry;
struct comp_msg *msg;
if (!completion) return;
entry = list_head( &completion->queue );
if (!entry)
set_error( STATUS_PENDING );
else
{
list_remove( entry );
completion->depth--;
msg = LIST_ENTRY( entry, struct comp_msg, queue_entry );
reply->ckey = msg->ckey;
reply->cvalue = msg->cvalue;
reply->status = msg->status;
reply->information = msg->information;
free( msg );
}
release_object( completion );
}
/* get queue depth for completion port */
DECL_HANDLER(query_completion)
{
struct completion* completion = get_completion_obj( current->process, req->handle, IO_COMPLETION_QUERY_STATE );
if (!completion) return;
reply->depth = completion->depth;
release_object( completion );
}