source: owl.c @ 6e90da8

release-1.10release-1.7release-1.8release-1.9
Last change on this file since 6e90da8 was 07b59ea, checked in by David Benjamin <davidben@mit.edu>, 14 years ago
Maintain the cursor location with the context stack
  • Property mode set to 100644
File size: 18.4 KB
Line 
1/*  Copyright (c) 2006-2010 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 <stdio.h>
10#include <unistd.h>
11#include <getopt.h>
12#include <stdlib.h>
13#include <string.h>
14#include <signal.h>
15#include <time.h>
16#include <sys/param.h>
17#include <sys/types.h>
18#include <sys/time.h>
19#include <termios.h>
20#include <sys/stat.h>
21#include <locale.h>
22#include "owl.h"
23#include "globalnotifier.h"
24
25
26#if OWL_STDERR_REDIR
27#ifdef HAVE_SYS_IOCTL_H
28#include <sys/ioctl.h>
29#endif
30#ifdef HAVE_SYS_FILIO_H
31#include <sys/filio.h>
32#endif
33int stderr_replace(void);
34#endif
35
36#define STDIN 0
37
38owl_global g;
39
40typedef struct _owl_options {
41  bool load_initial_subs;
42  char *configfile;
43  char *tty;
44  char *confdir;
45  bool debug;
46  bool rm_debug;
47} owl_options;
48
49void usage(void)
50{
51  fprintf(stderr, "Barnowl version %s\n", OWL_VERSION_STRING);
52  fprintf(stderr, "Usage: barnowl [-n] [-d] [-D] [-v] [-h] [-c <configfile>] [-s <confdir>] [-t <ttyname>]\n");
53  fprintf(stderr, "  -n,--no-subs        don't load zephyr subscriptions\n");
54  fprintf(stderr, "  -d,--debug          enable debugging\n");
55  fprintf(stderr, "  -D,--remove-debug   enable debugging and delete previous debug file\n");
56  fprintf(stderr, "  -v,--version        print the Barnowl version number and exit\n");
57  fprintf(stderr, "  -h,--help           print this help message\n");
58  fprintf(stderr, "  -c,--config-file    specify an alternate config file\n");
59  fprintf(stderr, "  -s,--config-dir     specify an alternate config dir (default ~/.owl)\n");
60  fprintf(stderr, "  -t,--tty            set the tty name\n");
61}
62
63/* TODO: free owl_options after init is done? */
64void owl_parse_options(int argc, char *argv[], owl_options *opts) {
65  static const struct option long_options[] = {
66    { "no-subs",         0, 0, 'n' },
67    { "config-file",     1, 0, 'c' },
68    { "config-dir",      1, 0, 's' },
69    { "tty",             1, 0, 't' },
70    { "debug",           0, 0, 'd' },
71    { "remove-debug",    0, 0, 'D' },
72    { "version",         0, 0, 'v' },
73    { "help",            0, 0, 'h' },
74    { NULL, 0, NULL, 0}
75  };
76  char c;
77
78  while((c = getopt_long(argc, argv, "nc:t:s:dDvh",
79                         long_options, NULL)) != -1) {
80    switch(c) {
81    case 'n':
82      opts->load_initial_subs = 0;
83      break;
84    case 'c':
85      opts->configfile = owl_strdup(optarg);
86      break;
87    case 's':
88      opts->confdir = owl_strdup(optarg);
89      break;
90    case 't':
91      opts->tty = owl_strdup(optarg);
92      break;
93    case 'D':
94      opts->rm_debug = 1;
95      /* fallthrough */
96    case 'd':
97      opts->debug = 1;
98      break;
99    case 'v':
100      printf("This is barnowl version %s\n", OWL_VERSION_STRING);
101      exit(0);
102    case 'h':
103    default:
104      usage();
105      exit(1);
106    }
107  }
108}
109
110void owl_start_color(void) {
111  start_color();
112#ifdef HAVE_USE_DEFAULT_COLORS
113  use_default_colors();
114#endif
115
116  /* define simple color pairs */
117  if (has_colors() && COLOR_PAIRS>=8) {
118    int bg = COLOR_BLACK;
119#ifdef HAVE_USE_DEFAULT_COLORS
120    bg = -1;
121#endif
122    init_pair(OWL_COLOR_BLACK,   COLOR_BLACK,   bg);
123    init_pair(OWL_COLOR_RED,     COLOR_RED,     bg);
124    init_pair(OWL_COLOR_GREEN,   COLOR_GREEN,   bg);
125    init_pair(OWL_COLOR_YELLOW,  COLOR_YELLOW,  bg);
126    init_pair(OWL_COLOR_BLUE,    COLOR_BLUE,    bg);
127    init_pair(OWL_COLOR_MAGENTA, COLOR_MAGENTA, bg);
128    init_pair(OWL_COLOR_CYAN,    COLOR_CYAN,    bg);
129    init_pair(OWL_COLOR_WHITE,   COLOR_WHITE,   bg);
130  }
131}
132
133void owl_start_curses(void) {
134  struct termios tio;
135  /* save initial terminal settings */
136  tcgetattr(0, owl_global_get_startup_tio(&g));
137
138  tcgetattr(0, &tio);
139  tio.c_iflag &= ~(ISTRIP|IEXTEN);
140  tio.c_cc[VQUIT] = 0;
141  tio.c_cc[VSUSP] = 0;
142  tcsetattr(0, TCSAFLUSH, &tio);
143
144  /* screen init */
145  initscr();
146  cbreak();
147  noecho();
148
149  owl_start_color();
150}
151
152static void owl_setup_default_filters(void)
153{
154  int i;
155  static const struct {
156    const char *name;
157    const char *desc;
158  } filters[] = {
159    { "personal",
160      "isprivate ^true$ and ( not type ^zephyr$ or ( class ^message  ) )" },
161    { "trash",
162      "class ^mail$ or opcode ^ping$ or type ^admin$ or ( not login ^none$ )" },
163    { "wordwrap", "not ( type ^admin$ or type ^zephyr$ )" },
164    { "ping", "opcode ^ping$" },
165    { "auto", "opcode ^auto$" },
166    { "login", "not login ^none$" },
167    { "reply-lockout", "class ^noc or class ^mail$" },
168    { "out", "direction ^out$" },
169    { "aim", "type ^aim$" },
170    { "zephyr", "type ^zephyr$" },
171    { "none", "false" },
172    { "all", "true" },
173    { NULL, NULL }
174  };
175
176  owl_function_debugmsg("startup: creating default filters");
177
178  for (i = 0; filters[i].name != NULL; i++)
179    owl_global_add_filter(&g, owl_filter_new_fromstring(filters[i].name,
180                                                        filters[i].desc));
181}
182
183/*
184 * Process a new message passed to us on the message queue from some
185 * protocol. This includes adding it to the message list, updating the
186 * view and scrolling if appropriate, logging it, and so on.
187 *
188 * Either a pointer is kept to the message internally, or it is freed
189 * if unneeded. The caller no longer ``owns'' the message's memory.
190 *
191 * Returns 1 if the message was added to the message list, and 0 if it
192 * was ignored due to user settings or otherwise.
193 */
194int owl_process_message(owl_message *m) {
195  const owl_filter *f;
196  /* if this message it on the puntlist, nuke it and continue */
197  if (owl_global_message_is_puntable(&g, m)) {
198    owl_message_delete(m);
199    return 0;
200  }
201
202  /*  login or logout that should be ignored? */
203  if (owl_global_is_ignorelogins(&g)
204      && owl_message_is_loginout(m)) {
205    owl_message_delete(m);
206    return 0;
207  }
208
209  if (!owl_global_is_displayoutgoing(&g)
210      && owl_message_is_direction_out(m)) {
211    owl_message_delete(m);
212    return 0;
213  }
214
215  /* add it to the global list */
216  owl_messagelist_append_element(owl_global_get_msglist(&g), m);
217  /* add it to any necessary views; right now there's only the current view */
218  owl_view_consider_message(owl_global_get_current_view(&g), m);
219
220  if(owl_message_is_direction_in(m)) {
221    /* let perl know about it*/
222    owl_perlconfig_getmsg(m, NULL);
223
224    /* do we need to autoreply? */
225    if (owl_global_is_zaway(&g) && !owl_message_get_attribute_value(m, "isauto")) {
226      if (owl_message_is_type_zephyr(m)) {
227        owl_zephyr_zaway(m);
228      } else if (owl_message_is_type_aim(m)) {
229        if (owl_message_is_private(m)) {
230          owl_function_send_aimawymsg(owl_message_get_sender(m), owl_global_get_zaway_msg(&g));
231        }
232      }
233    }
234
235    /* ring the bell if it's a personal */
236    if (!strcmp(owl_global_get_personalbell(&g), "on")) {
237      if (!owl_message_is_loginout(m) &&
238          !owl_message_is_mail(m) &&
239          owl_message_is_personal(m)) {
240        owl_function_beep();
241      }
242    } else if (!strcmp(owl_global_get_personalbell(&g), "off")) {
243      /* do nothing */
244    } else {
245      f=owl_global_get_filter(&g, owl_global_get_personalbell(&g));
246      if (f && owl_filter_message_match(f, m)) {
247        owl_function_beep();
248      }
249    }
250
251    /* if it matches the alert filter, do the alert action */
252    f=owl_global_get_filter(&g, owl_global_get_alert_filter(&g));
253    if (f && owl_filter_message_match(f, m)) {
254      owl_function_command(owl_global_get_alert_action(&g));
255    }
256
257    /* if it's a zephyr login or logout, update the zbuddylist */
258    if (owl_message_is_type_zephyr(m) && owl_message_is_loginout(m)) {
259      if (owl_message_is_login(m)) {
260        owl_zbuddylist_adduser(owl_global_get_zephyr_buddylist(&g), owl_message_get_sender(m));
261      } else if (owl_message_is_logout(m)) {
262        owl_zbuddylist_deluser(owl_global_get_zephyr_buddylist(&g), owl_message_get_sender(m));
263      } else {
264        owl_function_error("Internal error: received login notice that is neither login nor logout");
265      }
266    }
267  }
268
269  /* let perl know about it */
270  owl_perlconfig_newmsg(m, NULL);
271  /* log the message if we need to */
272  owl_log_message(m);
273  /* emit a signal for listeners */
274  owl_global_notifier_emit_message_received(g.gn, m);
275
276  return 1;
277}
278
279/*
280 * Process any new messages we have waiting in the message queue.
281 * Returns 1 if any messages were added to the message list, and 0 otherwise.
282 */
283int owl_process_messages(owl_ps_action *d, void *p)
284{
285  int newmsgs=0;
286  int followlast = owl_global_should_followlast(&g);
287  owl_message *m;
288
289  /* Grab incoming messages. */
290  while (owl_global_messagequeue_pending(&g)) {
291    m = owl_global_messagequeue_popmsg(&g);
292    if (owl_process_message(m))
293      newmsgs = 1;
294  }
295
296  if (newmsgs) {
297    /* follow the last message if we're supposed to */
298    if (followlast)
299      owl_function_lastmsg_noredisplay();
300
301    /* do the newmsgproc thing */
302    owl_function_do_newmsgproc();
303
304    /* redisplay if necessary */
305    /* this should be optimized to not run if the new messages won't be displayed */
306    owl_mainwin_redisplay(owl_global_get_mainwin(&g));
307  }
308  return newmsgs;
309}
310
311void owl_process_input(const owl_io_dispatch *d, void *data)
312{
313  owl_input j;
314
315  while (1) {
316    j.ch = wgetch(g.input_pad);
317    if (j.ch == ERR) return;
318
319    j.uch = '\0';
320    if (j.ch >= KEY_MIN && j.ch <= KEY_MAX) {
321      /* This is a curses control character. */
322    }
323    else if (j.ch > 0x7f && j.ch < 0xfe) {
324      /* Pull in a full utf-8 character. */
325      int bytes, i;
326      char utf8buf[7];
327      memset(utf8buf, '\0', 7);
328     
329      utf8buf[0] = j.ch;
330     
331      if ((j.ch & 0xc0) && (~j.ch & 0x20)) bytes = 2;
332      else if ((j.ch & 0xe0) && (~j.ch & 0x10)) bytes = 3;
333      else if ((j.ch & 0xf0) && (~j.ch & 0x08)) bytes = 4;
334      else if ((j.ch & 0xf8) && (~j.ch & 0x04)) bytes = 5;
335      else if ((j.ch & 0xfc) && (~j.ch & 0x02)) bytes = 6;
336      else bytes = 1;
337     
338      for (i = 1; i < bytes; i++) {
339        int tmp = wgetch(g.input_pad);
340        /* If what we got was not a byte, or not a continuation byte */
341        if (tmp > 0xff || !(tmp & 0x80 && ~tmp & 0x40)) {
342          /* ill-formed UTF-8 code unit subsequence, put back the
343             char we just got. */
344          ungetch(tmp);
345          j.ch = ERR;
346          break;
347        }
348        utf8buf[i] = tmp;
349      }
350     
351      if (j.ch != ERR) {
352        if (g_utf8_validate(utf8buf, -1, NULL)) {
353          j.uch = g_utf8_get_char(utf8buf);
354        }
355        else {
356          j.ch = ERR;
357        }
358      }
359    }
360    else if (j.ch <= 0x7f) {
361      j.uch = j.ch;
362    }
363
364    owl_process_input_char(j);
365  }
366}
367
368void sig_handler(int sig, siginfo_t *si, void *data)
369{
370  if (sig==SIGWINCH) {
371    /* we can't inturrupt a malloc here, so it just sets a flag
372     * schedulding a resize for later
373     */
374    owl_function_resize();
375  } else if (sig==SIGPIPE || sig==SIGCHLD) {
376    /* Set a flag and some info that we got the sigpipe
377     * so we can record that we got it and why... */
378    owl_global_set_errsignal(&g, sig, si);
379  } else if (sig==SIGTERM || sig==SIGHUP) {
380    owl_function_quit();
381  }
382}
383
384void sigint_handler(int sig, siginfo_t *si, void *data)
385{
386  owl_global_set_interrupted(&g);
387}
388
389void owl_register_signal_handlers(void) {
390  struct sigaction sigact;
391
392  /* signal handler */
393  /*sigact.sa_handler=sig_handler;*/
394  sigact.sa_sigaction=sig_handler;
395  sigemptyset(&sigact.sa_mask);
396  sigact.sa_flags=SA_SIGINFO;
397  sigaction(SIGWINCH, &sigact, NULL);
398  sigaction(SIGALRM, &sigact, NULL);
399  sigaction(SIGPIPE, &sigact, NULL);
400  sigaction(SIGTERM, &sigact, NULL);
401  sigaction(SIGHUP, &sigact, NULL);
402
403  sigact.sa_sigaction=sigint_handler;
404  sigaction(SIGINT, &sigact, NULL);
405}
406
407#if OWL_STDERR_REDIR
408
409/* Replaces stderr with a pipe so that we can read from it.
410 * Returns the fd of the pipe from which stderr can be read. */
411int stderr_replace(void)
412{
413  int pipefds[2];
414  if (0 != pipe(pipefds)) {
415    perror("pipe");
416    owl_function_debugmsg("stderr_replace: pipe FAILED\n");
417    return -1;
418  }
419    owl_function_debugmsg("stderr_replace: pipe: %d,%d\n", pipefds[0], pipefds[1]);
420  if (-1 == dup2(pipefds[1], 2 /*stderr*/)) {
421    owl_function_debugmsg("stderr_replace: dup2 FAILED (%s)\n", strerror(errno));
422    perror("dup2");
423    return -1;
424  }
425  return pipefds[0];
426}
427
428/* Sends stderr (read from rfd) messages to the error console */
429void stderr_redirect_handler(const owl_io_dispatch *d, void *data)
430{
431  int navail, bread;
432  char buf[4096];
433  int rfd = d->fd;
434  char *err;
435
436  if (rfd<0) return;
437  if (-1 == ioctl(rfd, FIONREAD, &navail)) {
438    return;
439  }
440  /*owl_function_debugmsg("stderr_redirect: navail = %d\n", navail);*/
441  if (navail <= 0) return;
442  if (navail > sizeof(buf)-1) {
443    navail = sizeof(buf)-1;
444  }
445  bread = read(rfd, buf, navail);
446  if (buf[navail-1] != '\0') {
447    buf[navail] = '\0';
448  }
449
450  err = owl_sprintf("[stderr]\n%s", buf);
451
452  owl_function_log_err(err);
453  owl_free(err);
454}
455
456#endif /* OWL_STDERR_REDIR */
457
458static int owl_refresh_pre_select_action(owl_ps_action *a, void *data)
459{
460  /* if a resize has been scheduled, deal with it */
461  owl_global_check_resize(&g);
462
463  /* update the terminal if we need to */
464  if (owl_global_is_needrefresh(&g)) {
465    /* Redraw the screen */
466    owl_window_redraw_scheduled();
467    doupdate();
468    owl_global_set_noneedrefresh(&g);
469  }
470  return 0;
471}
472
473
474int main(int argc, char **argv, char **env)
475{
476  int argcsave;
477  const char *const *argvsave;
478  char *perlout, *perlerr;
479  const owl_style *s;
480  const char *dir;
481  owl_options opts;
482
483  if (!GLIB_CHECK_VERSION (2, 12, 0))
484    g_error ("GLib version 2.12.0 or above is needed.");
485
486  argcsave=argc;
487  argvsave=strs(argv);
488
489  setlocale(LC_ALL, "");
490
491  g_type_init ();
492
493  memset(&opts, 0, sizeof opts);
494  opts.load_initial_subs = 1;
495  owl_parse_options(argc, argv, &opts);
496  g.load_initial_subs = opts.load_initial_subs;
497
498  owl_function_debugmsg("startup: Finished parsing arguments");
499
500  owl_register_signal_handlers();
501  owl_start_curses();
502
503  /* owl global init */
504  owl_global_init(&g);
505  if (opts.rm_debug) unlink(OWL_DEBUG_FILE);
506  if (opts.debug) owl_global_set_debug_on(&g);
507  if (opts.confdir) owl_global_set_confdir(&g, opts.confdir);
508  owl_function_debugmsg("startup: first available debugging message");
509  owl_global_set_startupargs(&g, argcsave, argvsave);
510  owl_global_set_haveaim(&g);
511
512  /* register STDIN dispatch; throw away return, we won't need it */
513  owl_select_add_io_dispatch(STDIN, OWL_IO_READ, &owl_process_input, NULL, NULL);
514  owl_zephyr_initialize();
515
516#if OWL_STDERR_REDIR
517  /* Do this only after we've started curses up... */
518  owl_function_debugmsg("startup: doing stderr redirection");
519  owl_select_add_io_dispatch(stderr_replace(), OWL_IO_READ, &stderr_redirect_handler, NULL, NULL);
520#endif
521
522  /* create the owl directory, in case it does not exist */
523  owl_function_debugmsg("startup: creating owl directory, if not present");
524  dir=owl_global_get_confdir(&g);
525  mkdir(dir, S_IRWXU);
526
527  /* set the tty, either from the command line, or by figuring it out */
528  owl_function_debugmsg("startup: setting tty name");
529  if (opts.tty) {
530    owl_global_set_tty(&g, opts.tty);
531  } else {
532    char *tty = owl_util_get_default_tty();
533    owl_global_set_tty(&g, tty);
534    owl_free(tty);
535  }
536
537  /* Initialize perl */
538  owl_function_debugmsg("startup: processing config file");
539
540  owl_global_pop_context(&g);
541  owl_global_push_context(&g, OWL_CTX_READCONFIG, NULL, NULL, NULL);
542
543  perlerr=owl_perlconfig_initperl(opts.configfile, &argc, &argv, &env);
544  if (perlerr) {
545    endwin();
546    fprintf(stderr, "Internal perl error: %s\n", perlerr);
547    fflush(stderr);
548    printf("Internal perl error: %s\n", perlerr);
549    fflush(stdout);
550    exit(1);
551  }
552
553  owl_global_complete_setup(&g);
554
555  owl_setup_default_filters();
556
557  /* set the current view */
558  owl_function_debugmsg("startup: setting the current view");
559  owl_view_create(owl_global_get_current_view(&g), "main",
560                  owl_global_get_filter(&g, "all"),
561                  owl_global_get_style_by_name(&g, "default"));
562
563  /* AIM init */
564  owl_function_debugmsg("startup: doing AIM initialization");
565  owl_aim_init();
566
567  /* execute the startup function in the configfile */
568  owl_function_debugmsg("startup: executing perl startup, if applicable");
569  perlout = owl_perlconfig_execute("BarnOwl::Hooks::_startup();");
570  if (perlout) owl_free(perlout);
571
572  /* welcome message */
573  owl_function_debugmsg("startup: creating splash message");
574  owl_function_adminmsg("",
575    "-----------------------------------------------------------------------\n"
576    "Welcome to barnowl version " OWL_VERSION_STRING ".  Press 'h' for on-line help.\n"
577    "To see a quick introduction, type ':show quickstart'.                  \n"
578    "                                                                       \n"
579    "BarnOwl is free software. Type ':show license' for more                \n"
580    "information.                                                     ^ ^   \n"
581    "                                                                 OvO   \n"
582    "Please report any bugs or suggestions to bug-barnowl@mit.edu    (   )  \n"
583    "-----------------------------------------------------------------m-m---\n"
584  );
585
586  /* process the startup file */
587  owl_function_debugmsg("startup: processing startup file");
588  owl_function_source(NULL);
589
590  /* Set the default style */
591  owl_function_debugmsg("startup: setting startup and default style");
592  if (0 != strcmp(owl_global_get_default_style(&g), "__unspecified__")) {
593    /* the style was set by the user: leave it alone */
594  } else {
595    owl_global_set_default_style(&g, "default");
596  }
597
598  owl_function_debugmsg("startup: set style for the view: %s", owl_global_get_default_style(&g));
599  s = owl_global_get_style_by_name(&g, owl_global_get_default_style(&g));
600  if(s)
601      owl_view_set_style(owl_global_get_current_view(&g), s);
602  else
603      owl_function_error("No such style: %s", owl_global_get_default_style(&g));
604
605  owl_function_debugmsg("startup: setting context interactive");
606
607  owl_global_pop_context(&g);
608  owl_global_push_context(&g, OWL_CTX_READCONFIG|OWL_CTX_RECV, NULL, "recv", NULL);
609
610  owl_select_add_pre_select_action(owl_refresh_pre_select_action, NULL, NULL);
611  owl_select_add_pre_select_action(owl_process_messages, NULL, NULL);
612
613  owl_function_debugmsg("startup: entering main loop");
614  /* main loop */
615  while (1) {
616    /* select on FDs we know about. */
617    owl_select();
618
619    /* Log any error signals */
620    {
621      siginfo_t si;
622      int signum;
623      if ((signum = owl_global_get_errsignal_and_clear(&g, &si)) > 0) {
624        owl_function_error("Got unexpected signal: %d %s  (code: %d band: %ld  errno: %d)",
625                           signum, signum==SIGPIPE?"SIGPIPE":"SIG????",
626                           si.si_code, si.si_band, si.si_errno);
627      }
628    }
629
630  }
631}
Note: See TracBrowser for help on using the repository browser.