move url.[hc] into the lib-lib.
[apps/madmutt.git] / imap / utf7.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 2000 Edmund Grimley Evans <edmundo@rano.org>
4  *
5  * This file is part of mutt-ng, see http://www.muttng.org/.
6  * It's licensed under the GNU General Public License,
7  * please see the file GPL in the top level source directory.
8  */
9
10 #if HAVE_CONFIG_H
11 # include "config.h"
12 #endif
13
14 #include <lib-lib/mem.h>
15
16 #include "mutt.h"
17 #include "charset.h"
18 #include "imap_private.h"
19
20 /*
21  * Convert the data (u7,u7len) from RFC 2060's UTF-7 to UTF-8.
22  * The result is null-terminated and returned, and also stored
23  * in (*u8,*u8len) if u8 or u8len is non-zero.
24  * If input data is invalid, return 0 and don't store anything.
25  * RFC 2060 obviously intends the encoding to be unique (see
26  * point 5 in section 5.1.3), so we reject any non-canonical
27  * form, such as &ACY- (instead of &-) or &AMA-&AMA- (instead
28  * of &AMAAwA-).
29  */
30 static char *utf7_to_utf8 (const char *u7, size_t u7len, char **u8,
31                            size_t * u8len)
32 {
33   char *buf, *p;
34   int b, ch, k;
35
36   p = buf = p_new(char, u7len + u7len / 8 + 1);
37
38   for (; u7len; u7++, u7len--) {
39     if (*u7 == '&') {
40       u7++, u7len--;
41
42       if (u7len && *u7 == '-') {
43         *p++ = '&';
44         continue;
45       }
46
47       ch = 0;
48       k = 10;
49       for (; u7len; u7++, u7len--) {
50         if ((b = base64val(*u7)) < 0)
51           break;
52         if (k > 0) {
53           ch |= b << k;
54           k -= 6;
55         }
56         else {
57           ch |= b >> (-k);
58           if (ch < 0x80) {
59             if (0x20 <= ch && ch < 0x7f)
60               /* Printable US-ASCII */
61               goto bail;
62             *p++ = ch;
63           }
64           else if (ch < 0x800) {
65             *p++ = 0xc0 | (ch >> 6);
66             *p++ = 0x80 | (ch & 0x3f);
67           }
68           else {
69             *p++ = 0xe0 | (ch >> 12);
70             *p++ = 0x80 | ((ch >> 6) & 0x3f);
71             *p++ = 0x80 | (ch & 0x3f);
72           }
73           ch = (b << (16 + k)) & 0xffff;
74           k += 10;
75         }
76       }
77       if (ch || k < 6)
78         /* Non-zero or too many extra bits */
79         goto bail;
80       if (!u7len || *u7 != '-')
81         /* BASE64 not properly terminated */
82         goto bail;
83       if (u7len > 2 && u7[1] == '&' && u7[2] != '-')
84         /* Adjacent BASE64 sections */
85         goto bail;
86     }
87     else if (*u7 < 0x20 || *u7 >= 0x7f)
88       /* Not printable US-ASCII */
89       goto bail;
90     else
91       *p++ = *u7;
92   }
93   *p++ = '\0';
94   if (u8len)
95     *u8len = p - buf;
96
97   p_realloc(&buf, p - buf);
98   if (u8)
99     *u8 = buf;
100   return buf;
101
102 bail:
103   p_delete(&buf);
104   return 0;
105 }
106
107 /*
108  * Convert the data (u8,u8len) from UTF-8 to RFC 2060's UTF-7.
109  * The result is null-terminated and returned, and also stored
110  * in (*u7,*u7len) if u7 or u7len is non-zero.
111  * Unicode characters above U+FFFF are replaced by U+FFFE.
112  * If input data is invalid, return 0 and don't store anything.
113  */
114 static char *utf8_to_utf7 (const char *u8, size_t u8len, char **u7,
115                            size_t * u7len)
116 {
117   char *buf, *p;
118   int ch;
119   int n, i, b = 0, k = 0;
120   int base64 = 0;
121
122   /*
123    * In the worst case we convert 2 chars to 7 chars. For example:
124    * "\x10&\x10&..." -> "&ABA-&-&ABA-&-...".
125    */
126   p = buf = p_new(char, (u8len / 2) * 7 + 6);
127
128   while (u8len) {
129     unsigned char c = *u8;
130
131     if (c < 0x80)
132       ch = c, n = 0;
133     else if (c < 0xc2)
134       goto bail;
135     else if (c < 0xe0)
136       ch = c & 0x1f, n = 1;
137     else if (c < 0xf0)
138       ch = c & 0x0f, n = 2;
139     else if (c < 0xf8)
140       ch = c & 0x07, n = 3;
141     else if (c < 0xfc)
142       ch = c & 0x03, n = 4;
143     else if (c < 0xfe)
144       ch = c & 0x01, n = 5;
145     else
146       goto bail;
147
148     u8++, u8len--;
149     if (n > u8len)
150       goto bail;
151     for (i = 0; i < n; i++) {
152       if ((u8[i] & 0xc0) != 0x80)
153         goto bail;
154       ch = (ch << 6) | (u8[i] & 0x3f);
155     }
156     if (n > 1 && !(ch >> (n * 5 + 1)))
157       goto bail;
158     u8 += n, u8len -= n;
159
160     if (ch < 0x20 || ch >= 0x7f) {
161       if (!base64) {
162         *p++ = '&';
163         base64 = 1;
164         b = 0;
165         k = 10;
166       }
167       if (ch & ~0xffff)
168         ch = 0xfffe;
169       *p++ = __m_b64chars[b | ch >> k];
170       k -= 6;
171       for (; k >= 0; k -= 6)
172         *p++ = __m_b64chars[(ch >> k) & 0x3f];
173       b = (ch << (-k)) & 0x3f;
174       k += 16;
175     }
176     else {
177       if (base64) {
178         if (k > 10)
179           *p++ = __m_b64chars[b];
180         *p++ = '-';
181         base64 = 0;
182       }
183       *p++ = ch;
184       if (ch == '&')
185         *p++ = '-';
186     }
187   }
188
189   if (u8len) {
190     p_delete(&buf);
191     return 0;
192   }
193
194   if (base64) {
195     if (k > 10)
196       *p++ = __m_b64chars[b];
197     *p++ = '-';
198   }
199
200   *p++ = '\0';
201   if (u7len)
202     *u7len = p - buf;
203   p_realloc(&buf, p - buf);
204   if (u7)
205     *u7 = buf;
206   return buf;
207
208 bail:
209   p_delete(&buf);
210   return 0;
211 }
212
213 void imap_utf7_encode (char **s)
214 {
215   if (Charset) {
216     char *t = m_strdup(*s);
217
218     if (!mutt_convert_string (&t, Charset, "UTF-8", 0)) {
219       char *u7 = utf8_to_utf7 (t, strlen (t), NULL, 0);
220       p_delete(s);
221       *s = u7;
222     }
223     p_delete(&t);
224   }
225 }
226
227 void imap_utf7_decode (char **s)
228 {
229   if (Charset) {
230     char *t = utf7_to_utf8 (*s, m_strlen(*s), 0, 0);
231
232     if (t && !mutt_convert_string (&t, "UTF-8", Charset, 0)) {
233       p_delete(s);
234       *s = t;
235     }
236   }
237 }