Annotation of qemu/tests/linux-test.c, revision 1.1.1.5

1.1       root        1: /*
                      2:  *  linux and CPU test
1.1.1.2   root        3:  *
1.1       root        4:  *  Copyright (c) 2003 Fabrice Bellard
                      5:  *
                      6:  *  This program is free software; you can redistribute it and/or modify
                      7:  *  it under the terms of the GNU General Public License as published by
                      8:  *  the Free Software Foundation; either version 2 of the License, or
                      9:  *  (at your option) any later version.
                     10:  *
                     11:  *  This program is distributed in the hope that it will be useful,
                     12:  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
                     13:  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
                     14:  *  GNU General Public License for more details.
                     15:  *
                     16:  *  You should have received a copy of the GNU General Public License
1.1.1.4   root       17:  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
1.1       root       18:  */
                     19: #include <stdarg.h>
                     20: #include <stdlib.h>
                     21: #include <stdio.h>
                     22: #include <unistd.h>
                     23: #include <fcntl.h>
                     24: #include <inttypes.h>
                     25: #include <string.h>
                     26: #include <sys/types.h>
                     27: #include <sys/stat.h>
                     28: #include <sys/wait.h>
                     29: #include <errno.h>
                     30: #include <utime.h>
                     31: #include <time.h>
                     32: #include <sys/time.h>
                     33: #include <sys/uio.h>
                     34: #include <sys/socket.h>
                     35: #include <netinet/in.h>
                     36: #include <arpa/inet.h>
                     37: #include <sched.h>
                     38: #include <dirent.h>
                     39: #include <setjmp.h>
                     40: #include <sys/shm.h>
                     41: 
                     42: #define TESTPATH "/tmp/linux-test.tmp"
                     43: #define TESTPORT 7654
                     44: #define STACK_SIZE 16384
                     45: 
                     46: void error1(const char *filename, int line, const char *fmt, ...)
                     47: {
                     48:     va_list ap;
                     49:     va_start(ap, fmt);
                     50:     fprintf(stderr, "%s:%d: ", filename, line);
                     51:     vfprintf(stderr, fmt, ap);
                     52:     fprintf(stderr, "\n");
                     53:     va_end(ap);
                     54:     exit(1);
                     55: }
                     56: 
                     57: int __chk_error(const char *filename, int line, int ret)
                     58: {
                     59:     if (ret < 0) {
1.1.1.2   root       60:         error1(filename, line, "%m (ret=%d, errno=%d)",
1.1       root       61:                ret, errno);
                     62:     }
                     63:     return ret;
                     64: }
                     65: 
