InitOnceExecuteOnce.c File Reference

#include "k32_vista.h"
#include <ndk/exfuncs.h>
#include <wine/config.h>
#include <wine/port.h>

Include dependency graph for InitOnceExecuteOnce.c:

Go to the source code of this file.

Functions
static DWORD WINAPI	RtlRunOnceBeginInitialize (RTL_RUN_ONCE *once, ULONG flags, void **context)
static DWORD WINAPI	RtlpRunOnceComplete (RTL_RUN_ONCE *once, ULONG flags, void *context)
static DWORD WINAPI	RtlpRunOnceExecuteOnce (RTL_RUN_ONCE *once, PRTL_RUN_ONCE_INIT_FN func, void *param, void **context)
BOOL NTAPI	InitOnceExecuteOnce (INIT_ONCE *once, PINIT_ONCE_FN func, void *param, void **context)

Variables
HANDLE	keyed_event = NULL

Function Documentation

InitOnceExecuteOnce()

BOOL NTAPI InitOnceExecuteOnce	(	INIT_ONCE *	once,
		PINIT_ONCE_FN	func,
		void *	param,
		void **	context
	)

Definition at line 115 of file InitOnceExecuteOnce.c.

{
    return !RtlpRunOnceExecuteOnce( once, (PRTL_RUN_ONCE_INIT_FN)func, param, context );
}

Referenced by apartment_createwindowifneeded(), DllGetClassObject(), get_notif_hwnd(), and init_winsock().

RtlpRunOnceComplete()

static DWORD WINAPI RtlpRunOnceComplete ( RTL_RUN_ONCE *  once, ULONG  flags, void *  context  )

static

Definition at line 55 of file InitOnceExecuteOnce.c.

{
    if ((ULONG_PTR)context & 3) return STATUS_INVALID_PARAMETER;

    if (flags & RTL_RUN_ONCE_INIT_FAILED)
    {
        if (context) return STATUS_INVALID_PARAMETER;
        if (flags & RTL_RUN_ONCE_ASYNC) return STATUS_INVALID_PARAMETER;
    }
    else context = (void *)((ULONG_PTR)context | 2);

    for (;;)
    {
        ULONG_PTR val = (ULONG_PTR)once->Ptr;

        switch (val & 3)
        {
        case 1:  /* in progress */
            if (interlocked_cmpxchg_ptr( &once->Ptr, context, (void *)val ) != (void *)val) break;
            val &= ~3;
            while (val)
            {
                ULONG_PTR next = *(ULONG_PTR *)val;
                NtReleaseKeyedEvent( keyed_event, (void *)val, FALSE, NULL );
                val = next;
            }
            return STATUS_SUCCESS;

        case 3:  /* in progress, async */
            if (!(flags & RTL_RUN_ONCE_ASYNC)) return STATUS_INVALID_PARAMETER;
            if (interlocked_cmpxchg_ptr( &once->Ptr, context, (void *)val ) != (void *)val) break;
            return STATUS_SUCCESS;

        default:
            return STATUS_UNSUCCESSFUL;
        }
    }
}

Referenced by RtlpRunOnceExecuteOnce().

RtlpRunOnceExecuteOnce()

static DWORD WINAPI RtlpRunOnceExecuteOnce ( RTL_RUN_ONCE *  once, PRTL_RUN_ONCE_INIT_FN  func, void *  param, void **  context  )

static

Definition at line 94 of file InitOnceExecuteOnce.c.

{
    DWORD ret = RtlRunOnceBeginInitialize( once, 0, context );

    if (ret != STATUS_PENDING) return ret;

    if (!func( once, param, context ))
    {
        RtlpRunOnceComplete( once, RTL_RUN_ONCE_INIT_FAILED, NULL );
        return STATUS_UNSUCCESSFUL;
    }

    return RtlpRunOnceComplete( once, 0, context ? *context : NULL );
}

Referenced by InitOnceExecuteOnce().

RtlRunOnceBeginInitialize()

static DWORD WINAPI RtlRunOnceBeginInitialize ( RTL_RUN_ONCE *  once, ULONG  flags, void **  context  )

static

Definition at line 12 of file InitOnceExecuteOnce.c.

{
    if (flags & RTL_RUN_ONCE_CHECK_ONLY)
    {
        ULONG_PTR val = (ULONG_PTR)once->Ptr;

        if (flags & RTL_RUN_ONCE_ASYNC) return STATUS_INVALID_PARAMETER;
        if ((val & 3) != 2) return STATUS_UNSUCCESSFUL;
        if (context) *context = (void *)(val & ~3);
        return STATUS_SUCCESS;
    }

    for (;;)
    {
        ULONG_PTR next, val = (ULONG_PTR)once->Ptr;

        switch (val & 3)
        {
        case 0:  /* first time */
            if (!interlocked_cmpxchg_ptr( &once->Ptr,
                                          (flags & RTL_RUN_ONCE_ASYNC) ? (void *)3 : (void *)1, 0 ))
                return STATUS_PENDING;
            break;

        case 1:  /* in progress, wait */
            if (flags & RTL_RUN_ONCE_ASYNC) return STATUS_INVALID_PARAMETER;
            next = val & ~3;
            if (interlocked_cmpxchg_ptr( &once->Ptr, (void *)((ULONG_PTR)&next | 1),
                                         (void *)val ) == (void *)val)
                NtWaitForKeyedEvent( keyed_event, &next, FALSE, NULL );
            break;

        case 2:  /* done */
            if (context) *context = (void *)(val & ~3);
            return STATUS_SUCCESS;

        case 3:  /* in progress, async */
            if (!(flags & RTL_RUN_ONCE_ASYNC)) return STATUS_INVALID_PARAMETER;
            return STATUS_PENDING;
        }
    }
}

Referenced by RtlpRunOnceExecuteOnce().

Variable Documentation

keyed_event

HANDLE keyed_event = NULL

Definition at line 10 of file InitOnceExecuteOnce.c.

Referenced by RtlpRunOnceComplete(), and RtlRunOnceBeginInitialize().