ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pcompos.c
Revision: 2.37
Committed: Thu Aug 2 18:33:43 2018 UTC (5 years, 9 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad5R2
Changes since 2.36: +2 -2 lines
Log Message:
Created MAXFMTLEN to guard against buffer overrun attacks in header input

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: pcompos.c,v 2.36 2018/04/16 19:40:37 greg Exp $";
3 #endif
4 /*
5 * pcompos.c - program to composite pictures.
6 *
7 * 6/30/87
8 */
9
10 #include <stdio.h>
11 #include <math.h>
12 #include <time.h>
13 #include <string.h>
14
15 #include "copyright.h"
16
17 #include "platform.h"
18 #include "paths.h"
19 #include "rterror.h"
20 #include "color.h"
21 #include "resolu.h"
22
23 #ifdef getc_unlocked /* avoid horrendous overhead of flockfile */
24 #undef getc
25 #define getc getc_unlocked
26 #endif
27
28 #define MAXFILE 1024
29
30 #define HASMIN 1
31 #define HASMAX 2
32
33 /* output picture size */
34 int xsiz = 0;
35 int ysiz = 0;
36 /* input dimensions */
37 int xmin = 0;
38 int ymin = 0;
39 int xmax = 0;
40 int ymax = 0;
41
42 COLR bgcolr = BLKCOLR; /* background color */
43
44 int labelht = 24; /* label height */
45
46 int checkthresh = 0; /* check threshold value */
47
48 char StandardInput[] = "<stdin>";
49 char Command[] = "<Command>";
50 char Label[] = "<Label>";
51
52 char *progname;
53
54 struct {
55 char *name; /* file or command name */
56 FILE *fp; /* stream pointer */
57 int xres, yres; /* picture size */
58 int xloc, yloc; /* anchor point */
59 int flags; /* HASMIN, HASMAX */
60 double thmin, thmax; /* thresholds */
61 } input[MAXFILE]; /* our input files */
62
63 int nfile; /* number of files */
64
65 int echoheader = 1;
66 char ourfmt[LPICFMT+1] = PICFMT;
67 int wrongformat = 0;
68
69
70 static gethfunc headline;
71 static void compos(void);
72 static int cmpcolr(COLR c1, double lv2);
73 static FILE * lblopen(char *s, int *xp, int *yp);
74
75
76
77 static int
78 headline( /* print line preceded by a tab */
79 char *s,
80 void *p
81 )
82 {
83 char fmt[MAXFMTLEN];
84
85 if (isheadid(s))
86 return(0);
87 if (formatval(fmt, s)) {
88 if (globmatch(ourfmt, fmt)) {
89 wrongformat = 0;
90 strcpy(ourfmt, fmt);
91 } else
92 wrongformat = 1;
93 } else if (echoheader) {
94 fputc('\t', stdout);
95 fputs(s, stdout);
96 }
97 return(0);
98 }
99
100
101 int
102 main(
103 int argc,
104 char *argv[]
105 )
106 {
107 int ncolumns = 0;
108 int autolabel = 0;
109 int curcol = 0, x0 = 0, curx = 0, cury = 0, spacing = 0;
110 int xsgn, ysgn;
111 char *thislabel;
112 int an;
113 SET_DEFAULT_BINARY();
114 SET_FILE_BINARY(stdin);
115 SET_FILE_BINARY(stdout);
116 progname = argv[0];
117
118 for (an = 1; an < argc && argv[an][0] == '-'; an++)
119 switch (argv[an][1]) {
120 case 'h':
121 echoheader = !echoheader;
122 break;
123 case 'x':
124 xsiz = atoi(argv[++an]);
125 break;
126 case 'y':
127 ysiz = atoi(argv[++an]);
128 break;
129 case 'b':
130 setcolr(bgcolr, atof(argv[an+1]),
131 atof(argv[an+2]),
132 atof(argv[an+3]));
133 an += 3;
134 break;
135 case 'a':
136 ncolumns = atoi(argv[++an]);
137 break;
138 case 's':
139 spacing = atoi(argv[++an]);
140 break;
141 case 'o':
142 curx = x0 = atoi(argv[++an]);
143 cury = atoi(argv[++an]);
144 break;
145 case 'l':
146 switch (argv[an][2]) {
147 case 'a':
148 autolabel++;
149 break;
150 case 'h':
151 labelht = atoi(argv[++an]);
152 break;
153 case '\0':
154 goto dofiles;
155 default:
156 goto userr;
157 }
158 break;
159 case '\0':
160 case 't':
161 goto dofiles;
162 default:
163 goto userr;
164 }
165 dofiles:
166 newheader("RADIANCE", stdout);
167 fputnow(stdout);
168 for (nfile = 0; an < argc; nfile++) {
169 if (nfile >= MAXFILE)
170 goto toomany;
171 thislabel = NULL;
172 input[nfile].flags = 0;
173 xsgn = ysgn = '-';
174 while (an < argc && (argv[an][0] == '-' || argv[an][0] == '+'
175 || argv[an][0] == '=')) {
176 switch (argv[an][1]) {
177 case 't':
178 checkthresh = 1;
179 if (argv[an][0] == '-') {
180 input[nfile].flags |= HASMIN;
181 input[nfile].thmin = atof(argv[an+1]);
182 } else if (argv[an][0] == '+') {
183 input[nfile].flags |= HASMAX;
184 input[nfile].thmax = atof(argv[an+1]);
185 } else
186 goto userr;
187 an++;
188 break;
189 case 'l':
190 if (strcmp(argv[an], "-l"))
191 goto userr;
192 thislabel = argv[++an];
193 break;
194 case '+':
195 case '-':
196 case '0':
197 if (argv[an][0] != '=')
198 goto userr;
199 xsgn = argv[an][1];
200 ysgn = argv[an][2];
201 if (ysgn != '+' && ysgn != '-' && ysgn != '0')
202 goto userr;
203 break;
204 case '\0':
205 if (argv[an][0] == '-')
206 goto getfile;
207 goto userr;
208 default:
209 goto userr;
210 }
211 an++;
212 }
213 getfile:
214 if (argc-an < (ncolumns ? 1 : 3))
215 goto userr;
216 if (autolabel && thislabel == NULL)
217 thislabel = argv[an];
218 if (!strcmp(argv[an], "-")) {
219 input[nfile].name = StandardInput;
220 input[nfile].fp = stdin;
221 } else {
222 if (argv[an][0] == '!') {
223 input[nfile].name = Command;
224 input[nfile].fp = popen(argv[an]+1, "r");
225 } else {
226 input[nfile].name = argv[an];
227 input[nfile].fp = fopen(argv[an], "r");
228 }
229 if (input[nfile].fp == NULL) {
230 perror(argv[an]);
231 quit(1);
232 }
233 }
234 an++;
235 /* get header */
236 if (echoheader)
237 printf("%s:\n", input[nfile].name);
238 getheader(input[nfile].fp, headline, NULL);
239 if (wrongformat) {
240 fprintf(stderr, "%s: incompatible input format\n",
241 input[nfile].name);
242 quit(1);
243 }
244 /* get picture size */
245 if (fgetresolu(&input[nfile].xres, &input[nfile].yres,
246 input[nfile].fp) < 0) {
247 fprintf(stderr, "%s: bad picture size\n",
248 input[nfile].name);
249 quit(1);
250 }
251 if (ncolumns) {
252 if (curcol >= abs(ncolumns)) {
253 cury = ymax + spacing;
254 curx = x0;
255 curcol = 0;
256 }
257 input[nfile].xloc = curx;
258 input[nfile].yloc = cury;
259 curx += input[nfile].xres + spacing;
260 curcol++;
261 } else {
262 input[nfile].xloc = atoi(argv[an++]);
263 if (xsgn == '+')
264 input[nfile].xloc -= input[nfile].xres;
265 else if (xsgn == '0')
266 input[nfile].xloc -= input[nfile].xres/2;
267 input[nfile].yloc = atoi(argv[an++]);
268 if (ysgn == '+')
269 input[nfile].yloc -= input[nfile].yres;
270 else if (ysgn == '0')
271 input[nfile].yloc -= input[nfile].yres/2;
272 }
273 if (input[nfile].xloc < xmin)
274 xmin = input[nfile].xloc;
275 if (input[nfile].yloc < ymin)
276 ymin = input[nfile].yloc;
277 if (input[nfile].xloc+input[nfile].xres > xmax)
278 xmax = input[nfile].xloc+input[nfile].xres;
279 if (input[nfile].yloc+input[nfile].yres > ymax)
280 ymax = input[nfile].yloc+input[nfile].yres;
281 if (thislabel != NULL) {
282 if (++nfile >= MAXFILE)
283 goto toomany;
284 input[nfile].name = Label;
285 input[nfile].flags = 0;
286 input[nfile].xres = input[nfile-1].xres;
287 input[nfile].yres = labelht;
288 if ((input[nfile].fp = lblopen(thislabel,
289 &input[nfile].xres,
290 &input[nfile].yres)) == NULL)
291 goto labelerr;
292 input[nfile].xloc = input[nfile-1].xloc;
293 input[nfile].yloc = input[nfile-1].yloc +
294 input[nfile-1].yres-input[nfile].yres;
295 }
296 }
297 if (xsiz <= 0)
298 xsiz = xmax;
299 else if (xsiz > xmax)
300 xmax = xsiz;
301 if (ysiz <= 0)
302 ysiz = ymax;
303 else if (ysiz > ymax)
304 ymax = ysiz;
305 if (ncolumns < 0) { /* reverse rows if requested */
306 int i = nfile;
307 while (i--)
308 input[i].yloc = ymax - input[i].yres - input[i].yloc;
309 }
310 /* add new header info. */
311 printargs(argc, argv, stdout);
312 if (strcmp(ourfmt, PICFMT))
313 fputformat(ourfmt, stdout); /* print format if known */
314 putchar('\n');
315 fprtresolu(xsiz, ysiz, stdout);
316
317 compos();
318
319 quit(0);
320 userr:
321 fprintf(stderr,
322 "Usage: %s [-h][-x xr][-y yr][-b r g b][-a n][-s p][-o x0 y0][-la][-lh h] ",
323 progname);
324 fprintf(stderr, "[-t min1][+t max1][-l lab][=SS] pic1 x1 y1 ..\n");
325 quit(1);
326 toomany:
327 fprintf(stderr, "%s: only %d files and labels allowed\n",
328 progname, MAXFILE);
329 quit(1);
330 labelerr:
331 fprintf(stderr, "%s: error opening label\n", progname);
332 quit(1);
333 return 1; /* pro forma return */
334 }
335
336
337 static void
338 compos(void) /* composite pictures */
339 {
340 COLR *scanin, *scanout;
341 int y;
342 register int x, i;
343
344 scanin = (COLR *)malloc((xmax-xmin)*sizeof(COLR));
345 if (scanin == NULL)
346 goto memerr;
347 scanin -= xmin;
348 if (checkthresh) {
349 scanout = (COLR *)malloc(xsiz*sizeof(COLR));
350 if (scanout == NULL)
351 goto memerr;
352 } else
353 scanout = scanin;
354 for (y = ymax-1; y >= 0; y--) {
355 for (x = 0; x < xsiz; x++)
356 copycolr(scanout[x], bgcolr);
357 for (i = 0; i < nfile; i++) {
358 if (input[i].yloc > y ||
359 input[i].yloc+input[i].yres <= y)
360 continue;
361 if (freadcolrs(scanin+input[i].xloc,
362 input[i].xres, input[i].fp) < 0) {
363 fprintf(stderr, "%s: read error (y==%d)\n",
364 input[i].name,
365 y-input[i].yloc);
366 quit(1);
367 }
368 if (y >= ysiz)
369 continue;
370 if (checkthresh) {
371 x = input[i].xloc+input[i].xres;
372 if (x > xsiz)
373 x = xsiz;
374 for (x--; x >= 0 && x >= input[i].xloc; x--) {
375 if (input[i].flags & HASMIN &&
376 cmpcolr(scanin[x], input[i].thmin) <= 0)
377 continue;
378 if (input[i].flags & HASMAX &&
379 cmpcolr(scanin[x], input[i].thmax) >= 0)
380 continue;
381 copycolr(scanout[x], scanin[x]);
382 }
383 }
384 }
385 if (y >= ysiz)
386 continue;
387 if (fwritecolrs(scanout, xsiz, stdout) < 0) {
388 perror(progname);
389 quit(1);
390 }
391 }
392 /* read remainders from streams */
393 for (i = 0; i < nfile; i++)
394 if (input[i].name[0] == '<')
395 while (getc(input[i].fp) != EOF)
396 ;
397 return;
398 memerr:
399 perror(progname);
400 quit(1);
401 }
402
403
404 static int
405 cmpcolr( /* compare COLR to luminance */
406 register COLR c1,
407 double lv2
408 )
409 {
410 double lv1 = .0;
411
412 if (c1[EXP])
413 lv1 = ldexp((double)normbright(c1), (int)c1[EXP]-(COLXS+8));
414 if (lv1 < lv2) return(-1);
415 if (lv1 > lv2) return(1);
416 return(0);
417 }
418
419
420 static FILE *
421 lblopen( /* open pipe to label generator */
422 char *s,
423 int *xp,
424 int *yp
425 )
426 {
427 char com[PATH_MAX];
428 FILE *fp;
429
430 sprintf(com, "psign -s -.15 -a 2 -x %d -y %d '%.90s'", *xp, *yp, s);
431 if ((fp = popen(com, "r")) == NULL)
432 return(NULL);
433 if (checkheader(fp, COLRFMT, NULL) < 0)
434 goto err;
435 if (fgetresolu(xp, yp, fp) < 0)
436 goto err;
437 return(fp);
438 err:
439 pclose(fp);
440 return(NULL);
441 }
442
443
444 void
445 quit(code) /* exit gracefully */
446 int code;
447 {
448 register int i;
449 /* close input files */
450 for (i = 0; i < nfile; i++)
451 if (input[i].name == Command || input[i].name == Label)
452 pclose(input[i].fp);
453 else
454 fclose(input[i].fp);
455 exit(code);
456 }