source: owl.c @ 5001a3d

Last change on this file since 5001a3d was 06e04a9, checked in by David Benjamin <davidben@mit.edu>, 9 years ago
perlvariables - iteration N Later commits reworked a lot of this, but the use of GValue over void* was kept.
  • Property mode set to 100644
File size: 19.0 KB
Line 
1/*  Copyright (c) 2006-2011 The BarnOwl Developers. All rights reserved.
2 *  Copyright (c) 2004 James Kretchmar. All rights reserved.
3 *
4 *  This program is free software. You can redistribute it and/or
5 *  modify under the terms of the Sleepycat License. See the COPYING
6 *  file included with the distribution for more information.
7 */
8
9#include "owl.h"
10#include <stdio.h>
11#include <getopt.h>
12#include <sys/stat.h>
13#include <locale.h>
14#include <unistd.h>
15
16#if OWL_STDERR_REDIR
17#ifdef HAVE_SYS_IOCTL_H
18#include <sys/ioctl.h>
19#endif
20#ifdef HAVE_SYS_FILIO_H
21#include <sys/filio.h>
22#endif
23int stderr_replace(void);
24#endif
25
26owl_global g;
27
28typedef struct _owl_options {
29  bool load_initial_subs;
30  char *configfile;
31  char *tty;
32  char *confdir;
33  bool debug;
34} owl_options;
35
36void usage(void)
37{
38  fprintf(stderr, "Barnowl version %s\n", OWL_VERSION_STRING);
39  fprintf(stderr, "Usage: barnowl [-n] [-d] [-D] [-v] [-h] [-c <configfile>] [-s <confdir>] [-t <ttyname>]\n");
40  fprintf(stderr, "  -n,--no-subs        don't load zephyr subscriptions\n");
41  fprintf(stderr, "  -d,--debug          enable debugging\n");
42  fprintf(stderr, "  -v,--version        print the Barnowl version number and exit\n");
43  fprintf(stderr, "  -h,--help           print this help message\n");
44  fprintf(stderr, "  -s,--config-dir     specify an alternate config dir (default ~/.owl)\n");
45  fprintf(stderr, "  -c,--config-file    specify an alternate config file (default ~/.owl/init.pl)\n");
46  fprintf(stderr, "  -t,--tty            set the tty name\n");
47}
48
49/* TODO: free owl_options after init is done? */
50void owl_parse_options(int argc, char *argv[], owl_options *opts) {
51  static const struct option long_options[] = {
52    { "no-subs",         0, 0, 'n' },
53    { "config-file",     1, 0, 'c' },
54    { "config-dir",      1, 0, 's' },
55    { "tty",             1, 0, 't' },
56    { "debug",           0, 0, 'd' },
57    { "version",         0, 0, 'v' },
58    { "help",            0, 0, 'h' },
59    { NULL, 0, NULL, 0}
60  };
61  char c;
62
63  while((c = getopt_long(argc, argv, "nc:t:s:dDvh",
64                         long_options, NULL)) != -1) {
65    switch(c) {
66    case 'n':
67      opts->load_initial_subs = 0;
68      break;
69    case 'c':
70      opts->configfile = g_strdup(optarg);
71      break;
72    case 's':
73      opts->confdir = g_strdup(optarg);
74      break;
75    case 't':
76      opts->tty = g_strdup(optarg);
77      break;
78    case 'd':
79      opts->debug = 1;
80      break;
81    case 'v':
82      printf("This is BarnOwl version %s\n", OWL_VERSION_STRING);
83      exit(0);
84    case 'h':
85    default:
86      usage();
87      exit(1);
88    }
89  }
90}
91
92void owl_start_color(void) {
93  start_color();
94#ifdef HAVE_USE_DEFAULT_COLORS
95  use_default_colors();
96#endif
97
98  /* define simple color pairs */
99  if (has_colors() && COLOR_PAIRS>=8) {
100    int bg = COLOR_BLACK;
101#ifdef HAVE_USE_DEFAULT_COLORS
102    bg = -1;
103#endif
104    init_pair(OWL_COLOR_BLACK,   COLOR_BLACK,   bg);
105    init_pair(OWL_COLOR_RED,     COLOR_RED,     bg);
106    init_pair(OWL_COLOR_GREEN,   COLOR_GREEN,   bg);
107    init_pair(OWL_COLOR_YELLOW,  COLOR_YELLOW,  bg);
108    init_pair(OWL_COLOR_BLUE,    COLOR_BLUE,    bg);
109    init_pair(OWL_COLOR_MAGENTA, COLOR_MAGENTA, bg);
110    init_pair(OWL_COLOR_CYAN,    COLOR_CYAN,    bg);
111    init_pair(OWL_COLOR_WHITE,   COLOR_WHITE,   bg);
112  }
113}
114
115void owl_start_curses(void) {
116  struct termios tio;
117  /* save initial terminal settings */
118  tcgetattr(STDIN_FILENO, owl_global_get_startup_tio(&g));
119
120  tcgetattr(STDIN_FILENO, &tio);
121  tio.c_iflag &= ~(ISTRIP|IEXTEN);
122  tio.c_cc[VQUIT] = fpathconf(STDIN_FILENO, _PC_VDISABLE);
123  tio.c_cc[VSUSP] = fpathconf(STDIN_FILENO, _PC_VDISABLE);
124  tio.c_cc[VSTART] = fpathconf(STDIN_FILENO, _PC_VDISABLE);
125  tio.c_cc[VSTOP] = fpathconf(STDIN_FILENO, _PC_VDISABLE);
126  tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio);
127
128  /* screen init */
129  initscr();
130  cbreak();
131  noecho();
132
133  owl_start_color();
134}
135
136void owl_shutdown_curses(void) {
137  endwin();
138  /* restore terminal settings */
139  tcsetattr(STDIN_FILENO, TCSAFLUSH, owl_global_get_startup_tio(&g));
140}
141
142/*
143 * Process a new message passed to us on the message queue from some
144 * protocol. This includes adding it to the message list, updating the
145 * view and scrolling if appropriate, logging it, and so on.
146 *
147 * Either a pointer is kept to the message internally, or it is freed
148 * if unneeded. The caller no longer ``owns'' the message's memory.
149 *
150 * Returns 1 if the message was added to the message list, and 0 if it
151 * was ignored due to user settings or otherwise.
152 */
153static int owl_process_message(owl_message *m) {
154  const owl_filter *f;
155  /* if this message it on the puntlist, nuke it and continue */
156  if (owl_global_message_is_puntable(&g, m)) {
157    owl_message_delete(m);
158    return 0;
159  }
160
161  /*  login or logout that should be ignored? */
162  if (owl_global_is_ignorelogins(&g)
163      && owl_message_is_loginout(m)) {
164    owl_message_delete(m);
165    return 0;
166  }
167
168  if (!owl_global_is_displayoutgoing(&g)
169      && owl_message_is_direction_out(m)) {
170    owl_message_delete(m);
171    return 0;
172  }
173
174  /* add it to the global list */
175  owl_messagelist_append_element(owl_global_get_msglist(&g), m);
176  /* add it to any necessary views; right now there's only the current view */
177  owl_view_consider_message(owl_global_get_current_view(&g), m);
178
179  if(owl_message_is_direction_in(m)) {
180    /* let perl know about it*/
181    owl_perlconfig_getmsg(m, NULL);
182
183    /* do we need to autoreply? */
184    if (owl_global_is_zaway(&g) && !owl_message_get_attribute_value(m, "isauto")) {
185      if (owl_message_is_type_zephyr(m)) {
186        owl_zephyr_zaway(m);
187      } else if (owl_message_is_type_aim(m)) {
188        if (owl_message_is_private(m)) {
189          owl_function_send_aimawymsg(owl_message_get_sender(m), owl_global_get_zaway_msg(&g));
190        }
191      }
192    }
193
194    /* ring the bell if it's a personal */
195    if (!strcmp(owl_global_get_personalbell(&g), "on")) {
196      if (!owl_message_is_loginout(m) &&
197          !owl_message_is_mail(m) &&
198          owl_message_is_personal(m)) {
199        owl_function_beep();
200      }
201    } else if (!strcmp(owl_global_get_personalbell(&g), "off")) {
202      /* do nothing */
203    } else {
204      f=owl_global_get_filter(&g, owl_global_get_personalbell(&g));
205      if (f && owl_filter_message_match(f, m)) {
206        owl_function_beep();
207      }
208    }
209
210    /* if it matches the alert filter, do the alert action */
211    f=owl_global_get_filter(&g, owl_global_get_alert_filter(&g));
212    if (f && owl_filter_message_match(f, m)) {
213      owl_function_command_norv(owl_global_get_alert_action(&g));
214    }
215
216    /* if it's a zephyr login or logout, update the zbuddylist */
217    if (owl_message_is_type_zephyr(m) && owl_message_is_loginout(m)) {
218      if (owl_message_is_login(m)) {
219        owl_zbuddylist_adduser(owl_global_get_zephyr_buddylist(&g), owl_message_get_sender(m));
220      } else if (owl_message_is_logout(m)) {
221        owl_zbuddylist_deluser(owl_global_get_zephyr_buddylist(&g), owl_message_get_sender(m));
222      } else {
223        owl_function_error("Internal error: received login notice that is neither login nor logout");
224      }
225    }
226  }
227
228  /* let perl know about it */
229  owl_perlconfig_newmsg(m, NULL);
230  /* log the message if we need to */
231  owl_log_message(m);
232  /* redraw the sepbar; TODO: don't violate layering */
233  owl_global_sepbar_dirty(&g);
234
235  return 1;
236}
237
238static gboolean owl_process_messages_prepare(GSource *source, int *timeout) {
239  *timeout = -1;
240  return owl_global_messagequeue_pending(&g);
241}
242
243static gboolean owl_process_messages_check(GSource *source) {
244  return owl_global_messagequeue_pending(&g);
245}
246
247/*
248 * Process any new messages we have waiting in the message queue.
249 */
250static gboolean owl_process_messages_dispatch(GSource *source, GSourceFunc callback, gpointer user_data) {
251  int newmsgs=0;
252  int followlast = owl_global_should_followlast(&g);
253  owl_message *m;
254
255  /* Grab incoming messages. */
256  while (owl_global_messagequeue_pending(&g)) {
257    m = owl_global_messagequeue_popmsg(&g);
258    if (owl_process_message(m))
259      newmsgs = 1;
260  }
261
262  if (newmsgs) {
263    /* follow the last message if we're supposed to */
264    if (followlast)
265      owl_function_lastmsg();
266
267    /* do the newmsgproc thing */
268    owl_function_do_newmsgproc();
269
270    /* redisplay if necessary */
271    /* this should be optimized to not run if the new messages won't be displayed */
272    owl_mainwin_redisplay(owl_global_get_mainwin(&g));
273  }
274  return TRUE;
275}
276
277static GSourceFuncs owl_process_messages_funcs = {
278  owl_process_messages_prepare,
279  owl_process_messages_check,
280  owl_process_messages_dispatch,
281  NULL
282};
283
284void owl_process_input_char(owl_input j)
285{
286  int ret;
287
288  owl_global_set_lastinputtime(&g, time(NULL));
289  ret = owl_keyhandler_process(owl_global_get_keyhandler(&g), j);
290  if (ret!=0 && ret!=1) {
291    owl_function_makemsg("Unable to handle keypress");
292  }
293}
294
295gboolean owl_process_input(GIOChannel *source, GIOCondition condition, void *data)
296{
297  owl_global *g = data;
298  owl_input j;
299
300  while (1) {
301    j.ch = wgetch(g->input_pad);
302    if (j.ch == ERR) return TRUE;
303
304    j.uch = '\0';
305    if (j.ch >= KEY_MIN && j.ch <= KEY_MAX) {
306      /* This is a curses control character. */
307    }
308    else if (j.ch > 0x7f && j.ch < 0xfe) {
309      /* Pull in a full utf-8 character. */
310      int bytes, i;
311      char utf8buf[7];
312      memset(utf8buf, '\0', 7);
313     
314      utf8buf[0] = j.ch;
315     
316      if ((j.ch & 0xc0) && (~j.ch & 0x20)) bytes = 2;
317      else if ((j.ch & 0xe0) && (~j.ch & 0x10)) bytes = 3;
318      else if ((j.ch & 0xf0) && (~j.ch & 0x08)) bytes = 4;
319      else if ((j.ch & 0xf8) && (~j.ch & 0x04)) bytes = 5;
320      else if ((j.ch & 0xfc) && (~j.ch & 0x02)) bytes = 6;
321      else bytes = 1;
322     
323      for (i = 1; i < bytes; i++) {
324        int tmp = wgetch(g->input_pad);
325        /* If what we got was not a byte, or not a continuation byte */
326        if (tmp > 0xff || !(tmp & 0x80 && ~tmp & 0x40)) {
327          /* ill-formed UTF-8 code unit subsequence, put back the
328             char we just got. */
329          ungetch(tmp);
330          j.ch = ERR;
331          break;
332        }
333        utf8buf[i] = tmp;
334      }
335     
336      if (j.ch != ERR) {
337        if (g_utf8_validate(utf8buf, -1, NULL)) {
338          j.uch = g_utf8_get_char(utf8buf);
339        }
340        else {
341          j.ch = ERR;
342        }
343      }
344    }
345    else if (j.ch <= 0x7f) {
346      j.uch = j.ch;
347    }
348
349    owl_process_input_char(j);
350  }
351  return TRUE;
352}
353
354static void sig_handler_main_thread(void *data) {
355  int sig = GPOINTER_TO_INT(data);
356
357  owl_function_debugmsg("Got signal %d", sig);
358  if (sig == SIGWINCH) {
359    owl_function_resize();
360  } else if (sig == SIGTERM || sig == SIGHUP) {
361    owl_function_quit();
362  } else if (sig == SIGINT && owl_global_take_interrupt(&g)) {
363    owl_input in;
364    in.ch = in.uch = owl_global_get_startup_tio(&g)->c_cc[VINTR];
365    owl_process_input_char(in);
366  }
367}
368
369static void sig_handler(const siginfo_t *siginfo, void *data) {
370  /* If it was an interrupt, set a flag so we can handle it earlier if
371   * needbe. sig_handler_main_thread will check the flag to make sure
372   * no one else took it. */
373  if (siginfo->si_signo == SIGINT) {
374    owl_global_add_interrupt(&g);
375  }
376  /* Send a message to the main thread. */
377  owl_select_post_task(sig_handler_main_thread,
378                       GINT_TO_POINTER(siginfo->si_signo), 
379                       NULL, g_main_context_default());
380}
381
382#define OR_DIE(s, syscall)       \
383  G_STMT_START {                 \
384    if ((syscall) == -1) {       \
385      perror((s));               \
386      exit(1);                   \
387    }                            \
388  } G_STMT_END
389
390void owl_register_signal_handlers(void) {
391  struct sigaction sig_ignore = { .sa_handler = SIG_IGN };
392  struct sigaction sig_default = { .sa_handler = SIG_DFL };
393  sigset_t sigset;
394  int ret, i;
395  const int reset_signals[] = { SIGABRT, SIGBUS, SIGCHLD, SIGFPE, SIGILL,
396                                SIGQUIT, SIGSEGV, };
397  /* Don't bother resetting watched ones because owl_signal_init will. */
398  const int watch_signals[] = { SIGWINCH, SIGTERM, SIGHUP, SIGINT, };
399
400  /* Sanitize our signals; the mask and dispositions from our parent
401   * aren't really useful. Signal list taken from equivalent code in
402   * Chromium. */
403  OR_DIE("sigemptyset", sigemptyset(&sigset));
404  if ((ret = pthread_sigmask(SIG_SETMASK, &sigset, NULL)) != 0) {
405    errno = ret;
406    perror("pthread_sigmask");
407    exit(1);
408  }
409  for (i = 0; i < G_N_ELEMENTS(reset_signals); i++) {
410    OR_DIE("sigaction", sigaction(reset_signals[i], &sig_default, NULL));
411  }
412
413  /* Turn off SIGPIPE; we check the return value of write. */
414  OR_DIE("sigaction", sigaction(SIGPIPE, &sig_ignore, NULL));
415
416  /* Register some signals with the signal thread. */
417  owl_signal_init(watch_signals, G_N_ELEMENTS(watch_signals),
418                  sig_handler, NULL);
419}
420
421#if OWL_STDERR_REDIR
422
423/* Replaces stderr with a pipe so that we can read from it.
424 * Returns the fd of the pipe from which stderr can be read. */
425int stderr_replace(void)
426{
427  int pipefds[2];
428  if (0 != pipe(pipefds)) {
429    perror("pipe");
430    owl_function_debugmsg("stderr_replace: pipe FAILED");
431    return -1;
432  }
433    owl_function_debugmsg("stderr_replace: pipe: %d,%d", pipefds[0], pipefds[1]);
434  if (-1 == dup2(pipefds[1], STDERR_FILENO)) {
435    owl_function_debugmsg("stderr_replace: dup2 FAILED (%s)", strerror(errno));
436    perror("dup2");
437    return -1;
438  }
439  return pipefds[0];
440}
441
442/* Sends stderr (read from rfd) messages to the error console */
443gboolean stderr_redirect_handler(GIOChannel *source, GIOCondition condition, void *data)
444{
445  int navail, bread;
446  char buf[4096];
447  int rfd = g_io_channel_unix_get_fd(source);
448  char *err;
449
450  /* TODO: Use g_io_channel_read_line? We'd have to be careful about
451   * blocking on the read. */
452
453  if (rfd<0) return TRUE;
454  if (-1 == ioctl(rfd, FIONREAD, &navail)) {
455    return TRUE;
456  }
457  /*owl_function_debugmsg("stderr_redirect: navail = %d\n", navail);*/
458  if (navail <= 0) return TRUE;
459  if (navail > sizeof(buf)-1) {
460    navail = sizeof(buf)-1;
461  }
462  bread = read(rfd, buf, navail);
463  if (bread == -1)
464    return TRUE;
465
466  err = g_strdup_printf("[stderr]\n%.*s", bread, buf);
467
468  owl_function_log_err(err);
469  g_free(err);
470  return TRUE;
471}
472
473#endif /* OWL_STDERR_REDIR */
474
475int main(int argc, char **argv, char **env)
476{
477  int argc_copy;
478  char **argv_copy;
479  char *perlout, *perlerr;
480  const char *dlerr;
481  const owl_style *s;
482  const char *dir;
483  owl_options opts;
484  GSource *source;
485  GIOChannel *channel;
486
487  argc_copy = argc;
488  argv_copy = g_strdupv(argv);
489
490  setlocale(LC_ALL, "");
491
492  memset(&opts, 0, sizeof opts);
493  opts.load_initial_subs = 1;
494  owl_parse_options(argc, argv, &opts);
495  g.load_initial_subs = opts.load_initial_subs;
496
497  owl_start_curses();
498
499  /* owl global init */
500  owl_global_init(&g);
501  if (opts.debug) owl_global_set_debug_on(&g);
502  if (opts.confdir) owl_global_set_confdir(&g, opts.confdir);
503  owl_function_debugmsg("startup: first available debugging message");
504  owl_global_set_startupargs(&g, argc_copy, argv_copy);
505  g_strfreev(argv_copy);
506  owl_global_set_haveaim(&g);
507
508  owl_register_signal_handlers();
509
510  /* register STDIN dispatch; throw away return, we won't need it */
511  channel = g_io_channel_unix_new(STDIN_FILENO);
512  g_io_add_watch(channel, G_IO_IN | G_IO_HUP | G_IO_ERR, &owl_process_input, &g);
513  g_io_channel_unref(channel);
514  owl_zephyr_initialize();
515
516#if OWL_STDERR_REDIR
517  /* Do this only after we've started curses up... */
518  if (isatty(STDERR_FILENO)) {
519    owl_function_debugmsg("startup: doing stderr redirection");
520    channel = g_io_channel_unix_new(stderr_replace());
521    g_io_add_watch(channel, G_IO_IN | G_IO_HUP | G_IO_ERR, &stderr_redirect_handler, NULL);
522    g_io_channel_unref(channel);
523  }
524#endif
525
526  /* create the owl directory, in case it does not exist */
527  owl_function_debugmsg("startup: creating owl directory, if not present");
528  dir=owl_global_get_confdir(&g);
529  mkdir(dir, S_IRWXU);
530
531  /* set the tty, either from the command line, or by figuring it out */
532  owl_function_debugmsg("startup: setting tty name");
533  if (opts.tty) {
534    owl_global_set_tty(&g, opts.tty);
535  } else {
536    char *tty = owl_util_get_default_tty();
537    owl_global_set_tty(&g, tty);
538    g_free(tty);
539  }
540
541  /* Initialize perl */
542  owl_function_debugmsg("startup: processing config file");
543
544  owl_global_pop_context(&g);
545  owl_global_push_context(&g, OWL_CTX_READCONFIG, NULL, NULL, NULL);
546
547  perlerr=owl_perlconfig_initperl(opts.configfile, &argc, &argv, &env);
548  if (perlerr) {
549    endwin();
550    fprintf(stderr, "Internal perl error: %s\n", perlerr);
551    fflush(stderr);
552    printf("Internal perl error: %s\n", perlerr);
553    fflush(stdout);
554    exit(1);
555  }
556
557  dlerr = owl_closure_init();
558  if(dlerr) {
559    endwin();
560    fprintf(stderr, "Error binding gtk2-perl functions from C: %s\n",dlerr);
561    fflush(stderr);
562    fprintf(stdout, "Error binding gtk2-perl functions from C: %s\n",dlerr);
563    fflush(stdout);
564    exit(1);
565  }
566 
567  owl_global_complete_setup(&g);
568
569  owl_global_setup_default_filters(&g);
570
571  /* set the current view */
572  owl_function_debugmsg("startup: setting the current view");
573  owl_view_create(owl_global_get_current_view(&g), "main",
574                  owl_global_get_filter(&g, "all"),
575                  owl_global_get_style_by_name(&g, "default"));
576
577  /* AIM init */
578  owl_function_debugmsg("startup: doing AIM initialization");
579  owl_aim_init();
580
581  /* execute the startup function in the configfile */
582  owl_function_debugmsg("startup: executing perl startup, if applicable");
583  perlout = owl_perlconfig_execute("BarnOwl::Hooks::_startup();");
584  g_free(perlout);
585
586  /* welcome message */
587  owl_function_debugmsg("startup: creating splash message");
588  owl_function_adminmsg("",
589    "-----------------------------------------------------------------------\n"
590    "Welcome to BarnOwl version " OWL_VERSION_STRING ".\n"
591    "To see a quick introduction, type ':show quickstart'.                  \n"
592    "Press 'h' for on-line help.                                            \n"
593    "                                                                       \n"
594    "BarnOwl is free software. Type ':show license' for more                \n"
595    "information.                                                     ^ ^   \n"
596    "                                                                 OvO   \n"
597    "Please report any bugs or suggestions to bug-barnowl@mit.edu    (   )  \n"
598    "-----------------------------------------------------------------m-m---\n"
599  );
600
601  owl_function_debugmsg("startup: setting context interactive");
602
603  owl_global_pop_context(&g);
604  owl_global_push_context(&g, OWL_CTX_INTERACTIVE|OWL_CTX_RECV, NULL, "recv", NULL);
605
606  /* process the startup file */
607  owl_function_debugmsg("startup: processing startup file");
608  owl_function_source(NULL);
609
610  owl_function_debugmsg("startup: set style for the view: %s", owl_global_get_default_style(&g));
611  s = owl_global_get_style_by_name(&g, owl_global_get_default_style(&g));
612  if(s)
613      owl_view_set_style(owl_global_get_current_view(&g), s);
614  else
615      owl_function_error("No such style: %s", owl_global_get_default_style(&g));
616
617  source = owl_window_redraw_source_new();
618  g_source_attach(source, NULL);
619  g_source_unref(source);
620
621  source = g_source_new(&owl_process_messages_funcs, sizeof(GSource));
622  g_source_attach(source, NULL);
623  g_source_unref(source);
624
625  owl_log_init();
626
627  owl_function_debugmsg("startup: entering main loop");
628  owl_select_run_loop();
629
630  /* Shut down everything. */
631  owl_zephyr_shutdown();
632  owl_signal_shutdown();
633  owl_shutdown_curses();
634  owl_log_shutdown();
635  return 0;
636}
Note: See TracBrowser for help on using the repository browser.