Archipelago » qemu

/*

 * Small test program to verify simulated mmap behaviour.

 *

 * When running qemu-linux-user with the -p flag, you may need to tell

 * this test program about the pagesize because getpagesize() will not reflect

 * the -p choice. Simply pass one argument beeing the pagesize.

 *

 * Copyright (c) 2007 AXIS Communications AB

 * Written by Edgar E. Iglesias.

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 * 

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

 */

#include <stdio.h>

#include <stdlib.h>

#include <stdint.h>

#include <string.h>

#include <unistd.h>

#include <sys/mman.h>

#define D(x)

#define fail_unless(x)                                         \

do                                                             \

{                                                              \

  if (!(x)) {                                                  \

    fprintf (stderr, "FAILED at %s:%d\n", __FILE__, __LINE__); \

    exit (EXIT_FAILURE);                                       \

  }                                                            \

} while (0);

unsigned char *dummybuf;

static unsigned int pagesize;

static unsigned int pagemask;

int test_fd;

size_t test_fsize;

void check_aligned_anonymous_unfixed_mmaps(void)

{

        void *p1;

        void *p2;

        void *p3;

        void *p4;

        void *p5;

        uintptr_t p;

        int i;

        fprintf (stderr, "%s", __func__);

        for (i = 0; i < 0x1fff; i++)

        {

                size_t len;

                len = pagesize + (pagesize * i & 7);

                p1 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                p2 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                p3 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                p4 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                p5 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                /* Make sure we get pages aligned with the pagesize. The

                   target expects this.  */

                fail_unless (p1 != MAP_FAILED);

                fail_unless (p2 != MAP_FAILED);

                fail_unless (p3 != MAP_FAILED);

                fail_unless (p4 != MAP_FAILED);

                fail_unless (p5 != MAP_FAILED);

                p = (uintptr_t) p1;

                D(printf ("p=%x\n", p));

                fail_unless ((p & pagemask) == 0);

                p = (uintptr_t) p2;

                fail_unless ((p & pagemask) == 0);

                p = (uintptr_t) p3;

                fail_unless ((p & pagemask) == 0);

                p = (uintptr_t) p4;

                fail_unless ((p & pagemask) == 0);

                p = (uintptr_t) p5;

                fail_unless ((p & pagemask) == 0);

                /* Make sure we can read from the entire area.  */

                memcpy (dummybuf, p1, pagesize);

                memcpy (dummybuf, p2, pagesize);

                memcpy (dummybuf, p3, pagesize);

                memcpy (dummybuf, p4, pagesize);

                memcpy (dummybuf, p5, pagesize);

                munmap (p1, len);

                munmap (p2, len);

                munmap (p3, len);

                munmap (p4, len);

                munmap (p5, len);

        }

        fprintf (stderr, " passed\n");

}

void check_large_anonymous_unfixed_mmap(void)

