X-Git-Url: http://git.marmaro.de/?p=mmh;a=blobdiff_plain;f=uip%2Frcvdist.c;h=1337c893fe5693e1e3334c423b1903f1a41ff241;hp=d3e41001af88b6f23b2c4c66ad888d2dd0c0ae88;hb=e16d5bfc58d8b0dfe61701cd4d6b1732e837d60a;hpb=fede6e42d81ce34fd5c1bbe7fb2757b281c2573a diff --git a/uip/rcvdist.c b/uip/rcvdist.c index d3e4100..1337c89 100644 --- a/uip/rcvdist.c +++ b/uip/rcvdist.c @@ -11,6 +11,10 @@ #include #include #include +#include +#include +#include +#include static struct swit switches[] = { #define FORMSW 0 @@ -30,7 +34,7 @@ static char tmpfil[BUFSIZ] = ""; ** prototypes */ static void rcvdistout(FILE *, char *, char *); -static void unlink_done(); +void unlink_done(); int @@ -42,7 +46,9 @@ main(int argc, char **argv) FILE *fp; char *tfile = NULL; - atexit(unlink_done); + if (atexit(unlink_done) != 0) { + adios(EX_OSERR, NULL, "atexit failed"); + } setlocale(LC_ALL, ""); invo_name = mhbasename(argv[0]); @@ -58,7 +64,7 @@ main(int argc, char **argv) switch (smatch(++cp, switches)) { case AMBIGSW: ambigsw(cp, switches); - exit(1); + exit(EX_USAGE); case UNKWNSW: vec[vecp++] = --cp; continue; @@ -66,14 +72,14 @@ main(int argc, char **argv) case HELPSW: snprintf(buf, sizeof(buf), "%s [switches] [switches for spost] address ...", invo_name); print_help(buf, switches, 1); - exit(0); + exit(argc == 2 ? EX_OK : EX_USAGE); case VERSIONSW: print_version(invo_name); - exit(0); + exit(argc == 2 ? EX_OK : EX_USAGE); case FORMSW: if (!(form = *argp++) || *form == '-') { - adios(NULL, "missing argument to %s", + adios(EX_USAGE, NULL, "missing argument to %s", argp[-2]); } continue; @@ -83,28 +89,27 @@ main(int argc, char **argv) } if (!addrs) { - adios(NULL, "usage: %s [switches] [switches for spost] address ...", invo_name); + adios(EX_USAGE, NULL, "usage: %s [switches] [switches for spost] address ...", invo_name); } umask(~m_gmprot()); tfile = m_mktemp2(NULL, invo_name, NULL, &fp); - if (tfile == NULL) adios("rcvdist", "unable to create temporary file"); + if (tfile == NULL) adios(EX_CANTCREAT, "rcvdist", "unable to create temporary file"); strncpy(tmpfil, tfile, sizeof(tmpfil)); cpydata(fileno(stdin), fileno(fp), "message", tmpfil); fseek(fp, 0L, SEEK_SET); tfile = m_mktemp2(NULL, invo_name, NULL, NULL); - if (tfile == NULL) adios("forw", "unable to create temporary file"); + if (tfile == NULL) adios(EX_CANTCREAT, "forw", "unable to create temporary file"); strncpy(drft, tfile, sizeof(tmpfil)); rcvdistout(fp, form, addrs); fclose(fp); if (distout(drft, tmpfil, backup) == NOTOK) { - /* sysexits.h EX_DATAERR */ - exit(1); + exit(EX_IOERR); } vec[0] = "spost"; @@ -115,7 +120,7 @@ main(int argc, char **argv) execvp(*vec, vec); fprintf(stderr, "unable to exec "); perror(*vec); - _exit(1); + _exit(EX_OSERR); return 0; /* dead code to satisfy the compiler */ } @@ -151,14 +156,14 @@ static char *addrcomps[] = { static void rcvdistout(FILE *inb, char *form, char *addrs) { - register int char_read = 0, format_len, i, state; - register char *tmpbuf, **nxtbuf, **ap; + int char_read = 0, format_len, i, state; + char *tmpbuf, **nxtbuf, **ap; char *cp, *scanl, name[NAMESZ]; - register struct comp *cptr, **savecomp; + struct comp *cptr, **savecomp; FILE *out; if (!(out = fopen(drft, "w"))) { - adios(drft, "unable to create"); + adios(EX_CANTCREAT, drft, "unable to create"); } /* get new format string */ @@ -166,13 +171,13 @@ rcvdistout(FILE *inb, char *form, char *addrs) format_len = strlen(cp); ncomps = fmt_compile(cp, &fmt) + 1; if (!(nxtbuf = compbuffers = - (char **) calloc((size_t) ncomps, sizeof(char *)))) { - adios(NULL, "unable to allocate component buffers"); + (char **) mh_xcalloc((size_t) ncomps, sizeof(char *)))) { + adios(EX_OSERR, NULL, "unable to allocate component buffers"); } if (!(savecomp = used_buf = - (struct comp **) calloc((size_t) (ncomps + 1), + (struct comp **) mh_xcalloc((size_t) (ncomps + 1), sizeof(struct comp *)))) { - adios(NULL, "unable to allocate component buffer stack"); + adios(EX_OSERR, NULL, "unable to allocate component buffer stack"); } savecomp += ncomps + 1; *--savecomp = 0; @@ -245,7 +250,7 @@ rcvdistout(FILE *inb, char *form, char *addrs) goto finished; default: - adios(NULL, "m_getfld() returned %d", state); + adios(EX_SOFTWARE, NULL, "m_getfld() returned %d", state); } } finished: ; @@ -258,7 +263,7 @@ finished: ; fputs(scanl, out); if (ferror(out)) { - adios(drft, "error writing"); + adios(EX_IOERR, drft, "error writing"); } fclose(out); @@ -275,7 +280,7 @@ finished: ; } -static void +void unlink_done() { if (*backup) {