diff options
author | Jianfeng Tan <henry.tjf@antfin.com> | 2019-11-18 06:59:50 +0000 |
---|---|---|
committer | Jianfeng Tan <henry.tjf@antfin.com> | 2020-03-05 01:31:33 +0800 |
commit | 78c896b3b3127515478090c19447e27dc406427e (patch) | |
tree | d6d67d4683e9ca0409f9984a834547a572fb5310 /test/packetdrill/fmemopen.c | |
parent | e4380f4866091fd92a7a57667dd938a99144f9cd (diff) |
TLDKv2dev-next-socket
Signed-off-by: Jianfeng Tan <henry.tjf@antfin.com>
Signed-off-by: Jielong Zhou <jielong.zjl@antfin.com>
Signed-off-by: Jian Zhang <wuzai.zj@antfin.com>
Signed-off-by: Chen Zhao <winters.zc@antfin.com>
Change-Id: I55c39de4c6cd30f991f35631eb507f770230f08e
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 */ |