server/class.c

   1 /*
   2  * Server-side window class management
   3  *
   4  * Copyright (C) 2002 Mike McCormack
   5  * Copyright (C) 2003 Alexandre Julliard
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "config.h"
  23 #include "wine/port.h"
  24
  25 #include <assert.h>
  26 #include <stdlib.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29
  30 #include "ntstatus.h"
  31 #define WIN32_NO_STATUS
  32
  33 #include "wine/list.h"
  34
  35 #include "request.h"
  36 #include "object.h"
  37 #include "process.h"
  38 #include "user.h"
  39 #include "winuser.h"
  40 #include "winternl.h"
  41
  42 struct window_class
  43 {
  44     struct list     entry;           /* entry in process list */
  45     struct process *process;         /* process owning the class */
  46     int             count;           /* reference count */
  47     int             local;           /* local class? */
  48     atom_t          atom;            /* class atom */
  49     void           *instance;        /* module instance */
  50     unsigned int    style;           /* class style */
  51     int             win_extra;       /* number of window extra bytes */
  52     void           *client_ptr;      /* pointer to class in client address space */
  53     int             nb_extra_bytes;  /* number of extra bytes */
  54     char            extra_bytes[1];  /* extra bytes storage */
  55 };
  56
  57 static struct window_class *create_class( struct process *process, int extra_bytes, int local )
  58 {
  59     struct window_class *class;
  60
  61     if (!(class = mem_alloc( sizeof(*class) + extra_bytes - 1 ))) return NULL;
  62
  63     class->process = (struct process *)grab_object( process );
  64     class->count = 0;
  65     class->local = local;
  66     class->nb_extra_bytes = extra_bytes;
  67     memset( class->extra_bytes, 0, extra_bytes );
  68     /* other fields are initialized by caller */
  69
  70     /* local classes have priority so we put them first in the list */
  71     if (local) list_add_head( &process->classes, &class->entry );
  72     else list_add_tail( &process->classes, &class->entry );
  73     return class;
  74 }
  75
  76 static void destroy_class( struct window_class *class )
  77 {
  78     list_remove( &class->entry );
  79     release_object( class->process );
  80     free( class );
  81 }
  82
  83 void destroy_process_classes( struct process *process )
  84 {
  85     struct list *ptr;
  86
  87     while ((ptr = list_head( &process->classes )))
  88     {
  89         struct window_class *class = LIST_ENTRY( ptr, struct window_class, entry );
  90         destroy_class( class );
  91     }
  92 }
  93
  94 static struct window_class *find_class( struct process *process, atom_t atom, void *instance )
  95 {
  96     struct list *ptr;
  97
  98     LIST_FOR_EACH( ptr, &process->classes )
  99     {
 100         struct window_class *class = LIST_ENTRY( ptr, struct window_class, entry );
 101         if (class->atom != atom) continue;
 102         if (!instance || !class->local || class->instance == instance) return class;
 103     }
 104     return NULL;
 105 }
 106
 107 struct window_class *grab_class( struct process *process, atom_t atom,
 108                                  void *instance, int *extra_bytes )
 109 {
 110     struct window_class *class = find_class( process, atom, instance );
 111     if (class)
 112     {
 113         class->count++;
 114         *extra_bytes = class->win_extra;
 115     }
 116     else set_error( STATUS_INVALID_HANDLE );
 117     return class;
 118 }
 119
 120 void release_class( struct window_class *class )
 121 {
 122     assert( class->count > 0 );
 123     class->count--;
 124 }
 125
 126 int is_desktop_class( struct window_class *class )
 127 {
 128     return (class->atom == DESKTOP_ATOM && !class->local);
 129 }
 130
 131 atom_t get_class_atom( struct window_class *class )
 132 {
 133     return class->atom;
 134 }
 135
 136 void *get_class_client_ptr( struct window_class *class )
 137 {
 138     return class->client_ptr;
 139 }
 140
 141 /* create a window class */
 142 DECL_HANDLER(create_class)
 143 {
 144     struct window_class *class;
 145     atom_t atom;
 146
 147     if (get_req_data_size())
 148     {
 149         atom = add_global_atom( NULL, get_req_data(), get_req_data_size() / sizeof(WCHAR) );
 150         if (!atom) return;
 151     }
 152     else
 153     {
 154         atom = req->atom;
 155         if (!grab_global_atom( NULL, atom )) return;
 156     }
 157
 158     class = find_class( current->process, atom, req->instance );
 159     if (class && !class->local == !req->local)
 160     {
 161         set_win32_error( ERROR_CLASS_ALREADY_EXISTS );
 162         release_global_atom( NULL, atom );
 163         return;
 164     }
 165     if (req->extra < 0 || req->extra > 4096 || req->win_extra < 0 || req->win_extra > 4096)
 166     {
 167         /* don't allow stupid values here */
 168         set_error( STATUS_INVALID_PARAMETER );
 169         release_global_atom( NULL, atom );
 170         return;
 171     }
 172
 173     if (!(class = create_class( current->process, req->extra, req->local )))
 174     {
 175         release_global_atom( NULL, atom );
 176         return;
 177     }
 178     class->atom       = atom;
 179     class->instance   = req->instance;
 180     class->style      = req->style;
 181     class->win_extra  = req->win_extra;
 182     class->client_ptr = req->client_ptr;
 183     reply->atom = atom;
 184 }
 185
 186 /* destroy a window class */
 187 DECL_HANDLER(destroy_class)
 188 {
 189     struct window_class *class;
 190     atom_t atom = req->atom;
 191
 192     if (get_req_data_size())
 193         atom = find_global_atom( NULL, get_req_data(), get_req_data_size() / sizeof(WCHAR) );
 194
 195     if (!(class = find_class( current->process, atom, req->instance )))
 196         set_win32_error( ERROR_CLASS_DOES_NOT_EXIST );
 197     else if (class->count)
 198         set_win32_error( ERROR_CLASS_HAS_WINDOWS );
 199     else
 200     {
 201         reply->client_ptr = class->client_ptr;
 202         destroy_class( class );
 203     }
 204 }
 205
 206
 207 /* set some information in a class */
 208 DECL_HANDLER(set_class_info)
 209 {
 210     struct window_class *class = get_window_class( req->window );
 211
 212     if (!class) return;
 213
 214     if (req->flags && class->process != current->process)
 215     {
 216         set_error( STATUS_ACCESS_DENIED );
 217         return;
 218     }
 219
 220     if (req->extra_size > sizeof(req->extra_value) ||
 221         req->extra_offset < -1 ||
 222         req->extra_offset > class->nb_extra_bytes - (int)req->extra_size)
 223     {
 224         set_win32_error( ERROR_INVALID_INDEX );
 225         return;
 226     }
 227     if ((req->flags & SET_CLASS_WINEXTRA) && (req->win_extra < 0 || req->win_extra > 4096))
 228     {
 229         set_error( STATUS_INVALID_PARAMETER );
 230         return;
 231     }
 232     if (req->extra_offset != -1)
 233     {
 234         memcpy( &reply->old_extra_value, class->extra_bytes + req->extra_offset, req->extra_size );
 235     }
 236     else if (req->flags & SET_CLASS_EXTRA)
 237     {
 238         set_win32_error( ERROR_INVALID_INDEX );
 239         return;
 240     }
 241
 242     reply->old_atom      = class->atom;
 243     reply->old_style     = class->style;
 244     reply->old_extra     = class->nb_extra_bytes;
 245     reply->old_win_extra = class->win_extra;
 246     reply->old_instance  = class->instance;
 247
 248     if (req->flags & SET_CLASS_ATOM)
 249     {
 250         if (!grab_global_atom( NULL, req->atom )) return;
 251         release_global_atom( NULL, class->atom );
 252         class->atom = req->atom;
 253     }
 254     if (req->flags & SET_CLASS_STYLE) class->style = req->style;
 255     if (req->flags & SET_CLASS_WINEXTRA) class->win_extra = req->win_extra;
 256     if (req->flags & SET_CLASS_INSTANCE) class->instance = req->instance;
 257     if (req->flags & SET_CLASS_EXTRA) memcpy( class->extra_bytes + req->extra_offset,
 258                                               &req->extra_value, req->extra_size );
 259 }