[disorder] / lib / mime.c

/*
 * This file is part of DisOrder
 * Copyright (C) 2005 Richard Kettlewell
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */


#include <config.h>
#include "types.h"

#include <string.h>
#include <ctype.h>

#include "mem.h"
#include "mime.h"
#include "vector.h"
#include "hex.h"

static int whitespace(int c) {
  switch(c) {
  case ' ':
  case '\t':
  case '\r':
  case '\n':
    return 1;
  default:
    return 0;
  }
}

static int tspecial(int c) {
  switch(c) {
  case '(':
  case ')':
  case '<':
  case '>':
  case '@':
  case ',':
  case ';':
  case ':':
  case '\\':
  case '"':
  case '/':
  case '[':
  case ']':
  case '?':
  case '=':
    return 1;
  default:
    return 0;
  }
}

static const char *skipwhite(const char *s) {
  int c, depth;
  
  for(;;) {
    switch(c = *s) {
    case ' ':
    case '\t':
    case '\r':
    case '\n':
      ++s;
      break;
    case '(':
      ++s;
      depth = 1;
      while(*s && depth) {
	c = *s++;
	switch(c) {
	case '(': ++depth; break;
	case ')': --depth; break;
	case '\\':
	  if(!*s) return 0;
	  ++s;
	  break;
	}
      }
      if(depth) return 0;
      break;
    default:
      return s;
    }
  }
}

static const char *parsestring(const char *s, char **valuep) {
  struct dynstr value;
  int c;

  dynstr_init(&value);
  ++s;
  while((c = *s++) != '"') {
    switch(c) {
    case '\\':
      if(!(c = *s++)) return 0;
    default:
      dynstr_append(&value, c);
      break;
    }
  }
  if(!c) return 0;
  dynstr_terminate(&value);
  *valuep = value.vec;
  return s;
}

int mime_content_type(const char *s,
		      char **typep,
		      char **parameternamep,
		      char **parametervaluep) {
  struct dynstr type, parametername, parametervalue;

  dynstr_init(&type);
  if(!(s = skipwhite(s))) return -1;
  if(!*s) return -1;
  while(*s && !tspecial(*s) && !whitespace(*s))
    dynstr_append(&type, tolower((unsigned char)*s++));
  if(!(s = skipwhite(s))) return -1;
  if(*s++ != '/') return -1;
  dynstr_append(&type, '/');
  if(!(s = skipwhite(s))) return -1;
  while(*s && !tspecial(*s) && !whitespace(*s))
    dynstr_append(&type, tolower((unsigned char)*s++));
  if(!(s = skipwhite(s))) return -1;

  if(*s == ';') {
    dynstr_init(&parametername);
    ++s;
    if(!(s = skipwhite(s))) return -1;
    if(!*s) return -1;
    while(*s && !tspecial(*s) && !whitespace(*s))
      dynstr_append(&parametername, tolower((unsigned char)*s++));
    if(!(s = skipwhite(s))) return -1;
    if(*s++ != '=') return -1;
    if(!(s = skipwhite(s))) return -1;
    if(*s == '"') {
      if(!(s = parsestring(s, parametervaluep))) return -1;
    } else {
      dynstr_init(&parametervalue);
      while(*s && !tspecial(*s) && !whitespace(*s))
	dynstr_append(&parametervalue, *s++);
      dynstr_terminate(&parametervalue);
      *parametervaluep = parametervalue.vec;
    }
    if(!(s = skipwhite(s))) return -1;
    dynstr_terminate(&parametername);
    *parameternamep = parametername.vec;
  } else
    *parametervaluep = *parameternamep = 0;
  dynstr_terminate(&type);
  *typep = type.vec;
  return 0;
}

static int iscrlf(const char *ptr) {
  return ptr[0] == '\r' && ptr[1] == '\n';
}

