diff options
Diffstat (limited to 'test/packetdrill/fmemopen.c')
-rw-r--r-- | test/packetdrill/fmemopen.c | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/test/packetdrill/fmemopen.c b/test/packetdrill/fmemopen.c new file mode 100644 index 0000000..5ed2be5 --- /dev/null +++ b/test/packetdrill/fmemopen.c @@ -0,0 +1,81 @@ +/* + * Copyright 2013 Google Inc. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + * 02110-1301, USA. + */ +/* + * Author: ncardwell@google.com (Neal Cardwell) + * + * FreeBSD does not have an fmemopen(), so we roll our own minimalist + * implementation here. + */ + +#include "types.h" + +#include <stdio.h> +#include <stdlib.h> + +#include "assert.h" +#include "fmemopen.h" + +#if !defined(HAVE_FMEMOPEN) + +struct fmemopen_read_state { + char *next; /* the next byte to return */ + char *end; /* the byte after the end of the string */ +}; + +static int fmemopen_readfn(void *cookie, char *buf, int len) +{ + struct fmemopen_read_state *read_cookie = + (struct fmemopen_read_state *)cookie; + int bytes = 0; + + assert(read_cookie->next <= read_cookie->end); + if (read_cookie->next == read_cookie->end) + return 0; + + bytes = read_cookie->end - read_cookie->next; + if (len < bytes) + bytes = len; + + memcpy(buf, read_cookie->next, bytes); + read_cookie->next += bytes; + + return bytes; +} + +FILE *fmemopen(char *str, size_t size, const char *mode) +{ + FILE *f = NULL; + struct fmemopen_read_state *read_cookie; + + assert(strcmp(mode, "r") == 0); /* only support read for now */ + + read_cookie = calloc(1, sizeof(struct fmemopen_read_state)); + read_cookie->next = str; + read_cookie->end = str + size; + + f = fropen(read_cookie, fmemopen_readfn); + if (!f) { + free(read_cookie); + return NULL; + } + + return f; +} + +#endif /* HAVE_FMEMOPEN */ |