1 /* 2 * Based on the OpenBSD test 3 * Copyright (c) 2011 Martin Pieuchot <mpi@openbsd.org> 4 * 5 * Permission to use, copy, modify, and distribute this software for any 6 * purpose with or without fee is hereby granted, provided that the above 7 * copyright notice and this permission notice appear in all copies. 8 * 9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 16 */ 17 18 #include <sys/cdefs.h> 19 __RCSID("$NetBSD: t_open_memstream.c,v 1.2 2014/10/19 11:17:43 justin Exp $"); 20 21 #include <atf-c.h> 22 #include <err.h> 23 #include <stdio.h> 24 #include <stdlib.h> 25 #include <string.h> 26 #include <unistd.h> 27 28 ATF_TC(test_open_memstream); 29 ATF_TC_HEAD(test_open_memstream, tc) 30 { 31 atf_tc_set_md_var(tc, "descr", "Test open_memstream functionality"); 32 } 33 34 #define OFFSET 16384 35 36 const char start[] = "start"; 37 const char hello[] = "hello"; 38 39 ATF_TC_BODY(test_open_memstream, tc) 40 { 41 FILE *fp; 42 char *buf = (char *)0xff; 43 size_t size = 0; 44 off_t off; 45 int i; 46 47 fp = open_memstream(&buf, &size); 48 ATF_REQUIRE(fp != NULL); 49 50 off = ftello(fp); 51 ATF_CHECK(off == 0); 52 53 ATF_CHECK(fflush(fp) == 0); 54 ATF_CHECK(size == 0); 55 ATF_CHECK(buf != (char *)0xff); 56 ATF_CHECK(fseek(fp, -6, SEEK_SET) == -1); 57 ATF_CHECK(fseek(fp, OFFSET, SEEK_SET) == 0); 58 ATF_CHECK(fprintf(fp, hello) != EOF); 59 ATF_CHECK(fflush(fp) != EOF); 60 ATF_CHECK(size == OFFSET + sizeof(hello)-1); 61 ATF_CHECK(fseek(fp, 0, SEEK_SET) == 0); 62 ATF_CHECK(fprintf(fp, start) != EOF); 63 ATF_CHECK(fflush(fp) != EOF); 64 ATF_CHECK(size == sizeof(start)-1); 65 66 /* Needed for sparse files */ 67 ATF_CHECK(strncmp(buf, start, sizeof(start)-1) == 0); 68 for (i = sizeof(start)-1; i < OFFSET; i++) 69 ATF_CHECK(buf[i] == '\0'); 70 71 ATF_CHECK(memcmp(buf + OFFSET, hello, sizeof(hello)-1) == 0); 72 73 /* verify that simply seeking past the end doesn't increase the size */ 74 ATF_CHECK(fseek(fp, 100, SEEK_END) == 0); 75 ATF_CHECK(fflush(fp) != EOF); 76 ATF_CHECK(size == OFFSET + sizeof(hello)-1); 77 ATF_CHECK(fseek(fp, 8, SEEK_SET) == 0); 78 ATF_CHECK(ftell(fp) == 8); 79 80 /* Try to seek backward */ 81 ATF_CHECK(fseek(fp, -1, SEEK_CUR) == 0); 82 ATF_CHECK(ftell(fp) == 7); 83 ATF_CHECK(fseek(fp, 5, SEEK_CUR) == 0); 84 ATF_CHECK(fclose(fp) != EOF); 85 ATF_CHECK(size == 12); 86 87 free(buf); 88 } 89 90 ATF_TP_ADD_TCS(tp) 91 { 92 93 ATF_TP_ADD_TC(tp, test_open_memstream); 94 95 return atf_no_error(); 96 } 97