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.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++, got = 1;
  97
  98         while (xisspace(*q))
  99             q++, got = 1;
 100
 101         if (got && prefix) {
 102             /* we could process list entries here if we had strcasestr(). */
 103             /* make sure we did not match a part of another field-value */
 104             got = !strncasecmp(q, prefix, preflen) && !xisalpha(q[preflen]);
 105         }
 106
 107         if (got) {
 108             debugs(25, 5, "mime_get_header: returning '" << q << "'");
 109             return q;
 110         }
 111     }
 112
 113     return NULL;
 114 }
 115
 116 size_t
 117 headersEnd(const char *mime, size_t l)
 118 {
 119     size_t e = 0;
 120     int state = 1;
 121
 122     PROF_start(headersEnd);
 123
 124     while (e < l && state < 3) {
 125         switch (state) {
 126
 127         case 0:
 128
 129             if ('\n' == mime[e])
 130                 state = 1;
 131
 132             break;
 133
 134         case 1:
 135             if ('\r' == mime[e])
 136                 state = 2;
 137             else if ('\n' == mime[e])
 138                 state = 3;
 139             else
 140                 state = 0;
 141
 142             break;
 143
 144         case 2:
 145             if ('\n' == mime[e])
 146                 state = 3;
 147             else
 148                 state = 0;
 149
 150             break;
 151
 152         default:
 153             break;
 154         }
 155
 156         e++;
 157     }
 158     PROF_stop(headersEnd);
 159
 160     if (3 == state)
 161         return e;
 162
 163     return 0;
 164 }