#include <sys/mman.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <err.h>

#define TESTSIZE        (128L*1024*1024)

char *TestBuf;
char *FlagBuf;

int
main(int argc, char **argv)
{
        size_t beg;
        size_t len;
        size_t fbeg;
        size_t flen;
        size_t i;
        long counter = 0;

        TestBuf = calloc(1, TESTSIZE);
        FlagBuf = calloc(1, TESTSIZE / 4096);

        for (;;) {
                beg = random() & (TESTSIZE - 1) & ~4095L;
                len = random() & (TESTSIZE - 1) & ~4095L;
                if (beg + len > TESTSIZE)
                        len = TESTSIZE - beg;
                fbeg = beg / 4096;
                flen = len / 4096;
                if (len == 0)
                        continue;

                if (random() & 1) {
                        if (mlock(TestBuf + beg, len) < 0) {
                                printf("%ld   mlock: %016lx-%016lx: ERROR %s\n",
                                        counter, beg, beg + len,
                                        strerror(errno));
                        } else {
                                memset(FlagBuf + fbeg, 1, flen);
                                printf("%ld   mlock: %016lx-%016lx: OK\n",
                                        counter, beg, beg + len);
                        }
                } else {
                        for (i = 0; i < flen; ++i) {
                                if (FlagBuf[fbeg+i] == 0)
                                        break;
                        }
                        if (i == flen) {
                                memset(FlagBuf + fbeg, 0, flen);
                                if (munlock(TestBuf + beg, len) < 0) {
                                        printf("%ld munlock: %016lx-%016lx: "
                                                "ERROR %s\n",
                                                counter, beg, beg + len,
                                                strerror(errno));
                                } else {
                                        printf("%ld munlock: %016lx-%016lx: OK\n",
                                                counter, beg, beg + len);
                                }
                        } else {
                                if (munlock(TestBuf + beg, len) == 0) {
                                        printf("%ld munlock: %016lx-%016lx: ERROR "
                                                "should have failed "
                                                "at %016lx\n",
                                                counter, beg, beg + len,
                                                (fbeg + i) * 4096L);
                                } else {
                                        printf("%ld munlock: %016lx-%016lx: OK "
                                                "(intentional failure)\n",
                                                counter, beg, beg + len);
                                }
                        }
                }
#if 0
                printf("%ld\r", counter);
#endif
                fflush(stdout);
                ++counter;
        }
        return (0);
}