2 Copyright (C) 2013 Pietro Cerutti <gahr@FreeBSD.org>
4 Redistribution and use in source and binary forms, with or without
5 modification, are permitted provided that the following conditions
7 1. Redistributions of source code must retain the above copyright
8 notice, this list of conditions and the following disclaimer.
9 2. Redistributions in binary form must reproduce the above copyright
10 notice, this list of conditions and the following disclaimer in the
11 documentation and/or other materials provided with the distribution.
13 THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND
14 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
16 ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
17 FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
18 DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
19 OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
20 HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
21 LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
22 OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * Test basic FILE * functions (fread, fwrite, fseek, fclose) against
28 * a FILE * retrieved using fmemopen()
31 #include <sys/cdefs.h>
32 __FBSDID("$FreeBSD$");
44 * Use a pre-existing buffer.
49 char str[] = "Test writing some stuff";
50 char str2[] = "AAAAAAAAA";
51 char str3[] = "AAAA writing some stuff";
56 /* Open a FILE * using fmemopen. */
57 fp = fmemopen(buf, sizeof(buf), "w");
60 /* Write to the buffer. */
61 nofw = fwrite(str, 1, sizeof(str), fp);
62 assert(nofw == sizeof(str));
64 /* Close the FILE *. */
68 /* Re-open the FILE * to read back the data. */
69 fp = fmemopen(buf, sizeof(buf), "r");
72 /* Read from the buffer. */
73 bzero(buf2, sizeof(buf2));
74 nofr = fread(buf2, 1, sizeof(buf2), fp);
75 assert(nofr == sizeof(buf2));
78 * Since a write on a FILE * retrieved by fmemopen
79 * will add a '\0' (if there's space), we can check
80 * the strings for equality.
82 assert(strcmp(str, buf2) == 0);
84 /* Close the FILE *. */
88 /* Now open a FILE * on the first 4 bytes of the string. */
89 fp = fmemopen(str, 4, "w");
93 * Try to write more bytes than we shoud, we'll get a short count (4).
95 nofw = fwrite(str2, 1, sizeof(str2), fp);
98 /* Close the FILE *. */
101 /* Check that the string was not modified after the first 4 bytes. */
102 assert(strcmp(str, str3) == 0);
109 * Let fmemopen allocate the buffer.
112 char str[] = "A quick test";
115 size_t nofw, nofr, i;
118 /* Open a FILE * using fmemopen. */
119 fp = fmemopen(NULL, 512, "w+");
122 /* fill the buffer */
123 for (i = 0; i < 512; i++) {
124 nofw = fwrite("a", 1, 1, fp);
128 /* Get the current position into the stream. */
133 * Try to write past the end, we should get a short object count (0)
135 nofw = fwrite("a", 1, 1, fp);
138 /* Close the FILE *. */
147 * Here we test that a read operation doesn't go past the end of the
148 * data actually written, and that a SEEK_END seeks from the end of the
149 * data, not of the whole buffer.
152 char buf[512] = {'\0'};
153 char str[] = "Test data length. ";
154 char str2[] = "Do we have two sentences?";
155 char str3[sizeof(str) + sizeof(str2) -1];
160 /* Open a FILE * for updating our buffer. */
161 fp = fmemopen(buf, sizeof(buf), "w+");
164 /* Write our string into the buffer. */
165 nofw = fwrite(str, 1, sizeof(str), fp);
166 assert(nofw == sizeof(str));
169 * Now seek to the end and check that ftell
170 * gives us sizeof(str).
172 rc = fseek(fp, 0, SEEK_END);
175 assert(pos == sizeof(str));
177 /* Close the FILE *. */
181 /* Reopen the buffer for appending. */
182 fp = fmemopen(buf, sizeof(buf), "a+");
185 /* We should now be writing after the first string. */
186 nofw = fwrite(str2, 1, sizeof(str2), fp);
187 assert(nofw == sizeof(str2));
189 /* Rewind the FILE *. */
190 rc = fseek(fp, 0, SEEK_SET);
193 /* Make sure we're at the beginning. */
197 /* Read the whole buffer. */
198 nofr = fread(str3, 1, sizeof(buf), fp);
199 assert(nofr == sizeof(str3));
201 /* Make sure the two strings are there. */
202 assert(strncmp(str3, str, sizeof(str) - 1) == 0);
203 assert(strncmp(str3 + sizeof(str) - 1, str2, sizeof(str2)) == 0);
205 /* Close the FILE *. */
214 * Make sure that NULL bytes are never appended when opening a buffer
224 /* Pre-fill the buffer. */
225 memset(buf, 'A', sizeof(buf));
227 /* Open a FILE * in binary mode. */
228 fp = fmemopen(buf, sizeof(buf), "w+b");
231 /* Write some data into it. */
232 nofw = fwrite(str, 1, strlen(str), fp);
233 assert(nofw == strlen(str));
235 /* Make sure that the buffer doesn't contain any NULL bytes. */
236 for (i = 0; i < sizeof(buf); i++)
237 assert(buf[i] != '\0');
239 /* Close the FILE *. */