dlls/user32/class.c

   1 /*
   2  * Window classes functions
   3  *
   4  * Copyright 1993, 1996, 2003 Alexandre Julliard
   5  * Copyright 1998 Juergen Schmied (jsch)
   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 <stdarg.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29
  30 #include "winerror.h"
  31 #include "windef.h"
  32 #include "winbase.h"
  33 #include "wingdi.h"
  34 #include "wine/winuser16.h"
  35 #include "wine/unicode.h"
  36 #include "win.h"
  37 #include "user_private.h"
  38 #include "controls.h"
  39 #include "winproc.h"
  40 #include "wine/server.h"
  41 #include "wine/list.h"
  42 #include "wine/debug.h"
  43
  44 WINE_DEFAULT_DEBUG_CHANNEL(class);
  45
  46 typedef struct tagCLASS
  47 {
  48     struct list      entry;         /* Entry in class list */
  49     UINT             style;         /* Class style */
  50     BOOL             local;         /* Local class? */
  51     WNDPROC          winproc;       /* Window procedure */
  52     INT              cbClsExtra;    /* Class extra bytes */
  53     INT              cbWndExtra;    /* Window extra bytes */
  54     LPWSTR           menuName;      /* Default menu name (Unicode followed by ASCII) */
  55     HINSTANCE        hInstance;     /* Module that created the task */
  56     HICON            hIcon;         /* Default icon */
  57     HICON            hIconSm;       /* Default small icon */
  58     HCURSOR          hCursor;       /* Default cursor */
  59     HBRUSH           hbrBackground; /* Default background */
  60     ATOM             atomName;      /* Name of the class */
  61 } CLASS;
  62
  63 static struct list class_list = LIST_INIT( class_list );
  64
  65 #define CLASS_OTHER_PROCESS ((CLASS *)1)
  66 #define MAX_ATOM_LEN 255 /* from dlls/kernel32/atom.c */
  67
  68 /***********************************************************************
  69  *           get_class_ptr
  70  */
  71 static CLASS *get_class_ptr( HWND hwnd, BOOL write_access )
  72 {
  73     WND *ptr = WIN_GetPtr( hwnd );
  74
  75     if (ptr)
  76     {
  77         if (ptr != WND_OTHER_PROCESS && ptr != WND_DESKTOP) return ptr->class;
  78         if (!write_access) return CLASS_OTHER_PROCESS;
  79
  80         /* modifying classes in other processes is not allowed */
  81         if (ptr == WND_DESKTOP || IsWindow( hwnd ))
  82         {
  83             SetLastError( ERROR_ACCESS_DENIED );
  84             return NULL;
  85         }
  86     }
  87     SetLastError( ERROR_INVALID_WINDOW_HANDLE );
  88     return NULL;
  89 }
  90
  91
  92 /***********************************************************************
  93  *           release_class_ptr
  94  */
  95 static inline void release_class_ptr( CLASS *ptr )
  96 {
  97     USER_Unlock();
  98 }
  99
 100
 101 /***********************************************************************
 102  *           set_server_info
 103  *
 104  * Set class info with the wine server.
 105  */
 106 static BOOL set_server_info( HWND hwnd, INT offset, LONG_PTR newval, UINT size )
 107 {
 108     BOOL ret;
 109
 110     SERVER_START_REQ( set_class_info )
 111     {
 112         req->window = hwnd;
 113         req->extra_offset = -1;
 114         switch(offset)
 115         {
 116         case GCW_ATOM:
 117             req->flags = SET_CLASS_ATOM;
 118             req->atom = newval;
 119         case GCL_STYLE:
 120             req->flags = SET_CLASS_STYLE;
 121             req->style = newval;
 122             break;
 123         case GCL_CBWNDEXTRA:
 124             req->flags = SET_CLASS_WINEXTRA;
 125             req->win_extra = newval;
 126             break;
 127         case GCLP_HMODULE:
 128             req->flags = SET_CLASS_INSTANCE;
 129             req->instance = (void *)newval;
 130             break;
 131         default:
 132             assert( offset >= 0 );
 133             req->flags = SET_CLASS_EXTRA;
 134             req->extra_offset = offset;
 135             req->extra_size = size;
 136             if ( size == sizeof(LONG) )
 137             {
 138                 LONG newlong = newval;
 139                 memcpy( &req->extra_value, &newlong, sizeof(LONG) );
 140             }
 141             else
 142                 memcpy( &req->extra_value, &newval, sizeof(LONG_PTR) );
 143             break;
 144         }
 145         ret = !wine_server_call_err( req );
 146     }
 147     SERVER_END_REQ;
 148     return ret;
 149 }
 150
 151
 152 /***********************************************************************
 153  *           CLASS_GetMenuNameA
 154  *
 155  * Get the menu name as a ASCII string.
 156  */
 157 static inline LPSTR CLASS_GetMenuNameA( CLASS *classPtr )
 158 {
 159     if (!HIWORD(classPtr->menuName)) return (LPSTR)classPtr->menuName;
 160     return (LPSTR)(classPtr->menuName + strlenW(classPtr->menuName) + 1);
 161 }
 162
 163
 164 /***********************************************************************
 165  *           CLASS_GetMenuNameW
 166  *
 167  * Get the menu name as a Unicode string.
 168  */
 169 static inline LPWSTR CLASS_GetMenuNameW( CLASS *classPtr )
 170 {
 171     return classPtr->menuName;
 172 }
 173
 174
 175 /***********************************************************************
 176  *           CLASS_SetMenuNameA
 177  *
 178  * Set the menu name in a class structure by copying the string.
 179  */
 180 static void CLASS_SetMenuNameA( CLASS *classPtr, LPCSTR name )
 181 {
 182     if (HIWORD(classPtr->menuName)) HeapFree( GetProcessHeap(), 0, classPtr->menuName );
 183     if (HIWORD(name))
 184     {
 185         DWORD lenA = strlen(name) + 1;
 186         DWORD lenW = MultiByteToWideChar( CP_ACP, 0, name, lenA, NULL, 0 );
 187         classPtr->menuName = HeapAlloc( GetProcessHeap(), 0, lenA + lenW*sizeof(WCHAR) );
 188         MultiByteToWideChar( CP_ACP, 0, name, lenA, classPtr->menuName, lenW );
 189         memcpy( classPtr->menuName + lenW, name, lenA );
 190     }
 191     else classPtr->menuName = (LPWSTR)name;
 192 }
 193
 194
 195 /***********************************************************************
 196  *           CLASS_SetMenuNameW
 197  *
 198  * Set the menu name in a class structure by copying the string.
 199  */
 200 static void CLASS_SetMenuNameW( CLASS *classPtr, LPCWSTR name )
 201 {
 202     if (HIWORD(classPtr->menuName)) HeapFree( GetProcessHeap(), 0, classPtr->menuName );
 203     if (HIWORD(name))
 204     {
 205         DWORD lenW = strlenW(name) + 1;
 206         DWORD lenA = WideCharToMultiByte( CP_ACP, 0, name, lenW, NULL, 0, NULL, NULL );
 207         classPtr->menuName = HeapAlloc( GetProcessHeap(), 0, lenA + lenW*sizeof(WCHAR) );
 208         memcpy( classPtr->menuName, name, lenW*sizeof(WCHAR) );
 209         WideCharToMultiByte( CP_ACP, 0, name, lenW,
 210                              (char *)(classPtr->menuName + lenW), lenA, NULL, NULL );
 211     }
 212     else classPtr->menuName = (LPWSTR)name;
 213 }
 214
 215
 216 /***********************************************************************
 217  *           CLASS_FreeClass
 218  *
 219  * Free a class structure.
 220  */
 221 static void CLASS_FreeClass( CLASS *classPtr )
 222 {
 223     TRACE("%p\n", classPtr);
 224
 225     USER_Lock();
 226
 227     list_remove( &classPtr->entry );
 228     if (classPtr->hbrBackground > (HBRUSH)(COLOR_GRADIENTINACTIVECAPTION + 1))
 229         DeleteObject( classPtr->hbrBackground );
 230     HeapFree( GetProcessHeap(), 0, classPtr->menuName );
 231     HeapFree( GetProcessHeap(), 0, classPtr );
 232     USER_Unlock();
 233 }
 234
 235
 236 /***********************************************************************
 237  *           CLASS_FreeModuleClasses
 238  */
 239 void CLASS_FreeModuleClasses( HMODULE16 hModule )
 240 {
 241     struct list *ptr, *next;
 242
 243     TRACE("0x%08x\n", hModule);
 244
 245     USER_Lock();
 246     for (ptr = list_head( &class_list ); ptr; ptr = next)
 247     {
 248         CLASS *class = LIST_ENTRY( ptr, CLASS, entry );
 249         next = list_next( &class_list, ptr );
 250         if (class->hInstance == HINSTANCE_32(hModule))
 251         {
 252             BOOL ret;
 253
 254             SERVER_START_REQ( destroy_class )
 255             {
 256                 req->atom = class->atomName;
 257                 req->instance = class->hInstance;
 258                 ret = !wine_server_call_err( req );
 259             }
 260             SERVER_END_REQ;
 261             if (ret) CLASS_FreeClass( class );
 262         }
 263     }
 264     USER_Unlock();
 265 }
 266
 267
 268 /***********************************************************************
 269  *           CLASS_FindClassByAtom
 270  *
 271  * Return a pointer to the class.
 272  * hinstance has been normalized by the caller.
 273  */
 274 static CLASS *CLASS_FindClassByAtom( ATOM atom, HINSTANCE hinstance )
 275 {
 276     struct list *ptr;
 277
 278     USER_Lock();
 279
 280     LIST_FOR_EACH( ptr, &class_list )
 281     {
 282         CLASS *class = LIST_ENTRY( ptr, CLASS, entry );
 283         if (class->atomName != atom) continue;
 284         if (!hinstance || !class->local || class->hInstance == hinstance)
 285         {
 286             TRACE("0x%04x %p -> %p\n", atom, hinstance, class);
 287             return class;
 288         }
 289     }
 290     USER_Unlock();
 291     TRACE("0x%04x %p -> not found\n", atom, hinstance);
 292     return NULL;
 293 }
 294
 295
 296 /***********************************************************************
 297  *           CLASS_RegisterClass
 298  *
 299  * The real RegisterClass() functionality.
 300  * The atom is deleted no matter what.
 301  */
 302 static CLASS *CLASS_RegisterClass( ATOM atom, HINSTANCE hInstance, BOOL local,
 303                                    DWORD style, INT classExtra, INT winExtra )
 304 {
 305     CLASS *classPtr;
 306     BOOL ret;
 307
 308     TRACE("atom=0x%x hinst=%p style=0x%x clExtr=0x%x winExtr=0x%x\n",
 309           atom, hInstance, style, classExtra, winExtra );
 310
 311     /* Fix the extra bytes value */
 312
 313     if (classExtra < 0 || winExtra < 0)
 314     {
 315          SetLastError( ERROR_INVALID_PARAMETER );
 316          return NULL;
 317     }
 318     if (classExtra > 40)  /* Extra bytes are limited to 40 in Win32 */
 319         WARN("Class extra bytes %d is > 40\n", classExtra);
 320     if (winExtra > 40)    /* Extra bytes are limited to 40 in Win32 */
 321         WARN("Win extra bytes %d is > 40\n", winExtra );
 322
 323     classPtr = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(CLASS) + classExtra );
 324     if (!classPtr)
 325     {
 326         GlobalDeleteAtom( atom );
 327         return NULL;
 328     }
 329
 330     SERVER_START_REQ( create_class )
 331     {
 332         req->local      = local;
 333         req->atom       = atom;
 334         req->style      = style;
 335         req->instance   = hInstance;
 336         req->extra      = classExtra;
 337         req->win_extra  = winExtra;
 338         req->client_ptr = classPtr;
 339         ret = !wine_server_call_err( req );
 340     }
 341     SERVER_END_REQ;
 342     GlobalDeleteAtom( atom );  /* the server increased the atom ref count */
 343     if (!ret)
 344     {
 345         HeapFree( GetProcessHeap(), 0, classPtr );
 346         return NULL;
 347     }
 348
 349     classPtr->style       = style;
 350     classPtr->local       = local;
 351     classPtr->cbWndExtra  = winExtra;
 352     classPtr->cbClsExtra  = classExtra;
 353     classPtr->hInstance   = hInstance;
 354     classPtr->atomName    = atom;
 355
 356     /* Other non-null values must be set by caller */
 357
 358     USER_Lock();
 359     if (local) list_add_head( &class_list, &classPtr->entry );
 360     else list_add_tail( &class_list, &classPtr->entry );
 361     return classPtr;
 362 }
 363
 364
 365 /***********************************************************************
 366  *           register_builtin
 367  *
 368  * Register a builtin control class.
 369  * This allows having both ASCII and Unicode winprocs for the same class.
 370  */
 371 static CLASS *register_builtin( const struct builtin_class_descr *descr )
 372 {
 373     ATOM atom;
 374     CLASS *classPtr;
 375
 376     if (!(atom = GlobalAddAtomA( descr->name ))) return 0;
 377
 378     if (!(classPtr = CLASS_RegisterClass( atom, user32_module, FALSE,
 379                                           descr->style, 0, descr->extra ))) return 0;
 380
 381     classPtr->hCursor       = LoadCursorA( 0, (LPSTR)descr->cursor );
 382     classPtr->hbrBackground = descr->brush;
 383     classPtr->winproc       = WINPROC_AllocProc( descr->procA, descr->procW );
 384     release_class_ptr( classPtr );
 385     return classPtr;
 386 }
 387
 388
 389 /***********************************************************************
 390  *           CLASS_RegisterBuiltinClasses
 391  */
 392 void CLASS_RegisterBuiltinClasses(void)
 393 {
 394     extern const struct builtin_class_descr BUTTON_builtin_class;
 395     extern const struct builtin_class_descr COMBO_builtin_class;
 396     extern const struct builtin_class_descr COMBOLBOX_builtin_class;
 397     extern const struct builtin_class_descr DIALOG_builtin_class;
 398     extern const struct builtin_class_descr DESKTOP_builtin_class;
 399     extern const struct builtin_class_descr EDIT_builtin_class;
 400     extern const struct builtin_class_descr ICONTITLE_builtin_class;
 401     extern const struct builtin_class_descr LISTBOX_builtin_class;
 402     extern const struct builtin_class_descr MDICLIENT_builtin_class;
 403     extern const struct builtin_class_descr MENU_builtin_class;
 404     extern const struct builtin_class_descr SCROLL_builtin_class;
 405     extern const struct builtin_class_descr STATIC_builtin_class;
 406
 407     register_builtin( &DESKTOP_builtin_class );
 408     register_builtin( &BUTTON_builtin_class );
 409     register_builtin( &COMBO_builtin_class );
 410     register_builtin( &COMBOLBOX_builtin_class );
 411     register_builtin( &DIALOG_builtin_class );
 412     register_builtin( &EDIT_builtin_class );
 413     register_builtin( &ICONTITLE_builtin_class );
 414     register_builtin( &LISTBOX_builtin_class );
 415     register_builtin( &MDICLIENT_builtin_class );
 416     register_builtin( &MENU_builtin_class );
 417     register_builtin( &SCROLL_builtin_class );
 418     register_builtin( &STATIC_builtin_class );
 419 }
 420
 421
 422 /***********************************************************************
 423  *           CLASS_AddWindow
 424  *
 425  * Add a new window using this class, and set the necessary
 426  * information inside the window structure.
 427  */
 428 void CLASS_AddWindow( CLASS *class, WND *win, BOOL unicode )
 429 {
 430     win->class    = class;
 431     win->clsStyle = class->style;
 432     win->winproc  = class->winproc;
 433     if (WINPROC_IsUnicode( win->winproc, unicode )) win->flags |= WIN_ISUNICODE;
 434 }
 435
 436
 437 /***********************************************************************
 438  *              RegisterClassA (USER32.@)
 439  *
 440  * Register a window class.
 441  *
 442  * RETURNS
 443  *      >0: Unique identifier
 444  *      0: Failure
 445  */
 446 ATOM WINAPI RegisterClassA( const WNDCLASSA* wc ) /* [in] Address of structure with class data */
 447 {
 448     WNDCLASSEXA wcex;
 449
 450     wcex.cbSize        = sizeof(wcex);
 451     wcex.style         = wc->style;
 452     wcex.lpfnWndProc   = wc->lpfnWndProc;
 453     wcex.cbClsExtra    = wc->cbClsExtra;
 454     wcex.cbWndExtra    = wc->cbWndExtra;
 455     wcex.hInstance     = wc->hInstance;
 456     wcex.hIcon         = wc->hIcon;
 457     wcex.hCursor       = wc->hCursor;
 458     wcex.hbrBackground = wc->hbrBackground;
 459     wcex.lpszMenuName  = wc->lpszMenuName;
 460     wcex.lpszClassName = wc->lpszClassName;
 461     wcex.hIconSm       = 0;
 462     return RegisterClassExA( &wcex );
 463 }
 464
 465
 466 /***********************************************************************
 467  *              RegisterClassW (USER32.@)
 468  *
 469  * See RegisterClassA.
 470  */
 471 ATOM WINAPI RegisterClassW( const WNDCLASSW* wc )
 472 {
 473     WNDCLASSEXW wcex;
 474
 475     wcex.cbSize        = sizeof(wcex);
 476     wcex.style         = wc->style;
 477     wcex.lpfnWndProc   = wc->lpfnWndProc;
 478     wcex.cbClsExtra    = wc->cbClsExtra;
 479     wcex.cbWndExtra    = wc->cbWndExtra;
 480     wcex.hInstance     = wc->hInstance;
 481     wcex.hIcon         = wc->hIcon;
 482     wcex.hCursor       = wc->hCursor;
 483     wcex.hbrBackground = wc->hbrBackground;
 484     wcex.lpszMenuName  = wc->lpszMenuName;
 485     wcex.lpszClassName = wc->lpszClassName;
 486     wcex.hIconSm       = 0;
 487     return RegisterClassExW( &wcex );
 488 }
 489
 490
 491 /***********************************************************************
 492  *              RegisterClassExA (USER32.@)
 493  */
 494 ATOM WINAPI RegisterClassExA( const WNDCLASSEXA* wc )
 495 {
 496     ATOM atom;
 497     CLASS *classPtr;
 498     HINSTANCE instance;
 499
 500     if (wc->hInstance == user32_module)
 501     {
 502         /* we can't register a class for user32 */
 503         SetLastError( ERROR_INVALID_PARAMETER );
 504         return 0;
 505     }
 506     if (!(instance = wc->hInstance)) instance = GetModuleHandleW( NULL );
 507
 508     if (!(atom = GlobalAddAtomA( wc->lpszClassName ))) return 0;
 509
 510     if (!(classPtr = CLASS_RegisterClass( atom, instance, !(wc->style & CS_GLOBALCLASS),
 511                                           wc->style, wc->cbClsExtra, wc->cbWndExtra )))
 512         return 0;
 513
 514     TRACE("atom=%04x wndproc=%p hinst=%p bg=%p style=%08x clsExt=%d winExt=%d class=%p\n",
 515           atom, wc->lpfnWndProc, instance, wc->hbrBackground,
 516           wc->style, wc->cbClsExtra, wc->cbWndExtra, classPtr );
 517
 518     classPtr->hIcon         = wc->hIcon;
 519     classPtr->hIconSm       = wc->hIconSm;
 520     classPtr->hCursor       = wc->hCursor;
 521     classPtr->hbrBackground = wc->hbrBackground;
 522     classPtr->winproc       = WINPROC_AllocProc( wc->lpfnWndProc, NULL );
 523     CLASS_SetMenuNameA( classPtr, wc->lpszMenuName );
 524     release_class_ptr( classPtr );
 525     return atom;
 526 }
 527
 528
 529 /***********************************************************************
 530  *              RegisterClassExW (USER32.@)
 531  */
 532 ATOM WINAPI RegisterClassExW( const WNDCLASSEXW* wc )
 533 {
 534     ATOM atom;
 535     CLASS *classPtr;
 536     HINSTANCE instance;
 537
 538     if (wc->hInstance == user32_module)
 539     {
 540         /* we can't register a class for user32 */
 541         SetLastError( ERROR_INVALID_PARAMETER );
 542         return 0;
 543     }
 544     if (!(instance = wc->hInstance)) instance = GetModuleHandleW( NULL );
 545
 546     if (!(atom = GlobalAddAtomW( wc->lpszClassName ))) return 0;
 547
 548     if (!(classPtr = CLASS_RegisterClass( atom, instance, !(wc->style & CS_GLOBALCLASS),
 549                                           wc->style, wc->cbClsExtra, wc->cbWndExtra )))
 550         return 0;
 551
 552     TRACE("atom=%04x wndproc=%p hinst=%p bg=%p style=%08x clsExt=%d winExt=%d class=%p\n",
 553           atom, wc->lpfnWndProc, instance, wc->hbrBackground,
 554           wc->style, wc->cbClsExtra, wc->cbWndExtra, classPtr );
 555
 556     classPtr->hIcon         = wc->hIcon;
 557     classPtr->hIconSm       = wc->hIconSm;
 558     classPtr->hCursor       = wc->hCursor;
 559     classPtr->hbrBackground = wc->hbrBackground;
 560     classPtr->winproc       = WINPROC_AllocProc( NULL, wc->lpfnWndProc );
 561     CLASS_SetMenuNameW( classPtr, wc->lpszMenuName );
 562     release_class_ptr( classPtr );
 563     return atom;
 564 }
 565
 566
 567 /***********************************************************************
 568  *              UnregisterClassA (USER32.@)
 569  */
 570 BOOL WINAPI UnregisterClassA( LPCSTR className, HINSTANCE hInstance )
 571 {
 572     ATOM atom = HIWORD(className) ? GlobalFindAtomA( className ) : LOWORD(className);
 573     return UnregisterClassW( (LPCWSTR)MAKEINTATOM(atom), hInstance );
 574 }
 575
 576 /***********************************************************************
 577  *              UnregisterClassW (USER32.@)
 578  */
 579 BOOL WINAPI UnregisterClassW( LPCWSTR className, HINSTANCE hInstance )
 580 {
 581     CLASS *classPtr = NULL;
 582     ATOM atom = HIWORD(className) ? GlobalFindAtomW( className ) : LOWORD(className);
 583
 584     TRACE("%s %p %x\n",debugstr_w(className), hInstance, atom);
 585
 586     if (!atom)
 587     {
 588         SetLastError( ERROR_CLASS_DOES_NOT_EXIST );
 589         return FALSE;
 590     }
 591
 592     SERVER_START_REQ( destroy_class )
 593     {
 594         req->atom = atom;
 595         req->instance = hInstance;
 596         if (!wine_server_call_err( req )) classPtr = reply->client_ptr;
 597     }
 598     SERVER_END_REQ;
 599
 600     if (classPtr) CLASS_FreeClass( classPtr );
 601     return (classPtr != NULL);
 602 }
 603
 604
 605 /***********************************************************************
 606  *              GetClassWord (USER32.@)
 607  */
 608 WORD WINAPI GetClassWord( HWND hwnd, INT offset )
 609 {
 610     CLASS *class;
 611     WORD retvalue = 0;
 612
 613     if (offset < 0) return GetClassLongA( hwnd, offset );
 614
 615     TRACE("%p %x\n",hwnd, offset);
 616
 617     if (!(class = get_class_ptr( hwnd, FALSE ))) return 0;
 618
 619     if (class == CLASS_OTHER_PROCESS)
 620     {
 621         SERVER_START_REQ( set_class_info )
 622         {
 623             req->window = hwnd;
 624             req->flags = 0;
 625             req->extra_offset = offset;
 626             req->extra_size = sizeof(retvalue);
 627             if (!wine_server_call_err( req ))
 628                 memcpy( &retvalue, &reply->old_extra_value, sizeof(retvalue) );
 629         }
 630         SERVER_END_REQ;
 631         return retvalue;
 632     }
 633
 634     if (offset <= class->cbClsExtra - sizeof(WORD))
 635         memcpy( &retvalue, (char *)(class + 1) + offset, sizeof(retvalue) );
 636     else
 637         SetLastError( ERROR_INVALID_INDEX );
 638     release_class_ptr( class );
 639     return retvalue;
 640 }
 641
 642
 643 /***********************************************************************
 644  *             CLASS_GetClassLong
 645  *
 646  * Implementation of GetClassLong(Ptr)A/W
 647  */
 648 static ULONG_PTR CLASS_GetClassLong( HWND hwnd, INT offset, UINT size,
 649                                      BOOL unicode )
 650 {
 651     CLASS *class;
 652     ULONG_PTR retvalue = 0;
 653
 654     TRACE("%p %d\n", hwnd, offset);
 655
 656     if (!(class = get_class_ptr( hwnd, FALSE ))) return 0;
 657
 658     if (class == CLASS_OTHER_PROCESS)
 659     {
 660         SERVER_START_REQ( set_class_info )
 661         {
 662             req->window = hwnd;
 663             req->flags = 0;
 664             req->extra_offset = (offset >= 0) ? offset : -1;
 665             req->extra_size = (offset >= 0) ? size : 0;
 666             if (!wine_server_call_err( req ))
 667             {
 668                 switch(offset)
 669                 {
 670                 case GCLP_HBRBACKGROUND:
 671                 case GCLP_HCURSOR:
 672                 case GCLP_HICON:
 673                 case GCLP_HICONSM:
 674                 case GCLP_WNDPROC:
 675                 case GCLP_MENUNAME:
 676                     FIXME( "offset %d (%s) not supported on other process window %p\n",
 677                            offset, SPY_GetClassLongOffsetName(offset), hwnd );
 678                     SetLastError( ERROR_INVALID_HANDLE );
 679                     break;
 680                 case GCL_STYLE:
 681                     retvalue = reply->old_style;
 682                     break;
 683                 case GCL_CBWNDEXTRA:
 684                     retvalue = reply->old_win_extra;
 685                     break;
 686                 case GCL_CBCLSEXTRA:
 687                     retvalue = reply->old_extra;
 688                     break;
 689                 case GCLP_HMODULE:
 690                     retvalue = (ULONG_PTR)reply->old_instance;
 691                     break;
 692                 case GCW_ATOM:
 693                     retvalue = reply->old_atom;
 694                     break;
 695                 default:
 696                     if (offset >= 0)
 697                     {
 698                         if (size == sizeof(DWORD))
 699                         {
 700                             DWORD retdword;
 701                             memcpy( &retdword, &reply->old_extra_value, sizeof(DWORD) );
 702                             retvalue = retdword;
 703                         }
 704                         else
 705                             memcpy( &retvalue, &reply->old_extra_value,
 706                                     sizeof(ULONG_PTR) );
 707                     }
 708                     else SetLastError( ERROR_INVALID_INDEX );
 709                     break;
 710                 }
 711             }
 712         }
 713         SERVER_END_REQ;
 714         return retvalue;
 715     }
 716
 717     if (offset >= 0)
 718     {
 719         if (offset <= class->cbClsExtra - size)
 720         {
 721             if (size == sizeof(DWORD))
 722             {
 723                 DWORD retdword;
 724                 memcpy( &retdword, (char *)(class + 1) + offset, sizeof(DWORD) );
 725                 retvalue = retdword;
 726             }
 727             else
 728                 memcpy( &retvalue, (char *)(class + 1) + offset, sizeof(ULONG_PTR) );
 729         }
 730         else
 731             SetLastError( ERROR_INVALID_INDEX );
 732         release_class_ptr( class );
 733         return retvalue;
 734     }
 735
 736     switch(offset)
 737     {
 738     case GCLP_HBRBACKGROUND:
 739         retvalue = (ULONG_PTR)class->hbrBackground;
 740         break;
 741     case GCLP_HCURSOR:
 742         retvalue = (ULONG_PTR)class->hCursor;
 743         break;
 744     case GCLP_HICON:
 745         retvalue = (ULONG_PTR)class->hIcon;
 746         break;
 747     case GCLP_HICONSM:
 748         retvalue = (ULONG_PTR)class->hIconSm;
 749         break;
 750     case GCL_STYLE:
 751         retvalue = class->style;
 752         break;
 753     case GCL_CBWNDEXTRA:
 754         retvalue = class->cbWndExtra;
 755         break;
 756     case GCL_CBCLSEXTRA:
 757         retvalue = class->cbClsExtra;
 758         break;
 759     case GCLP_HMODULE:
 760         retvalue = (ULONG_PTR)class->hInstance;
 761         break;
 762     case GCLP_WNDPROC:
 763         retvalue = (ULONG_PTR)WINPROC_GetProc( class->winproc, unicode );
 764         break;
 765     case GCLP_MENUNAME:
 766         retvalue = (ULONG_PTR)CLASS_GetMenuNameW( class );
 767         if (unicode)
 768             retvalue = (ULONG_PTR)CLASS_GetMenuNameW( class );
 769         else
 770             retvalue = (ULONG_PTR)CLASS_GetMenuNameA( class );
 771         break;
 772     case GCW_ATOM:
 773         retvalue = class->atomName;
 774         break;
 775     default:
 776         SetLastError( ERROR_INVALID_INDEX );
 777         break;
 778     }
 779     release_class_ptr( class );
 780     return retvalue;
 781 }
 782
 783
 784 /***********************************************************************
 785  *              GetClassLongW (USER32.@)
 786  */
 787 DWORD WINAPI GetClassLongW( HWND hwnd, INT offset )
 788 {
 789     return CLASS_GetClassLong( hwnd, offset, sizeof(DWORD), TRUE );
 790 }
 791
 792
 793
 794 /***********************************************************************
 795  *              GetClassLongA (USER32.@)
 796  */
 797 DWORD WINAPI GetClassLongA( HWND hwnd, INT offset )
 798 {
 799     return CLASS_GetClassLong( hwnd, offset, sizeof(DWORD), FALSE );
 800 }
 801
 802
 803 /***********************************************************************
 804  *              SetClassWord (USER32.@)
 805  */
 806 WORD WINAPI SetClassWord( HWND hwnd, INT offset, WORD newval )
 807 {
 808     CLASS *class;
 809     WORD retval = 0;
 810
 811     if (offset < 0) return SetClassLongA( hwnd, offset, (DWORD)newval );
 812
 813     TRACE("%p %d %x\n", hwnd, offset, newval);
 814
 815     if (!(class = get_class_ptr( hwnd, TRUE ))) return 0;
 816
 817     SERVER_START_REQ( set_class_info )
 818     {
 819         req->window = hwnd;
 820         req->flags = SET_CLASS_EXTRA;
 821         req->extra_offset = offset;
 822         req->extra_size = sizeof(newval);
 823         memcpy( &req->extra_value, &newval, sizeof(newval) );
 824         if (!wine_server_call_err( req ))
 825         {
 826             void *ptr = (char *)(class + 1) + offset;
 827             memcpy( &retval, ptr, sizeof(retval) );
 828             memcpy( ptr, &newval, sizeof(newval) );
 829         }
 830     }
 831     SERVER_END_REQ;
 832     release_class_ptr( class );
 833     return retval;
 834 }
 835
 836
 837 /***********************************************************************
 838  *             CLASS_SetClassLong
 839  *
 840  * Implementation of SetClassLong(Ptr)A/W
 841  */
 842 static ULONG_PTR CLASS_SetClassLong( HWND hwnd, INT offset, LONG_PTR newval,
 843                                      UINT size, BOOL unicode )
 844 {
 845     CLASS *class;
 846     ULONG_PTR retval = 0;
 847
 848     TRACE("%p %d %lx\n", hwnd, offset, newval);
 849
 850     if (!(class = get_class_ptr( hwnd, TRUE ))) return 0;
 851
 852     if (offset >= 0)
 853     {
 854         if (set_server_info( hwnd, offset, newval, size ))
 855         {
 856             void *ptr = (char *)(class + 1) + offset;
 857             if ( size == sizeof(LONG) )
 858             {
 859                 DWORD retdword;
 860                 LONG newlong = newval;
 861                 memcpy( &retdword, ptr, sizeof(DWORD) );
 862                 memcpy( ptr, &newlong, sizeof(LONG) );
 863                 retval = retdword;
 864             }
 865             else
 866             {
 867                 memcpy( &retval, ptr, sizeof(ULONG_PTR) );
 868                 memcpy( ptr, &newval, sizeof(LONG_PTR) );
 869             }
 870         }
 871     }
 872     else switch(offset)
 873     {
 874     case GCLP_MENUNAME:
 875         if ( unicode )
 876             CLASS_SetMenuNameW( class, (LPCWSTR)newval );
 877         else
 878             CLASS_SetMenuNameA( class, (LPCSTR)newval );
 879         retval = 0;  /* Old value is now meaningless anyway */
 880         break;
 881     case GCLP_WNDPROC:
 882         retval = (ULONG_PTR)WINPROC_GetProc( class->winproc, unicode );
 883         class->winproc = WINPROC_AllocProc( unicode ? NULL : (WNDPROC)newval,
 884                                             unicode ? (WNDPROC)newval : NULL );
 885         break;
 886     case GCLP_HBRBACKGROUND:
 887         retval = (ULONG_PTR)class->hbrBackground;
 888         class->hbrBackground = (HBRUSH)newval;
 889         break;
 890     case GCLP_HCURSOR:
 891         retval = (ULONG_PTR)class->hCursor;
 892         class->hCursor = (HCURSOR)newval;
 893         break;
 894     case GCLP_HICON:
 895         retval = (ULONG_PTR)class->hIcon;
 896         class->hIcon = (HICON)newval;
 897         break;
 898     case GCLP_HICONSM:
 899         retval = (ULONG_PTR)class->hIconSm;
 900         class->hIconSm = (HICON)newval;
 901         break;
 902     case GCL_STYLE:
 903         if (!set_server_info( hwnd, offset, newval, size )) break;
 904         retval = class->style;
 905         class->style = newval;
 906         break;
 907     case GCL_CBWNDEXTRA:
 908         if (!set_server_info( hwnd, offset, newval, size )) break;
 909         retval = class->cbWndExtra;
 910         class->cbWndExtra = newval;
 911         break;
 912     case GCLP_HMODULE:
 913         if (!set_server_info( hwnd, offset, newval, size )) break;
 914         retval = (ULONG_PTR)class->hInstance;
 915         class->hInstance = (HINSTANCE)newval;
 916         break;
 917     case GCW_ATOM:
 918         if (!set_server_info( hwnd, offset, newval, size )) break;
 919         retval = class->atomName;
 920         class->atomName = newval;
 921         break;
 922     case GCL_CBCLSEXTRA:  /* cannot change this one */
 923         SetLastError( ERROR_INVALID_PARAMETER );
 924         break;
 925     default:
 926         SetLastError( ERROR_INVALID_INDEX );
 927         break;
 928     }
 929     release_class_ptr( class );
 930     return retval;
 931 }
 932
 933
 934 /***********************************************************************
 935  *              SetClassLongW (USER32.@)
 936  */
 937 DWORD WINAPI SetClassLongW( HWND hwnd, INT offset, LONG newval )
 938 {
 939     TRACE("%p %d %x\n", hwnd, offset, newval);
 940
 941     return CLASS_SetClassLong( hwnd, offset, newval, sizeof(LONG), TRUE );
 942 }
 943
 944
 945 /***********************************************************************
 946  *              SetClassLongA (USER32.@)
 947  */
 948 DWORD WINAPI SetClassLongA( HWND hwnd, INT offset, LONG newval )
 949 {
 950     TRACE("%p %d %x\n", hwnd, offset, newval);
 951
 952     return CLASS_SetClassLong( hwnd, offset, newval, sizeof(LONG), FALSE );
 953 }
 954
 955
 956 /***********************************************************************
 957  *              GetClassNameA (USER32.@)
 958  */
 959 INT WINAPI GetClassNameA( HWND hwnd, LPSTR buffer, INT count )
 960 {
 961     char tmpbuf[MAX_ATOM_LEN + 1];
 962     INT ret;
 963
 964     TRACE("%p %p %d\n", hwnd, buffer, count);
 965
 966     if (count <= 0) return 0;
 967
 968     ret = GlobalGetAtomNameA( GetClassLongW( hwnd, GCW_ATOM ), tmpbuf, MAX_ATOM_LEN + 1 );
 969     if (ret)
 970     {
 971         ret = min(count - 1, ret);
 972         memcpy(buffer, tmpbuf, ret);
 973         buffer[ret] = 0;
 974     }
 975     return ret;
 976 }
 977
 978
 979 /***********************************************************************
 980  *              GetClassNameW (USER32.@)
 981  */
 982 INT WINAPI GetClassNameW( HWND hwnd, LPWSTR buffer, INT count )
 983 {
 984     WCHAR tmpbuf[MAX_ATOM_LEN + 1];
 985     INT ret;
 986
 987     TRACE("%p %p %d\n", hwnd, buffer, count);
 988
 989     if (count <= 0) return 0;
 990
 991     ret = GlobalGetAtomNameW( GetClassLongW( hwnd, GCW_ATOM ), tmpbuf, MAX_ATOM_LEN + 1 );
 992     if (ret)
 993     {
 994         ret = min(count - 1, ret);
 995         memcpy(buffer, tmpbuf, ret * sizeof(WCHAR));
 996         buffer[ret] = 0;
 997     }
 998     return ret;
 999 }
