1/* 2 * Check that fault injection works properly. 3 * 4 * Copyright (c) 2016 Dmitry V. Levin <ldv@altlinux.org> 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. The name of the author may not be used to endorse or promote products 16 * derived from this software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 */ 29 30#include "tests.h" 31 32#include <assert.h> 33#include <errno.h> 34#include <stdio.h> 35#include <stdlib.h> 36#include <string.h> 37#include <unistd.h> 38#include <sys/stat.h> 39#include <sys/uio.h> 40 41static const int expfd = 4; 42static const int gotfd = 5; 43 44#define DEFAULT_ERRNO ENOSYS 45 46static const char *errstr; 47static int is_raw, err, first, step, iter, try; 48 49static void 50invoke(int fail) 51{ 52 static char buf[sizeof(int) * 3 + 3]; 53 const struct iovec io = { 54 .iov_base = buf, 55 .iov_len = sprintf(buf, "%d.", ++try) 56 }; 57 int rc; 58 59 if (!fail) { 60 rc = write(expfd, io.iov_base, io.iov_len); 61 if (rc != (int) io.iov_len) 62 perror_msg_and_fail("write"); 63 } 64 65 errno = 0; 66 rc = writev(gotfd, &io, 1); 67 68 if (fail) { 69 if (!(rc == -1 && errno == err)) 70 perror_msg_and_fail("expected errno %d" 71 ", got rc == %d, errno == %d", 72 err, rc, errno); 73 74 if (is_raw) 75 tprintf("writev(%#x, %p, 0x1) = -1 (errno %d)" 76 " (INJECTED)\n", gotfd, &io, err); 77 else 78 tprintf("writev(%d, [{iov_base=\"%s\", iov_len=%d}], 1)" 79 " = -1 %s (%m) (INJECTED)\n", 80 gotfd, buf, (int) io.iov_len, errstr); 81 } else { 82 if (rc != (int) io.iov_len) 83 perror_msg_and_fail("expected %d" 84 ", got rc == %d, errno == %d", 85 (int) io.iov_len, rc, errno); 86 87 if (is_raw) 88 tprintf("writev(%#x, %p, 0x1) = %#x\n", gotfd, &io, rc); 89 else 90 tprintf("writev(%d, [{iov_base=\"%s\", iov_len=%d}], 1)" 91 " = %d\n", 92 gotfd, buf, (int) io.iov_len, (int) io.iov_len); 93 } 94} 95 96int 97main(int argc, char *argv[]) 98{ 99 struct stat st; 100 101 assert(fstat(expfd, &st) == 0); 102 assert(fstat(gotfd, &st) == 0); 103 104 assert(argc == 6); 105 106 is_raw = !strcmp("raw", argv[1]); 107 108 errstr = argv[2]; 109 err = atoi(errstr); 110 assert(err >= 0); 111 112 if (!err) { 113 if (!*errstr) 114 err = DEFAULT_ERRNO; 115 else if (!strcasecmp(errstr, "EINVAL")) 116 err = EINVAL; 117 else 118 err = ENOSYS; 119 } 120 121 errno = err; 122 errstr = errno2name(); 123 124 first = atoi(argv[3]); 125 step = atoi(argv[4]); 126 iter = atoi(argv[5]); 127 128 assert(first > 0); 129 assert(step >= 0); 130 131 tprintf("%s", ""); 132 133 int i; 134 for (i = 1; i <= iter; ++i) { 135 int fail = 0; 136 if (first > 0) { 137 --first; 138 if (first == 0) { 139 fail = 1; 140 first = step; 141 } 142 } 143 invoke(fail); 144 } 145 146 tprintf("%s\n", "+++ exited with 0 +++"); 147 return 0; 148} 149