dlls/rtworkq/queue.c

   1 /*
   2  * Copyright 2019-2020 Nikolay Sivov for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <assert.h>
  20
  21 #define COBJMACROS
  22 #define NONAMELESSUNION
  23
  24 #include "initguid.h"
  25 #include "rtworkq.h"
  26 #include "wine/debug.h"
  27 #include "wine/heap.h"
  28 #include "wine/list.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(mfplat);
  31
  32 #define FIRST_USER_QUEUE_HANDLE 5
  33 #define MAX_USER_QUEUE_HANDLES 124
  34
  35 #define WAIT_ITEM_KEY_MASK      (0x82000000)
  36 #define SCHEDULED_ITEM_KEY_MASK (0x80000000)
  37
  38 static LONG next_item_key;
  39
  40 static RTWQWORKITEM_KEY get_item_key(DWORD mask, DWORD key)
  41 {
  42     return ((RTWQWORKITEM_KEY)mask << 32) | key;
  43 }
  44
  45 static RTWQWORKITEM_KEY generate_item_key(DWORD mask)
  46 {
  47     return get_item_key(mask, InterlockedIncrement(&next_item_key));
  48 }
  49
  50 struct queue_handle
  51 {
  52     void *obj;
  53     LONG refcount;
  54     WORD generation;
  55 };
  56
  57 static struct queue_handle user_queues[MAX_USER_QUEUE_HANDLES];
  58 static struct queue_handle *next_free_user_queue;
  59 static struct queue_handle *next_unused_user_queue = user_queues;
  60 static WORD queue_generation;
  61
  62 static CRITICAL_SECTION queues_section;
  63 static CRITICAL_SECTION_DEBUG queues_critsect_debug =
  64 {
  65     0, 0, &queues_section,
  66     { &queues_critsect_debug.ProcessLocksList, &queues_critsect_debug.ProcessLocksList },
  67       0, 0, { (DWORD_PTR)(__FILE__ ": queues_section") }
  68 };
  69 static CRITICAL_SECTION queues_section = { &queues_critsect_debug, -1, 0, 0, 0, 0 };
  70
  71 static LONG platform_lock;
  72 static CO_MTA_USAGE_COOKIE mta_cookie;
  73
  74 static struct queue_handle *get_queue_obj(DWORD handle)
  75 {
  76     unsigned int idx = HIWORD(handle) - FIRST_USER_QUEUE_HANDLE;
  77
  78     if (idx < MAX_USER_QUEUE_HANDLES && user_queues[idx].refcount)
  79     {
  80         if (LOWORD(handle) == user_queues[idx].generation)
  81             return &user_queues[idx];
  82     }
  83
  84     return NULL;
  85 }
  86
  87 /* Should be kept in sync with corresponding MFASYNC_CALLBACK_ constants. */
  88 enum rtwq_callback_queue_id
  89 {
  90     RTWQ_CALLBACK_QUEUE_UNDEFINED     = 0x00000000,
  91     RTWQ_CALLBACK_QUEUE_STANDARD      = 0x00000001,
  92     RTWQ_CALLBACK_QUEUE_RT            = 0x00000002,
  93     RTWQ_CALLBACK_QUEUE_IO            = 0x00000003,
  94     RTWQ_CALLBACK_QUEUE_TIMER         = 0x00000004,
  95     RTWQ_CALLBACK_QUEUE_MULTITHREADED = 0x00000005,
  96     RTWQ_CALLBACK_QUEUE_LONG_FUNCTION = 0x00000007,
  97     RTWQ_CALLBACK_QUEUE_PRIVATE_MASK  = 0xffff0000,
  98     RTWQ_CALLBACK_QUEUE_ALL           = 0xffffffff,
  99 };
 100
 101 /* Should be kept in sync with corresponding MFASYNC_ constants. */
 102 enum rtwq_callback_flags
 103 {
 104     RTWQ_FAST_IO_PROCESSING_CALLBACK = 0x00000001,
 105     RTWQ_SIGNAL_CALLBACK             = 0x00000002,
 106     RTWQ_BLOCKING_CALLBACK           = 0x00000004,
 107     RTWQ_REPLY_CALLBACK              = 0x00000008,
 108     RTWQ_LOCALIZE_REMOTE_CALLBACK    = 0x00000010,
 109 };
 110
 111 enum system_queue_index
 112 {
 113     SYS_QUEUE_STANDARD = 0,
 114     SYS_QUEUE_RT,
 115     SYS_QUEUE_IO,
 116     SYS_QUEUE_TIMER,
 117     SYS_QUEUE_MULTITHREADED,
 118     SYS_QUEUE_DO_NOT_USE,
 119     SYS_QUEUE_LONG_FUNCTION,
 120     SYS_QUEUE_COUNT,
 121 };
 122
 123 struct work_item
 124 {
 125     IUnknown IUnknown_iface;
 126     LONG refcount;
 127     struct list entry;
 128     IRtwqAsyncResult *result;
 129     IRtwqAsyncResult *reply_result;
 130     struct queue *queue;
 131     RTWQWORKITEM_KEY key;
 132     LONG priority;
 133     DWORD flags;
 134     PTP_SIMPLE_CALLBACK finalization_callback;
 135     union
 136     {
 137         TP_WAIT *wait_object;
 138         TP_TIMER *timer_object;
 139     } u;
 140 };
 141
 142 static struct work_item *work_item_impl_from_IUnknown(IUnknown *iface)
 143 {
 144     return CONTAINING_RECORD(iface, struct work_item, IUnknown_iface);
 145 }
 146
 147 static const TP_CALLBACK_PRIORITY priorities[] =
 148 {
 149     TP_CALLBACK_PRIORITY_HIGH,
 150     TP_CALLBACK_PRIORITY_NORMAL,
 151     TP_CALLBACK_PRIORITY_LOW,
 152 };
 153
 154 struct queue;
 155 struct queue_desc;
 156
 157 struct queue_ops
 158 {
 159     HRESULT (*init)(const struct queue_desc *desc, struct queue *queue);
 160     BOOL (*shutdown)(struct queue *queue);
 161     void (*submit)(struct queue *queue, struct work_item *item);
 162 };
 163
 164 struct queue_desc
 165 {
 166     RTWQ_WORKQUEUE_TYPE queue_type;
 167     const struct queue_ops *ops;
 168     DWORD target_queue;
 169 };
 170
 171 struct queue
 172 {
 173     IRtwqAsyncCallback IRtwqAsyncCallback_iface;
 174     const struct queue_ops *ops;
 175     TP_POOL *pool;
 176     TP_CALLBACK_ENVIRON_V3 envs[ARRAY_SIZE(priorities)];
 177     CRITICAL_SECTION cs;
 178     struct list pending_items;
 179     DWORD id;
 180     /* Data used for serial queues only. */
 181     PTP_SIMPLE_CALLBACK finalization_callback;
 182     DWORD target_queue;
 183 };
 184
 185 static void shutdown_queue(struct queue *queue);
 186
 187 static HRESULT lock_user_queue(DWORD queue)
 188 {
 189     HRESULT hr = RTWQ_E_INVALID_WORKQUEUE;
 190     struct queue_handle *entry;
 191
 192     if (!(queue & RTWQ_CALLBACK_QUEUE_PRIVATE_MASK))
 193         return S_OK;
 194
 195     EnterCriticalSection(&queues_section);
 196     entry = get_queue_obj(queue);
 197     if (entry && entry->refcount)
 198     {
 199         entry->refcount++;
 200         hr = S_OK;
 201     }
 202     LeaveCriticalSection(&queues_section);
 203     return hr;
 204 }
 205
 206 static HRESULT unlock_user_queue(DWORD queue)
 207 {
 208     HRESULT hr = RTWQ_E_INVALID_WORKQUEUE;
 209     struct queue_handle *entry;
 210
 211     if (!(queue & RTWQ_CALLBACK_QUEUE_PRIVATE_MASK))
 212         return S_OK;
 213
 214     EnterCriticalSection(&queues_section);
 215     entry = get_queue_obj(queue);
 216     if (entry && entry->refcount)
 217     {
 218         if (--entry->refcount == 0)
 219         {
 220             shutdown_queue((struct queue *)entry->obj);
 221             heap_free(entry->obj);
 222             entry->obj = next_free_user_queue;
 223             next_free_user_queue = entry;
 224         }
 225         hr = S_OK;
 226     }
 227     LeaveCriticalSection(&queues_section);
 228     return hr;
 229 }
 230
 231 static struct queue *queue_impl_from_IRtwqAsyncCallback(IRtwqAsyncCallback *iface)
 232 {
 233     return CONTAINING_RECORD(iface, struct queue, IRtwqAsyncCallback_iface);
 234 }
 235
 236 static HRESULT WINAPI queue_serial_callback_QueryInterface(IRtwqAsyncCallback *iface, REFIID riid, void **obj)
 237 {
 238     if (IsEqualIID(riid, &IID_IRtwqAsyncCallback) ||
 239             IsEqualIID(riid, &IID_IUnknown))
 240     {
 241         *obj = iface;
 242         IRtwqAsyncCallback_AddRef(iface);
 243         return S_OK;
 244     }
 245
 246     *obj = NULL;
 247     return E_NOINTERFACE;
 248 }
 249
 250 static ULONG WINAPI queue_serial_callback_AddRef(IRtwqAsyncCallback *iface)
 251 {
 252     return 2;
 253 }
 254
 255 static ULONG WINAPI queue_serial_callback_Release(IRtwqAsyncCallback *iface)
 256 {
 257     return 1;
 258 }
 259
 260 static HRESULT WINAPI queue_serial_callback_GetParameters(IRtwqAsyncCallback *iface, DWORD *flags, DWORD *queue_id)
 261 {
 262     struct queue *queue = queue_impl_from_IRtwqAsyncCallback(iface);
 263
 264     *flags = 0;
 265     *queue_id = queue->id;
 266
 267     return S_OK;
 268 }
 269
 270 static HRESULT WINAPI queue_serial_callback_Invoke(IRtwqAsyncCallback *iface, IRtwqAsyncResult *result)
 271 {
 272     /* Reply callback won't be called in a regular way, pending items and chained queues will make it
 273        unnecessary complicated to reach actual work queue that's able to execute this item. Instead
 274        serial queues are cleaned up right away on submit(). */
 275     return S_OK;
 276 }
 277
 278 static const IRtwqAsyncCallbackVtbl queue_serial_callback_vtbl =
 279 {
 280     queue_serial_callback_QueryInterface,
 281     queue_serial_callback_AddRef,
 282     queue_serial_callback_Release,
 283     queue_serial_callback_GetParameters,
 284     queue_serial_callback_Invoke,
 285 };
 286
 287 static struct queue system_queues[SYS_QUEUE_COUNT];
 288
 289 static struct queue *get_system_queue(DWORD queue_id)
 290 {
 291     switch (queue_id)
 292     {
 293         case RTWQ_CALLBACK_QUEUE_STANDARD:
 294         case RTWQ_CALLBACK_QUEUE_RT:
 295         case RTWQ_CALLBACK_QUEUE_IO:
 296         case RTWQ_CALLBACK_QUEUE_TIMER:
 297         case RTWQ_CALLBACK_QUEUE_MULTITHREADED:
 298         case RTWQ_CALLBACK_QUEUE_LONG_FUNCTION:
 299             return &system_queues[queue_id - 1];
 300         default:
 301             return NULL;
 302     }
 303 }
 304
 305 static HRESULT grab_queue(DWORD queue_id, struct queue **ret);
 306
 307 static void CALLBACK standard_queue_cleanup_callback(void *object_data, void *group_data)
 308 {
 309 }
 310
 311 static HRESULT pool_queue_init(const struct queue_desc *desc, struct queue *queue)
 312 {
 313     TP_CALLBACK_ENVIRON_V3 env;
 314     unsigned int max_thread, i;
 315
 316     queue->pool = CreateThreadpool(NULL);
 317
 318     memset(&env, 0, sizeof(env));
 319     env.Version = 3;
 320     env.Size = sizeof(env);
 321     env.Pool = queue->pool;
 322     env.CleanupGroup = CreateThreadpoolCleanupGroup();
 323     env.CleanupGroupCancelCallback = standard_queue_cleanup_callback;
 324     env.CallbackPriority = TP_CALLBACK_PRIORITY_NORMAL;
 325     for (i = 0; i < ARRAY_SIZE(queue->envs); ++i)
 326     {
 327         queue->envs[i] = env;
 328         queue->envs[i].CallbackPriority = priorities[i];
 329     }
 330     list_init(&queue->pending_items);
 331     InitializeCriticalSection(&queue->cs);
 332
 333     max_thread = (desc->queue_type == RTWQ_STANDARD_WORKQUEUE || desc->queue_type == RTWQ_WINDOW_WORKQUEUE) ? 1 : 4;
 334
 335     SetThreadpoolThreadMinimum(queue->pool, 1);
 336     SetThreadpoolThreadMaximum(queue->pool, max_thread);
 337
 338     if (desc->queue_type == RTWQ_WINDOW_WORKQUEUE)
 339         FIXME("RTWQ_WINDOW_WORKQUEUE is not supported.\n");
 340
 341     return S_OK;
 342 }
 343
 344 static BOOL pool_queue_shutdown(struct queue *queue)
 345 {
 346     if (!queue->pool)
 347         return FALSE;
 348
 349     CloseThreadpoolCleanupGroupMembers(queue->envs[0].CleanupGroup, TRUE, NULL);
 350     CloseThreadpool(queue->pool);
 351     queue->pool = NULL;
 352
 353     return TRUE;
 354 }
 355
 356 static void CALLBACK standard_queue_worker(TP_CALLBACK_INSTANCE *instance, void *context, TP_WORK *work)
 357 {
 358     struct work_item *item = context;
 359     RTWQASYNCRESULT *result = (RTWQASYNCRESULT *)item->result;
 360
 361     TRACE("result object %p.\n", result);
 362
 363     /* Submitting from serial queue in reply mode, use different result object acting as receipt token.
 364        It's submitted to user callback still, but when invoked, special serial queue callback will be used
 365        to ensure correct destination queue. */
 366
 367     IRtwqAsyncCallback_Invoke(result->pCallback, item->reply_result ? item->reply_result : item->result);
 368
 369     IUnknown_Release(&item->IUnknown_iface);
 370 }
 371
 372 static void pool_queue_submit(struct queue *queue, struct work_item *item)
 373 {
 374     TP_CALLBACK_PRIORITY callback_priority;
 375     TP_CALLBACK_ENVIRON_V3 env;
 376     TP_WORK *work_object;
 377
 378     if (item->priority == 0)
 379         callback_priority = TP_CALLBACK_PRIORITY_NORMAL;
 380     else if (item->priority < 0)
 381         callback_priority = TP_CALLBACK_PRIORITY_LOW;
 382     else
 383         callback_priority = TP_CALLBACK_PRIORITY_HIGH;
 384
 385     env = queue->envs[callback_priority];
 386     env.FinalizationCallback = item->finalization_callback;
 387     /* Worker pool callback will release one reference. Grab one more to keep object alive when
 388        we need finalization callback. */
 389     if (item->finalization_callback)
 390         IUnknown_AddRef(&item->IUnknown_iface);
 391     work_object = CreateThreadpoolWork(standard_queue_worker, item, (TP_CALLBACK_ENVIRON *)&env);
 392     SubmitThreadpoolWork(work_object);
 393
 394     TRACE("dispatched %p.\n", item->result);
 395 }
 396
 397 static const struct queue_ops pool_queue_ops =
 398 {
 399     pool_queue_init,
 400     pool_queue_shutdown,
 401     pool_queue_submit,
 402 };
 403
 404 static struct work_item * serial_queue_get_next(struct queue *queue, struct work_item *item)
 405 {
 406     struct work_item *next_item = NULL;
 407
 408     list_remove(&item->entry);
 409     if (!list_empty(&item->queue->pending_items))
 410         next_item = LIST_ENTRY(list_head(&item->queue->pending_items), struct work_item, entry);
 411
 412     return next_item;
 413 }
 414
 415 static void CALLBACK serial_queue_finalization_callback(PTP_CALLBACK_INSTANCE instance, void *user_data)
 416 {
 417     struct work_item *item = (struct work_item *)user_data, *next_item;
 418     struct queue *target_queue, *queue = item->queue;
 419     HRESULT hr;
 420
 421     EnterCriticalSection(&queue->cs);
 422
 423     if ((next_item = serial_queue_get_next(queue, item)))
 424     {
 425         if (SUCCEEDED(hr = grab_queue(queue->target_queue, &target_queue)))
 426             target_queue->ops->submit(target_queue, next_item);
 427         else
 428             WARN("Failed to grab queue for id %#x, hr %#x.\n", queue->target_queue, hr);
 429     }
 430
 431     LeaveCriticalSection(&queue->cs);
 432
 433     IUnknown_Release(&item->IUnknown_iface);
 434 }
 435
 436 static HRESULT serial_queue_init(const struct queue_desc *desc, struct queue *queue)
 437 {
 438     queue->IRtwqAsyncCallback_iface.lpVtbl = &queue_serial_callback_vtbl;
 439     queue->target_queue = desc->target_queue;
 440     lock_user_queue(queue->target_queue);
 441     queue->finalization_callback = serial_queue_finalization_callback;
 442
 443     return S_OK;
 444 }
 445
 446 static BOOL serial_queue_shutdown(struct queue *queue)
 447 {
 448     unlock_user_queue(queue->target_queue);
 449
 450     return TRUE;
 451 }
 452
 453 static struct work_item * serial_queue_is_ack_token(struct queue *queue, struct work_item *item)
 454 {
 455     RTWQASYNCRESULT *async_result = (RTWQASYNCRESULT *)item->result;
 456     struct work_item *head;
 457
 458     if (list_empty(&queue->pending_items))
 459         return NULL;
 460
 461     head = LIST_ENTRY(list_head(&queue->pending_items), struct work_item, entry);
 462     if (head->reply_result == item->result && async_result->pCallback == &queue->IRtwqAsyncCallback_iface)
 463         return head;
 464
 465     return NULL;
 466 }
 467
 468 static void serial_queue_submit(struct queue *queue, struct work_item *item)
 469 {
 470     struct work_item *head, *next_item = NULL;
 471     struct queue *target_queue;
 472     HRESULT hr;
 473
 474     /* In reply mode queue will advance when 'reply_result' is invoked, in regular mode it will advance automatically,
 475        via finalization callback. */
 476
 477     if (item->flags & RTWQ_REPLY_CALLBACK)
 478     {
 479         if (FAILED(hr = RtwqCreateAsyncResult(NULL, &queue->IRtwqAsyncCallback_iface, NULL, &item->reply_result)))
 480             WARN("Failed to create reply object, hr %#x.\n", hr);
 481     }
 482     else
 483         item->finalization_callback = queue->finalization_callback;
 484
 485     /* Serial queues could be chained together, detach from current queue before transitioning item to this one.
 486        Items are not detached when submitted to pool queues, because pool queues won't forward them further. */
 487     EnterCriticalSection(&item->queue->cs);
 488     list_remove(&item->entry);
 489     LeaveCriticalSection(&item->queue->cs);
 490
 491     EnterCriticalSection(&queue->cs);
 492
 493     item->queue = queue;
 494
 495     if ((head = serial_queue_is_ack_token(queue, item)))
 496     {
 497         /* Ack receipt token - pop waiting item, advance. */
 498         next_item = serial_queue_get_next(queue, head);
 499         IUnknown_Release(&head->IUnknown_iface);
 500     }
 501     else
 502     {
 503         if (list_empty(&queue->pending_items))
 504             next_item = item;
 505         list_add_tail(&queue->pending_items, &item->entry);
 506         IUnknown_AddRef(&item->IUnknown_iface);
 507     }
 508
 509     if (next_item)
 510     {
 511         if (SUCCEEDED(hr = grab_queue(queue->target_queue, &target_queue)))
 512             target_queue->ops->submit(target_queue, next_item);
 513         else
 514             WARN("Failed to grab queue for id %#x, hr %#x.\n", queue->target_queue, hr);
 515     }
 516
 517     LeaveCriticalSection(&queue->cs);
 518 }
 519
 520 static const struct queue_ops serial_queue_ops =
 521 {
 522     serial_queue_init,
 523     serial_queue_shutdown,
 524     serial_queue_submit,
 525 };
 526
 527 static HRESULT WINAPI work_item_QueryInterface(IUnknown *iface, REFIID riid, void **obj)
 528 {
 529     if (IsEqualIID(riid, &IID_IUnknown))
 530     {
 531         *obj = iface;
 532         IUnknown_AddRef(iface);
 533         return S_OK;
 534     }
 535
 536     *obj = NULL;
 537     return E_NOINTERFACE;
 538 }
 539
 540 static ULONG WINAPI work_item_AddRef(IUnknown *iface)
 541 {
 542     struct work_item *item = work_item_impl_from_IUnknown(iface);
 543     return InterlockedIncrement(&item->refcount);
 544 }
 545
 546 static ULONG WINAPI work_item_Release(IUnknown *iface)
 547 {
 548     struct work_item *item = work_item_impl_from_IUnknown(iface);
 549     ULONG refcount = InterlockedDecrement(&item->refcount);
 550
 551     if (!refcount)
 552     {
 553         if (item->reply_result)
 554             IRtwqAsyncResult_Release(item->reply_result);
 555         IRtwqAsyncResult_Release(item->result);
 556         heap_free(item);
 557     }
 558
 559     return refcount;
 560 }
 561
 562 static const IUnknownVtbl work_item_vtbl =
 563 {
 564     work_item_QueryInterface,
 565     work_item_AddRef,
 566     work_item_Release,
 567 };
 568
 569 static struct work_item * alloc_work_item(struct queue *queue, LONG priority, IRtwqAsyncResult *result)
 570 {
 571     RTWQASYNCRESULT *async_result = (RTWQASYNCRESULT *)result;
 572     DWORD flags = 0, queue_id = 0;
 573     struct work_item *item;
 574
 575     item = heap_alloc_zero(sizeof(*item));
 576
 577     item->IUnknown_iface.lpVtbl = &work_item_vtbl;
 578     item->result = result;
 579     IRtwqAsyncResult_AddRef(item->result);
 580     item->refcount = 1;
 581     item->queue = queue;
 582     list_init(&item->entry);
 583     item->priority = priority;
 584
 585     if (SUCCEEDED(IRtwqAsyncCallback_GetParameters(async_result->pCallback, &flags, &queue_id)))
 586         item->flags = flags;
 587
 588     return item;
 589 }
 590
 591 static void init_work_queue(const struct queue_desc *desc, struct queue *queue)
 592 {
 593     assert(desc->ops != NULL);
 594
 595     queue->ops = desc->ops;
 596     if (SUCCEEDED(queue->ops->init(desc, queue)))
 597     {
 598         list_init(&queue->pending_items);
 599         InitializeCriticalSection(&queue->cs);
 600     }
 601 }
 602
 603 static HRESULT grab_queue(DWORD queue_id, struct queue **ret)
 604 {
 605     struct queue *queue = get_system_queue(queue_id);
 606     RTWQ_WORKQUEUE_TYPE queue_type;
 607     struct queue_handle *entry;
 608
 609     *ret = NULL;
 610
 611     if (!system_queues[SYS_QUEUE_STANDARD].pool)
 612         return RTWQ_E_SHUTDOWN;
 613
 614     if (queue && queue->pool)
 615     {
 616         *ret = queue;
 617         return S_OK;
 618     }
 619     else if (queue)
 620     {
 621         struct queue_desc desc;
 622
 623         EnterCriticalSection(&queues_section);
 624         switch (queue_id)
 625         {
 626             case RTWQ_CALLBACK_QUEUE_IO:
 627             case RTWQ_CALLBACK_QUEUE_MULTITHREADED:
 628             case RTWQ_CALLBACK_QUEUE_LONG_FUNCTION:
 629                 queue_type = RTWQ_MULTITHREADED_WORKQUEUE;
 630                 break;
 631             default:
 632                 queue_type = RTWQ_STANDARD_WORKQUEUE;
 633         }
 634
 635         desc.queue_type = queue_type;
 636         desc.ops = &pool_queue_ops;
 637         desc.target_queue = 0;
 638         init_work_queue(&desc, queue);
 639         LeaveCriticalSection(&queues_section);
 640         *ret = queue;
 641         return S_OK;
 642     }
 643
 644     /* Handles user queues. */
 645     if ((entry = get_queue_obj(queue_id)))
 646         *ret = entry->obj;
 647
 648     return *ret ? S_OK : RTWQ_E_INVALID_WORKQUEUE;
 649 }
 650
 651 static void shutdown_queue(struct queue *queue)
 652 {
 653     struct work_item *item, *item2;
 654
 655     if (!queue->ops || !queue->ops->shutdown(queue))
 656         return;
 657
 658     EnterCriticalSection(&queue->cs);
 659     LIST_FOR_EACH_ENTRY_SAFE(item, item2, &queue->pending_items, struct work_item, entry)
 660     {
 661         list_remove(&item->entry);
 662         IUnknown_Release(&item->IUnknown_iface);
 663     }
 664     LeaveCriticalSection(&queue->cs);
 665
 666     DeleteCriticalSection(&queue->cs);
 667
 668     memset(queue, 0, sizeof(*queue));
 669 }
 670
 671 static HRESULT queue_submit_item(struct queue *queue, LONG priority, IRtwqAsyncResult *result)
 672 {
 673     struct work_item *item;
 674
 675     if (!(item = alloc_work_item(queue, priority, result)))
 676         return E_OUTOFMEMORY;
 677
 678     queue->ops->submit(queue, item);
 679
 680     return S_OK;
 681 }
 682
 683 static HRESULT queue_put_work_item(DWORD queue_id, LONG priority, IRtwqAsyncResult *result)
 684 {
 685     struct queue *queue;
 686     HRESULT hr;
 687
 688     if (FAILED(hr = grab_queue(queue_id, &queue)))
 689         return hr;
 690
 691     return queue_submit_item(queue, priority, result);
 692 }
 693
 694 static HRESULT invoke_async_callback(IRtwqAsyncResult *result)
 695 {
 696     RTWQASYNCRESULT *result_data = (RTWQASYNCRESULT *)result;
 697     DWORD queue = RTWQ_CALLBACK_QUEUE_STANDARD, flags;
 698     HRESULT hr;
 699
 700     if (FAILED(IRtwqAsyncCallback_GetParameters(result_data->pCallback, &flags, &queue)))
 701         queue = RTWQ_CALLBACK_QUEUE_STANDARD;
 702
 703     if (FAILED(lock_user_queue(queue)))
 704         queue = RTWQ_CALLBACK_QUEUE_STANDARD;
 705
 706     hr = queue_put_work_item(queue, 0, result);
 707
 708     unlock_user_queue(queue);
 709
 710     return hr;
 711 }
 712
 713 static void queue_release_pending_item(struct work_item *item)
 714 {
 715     EnterCriticalSection(&item->queue->cs);
 716     if (item->key)
 717     {
 718         list_remove(&item->entry);
 719         item->key = 0;
 720         IUnknown_Release(&item->IUnknown_iface);
 721     }
 722     LeaveCriticalSection(&item->queue->cs);
 723 }
 724
 725 static void CALLBACK waiting_item_callback(TP_CALLBACK_INSTANCE *instance, void *context, TP_WAIT *wait,
 726         TP_WAIT_RESULT wait_result)
 727 {
 728     struct work_item *item = context;
 729
 730     TRACE("result object %p.\n", item->result);
 731
 732     invoke_async_callback(item->result);
 733
 734     IUnknown_Release(&item->IUnknown_iface);
 735 }
 736
 737 static void CALLBACK waiting_item_cancelable_callback(TP_CALLBACK_INSTANCE *instance, void *context, TP_WAIT *wait,
 738         TP_WAIT_RESULT wait_result)
 739 {
 740     struct work_item *item = context;
 741
 742     TRACE("result object %p.\n", item->result);
 743
 744     queue_release_pending_item(item);
 745
 746     invoke_async_callback(item->result);
 747
 748     IUnknown_Release(&item->IUnknown_iface);
 749 }
 750
 751 static void CALLBACK scheduled_item_callback(TP_CALLBACK_INSTANCE *instance, void *context, TP_TIMER *timer)
 752 {
 753     struct work_item *item = context;
 754
 755     TRACE("result object %p.\n", item->result);
 756
 757     invoke_async_callback(item->result);
 758
 759     IUnknown_Release(&item->IUnknown_iface);
 760 }
 761
 762 static void CALLBACK scheduled_item_cancelable_callback(TP_CALLBACK_INSTANCE *instance, void *context, TP_TIMER *timer)
 763 {
 764     struct work_item *item = context;
 765
 766     TRACE("result object %p.\n", item->result);
 767
 768     queue_release_pending_item(item);
 769
 770     invoke_async_callback(item->result);
 771
 772     IUnknown_Release(&item->IUnknown_iface);
 773 }
 774
 775 static void CALLBACK periodic_item_callback(TP_CALLBACK_INSTANCE *instance, void *context, TP_TIMER *timer)
 776 {
 777     struct work_item *item = context;
 778
 779     IUnknown_AddRef(&item->IUnknown_iface);
 780
 781     invoke_async_callback(item->result);
 782
 783     IUnknown_Release(&item->IUnknown_iface);
 784 }
 785
 786 static void queue_mark_item_pending(DWORD mask, struct work_item *item, RTWQWORKITEM_KEY *key)
 787 {
 788     *key = generate_item_key(mask);
 789     item->key = *key;
 790
 791     EnterCriticalSection(&item->queue->cs);
 792     list_add_tail(&item->queue->pending_items, &item->entry);
 793     IUnknown_AddRef(&item->IUnknown_iface);
 794     LeaveCriticalSection(&item->queue->cs);
 795 }
 796
 797 static HRESULT queue_submit_wait(struct queue *queue, HANDLE event, LONG priority, IRtwqAsyncResult *result,
 798         RTWQWORKITEM_KEY *key)
 799 {
 800     PTP_WAIT_CALLBACK callback;
 801     struct work_item *item;
 802
 803     if (!(item = alloc_work_item(queue, priority, result)))
 804         return E_OUTOFMEMORY;
 805
 806     if (key)
 807     {
 808         queue_mark_item_pending(WAIT_ITEM_KEY_MASK, item, key);
 809         callback = waiting_item_cancelable_callback;
 810     }
 811     else
 812         callback = waiting_item_callback;
 813
 814     item->u.wait_object = CreateThreadpoolWait(callback, item,
 815             (TP_CALLBACK_ENVIRON *)&queue->envs[TP_CALLBACK_PRIORITY_NORMAL]);
 816     SetThreadpoolWait(item->u.wait_object, event, NULL);
 817
 818     TRACE("dispatched %p.\n", result);
 819
 820     return S_OK;
 821 }
 822
 823 static HRESULT queue_submit_timer(struct queue *queue, IRtwqAsyncResult *result, INT64 timeout, DWORD period,
 824         RTWQWORKITEM_KEY *key)
 825 {
 826     PTP_TIMER_CALLBACK callback;
 827     struct work_item *item;
 828     FILETIME filetime;
 829     LARGE_INTEGER t;
 830
 831     if (!(item = alloc_work_item(queue, 0, result)))
 832         return E_OUTOFMEMORY;
 833
 834     if (key)
 835     {
 836         queue_mark_item_pending(SCHEDULED_ITEM_KEY_MASK, item, key);
 837     }
 838
 839     if (period)
 840         callback = periodic_item_callback;
 841     else
 842         callback = key ? scheduled_item_cancelable_callback : scheduled_item_callback;
 843
 844     t.QuadPart = timeout * 1000 * 10;
 845     filetime.dwLowDateTime = t.u.LowPart;
 846     filetime.dwHighDateTime = t.u.HighPart;
 847
 848     item->u.timer_object = CreateThreadpoolTimer(callback, item,
 849             (TP_CALLBACK_ENVIRON *)&queue->envs[TP_CALLBACK_PRIORITY_NORMAL]);
 850     SetThreadpoolTimer(item->u.timer_object, &filetime, period, 0);
 851
 852     TRACE("dispatched %p.\n", result);
 853
 854     return S_OK;
 855 }
 856
 857 static HRESULT queue_cancel_item(struct queue *queue, RTWQWORKITEM_KEY key)
 858 {
 859     HRESULT hr = RTWQ_E_NOT_FOUND;
 860     struct work_item *item;
 861
 862     EnterCriticalSection(&queue->cs);
 863     LIST_FOR_EACH_ENTRY(item, &queue->pending_items, struct work_item, entry)
 864     {
 865         if (item->key == key)
 866         {
 867             key >>= 32;
 868             if ((key & WAIT_ITEM_KEY_MASK) == WAIT_ITEM_KEY_MASK)
 869             {
 870                 IRtwqAsyncResult_SetStatus(item->result, RTWQ_E_OPERATION_CANCELLED);
 871                 invoke_async_callback(item->result);
 872                 CloseThreadpoolWait(item->u.wait_object);
 873             }
 874             else if ((key & SCHEDULED_ITEM_KEY_MASK) == SCHEDULED_ITEM_KEY_MASK)
 875                 CloseThreadpoolTimer(item->u.timer_object);
 876             else
 877                 WARN("Unknown item key mask %#x.\n", (DWORD)key);
 878             queue_release_pending_item(item);
 879             hr = S_OK;
 880             break;
 881         }
 882     }
 883     LeaveCriticalSection(&queue->cs);
 884
 885     return hr;
 886 }
 887
 888 static HRESULT alloc_user_queue(const struct queue_desc *desc, DWORD *queue_id)
 889 {
 890     struct queue_handle *entry;
 891     struct queue *queue;
 892     unsigned int idx;
 893
 894     *queue_id = RTWQ_CALLBACK_QUEUE_UNDEFINED;
 895
 896     if (platform_lock <= 0)
 897         return RTWQ_E_SHUTDOWN;
 898
 899     queue = heap_alloc_zero(sizeof(*queue));
 900     if (!queue)
 901         return E_OUTOFMEMORY;
 902
 903     init_work_queue(desc, queue);
 904
 905     EnterCriticalSection(&queues_section);
 906
 907     entry = next_free_user_queue;
 908     if (entry)
 909         next_free_user_queue = entry->obj;
 910     else if (next_unused_user_queue < user_queues + MAX_USER_QUEUE_HANDLES)
 911         entry = next_unused_user_queue++;
 912     else
 913     {
 914         LeaveCriticalSection(&queues_section);
 915         heap_free(queue);
 916         WARN("Out of user queue handles.\n");
 917         return E_OUTOFMEMORY;
 918     }
 919
 920     entry->refcount = 1;
 921     entry->obj = queue;
 922     if (++queue_generation == 0xffff) queue_generation = 1;
 923     entry->generation = queue_generation;
 924     idx = entry - user_queues + FIRST_USER_QUEUE_HANDLE;
 925     *queue_id = (idx << 16) | entry->generation;
 926
 927     LeaveCriticalSection(&queues_section);
 928
 929     return S_OK;
 930 }
 931
 932 struct async_result
 933 {
 934     RTWQASYNCRESULT result;
 935     LONG refcount;
 936     IUnknown *object;
 937     IUnknown *state;
 938 };
 939
 940 static struct async_result *impl_from_IRtwqAsyncResult(IRtwqAsyncResult *iface)
 941 {
 942     return CONTAINING_RECORD(iface, struct async_result, result.AsyncResult);
 943 }
 944
 945 static HRESULT WINAPI async_result_QueryInterface(IRtwqAsyncResult *iface, REFIID riid, void **obj)
 946 {
 947     TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), obj);
 948
 949     if (IsEqualIID(riid, &IID_IRtwqAsyncResult) ||
 950             IsEqualIID(riid, &IID_IUnknown))
 951     {
 952         *obj = iface;
 953         IRtwqAsyncResult_AddRef(iface);
 954         return S_OK;
 955     }
 956
 957     *obj = NULL;
 958     WARN("Unsupported interface %s.\n", debugstr_guid(riid));
 959     return E_NOINTERFACE;
 960 }
 961
 962 static ULONG WINAPI async_result_AddRef(IRtwqAsyncResult *iface)
 963 {
 964     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
 965     ULONG refcount = InterlockedIncrement(&result->refcount);
 966
 967     TRACE("%p, %u.\n", iface, refcount);
 968
 969     return refcount;
 970 }
 971
 972 static ULONG WINAPI async_result_Release(IRtwqAsyncResult *iface)
 973 {
 974     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
 975     ULONG refcount = InterlockedDecrement(&result->refcount);
 976
 977     TRACE("%p, %u.\n", iface, refcount);
 978
 979     if (!refcount)
 980     {
 981         if (result->result.pCallback)
 982             IRtwqAsyncCallback_Release(result->result.pCallback);
 983         if (result->object)
 984             IUnknown_Release(result->object);
 985         if (result->state)
 986             IUnknown_Release(result->state);
 987         if (result->result.hEvent)
 988             CloseHandle(result->result.hEvent);
 989         heap_free(result);
 990
 991         RtwqUnlockPlatform();
 992     }
 993
 994     return refcount;
 995 }
 996
 997 static HRESULT WINAPI async_result_GetState(IRtwqAsyncResult *iface, IUnknown **state)
 998 {
 999     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
1000
1001     TRACE("%p, %p.\n", iface, state);
1002
1003     if (!result->state)
1004         return E_POINTER;
1005
1006     *state = result->state;
1007     IUnknown_AddRef(*state);
1008
1009     return S_OK;
1010 }
1011
1012 static HRESULT WINAPI async_result_GetStatus(IRtwqAsyncResult *iface)
1013 {
1014     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
1015
1016     TRACE("%p.\n", iface);
1017
1018     return result->result.hrStatusResult;
1019 }
1020
1021 static HRESULT WINAPI async_result_SetStatus(IRtwqAsyncResult *iface, HRESULT status)
1022 {
1023     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
1024
1025     TRACE("%p, %#x.\n", iface, status);
1026
1027     result->result.hrStatusResult = status;
1028
1029     return S_OK;
1030 }
1031
1032 static HRESULT WINAPI async_result_GetObject(IRtwqAsyncResult *iface, IUnknown **object)
1033 {
1034     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
1035
1036     TRACE("%p, %p.\n", iface, object);
1037
1038     if (!result->object)
1039         return E_POINTER;
1040
1041     *object = result->object;
1042     IUnknown_AddRef(*object);
1043
1044     return S_OK;
1045 }
1046
1047 static IUnknown * WINAPI async_result_GetStateNoAddRef(IRtwqAsyncResult *iface)
1048 {
1049     struct async_result *result = impl_from_IRtwqAsyncResult(iface);
1050
1051     TRACE("%p.\n", iface);
1052
1053     return result->state;
1054 }
1055
1056 static const IRtwqAsyncResultVtbl async_result_vtbl =
1057 {
1058     async_result_QueryInterface,
1059     async_result_AddRef,
1060     async_result_Release,
1061     async_result_GetState,
1062     async_result_GetStatus,
1063     async_result_SetStatus,
1064     async_result_GetObject,
1065     async_result_GetStateNoAddRef,
1066 };
1067
1068 static HRESULT create_async_result(IUnknown *object, IRtwqAsyncCallback *callback, IUnknown *state, IRtwqAsyncResult **out)
1069 {
1070     struct async_result *result;
1071
1072     if (!out)
1073         return E_INVALIDARG;
1074
1075     result = heap_alloc_zero(sizeof(*result));
1076     if (!result)
1077         return E_OUTOFMEMORY;
1078
1079     RtwqLockPlatform();
1080
1081     result->result.AsyncResult.lpVtbl = &async_result_vtbl;
1082     result->refcount = 1;
1083     result->object = object;
1084     if (result->object)
1085         IUnknown_AddRef(result->object);
1086     result->result.pCallback = callback;
1087     if (result->result.pCallback)
1088         IRtwqAsyncCallback_AddRef(result->result.pCallback);
1089     result->state = state;
1090     if (result->state)
1091         IUnknown_AddRef(result->state);
1092
1093     *out = &result->result.AsyncResult;
1094
1095     TRACE("Created async result object %p.\n", *out);
1096
1097     return S_OK;
1098 }
1099
1100 HRESULT WINAPI RtwqCreateAsyncResult(IUnknown *object, IRtwqAsyncCallback *callback, IUnknown *state,
1101         IRtwqAsyncResult **out)
1102 {
1103     TRACE("%p, %p, %p, %p.\n", object, callback, state, out);
1104
1105     return create_async_result(object, callback, state, out);
1106 }
1107
1108 HRESULT WINAPI RtwqLockPlatform(void)
1109 {
1110     InterlockedIncrement(&platform_lock);
1111
1112     return S_OK;
1113 }
1114
1115 HRESULT WINAPI RtwqUnlockPlatform(void)
1116 {
1117     InterlockedDecrement(&platform_lock);
1118
1119     return S_OK;
1120 }
1121
1122 static void init_system_queues(void)
1123 {
1124     struct queue_desc desc;
1125     HRESULT hr;
1126
1127     /* Always initialize standard queue, keep the rest lazy. */
1128
1129     EnterCriticalSection(&queues_section);
1130
1131     if (system_queues[SYS_QUEUE_STANDARD].pool)
1132     {
1133         LeaveCriticalSection(&queues_section);
1134         return;
1135     }
1136
1137     if (FAILED(hr = CoIncrementMTAUsage(&mta_cookie)))
1138         WARN("Failed to initialize MTA, hr %#x.\n", hr);
1139
1140     desc.queue_type = RTWQ_STANDARD_WORKQUEUE;
1141     desc.ops = &pool_queue_ops;
1142     desc.target_queue = 0;
1143     init_work_queue(&desc, &system_queues[SYS_QUEUE_STANDARD]);
1144
1145     LeaveCriticalSection(&queues_section);
1146 }
1147
1148 HRESULT WINAPI RtwqStartup(void)
1149 {
1150     if (InterlockedIncrement(&platform_lock) == 1)
1151     {
1152         init_system_queues();
1153     }
1154
1155     return S_OK;
1156 }
1157
1158 static void shutdown_system_queues(void)
1159 {
1160     unsigned int i;
1161     HRESULT hr;
1162
1163     EnterCriticalSection(&queues_section);
1164
1165     for (i = 0; i < ARRAY_SIZE(system_queues); ++i)
1166     {
1167         shutdown_queue(&system_queues[i]);
1168     }
1169
1170     if (FAILED(hr = CoDecrementMTAUsage(mta_cookie)))
1171         WARN("Failed to uninitialize MTA, hr %#x.\n", hr);
1172
1173     LeaveCriticalSection(&queues_section);
1174 }
1175
1176 HRESULT WINAPI RtwqShutdown(void)
1177 {
1178     if (platform_lock <= 0)
1179         return S_OK;
1180
1181     if (InterlockedExchangeAdd(&platform_lock, -1) == 1)
1182     {
1183         shutdown_system_queues();
1184     }
1185
1186     return S_OK;
1187 }
1188
1189 HRESULT WINAPI RtwqPutWaitingWorkItem(HANDLE event, LONG priority, IRtwqAsyncResult *result, RTWQWORKITEM_KEY *key)
1190 {
1191     struct queue *queue;
1192     HRESULT hr;
1193
1194     TRACE("%p, %d, %p, %p.\n", event, priority, result, key);
1195
1196     if (FAILED(hr = grab_queue(RTWQ_CALLBACK_QUEUE_TIMER, &queue)))
1197         return hr;
1198
1199     hr = queue_submit_wait(queue, event, priority, result, key);
1200
1201     return hr;
1202 }
1203
1204 static HRESULT schedule_work_item(IRtwqAsyncResult *result, INT64 timeout, RTWQWORKITEM_KEY *key)
1205 {
1206     struct queue *queue;
1207     HRESULT hr;
1208
1209     if (FAILED(hr = grab_queue(RTWQ_CALLBACK_QUEUE_TIMER, &queue)))
1210         return hr;
1211
1212     TRACE("%p, %s, %p.\n", result, wine_dbgstr_longlong(timeout), key);
1213
1214     return queue_submit_timer(queue, result, timeout, 0, key);
1215 }
1216
1217 HRESULT WINAPI RtwqScheduleWorkItem(IRtwqAsyncResult *result, INT64 timeout, RTWQWORKITEM_KEY *key)
1218 {
1219     TRACE("%p, %s, %p.\n", result, wine_dbgstr_longlong(timeout), key);
1220
1221     return schedule_work_item(result, timeout, key);
1222 }
1223
1224 struct periodic_callback
1225 {
1226     IRtwqAsyncCallback IRtwqAsyncCallback_iface;
1227     LONG refcount;
1228     RTWQPERIODICCALLBACK callback;
1229 };
1230
1231 static struct periodic_callback *impl_from_IRtwqAsyncCallback(IRtwqAsyncCallback *iface)
1232 {
1233     return CONTAINING_RECORD(iface, struct periodic_callback, IRtwqAsyncCallback_iface);
1234 }
1235
1236 static HRESULT WINAPI periodic_callback_QueryInterface(IRtwqAsyncCallback *iface, REFIID riid, void **obj)
1237 {
1238     if (IsEqualIID(riid, &IID_IRtwqAsyncCallback) ||
1239             IsEqualIID(riid, &IID_IUnknown))
1240     {
1241         *obj = iface;
1242         IRtwqAsyncCallback_AddRef(iface);
1243         return S_OK;
1244     }
1245
1246     *obj = NULL;
1247     return E_NOINTERFACE;
1248 }
1249
1250 static ULONG WINAPI periodic_callback_AddRef(IRtwqAsyncCallback *iface)
1251 {
1252     struct periodic_callback *callback = impl_from_IRtwqAsyncCallback(iface);
1253     ULONG refcount = InterlockedIncrement(&callback->refcount);
1254
1255     TRACE("%p, %u.\n", iface, refcount);
1256
1257     return refcount;
1258 }
1259
1260 static ULONG WINAPI periodic_callback_Release(IRtwqAsyncCallback *iface)
1261 {
1262     struct periodic_callback *callback = impl_from_IRtwqAsyncCallback(iface);
1263     ULONG refcount = InterlockedDecrement(&callback->refcount);
1264
1265     TRACE("%p, %u.\n", iface, refcount);
1266
1267     if (!refcount)
1268         heap_free(callback);
1269
1270     return refcount;
1271 }
1272
1273 static HRESULT WINAPI periodic_callback_GetParameters(IRtwqAsyncCallback *iface, DWORD *flags, DWORD *queue)
1274 {
1275     return E_NOTIMPL;
1276 }
1277
1278 static HRESULT WINAPI periodic_callback_Invoke(IRtwqAsyncCallback *iface, IRtwqAsyncResult *result)
1279 {
1280     struct periodic_callback *callback = impl_from_IRtwqAsyncCallback(iface);
1281     IUnknown *context = NULL;
1282
1283     if (FAILED(IRtwqAsyncResult_GetObject(result, &context)))
1284         WARN("Expected object to be set for result object.\n");
1285
1286     callback->callback(context);
1287
1288     if (context)
1289         IUnknown_Release(context);
1290
1291     return S_OK;
1292 }
1293
1294 static const IRtwqAsyncCallbackVtbl periodic_callback_vtbl =
1295 {
1296     periodic_callback_QueryInterface,
1297     periodic_callback_AddRef,
1298     periodic_callback_Release,
1299     periodic_callback_GetParameters,
1300     periodic_callback_Invoke,
1301 };
1302
1303 static HRESULT create_periodic_callback_obj(RTWQPERIODICCALLBACK callback, IRtwqAsyncCallback **out)
1304 {
1305     struct periodic_callback *object;
1306
1307     object = heap_alloc(sizeof(*object));
1308     if (!object)
1309         return E_OUTOFMEMORY;
1310
1311     object->IRtwqAsyncCallback_iface.lpVtbl = &periodic_callback_vtbl;
1312     object->refcount = 1;
1313     object->callback = callback;
1314
1315     *out = &object->IRtwqAsyncCallback_iface;
1316
1317     return S_OK;
1318 }
1319
1320 HRESULT WINAPI RtwqAddPeriodicCallback(RTWQPERIODICCALLBACK callback, IUnknown *context, DWORD *key)
1321 {
1322     IRtwqAsyncCallback *periodic_callback;
1323     RTWQWORKITEM_KEY workitem_key;
1324     IRtwqAsyncResult *result;
1325     struct queue *queue;
1326     HRESULT hr;
1327
1328     TRACE("%p, %p, %p.\n", callback, context, key);
1329
1330     if (FAILED(hr = grab_queue(RTWQ_CALLBACK_QUEUE_TIMER, &queue)))
1331         return hr;
1332
1333     if (FAILED(hr = create_periodic_callback_obj(callback, &periodic_callback)))
1334         return hr;
1335
1336     hr = create_async_result(context, periodic_callback, NULL, &result);
1337     IRtwqAsyncCallback_Release(periodic_callback);
1338     if (FAILED(hr))
1339         return hr;
1340
1341     /* Same period MFGetTimerPeriodicity() returns. */
1342     hr = queue_submit_timer(queue, result, 0, 10, key ? &workitem_key : NULL);
1343
1344     IRtwqAsyncResult_Release(result);
1345
1346     if (key)
1347         *key = workitem_key;
1348
1349     return S_OK;
1350 }
1351
1352 HRESULT WINAPI RtwqRemovePeriodicCallback(DWORD key)
1353 {
1354     struct queue *queue;
1355     HRESULT hr;
1356
1357     TRACE("%#x.\n", key);
1358
1359     if (FAILED(hr = grab_queue(RTWQ_CALLBACK_QUEUE_TIMER, &queue)))
1360         return hr;
1361
1362     return queue_cancel_item(queue, get_item_key(SCHEDULED_ITEM_KEY_MASK, key));
1363 }
1364
1365 HRESULT WINAPI RtwqCancelWorkItem(RTWQWORKITEM_KEY key)
1366 {
1367     struct queue *queue;
1368     HRESULT hr;
1369
1370     TRACE("%s.\n", wine_dbgstr_longlong(key));
1371
1372     if (FAILED(hr = grab_queue(RTWQ_CALLBACK_QUEUE_TIMER, &queue)))
1373         return hr;
1374
1375     return queue_cancel_item(queue, key);
1376 }
1377
1378 HRESULT WINAPI RtwqInvokeCallback(IRtwqAsyncResult *result)
1379 {
1380     TRACE("%p.\n", result);
1381
1382     return invoke_async_callback(result);
1383 }
1384
1385 HRESULT WINAPI RtwqPutWorkItem(DWORD queue, LONG priority, IRtwqAsyncResult *result)
1386 {
1387     TRACE("%#x, %d, %p.\n", queue, priority, result);
1388
1389     return queue_put_work_item(queue, priority, result);
1390 }
1391
1392 HRESULT WINAPI RtwqAllocateWorkQueue(RTWQ_WORKQUEUE_TYPE queue_type, DWORD *queue)
1393 {
1394     struct queue_desc desc;
1395
1396     TRACE("%d, %p.\n", queue_type, queue);
1397
1398     desc.queue_type = queue_type;
1399     desc.ops = &pool_queue_ops;
1400     desc.target_queue = 0;
1401     return alloc_user_queue(&desc, queue);
1402 }
1403
1404 HRESULT WINAPI RtwqLockWorkQueue(DWORD queue)
1405 {
1406     TRACE("%#x.\n", queue);
1407
1408     return lock_user_queue(queue);
1409 }
1410
1411 HRESULT WINAPI RtwqUnlockWorkQueue(DWORD queue)
1412 {
1413     TRACE("%#x.\n", queue);
1414
1415     return unlock_user_queue(queue);
1416 }
1417
1418 HRESULT WINAPI RtwqSetLongRunning(DWORD queue_id, BOOL enable)
1419 {
1420     struct queue *queue;
1421     HRESULT hr;
1422     int i;
1423
1424     TRACE("%#x, %d.\n", queue_id, enable);
1425
1426     lock_user_queue(queue_id);
1427
1428     if (SUCCEEDED(hr = grab_queue(queue_id, &queue)))
1429     {
1430         for (i = 0; i < ARRAY_SIZE(queue->envs); ++i)
1431             queue->envs[i].u.s.LongFunction = !!enable;
1432     }
1433
1434     unlock_user_queue(queue_id);
1435
1436     return hr;
1437 }
1438
1439 HRESULT WINAPI RtwqLockSharedWorkQueue(const WCHAR *usageclass, LONG priority, DWORD *taskid, DWORD *queue)
1440 {
1441     FIXME("%s, %d, %p, %p.\n", debugstr_w(usageclass), priority, taskid, queue);
1442
1443     return RtwqAllocateWorkQueue(RTWQ_STANDARD_WORKQUEUE, queue);
1444 }
1445
1446 HRESULT WINAPI RtwqSetDeadline(DWORD queue_id, LONGLONG deadline, HANDLE *request)
1447 {
1448     FIXME("%#x, %s, %p.\n", queue_id, wine_dbgstr_longlong(deadline), request);
1449
1450     return E_NOTIMPL;
1451 }
1452
1453 HRESULT WINAPI RtwqSetDeadline2(DWORD queue_id, LONGLONG deadline, LONGLONG predeadline, HANDLE *request)
1454 {
1455     FIXME("%#x, %s, %s, %p.\n", queue_id, wine_dbgstr_longlong(deadline), wine_dbgstr_longlong(predeadline), request);
1456
1457     return E_NOTIMPL;
1458 }
1459
1460 HRESULT WINAPI RtwqCancelDeadline(HANDLE request)
1461 {
1462     FIXME("%p.\n", request);
1463
1464     return E_NOTIMPL;
1465 }
1466
1467 HRESULT WINAPI RtwqAllocateSerialWorkQueue(DWORD target_queue, DWORD *queue)
1468 {
1469     struct queue_desc desc;
1470
1471     TRACE("%#x, %p.\n", target_queue, queue);
1472
1473     desc.queue_type = RTWQ_STANDARD_WORKQUEUE;
1474     desc.ops = &serial_queue_ops;
1475     desc.target_queue = target_queue;
1476     return alloc_user_queue(&desc, queue);
1477 }
1478
1479 HRESULT WINAPI RtwqJoinWorkQueue(DWORD queue, HANDLE hFile, HANDLE *cookie)
1480 {
1481     FIXME("%#x, %p, %p.\n", queue, hFile, cookie);
1482
1483     return E_NOTIMPL;
1484 }
1485
1486 HRESULT WINAPI RtwqUnjoinWorkQueue(DWORD queue, HANDLE cookie)
1487 {
1488     FIXME("%#x, %p.\n", queue, cookie);
1489
1490     return E_NOTIMPL;
1491 }
1492
1493 HRESULT WINAPI RtwqGetWorkQueueMMCSSClass(DWORD queue, WCHAR *class, DWORD *length)
1494 {
1495     FIXME("%#x, %p, %p.\n", queue, class, length);
1496
1497     return E_NOTIMPL;
1498 }
1499
1500 HRESULT WINAPI RtwqGetWorkQueueMMCSSTaskId(DWORD queue, DWORD *taskid)
1501 {
1502     FIXME("%#x, %p.\n", queue, taskid);
1503
1504     return E_NOTIMPL;
1505 }
1506
1507 HRESULT WINAPI RtwqGetWorkQueueMMCSSPriority(DWORD queue, LONG *priority)
1508 {
1509     FIXME("%#x, %p.\n", queue, priority);
1510
1511     return E_NOTIMPL;
1512 }
1513
1514 HRESULT WINAPI RtwqRegisterPlatformWithMMCSS(const WCHAR *class, DWORD *taskid, LONG priority)
1515 {
1516     FIXME("%s, %p, %d.\n", debugstr_w(class), taskid, priority);
1517
1518     return E_NOTIMPL;
1519 }
1520
1521 HRESULT WINAPI RtwqUnregisterPlatformFromMMCSS(void)
1522 {
1523     FIXME("\n");
1524
1525     return E_NOTIMPL;
1526 }
1527
1528 HRESULT WINAPI RtwqBeginRegisterWorkQueueWithMMCSS(DWORD queue, const WCHAR *class, DWORD taskid, LONG priority,
1529         IRtwqAsyncCallback *callback, IUnknown *state)
1530 {
1531     FIXME("%#x, %s, %u, %d, %p, %p.\n", queue, debugstr_w(class), taskid, priority, callback, state);
1532
1533     return E_NOTIMPL;
1534 }
1535
1536 HRESULT WINAPI RtwqEndRegisterWorkQueueWithMMCSS(IRtwqAsyncResult *result, DWORD *taskid)
1537 {
1538     FIXME("%p, %p.\n", result, taskid);
1539
1540     return E_NOTIMPL;
1541 }
1542
1543 HRESULT WINAPI RtwqBeginUnregisterWorkQueueWithMMCSS(DWORD queue, IRtwqAsyncCallback *callback, IUnknown *state)
1544 {
1545     FIXME("%#x, %p, %p.\n", queue, callback, state);
1546
1547     return E_NOTIMPL;
1548 }
1549
1550 HRESULT WINAPI RtwqEndUnregisterWorkQueueWithMMCSS(IRtwqAsyncResult *result)
1551 {
1552     FIXME("%p.\n", result);
1553
1554     return E_NOTIMPL;
1555 }
1556
1557 HRESULT WINAPI RtwqRegisterPlatformEvents(IRtwqPlatformEvents *events)
1558 {
1559     FIXME("%p.\n", events);
1560
1561     return E_NOTIMPL;
1562 }
1563
1564 HRESULT WINAPI RtwqUnregisterPlatformEvents(IRtwqPlatformEvents *events)
1565 {
1566     FIXME("%p.\n", events);
1567
1568     return E_NOTIMPL;
1569 }