/* msg.cpp -- info and error messages

   This file is part of the UPX executable compressor.

   Copyright (C) Markus Franz Xaver Johannes Oberhumer

   Copyright (C) Laszlo Molnar

   All Rights Reserved.

   UPX and the UCL library are free software; you can redistribute them

   and/or modify them 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; see the file COPYING.

   If not, write to the Free Software Foundation, Inc.,

   59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

   Markus F.X.J. Oberhumer              Laszlo Molnar

   <markus@oberhumer.com>               <ezerotven+github@gmail.com>

 */

#include "util/system_headers.h"

#include <typeinfo> // typeid()

#include "conf.h"

/*************************************************************************

//

**************************************************************************/

static int pr_need_nl = 0;

void printSetNl(int need_nl) noexcept { pr_need_nl = need_nl; }

void printClearLine(FILE *f) noexcept {

    static char clear_line_msg[1 + 79 + 1 + 1];

    if (!clear_line_msg[0]) {

        char *msg = clear_line_msg;

        msg[0] = '\r';

        memset(msg + 1, ' ', 79);

        msg[80] = '\r';

        msg[81] = 0;

    }

    fflush(stdout);

    fflush(stderr);

    if (f == nullptr)

        f = stdout;

    con_fprintf(f, "%s", clear_line_msg); // NOLINT(clang-analyzer-unix.Stream)

    fflush(f);

    printSetNl(0);

}

static void pr_print(bool c, const char *msg) noexcept {

    if (c && !opt->to_stdout)

        con_fprintf(stderr, "%s", msg);

    else

        fprintf(stderr, "%s", msg);

}

static void pr_error(const char *iname, const char *msg, bool is_warning) noexcept {

    fflush(stdout);

    fflush(stderr);

    char buf[1024];

    buf[0] = 0;

    if (pr_need_nl == 2)

        printClearLine(stdout);

    else if (pr_need_nl) {

        buf[0] = '\n';

        buf[1] = 0;

        printSetNl(0);

    }

    // This hack is needed, otherwise error messages may get lost

    // when the cursor is not yet at the bottom of the screen.

    // At least I can use some colors then...

    bool c = acc_isatty(STDERR_FILENO) ? 1 : 0;

    int fg = con_fg(stderr, FG_BRTRED);

    upx_safe_snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), "%s: ", progname);

    pr_print(c, buf);

    //(void)con_fg(stderr,FG_RED);

    upx_safe_snprintf(buf, sizeof(buf), "%s: ", iname);

    pr_print(c, buf);

    //(void)con_fg(stderr,FG_BRTRED);

    pr_print(c, msg);

    pr_print(c, "\n");

    fflush(stdout);

    fflush(stderr);

    fg = con_fg(stderr, fg);

    UNUSED(is_warning);

    UNUSED(fg);

}

void printErr(const char *iname, const Throwable &e) noexcept {

    char buf[1024];

    size_t l;

    upx_safe_snprintf(buf, sizeof(buf), "%s", prettyExceptionName(typeid(e).name()));

    l = strlen(buf);

    if (l < sizeof(buf) && e.getMsg())

        upx_safe_snprintf(buf + l, sizeof(buf) - l, ": %s", e.getMsg());

    l = strlen(buf);

    if (l < sizeof(buf) && e.getErrno()) {

        upx_safe_snprintf(buf + l, sizeof(buf) - l, ": %s", strerror(e.getErrno()));

#if 1

        // some compilers (e.g. Borland C++) put a trailing '\n'

        // into the strerror() result

        l = strlen(buf);

        while (l-- > 0 && (buf[l] == '\n' || buf[l] == ' '))

            buf[l] = 0;

#endif

    }

    pr_error(iname, buf, e.isWarning());

}

void printErr(const char *iname, const char *format, ...) noexcept {

    va_list args;

    char buf[1024];

    va_start(args, format);

    upx_safe_vsnprintf(buf, sizeof(buf), format, args);

    va_end(args);

    pr_error(iname, buf, false);

}

void printWarn(const char *iname, const char *format, ...) noexcept {

    va_list args;

    char buf[1024];

    va_start(args, format);

    upx_safe_vsnprintf(buf, sizeof(buf), format, args);

    va_end(args);

    pr_error(iname, buf, true);

}

void printUnhandledException(const char *iname, const std::exception *e) noexcept {

    if (e != nullptr)

        printErr(iname, "unhandled exception: %s\n", prettyExceptionName(e->what()));

    else

        printErr(iname, "internal error: unhandled exception!\n");

    if (opt->cmd != CMD_COMPRESS) {

        printErr(iname, "  this file has possibly been modified/hacked; take care!\n");

    }

}

/*************************************************************************

// info

**************************************************************************/

static int info_header = 0;

static void info_print(const char *msg) {

    if (opt->info_mode <= 0)

        return;

    FILE *f = opt->to_stdout ? stderr : stdout;

    if (pr_need_nl) {

        printClearLine(f);

        con_fprintf(f, "%s\n", msg);

    } else if (pr_need_nl)

        con_fprintf(f, "\n%s\n", msg);

    else

        con_fprintf(f, "%s\n", msg);

    fflush(f);

    printSetNl(0);

}

void infoHeader() { info_header = 0; }

void infoHeader(const char *format, ...) {

    if (opt->info_mode <= 0)

        return;

    va_list args;

    char buf[1024];

    va_start(args, format);

    upx_safe_vsnprintf(buf, sizeof(buf), format, args);

    va_end(args);

    info_print(buf);

    info_header = 1;

}

void info(const char *format, ...) {

    if (opt->info_mode <= 0)

        return;

    va_list args;

    char buf[1024];

    const int n = 4 * info_header;

    memset(buf, ' ', n);

    va_start(args, format);

    upx_safe_vsnprintf(buf + n, sizeof(buf) - n, format, args);

    va_end(args);

    info_print(buf);

}

void infoWarning(const char *format, ...) {

    if (opt->info_mode <= 0)

        return;

    va_list args;

    char buf[1024];

    va_start(args, format);

    upx_safe_vsnprintf(buf, sizeof(buf), format, args);

    va_end(args);

    info("[WARNING] %s\n", buf);

}

void infoWriting(const char *what, upx_int64_t size) {

    if (opt->info_mode <= 0)

        return;

    info("Writing %s: %lld bytes", what, size);

}

/* vim:set ts=4 sw=4 et: */