1922aaf2a367668a2495397a6de08c4a6cca7f44
[doldaconnect.git] / clients / gui-shell / dsh.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 <string.h>
24 #include <errno.h>
25 #include <pwd.h>
26 #include <locale.h>
27 #include <libintl.h>
28 #include <signal.h>
29 #include <sys/wait.h>
30 #include <gtk/gtk.h>
31 #include <gdk/gdkkeysyms.h>
32 #include <stdarg.h>
33 #include <doldaconnect/uilib.h>
34 #include <doldaconnect/uimisc.h>
35 #include <doldaconnect/utils.h>
36
37 #ifdef HAVE_CONFIG_H
38 #include <config.h>
39 #endif
40
41 #ifdef HAVE_NOTIFY
42 #include <libnotify/notify.h>
43 #endif
44
45 #define _(text) gettext(text)
46
47 struct trinfo {
48     int ostate;
49     int opos, spos, speed;
50     time_t lastprog;
51     int warned;
52     double sprog;
53 };
54
55 void updatewrite(void);
56
57 int remote = 0;
58 char *server;
59 GtkStatusIcon *tray;
60 pid_t dpid = 0, dcpid = 0;
61 int connected = 0;
62 int dstat;
63 int derrfd, derrtag;
64 char *derrbuf = NULL;
65 size_t derrbufsize = 0, derrbufdata = 0;
66 int dcfd, dcfdrtag, dcfdwtag = -1;
67 GdkPixbuf *dcicon;
68 #ifdef HAVE_NOTIFY
69 NotifyNotification *trnote = NULL;
70 #endif
71
72 #include "dsh-start.gtkh"
73 #include "dsh-menu.gtkh"
74
75 int running(char *pf)
76 {
77     FILE *pfs;
78     char buf[1024];
79     int pid;
80     
81     if((pfs = fopen(pf, "r")) == NULL) {
82         perror(pf);
83         return(0);
84     }
85     fgets(buf, sizeof(buf), pfs);
86     fclose(pfs);
87     if((pid = atoi(buf)) == 0)
88         return(0);
89     return(!kill(pid, 0));
90 }
91
92 void derrcb(gpointer data, gint source, GdkInputCondition cond)
93 {
94     int ret = 0;
95     
96     sizebuf2(derrbuf, derrbufdata + 1024, 1);
97     ret = read(derrfd, derrbuf + derrbufdata, derrbufsize - derrbufdata);
98     if(ret <= 0) {
99         if(ret < 0)
100             bprintf(derrbuf, "\ncould not read from daemon: %s\n", strerror(errno));
101         gdk_input_remove(derrtag);
102         close(derrfd);
103         derrfd = -1;
104     } else {
105         derrbufdata += ret;
106     }
107 }
108
109 int msgbox(int type, int buttons, char *format, ...)
110 {
111     GtkWidget *swnd;
112     va_list args;
113     char *buf;
114     int resp;
115     
116     va_start(args, format);
117     buf = vsprintf2(format, args);
118     va_end(args);
119     swnd = gtk_message_dialog_new(NULL, 0, type, buttons, "%s", buf);
120     resp = gtk_dialog_run(GTK_DIALOG(swnd));
121     gtk_widget_destroy(swnd);
122     free(buf);
123     return(resp);
124 }
125
126 void destroytr(struct dc_transfer *tr)
127 {
128     struct trinfo *tri;
129     
130     tri = tr->udata;
131     free(tri);
132 }
133
134 void inittr(struct dc_transfer *tr)
135 {
136     struct trinfo *tri;
137     
138     tr->udata = tri = memset(smalloc(sizeof(*tri)), 0, sizeof(*tri));
139     tr->destroycb = destroytr;
140     tri->ostate = tr->state;
141     tri->spos = tri->opos = tr->curpos;
142     tri->speed = -1;
143     tri->lastprog = time(NULL);
144     tri->sprog = ntime();
145 }
146
147 #ifdef HAVE_NOTIFY
148 void notify(NotifyNotification **n, char *cat, char *title, char *body, ...)
149 {
150     va_list args;
151     char *bbuf;
152     
153     va_start(args, body);
154     bbuf = vsprintf2(body, args);
155     va_end(args);
156     if(*n == NULL) {
157         *n = notify_notification_new_with_status_icon(title, bbuf, NULL, tray);
158         notify_notification_set_icon_from_pixbuf(*n, dcicon);
159     } else {
160         notify_notification_update(*n, title, bbuf, NULL);
161     }
162     notify_notification_show(*n, NULL);
163 }
164 #endif
165
166 /* XXX: Achtung! Too DC-specific! */
167 wchar_t *getfilename(wchar_t *path)
168 {
169     wchar_t *p;
170     
171     if((p = wcsrchr(path, L'\\')) == NULL)
172         return(path);
173     else
174         return(p + 1);
175 }
176
177 char *bytes2si(long long bytes)
178 {
179     int i;
180     double b;
181     char *sd;
182     static char ret[64];
183     
184     b = bytes;
185     for(i = 0; (b >= 1024) && (i < 4); i++)
186         b /= 1024;
187     if(i == 0)
188         sd = "B";
189     else if(i == 1)
190         sd = "kiB";
191     else if(i == 2)
192         sd = "MiB";
193     else if(i == 3)
194         sd = "GiB";
195     else
196         sd = "TiB";
197     snprintf(ret, 64, "%.1f %s", b, sd);
198     return(ret);
199 }
200
201 void updatetooltip(void)
202 {
203     struct dc_transfer *tr;
204     struct trinfo *tri;
205     int t, i, a, st, bc, bt;
206     char *buf;
207     size_t bufsize, bufdata;
208     
209     t = i = a = 0;
210     st = bc = bt = -1;
211     for(tr = dc_transfers; tr != NULL; tr = tr->next) {
212         if(tr->dir != DC_TRNSD_DOWN)
213             continue;
214         tri = tr->udata;
215         t++;
216         if(tr->state == DC_TRNS_WAITING)
217             i++;
218         else if((tr->state == DC_TRNS_HS) || (tr->state == DC_TRNS_MAIN))
219             a++;
220         if((tr->state == DC_TRNS_MAIN)) {
221             if(bt == -1)
222                 bc = bt = 0;
223             bc += tr->curpos;
224             bt += tr->size;
225             if(tri->speed != -1) {
226                 if(st == -1)
227                     st = 0;
228                 st += tri->speed;
229             }
230         }
231     }
232     buf = NULL;
233     bufsize = bufdata = 0;
234     bprintf(buf, "%s: %i", _("Transfers"), t);
235     if(t > 0)
236         bprintf(buf, " (%i/%i)", i, a);
237     if(bt > 0)
238         bprintf(buf, ", %.1f%%", ((double)bc / (double)bt) * 100.0);
239     if(st != -1)
240         bprintf(buf, ", %s/s", bytes2si(st));
241     addtobuf(buf, 0);
242     gtk_status_icon_set_tooltip(tray, buf);
243     free(buf);
244 }
245
246 void trstatechange(struct dc_transfer *tr, int ostate)
247 {
248     struct trinfo *tri;
249     
250     tri = tr->udata;
251     if((ostate == DC_TRNS_MAIN) && (tr->dir == DC_TRNSD_DOWN)) {
252         if(tr->state == DC_TRNS_DONE) {
253 #ifdef HAVE_NOTIFY
254             if(dcpid == 0)
255                 notify(&trnote, "transfer.complete", _("Transfer complete"), _("Finished downloading %ls from %ls"), getfilename(tr->path), tr->peernick);
256 #endif
257         } else {
258 #ifdef HAVE_NOTIFY
259             if(dcpid == 0)
260                 notify(&trnote, "transfer.error", _("Transfer interrupted"), _("The transfer of %ls from %ls was interrupted from the other side"), getfilename(tr->path), tr->peernick);
261 #endif
262         }
263     }
264     if(tr->state == DC_TRNS_MAIN) {
265         tri->speed = -1;
266         tri->spos = tr->curpos;
267         tri->sprog = ntime();
268     }
269 }
270
271 void updatetrinfo(void)
272 {
273     struct dc_transfer *tr;
274     struct trinfo *tri;
275     time_t now;
276     double dnow;
277     
278     now = time(NULL);
279     dnow = ntime();
280     for(tr = dc_transfers; tr != NULL; tr = tr->next) {
281         if(tr->udata == NULL) {
282             inittr(tr);
283         } else {
284             tri = tr->udata;
285             if(tri->ostate != tr->state) {
286                 trstatechange(tr, tri->ostate);
287                 tri->ostate = tr->state;
288             }
289             if(tri->opos != tr->curpos) {
290                 tri->opos = tr->curpos;
291                 tri->lastprog = now;
292                 tri->warned = 0;
293             }
294 #ifdef HAVE_NOTIFY
295             if((tr->state = DC_TRNS_MAIN) && (now - tri->lastprog > 600) && !tri->warned) {
296                 if(dcpid == 0) {
297                     notify(&trnote, "transfer.error", _("Transfer stalled"), _("The transfer of %ls from %ls has not made progress for 10 minutes"), getfilename(tr->path), tr->peernick);
298                     tri->warned = 1;
299                 }
300             }
301 #endif
302             if((tr->state == DC_TRNS_MAIN) && (dnow - tri->sprog > 10)) {
303                 tri->speed = ((double)(tr->curpos - tri->spos) / (dnow - tri->sprog));
304                 tri->spos = tr->curpos;
305                 tri->sprog = dnow;
306             }
307         }
308     }
309     updatetooltip();
310 }
311
312 void trlscb(int resp, void *data)
313 {
314     updatetrinfo();
315 }
316
317 gint trupdatecb(gpointer data)
318 {
319     updatetrinfo();
320     return(TRUE);
321 }
322
323 void logincb(int err, wchar_t *reason, void *data)
324 {
325     if(err != DC_LOGIN_ERR_SUCCESS) {
326         msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not connect to server"));
327         exit(1);
328     }
329     dc_queuecmd(NULL, NULL, L"notify", L"trans:act", L"on", L"trans:prog", L"on", NULL);
330     dc_gettrlistasync(trlscb, NULL);
331     connected = 1;
332     updatewrite();
333 }
334
335 void dcfdcb(gpointer data, gint source, GdkInputCondition cond)
336 {
337     struct dc_response *resp;
338     
339     if(((cond & GDK_INPUT_READ) && dc_handleread()) || ((cond & GDK_INPUT_WRITE) && dc_handlewrite())) {
340         if(errno == 0) {
341             gtk_main_quit();
342         } else {
343             msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not connect to server: %s"), strerror(errno));
344             exit(1);
345         }
346         return;
347     }
348     while((resp = dc_getresp()) != NULL) {
349         if(!wcscmp(resp->cmdname, L".connect")) {
350             if(resp->code != 201) {
351                 msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("The server refused the connection"));
352                 exit(1);
353             } else if(dc_checkprotocol(resp, DC_LATEST)) {
354                 msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Server protocol revision mismatch"));
355                 exit(1);
356             } else {
357                 gtk_status_icon_set_tooltip(tray, _("Authenticating..."));
358                 dc_loginasync(NULL, 1, dc_convnone, logincb, NULL);
359             }
360         } else if(!wcscmp(resp->cmdname, L".notify")) {
361             dc_uimisc_handlenotify(resp);
362             updatetrinfo();
363         }
364         dc_freeresp(resp);
365     }
366     updatewrite();
367 }
368
369 void updatewrite(void)
370 {
371     if(dcfd < 0)
372         return;
373     if(dc_wantwrite()) {
374         if(dcfdwtag == -1)
375             dcfdwtag = gdk_input_add(dcfd, GDK_INPUT_WRITE, dcfdcb, NULL);
376     } else {
377         if(dcfdwtag != -1) {
378             gdk_input_remove(dcfdwtag);
379             dcfdwtag = -1;
380         }
381     }
382 }
383
384 void connectdc(void)
385 {
386     if((dcfd = dc_connect(server)) < 0) {
387         msgbox(GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, _("Could not connect to server: %s"), strerror(errno));
388         exit(1);
389     }
390     dcfdrtag = gdk_input_add(dcfd, GDK_INPUT_READ, dcfdcb, NULL);
391     updatewrite();
392     gtk_status_icon_set_tooltip(tray, _("Connecting to server..."));
393 }
394
395 void startdaemon(void)
396 {
397     char pf[1024];
398     int pfd[2], i;
399     
400     if(getenv("HOME") != NULL)
401         snprintf(pf, sizeof(pf), "%s/.doldacond.pid", getenv("HOME"));
402     else
403         snprintf(pf, sizeof(pf), "%s/.doldacond.pid", getpwuid(getuid())->pw_dir);
404     if(access(pf, F_OK) || !running(pf)) {
405         pipe(pfd);
406         if((dpid = fork()) == 0) {
407             dup2(pfd[1], 2);
408             for(i = 3; i < FD_SETSIZE; i++)
409                 close(i);
410             execlp("doldacond", "doldacond", "-p", pf, NULL);
411             perror("doldacond");
412             exit(127);
413         }
414         if(dpid == -1)
415             abort();
416         close(pfd[1]);
417         derrfd = pfd[0];
418         derrtag = gdk_input_add(derrfd, GDK_INPUT_READ, derrcb, NULL);
419         create_start_wnd();
420         gtk_widget_show(start_wnd);
421         gtk_status_icon_set_tooltip(tray, _("Starting..."));
422     } else {
423         connectdc();
424     }
425 }
426
427 gboolean daemonized(gpointer uu)
428 {
429     gtk_widget_hide(start_wnd);
430     dpid = 0;
431     if(derrfd != -1) {
432         gdk_input_remove(derrtag);
433         close(derrfd);
434     }
435     if(dstat != 0) {
436         gtk_status_icon_set_visible(tray, FALSE);
437         gtk_text_buffer_set_text(gtk_text_view_get_buffer(GTK_TEXT_VIEW(start_log)), derrbuf, derrbufdata);
438         gtk_widget_show(start_errwnd);
439     } else {
440         connectdc();
441     }
442     return(FALSE);
443 }
444
445 void sighandler(int sig)
446 {
447     pid_t p;
448     int status;
449     
450     if(sig == SIGCHLD) {
451         while((p = waitpid(-1, &status, WNOHANG)) > 0) {
452             if(p == dpid) {
453                 dstat = status;
454                 gtk_timeout_add(1, daemonized, NULL);
455             } else if(p == dcpid) {
456                 dcpid = 0;
457             }
458         }
459     }
460 }
461
462 void dolcon(void)
463 {
464     int i;
465     
466     if((dcpid = fork()) == 0) {
467         for(i = 3; i < FD_SETSIZE; i++)
468             close(i);
469         if(remote)
470             execlp("dolcon", "dolcon", NULL);
471         else
472             execlp("dolcon", "dolcon", "-l", NULL);
473         perror("dolcon");
474         exit(127);
475     }
476 }
477
478 void cb_shm_dolconf_activate(GtkWidget *uu1, gpointer uu2)
479 {
480     int i;
481     
482     if((dcpid = fork()) == 0) {
483         for(i = 3; i < FD_SETSIZE; i++)
484             close(i);
485         execlp("dolconf", "dolconf", NULL);
486         perror("dolconf");
487         exit(127);
488     }
489 }
490
491 void cb_shm_dolcon_activate(GtkWidget *uu1, gpointer uu2)
492 {
493     dolcon();
494 }
495
496 void cb_shm_quit_activate(GtkWidget *uu1, gpointer uu2)
497 {
498     dc_queuecmd(NULL, NULL, L"shutdown", NULL);
499     updatewrite();
500 }
501
502 void tray_activate(GtkStatusIcon *uu1, gpointer uu2)
503 {
504     if(dpid != 0) {
505         gtk_widget_show(start_wnd);
506     } else if(connected) {
507         dolcon();
508     }
509 }
510
511 void tray_popup(GtkStatusIcon *uu1, guint button, guint time, gpointer uu2)
512 {
513     gtk_menu_popup(GTK_MENU(shm_menu), NULL, NULL, NULL, NULL, button, time);
514 }
515
516 void cb_start_hide_clicked(GtkWidget *uu1, gpointer uu2)
517 {
518     gtk_widget_hide(start_wnd);
519 }
520
521 void cb_start_abort_clicked(GtkWidget *uu1, gpointer uu2)
522 {
523     kill(dpid, SIGINT);
524     exit(0);
525 }
526
527 void cb_start_errok_clicked(GtkWidget *uu1, gpointer uu2)
528 {
529     gtk_main_quit();
530 }
531
532 #include "../dolda-icon.xpm"
533
534 void inittray(void)
535 {
536     tray = gtk_status_icon_new_from_pixbuf(gdk_pixbuf_scale_simple(dcicon, 24, 24, GDK_INTERP_BILINEAR));
537     gtk_status_icon_set_tooltip(tray, "");
538     g_signal_connect(G_OBJECT(tray), "activate", G_CALLBACK(tray_activate), (gpointer)NULL);
539     g_signal_connect(G_OBJECT(tray), "popup-menu", G_CALLBACK(tray_popup), (gpointer)NULL);
540 }
541
542 int main(int argc, char **argv)
543 {
544     int c;
545     
546     setlocale(LC_ALL, "");
547     bindtextdomain(PACKAGE, LOCALEDIR);
548     textdomain(PACKAGE);
549     signal(SIGCHLD, sighandler);
550     dc_init();
551     server = dc_srv_local;
552     gtk_init(&argc, &argv);
553 #ifdef HAVE_NOTIFY
554     notify_init("Dolda Connect");
555 #endif
556     while((c = getopt(argc, argv, "rhs:")) != -1) {
557         switch(c) {
558         case 'r':
559             remote = 1;
560             server = NULL;
561             break;
562         case 's':
563             remote = 1;
564             server = optarg;
565             break;
566         case 'h':
567             printf("usage: doldacond-shell [-hr]\n");
568             printf("\t-h\tDisplay this help message\n");
569             printf("\t-r\tConnect to a remote host\n");
570             exit(0);
571         default:
572             fprintf(stderr, "usage: doldacond-shell [-hr]\n");
573             exit(1);
574         }
575     }
576
577     create_shm_wnd();
578     dcicon = gdk_pixbuf_new_from_xpm_data((const char **)dolda_icon_xpm);
579     gtk_window_set_default_icon(dcicon);
580     inittray();
581     if(remote)
582         connectdc();
583     else
584         startdaemon();
585     
586     g_timeout_add(10000, trupdatecb, NULL);
587     gtk_main();
588
589     return(0);
590 }
591
592 #include "dsh-start.gtk"
593 #include "dsh-menu.gtk"