* libio/fmemopen.c (fmemopen): Free stream memory in case of
[platform/upstream/glibc.git] / libio / fmemopen.c
1 /* Fmemopen implementation.
2    Copyright (C) 2000, 2002, 2005, 2006 Free Software Foundation, Inc.
3    This file is part of the GNU C Library.
4    Contributed by  Hanno Mueller, kontakt@hanno.de, 2000.
5
6    The GNU C Library is free software; you can redistribute it and/or
7    modify it under the terms of the GNU Lesser General Public
8    License as published by the Free Software Foundation; either
9    version 2.1 of the License, or (at your option) any later version.
10
11    The GNU C Library 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 GNU
14    Lesser General Public License for more details.
15
16    You should have received a copy of the GNU Lesser General Public
17    License along with the GNU C Library; if not, write to the Free
18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19    02111-1307 USA.  */
20
21 /*
22  * fmemopen() - "my" version of a string stream
23  * Hanno Mueller, kontakt@hanno.de
24  *
25  *
26  * I needed fmemopen() for an application that I currently work on,
27  * but couldn't find it in libio. The following snippet of code is an
28  * attempt to implement what glibc's documentation describes.
29  *
30  *
31  *
32  * I already see some potential problems:
33  *
34  * - I never used the "original" fmemopen(). I am sure that "my"
35  *   fmemopen() behaves differently than the original version.
36  *
37  * - The documentation doesn't say wether a string stream allows
38  *   seeks. I checked the old fmemopen implementation in glibc's stdio
39  *   directory, wasn't quite able to see what is going on in that
40  *   source, but as far as I understand there was no seek there. For
41  *   my application, I needed fseek() and ftell(), so it's here.
42  *
43  * - "append" mode and fseek(p, SEEK_END) have two different ideas
44  *   about the "end" of the stream.
45  *
46  *   As described in the documentation, when opening the file in
47  *   "append" mode, the position pointer will be set to the first null
48  *   character of the string buffer (yet the buffer may already
49  *   contain more data). For fseek(), the last byte of the buffer is
50  *   used as the end of the stream.
51  *
52  * - It is unclear to me what the documentation tries to say when it
53  *   explains what happens when you use fmemopen with a NULL
54  *   buffer.
55  *
56  *   Quote: "fmemopen [then] allocates an array SIZE bytes long. This
57  *   is really only useful if you are going to write things to the
58  *   buffer and then read them back in again."
59  *
60  *   What does that mean if the original fmemopen() did not allow
61  *   seeking? How do you read what you just wrote without seeking back
62  *   to the beginning of the stream?
63  *
64  * - I think there should be a second version of fmemopen() that does
65  *   not add null characters for each write. (At least in my
66  *   application, I am not actually using strings but binary data and
67  *   so I don't need the stream to add null characters on its own.)
68  */
69
70 #include <errno.h>
71 #include <libio.h>
72 #include <stdio.h>
73 #include <stdlib.h>
74 #include <stdint.h>
75 #include <string.h>
76 #include <sys/types.h>
77 #include "libioP.h"
78
79
80 typedef struct fmemopen_cookie_struct fmemopen_cookie_t;
81 struct fmemopen_cookie_struct
82 {
83   char *buffer;
84   int mybuffer;
85   size_t size;
86   _IO_off64_t pos;
87   size_t maxpos;
88 };
89
90
91 static ssize_t
92 fmemopen_read (void *cookie, char *b, size_t s)
93 {
94   fmemopen_cookie_t *c;
95
96   c = (fmemopen_cookie_t *) cookie;
97
98   if (c->pos + s > c->size)
99     {
100       if ((size_t) c->pos == c->size)
101         return 0;
102       s = c->size - c->pos;
103     }
104
105   memcpy (b, &(c->buffer[c->pos]), s);
106
107   c->pos += s;
108   if ((size_t) c->pos > c->maxpos)
109     c->maxpos = c->pos;
110
111   return s;
112 }
113
114
115 static ssize_t
116 fmemopen_write (void *cookie, const char *b, size_t s)
117 {
118   fmemopen_cookie_t *c;
119   int addnullc;
120
121   c = (fmemopen_cookie_t *) cookie;
122
123   addnullc = s == 0 || b[s - 1] != '\0';
124
125   if (c->pos + s + addnullc > c->size)
126     {
127       if ((size_t) (c->pos + addnullc) == c->size)
128         {
129           __set_errno (ENOSPC);
130           return 0;
131         }
132       s = c->size - c->pos - addnullc;
133     }
134
135   memcpy (&(c->buffer[c->pos]), b, s);
136
137   c->pos += s;
138   if ((size_t) c->pos > c->maxpos)
139     {
140       c->maxpos = c->pos;
141       if (addnullc)
142         c->buffer[c->maxpos] = '\0';
143     }
144
145   return s;
146 }
147
148
149 static int
150 fmemopen_seek (void *cookie, _IO_off64_t *p, int w)
151 {
152   _IO_off64_t np;
153   fmemopen_cookie_t *c;
154
155   c = (fmemopen_cookie_t *) cookie;
156
157   switch (w)
158     {
159     case SEEK_SET:
160       np = *p;
161       break;
162
163     case SEEK_CUR:
164       np = c->pos + *p;
165       break;
166
167     case SEEK_END:
168       np = c->maxpos - *p;
169       break;
170
171     default:
172       return -1;
173     }
174
175   if (np < 0 || (size_t) np > c->size)
176     return -1;
177
178   *p = c->pos = np;
179
180   return 0;
181 }
182
183
184 static int
185 fmemopen_close (void *cookie)
186 {
187   fmemopen_cookie_t *c;
188
189   c = (fmemopen_cookie_t *) cookie;
190
191   if (c->mybuffer)
192     free (c->buffer);
193   free (c);
194
195   return 0;
196 }
197
198
199 FILE *
200 fmemopen (void *buf, size_t len, const char *mode)
201 {
202   cookie_io_functions_t iof;
203   fmemopen_cookie_t *c;
204
205   if (__builtin_expect (len == 0, 0))
206     {
207     einval:
208       __set_errno (EINVAL);
209       return NULL;
210     }
211
212   c = (fmemopen_cookie_t *) malloc (sizeof (fmemopen_cookie_t));
213   if (c == NULL)
214     return NULL;
215
216   c->mybuffer = (buf == NULL);
217
218   if (c->mybuffer)
219     {
220       c->buffer = (char *) malloc (len);
221       if (c->buffer == NULL)
222         {
223           free (c);
224           return NULL;
225         }
226       c->buffer[0] = '\0';
227     }
228   else
229     {
230       if (__builtin_expect ((uintptr_t) len > -(uintptr_t) buf, 0))
231         {
232           free (c);
233           goto einval;
234         }
235
236       c->buffer = buf;
237     }
238
239   c->size = len;
240
241   if (mode[0] == 'w')
242     c->buffer[0] = '\0';
243
244   c->maxpos = strlen (c->buffer);
245
246   if (mode[0] == 'a')
247     c->pos = c->maxpos;
248   else
249     c->pos = 0;
250
251   iof.read = fmemopen_read;
252   iof.write = fmemopen_write;
253   iof.seek = fmemopen_seek;
254   iof.close = fmemopen_close;
255
256   return _IO_fopencookie (c, mode, iof);
257 }