Mercurial > hg
view mcabber/src/main.c @ 1617:9ca672ee884f
Fix previous commit :)
author | Mikael Berthe <mikael@lilotux.net> |
---|---|
date | Sun, 11 Oct 2009 22:47:12 +0200 |
parents | 6db9f403f707 |
children | 9296987856d9 |
line wrap: on
line source
/* * main.c * * Copyright (C) 2005-2009 Mikael Berthe <mikael@lilotux.net> * Parts of this file come from Cabber <cabber@ajmacias.com> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or (at * your option) any later version. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * USA */ #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <string.h> #include <signal.h> #include <termios.h> #include <sys/types.h> #include <sys/wait.h> #include <glib.h> #include <config.h> #include "caps.h" #include "screen.h" #include "settings.h" #include "roster.h" #include "commands.h" #include "histolog.h" #include "hooks.h" #include "utils.h" #include "pgp.h" #include "otr.h" #include "fifo.h" #include "xmpp.h" #ifdef ENABLE_HGCSET # include "hgcset.h" #endif #ifndef WAIT_ANY # define WAIT_ANY -1 #endif static unsigned int terminate_ui; GMainLoop *main_loop = NULL; static struct termios *backup_termios; char *mcabber_version(void) { char *ver; #ifdef HGCSET ver = g_strdup_printf("%s (%s)", PACKAGE_VERSION, HGCSET); #else ver = g_strdup(PACKAGE_VERSION); #endif return ver; } static void mcabber_terminate(const char *msg) { fifo_deinit(); xmpp_disconnect(); scr_TerminateCurses(); // Restore term settings, if needed. if (backup_termios) tcsetattr(fileno(stdin), TCSAFLUSH, backup_termios); if (msg) fprintf(stderr, "%s\n", msg); printf("Bye!\n"); exit(EXIT_SUCCESS); } void sig_handler(int signum) { if (signum == SIGCHLD) { int status; pid_t pid; do { pid = waitpid (WAIT_ANY, &status, WNOHANG); // Check the exit status value if 'eventcmd_checkstatus' is set if (settings_opt_get_int("eventcmd_checkstatus")) { if (pid > 0) { // exit status 2 -> beep if (WIFEXITED(status) && WEXITSTATUS(status) == 2) { scr_Beep(); } } } } while (pid > 0); signal(SIGCHLD, sig_handler); } else if (signum == SIGTERM) { mcabber_terminate("Killed by SIGTERM"); } else if (signum == SIGINT) { mcabber_terminate("Killed by SIGINT"); #ifdef USE_SIGWINCH } else if (signum == SIGWINCH) { ungetch(KEY_RESIZE); #endif } else { scr_LogPrint(LPRINT_LOGNORM, "Caught signal: %d", signum); } } // ask_password(what) // Return the password, or NULL. // The string must be freed after use. static char *ask_password(const char *what) { char *password, *p; size_t passsize = 128; struct termios orig, new; password = g_new0(char, passsize); /* Turn echoing off and fail if we can't. */ if (tcgetattr(fileno(stdin), &orig) != 0) return NULL; backup_termios = &orig; new = orig; new.c_lflag &= ~ECHO; if (tcsetattr(fileno(stdin), TCSAFLUSH, &new) != 0) return NULL; /* Read the password. */ printf("Please enter %s: ", what); if (fgets(password, passsize, stdin) == NULL) return NULL; /* Restore terminal. */ tcsetattr(fileno(stdin), TCSAFLUSH, &orig); printf("\n"); backup_termios = NULL; for (p = (char*)password; *p; p++) ; for ( ; p > (char*)password ; p--) if (*p == '\n' || *p == '\r') *p = 0; return password; } static void credits(void) { const char *v_fmt = "MCabber %s -- Email: mcabber [at] lilotux [dot] net\n"; char *v = mcabber_version(); printf(v_fmt, v); scr_LogPrint(LPRINT_LOGNORM|LPRINT_NOTUTF8, v_fmt, v); g_free(v); } static void compile_options(void) { puts("Installation data directory: " DATA_DIR "\n"); #ifdef HAVE_UNICODE puts("Compiled with unicode support."); #endif #ifdef HAVE_GPGME puts("Compiled with GPG support."); #endif #ifdef MODULES_ENABLE puts ("Compiled with modules support"); #endif #ifdef HAVE_LIBOTR puts("Compiled with OTR support."); #endif #ifdef WITH_ENCHANT puts("Compiled with Enchant support."); #endif #ifdef WITH_ASPELL puts("Compiled with Aspell support."); #endif #ifdef ENABLE_DEBUG puts("Compiled with debugging support."); #endif } static void main_init_pgp(void) { #ifdef HAVE_GPGME const char *pk, *pp; char *typed_passwd = NULL; char *p; bool pgp_invalid = FALSE; bool pgp_agent; int retries; p = getenv("GPG_AGENT_INFO"); pgp_agent = (p && strchr(p, ':')); pk = settings_opt_get("pgp_private_key"); pp = settings_opt_get("pgp_passphrase"); if (settings_opt_get("pgp_passphrase_retries")) retries = settings_opt_get_int("pgp_passphrase_retries"); else retries = 2; if (!pk) { scr_LogPrint(LPRINT_LOGNORM, "WARNING: unknown PGP private key"); pgp_invalid = TRUE; } else if (!(pp || pgp_agent)) { // Request PGP passphrase pp = typed_passwd = ask_password("PGP passphrase"); } gpg_init(pk, pp); // Erase password from the settings array if (pp) { memset((char*)pp, 0, strlen(pp)); if (typed_passwd) g_free(typed_passwd); else settings_set(SETTINGS_TYPE_OPTION, "pgp_passphrase", NULL); } if (!pgp_agent && pk && pp && gpg_test_passphrase()) { // Let's check the pasphrase int i; for (i = 1; retries < 0 || i <= retries; i++) { typed_passwd = ask_password("PGP passphrase"); // Ask again... if (typed_passwd) { gpg_set_passphrase(typed_passwd); memset(typed_passwd, 0, strlen(typed_passwd)); g_free(typed_passwd); } if (!gpg_test_passphrase()) break; // Ok } if (i > retries) pgp_invalid = TRUE; } if (pgp_invalid) scr_LogPrint(LPRINT_LOGNORM, "WARNING: PGP key/pass invalid"); #else /* not HAVE_GPGME */ scr_LogPrint(LPRINT_LOGNORM, "WARNING: not compiled with PGP support"); #endif /* HAVE_GPGME */ } void mcabber_set_terminate_ui(void) { terminate_ui = TRUE; } gboolean mcabber_loop() { keycode kcode; if (terminate_ui) { g_main_loop_quit(main_loop); return FALSE; } scr_DoUpdate(); scr_Getch(&kcode); if (kcode.value != ERR) { process_key(kcode); } else { scr_CheckAutoAway(FALSE); if (update_roster) scr_DrawRoster(); hk_mainloop(); } return TRUE; } int main(int argc, char **argv) { char *configFile = NULL; const char *optstring; int optval, optval2; int ret; credits(); signal(SIGTERM, sig_handler); signal(SIGINT, sig_handler); signal(SIGCHLD, sig_handler); #ifdef USE_SIGWINCH signal(SIGWINCH, sig_handler); #endif signal(SIGPIPE, SIG_IGN); /* Parse command line options */ while (1) { int c = getopt(argc, argv, "hVf:"); if (c == -1) { break; } else switch (c) { case 'h': case '?': printf("Usage: %s [-h|-V|-f mcabberrc_file]\n\n", argv[0]); return (c == 'h' ? 0 : -1); case 'V': compile_options(); return 0; case 'f': configFile = g_strdup(optarg); break; } } if (optind < argc) { fprintf(stderr, "Usage: %s [-h|-V|-f mcabberrc_file]\n\n", argv[0]); return -1; } /* Initialize command system, roster and default key bindings */ cmd_init(); roster_init(); settings_init(); scr_init_bindings(); caps_init(); /* Initialize charset */ scr_InitLocaleCharSet(); /* Parsing config file... */ ret = cfg_read_file(configFile, TRUE); /* free() configFile if it has been allocated during options parsing */ g_free(configFile); /* Leave if there was an error in the config. file */ if (ret == -2) exit(EXIT_FAILURE); optstring = settings_opt_get("tracelog_file"); if (optstring) ut_InitDebug(settings_opt_get_int("tracelog_level"), optstring); /* If no password is stored, we ask for it before entering ncurses mode -- unless the username is unknown. */ if (settings_opt_get("jid") && !settings_opt_get("password")) { const char *p; char *pwd; p = settings_opt_get("server"); if (p) printf("Server: %s\n", p); p = settings_opt_get("jid"); if (p) printf("User JID: %s\n", p); pwd = ask_password("Jabber password"); settings_set(SETTINGS_TYPE_OPTION, "password", pwd); g_free(pwd); } /* Initialize PGP system We do it before ncurses initialization because we may need to request a passphrase. */ if (settings_opt_get_int("pgp")) main_init_pgp(); /* Initialize N-Curses */ scr_LogPrint(LPRINT_DEBUG, "Initializing N-Curses..."); scr_InitCurses(); scr_DrawMainWindow(TRUE); optval = (settings_opt_get_int("logging") > 0); optval2 = (settings_opt_get_int("load_logs") > 0); if (optval || optval2) hlog_enable(optval, settings_opt_get("logging_dir"), optval2); #if defined(WITH_ENCHANT) || defined(WITH_ASPELL) /* Initialize spelling */ if (settings_opt_get_int("spell_enable")) { spellcheck_init(); } #endif optstring = settings_opt_get("events_command"); if (optstring) hk_ext_cmd_init(optstring); if (settings_opt_get_int("hide_offline_buddies") > 0) { // XXX Deprecated scr_RosterDisplay("ofdna"); scr_LogPrint(LPRINT_LOGNORM, "* Warning: 'hide_offline_buddies' is deprecated."); } else { optstring = settings_opt_get("roster_display_filter"); if (optstring) scr_RosterDisplay(optstring); // Empty filter isn't allowed... if (!buddylist_get_filter()) scr_RosterDisplay("*"); } chatstates_disabled = settings_opt_get_int("disable_chatstates"); /* Initialize FIFO named pipe */ fifo_init(settings_opt_get("fifo_name")); /* Load previous roster state */ hlog_load_state(); main_loop = g_main_loop_new(NULL, TRUE); if (ret < 0) { scr_LogPrint(LPRINT_NORMAL, "No configuration file has been found."); scr_ShowBuddyWindow(); } else { /* Connection */ xmpp_connect(); } scr_LogPrint(LPRINT_DEBUG, "Entering into main loop..."); g_timeout_add(10, mcabber_loop, NULL); g_main_loop_run(main_loop); scr_TerminateCurses(); #ifdef MODULES_ENABLE cmd_deinit(); #endif fifo_deinit(); #ifdef HAVE_LIBOTR otr_terminate(); #endif xmpp_disconnect(); #ifdef HAVE_GPGME gpg_terminate(); #endif #if defined(WITH_ENCHANT) || defined(WITH_ASPELL) /* Deinitialize spelling */ if (settings_opt_get_int("spell_enable")) spellcheck_deinit(); #endif /* Save pending message state */ hlog_save_state(); caps_free(); printf("\n\nThanks for using mcabber!\n"); return 0; } /* vim: set expandtab cindent cinoptions=>2\:2(0: For Vim users... */