git.oblomov.eu Git - wine/blob - dlls/ntdll/sync.c

   1 /*
   2  *      Process synchronisation
   3  *
   4  * Copyright 1996, 1997, 1998 Marcus Meissner
   5  * Copyright 1997, 1999 Alexandre Julliard
   6  * Copyright 1999, 2000 Juergen Schmied
   7  * Copyright 2003 Eric Pouech
   8  *
   9  * This library is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU Lesser General Public
  11  * License as published by the Free Software Foundation; either
  12  * version 2.1 of the License, or (at your option) any later version.
  13  *
  14  * This library is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17  * Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public
  20  * License along with this library; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  22  */
  23
  24 #include "config.h"
  25
  26 #include <assert.h>
  27 #include <errno.h>
  28 #include <signal.h>
  29 #ifdef HAVE_SYS_TIME_H
  30 # include <sys/time.h>
  31 #endif
  32 #ifdef HAVE_POLL_H
  33 #include <poll.h>
  34 #endif
  35 #ifdef HAVE_SYS_POLL_H
  36 # include <sys/poll.h>
  37 #endif
  38 #ifdef HAVE_UNISTD_H
  39 # include <unistd.h>
  40 #endif
  41 #ifdef HAVE_SCHED_H
  42 # include <sched.h>
  43 #endif
  44 #include <string.h>
  45 #include <stdarg.h>
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <time.h>
  49
  50 #define NONAMELESSUNION
  51 #define NONAMELESSSTRUCT
  52
  53 #include "ntstatus.h"
  54 #define WIN32_NO_STATUS
  55 #include "windef.h"
  56 #include "thread.h"
  57 #include "wine/server.h"
  58 #include "wine/debug.h"
  59 #include "ntdll_misc.h"
  60
  61 WINE_DEFAULT_DEBUG_CHANNEL(ntdll);
  62
  63
  64 /*
  65  *      Semaphores
  66  */
  67
  68 /******************************************************************************
  69  *  NtCreateSemaphore (NTDLL.@)
  70  */
  71 NTSTATUS WINAPI NtCreateSemaphore( OUT PHANDLE SemaphoreHandle,
  72                                    IN ACCESS_MASK access,
  73                                    IN const OBJECT_ATTRIBUTES *attr OPTIONAL,
  74                                    IN LONG InitialCount,
  75                                    IN LONG MaximumCount )
  76 {
  77     DWORD len = attr && attr->ObjectName ? attr->ObjectName->Length : 0;
  78     NTSTATUS ret;
  79
  80     if (MaximumCount <= 0 || InitialCount < 0 || InitialCount > MaximumCount)
  81         return STATUS_INVALID_PARAMETER;
  82     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
  83
  84     SERVER_START_REQ( create_semaphore )
  85     {
  86         req->access  = access;
  87         req->attributes = (attr) ? attr->Attributes : 0;
  88         req->rootdir = attr ? attr->RootDirectory : 0;
  89         req->initial = InitialCount;
  90         req->max     = MaximumCount;
  91         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
  92         ret = wine_server_call( req );
  93         *SemaphoreHandle = reply->handle;
  94     }
  95     SERVER_END_REQ;
  96     return ret;
  97 }
  98
  99 /******************************************************************************
 100  *  NtOpenSemaphore (NTDLL.@)
 101  */
 102 NTSTATUS WINAPI NtOpenSemaphore( OUT PHANDLE SemaphoreHandle,
 103                                  IN ACCESS_MASK access,
 104                                  IN const OBJECT_ATTRIBUTES *attr )
 105 {
 106     DWORD len = attr && attr->ObjectName ? attr->ObjectName->Length : 0;
 107     NTSTATUS ret;
 108
 109     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 110
 111     SERVER_START_REQ( open_semaphore )
 112     {
 113         req->access  = access;
 114         req->attributes = (attr) ? attr->Attributes : 0;
 115         req->rootdir = attr ? attr->RootDirectory : 0;
 116         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 117         ret = wine_server_call( req );
 118         *SemaphoreHandle = reply->handle;
 119     }
 120     SERVER_END_REQ;
 121     return ret;
 122 }
 123
 124 /******************************************************************************
 125  *  NtQuerySemaphore (NTDLL.@)
 126  */
 127 NTSTATUS WINAPI NtQuerySemaphore(
 128         HANDLE SemaphoreHandle,
 129         SEMAPHORE_INFORMATION_CLASS SemaphoreInformationClass,
 130         PVOID SemaphoreInformation,
 131         ULONG Length,
 132         PULONG ReturnLength)
 133 {
 134         FIXME("(%p,%d,%p,0x%08x,%p) stub!\n",
 135         SemaphoreHandle, SemaphoreInformationClass, SemaphoreInformation, Length, ReturnLength);
 136         return STATUS_SUCCESS;
 137 }
 138
 139 /******************************************************************************
 140  *  NtReleaseSemaphore (NTDLL.@)
 141  */
 142 NTSTATUS WINAPI NtReleaseSemaphore( HANDLE handle, ULONG count, PULONG previous )
 143 {
 144     NTSTATUS ret;
 145     SERVER_START_REQ( release_semaphore )
 146     {
 147         req->handle = handle;
 148         req->count  = count;
 149         if (!(ret = wine_server_call( req )))
 150         {
 151             if (previous) *previous = reply->prev_count;
 152         }
 153     }
 154     SERVER_END_REQ;
 155     return ret;
 156 }
 157
 158 /*
 159  *      Events
 160  */
 161
 162 /**************************************************************************
 163  * NtCreateEvent (NTDLL.@)
 164  * ZwCreateEvent (NTDLL.@)
 165  */
 166 NTSTATUS WINAPI NtCreateEvent(
 167         OUT PHANDLE EventHandle,
 168         IN ACCESS_MASK DesiredAccess,
 169         IN const OBJECT_ATTRIBUTES *attr,
 170         IN BOOLEAN ManualReset,
 171         IN BOOLEAN InitialState)
 172 {
 173     DWORD len = attr && attr->ObjectName ? attr->ObjectName->Length : 0;
 174     NTSTATUS ret;
 175
 176     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 177
 178     SERVER_START_REQ( create_event )
 179     {
 180         req->access = DesiredAccess;
 181         req->attributes = (attr) ? attr->Attributes : 0;
 182         req->rootdir = attr ? attr->RootDirectory : 0;
 183         req->manual_reset = ManualReset;
 184         req->initial_state = InitialState;
 185         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 186         ret = wine_server_call( req );
 187         *EventHandle = reply->handle;
 188     }
 189     SERVER_END_REQ;
 190     return ret;
 191 }
 192
 193 /******************************************************************************
 194  *  NtOpenEvent (NTDLL.@)
 195  *  ZwOpenEvent (NTDLL.@)
 196  */
 197 NTSTATUS WINAPI NtOpenEvent(
 198         OUT PHANDLE EventHandle,
 199         IN ACCESS_MASK DesiredAccess,
 200         IN const OBJECT_ATTRIBUTES *attr )
 201 {
 202     DWORD len = attr && attr->ObjectName ? attr->ObjectName->Length : 0;
 203     NTSTATUS ret;
 204
 205     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 206
 207     SERVER_START_REQ( open_event )
 208     {
 209         req->access  = DesiredAccess;
 210         req->attributes = (attr) ? attr->Attributes : 0;
 211         req->rootdir = attr ? attr->RootDirectory : 0;
 212         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 213         ret = wine_server_call( req );
 214         *EventHandle = reply->handle;
 215     }
 216     SERVER_END_REQ;
 217     return ret;
 218 }
 219
 220
 221 /******************************************************************************
 222  *  NtSetEvent (NTDLL.@)
 223  *  ZwSetEvent (NTDLL.@)
 224  */
 225 NTSTATUS WINAPI NtSetEvent( HANDLE handle, PULONG NumberOfThreadsReleased )
 226 {
 227     NTSTATUS ret;
 228
 229     /* FIXME: set NumberOfThreadsReleased */
 230
 231     SERVER_START_REQ( event_op )
 232     {
 233         req->handle = handle;
 234         req->op     = SET_EVENT;
 235         ret = wine_server_call( req );
 236     }
 237     SERVER_END_REQ;
 238     return ret;
 239 }
 240
 241 /******************************************************************************
 242  *  NtResetEvent (NTDLL.@)
 243  */
 244 NTSTATUS WINAPI NtResetEvent( HANDLE handle, PULONG NumberOfThreadsReleased )
 245 {
 246     NTSTATUS ret;
 247
 248     /* resetting an event can't release any thread... */
 249     if (NumberOfThreadsReleased) *NumberOfThreadsReleased = 0;
 250
 251     SERVER_START_REQ( event_op )
 252     {
 253         req->handle = handle;
 254         req->op     = RESET_EVENT;
 255         ret = wine_server_call( req );
 256     }
 257     SERVER_END_REQ;
 258     return ret;
 259 }
 260
 261 /******************************************************************************
 262  *  NtClearEvent (NTDLL.@)
 263  *
 264  * FIXME
 265  *   same as NtResetEvent ???
 266  */
 267 NTSTATUS WINAPI NtClearEvent ( HANDLE handle )
 268 {
 269     return NtResetEvent( handle, NULL );
 270 }
 271
 272 /******************************************************************************
 273  *  NtPulseEvent (NTDLL.@)
 274  *
 275  * FIXME
 276  *   PulseCount
 277  */
 278 NTSTATUS WINAPI NtPulseEvent( HANDLE handle, PULONG PulseCount )
 279 {
 280     NTSTATUS ret;
 281
 282     if (PulseCount)
 283       FIXME("(%p,%d)\n", handle, *PulseCount);
 284
 285     SERVER_START_REQ( event_op )
 286     {
 287         req->handle = handle;
 288         req->op     = PULSE_EVENT;
 289         ret = wine_server_call( req );
 290     }
 291     SERVER_END_REQ;
 292     return ret;
 293 }
 294
 295 /******************************************************************************
 296  *  NtQueryEvent (NTDLL.@)
 297  */
 298 NTSTATUS WINAPI NtQueryEvent (
 299         IN  HANDLE EventHandle,
 300         IN  EVENT_INFORMATION_CLASS EventInformationClass,
 301         OUT PVOID EventInformation,
 302         IN  ULONG EventInformationLength,
 303         OUT PULONG  ReturnLength)
 304 {
 305         FIXME("(%p)\n", EventHandle);
 306         return STATUS_SUCCESS;
 307 }
 308
 309 /*
 310  *      Mutants (known as Mutexes in Kernel32)
 311  */
 312
 313 /******************************************************************************
 314  *              NtCreateMutant                          [NTDLL.@]
 315  *              ZwCreateMutant                          [NTDLL.@]
 316  */
 317 NTSTATUS WINAPI NtCreateMutant(OUT HANDLE* MutantHandle,
 318                                IN ACCESS_MASK access,
 319                                IN const OBJECT_ATTRIBUTES* attr OPTIONAL,
 320                                IN BOOLEAN InitialOwner)
 321 {
 322     NTSTATUS    status;
 323     DWORD       len = attr && attr->ObjectName ? attr->ObjectName->Length : 0;
 324
 325     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 326
 327     SERVER_START_REQ( create_mutex )
 328     {
 329         req->access  = access;
 330         req->attributes = (attr) ? attr->Attributes : 0;
 331         req->rootdir = attr ? attr->RootDirectory : 0;
 332         req->owned   = InitialOwner;
 333         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 334         status = wine_server_call( req );
 335         *MutantHandle = reply->handle;
 336     }
 337     SERVER_END_REQ;
 338     return status;
 339 }
 340
 341 /**************************************************************************
 342  *              NtOpenMutant                            [NTDLL.@]
 343  *              ZwOpenMutant                            [NTDLL.@]
 344  */
 345 NTSTATUS WINAPI NtOpenMutant(OUT HANDLE* MutantHandle,
 346                              IN ACCESS_MASK access,
 347                              IN const OBJECT_ATTRIBUTES* attr )
 348 {
 349     NTSTATUS    status;
 350     DWORD       len = attr && attr->ObjectName ? attr->ObjectName->Length : 0;
 351
 352     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 353
 354     SERVER_START_REQ( open_mutex )
 355     {
 356         req->access  = access;
 357         req->attributes = (attr) ? attr->Attributes : 0;
 358         req->rootdir = attr ? attr->RootDirectory : 0;
 359         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 360         status = wine_server_call( req );
 361         *MutantHandle = reply->handle;
 362     }
 363     SERVER_END_REQ;
 364     return status;
 365 }
 366
 367 /**************************************************************************
 368  *              NtReleaseMutant                         [NTDLL.@]
 369  *              ZwReleaseMutant                         [NTDLL.@]
 370  */
 371 NTSTATUS WINAPI NtReleaseMutant( IN HANDLE handle, OUT PLONG prev_count OPTIONAL)
 372 {
 373     NTSTATUS    status;
 374
 375     SERVER_START_REQ( release_mutex )
 376     {
 377         req->handle = handle;
 378         status = wine_server_call( req );
 379         if (prev_count) *prev_count = reply->prev_count;
 380     }
 381     SERVER_END_REQ;
 382     return status;
 383 }
 384
 385 /******************************************************************
 386  *              NtQueryMutant                   [NTDLL.@]
 387  *              ZwQueryMutant                   [NTDLL.@]
 388  */
 389 NTSTATUS WINAPI NtQueryMutant(IN HANDLE handle,
 390                               IN MUTANT_INFORMATION_CLASS MutantInformationClass,
 391                               OUT PVOID MutantInformation,
 392                               IN ULONG MutantInformationLength,
 393                               OUT PULONG ResultLength OPTIONAL )
 394 {
 395     FIXME("(%p %u %p %u %p): stub!\n",
 396           handle, MutantInformationClass, MutantInformation, MutantInformationLength, ResultLength);
 397     return STATUS_NOT_IMPLEMENTED;
 398 }
 399
 400 /*
 401  *      Timers
 402  */
 403
 404 /**************************************************************************
 405  *              NtCreateTimer                           [NTDLL.@]
 406  *              ZwCreateTimer                           [NTDLL.@]
 407  */
 408 NTSTATUS WINAPI NtCreateTimer(OUT HANDLE *handle,
 409                               IN ACCESS_MASK access,
 410                               IN const OBJECT_ATTRIBUTES *attr OPTIONAL,
 411                               IN TIMER_TYPE timer_type)
 412 {
 413     DWORD       len = (attr && attr->ObjectName) ? attr->ObjectName->Length : 0;
 414     NTSTATUS    status;
 415
 416     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 417
 418     if (timer_type != NotificationTimer && timer_type != SynchronizationTimer)
 419         return STATUS_INVALID_PARAMETER;
 420
 421     SERVER_START_REQ( create_timer )
 422     {
 423         req->access  = access;
 424         req->attributes = (attr) ? attr->Attributes : 0;
 425         req->rootdir = attr ? attr->RootDirectory : 0;
 426         req->manual  = (timer_type == NotificationTimer) ? TRUE : FALSE;
 427         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 428         status = wine_server_call( req );
 429         *handle = reply->handle;
 430     }
 431     SERVER_END_REQ;
 432     return status;
 433
 434 }
 435
 436 /**************************************************************************
 437  *              NtOpenTimer                             [NTDLL.@]
 438  *              ZwOpenTimer                             [NTDLL.@]
 439  */
 440 NTSTATUS WINAPI NtOpenTimer(OUT PHANDLE handle,
 441                             IN ACCESS_MASK access,
 442                             IN const OBJECT_ATTRIBUTES* attr )
 443 {
 444     DWORD       len = (attr && attr->ObjectName) ? attr->ObjectName->Length : 0;
 445     NTSTATUS    status;
 446
 447     if (len >= MAX_PATH * sizeof(WCHAR)) return STATUS_NAME_TOO_LONG;
 448
 449     SERVER_START_REQ( open_timer )
 450     {
 451         req->access  = access;
 452         req->attributes = (attr) ? attr->Attributes : 0;
 453         req->rootdir = attr ? attr->RootDirectory : 0;
 454         if (len) wine_server_add_data( req, attr->ObjectName->Buffer, len );
 455         status = wine_server_call( req );
 456         *handle = reply->handle;
 457     }
 458     SERVER_END_REQ;
 459     return status;
 460 }
 461
 462 /**************************************************************************
 463  *              NtSetTimer                              [NTDLL.@]
 464  *              ZwSetTimer                              [NTDLL.@]
 465  */
 466 NTSTATUS WINAPI NtSetTimer(IN HANDLE handle,
 467                            IN const LARGE_INTEGER* when,
 468                            IN PTIMER_APC_ROUTINE callback,
 469                            IN PVOID callback_arg,
 470                            IN BOOLEAN resume,
 471                            IN ULONG period OPTIONAL,
 472                            OUT PBOOLEAN state OPTIONAL)
 473 {
 474     NTSTATUS    status = STATUS_SUCCESS;
 475
 476     TRACE("(%p,%p,%p,%p,%08x,0x%08x,%p) stub\n",
 477           handle, when, callback, callback_arg, resume, period, state);
 478
 479     SERVER_START_REQ( set_timer )
 480     {
 481         req->handle   = handle;
 482         req->period   = period;
 483         req->expire   = when->QuadPart;
 484         req->callback = callback;
 485         req->arg      = callback_arg;
 486         status = wine_server_call( req );
 487         if (state) *state = reply->signaled;
 488     }
 489     SERVER_END_REQ;
 490
 491     /* set error but can still succeed */
 492     if (resume && status == STATUS_SUCCESS) return STATUS_TIMER_RESUME_IGNORED;
 493     return status;
 494 }
 495
 496 /**************************************************************************
 497  *              NtCancelTimer                           [NTDLL.@]
 498  *              ZwCancelTimer                           [NTDLL.@]
 499  */
 500 NTSTATUS WINAPI NtCancelTimer(IN HANDLE handle, OUT BOOLEAN* state)
 501 {
 502     NTSTATUS    status;
 503
 504     SERVER_START_REQ( cancel_timer )
 505     {
 506         req->handle = handle;
 507         status = wine_server_call( req );
 508         if (state) *state = reply->signaled;
 509     }
 510     SERVER_END_REQ;
 511     return status;
 512 }
 513
 514 /******************************************************************************
 515  *  NtQueryTimer (NTDLL.@)
 516  *
 517  * Retrieves information about a timer.
 518  *
 519  * PARAMS
 520  *  TimerHandle           [I] The timer to retrieve information about.
 521  *  TimerInformationClass [I] The type of information to retrieve.
 522  *  TimerInformation      [O] Pointer to buffer to store information in.
 523  *  Length                [I] The length of the buffer pointed to by TimerInformation.
 524  *  ReturnLength          [O] Optional. The size of buffer actually used.
 525  *
 526  * RETURNS
 527  *  Success: STATUS_SUCCESS
 528  *  Failure: STATUS_INFO_LENGTH_MISMATCH, if Length doesn't match the required data
 529  *           size for the class specified.
 530  *           STATUS_INVALID_INFO_CLASS, if an invalid TimerInformationClass was specified.
 531  *           STATUS_ACCESS_DENIED, if TimerHandle does not have TIMER_QUERY_STATE access
 532  *           to the timer.
 533  */
 534 NTSTATUS WINAPI NtQueryTimer(
 535     HANDLE TimerHandle,
 536     TIMER_INFORMATION_CLASS TimerInformationClass,
 537     PVOID TimerInformation,
 538     ULONG Length,
 539     PULONG ReturnLength)
 540 {
 541     TIMER_BASIC_INFORMATION * basic_info = (TIMER_BASIC_INFORMATION *)TimerInformation;
 542     NTSTATUS status;
 543     LARGE_INTEGER now;
 544
 545     TRACE("(%p,%d,%p,0x%08x,%p)\n", TimerHandle, TimerInformationClass,
 546        TimerInformation, Length, ReturnLength);
 547
 548     switch (TimerInformationClass)
 549     {
 550     case TimerBasicInformation:
 551         if (Length < sizeof(TIMER_BASIC_INFORMATION))
 552             return STATUS_INFO_LENGTH_MISMATCH;
 553
 554         SERVER_START_REQ(get_timer_info)
 555         {
 556             req->handle = TimerHandle;
 557             status = wine_server_call(req);
 558
 559             /* convert server time to absolute NTDLL time */
 560             basic_info->RemainingTime.QuadPart = reply->when;
 561             basic_info->TimerState = reply->signaled;
 562         }
 563         SERVER_END_REQ;
 564
 565         /* convert from absolute into relative time */
 566         NtQuerySystemTime(&now);
 567         if (now.QuadPart > basic_info->RemainingTime.QuadPart)
 568             basic_info->RemainingTime.QuadPart = 0;
 569         else
 570             basic_info->RemainingTime.QuadPart -= now.QuadPart;
 571
 572         if (ReturnLength) *ReturnLength = sizeof(TIMER_BASIC_INFORMATION);
 573
 574         return status;
 575     }
 576
 577     FIXME("Unhandled class %d\n", TimerInformationClass);
 578     return STATUS_INVALID_INFO_CLASS;
 579 }
 580
 581
 582 /******************************************************************************
 583  * NtQueryTimerResolution [NTDLL.@]
 584  */
 585 NTSTATUS WINAPI NtQueryTimerResolution(OUT ULONG* min_resolution,
 586                                        OUT ULONG* max_resolution,
 587                                        OUT ULONG* current_resolution)
 588 {
 589     FIXME("(%p,%p,%p), stub!\n",
 590           min_resolution, max_resolution, current_resolution);
 591
 592     return STATUS_NOT_IMPLEMENTED;
 593 }
 594
 595 /******************************************************************************
 596  * NtSetTimerResolution [NTDLL.@]
 597  */
 598 NTSTATUS WINAPI NtSetTimerResolution(IN ULONG resolution,
 599                                      IN BOOLEAN set_resolution,
 600                                      OUT ULONG* current_resolution )
 601 {
 602     FIXME("(%u,%u,%p), stub!\n",
 603           resolution, set_resolution, current_resolution);
 604
 605     return STATUS_NOT_IMPLEMENTED;
 606 }
 607
 608
 609 /***********************************************************************
 610  *              wait_reply
 611  *
 612  * Wait for a reply on the waiting pipe of the current thread.
 613  */
 614 static int wait_reply( void *cookie )
 615 {
 616     int signaled;
 617     struct wake_up_reply reply;
 618     for (;;)
 619     {
 620         int ret;
 621         ret = read( ntdll_get_thread_data()->wait_fd[0], &reply, sizeof(reply) );
 622         if (ret == sizeof(reply))
 623         {
 624             if (!reply.cookie) break;  /* thread got killed */
 625             if (reply.cookie == cookie) return reply.signaled;
 626             /* we stole another reply, wait for the real one */
 627             signaled = wait_reply( cookie );
 628             /* and now put the wrong one back in the pipe */
 629             for (;;)
 630             {
 631                 ret = write( ntdll_get_thread_data()->wait_fd[1], &reply, sizeof(reply) );
 632                 if (ret == sizeof(reply)) break;
 633                 if (ret >= 0) server_protocol_error( "partial wakeup write %d\n", ret );
 634                 if (errno == EINTR) continue;
 635                 server_protocol_perror("wakeup write");
 636             }
 637             return signaled;
 638         }
 639         if (ret >= 0) server_protocol_error( "partial wakeup read %d\n", ret );
 640         if (errno == EINTR) continue;
 641         server_protocol_perror("wakeup read");
 642     }
 643     /* the server closed the connection; time to die... */
 644     server_abort_thread(0);
 645 }
 646
 647
 648 /***********************************************************************
 649  *              invoke_apc
 650  *
 651  * Invoke a single APC. Return TRUE if a user APC has been run.
 652  */
 653 static BOOL invoke_apc( const apc_call_t *call, apc_result_t *result )
 654 {
 655     BOOL user_apc = FALSE;
 656
 657     memset( result, 0, sizeof(*result) );
 658
 659     switch (call->type)
 660     {
 661     case APC_NONE:
 662         break;
 663     case APC_USER:
 664         call->user.func( call->user.args[0], call->user.args[1], call->user.args[2] );
 665         user_apc = TRUE;
 666         break;
 667     case APC_TIMER:
 668         call->timer.func( call->timer.arg, (DWORD)call->timer.time, (DWORD)(call->timer.time >> 32) );
 669         user_apc = TRUE;
 670         break;
 671     case APC_ASYNC_IO:
 672         result->type = call->type;
 673         result->async_io.status = call->async_io.func( call->async_io.user,
 674                                                        call->async_io.sb,
 675                                                        call->async_io.status );
 676         break;
 677     case APC_VIRTUAL_ALLOC:
 678         result->type = call->type;
 679         result->virtual_alloc.addr = call->virtual_alloc.addr;
 680         result->virtual_alloc.size = call->virtual_alloc.size;
 681         result->virtual_alloc.status = NtAllocateVirtualMemory( NtCurrentProcess(),
 682                                                                 &result->virtual_alloc.addr,
 683                                                                 call->virtual_alloc.zero_bits,
 684                                                                 &result->virtual_alloc.size,
 685                                                                 call->virtual_alloc.op_type,
 686                                                                 call->virtual_alloc.prot );
 687         break;
 688     case APC_VIRTUAL_FREE:
 689         result->type = call->type;
 690         result->virtual_free.addr = call->virtual_free.addr;
 691         result->virtual_free.size = call->virtual_free.size;
 692         result->virtual_free.status = NtFreeVirtualMemory( NtCurrentProcess(),
 693                                                            &result->virtual_free.addr,
 694                                                            &result->virtual_free.size,
 695                                                            call->virtual_free.op_type );
 696         break;
 697     case APC_VIRTUAL_QUERY:
 698     {
 699         MEMORY_BASIC_INFORMATION info;
 700         result->type = call->type;
 701         result->virtual_query.status = NtQueryVirtualMemory( NtCurrentProcess(),
 702                                                              call->virtual_query.addr,
 703                                                              MemoryBasicInformation, &info,
 704                                                              sizeof(info), NULL );
 705         if (result->virtual_query.status == STATUS_SUCCESS)
 706         {
 707             result->virtual_query.base       = info.BaseAddress;
 708             result->virtual_query.alloc_base = info.AllocationBase;
 709             result->virtual_query.size       = info.RegionSize;
 710             result->virtual_query.state      = info.State;
 711             result->virtual_query.prot       = info.Protect;
 712             result->virtual_query.alloc_prot = info.AllocationProtect;
 713             result->virtual_query.alloc_type = info.Type;
 714         }
 715         break;
 716     }
 717     case APC_VIRTUAL_PROTECT:
 718         result->type = call->type;
 719         result->virtual_protect.addr = call->virtual_protect.addr;
 720         result->virtual_protect.size = call->virtual_protect.size;
 721         result->virtual_protect.status = NtProtectVirtualMemory( NtCurrentProcess(),
 722                                                                  &result->virtual_protect.addr,
 723                                                                  &result->virtual_protect.size,
 724                                                                  call->virtual_protect.prot,
 725                                                                  &result->virtual_protect.prot );
 726         break;
 727     case APC_VIRTUAL_FLUSH:
 728         result->type = call->type;
 729         result->virtual_flush.addr = call->virtual_flush.addr;
 730         result->virtual_flush.size = call->virtual_flush.size;
 731         result->virtual_flush.status = NtFlushVirtualMemory( NtCurrentProcess(),
 732                                                              &result->virtual_flush.addr,
 733                                                              &result->virtual_flush.size, 0 );
 734         break;
 735     case APC_VIRTUAL_LOCK:
 736         result->type = call->type;
 737         result->virtual_lock.addr = call->virtual_lock.addr;
 738         result->virtual_lock.size = call->virtual_lock.size;
 739         result->virtual_lock.status = NtLockVirtualMemory( NtCurrentProcess(),
 740                                                            &result->virtual_lock.addr,
 741                                                            &result->virtual_lock.size, 0 );
 742         break;
 743     case APC_VIRTUAL_UNLOCK:
 744         result->type = call->type;
 745         result->virtual_unlock.addr = call->virtual_unlock.addr;
 746         result->virtual_unlock.size = call->virtual_unlock.size;
 747         result->virtual_unlock.status = NtUnlockVirtualMemory( NtCurrentProcess(),
 748                                                                &result->virtual_unlock.addr,
 749                                                                &result->virtual_unlock.size, 0 );
 750         break;
 751     case APC_MAP_VIEW:
 752     {
 753         LARGE_INTEGER offset;
 754         result->type = call->type;
 755         result->map_view.addr   = call->map_view.addr;
 756         result->map_view.size   = call->map_view.size;
 757         offset.u.LowPart        = call->map_view.offset_low;
 758         offset.u.HighPart       = call->map_view.offset_high;
 759         result->map_view.status = NtMapViewOfSection( call->map_view.handle, NtCurrentProcess(),
 760                                                       &result->map_view.addr, call->map_view.zero_bits,
 761                                                       0, &offset, &result->map_view.size, ViewShare,
 762                                                       call->map_view.alloc_type, call->map_view.prot );
 763         NtClose( call->map_view.handle );
 764         break;
 765     }
 766     case APC_UNMAP_VIEW:
 767         result->type = call->type;
 768         result->unmap_view.status = NtUnmapViewOfSection( NtCurrentProcess(), call->unmap_view.addr );
 769         break;
 770     case APC_CREATE_THREAD:
 771     {
 772         CLIENT_ID id;
 773         result->type = call->type;
 774         result->create_thread.status = RtlCreateUserThread( NtCurrentProcess(), NULL,
 775                                                             call->create_thread.suspend, NULL,
 776                                                             call->create_thread.reserve,
 777                                                             call->create_thread.commit,
 778                                                             call->create_thread.func,
 779                                                             call->create_thread.arg,
 780                                                             &result->create_thread.handle, &id );
 781         result->create_thread.tid = HandleToULong(id.UniqueThread);
 782         break;
 783     }
 784     default:
 785         server_protocol_error( "get_apc_request: bad type %d\n", call->type );
 786         break;
 787     }
 788     return user_apc;
 789 }
 790
 791 /***********************************************************************
 792  *           NTDLL_queue_process_apc
 793  */
 794 NTSTATUS NTDLL_queue_process_apc( HANDLE process, const apc_call_t *call, apc_result_t *result )
 795 {
 796     for (;;)
 797     {
 798         NTSTATUS ret;
 799         HANDLE handle = 0;
 800         BOOL self = FALSE;
 801
 802         SERVER_START_REQ( queue_apc )
 803         {
 804             req->process = process;
 805             req->call = *call;
 806             if (!(ret = wine_server_call( req )))
 807             {
 808                 handle = reply->handle;
 809                 self = reply->self;
 810             }
 811         }
 812         SERVER_END_REQ;
 813         if (ret != STATUS_SUCCESS) return ret;
 814
 815         if (self)
 816         {
 817             invoke_apc( call, result );
 818         }
 819         else
 820         {
 821             NtWaitForSingleObject( handle, FALSE, NULL );
 822
 823             SERVER_START_REQ( get_apc_result )
 824             {
 825                 req->handle = handle;
 826                 if (!(ret = wine_server_call( req ))) *result = reply->result;
 827             }
 828             SERVER_END_REQ;
 829
 830             if (!ret && result->type == APC_NONE) continue;  /* APC didn't run, try again */
 831             if (ret) NtClose( handle );
 832         }
 833         return ret;
 834     }
 835 }
 836
 837
 838 /***********************************************************************
 839  *              NTDLL_wait_for_multiple_objects
 840  *
 841  * Implementation of NtWaitForMultipleObjects
 842  */
 843 NTSTATUS NTDLL_wait_for_multiple_objects( UINT count, const HANDLE *handles, UINT flags,
 844                                           const LARGE_INTEGER *timeout, HANDLE signal_object )
 845 {
 846     NTSTATUS ret;
 847     int cookie;
 848     BOOL user_apc = FALSE;
 849     obj_handle_t apc_handle = 0;
 850     apc_call_t call;
 851     apc_result_t result;
 852     timeout_t abs_timeout = timeout ? timeout->QuadPart : TIMEOUT_INFINITE;
 853
 854     memset( &result, 0, sizeof(result) );
 855
 856     for (;;)
 857     {
 858         SERVER_START_REQ( select )
 859         {
 860             req->flags    = flags;
 861             req->cookie   = &cookie;
 862             req->signal   = signal_object;
 863             req->prev_apc = apc_handle;
 864             req->timeout  = abs_timeout;
 865             wine_server_add_data( req, &result, sizeof(result) );
 866             wine_server_add_data( req, handles, count * sizeof(HANDLE) );
 867             ret = wine_server_call( req );
 868             abs_timeout = reply->timeout;
 869             apc_handle  = reply->apc_handle;
 870             call        = reply->call;
 871         }
 872         SERVER_END_REQ;
 873         if (ret == STATUS_PENDING) ret = wait_reply( &cookie );
 874         if (ret != STATUS_USER_APC) break;
 875         if (invoke_apc( &call, &result ))
 876         {
 877             /* if we ran a user apc we have to check once more if an object got signaled,
 878              * but we don't want to wait */
 879             abs_timeout = 0;
 880             user_apc = TRUE;
 881         }
 882         signal_object = 0;  /* don't signal it multiple times */
 883     }
 884
 885     if (ret == STATUS_TIMEOUT && user_apc) ret = STATUS_USER_APC;
 886
 887     /* A test on Windows 2000 shows that Windows always yields during
 888        a wait, but a wait that is hit by an event gets a priority
 889        boost as well.  This seems to model that behavior the closest.  */
 890     if (ret == STATUS_TIMEOUT) NtYieldExecution();
 891
 892     return ret;
 893 }
 894
 895
 896 /* wait operations */
 897
 898 /******************************************************************
 899  *              NtWaitForMultipleObjects (NTDLL.@)
 900  */
 901 NTSTATUS WINAPI NtWaitForMultipleObjects( DWORD count, const HANDLE *handles,
 902                                           BOOLEAN wait_all, BOOLEAN alertable,
 903                                           const LARGE_INTEGER *timeout )
 904 {
 905     UINT flags = SELECT_INTERRUPTIBLE;
 906
 907     if (!count || count > MAXIMUM_WAIT_OBJECTS) return STATUS_INVALID_PARAMETER_1;
 908
 909     if (wait_all) flags |= SELECT_ALL;
 910     if (alertable) flags |= SELECT_ALERTABLE;
 911     return NTDLL_wait_for_multiple_objects( count, handles, flags, timeout, 0 );
 912 }
 913
 914
 915 /******************************************************************
 916  *              NtWaitForSingleObject (NTDLL.@)
 917  */
 918 NTSTATUS WINAPI NtWaitForSingleObject(HANDLE handle, BOOLEAN alertable, const LARGE_INTEGER *timeout )
 919 {
 920     return NtWaitForMultipleObjects( 1, &handle, FALSE, alertable, timeout );
 921 }
 922
 923
 924 /******************************************************************
 925  *              NtSignalAndWaitForSingleObject (NTDLL.@)
 926  */
 927 NTSTATUS WINAPI NtSignalAndWaitForSingleObject( HANDLE hSignalObject, HANDLE hWaitObject,
 928                                                 BOOLEAN alertable, const LARGE_INTEGER *timeout )
 929 {
 930     UINT flags = SELECT_INTERRUPTIBLE;
 931
 932     if (!hSignalObject) return STATUS_INVALID_HANDLE;
 933     if (alertable) flags |= SELECT_ALERTABLE;
 934     return NTDLL_wait_for_multiple_objects( 1, &hWaitObject, flags, timeout, hSignalObject );
 935 }
 936
 937
 938 /******************************************************************
 939  *              NtYieldExecution (NTDLL.@)
 940  */
 941 NTSTATUS WINAPI NtYieldExecution(void)
 942 {
 943 #ifdef HAVE_SCHED_YIELD
 944     sched_yield();
 945     return STATUS_SUCCESS;
 946 #else
 947     return STATUS_NO_YIELD_PERFORMED;
 948 #endif
 949 }
 950
 951
 952 /******************************************************************
 953  *              NtDelayExecution (NTDLL.@)
 954  */
 955 NTSTATUS WINAPI NtDelayExecution( BOOLEAN alertable, const LARGE_INTEGER *timeout )
 956 {
 957     /* if alertable, we need to query the server */
 958     if (alertable)
 959         return NTDLL_wait_for_multiple_objects( 0, NULL, SELECT_INTERRUPTIBLE | SELECT_ALERTABLE,
 960                                                 timeout, 0 );
 961
 962     if (!timeout || timeout->QuadPart == TIMEOUT_INFINITE)  /* sleep forever */
 963     {
 964         for (;;) select( 0, NULL, NULL, NULL, NULL );
 965     }
 966     else
 967     {
 968         LARGE_INTEGER now;
 969         timeout_t when, diff;
 970
 971         if ((when = timeout->QuadPart) < 0)
 972         {
 973             NtQuerySystemTime( &now );
 974             when = now.QuadPart - when;
 975         }
 976
 977         /* Note that we yield after establishing the desired timeout */
 978         NtYieldExecution();
 979         if (!when) return STATUS_SUCCESS;
 980
 981         for (;;)
 982         {
 983             struct timeval tv;
 984             NtQuerySystemTime( &now );
 985             diff = (when - now.QuadPart + 9) / 10;
 986             if (diff <= 0) break;
 987             tv.tv_sec  = diff / 1000000;
 988             tv.tv_usec = diff % 1000000;
 989             if (select( 0, NULL, NULL, NULL, &tv ) != -1) break;
 990         }
 991     }
 992     return STATUS_SUCCESS;
 993 }
 994
 995 /******************************************************************
 996  *              NtCreateIoCompletion (NTDLL.@)
 997  *              ZwCreateIoCompletion (NTDLL.@)
 998  *
 999  * Creates I/O completion object.
1000  *
1001  * PARAMS
1002  *      CompletionPort            [O] created completion object handle will be placed there
1003  *      DesiredAccess             [I] desired access to a handle (combination of IO_COMPLETION_*)
1004  *      ObjectAttributes          [I] completion object attributes
1005  *      NumberOfConcurrentThreads [I] desired number of concurrent active worker threads
1006  *
1007  */
1008 NTSTATUS WINAPI NtCreateIoCompletion( PHANDLE CompletionPort, ACCESS_MASK DesiredAccess,
1009                                       POBJECT_ATTRIBUTES ObjectAttributes, ULONG NumberOfConcurrentThreads )
1010 {
1011     NTSTATUS status;
1012
1013     TRACE("(%p, %x, %p, %d)\n", CompletionPort, DesiredAccess,
1014           ObjectAttributes, NumberOfConcurrentThreads);
1015
1016     if (!CompletionPort)
1017         return STATUS_INVALID_PARAMETER;
1018
1019     SERVER_START_REQ( create_completion )
1020     {
1021         req->access     = DesiredAccess;
1022         req->attributes = ObjectAttributes ? ObjectAttributes->Attributes : 0;
1023         req->rootdir    = ObjectAttributes ? ObjectAttributes->RootDirectory : NULL;
1024         req->concurrent = NumberOfConcurrentThreads;
1025         if (ObjectAttributes && ObjectAttributes->ObjectName)
1026             wine_server_add_data( req, ObjectAttributes->ObjectName->Buffer,
1027                                        ObjectAttributes->ObjectName->Length );
1028         if (!(status = wine_server_call( req )))
1029             *CompletionPort = reply->handle;
1030     }
1031     SERVER_END_REQ;
1032     return status;
1033 }
1034
1035 /******************************************************************
1036  *              NtSetIoCompletion (NTDLL.@)
1037  *              ZwSetIoCompletion (NTDLL.@)
1038  *
1039  * Inserts completion message into queue
1040  *
1041  * PARAMS
1042  *      CompletionPort           [I] HANDLE to completion object
1043  *      CompletionKey            [I] completion key
1044  *      CompletionValue          [I] completion value (usually pointer to OVERLAPPED)
1045  *      Status                   [I] operation status
1046  *      NumberOfBytesTransferred [I] number of bytes transferred
1047  */
1048 NTSTATUS WINAPI NtSetIoCompletion( HANDLE CompletionPort, ULONG_PTR CompletionKey,
1049                                    ULONG_PTR CompletionValue, NTSTATUS Status,
1050                                    ULONG NumberOfBytesTransferred )
1051 {
1052     NTSTATUS status;
1053
1054     TRACE("(%p, %lx, %lx, %x, %d)\n", CompletionPort, CompletionKey,
1055           CompletionValue, Status, NumberOfBytesTransferred);
1056
1057     SERVER_START_REQ( add_completion )
1058     {
1059         req->handle      = CompletionPort;
1060         req->ckey        = CompletionKey;
1061         req->cvalue      = CompletionValue;
1062         req->status      = Status;
1063         req->information = NumberOfBytesTransferred;
1064         status = wine_server_call( req );
1065     }
1066     SERVER_END_REQ;
1067     return status;
1068 }
1069
1070 /******************************************************************
1071  *              NtRemoveIoCompletion (NTDLL.@)
1072  *              ZwRemoveIoCompletion (NTDLL.@)
1073  *
1074  * (Wait for and) retrieve first completion message from completion object's queue
1075  *
1076  * PARAMS
1077  *      CompletionPort  [I] HANDLE to I/O completion object
1078  *      CompletionKey   [O] completion key
1079  *      CompletionValue [O] Completion value given in NtSetIoCompletion or in async operation
1080  *      iosb            [O] IO_STATUS_BLOCK of completed asynchronous operation
1081  *      WaitTime        [I] optional wait time in NTDLL format
1082  *
1083  */
1084 NTSTATUS WINAPI NtRemoveIoCompletion( HANDLE CompletionPort, PULONG_PTR CompletionKey,
1085                                       PULONG_PTR CompletionValue, PIO_STATUS_BLOCK iosb,
1086                                       PLARGE_INTEGER WaitTime )
1087 {
1088     NTSTATUS status;
1089
1090     TRACE("(%p, %p, %p, %p, %p)\n", CompletionPort, CompletionKey,
1091           CompletionValue, iosb, WaitTime);
1092
1093     for(;;)
1094     {
1095         SERVER_START_REQ( remove_completion )
1096         {
1097             req->handle = CompletionPort;
1098             if (!(status = wine_server_call( req )))
1099             {
1100                 *CompletionKey    = reply->ckey;
1101                 *CompletionValue  = reply->cvalue;
1102                 iosb->Information = reply->information;
1103                 iosb->u.Status    = reply->status;
1104             }
1105         }
1106         SERVER_END_REQ;
1107         if (status != STATUS_PENDING) break;
1108
1109         status = NtWaitForSingleObject( CompletionPort, FALSE, WaitTime );
1110         if (status != WAIT_OBJECT_0) break;
1111     }
1112     return status;
1113 }
1114
1115 /******************************************************************
1116  *              NtOpenIoCompletion (NTDLL.@)
1117  *              ZwOpenIoCompletion (NTDLL.@)
1118  *
1119  * Opens I/O completion object
1120  *
1121  * PARAMS
1122  *      CompletionPort     [O] completion object handle will be placed there
1123  *      DesiredAccess      [I] desired access to a handle (combination of IO_COMPLETION_*)
1124  *      ObjectAttributes   [I] completion object name
1125  *
1126  */
1127 NTSTATUS WINAPI NtOpenIoCompletion( PHANDLE CompletionPort, ACCESS_MASK DesiredAccess,
1128                                     POBJECT_ATTRIBUTES ObjectAttributes )
1129 {
1130     NTSTATUS status;
1131
1132     TRACE("(%p, 0x%x, %p)\n", CompletionPort, DesiredAccess, ObjectAttributes);
1133
1134     if (!CompletionPort || !ObjectAttributes || !ObjectAttributes->ObjectName)
1135         return STATUS_INVALID_PARAMETER;
1136
1137     SERVER_START_REQ( open_completion )
1138     {
1139         req->access     = DesiredAccess;
1140         req->rootdir    = ObjectAttributes->RootDirectory;
1141         wine_server_add_data( req, ObjectAttributes->ObjectName->Buffer,
1142                                    ObjectAttributes->ObjectName->Length );
1143         if (!(status = wine_server_call( req )))
1144             *CompletionPort = reply->handle;
1145     }
1146     SERVER_END_REQ;
1147     return status;
1148 }
1149
1150 /******************************************************************
1151  *              NtQueryIoCompletion (NTDLL.@)
1152  *              ZwQueryIoCompletion (NTDLL.@)
1153  *
1154  * Requests information about given I/O completion object
1155  *
1156  * PARAMS
1157  *      CompletionPort        [I] HANDLE to completion port to request
1158  *      InformationClass      [I] information class
1159  *      CompletionInformation [O] user-provided buffer for data
1160  *      BufferLength          [I] buffer length
1161  *      RequiredLength        [O] required buffer length
1162  *
1163  */
1164 NTSTATUS WINAPI NtQueryIoCompletion( HANDLE CompletionPort, IO_COMPLETION_INFORMATION_CLASS InformationClass,
1165                                      PVOID CompletionInformation, ULONG BufferLength, PULONG RequiredLength )
1166 {
1167     NTSTATUS status;
1168
1169     TRACE("(%p, %d, %p, 0x%x, %p)\n", CompletionPort, InformationClass, CompletionInformation,
1170           BufferLength, RequiredLength);
1171
1172     if (!CompletionInformation) return STATUS_INVALID_PARAMETER;
1173     switch( InformationClass )
1174     {
1175         case IoCompletionBasicInformation:
1176             {
1177                 ULONG *info = (ULONG *)CompletionInformation;
1178
1179                 if (RequiredLength) *RequiredLength = sizeof(*info);
1180                 if (BufferLength != sizeof(*info))
1181                     status = STATUS_INFO_LENGTH_MISMATCH;
1182                 else
1183                 {
1184                     SERVER_START_REQ( query_completion )
1185                     {
1186                         req->handle = CompletionPort;
1187                         if (!(status = wine_server_call( req )))
1188                             *info = reply->depth;
1189                     }
1190                     SERVER_END_REQ;
1191                 }
1192             }
1193             break;
1194         default:
1195             status = STATUS_INVALID_PARAMETER;
1196             break;
1197     }
1198     return status;
1199 }