Fixes to progress message handling in xz:
- Don't use Windows-specific code on Windows. The old code required at least Windows 2000. Now it should work on Windows 98 and later, and maybe on Windows 95 too. - Use less precision when showing estimated remaining time. - Fix some small design issues.
This commit is contained in:
parent
47c2e21f82
commit
68bf7ac298
491
src/xz/message.c
491
src/xz/message.c
|
@ -23,13 +23,6 @@
|
||||||
# include <sys/time.h>
|
# include <sys/time.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
# ifndef _WIN32_WINNT
|
|
||||||
# define _WIN32_WINNT 0x0500
|
|
||||||
# endif
|
|
||||||
# include <windows.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <stdarg.h>
|
#include <stdarg.h>
|
||||||
|
|
||||||
|
|
||||||
|
@ -51,77 +44,53 @@ static const char *filename;
|
||||||
/// True once the a filename has been printed to stderr as part of progress
|
/// True once the a filename has been printed to stderr as part of progress
|
||||||
/// message. If automatic progress updating isn't enabled, this becomes true
|
/// message. If automatic progress updating isn't enabled, this becomes true
|
||||||
/// after the first progress message has been printed due to user sending
|
/// after the first progress message has been printed due to user sending
|
||||||
/// SIGALRM. Once this variable is true, we will print an empty line before
|
/// SIGINFO, SIGUSR1, or SIGALRM. Once this variable is true, we will print
|
||||||
/// the next filename to make the output more readable.
|
/// an empty line before the next filename to make the output more readable.
|
||||||
static bool first_filename_printed = false;
|
static bool first_filename_printed = false;
|
||||||
|
|
||||||
/// This is set to true when we have printed the current filename to stderr
|
/// This is set to true when we have printed the current filename to stderr
|
||||||
/// as part of a progress message. This variable is useful only if not
|
/// as part of a progress message. This variable is useful only if not
|
||||||
/// updating progress automatically: if user sends many SIGALRM signals,
|
/// updating progress automatically: if user sends many SIGINFO, SIGUSR1, or
|
||||||
/// we won't print the name of the same file multiple times.
|
/// SIGALRM signals, we won't print the name of the same file multiple times.
|
||||||
static bool current_filename_printed = false;
|
static bool current_filename_printed = false;
|
||||||
|
|
||||||
/// True if we should print progress indicator and update it automatically.
|
/// True if we should print progress indicator and update it automatically
|
||||||
|
/// if also verbose >= V_VERBOSE.
|
||||||
static bool progress_automatic;
|
static bool progress_automatic;
|
||||||
|
|
||||||
|
/// True if message_progress_start() has been called but
|
||||||
|
/// message_progress_end() hasn't been called yet.
|
||||||
|
static bool progress_started = false;
|
||||||
|
|
||||||
/// This is true when a progress message was printed and the cursor is still
|
/// This is true when a progress message was printed and the cursor is still
|
||||||
/// on the same line with the progress message. In that case, a newline has
|
/// on the same line with the progress message. In that case, a newline has
|
||||||
/// to be printed before any error messages.
|
/// to be printed before any error messages.
|
||||||
static bool progress_active = false;
|
static bool progress_active = false;
|
||||||
|
|
||||||
|
/// Pointer to lzma_stream used to do the encoding or decoding.
|
||||||
|
static lzma_stream *progress_strm;
|
||||||
|
|
||||||
/// Expected size of the input stream is needed to show completion percentage
|
/// Expected size of the input stream is needed to show completion percentage
|
||||||
/// and estimate remaining time.
|
/// and estimate remaining time.
|
||||||
static uint64_t expected_in_size;
|
static uint64_t expected_in_size;
|
||||||
|
|
||||||
/// Time when we started processing the file
|
/// Time when we started processing the file
|
||||||
static double start_time;
|
static uint64_t start_time;
|
||||||
|
|
||||||
|
|
||||||
|
// Use alarm() and SIGALRM when they are supported. This has two minor
|
||||||
|
// advantages over the alternative of polling gettimeofday():
|
||||||
|
// - It is possible for the user to send SIGINFO, SIGUSR1, or SIGALRM to
|
||||||
|
// get intermediate progress information even when --verbose wasn't used
|
||||||
|
// or stderr is not a terminal.
|
||||||
|
// - alarm() + SIGALRM seems to have slightly less overhead than polling
|
||||||
|
// gettimeofday().
|
||||||
|
#ifdef SIGALRM
|
||||||
|
|
||||||
/// The signal handler for SIGALRM sets this to true. It is set back to false
|
/// The signal handler for SIGALRM sets this to true. It is set back to false
|
||||||
/// once the progress message has been updated.
|
/// once the progress message has been updated.
|
||||||
static volatile sig_atomic_t progress_needs_updating = false;
|
static volatile sig_atomic_t progress_needs_updating = false;
|
||||||
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
static HANDLE timer_queue = NULL;
|
|
||||||
static HANDLE timer_timer = NULL;
|
|
||||||
|
|
||||||
|
|
||||||
static void CALLBACK
|
|
||||||
timer_callback(PVOID dummy1 lzma_attribute((unused)),
|
|
||||||
BOOLEAN dummy2 lzma_attribute((unused)))
|
|
||||||
{
|
|
||||||
progress_needs_updating = true;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
/// Emulate alarm() on Windows.
|
|
||||||
static void
|
|
||||||
my_alarm(unsigned int seconds)
|
|
||||||
{
|
|
||||||
// Just in case creating the queue has failed.
|
|
||||||
if (timer_queue == NULL)
|
|
||||||
return;
|
|
||||||
|
|
||||||
// If an old timer_timer exists, get rid of it first.
|
|
||||||
if (timer_timer != NULL) {
|
|
||||||
(void)DeleteTimerQueueTimer(timer_queue, timer_timer, NULL);
|
|
||||||
timer_timer = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
// If it fails, tough luck. It's not that important.
|
|
||||||
(void)CreateTimerQueueTimer(&timer_timer, timer_queue, &timer_callback,
|
|
||||||
NULL, 1000U * seconds, 0,
|
|
||||||
WT_EXECUTEINTIMERTHREAD | WT_EXECUTEONLYONCE);
|
|
||||||
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
#define my_alarm alarm
|
|
||||||
|
|
||||||
/// Signal handler for SIGALRM
|
/// Signal handler for SIGALRM
|
||||||
static void
|
static void
|
||||||
progress_signal_handler(int sig lzma_attribute((unused)))
|
progress_signal_handler(int sig lzma_attribute((unused)))
|
||||||
|
@ -130,21 +99,25 @@ progress_signal_handler(int sig lzma_attribute((unused)))
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
/// This is true when progress message printing is wanted. Using the same
|
||||||
|
/// variable name as above to avoid some ifdefs.
|
||||||
|
static bool progress_needs_updating = false;
|
||||||
|
|
||||||
|
/// Elapsed time when the next progress message update should be done.
|
||||||
|
static uint64_t progress_next_update;
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
/// Get the current time as double
|
|
||||||
static double
|
/// Get the current time as microseconds since epoch
|
||||||
|
static uint64_t
|
||||||
my_time(void)
|
my_time(void)
|
||||||
{
|
{
|
||||||
struct timeval tv;
|
struct timeval tv;
|
||||||
|
gettimeofday(&tv, NULL);
|
||||||
// This really shouldn't fail. I'm not sure what to return if it
|
return (uint64_t)(tv.tv_sec) * UINT64_C(1000000) + tv.tv_usec;
|
||||||
// still fails. It doesn't look so useful to check the return value
|
|
||||||
// everywhere. FIXME?
|
|
||||||
if (gettimeofday(&tv, NULL))
|
|
||||||
return -1.0;
|
|
||||||
|
|
||||||
return (double)(tv.tv_sec) + (double)(tv.tv_usec) / 1.0e6;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -204,20 +177,36 @@ message_init(const char *given_argv0)
|
||||||
}
|
}
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef SIGALRM
|
||||||
timer_queue = CreateTimerQueue();
|
// At least DJGPP lacks SA_RESTART. It's not essential for us (the
|
||||||
#else
|
// rest of the code can handle interrupted system calls), so just
|
||||||
|
// define it zero.
|
||||||
# ifndef SA_RESTART
|
# ifndef SA_RESTART
|
||||||
# define SA_RESTART 0
|
# define SA_RESTART 0
|
||||||
# endif
|
# endif
|
||||||
// Establish the signal handler for SIGALRM. Since this signal
|
// Establish the signal handlers which set a flag to tell us that
|
||||||
// doesn't require any quick action, we set SA_RESTART.
|
// progress info should be updated. Since these signals don't
|
||||||
|
// require any quick action, we set SA_RESTART.
|
||||||
|
static const int sigs[] = {
|
||||||
|
#ifdef SIGALRM
|
||||||
|
SIGALRM,
|
||||||
|
#endif
|
||||||
|
#ifdef SIGINFO
|
||||||
|
SIGINFO,
|
||||||
|
#endif
|
||||||
|
#ifdef SIGUSR1
|
||||||
|
SIGUSR1,
|
||||||
|
#endif
|
||||||
|
};
|
||||||
|
|
||||||
struct sigaction sa;
|
struct sigaction sa;
|
||||||
sigemptyset(&sa.sa_mask);
|
sigemptyset(&sa.sa_mask);
|
||||||
sa.sa_flags = SA_RESTART;
|
sa.sa_flags = SA_RESTART;
|
||||||
sa.sa_handler = &progress_signal_handler;
|
sa.sa_handler = &progress_signal_handler;
|
||||||
if (sigaction(SIGALRM, &sa, NULL))
|
|
||||||
message_signal_handler();
|
for (size_t i = 0; i < ARRAY_SIZE(sigs); ++i)
|
||||||
|
if (sigaction(sigs[i], &sa, NULL))
|
||||||
|
message_signal_handler();
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
return;
|
return;
|
||||||
|
@ -288,16 +277,25 @@ print_filename(void)
|
||||||
|
|
||||||
|
|
||||||
extern void
|
extern void
|
||||||
message_progress_start(const char *src_name, uint64_t in_size)
|
message_progress_start(
|
||||||
|
lzma_stream *strm, const char *src_name, uint64_t in_size)
|
||||||
{
|
{
|
||||||
|
// Store the pointer to the lzma_stream used to do the coding.
|
||||||
|
// It is needed to find out the position in the stream.
|
||||||
|
progress_strm = strm;
|
||||||
|
|
||||||
// Store the processing start time of the file and its expected size.
|
// Store the processing start time of the file and its expected size.
|
||||||
// If we aren't printing any statistics, then these are unused. But
|
// If we aren't printing any statistics, then these are unused. But
|
||||||
// since it is possible that the user tells us with SIGALRM to show
|
// since it is possible that the user sends us a signal to show
|
||||||
// statistics, we need to have these available anyway.
|
// statistics, we need to have these available anyway.
|
||||||
start_time = my_time();
|
start_time = my_time();
|
||||||
filename = src_name;
|
filename = src_name;
|
||||||
expected_in_size = in_size;
|
expected_in_size = in_size;
|
||||||
|
|
||||||
|
// Indicate that progress info may need to be printed before
|
||||||
|
// printing error messages.
|
||||||
|
progress_started = true;
|
||||||
|
|
||||||
// Indicate the name of this file hasn't been printed to
|
// Indicate the name of this file hasn't been printed to
|
||||||
// stderr yet.
|
// stderr yet.
|
||||||
current_filename_printed = false;
|
current_filename_printed = false;
|
||||||
|
@ -306,19 +304,26 @@ message_progress_start(const char *src_name, uint64_t in_size)
|
||||||
++files_pos;
|
++files_pos;
|
||||||
|
|
||||||
// If progress indicator is wanted, print the filename and possibly
|
// If progress indicator is wanted, print the filename and possibly
|
||||||
// the file count now. As an exception, if there is exactly one file,
|
// the file count now.
|
||||||
// do not print the filename at all.
|
|
||||||
if (verbosity >= V_VERBOSE && progress_automatic) {
|
if (verbosity >= V_VERBOSE && progress_automatic) {
|
||||||
// Print the filename to stderr if that is appropriate with
|
// Print the filename to stderr if that is appropriate with
|
||||||
// the current settings.
|
// the current settings.
|
||||||
print_filename();
|
print_filename();
|
||||||
|
|
||||||
// Start the timer to set progress_needs_updating to true
|
// Start the timer to display the first progress message
|
||||||
// after about one second. An alternative would to be set
|
// after one second. An alternative would be to show the
|
||||||
// progress_needs_updating to true here immediatelly, but
|
// first message almost immediatelly, but delaying by one
|
||||||
// setting the timer looks better to me, since extremely
|
// second looks better to me, since extremely early
|
||||||
// early progress info is pretty much useless.
|
// progress info is pretty much useless.
|
||||||
my_alarm(1);
|
#ifdef SIGALRM
|
||||||
|
// First disable a possibly existing alarm.
|
||||||
|
alarm(0);
|
||||||
|
progress_needs_updating = false;
|
||||||
|
alarm(1);
|
||||||
|
#else
|
||||||
|
progress_needs_updating = true;
|
||||||
|
progress_next_update = 1000000;
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
return;
|
return;
|
||||||
|
@ -327,21 +332,31 @@ message_progress_start(const char *src_name, uint64_t in_size)
|
||||||
|
|
||||||
/// Make the string indicating completion percentage.
|
/// Make the string indicating completion percentage.
|
||||||
static const char *
|
static const char *
|
||||||
progress_percentage(uint64_t in_pos)
|
progress_percentage(uint64_t in_pos, bool final)
|
||||||
{
|
{
|
||||||
// If the size of the input file is unknown or the size told us is
|
static char buf[sizeof("100.0 %")];
|
||||||
// clearly wrong since we have processed more data than the alleged
|
|
||||||
// size of the file, show a static string indicating that we have
|
|
||||||
// no idea of the completion percentage.
|
|
||||||
if (expected_in_size == 0 || in_pos > expected_in_size)
|
|
||||||
return "--- %";
|
|
||||||
|
|
||||||
static char buf[sizeof("99.9 %")];
|
double percentage;
|
||||||
|
|
||||||
// Never show 100.0 % before we actually are finished (that case is
|
if (final) {
|
||||||
// handled separately in message_progress_end()).
|
// Use floating point conversion of snprintf() also for
|
||||||
snprintf(buf, sizeof(buf), "%.1f %%",
|
// 100.0 % instead of fixed string, because the decimal
|
||||||
(double)(in_pos) / (double)(expected_in_size) * 99.9);
|
// separator isn't a dot in all locales.
|
||||||
|
percentage = 100.0;
|
||||||
|
} else {
|
||||||
|
// If the size of the input file is unknown or the size told us is
|
||||||
|
// clearly wrong since we have processed more data than the alleged
|
||||||
|
// size of the file, show a static string indicating that we have
|
||||||
|
// no idea of the completion percentage.
|
||||||
|
if (expected_in_size == 0 || in_pos > expected_in_size)
|
||||||
|
return "--- %";
|
||||||
|
|
||||||
|
// Never show 100.0 % before we actually are finished.
|
||||||
|
percentage = (double)(in_pos) / (double)(expected_in_size)
|
||||||
|
* 99.9;
|
||||||
|
}
|
||||||
|
|
||||||
|
snprintf(buf, sizeof(buf), "%.1f %%", percentage);
|
||||||
|
|
||||||
return buf;
|
return buf;
|
||||||
}
|
}
|
||||||
|
@ -350,6 +365,8 @@ progress_percentage(uint64_t in_pos)
|
||||||
static void
|
static void
|
||||||
progress_sizes_helper(char **pos, size_t *left, uint64_t value, bool final)
|
progress_sizes_helper(char **pos, size_t *left, uint64_t value, bool final)
|
||||||
{
|
{
|
||||||
|
// Allow high precision only for the final message, since it looks
|
||||||
|
// stupid for in-progress information.
|
||||||
if (final) {
|
if (final) {
|
||||||
// At maximum of four digits is allowed for exact byte count.
|
// At maximum of four digits is allowed for exact byte count.
|
||||||
if (value < 10000) {
|
if (value < 10000) {
|
||||||
|
@ -368,6 +385,7 @@ progress_sizes_helper(char **pos, size_t *left, uint64_t value, bool final)
|
||||||
// Otherwise we use MiB.
|
// Otherwise we use MiB.
|
||||||
my_snprintf(pos, left, "%'.1f MiB",
|
my_snprintf(pos, left, "%'.1f MiB",
|
||||||
(double)(value) / (1024.0 * 1024.0));
|
(double)(value) / (1024.0 * 1024.0));
|
||||||
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -412,11 +430,11 @@ progress_sizes(uint64_t compressed_pos, uint64_t uncompressed_pos, bool final)
|
||||||
|
|
||||||
/// Make the string containing the processing speed of uncompressed data.
|
/// Make the string containing the processing speed of uncompressed data.
|
||||||
static const char *
|
static const char *
|
||||||
progress_speed(uint64_t uncompressed_pos, double elapsed)
|
progress_speed(uint64_t uncompressed_pos, uint64_t elapsed)
|
||||||
{
|
{
|
||||||
// Don't print the speed immediatelly, since the early values look
|
// Don't print the speed immediatelly, since the early values look
|
||||||
// like somewhat random.
|
// like somewhat random.
|
||||||
if (elapsed < 3.0)
|
if (elapsed < 3000000)
|
||||||
return "";
|
return "";
|
||||||
|
|
||||||
static const char unit[][8] = {
|
static const char unit[][8] = {
|
||||||
|
@ -428,17 +446,24 @@ progress_speed(uint64_t uncompressed_pos, double elapsed)
|
||||||
size_t unit_index = 0;
|
size_t unit_index = 0;
|
||||||
|
|
||||||
// Calculate the speed as KiB/s.
|
// Calculate the speed as KiB/s.
|
||||||
double speed = (double)(uncompressed_pos) / (elapsed * 1024.0);
|
double speed = (double)(uncompressed_pos)
|
||||||
|
/ ((double)(elapsed) * (1024.0 / 1e6));
|
||||||
|
|
||||||
// Adjust the unit of the speed if needed.
|
// Adjust the unit of the speed if needed.
|
||||||
while (speed > 999.9) {
|
while (speed > 999.0) {
|
||||||
speed /= 1024.0;
|
speed /= 1024.0;
|
||||||
if (++unit_index == ARRAY_SIZE(unit))
|
if (++unit_index == ARRAY_SIZE(unit))
|
||||||
return ""; // Way too fast ;-)
|
return ""; // Way too fast ;-)
|
||||||
}
|
}
|
||||||
|
|
||||||
static char buf[sizeof("999.9 GiB/s")];
|
// Use decimal point only if the number is small. Examples:
|
||||||
snprintf(buf, sizeof(buf), "%.1f %s", speed, unit[unit_index]);
|
// - 0.1 KiB/s
|
||||||
|
// - 9.9 KiB/s
|
||||||
|
// - 99 KiB/s
|
||||||
|
// - 999 KiB/s
|
||||||
|
static char buf[sizeof("999 GiB/s")];
|
||||||
|
snprintf(buf, sizeof(buf), "%.*f %s",
|
||||||
|
speed > 9.9 ? 0 : 1, speed, unit[unit_index]);
|
||||||
return buf;
|
return buf;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -446,13 +471,15 @@ progress_speed(uint64_t uncompressed_pos, double elapsed)
|
||||||
/// Make a string indicating elapsed or remaining time. The format is either
|
/// Make a string indicating elapsed or remaining time. The format is either
|
||||||
/// M:SS or H:MM:SS depending on if the time is an hour or more.
|
/// M:SS or H:MM:SS depending on if the time is an hour or more.
|
||||||
static const char *
|
static const char *
|
||||||
progress_time(uint32_t seconds)
|
progress_time(uint64_t useconds)
|
||||||
{
|
{
|
||||||
// 9999 hours = 416 days
|
// 9999 hours = 416 days
|
||||||
static char buf[sizeof("9999:59:59")];
|
static char buf[sizeof("9999:59:59")];
|
||||||
|
|
||||||
|
uint32_t seconds = useconds / 1000000;
|
||||||
|
|
||||||
// Don't show anything if the time is zero or ridiculously big.
|
// Don't show anything if the time is zero or ridiculously big.
|
||||||
if (seconds == 0 || seconds > ((UINT32_C(9999) * 60) + 59) * 60 + 59)
|
if (seconds == 0 || seconds > ((9999 * 60) + 59) * 60 + 59)
|
||||||
return "";
|
return "";
|
||||||
|
|
||||||
uint32_t minutes = seconds / 60;
|
uint32_t minutes = seconds / 60;
|
||||||
|
@ -476,87 +503,187 @@ progress_time(uint32_t seconds)
|
||||||
/// Make the string to contain the estimated remaining time, or if the amount
|
/// Make the string to contain the estimated remaining time, or if the amount
|
||||||
/// of input isn't known, how much time has elapsed.
|
/// of input isn't known, how much time has elapsed.
|
||||||
static const char *
|
static const char *
|
||||||
progress_remaining(uint64_t in_pos, double elapsed)
|
progress_remaining(uint64_t in_pos, uint64_t elapsed)
|
||||||
{
|
{
|
||||||
// If we don't know the size of the input, we indicate the time
|
// Show the amount of time spent so far when making an estimate of
|
||||||
// spent so far.
|
// remaining time wouldn't be reasonable:
|
||||||
if (expected_in_size == 0 || in_pos > expected_in_size)
|
// - Input size is unknown.
|
||||||
return progress_time((uint32_t)(elapsed));
|
// - Input has grown bigger since we started (de)compressing.
|
||||||
|
// - We haven't processed much data yet, so estimate would be
|
||||||
// If we are at the very beginning of the file or the file is very
|
// too inaccurate.
|
||||||
// small, don't give any estimate to avoid far too wrong estimations.
|
// - Only a few seconds has passed since we started (de)compressing,
|
||||||
if (in_pos < (UINT64_C(1) << 19) || elapsed < 8.0)
|
// so estimate would be too inaccurate.
|
||||||
return "";
|
if (expected_in_size == 0 || in_pos > expected_in_size
|
||||||
|
|| in_pos < (UINT64_C(1) << 19) || elapsed < 8000000)
|
||||||
|
return progress_time(elapsed);
|
||||||
|
|
||||||
// Calculate the estimate. Don't give an estimate of zero seconds,
|
// Calculate the estimate. Don't give an estimate of zero seconds,
|
||||||
// since it is possible that all the input has been already passed
|
// since it is possible that all the input has been already passed
|
||||||
// to the library, but there is still quite a bit of output pending.
|
// to the library, but there is still quite a bit of output pending.
|
||||||
uint32_t remaining = (double)(expected_in_size - in_pos)
|
uint32_t remaining = (double)(expected_in_size - in_pos)
|
||||||
* elapsed / (double)(in_pos);
|
* ((double)(elapsed) / 1e6) / (double)(in_pos);
|
||||||
if (remaining == 0)
|
if (remaining < 1)
|
||||||
remaining = 1;
|
remaining = 1;
|
||||||
|
|
||||||
return progress_time(remaining);
|
static char buf[sizeof("9 h 55 min")];
|
||||||
|
|
||||||
|
// Select appropriate precision for the estimated remaining time.
|
||||||
|
if (remaining <= 10) {
|
||||||
|
// At maximum of 10 seconds remaining.
|
||||||
|
// Show the number of seconds as is.
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " s", remaining);
|
||||||
|
|
||||||
|
} else if (remaining <= 50) {
|
||||||
|
// At maximum of 50 seconds remaining.
|
||||||
|
// Round up to the next multiple of five seconds.
|
||||||
|
remaining = (remaining + 4) / 5 * 5;
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " s", remaining);
|
||||||
|
|
||||||
|
} else if (remaining <= 590) {
|
||||||
|
// At maximum of 9 minutes and 50 seconds remaining.
|
||||||
|
// Round up to the next multiple of ten seconds.
|
||||||
|
remaining = (remaining + 9) / 10 * 10;
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " min %" PRIu32 " s",
|
||||||
|
remaining / 60, remaining % 60);
|
||||||
|
|
||||||
|
} else if (remaining <= 59 * 60) {
|
||||||
|
// At maximum of 59 minutes remaining.
|
||||||
|
// Round up to the next multiple of a minute.
|
||||||
|
remaining = (remaining + 59) / 60;
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " min", remaining);
|
||||||
|
|
||||||
|
} else if (remaining <= 9 * 3600 + 50 * 60) {
|
||||||
|
// At maximum of 9 hours and 50 minutes left.
|
||||||
|
// Round up to the next multiple of ten minutes.
|
||||||
|
remaining = (remaining + 599) / 600 * 10;
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " h %" PRIu32 " min",
|
||||||
|
remaining / 60, remaining % 60);
|
||||||
|
|
||||||
|
} else if (remaining <= 23 * 3600) {
|
||||||
|
// At maximum of 23 hours remaining.
|
||||||
|
// Round up to the next multiple of an hour.
|
||||||
|
remaining = (remaining + 3599) / 3600;
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " h", remaining);
|
||||||
|
|
||||||
|
} else if (remaining <= 9 * 24 * 3600 + 23 * 3600) {
|
||||||
|
// At maximum of 9 days and 23 hours remaining.
|
||||||
|
// Round up to the next multiple of an hour.
|
||||||
|
remaining = (remaining + 3599) / 3600;
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " d %" PRIu32 " h",
|
||||||
|
remaining / 24, remaining % 24);
|
||||||
|
|
||||||
|
} else if (remaining <= 999 * 24 * 3600) {
|
||||||
|
// At maximum of 999 days remaining. ;-)
|
||||||
|
// Round up to the next multiple of a day.
|
||||||
|
remaining = (remaining + 24 * 3600 - 1) / (24 * 3600);
|
||||||
|
snprintf(buf, sizeof(buf), "%" PRIu32 " d", remaining);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
// The estimated remaining time is so big that it's better
|
||||||
|
// that we just show the elapsed time.
|
||||||
|
return progress_time(elapsed);
|
||||||
|
}
|
||||||
|
|
||||||
|
return buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// Calculate the elapsed time as microseconds.
|
||||||
|
static uint64_t
|
||||||
|
progress_elapsed(void)
|
||||||
|
{
|
||||||
|
return my_time() - start_time;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// Get information about position in the stream. This is currently simple,
|
||||||
|
/// but it will become more complicated once we have multithreading support.
|
||||||
|
static void
|
||||||
|
progress_pos(uint64_t *in_pos,
|
||||||
|
uint64_t *compressed_pos, uint64_t *uncompressed_pos)
|
||||||
|
{
|
||||||
|
*in_pos = progress_strm->total_in;
|
||||||
|
|
||||||
|
if (opt_mode == MODE_COMPRESS) {
|
||||||
|
*compressed_pos = progress_strm->total_out;
|
||||||
|
*uncompressed_pos = progress_strm->total_in;
|
||||||
|
} else {
|
||||||
|
*compressed_pos = progress_strm->total_in;
|
||||||
|
*uncompressed_pos = progress_strm->total_out;
|
||||||
|
}
|
||||||
|
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
extern void
|
extern void
|
||||||
message_progress_update(uint64_t in_pos, uint64_t out_pos)
|
message_progress_update(void)
|
||||||
{
|
{
|
||||||
// If there's nothing to do, return immediatelly.
|
if (!progress_needs_updating)
|
||||||
if (!progress_needs_updating || in_pos == 0)
|
|
||||||
return;
|
return;
|
||||||
|
|
||||||
|
// Calculate how long we have been processing this file.
|
||||||
|
const uint64_t elapsed = progress_elapsed();
|
||||||
|
|
||||||
|
#ifndef SIGALRM
|
||||||
|
if (progress_next_update > elapsed)
|
||||||
|
return;
|
||||||
|
|
||||||
|
progress_next_update = elapsed + 1000000;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Get our current position in the stream.
|
||||||
|
uint64_t in_pos;
|
||||||
|
uint64_t compressed_pos;
|
||||||
|
uint64_t uncompressed_pos;
|
||||||
|
progress_pos(&in_pos, &compressed_pos, &uncompressed_pos);
|
||||||
|
|
||||||
|
// Block signals so that fprintf() doesn't get interrupted.
|
||||||
|
signals_block();
|
||||||
|
|
||||||
// Print the filename if it hasn't been printed yet.
|
// Print the filename if it hasn't been printed yet.
|
||||||
print_filename();
|
print_filename();
|
||||||
|
|
||||||
// Calculate how long we have been processing this file.
|
|
||||||
const double elapsed = my_time() - start_time;
|
|
||||||
|
|
||||||
// Set compressed_pos and uncompressed_pos.
|
|
||||||
uint64_t compressed_pos;
|
|
||||||
uint64_t uncompressed_pos;
|
|
||||||
if (opt_mode == MODE_COMPRESS) {
|
|
||||||
compressed_pos = out_pos;
|
|
||||||
uncompressed_pos = in_pos;
|
|
||||||
} else {
|
|
||||||
compressed_pos = in_pos;
|
|
||||||
uncompressed_pos = out_pos;
|
|
||||||
}
|
|
||||||
|
|
||||||
signals_block();
|
|
||||||
|
|
||||||
// Print the actual progress message. The idea is that there is at
|
// Print the actual progress message. The idea is that there is at
|
||||||
// least three spaces between the fields in typical situations, but
|
// least three spaces between the fields in typical situations, but
|
||||||
// even in rare situations there is at least one space.
|
// even in rare situations there is at least one space.
|
||||||
fprintf(stderr, " %7s %43s %11s %10s\r",
|
fprintf(stderr, " %7s %43s %9s %10s\r",
|
||||||
progress_percentage(in_pos),
|
progress_percentage(in_pos, false),
|
||||||
progress_sizes(compressed_pos, uncompressed_pos, false),
|
progress_sizes(compressed_pos, uncompressed_pos, false),
|
||||||
progress_speed(uncompressed_pos, elapsed),
|
progress_speed(uncompressed_pos, elapsed),
|
||||||
progress_remaining(in_pos, elapsed));
|
progress_remaining(in_pos, elapsed));
|
||||||
|
|
||||||
|
#ifdef SIGALRM
|
||||||
// Updating the progress info was finished. Reset
|
// Updating the progress info was finished. Reset
|
||||||
// progress_needs_updating to wait for the next SIGALRM.
|
// progress_needs_updating to wait for the next SIGALRM.
|
||||||
//
|
//
|
||||||
// NOTE: This has to be done before my_alarm() call or with (very) bad
|
// NOTE: This has to be done before alarm(1) or with (very) bad
|
||||||
// luck we could be setting this to false after the alarm has already
|
// luck we could be setting this to false after the alarm has already
|
||||||
// been triggered.
|
// been triggered.
|
||||||
progress_needs_updating = false;
|
progress_needs_updating = false;
|
||||||
|
|
||||||
if (progress_automatic) {
|
if (verbosity >= V_VERBOSE && progress_automatic) {
|
||||||
// Mark that the progress indicator is active, so if an error
|
// Mark that the progress indicator is active, so if an error
|
||||||
// occurs, the error message gets printed cleanly.
|
// occurs, the error message gets printed cleanly.
|
||||||
progress_active = true;
|
progress_active = true;
|
||||||
|
|
||||||
// Restart the timer so that progress_needs_updating gets
|
// Restart the timer so that progress_needs_updating gets
|
||||||
// set to true after about one second.
|
// set to true after about one second.
|
||||||
my_alarm(1);
|
alarm(1);
|
||||||
} else {
|
} else {
|
||||||
// The progress message was printed because user had sent us
|
// The progress message was printed because user had sent us
|
||||||
// SIGALRM. In this case, each progress message is printed
|
// SIGALRM. In this case, each progress message is printed
|
||||||
// on its own line.
|
// on its own line.
|
||||||
fputc('\n', stderr);
|
fputc('\n', stderr);
|
||||||
}
|
}
|
||||||
|
#else
|
||||||
|
// When SIGALRM isn't supported and we get here, it's always due to
|
||||||
|
// automatic progress update. We set progress_active here too like
|
||||||
|
// described above.
|
||||||
|
assert(verbosity >= V_VERBOSE);
|
||||||
|
assert(progress_automatic);
|
||||||
|
progress_active = true;
|
||||||
|
#endif
|
||||||
|
|
||||||
signals_unblock();
|
signals_unblock();
|
||||||
|
|
||||||
|
@ -564,57 +691,58 @@ message_progress_update(uint64_t in_pos, uint64_t out_pos)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
extern void
|
static void
|
||||||
message_progress_end(uint64_t in_pos, uint64_t out_pos, bool success)
|
progress_flush(bool finished)
|
||||||
{
|
{
|
||||||
// If we are not in verbose mode, we have nothing to do.
|
if (!progress_started || verbosity < V_VERBOSE)
|
||||||
if (verbosity < V_VERBOSE || user_abort)
|
|
||||||
return;
|
return;
|
||||||
|
|
||||||
// Cancel a pending alarm, if any.
|
uint64_t in_pos;
|
||||||
if (progress_automatic) {
|
|
||||||
my_alarm(0);
|
|
||||||
progress_active = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const double elapsed = my_time() - start_time;
|
|
||||||
|
|
||||||
uint64_t compressed_pos;
|
uint64_t compressed_pos;
|
||||||
uint64_t uncompressed_pos;
|
uint64_t uncompressed_pos;
|
||||||
if (opt_mode == MODE_COMPRESS) {
|
progress_pos(&in_pos, &compressed_pos, &uncompressed_pos);
|
||||||
compressed_pos = out_pos;
|
|
||||||
uncompressed_pos = in_pos;
|
|
||||||
} else {
|
|
||||||
compressed_pos = in_pos;
|
|
||||||
uncompressed_pos = out_pos;
|
|
||||||
}
|
|
||||||
|
|
||||||
// If it took less than a second, don't display the time.
|
// Avoid printing intermediate progress info if some error occurs
|
||||||
const char *elapsed_str = progress_time((double)(elapsed));
|
// in the beginning of the stream. (If something goes wrong later in
|
||||||
|
// the stream, it is sometimes useful to tell the user where the
|
||||||
|
// error approximately occurred, especially if the error occurs
|
||||||
|
// after a time-consuming operation.)
|
||||||
|
if (!finished && !progress_active
|
||||||
|
&& (compressed_pos == 0 || uncompressed_pos == 0))
|
||||||
|
return;
|
||||||
|
|
||||||
|
progress_active = false;
|
||||||
|
|
||||||
|
const uint64_t elapsed = progress_elapsed();
|
||||||
|
const char *elapsed_str = progress_time(elapsed);
|
||||||
|
|
||||||
signals_block();
|
signals_block();
|
||||||
|
|
||||||
// When using the auto-updating progress indicator, the final
|
// When using the auto-updating progress indicator, the final
|
||||||
// statistics are printed in the same format as the progress
|
// statistics are printed in the same format as the progress
|
||||||
// indicator itself.
|
// indicator itself.
|
||||||
if (progress_automatic && in_pos > 0) {
|
if (progress_automatic) {
|
||||||
// Using floating point conversion for the percentage instead
|
// Using floating point conversion for the percentage instead
|
||||||
// of static "100.0 %" string, because the decimal separator
|
// of static "100.0 %" string, because the decimal separator
|
||||||
// isn't a dot in all locales.
|
// isn't a dot in all locales.
|
||||||
fprintf(stderr, " %5.1f %% %43s %11s %10s\n",
|
fprintf(stderr, " %7s %43s %9s %10s\n",
|
||||||
100.0,
|
progress_percentage(in_pos, finished),
|
||||||
progress_sizes(compressed_pos, uncompressed_pos, true),
|
progress_sizes(compressed_pos, uncompressed_pos, true),
|
||||||
progress_speed(uncompressed_pos, elapsed),
|
progress_speed(uncompressed_pos, elapsed),
|
||||||
elapsed_str);
|
elapsed_str);
|
||||||
|
} else {
|
||||||
|
// The filename is always printed.
|
||||||
|
fprintf(stderr, "%s: ", filename);
|
||||||
|
|
||||||
// When no automatic progress indicator is used, don't print a verbose
|
// Percentage is printed only if we didn't finish yet.
|
||||||
// message at all if we something went wrong and we couldn't produce
|
// FIXME: This may look weird when size of the input
|
||||||
// any output. If we did produce output, then it is sometimes useful
|
// isn't known.
|
||||||
// to tell that to the user, especially if we detected an error after
|
if (!finished)
|
||||||
// a time-consuming operation.
|
fprintf(stderr, "%s, ",
|
||||||
} else if (success || out_pos > 0) {
|
progress_percentage(in_pos, false));
|
||||||
// The filename and size information are always printed.
|
|
||||||
fprintf(stderr, "%s: %s", filename, progress_sizes(
|
// Size information is always printed.
|
||||||
|
fprintf(stderr, "%s", progress_sizes(
|
||||||
compressed_pos, uncompressed_pos, true));
|
compressed_pos, uncompressed_pos, true));
|
||||||
|
|
||||||
// The speed and elapsed time aren't always shown.
|
// The speed and elapsed time aren't always shown.
|
||||||
|
@ -634,22 +762,23 @@ message_progress_end(uint64_t in_pos, uint64_t out_pos, bool success)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
extern void
|
||||||
|
message_progress_end(bool success)
|
||||||
|
{
|
||||||
|
assert(progress_started);
|
||||||
|
progress_flush(success);
|
||||||
|
progress_started = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
static void
|
static void
|
||||||
vmessage(enum message_verbosity v, const char *fmt, va_list ap)
|
vmessage(enum message_verbosity v, const char *fmt, va_list ap)
|
||||||
{
|
{
|
||||||
if (v <= verbosity) {
|
if (v <= verbosity) {
|
||||||
signals_block();
|
signals_block();
|
||||||
|
|
||||||
// If there currently is a progress message on the screen,
|
progress_flush(false);
|
||||||
// print a newline so that the progress message is left
|
|
||||||
// readable. This is good, because it is nice to be able to
|
|
||||||
// see where the error occurred. (The alternative would be
|
|
||||||
// to clear the progress message and replace it with the
|
|
||||||
// error message.)
|
|
||||||
if (progress_active) {
|
|
||||||
progress_active = false;
|
|
||||||
fputc('\n', stderr);
|
|
||||||
}
|
|
||||||
|
|
||||||
fprintf(stderr, "%s: ", argv0);
|
fprintf(stderr, "%s: ", argv0);
|
||||||
vfprintf(stderr, fmt, ap);
|
vfprintf(stderr, fmt, ap);
|
||||||
|
|
|
@ -111,21 +111,29 @@ extern void message_version(void) lzma_attribute((noreturn));
|
||||||
extern void message_help(bool long_help) lzma_attribute((noreturn));
|
extern void message_help(bool long_help) lzma_attribute((noreturn));
|
||||||
|
|
||||||
|
|
||||||
|
/// \brief Start progress info handling
|
||||||
///
|
///
|
||||||
extern void message_progress_start(const char *filename, uint64_t in_size);
|
/// This must be paired with a call to message_progress_end() before the
|
||||||
|
/// given *strm becomes invalid.
|
||||||
|
///
|
||||||
|
/// \param strm Pointer to lzma_stream used for the coding.
|
||||||
|
/// \param filename Name of the input file. stdin_filename is
|
||||||
|
/// handled specially.
|
||||||
|
/// \param in_size Size of the input file, or zero if unknown.
|
||||||
|
///
|
||||||
|
extern void message_progress_start(
|
||||||
|
lzma_stream *strm, const char *filename, uint64_t in_size);
|
||||||
|
|
||||||
|
|
||||||
///
|
/// Update the progress info if in verbose mode and enough time has passed
|
||||||
extern void message_progress_update(uint64_t in_pos, uint64_t out_pos);
|
/// since the previous update. This can be called only when
|
||||||
|
/// message_progress_start() has already been used.
|
||||||
|
extern void message_progress_update(void);
|
||||||
|
|
||||||
|
|
||||||
/// \brief Finishes the progress message if we were in verbose mode
|
/// \brief Finishes the progress message if we were in verbose mode
|
||||||
///
|
///
|
||||||
/// \param in_pos Final input position i.e. how much input there was.
|
/// \param finished True if the whole stream was successfully coded
|
||||||
/// \param out_pos Final output position
|
/// and output written to the output stream.
|
||||||
/// \param success True if the operation was successful. We don't
|
|
||||||
/// print the final progress message if the operation
|
|
||||||
/// wasn't successful.
|
|
||||||
///
|
///
|
||||||
extern void message_progress_end(
|
extern void message_progress_end(bool finished);
|
||||||
uint64_t in_pos, uint64_t out_pos, bool success);
|
|
||||||
|
|
|
@ -337,10 +337,11 @@ coder_run(file_pair *pair)
|
||||||
// Initialize the progress indicator.
|
// Initialize the progress indicator.
|
||||||
const uint64_t in_size = pair->src_st.st_size <= (off_t)(0)
|
const uint64_t in_size = pair->src_st.st_size <= (off_t)(0)
|
||||||
? 0 : (uint64_t)(pair->src_st.st_size);
|
? 0 : (uint64_t)(pair->src_st.st_size);
|
||||||
message_progress_start(pair->src_name, in_size);
|
message_progress_start(&strm, pair->src_name, in_size);
|
||||||
|
|
||||||
lzma_action action = LZMA_RUN;
|
lzma_action action = LZMA_RUN;
|
||||||
lzma_ret ret;
|
lzma_ret ret;
|
||||||
|
bool success = false; // Assume that something goes wrong.
|
||||||
|
|
||||||
strm.avail_in = 0;
|
strm.avail_in = 0;
|
||||||
strm.next_out = out_buf;
|
strm.next_out = out_buf;
|
||||||
|
@ -370,7 +371,7 @@ coder_run(file_pair *pair)
|
||||||
if (strm.avail_out == 0) {
|
if (strm.avail_out == 0) {
|
||||||
if (opt_mode != MODE_TEST && io_write(pair, out_buf,
|
if (opt_mode != MODE_TEST && io_write(pair, out_buf,
|
||||||
IO_BUFFER_SIZE - strm.avail_out))
|
IO_BUFFER_SIZE - strm.avail_out))
|
||||||
return false;
|
break;
|
||||||
|
|
||||||
strm.next_out = out_buf;
|
strm.next_out = out_buf;
|
||||||
strm.avail_out = IO_BUFFER_SIZE;
|
strm.avail_out = IO_BUFFER_SIZE;
|
||||||
|
@ -383,18 +384,6 @@ coder_run(file_pair *pair)
|
||||||
&& ret != LZMA_UNSUPPORTED_CHECK;
|
&& ret != LZMA_UNSUPPORTED_CHECK;
|
||||||
|
|
||||||
if (stop) {
|
if (stop) {
|
||||||
// First print the final progress info.
|
|
||||||
// This way the user sees more accurately
|
|
||||||
// where the error occurred. Note that we
|
|
||||||
// print this *before* the possible error
|
|
||||||
// message.
|
|
||||||
//
|
|
||||||
// FIXME: What if something goes wrong
|
|
||||||
// after this?
|
|
||||||
message_progress_end(strm.total_in,
|
|
||||||
strm.total_out,
|
|
||||||
ret == LZMA_STREAM_END);
|
|
||||||
|
|
||||||
// Write the remaining bytes even if something
|
// Write the remaining bytes even if something
|
||||||
// went wrong, because that way the user gets
|
// went wrong, because that way the user gets
|
||||||
// as much data as possible, which can be good
|
// as much data as possible, which can be good
|
||||||
|
@ -403,21 +392,32 @@ coder_run(file_pair *pair)
|
||||||
if (opt_mode != MODE_TEST && io_write(pair,
|
if (opt_mode != MODE_TEST && io_write(pair,
|
||||||
out_buf, IO_BUFFER_SIZE
|
out_buf, IO_BUFFER_SIZE
|
||||||
- strm.avail_out))
|
- strm.avail_out))
|
||||||
return false;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ret == LZMA_STREAM_END) {
|
if (ret == LZMA_STREAM_END) {
|
||||||
// Check that there is no trailing garbage.
|
// Check that there is no trailing garbage.
|
||||||
// This is needed for LZMA_Alone and raw
|
// This is needed for LZMA_Alone and raw
|
||||||
// streams.
|
// streams.
|
||||||
if (strm.avail_in == 0 && (pair->src_eof
|
if (strm.avail_in == 0 && !pair->src_eof) {
|
||||||
|| io_read(pair, in_buf, 1)
|
// Try reading one more byte.
|
||||||
== 0)) {
|
// Hopefully we don't get any more
|
||||||
assert(pair->src_eof);
|
// input, and thus pair->src_eof
|
||||||
return true;
|
// becomes true.
|
||||||
|
strm.avail_in = io_read(
|
||||||
|
pair, in_buf, 1);
|
||||||
|
if (strm.avail_in == SIZE_MAX)
|
||||||
|
break;
|
||||||
|
|
||||||
|
assert(strm.avail_in == 0
|
||||||
|
|| strm.avail_in == 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
// FIXME: What about io_read() failing?
|
if (strm.avail_in == 0) {
|
||||||
|
assert(pair->src_eof);
|
||||||
|
success = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
// We hadn't reached the end of the file.
|
// We hadn't reached the end of the file.
|
||||||
ret = LZMA_DATA_ERROR;
|
ret = LZMA_DATA_ERROR;
|
||||||
|
@ -461,15 +461,16 @@ coder_run(file_pair *pair)
|
||||||
}
|
}
|
||||||
|
|
||||||
if (stop)
|
if (stop)
|
||||||
return false;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Show progress information if --verbose was specified and
|
// Show progress information under certain conditions.
|
||||||
// stderr is a terminal.
|
message_progress_update();
|
||||||
message_progress_update(strm.total_in, strm.total_out);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return false;
|
message_progress_end(success);
|
||||||
|
|
||||||
|
return success;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue