1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/*
* 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;
}
|