1000
1001
1002 /***********************************************************************
1003  *              RealGetWindowClassA (USER32.@)
1004  */
1005 UINT WINAPI RealGetWindowClassA( HWND hwnd, LPSTR buffer, UINT count )
1006 {
1007     return GetClassNameA( hwnd, buffer, count );
1008 }
1009
1010
1011 /***********************************************************************
1012  *              RealGetWindowClassW (USER32.@)
1013  */
1014 UINT WINAPI RealGetWindowClassW( HWND hwnd, LPWSTR buffer, UINT count )
1015 {
1016     return GetClassNameW( hwnd, buffer, count );
1017 }
1018
1019
1020 /***********************************************************************
1021  *              GetClassInfoA (USER32.@)
1022  */
1023 BOOL WINAPI GetClassInfoA( HINSTANCE hInstance, LPCSTR name, WNDCLASSA *wc )
1024 {
1025     WNDCLASSEXA wcex;
1026     UINT ret = GetClassInfoExA( hInstance, name, &wcex );
1027
1028     if (ret)
1029     {
1030         wc->style         = wcex.style;
1031         wc->lpfnWndProc   = wcex.lpfnWndProc;
1032         wc->cbClsExtra    = wcex.cbClsExtra;
1033         wc->cbWndExtra    = wcex.cbWndExtra;
1034         wc->hInstance     = wcex.hInstance;
1035         wc->hIcon         = wcex.hIcon;
1036         wc->hCursor       = wcex.hCursor;
1037         wc->hbrBackground = wcex.hbrBackground;
1038         wc->lpszMenuName  = wcex.lpszMenuName;
1039         wc->lpszClassName = wcex.lpszClassName;
1040     }
1041     return ret;
1042 }
1043
1044
1045 /***********************************************************************
1046  *              GetClassInfoW (USER32.@)
1047  */
1048 BOOL WINAPI GetClassInfoW( HINSTANCE hInstance, LPCWSTR name, WNDCLASSW *wc )
1049 {
1050     WNDCLASSEXW wcex;
1051     UINT ret = GetClassInfoExW( hInstance, name, &wcex );
1052
1053     if (ret)
1054     {
1055         wc->style         = wcex.style;
1056         wc->lpfnWndProc   = wcex.lpfnWndProc;
1057         wc->cbClsExtra    = wcex.cbClsExtra;
1058         wc->cbWndExtra    = wcex.cbWndExtra;
1059         wc->hInstance     = wcex.hInstance;
1060         wc->hIcon         = wcex.hIcon;
1061         wc->hCursor       = wcex.hCursor;
1062         wc->hbrBackground = wcex.hbrBackground;
1063         wc->lpszMenuName  = wcex.lpszMenuName;
1064         wc->lpszClassName = wcex.lpszClassName;
1065     }
1066     return ret;
1067 }
1068
1069
1070 /***********************************************************************
1071  *              GetClassInfoExA (USER32.@)
1072  */
1073 BOOL WINAPI GetClassInfoExA( HINSTANCE hInstance, LPCSTR name, WNDCLASSEXA *wc )
1074 {
1075     ATOM atom = HIWORD(name) ? GlobalFindAtomA( name ) : LOWORD(name);
1076     CLASS *classPtr;
1077
1078     TRACE("%p %s %x %p\n", hInstance, debugstr_a(name), atom, wc);
1079
1080     if (!hInstance) hInstance = user32_module;
1081
1082     if (!atom || !(classPtr = CLASS_FindClassByAtom( atom, hInstance )))
1083     {
1084         SetLastError( ERROR_CLASS_DOES_NOT_EXIST );
1085         return FALSE;
1086     }
1087     wc->style         = classPtr->style;
1088     wc->lpfnWndProc   = WINPROC_GetProc( classPtr->winproc, FALSE );
1089     wc->cbClsExtra    = classPtr->cbClsExtra;
1090     wc->cbWndExtra    = classPtr->cbWndExtra;
1091     wc->hInstance     = (hInstance == user32_module) ? 0 : hInstance;
1092     wc->hIcon         = (HICON)classPtr->hIcon;
1093     wc->hIconSm       = (HICON)classPtr->hIconSm;
1094     wc->hCursor       = (HCURSOR)classPtr->hCursor;
1095     wc->hbrBackground = (HBRUSH)classPtr->hbrBackground;
1096     wc->lpszMenuName  = CLASS_GetMenuNameA( classPtr );
1097     wc->lpszClassName = name;
1098     release_class_ptr( classPtr );
1099
1100     /* We must return the atom of the class here instead of just TRUE. */
1101     return atom;
1102 }
1103
1104
1105 /***********************************************************************
1106  *              GetClassInfoExW (USER32.@)
1107  */
1108 BOOL WINAPI GetClassInfoExW( HINSTANCE hInstance, LPCWSTR name, WNDCLASSEXW *wc )
1109 {
1110     ATOM atom = HIWORD(name) ? GlobalFindAtomW( name ) : LOWORD(name);
1111     CLASS *classPtr;
1112
1113     TRACE("%p %s %x %p\n", hInstance, debugstr_w(name), atom, wc);
1114
1115     if (!hInstance) hInstance = user32_module;
1116
1117     if (!atom || !(classPtr = CLASS_FindClassByAtom( atom, hInstance )))
1118     {
1119         SetLastError( ERROR_CLASS_DOES_NOT_EXIST );
1120         return FALSE;
1121     }
1122     wc->style         = classPtr->style;
1123     wc->lpfnWndProc   = WINPROC_GetProc( classPtr->winproc, TRUE );
1124     wc->cbClsExtra    = classPtr->cbClsExtra;
1125     wc->cbWndExtra    = classPtr->cbWndExtra;
1126     wc->hInstance     = (hInstance == user32_module) ? 0 : hInstance;
1127     wc->hIcon         = (HICON)classPtr->hIcon;
1128     wc->hIconSm       = (HICON)classPtr->hIconSm;
1129     wc->hCursor       = (HCURSOR)classPtr->hCursor;
1130     wc->hbrBackground = (HBRUSH)classPtr->hbrBackground;
1131     wc->lpszMenuName  = CLASS_GetMenuNameW( classPtr );
1132     wc->lpszClassName = name;
1133     release_class_ptr( classPtr );
1134
1135     /* We must return the atom of the class here instead of just TRUE. */
1136     return atom;
1137 }
1138
1139
1140 #if 0  /* toolhelp is in kernel, so this cannot work */
1141
1142 /***********************************************************************
1143  *              ClassFirst (TOOLHELP.69)
1144  */
1145 BOOL16 WINAPI ClassFirst16( CLASSENTRY *pClassEntry )
1146 {
1147     TRACE("%p\n",pClassEntry);
1148     pClassEntry->wNext = 1;
1149     return ClassNext16( pClassEntry );
1150 }
1151
1152
1153 /***********************************************************************
1154  *              ClassNext (TOOLHELP.70)
1155  */
1156 BOOL16 WINAPI ClassNext16( CLASSENTRY *pClassEntry )
1157 {
1158     int i;
1159     CLASS *class = firstClass;
1160
1161     TRACE("%p\n",pClassEntry);
1162
1163     if (!pClassEntry->wNext) return FALSE;
1164     for (i = 1; (i < pClassEntry->wNext) && class; i++) class = class->next;
1165     if (!class)
1166     {
1167         pClassEntry->wNext = 0;
1168         return FALSE;
1169     }
1170     pClassEntry->hInst = class->hInstance;
1171     pClassEntry->wNext++;
1172     GlobalGetAtomNameA( class->atomName, pClassEntry->szClassName,
1173                           sizeof(pClassEntry->szClassName) );
1174     return TRUE;
1175 }
1176 #endif
1177
1178 /* 64bit versions */
1179
1180 #ifdef GetClassLongPtrA
1181 #undef GetClassLongPtrA
1182 #endif
1183
1184 #ifdef GetClassLongPtrW
1185 #undef GetClassLongPtrW
1186 #endif
1187
1188 #ifdef SetClassLongPtrA
1189 #undef SetClassLongPtrA
1190 #endif
1191
1192 #ifdef SetClassLongPtrW
1193 #undef SetClassLongPtrW
1194 #endif
1195
1196 /***********************************************************************
1197  *              GetClassLongPtrA (USER32.@)
1198  */
1199 ULONG_PTR WINAPI GetClassLongPtrA( HWND hwnd, INT offset )
1200 {
1201     return CLASS_GetClassLong( hwnd, offset, sizeof(ULONG_PTR), FALSE );
1202 }
1203
1204 /***********************************************************************
1205  *              GetClassLongPtrW (USER32.@)
1206  */
1207 ULONG_PTR WINAPI GetClassLongPtrW( HWND hwnd, INT offset )
1208 {
1209     return CLASS_GetClassLong( hwnd, offset, sizeof(ULONG_PTR), TRUE );
1210 }
1211
1212 /***********************************************************************
1213  *              SetClassLongPtrW (USER32.@)
1214  */
1215 ULONG_PTR WINAPI SetClassLongPtrW( HWND hwnd, INT offset, LONG_PTR newval )
1216 {
1217     return CLASS_SetClassLong( hwnd, offset, newval, sizeof(LONG_PTR), TRUE );
1218 }
1219
1220 /***********************************************************************
1221  *              SetClassLongPtrA (USER32.@)
1222  */
1223 ULONG_PTR WINAPI SetClassLongPtrA( HWND hwnd, INT offset, LONG_PTR newval )
1224 {
1225     return CLASS_SetClassLong( hwnd, offset, newval, sizeof(LONG_PTR), FALSE );
1226 }