Clarified the dscp-tos documentation a bit.
[doldaconnect.git] / clients / gtk2 / hublist.c
1 /*
2  *  Dolda Connect - Modular multiuser Direct Connect-style client
3  *  Copyright (C) 2007 Fredrik Tolf <fredrik@dolda2000.com>
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 #include <stdlib.h>
21 #include <stdio.h>
22 #include <unistd.h>
23 #include <regex.h>
24 #include <signal.h>
25 #include <string.h>
26 #include <stdarg.h>
27 #include <gtk/gtk.h>
28 #include <bzlib.h>
29 #include <errno.h>
30 #include <sys/poll.h>
31
32 #ifdef HAVE_CONFIG_H
33 #include <config.h>
34 #endif
35 #include "dolcon.h"
36 #include "hublist.h"
37 #include <http.h>
38
39 static void settags(void);
40
41 static regex_t *filter = NULL;
42 static int itag = -1, otag = -1;
43 static int (*handler)(int, char *, size_t) = NULL;
44 static int state;
45 static struct htconn *hc;
46 static bz_stream *bzs;
47 /* Only needed because of bzlib: */
48 static char *mybuf;
49 static size_t mybufsize, mybufdata;
50
51 #include "mainwnd.gtkh"
52
53 void aborthublist(void)
54 {
55     if(mybuf != NULL) {
56         free(mybuf);
57         mybuf = NULL;
58         mybufsize = mybufdata = 0;
59     }
60     if(hc != NULL) {
61         gtk_widget_hide(main_pubhubbarbox);
62         if(itag != -1)
63             gdk_input_remove(itag);
64         if(otag != -1)
65             gdk_input_remove(otag);
66         itag = otag = -1;
67         freehtconn(hc);
68         hc = NULL;
69         if(bzs != NULL) {
70             BZ2_bzDecompressEnd(bzs);
71             free(bzs);
72             bzs = NULL;
73         }
74         if(filter != NULL) {
75             regfree(filter);
76             free(filter);
77             filter = NULL;
78         }
79         if(handler != NULL) {
80             handler(PHO_FINI, NULL, 0);
81             handler = NULL;
82         }
83     }
84 }
85
86 int validhub(char *field, ...)
87 {
88     int match;
89     va_list args;
90     
91     if(filter == NULL)
92         return(1);
93     match = 0;
94     va_start(args, field);
95     do {
96         if(!regexec(filter, field, 0, NULL, 0)) {
97             match = 1;
98             break;
99         }
100     } while((field = va_arg(args, char *)) != NULL);
101     va_end(args);
102     return(match);
103 }
104
105 static void fdcb(gpointer data, gint source, GdkInputCondition cond)
106 {
107     int ret, bzret, hret;
108     
109     if(itag != -1)
110         gdk_input_remove(itag);
111     if(otag != -1)
112         gdk_input_remove(otag);
113     itag = otag = -1;
114     ret = htprocess(hc, ((cond & GDK_INPUT_READ)?POLLIN:0) | ((cond & GDK_INPUT_WRITE)?POLLOUT:0));
115     if(ret < 0) {
116         msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not read hublist from server: %s"), strerror(errno));
117         aborthublist();
118         return;
119     }
120     if(state == 0) {
121         if(hc->rescode != 0) {
122             if(hc->rescode != 200) {
123                 msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("The hublist server returned an error: \"%i %s\""), hc->rescode, hc->resstr);
124                 aborthublist();
125                 return;
126             }
127             state = 1;
128             gtk_progress_bar_set_text(GTK_PROGRESS_BAR(main_pubhubbar), _("Getting list..."));
129         }
130     }
131     if(state == 1) {
132         if(hc->tlen > 0) {
133             gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(main_pubhubbar), ((double)hc->rxd) / ((double)hc->tlen));
134         } else {
135             gtk_progress_bar_set_pulse_step(GTK_PROGRESS_BAR(main_pubhubbar), ((double)hc->databufdata) / 10000.0);
136             gtk_progress_bar_pulse(GTK_PROGRESS_BAR(main_pubhubbar));
137         }
138         if(hc->databufdata > 0) {
139             if(bzs == NULL) {
140                 bufcat(mybuf, hc->databuf, hc->databufdata);
141                 hc->databufdata = 0;
142             } else {
143                 bzs->next_in = hc->databuf;
144                 bzs->avail_in = hc->databufdata;
145                 do {
146                     sizebuf2(mybuf, mybufdata + 1024, 1);
147                     bzs->next_out = mybuf + mybufdata;
148                     bzs->avail_out = mybufsize - mybufdata;
149                     bzret = BZ2_bzDecompress(bzs);
150                     if((bzret != BZ_OK) && (bzret != BZ_STREAM_END)) {
151                         msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not decompress hublist (%i)"), hret);
152                         aborthublist();
153                         return;
154                     }
155                     mybufdata = mybufsize - bzs->avail_out;
156                     if(bzret == BZ_STREAM_END) {
157                         ret = 1;
158                         break;
159                     }
160                 } while(bzs->avail_out == 0);
161                 memmove(hc->databuf, bzs->next_in, hc->databufdata -= (bzs->next_in - hc->databuf));
162             }
163             if((hret = handler(PHO_DATA, mybuf, mybufdata)) < 0)
164                 aborthublist();
165             else
166                 memmove(mybuf, mybuf + hret, mybufdata -= hret);
167         }
168         if(ret) {
169             gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(main_pubhubbar), 1);
170             gtk_progress_bar_set_text(GTK_PROGRESS_BAR(main_pubhubbar), _("Finalizing list..."));
171             gdk_window_process_updates(main_pubhubbar->window, FALSE);
172             handler(PHO_EOF, NULL, 0);
173             aborthublist();
174         }
175     }
176     if(hc != NULL)
177         settags();
178 }
179
180 static void settags(void)
181 {
182     if(htpollflags(hc) & POLLIN)
183         itag = gdk_input_add(hc->fd, GDK_INPUT_READ, fdcb, NULL);
184     if(htpollflags(hc) & POLLOUT)
185         otag = gdk_input_add(hc->fd, GDK_INPUT_WRITE, fdcb, NULL);
186 }
187
188 void fetchhublist(char *url, regex_t *flt)
189 {
190     int len;
191     char *p;
192     struct hturlinfo *u;
193     
194     aborthublist();
195     filter = flt;
196     u = parseurl(url);
197     hc = htconnect(u);
198     freeurl(u);
199     if(hc == NULL) {
200         msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not read hublist from server: %s"), strerror(errno));
201         return;
202     }
203     state = 0;
204     settags();
205     gtk_widget_show(main_pubhubbarbox);
206     gtk_progress_bar_set_text(GTK_PROGRESS_BAR(main_pubhubbar), _("Connecting..."));
207     gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(main_pubhubbar), 0);
208     gdk_window_process_updates(main_pubhubbarbox->window, TRUE);
209
210     len = strlen(url);
211     p = url + len;
212     if((len > 4) && !strncmp(p - 4, ".bz2", 4)) {
213         /* Because using Transfer-Encoding would just be too good! */
214         p -= 4;
215         len -= 4;
216         bzs = memset(smalloc(sizeof(*bzs)), 0, sizeof(*bzs));
217         if(BZ2_bzDecompressInit(bzs, 0, 0) != BZ_OK) {
218             msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not initialize decompression library"));
219             free(bzs);
220             bzs = NULL;
221             aborthublist();
222             return;
223         }
224     }
225     if((len > 4) && !strncmp(p - 4, ".xml", 4)) {
226         /* Because using Content-Type would just be too good! */
227         p -= 4;
228         len -= 4;
229         handler = pubhubxmlhandler;
230     } else {
231         handler = pubhuboldhandler;
232     }
233     handler(PHO_INIT, NULL, 0);
234 }