server: Support using a name to destroy a window class too.
[wine] / 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 }