/*      $OpenBSD: atexit_test.c,v 1.1 2014/11/23 08:46:49 guenther Exp $       */

/*
 * Copyright (c) 2014 Philip Guenther <guenther@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <dlfcn.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>

void *libaa, *libab;

#define CALLBACK(name)  static void name(void) { printf("exe "#name"\n"); }

CALLBACK(cleanup1)
CALLBACK(cleanup2)
CALLBACK(cleanup3)

static void
cleanup_dlclose(void)
{
        printf("exe cleanup_dlclose begin\n");
        dlclose(libaa);
        dlclose(libab);
        printf("exe cleanup_dlclose end\n");
}

static void
aa(void)
{
        void (*func)(void) = dlsym(libaa, "aa");
        if (func == NULL)
                errx(1, "dlsym(libaa, aa): %s", dlerror());
        func();
}

static void
ab(void)
{
        void (*func)(void) = dlsym(libab, "ab");
        if (func == NULL)
                errx(1, "dlsym(libab, ab): %s", dlerror());
        func();
}

int
main(int argc, char **argv)
{
        int test;

        libaa = dlopen(LIBAA, RTLD_LAZY);
        if (libaa == NULL)
                errx(1, "dlopen(%s, RTLD_LAZY): %s", LIBAA, dlerror());

        libab = dlopen(LIBAB, RTLD_LAZY);
        if (libab == NULL)
                errx(1, "dlopen(%s, RTLD_LAZY): %s", LIBAB, dlerror());

        if (argc != 2)
                test = 0;
        else
                test = atoi(argv[1]);

        switch (test) {
        case 0:
                /* 1, aa, 2, ab, 3, then exit */
                atexit(cleanup1);
                aa();
                atexit(cleanup2);
                ab();
                atexit(cleanup3);
                exit(0);

        case 1:
                /* 1, aa, 2, ab, 3, then dlclose aa, then bb */
                atexit(cleanup1);
                aa();
                atexit(cleanup2);
                ab();
                atexit(cleanup3);
                dlclose(libaa);
                dlclose(libab);
                exit(0);

        case 2:
                /* 1, aa, cleanup_dlclose, ab, 3, then exit */
                atexit(cleanup1);
                aa();
                atexit(cleanup_dlclose);
                ab();
                atexit(cleanup3);
                exit(0);

        case 3:
                /* 1, aa, 2, ab, cleanup_dlclose, then exit */
                atexit(cleanup1);
                aa();
                atexit(cleanup2);
                ab();
                atexit(cleanup_dlclose);
                exit(0);

        }

        return (0);
}