/* This file is part of telegram-client. Telegram-client 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. Telegram-client 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 telegram-client. If not, see . Copyright Vitaly Valtman 2013 */ #define _GNU_SOURCE #include "config.h" #include #include #include #include #include #include #include #include #if (READLINE == GNU) #include #else #include #endif #include #include #include #include #include #ifdef ENABLE_LIBCONFIG #include #endif #include "loop.h" #include "mtproto-client.h" #define PROGNAME "telegram-client" #define VERSION "0.01" #define CONFIG_DIRECTORY ".telegram" #define CONFIG_FILE "config" #define AUTH_KEY_FILE "auth" #define STATE_FILE "state" #define SECRET_CHAT_FILE "secret" #define DOWNLOADS_DIRECTORY "downloads" #define CONFIG_DIRECTORY_MODE 0700 #define DEFAULT_CONFIG_CONTENTS \ "# This is an empty config file\n" \ "# Feel free to put something here\n" char *default_username; char *auth_token; int msg_num_mode; char *config_filename; char *prefix; int test_dc; char *auth_file_name; char *state_file_name; char *secret_chat_file_name; char *downloads_directory; char *config_directory; int binlog_enabled; void set_default_username (const char *s) { if (default_username) { free (default_username); } default_username = strdup (s); } /* {{{ TERMINAL */ tcflag_t old_lflag; cc_t old_vtime; struct termios term; void get_terminal_attributes (void) { if (tcgetattr (STDIN_FILENO, &term) < 0) { perror ("tcgetattr()"); exit (EXIT_FAILURE); } old_lflag = term.c_lflag; old_vtime = term.c_cc[VTIME]; } void set_terminal_attributes (void) { if (tcsetattr (STDIN_FILENO, 0, &term) < 0) { perror ("tcsetattr()"); exit (EXIT_FAILURE); } } /* }}} */ char *get_home_directory (void) { struct passwd *current_passwd; uid_t user_id; setpwent (); user_id = getuid (); while ((current_passwd = getpwent ())) { if (current_passwd->pw_uid == user_id) { return current_passwd->pw_dir; } } return ""; } char *get_config_directory (void) { char *config_directory; int length = strlen (get_home_directory ()) + strlen (CONFIG_DIRECTORY) + 2; config_directory = (char *) calloc (length, sizeof (char)); sprintf (config_directory, "%s/" CONFIG_DIRECTORY, get_home_directory ()); return config_directory; } char *get_config_filename (void) { return config_filename; } char *get_auth_key_filename (void) { return auth_file_name; } char *get_state_filename (void) { return state_file_name; } char *get_secret_chat_filename (void) { return secret_chat_file_name; } char *get_downloads_directory (void) { return downloads_directory; } char *make_full_path (char *s) { if (*s != '/') { char *t = s; assert (asprintf (&s, "%s/%s", get_home_directory (), s) >= 0); free (t); } return s; } void running_for_first_time (void) { if (config_filename) { return; // Do not create custom config file } assert (asprintf (&config_filename, "%s/%s/%s", get_home_directory (), CONFIG_DIRECTORY, CONFIG_FILE) >= 0); config_filename = make_full_path (config_filename); struct stat *config_file_stat = NULL; int config_file_fd; char *config_directory = get_config_directory (); //char *downloads_directory = get_downloads_directory (); if (!mkdir (config_directory, CONFIG_DIRECTORY_MODE)) { printf ("[%s] created\n", config_directory); } // see if config file is there if (stat (config_filename, config_file_stat) != 0) { // config file missing, so touch it config_file_fd = open (config_filename, O_CREAT | O_RDWR, 0600); if (config_file_fd == -1) { perror ("open[config_file]"); exit (EXIT_FAILURE); } if (fchmod (config_file_fd, CONFIG_DIRECTORY_MODE) != 0) { perror ("fchmod[" CONFIG_FILE "]"); exit (EXIT_FAILURE); } if (write (config_file_fd, DEFAULT_CONFIG_CONTENTS, strlen (DEFAULT_CONFIG_CONTENTS)) <= 0) { perror ("write[config_file]"); exit (EXIT_FAILURE); } close (config_file_fd); /*int auth_file_fd = open (get_auth_key_filename (), O_CREAT | O_RDWR, 0600); int x = -1; assert (write (auth_file_fd, &x, 4) == 4); close (auth_file_fd); printf ("[%s] created\n", config_filename);*/ /* create downloads directory */ /*if (mkdir (downloads_directory, 0755) !=0) { perror ("creating download directory"); exit (EXIT_FAILURE); }*/ } } #ifdef ENABLE_LIBCONFIG void parse_config_val (config_t *conf, char **s, char *param_name, const char *default_name, const char *path) { static char buf[1000]; int l = 0; if (prefix) { l = strlen (prefix); memcpy (buf, prefix, l); buf[l ++] = '.'; } *s = 0; const char *r = 0; strcpy (buf + l, param_name); config_lookup_string (conf, buf, &r); if (r) { if (path) { assert (asprintf (s, "%s/%s", path, r) >= 0); } else { *s = strdup (r); } } else { if (path) { assert (asprintf (s, "%s/%s", path, default_name) >= 0); } else { *s = strdup (default_name); } } } void parse_config (void) { config_filename = make_full_path (config_filename); config_t conf; config_init (&conf); if (config_read_file (&conf, config_filename) != CONFIG_TRUE) { fprintf (stderr, "Can not read config '%s': error '%s' on the line %d\n", config_filename, config_error_text (&conf), config_error_line (&conf)); exit (2); } if (!prefix) { config_lookup_string (&conf, "default_profile", (void *)&prefix); } static char buf[1000]; int l = 0; if (prefix) { l = strlen (prefix); memcpy (buf, prefix, l); buf[l ++] = '.'; } test_dc = 0; strcpy (buf + l, "test"); config_lookup_bool (&conf, buf, &test_dc); if (!msg_num_mode) { strcpy (buf + l, "msg_num"); config_lookup_bool (&conf, buf, &msg_num_mode); } parse_config_val (&conf, &config_directory, "config_directory", CONFIG_DIRECTORY, 0); config_directory = make_full_path (config_directory); parse_config_val (&conf, &auth_file_name, "auth_file", AUTH_KEY_FILE, config_directory); parse_config_val (&conf, &state_file_name, "state_file", STATE_FILE, config_directory); parse_config_val (&conf, &secret_chat_file_name, "secret", SECRET_CHAT_FILE, config_directory); parse_config_val (&conf, &downloads_directory, "downloads", DOWNLOADS_DIRECTORY, config_directory); strcpy (buf + l, "binlog_enabled"); config_lookup_bool (&conf, buf, &binlog_enabled); if (!mkdir (config_directory, CONFIG_DIRECTORY_MODE)) { printf ("[%s] created\n", config_directory); } if (!mkdir (downloads_directory, CONFIG_DIRECTORY_MODE)) { printf ("[%s] created\n", downloads_directory); } } #else void parse_config (void) { printf ("libconfig not enabled\n"); } #endif void inner_main (void) { loop (); } void usage (void) { printf ("%s [-u username] [-h] [-k public key name] [-N] [-v]\n", PROGNAME); exit (1); } extern char *rsa_public_key_name; extern int verbosity; extern int default_dc_num; void args_parse (int argc, char **argv) { int opt = 0; while ((opt = getopt (argc, argv, "u:hk:vn:Nc:p:")) != -1) { switch (opt) { case 'u': set_default_username (optarg); break; case 'k': rsa_public_key_name = strdup (optarg); break; case 'v': verbosity ++; break; case 'N': msg_num_mode ++; break; case 'c': config_filename = strdup (optarg); break; case 'p': prefix = strdup (optarg); assert (strlen (prefix) <= 100); break; case 'h': default: usage (); break; } } } void print_backtrace (void) { void *buffer[255]; const int calls = backtrace (buffer, sizeof (buffer) / sizeof (void *)); backtrace_symbols_fd (buffer, calls, 1); exit(EXIT_FAILURE); } void sig_handler (int signum) { set_terminal_attributes (); printf ("Signal %d received\n", signum); print_backtrace (); } int main (int argc, char **argv) { signal (SIGSEGV, sig_handler); signal (SIGABRT, sig_handler); args_parse (argc, argv); printf ( "Telegram-client version " TG_VERSION ", Copyright (C) 2013 Vitaly Valtman\n" "Telegram-client comes with ABSOLUTELY NO WARRANTY; for details type `show_license'.\n" "This is free software, and you are welcome to redistribute it\n" "under certain conditions; type `show_license' for details.\n" ); running_for_first_time (); parse_config (); get_terminal_attributes (); inner_main (); return 0; }