src/mime_header.cc

   1
   2 /*
   3  * $Id$
   4  *
   5  * DEBUG: section 25    MiME Header Parsing
   6  * AUTHOR: Harvest Derived
   7  *
   8  * SQUID Web Proxy Cache          http://www.squid-cache.org/
   9  * ----------------------------------------------------------
  10  *
  11  *  Squid is the result of efforts by numerous individuals from
  12  *  the Internet community; see the CONTRIBUTORS file for full
  13  *  details.   Many organizations have provided support for Squid's
  14  *  development; see the SPONSORS file for full details.  Squid is
  15  *  Copyrighted (C) 2001 by the Regents of the University of
  16  *  California; see the COPYRIGHT file for full details.  Squid
  17  *  incorporates software developed and/or copyrighted by other
  18  *  sources; see the CREDITS file for full details.
  19  *
  20  *  This program is free software; you can redistribute it and/or modify
  21  *  it under the terms of the GNU General Public License as published by
  22  *  the Free Software Foundation; either version 2 of the License, or
  23  *  (at your option) any later version.
  24  *
  25  *  This program is distributed in the hope that it will be useful,
  26  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  27  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  28  *  GNU General Public License for more details.
  29  *
  30  *  You should have received a copy of the GNU General Public License
  31  *  along with this program; if not, write to the Free Software
  32  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  33  *
  34  */
  35
  36 #include "squid-old.h"
  37
  38 #define GET_HDR_SZ 1024
  39
  40 /* returns a pointer to a field-value of the first matching field-name */
  41 char *
  42 mime_get_header(const char *mime, const char *name)
  43 {
  44     return mime_get_header_field(mime, name, NULL);
  45 }
  46
  47 /*
  48  * returns a pointer to a field-value of the first matching field-name where
  49  * field-value matches prefix if any
  50  */
  51 char *
  52 mime_get_header_field(const char *mime, const char *name, const char *prefix)
  53 {
  54     LOCAL_ARRAY(char, header, GET_HDR_SZ);
  55     const char *p = NULL;
  56     char *q = NULL;
  57     char got = 0;
  58     const int namelen = name ? strlen(name) : 0;
  59     const int preflen = prefix ? strlen(prefix) : 0;
  60     int l;
  61
  62     if (NULL == mime)
  63         return NULL;
  64
  65     assert(NULL != name);
  66
  67     debugs(25, 5, "mime_get_header: looking for '" << name << "'");
  68
  69     for (p = mime; *p; p += strcspn(p, "\n\r")) {
  70         if (strcmp(p, "\r\n\r\n") == 0 || strcmp(p, "\n\n") == 0)
  71             return NULL;
  72
  73         while (xisspace(*p))
  74             ++p;
  75
  76         if (strncasecmp(p, name, namelen))
  77             continue;
  78
  79         if (!xisspace(p[namelen]) && p[namelen] != ':')
  80             continue;
  81
  82         l = strcspn(p, "\n\r") + 1;
  83
  84         if (l > GET_HDR_SZ)
  85             l = GET_HDR_SZ;
  86
  87         xstrncpy(header, p, l);
  88
  89         debugs(25, 5, "mime_get_header: checking '" << header << "'");
  90
  91         q = header;
  92
  93         q += namelen;
  94
  95         if (*q == ':') {
  96             ++q;
  97             got = 1;
  98         }
  99
 100         while (xisspace(*q)) {
 101             ++q;
 102             got = 1;
 103         }
 104
 105         if (got && prefix) {
 106             /* we could process list entries here if we had strcasestr(). */
 107             /* make sure we did not match a part of another field-value */
 108             got = !strncasecmp(q, prefix, preflen) && !xisalpha(q[preflen]);
 109         }
 110
 111         if (got) {
 112             debugs(25, 5, "mime_get_header: returning '" << q << "'");
 113             return q;
 114         }
 115     }
 116
 117     return NULL;
 118 }
 119
 120 size_t
 121 headersEnd(const char *mime, size_t l)
 122 {
 123     size_t e = 0;
 124     int state = 1;
 125
 126     PROF_start(headersEnd);
 127
 128     while (e < l && state < 3) {
 129         switch (state) {
 130
 131         case 0:
 132
 133             if ('\n' == mime[e])
 134                 state = 1;
 135
 136             break;
 137
 138         case 1:
 139             if ('\r' == mime[e])
 140                 state = 2;
 141             else if ('\n' == mime[e])
 142                 state = 3;
 143             else
 144                 state = 0;
 145
 146             break;
 147
 148         case 2:
 149             if ('\n' == mime[e])
 150                 state = 3;
 151             else
 152                 state = 0;
 153
 154             break;
 155
 156         default:
 157             break;
 158         }
 159
 160         ++e;
 161     }
 162     PROF_stop(headersEnd);
 163
 164     if (3 == state)
 165         return e;
 166
 167     return 0;
 168 }