1.1.1.4   root       66: #define error(fmt, ...) error1(__FILE__, __LINE__, fmt, ## __VA_ARGS__)
1.1       root       67: 
                     68: #define chk_error(ret) __chk_error(__FILE__, __LINE__, (ret))
                     69: 
                     70: /*******************************************************/
                     71: 
                     72: #define FILE_BUF_SIZE 300
                     73: 
                     74: void test_file(void)
                     75: {
                     76:     int fd, i, len, ret;
                     77:     uint8_t buf[FILE_BUF_SIZE];
                     78:     uint8_t buf2[FILE_BUF_SIZE];
                     79:     uint8_t buf3[FILE_BUF_SIZE];
                     80:     char cur_dir[1024];
                     81:     struct stat st;
                     82:     struct utimbuf tbuf;
                     83:     struct iovec vecs[2];
                     84:     DIR *dir;
                     85:     struct dirent *de;
                     86: 
                     87:     /* clean up, just in case */
                     88:     unlink(TESTPATH "/file1");
                     89:     unlink(TESTPATH "/file2");
                     90:     unlink(TESTPATH "/file3");
                     91:     rmdir(TESTPATH);
                     92: 
                     93:     if (getcwd(cur_dir, sizeof(cur_dir)) == NULL)
                     94:         error("getcwd");
1.1.1.2   root       95: 
1.1       root       96:     chk_error(mkdir(TESTPATH, 0755));
1.1.1.2   root       97: 
1.1       root       98:     chk_error(chdir(TESTPATH));
1.1.1.2   root       99: 
1.1       root      100:     /* open/read/write/close/readv/writev/lseek */
                    101: 
                    102:     fd = chk_error(open("file1", O_WRONLY | O_TRUNC | O_CREAT, 0644));
                    103:     for(i=0;i < FILE_BUF_SIZE; i++)
                    104:         buf[i] = i;
                    105:     len = chk_error(write(fd, buf, FILE_BUF_SIZE / 2));
                    106:     if (len != (FILE_BUF_SIZE / 2))
                    107:         error("write");
                    108:     vecs[0].iov_base = buf + (FILE_BUF_SIZE / 2);
                    109:     vecs[0].iov_len = 16;
                    110:     vecs[1].iov_base = buf + (FILE_BUF_SIZE / 2) + 16;
                    111:     vecs[1].iov_len = (FILE_BUF_SIZE / 2) - 16;
                    112:     len = chk_error(writev(fd, vecs, 2));
                    113:     if (len != (FILE_BUF_SIZE / 2))
                    114:      error("writev");
                    115:     chk_error(close(fd));
                    116: 
                    117:     chk_error(rename("file1", "file2"));
                    118: 
                    119:     fd = chk_error(open("file2", O_RDONLY));
                    120: 
                    121:     len = chk_error(read(fd, buf2, FILE_BUF_SIZE));
                    122:     if (len != FILE_BUF_SIZE)
                    123:         error("read");
                    124:     if (memcmp(buf, buf2, FILE_BUF_SIZE) != 0)
                    125:         error("memcmp");
1.1.1.2   root      126: 
1.1       root      127: #define FOFFSET 16
                    128:     ret = chk_error(lseek(fd, FOFFSET, SEEK_SET));
                    129:     if (ret != 16)
                    130:         error("lseek");
                    131:     vecs[0].iov_base = buf3;
                    132:     vecs[0].iov_len = 32;
                    133:     vecs[1].iov_base = buf3 + 32;
                    134:     vecs[1].iov_len = FILE_BUF_SIZE - FOFFSET - 32;
                    135:     len = chk_error(readv(fd, vecs, 2));
                    136:     if (len != FILE_BUF_SIZE - FOFFSET)
                    137:         error("readv");
                    138:     if (memcmp(buf + FOFFSET, buf3, FILE_BUF_SIZE - FOFFSET) != 0)
                    139:         error("memcmp");
1.1.1.2   root      140: 
1.1       root      141:     chk_error(close(fd));
                    142: 
                    143:     /* access */
                    144:     chk_error(access("file2", R_OK));
                    145: 
                    146:     /* stat/chmod/utime/truncate */
                    147: 
                    148:     chk_error(chmod("file2", 0600));
                    149:     tbuf.actime = 1001;
                    150:     tbuf.modtime = 1000;
                    151:     chk_error(truncate("file2", 100));
                    152:     chk_error(utime("file2", &tbuf));
                    153:     chk_error(stat("file2", &st));
                    154:     if (st.st_size != 100)
                    155:         error("stat size");
                    156:     if (!S_ISREG(st.st_mode))
                    157:         error("stat mode");
                    158:     if ((st.st_mode & 0777) != 0600)
                    159:         error("stat mode2");
                    160:     if (st.st_atime != 1001 ||
                    161:         st.st_mtime != 1000)
                    162:         error("stat time");
                    163: 
                    164:     chk_error(stat(TESTPATH, &st));
                    165:     if (!S_ISDIR(st.st_mode))
                    166:         error("stat mode");
                    167: 
                    168:     /* fstat */
                    169:     fd = chk_error(open("file2", O_RDWR));
                    170:     chk_error(ftruncate(fd, 50));
                    171:     chk_error(fstat(fd, &st));
                    172:     chk_error(close(fd));
1.1.1.2   root      173: 
1.1       root      174:     if (st.st_size != 50)
                    175:         error("stat size");
                    176:     if (!S_ISREG(st.st_mode))
                    177:         error("stat mode");
1.1.1.2   root      178: 
1.1       root      179:     /* symlink/lstat */
                    180:     chk_error(symlink("file2", "file3"));
                    181:     chk_error(lstat("file3", &st));
                    182:     if (!S_ISLNK(st.st_mode))
                    183:         error("stat mode");
1.1.1.2   root      184: 
1.1       root      185:     /* getdents */
                    186:     dir = opendir(TESTPATH);
                    187:     if (!dir)
                    188:         error("opendir");
                    189:     len = 0;
                    190:     for(;;) {
                    191:         de = readdir(dir);
                    192:         if (!de)
                    193:             break;
                    194:         if (strcmp(de->d_name, ".") != 0 &&
                    195:             strcmp(de->d_name, "..") != 0 &&
                    196:             strcmp(de->d_name, "file2") != 0 &&
                    197:             strcmp(de->d_name, "file3") != 0)
                    198:             error("readdir");
                    199:         len++;
                    200:     }
                    201:     closedir(dir);
                    202:     if (len != 4)
                    203:         error("readdir");
                    204: 
                    205:     chk_error(unlink("file3"));
                    206:     chk_error(unlink("file2"));
                    207:     chk_error(chdir(cur_dir));
                    208:     chk_error(rmdir(TESTPATH));
                    209: }
                    210: 
                    211: void test_fork(void)
                    212: {
                    213:     int pid, status;
                    214: 
                    215:     pid = chk_error(fork());
                    216:     if (pid == 0) {
                    217:         /* child */
                    218:         exit(2);
                    219:     }
                    220:     chk_error(waitpid(pid, &status, 0));
                    221:     if (!WIFEXITED(status) || WEXITSTATUS(status) != 2)
                    222:         error("waitpid status=0x%x", status);
                    223: }
                    224: 
                    225: void test_time(void)
                    226: {
                    227:     struct timeval tv, tv2;
                    228:     struct timespec ts, rem;
                    229:     struct rusage rusg1, rusg2;
                    230:     int ti, i;
                    231: 
                    232:     chk_error(gettimeofday(&tv, NULL));
                    233:     rem.tv_sec = 1;
                    234:     ts.tv_sec = 0;
                    235:     ts.tv_nsec = 20 * 1000000;
                    236:     chk_error(nanosleep(&ts, &rem));
                    237:     if (rem.tv_sec != 1)
                    238:         error("nanosleep");
                    239:     chk_error(gettimeofday(&tv2, NULL));
                    240:     ti = tv2.tv_sec - tv.tv_sec;
                    241:     if (ti >= 2)
                    242:         error("gettimeofday");
1.1.1.2   root      243: 
1.1       root      244:     chk_error(getrusage(RUSAGE_SELF, &rusg1));
                    245:     for(i = 0;i < 10000; i++);
                    246:     chk_error(getrusage(RUSAGE_SELF, &rusg2));
                    247:     if ((rusg2.ru_utime.tv_sec - rusg1.ru_utime.tv_sec) < 0 ||
                    248:         (rusg2.ru_stime.tv_sec - rusg1.ru_stime.tv_sec) < 0)
                    249:         error("getrusage");
                    250: }
                    251: 
                    252: void pstrcpy(char *buf, int buf_size, const char *str)
                    253: {
                    254:     int c;
                    255:     char *q = buf;
                    256: 
                    257:     if (buf_size <= 0)
                    258:         return;
                    259: 
                    260:     for(;;) {
                    261:         c = *str++;
                    262:         if (c == 0 || q >= buf + buf_size - 1)
                    263:             break;
                    264:         *q++ = c;
                    265:     }
                    266:     *q = '\0';
                    267: }
                    268: 
                    269: /* strcat and truncate. */
                    270: char *pstrcat(char *buf, int buf_size, const char *s)
                    271: {
                    272:     int len;
                    273:     len = strlen(buf);
1.1.1.2   root      274:     if (len < buf_size)
1.1       root      275:         pstrcpy(buf + len, buf_size - len, s);
                    276:     return buf;
                    277: }
                    278: 
                    279: int server_socket(void)
                    280: {
                    281:     int val, fd;
                    282:     struct sockaddr_in sockaddr;
                    283: 
                    284:     /* server socket */
                    285:     fd = chk_error(socket(PF_INET, SOCK_STREAM, 0));
                    286: 
                    287:     val = 1;
                    288:     chk_error(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)));
                    289: 
                    290:     sockaddr.sin_family = AF_INET;
                    291:     sockaddr.sin_port = htons(TESTPORT);
                    292:     sockaddr.sin_addr.s_addr = 0;
                    293:     chk_error(bind(fd, (struct sockaddr *)&sockaddr, sizeof(sockaddr)));
                    294:     chk_error(listen(fd, 0));
                    295:     return fd;
                    296: 
                    297: }
                    298: 
                    299: int client_socket(void)
                    300: {
                    301:     int fd;
                    302:     struct sockaddr_in sockaddr;
                    303: 
                    304:     /* server socket */
                    305:     fd = chk_error(socket(PF_INET, SOCK_STREAM, 0));
                    306:     sockaddr.sin_family = AF_INET;
                    307:     sockaddr.sin_port = htons(TESTPORT);
                    308:     inet_aton("127.0.0.1", &sockaddr.sin_addr);
                    309:     chk_error(connect(fd, (struct sockaddr *)&sockaddr, sizeof(sockaddr)));
                    310:     return fd;
                    311: }
                    312: 
                    313: const char socket_msg[] = "hello socket\n";
                    314: 
                    315: void test_socket(void)
                    316: {
                    317:     int server_fd, client_fd, fd, pid, ret, val;
                    318:     struct sockaddr_in sockaddr;
                    319:     socklen_t len;
                    320:     char buf[512];
                    321: 
                    322:     server_fd = server_socket();
                    323: 
                    324:     /* test a few socket options */
                    325:     len = sizeof(val);
                    326:     chk_error(getsockopt(server_fd, SOL_SOCKET, SO_TYPE, &val, &len));
                    327:     if (val != SOCK_STREAM)
                    328:         error("getsockopt");
1.1.1.2   root      329: 
1.1       root      330:     pid = chk_error(fork());
                    331:     if (pid == 0) {
                    332:         client_fd = client_socket();
                    333:         send(client_fd, socket_msg, sizeof(socket_msg), 0);
                    334:         close(client_fd);
                    335:         exit(0);
                    336:     }
                    337:     len = sizeof(sockaddr);
                    338:     fd = chk_error(accept(server_fd, (struct sockaddr *)&sockaddr, &len));
                    339: 
                    340:     ret = chk_error(recv(fd, buf, sizeof(buf), 0));
                    341:     if (ret != sizeof(socket_msg))
                    342:         error("recv");
                    343:     if (memcmp(buf, socket_msg, sizeof(socket_msg)) != 0)
                    344:         error("socket_msg");
                    345:     chk_error(close(fd));
                    346:     chk_error(close(server_fd));
                    347: }
                    348: 
                    349: #define WCOUNT_MAX 512
                    350: 
                    351: void test_pipe(void)
                    352: {
                    353:     fd_set rfds, wfds;
                    354:     int fds[2], fd_max, ret;
                    355:     uint8_t ch;
                    356:     int wcount, rcount;
                    357: 
                    358:     chk_error(pipe(fds));
                    359:     chk_error(fcntl(fds[0], F_SETFL, O_NONBLOCK));
                    360:     chk_error(fcntl(fds[1], F_SETFL, O_NONBLOCK));
                    361:     wcount = 0;
                    362:     rcount = 0;
                    363:     for(;;) {
                    364:         FD_ZERO(&rfds);
                    365:         fd_max = fds[0];
                    366:         FD_SET(fds[0], &rfds);
                    367: 
                    368:         FD_ZERO(&wfds);
                    369:         FD_SET(fds[1], &wfds);
                    370:         if (fds[1] > fd_max)
                    371:             fd_max = fds[1];
                    372: 
                    373:         ret = chk_error(select(fd_max + 1, &rfds, &wfds, NULL, NULL));
                    374:         if (ret > 0) {
                    375:             if (FD_ISSET(fds[0], &rfds)) {
                    376:                 chk_error(read(fds[0], &ch, 1));
                    377:                 rcount++;
                    378:                 if (rcount >= WCOUNT_MAX)
                    379:                     break;
                    380:             }
                    381:             if (FD_ISSET(fds[1], &wfds)) {
                    382:                 ch = 'a';
                    383:                 chk_error(write(fds[0], &ch, 1));
                    384:                 wcount++;
                    385:             }
                    386:         }
                    387:     }
                    388:     chk_error(close(fds[0]));
                    389:     chk_error(close(fds[1]));
                    390: }
                    391: 
                    392: int thread1_res;
                    393: int thread2_res;
                    394: 
                    395: int thread1_func(void *arg)
                    396: {
                    397:     int i;
                    398:     for(i=0;i<5;i++) {
                    399:         thread1_res++;
                    400:         usleep(10 * 1000);
                    401:     }
                    402:     return 0;
                    403: }
                    404: 
                    405: int thread2_func(void *arg)
                    406: {
                    407:     int i;
                    408:     for(i=0;i<6;i++) {
                    409:         thread2_res++;
                    410:         usleep(10 * 1000);
                    411:     }
                    412:     return 0;
                    413: }
                    414: 
                    415: void test_clone(void)
                    416: {
                    417:     uint8_t *stack1, *stack2;
                    418:     int pid1, pid2, status1, status2;
                    419: 
                    420:     stack1 = malloc(STACK_SIZE);
1.1.1.2   root      421:     pid1 = chk_error(clone(thread1_func, stack1 + STACK_SIZE,
1.1       root      422:                            CLONE_VM | CLONE_FS | CLONE_FILES | SIGCHLD, "hello1"));
                    423: 
                    424:     stack2 = malloc(STACK_SIZE);
1.1.1.2   root      425:     pid2 = chk_error(clone(thread2_func, stack2 + STACK_SIZE,
1.1       root      426:                            CLONE_VM | CLONE_FS | CLONE_FILES | SIGCHLD, "hello2"));
                    427: 
                    428:     while (waitpid(pid1, &status1, 0) != pid1);
1.1.1.5 ! root      429:     free(stack1);
1.1       root      430:     while (waitpid(pid2, &status2, 0) != pid2);
1.1.1.5 ! root      431:     free(stack2);
1.1       root      432:     if (thread1_res != 5 ||
                    433:         thread2_res != 6)
                    434:         error("clone");
                    435: }
                    436: 
                    437: /***********************************/
                    438: 
                    439: volatile int alarm_count;
                    440: jmp_buf jmp_env;
                    441: 
                    442: void sig_alarm(int sig)
                    443: {
                    444:     if (sig != SIGALRM)
                    445:         error("signal");
                    446:     alarm_count++;
                    447: }
                    448: 
                    449: void sig_segv(int sig, siginfo_t *info, void *puc)
                    450: {
                    451:     if (sig != SIGSEGV)
                    452:         error("signal");
                    453:     longjmp(jmp_env, 1);
                    454: }
                    455: 
                    456: void test_signal(void)
                    457: {
                    458:     struct sigaction act;
                    459:     struct itimerval it, oit;
                    460: 
                    461:     /* timer test */
                    462: 
                    463:     alarm_count = 0;
                    464: 
                    465:     act.sa_handler = sig_alarm;
                    466:     sigemptyset(&act.sa_mask);
                    467:     act.sa_flags = 0;
                    468:     chk_error(sigaction(SIGALRM, &act, NULL));
1.1.1.2   root      469: 
1.1       root      470:     it.it_interval.tv_sec = 0;
                    471:     it.it_interval.tv_usec = 10 * 1000;
                    472:     it.it_value.tv_sec = 0;
                    473:     it.it_value.tv_usec = 10 * 1000;
                    474:     chk_error(setitimer(ITIMER_REAL, &it, NULL));
                    475:     chk_error(getitimer(ITIMER_REAL, &oit));
                    476:     if (oit.it_value.tv_sec != it.it_value.tv_sec ||
                    477:         oit.it_value.tv_usec != it.it_value.tv_usec)
                    478:         error("itimer");
1.1.1.2   root      479: 
1.1       root      480:     while (alarm_count < 5) {
                    481:         usleep(10 * 1000);
                    482:     }
                    483: 
                    484:     it.it_interval.tv_sec = 0;
                    485:     it.it_interval.tv_usec = 0;
                    486:     it.it_value.tv_sec = 0;
                    487:     it.it_value.tv_usec = 0;
                    488:     memset(&oit, 0xff, sizeof(oit));
                    489:     chk_error(setitimer(ITIMER_REAL, &it, &oit));
                    490:     if (oit.it_value.tv_sec != 0 ||
                    491:         oit.it_value.tv_usec != 10 * 1000)
                    492:         error("setitimer");
                    493: 
                    494:     /* SIGSEGV test */
                    495:     act.sa_sigaction = sig_segv;
                    496:     sigemptyset(&act.sa_mask);
                    497:     act.sa_flags = SA_SIGINFO;
                    498:     chk_error(sigaction(SIGSEGV, &act, NULL));
                    499:     if (setjmp(jmp_env) == 0) {
                    500:         *(uint8_t *)0 = 0;
                    501:     }
1.1.1.2   root      502: 
1.1       root      503:     act.sa_handler = SIG_DFL;
                    504:     sigemptyset(&act.sa_mask);
                    505:     act.sa_flags = 0;
                    506:     chk_error(sigaction(SIGSEGV, &act, NULL));
                    507: }
                    508: 
                    509: #define SHM_SIZE 32768
                    510: 
                    511: void test_shm(void)
                    512: {
                    513:     void *ptr;
                    514:     int shmid;
                    515: 
                    516:     shmid = chk_error(shmget(IPC_PRIVATE, SHM_SIZE, IPC_CREAT | 0777));
                    517:     ptr = shmat(shmid, NULL, 0);
                    518:     if (!ptr)
                    519:         error("shmat");
                    520: 
                    521:     memset(ptr, 0, SHM_SIZE);
                    522: 
                    523:     chk_error(shmctl(shmid, IPC_RMID, 0));
                    524:     chk_error(shmdt(ptr));
                    525: }
                    526: 
                    527: int main(int argc, char **argv)
                    528: {
                    529:     test_file();
                    530:     test_fork();
                    531:     test_time();
                    532:     test_socket();
                    533:     //    test_clone();
                    534:     test_signal();
                    535:     test_shm();
                    536:     return 0;
                    537: }

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.