int main(int argc, char **argv)
{
token_t t;
fpo = stdout;
mbox_mode = true; /* to allow multiple messages */
process_arglist(argc, argv);
process_config_files(false, longopts_bogolexer);
if (encoding == E_UNKNOWN)
encoding = E_DEFAULT;
textblock_init();
if (!passthrough)
{
if (quiet)
fprintf(fpo, "quiet mode.\n");
else
fprintf(fpo, "normal mode.\n");
}
bogoreader_init(argc, (const char * const *) argv);
while ((*reader_more)()) {
word_t token;
lexer_init();
while ((t = get_token( &token )) != NONE)
{
count += 1;
if (passthrough) {
fprintf(fpo, "%s\n", token.u.text);
}
else if (!quiet)
fprintf(fpo, "get_token: %d \"%s\"\n", (int)t, token.u.text);
}
}
if ( !passthrough )
fprintf(fpo, "%d tokens read.\n", count);
/* cleanup storage */
token_cleanup();
mime_cleanup();
textblock_free();
MEMDISPLAY;
return 0;
}
static int process_arglist(int argc, char **argv)
{
int count = 1;
bulk_mode = B_CMDLINE;
#ifdef __EMX__
_response (&argc, &argv); /* expand response files (@filename) */
_wildcard (&argc, &argv); /* expand wildcards (*.*) */
#endif
#define OPTIONS ":-:c:Cd:DeEM:n:qr:s:tT:vVx:"
while (1)
{
int option;
int option_index = 0;
const char *val;
option = getopt_long_chk(argc, argv, OPTIONS,
longopts_bogotune, &option_index);
if (option == -1)
break;
val = optarg;
process_arg(option, NULL, val, PR_NONE, PASS_1_CLI);
}
if (ds_flag == DS_NONE) /* default is "wordlist on disk" */
ds_flag = DS_DSK;
if (ds_flag == DS_ERR) {
fprintf(stderr, "Only one '-d dir' or '-D' option is allowed.\n");
exit(EX_ERROR);
}
if (bogolex_file == NULL &&
(spam_files->count == 0 || ham_files->count == 0)) {
fprintf(stderr,
"Bogotune needs both non-spam and spam message sets for its parameter testing.\n");
exit(EX_ERROR);
}
if (!suppress_config_file)
process_config_files(false, longopts_bogotune);
return count;
}
int main(int argc, char *argv[])
{
ex_t rc = EX_OK;
bfpath *bfp;
bfpath_mode mode;
fBogoutil = true;
signal_setup(); /* setup to catch signals */
atexit(bf_exit);
progtype = build_progtype(progname, DB_TYPE);
set_today(); /* compute current date for token age */
process_arglist(argc, argv);
process_config_files(false, longopts_bogoutil); /* need to read lock sizes */
/* Extra or missing parameters */
if (flag != M_WORD && flag != M_LIST_LOGFILES && argc != optind) {
fprintf(stderr, "Missing or extraneous argument.\n");
usage(stderr);
exit(EX_ERROR);
}
bfp = bfpath_create(ds_file);
if (bogohome == NULL)
set_bogohome( "." ); /* set default */
bfpath_set_bogohome(bfp);
mode = get_mode(flag);
if (bfpath_check_mode(bfp, mode)) {
if (bfp->isdir)
bfpath_set_filename(bfp, WORDLIST);
}
if (!bfpath_check_mode(bfp, mode)) {
fprintf(stderr, "Can't open wordlist '%s'\n", bfp->filepath);
exit(EX_ERROR);
}
errno = 0; /* clear error status */
switch (flag) {
case M_RECOVER:
ds_init(bfp);
rc = ds_recover(bfp, false);
break;
case M_CRECOVER:
ds_init(bfp);
rc = ds_recover(bfp, true);
break;
case M_CHECKPOINT:
ds_init(bfp);
rc = ds_checkpoint(bfp);
break;
case M_LIST_LOGFILES:
dsm_init(bfp);
rc = ds_list_logfiles(bfp, argc - optind, argv + optind);
break;
case M_PURGELOGS:
ds_init(bfp);
rc = ds_purgelogs(bfp);
break;
case M_REMOVEENV:
dsm_init(bfp);
rc = ds_remove(bfp);
break;
case M_VERIFY:
dsm_init(bfp);
rc = ds_verify(bfp);
break;
case M_LEAFPAGES:
{
u_int32_t c;
dsm_init(bfp);
c = ds_leafpages(bfp);
if (c == 0xffffffff) {
fprintf(stderr, "%s: error getting leaf page count.\n", ds_file);
rc = EX_ERROR;
} else if (c == 0) {
puts("UNKNOWN");
} else {
printf("%lu\n", (unsigned long)c);
}
}
break;
case M_PAGESIZE:
{
u_int32_t s;
dsm_init(bfp);
s = ds_pagesize(bfp);
if (s == 0xffffffff) {
fprintf(stderr, "%s: error getting page size.\n", ds_file);
} else if (s == 0) {
puts("UNKNOWN");
} else {
printf("%lu\n", (unsigned long)s);
}
}
break;
case M_DUMP:
rc = dump_wordlist(bfp);
break;
case M_LOAD:
rc = load_wordlist(bfp) ? EX_ERROR : EX_OK;
break;
case M_MAINTAIN:
maintain = true;
rc = maintain_wordlist_file(bfp);
break;
case M_WORD:
argc -= optind;
argv += optind;
rc = display_words(bfp, argc, argv, prob);
break;
case M_HIST:
rc = histogram(bfp);
break;
case M_ROBX:
rc = get_robx(bfp);
break;
case M_NONE:
default:
/* should have been handled above */
abort();
break;
}
bfpath_free(bfp);
return rc;
}