1 /*
2  * cdjpeg.c
3  *
4  * This file was part of the Independent JPEG Group's software:
5  * Copyright (C) 1991-1997, Thomas G. Lane.
6  * libjpeg-turbo Modifications:
7  * Copyright (C) 2019, D. R. Commander.
8  * For conditions of distribution and use, see the accompanying README.ijg
9  * file.
10  *
11  * This file contains common support routines used by the IJG application
12  * programs (cjpeg, djpeg, jpegtran).
13  */
14 
15 #include "cdjpeg.h"             /* Common decls for cjpeg/djpeg applications */
16 #include <ctype.h>              /* to declare isupper(), tolower() */
17 #ifdef USE_SETMODE
18 #include <fcntl.h>              /* to declare setmode()'s parameter macros */
19 /* If you have setmode() but not <io.h>, just delete this line: */
20 #include <io.h>                 /* to declare setmode() */
21 #endif
22 
23 
24 /*
25  * Optional progress monitor: display a percent-done figure on stderr.
26  */
27 
28 METHODDEF(void)
progress_monitor(j_common_ptr cinfo)29 progress_monitor(j_common_ptr cinfo)
30 {
31   cd_progress_ptr prog = (cd_progress_ptr)cinfo->progress;
32 
33   if (prog->max_scans != 0 && cinfo->is_decompressor) {
34     int scan_no = ((j_decompress_ptr)cinfo)->input_scan_number;
35 
36     if (scan_no > (int)prog->max_scans) {
37       fprintf(stderr, "Scan number %d exceeds maximum scans (%d)\n", scan_no,
38               prog->max_scans);
39       exit(EXIT_FAILURE);
40     }
41   }
42 
43   if (prog->report) {
44     int total_passes = prog->pub.total_passes + prog->total_extra_passes;
45     int percent_done =
46       (int)(prog->pub.pass_counter * 100L / prog->pub.pass_limit);
47 
48     if (percent_done != prog->percent_done) {
49       prog->percent_done = percent_done;
50       if (total_passes > 1) {
51         fprintf(stderr, "\rPass %d/%d: %3d%% ",
52                 prog->pub.completed_passes + prog->completed_extra_passes + 1,
53                 total_passes, percent_done);
54       } else {
55         fprintf(stderr, "\r %3d%% ", percent_done);
56       }
57       fflush(stderr);
58     }
59   }
60 }
61 
62 
63 GLOBAL(void)
start_progress_monitor(j_common_ptr cinfo,cd_progress_ptr progress)64 start_progress_monitor(j_common_ptr cinfo, cd_progress_ptr progress)
65 {
66   /* Enable progress display, unless trace output is on */
67   if (cinfo->err->trace_level == 0) {
68     progress->pub.progress_monitor = progress_monitor;
69     progress->completed_extra_passes = 0;
70     progress->total_extra_passes = 0;
71     progress->max_scans = 0;
72     progress->report = FALSE;
73     progress->percent_done = -1;
74     cinfo->progress = &progress->pub;
75   }
76 }
77 
78 
79 GLOBAL(void)
end_progress_monitor(j_common_ptr cinfo)80 end_progress_monitor(j_common_ptr cinfo)
81 {
82   /* Clear away progress display */
83   if (cinfo->err->trace_level == 0) {
84     fprintf(stderr, "\r                \r");
85     fflush(stderr);
86   }
87 }
88 
89 
90 /*
91  * Case-insensitive matching of possibly-abbreviated keyword switches.
92  * keyword is the constant keyword (must be lower case already),
93  * minchars is length of minimum legal abbreviation.
94  */
95 
96 GLOBAL(boolean)
keymatch(char * arg,const char * keyword,int minchars)97 keymatch(char *arg, const char *keyword, int minchars)
98 {
99   register int ca, ck;
100   register int nmatched = 0;
101 
102   while ((ca = *arg++) != '\0') {
103     if ((ck = *keyword++) == '\0')
104       return FALSE;             /* arg longer than keyword, no good */
105     if (isupper(ca))            /* force arg to lcase (assume ck is already) */
106       ca = tolower(ca);
107     if (ca != ck)
108       return FALSE;             /* no good */
109     nmatched++;                 /* count matched characters */
110   }
111   /* reached end of argument; fail if it's too short for unique abbrev */
112   if (nmatched < minchars)
113     return FALSE;
114   return TRUE;                  /* A-OK */
115 }
116 
117 
118 /*
119  * Routines to establish binary I/O mode for stdin and stdout.
120  * Non-Unix systems often require some hacking to get out of text mode.
121  */
122 
123 GLOBAL(FILE *)
read_stdin(void)124 read_stdin(void)
125 {
126   FILE *input_file = stdin;
127 
128 #ifdef USE_SETMODE              /* need to hack file mode? */
129   setmode(fileno(stdin), O_BINARY);
130 #endif
131 #ifdef USE_FDOPEN               /* need to re-open in binary mode? */
132   if ((input_file = fdopen(fileno(stdin), READ_BINARY)) == NULL) {
133     fprintf(stderr, "Cannot reopen stdin\n");
134     exit(EXIT_FAILURE);
135   }
136 #endif
137   return input_file;
138 }
139 
140 
141 GLOBAL(FILE *)
write_stdout(void)142 write_stdout(void)
143 {
144   FILE *output_file = stdout;
145 
146 #ifdef USE_SETMODE              /* need to hack file mode? */
147   setmode(fileno(stdout), O_BINARY);
148 #endif
149 #ifdef USE_FDOPEN               /* need to re-open in binary mode? */
150   if ((output_file = fdopen(fileno(stdout), WRITE_BINARY)) == NULL) {
151     fprintf(stderr, "Cannot reopen stdout\n");
152     exit(EXIT_FAILURE);
153   }
154 #endif
155   return output_file;
156 }
157