const char *mime_parse(const char *s,
		       int (*callback)(const char *name, const char *value,
				       void *u),
		       void *u) {
  struct dynstr name, value;
  char *cte = 0, *p;
  
  while(*s && !iscrlf(s)) {
    dynstr_init(&name);
    dynstr_init(&value);
    while(*s && !tspecial(*s) && !whitespace(*s))
      dynstr_append(&name, tolower((unsigned char)*s++));
    if(!(s = skipwhite(s))) return 0;
    if(*s != ':') return 0;
    ++s;
    while(*s && !(*s == '\n' && !(s[1] == ' ' || s[1] == '\t')))
      dynstr_append(&value, *s++);
    if(*s) ++s;
    dynstr_terminate(&name);
    dynstr_terminate(&value);
    if(!strcmp(name.vec, "content-transfer-encoding")) {
      cte = xstrdup(value.vec);
      for(p = cte; *p; p++)
	*p = tolower((unsigned char)*p);
    }
    if(callback(name.vec, value.vec, u)) return 0;
  }
  if(*s) s += 2;
  if(cte) {
    if(!strcmp(cte, "base64")) return mime_base64(s);
    if(!strcmp(cte, "quoted-printable")) return mime_qp(s);
  }
  return s;
}

static int isboundary(const char *ptr, const char *boundary, size_t bl) {
  return (ptr[0] == '-'
	  && ptr[1] == '-'
	  && !strncmp(ptr + 2, boundary, bl)
	  && (iscrlf(ptr + bl + 2)
	      || (ptr[bl + 2] == '-'
		  && ptr[bl + 3] == '-'
		  && iscrlf(ptr + bl + 4))));
}

static int isfinal(const char *ptr, const char *boundary, size_t bl) {
  return (ptr[0] == '-'
	  && ptr[1] == '-'
	  && !strncmp(ptr + 2, boundary, bl)
	  && ptr[bl + 2] == '-'
	  && ptr[bl + 3] == '-'
	  && iscrlf(ptr + bl + 4));
}

int mime_multipart(const char *s,
		   int (*callback)(const char *s, void *u),
		   const char *boundary,
		   void *u) {
  size_t bl = strlen(boundary);
  const char *start, *e;
  int ret;

  if(!isboundary(s, boundary, bl)) return -1;
  while(!isfinal(s, boundary, bl)) {
    s = strstr(s, "\r\n") + 2;
    start = s;
    while(!isboundary(s, boundary, bl)) {
      if(!(e = strstr(s, "\r\n"))) return -1;
      s = e + 2;
    }
    if((ret = callback(xstrndup(start,
				s == start ? 0 : s - start - 2),
		       u)))
      return ret;
  }
  return 0;
}

int mime_rfc2388_content_disposition(const char *s,
				     char **dispositionp,
				     char **parameternamep,
				     char **parametervaluep) {
  struct dynstr disposition, parametername, parametervalue;

  dynstr_init(&disposition);
  if(!(s = skipwhite(s))) return -1;
  if(!*s) return -1;
  while(*s && !tspecial(*s) && !whitespace(*s))
    dynstr_append(&disposition, tolower((unsigned char)*s++));
  if(!(s = skipwhite(s))) return -1;

  if(*s == ';') {
    dynstr_init(&parametername);
    ++s;
    if(!(s = skipwhite(s))) return -1;
    if(!*s) return -1;
    while(*s && !tspecial(*s) && !whitespace(*s))
      dynstr_append(&parametername, tolower((unsigned char)*s++));
    if(!(s = skipwhite(s))) return -1;
    if(*s++ != '=') return -1;
    if(!(s = skipwhite(s))) return -1;
    if(*s == '"') {
      if(!(s = parsestring(s, parametervaluep))) return -1;
    } else {
      dynstr_init(&parametervalue);
      while(*s && !tspecial(*s) && !whitespace(*s))
	dynstr_append(&parametervalue, *s++);
      dynstr_terminate(&parametervalue);
      *parametervaluep = parametervalue.vec;
    }
    if(!(s = skipwhite(s))) return -1;
    dynstr_terminate(&parametername);
    *parameternamep = parametername.vec;
  } else
    *parametervaluep = *parameternamep = 0;
  dynstr_terminate(&disposition);
  *dispositionp = disposition.vec;
  return 0;
}

char *mime_qp(const char *s) {
  struct dynstr d;
  int c, a, b;
  const char *t;

  dynstr_init(&d);
  while((c = *s++)) {
    switch(c) {
    case '=':
      if((a = unhexdigitq(s[0])) != -1
	 && (b = unhexdigitq(s[1])) != -1) {
	dynstr_append(&d, a * 16 + b);
	s += 2;
      } else {
	t = s;
	while(*t == ' ' || *t == '\t') ++t;
	if(iscrlf(t)) {
	  /* soft line break */
	  s = t + 2;
	} else
	  return 0;
      }
      break;
    case ' ':
    case '\t':
      t = s;
      while(*t == ' ' || *t == '\t') ++t;
      if(iscrlf(t))
	/* trailing space is always eliminated */
	s = t;
      else
	dynstr_append(&d, c);
      break;
    default:
      dynstr_append(&d, c);
      break;
    }
  }
  dynstr_terminate(&d);
  return d.vec;
}

