开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 代理服务器 > 查看源码
mime.c
资源名称：squid-2.2.devel3-src.tar.gz　[点击查看]
上传用户：liugui
上传日期：2007-01-04
资源大小：822k
文件大小：11k
源码类别：

代理服务器

开发平台：

Unix_Linux

mime.c：源码内容
  2. /*
  3.  * $Id: mime.c,v 1.84.2.3 1999/02/12 22:18:59 wessels Exp $
  4.  *
  5.  * DEBUG: section 25    MIME Parsing
  6.  * AUTHOR: Harvest Derived
  7.  *
  8.  * SQUID Internet Object Cache  http://squid.nlanr.net/Squid/
  9.  * ----------------------------------------------------------
  10.  *
  11.  *  Squid is the result of efforts by numerous individuals from the
  12.  *  Internet community.  Development is led by Duane Wessels of the
  13.  *  National Laboratory for Applied Network Research and funded by the
  14.  *  National Science Foundation.  Squid is Copyrighted (C) 1998 by
  15.  *  Duane Wessels and the University of California San Diego.  Please
  16.  *  see the COPYRIGHT file for full details.  Squid incorporates
  17.  *  software developed and/or copyrighted by other sources.  Please see
  18.  *  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.  */
  36. #include "squid.h"
  38. #define GET_HDR_SZ 1024
  40. typedef struct _mime_entry {
  41.     char *pattern;
  42.     regex_t compiled_pattern;
  43.     char *icon;
  44.     char *content_type;
  45.     char *content_encoding;
  46.     char transfer_mode;
  47.     unsigned int view_option:1, download_option:1;
  48.     struct _mime_entry *next;
  49. } mimeEntry;
  51. static mimeEntry *MimeTable = NULL;
  52. static mimeEntry **MimeTableTail = NULL;
  54. static void mimeLoadIconFile(const char *icon);
  56. /* returns a pointer to a field-value of the first matching field-name */
  57. char *
  58. mime_get_header(const char *mime, const char *name)
  59. {
  60.     return mime_get_header_field(mime, name, NULL);
  61. }
  63. /*
  64.  * returns a pointer to a field-value of the first matching field-name where
  65.  * field-value matches prefix if any
  66.  */
  67. char *
  68. mime_get_header_field(const char *mime, const char *name, const char *prefix)
  69. {
  70.     LOCAL_ARRAY(char, header, GET_HDR_SZ);
  71.     const char *p = NULL;
  72.     char *q = NULL;
  73.     char got = 0;
  74.     const int namelen = name ? strlen(name) : 0;
  75.     const int preflen = prefix ? strlen(prefix) : 0;
  76.     int l;
  78.     if (NULL == mime)
  79. return NULL;
  80.     assert(NULL != name);
  82.     debug(25, 5) ("mime_get_header: looking for '%s'n", name);
  84.     for (p = mime; *p; p += strcspn(p, "nr")) {
  85. if (strcmp(p, "rnrn") == 0 || strcmp(p, "nn") == 0)
  86.     return NULL;
  87. while (xisspace(*p))
  88.     p++;
  89. if (strncasecmp(p, name, namelen))
  90.     continue;
  91. if (!xisspace(p[namelen]) && p[namelen] != ':')
  92.     continue;
  93. l = strcspn(p, "nr") + 1;
  94. if (l > GET_HDR_SZ)
  95.     l = GET_HDR_SZ;
  96. xstrncpy(header, p, l);
  97. debug(25, 5) ("mime_get_header: checking '%s'n", header);
  98. q = header;
  99. q += namelen;
  100. if (*q == ':')
  101.     q++, got = 1;
  102. while (xisspace(*q))
  103.     q++, got = 1;
  104. if (got && prefix) {
  105.     /* we could process list entries here if we had strcasestr(). */
  106.     /* make sure we did not match a part of another field-value */
  107.     got = !strncasecmp(q, prefix, preflen) && !xisalpha(q[preflen]);
  108. }
  109. if (got) {
  110.     debug(25, 5) ("mime_get_header: returning '%s'n", q);
  111.     return q;
  112. }
  113.     }
  114.     return NULL;
  115. }
  117. size_t
  118. headersEnd(const char *mime, size_t l)
  119. {
  120.     size_t e = 0;
  121.     int state = 0;
  122.     while (e < l && state < 3) {
  123. switch (state) {
  124. case 0:
  125.     if ('n' == mime[e])
  126. state = 1;
  127.     break;
  128. case 1:
  129.     if ('r' == mime[e])
  130. state = 2;
  131.     else if ('n' == mime[e])
  132. state = 3;
  133.     else
  134. state = 0;
  135.     break;
  136. case 2:
  137.     if ('r' == mime[e]) /* ignore repeated CR */
  138. (void) 0;
  139.     else if ('n' == mime[e])
  140. state = 3;
  141.     else
  142. state = 0;
  143.     break;
  144. default:
  145.     break;
  146. }
  147. e++;
  148.     }
  149.     if (3 == state)
  150. return e;
  151.     return 0;
  152. }
  154. const char *
  155. mime_get_auth(const char *hdr, const char *auth_scheme, const char **auth_field)
  156. {
  157.     char *auth_hdr;
  158.     char *t;
  159.     if (auth_field)
  160. *auth_field = NULL;
  161.     if (hdr == NULL)
  162. return NULL;
  163.     if ((auth_hdr = mime_get_header(hdr, "Authorization")) == NULL)
  164. return NULL;
  165.     if (auth_field)
  166. *auth_field = auth_hdr;
  167.     if ((t = strtok(auth_hdr, " t")) == NULL)
  168. return NULL;
  169.     if (strcasecmp(t, auth_scheme) != 0)
  170. return NULL;
  171.     if ((t = strtok(NULL, " t")) == NULL)
  172. return NULL;
  173.     return base64_decode(t);
  174. }
  176. static mimeEntry *
  177. mimeGetEntry(const char *fn, int skip_encodings)
  178. {
  179.     mimeEntry *m;
  180.     char *t;
  181.     char *name = xstrdup(fn);
  182.   try_again:
  183.     for (m = MimeTable; m; m = m->next) {
  184. if (regexec(&m->compiled_pattern, name, 0, 0, 0) == 0)
  185.     break;
  186.     }
  187.     if (!skip_encodings)
  188. (void) 0;
  189.     else if (m == NULL)
  190. (void) 0;
  191.     else if (strcmp(m->content_type, dash_str))
  192. (void) 0;
  193.     else if (!strcmp(m->content_encoding, dash_str))
  194. (void) 0;
  195.     else {
  196. /* Assume we matched /.w$/ and cut off the last extension */
  197. if ((t = strrchr(name, '.'))) {
  198.     *t = '';
  199.     goto try_again;
  200. }
  201. /* What? A encoding without a extension? */
  202. m = NULL;
  203.     }
  204.     xfree(name);
  205.     return m;
  206. }
  208. char *
  209. mimeGetIcon(const char *fn)
  210. {
  211.     mimeEntry *m = mimeGetEntry(fn, 1);
  212.     if (m == NULL)
  213. return NULL;
  214.     if (!strcmp(m->icon, dash_str))
  215. return NULL;
  216.     return m->icon;
  217. }
  219. char *
  220. mimeGetIconURL(const char *fn)
  221. {
  222.     char *icon = mimeGetIcon(fn);
  223.     if (icon == NULL)
  224. return NULL;
  225.     return internalLocalUri("/squid-internal-static/icons/", icon);
  226. }
  228. char *
  229. mimeGetContentType(const char *fn)
  230. {
  231.     mimeEntry *m = mimeGetEntry(fn, 1);
  232.     if (m == NULL)
  233. return NULL;
  234.     if (!strcmp(m->content_type, dash_str))
  235. return NULL;
  236.     return m->content_type;
  237. }
  239. char *
  240. mimeGetContentEncoding(const char *fn)
  241. {
  242.     mimeEntry *m = mimeGetEntry(fn, 0);
  243.     if (m == NULL)
  244. return NULL;
  245.     if (!strcmp(m->content_encoding, dash_str))
  246. return NULL;
  247.     return m->content_encoding;
  248. }
  250. char
  251. mimeGetTransferMode(const char *fn)
  252. {
  253.     mimeEntry *m = mimeGetEntry(fn, 0);
  254.     return m ? m->transfer_mode : 'I';
  255. }
  257. int
  258. mimeGetDownloadOption(const char *fn)
  259. {
  260.     mimeEntry *m = mimeGetEntry(fn, 1);
  261.     return m ? m->download_option : 0;
  262. }
  264. int
  265. mimeGetViewOption(const char *fn)
  266. {
  267.     mimeEntry *m = mimeGetEntry(fn, 0);
  268.     return m ? m->view_option : 0;
  269. }
  271. void
  272. mimeInit(char *filename)
  273. {
  274.     FILE *fp;
  275.     char buf[BUFSIZ];
  276.     char chopbuf[BUFSIZ];
  277.     char *t;
  278.     char *pattern;
  279.     char *icon;
  280.     char *type;
  281.     char *encoding;
  282.     char *mode;
  283.     char *option;
  284.     int view_option;
  285.     int download_option;
  286.     regex_t re;
  287.     mimeEntry *m;
  288.     int re_flags = REG_EXTENDED | REG_NOSUB | REG_ICASE;
  289.     if (filename == NULL)
  290. return;
  291.     if ((fp = fopen(filename, "r")) == NULL) {
  292. debug(50, 1) ("mimeInit: %s: %sn", filename, xstrerror());
  293. return;
  294.     }
  295.     if (MimeTableTail == NULL)
  296. MimeTableTail = &MimeTable;
  297.     while (fgets(buf, BUFSIZ, fp)) {
  298. if ((t = strchr(buf, '#')))
  299.     *t = '';
  300. if ((t = strchr(buf, 'r')))
  301.     *t = '';
  302. if ((t = strchr(buf, 'n')))
  303.     *t = '';
  304. if (buf[0] == '')
  305.     continue;
  306. xstrncpy(chopbuf, buf, BUFSIZ);
  307. if ((pattern = strtok(chopbuf, w_space)) == NULL) {
  308.     debug(25, 1) ("mimeInit: parse error: '%s'n", buf);
  309.     continue;
  310. }
  311. if ((type = strtok(NULL, w_space)) == NULL) {
  312.     debug(25, 1) ("mimeInit: parse error: '%s'n", buf);
  313.     continue;
  314. }
  315. if ((icon = strtok(NULL, w_space)) == NULL) {
  316.     debug(25, 1) ("mimeInit: parse error: '%s'n", buf);
  317.     continue;
  318. }
  319. if ((encoding = strtok(NULL, w_space)) == NULL) {
  320.     debug(25, 1) ("mimeInit: parse error: '%s'n", buf);
  321.     continue;
  322. }
  323. if ((mode = strtok(NULL, w_space)) == NULL) {
  324.     debug(25, 1) ("mimeInit: parse error: '%s'n", buf);
  325.     continue;
  326. }
  327. download_option = 0;
  328. view_option = 0;
  329. while ((option = strtok(NULL, w_space)) != NULL) {
  330.     if (!strcmp(option, "+download"))
  331. download_option = 1;
  332.     else if (!strcmp(option, "+view"))
  333. view_option = 1;
  334.     else
  335. debug(25, 1) ("mimeInit: unknown option: '%s' (%s)n", buf, option);
  336. }
  337. if (regcomp(&re, pattern, re_flags) != 0) {
  338.     debug(25, 1) ("mimeInit: regcomp error: '%s'n", buf);
  339.     continue;
  340. }
  341. m = xcalloc(1, sizeof(mimeEntry));
  342. m->pattern = xstrdup(pattern);
  343. m->content_type = xstrdup(type);
  344. m->icon = xstrdup(icon);
  345. m->content_encoding = xstrdup(encoding);
  346. m->compiled_pattern = re;
  347. if (!strcasecmp(mode, "ascii"))
  348.     m->transfer_mode = 'A';
  349. else if (!strcasecmp(mode, "text"))
  350.     m->transfer_mode = 'A';
  351. else
  352.     m->transfer_mode = 'I';
  353. m->view_option = view_option;
  354. m->download_option = download_option;
  355. *MimeTableTail = m;
  356. MimeTableTail = &m->next;
  357. debug(25, 5) ("mimeInit: added '%s'n", buf);
  358.     }
  359.     fclose(fp);
  360.     /*
  361.      * Create Icon StoreEntry's
  362.      */
  363.     for (m = MimeTable; m != NULL; m = m->next)
  364. mimeLoadIconFile(m->icon);
  365.     debug(25, 1) ("Loaded Icons.n");
  366. }
  368. void
  369. mimeFreeMemory(void)
  370. {
  371.     mimeEntry *m;
  372.     while ((m = MimeTable)) {
  373. MimeTable = m->next;
  374. safe_free(m->pattern);
  375. safe_free(m->content_type);
  376. safe_free(m->icon);
  377. safe_free(m->content_encoding);
  378. regfree(&m->compiled_pattern);
  379. safe_free(m);
  380.     }
  381.     MimeTableTail = &MimeTable;
  382. }
  384. static void
  385. mimeLoadIconFile(const char *icon)
  386. {
  387.     int fd;
  388.     int n;
  389.     request_flags flags;
  390.     struct stat sb;
  391.     StoreEntry *e;
  392.     LOCAL_ARRAY(char, path, MAXPATHLEN);
  393.     LOCAL_ARRAY(char, url, MAX_URL);
  394.     char *buf;
  395.     const char *type = mimeGetContentType(icon);
  396.     HttpReply *reply;
  397.     if (type == NULL)
  398. fatal("Unknown icon format while reading mime.confn");
  399.     buf = internalLocalUri("/squid-internal-static/icons/", icon);
  400.     xstrncpy(url, buf, MAX_URL);
  401.     if (storeGetPublic(url, METHOD_GET))
  402. return;
  403.     snprintf(path, MAXPATHLEN, "%s/%s", Config.icons.directory, icon);
  404.     fd = file_open(path, O_RDONLY, NULL, NULL, NULL);
  405.     if (fd < 0) {
  406. debug(25, 0) ("mimeLoadIconFile: %s: %sn", path, xstrerror());
  407. return;
  408.     }
  409.     if (fstat(fd, &sb) < 0) {
  410. debug(50, 0) ("mimeLoadIconFile: FD %d: fstat: %sn", fd, xstrerror());
  411. return;
  412.     }
  413.     flags = null_request_flags;
  414.     flags.cachable = 1;
  415.     e = storeCreateEntry(url,
  416. url,
  417. flags,
  418. METHOD_GET);
  419.     assert(e != NULL);
  420.     storeSetPublicKey(e);
  421.     e->mem_obj->request = requestLink(urlParse(METHOD_GET, url));
  422.     httpReplyReset(reply = e->mem_obj->reply);
  423.     httpReplySetHeaders(reply, 1.0, HTTP_OK, NULL,
  424. type, (int) sb.st_size, sb.st_mtime, -1);
  425.     reply->cache_control = httpHdrCcCreate();
  426.     httpHdrCcSetMaxAge(reply->cache_control, 86400);
  427.     httpHeaderPutCc(&reply->header, reply->cache_control);
  428.     httpReplySwapOut(reply, e);
  429.     /* read the file into the buffer and append it to store */
  430.     buf = memAllocate(MEM_4K_BUF);
  431.     while ((n = read(fd, buf, 4096)) > 0)
  432. storeAppend(e, buf, n);
  433.     file_close(fd);
  434.     storeComplete(e);
  435.     storeTimestampsSet(e);
  436.     EBIT_SET(e->flags, ENTRY_SPECIAL);
  437.     debug(25, 3) ("Loaded icon %sn", url);
  438.     storeUnlockObject(e);
  439.     memFree(buf, MEM_4K_BUF);
  440. }