X-Git-Url: http://git.droids-corp.org/?p=protos%2Flibecoli.git;a=blobdiff_plain;f=lib%2Fmain.c;h=2a08cdd760926d1899e70338da0f3976d12b0845;hp=d3d35b36626c11a1f8c6505f8d67cc29fa098c2e;hb=39a8be563b6f1cb25bd9158bb7c48979f7b98290;hpb=89d4ea2128501a119d835fccbf2500076c69695d diff --git a/lib/main.c b/lib/main.c index d3d35b3..2a08cdd 100644 --- a/lib/main.c +++ b/lib/main.c @@ -1,77 +1,68 @@ -/* - * Copyright (c) 2016, Olivier MATZ - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are met: - * - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * * Neither the name of the University of California, Berkeley nor the - * names of its contributors may be used to endorse or promote products - * derived from this software without specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY - * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED - * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE - * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY - * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES - * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; - * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND - * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT - * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS - * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +/* SPDX-License-Identifier: BSD-3-Clause + * Copyright 2016, Olivier MATZ */ #include #include #include #include +#include #include +#include +#include #include #include #include +/* LCOV_EXCL_START */ +EC_LOG_TYPE_REGISTER(main); + #define COUNT_OF(x) ((sizeof(x)/sizeof(0[x])) / \ ((size_t)(!(sizeof(x) % sizeof(0[x]))))) static int log_level = EC_LOG_INFO; static int alloc_fail_proba = 0; +static int seed = 0; +static size_t alloc_success = 0; static const char ec_short_options[] = "h" /* help */ "l:" /* log-level */ "r:" /* random-alloc-fail */ + "s:" /* seed */ ; #define EC_OPT_HELP "help" #define EC_OPT_LOG_LEVEL "log-level" #define EC_OPT_RANDOM_ALLOC_FAIL "random-alloc-fail" +#define EC_OPT_SEED "seed" static const struct option ec_long_options[] = { {EC_OPT_HELP, 1, NULL, 'h'}, {EC_OPT_LOG_LEVEL, 1, NULL, 'l'}, {EC_OPT_RANDOM_ALLOC_FAIL, 1, NULL, 'r'}, + {EC_OPT_SEED, 1, NULL, 's'}, {NULL, 0, NULL, 0} }; static void usage(const char *prgname) { - printf("%s [options]\n" + printf("%s [options] [test1 test2 test3...]\n" " -h\n" " --"EC_OPT_HELP"\n" " Show this help.\n" " -l \n" " --"EC_OPT_LOG_LEVEL"=\n" - " Set log level (0 = no log, 6 = verbose).\n" + " Set log level (0 = no log, 7 = verbose).\n" " -r \n" " --"EC_OPT_RANDOM_ALLOC_FAIL"=\n" " Cause malloc to fail randomly. This helps to debug\n" " leaks or crashes in error cases. The probability is\n" " between 0 and 100.\n" + " -s \n" + " --seed=\n" + " Seeds the random number generator. Default is 0.\n" , prgname); } @@ -93,9 +84,9 @@ parse_int(const char *s, int min, int max, int *ret, unsigned int base) return 0; } -static void parse_args(int argc, char **argv) +static int parse_args(int argc, char **argv) { - int opt; + int ret, opt; while ((opt = getopt_long(argc, argv, ec_short_options, ec_long_options, NULL)) != EOF) { @@ -123,11 +114,24 @@ static void parse_args(int argc, char **argv) } break; + case 's': /* seed */ + if (parse_int(optarg, 0, INT_MAX, &seed, 10) < 0) { + printf("Invalid seed value\n"); + usage(argv[0]); + exit(1); + } + break; + default: usage(argv[0]); - exit(1); + return -1; } } + + ret = optind - 1; + optind = 1; + + return ret; } TAILQ_HEAD(debug_alloc_hdr_list, debug_alloc_hdr); @@ -149,15 +153,17 @@ struct debug_alloc_ftr { unsigned int cookie; } __attribute__((packed)); +static int malloc_seq; + static void *debug_malloc(size_t size, const char *file, unsigned int line) { struct debug_alloc_hdr *hdr; struct debug_alloc_ftr *ftr; size_t new_size = size + sizeof(*hdr) + sizeof(*ftr); void *ret; + int r = random(); - - if (alloc_fail_proba != 0 && (random() % 100) < alloc_fail_proba) + if (alloc_fail_proba != 0 && (r % 100) < alloc_fail_proba) hdr = NULL; else hdr = malloc(new_size); @@ -177,9 +183,11 @@ static void *debug_malloc(size_t size, const char *file, unsigned int line) ftr->cookie = 0x87654321; } - ec_log(EC_LOG_DEBUG, "%s:%d: info: malloc(%zd) -> %p\n", - file, line, size, ret); + EC_LOG(EC_LOG_DEBUG, "%s:%d: info: malloc(%zd) -> %p seq=%d\n", + file, line, size, ret, malloc_seq++); + if (ret) + alloc_success++; return ret; } @@ -191,21 +199,21 @@ static void debug_free(void *ptr, const char *file, unsigned int line) (void)file; (void)line; - ec_log(EC_LOG_DEBUG, "%s:%d: info: free(%p)\n", file, line, ptr); + EC_LOG(EC_LOG_DEBUG, "%s:%d: info: free(%p)\n", file, line, ptr); if (ptr == NULL) return; hdr = (ptr - sizeof(*hdr)); if (hdr->cookie != 0x12345678) { - ec_log(EC_LOG_ERR, "%s:%d: error: free(%p): bad start cookie\n", + EC_LOG(EC_LOG_ERR, "%s:%d: error: free(%p): bad start cookie\n", file, line, ptr); abort(); } ftr = (ptr + hdr->size); if (ftr->cookie != 0x87654321) { - ec_log(EC_LOG_ERR, "%s:%d: error: free(%p): bad end cookie\n", + EC_LOG(EC_LOG_ERR, "%s:%d: error: free(%p): bad end cookie\n", file, line, ptr); abort(); } @@ -216,7 +224,7 @@ static void debug_free(void *ptr, const char *file, unsigned int line) } if (h == NULL) { - ec_log(EC_LOG_ERR, "%s:%d: error: free(%p): bad ptr\n", + EC_LOG(EC_LOG_ERR, "%s:%d: error: free(%p): bad ptr\n", file, line, ptr); abort(); } @@ -236,7 +244,7 @@ static void *debug_realloc(void *ptr, size_t size, const char *file, if (ptr != NULL) { hdr = (ptr - sizeof(*hdr)); if (hdr->cookie != 0x12345678) { - ec_log(EC_LOG_ERR, + EC_LOG(EC_LOG_ERR, "%s:%d: error: realloc(%p): bad start cookie\n", file, line, ptr); abort(); @@ -244,7 +252,7 @@ static void *debug_realloc(void *ptr, size_t size, const char *file, ftr = (ptr + hdr->size); if (ftr->cookie != 0x87654321) { - ec_log(EC_LOG_ERR, + EC_LOG(EC_LOG_ERR, "%s:%d: error: realloc(%p): bad end cookie\n", file, line, ptr); abort(); @@ -256,7 +264,7 @@ static void *debug_realloc(void *ptr, size_t size, const char *file, } if (h == NULL) { - ec_log(EC_LOG_ERR, "%s:%d: error: realloc(%p): bad ptr\n", + EC_LOG(EC_LOG_ERR, "%s:%d: error: realloc(%p): bad ptr\n", file, line, ptr); abort(); } @@ -289,9 +297,11 @@ static void *debug_realloc(void *ptr, size_t size, const char *file, ftr->cookie = 0x87654321; } - ec_log(EC_LOG_DEBUG, "%s:%d: info: realloc(%p, %zd) -> %p\n", - file, line, ptr, size, ret); + EC_LOG(EC_LOG_DEBUG, "%s:%d: info: realloc(%p, %zd) -> %p seq=%d\n", + file, line, ptr, size, ret, malloc_seq++); + if (ret) + alloc_success++; return ret; } @@ -301,59 +311,82 @@ static int debug_alloc_dump_leaks(void) int i; char **buffer; + EC_LOG(EC_LOG_INFO, "%zd successful allocations\n", alloc_success); + if (TAILQ_EMPTY(&debug_alloc_hdr_list)) return 0; TAILQ_FOREACH(hdr, &debug_alloc_hdr_list, next) { - ec_log(EC_LOG_ERR, + EC_LOG(EC_LOG_ERR, "%s:%d: error: memory leak size=%zd ptr=%p\n", hdr->file, hdr->line, hdr->size, hdr + 1); buffer = backtrace_symbols(hdr->stack, hdr->stacklen); if (buffer == NULL) { for (i = 0; i < hdr->stacklen; i++) - ec_log(EC_LOG_ERR, " %p\n", hdr->stack[i]); + EC_LOG(EC_LOG_ERR, " %p\n", hdr->stack[i]); } else { for (i = 0; i < hdr->stacklen; i++) - ec_log(EC_LOG_ERR, " %s\n", + EC_LOG(EC_LOG_ERR, " %s\n", buffer ? buffer[i] : "unknown"); } free(buffer); } - ec_log(EC_LOG_ERR, + EC_LOG(EC_LOG_ERR, " missing static syms, use: addr2line -f -e \n"); return -1; } -static int debug_log(unsigned int level, void *opaque, const char *str) +static int debug_log(int type, unsigned int level, void *opaque, + const char *str) { + (void)type; (void)opaque; if (level > (unsigned int)log_level) return 0; - return printf("%s", str); + if (printf("%s", str) < 0) + return -1; + + return 0; } int main(int argc, char **argv) { - int ret, leaks; + int i, ret = 0, leaks; - parse_args(argc, argv); + ret = parse_args(argc, argv); + if (ret < 0) + return 1; + + argc -= ret; + argv += ret; - ec_log_register(debug_log, NULL); + srandom(seed); /* register a new malloc to track memleaks */ TAILQ_INIT(&debug_alloc_hdr_list); if (ec_malloc_register(debug_malloc, debug_free, debug_realloc) < 0) { - ec_log(EC_LOG_ERR, "cannot register new malloc\n"); - return -1; + EC_LOG(EC_LOG_ERR, "cannot register new malloc\n"); + return 1; + } + + if (ec_init() < 0) { + fprintf(stderr, "cannot init ecoli: %s\n", strerror(errno)); + return 1; } + ec_log_fct_register(debug_log, NULL); - ret = ec_test_all(); + ret = 0; + if (argc <= 1) { + ret = ec_test_all(); + } else { + for (i = 1; i < argc; i++) + ret |= ec_test_one(argv[i]); + } - ec_malloc_unregister(); leaks = debug_alloc_dump_leaks(); if (alloc_fail_proba == 0 && ret != 0) { @@ -368,3 +401,4 @@ int main(int argc, char **argv) return 0; } +/* LCOV_EXCL_STOP */