summaryrefslogtreecommitdiff
path: root/testsuite/tests/rts/linker_unload.c
blob: f1cc891df187dd9e73ee3e0d9681760a06abb4ea (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#include "ghcconfig.h"
#include <stdio.h>
#include <stdlib.h>
#include "Rts.h"

#define ITERATIONS 10000

#if defined(mingw32_HOST_OS)
#define OBJPATH L"Test.o"
#else
#define OBJPATH "Test.o"
#endif

typedef int testfun(int);

void loadPkg(pathchar *path)
{
    int r;

    r = loadArchive(path);
    if (!r) {
        errorBelch("loadObjs(%s) failed", path);
        exit(1);
    }
}

int main (int argc, char *argv[])
{
    testfun *f;
    int i, r;

    hs_init(&argc, &argv);

    initLinker_(0);

    for (i=1; i < argc; i++) {
#if defined(mingw32_HOST_OS)
        size_t len = mbstowcs(NULL, argv[i], 0) + 1;
        if (len == -1) {
            errorBelch("invalid multibyte sequence in argument %d: %s", i, argv[i]);
            exit(1);
        }
        wchar_t *buf = (wchar_t*)_alloca(len * sizeof(wchar_t));
        size_t len2 = mbstowcs(buf, argv[i], len);
        if (len != len2 + 1) {
            errorBelch("something fishy is going on in argument %d: %s", i, argv[i]);
            exit(1);
        }
        loadPkg(buf);
#else
        loadPkg(argv[i]);
#endif
    }

    for (i=0; i < ITERATIONS; i++) {
        r = loadObj(OBJPATH);
        if (!r) {
            errorBelch("loadObj(%s) failed", OBJPATH);
            exit(1);
        }
        r = resolveObjs();
        if (!r) {
            errorBelch("resolveObjs failed");
            exit(1);
        }
#if LEADING_UNDERSCORE
        f = lookupSymbol("_f");
#else
        f = lookupSymbol("f");
#endif
        if (!f) {
            errorBelch("lookupSymbol failed");
            exit(1);
        }
        r = f(3);
        if (r != 4) {
            errorBelch("call failed; %d", r);
            exit(1);
        }
        unloadObj(OBJPATH);
        performMajorGC();
        printf("%d ", i);
        fflush(stdout);
    }
}