blob: 88ee93ad3e9fd1b02e02319a03cc0a8df24f2081 [file] [log] [blame]
/*
* This file is part of the flashrom project.
*
* Copyright (C) 2009 Sean Nelson <audiohacked@gmail.com>
* Copyright (C) 2011 Carl-Daniel Hailfinger
*
* This program is free software; you can redistribute it and/or modify
* it 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.
*/
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <errno.h>
#include "flash.h"
#include "cli.h"
enum flashprog_log_level verbose_screen = FLASHPROG_MSG_INFO;
enum flashprog_log_level verbose_logfile = FLASHPROG_MSG_DEBUG2;
static FILE *logfile = NULL;
int close_logfile(void)
{
if (!logfile)
return 0;
/* No need to call fflush() explicitly, fclose() already does that. */
if (fclose(logfile)) {
/* fclose returned an error. Stop writing to be safe. */
logfile = NULL;
msg_gerr("Closing the log file returned error %s\n", strerror(errno));
return 1;
}
logfile = NULL;
return 0;
}
int open_logfile(const char * const filename)
{
if (!filename) {
msg_gerr("No logfile name specified.\n");
return 1;
}
if ((logfile = fopen(filename, "w")) == NULL) {
msg_gerr("Error: opening log file \"%s\" failed: %s\n", filename, strerror(errno));
return 1;
}
return 0;
}
void start_logging(void)
{
enum flashprog_log_level oldverbose_screen = verbose_screen;
/* Shut up the console. */
verbose_screen = FLASHPROG_MSG_ERROR;
print_version();
verbose_screen = oldverbose_screen;
}
static const char *flashprog_progress_stage_to_string(enum flashprog_progress_stage stage)
{
if (stage == FLASHPROG_PROGRESS_READ)
return "Reading";
if (stage == FLASHPROG_PROGRESS_WRITE)
return "Writing";
if (stage == FLASHPROG_PROGRESS_ERASE)
return "Erasing";
return "Progress";
}
static void print_progress_bar(enum flashprog_progress_stage stage, unsigned int pc)
{
char progress_line[73], *bar;
unsigned int i;
const int bar_start = snprintf(progress_line, sizeof(progress_line), "%s... [",
flashprog_progress_stage_to_string(stage));
for (bar = progress_line + bar_start, i = 0; i < pc; i += 2)
*bar++ = '=';
if (i < 100)
*bar++ = '>', i += 2;
for (; i < 100; i += 2)
*bar++ = ' ';
snprintf(bar, sizeof(progress_line) - (bar - progress_line), "] %3u%% ", pc);
printf("\r%s", progress_line);
}
void flashprog_progress_cb(enum flashprog_progress_stage stage, size_t current, size_t total, void *user_data)
{
static enum flashprog_progress_stage last_stage = (enum flashprog_progress_stage)-1;
static unsigned int last_pc = (unsigned int)-1;
const unsigned int pc = total ? (current * 100ull) / total : 100;
if (last_stage == stage && last_pc == pc)
return;
if (last_stage != stage || pc == 0)
printf("\n");
print_progress_bar(stage, pc);
last_stage = stage;
last_pc = pc;
}
/* Please note that level is the verbosity, not the importance of the message. */
int flashprog_print_cb(enum flashprog_log_level level, const char *fmt, va_list ap)
{
int ret = 0;
FILE *output_type = stdout;
va_list logfile_args;
va_copy(logfile_args, ap);
if (level < FLASHPROG_MSG_INFO)
output_type = stderr;
if (level <= verbose_screen) {
ret = vfprintf(output_type, fmt, ap);
/* msg_*spew often happens inside chip accessors in possibly
* time-critical operations. Don't slow them down by flushing. */
if (level != FLASHPROG_MSG_SPEW)
fflush(output_type);
}
if ((level <= verbose_logfile) && logfile) {
ret = vfprintf(logfile, fmt, logfile_args);
if (level != FLASHPROG_MSG_SPEW)
fflush(logfile);
}
va_end(logfile_args);
return ret;
}