New NNTP authentication code.
[claws.git] / src / nntp.c
1 /*
2  * Sylpheed -- a GTK+ based, lightweight, and fast e-mail client
3  * Copyright (C) 1999,2000 Hiroyuki Yamamoto
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,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU 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, USA.
18  */
19
20 #ifdef HAVE_CONFIG_H
21 #  include "config.h"
22 #endif
23
24 #include <glib.h>
25 #include <stdio.h>
26 #include <string.h>
27
28 #include "intl.h"
29 #include "nntp.h"
30 #include "socket.h"
31 #include "utils.h"
32
33 static gint verbose = 1;
34
35 static void nntp_gen_send(NNTPSockInfo *sock, const gchar *format, ...);
36 static gint nntp_gen_recv(NNTPSockInfo *sock, gchar *buf, gint size);
37 static gint nntp_gen_command(NNTPSockInfo *sock, gchar *argbuf,
38                              const gchar *format, ...);
39
40 NNTPSockInfo *nntp_open(const gchar *server, gushort port, gchar *buf)
41 {
42         SockInfo *sock;
43         NNTPSockInfo *nntp_sock;
44
45         nntp_sock = g_new0(NNTPSockInfo, 1);
46         if ((sock = sock_connect(server, port)) == NULL) {
47                 log_warning(_("Can't connect to NNTP server: %s:%d\n"),
48                             server, port);
49                 g_free(nntp_sock);
50                 return NULL;
51         }
52         nntp_sock->sock = sock;
53
54         if (nntp_ok(nntp_sock, buf) == NN_SUCCESS)
55                 return nntp_sock;
56         else {
57                 sock_close(sock);
58                 g_free(nntp_sock);
59                 return NULL;
60         }
61 }
62
63 NNTPSockInfo *nntp_open_auth(const gchar *server, gushort port, gchar *buf,
64                              const gchar *userid, const gchar *passwd)
65 {
66         NNTPSockInfo *sock;
67
68         sock = nntp_open(server, port, buf);
69         if (!sock)
70                 return NULL;
71         sock->userid = g_strdup(userid);
72         sock->passwd = g_strdup(passwd);
73         return sock;
74 }
75
76
77 void nntp_close(NNTPSockInfo *sock)
78 {
79         if (!sock)
80                 return;
81
82         sock_close(sock->sock);
83         g_free(sock->passwd);
84         g_free(sock->userid);
85         g_free(sock);
86 }
87
88 gint nntp_group(NNTPSockInfo *sock, const gchar *group,
89                 gint *num, gint *first, gint *last)
90 {
91         gint ok;
92         gint resp;
93         gchar buf[NNTPBUFSIZE];
94
95         ok = nntp_gen_command(sock, buf, "GROUP %s", group);
96
97         if (ok != NN_SUCCESS)
98                 return ok;
99
100         if (sscanf(buf, "%d %d %d %d", &resp, num, first, last)
101             != 4) {
102                 log_warning(_("protocol error: %s\n"), buf);
103                 return NN_PROTOCOL;
104         }
105
106         return NN_SUCCESS;
107 }
108
109 gint nntp_get_article(NNTPSockInfo *sock, const gchar *cmd, gint num, gchar **msgid)
110 {
111         gint ok;
112         gchar buf[NNTPBUFSIZE];
113
114         if (num > 0)
115                 ok = nntp_gen_command(sock, buf, "%s %d", cmd, num);
116         else
117                 ok = nntp_gen_command(sock, buf, cmd);
118
119         if (ok != NN_SUCCESS)
120                 return ok;
121
122         extract_parenthesis(buf, '<', '>');
123         if (buf[0] == '\0') {
124                 log_warning(_("protocol error\n"));
125                 return NN_PROTOCOL;
126         }
127         *msgid = g_strdup(buf);
128
129         return NN_SUCCESS;
130 }
131
132 gint nntp_article(NNTPSockInfo *sock, gint num, gchar **msgid)
133 {
134         return nntp_get_article(sock, "ARTICLE", num, msgid);
135 }
136
137 gint nntp_body(NNTPSockInfo *sock, gint num, gchar **msgid)
138 {
139         return nntp_get_article(sock, "BODY", num, msgid);
140 }
141
142 gint nntp_head(NNTPSockInfo *sock, gint num, gchar **msgid)
143 {
144         return nntp_get_article(sock, "HEAD", num, msgid);
145 }
146
147 gint nntp_stat(NNTPSockInfo *sock, gint num, gchar **msgid)
148 {
149         return nntp_get_article(sock, "STAT", num, msgid);
150 }
151
152 gint nntp_next(NNTPSockInfo *sock, gint *num, gchar **msgid)
153 {
154         gint ok;
155         gint resp;
156         gchar buf[NNTPBUFSIZE];
157
158         ok = nntp_gen_command(sock, buf, "NEXT");
159
160         if (ok != NN_SUCCESS)
161                 return ok;
162
163         if (sscanf(buf, "%d %d", &resp, num) != 2) {
164                 log_warning(_("protocol error: %s\n"), buf);
165                 return NN_PROTOCOL;
166         }
167
168         extract_parenthesis(buf, '<', '>');
169         if (buf[0] == '\0') {
170                 log_warning(_("protocol error\n"));
171                 return NN_PROTOCOL;
172         }
173         *msgid = g_strdup(buf);
174
175         return NN_SUCCESS;
176 }
177
178 gint nntp_xover(NNTPSockInfo *sock, gint first, gint last)
179 {
180         gint ok;
181         gchar buf[NNTPBUFSIZE];
182
183         ok = nntp_gen_command(sock, buf, "XOVER %d-%d", first, last);
184         if (ok != NN_SUCCESS)
185                 return ok;
186
187         return NN_SUCCESS;
188 }
189
190 gint nntp_post(NNTPSockInfo *sock, FILE *fp)
191 {
192         gint ok;
193         gchar buf[NNTPBUFSIZE];
194
195         ok = nntp_gen_command(sock, buf, "POST");
196         if (ok != NN_SUCCESS)
197                 return ok;
198
199         while (fgets(buf, sizeof(buf), fp) != NULL) {
200                 strretchomp(buf);
201                 if (buf[0] == '.') {
202                         if (sock_write(sock->sock, ".", 1) < 0) {
203                                 log_warning(_("Error occurred while posting\n"));
204                                 return NN_SOCKET;
205                         }
206                 }
207
208                 if (sock_puts(sock->sock, buf) < 0) {
209                         log_warning(_("Error occurred while posting\n"));
210                         return NN_SOCKET;
211                 }
212         }
213
214         sock_write(sock->sock, ".\r\n", 3);
215         if ((ok = nntp_ok(sock, buf)) != NN_SUCCESS)
216                 return ok;
217
218         return NN_SUCCESS;
219 }
220
221 gint nntp_newgroups(NNTPSockInfo *sock)
222 {
223         return NN_SUCCESS;
224 }
225
226 gint nntp_newnews(NNTPSockInfo *sock)
227 {
228         return NN_SUCCESS;
229 }
230
231 gint nntp_mode(NNTPSockInfo *sock, gboolean stream)
232 {
233         gint ok;
234
235         ok = nntp_gen_command(sock, NULL, "MODE %s",
236                               stream ? "STREAM" : "READER");
237
238         return ok;
239 }
240
241 gint nntp_ok(NNTPSockInfo *sock, gchar *argbuf)
242 {
243         gint ok;
244         gchar buf[NNTPBUFSIZE];
245
246         if ((ok = nntp_gen_recv(sock, buf, sizeof(buf))) == NN_SUCCESS) {
247                 if (strlen(buf) < 4)
248                         return NN_ERROR;
249
250                 if ((buf[0] == '1' || buf[0] == '2' || buf[0] == '3') &&
251                     buf[3] == ' ') {
252                         if (argbuf)
253                                 strcpy(argbuf, buf);
254
255                         if (!strncmp(buf, "381 ", 4))
256                                 return NN_AUTHCONT;
257                         return NN_SUCCESS;
258                 } else if (!strncmp(buf, "480 ", 4))
259                         return NN_AUTHREQ;
260                 else
261                         return NN_ERROR;
262         }
263
264         return ok;
265 }
266
267 static void nntp_gen_send(NNTPSockInfo *sock, const gchar *format, ...)
268 {
269         gchar buf[NNTPBUFSIZE];
270         va_list args;
271
272         va_start(args, format);
273         g_vsnprintf(buf, sizeof(buf), format, args);
274         va_end(args);
275
276         if (verbose) {
277                 if (!g_strncasecmp(buf, "AUTHINFO PASS", 13))
278                         log_print("NNTP> AUTHINFO PASS ***\n");
279                 else
280                         log_print("NNTP> %s\n", buf);
281         }
282
283         strcat(buf, "\r\n");
284         sock_write(sock->sock, buf, strlen(buf));
285 }
286
287 static gint nntp_gen_recv(NNTPSockInfo *sock, gchar *buf, gint size)
288 {
289         if (sock_read(sock->sock, buf, size) == -1)
290                 return NN_SOCKET;
291
292         strretchomp(buf);
293
294         if (verbose)
295                 log_print("NNTP< %s\n", buf);
296
297         return NN_SUCCESS;
298 }
299
300 static gint nntp_gen_command(NNTPSockInfo *sock, gchar *argbuf,
301                              const gchar *format, ...)
302 {
303         gchar buf[NNTPBUFSIZE];
304         va_list args;
305         gint ok;
306
307         va_start(args, format);
308         g_vsnprintf(buf, sizeof(buf) - 2, format, args);
309         va_end(args);
310
311         nntp_gen_send(sock, "%s", buf);
312         ok = nntp_ok(sock, argbuf);
313         if (ok == NN_AUTHREQ && sock->userid && sock->passwd) {
314                 nntp_gen_send(sock, "AUTHINFO USER %s", sock->userid);
315                 ok = nntp_ok(sock, NULL);
316                 if (ok == NN_AUTHCONT) {
317                         nntp_gen_send(sock, "AUTHINFO PASS %s", sock->passwd);
318                         ok = nntp_ok(sock, NULL);
319                 }
320                 if (ok != NN_SUCCESS)
321                         return ok;
322                 nntp_gen_send(sock, "%s", buf);
323                 ok = nntp_ok(sock, argbuf);
324         }
325         return ok;
326 }
327
328 /*
329   nntp_list sends the command "LIST" to the news server,
330   a function is needed to read the newsgroups list.
331  */
332
333 gint nntp_list(NNTPSockInfo *sock)
334 {
335         return nntp_gen_command(sock, NULL, "LIST");
336 }