| /* IBM Corporation */ |
| /* 01/02/2003 Port to LTP avenkat@us.ibm.com */ |
| /* 06/30/2001 Port to Linux nsharoff@us.ibm.com */ |
| /* |
| * Copyright (c) International Business Machines Corp., 2003 |
| * |
| * |
| * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| #define _GNU_SOURCE 1 |
| #include <stdio.h> |
| #include <fcntl.h> |
| #include <signal.h> |
| #include <sys/mman.h> |
| #include <sys/wait.h> |
| #include <sys/stat.h> |
| #include <unistd.h> |
| #include <stdlib.h> |
| #include <errno.h> |
| #include <sys/types.h> |
| #include <limits.h> |
| /***** LTP Port *****/ |
| #include "test.h" |
| #define FAILED 0 |
| #define PASSED 1 |
| |
| int local_flag = PASSED; |
| char *TCID = "mmapstress01"; //tmnoextend |
| FILE *temp; |
| int TST_TOTAL = 1; |
| |
| int anyfail(); |
| void ok_exit(); |
| /***** ** ** *****/ |
| |
| /* |
| * This test stresses mmaps, without dealing with fragments or anything! |
| * It forks a specified number of children, |
| * all of whom mmap the same file, make a given number of accesses |
| * to random pages in the map (reading & writing and comparing data). |
| * Then the child exits and the parent forks another to take its place. |
| * Each time a child is forked, it stats the file and maps the full |
| * length of the file. |
| * |
| * This program continues to run until it either receives a SIGINT, |
| * or times out (if a timeout value is specified). When either of |
| * these things happens, it cleans up its kids, then checks the |
| * file to make sure it has the correct data. |
| * |
| * usage: |
| * tmnoextend -p nprocs [-t minutes -f filesize -S sparseoffset |
| * -r -o -m -l -d] |
| * where: |
| * -p nprocs - specifies the number of mapping children |
| * to create. (nprocs + 1 children actually |
| * get created, since one is the writer child) |
| * -t minutes - specifies minutes to run. If not specified, |
| * default is to run forever until a SIGINT |
| * is received. |
| * -f filesize - initial filesize (defaults to FILESIZE) |
| * -S sparseoffset - when non-zero, causes a sparse area to |
| * be left before the data, meaning that the |
| * actual initial file size is sparseoffset + |
| * filesize. Useful for testing large files. |
| * (default is 0). |
| * -r - randomize number of pages map children check. |
| * (random % MAXLOOPS). If not specified, each |
| * child checks MAXLOOPS pages. |
| * -o - randomize offset of file to map. (default is 0) |
| * -m - do random msync/fsyncs as well |
| * -l - if set, the output file is not removed on |
| * program exit. |
| * -d - enable debug output |
| * |
| * Compile with -DLARGE_FILE to enable file sizes > 2 GB. |
| */ |
| |
| #define MAXLOOPS 500 /* max pages for map children to write */ |
| #define FILESIZE 4096 /* initial filesize set up by parent */ |
| |
| #ifdef roundup |
| #undef roundup |
| #endif |
| #define roundup(x, y) ((((x)+((y)-1))/(y))*(y)) |
| #define min(x, y) (((x) < (y)) ? (x) : (y)) |
| |
| extern time_t time(time_t *); |
| extern char *ctime(const time_t *); |
| extern void *malloc(size_t); |
| extern long lrand48(void); |
| extern void srand(unsigned); |
| extern void srand48(long); |
| extern int rand(void); |
| extern int atoi(const char *); |
| |
| char *usage = |
| "-p nprocs [-t minutes -f filesize -S sparseoffset -r -o -m -l -d]"; |
| |
| typedef unsigned char uchar_t; |
| #define SIZE_MAX UINT_MAX |
| |
| unsigned int initrand(void); |
| void finish(int sig); |
| void child_mapper(char *file, unsigned procno, unsigned nprocs); |
| int fileokay(char *file, uchar_t * expbuf); |
| int finished = 0; |
| int leavefile = 0; |
| |
| int debug = 0; |
| #ifdef LARGE_FILE |
| off64_t filesize = FILESIZE; |
| off64_t sparseoffset = 0; |
| #else /* LARGE_FILE */ |
| off_t filesize = FILESIZE; |
| off_t sparseoffset = 0; |
| #endif /* LARGE_FILE */ |
| unsigned randloops = 0; |
| unsigned dosync = 0; |
| unsigned do_offset = 0; |
| unsigned pattern = 0; |
| char filename[64]; |
| |
| int main(int argc, char *argv[]) |
| { |
| char *progname; |
| int fd; |
| int c; |
| extern char *optarg; |
| unsigned nprocs = 0; |
| unsigned procno; |
| pid_t *pidarray = NULL; |
| pid_t pid; |
| uchar_t *buf = NULL; |
| unsigned int seed; |
| int pagesize = sysconf(_SC_PAGE_SIZE); |
| float alarmtime = 0; |
| struct sigaction sa; |
| unsigned i; |
| int write_cnt; |
| uchar_t data; |
| int no_prob = 0; |
| int wait_stat; |
| time_t t; |
| #ifdef LARGE_FILE |
| off64_t bytes_left; |
| #else /* LARGE_FILE */ |
| off_t bytes_left; |
| #endif /* LARGE_FILE */ |
| |
| progname = *argv; |
| tst_tmpdir(); |
| if (argc < 2) { |
| tst_brkm(TBROK, NULL, "usage: %s %s\n", progname, usage); |
| } |
| |
| while ((c = getopt(argc, argv, "S:omdlrf:p:t:")) != -1) { |
| switch (c) { |
| case 'd': |
| debug = 1; |
| break; |
| case 't': |
| alarmtime = atof(optarg) * 60; |
| break; |
| case 'p': |
| nprocs = atoi(optarg); |
| break; |
| case 'l': |
| leavefile = 1; |
| break; |
| case 'f': |
| #ifdef LARGE_FILE |
| filesize = atoll(optarg); |
| #else /* LARGE_FILE */ |
| filesize = atoi(optarg); |
| #endif /* LARGE_FILE */ |
| if (filesize < 0) { |
| (void)fprintf(stderr, "error: negative " |
| "filesize\n"); |
| anyfail(); |
| } |
| break; |
| case 'r': |
| randloops = 1; |
| break; |
| case 'm': |
| dosync = 1; |
| break; |
| case 'o': |
| do_offset = 1; |
| break; |
| case 'S': |
| #ifdef LARGE_FILE |
| sparseoffset = atoll(optarg); |
| #else /* LARGE_FILE */ |
| sparseoffset = atoi(optarg); |
| #endif /* LARGE_FILE */ |
| if (sparseoffset % pagesize != 0) { |
| fprintf(stderr, |
| "sparseoffset must be pagesize multiple\n"); |
| anyfail(); |
| } |
| break; |
| default: |
| (void)fprintf(stderr, "usage: %s %s\n", progname, |
| usage); |
| tst_exit(); |
| } |
| } |
| |
| /* nprocs is >= 0 since it's unsigned */ |
| if (nprocs > 255) { |
| (void)fprintf(stderr, "invalid nprocs %d - (range 0-255)\n", |
| nprocs); |
| anyfail(); |
| } |
| |
| (void)time(&t); |
| //(void)printf("%s: Started %s", argv[0], ctime(&t)); LTP Port |
| (void)sprintf(filename, "%sout.%d", progname, getpid()); |
| seed = initrand(); |
| pattern = seed & 0xff; |
| |
| if (debug) { |
| #ifdef LARGE_FILE |
| (void)printf("creating file <%s> with %Ld bytes, pattern %d\n", |
| filename, filesize, pattern); |
| #else /* LARGE_FILE */ |
| (void)printf("creating file <%s> with %ld bytes, pattern %d\n", |
| filename, filesize, pattern); |
| #endif /* LARGE_FILE */ |
| if (alarmtime) |
| (void)printf("running for %f minutes\n", |
| alarmtime / 60); |
| else |
| (void)printf("running with no time limit\n"); |
| } |
| |
| /* |
| * Plan for death by signal. User may have specified |
| * a time limit, in which set an alarm and catch SIGALRM. |
| * Also catch and cleanup with SIGINT. |
| */ |
| sa.sa_handler = finish; |
| sa.sa_flags = 0; |
| if (sigemptyset(&sa.sa_mask)) { |
| perror("sigemptyset error"); |
| goto cleanup; |
| } |
| |
| if (sigaction(SIGINT, &sa, 0) == -1) { |
| perror("sigaction error SIGINT"); |
| goto cleanup; |
| } |
| if (sigaction(SIGQUIT, &sa, 0) == -1) { |
| perror("sigaction error SIGQUIT"); |
| goto cleanup; |
| } |
| if (sigaction(SIGTERM, &sa, 0) == -1) { |
| perror("sigaction error SIGTERM"); |
| goto cleanup; |
| } |
| |
| if (alarmtime) { |
| if (sigaction(SIGALRM, &sa, 0) == -1) { |
| perror("sigaction error"); |
| goto cleanup; |
| } |
| (void)alarm(alarmtime); |
| } |
| #ifdef LARGE_FILE |
| if ((fd = open64(filename, O_CREAT | O_TRUNC | O_RDWR, 0664)) == -1) { |
| #else /* LARGE_FILE */ |
| if ((fd = open(filename, O_CREAT | O_TRUNC | O_RDWR, 0664)) == -1) { |
| #endif /* LARGE_FILE */ |
| perror("open error"); |
| anyfail(); |
| } |
| |
| if ((buf = malloc(pagesize)) == NULL |
| || (pidarray = malloc(nprocs * sizeof(pid_t))) == NULL) { |
| perror("malloc error"); |
| anyfail(); |
| } |
| |
| for (i = 0; i < nprocs; i++) |
| *(pidarray + i) = 0; |
| |
| for (i = 0, data = 0; i < pagesize; i++) { |
| *(buf + i) = (data + pattern) & 0xff; |
| if (++data == nprocs) |
| data = 0; |
| } |
| #ifdef LARGE_FILE |
| if (lseek64(fd, sparseoffset, SEEK_SET) < 0) { |
| #else /* LARGE_FILE */ |
| if (lseek(fd, sparseoffset, SEEK_SET) < 0) { |
| #endif /* LARGE_FILE */ |
| perror("lseek"); |
| anyfail(); |
| } |
| for (bytes_left = filesize; bytes_left; bytes_left -= c) { |
| write_cnt = min(pagesize, bytes_left); |
| if ((c = write(fd, buf, write_cnt)) != write_cnt) { |
| if (c == -1) { |
| perror("write error"); |
| } else { |
| (void)fprintf(stderr, "write: wrote %d of %d " |
| "bytes\n", c, write_cnt); |
| } |
| (void)close(fd); |
| (void)unlink(filename); |
| anyfail(); |
| } |
| } |
| |
| (void)close(fd); |
| |
| /* |
| * Fork off mmap children. |
| */ |
| for (procno = 0; procno < nprocs; procno++) { |
| switch (pid = fork()) { |
| |
| case -1: |
| perror("fork error"); |
| goto cleanup; |
| |
| case 0: |
| child_mapper(filename, procno, nprocs); |
| exit(0); |
| |
| default: |
| pidarray[procno] = pid; |
| } |
| } |
| |
| /* |
| * Now wait for children and refork them as needed. |
| */ |
| |
| while (!finished) { |
| pid = wait(&wait_stat); |
| /* |
| * Block signals while processing child exit. |
| */ |
| |
| if (sighold(SIGALRM) || sighold(SIGINT)) { |
| perror("sighold error"); |
| goto cleanup; |
| } |
| |
| if (pid != -1) { |
| /* |
| * Check exit status, then refork with the |
| * appropriate procno. |
| */ |
| if (!WIFEXITED(wait_stat) |
| || WEXITSTATUS(wait_stat) != 0) { |
| (void)fprintf(stderr, "child exit with err " |
| "<x%x>\n", wait_stat); |
| goto cleanup; |
| } |
| for (i = 0; i < nprocs; i++) |
| if (pid == pidarray[i]) |
| break; |
| if (i == nprocs) { |
| (void)fprintf(stderr, "unknown child pid %d, " |
| "<x%x>\n", pid, wait_stat); |
| goto cleanup; |
| } |
| |
| if ((pid = fork()) == -1) { |
| perror("fork error"); |
| pidarray[i] = 0; |
| goto cleanup; |
| } else if (pid == 0) { /* child */ |
| child_mapper(filename, i, nprocs); |
| exit(0); |
| } else |
| pidarray[i] = pid; |
| } else { |
| /* |
| * wait returned an error. If EINTR, then |
| * normal finish, else it's an unexpected |
| * error... |
| */ |
| if (errno != EINTR || !finished) { |
| perror("unexpected wait error"); |
| goto cleanup; |
| } |
| } |
| if (sigrelse(SIGALRM) || sigrelse(SIGINT)) { |
| perror("sigrelse error"); |
| goto cleanup; |
| } |
| } |
| |
| /* |
| * Finished! Check the file for sanity, then kill all |
| * the children and done!. |
| */ |
| |
| if (sighold(SIGALRM)) { |
| perror("sighold error"); |
| goto cleanup; |
| } |
| (void)alarm(0); |
| no_prob = 1; |
| |
| cleanup: |
| for (i = 0; i < nprocs; i++) |
| (void)kill(pidarray[i], SIGKILL); |
| |
| while (wait(&wait_stat) != -1 || errno != ECHILD) |
| continue; |
| |
| if (no_prob) { /* only check file if no errors */ |
| if (!fileokay(filename, buf)) { |
| (void)fprintf(stderr, "file data incorrect!\n"); |
| (void)printf(" leaving file <%s>\n", filename); |
| /***** LTP Port *****/ |
| local_flag = FAILED; |
| anyfail(); |
| /***** ** *****/ |
| } else { |
| (void)printf("file data okay\n"); |
| if (!leavefile) |
| (void)unlink(filename); |
| } |
| } else |
| (void)printf(" leaving file <%s>\n", filename); |
| |
| (void)time(&t); |
| //(void)printf("%s: Finished %s", argv[0], ctime(&t)); LTP Port |
| ok_exit(); |
| tst_exit(); |
| } |
| |
| /* |
| * Child process that reads/writes map. The child stats the file |
| * to determine the size, maps the size of the file, then reads/writes |
| * its own locations on random pages of the map (its locations being |
| * determined based on nprocs & procno). After a specific number of |
| * iterations, it exits. |
| */ |
| void child_mapper(char *file, unsigned procno, unsigned nprocs) |
| { |
| #ifdef LARGE_FILE |
| struct stat64 statbuf; |
| off64_t filesize; |
| off64_t offset; |
| #else /* LARGE_FILE */ |
| struct stat statbuf; |
| off_t filesize; |
| off_t offset; |
| #endif /* LARGE_FILE */ |
| size_t validsize; |
| size_t mapsize; |
| char *maddr = NULL, *paddr; |
| int fd; |
| size_t pagesize = sysconf(_SC_PAGE_SIZE); |
| unsigned randpage; |
| unsigned int seed; |
| unsigned loopcnt; |
| unsigned nloops; |
| unsigned mappages; |
| unsigned i; |
| |
| seed = initrand(); /* initialize random seed */ |
| |
| #ifdef LARGE_FILE |
| if (stat64(file, &statbuf) == -1) { |
| #else /* LARGE_FILE */ |
| if (stat(file, &statbuf) == -1) { |
| #endif /* LARGE_FILE */ |
| perror("stat error"); |
| anyfail(); |
| } |
| filesize = statbuf.st_size; |
| |
| #ifdef LARGE_FILE |
| if ((fd = open64(file, O_RDWR)) == -1) { |
| #else /* LARGE_FILE */ |
| if ((fd = open(file, O_RDWR)) == -1) { |
| #endif /* LARGE_FILE */ |
| perror("open error"); |
| anyfail(); |
| } |
| |
| if (statbuf.st_size - sparseoffset > SIZE_MAX) { |
| fprintf(stderr, "size_t overflow when setting up map\n"); |
| anyfail(); |
| } |
| mapsize = (size_t) (statbuf.st_size - sparseoffset); |
| mappages = roundup(mapsize, pagesize) / pagesize; |
| offset = sparseoffset; |
| if (do_offset) { |
| int pageoffset = lrand48() % mappages; |
| int byteoffset = pageoffset * pagesize; |
| offset += byteoffset; |
| mapsize -= byteoffset; |
| mappages -= pageoffset; |
| } |
| nloops = (randloops) ? (lrand48() % MAXLOOPS) : MAXLOOPS; |
| |
| if (debug) { |
| #ifdef LARGE_FILE |
| (void)printf("child %d (pid %ld): seed %d, fsize %Ld, " |
| "mapsize %d, off %Ld, loop %d\n", |
| procno, getpid(), seed, filesize, mapsize, |
| offset / pagesize, nloops); |
| #else /* LARGE_FILE */ |
| (void)printf("child %d (pid %d): seed %d, fsize %ld, " |
| "mapsize %ld, off %ld, loop %d\n", |
| procno, getpid(), seed, filesize, (long)mapsize, |
| offset / pagesize, nloops); |
| #endif /* LARGE_FILE */ |
| } |
| #ifdef LARGE_FILE |
| if ((maddr = mmap64(0, mapsize, PROT_READ | PROT_WRITE, MAP_SHARED, |
| fd, offset)) == (caddr_t) - 1) { |
| #else /* LARGE_FILE */ |
| if ((maddr = mmap(0, mapsize, PROT_READ | PROT_WRITE, MAP_SHARED, |
| fd, offset)) == (caddr_t) - 1) { |
| #endif /* LARGE_FILE */ |
| perror("mmap error"); |
| anyfail(); |
| } |
| |
| (void)close(fd); |
| |
| /* |
| * Now loop read/writing random pages. |
| */ |
| for (loopcnt = 0; loopcnt < nloops; loopcnt++) { |
| randpage = lrand48() % mappages; |
| paddr = maddr + (randpage * pagesize); /* page address */ |
| |
| if (randpage < mappages - 1 || !(mapsize % pagesize)) |
| validsize = pagesize; |
| else |
| validsize = mapsize % pagesize; |
| |
| for (i = procno; i < validsize; i += nprocs) { |
| if (*((unsigned char *)(paddr + i)) |
| != ((procno + pattern) & 0xff)) { |
| (void)fprintf(stderr, "child %d: invalid data " |
| "<x%x>", procno, |
| *((unsigned char *)(paddr + i))); |
| (void)fprintf(stderr, " at pg %d off %d, exp " |
| "<x%x>\n", randpage, i, |
| (procno + pattern) & 0xff); |
| anyfail(); |
| } |
| |
| /* |
| * Now write it. |
| */ |
| *(paddr + i) = (procno + pattern) & 0xff; |
| } |
| } |
| if (dosync) { |
| /* |
| * Exercise msync() as well! |
| */ |
| randpage = lrand48() % mappages; |
| paddr = maddr + (randpage * pagesize); /* page address */ |
| if (msync(paddr, (mappages - randpage) * pagesize, |
| MS_SYNC) == -1) { |
| anyfail(); |
| } |
| } |
| if (munmap(maddr, mapsize) == -1) { |
| perror("munmap failed"); |
| local_flag = FAILED; |
| anyfail(); |
| } |
| exit(0); |
| } |
| |
| /* |
| * Make sure file has all the correct data. |
| */ |
| int fileokay(char *file, uchar_t * expbuf) |
| { |
| #ifdef LARGE_FILE |
| struct stat64 statbuf; |
| #else /* LARGE_FILE */ |
| struct stat statbuf; |
| #endif /* LARGE_FILE */ |
| size_t mapsize; |
| uchar_t *readbuf; |
| unsigned mappages; |
| unsigned pagesize = sysconf(_SC_PAGE_SIZE); |
| int fd; |
| int cnt; |
| unsigned i, j; |
| |
| #ifdef LARGE_FILE |
| if ((fd = open64(file, O_RDONLY)) == -1) { |
| #else /* LARGE_FILE */ |
| if ((fd = open(file, O_RDONLY)) == -1) { |
| #endif /* LARGE_FILE */ |
| perror("open error"); |
| /***** LTP Port *****/ |
| local_flag = FAILED; |
| anyfail(); |
| /***** ** *****/ |
| return 0; |
| } |
| #ifdef LARGE_FILE |
| if (fstat64(fd, &statbuf) == -1) { |
| #else /* LARGE_FILE */ |
| if (fstat(fd, &statbuf) == -1) { |
| #endif /* LARGE_FILE */ |
| perror("stat error"); |
| /***** LTP Port *****/ |
| local_flag = FAILED; |
| anyfail(); |
| /***** ** *****/ |
| return 0; |
| } |
| #ifdef LARGE_FILE |
| if (lseek64(fd, sparseoffset, SEEK_SET) < 0) { |
| #else /* LARGE_FILE */ |
| if (lseek(fd, sparseoffset, SEEK_SET) < 0) { |
| #endif /* LARGE_FILE */ |
| perror("lseek"); |
| anyfail(); |
| } |
| readbuf = malloc(pagesize); |
| |
| if (statbuf.st_size - sparseoffset > SIZE_MAX) { |
| fprintf(stderr, "size_t overflow when setting up map\n"); |
| anyfail(); |
| } |
| mapsize = (size_t) (statbuf.st_size - sparseoffset); |
| |
| mappages = roundup(mapsize, pagesize) / pagesize; |
| |
| for (i = 0; i < mappages; i++) { |
| cnt = read(fd, readbuf, pagesize); |
| if (cnt == -1) { |
| perror("read error"); |
| /***** LTP Port *****/ |
| local_flag = FAILED; |
| anyfail(); |
| /***** ** *****/ |
| return 0; |
| } else if (cnt != pagesize) { |
| /* |
| * Okay if at last page in file... |
| */ |
| if ((i * pagesize) + cnt != mapsize) { |
| (void)fprintf(stderr, "read %d of %ld bytes\n", |
| (i * pagesize) + cnt, |
| (long)mapsize); |
| return 0; |
| } |
| } |
| /* |
| * Compare read bytes of data. |
| */ |
| for (j = 0; j < cnt; j++) { |
| if (expbuf[j] != readbuf[j]) { |
| (void)fprintf(stderr, |
| "read bad data: exp %c got %c)", |
| expbuf[j], readbuf[j]); |
| #ifdef LARGE_FILE |
| (void)fprintf(stderr, ", pg %d off %d, " |
| "(fsize %Ld)\n", i, j, |
| statbuf.st_size); |
| #else /* LARGE_FILE */ |
| (void)fprintf(stderr, ", pg %d off %d, " |
| "(fsize %ld)\n", i, j, |
| statbuf.st_size); |
| #endif /* LARGE_FILE */ |
| return 0; |
| } |
| } |
| } |
| close(fd); |
| |
| return 1; |
| } |
| |
| /*ARGSUSED*/ void finish(int sig) |
| { |
| finished++; |
| return; |
| } |
| |
| unsigned int initrand(void) |
| { |
| unsigned int seed; |
| |
| /* |
| * Initialize random seed... Got this from a test written |
| * by scooter: |
| * Use srand/rand to diffuse the information from the |
| * time and pid. If you start several processes, then |
| * the time and pid information don't provide much |
| * variation. |
| */ |
| srand((unsigned int)getpid()); |
| seed = rand(); |
| srand((unsigned int)time(NULL)); |
| seed = (seed ^ rand()) % 100000; |
| srand48((long int)seed); |
| return (seed); |
| } |
| |
| /***** LTP Port *****/ |
| void ok_exit() |
| { |
| tst_resm(TPASS, "Test passed"); |
| tst_rmdir(); |
| tst_exit(); |
| } |
| |
| int anyfail() |
| { |
| tst_brkm(TFAIL, tst_rmdir, "Test failed"); |
| } |
| |
| /***** ** ** *****/ |