You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
mblaze/magrep.c

117 lines
2.1 KiB
C

8 years ago
#include <sys/stat.h>
#include <sys/types.h>
#include <ctype.h>
#include <errno.h>
#include <regex.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "blaze822.h"
static int aflag;
static int cflag;
static int dflag;
static int iflag;
static int qflag;
static int vflag;
static long matches;
static regex_t pattern;
static char *header;
int
match(char *file, char *s)
{
if (vflag ^ (regexec(&pattern, s, 0, 0, 0) == 0)) {
if (qflag)
exit(0);
if (!cflag)
printf("%s\n", file);
matches++;
return 1;
}
return 0;
}
void
magrep(char *file)
{
char *filename = file;
while (*filename == ' ' || *filename == '\t')
filename++;
struct message *msg = blaze822(filename);
if (!msg)
return;
char *v = blaze822_chdr(msg, header);
if (v) {
if (dflag) {
char d[4096];
blaze822_decode_rfc2047(d, v, sizeof d, "UTF-8");
match(file, d);
} else if (aflag) {
char *disp, *addr;
while ((v = blaze822_addr(v, &disp, &addr))) {
if (addr && match(file, addr))
break;
}
} else {
match(file, v);
}
}
blaze822_free(msg);
}
int
main(int argc, char *argv[])
{
int c;
while ((c = getopt(argc, argv, "acdiqv")) != -1)
switch(c) {
case 'a': aflag = 1; break;
case 'c': cflag = 1; break;
case 'd': dflag = 1; break;
case 'i': iflag = REG_ICASE; break;
case 'q': qflag = 1; break;
case 'v': vflag = 1; break;
default:
usage:
fprintf(stderr,
"Usage: magrep [-c|-q] [-v] [-i] [-a|-d] header:regex [msgs...]\n");
exit(2);
}
if (argc == optind)
goto usage;
header = argv[optind++];
char *rx = strchr(header, ':');
if (!rx)
goto usage;
*rx++ = 0;
int r = regcomp(&pattern, rx, REG_EXTENDED | iflag);
if (r != 0) {
char buf[256];
regerror(r, &pattern, buf, sizeof buf);
fprintf(stderr, "magrep: regex error: %s\n", buf);
exit(2);
}
if (argc == optind && isatty(0))
blaze822_loop1(":", magrep);
else
blaze822_loop(argc-optind, argv+optind, magrep);
if (cflag)
printf("%ld\n", matches);
return !matches;
}