char *mime_base64(const char *s) {
  struct dynstr d;
  const char *t;
  int b[4], n, c;
  static const char table[] =
    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";

  dynstr_init(&d);
  n = 0;
  while((c = (unsigned char)*s++)) {
    if((t = strchr(table, c))) {
      b[n++] = t - table;
      if(n == 4) {
	dynstr_append(&d, (b[0] << 2) + (b[1] >> 4));
	dynstr_append(&d, (b[1] << 4) + (b[2] >> 2));
	dynstr_append(&d, (b[2] << 6) + b[3]);
	n = 0;
      }
    } else if(c == '=') {
      if(n >= 2) {
	dynstr_append(&d, (b[0] << 2) + (b[1] >> 4));
	if(n == 3)
	  dynstr_append(&d, (b[1] << 4) + (b[2] >> 2));
      }
      break;
    }
  }
  dynstr_terminate(&d);
  return d.vec;
}

/*
Local Variables:
c-basic-offset:2
comment-column:40
fill-column:79
End:
*/
Commit	Line	Data
460b9539	1	/*
	2	* This file is part of DisOrder
	3	* Copyright (C) 2005 Richard Kettlewell
	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License as published by
	7	* the Free Software Foundation; either version 2 of the License, or
	8	* (at your option) any later version.
	9	*
	10	* This program is distributed in the hope that it will be useful, but
	11	* WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	13	* General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License
	16	* along with this program; if not, write to the Free Software
	17	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
	18	* USA
	19	*/
	20
	21
	22	#include <config.h>
	23	#include "types.h"
	24
	25	#include <string.h>
	26	#include <ctype.h>
	27
	28	#include "mem.h"
	29	#include "mime.h"
	30	#include "vector.h"
	31	#include "hex.h"
	32
	33	static int whitespace(int c) {
	34	switch(c) {
	35	case ' ':
	36	case '\t':
	37	case '\r':
	38	case '\n':
	39	return 1;
	40	default:
	41	return 0;
	42	}
	43	}
	44
	45	static int tspecial(int c) {
	46	switch(c) {
	47	case '(':
	48	case ')':
	49	case '<':
	50	case '>':
	51	case '@':
	52	case ',':
	53	case ';':
	54	case ':':
	55	case '\\':
	56	case '"':
	57	case '/':
	58	case '[':
	59	case ']':
	60	case '?':
	61	case '=':
	62	return 1;
	63	default:
	64	return 0;
65	}
66	}
67
68	static const char skipwhite(const char s) {
69	int c, depth;
70
71	for(;;) {
72	switch(c = *s) {
73	case ' ':
74	case '\t':
75	case '\r':
76	case '\n':
77	++s;
78	break;
79	case '(':
80	++s;
81	depth = 1;
82	while(*s && depth) {
83	c = *s++;
84	switch(c) {
85	case '(': ++depth; break;
86	case ')': --depth; break;
87	case '\\':
88	if(!*s) return 0;
89	++s;
90	break;
91	}
92	}
93	if(depth) return 0;
94	break;
95	default:
96	return s;
97	}
98	}
99	}
100
101	static const char parsestring(const char s, char **valuep) {
102	struct dynstr value;
103	int c;
104
105	dynstr_init(&value);
106	++s;
107	while((c = *s++) != '"') {
108	switch(c) {
109	case '\\':
110	if(!(c = *s++)) return 0;
111	default:
112	dynstr_append(&value, c);
113	break;
114	}
115	}
116	if(!c) return 0;
117	dynstr_terminate(&value);
118	*valuep = value.vec;
119	return s;
120	}
121
122	int mime_content_type(const char *s,
123	char **typep,
124	char **parameternamep,
125	char **parametervaluep) {
126	struct dynstr type, parametername, parametervalue;
127
128	dynstr_init(&type);
129	if(!(s = skipwhite(s))) return -1;
130	if(!*s) return -1;
131	while(s && !tspecial(s) && !whitespace(*s))
132	dynstr_append(&type, tolower((unsigned char)*s++));
133	if(!(s = skipwhite(s))) return -1;
134	if(*s++ != '/') return -1;
135	dynstr_append(&type, '/');
136	if(!(s = skipwhite(s))) return -1;
137	while(s && !tspecial(s) && !whitespace(*s))
138	dynstr_append(&type, tolower((unsigned char)*s++));
139	if(!(s = skipwhite(s))) return -1;
140
141	if(*s == ';') {
142	dynstr_init(&parametername);
143	++s;
144	if(!(s = skipwhite(s))) return -1;
145	if(!*s) return -1;
146	while(s && !tspecial(s) && !whitespace(*s))
147	dynstr_append(&parametername, tolower((unsigned char)*s++));
148	if(!(s = skipwhite(s))) return -1;
149	if(*s++ != '=') return -1;
150	if(!(s = skipwhite(s))) return -1;
151	if(*s == '"') {
152	if(!(s = parsestring(s, parametervaluep))) return -1;
153	} else {
154	dynstr_init(&parametervalue);
155	while(s && !tspecial(s) && !whitespace(*s))
156	dynstr_append(&parametervalue, *s++);
157	dynstr_terminate(&parametervalue);
158	*parametervaluep = parametervalue.vec;
159	}
160	if(!(s = skipwhite(s))) return -1;
161	dynstr_terminate(&parametername);
162	*parameternamep = parametername.vec;
163	} else
164	parametervaluep = parameternamep = 0;
165	dynstr_terminate(&type);
166	*typep = type.vec;
167	return 0;
168	}
169
170	static int iscrlf(const char *ptr) {
171	return ptr[0] == '\r' && ptr[1] == '\n';
172	}
173
174	const char mime_parse(const char s,
175	int (callback)(const char name, const char *value,
176	void *u),
177	void *u) {
178	struct dynstr name, value;
179	char cte = 0, p;
180
181	while(*s && !iscrlf(s)) {
182	dynstr_init(&name);
183	dynstr_init(&value);
184	while(s && !tspecial(s) && !whitespace(*s))
185	dynstr_append(&name, tolower((unsigned char)*s++));
186	if(!(s = skipwhite(s))) return 0;
187	if(*s != ':') return 0;
188	++s;
189	while(s && !(s == '\n' && !(s[1] == ' ' \|\| s[1] == '\t')))
190	dynstr_append(&value, *s++);
191	if(*s) ++s;
192	dynstr_terminate(&name);
193	dynstr_terminate(&value);
194	if(!strcmp(name.vec, "content-transfer-encoding")) {
195	cte = xstrdup(value.vec);
196	for(p = cte; *p; p++)
197	p = tolower((unsigned char)p);
198	}
199	if(callback(name.vec, value.vec, u)) return 0;
200	}
201	if(*s) s += 2;
202	if(cte) {
203	if(!strcmp(cte, "base64")) return mime_base64(s);
204	if(!strcmp(cte, "quoted-printable")) return mime_qp(s);
205	}
206	return s;
207	}
208
209	static int isboundary(const char ptr, const char boundary, size_t bl) {
210	return (ptr[0] == '-'
211	&& ptr[1] == '-'
212	&& !strncmp(ptr + 2, boundary, bl)
213	&& (iscrlf(ptr + bl + 2)
214	\|\| (ptr[bl + 2] == '-'
215	&& ptr[bl + 3] == '-'
216	&& iscrlf(ptr + bl + 4))));
217	}
218
219	static int isfinal(const char ptr, const char boundary, size_t bl) {
220	return (ptr[0] == '-'
221	&& ptr[1] == '-'
222	&& !strncmp(ptr + 2, boundary, bl)
223	&& ptr[bl + 2] == '-'
224	&& ptr[bl + 3] == '-'
225	&& iscrlf(ptr + bl + 4));
226	}
227
228	int mime_multipart(const char *s,
229	int (callback)(const char s, void *u),
230	const char *boundary,
231	void *u) {
232	size_t bl = strlen(boundary);
233	const char start, e;
234	int ret;
235
236	if(!isboundary(s, boundary, bl)) return -1;
237	while(!isfinal(s, boundary, bl)) {
238	s = strstr(s, "\r\n") + 2;
239	start = s;
240	while(!isboundary(s, boundary, bl)) {
241	if(!(e = strstr(s, "\r\n"))) return -1;
242	s = e + 2;
243	}
244	if((ret = callback(xstrndup(start,
245	s == start ? 0 : s - start - 2),
246	u)))
247	return ret;
248	}
249	return 0;
250	}
251
252	int mime_rfc2388_content_disposition(const char *s,
253	char **dispositionp,
254	char **parameternamep,
255	char **parametervaluep) {
256	struct dynstr disposition, parametername, parametervalue;
257
258	dynstr_init(&disposition);
259	if(!(s = skipwhite(s))) return -1;
260	if(!*s) return -1;
261	while(s && !tspecial(s) && !whitespace(*s))
262	dynstr_append(&disposition, tolower((unsigned char)*s++));
263	if(!(s = skipwhite(s))) return -1;
264
265	if(*s == ';') {
266	dynstr_init(&parametername);
267	++s;
268	if(!(s = skipwhite(s))) return -1;
269	if(!*s) return -1;
270	while(s && !tspecial(s) && !whitespace(*s))
271	dynstr_append(&parametername, tolower((unsigned char)*s++));
272	if(!(s = skipwhite(s))) return -1;
273	if(*s++ != '=') return -1;
274	if(!(s = skipwhite(s))) return -1;
275	if(*s == '"') {
276	if(!(s = parsestring(s, parametervaluep))) return -1;
277	} else {
278	dynstr_init(&parametervalue);
279	while(s && !tspecial(s) && !whitespace(*s))
280	dynstr_append(&parametervalue, *s++);
281	dynstr_terminate(&parametervalue);
282	*parametervaluep = parametervalue.vec;
283	}
284	if(!(s = skipwhite(s))) return -1;
285	dynstr_terminate(&parametername);
286	*parameternamep = parametername.vec;
287	} else
288	parametervaluep = parameternamep = 0;
289	dynstr_terminate(&disposition);
290	*dispositionp = disposition.vec;
291	return 0;
292	}
293
294	char mime_qp(const char s) {
295	struct dynstr d;
296	int c, a, b;
297	const char *t;
298
299	dynstr_init(&d);
300	while((c = *s++)) {
301	switch(c) {
302	case '=':
303	if((a = unhexdigitq(s[0])) != -1
304	&& (b = unhexdigitq(s[1])) != -1) {
305	dynstr_append(&d, a * 16 + b);
306	s += 2;
307	} else {
308	t = s;
309	while(t == ' ' \|\| t == '\t') ++t;
310	if(iscrlf(t)) {
311	/* soft line break */
312	s = t + 2;
313	} else
314	return 0;
315	}
316	break;
317	case ' ':
318	case '\t':
319	t = s;
320	while(t == ' ' \|\| t == '\t') ++t;
321	if(iscrlf(t))
322	/* trailing space is always eliminated */
323	s = t;
324	else
325	dynstr_append(&d, c);
326	break;
327	default:
328	dynstr_append(&d, c);
329	break;
330	}
331	}
332	dynstr_terminate(&d);
333	return d.vec;
334	}
335
336	char mime_base64(const char s) {
337	struct dynstr d;
338	const char *t;
339	int b[4], n, c;
340	static const char table[] =
341	"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
342
343	dynstr_init(&d);
344	n = 0;
345	while((c = (unsigned char)*s++)) {
346	if((t = strchr(table, c))) {
347	b[n++] = t - table;
348	if(n == 4) {
349	dynstr_append(&d, (b[0] << 2) + (b[1] >> 4));
350	dynstr_append(&d, (b[1] << 4) + (b[2] >> 2));
351	dynstr_append(&d, (b[2] << 6) + b[3]);
352	n = 0;
353	}
354	} else if(c == '=') {
355	if(n >= 2) {
356	dynstr_append(&d, (b[0] << 2) + (b[1] >> 4));
357	if(n == 3)
358	dynstr_append(&d, (b[1] << 4) + (b[2] >> 2));
359	}
360	break;
361	}
362	}
363	dynstr_terminate(&d);
364	return d.vec;
365	}
366
367	/*
368	Local Variables:
369	c-basic-offset:2
370	comment-column:40
371	fill-column:79
372	End:
373	*/