]> git.ipfire.org Git - thirdparty/cups.git/blob - cgi-bin/var.c
Move debug printfs to internal usage only.
[thirdparty/cups.git] / cgi-bin / var.c
1 /*
2 * CGI form variable and array functions for CUPS.
3 *
4 * Copyright 2007-2015 by Apple Inc.
5 * Copyright 1997-2005 by Easy Software Products.
6 *
7 * Licensed under Apache License v2.0. See the file "LICENSE" for more information.
8 */
9
10 /*
11 * Include necessary headers...
12 */
13
14 /*#define DEBUG*/
15 #include "cgi-private.h"
16 #include <cups/http.h>
17
18
19 /*
20 * Session ID name
21 */
22
23 #define CUPS_SID "org.cups.sid"
24
25
26 /*
27 * Data structure to hold all the CGI form variables and arrays...
28 */
29
30 typedef struct /**** Form variable structure ****/
31 {
32 const char *name; /* Name of variable */
33 int nvalues, /* Number of values */
34 avalues; /* Number of values allocated */
35 const char **values; /* Value(s) of variable */
36 } _cgi_var_t;
37
38
39 /*
40 * Local globals...
41 */
42
43 static int num_cookies = 0;/* Number of cookies */
44 static cups_option_t *cookies = NULL;/* Cookies */
45 static int form_count = 0, /* Form variable count */
46 form_alloc = 0; /* Number of variables allocated */
47 static _cgi_var_t *form_vars = NULL;
48 /* Form variables */
49 static cgi_file_t *form_file = NULL;
50 /* Uploaded file */
51
52
53 /*
54 * Local functions...
55 */
56
57 static void cgi_add_variable(const char *name, int element,
58 const char *value);
59 static int cgi_compare_variables(const _cgi_var_t *v1,
60 const _cgi_var_t *v2);
61 static _cgi_var_t *cgi_find_variable(const char *name);
62 static void cgi_initialize_cookies(void);
63 static int cgi_initialize_get(void);
64 static int cgi_initialize_multipart(const char *boundary);
65 static int cgi_initialize_post(void);
66 static int cgi_initialize_string(const char *data);
67 static const char *cgi_passwd(const char *prompt);
68 static const char *cgi_set_sid(void);
69 static void cgi_sort_variables(void);
70 static void cgi_unlink_file(void);
71
72
73 /*
74 * 'cgiCheckVariables()' - Check for the presence of "required" variables.
75 *
76 * Names may be separated by spaces and/or commas.
77 */
78
79 int /* O - 1 if all variables present, 0 otherwise */
80 cgiCheckVariables(const char *names) /* I - Variables to look for */
81 {
82 char name[255], /* Current variable name */
83 *s; /* Pointer in string */
84 const char *val; /* Value of variable */
85 int element; /* Array element number */
86
87
88 if (names == NULL)
89 return (1);
90
91 while (*names != '\0')
92 {
93 while (*names == ' ' || *names == ',')
94 names ++;
95
96 for (s = name; *names != '\0' && *names != ' ' && *names != ','; s ++, names ++)
97 *s = *names;
98
99 *s = 0;
100 if (name[0] == '\0')
101 break;
102
103 if ((s = strrchr(name, '-')) != NULL)
104 {
105 *s = '\0';
106 element = atoi(s + 1) - 1;
107 val = cgiGetArray(name, element);
108 }
109 else
110 val = cgiGetVariable(name);
111
112 if (val == NULL)
113 return (0);
114
115 if (*val == '\0')
116 return (0); /* Can't be blank, either! */
117 }
118
119 return (1);
120 }
121
122
123 /*
124 * 'cgiClearVariables()' - Clear all form variables.
125 */
126
127 void
128 cgiClearVariables(void)
129 {
130 int i, j; /* Looping vars */
131 _cgi_var_t *v; /* Current variable */
132
133
134 fputs("DEBUG: cgiClearVariables called.\n", stderr);
135
136 for (v = form_vars, i = form_count; i > 0; v ++, i --)
137 {
138 _cupsStrFree(v->name);
139 for (j = 0; j < v->nvalues; j ++)
140 if (v->values[j])
141 _cupsStrFree(v->values[j]);
142 }
143
144 form_count = 0;
145
146 cgi_unlink_file();
147 }
148
149
150 /*
151 * 'cgiGetArray()' - Get an element from a form array.
152 */
153
154 const char * /* O - Element value or NULL */
155 cgiGetArray(const char *name, /* I - Name of array variable */
156 int element) /* I - Element number (0 to N) */
157 {
158 _cgi_var_t *var; /* Pointer to variable */
159
160
161 if ((var = cgi_find_variable(name)) == NULL)
162 return (NULL);
163
164 if (element < 0 || element >= var->nvalues)
165 return (NULL);
166
167 return (_cupsStrRetain(var->values[element]));
168 }
169
170
171 /*
172 * 'cgiGetCookie()' - Get a cookie value.
173 */
174
175 const char * /* O - Value or NULL */
176 cgiGetCookie(const char *name) /* I - Name of cookie */
177 {
178 return (cupsGetOption(name, num_cookies, cookies));
179 }
180
181
182 /*
183 * 'cgiGetFile()' - Get the file (if any) that was submitted in the form.
184 */
185
186 const cgi_file_t * /* O - Attached file or NULL */
187 cgiGetFile(void)
188 {
189 return (form_file);
190 }
191
192
193 /*
194 * 'cgiGetSize()' - Get the size of a form array value.
195 */
196
197 int /* O - Number of elements */
198 cgiGetSize(const char *name) /* I - Name of variable */
199 {
200 _cgi_var_t *var; /* Pointer to variable */
201
202
203 if ((var = cgi_find_variable(name)) == NULL)
204 return (0);
205
206 return (var->nvalues);
207 }
208
209
210 /*
211 * 'cgiGetVariable()' - Get a CGI variable from the database.
212 *
213 * Returns NULL if the variable doesn't exist. If the variable is an
214 * array of values, returns the last element.
215 */
216
217 const char * /* O - Value of variable */
218 cgiGetVariable(const char *name) /* I - Name of variable */
219 {
220 const _cgi_var_t *var; /* Returned variable */
221
222
223 var = cgi_find_variable(name);
224
225 return ((var == NULL) ? NULL : _cupsStrRetain(var->values[var->nvalues - 1]));
226 }
227
228
229 /*
230 * 'cgiInitialize()' - Initialize the CGI variable "database".
231 */
232
233 int /* O - Non-zero if there was form data */
234 cgiInitialize(void)
235 {
236 const char *method, /* Form posting method */
237 *content_type, /* Content-Type of post data */
238 *cups_sid_cookie, /* SID cookie */
239 *cups_sid_form; /* SID form variable */
240
241
242 /*
243 * Setup a password callback for authentication...
244 */
245
246 cupsSetPasswordCB(cgi_passwd);
247
248 /*
249 * Set the locale so that times, etc. are formatted properly...
250 */
251
252 setlocale(LC_ALL, "");
253
254 #ifdef DEBUG
255 /*
256 * Disable output buffering to find bugs...
257 */
258
259 setbuf(stdout, NULL);
260 #endif /* DEBUG */
261
262 /*
263 * Get cookies...
264 */
265
266 cgi_initialize_cookies();
267
268 if ((cups_sid_cookie = cgiGetCookie(CUPS_SID)) == NULL)
269 {
270 fputs("DEBUG: " CUPS_SID " cookie not found, initializing!\n", stderr);
271 cups_sid_cookie = cgi_set_sid();
272 }
273
274 fprintf(stderr, "DEBUG: " CUPS_SID " cookie is \"%s\"\n", cups_sid_cookie);
275
276 /*
277 * Get the request method (GET or POST)...
278 */
279
280 method = getenv("REQUEST_METHOD");
281 content_type = getenv("CONTENT_TYPE");
282 if (!method)
283 return (0);
284
285 /*
286 * Grab form data from the corresponding location...
287 */
288
289 if (!_cups_strcasecmp(method, "GET"))
290 return (cgi_initialize_get());
291 else if (!_cups_strcasecmp(method, "POST") && content_type)
292 {
293 const char *boundary = strstr(content_type, "boundary=");
294
295 if (boundary)
296 boundary += 9;
297
298 if (content_type && !strncmp(content_type, "multipart/form-data; ", 21))
299 {
300 if (!cgi_initialize_multipart(boundary))
301 return (0);
302 }
303 else if (!cgi_initialize_post())
304 return (0);
305
306 if ((cups_sid_form = cgiGetVariable(CUPS_SID)) == NULL ||
307 strcmp(cups_sid_cookie, cups_sid_form))
308 {
309 if (cups_sid_form)
310 fprintf(stderr, "DEBUG: " CUPS_SID " form variable is \"%s\"\n",
311 cups_sid_form);
312 else
313 fputs("DEBUG: " CUPS_SID " form variable is not present.\n", stderr);
314
315 cgiClearVariables();
316 return (0);
317 }
318 else
319 return (1);
320 }
321 else
322 return (0);
323 }
324
325
326 /*
327 * 'cgiIsPOST()' - Determine whether this page was POSTed.
328 */
329
330 int /* O - 1 if POST, 0 if GET */
331 cgiIsPOST(void)
332 {
333 const char *method; /* REQUEST_METHOD environment variable */
334
335
336 if ((method = getenv("REQUEST_METHOD")) == NULL)
337 return (0);
338 else
339 return (!strcmp(method, "POST"));
340 }
341
342
343 /*
344 * 'cgiSetArray()' - Set array element N to the specified string.
345 *
346 * If the variable array is smaller than (element + 1), the intervening
347 * elements are set to NULL.
348 */
349
350 void
351 cgiSetArray(const char *name, /* I - Name of variable */
352 int element, /* I - Element number (0 to N) */
353 const char *value) /* I - Value of variable */
354 {
355 int i; /* Looping var */
356 _cgi_var_t *var; /* Returned variable */
357
358
359 if (name == NULL || value == NULL || element < 0 || element > 100000)
360 return;
361
362 fprintf(stderr, "DEBUG: cgiSetArray: %s[%d]=\"%s\"\n", name, element, value);
363
364 if ((var = cgi_find_variable(name)) == NULL)
365 {
366 cgi_add_variable(name, element, value);
367 cgi_sort_variables();
368 }
369 else
370 {
371 if (element >= var->avalues)
372 {
373 const char **temp; /* Temporary pointer */
374
375 temp = (const char **)realloc((void *)(var->values),
376 sizeof(char *) * (size_t)(element + 16));
377 if (!temp)
378 return;
379
380 var->avalues = element + 16;
381 var->values = temp;
382 }
383
384 if (element >= var->nvalues)
385 {
386 for (i = var->nvalues; i < element; i ++)
387 var->values[i] = NULL;
388
389 var->nvalues = element + 1;
390 }
391 else if (var->values[element])
392 _cupsStrFree((char *)var->values[element]);
393
394 var->values[element] = _cupsStrAlloc(value);
395 }
396 }
397
398
399 /*
400 * 'cgiSetCookie()' - Set a cookie value.
401 */
402
403 void
404 cgiSetCookie(const char *name, /* I - Name */
405 const char *value, /* I - Value */
406 const char *path, /* I - Path (typically "/") */
407 const char *domain, /* I - Domain name */
408 time_t expires, /* I - Expiration date (0 for session) */
409 int secure) /* I - Require SSL */
410 {
411 num_cookies = cupsAddOption(name, value, num_cookies, &cookies);
412
413 printf("Set-Cookie: %s=%s;", name, value);
414 if (path)
415 printf(" path=%s;", path);
416 if (domain)
417 printf(" domain=%s;", domain);
418 if (expires)
419 {
420 char date[256]; /* Date string */
421
422 printf(" expires=%s;", httpGetDateString2(expires, date, sizeof(date)));
423 }
424 if (secure)
425 puts(" httponly; secure;");
426 else
427 puts(" httponly;");
428 }
429
430
431 /*
432 * 'cgiSetSize()' - Set the array size.
433 */
434
435 void
436 cgiSetSize(const char *name, /* I - Name of variable */
437 int size) /* I - Number of elements (0 to N) */
438 {
439 int i; /* Looping var */
440 _cgi_var_t *var; /* Returned variable */
441
442
443 if (name == NULL || size < 0 || size > 100000)
444 return;
445
446 if ((var = cgi_find_variable(name)) == NULL)
447 return;
448
449 if (size >= var->avalues)
450 {
451 const char **temp; /* Temporary pointer */
452
453 temp = (const char **)realloc((void *)(var->values),
454 sizeof(char *) * (size_t)(size + 16));
455 if (!temp)
456 return;
457
458 var->avalues = size + 16;
459 var->values = temp;
460 }
461
462 if (size > var->nvalues)
463 {
464 for (i = var->nvalues; i < size; i ++)
465 var->values[i] = NULL;
466 }
467 else if (size < var->nvalues)
468 {
469 for (i = size; i < var->nvalues; i ++)
470 if (var->values[i])
471 _cupsStrFree((void *)(var->values[i]));
472 }
473
474 var->nvalues = size;
475 }
476
477
478 /*
479 * 'cgiSetVariable()' - Set a CGI variable in the database.
480 *
481 * If the variable is an array, this truncates the array to a single element.
482 */
483
484 void
485 cgiSetVariable(const char *name, /* I - Name of variable */
486 const char *value) /* I - Value of variable */
487 {
488 int i; /* Looping var */
489 _cgi_var_t *var; /* Returned variable */
490
491
492 if (name == NULL || value == NULL)
493 return;
494
495 fprintf(stderr, "cgiSetVariable: %s=\"%s\"\n", name, value);
496
497 if ((var = cgi_find_variable(name)) == NULL)
498 {
499 cgi_add_variable(name, 0, value);
500 cgi_sort_variables();
501 }
502 else
503 {
504 for (i = 0; i < var->nvalues; i ++)
505 if (var->values[i])
506 _cupsStrFree((char *)var->values[i]);
507
508 var->values[0] = _cupsStrAlloc(value);
509 var->nvalues = 1;
510 }
511 }
512
513
514 /*
515 * 'cgi_add_variable()' - Add a form variable.
516 */
517
518 static void
519 cgi_add_variable(const char *name, /* I - Variable name */
520 int element, /* I - Array element number */
521 const char *value) /* I - Variable value */
522 {
523 _cgi_var_t *var; /* New variable */
524
525
526 if (name == NULL || value == NULL || element < 0 || element > 100000)
527 return;
528
529 if (form_count >= form_alloc)
530 {
531 _cgi_var_t *temp_vars; /* Temporary form pointer */
532
533
534 if (form_alloc == 0)
535 temp_vars = malloc(sizeof(_cgi_var_t) * 16);
536 else
537 temp_vars = realloc(form_vars, (size_t)(form_alloc + 16) * sizeof(_cgi_var_t));
538
539 if (!temp_vars)
540 return;
541
542 form_vars = temp_vars;
543 form_alloc += 16;
544 }
545
546 var = form_vars + form_count;
547
548 if ((var->values = calloc((size_t)element + 1, sizeof(char *))) == NULL)
549 return;
550
551 var->name = _cupsStrAlloc(name);
552 var->nvalues = element + 1;
553 var->avalues = element + 1;
554 var->values[element] = _cupsStrAlloc(value);
555
556 form_count ++;
557 }
558
559
560 /*
561 * 'cgi_compare_variables()' - Compare two variables.
562 */
563
564 static int /* O - Result of comparison */
565 cgi_compare_variables(
566 const _cgi_var_t *v1, /* I - First variable */
567 const _cgi_var_t *v2) /* I - Second variable */
568 {
569 return (_cups_strcasecmp(v1->name, v2->name));
570 }
571
572
573 /*
574 * 'cgi_find_variable()' - Find a variable.
575 */
576
577 static _cgi_var_t * /* O - Variable pointer or NULL */
578 cgi_find_variable(const char *name) /* I - Name of variable */
579 {
580 _cgi_var_t key; /* Search key */
581
582
583 if (form_count < 1 || name == NULL)
584 return (NULL);
585
586 key.name = name;
587
588 return ((_cgi_var_t *)bsearch(&key, form_vars, (size_t)form_count, sizeof(_cgi_var_t),
589 (int (*)(const void *, const void *))cgi_compare_variables));
590 }
591
592
593 /*
594 * 'cgi_initialize_cookies()' - Initialize cookies.
595 */
596
597 static void
598 cgi_initialize_cookies(void)
599 {
600 const char *cookie; /* HTTP_COOKIE environment variable */
601 char name[128], /* Name string */
602 value[512], /* Value string */
603 *ptr; /* Pointer into name/value */
604
605
606 if ((cookie = getenv("HTTP_COOKIE")) == NULL)
607 return;
608
609 while (*cookie)
610 {
611 int skip = 0; /* Skip this cookie? */
612
613 /*
614 * Skip leading whitespace...
615 */
616
617 while (isspace(*cookie & 255))
618 cookie ++;
619 if (!*cookie)
620 break;
621
622 /*
623 * Copy the name...
624 */
625
626 for (ptr = name; *cookie && *cookie != '=';)
627 if (ptr < (name + sizeof(name) - 1))
628 {
629 *ptr++ = *cookie++;
630 }
631 else
632 {
633 skip = 1;
634 cookie ++;
635 }
636
637 if (*cookie != '=')
638 break;
639
640 *ptr = '\0';
641 cookie ++;
642
643 /*
644 * Then the value...
645 */
646
647 if (*cookie == '\"')
648 {
649 for (cookie ++, ptr = value; *cookie && *cookie != '\"';)
650 if (ptr < (value + sizeof(value) - 1))
651 {
652 *ptr++ = *cookie++;
653 }
654 else
655 {
656 skip = 1;
657 cookie ++;
658 }
659
660 if (*cookie == '\"')
661 cookie ++;
662 else
663 skip = 1;
664 }
665 else
666 {
667 for (ptr = value; *cookie && *cookie != ';';)
668 if (ptr < (value + sizeof(value) - 1))
669 {
670 *ptr++ = *cookie++;
671 }
672 else
673 {
674 skip = 1;
675 cookie ++;
676 }
677 }
678
679 if (*cookie == ';')
680 cookie ++;
681 else if (*cookie)
682 skip = 1;
683
684 *ptr = '\0';
685
686 /*
687 * Then add the cookie to an array as long as the name doesn't start with
688 * "$"...
689 */
690
691 if (name[0] != '$' && !skip)
692 num_cookies = cupsAddOption(name, value, num_cookies, &cookies);
693 }
694 }
695
696
697 /*
698 * 'cgi_initialize_get()' - Initialize form variables using the GET method.
699 */
700
701 static int /* O - 1 if form data read */
702 cgi_initialize_get(void)
703 {
704 char *data; /* Pointer to form data string */
705
706
707 /*
708 * Check to see if there is anything for us to read...
709 */
710
711 data = getenv("QUERY_STRING");
712 if (data == NULL || strlen(data) == 0)
713 return (0);
714
715 /*
716 * Parse it out and return...
717 */
718
719 return (cgi_initialize_string(data));
720 }
721
722
723 /*
724 * 'cgi_initialize_multipart()' - Initialize variables and file using the POST
725 * method.
726 *
727 * TODO: Update to support files > 2GB.
728 */
729
730 static int /* O - 1 if form data was read */
731 cgi_initialize_multipart(
732 const char *boundary) /* I - Boundary string */
733 {
734 char line[10240], /* MIME header line */
735 name[1024], /* Form variable name */
736 filename[1024], /* Form filename */
737 mimetype[1024], /* MIME media type */
738 bstring[256], /* Boundary string to look for */
739 *ptr, /* Pointer into name/filename */
740 *end; /* End of buffer */
741 int ch, /* Character from file */
742 fd; /* Temporary file descriptor */
743 size_t blen; /* Length of boundary string */
744
745
746 /*
747 * Read multipart form data until we run out...
748 */
749
750 name[0] = '\0';
751 filename[0] = '\0';
752 mimetype[0] = '\0';
753
754 snprintf(bstring, sizeof(bstring), "\r\n--%s", boundary);
755 blen = strlen(bstring);
756
757 while (fgets(line, sizeof(line), stdin))
758 {
759 if (!strcmp(line, "\r\n"))
760 {
761 /*
762 * End of headers, grab value...
763 */
764
765 if (filename[0])
766 {
767 /*
768 * Read an embedded file...
769 */
770
771 if (form_file)
772 {
773 /*
774 * Remove previous file...
775 */
776
777 cgi_unlink_file();
778 }
779
780 /*
781 * Allocate memory for the new file...
782 */
783
784 if ((form_file = calloc(1, sizeof(cgi_file_t))) == NULL)
785 return (0);
786
787 form_file->name = strdup(name);
788 form_file->filename = strdup(filename);
789 form_file->mimetype = strdup(mimetype);
790
791 fd = cupsTempFd(form_file->tempfile, sizeof(form_file->tempfile));
792
793 if (fd < 0)
794 return (0);
795
796 atexit(cgi_unlink_file);
797
798 /*
799 * Copy file data to the temp file...
800 */
801
802 ptr = line;
803
804 while ((ch = getchar()) != EOF)
805 {
806 *ptr++ = (char)ch;
807
808 if ((size_t)(ptr - line) >= blen && !memcmp(ptr - blen, bstring, blen))
809 {
810 ptr -= blen;
811 break;
812 }
813
814 if ((ptr - line - (int)blen) >= 8192)
815 {
816 /*
817 * Write out the first 8k of the buffer...
818 */
819
820 write(fd, line, 8192);
821 memmove(line, line + 8192, (size_t)(ptr - line - 8192));
822 ptr -= 8192;
823 }
824 }
825
826 /*
827 * Write the rest of the data and close the temp file...
828 */
829
830 if (ptr > line)
831 write(fd, line, (size_t)(ptr - line));
832
833 close(fd);
834 }
835 else
836 {
837 /*
838 * Just get a form variable; the current code only handles
839 * form values up to 10k in size...
840 */
841
842 ptr = line;
843 end = line + sizeof(line) - 1;
844
845 while ((ch = getchar()) != EOF)
846 {
847 if (ptr < end)
848 *ptr++ = (char)ch;
849
850 if ((size_t)(ptr - line) >= blen && !memcmp(ptr - blen, bstring, blen))
851 {
852 ptr -= blen;
853 break;
854 }
855 }
856
857 *ptr = '\0';
858
859 /*
860 * Set the form variable...
861 */
862
863 if ((ptr = strrchr(name, '-')) != NULL && isdigit(ptr[1] & 255))
864 {
865 /*
866 * Set a specific index in the array...
867 */
868
869 *ptr++ = '\0';
870 if (line[0])
871 cgiSetArray(name, atoi(ptr) - 1, line);
872 }
873 else if (cgiGetVariable(name))
874 {
875 /*
876 * Add another element in the array...
877 */
878
879 cgiSetArray(name, cgiGetSize(name), line);
880 }
881 else
882 {
883 /*
884 * Just set the line...
885 */
886
887 cgiSetVariable(name, line);
888 }
889 }
890
891 /*
892 * Read the rest of the current line...
893 */
894
895 fgets(line, sizeof(line), stdin);
896
897 /*
898 * Clear the state vars...
899 */
900
901 name[0] = '\0';
902 filename[0] = '\0';
903 mimetype[0] = '\0';
904 }
905 else if (!_cups_strncasecmp(line, "Content-Disposition:", 20))
906 {
907 if ((ptr = strstr(line + 20, " name=\"")) != NULL)
908 {
909 strlcpy(name, ptr + 7, sizeof(name));
910
911 if ((ptr = strchr(name, '\"')) != NULL)
912 *ptr = '\0';
913 }
914
915 if ((ptr = strstr(line + 20, " filename=\"")) != NULL)
916 {
917 strlcpy(filename, ptr + 11, sizeof(filename));
918
919 if ((ptr = strchr(filename, '\"')) != NULL)
920 *ptr = '\0';
921 }
922 }
923 else if (!_cups_strncasecmp(line, "Content-Type:", 13))
924 {
925 for (ptr = line + 13; isspace(*ptr & 255); ptr ++);
926
927 strlcpy(mimetype, ptr, sizeof(mimetype));
928
929 for (ptr = mimetype + strlen(mimetype) - 1;
930 ptr > mimetype && isspace(*ptr & 255);
931 *ptr-- = '\0');
932 }
933 }
934
935 /*
936 * Return 1 for "form data found"...
937 */
938
939 return (1);
940 }
941
942
943 /*
944 * 'cgi_initialize_post()' - Initialize variables using the POST method.
945 */
946
947 static int /* O - 1 if form data was read */
948 cgi_initialize_post(void)
949 {
950 char *content_length, /* Length of input data (string) */
951 *data; /* Pointer to form data string */
952 size_t length, /* Length of input data */
953 tbytes; /* Total number of bytes read */
954 ssize_t nbytes; /* Number of bytes read this read() */
955 int status; /* Return status */
956
957
958 /*
959 * Check to see if there is anything for us to read...
960 */
961
962 content_length = getenv("CONTENT_LENGTH");
963 if (content_length == NULL || atoi(content_length) <= 0)
964 return (0);
965
966 /*
967 * Get the length of the input stream and allocate a buffer for it...
968 */
969
970 length = (size_t)strtol(content_length, NULL, 10);
971 data = malloc(length + 1);
972
973 if (data == NULL)
974 return (0);
975
976 /*
977 * Read the data into the buffer...
978 */
979
980 for (tbytes = 0; tbytes < length; tbytes += (size_t)nbytes)
981 if ((nbytes = read(0, data + tbytes, (size_t)(length - tbytes))) < 0)
982 {
983 if (errno != EAGAIN)
984 {
985 free(data);
986 return (0);
987 }
988 else
989 nbytes = 0;
990 }
991 else if (nbytes == 0)
992 {
993 /*
994 * CUPS STR #3176: OpenBSD: Early end-of-file on POST data causes 100% CPU
995 *
996 * This should never happen, but does on OpenBSD. If we see early end-of-
997 * file, treat this as an error and process no data.
998 */
999
1000 free(data);
1001 return (0);
1002 }
1003
1004 data[length] = '\0';
1005
1006 /*
1007 * Parse it out...
1008 */
1009
1010 status = cgi_initialize_string(data);
1011
1012 /*
1013 * Free the data and return...
1014 */
1015
1016 free(data);
1017
1018 return (status);
1019 }
1020
1021
1022 /*
1023 * 'cgi_initialize_string()' - Initialize form variables from a string.
1024 */
1025
1026 static int /* O - 1 if form data was processed */
1027 cgi_initialize_string(const char *data) /* I - Form data string */
1028 {
1029 int done; /* True if we're done reading a form variable */
1030 char *s, /* Pointer to current form string */
1031 ch, /* Temporary character */
1032 name[255], /* Name of form variable */
1033 value[65536]; /* Variable value */
1034
1035
1036 /*
1037 * Check input...
1038 */
1039
1040 if (data == NULL)
1041 return (0);
1042
1043 /*
1044 * Loop until we've read all the form data...
1045 */
1046
1047 while (*data != '\0')
1048 {
1049 /*
1050 * Get the variable name...
1051 */
1052
1053 for (s = name; *data != '\0'; data ++)
1054 if (*data == '=')
1055 break;
1056 else if (*data >= ' ' && s < (name + sizeof(name) - 1))
1057 *s++ = *data;
1058
1059 *s = '\0';
1060 if (*data == '=')
1061 data ++;
1062 else
1063 return (0);
1064
1065 /*
1066 * Read the variable value...
1067 */
1068
1069 for (s = value, done = 0; !done && *data != '\0'; data ++)
1070 switch (*data)
1071 {
1072 case '&' : /* End of data... */
1073 done = 1;
1074 break;
1075
1076 case '+' : /* Escaped space character */
1077 if (s < (value + sizeof(value) - 1))
1078 *s++ = ' ';
1079 break;
1080
1081 case '%' : /* Escaped control character */
1082 /*
1083 * Read the hex code...
1084 */
1085
1086 if (!isxdigit(data[1] & 255) || !isxdigit(data[2] & 255))
1087 return (0);
1088
1089 if (s < (value + sizeof(value) - 1))
1090 {
1091 data ++;
1092 ch = *data - '0';
1093 if (ch > 9)
1094 ch -= 7;
1095 *s = (char)(ch << 4);
1096
1097 data ++;
1098 ch = *data - '0';
1099 if (ch > 9)
1100 ch -= 7;
1101 *s++ |= ch;
1102 }
1103 else
1104 data += 2;
1105 break;
1106
1107 default : /* Other characters come straight through */
1108 if (*data >= ' ' && s < (value + sizeof(value) - 1))
1109 *s++ = *data;
1110 break;
1111 }
1112
1113 *s = '\0'; /* nul terminate the string */
1114
1115 /*
1116 * Remove trailing whitespace...
1117 */
1118
1119 if (s > value)
1120 s --;
1121
1122 while (s >= value && isspace(*s & 255))
1123 *s-- = '\0';
1124
1125 /*
1126 * Add the string to the variable "database"...
1127 */
1128
1129 if ((s = strrchr(name, '-')) != NULL && isdigit(s[1] & 255))
1130 {
1131 *s++ = '\0';
1132 if (value[0])
1133 cgiSetArray(name, atoi(s) - 1, value);
1134 }
1135 else if (cgiGetVariable(name) != NULL)
1136 cgiSetArray(name, cgiGetSize(name), value);
1137 else
1138 cgiSetVariable(name, value);
1139 }
1140
1141 return (1);
1142 }
1143
1144
1145 /*
1146 * 'cgi_passwd()' - Catch authentication requests and notify the server.
1147 *
1148 * This function sends a Status header and exits, forcing authentication
1149 * for this request.
1150 */
1151
1152 static const char * /* O - NULL (no return) */
1153 cgi_passwd(const char *prompt) /* I - Prompt (not used) */
1154 {
1155 (void)prompt;
1156
1157 fprintf(stderr, "DEBUG: cgi_passwd(prompt=\"%s\") called!\n",
1158 prompt ? prompt : "(null)");
1159
1160 /*
1161 * Send a 401 (unauthorized) status to the server, so it can notify
1162 * the client that authentication is required.
1163 */
1164
1165 puts("Status: 401\n");
1166 exit(0);
1167
1168 /*
1169 * This code is never executed, but is present to satisfy the compiler.
1170 */
1171
1172 return (NULL);
1173 }
1174
1175
1176 /*
1177 * 'cgi_set_sid()' - Set the CUPS session ID.
1178 */
1179
1180 static const char * /* O - New session ID */
1181 cgi_set_sid(void)
1182 {
1183 char buffer[512], /* SID data */
1184 sid[33]; /* SID string */
1185 unsigned char sum[16]; /* MD5 sum */
1186 const char *remote_addr, /* REMOTE_ADDR */
1187 *server_name, /* SERVER_NAME */
1188 *server_port; /* SERVER_PORT */
1189
1190
1191 if ((remote_addr = getenv("REMOTE_ADDR")) == NULL)
1192 remote_addr = "REMOTE_ADDR";
1193 if ((server_name = getenv("SERVER_NAME")) == NULL)
1194 server_name = "SERVER_NAME";
1195 if ((server_port = getenv("SERVER_PORT")) == NULL)
1196 server_port = "SERVER_PORT";
1197
1198 CUPS_SRAND(time(NULL));
1199 snprintf(buffer, sizeof(buffer), "%s:%s:%s:%02X%02X%02X%02X%02X%02X%02X%02X",
1200 remote_addr, server_name, server_port,
1201 (unsigned)CUPS_RAND() & 255, (unsigned)CUPS_RAND() & 255,
1202 (unsigned)CUPS_RAND() & 255, (unsigned)CUPS_RAND() & 255,
1203 (unsigned)CUPS_RAND() & 255, (unsigned)CUPS_RAND() & 255,
1204 (unsigned)CUPS_RAND() & 255, (unsigned)CUPS_RAND() & 255);
1205 cupsHashData("md5", (unsigned char *)buffer, strlen(buffer), sum, sizeof(sum));
1206
1207 cgiSetCookie(CUPS_SID, cupsHashString(sum, sizeof(sum), sid, sizeof(sid)), "/", NULL, 0, 0);
1208
1209 return (cupsGetOption(CUPS_SID, num_cookies, cookies));
1210 }
1211
1212
1213 /*
1214 * 'cgi_sort_variables()' - Sort all form variables for faster lookup.
1215 */
1216
1217 static void
1218 cgi_sort_variables(void)
1219 {
1220 if (form_count < 2)
1221 return;
1222
1223 qsort(form_vars, (size_t)form_count, sizeof(_cgi_var_t),
1224 (int (*)(const void *, const void *))cgi_compare_variables);
1225 }
1226
1227
1228 /*
1229 * 'cgi_unlink_file()' - Remove the uploaded form.
1230 */
1231
1232 static void
1233 cgi_unlink_file(void)
1234 {
1235 if (form_file)
1236 {
1237 /*
1238 * Remove the temporary file...
1239 */
1240
1241 unlink(form_file->tempfile);
1242
1243 /*
1244 * Free memory used...
1245 */
1246
1247 free(form_file->name);
1248 free(form_file->filename);
1249 free(form_file->mimetype);
1250 free(form_file);
1251
1252 form_file = NULL;
1253 }
1254 }