diff options
Diffstat (limited to 'libctf/testsuite/libctf-regression/zrewrite.c')
| -rw-r--r-- | libctf/testsuite/libctf-regression/zrewrite.c | 156 |
1 files changed, 156 insertions, 0 deletions
diff --git a/libctf/testsuite/libctf-regression/zrewrite.c b/libctf/testsuite/libctf-regression/zrewrite.c new file mode 100644 index 0000000..4d5d15e --- /dev/null +++ b/libctf/testsuite/libctf-regression/zrewrite.c @@ -0,0 +1,156 @@ +/* Make sure that you can modify then ctf_compress_write() a dict + and it changes after modification. */ + +#include <ctf-api.h> +#include <errno.h> +#include <fcntl.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + +char *read_file(const char *path, size_t *len) +{ + char *in = NULL; + char buf[4096]; + int foo; + size_t ret; + + if ((foo = open (path, O_RDONLY)) < 0) + { + fprintf (stderr, "error opening %s: %s\n", path, strerror(errno)); + exit (1); + } + + *len = 0; + while ((ret = read (foo, buf, 4096)) > 0) + { + if ((in = realloc (in, *len + ret)) == NULL) + { + fprintf (stderr, "Out of memory\n"); + exit (1); + } + + memcpy (&in[*len], buf, ret); + *len += ret; + } + + if (ret < 0) + { + fprintf (stderr, "error reading %s: %s\n", path, strerror(errno)); + exit (1); + } + close (foo); + return in; +} + +int +main (int argc, char *argv[]) +{ + ctf_dict_t *fp, *fp_b; + ctf_archive_t *ctf, *ctf_b; + int foo; + char *a, *b; + size_t a_len, b_len; + ctf_id_t type, ptrtype; + int err; + + if (argc != 2) + { + fprintf (stderr, "Syntax: %s PROGRAM\n", argv[0]); + exit(1); + } + + if ((ctf = ctf_open (argv[1], NULL, &err)) == NULL) + goto open_err; + if ((fp = ctf_dict_open (ctf, NULL, &err)) == NULL) + goto open_err; + + if ((foo = open ("tmpdir/one", O_CREAT | O_TRUNC | O_WRONLY, 0666)) < 0) + goto write_stderr; + if (ctf_compress_write (fp, foo) < 0) + goto write_err; + close (foo); + + if ((foo = open ("tmpdir/two", O_CREAT | O_TRUNC | O_WRONLY, 0666)) < 0) + goto write_stderr; + if (ctf_compress_write (fp, foo) < 0) + goto write_err; + close (foo); + + a = read_file ("tmpdir/one", &a_len); + b = read_file ("tmpdir/two", &b_len); + + if (a_len != b_len || memcmp (a, b, a_len) != 0) + { + fprintf (stderr, "consecutive compress_writes are different: lengths %i and %i\n", a_len, b_len); + return 1; + } + + free (b); + + /* Add some new types to the dict and write it out, then read it back in and + make sure they're still there, and that at least some of the + originally-present data objects are still there too. */ + + if ((type = ctf_lookup_by_name (fp, "struct a_struct")) == CTF_ERR) + fprintf (stderr, "Lookup of struct a_struct failed: %s\n", ctf_errmsg (ctf_errno (fp))); + + if ((ptrtype = ctf_add_pointer (fp, CTF_ADD_ROOT, type)) == CTF_ERR) + fprintf (stderr, "Cannot add pointer to ctf_opened dict: %s\n", ctf_errmsg (ctf_errno (fp))); + + unlink ("tmpdir/two"); + + if ((foo = open ("tmpdir/two", O_CREAT | O_TRUNC | O_WRONLY, 0666)) < 0) + goto write_stderr; + if (ctf_compress_write (fp, foo) < 0) + goto write_err; + close (foo); + + b = read_file ("tmpdir/two", &b_len); + + if (memcmp (a, b, b_len) == 0) + { + fprintf (stderr, "compress_writes after adding types does not change the dict\n"); + return 1; + } + + free (a); + free (b); + + if ((ctf_b = ctf_open ("tmpdir/two", NULL, &err)) == NULL) + goto open_err; + if ((fp_b = ctf_dict_open (ctf_b, NULL, &err)) == NULL) + goto open_err; + + if (ctf_type_reference (fp_b, ptrtype) == CTF_ERR) + fprintf (stderr, "Lookup of pointer preserved across writeout failed: %s\n", ctf_errmsg (ctf_errno (fp_b))); + + if (ctf_type_reference (fp_b, ptrtype) != type) + fprintf (stderr, "Look up of newly-added type in serialized dict yields ID %lx, expected %lx\n", ctf_type_reference (fp_b, ptrtype), type); + + if (ctf_lookup_by_symbol_name (fp_b, "an_int") == CTF_ERR) + fprintf (stderr, "Lookup of symbol an_int failed: %s\n", ctf_errmsg (ctf_errno (fp_b))); + + ctf_dict_close (fp); + ctf_close (ctf); + + ctf_dict_close (fp_b); + ctf_close (ctf_b); + + printf ("All done.\n"); + return 0; + + open_err: + fprintf (stderr, "%s: cannot open: %s\n", argv[0], ctf_errmsg (err)); + return 1; + write_err: + fprintf (stderr, "%s: cannot write: %s\n", argv[0], ctf_errmsg (ctf_errno (fp))); + return 1; + write_stderr: + fprintf (stderr, "%s: cannot open for writing: %s\n", argv[0], strerror (errno)); + return 1; + read_err: + fprintf (stderr, "%s: cannot read\n", argv[0]); + return 1; +} |