From 963cfc9a6f6e721f52aa949e6d1af0c3e8dc2ecc Mon Sep 17 00:00:00 2001 From: cinap_lenrek Date: Sun, 12 Mar 2017 17:15:03 +0100 Subject: merging erik quanstros nupas --- sys/src/cmd/upas/scanmail/testscan.c | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) (limited to 'sys/src/cmd/upas/scanmail/testscan.c') diff --git a/sys/src/cmd/upas/scanmail/testscan.c b/sys/src/cmd/upas/scanmail/testscan.c index e5ea59ad5..61ea70a15 100644 --- a/sys/src/cmd/upas/scanmail/testscan.c +++ b/sys/src/cmd/upas/scanmail/testscan.c @@ -1,4 +1,5 @@ -#include "sys.h" +#include "common.h" +#include #include "spam.h" int debug; @@ -13,7 +14,7 @@ int matchaction(Patterns*, char*); void usage(void) { - fprint(2, "missing or bad arguments to qer\n"); + fprint(2, "usage: testscan -avd [-p pattern] ...\n"); exits("usage"); } @@ -23,10 +24,9 @@ Malloc(long n) void *p; p = malloc(n); - if(p == 0){ - fprint(2, "malloc error"); - exits("malloc"); - } + if(p == nil) + sysfatal("malloc: %r"); + setmalloctag(p, getcallerpc(&n)); return p; } @@ -34,10 +34,9 @@ void* Realloc(void *p, ulong n) { p = realloc(p, n); - if(p == 0){ - fprint(2, "realloc error"); - exits("realloc"); - } + if(p == nil) + sysfatal("malloc: %r"); + setrealloctag(p, getcallerpc(&p)); return p; } @@ -79,7 +78,7 @@ main(int argc, char *argv[]) char body[Bodysize+2], *raw, *ret; Biobuf *bp; - sprint(patfile, "%s/patterns", UPASLIB); + snprint(patfile, sizeof patfile, "%s/patterns", UPASLIB); aflag = -1; vflag = 0; ARGBEGIN { @@ -93,7 +92,7 @@ main(int argc, char *argv[]) debug++; break; case 'p': - strcpy(patfile,ARGF()); + snprint(patfile, sizeof patfile, "%s", EARGF(usage())); break; } ARGEND -- cgit v1.2.3