1 /* Copyright (c) 2013-2018 Dovecot authors, see the included COPYING file */
2
3 #include "lib.h"
4 #include "net.h"
5 #include "istream.h"
6 #include "message-parser.h"
7 #include "mail-storage.h"
8 #include "mail-namespace.h"
9 #include "imap-url.h"
10 #include "imap-msgpart.h"
11 #include "imap-msgpart-url.h"
12
13 struct imap_msgpart_url {
14 char *mailbox;
15 uint32_t uidvalidity;
16 uint32_t uid;
17 char *section;
18 uoff_t partial_offset, partial_size;
19
20 struct imap_msgpart *part;
21
22 struct mail_user *user;
23 struct mailbox *selected_box;
24 struct mailbox *box;
25 struct mailbox_transaction_context *trans;
26 struct mail *mail;
27
28 struct imap_msgpart_open_result result;
29
30 bool decode_cte_to_binary:1;
31 };
32
imap_msgpart_url_create(struct mail_user * user,const struct imap_url * url,struct imap_msgpart_url ** mpurl_r,const char ** client_error_r)33 int imap_msgpart_url_create(struct mail_user *user, const struct imap_url *url,
34 struct imap_msgpart_url **mpurl_r,
35 const char **client_error_r)
36 {
37 const char *section = url->section == NULL ? "" : url->section;
38 struct imap_msgpart_url *mpurl;
39 struct imap_msgpart *msgpart;
40
41 if (url->mailbox == NULL || url->uid == 0 ||
42 url->search_program != NULL) {
43 *client_error_r = "Invalid messagepart IMAP URL";
44 return -1;
45 }
46 if (imap_msgpart_parse(section, &msgpart) < 0) {
47 *client_error_r = "Invalid section";
48 return -1;
49 }
50
51 mpurl = i_new(struct imap_msgpart_url, 1);
52 mpurl->part = msgpart;
53 mpurl->user = user;
54 mpurl->mailbox = i_strdup(url->mailbox);
55 mpurl->uidvalidity = url->uidvalidity;
56 mpurl->uid = url->uid;
57 if (url->section != NULL)
58 mpurl->section = i_strdup(url->section);
59 mpurl->partial_offset = url->partial_offset;
60 mpurl->partial_size = url->partial_size;
61
62 imap_msgpart_set_partial(msgpart, url->partial_offset,
63 url->partial_size == 0 ?
64 UOFF_T_MAX : url->partial_size);
65
66 *mpurl_r = mpurl;
67 return 0;
68 }
69
imap_msgpart_url_parse(struct mail_user * user,struct mailbox * selected_box,const char * urlstr,struct imap_msgpart_url ** mpurl_r,const char ** client_error_r)70 int imap_msgpart_url_parse(struct mail_user *user, struct mailbox *selected_box,
71 const char *urlstr, struct imap_msgpart_url **mpurl_r,
72 const char **client_error_r)
73 {
74 struct mailbox_status box_status;
75 struct imap_url base_url, *url;
76 const char *error;
77
78 /* build base url */
79 i_zero(&base_url);
80 if (selected_box != NULL) {
81 mailbox_get_open_status(selected_box, STATUS_UIDVALIDITY,
82 &box_status);
83 base_url.mailbox = mailbox_get_vname(selected_box);
84 base_url.uidvalidity = box_status.uidvalidity;
85 }
86
87 /* parse url */
88 if (imap_url_parse(urlstr, &base_url,
89 IMAP_URL_PARSE_REQUIRE_RELATIVE, &url, &error) < 0) {
90 *client_error_r = t_strconcat("Invalid IMAP URL: ", error, NULL);
91 return 0;
92 }
93 if (url->mailbox == NULL) {
94 *client_error_r = "Mailbox-relative IMAP URL, but no mailbox selected";
95 return 0;
96 }
97 if (imap_msgpart_url_create(user, url, mpurl_r, client_error_r) < 0)
98 return -1;
99 (*mpurl_r)->selected_box = selected_box;
100 return 1;
101 }
102
imap_msgpart_url_get_mailbox(struct imap_msgpart_url * mpurl)103 struct mailbox *imap_msgpart_url_get_mailbox(struct imap_msgpart_url *mpurl)
104 {
105 return mpurl->box;
106 }
107
imap_msgpart_url_open_mailbox(struct imap_msgpart_url * mpurl,struct mailbox ** box_r,enum mail_error * error_code_r,const char ** client_error_r)108 int imap_msgpart_url_open_mailbox(struct imap_msgpart_url *mpurl,
109 struct mailbox **box_r, enum mail_error *error_code_r,
110 const char **client_error_r)
111 {
112 struct mailbox_status box_status;
113 enum mailbox_flags flags = MAILBOX_FLAG_READONLY;
114 struct mail_namespace *ns;
115 struct mailbox *box;
116
117 if (mpurl->box != NULL) {
118 *box_r = mpurl->box;
119 *error_code_r = MAIL_ERROR_NONE;
120 return 1;
121 }
122
123 /* find mailbox namespace */
124 ns = mail_namespace_find(mpurl->user->namespaces, mpurl->mailbox);
125
126 /* open mailbox */
127 if (mpurl->selected_box != NULL &&
128 mailbox_equals(mpurl->selected_box, ns, mpurl->mailbox))
129 box = mpurl->selected_box;
130 else {
131 box = mailbox_alloc(ns->list, mpurl->mailbox, flags);
132 mailbox_set_reason(box, "Read IMAP URL");
133 }
134 if (mailbox_open(box) < 0) {
135 *client_error_r = mail_storage_get_last_error(mailbox_get_storage(box),
136 error_code_r);
137 if (box != mpurl->selected_box)
138 mailbox_free(&box);
139 return *error_code_r == MAIL_ERROR_TEMP ? -1 : 0;
140 }
141
142 /* verify UIDVALIDITY */
143 mailbox_get_open_status(box, STATUS_UIDVALIDITY, &box_status);
144 if (mpurl->uidvalidity > 0 &&
145 box_status.uidvalidity != mpurl->uidvalidity) {
146 *client_error_r = "Invalid UIDVALIDITY";
147 *error_code_r = MAIL_ERROR_EXPUNGED;
148 if (box != mpurl->selected_box)
149 mailbox_free(&box);
150 return 0;
151 }
152 mpurl->box = box;
153 *box_r = box;
154 return 1;
155 }
156
imap_msgpart_url_open_mail(struct imap_msgpart_url * mpurl,struct mail ** mail_r,const char ** client_error_r)157 int imap_msgpart_url_open_mail(struct imap_msgpart_url *mpurl,
158 struct mail **mail_r,
159 const char **client_error_r)
160 {
161 struct mailbox_transaction_context *t;
162 struct mailbox *box;
163 enum mail_error error_code;
164 struct mail *mail;
165 int ret;
166
167 if (mpurl->mail != NULL) {
168 *mail_r = mpurl->mail;
169 return 1;
170 }
171
172 /* open mailbox if it is not yet open */
173 if ((ret = imap_msgpart_url_open_mailbox(mpurl, &box, &error_code,
174 client_error_r)) <= 0)
175 return ret;
176
177 /* start transaction */
178 t = mailbox_transaction_begin(box, 0, __func__);
179 mail = mail_alloc(t, MAIL_FETCH_MESSAGE_PARTS |
180 MAIL_FETCH_IMAP_BODYSTRUCTURE, NULL);
181
182 /* find the message */
183 if (!mail_set_uid(mail, mpurl->uid)) {
184 *client_error_r = "Message not found";
185 mail_free(&mail);
186 mailbox_transaction_rollback(&t);
187 return 0;
188 }
189
190 mpurl->trans = t;
191 mpurl->mail = mail;
192 *mail_r = mail;
193 return 1;
194 }
195
196 struct imap_msgpart *
imap_msgpart_url_get_part(struct imap_msgpart_url * mpurl)197 imap_msgpart_url_get_part(struct imap_msgpart_url *mpurl)
198 {
199 return mpurl->part;
200 }
201
imap_msgpart_url_set_decode_to_binary(struct imap_msgpart_url * mpurl)202 void imap_msgpart_url_set_decode_to_binary(struct imap_msgpart_url *mpurl)
203 {
204 imap_msgpart_set_decode_to_binary(mpurl->part);
205 }
206
imap_msgpart_url_read_part(struct imap_msgpart_url * mpurl,struct imap_msgpart_open_result * result_r,const char ** client_error_r)207 int imap_msgpart_url_read_part(struct imap_msgpart_url *mpurl,
208 struct imap_msgpart_open_result *result_r,
209 const char **client_error_r)
210 {
211 struct mail *mail;
212 int ret;
213
214 if (mpurl->result.input != NULL) {
215 i_stream_seek(mpurl->result.input, 0);
216 *result_r = mpurl->result;
217 return 1;
218 }
219
220 /* open mail if it is not yet open */
221 ret = imap_msgpart_url_open_mail(mpurl, &mail, client_error_r);
222 if (ret <= 0)
223 return ret;
224
225 /* open the referenced part as a stream */
226 ret = imap_msgpart_open(mail, mpurl->part, result_r);
227 if (ret < 0) {
228 *client_error_r = mailbox_get_last_error(mpurl->box, NULL);
229 return ret;
230 }
231
232 mpurl->result = *result_r;
233 return 1;
234 }
235
imap_msgpart_url_verify(struct imap_msgpart_url * mpurl,const char ** client_error_r)236 int imap_msgpart_url_verify(struct imap_msgpart_url *mpurl,
237 const char **client_error_r)
238 {
239 struct mail *mail;
240 int ret;
241
242 if (mpurl->result.input != NULL)
243 return 1;
244
245 /* open mail if it is not yet open */
246 ret = imap_msgpart_url_open_mail(mpurl, &mail, client_error_r);
247 return ret;
248 }
249
imap_msgpart_url_get_bodypartstructure(struct imap_msgpart_url * mpurl,const char ** bpstruct_r,const char ** client_error_r)250 int imap_msgpart_url_get_bodypartstructure(struct imap_msgpart_url *mpurl,
251 const char **bpstruct_r,
252 const char **client_error_r)
253 {
254 struct mail *mail;
255 int ret;
256
257 /* open mail if it is not yet open */
258 ret = imap_msgpart_url_open_mail(mpurl, &mail, client_error_r);
259 if (ret <= 0)
260 return ret;
261
262 ret = imap_msgpart_bodypartstructure(mail, mpurl->part, bpstruct_r);
263 if (ret < 0)
264 *client_error_r = mailbox_get_last_error(mpurl->box, NULL);
265 else if (ret == 0)
266 *client_error_r = "Message part not found";
267 return ret;
268 }
269
imap_msgpart_url_free(struct imap_msgpart_url ** _mpurl)270 void imap_msgpart_url_free(struct imap_msgpart_url **_mpurl)
271 {
272 struct imap_msgpart_url *mpurl = *_mpurl;
273
274 *_mpurl = NULL;
275
276 i_stream_unref(&mpurl->result.input);
277 if (mpurl->part != NULL)
278 imap_msgpart_free(&mpurl->part);
279 if (mpurl->mail != NULL)
280 mail_free(&mpurl->mail);
281 if (mpurl->trans != NULL)
282 mailbox_transaction_rollback(&mpurl->trans);
283 if (mpurl->box != NULL && mpurl->box != mpurl->selected_box)
284 mailbox_free(&mpurl->box);
285 if (mpurl->section != NULL)
286 i_free(mpurl->section);
287 i_free(mpurl->mailbox);
288 i_free(mpurl);
289 }
290