atexit.c File Reference

#include <apitest.h>
#include <stdlib.h>
#include <stdio.h>
#include <ntndk.h>

Include dependency graph for atexit.c:

Go to the source code of this file.

Typedefs
typedef int(__cdecl *	PFN_atexit) (void(__cdecl *) (void))

Functions
void	exitfunc1 (void)
void	exitfunc2 (void)
void	exitfunc3 (void)
void	Test_atexit ()
	START_TEST (atexit)

Variables
int	g_sequence = 0
HANDLE	g_hSemaphore

Typedef Documentation

PFN_atexit

typedef int(__cdecl * PFN_atexit) (void(__cdecl *)(void))

Definition at line 39 of file atexit.c.

Function Documentation

exitfunc1()

void exitfunc1

(

void

)

Definition at line 17 of file atexit.c.

{
    ok_int(g_sequence, 1);
    g_sequence++;
    ReleaseSemaphore(g_hSemaphore, 1, NULL);
}

Referenced by Test_atexit().

exitfunc2()

void exitfunc2

(

void

)

Definition at line 24 of file atexit.c.

{
    ok_int(g_sequence, 2);
    g_sequence++;
    ReleaseSemaphore(g_hSemaphore, 1, NULL);
}

Referenced by Test_atexit().

exitfunc3()

void exitfunc3

(

void

)

Definition at line 31 of file atexit.c.

{
    ok_int(g_sequence, 0);
    g_sequence++;
    ReleaseSemaphore(g_hSemaphore, 1, NULL);
    printf("exitfunc3\n");
}

Referenced by Test_atexit().

START_TEST()

START_TEST

(

atexit

)

Definition at line 64 of file atexit.c.

{
    CHAR Buffer[MAX_PATH];
    PSTR CommandLine;
    int result;
    HANDLE hSemaphore;
    SEMAPHORE_BASIC_INFORMATION SemInfo;
    NTSTATUS Status;

    /* Check recursive call */
    CommandLine = GetCommandLineA();
    if (strstr(CommandLine, "-run") != NULL)
    {
        Test_atexit();
        return;
    }

    /* Create a named semaphore to count atexit callbacks in remote process */
    hSemaphore = CreateSemaphoreA(NULL, 1, 20, "atext_apitest_sempahore");

    /* Run the actual test in a new process */
    sprintf(Buffer, "%s -run", CommandLine);
    result = system(Buffer);
    ok_int(result, 0);

    /* Check the new semaphore state */
    Status = NtQuerySemaphore(hSemaphore, SemaphoreBasicInformation, &SemInfo, sizeof(SemInfo), NULL);
    ok(NT_SUCCESS(Status), "NtQuerySemaphore failed: 0x%lx\n", Status);
    ok_int(SemInfo.CurrentCount, 4);
}

Test_atexit()

void Test_atexit

(

)

Definition at line 41 of file atexit.c.

{
    HMODULE hmod;
    PFN_atexit patexit;

    /* Open the named sempahore to count atexit callbacks */
    g_hSemaphore = OpenSemaphoreA(SEMAPHORE_ALL_ACCESS, FALSE, "atext_apitest_sempahore");
    ok(g_hSemaphore != NULL, "couldn't open semaphore.\n");

    /* Load atexit from msvcrt.dll */
    hmod = GetModuleHandleA("msvcrt.dll");
    patexit = (PFN_atexit)GetProcAddress(hmod, "atexit");
    ok(patexit != NULL, "failed to get atexit from msvcrt.dll\n");

    /* Register 3 exit functions, the second one in msvcrt. */
    ok_int(atexit(exitfunc1), 0);
    if (patexit != NULL)
    {
        ok_int(patexit(exitfunc2), 0);
    }
    ok_int(atexit(exitfunc3), 0);
}

Referenced by START_TEST().

Variable Documentation

g_hSemaphore

HANDLE g_hSemaphore

Definition at line 15 of file atexit.c.

Referenced by exitfunc1(), exitfunc2(), exitfunc3(), and Test_atexit().

g_sequence

int g_sequence = 0

Definition at line 14 of file atexit.c.

Referenced by exitfunc1(), exitfunc2(), and exitfunc3().