]> git.ipfire.org Git - thirdparty/cups.git/blame - cups/file.c
Several libcups files did not have the Apple license exception notice (STR #4361)
[thirdparty/cups.git] / cups / file.c
CommitLineData
ef416fc2 1/*
f2d18633 2 * "$Id$"
ef416fc2 3 *
da003234 4 * File functions for CUPS.
ef416fc2 5 *
da003234
MS
6 * Since stdio files max out at 256 files on many systems, we have to
7 * write similar functions without this limit. At the same time, using
8 * our own file functions allows us to provide transparent support of
9 * gzip'd print files, PPD files, etc.
ef416fc2 10 *
7e86f2f6 11 * Copyright 2007-2014 by Apple Inc.
da003234 12 * Copyright 1997-2007 by Easy Software Products, all rights reserved.
ef416fc2 13 *
da003234
MS
14 * These coded instructions, statements, and computer programs are the
15 * property of Apple Inc. and are protected by Federal copyright
16 * law. Distribution and use rights are outlined in the file "LICENSE.txt"
17 * which should have been included with this file. If this file is
18 * file is missing or damaged, see the license at "http://www.cups.org/".
6539a0af
MS
19 *
20 * This file is subject to the Apple OS-Developed Software exception.
ef416fc2 21 */
22
23/*
24 * Include necessary headers...
25 */
26
b9faaae1 27#include "file-private.h"
c7017ecc
MS
28#include <sys/stat.h>
29#include <sys/types.h>
ef416fc2 30
31
32/*
33 * Local functions...
34 */
35
36#ifdef HAVE_LIBZ
37static ssize_t cups_compress(cups_file_t *fp, const char *buf, size_t bytes);
38#endif /* HAVE_LIBZ */
39static ssize_t cups_fill(cups_file_t *fp);
c7017ecc 40static int cups_open(const char *filename, int mode);
ef416fc2 41static ssize_t cups_read(cups_file_t *fp, char *buf, size_t bytes);
42static ssize_t cups_write(cups_file_t *fp, const char *buf, size_t bytes);
43
44
83e08001 45#ifndef WIN32
22c9029b
MS
46/*
47 * '_cupsFileCheck()' - Check the permissions of the given filename.
48 */
49
50_cups_fc_result_t /* O - Check result */
51_cupsFileCheck(
52 const char *filename, /* I - Filename to check */
53 _cups_fc_filetype_t filetype, /* I - Type of file checks? */
54 int dorootchecks, /* I - Check for root permissions? */
55 _cups_fc_func_t cb, /* I - Callback function */
56 void *context) /* I - Context pointer for callback */
f228370c 57
22c9029b
MS
58{
59 struct stat fileinfo; /* File information */
60 char message[1024], /* Message string */
61 temp[1024], /* Parent directory filename */
62 *ptr; /* Pointer into parent directory */
63 _cups_fc_result_t result; /* Check result */
64
65
88f9aafc
MS
66 /*
67 * Does the filename contain a relative path ("../")?
68 */
69
70 if (strstr(filename, "../"))
71 {
72 /*
73 * Yes, fail it!
74 */
75
76 result = _CUPS_FILE_CHECK_RELATIVE_PATH;
77 goto finishup;
78 }
79
22c9029b
MS
80 /*
81 * Does the program even exist and is it accessible?
82 */
83
84 if (stat(filename, &fileinfo))
85 {
86 /*
87 * Nope...
88 */
89
90 result = _CUPS_FILE_CHECK_MISSING;
91 goto finishup;
92 }
93
94 /*
95 * Check the execute bit...
96 */
97
98 result = _CUPS_FILE_CHECK_OK;
99
100 switch (filetype)
101 {
102 case _CUPS_FILE_CHECK_DIRECTORY :
103 if (!S_ISDIR(fileinfo.st_mode))
104 result = _CUPS_FILE_CHECK_WRONG_TYPE;
105 break;
106
107 default :
108 if (!S_ISREG(fileinfo.st_mode))
109 result = _CUPS_FILE_CHECK_WRONG_TYPE;
110 break;
111 }
112
113 if (result)
114 goto finishup;
115
116 /*
117 * Are we doing root checks?
118 */
119
120 if (!dorootchecks)
121 {
122 /*
123 * Nope, so anything (else) goes...
124 */
125
126 goto finishup;
127 }
128
129 /*
130 * Verify permission of the file itself:
131 *
132 * 1. Must be owned by root
85dda01c 133 * 2. Must not be writable by group
22c9029b
MS
134 * 3. Must not be setuid
135 * 4. Must not be writable by others
136 */
137
138 if (fileinfo.st_uid || /* 1. Must be owned by root */
85dda01c 139 (fileinfo.st_mode & S_IWGRP) || /* 2. Must not be writable by group */
22c9029b
MS
140 (fileinfo.st_mode & S_ISUID) || /* 3. Must not be setuid */
141 (fileinfo.st_mode & S_IWOTH)) /* 4. Must not be writable by others */
142 {
143 result = _CUPS_FILE_CHECK_PERMISSIONS;
144 goto finishup;
145 }
146
147 if (filetype == _CUPS_FILE_CHECK_DIRECTORY ||
148 filetype == _CUPS_FILE_CHECK_FILE_ONLY)
149 goto finishup;
150
151 /*
152 * Now check the containing directory...
153 */
154
155 strlcpy(temp, filename, sizeof(temp));
156 if ((ptr = strrchr(temp, '/')) != NULL)
157 {
158 if (ptr == temp)
159 ptr[1] = '\0';
160 else
161 *ptr = '\0';
162 }
163
164 if (stat(temp, &fileinfo))
165 {
166 /*
167 * Doesn't exist?!?
168 */
169
170 result = _CUPS_FILE_CHECK_MISSING;
171 filetype = _CUPS_FILE_CHECK_DIRECTORY;
172 filename = temp;
173
174 goto finishup;
175 }
176
177 if (fileinfo.st_uid || /* 1. Must be owned by root */
85dda01c 178 (fileinfo.st_mode & S_IWGRP) || /* 2. Must not be writable by group */
22c9029b
MS
179 (fileinfo.st_mode & S_ISUID) || /* 3. Must not be setuid */
180 (fileinfo.st_mode & S_IWOTH)) /* 4. Must not be writable by others */
181 {
182 result = _CUPS_FILE_CHECK_PERMISSIONS;
183 filetype = _CUPS_FILE_CHECK_DIRECTORY;
184 filename = temp;
185 }
186
187 /*
188 * Common return point...
189 */
190
191 finishup:
192
193 if (cb)
194 {
195 cups_lang_t *lang = cupsLangDefault();
196 /* Localization information */
197
198 switch (result)
199 {
200 case _CUPS_FILE_CHECK_OK :
201 if (filetype == _CUPS_FILE_CHECK_DIRECTORY)
202 snprintf(message, sizeof(message),
203 _cupsLangString(lang, _("Directory \"%s\" permissions OK "
204 "(0%o/uid=%d/gid=%d).")),
205 filename, fileinfo.st_mode, (int)fileinfo.st_uid,
206 (int)fileinfo.st_gid);
207 else
208 snprintf(message, sizeof(message),
209 _cupsLangString(lang, _("File \"%s\" permissions OK "
210 "(0%o/uid=%d/gid=%d).")),
211 filename, fileinfo.st_mode, (int)fileinfo.st_uid,
212 (int)fileinfo.st_gid);
213 break;
214
215 case _CUPS_FILE_CHECK_MISSING :
216 if (filetype == _CUPS_FILE_CHECK_DIRECTORY)
217 snprintf(message, sizeof(message),
218 _cupsLangString(lang, _("Directory \"%s\" not available: "
219 "%s")),
220 filename, strerror(errno));
221 else
222 snprintf(message, sizeof(message),
223 _cupsLangString(lang, _("File \"%s\" not available: %s")),
224 filename, strerror(errno));
225 break;
226
227 case _CUPS_FILE_CHECK_PERMISSIONS :
228 if (filetype == _CUPS_FILE_CHECK_DIRECTORY)
229 snprintf(message, sizeof(message),
230 _cupsLangString(lang, _("Directory \"%s\" has insecure "
231 "permissions "
232 "(0%o/uid=%d/gid=%d).")),
233 filename, fileinfo.st_mode, (int)fileinfo.st_uid,
234 (int)fileinfo.st_gid);
235 else
236 snprintf(message, sizeof(message),
237 _cupsLangString(lang, _("File \"%s\" has insecure "
238 "permissions "
239 "(0%o/uid=%d/gid=%d).")),
240 filename, fileinfo.st_mode, (int)fileinfo.st_uid,
241 (int)fileinfo.st_gid);
242 break;
243
244 case _CUPS_FILE_CHECK_WRONG_TYPE :
245 if (filetype == _CUPS_FILE_CHECK_DIRECTORY)
246 snprintf(message, sizeof(message),
247 _cupsLangString(lang, _("Directory \"%s\" is a file.")),
248 filename);
249 else
250 snprintf(message, sizeof(message),
251 _cupsLangString(lang, _("File \"%s\" is a directory.")),
252 filename);
253 break;
88f9aafc
MS
254
255 case _CUPS_FILE_CHECK_RELATIVE_PATH :
256 if (filetype == _CUPS_FILE_CHECK_DIRECTORY)
257 snprintf(message, sizeof(message),
258 _cupsLangString(lang, _("Directory \"%s\" contains a "
259 "relative path.")), filename);
260 else
261 snprintf(message, sizeof(message),
262 _cupsLangString(lang, _("File \"%s\" contains a relative "
263 "path.")), filename);
264 break;
22c9029b
MS
265 }
266
267 (*cb)(context, result, message);
268 }
269
270 return (result);
271}
272
273
274/*
275 * '_cupsFileCheckFilter()' - Report file check results as CUPS filter messages.
276 */
277
278void
279_cupsFileCheckFilter(
280 void *context, /* I - Context pointer (unused) */
281 _cups_fc_result_t result, /* I - Result code */
282 const char *message) /* I - Message text */
283{
284 const char *prefix; /* Messaging prefix */
285
286
321d8d57
MS
287 (void)context;
288
22c9029b
MS
289 switch (result)
290 {
291 case _CUPS_FILE_CHECK_OK :
292 prefix = "DEBUG2";
293 break;
294
295 case _CUPS_FILE_CHECK_MISSING :
296 case _CUPS_FILE_CHECK_WRONG_TYPE :
297 prefix = "ERROR";
298 fputs("STATE: +cups-missing-filter-warning\n", stderr);
299 break;
300
301 case _CUPS_FILE_CHECK_PERMISSIONS :
88f9aafc 302 case _CUPS_FILE_CHECK_RELATIVE_PATH :
22c9029b
MS
303 prefix = "ERROR";
304 fputs("STATE: +cups-insecure-filter-warning\n", stderr);
305 break;
306 }
307
308 fprintf(stderr, "%s: %s\n", prefix, message);
309}
83e08001 310#endif /* !WIN32 */
22c9029b
MS
311
312
ef416fc2 313/*
314 * 'cupsFileClose()' - Close a CUPS file.
5a738aea 315 *
f3c17241 316 * @since CUPS 1.2/OS X 10.5@
ef416fc2 317 */
318
319int /* O - 0 on success, -1 on error */
320cupsFileClose(cups_file_t *fp) /* I - CUPS file */
321{
322 int fd; /* File descriptor */
323 char mode; /* Open mode */
324 int status; /* Return status */
80ca4592 325 int is_stdio; /* Is a stdio file? */
ef416fc2 326
327
e07d4801 328 DEBUG_printf(("cupsFileClose(fp=%p)", fp));
ef416fc2 329
330 /*
331 * Range check...
332 */
333
334 if (!fp)
335 return (-1);
336
337 /*
338 * Flush pending write data...
339 */
340
341 if (fp->mode == 'w')
342 status = cupsFileFlush(fp);
343 else
344 status = 0;
345
346#ifdef HAVE_LIBZ
347 if (fp->compressed && status >= 0)
348 {
349 if (fp->mode == 'r')
350 {
351 /*
352 * Free decompression data...
353 */
354
355 inflateEnd(&fp->stream);
356 }
357 else
358 {
359 /*
360 * Flush any remaining compressed data...
361 */
362
363 unsigned char trailer[8]; /* Trailer CRC and length */
364 int done; /* Done writing... */
365
366
367 fp->stream.avail_in = 0;
368
369 for (done = 0;;)
370 {
371 if (fp->stream.next_out > fp->cbuf)
372 {
373 if (cups_write(fp, (char *)fp->cbuf,
7e86f2f6 374 (size_t)(fp->stream.next_out - fp->cbuf)) < 0)
ef416fc2 375 status = -1;
376
377 fp->stream.next_out = fp->cbuf;
378 fp->stream.avail_out = sizeof(fp->cbuf);
379 }
380
381 if (done || status < 0)
382 break;
383
384 done = deflate(&fp->stream, Z_FINISH) == Z_STREAM_END &&
385 fp->stream.next_out == fp->cbuf;
386 }
387
388 /*
389 * Write the CRC and length...
390 */
391
7e86f2f6
MS
392 trailer[0] = (unsigned char)fp->crc;
393 trailer[1] = (unsigned char)(fp->crc >> 8);
394 trailer[2] = (unsigned char)(fp->crc >> 16);
395 trailer[3] = (unsigned char)(fp->crc >> 24);
396 trailer[4] = (unsigned char)fp->pos;
397 trailer[5] = (unsigned char)(fp->pos >> 8);
398 trailer[6] = (unsigned char)(fp->pos >> 16);
399 trailer[7] = (unsigned char)(fp->pos >> 24);
ef416fc2 400
401 if (cups_write(fp, (char *)trailer, 8) < 0)
402 status = -1;
403
404 /*
405 * Free all memory used by the compression stream...
406 */
407
408 deflateEnd(&(fp->stream));
409 }
410 }
411#endif /* HAVE_LIBZ */
412
413 /*
414 * Save the file descriptor we used and free memory...
415 */
416
80ca4592 417 fd = fp->fd;
418 mode = fp->mode;
419 is_stdio = fp->is_stdio;
ef416fc2 420
75bd9771
MS
421 if (fp->printf_buffer)
422 free(fp->printf_buffer);
423
ef416fc2 424 free(fp);
425
426 /*
427 * Close the file, returning the close status...
428 */
429
430 if (mode == 's')
431 {
87e98392 432 if (httpAddrClose(NULL, fd) < 0)
ef416fc2 433 status = -1;
434 }
80ca4592 435 else if (!is_stdio)
ef416fc2 436 {
437 if (close(fd) < 0)
438 status = -1;
439 }
440
441 return (status);
442}
443
444
445/*
446 * 'cupsFileCompression()' - Return whether a file is compressed.
5a738aea 447 *
f3c17241 448 * @since CUPS 1.2/OS X 10.5@
ef416fc2 449 */
450
5a738aea 451int /* O - @code CUPS_FILE_NONE@ or @code CUPS_FILE_GZIP@ */
ef416fc2 452cupsFileCompression(cups_file_t *fp) /* I - CUPS file */
453{
80ca4592 454 return (fp ? fp->compressed : CUPS_FILE_NONE);
ef416fc2 455}
456
457
458/*
459 * 'cupsFileEOF()' - Return the end-of-file status.
5a738aea 460 *
f3c17241 461 * @since CUPS 1.2/OS X 10.5@
ef416fc2 462 */
463
5a738aea 464int /* O - 1 on end of file, 0 otherwise */
ef416fc2 465cupsFileEOF(cups_file_t *fp) /* I - CUPS file */
466{
80ca4592 467 return (fp ? fp->eof : 1);
ef416fc2 468}
469
470
fa73b229 471/*
472 * 'cupsFileFind()' - Find a file using the specified path.
473 *
474 * This function allows the paths in the path string to be separated by
475 * colons (UNIX standard) or semicolons (Windows standard) and stores the
476 * result in the buffer supplied. If the file cannot be found in any of
5a738aea
MS
477 * the supplied paths, @code NULL@ is returned. A @code NULL@ path only
478 * matches the current directory.
479 *
f3c17241 480 * @since CUPS 1.2/OS X 10.5@
fa73b229 481 */
482
5a738aea 483const char * /* O - Full path to file or @code NULL@ if not found */
fa73b229 484cupsFileFind(const char *filename, /* I - File to find */
485 const char *path, /* I - Colon/semicolon-separated path */
4400e98d 486 int executable, /* I - 1 = executable files, 0 = any file/dir */
487 char *buffer, /* I - Filename buffer */
fa73b229 488 int bufsize) /* I - Size of filename buffer */
489{
490 char *bufptr, /* Current position in buffer */
491 *bufend; /* End of buffer */
492
493
494 /*
495 * Range check input...
496 */
497
e07d4801
MS
498 DEBUG_printf(("cupsFileFind(filename=\"%s\", path=\"%s\", executable=%d, "
499 "buffer=%p, bufsize=%d)", filename, path, executable, buffer,
500 bufsize));
501
fa73b229 502 if (!filename || !buffer || bufsize < 2)
503 return (NULL);
504
505 if (!path)
506 {
507 /*
508 * No path, so check current directory...
509 */
510
511 if (!access(filename, 0))
512 {
07623986 513 strlcpy(buffer, filename, (size_t)bufsize);
fa73b229 514 return (buffer);
515 }
516 else
517 return (NULL);
518 }
519
520 /*
521 * Now check each path and return the first match...
522 */
523
524 bufend = buffer + bufsize - 1;
525 bufptr = buffer;
526
527 while (*path)
528 {
b86bc4cf 529#ifdef WIN32
530 if (*path == ';' || (*path == ':' && ((bufptr - buffer) > 1 || !isalpha(buffer[0] & 255))))
531#else
fa73b229 532 if (*path == ';' || *path == ':')
b86bc4cf 533#endif /* WIN32 */
fa73b229 534 {
535 if (bufptr > buffer && bufptr[-1] != '/' && bufptr < bufend)
536 *bufptr++ = '/';
537
07623986 538 strlcpy(bufptr, filename, (size_t)(bufend - bufptr));
fa73b229 539
4400e98d 540#ifdef WIN32
fa73b229 541 if (!access(buffer, 0))
4400e98d 542#else
543 if (!access(buffer, executable ? X_OK : 0))
544#endif /* WIN32 */
b86bc4cf 545 {
e07d4801 546 DEBUG_printf(("1cupsFileFind: Returning \"%s\"", buffer));
fa73b229 547 return (buffer);
b86bc4cf 548 }
fa73b229 549
550 bufptr = buffer;
551 }
552 else if (bufptr < bufend)
553 *bufptr++ = *path;
554
555 path ++;
556 }
557
558 /*
559 * Check the last path...
560 */
561
562 if (bufptr > buffer && bufptr[-1] != '/' && bufptr < bufend)
563 *bufptr++ = '/';
564
07623986 565 strlcpy(bufptr, filename, (size_t)(bufend - bufptr));
fa73b229 566
567 if (!access(buffer, 0))
b86bc4cf 568 {
e07d4801 569 DEBUG_printf(("1cupsFileFind: Returning \"%s\"", buffer));
fa73b229 570 return (buffer);
b86bc4cf 571 }
fa73b229 572 else
b86bc4cf 573 {
e07d4801 574 DEBUG_puts("1cupsFileFind: Returning NULL");
fa73b229 575 return (NULL);
b86bc4cf 576 }
fa73b229 577}
578
579
ef416fc2 580/*
581 * 'cupsFileFlush()' - Flush pending output.
5a738aea 582 *
f3c17241 583 * @since CUPS 1.2/OS X 10.5@
ef416fc2 584 */
585
586int /* O - 0 on success, -1 on error */
587cupsFileFlush(cups_file_t *fp) /* I - CUPS file */
588{
2abf387c 589 ssize_t bytes; /* Bytes to write */
ef416fc2 590
591
e07d4801 592 DEBUG_printf(("cupsFileFlush(fp=%p)", fp));
ef416fc2 593
594 /*
595 * Range check input...
596 */
597
598 if (!fp || fp->mode != 'w')
599 {
e07d4801 600 DEBUG_puts("1cupsFileFlush: Attempt to flush a read-only file...");
ef416fc2 601 return (-1);
602 }
603
b86bc4cf 604 bytes = (ssize_t)(fp->ptr - fp->buf);
ef416fc2 605
e07d4801 606 DEBUG_printf(("2cupsFileFlush: Flushing " CUPS_LLFMT " bytes...",
634763e8 607 CUPS_LLCAST bytes));
ecdc0628 608
ef416fc2 609 if (bytes > 0)
610 {
611#ifdef HAVE_LIBZ
612 if (fp->compressed)
7e86f2f6 613 bytes = cups_compress(fp, fp->buf, (size_t)bytes);
ef416fc2 614 else
615#endif /* HAVE_LIBZ */
7e86f2f6 616 bytes = cups_write(fp, fp->buf, (size_t)bytes);
ef416fc2 617
618 if (bytes < 0)
619 return (-1);
620
621 fp->ptr = fp->buf;
622 }
f14324a7 623
ef416fc2 624 return (0);
625}
626
627
628/*
629 * 'cupsFileGetChar()' - Get a single character from a file.
5a738aea 630 *
f3c17241 631 * @since CUPS 1.2/OS X 10.5@
ef416fc2 632 */
633
5a738aea 634int /* O - Character or -1 on end of file */
ef416fc2 635cupsFileGetChar(cups_file_t *fp) /* I - CUPS file */
636{
637 /*
638 * Range check input...
639 */
640
641 if (!fp || (fp->mode != 'r' && fp->mode != 's'))
b86bc4cf 642 {
f14324a7 643 DEBUG_puts("5cupsFileGetChar: Bad arguments!");
ef416fc2 644 return (-1);
b86bc4cf 645 }
ef416fc2 646
647 /*
648 * If the input buffer is empty, try to read more data...
649 */
650
651 if (fp->ptr >= fp->end)
652 if (cups_fill(fp) < 0)
b86bc4cf 653 {
f14324a7 654 DEBUG_puts("5cupsFileGetChar: Unable to fill buffer!");
ef416fc2 655 return (-1);
b86bc4cf 656 }
ef416fc2 657
658 /*
659 * Return the next character in the buffer...
660 */
661
f14324a7 662 DEBUG_printf(("5cupsFileGetChar: Returning %d...", *(fp->ptr) & 255));
b86bc4cf 663
634763e8
MS
664 fp->pos ++;
665
f14324a7 666 DEBUG_printf(("6cupsFileGetChar: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 667
ef416fc2 668 return (*(fp->ptr)++ & 255);
669}
670
671
672/*
22c9029b 673 * 'cupsFileGetConf()' - Get a line from a configuration file.
5a738aea 674 *
f3c17241 675 * @since CUPS 1.2/OS X 10.5@
ef416fc2 676 */
677
5a738aea 678char * /* O - Line read or @code NULL@ on end of file or error */
ef416fc2 679cupsFileGetConf(cups_file_t *fp, /* I - CUPS file */
680 char *buf, /* O - String buffer */
681 size_t buflen, /* I - Size of string buffer */
682 char **value, /* O - Pointer to value */
683 int *linenum) /* IO - Current line number */
684{
685 char *ptr; /* Pointer into line */
686
687
688 /*
689 * Range check input...
690 */
691
e07d4801
MS
692 DEBUG_printf(("2cupsFileGetConf(fp=%p, buf=%p, buflen=" CUPS_LLFMT
693 ", value=%p, linenum=%p)", fp, buf, CUPS_LLCAST buflen,
634763e8
MS
694 value, linenum));
695
ef416fc2 696 if (!fp || (fp->mode != 'r' && fp->mode != 's') ||
697 !buf || buflen < 2 || !value)
698 {
699 if (value)
700 *value = NULL;
701
702 return (NULL);
703 }
704
705 /*
706 * Read the next non-comment line...
707 */
708
709 *value = NULL;
f7deaa1a 710
ef416fc2 711 while (cupsFileGets(fp, buf, buflen))
712 {
713 (*linenum) ++;
714
715 /*
716 * Strip any comments...
717 */
718
719 if ((ptr = strchr(buf, '#')) != NULL)
720 {
f7deaa1a 721 if (ptr > buf && ptr[-1] == '\\')
ef416fc2 722 {
f7deaa1a 723 // Unquote the #...
724 _cups_strcpy(ptr - 1, ptr);
ef416fc2 725 }
f7deaa1a 726 else
727 {
728 // Strip the comment and any trailing whitespace...
729 while (ptr > buf)
730 {
7cf5915e 731 if (!_cups_isspace(ptr[-1]))
f7deaa1a 732 break;
733
734 ptr --;
735 }
ef416fc2 736
f7deaa1a 737 *ptr = '\0';
738 }
ef416fc2 739 }
740
741 /*
742 * Strip leading whitespace...
743 */
744
7cf5915e 745 for (ptr = buf; _cups_isspace(*ptr); ptr ++);
ef416fc2 746
747 if (ptr > buf)
748 _cups_strcpy(buf, ptr);
749
750 /*
751 * See if there is anything left...
752 */
753
754 if (buf[0])
755 {
756 /*
757 * Yes, grab any value and return...
758 */
759
760 for (ptr = buf; *ptr; ptr ++)
7cf5915e 761 if (_cups_isspace(*ptr))
ef416fc2 762 break;
763
764 if (*ptr)
765 {
766 /*
767 * Have a value, skip any other spaces...
768 */
769
7cf5915e 770 while (_cups_isspace(*ptr))
ef416fc2 771 *ptr++ = '\0';
772
773 if (*ptr)
774 *value = ptr;
775
776 /*
777 * Strip trailing whitespace and > for lines that begin with <...
778 */
779
780 ptr += strlen(ptr) - 1;
781
782 if (buf[0] == '<' && *ptr == '>')
783 *ptr-- = '\0';
784 else if (buf[0] == '<' && *ptr != '>')
785 {
786 /*
787 * Syntax error...
788 */
789
790 *value = NULL;
791 return (buf);
792 }
793
7cf5915e 794 while (ptr > *value && _cups_isspace(*ptr))
ef416fc2 795 *ptr-- = '\0';
796 }
797
798 /*
799 * Return the line...
800 */
801
802 return (buf);
803 }
804 }
805
806 return (NULL);
807}
808
809
80ca4592 810/*
811 * 'cupsFileGetLine()' - Get a CR and/or LF-terminated line that may
812 * contain binary data.
813 *
5a738aea
MS
814 * This function differs from @link cupsFileGets@ in that the trailing CR
815 * and LF are preserved, as is any binary data on the line. The buffer is
816 * nul-terminated, however you should use the returned length to determine
80ca4592 817 * the number of bytes on the line.
5a738aea 818 *
f3c17241 819 * @since CUPS 1.2/OS X 10.5@
80ca4592 820 */
821
5a738aea 822size_t /* O - Number of bytes on line or 0 on end of file */
80ca4592 823cupsFileGetLine(cups_file_t *fp, /* I - File to read from */
824 char *buf, /* I - Buffer */
825 size_t buflen) /* I - Size of buffer */
826{
827 int ch; /* Character from file */
828 char *ptr, /* Current position in line buffer */
829 *end; /* End of line buffer */
830
831
832 /*
833 * Range check input...
834 */
835
e07d4801 836 DEBUG_printf(("2cupsFileGetLine(fp=%p, buf=%p, buflen=" CUPS_LLFMT ")",
634763e8
MS
837 fp, buf, CUPS_LLCAST buflen));
838
80ca4592 839 if (!fp || (fp->mode != 'r' && fp->mode != 's') || !buf || buflen < 3)
840 return (0);
841
842 /*
843 * Now loop until we have a valid line...
844 */
845
846 for (ptr = buf, end = buf + buflen - 2; ptr < end ;)
847 {
848 if (fp->ptr >= fp->end)
849 if (cups_fill(fp) <= 0)
850 break;
851
852 *ptr++ = ch = *(fp->ptr)++;
634763e8 853 fp->pos ++;
80ca4592 854
855 if (ch == '\r')
856 {
857 /*
858 * Check for CR LF...
859 */
860
861 if (fp->ptr >= fp->end)
862 if (cups_fill(fp) <= 0)
863 break;
864
865 if (*(fp->ptr) == '\n')
634763e8 866 {
80ca4592 867 *ptr++ = *(fp->ptr)++;
634763e8
MS
868 fp->pos ++;
869 }
80ca4592 870
871 break;
872 }
873 else if (ch == '\n')
874 {
875 /*
876 * Line feed ends a line...
877 */
878
879 break;
880 }
881 }
882
883 *ptr = '\0';
884
e07d4801 885 DEBUG_printf(("4cupsFileGetLine: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 886
7e86f2f6 887 return ((size_t)(ptr - buf));
80ca4592 888}
889
890
ef416fc2 891/*
892 * 'cupsFileGets()' - Get a CR and/or LF-terminated line.
5a738aea 893 *
f3c17241 894 * @since CUPS 1.2/OS X 10.5@
ef416fc2 895 */
896
5a738aea 897char * /* O - Line read or @code NULL@ on end of file or error */
ef416fc2 898cupsFileGets(cups_file_t *fp, /* I - CUPS file */
899 char *buf, /* O - String buffer */
900 size_t buflen) /* I - Size of string buffer */
901{
902 int ch; /* Character from file */
903 char *ptr, /* Current position in line buffer */
904 *end; /* End of line buffer */
905
906
907 /*
908 * Range check input...
909 */
910
e07d4801 911 DEBUG_printf(("2cupsFileGets(fp=%p, buf=%p, buflen=" CUPS_LLFMT ")", fp, buf,
634763e8
MS
912 CUPS_LLCAST buflen));
913
ef416fc2 914 if (!fp || (fp->mode != 'r' && fp->mode != 's') || !buf || buflen < 2)
915 return (NULL);
916
917 /*
918 * Now loop until we have a valid line...
919 */
920
921 for (ptr = buf, end = buf + buflen - 1; ptr < end ;)
922 {
923 if (fp->ptr >= fp->end)
924 if (cups_fill(fp) <= 0)
925 {
926 if (ptr == buf)
927 return (NULL);
928 else
929 break;
930 }
931
932 ch = *(fp->ptr)++;
634763e8 933 fp->pos ++;
ef416fc2 934
935 if (ch == '\r')
936 {
937 /*
938 * Check for CR LF...
939 */
940
941 if (fp->ptr >= fp->end)
942 if (cups_fill(fp) <= 0)
943 break;
944
945 if (*(fp->ptr) == '\n')
634763e8
MS
946 {
947 fp->ptr ++;
948 fp->pos ++;
949 }
ef416fc2 950
951 break;
952 }
953 else if (ch == '\n')
954 {
955 /*
956 * Line feed ends a line...
957 */
958
959 break;
960 }
961 else
7e86f2f6 962 *ptr++ = (char)ch;
ef416fc2 963 }
964
965 *ptr = '\0';
966
e07d4801 967 DEBUG_printf(("4cupsFileGets: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 968
ef416fc2 969 return (buf);
970}
971
972
973/*
974 * 'cupsFileLock()' - Temporarily lock access to a file.
5a738aea 975 *
f3c17241 976 * @since CUPS 1.2/OS X 10.5@
ef416fc2 977 */
978
979int /* O - 0 on success, -1 on error */
5a738aea 980cupsFileLock(cups_file_t *fp, /* I - CUPS file */
ef416fc2 981 int block) /* I - 1 to wait for the lock, 0 to fail right away */
982{
983 /*
984 * Range check...
985 */
986
987 if (!fp || fp->mode == 's')
988 return (-1);
989
990 /*
991 * Try the lock...
992 */
993
994#ifdef WIN32
536bc2c6 995 return (_locking(fp->fd, block ? _LK_LOCK : _LK_NBLCK, 0));
ef416fc2 996#else
997 return (lockf(fp->fd, block ? F_LOCK : F_TLOCK, 0));
998#endif /* WIN32 */
999}
1000
1001
1002/*
1003 * 'cupsFileNumber()' - Return the file descriptor associated with a CUPS file.
5a738aea 1004 *
f3c17241 1005 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1006 */
1007
1008int /* O - File descriptor */
1009cupsFileNumber(cups_file_t *fp) /* I - CUPS file */
1010{
5a738aea
MS
1011 if (fp)
1012 return (fp->fd);
1013 else
1014 return (-1);
ef416fc2 1015}
1016
1017
1018/*
1019 * 'cupsFileOpen()' - Open a CUPS file.
5a738aea
MS
1020 *
1021 * The "mode" parameter can be "r" to read, "w" to write, overwriting any
1022 * existing file, "a" to append to an existing file or create a new file,
1023 * or "s" to open a socket connection.
1024 *
634763e8
MS
1025 * When opening for writing ("w"), an optional number from 1 to 9 can be
1026 * supplied which enables Flate compression of the file. Compression is
1027 * not supported for the "a" (append) mode.
5a738aea
MS
1028 *
1029 * When opening a socket connection, the filename is a string of the form
1030 * "address:port" or "hostname:port". The socket will make an IPv4 or IPv6
1031 * connection as needed, generally preferring IPv6 connections when there is
1032 * a choice.
1033 *
f3c17241 1034 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1035 */
1036
5a738aea 1037cups_file_t * /* O - CUPS file or @code NULL@ if the file or socket cannot be opened */
ef416fc2 1038cupsFileOpen(const char *filename, /* I - Name of file */
1039 const char *mode) /* I - Open mode */
1040{
1041 cups_file_t *fp; /* New CUPS file */
1042 int fd; /* File descriptor */
1043 char hostname[1024], /* Hostname */
1044 *portname; /* Port "name" (number or service) */
1045 http_addrlist_t *addrlist; /* Host address list */
1046
1047
e07d4801 1048 DEBUG_printf(("cupsFileOpen(filename=\"%s\", mode=\"%s\")", filename,
b423cd4c 1049 mode));
1050
ef416fc2 1051 /*
1052 * Range check input...
1053 */
1054
1055 if (!filename || !mode ||
634763e8
MS
1056 (*mode != 'r' && *mode != 'w' && *mode != 'a' && *mode != 's') ||
1057 (*mode == 'a' && isdigit(mode[1] & 255)))
ef416fc2 1058 return (NULL);
1059
1060 /*
1061 * Open the file...
1062 */
1063
1064 switch (*mode)
1065 {
1066 case 'a' : /* Append file */
c7017ecc
MS
1067 fd = cups_open(filename,
1068 O_RDWR | O_CREAT | O_APPEND | O_LARGEFILE | O_BINARY);
ef416fc2 1069 break;
1070
1071 case 'r' : /* Read file */
b86bc4cf 1072 fd = open(filename, O_RDONLY | O_LARGEFILE | O_BINARY, 0);
ef416fc2 1073 break;
1074
1075 case 'w' : /* Write file */
c7017ecc
MS
1076 fd = cups_open(filename, O_WRONLY | O_LARGEFILE | O_BINARY);
1077 if (fd < 0 && errno == ENOENT)
1078 {
1079 fd = cups_open(filename,
1080 O_WRONLY | O_CREAT | O_EXCL | O_LARGEFILE | O_BINARY);
1081 if (fd < 0 && errno == EEXIST)
1082 fd = cups_open(filename, O_WRONLY | O_LARGEFILE | O_BINARY);
1083 }
1084
1085 if (fd >= 0)
1086#ifdef WIN32
1087 _chsize(fd, 0);
1088#else
1089 ftruncate(fd, 0);
1090#endif /* WIN32 */
ef416fc2 1091 break;
1092
1093 case 's' : /* Read/write socket */
1094 strlcpy(hostname, filename, sizeof(hostname));
1095 if ((portname = strrchr(hostname, ':')) != NULL)
1096 *portname++ = '\0';
1097 else
1098 return (NULL);
1099
1100 /*
1101 * Lookup the hostname and service...
1102 */
1103
1104 if ((addrlist = httpAddrGetList(hostname, AF_UNSPEC, portname)) == NULL)
1105 return (NULL);
1106
1107 /*
1108 * Connect to the server...
1109 */
1110
1111 if (!httpAddrConnect(addrlist, &fd))
1112 {
1113 httpAddrFreeList(addrlist);
1114 return (NULL);
1115 }
1116
1117 httpAddrFreeList(addrlist);
1118 break;
1119
1120 default : /* Remove bogus compiler warning... */
1121 return (NULL);
1122 }
1123
1124 if (fd < 0)
1125 return (NULL);
1126
1127 /*
1128 * Create the CUPS file structure...
1129 */
1130
1131 if ((fp = cupsFileOpenFd(fd, mode)) == NULL)
1132 {
1133 if (*mode == 's')
87e98392 1134 httpAddrClose(NULL, fd);
ef416fc2 1135 else
1136 close(fd);
1137 }
1138
1139 /*
1140 * Return it...
1141 */
1142
1143 return (fp);
1144}
1145
1146/*
1147 * 'cupsFileOpenFd()' - Open a CUPS file using a file descriptor.
5a738aea 1148 *
634763e8
MS
1149 * The "mode" parameter can be "r" to read, "w" to write, "a" to append,
1150 * or "s" to treat the file descriptor as a bidirectional socket connection.
5a738aea 1151 *
634763e8
MS
1152 * When opening for writing ("w"), an optional number from 1 to 9 can be
1153 * supplied which enables Flate compression of the file. Compression is
1154 * not supported for the "a" (append) mode.
5a738aea 1155 *
f3c17241 1156 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1157 */
1158
5a738aea 1159cups_file_t * /* O - CUPS file or @code NULL@ if the file could not be opened */
ef416fc2 1160cupsFileOpenFd(int fd, /* I - File descriptor */
1161 const char *mode) /* I - Open mode */
1162{
1163 cups_file_t *fp; /* New CUPS file */
1164
1165
e07d4801 1166 DEBUG_printf(("cupsFileOpenFd(fd=%d, mode=\"%s\")", fd, mode));
b423cd4c 1167
ef416fc2 1168 /*
1169 * Range check input...
1170 */
1171
1172 if (fd < 0 || !mode ||
634763e8
MS
1173 (*mode != 'r' && *mode != 'w' && *mode != 'a' && *mode != 's') ||
1174 (*mode == 'a' && isdigit(mode[1] & 255)))
ef416fc2 1175 return (NULL);
1176
1177 /*
1178 * Allocate memory...
1179 */
1180
1181 if ((fp = calloc(1, sizeof(cups_file_t))) == NULL)
1182 return (NULL);
1183
1184 /*
1185 * Open the file...
1186 */
1187
1188 fp->fd = fd;
1189
1190 switch (*mode)
1191 {
ef416fc2 1192 case 'a' :
634763e8
MS
1193 fp->pos = lseek(fd, 0, SEEK_END);
1194
1195 case 'w' :
ef416fc2 1196 fp->mode = 'w';
1197 fp->ptr = fp->buf;
1198 fp->end = fp->buf + sizeof(fp->buf);
1199
1200#ifdef HAVE_LIBZ
1201 if (mode[1] >= '1' && mode[1] <= '9')
1202 {
1203 /*
1204 * Open a compressed stream, so write the standard gzip file
1205 * header...
1206 */
1207
1208 unsigned char header[10]; /* gzip file header */
1209 time_t curtime; /* Current time */
1210
1211
1212 curtime = time(NULL);
1213 header[0] = 0x1f;
1214 header[1] = 0x8b;
1215 header[2] = Z_DEFLATED;
1216 header[3] = 0;
7e86f2f6
MS
1217 header[4] = (unsigned char)curtime;
1218 header[5] = (unsigned char)(curtime >> 8);
1219 header[6] = (unsigned char)(curtime >> 16);
1220 header[7] = (unsigned char)(curtime >> 24);
ef416fc2 1221 header[8] = 0;
1222 header[9] = 0x03;
1223
1224 cups_write(fp, (char *)header, 10);
1225
1226 /*
1227 * Initialize the compressor...
1228 */
1229
1230 deflateInit2(&(fp->stream), mode[1] - '0', Z_DEFLATED, -15, 8,
1231 Z_DEFAULT_STRATEGY);
1232
1233 fp->stream.next_out = fp->cbuf;
1234 fp->stream.avail_out = sizeof(fp->cbuf);
1235 fp->compressed = 1;
1236 fp->crc = crc32(0L, Z_NULL, 0);
1237 }
1238#endif /* HAVE_LIBZ */
1239 break;
1240
1241 case 'r' :
1242 fp->mode = 'r';
1243 break;
1244
1245 case 's' :
1246 fp->mode = 's';
1247 break;
1248
1249 default : /* Remove bogus compiler warning... */
1250 return (NULL);
1251 }
1252
1253 /*
1254 * Don't pass this file to child processes...
1255 */
1256
1257#ifndef WIN32
1258 fcntl(fp->fd, F_SETFD, fcntl(fp->fd, F_GETFD) | FD_CLOEXEC);
1259#endif /* !WIN32 */
1260
1261 return (fp);
1262}
1263
1264
1265/*
1266 * 'cupsFilePeekChar()' - Peek at the next character from a file.
5a738aea 1267 *
f3c17241 1268 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1269 */
1270
5a738aea 1271int /* O - Character or -1 on end of file */
ef416fc2 1272cupsFilePeekChar(cups_file_t *fp) /* I - CUPS file */
1273{
1274 /*
1275 * Range check input...
1276 */
1277
1278 if (!fp || (fp->mode != 'r' && fp->mode != 's'))
1279 return (-1);
1280
1281 /*
1282 * If the input buffer is empty, try to read more data...
1283 */
1284
1285 if (fp->ptr >= fp->end)
1286 if (cups_fill(fp) < 0)
1287 return (-1);
1288
1289 /*
1290 * Return the next character in the buffer...
1291 */
1292
1293 return (*(fp->ptr) & 255);
1294}
1295
1296
1297/*
1298 * 'cupsFilePrintf()' - Write a formatted string.
5a738aea 1299 *
f3c17241 1300 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1301 */
1302
5a738aea 1303int /* O - Number of bytes written or -1 on error */
ef416fc2 1304cupsFilePrintf(cups_file_t *fp, /* I - CUPS file */
1305 const char *format, /* I - Printf-style format string */
1306 ...) /* I - Additional args as necessary */
1307{
1308 va_list ap; /* Argument list */
2abf387c 1309 ssize_t bytes; /* Formatted size */
ecdc0628 1310
ef416fc2 1311
e07d4801 1312 DEBUG_printf(("2cupsFilePrintf(fp=%p, format=\"%s\", ...)", fp, format));
ef416fc2 1313
1314 if (!fp || !format || (fp->mode != 'w' && fp->mode != 's'))
1315 return (-1);
1316
75bd9771
MS
1317 if (!fp->printf_buffer)
1318 {
1319 /*
1320 * Start with an 1k printf buffer...
1321 */
1322
1323 if ((fp->printf_buffer = malloc(1024)) == NULL)
1324 return (-1);
1325
1326 fp->printf_size = 1024;
1327 }
1328
ef416fc2 1329 va_start(ap, format);
75bd9771 1330 bytes = vsnprintf(fp->printf_buffer, fp->printf_size, format, ap);
ef416fc2 1331 va_end(ap);
1332
536bc2c6 1333 if (bytes >= (ssize_t)fp->printf_size)
75bd9771
MS
1334 {
1335 /*
1336 * Expand the printf buffer...
1337 */
1338
1339 char *temp; /* Temporary buffer pointer */
1340
1341
1342 if (bytes > 65535)
1343 return (-1);
1344
7e86f2f6 1345 if ((temp = realloc(fp->printf_buffer, (size_t)(bytes + 1))) == NULL)
75bd9771
MS
1346 return (-1);
1347
1348 fp->printf_buffer = temp;
7e86f2f6 1349 fp->printf_size = (size_t)(bytes + 1);
75bd9771
MS
1350
1351 va_start(ap, format);
1352 bytes = vsnprintf(fp->printf_buffer, fp->printf_size, format, ap);
1353 va_end(ap);
1354 }
ecdc0628 1355
ef416fc2 1356 if (fp->mode == 's')
634763e8 1357 {
7e86f2f6 1358 if (cups_write(fp, fp->printf_buffer, (size_t)bytes) < 0)
634763e8
MS
1359 return (-1);
1360
1361 fp->pos += bytes;
1362
e07d4801 1363 DEBUG_printf(("4cupsFilePrintf: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1364
7e86f2f6 1365 return ((int)bytes);
634763e8 1366 }
ef416fc2 1367
1368 if ((fp->ptr + bytes) > fp->end)
1369 if (cupsFileFlush(fp))
1370 return (-1);
1371
1372 fp->pos += bytes;
1373
e07d4801 1374 DEBUG_printf(("4cupsFilePrintf: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1375
7e86f2f6 1376 if ((size_t)bytes > sizeof(fp->buf))
ef416fc2 1377 {
1378#ifdef HAVE_LIBZ
1379 if (fp->compressed)
7e86f2f6 1380 return ((int)cups_compress(fp, fp->printf_buffer, (size_t)bytes));
ef416fc2 1381 else
1382#endif /* HAVE_LIBZ */
7e86f2f6 1383 return ((int)cups_write(fp, fp->printf_buffer, (size_t)bytes));
ef416fc2 1384 }
1385 else
1386 {
07623986 1387 memcpy(fp->ptr, fp->printf_buffer, (size_t)bytes);
ef416fc2 1388 fp->ptr += bytes;
7e86f2f6 1389 return ((int)bytes);
ef416fc2 1390 }
1391}
1392
1393
1394/*
1395 * 'cupsFilePutChar()' - Write a character.
5a738aea 1396 *
f3c17241 1397 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1398 */
1399
1400int /* O - 0 on success, -1 on error */
1401cupsFilePutChar(cups_file_t *fp, /* I - CUPS file */
1402 int c) /* I - Character to write */
1403{
1404 /*
1405 * Range check input...
1406 */
1407
1408 if (!fp || (fp->mode != 'w' && fp->mode != 's'))
1409 return (-1);
1410
1411 if (fp->mode == 's')
1412 {
1413 /*
1414 * Send character immediately over socket...
1415 */
1416
1417 char ch; /* Output character */
1418
1419
7e86f2f6 1420 ch = (char)c;
ef416fc2 1421
1422 if (send(fp->fd, &ch, 1, 0) < 1)
1423 return (-1);
1424 }
1425 else
1426 {
1427 /*
1428 * Buffer it up...
1429 */
1430
1431 if (fp->ptr >= fp->end)
1432 if (cupsFileFlush(fp))
1433 return (-1);
1434
7e86f2f6 1435 *(fp->ptr) ++ = (char)c;
ef416fc2 1436 }
1437
1438 fp->pos ++;
1439
e07d4801 1440 DEBUG_printf(("4cupsFilePutChar: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1441
ef416fc2 1442 return (0);
1443}
1444
1445
58dc1933
MS
1446/*
1447 * 'cupsFilePutConf()' - Write a configuration line.
1448 *
1449 * This function handles any comment escaping of the value.
1450 *
f3c17241 1451 * @since CUPS 1.4/OS X 10.6@
58dc1933
MS
1452 */
1453
1454ssize_t /* O - Number of bytes written or -1 on error */
1455cupsFilePutConf(cups_file_t *fp, /* I - CUPS file */
1456 const char *directive, /* I - Directive */
1457 const char *value) /* I - Value */
1458{
1459 ssize_t bytes, /* Number of bytes written */
1460 temp; /* Temporary byte count */
1461 const char *ptr; /* Pointer into value */
1462
1463
1464 if (!fp || !directive || !*directive)
1465 return (-1);
1466
1467 if ((bytes = cupsFilePuts(fp, directive)) < 0)
1468 return (-1);
1469
1470 if (cupsFilePutChar(fp, ' ') < 0)
1471 return (-1);
1472 bytes ++;
1473
1474 if (value && *value)
1475 {
1476 if ((ptr = strchr(value, '#')) != NULL)
1477 {
1478 /*
1479 * Need to quote the first # in the info string...
1480 */
1481
7e86f2f6 1482 if ((temp = cupsFileWrite(fp, value, (size_t)(ptr - value))) < 0)
58dc1933
MS
1483 return (-1);
1484 bytes += temp;
1485
1486 if (cupsFilePutChar(fp, '\\') < 0)
1487 return (-1);
1488 bytes ++;
1489
1490 if ((temp = cupsFilePuts(fp, ptr)) < 0)
1491 return (-1);
1492 bytes += temp;
1493 }
1494 else if ((temp = cupsFilePuts(fp, value)) < 0)
1495 return (-1);
1496 else
1497 bytes += temp;
1498 }
1499
1500 if (cupsFilePutChar(fp, '\n') < 0)
1501 return (-1);
1502 else
1503 return (bytes + 1);
1504}
1505
1506
ef416fc2 1507/*
1508 * 'cupsFilePuts()' - Write a string.
5a738aea
MS
1509 *
1510 * Like the @code fputs@ function, no newline is appended to the string.
1511 *
f3c17241 1512 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1513 */
1514
5a738aea 1515int /* O - Number of bytes written or -1 on error */
ef416fc2 1516cupsFilePuts(cups_file_t *fp, /* I - CUPS file */
1517 const char *s) /* I - String to write */
1518{
2abf387c 1519 ssize_t bytes; /* Bytes to write */
ef416fc2 1520
1521
1522 /*
1523 * Range check input...
1524 */
1525
1526 if (!fp || !s || (fp->mode != 'w' && fp->mode != 's'))
1527 return (-1);
1528
1529 /*
1530 * Write the string...
1531 */
1532
7e86f2f6 1533 bytes = (ssize_t)strlen(s);
ef416fc2 1534
1535 if (fp->mode == 's')
1536 {
7e86f2f6 1537 if (cups_write(fp, s, (size_t)bytes) < 0)
ef416fc2 1538 return (-1);
1539
1540 fp->pos += bytes;
1541
e07d4801 1542 DEBUG_printf(("4cupsFilePuts: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1543
7e86f2f6 1544 return ((int)bytes);
ef416fc2 1545 }
1546
1547 if ((fp->ptr + bytes) > fp->end)
1548 if (cupsFileFlush(fp))
1549 return (-1);
1550
1551 fp->pos += bytes;
1552
e07d4801 1553 DEBUG_printf(("4cupsFilePuts: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1554
7e86f2f6 1555 if ((size_t)bytes > sizeof(fp->buf))
ef416fc2 1556 {
1557#ifdef HAVE_LIBZ
1558 if (fp->compressed)
7e86f2f6 1559 return ((int)cups_compress(fp, s, (size_t)bytes));
ef416fc2 1560 else
1561#endif /* HAVE_LIBZ */
7e86f2f6 1562 return ((int)cups_write(fp, s, (size_t)bytes));
ef416fc2 1563 }
1564 else
1565 {
07623986 1566 memcpy(fp->ptr, s, (size_t)bytes);
ef416fc2 1567 fp->ptr += bytes;
7e86f2f6 1568 return ((int)bytes);
ef416fc2 1569 }
1570}
1571
1572
1573/*
1574 * 'cupsFileRead()' - Read from a file.
5a738aea 1575 *
f3c17241 1576 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1577 */
1578
5a738aea 1579ssize_t /* O - Number of bytes read or -1 on error */
ef416fc2 1580cupsFileRead(cups_file_t *fp, /* I - CUPS file */
1581 char *buf, /* O - Buffer */
1582 size_t bytes) /* I - Number of bytes to read */
1583{
2abf387c 1584 size_t total; /* Total bytes read */
1585 ssize_t count; /* Bytes read */
ef416fc2 1586
1587
e07d4801 1588 DEBUG_printf(("2cupsFileRead(fp=%p, buf=%p, bytes=" CUPS_LLFMT ")", fp, buf,
634763e8 1589 CUPS_LLCAST bytes));
ef416fc2 1590
1591 /*
1592 * Range check input...
1593 */
1594
82f97232 1595 if (!fp || !buf || (fp->mode != 'r' && fp->mode != 's'))
ef416fc2 1596 return (-1);
1597
1598 if (bytes == 0)
1599 return (0);
1600
1601 /*
1602 * Loop until all bytes are read...
1603 */
1604
1605 total = 0;
1606 while (bytes > 0)
1607 {
1608 if (fp->ptr >= fp->end)
1609 if (cups_fill(fp) <= 0)
1610 {
e07d4801
MS
1611 DEBUG_printf(("4cupsFileRead: cups_fill() returned -1, total="
1612 CUPS_LLFMT, CUPS_LLCAST total));
ef416fc2 1613
1614 if (total > 0)
b86bc4cf 1615 return ((ssize_t)total);
ef416fc2 1616 else
1617 return (-1);
1618 }
1619
b86bc4cf 1620 count = (ssize_t)(fp->end - fp->ptr);
1621 if (count > (ssize_t)bytes)
1622 count = (ssize_t)bytes;
ef416fc2 1623
07623986 1624 memcpy(buf, fp->ptr,(size_t) count);
ef416fc2 1625 fp->ptr += count;
634763e8
MS
1626 fp->pos += count;
1627
e07d4801 1628 DEBUG_printf(("4cupsFileRead: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
ef416fc2 1629
1630 /*
1631 * Update the counts for the last read...
1632 */
1633
7e86f2f6
MS
1634 bytes -= (size_t)count;
1635 total += (size_t)count;
ef416fc2 1636 buf += count;
1637 }
1638
1639 /*
1640 * Return the total number of bytes read...
1641 */
1642
e07d4801 1643 DEBUG_printf(("3cupsFileRead: total=" CUPS_LLFMT, CUPS_LLCAST total));
ef416fc2 1644
b86bc4cf 1645 return ((ssize_t)total);
ef416fc2 1646}
1647
1648
1649/*
5a738aea
MS
1650 * 'cupsFileRewind()' - Set the current file position to the beginning of the
1651 * file.
1652 *
f3c17241 1653 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1654 */
1655
5a738aea 1656off_t /* O - New file position or -1 on error */
ef416fc2 1657cupsFileRewind(cups_file_t *fp) /* I - CUPS file */
1658{
80ca4592 1659 /*
1660 * Range check input...
1661 */
1662
e07d4801
MS
1663 DEBUG_printf(("cupsFileRewind(fp=%p)", fp));
1664 DEBUG_printf(("2cupsFileRewind: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1665
80ca4592 1666 if (!fp || fp->mode != 'r')
1667 return (-1);
1668
1669 /*
1670 * Handle special cases...
1671 */
1672
634763e8 1673 if (fp->bufpos == 0)
80ca4592 1674 {
1675 /*
1676 * No seeking necessary...
1677 */
1678
634763e8
MS
1679 fp->pos = 0;
1680
80ca4592 1681 if (fp->ptr)
1682 {
1683 fp->ptr = fp->buf;
1684 fp->eof = 0;
1685 }
1686
e07d4801 1687 DEBUG_printf(("2cupsFileRewind: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 1688
80ca4592 1689 return (0);
1690 }
1691
1692 /*
1693 * Otherwise, seek in the file and cleanup any compression buffers...
1694 */
1695
1696#ifdef HAVE_LIBZ
1697 if (fp->compressed)
1698 {
1699 inflateEnd(&fp->stream);
1700 fp->compressed = 0;
1701 }
1702#endif /* HAVE_LIBZ */
1703
c9fc04c6
MS
1704 if (lseek(fp->fd, 0, SEEK_SET))
1705 {
e07d4801 1706 DEBUG_printf(("1cupsFileRewind: lseek failed: %s", strerror(errno)));
c9fc04c6
MS
1707 return (-1);
1708 }
80ca4592 1709
634763e8
MS
1710 fp->bufpos = 0;
1711 fp->pos = 0;
1712 fp->ptr = NULL;
1713 fp->end = NULL;
1714 fp->eof = 0;
1715
e07d4801 1716 DEBUG_printf(("2cupsFileRewind: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
80ca4592 1717
1718 return (0);
ef416fc2 1719}
1720
1721
1722/*
1723 * 'cupsFileSeek()' - Seek in a file.
5a738aea 1724 *
f3c17241 1725 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1726 */
1727
5a738aea 1728off_t /* O - New file position or -1 on error */
ef416fc2 1729cupsFileSeek(cups_file_t *fp, /* I - CUPS file */
1730 off_t pos) /* I - Position in file */
1731{
2abf387c 1732 ssize_t bytes; /* Number bytes in buffer */
ef416fc2 1733
1734
e07d4801 1735 DEBUG_printf(("cupsFileSeek(fp=%p, pos=" CUPS_LLFMT ")", fp,
634763e8 1736 CUPS_LLCAST pos));
e07d4801
MS
1737 DEBUG_printf(("2cupsFileSeek: fp->pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
1738 DEBUG_printf(("2cupsFileSeek: fp->ptr=%p, fp->end=%p", fp->ptr, fp->end));
ef416fc2 1739
1740 /*
1741 * Range check input...
1742 */
1743
1744 if (!fp || pos < 0 || fp->mode != 'r')
1745 return (-1);
1746
80ca4592 1747 /*
1748 * Handle special cases...
1749 */
1750
1751 if (pos == 0)
1752 return (cupsFileRewind(fp));
1753
634763e8 1754 if (fp->ptr)
b423cd4c 1755 {
634763e8 1756 bytes = (ssize_t)(fp->end - fp->buf);
b423cd4c 1757
e07d4801 1758 DEBUG_printf(("2cupsFileSeek: bytes=" CUPS_LLFMT, CUPS_LLCAST bytes));
c168a833 1759
634763e8 1760 if (pos >= fp->bufpos && pos < (fp->bufpos + bytes))
b423cd4c 1761 {
634763e8
MS
1762 /*
1763 * No seeking necessary...
1764 */
1765
1766 fp->pos = pos;
1767 fp->ptr = fp->buf + pos - fp->bufpos;
b423cd4c 1768 fp->eof = 0;
b423cd4c 1769
634763e8
MS
1770 return (pos);
1771 }
b423cd4c 1772 }
1773
80ca4592 1774#ifdef HAVE_LIBZ
1775 if (!fp->compressed && !fp->ptr)
1776 {
1777 /*
1778 * Preload a buffer to determine whether the file is compressed...
1779 */
1780
1781 if (cups_fill(fp) < 0)
1782 return (-1);
1783 }
1784#endif /* HAVE_LIBZ */
1785
ef416fc2 1786 /*
634763e8 1787 * Seek forwards or backwards...
ef416fc2 1788 */
1789
ef416fc2 1790 fp->eof = 0;
1791
634763e8 1792 if (pos < fp->bufpos)
ef416fc2 1793 {
1794 /*
1795 * Need to seek backwards...
1796 */
1797
e07d4801 1798 DEBUG_puts("2cupsFileSeek: SEEK BACKWARDS");
80ca4592 1799
ef416fc2 1800#ifdef HAVE_LIBZ
1801 if (fp->compressed)
1802 {
1803 inflateEnd(&fp->stream);
1804
1805 lseek(fp->fd, 0, SEEK_SET);
634763e8
MS
1806 fp->bufpos = 0;
1807 fp->pos = 0;
1808 fp->ptr = NULL;
1809 fp->end = NULL;
ef416fc2 1810
1811 while ((bytes = cups_fill(fp)) > 0)
634763e8 1812 if (pos >= fp->bufpos && pos < (fp->bufpos + bytes))
ef416fc2 1813 break;
1814
1815 if (bytes <= 0)
1816 return (-1);
80ca4592 1817
634763e8
MS
1818 fp->ptr = fp->buf + pos - fp->bufpos;
1819 fp->pos = pos;
ef416fc2 1820 }
1821 else
1822#endif /* HAVE_LIBZ */
1823 {
634763e8
MS
1824 fp->bufpos = lseek(fp->fd, pos, SEEK_SET);
1825 fp->pos = fp->bufpos;
1826 fp->ptr = NULL;
1827 fp->end = NULL;
80ca4592 1828
e07d4801 1829 DEBUG_printf(("2cupsFileSeek: lseek() returned " CUPS_LLFMT,
634763e8 1830 CUPS_LLCAST fp->pos));
ef416fc2 1831 }
1832 }
634763e8 1833 else
ef416fc2 1834 {
1835 /*
1836 * Need to seek forwards...
1837 */
1838
e07d4801 1839 DEBUG_puts("2cupsFileSeek: SEEK FORWARDS");
80ca4592 1840
ef416fc2 1841#ifdef HAVE_LIBZ
80ca4592 1842 if (fp->compressed)
ef416fc2 1843 {
1844 while ((bytes = cups_fill(fp)) > 0)
80ca4592 1845 {
634763e8 1846 if (pos >= fp->bufpos && pos < (fp->bufpos + bytes))
ef416fc2 1847 break;
80ca4592 1848 }
ef416fc2 1849
1850 if (bytes <= 0)
1851 return (-1);
80ca4592 1852
634763e8
MS
1853 fp->ptr = fp->buf + pos - fp->bufpos;
1854 fp->pos = pos;
ef416fc2 1855 }
1856 else
1857#endif /* HAVE_LIBZ */
1858 {
634763e8
MS
1859 fp->bufpos = lseek(fp->fd, pos, SEEK_SET);
1860 fp->pos = fp->bufpos;
1861 fp->ptr = NULL;
1862 fp->end = NULL;
80ca4592 1863
e07d4801 1864 DEBUG_printf(("2cupsFileSeek: lseek() returned " CUPS_LLFMT,
634763e8 1865 CUPS_LLCAST fp->pos));
ef416fc2 1866 }
1867 }
ef416fc2 1868
e07d4801 1869 DEBUG_printf(("2cupsFileSeek: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
ef416fc2 1870
1871 return (fp->pos);
1872}
1873
1874
80ca4592 1875/*
1876 * 'cupsFileStderr()' - Return a CUPS file associated with stderr.
5a738aea 1877 *
f3c17241 1878 * @since CUPS 1.2/OS X 10.5@
80ca4592 1879 */
1880
5a738aea 1881cups_file_t * /* O - CUPS file */
80ca4592 1882cupsFileStderr(void)
1883{
1884 _cups_globals_t *cg = _cupsGlobals(); /* Pointer to library globals... */
1885
1886
1887 /*
1888 * Open file descriptor 2 as needed...
1889 */
1890
1891 if (!cg->stdio_files[2])
1892 {
1893 /*
1894 * Flush any pending output on the stdio file...
1895 */
1896
1897 fflush(stderr);
1898
1899 /*
1900 * Open file descriptor 2...
1901 */
1902
1903 if ((cg->stdio_files[2] = cupsFileOpenFd(2, "w")) != NULL)
1904 cg->stdio_files[2]->is_stdio = 1;
1905 }
1906
1907 return (cg->stdio_files[2]);
1908}
1909
1910
1911/*
1912 * 'cupsFileStdin()' - Return a CUPS file associated with stdin.
5a738aea 1913 *
f3c17241 1914 * @since CUPS 1.2/OS X 10.5@
80ca4592 1915 */
1916
5a738aea 1917cups_file_t * /* O - CUPS file */
80ca4592 1918cupsFileStdin(void)
1919{
1920 _cups_globals_t *cg = _cupsGlobals(); /* Pointer to library globals... */
1921
1922
1923 /*
1924 * Open file descriptor 0 as needed...
1925 */
1926
1927 if (!cg->stdio_files[0])
1928 {
1929 /*
1930 * Open file descriptor 0...
1931 */
1932
1933 if ((cg->stdio_files[0] = cupsFileOpenFd(0, "r")) != NULL)
1934 cg->stdio_files[0]->is_stdio = 1;
1935 }
1936
1937 return (cg->stdio_files[0]);
1938}
1939
1940
1941/*
1942 * 'cupsFileStdout()' - Return a CUPS file associated with stdout.
5a738aea 1943 *
f3c17241 1944 * @since CUPS 1.2/OS X 10.5@
80ca4592 1945 */
1946
5a738aea 1947cups_file_t * /* O - CUPS file */
80ca4592 1948cupsFileStdout(void)
1949{
1950 _cups_globals_t *cg = _cupsGlobals(); /* Pointer to library globals... */
1951
1952
1953 /*
1954 * Open file descriptor 1 as needed...
1955 */
1956
1957 if (!cg->stdio_files[1])
1958 {
1959 /*
1960 * Flush any pending output on the stdio file...
1961 */
1962
1963 fflush(stdout);
1964
1965 /*
1966 * Open file descriptor 1...
1967 */
1968
1969 if ((cg->stdio_files[1] = cupsFileOpenFd(1, "w")) != NULL)
1970 cg->stdio_files[1]->is_stdio = 1;
1971 }
1972
1973 return (cg->stdio_files[1]);
1974}
1975
1976
ef416fc2 1977/*
1978 * 'cupsFileTell()' - Return the current file position.
5a738aea 1979 *
f3c17241 1980 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1981 */
1982
1983off_t /* O - File position */
1984cupsFileTell(cups_file_t *fp) /* I - CUPS file */
1985{
e07d4801
MS
1986 DEBUG_printf(("2cupsFileTell(fp=%p)", fp));
1987 DEBUG_printf(("3cupsFileTell: pos=" CUPS_LLFMT,
1988 CUPS_LLCAST (fp ? fp->pos : -1)));
634763e8 1989
80ca4592 1990 return (fp ? fp->pos : 0);
ef416fc2 1991}
1992
1993
1994/*
1995 * 'cupsFileUnlock()' - Unlock access to a file.
5a738aea 1996 *
f3c17241 1997 * @since CUPS 1.2/OS X 10.5@
ef416fc2 1998 */
1999
2000int /* O - 0 on success, -1 on error */
5a738aea 2001cupsFileUnlock(cups_file_t *fp) /* I - CUPS file */
ef416fc2 2002{
2003 /*
2004 * Range check...
2005 */
2006
e07d4801 2007 DEBUG_printf(("cupsFileUnlock(fp=%p)", fp));
634763e8 2008
ef416fc2 2009 if (!fp || fp->mode == 's')
2010 return (-1);
2011
2012 /*
2013 * Unlock...
2014 */
2015
2016#ifdef WIN32
536bc2c6 2017 return (_locking(fp->fd, _LK_UNLCK, 0));
ef416fc2 2018#else
2019 return (lockf(fp->fd, F_ULOCK, 0));
2020#endif /* WIN32 */
2021}
2022
2023
2024/*
2025 * 'cupsFileWrite()' - Write to a file.
5a738aea 2026 *
f3c17241 2027 * @since CUPS 1.2/OS X 10.5@
ef416fc2 2028 */
2029
5a738aea 2030ssize_t /* O - Number of bytes written or -1 on error */
ef416fc2 2031cupsFileWrite(cups_file_t *fp, /* I - CUPS file */
2032 const char *buf, /* I - Buffer */
2033 size_t bytes) /* I - Number of bytes to write */
2034{
2035 /*
2036 * Range check input...
2037 */
2038
e07d4801 2039 DEBUG_printf(("2cupsFileWrite(fp=%p, buf=%p, bytes=" CUPS_LLFMT ")",
634763e8
MS
2040 fp, buf, CUPS_LLCAST bytes));
2041
82f97232 2042 if (!fp || !buf || (fp->mode != 'w' && fp->mode != 's'))
ef416fc2 2043 return (-1);
2044
2045 if (bytes == 0)
2046 return (0);
2047
2048 /*
2049 * Write the buffer...
2050 */
2051
2052 if (fp->mode == 's')
2053 {
2054 if (cups_write(fp, buf, bytes) < 0)
2055 return (-1);
2056
b86bc4cf 2057 fp->pos += (off_t)bytes;
ef416fc2 2058
e07d4801 2059 DEBUG_printf(("4cupsFileWrite: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 2060
b86bc4cf 2061 return ((ssize_t)bytes);
ef416fc2 2062 }
2063
2064 if ((fp->ptr + bytes) > fp->end)
2065 if (cupsFileFlush(fp))
2066 return (-1);
2067
b86bc4cf 2068 fp->pos += (off_t)bytes;
ef416fc2 2069
e07d4801 2070 DEBUG_printf(("4cupsFileWrite: pos=" CUPS_LLFMT, CUPS_LLCAST fp->pos));
634763e8 2071
ef416fc2 2072 if (bytes > sizeof(fp->buf))
2073 {
2074#ifdef HAVE_LIBZ
2075 if (fp->compressed)
2076 return (cups_compress(fp, buf, bytes));
2077 else
2078#endif /* HAVE_LIBZ */
2079 return (cups_write(fp, buf, bytes));
2080 }
2081 else
2082 {
2083 memcpy(fp->ptr, buf, bytes);
2084 fp->ptr += bytes;
b86bc4cf 2085 return ((ssize_t)bytes);
ef416fc2 2086 }
2087}
2088
2089
2090#ifdef HAVE_LIBZ
2091/*
22c9029b 2092 * 'cups_compress()' - Compress a buffer of data.
ef416fc2 2093 */
2094
2095static ssize_t /* O - Number of bytes written or -1 */
2096cups_compress(cups_file_t *fp, /* I - CUPS file */
2097 const char *buf, /* I - Buffer */
2098 size_t bytes) /* I - Number bytes */
2099{
e07d4801 2100 DEBUG_printf(("7cups_compress(fp=%p, buf=%p, bytes=" CUPS_LLFMT ")", fp, buf,
634763e8 2101 CUPS_LLCAST bytes));
ecdc0628 2102
ef416fc2 2103 /*
2104 * Update the CRC...
2105 */
2106
7e86f2f6 2107 fp->crc = crc32(fp->crc, (const Bytef *)buf, (uInt)bytes);
ef416fc2 2108
2109 /*
2110 * Deflate the bytes...
2111 */
2112
2113 fp->stream.next_in = (Bytef *)buf;
7e86f2f6 2114 fp->stream.avail_in = (uInt)bytes;
ef416fc2 2115
2116 while (fp->stream.avail_in > 0)
2117 {
2118 /*
2119 * Flush the current buffer...
2120 */
2121
e07d4801 2122 DEBUG_printf(("9cups_compress: avail_in=%d, avail_out=%d",
634763e8 2123 fp->stream.avail_in, fp->stream.avail_out));
ecdc0628 2124
7e86f2f6 2125 if (fp->stream.avail_out < (uInt)(sizeof(fp->cbuf) / 8))
ef416fc2 2126 {
7e86f2f6 2127 if (cups_write(fp, (char *)fp->cbuf, (size_t)(fp->stream.next_out - fp->cbuf)) < 0)
ef416fc2 2128 return (-1);
ecdc0628 2129
2130 fp->stream.next_out = fp->cbuf;
2131 fp->stream.avail_out = sizeof(fp->cbuf);
ef416fc2 2132 }
2133
2134 deflate(&(fp->stream), Z_NO_FLUSH);
2135 }
2136
7e86f2f6 2137 return ((ssize_t)bytes);
ef416fc2 2138}
2139#endif /* HAVE_LIBZ */
2140
2141
2142/*
22c9029b 2143 * 'cups_fill()' - Fill the input buffer.
ef416fc2 2144 */
2145
2146static ssize_t /* O - Number of bytes or -1 */
2147cups_fill(cups_file_t *fp) /* I - CUPS file */
2148{
2149 ssize_t bytes; /* Number of bytes read */
2150#ifdef HAVE_LIBZ
c277e2f8 2151 int status; /* Decompression status */
ef416fc2 2152 const unsigned char *ptr, /* Pointer into buffer */
2153 *end; /* End of buffer */
2154#endif /* HAVE_LIBZ */
2155
2156
e07d4801
MS
2157 DEBUG_printf(("7cups_fill(fp=%p)", fp));
2158 DEBUG_printf(("9cups_fill: fp->ptr=%p, fp->end=%p, fp->buf=%p, "
2159 "fp->bufpos=" CUPS_LLFMT ", fp->eof=%d",
634763e8 2160 fp->ptr, fp->end, fp->buf, CUPS_LLCAST fp->bufpos, fp->eof));
ef416fc2 2161
2162 if (fp->ptr && fp->end)
c9fc04c6 2163 fp->bufpos += fp->end - fp->buf;
ef416fc2 2164
2165#ifdef HAVE_LIBZ
e07d4801 2166 DEBUG_printf(("9cups_fill: fp->compressed=%d", fp->compressed));
b423cd4c 2167
ef416fc2 2168 while (!fp->ptr || fp->compressed)
2169 {
2170 /*
2171 * Check to see if we have read any data yet; if not, see if we have a
2172 * compressed file...
2173 */
2174
2175 if (!fp->ptr)
2176 {
2177 /*
2178 * Reset the file position in case we are seeking...
2179 */
2180
2181 fp->compressed = 0;
ef416fc2 2182
2183 /*
2184 * Read the first bytes in the file to determine if we have a gzip'd
2185 * file...
2186 */
2187
fa73b229 2188 if ((bytes = cups_read(fp, (char *)fp->buf, sizeof(fp->buf))) < 0)
ef416fc2 2189 {
2190 /*
2191 * Can't read from file!
2192 */
2193
e07d4801 2194 DEBUG_printf(("9cups_fill: cups_read() returned " CUPS_LLFMT,
b423cd4c 2195 CUPS_LLCAST bytes));
2196
ef416fc2 2197 return (-1);
2198 }
2199
fa73b229 2200 if (bytes < 10 || fp->buf[0] != 0x1f ||
e00b005a 2201 (fp->buf[1] & 255) != 0x8b ||
fa73b229 2202 fp->buf[2] != 8 || (fp->buf[3] & 0xe0) != 0)
ef416fc2 2203 {
2204 /*
2205 * Not a gzip'd file!
2206 */
2207
ef416fc2 2208 fp->ptr = fp->buf;
2209 fp->end = fp->buf + bytes;
2210
e07d4801 2211 DEBUG_printf(("9cups_fill: Returning " CUPS_LLFMT,
c9fc04c6 2212 CUPS_LLCAST bytes));
b423cd4c 2213
ef416fc2 2214 return (bytes);
2215 }
2216
2217 /*
2218 * Parse header junk: extra data, original name, and comment...
2219 */
2220
fa73b229 2221 ptr = (unsigned char *)fp->buf + 10;
2222 end = (unsigned char *)fp->buf + bytes;
ef416fc2 2223
fa73b229 2224 if (fp->buf[3] & 0x04)
ef416fc2 2225 {
2226 /*
2227 * Skip extra data...
2228 */
2229
2230 if ((ptr + 2) > end)
2231 {
2232 /*
2233 * Can't read from file!
2234 */
2235
2236 return (-1);
2237 }
2238
2239 bytes = ((unsigned char)ptr[1] << 8) | (unsigned char)ptr[0];
2240 ptr += 2 + bytes;
2241
2242 if (ptr > end)
2243 {
2244 /*
2245 * Can't read from file!
2246 */
2247
2248 return (-1);
2249 }
2250 }
2251
fa73b229 2252 if (fp->buf[3] & 0x08)
ef416fc2 2253 {
2254 /*
2255 * Skip original name data...
2256 */
2257
2258 while (ptr < end && *ptr)
2259 ptr ++;
2260
2261 if (ptr < end)
2262 ptr ++;
2263 else
2264 {
2265 /*
2266 * Can't read from file!
2267 */
2268
2269 return (-1);
2270 }
2271 }
2272
fa73b229 2273 if (fp->buf[3] & 0x10)
ef416fc2 2274 {
2275 /*
2276 * Skip comment data...
2277 */
2278
2279 while (ptr < end && *ptr)
2280 ptr ++;
2281
2282 if (ptr < end)
2283 ptr ++;
2284 else
2285 {
2286 /*
2287 * Can't read from file!
2288 */
2289
2290 return (-1);
2291 }
2292 }
2293
fa73b229 2294 if (fp->buf[3] & 0x02)
ef416fc2 2295 {
2296 /*
2297 * Skip header CRC data...
2298 */
2299
2300 ptr += 2;
2301
2302 if (ptr > end)
2303 {
2304 /*
2305 * Can't read from file!
2306 */
2307
2308 return (-1);
2309 }
2310 }
2311
fa73b229 2312 /*
2313 * Copy the flate-compressed data to the compression buffer...
2314 */
2315
2316 if ((bytes = end - ptr) > 0)
07623986 2317 memcpy(fp->cbuf, ptr, (size_t)bytes);
fa73b229 2318
ef416fc2 2319 /*
2320 * Setup the decompressor data...
2321 */
2322
2323 fp->stream.zalloc = (alloc_func)0;
2324 fp->stream.zfree = (free_func)0;
2325 fp->stream.opaque = (voidpf)0;
fa73b229 2326 fp->stream.next_in = (Bytef *)fp->cbuf;
ef416fc2 2327 fp->stream.next_out = NULL;
7e86f2f6 2328 fp->stream.avail_in = (uInt)bytes;
ef416fc2 2329 fp->stream.avail_out = 0;
2330 fp->crc = crc32(0L, Z_NULL, 0);
2331
2332 if (inflateInit2(&(fp->stream), -15) != Z_OK)
2333 return (-1);
2334
2335 fp->compressed = 1;
2336 }
2337
2338 if (fp->compressed)
2339 {
2340 /*
2341 * If we have reached end-of-file, return immediately...
2342 */
2343
2344 if (fp->eof)
2345 return (-1);
2346
2347 /*
2348 * Fill the decompression buffer as needed...
2349 */
2350
2351 if (fp->stream.avail_in == 0)
2352 {
2353 if ((bytes = cups_read(fp, (char *)fp->cbuf, sizeof(fp->cbuf))) <= 0)
2354 return (-1);
2355
2356 fp->stream.next_in = fp->cbuf;
7e86f2f6 2357 fp->stream.avail_in = (uInt)bytes;
ef416fc2 2358 }
2359
2360 /*
2361 * Decompress data from the buffer...
2362 */
2363
2364 fp->stream.next_out = (Bytef *)fp->buf;
2365 fp->stream.avail_out = sizeof(fp->buf);
2366
c277e2f8
MS
2367 status = inflate(&(fp->stream), Z_NO_FLUSH);
2368
2369 if (fp->stream.next_out > (Bytef *)fp->buf)
2370 fp->crc = crc32(fp->crc, (Bytef *)fp->buf,
7e86f2f6 2371 (uInt)(fp->stream.next_out - (Bytef *)fp->buf));
c277e2f8
MS
2372
2373 if (status == Z_STREAM_END)
ef416fc2 2374 {
2375 /*
2376 * Read the CRC and length...
2377 */
2378
2379 unsigned char trailer[8]; /* Trailer bytes */
2380 uLong tcrc; /* Trailer CRC */
2381
2382
7e86f2f6 2383 if (read(fp->fd, trailer, sizeof(trailer)) < (ssize_t)sizeof(trailer))
ef416fc2 2384 {
2385 /*
2386 * Can't get it, so mark end-of-file...
2387 */
2388
2389 fp->eof = 1;
ef416fc2 2390 }
fa73b229 2391 else
2392 {
da003234
MS
2393 tcrc = ((((((uLong)trailer[3] << 8) | (uLong)trailer[2]) << 8) |
2394 (uLong)trailer[1]) << 8) | (uLong)trailer[0];
ef416fc2 2395
fa73b229 2396 if (tcrc != fp->crc)
2397 {
2398 /*
2399 * Bad CRC, mark end-of-file...
2400 */
2401
da003234 2402 DEBUG_printf(("9cups_fill: tcrc=%08x != fp->crc=%08x",
c277e2f8
MS
2403 (unsigned int)tcrc, (unsigned int)fp->crc));
2404
fa73b229 2405 fp->eof = 1;
2406
2407 return (-1);
2408 }
ef416fc2 2409
ef416fc2 2410 /*
fa73b229 2411 * Otherwise, reset the compressed flag so that we re-read the
2412 * file header...
ef416fc2 2413 */
ef416fc2 2414
fa73b229 2415 fp->compressed = 0;
ef416fc2 2416 }
ef416fc2 2417 }
2418
7e86f2f6 2419 bytes = (ssize_t)sizeof(fp->buf) - (ssize_t)fp->stream.avail_out;
ef416fc2 2420
2421 /*
2422 * Return the decompressed data...
2423 */
2424
2425 fp->ptr = fp->buf;
2426 fp->end = fp->buf + bytes;
2427
fa73b229 2428 if (bytes)
2429 return (bytes);
ef416fc2 2430 }
2431 }
2432#endif /* HAVE_LIBZ */
2433
2434 /*
2435 * Read a buffer's full of data...
2436 */
2437
2438 if ((bytes = cups_read(fp, fp->buf, sizeof(fp->buf))) <= 0)
2439 {
2440 /*
2441 * Can't read from file!
2442 */
2443
2444 fp->eof = 1;
2445 fp->ptr = fp->buf;
2446 fp->end = fp->buf;
2447
2448 return (-1);
2449 }
2450
2451 /*
2452 * Return the bytes we read...
2453 */
2454
2455 fp->eof = 0;
2456 fp->ptr = fp->buf;
2457 fp->end = fp->buf + bytes;
2458
2459 return (bytes);
2460}
2461
2462
c7017ecc
MS
2463/*
2464 * 'cups_open()' - Safely open a file for writing.
2465 *
2466 * We don't allow appending to directories or files that are hard-linked or
2467 * symlinked.
2468 */
2469
2470static int /* O - File descriptor or -1 otherwise */
2471cups_open(const char *filename, /* I - Filename */
2472 int mode) /* I - Open mode */
2473{
2474 int fd; /* File descriptor */
2475 struct stat fileinfo; /* File information */
2476#ifndef WIN32
2477 struct stat linkinfo; /* Link information */
2478#endif /* !WIN32 */
2479
2480
2481 /*
2482 * Open the file...
2483 */
2484
2485 if ((fd = open(filename, mode, 0666)) < 0)
2486 return (-1);
2487
2488 /*
2489 * Then verify that the file descriptor doesn't point to a directory or hard-
2490 * linked file.
2491 */
2492
2493 if (fstat(fd, &fileinfo))
2494 {
2495 close(fd);
2496 return (-1);
2497 }
2498
2499 if (fileinfo.st_nlink != 1)
2500 {
2501 close(fd);
2502 errno = EPERM;
2503 return (-1);
2504 }
2505
2506#ifdef WIN32
2507 if (fileinfo.st_mode & _S_IFDIR)
2508#else
2509 if (S_ISDIR(fileinfo.st_mode))
2510#endif /* WIN32 */
2511 {
2512 close(fd);
2513 errno = EISDIR;
2514 return (-1);
2515 }
2516
2517#ifndef WIN32
2518 /*
2519 * Then use lstat to determine whether the filename is a symlink...
2520 */
2521
2522 if (lstat(filename, &linkinfo))
2523 {
2524 close(fd);
2525 return (-1);
2526 }
2527
2528 if (S_ISLNK(linkinfo.st_mode) ||
2529 fileinfo.st_dev != linkinfo.st_dev ||
2530 fileinfo.st_ino != linkinfo.st_ino ||
2531#ifdef HAVE_ST_GEN
2532 fileinfo.st_gen != linkinfo.st_gen ||
2533#endif /* HAVE_ST_GEN */
2534 fileinfo.st_nlink != linkinfo.st_nlink ||
2535 fileinfo.st_mode != linkinfo.st_mode)
2536 {
2537 /*
2538 * Yes, don't allow!
2539 */
2540
2541 close(fd);
2542 errno = EPERM;
2543 return (-1);
2544 }
2545#endif /* !WIN32 */
2546
2547 return (fd);
2548}
2549
2550
ef416fc2 2551/*
2552 * 'cups_read()' - Read from a file descriptor.
2553 */
2554
2555static ssize_t /* O - Number of bytes read or -1 */
2556cups_read(cups_file_t *fp, /* I - CUPS file */
2557 char *buf, /* I - Buffer */
2558 size_t bytes) /* I - Number bytes */
2559{
2560 ssize_t total; /* Total bytes read */
2561
2562
e07d4801 2563 DEBUG_printf(("7cups_read(fp=%p, buf=%p, bytes=" CUPS_LLFMT ")", fp, buf,
634763e8
MS
2564 CUPS_LLCAST bytes));
2565
ef416fc2 2566 /*
2567 * Loop until we read at least 0 bytes...
2568 */
2569
2570 for (;;)
2571 {
b86bc4cf 2572#ifdef WIN32
2573 if (fp->mode == 's')
2574 total = (ssize_t)recv(fp->fd, buf, (unsigned)bytes, 0);
2575 else
2576 total = (ssize_t)read(fp->fd, buf, (unsigned)bytes);
2577#else
ef416fc2 2578 if (fp->mode == 's')
2579 total = recv(fp->fd, buf, bytes, 0);
2580 else
2581 total = read(fp->fd, buf, bytes);
b86bc4cf 2582#endif /* WIN32 */
ef416fc2 2583
e07d4801 2584 DEBUG_printf(("9cups_read: total=" CUPS_LLFMT, CUPS_LLCAST total));
634763e8 2585
ef416fc2 2586 if (total >= 0)
2587 break;
2588
2589 /*
2590 * Reads can be interrupted by signals and unavailable resources...
2591 */
2592
2593 if (errno == EAGAIN || errno == EINTR)
2594 continue;
2595 else
2596 return (-1);
2597 }
2598
2599 /*
2600 * Return the total number of bytes read...
2601 */
2602
2603 return (total);
2604}
2605
2606
2607/*
2608 * 'cups_write()' - Write to a file descriptor.
2609 */
2610
2611static ssize_t /* O - Number of bytes written or -1 */
2612cups_write(cups_file_t *fp, /* I - CUPS file */
2613 const char *buf, /* I - Buffer */
2614 size_t bytes) /* I - Number bytes */
2615{
2abf387c 2616 size_t total; /* Total bytes written */
2617 ssize_t count; /* Count this time */
ef416fc2 2618
2619
e07d4801 2620 DEBUG_printf(("7cups_write(fp=%p, buf=%p, bytes=" CUPS_LLFMT ")", fp, buf,
634763e8 2621 CUPS_LLCAST bytes));
ecdc0628 2622
ef416fc2 2623 /*
2624 * Loop until all bytes are written...
2625 */
2626
2627 total = 0;
2628 while (bytes > 0)
2629 {
b86bc4cf 2630#ifdef WIN32
2631 if (fp->mode == 's')
2632 count = (ssize_t)send(fp->fd, buf, (unsigned)bytes, 0);
2633 else
2634 count = (ssize_t)write(fp->fd, buf, (unsigned)bytes);
2635#else
ef416fc2 2636 if (fp->mode == 's')
2637 count = send(fp->fd, buf, bytes, 0);
2638 else
2639 count = write(fp->fd, buf, bytes);
b86bc4cf 2640#endif /* WIN32 */
ef416fc2 2641
e07d4801 2642 DEBUG_printf(("9cups_write: count=" CUPS_LLFMT, CUPS_LLCAST count));
634763e8 2643
ef416fc2 2644 if (count < 0)
2645 {
2646 /*
2647 * Writes can be interrupted by signals and unavailable resources...
2648 */
2649
2650 if (errno == EAGAIN || errno == EINTR)
2651 continue;
2652 else
2653 return (-1);
2654 }
2655
2656 /*
2657 * Update the counts for the last write call...
2658 */
2659
7e86f2f6
MS
2660 bytes -= (size_t)count;
2661 total += (size_t)count;
ef416fc2 2662 buf += count;
2663 }
2664
2665 /*
2666 * Return the total number of bytes written...
2667 */
2668
b86bc4cf 2669 return ((ssize_t)total);
ef416fc2 2670}
2671
2672
2673/*
f2d18633 2674 * End of "$Id$".
ef416fc2 2675 */