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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > WEB邮件程序 > 查看源码
rfc2045rewrite.c
资源名称：sqwebmail-0.37a.38pre2.tar.gz　[点击查看]
上传用户：s81996212
上传日期：2007-01-04
资源大小：722k
文件大小：9k
源码类别：

WEB邮件程序

开发平台：

C/C++

rfc2045rewrite.c：源码内容
  1. /*
  2. ** Copyright 1998 - 1999 Double Precision, Inc.  See COPYING for
  3. ** distribution information.
  4. */
  6. #if HAVE_CONFIG_H
  7. #include "config.h"
  8. #endif
  9. #include "rfc2045.h"
  10. #include "rfc2045charset.h"
  11. #if HAVE_UNISTD_H
  12. #include <unistd.h>
  13. #endif
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #include <ctype.h>
  17. #include       <string.h>
  18. #if    HAVE_STRINGS_H
  19. #include       <strings.h>
  20. #endif
  22. /* $Id: rfc2045rewrite.c,v 1.9 2000/01/31 04:49:39 mrsam Exp $ */
  24. static char *rw_boundary_root;
  25. static int rw_boundary_cnt;
  26. static const char *rw_appname;
  28. static FILE *fdin;
  29. static int fdout;
  30. static int (*fdout_func)(const char *, int, void *);
  31. static void *fdout_arg;
  33. static char fdout_buf[512];
  34. static char *fdout_ptr;
  35. static size_t fdout_left;
  36. extern void rfc2045_enomem();
  38. /* Quoted printable encoding */
  39. static void qpe_start();
  40. static int qpe_do(const char *, size_t, void *);
  41. static void qpe_end();
  42. static int conv_err;
  44. static int fdout_flush()
  45. {
  46. int n=fdout_ptr-fdout_buf;
  47. int i=0;
  48. char *p=fdout_buf;
  50. while (n)
  51. {
  52. i=fdout_func ? (*fdout_func)(p, n, fdout_arg):
  53. write(fdout, p, n);
  54. if (i <= 0) return (-1);
  55. p += i;
  56. n -= i;
  57. }
  58. fdout_ptr=fdout_buf;
  59. fdout_left=sizeof(fdout_buf);
  60. return (0);
  61. }
  63. static int fdout_add(const char *p, size_t cnt)
  64. {
  65. while (cnt)
  66. {
  67. if (cnt < fdout_left)
  68. {
  69. memcpy(fdout_ptr, p, cnt);
  70. fdout_ptr += cnt;
  71. fdout_left -= cnt;
  72. return (0);
  73. }
  74. if (fdout_left == 0)
  75. {
  76. if (fdout_flush()) return (-1);
  77. continue;
  78. }
  79. memcpy(fdout_ptr, p, fdout_left);
  80. p += fdout_left;
  81. cnt -= fdout_left;
  82. fdout_ptr += fdout_left;
  83. fdout_left=0;
  84. }
  85. return (0);
  86. }
  88. static int do_8bit(const char *p, size_t cnt, void *ptr)
  89. {
  90. if (fdout_add(p, cnt))
  91. conv_err=1;
  92. return (0);
  93. }
  95. static int fdout_autoconverted(const char *oldte, const char *newte)
  96. {
  97. if (fdout_add("X-Mime-Autoconverted: from ", 27) ||
  98. fdout_add(oldte, strlen(oldte)) ||
  99. fdout_add(" to ", 4) ||
  100. fdout_add(newte, strlen(newte)) ||
  101. (rw_appname && (fdout_add(" by ", 4) ||
  102. fdout_add(rw_appname, strlen(rw_appname)))) ||
  103. fdout_add("n", 1)) return (-1);
  104. return (0);
  105. }
  107. static int fdout_value(const char *);
  109. static int fdout_attr(const struct rfc2045attr *a)
  110. {
  111. if (fdout_add(a->name, strlen(a->name))) return (-1);
  112. if (a->value && (fdout_add("=", 1) || fdout_value(a->value)))
  113. return (-1);
  114. return (0);
  115. }
  117. static int fdout_value(const char *v)
  118. {
  119. size_t i,j;
  121. for (i=0; v[i]; i++)
  122. {
  123. if ( !isalnum((int)(unsigned char)v[i]) && v[i] != '-')
  124. {
  125. if (fdout_add(""", 1)) return (-1);
  126. for (j=i=0; v[i]; i++)
  127. if (v[i] == '\' || v[i] == '"')
  128. {
  129. if (fdout_add(v+j, i-j) ||
  130. fdout_add("\", 1))
  131. return (-1);
  132. j=i;
  133. }
  134. if (fdout_add(v+j, i-j) || fdout_add(""", 1))
  135. return (-1);
  136. return (0);
  137. }
  138. }
  139. return (fdout_add(v, i));
  140. }
  142. #define TE(p) ((p)->rw_transfer_encoding ? 
  143. (p)->rw_transfer_encoding: (p)->content_transfer_encoding)
  145. static int rwmime(struct rfc2045 *p)
  146. {
  147. static char mimever[]="Mime-Version: 1.0n";
  148. const char *te;
  149. struct rfc2045attr *a;
  151. if (!p->parent)
  152. if (fdout_add(mimever, sizeof(mimever)-1)) return (-1);
  154. if (p->content_type)
  155. {
  156. if (fdout_add("Content-Type: ", 14) ||
  157. fdout_add(p->content_type, strlen(p->content_type)))
  158. return (-1);
  160. for (a=p->content_type_attr; a; a=a->next)
  161. {
  162. if (!a->name || strcmp(a->name, "boundary") == 0)
  163. continue;
  164. if ( fdout_add("; ", 2) ||
  165. fdout_attr(a)) return (-1);
  166. }
  167. }
  169. if (p->firstpart
  170. && p->firstpart->next /* ADDED 8/30/99, see below */)
  171. {
  172. char buf[80];
  174. ++rw_boundary_cnt;
  175. sprintf(buf, "-%d", rw_boundary_cnt);
  176. if ( fdout_add("; boundary="", 12) ||
  177. fdout_add(rw_boundary_root, strlen(rw_boundary_root)) ||
  178. fdout_add(buf, strlen(buf)) ||
  179. fdout_add(""", 1)) return (-1);
  180. }
  181. if (fdout_add("n", 1)) return (-1);
  183. /* Show content transfer encoding for top section, or if it's
  184. ** different than the parent.
  185. */
  186. te=TE(p);
  187. if (te && (!p->parent || strcmp(te, TE(p->parent))))
  188. {
  189. if (fdout_add("Content-Transfer-Encoding: ", 27) ||
  190. fdout_add(te, strlen(te)) ||
  191. fdout_add("n", 1)) return (-1);
  192. }
  193. return (0);
  194. }
  196. static int dorw(struct rfc2045 *p)
  197. {
  198. /* WTF STATIC??? */ int seen_mime=0;
  199. char buf[256];
  200. int c;
  201. int bcnt;
  203. if (fseek(fdin, p->startpos, SEEK_SET) == -1) return (-1);
  204. if (p->parent)
  205. {
  206. seen_mime=1;
  207. if (rwmime(p)) return (-1);
  208. }
  209. while (fgets(buf, sizeof(buf), fdin))
  210. {
  211. if (buf[0] == 'n') break;
  213. if (RFC2045_ISMIME1DEF(p->mime_version) &&
  214. strncasecmp(buf, "mime-version:", 13) == 0 &&
  215. !seen_mime)
  216. {
  217. seen_mime=1;
  218. rwmime(p);
  219. if (strchr(buf, 'n') == NULL)
  220. while ((c=getc(fdin)) >= 0 && c != 'n')
  221. ;
  222. while ((c=getc(fdin)) >= 0 && c != 'n' && isspace(c))
  223. while ((c=getc(fdin)) >= 0 && c != 'n')
  224. ;
  225. if (c >= 0) ungetc(c, fdin);
  226. continue;
  227. }
  229. if (!RFC2045_ISMIME1DEF(p->mime_version) || (
  230. strncasecmp(buf, "mime-version:", 13) &&
  231. strncasecmp(buf, "content-type:", 13) &&
  232. strncasecmp(buf, "content-transfer-encoding:", 26))
  233. )
  234. {
  235. do
  236. {
  237. do
  238. {
  239. if (fdout_add(buf, strlen(buf)))
  240. return (-1);
  241. } while (strchr(buf, 'n') == NULL &&
  242. fgets(buf, sizeof(buf), fdin));
  244. c=getc(fdin);
  245. if (c >= 0) ungetc(c, fdin);
  246. } while (c >= 0 && c != 'n' && isspace(c) &&
  247.     fgets(buf, sizeof(buf), fdin));
  248. }
  249. else
  250. while ( (c=getc(fdin)) >= 0 &&
  251. (ungetc(c, fdin), c) != 'n' && isspace(c))
  252. {
  253. while (fgets(buf, sizeof(buf), fdin) &&
  254. strchr(buf, 'n') == NULL)
  255. ;
  256. }
  257. }
  258. if (RFC2045_ISMIME1DEF(p->mime_version))
  259. {
  260. if (!seen_mime)
  261. if (rwmime(p)) return (-1);
  263. if (!p->firstpart && p->rw_transfer_encoding)
  264. if (fdout_autoconverted(p->content_transfer_encoding,
  265. p->rw_transfer_encoding)) return (-1);
  266. }
  268. if (fdout_add("n", 1)) return (-1);
  269. if (fseek(fdin, p->startbody, SEEK_SET) == -1) return (-1);
  271. /* For non-multipart section, just print the body */
  273. if (!p->firstpart)
  274. {
  275. off_t ps=p->startbody;
  276. int convmode=0;
  278. if (p->rw_transfer_encoding)
  279. {
  280. if ( strcasecmp(p->rw_transfer_encoding,
  281. "quoted-printable") == 0)
  282. convmode=RFC2045_RW_7BIT;
  283. else
  284. convmode=RFC2045_RW_8BIT;
  285. }
  287. conv_err=0;
  288. if (convmode == RFC2045_RW_7BIT)
  289. {
  290. qpe_start();
  291. rfc2045_cdecode_start(p, &qpe_do, 0);
  292. }
  294. if (convmode == RFC2045_RW_8BIT)
  295. {
  296. rfc2045_cdecode_start(p, &do_8bit, 0);
  297. }
  299. while (ps < p->endbody)
  300. {
  301. int n;
  303. if (p->endbody - ps > sizeof(buf))
  304. n=sizeof(buf);
  305. else n=p->endbody-ps;
  306. n=fread(buf, 1, n, fdin);
  307. if (n <= 0) return (-1);
  308. if (convmode)
  309. rfc2045_cdecode(p, buf, n);
  310. else if (fdout_add(buf, n)) conv_err=1;
  311. ps += n;
  312. if (conv_err) break;
  313. }
  314. if (convmode == RFC2045_RW_7BIT)
  315. {
  316. rfc2045_cdecode_end(p);
  317. qpe_end();
  318. }
  319. if (convmode == RFC2045_RW_8BIT)
  320. {
  321. rfc2045_cdecode_end(p);
  322. }
  323. if (conv_err) return (-1);
  324. return (0);
  325. }
  327. bcnt=rw_boundary_cnt;
  329. /* Sam 8/30/99 fix - handle message/rfc822:
  331.             --boundary
  332.             Content-Type: message/rfc822
  334.          --><-- we're here, DON'T add RFC2045MIMEMSG and rest of crap here
  335. */
  336. if (p->firstpart->next == 0)
  337. {
  338. int rc;
  340. p->firstpart->parent=0;
  341. rc=dorw(p->firstpart);
  342. p->firstpart->parent=p;
  343. return (rc);
  344. }
  346. if (fdout_add(RFC2045MIMEMSG, sizeof(RFC2045MIMEMSG)-1))
  347. return (-1);
  348. for (p=p->firstpart; p; p=p->next)
  349. {
  350. if (p->isdummy) continue;
  351. sprintf(buf, "n--%s-%dn", rw_boundary_root, bcnt);
  352. if (fdout_add(buf, strlen(buf))) return (-1);
  353. if (dorw(p) != 0) return(-1);
  354. }
  355. sprintf(buf, "n--%s-%d--n", rw_boundary_root, bcnt);
  356. if (fdout_add(buf, strlen(buf))) return (-1);
  357. return (0);
  358. }
  360. static int rfc2045_rewrite_common(struct rfc2045 *, int, const char *);
  362. int rfc2045_rewrite(struct rfc2045 *p, int fdin_arg, int fdout_arg,
  363. const char *appname)
  364. {
  365. fdout=fdout_arg;
  366. fdout_func=0;
  367. return (rfc2045_rewrite_common(p, fdin_arg, appname));
  368. }
  370. int rfc2045_rewrite_func(struct rfc2045 *p, int fdin_arg,
  371. int (*funcarg)(const char *, int, void *), void *funcargarg,
  372. const char *appname)
  373. {
  374. fdout= -1;
  375. fdout_func=funcarg;
  376. fdout_arg=funcargarg;
  377. return (rfc2045_rewrite_common(p, fdin_arg, appname));
  378. }
  380. static int rfc2045_rewrite_common(struct rfc2045 *p,
  381. int fdin_arg, const char *appname)
  382. {
  383. int rc;
  384. int fd=dup(fdin_arg);
  386. if (fd < 0) return (-1);
  387. rw_appname=appname;
  388. fdin=fdopen(fd, "r");
  389. if (!fdin)
  390. {
  391. close(fd);
  392. return (-1);
  393. }
  395. fdout_ptr=fdout_buf;
  396. fdout_left=sizeof(fdout_buf);
  398. rw_boundary_root=rfc2045_mk_boundary(p, fd);
  399. if (rw_boundary_root == 0)
  400. rc= -1;
  401. else
  402. {
  403. rw_boundary_cnt=1;
  404. rc=dorw(p);
  405. free(rw_boundary_root);
  406. }
  407. if (rc == 0 && fdout_ptr > fdout_buf)
  408. rc=fdout_flush();
  409. fclose(fdin);
  410. return (rc);
  411. }
  413. static int qpe_pos;
  415. static void qpe_start()
  416. {
  417. qpe_pos=0;
  418. }
  420. static int qpe_do(const char *p, size_t i, void *ptr)
  421. {
  422. size_t j,k;
  424. if (conv_err) return (0);
  425. for (j=k=0; j<i; j++)
  426. {
  427. if (p[j] == 'n')
  428. {
  429. if (fdout_add(p+k, j+1-k)) conv_err=1;
  430. k=j+1;
  431. qpe_pos=0;
  432. continue;
  433. }
  434. if (qpe_pos >= 72)
  435. {
  436. if (fdout_add(p+k, j-k) ||
  437. fdout_add("=n", 2)) conv_err=1;
  438. k=j;
  439. qpe_pos=0;
  440. }
  442. if (p[j] < 32 || p[j] >= 127 || p[j] == '=')
  443. {
  444. char buf[3];
  445. static char xdigit[16]="0123456789ABCDEF";
  446. unsigned n= (unsigned char)p[j];
  448. buf[0]='=';
  449. buf[1]=xdigit[ n / 16];
  450. buf[2]=xdigit[ n % 16];
  451. if (fdout_add(p+k, j-k) ||
  452. fdout_add(buf, 3)) conv_err=1;
  453. qpe_pos += 2;
  454. k=j+1;
  455. }
  456. ++qpe_pos;
  457. }
  458. if (fdout_add(p+k, j-k)) conv_err=1;
  459. return (0);
  460. }
  462. static void qpe_end()
  463. {
  464. }