]> icculus.org git repositories - icculus/xz.git/blob - src/xz/message.c
Put the interesting parts of XZ Utils into the public domain.
[icculus/xz.git] / src / xz / message.c
1 ///////////////////////////////////////////////////////////////////////////////
2 //
3 /// \file       message.c
4 /// \brief      Printing messages to stderr
5 //
6 //  Author:     Lasse Collin
7 //
8 //  This file has been put into the public domain.
9 //  You can do whatever you want with this file.
10 //
11 ///////////////////////////////////////////////////////////////////////////////
12
13 #include "private.h"
14
15 #ifdef HAVE_SYS_TIME_H
16 #       include <sys/time.h>
17 #endif
18
19 #include <stdarg.h>
20
21
22 /// Name of the program which is prefixed to the error messages.
23 static const char *argv0;
24
25 /// Number of the current file
26 static unsigned int files_pos = 0;
27
28 /// Total number of input files; zero if unknown.
29 static unsigned int files_total;
30
31 /// Verbosity level
32 static enum message_verbosity verbosity = V_WARNING;
33
34 /// Filename which we will print with the verbose messages
35 static const char *filename;
36
37 /// True once the a filename has been printed to stderr as part of progress
38 /// message. If automatic progress updating isn't enabled, this becomes true
39 /// after the first progress message has been printed due to user sending
40 /// SIGINFO, SIGUSR1, or SIGALRM. Once this variable is true, we will print
41 /// an empty line before the next filename to make the output more readable.
42 static bool first_filename_printed = false;
43
44 /// This is set to true when we have printed the current filename to stderr
45 /// as part of a progress message. This variable is useful only if not
46 /// updating progress automatically: if user sends many SIGINFO, SIGUSR1, or
47 /// SIGALRM signals, we won't print the name of the same file multiple times.
48 static bool current_filename_printed = false;
49
50 /// True if we should print progress indicator and update it automatically
51 /// if also verbose >= V_VERBOSE.
52 static bool progress_automatic;
53
54 /// True if message_progress_start() has been called but
55 /// message_progress_end() hasn't been called yet.
56 static bool progress_started = false;
57
58 /// This is true when a progress message was printed and the cursor is still
59 /// on the same line with the progress message. In that case, a newline has
60 /// to be printed before any error messages.
61 static bool progress_active = false;
62
63 /// Pointer to lzma_stream used to do the encoding or decoding.
64 static lzma_stream *progress_strm;
65
66 /// Expected size of the input stream is needed to show completion percentage
67 /// and estimate remaining time.
68 static uint64_t expected_in_size;
69
70 /// Time when we started processing the file
71 static uint64_t start_time;
72
73
74 // Use alarm() and SIGALRM when they are supported. This has two minor
75 // advantages over the alternative of polling gettimeofday():
76 //  - It is possible for the user to send SIGINFO, SIGUSR1, or SIGALRM to
77 //    get intermediate progress information even when --verbose wasn't used
78 //    or stderr is not a terminal.
79 //  - alarm() + SIGALRM seems to have slightly less overhead than polling
80 //    gettimeofday().
81 #ifdef SIGALRM
82
83 /// The signal handler for SIGALRM sets this to true. It is set back to false
84 /// once the progress message has been updated.
85 static volatile sig_atomic_t progress_needs_updating = false;
86
87 /// Signal handler for SIGALRM
88 static void
89 progress_signal_handler(int sig lzma_attribute((unused)))
90 {
91         progress_needs_updating = true;
92         return;
93 }
94
95 #else
96
97 /// This is true when progress message printing is wanted. Using the same
98 /// variable name as above to avoid some ifdefs.
99 static bool progress_needs_updating = false;
100
101 /// Elapsed time when the next progress message update should be done.
102 static uint64_t progress_next_update;
103
104 #endif
105
106
107 /// Get the current time as microseconds since epoch
108 static uint64_t
109 my_time(void)
110 {
111         struct timeval tv;
112         gettimeofday(&tv, NULL);
113         return (uint64_t)(tv.tv_sec) * UINT64_C(1000000) + tv.tv_usec;
114 }
115
116
117 /// Wrapper for snprintf() to help constructing a string in pieces.
118 static void lzma_attribute((format(printf, 3, 4)))
119 my_snprintf(char **pos, size_t *left, const char *fmt, ...)
120 {
121         va_list ap;
122         va_start(ap, fmt);
123         const int len = vsnprintf(*pos, *left, fmt, ap);
124         va_end(ap);
125
126         // If an error occurred, we want the caller to think that the whole
127         // buffer was used. This way no more data will be written to the
128         // buffer. We don't need better error handling here.
129         if (len < 0 || (size_t)(len) >= *left) {
130                 *left = 0;
131         } else {
132                 *pos += len;
133                 *left -= len;
134         }
135
136         return;
137 }
138
139
140 extern void
141 message_init(const char *given_argv0)
142 {
143         // Name of the program
144         argv0 = given_argv0;
145
146         // If --verbose is used, we use a progress indicator if and only
147         // if stderr is a terminal. If stderr is not a terminal, we print
148         // verbose information only after finishing the file. As a special
149         // exception, even if --verbose was not used, user can send SIGALRM
150         // to make us print progress information once without automatic
151         // updating.
152         progress_automatic = isatty(STDERR_FILENO);
153
154         // Commented out because COLUMNS is rarely exported to environment.
155         // Most users have at least 80 columns anyway, let's think something
156         // fancy here if enough people complain.
157 /*
158         if (progress_automatic) {
159                 // stderr is a terminal. Check the COLUMNS environment
160                 // variable to see if the terminal is wide enough. If COLUMNS
161                 // doesn't exist or it has some unparseable value, we assume
162                 // that the terminal is wide enough.
163                 const char *columns_str = getenv("COLUMNS");
164                 if (columns_str != NULL) {
165                         char *endptr;
166                         const long columns = strtol(columns_str, &endptr, 10);
167                         if (*endptr != '\0' || columns < 80)
168                                 progress_automatic = false;
169                 }
170         }
171 */
172
173 #ifdef SIGALRM
174         // At least DJGPP lacks SA_RESTART. It's not essential for us (the
175         // rest of the code can handle interrupted system calls), so just
176         // define it zero.
177 #       ifndef SA_RESTART
178 #               define SA_RESTART 0
179 #       endif
180         // Establish the signal handlers which set a flag to tell us that
181         // progress info should be updated. Since these signals don't
182         // require any quick action, we set SA_RESTART.
183         static const int sigs[] = {
184 #ifdef SIGALRM
185                 SIGALRM,
186 #endif
187 #ifdef SIGINFO
188                 SIGINFO,
189 #endif
190 #ifdef SIGUSR1
191                 SIGUSR1,
192 #endif
193         };
194
195         struct sigaction sa;
196         sigemptyset(&sa.sa_mask);
197         sa.sa_flags = SA_RESTART;
198         sa.sa_handler = &progress_signal_handler;
199
200         for (size_t i = 0; i < ARRAY_SIZE(sigs); ++i)
201                 if (sigaction(sigs[i], &sa, NULL))
202                         message_signal_handler();
203 #endif
204
205         return;
206 }
207
208
209 extern void
210 message_verbosity_increase(void)
211 {
212         if (verbosity < V_DEBUG)
213                 ++verbosity;
214
215         return;
216 }
217
218
219 extern void
220 message_verbosity_decrease(void)
221 {
222         if (verbosity > V_SILENT)
223                 --verbosity;
224
225         return;
226 }
227
228
229 extern void
230 message_set_files(unsigned int files)
231 {
232         files_total = files;
233         return;
234 }
235
236
237 /// Prints the name of the current file if it hasn't been printed already,
238 /// except if we are processing exactly one stream from stdin to stdout.
239 /// I think it looks nicer to not print "(stdin)" when --verbose is used
240 /// in a pipe and no other files are processed.
241 static void
242 print_filename(void)
243 {
244         if (!current_filename_printed
245                         && (files_total != 1 || filename != stdin_filename)) {
246                 signals_block();
247
248                 // If a file was already processed, put an empty line
249                 // before the next filename to improve readability.
250                 if (first_filename_printed)
251                         fputc('\n', stderr);
252
253                 first_filename_printed = true;
254                 current_filename_printed = true;
255
256                 // If we don't know how many files there will be due
257                 // to usage of --files or --files0.
258                 if (files_total == 0)
259                         fprintf(stderr, "%s (%u)\n", filename,
260                                         files_pos);
261                 else
262                         fprintf(stderr, "%s (%u/%u)\n", filename,
263                                         files_pos, files_total);
264
265                 signals_unblock();
266         }
267
268         return;
269 }
270
271
272 extern void
273 message_progress_start(
274                 lzma_stream *strm, const char *src_name, uint64_t in_size)
275 {
276         // Store the pointer to the lzma_stream used to do the coding.
277         // It is needed to find out the position in the stream.
278         progress_strm = strm;
279
280         // Store the processing start time of the file and its expected size.
281         // If we aren't printing any statistics, then these are unused. But
282         // since it is possible that the user sends us a signal to show
283         // statistics, we need to have these available anyway.
284         start_time = my_time();
285         filename = src_name;
286         expected_in_size = in_size;
287
288         // Indicate that progress info may need to be printed before
289         // printing error messages.
290         progress_started = true;
291
292         // Indicate the name of this file hasn't been printed to
293         // stderr yet.
294         current_filename_printed = false;
295
296         // Start numbering the files starting from one.
297         ++files_pos;
298
299         // If progress indicator is wanted, print the filename and possibly
300         // the file count now.
301         if (verbosity >= V_VERBOSE && progress_automatic) {
302                 // Print the filename to stderr if that is appropriate with
303                 // the current settings.
304                 print_filename();
305
306                 // Start the timer to display the first progress message
307                 // after one second. An alternative would be to show the
308                 // first message almost immediatelly, but delaying by one
309                 // second looks better to me, since extremely early
310                 // progress info is pretty much useless.
311 #ifdef SIGALRM
312                 // First disable a possibly existing alarm.
313                 alarm(0);
314                 progress_needs_updating = false;
315                 alarm(1);
316 #else
317                 progress_needs_updating = true;
318                 progress_next_update = 1000000;
319 #endif
320         }
321
322         return;
323 }
324
325
326 /// Make the string indicating completion percentage.
327 static const char *
328 progress_percentage(uint64_t in_pos, bool final)
329 {
330         static char buf[sizeof("100.0 %")];
331
332         double percentage;
333
334         if (final) {
335                 // Use floating point conversion of snprintf() also for
336                 // 100.0 % instead of fixed string, because the decimal
337                 // separator isn't a dot in all locales.
338                 percentage = 100.0;
339         } else {
340                 // If the size of the input file is unknown or the size told us is
341                 // clearly wrong since we have processed more data than the alleged
342                 // size of the file, show a static string indicating that we have
343                 // no idea of the completion percentage.
344                 if (expected_in_size == 0 || in_pos > expected_in_size)
345                         return "--- %";
346
347                 // Never show 100.0 % before we actually are finished.
348                 percentage = (double)(in_pos) / (double)(expected_in_size)
349                                 * 99.9;
350         }
351
352         snprintf(buf, sizeof(buf), "%.1f %%", percentage);
353
354         return buf;
355 }
356
357
358 static void
359 progress_sizes_helper(char **pos, size_t *left, uint64_t value, bool final)
360 {
361         // Allow high precision only for the final message, since it looks
362         // stupid for in-progress information.
363         if (final) {
364                 // At maximum of four digits is allowed for exact byte count.
365                 if (value < 10000) {
366                         my_snprintf(pos, left, "%'" PRIu64 " B", value);
367                         return;
368                 }
369
370                 // At maximum of five significant digits is allowed for KiB.
371                 if (value < UINT64_C(10239900)) {
372                         my_snprintf(pos, left, "%'.1f KiB",
373                                         (double)(value) / 1024.0);
374                         return;
375                 }
376         }
377
378         // Otherwise we use MiB.
379         my_snprintf(pos, left, "%'.1f MiB",
380                         (double)(value) / (1024.0 * 1024.0));
381
382         return;
383 }
384
385
386 /// Make the string containing the amount of input processed, amount of
387 /// output produced, and the compression ratio.
388 static const char *
389 progress_sizes(uint64_t compressed_pos, uint64_t uncompressed_pos, bool final)
390 {
391         // This is enough to hold sizes up to about 99 TiB if thousand
392         // separator is used, or about 1 PiB without thousand separator.
393         // After that the progress indicator will look a bit silly, since
394         // the compression ratio no longer fits with three decimal places.
395         static char buf[44];
396
397         char *pos = buf;
398         size_t left = sizeof(buf);
399
400         // Print the sizes. If this the final message, use more reasonable
401         // units than MiB if the file was small.
402         progress_sizes_helper(&pos, &left, compressed_pos, final);
403         my_snprintf(&pos, &left, " / ");
404         progress_sizes_helper(&pos, &left, uncompressed_pos, final);
405
406         // Avoid division by zero. If we cannot calculate the ratio, set
407         // it to some nice number greater than 10.0 so that it gets caught
408         // in the next if-clause.
409         const double ratio = uncompressed_pos > 0
410                         ? (double)(compressed_pos) / (double)(uncompressed_pos)
411                         : 16.0;
412
413         // If the ratio is very bad, just indicate that it is greater than
414         // 9.999. This way the length of the ratio field stays fixed.
415         if (ratio > 9.999)
416                 snprintf(pos, left, " > %.3f", 9.999);
417         else
418                 snprintf(pos, left, " = %.3f", ratio);
419
420         return buf;
421 }
422
423
424 /// Make the string containing the processing speed of uncompressed data.
425 static const char *
426 progress_speed(uint64_t uncompressed_pos, uint64_t elapsed)
427 {
428         // Don't print the speed immediatelly, since the early values look
429         // like somewhat random.
430         if (elapsed < 3000000)
431                 return "";
432
433         static const char unit[][8] = {
434                 "KiB/s",
435                 "MiB/s",
436                 "GiB/s",
437         };
438
439         size_t unit_index = 0;
440
441         // Calculate the speed as KiB/s.
442         double speed = (double)(uncompressed_pos)
443                         / ((double)(elapsed) * (1024.0 / 1e6));
444
445         // Adjust the unit of the speed if needed.
446         while (speed > 999.0) {
447                 speed /= 1024.0;
448                 if (++unit_index == ARRAY_SIZE(unit))
449                         return ""; // Way too fast ;-)
450         }
451
452         // Use decimal point only if the number is small. Examples:
453         //  - 0.1 KiB/s
454         //  - 9.9 KiB/s
455         //  - 99 KiB/s
456         //  - 999 KiB/s
457         static char buf[sizeof("999 GiB/s")];
458         snprintf(buf, sizeof(buf), "%.*f %s",
459                         speed > 9.9 ? 0 : 1, speed, unit[unit_index]);
460         return buf;
461 }
462
463
464 /// Make a string indicating elapsed or remaining time. The format is either
465 /// M:SS or H:MM:SS depending on if the time is an hour or more.
466 static const char *
467 progress_time(uint64_t useconds)
468 {
469         // 9999 hours = 416 days
470         static char buf[sizeof("9999:59:59")];
471
472         uint32_t seconds = useconds / 1000000;
473
474         // Don't show anything if the time is zero or ridiculously big.
475         if (seconds == 0 || seconds > ((9999 * 60) + 59) * 60 + 59)
476                 return "";
477
478         uint32_t minutes = seconds / 60;
479         seconds %= 60;
480
481         if (minutes >= 60) {
482                 const uint32_t hours = minutes / 60;
483                 minutes %= 60;
484                 snprintf(buf, sizeof(buf),
485                                 "%" PRIu32 ":%02" PRIu32 ":%02" PRIu32,
486                                 hours, minutes, seconds);
487         } else {
488                 snprintf(buf, sizeof(buf), "%" PRIu32 ":%02" PRIu32,
489                                 minutes, seconds);
490         }
491
492         return buf;
493 }
494
495
496 /// Make the string to contain the estimated remaining time, or if the amount
497 /// of input isn't known, how much time has elapsed.
498 static const char *
499 progress_remaining(uint64_t in_pos, uint64_t elapsed)
500 {
501         // Show the amount of time spent so far when making an estimate of
502         // remaining time wouldn't be reasonable:
503         //  - Input size is unknown.
504         //  - Input has grown bigger since we started (de)compressing.
505         //  - We haven't processed much data yet, so estimate would be
506         //    too inaccurate.
507         //  - Only a few seconds has passed since we started (de)compressing,
508         //    so estimate would be too inaccurate.
509         if (expected_in_size == 0 || in_pos > expected_in_size
510                         || in_pos < (UINT64_C(1) << 19) || elapsed < 8000000)
511                 return progress_time(elapsed);
512
513         // Calculate the estimate. Don't give an estimate of zero seconds,
514         // since it is possible that all the input has been already passed
515         // to the library, but there is still quite a bit of output pending.
516         uint32_t remaining = (double)(expected_in_size - in_pos)
517                         * ((double)(elapsed) / 1e6) / (double)(in_pos);
518         if (remaining < 1)
519                 remaining = 1;
520
521         static char buf[sizeof("9 h 55 min")];
522
523         // Select appropriate precision for the estimated remaining time.
524         if (remaining <= 10) {
525                 // At maximum of 10 seconds remaining.
526                 // Show the number of seconds as is.
527                 snprintf(buf, sizeof(buf), "%" PRIu32 " s", remaining);
528
529         } else if (remaining <= 50) {
530                 // At maximum of 50 seconds remaining.
531                 // Round up to the next multiple of five seconds.
532                 remaining = (remaining + 4) / 5 * 5;
533                 snprintf(buf, sizeof(buf), "%" PRIu32 " s", remaining);
534
535         } else if (remaining <= 590) {
536                 // At maximum of 9 minutes and 50 seconds remaining.
537                 // Round up to the next multiple of ten seconds.
538                 remaining = (remaining + 9) / 10 * 10;
539                 snprintf(buf, sizeof(buf), "%" PRIu32 " min %" PRIu32 " s",
540                                 remaining / 60, remaining % 60);
541
542         } else if (remaining <= 59 * 60) {
543                 // At maximum of 59 minutes remaining.
544                 // Round up to the next multiple of a minute.
545                 remaining = (remaining + 59) / 60;
546                 snprintf(buf, sizeof(buf), "%" PRIu32 " min", remaining);
547
548         } else if (remaining <= 9 * 3600 + 50 * 60) {
549                 // At maximum of 9 hours and 50 minutes left.
550                 // Round up to the next multiple of ten minutes.
551                 remaining = (remaining + 599) / 600 * 10;
552                 snprintf(buf, sizeof(buf), "%" PRIu32 " h %" PRIu32 " min",
553                                 remaining / 60, remaining % 60);
554
555         } else if (remaining <= 23 * 3600) {
556                 // At maximum of 23 hours remaining.
557                 // Round up to the next multiple of an hour.
558                 remaining = (remaining + 3599) / 3600;
559                 snprintf(buf, sizeof(buf), "%" PRIu32 " h", remaining);
560
561         } else if (remaining <= 9 * 24 * 3600 + 23 * 3600) {
562                 // At maximum of 9 days and 23 hours remaining.
563                 // Round up to the next multiple of an hour.
564                 remaining = (remaining + 3599) / 3600;
565                 snprintf(buf, sizeof(buf), "%" PRIu32 " d %" PRIu32 " h",
566                                 remaining / 24, remaining % 24);
567
568         } else if (remaining <= 999 * 24 * 3600) {
569                 // At maximum of 999 days remaining. ;-)
570                 // Round up to the next multiple of a day.
571                 remaining = (remaining + 24 * 3600 - 1) / (24 * 3600);
572                 snprintf(buf, sizeof(buf), "%" PRIu32 " d", remaining);
573
574         } else {
575                 // The estimated remaining time is so big that it's better
576                 // that we just show the elapsed time.
577                 return progress_time(elapsed);
578         }
579
580         return buf;
581 }
582
583
584 /// Calculate the elapsed time as microseconds.
585 static uint64_t
586 progress_elapsed(void)
587 {
588         return my_time() - start_time;
589 }
590
591
592 /// Get information about position in the stream. This is currently simple,
593 /// but it will become more complicated once we have multithreading support.
594 static void
595 progress_pos(uint64_t *in_pos,
596                 uint64_t *compressed_pos, uint64_t *uncompressed_pos)
597 {
598         *in_pos = progress_strm->total_in;
599
600         if (opt_mode == MODE_COMPRESS) {
601                 *compressed_pos = progress_strm->total_out;
602                 *uncompressed_pos = progress_strm->total_in;
603         } else {
604                 *compressed_pos = progress_strm->total_in;
605                 *uncompressed_pos = progress_strm->total_out;
606         }
607
608         return;
609 }
610
611
612 extern void
613 message_progress_update(void)
614 {
615         if (!progress_needs_updating)
616                 return;
617
618         // Calculate how long we have been processing this file.
619         const uint64_t elapsed = progress_elapsed();
620
621 #ifndef SIGALRM
622         if (progress_next_update > elapsed)
623                 return;
624
625         progress_next_update = elapsed + 1000000;
626 #endif
627
628         // Get our current position in the stream.
629         uint64_t in_pos;
630         uint64_t compressed_pos;
631         uint64_t uncompressed_pos;
632         progress_pos(&in_pos, &compressed_pos, &uncompressed_pos);
633
634         // Block signals so that fprintf() doesn't get interrupted.
635         signals_block();
636
637         // Print the filename if it hasn't been printed yet.
638         print_filename();
639
640         // Print the actual progress message. The idea is that there is at
641         // least three spaces between the fields in typical situations, but
642         // even in rare situations there is at least one space.
643         fprintf(stderr, "  %7s %43s   %9s   %10s\r",
644                 progress_percentage(in_pos, false),
645                 progress_sizes(compressed_pos, uncompressed_pos, false),
646                 progress_speed(uncompressed_pos, elapsed),
647                 progress_remaining(in_pos, elapsed));
648
649 #ifdef SIGALRM
650         // Updating the progress info was finished. Reset
651         // progress_needs_updating to wait for the next SIGALRM.
652         //
653         // NOTE: This has to be done before alarm(1) or with (very) bad
654         // luck we could be setting this to false after the alarm has already
655         // been triggered.
656         progress_needs_updating = false;
657
658         if (verbosity >= V_VERBOSE && progress_automatic) {
659                 // Mark that the progress indicator is active, so if an error
660                 // occurs, the error message gets printed cleanly.
661                 progress_active = true;
662
663                 // Restart the timer so that progress_needs_updating gets
664                 // set to true after about one second.
665                 alarm(1);
666         } else {
667                 // The progress message was printed because user had sent us
668                 // SIGALRM. In this case, each progress message is printed
669                 // on its own line.
670                 fputc('\n', stderr);
671         }
672 #else
673         // When SIGALRM isn't supported and we get here, it's always due to
674         // automatic progress update. We set progress_active here too like
675         // described above.
676         assert(verbosity >= V_VERBOSE);
677         assert(progress_automatic);
678         progress_active = true;
679 #endif
680
681         signals_unblock();
682
683         return;
684 }
685
686
687 static void
688 progress_flush(bool finished)
689 {
690         if (!progress_started || verbosity < V_VERBOSE)
691                 return;
692
693         uint64_t in_pos;
694         uint64_t compressed_pos;
695         uint64_t uncompressed_pos;
696         progress_pos(&in_pos, &compressed_pos, &uncompressed_pos);
697
698         // Avoid printing intermediate progress info if some error occurs
699         // in the beginning of the stream. (If something goes wrong later in
700         // the stream, it is sometimes useful to tell the user where the
701         // error approximately occurred, especially if the error occurs
702         // after a time-consuming operation.)
703         if (!finished && !progress_active
704                         && (compressed_pos == 0 || uncompressed_pos == 0))
705                 return;
706
707         progress_active = false;
708
709         const uint64_t elapsed = progress_elapsed();
710         const char *elapsed_str = progress_time(elapsed);
711
712         signals_block();
713
714         // When using the auto-updating progress indicator, the final
715         // statistics are printed in the same format as the progress
716         // indicator itself.
717         if (progress_automatic) {
718                 // Using floating point conversion for the percentage instead
719                 // of static "100.0 %" string, because the decimal separator
720                 // isn't a dot in all locales.
721                 fprintf(stderr, "  %7s %43s   %9s   %10s\n",
722                         progress_percentage(in_pos, finished),
723                         progress_sizes(compressed_pos, uncompressed_pos, true),
724                         progress_speed(uncompressed_pos, elapsed),
725                         elapsed_str);
726         } else {
727                 // The filename is always printed.
728                 fprintf(stderr, "%s: ", filename);
729
730                 // Percentage is printed only if we didn't finish yet.
731                 // FIXME: This may look weird when size of the input
732                 // isn't known.
733                 if (!finished)
734                         fprintf(stderr, "%s, ",
735                                         progress_percentage(in_pos, false));
736
737                 // Size information is always printed.
738                 fprintf(stderr, "%s", progress_sizes(
739                                 compressed_pos, uncompressed_pos, true));
740
741                 // The speed and elapsed time aren't always shown.
742                 const char *speed = progress_speed(uncompressed_pos, elapsed);
743                 if (speed[0] != '\0')
744                         fprintf(stderr, ", %s", speed);
745
746                 if (elapsed_str[0] != '\0')
747                         fprintf(stderr, ", %s", elapsed_str);
748
749                 fputc('\n', stderr);
750         }
751
752         signals_unblock();
753
754         return;
755 }
756
757
758 extern void
759 message_progress_end(bool success)
760 {
761         assert(progress_started);
762         progress_flush(success);
763         progress_started = false;
764         return;
765 }
766
767
768 static void
769 vmessage(enum message_verbosity v, const char *fmt, va_list ap)
770 {
771         if (v <= verbosity) {
772                 signals_block();
773
774                 progress_flush(false);
775
776                 fprintf(stderr, "%s: ", argv0);
777                 vfprintf(stderr, fmt, ap);
778                 fputc('\n', stderr);
779
780                 signals_unblock();
781         }
782
783         return;
784 }
785
786
787 extern void
788 message(enum message_verbosity v, const char *fmt, ...)
789 {
790         va_list ap;
791         va_start(ap, fmt);
792         vmessage(v, fmt, ap);
793         va_end(ap);
794         return;
795 }
796
797
798 extern void
799 message_warning(const char *fmt, ...)
800 {
801         va_list ap;
802         va_start(ap, fmt);
803         vmessage(V_WARNING, fmt, ap);
804         va_end(ap);
805
806         set_exit_status(E_WARNING);
807         return;
808 }
809
810
811 extern void
812 message_error(const char *fmt, ...)
813 {
814         va_list ap;
815         va_start(ap, fmt);
816         vmessage(V_ERROR, fmt, ap);
817         va_end(ap);
818
819         set_exit_status(E_ERROR);
820         return;
821 }
822
823
824 extern void
825 message_fatal(const char *fmt, ...)
826 {
827         va_list ap;
828         va_start(ap, fmt);
829         vmessage(V_ERROR, fmt, ap);
830         va_end(ap);
831
832         my_exit(E_ERROR);
833 }
834
835
836 extern void
837 message_bug(void)
838 {
839         message_fatal(_("Internal error (bug)"));
840 }
841
842
843 extern void
844 message_signal_handler(void)
845 {
846         message_fatal(_("Cannot establish signal handlers"));
847 }
848
849
850 extern const char *
851 message_strm(lzma_ret code)
852 {
853         switch (code) {
854         case LZMA_NO_CHECK:
855                 return _("No integrity check; not verifying file integrity");
856
857         case LZMA_UNSUPPORTED_CHECK:
858                 return _("Unsupported type of integrity check; "
859                                 "not verifying file integrity");
860
861         case LZMA_MEM_ERROR:
862                 return strerror(ENOMEM);
863
864         case LZMA_MEMLIMIT_ERROR:
865                 return _("Memory usage limit reached");
866
867         case LZMA_FORMAT_ERROR:
868                 return _("File format not recognized");
869
870         case LZMA_OPTIONS_ERROR:
871                 return _("Unsupported options");
872
873         case LZMA_DATA_ERROR:
874                 return _("Compressed data is corrupt");
875
876         case LZMA_BUF_ERROR:
877                 return _("Unexpected end of input");
878
879         case LZMA_OK:
880         case LZMA_STREAM_END:
881         case LZMA_GET_CHECK:
882         case LZMA_PROG_ERROR:
883                 return _("Internal error (bug)");
884         }
885
886         return NULL;
887 }
888
889
890 extern void
891 message_filters(enum message_verbosity v, const lzma_filter *filters)
892 {
893         if (v > verbosity)
894                 return;
895
896         fprintf(stderr, _("%s: Filter chain:"), argv0);
897
898         for (size_t i = 0; filters[i].id != LZMA_VLI_UNKNOWN; ++i) {
899                 fprintf(stderr, " --");
900
901                 switch (filters[i].id) {
902                 case LZMA_FILTER_LZMA1:
903                 case LZMA_FILTER_LZMA2: {
904                         const lzma_options_lzma *opt = filters[i].options;
905                         const char *mode;
906                         const char *mf;
907
908                         switch (opt->mode) {
909                         case LZMA_MODE_FAST:
910                                 mode = "fast";
911                                 break;
912
913                         case LZMA_MODE_NORMAL:
914                                 mode = "normal";
915                                 break;
916
917                         default:
918                                 mode = "UNKNOWN";
919                                 break;
920                         }
921
922                         switch (opt->mf) {
923                         case LZMA_MF_HC3:
924                                 mf = "hc3";
925                                 break;
926
927                         case LZMA_MF_HC4:
928                                 mf = "hc4";
929                                 break;
930
931                         case LZMA_MF_BT2:
932                                 mf = "bt2";
933                                 break;
934
935                         case LZMA_MF_BT3:
936                                 mf = "bt3";
937                                 break;
938
939                         case LZMA_MF_BT4:
940                                 mf = "bt4";
941                                 break;
942
943                         default:
944                                 mf = "UNKNOWN";
945                                 break;
946                         }
947
948                         fprintf(stderr, "lzma%c=dict=%" PRIu32
949                                         ",lc=%" PRIu32 ",lp=%" PRIu32
950                                         ",pb=%" PRIu32
951                                         ",mode=%s,nice=%" PRIu32 ",mf=%s"
952                                         ",depth=%" PRIu32,
953                                         filters[i].id == LZMA_FILTER_LZMA2
954                                                 ? '2' : '1',
955                                         opt->dict_size,
956                                         opt->lc, opt->lp, opt->pb,
957                                         mode, opt->nice_len, mf, opt->depth);
958                         break;
959                 }
960
961                 case LZMA_FILTER_X86:
962                         fprintf(stderr, "x86");
963                         break;
964
965                 case LZMA_FILTER_POWERPC:
966                         fprintf(stderr, "powerpc");
967                         break;
968
969                 case LZMA_FILTER_IA64:
970                         fprintf(stderr, "ia64");
971                         break;
972
973                 case LZMA_FILTER_ARM:
974                         fprintf(stderr, "arm");
975                         break;
976
977                 case LZMA_FILTER_ARMTHUMB:
978                         fprintf(stderr, "armthumb");
979                         break;
980
981                 case LZMA_FILTER_SPARC:
982                         fprintf(stderr, "sparc");
983                         break;
984
985                 case LZMA_FILTER_DELTA: {
986                         const lzma_options_delta *opt = filters[i].options;
987                         fprintf(stderr, "delta=dist=%" PRIu32, opt->dist);
988                         break;
989                 }
990
991                 default:
992                         fprintf(stderr, "UNKNOWN");
993                         break;
994                 }
995         }
996
997         fputc('\n', stderr);
998         return;
999 }
1000
1001
1002 extern void
1003 message_try_help(void)
1004 {
1005         // Print this with V_WARNING instead of V_ERROR to prevent it from
1006         // showing up when --quiet has been specified.
1007         message(V_WARNING, _("Try `%s --help' for more information."), argv0);
1008         return;
1009 }
1010
1011
1012 extern void
1013 message_version(void)
1014 {
1015         // It is possible that liblzma version is different than the command
1016         // line tool version, so print both.
1017         printf("xz " LZMA_VERSION_STRING "\n");
1018         printf("liblzma %s\n", lzma_version_string());
1019         my_exit(E_SUCCESS);
1020 }
1021
1022
1023 extern void
1024 message_help(bool long_help)
1025 {
1026         printf(_("Usage: %s [OPTION]... [FILE]...\n"
1027                         "Compress or decompress FILEs in the .xz format.\n\n"),
1028                         argv0);
1029
1030         puts(_("Mandatory arguments to long options are mandatory for "
1031                         "short options too.\n"));
1032
1033         if (long_help)
1034                 puts(_(" Operation mode:\n"));
1035
1036         puts(_(
1037 "  -z, --compress      force compression\n"
1038 "  -d, --decompress    force decompression\n"
1039 "  -t, --test          test compressed file integrity\n"
1040 "  -l, --list          list information about files"));
1041
1042         if (long_help)
1043                 puts(_("\n Operation modifiers:\n"));
1044
1045         puts(_(
1046 "  -k, --keep          keep (don't delete) input files\n"
1047 "  -f, --force         force overwrite of output file and (de)compress links\n"
1048 "  -c, --stdout        write to standard output and don't delete input files"));
1049
1050         if (long_help)
1051                 puts(_(
1052 "  -S, --suffix=.SUF   use the suffix `.SUF' on compressed files\n"
1053 "      --files=[FILE]  read filenames to process from FILE; if FILE is\n"
1054 "                      omitted, filenames are read from the standard input;\n"
1055 "                      filenames must be terminated with the newline character\n"
1056 "      --files0=[FILE] like --files but use the null character as terminator"));
1057
1058         if (long_help) {
1059                 puts(_("\n Basic file format and compression options:\n"));
1060                 puts(_(
1061 "  -F, --format=FMT    file format to encode or decode; possible values are\n"
1062 "                      `auto' (default), `xz', `lzma', and `raw'\n"
1063 "  -C, --check=CHECK   integrity check type: `crc32', `crc64' (default),\n"
1064 "                      or `sha256'"));
1065         }
1066
1067         puts(_(
1068 "  -0 .. -9            compression preset; 0-2 fast compression, 3-5 good\n"
1069 "                      compression, 6-9 excellent compression; default is 6"));
1070
1071         puts(_(
1072 "  -e, --extreme       use more CPU time when encoding to increase compression\n"
1073 "                      ratio without increasing memory usage of the decoder"));
1074
1075         puts(_(
1076 "  -M, --memory=NUM    use roughly NUM bytes of memory at maximum; 0 indicates\n"
1077 "                      the default setting, which depends on the operation mode\n"
1078 "                      and the amount of physical memory (RAM)"));
1079
1080         if (long_help) {
1081                 puts(_(
1082 "\n Custom filter chain for compression (alternative for using presets):"));
1083
1084 #if defined(HAVE_ENCODER_LZMA1) || defined(HAVE_DECODER_LZMA1) \
1085                 || defined(HAVE_ENCODER_LZMA2) || defined(HAVE_DECODER_LZMA2)
1086                 puts(_(
1087 "\n"
1088 "  --lzma1=[OPTS]      LZMA1 or LZMA2; OPTS is a comma-separated list of zero or\n"
1089 "  --lzma2=[OPTS]      more of the following options (valid values; default):\n"
1090 "                        preset=NUM reset options to preset number NUM (1-9)\n"
1091 "                        dict=NUM   dictionary size (4KiB - 1536MiB; 8MiB)\n"
1092 "                        lc=NUM     number of literal context bits (0-4; 3)\n"
1093 "                        lp=NUM     number of literal position bits (0-4; 0)\n"
1094 "                        pb=NUM     number of position bits (0-4; 2)\n"
1095 "                        mode=MODE  compression mode (fast, normal; normal)\n"
1096 "                        nice=NUM   nice length of a match (2-273; 64)\n"
1097 "                        mf=NAME    match finder (hc3, hc4, bt2, bt3, bt4; bt4)\n"
1098 "                        depth=NUM  maximum search depth; 0=automatic (default)"));
1099 #endif
1100
1101                 puts(_(
1102 "\n"
1103 "  --x86               x86 filter (sometimes called BCJ filter)\n"
1104 "  --powerpc           PowerPC (big endian) filter\n"
1105 "  --ia64              IA64 (Itanium) filter\n"
1106 "  --arm               ARM filter\n"
1107 "  --armthumb          ARM-Thumb filter\n"
1108 "  --sparc             SPARC filter"));
1109
1110 #if defined(HAVE_ENCODER_DELTA) || defined(HAVE_DECODER_DELTA)
1111                 puts(_(
1112 "\n"
1113 "  --delta=[OPTS]      Delta filter; valid OPTS (valid values; default):\n"
1114 "                        dist=NUM   distance between bytes being subtracted\n"
1115 "                                   from each other (1-256; 1)"));
1116 #endif
1117
1118 #if defined(HAVE_ENCODER_SUBBLOCK) || defined(HAVE_DECODER_SUBBLOCK)
1119                 puts(_(
1120 "\n"
1121 "  --subblock=[OPTS]   Subblock filter; valid OPTS (valid values; default):\n"
1122 "                        size=NUM   number of bytes of data per subblock\n"
1123 "                                   (1 - 256Mi; 4Ki)\n"
1124 "                        rle=NUM    run-length encoder chunk size (0-256; 0)"));
1125 #endif
1126         }
1127
1128         if (long_help)
1129                 puts(_("\n Other options:\n"));
1130
1131         puts(_(
1132 "  -q, --quiet         suppress warnings; specify twice to suppress errors too\n"
1133 "  -v, --verbose       be verbose; specify twice for even more verbose"));
1134
1135         if (long_help)
1136                 puts(_(
1137 "\n"
1138 "  -h, --help          display the short help (lists only the basic options)\n"
1139 "  -H, --long-help     display this long help"));
1140         else
1141                 puts(_(
1142 "  -h, --help          display this short help\n"
1143 "  -H, --long-help     display the long help (lists also the advanced options)"));
1144
1145         puts(_(
1146 "  -V, --version       display the version number"));
1147
1148         puts(_("\nWith no FILE, or when FILE is -, read standard input.\n"));
1149
1150         if (long_help) {
1151                 printf(_(
1152 "On this system and configuration, the tool will use at maximum of\n"
1153 "  * roughly %'" PRIu64 " MiB RAM for compression;\n"
1154 "  * roughly %'" PRIu64 " MiB RAM for decompression; and\n"),
1155                                 hardware_memlimit_encoder() / (1024 * 1024),
1156                                 hardware_memlimit_decoder() / (1024 * 1024));
1157                 printf(N_("  * one thread for (de)compression.\n\n",
1158                         "  * %'" PRIu32 " threads for (de)compression.\n\n",
1159                         hardware_threadlimit_get()),
1160                         hardware_threadlimit_get());
1161         }
1162
1163         printf(_("Report bugs to <%s> (in English or Finnish).\n"),
1164                         PACKAGE_BUGREPORT);
1165
1166         my_exit(E_SUCCESS);
1167 }