* Copyright (C) Jaakko Heinonen
*/
+#include <errno.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
#include <string.h>
#include <unistd.h>
-#include <stdlib.h>
#include <sys/stat.h>
-#include <signal.h>
-#include <fcntl.h>
-#include <errno.h>
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#if defined(HAVE_LOCALE_H) && defined(HAVE_SETLOCALE)
# include <locale.h>
#endif
+#include <assert.h>
#include "abook.h"
#include "ui.h"
#include "database.h"
#include "options.h"
#include "getname.h"
#include "getopt.h"
-#include <assert.h>
+#include "xmalloc.h"
static void init_abook();
static void quit_abook_sig(int i);
free(dir);
}
+static void
+xmalloc_error_handler(int err)
+{
+ if(is_ui_initialized())
+ quit_abook(QUIT_SAVE);
+
+ fprintf(stderr, "Memory allocation failure: %s\n", strerror(err));
+ exit(EXIT_FAILURE);
+}
+
static void
init_abook()
{
set_filenames();
check_abook_directory();
init_opts();
- load_opts(rcfile);
+ if(load_opts(rcfile) > 0) {
+ printf("Press enter to continue...\n");
+ fgetc(stdin);
+ }
signal(SIGKILL, quit_abook_sig);
signal(SIGTERM, quit_abook_sig);
}
void
-quit_abook()
+quit_abook(int save_db)
{
- if( opt_get_bool(BOOL_AUTOSAVE) )
- save_database();
- else if( statusline_ask_boolean("Save database", TRUE) )
- save_database();
+ if(save_db) {
+ if(opt_get_bool(BOOL_AUTOSAVE))
+ save_database();
+ else if(statusline_ask_boolean("Save database", TRUE))
+ save_database();
+ } else if(!statusline_ask_boolean("Quit without saving", FALSE))
+ return;
free_opts();
close_database();
static void
quit_abook_sig(int i)
{
- quit_abook();
+ quit_abook(QUIT_SAVE);
}
int
main(int argc, char **argv)
{
#if defined(HAVE_SETLOCALE) && defined(HAVE_LOCALE_H)
- setlocale(LC_ALL, "" );
+ setlocale(LC_ALL, "");
#endif
+ xmalloc_set_error_handler(xmalloc_error_handler);
parse_command_line(argc, argv);
get_commands();
- quit_abook();
+ quit_abook(QUIT_SAVE);
return 0;
}
static void
free_filenames()
{
- my_free(rcfile);
- my_free(datafile);
+ xfree(rcfile);
+ xfree(datafile);
}
putchar('\n');
while(i >= 0) {
muttq_print_item(stdout, i);
- i = find_item(str, i+1, search_fields);
+ i = find_item(str, i + 1, search_fields);
}
}
}
}
- cmd = strconcat(mutt_command, " \'", mailstr,
- "\'", NULL);
+ cmd = strconcat(mutt_command, " \'", mailstr, "\'", NULL);
free(mailstr);
#ifdef DEBUG
fprintf(stderr, "cmd: %s\n", cmd);
ui_init_curses();
}
-void *
-abook_malloc(size_t size)
-{
- void *ptr;
-
- if ( (ptr = malloc(size)) == NULL ) {
- if( is_ui_initialized() )
- quit_abook();
- perror("malloc() failed");
- exit(1);
- }
-
- return ptr;
-}
-
-void *
-abook_realloc(void *ptr, size_t size)
-{
- ptr = realloc(ptr, size);
-
- if( size == 0 )
- return NULL;
-
- if( ptr == NULL ) {
- if( is_ui_initialized() )
- quit_abook();
- perror("realloc() failed");
- exit(1);
- }
-
- return ptr;
-}
-
FILE *
abook_fopen (const char *path, const char *mode)
{
init_opts();
load_opts(rcfile);
- switch( import_file(srcformat, srcfile) ) {
+ switch(import_file(srcformat, srcfile)) {
case -1:
fprintf(stderr,
"input format %s not supported\n", srcformat);
}
if(!ret)
- switch( export_file(dstformat, dstfile) ) {
+ switch(export_file(dstformat, dstfile)) {
case -1:
fprintf(stderr,
"output format %s not supported\n",
{
set_filenames();
atexit(free_filenames);
+ check_abook_directory();
init_opts();
load_opts(rcfile);
atexit(free_opts);
{
list_item item;
+ if(opt_get_bool(BOOL_ADD_EMAIL_PREVENT_DUPLICATES)) {
+ int search_fields[] = { EMAIL, -1 };
+ if(find_item(email, 0, search_fields) >= 0) {
+ if(!quiet)
+ printf("Address %s already in addressbook\n",
+ email);
+ return 0;
+ }
+ }
+
if(!quiet) {
FILE *in = fopen("/dev/tty", "r");
char c;
"you may want to use --add-email-quiet\n");
exit(1);
}
- printf("Add ``%s <%s>'' to %s ? (y/n)\n",
- name,
- email,
- datafile
- );
+
do {
- c = fgetc(in);
+ printf("Add ``%s <%s>'' to %s ? (y/n)\n",
+ name,
+ email,
+ datafile);
+ c = getc(in);
if(c == 'n' || c == 'N') {
fclose(in);
return 0;
char *name = NULL, *email = NULL;
struct stat s;
- if( (fstat(fileno(stdin), &s)) == -1 || S_ISDIR(s.st_mode)) {
+ if( (fstat(fileno(stdin), &s)) == -1 || S_ISDIR(s.st_mode) ) {
fprintf(stderr, "stdin is a directory or cannot stat stdin\n");
exit(1);
}
getname(line, &name, &email);
add_email_count += add_email_add_item(quiet,
name, email);
- my_free(name);
- my_free(email);
+ xfree(name);
+ xfree(email);
}
- my_free(line);
+ xfree(line);
} while( !feof(stdin) );
quit_add_email();