diff --git a/include/mdso/mdso.h b/include/mdso/mdso.h index ed74243..4982bb8 100644 --- a/include/mdso/mdso.h +++ b/include/mdso/mdso.h @@ -37,6 +37,10 @@ extern "C" { #define MDSO_DRIVER_DRY_RUN 0x0020 #define MDSO_DRIVER_QUAD_PTR 0x0040 +#define MDSO_DRIVER_ANNOTATE_ALWAYS 0x1000 +#define MDSO_DRIVER_ANNOTATE_NEVER 0x2000 +#define MDSO_DRIVER_ANNOTATE_FULL 0x4000 + /* error flags */ #define MDSO_ERROR_TOP_LEVEL 0x0001 #define MDSO_ERROR_NESTED 0x0002 @@ -112,8 +116,10 @@ mdso_api int mdso_unmap_input (struct mdso_input *); /* utility api */ mdso_api int mdso_main (int, char **, char **); -mdso_api int mdso_output_export_symbols(const struct mdso_driver_ctx *, const struct mdso_unit_ctx *, FILE *); mdso_api int mdso_create_implib_sources(const struct mdso_driver_ctx *); +mdso_api int mdso_output_export_symbols(const struct mdso_driver_ctx *, const struct mdso_unit_ctx *, FILE *); +mdso_api int mdso_output_error_record (const struct mdso_driver_ctx *, const struct mdso_error_info *); +mdso_api int mdso_output_error_vector (const struct mdso_driver_ctx *); /* low-level api */ mdso_api uint32_t mdso_crc32_mbstr (const unsigned char * str, size_t * symlen); diff --git a/project/common.mk b/project/common.mk index 43c6a3b..260e1a6 100644 --- a/project/common.mk +++ b/project/common.mk @@ -11,6 +11,7 @@ COMMON_SRCS = \ src/logic/mdso_generate_symfn.c \ src/logic/mdso_map_input.c \ src/output/mdso_create_output.c \ + src/output/mdso_output_error.c \ src/output/mdso_output_export_symbols.c \ src/skin/mdso_skin_default.c \ diff --git a/src/output/mdso_output_error.c b/src/output/mdso_output_error.c new file mode 100644 index 0000000..fd94592 --- /dev/null +++ b/src/output/mdso_output_error.c @@ -0,0 +1,158 @@ +/****************************************************************/ +/* mdso: midipix dso scavenger */ +/* Copyright (C) 2015--2016 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.MDSO. */ +/****************************************************************/ + +#include +#include +#include +#include +#include + +static const char aclr_reset[] = "\x1b[0m"; +static const char aclr_bold[] = "\x1b[1m"; + +static const char aclr_green[] = "\x1b[32m"; +static const char aclr_blue[] = "\x1b[34m"; +static const char aclr_magenta[] = "\x1b[35m"; + +static const char * mdso_output_error_header(const struct mdso_error_info * erri) +{ + if (erri->flags & MDSO_ERROR_CHILD) + return "exec error upon"; + + else if (erri->flags & MDSO_ERROR_TOP_LEVEL) + return "error logged in"; + + else if (erri->flags & MDSO_ERROR_NESTED) + return "< returned to >"; + + else + return "distorted state"; +} + +static const char * mdso_output_strerror(const struct mdso_error_info * erri) +{ + if (erri->flags & MDSO_ERROR_CUSTOM) + return "flow error: unexpected condition or other."; + + else if (erri->flags & MDSO_ERROR_NESTED) + return ""; + + else if (erri->flags & MDSO_ERROR_CHILD) + return "(see child process error messages)"; + + else if (erri->syserror == ENOBUFS) + return "input error: string length exceeds buffer size."; + + else + return strerror(erri->syserror); +} + +static int mdso_output_error_record_plain( + const struct mdso_driver_ctx * dctx, + const struct mdso_error_info * erri) +{ + const char * errdesc = mdso_output_strerror(erri); + + if (fprintf(stderr,"%s: %s %s(), line %d%s%s.\n", + dctx->program, + mdso_output_error_header(erri), + erri->function, + erri->line, + strlen(errdesc) ? ": " : "", + errdesc) < 0) + return -1; + + return fflush(stderr); +} + +static int mdso_output_error_record_annotated( + const struct mdso_driver_ctx * dctx, + const struct mdso_error_info * erri) +{ + const char * errdesc = mdso_output_strerror(erri); + + if (fprintf( + stderr, + "%s%s%s:%s %s%s%s %s%s%s()%s, %s%sline %d%s%s%s%s%s.\n", + + aclr_bold,aclr_magenta, + dctx->program, + aclr_reset, + + aclr_bold, + mdso_output_error_header(erri), + aclr_reset, + + aclr_bold,aclr_blue, + erri->function, + aclr_reset, + + aclr_bold,aclr_green, + erri->line, + aclr_reset, + strlen(errdesc) ? ": " : "", + + aclr_bold, + mdso_output_strerror(erri), + aclr_reset) < 0) + return -1; + + return fflush(stderr); +} + +int mdso_output_error_record( + const struct mdso_driver_ctx * dctx, + const struct mdso_error_info * erri) +{ + if (dctx->cctx->drvflags & MDSO_DRIVER_ANNOTATE_NEVER) + return mdso_output_error_record_plain(dctx,erri); + + else if (dctx->cctx->drvflags & MDSO_DRIVER_ANNOTATE_ALWAYS) + return mdso_output_error_record_annotated(dctx,erri); + + else if (isatty(STDERR_FILENO)) + return mdso_output_error_record_annotated(dctx,erri); + + else + return mdso_output_error_record_plain(dctx,erri); +} + +static int mdso_output_error_vector_plain(const struct mdso_driver_ctx * dctx) +{ + struct mdso_error_info ** perr; + + for (perr=dctx->errv; *perr; perr++) + if (mdso_output_error_record_plain(dctx,*perr)) + return -1; + + return 0; +} + +static int mdso_output_error_vector_annotated(const struct mdso_driver_ctx * dctx) +{ + struct mdso_error_info ** perr; + + for (perr=dctx->errv; *perr; perr++) + if (mdso_output_error_record_annotated(dctx,*perr)) + return -1; + + return 0; +} + +int mdso_output_error_vector(const struct mdso_driver_ctx * dctx) +{ + if (dctx->cctx->drvflags & MDSO_DRIVER_ANNOTATE_NEVER) + return mdso_output_error_vector_plain(dctx); + + else if (dctx->cctx->drvflags & MDSO_DRIVER_ANNOTATE_ALWAYS) + return mdso_output_error_vector_annotated(dctx); + + else if (isatty(STDERR_FILENO)) + return mdso_output_error_vector_annotated(dctx); + + else + return mdso_output_error_vector_plain(dctx); +}