{

        void *p1;

        uintptr_t p;

        size_t len;

        fprintf (stderr, "%s", __func__);

        len = 0x02000000;

        p1 = mmap(NULL, len, PROT_READ, 

                  MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

        /* Make sure we get pages aligned with the pagesize. The

           target expects this.  */

        fail_unless (p1 != MAP_FAILED);

        p = (uintptr_t) p1;

        fail_unless ((p & pagemask) == 0);

        /* Make sure we can read from the entire area.  */

        memcpy (dummybuf, p1, pagesize);

        munmap (p1, len);

        fprintf (stderr, " passed\n");

}

void check_aligned_anonymous_unfixed_colliding_mmaps(void)

{

        char *p1;

        char *p2;

        char *p3;

        uintptr_t p;

        int i;

        fprintf (stderr, "%s", __func__);

        for (i = 0; i < 0x2fff; i++)

        {

                int nlen;

                p1 = mmap(NULL, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                fail_unless (p1 != MAP_FAILED);

                p = (uintptr_t) p1;

                fail_unless ((p & pagemask) == 0);

                memcpy (dummybuf, p1, pagesize);

                p2 = mmap(NULL, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                fail_unless (p2 != MAP_FAILED);

                p = (uintptr_t) p2;

                fail_unless ((p & pagemask) == 0);

                memcpy (dummybuf, p2, pagesize);

                munmap (p1, pagesize);

                nlen = pagesize * 8;

                p3 = mmap(NULL, nlen, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);

                /* Check if the mmaped areas collide.  */

                if (p3 < p2 

                    && (p3 + nlen) > p2)

                        fail_unless (0);

                memcpy (dummybuf, p3, pagesize);

                /* Make sure we get pages aligned with the pagesize. The

                   target expects this.  */

                fail_unless (p3 != MAP_FAILED);

                p = (uintptr_t) p3;

                fail_unless ((p & pagemask) == 0);

                munmap (p2, pagesize);

                munmap (p3, nlen);

        }

        fprintf (stderr, " passed\n");

}

void check_aligned_anonymous_fixed_mmaps(void)

{

        char *addr;

        void *p1;

        uintptr_t p;

        int i;

        /* Find a suitable address to start with.  */

        addr = mmap(NULL, pagesize * 40, PROT_READ | PROT_WRITE, 

                    MAP_PRIVATE | MAP_ANONYMOUS,

                    -1, 0);

        fprintf (stderr, "%s addr=%p", __func__, addr);

        fail_unless (addr != MAP_FAILED);

        for (i = 0; i < 40; i++)

        {

                /* Create submaps within our unfixed map.  */

                p1 = mmap(addr, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED,

                          -1, 0);

                /* Make sure we get pages aligned with the pagesize. 

                   The target expects this.  */

                p = (uintptr_t) p1;

                fail_unless (p1 == addr);

                fail_unless ((p & pagemask) == 0);                

                memcpy (dummybuf, p1, pagesize);

                munmap (p1, pagesize);

                addr += pagesize;

        }

        fprintf (stderr, " passed\n");

}

void check_aligned_anonymous_fixed_mmaps_collide_with_host(void)

{

        char *addr;

        void *p1;

        uintptr_t p;

        int i;

        /* Find a suitable address to start with.  Right were the x86 hosts

         stack is.  */

        addr = ((void *)0x80000000);

        fprintf (stderr, "%s addr=%p", __func__, addr);

        fprintf (stderr, "FIXME: QEMU fails to track pages used by the host.");

        for (i = 0; i < 20; i++)

        {

                /* Create submaps within our unfixed map.  */

                p1 = mmap(addr, pagesize, PROT_READ | PROT_WRITE, 

                          MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED,

                          -1, 0);

                /* Make sure we get pages aligned with the pagesize. 

                   The target expects this.  */

                p = (uintptr_t) p1;

                fail_unless (p1 == addr);

                fail_unless ((p & pagemask) == 0);                

                memcpy (p1, dummybuf, pagesize);

                munmap (p1, pagesize);

                addr += pagesize;

        }

        fprintf (stderr, " passed\n");

}

void check_file_unfixed_mmaps(void)

{

        unsigned int *p1, *p2, *p3;

        uintptr_t p;

        int i;

        fprintf (stderr, "%s", __func__);

        for (i = 0; i < 0x10; i++)

        {

                size_t len;

                len = pagesize;

                p1 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE, 

                          test_fd, 0);

                p2 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE, 

                          test_fd, pagesize);

                p3 = mmap(NULL, len, PROT_READ, 

                          MAP_PRIVATE, 

                          test_fd, pagesize * 2);

                fail_unless (p1 != MAP_FAILED);

                fail_unless (p2 != MAP_FAILED);

                fail_unless (p3 != MAP_FAILED);

                /* Make sure we get pages aligned with the pagesize. The

                   target expects this.  */

                p = (uintptr_t) p1;

                fail_unless ((p & pagemask) == 0);

                p = (uintptr_t) p2;

                fail_unless ((p & pagemask) == 0);

                p = (uintptr_t) p3;

                fail_unless ((p & pagemask) == 0);

                /* Verify that the file maps was made correctly.  */

                D(printf ("p1=%d p2=%d p3=%d\n", *p1, *p2, *p3));

                fail_unless (*p1 == 0);

                fail_unless (*p2 == (pagesize / sizeof *p2));

                fail_unless (*p3 == ((pagesize * 2) / sizeof *p3));

                memcpy (dummybuf, p1, pagesize);

                memcpy (dummybuf, p2, pagesize);

                memcpy (dummybuf, p3, pagesize);

                munmap (p1, len);

                munmap (p2, len);

                munmap (p3, len);

        }

        fprintf (stderr, " passed\n");

}

void check_file_unfixed_eof_mmaps(void)

{

        char *cp;

        unsigned int *p1;

        uintptr_t p;

        int i;

        fprintf (stderr, "%s", __func__);

        for (i = 0; i < 0x10; i++)

        {

                p1 = mmap(NULL, pagesize, PROT_READ, 

                          MAP_PRIVATE, 

                          test_fd, 

                          (test_fsize - sizeof *p1) & ~pagemask);

                fail_unless (p1 != MAP_FAILED);

                /* Make sure we get pages aligned with the pagesize. The

                   target expects this.  */

                p = (uintptr_t) p1;

                fail_unless ((p & pagemask) == 0);

                /* Verify that the file maps was made correctly.  */

                fail_unless (p1[(test_fsize & pagemask) / sizeof *p1 - 1]

                             == ((test_fsize - sizeof *p1) / sizeof *p1));

                /* Verify that the end of page is accessable and zeroed.  */

                cp = (void *) p1;

                fail_unless (cp[pagesize - 4] == 0);

                munmap (p1, pagesize);

        }

        fprintf (stderr, " passed\n");

}

void check_file_fixed_eof_mmaps(void)

{

        char *addr;

        char *cp;

        unsigned int *p1;

        uintptr_t p;

        int i;

        /* Find a suitable address to start with.  */

        addr = mmap(NULL, pagesize * 44, PROT_READ, 

                    MAP_PRIVATE | MAP_ANONYMOUS,

                    -1, 0);

        fprintf (stderr, "%s addr=%p", __func__, (void *)addr);

        fail_unless (addr != MAP_FAILED);

        for (i = 0; i < 0x10; i++)

        {

                /* Create submaps within our unfixed map.  */

                p1 = mmap(addr, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_FIXED, 

                          test_fd, 

                          (test_fsize - sizeof *p1) & ~pagemask);

                fail_unless (p1 != MAP_FAILED);

                /* Make sure we get pages aligned with the pagesize. The

                   target expects this.  */

                p = (uintptr_t) p1;

                fail_unless ((p & pagemask) == 0);

                /* Verify that the file maps was made correctly.  */

                fail_unless (p1[(test_fsize & pagemask) / sizeof *p1 - 1]

                             == ((test_fsize - sizeof *p1) / sizeof *p1));

                /* Verify that the end of page is accessable and zeroed.  */

                cp = (void *)p1;

                fail_unless (cp[pagesize - 4] == 0);

                munmap (p1, pagesize);

                addr += pagesize;

        }

        fprintf (stderr, " passed\n");

}

void check_file_fixed_mmaps(void)

{

        unsigned char *addr;

        unsigned int *p1, *p2, *p3, *p4;

        int i;

        /* Find a suitable address to start with.  */

        addr = mmap(NULL, pagesize * 40 * 4, PROT_READ, 

                    MAP_PRIVATE | MAP_ANONYMOUS,

                    -1, 0);

        fprintf (stderr, "%s addr=%p", __func__, (void *)addr);

        fail_unless (addr != MAP_FAILED);

        for (i = 0; i < 40; i++)

        {

                p1 = mmap(addr, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_FIXED,

                          test_fd, 0);

                p2 = mmap(addr + pagesize, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_FIXED,

                          test_fd, pagesize);

                p3 = mmap(addr + pagesize * 2, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_FIXED,

                          test_fd, pagesize * 2);

                p4 = mmap(addr + pagesize * 3, pagesize, PROT_READ, 

                          MAP_PRIVATE | MAP_FIXED,

                          test_fd, pagesize * 3);

                /* Make sure we get pages aligned with the pagesize. 

                   The target expects this.  */

                fail_unless (p1 == (void *)addr);

                fail_unless (p2 == (void *)addr + pagesize);

                fail_unless (p3 == (void *)addr + pagesize * 2);

                fail_unless (p4 == (void *)addr + pagesize * 3);

                /* Verify that the file maps was made correctly.  */

                fail_unless (*p1 == 0);

                fail_unless (*p2 == (pagesize / sizeof *p2));

                fail_unless (*p3 == ((pagesize * 2) / sizeof *p3));

                fail_unless (*p4 == ((pagesize * 3) / sizeof *p4));

                memcpy (dummybuf, p1, pagesize);

                memcpy (dummybuf, p2, pagesize);

                memcpy (dummybuf, p3, pagesize);

                memcpy (dummybuf, p4, pagesize);

                munmap (p1, pagesize);

                munmap (p2, pagesize);

                munmap (p3, pagesize);

                munmap (p4, pagesize);

                addr += pagesize * 4;

        }

        fprintf (stderr, " passed\n");

}

int main(int argc, char **argv)

{

        char tempname[] = "/tmp/.cmmapXXXXXX";

        unsigned int i;

        /* Trust the first argument, otherwise probe the system for our

           pagesize.  */

        if (argc > 1)

                pagesize = strtoul(argv[1], NULL, 0);

        else

                pagesize = sysconf(_SC_PAGESIZE);

        /* Assume pagesize is a power of two.  */

        pagemask = pagesize - 1;

        dummybuf = malloc (pagesize);

        printf ("pagesize=%u pagemask=%x\n", pagesize, pagemask);

        test_fd = mkstemp(tempname);

        unlink(tempname);

        /* Fill the file with int's counting from zero and up.  */

        for (i = 0; i < (pagesize * 4) / sizeof i; i++)

                write (test_fd, &i, sizeof i);

        /* Append a few extra writes to make the file end at non 

           page boundary.  */

        write (test_fd, &i, sizeof i); i++;

        write (test_fd, &i, sizeof i); i++;

        write (test_fd, &i, sizeof i); i++;

        test_fsize = lseek(test_fd, 0, SEEK_CUR);

        /* Run the tests.  */

        check_aligned_anonymous_unfixed_mmaps();

        check_aligned_anonymous_unfixed_colliding_mmaps();

        check_aligned_anonymous_fixed_mmaps();

        check_file_unfixed_mmaps();

        check_file_fixed_mmaps();

        check_file_fixed_eof_mmaps();

        check_file_unfixed_eof_mmaps();

        /* Fails at the moment.  */

        /* check_aligned_anonymous_fixed_mmaps_collide_with_host(); */

        return EXIT_SUCCESS;

}