Blob Blame History Raw
/***************************************************************/
/* perk: PE Resource Kit */
/* Copyright (C) 2015--2017 Z. Gilboa */
/* Released under GPLv2 and GPLv3; see COPYING.PERK. */
/***************************************************************/
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <perk/perk.h>
#include "perk_driver_impl.h"
#include "perk_dprintf_impl.h"
static const char aclr_reset[] = "\x1b[0m";
static const char aclr_bold[] = "\x1b[1m";
static const char aclr_red[] = "\x1b[31m";
static const char aclr_green[] = "\x1b[32m";
static const char aclr_blue[] = "\x1b[34m";
static const char aclr_magenta[] = "\x1b[35m";
static const char * const pe_error_strings[PERK_ERR_CAP] = {
[PERK_ERR_FLOW_ERROR] = "flow error: unexpected condition or other",
[PERK_ERR_FLEE_ERROR] = "flees and bugs and cats and mice",
[PERK_ERR_NULL_CONTEXT] = "null driver or unit context",
[PERK_ERR_NULL_IMAGE] = "null image base pointer",
[PERK_ERR_INVALID_CONTEXT] = "invalid driver or unit context",
[PERK_ERR_INVALID_IMAGE] = "invalid PE image",
[PERK_ERR_IMAGE_SIZE_ZERO] = "PE image size cannot be zero",
[PERK_ERR_IMAGE_MALFORMED] = "malformed PE image detected",
[PERK_ERR_BAD_DOS_HEADER] = "bad DOS header",
[PERK_ERR_BAD_COFF_HEADER] = "bad COFF header",
[PERK_ERR_BAD_IMAGE_TYPE] = "bad PE image type",
[PERK_ERR_UNSUPPORTED_ABI] = "unsupported image abi",
};
static const char * pe_output_error_header(const struct pe_error_info * erri)
{
if (erri->eflags & PERK_ERROR_CHILD)
return "exec error upon";
else if (erri->eflags & PERK_ERROR_TOP_LEVEL)
return "error logged in";
else if (erri->eflags & PERK_ERROR_NESTED)
return "< returned to >";
else
return "distorted state";
}
static const char * pe_output_unit_header(const struct pe_error_info * erri)
{
if (!(erri->eflags & PERK_ERROR_CUSTOM))
return "while opening";
else if (erri->elibcode == PERK_ERR_IMAGE_SIZE_ZERO)
return "while mapping";
else
return "while parsing";
}
static const char * pe_output_strerror(
const struct pe_error_info * erri,
char (*errbuf)[256])
{
if (erri->eflags & PERK_ERROR_CUSTOM)
return ((erri->elibcode < 0) || (erri->elibcode >= PERK_ERR_CAP))
? "internal error: please report to the maintainer"
: pe_error_strings[erri->elibcode];
else if (erri->eflags & PERK_ERROR_NESTED)
return "";
else if (erri->eflags & PERK_ERROR_CHILD)
return "(see child process error messages)";
else if (erri->esyscode == ENOBUFS)
return "input error: string length exceeds buffer size.";
else
return strerror_r(erri->esyscode,*errbuf,sizeof(*errbuf))
? "internal error: strerror_r(3) call failed"
: *errbuf;
}
static int pe_output_error_record_plain(
const struct pe_driver_ctx * dctx,
const struct pe_error_info * erri)
{
const char * epath;
char errbuf[256];
int fderr = pe_driver_fderr(dctx);
const char * errdesc = pe_output_strerror(erri,&errbuf);
epath = erri->euctx
? *erri->euctx->path
: erri->eunit;
if (epath && !(erri->eflags & PERK_ERROR_NESTED))
if (pe_dprintf(
fderr,
"%s: [%s] '%s':\n",
dctx->program,
pe_output_unit_header(erri),
epath) < 0)
return -1;
if (pe_dprintf(
fderr,
"%s: %s %s(), line %d%s%s.\n",
dctx->program,
pe_output_error_header(erri),
erri->efunction,
erri->eline,
strlen(errdesc) ? ": " : "",
errdesc) < 0)
return -1;
return 0;
}
static int pe_output_error_record_annotated(
const struct pe_driver_ctx * dctx,
const struct pe_error_info * erri)
{
const char * epath;
char errbuf[256];
int fderr = pe_driver_fderr(dctx);
const char * errdesc = pe_output_strerror(erri,&errbuf);
epath = erri->euctx
? *erri->euctx->path
: erri->eunit;
if (epath && !(erri->eflags & PERK_ERROR_NESTED))
if (pe_dprintf(
fderr,
"%s%s%s:%s %s[%s]%s %s%s'%s'%s:\n",
aclr_bold,aclr_magenta,
dctx->program,
aclr_reset,
aclr_bold,
pe_output_unit_header(erri),
aclr_reset,
aclr_bold,aclr_red,
epath,
aclr_reset) < 0)
return -1;
if (pe_dprintf(
fderr,
"%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,
pe_output_error_header(erri),
aclr_reset,
aclr_bold,aclr_blue,
erri->efunction,
aclr_reset,
aclr_bold,aclr_green,
erri->eline,
aclr_reset,
strlen(errdesc) ? ": " : "",
aclr_bold,
errdesc,
aclr_reset) < 0)
return -1;
return 0;
}
int pe_output_error_record(
const struct pe_driver_ctx * dctx,
const struct pe_error_info * erri)
{
if (dctx->cctx->drvflags & PERK_DRIVER_ANNOTATE_NEVER)
return pe_output_error_record_plain(dctx,erri);
else if (dctx->cctx->drvflags & PERK_DRIVER_ANNOTATE_ALWAYS)
return pe_output_error_record_annotated(dctx,erri);
else if (isatty(pe_driver_fderr(dctx)))
return pe_output_error_record_annotated(dctx,erri);
else
return pe_output_error_record_plain(dctx,erri);
}
static int pe_output_error_vector_plain(const struct pe_driver_ctx * dctx)
{
struct pe_error_info ** perr;
for (perr=dctx->errv; *perr; perr++)
if (pe_output_error_record_plain(dctx,*perr))
return -1;
return 0;
}
static int pe_output_error_vector_annotated(const struct pe_driver_ctx * dctx)
{
struct pe_error_info ** perr;
for (perr=dctx->errv; *perr; perr++)
if (pe_output_error_record_annotated(dctx,*perr))
return -1;
return 0;
}
int pe_output_error_vector(const struct pe_driver_ctx * dctx)
{
if (dctx->cctx->drvflags & PERK_DRIVER_ANNOTATE_NEVER)
return pe_output_error_vector_plain(dctx);
else if (dctx->cctx->drvflags & PERK_DRIVER_ANNOTATE_ALWAYS)
return pe_output_error_vector_annotated(dctx);
else if (isatty(pe_driver_fderr(dctx)))
return pe_output_error_vector_annotated(dctx);
else
return pe_output_error_vector_plain(dctx);
}