-
-/*
- * rcvtty.c -- a rcvmail program (a lot like rcvalert) handling IPC ttys
+/* rcvtty.c -- a rcvmail program (a lot like rcvalert) handling IPC ttys
*
- * $Id$
+ * This code is Copyright (c) 2002, by the authors of nmh. See the
+ * COPYRIGHT file in the root directory of the nmh distribution for
+ * complete copyright information.
*/
-#include <h/mh.h>
-#include <h/signals.h>
-#include <h/rcvmail.h>
-#include <h/scansbr.h>
-#include <zotnet/tws/tws.h>
-#include <zotnet/mts/mts.h>
-#include <signal.h>
+/* Changed to use getutent() and friends. Assumes that when getutent() exists,
+ * a number of other things also exist. Please check.
+ * Ruud de Rooij <ruud@ruud.org> Sun, 28 May 2000 17:28:55 +0200
+ */
+
+#include "h/mh.h"
+#include "sbr/fmt_new.h"
+#include "scansbr.h"
+#include "sbr/getarguments.h"
+#include "sbr/smatch.h"
+#include "sbr/closefds.h"
+#include "sbr/ambigsw.h"
+#include "sbr/print_version.h"
+#include "sbr/print_help.h"
+#include "sbr/error.h"
+#include "h/signals.h"
+#include <setjmp.h>
+#include "h/fmt_scan.h"
+#include "h/tws.h"
+#include "h/mts.h"
+#include "h/done.h"
+#include "h/utils.h"
+#include "sbr/m_mktemp.h"
#include <fcntl.h>
-#include <utmp.h>
-#ifndef UTMP_FILE
-# ifdef _PATH_UTMP
-# define UTMP_FILE _PATH_UTMP
-# else
-# define UTMP_FILE "/etc/utmp"
-# endif
-#endif
+#ifdef HAVE_GETUTXENT
+#include <utmpx.h>
+#endif /* HAVE_GETUTXENT */
#define SCANFMT \
"%2(hour{dtimenow}):%02(min{dtimenow}): %<(size)%5(size) %>%<{encrypted}E%>\
%<(mymbox{from})%<{to}To:%14(friendly{to})%>%>%<(zero)%17(friendly{from})%> \
%{subject}%<{body}<<%{body}>>%>"
-static struct swit switches[] = {
-#define BIFFSW 0
- { "biff", 0 },
-#define FORMSW 1
- { "form formatfile", 0 },
-#define FMTSW 2
- { "format string", 5 },
-#define WIDTHSW 3
- { "width columns", 0 },
-#define NLSW 4
- { "newline", 0 },
-#define NNLSW 5
- { "nonewline", 0 },
-#define BELSW 6
- { "bell", 0 },
-#define NBELSW 7
- { "nobell", 0 },
-#define VERSIONSW 8
- { "version", 0 },
-#define HELPSW 9
- { "help", 0 },
- { NULL, 0 }
-};
+#define RCVTTY_SWITCHES \
+ X("biff", 0, BIFFSW) \
+ X("form formatfile", 0, FORMSW) \
+ X("format string", 5, FMTSW) \
+ X("width columns", 0, WIDTHSW) \
+ X("newline", 0, NLSW) \
+ X("nonewline", 0, NNLSW) \
+ X("bell", 0, BELSW) \
+ X("nobell", 0, NBELSW) \
+ X("version", 0, VERSIONSW) \
+ X("help", 0, HELPSW) \
+
+#define X(sw, minchars, id) id,
+DEFINE_SWITCH_ENUM(RCVTTY);
+#undef X
+
+#define X(sw, minchars, id) { sw, minchars, id },
+DEFINE_SWITCH_ARRAY(RCVTTY, switches);
+#undef X
static jmp_buf myctx;
-static int bell = 1;
-static int newline = 1;
-static int biff = 0;
-static int width = 0;
+static bool bell = true;
+static bool newline = true;
+static bool biff;
+static int width = -1;
static char *form = NULL;
static char *format = NULL;
-/*
- * external prototypes
- */
-char *getusername(void);
-
/*
* static prototypes
*/
-static RETSIGTYPE alrmser (int);
+static void alrmser (int);
static int message_fd (char **);
static int header_fd (void);
+#if HAVE_GETUTXENT
static void alert (char *, int);
+#endif /* HAVE_GETUTXENT */
int
main (int argc, char **argv)
{
int md, vecp = 0;
- char *cp, *user, buf[BUFSIZ], tty[BUFSIZ];
+ char *cp, *user, buf[BUFSIZ];
char **argp, **arguments, *vec[MAXARGS];
- struct utmp ut;
- register FILE *uf;
+ struct utmpx *utp;
-#ifdef LOCALE
- setlocale(LC_ALL, "");
-#endif
- invo_name = r1bindex (argv[0], '/');
+ if (nmh_init(argv[0], true, false)) { return 1; }
- /* read user profile/context */
- context_read();
-
- mts_init (invo_name);
+ mts_init ();
arguments = getarguments (invo_name, argc, argv, 1);
argp = arguments;
while ((cp = *argp++)) {
if (*cp == '-') {
switch (smatch (++cp, switches)) {
- case AMBIGSW:
+ case AMBIGSW:
ambigsw (cp, switches);
done (1);
- case UNKWNSW:
+ case UNKWNSW:
vec[vecp++] = --cp;
continue;
- case HELPSW:
+ case HELPSW:
snprintf (buf, sizeof(buf), "%s [command ...]", invo_name);
print_help (buf, switches, 1);
- done (1);
+ done (0);
case VERSIONSW:
print_version(invo_name);
- done (1);
+ done (0);
case BIFFSW:
- biff = 1;
+ biff = true;
continue;
- case FORMSW:
+ case FORMSW:
if (!(form = *argp++) || *form == '-')
- adios (NULL, "missing argument to %s", argp[-2]);
+ die("missing argument to %s", argp[-2]);
format = NULL;
continue;
- case FMTSW:
+ case FMTSW:
if (!(format = *argp++) || *format == '-')
- adios (NULL, "missing argument to %s", argp[-2]);
+ die("missing argument to %s", argp[-2]);
form = NULL;
continue;
case WIDTHSW:
if (!(cp = *argp++) || *cp == '-')
- adios(NULL, "missing argument to %s", argp[-2]);
+ die("missing argument to %s", argp[-2]);
width = atoi(cp);
continue;
case NLSW:
- newline = 1;
+ newline = true;
continue;
case NNLSW:
- newline = 0;
+ newline = false;
continue;
case BELSW:
- bell = 1;
+ bell = true;
continue;
case NBELSW:
- bell = 0;
+ bell = false;
continue;
}
vec[vecp] = 0;
if ((md = vecp ? message_fd (vec) : header_fd ()) == NOTOK)
- exit (RCV_MBX);
+ exit(1);
user = getusername();
- if ((uf = fopen (UTMP_FILE, "r")) == NULL)
- exit (RCV_MBX);
-
- while (fread ((char *) &ut, sizeof(ut), 1, uf) == 1)
- if (ut.ut_name[0] != 0
- && strncmp (user, ut.ut_name, sizeof(ut.ut_name)) == 0) {
- strncpy (tty, ut.ut_line, sizeof(ut.ut_line));
- alert (tty, md);
- }
- fclose (uf);
- exit (RCV_MOK);
- return 0; /* dead code to satisfy the compiler */
+#if HAVE_GETUTXENT
+ setutxent();
+ while ((utp = getutxent()) != NULL) {
+ if (utp->ut_type == USER_PROCESS && utp->ut_user[0] != 0
+ && utp->ut_line[0] != 0
+ && strncmp (user, utp->ut_user, sizeof(utp->ut_user)) == 0) {
+ alert(utp->ut_line, md);
+ }
+ }
+ endutxent();
+#else
+ NMH_UNUSED (utp);
+#endif /* HAVE_GETUTXENT */
+
+ exit(0);
}
-static RETSIGTYPE
+static void
alrmser (int i)
{
-#ifndef RELIABLE_SIGNALS
- SIGNAL (SIGALRM, alrmser);
-#endif
+ NMH_UNUSED (i);
longjmp (myctx, 1);
}
message_fd (char **vec)
{
pid_t child_id;
- int bytes, fd, seconds;
- char tmpfil[BUFSIZ];
+ int bytes, seconds;
+ int fd;
+ char *tfile;
struct stat st;
-#ifdef HAVE_MKSTEMP
- fd = mkstemp (strncpy (tmpfil, "/tmp/rcvttyXXXXX", sizeof(tmpfil)));
-#else
- unlink (mktemp (strncpy (tmpfil, "/tmp/rcvttyXXXXX", sizeof(tmpfil))));
- if ((fd = open (tmpfil, O_RDWR | O_CREAT | O_TRUNC, 0600)) == NOTOK)
- return header_fd ();
-#endif
- unlink (tmpfil);
+ if ((tfile = m_mktemp2(NULL, invo_name, &fd, NULL)) == NULL) {
+ inform("unable to create temporary file in %s", get_temp_dir());
+ return NOTOK;
+ }
+ (void) m_unlink(tfile); /* Use fd, no longer need the file name. */
- if ((child_id = vfork()) == NOTOK) {
+ if ((child_id = fork()) == NOTOK) {
/* fork error */
close (fd);
return header_fd ();
- } else if (child_id) {
+ }
+ if (child_id) {
/* parent process */
if (!setjmp (myctx)) {
SIGNAL (SIGALRM, alrmser);
pidwait(child_id, OK);
alarm (0);
- if (fstat (fd, &st) != NOTOK && st.st_size > (off_t) 0)
+ if (fstat (fd, &st) != NOTOK && st.st_size > 0)
return fd;
} else {
/*
* Ruthlessly kill the child and anything
* else in its process group.
*/
- KILLPG(child_id, SIGKILL);
+ killpg(child_id, SIGKILL);
}
close (fd);
return header_fd ();
/* child process */
rewind (stdin);
if (dup2 (fd, 1) == NOTOK || dup2 (fd, 2) == NOTOK)
- _exit (-1);
+ _exit(1);
closefds (3);
- setpgid ((pid_t) 0, getpid ()); /* put in own process group */
- execvp (vec[0], vec);
- _exit (-1);
- return 1; /* dead code to satisfy compiler */
+ setpgid(0, getpid()); /* put in own process group */
+ if (execvp (vec[0], vec) == NOTOK) {
+ _exit(1);
+ }
+
+ return NOTOK;
}
header_fd (void)
{
int fd;
- char *nfs, tmpfil[BUFSIZ];
+ char *nfs;
+ char *tfile = NULL;
+ charstring_t scanl = NULL;
- strncpy (tmpfil, m_tmpfil (invo_name), sizeof(tmpfil));
- if ((fd = open (tmpfil, O_RDWR | O_CREAT | O_TRUNC, 0600)) == NOTOK)
- return NOTOK;
- unlink (tmpfil);
+ if ((tfile = m_mktemp2(NULL, invo_name, &fd, NULL)) == NULL) {
+ inform("unable to create temporary file in %s", get_temp_dir());
+ return NOTOK;
+ }
+ (void) m_unlink(tfile); /* Use fd, no longer need the file name. */
rewind (stdin);
/* get new format string */
nfs = new_fs (form, format, SCANFMT);
- scan (stdin, 0, 0, nfs, width, 0, 0, NULL, 0L, 0);
- if (newline)
- write (fd, "\n\r", 2);
- write (fd, scanl, strlen (scanl));
- if (bell)
- write (fd, "\007", 1);
+ scan (stdin, 0, 0, nfs, width, 0, 0, NULL, 0L, 0, &scanl);
+ scan_finished ();
+ if (newline) {
+ if (write (fd, "\n\r", 2) < 0) {
+ advise (tfile, "write LF/CR");
+ }
+ }
+ if (write (fd, charstring_buffer (scanl), charstring_bytes (scanl)) < 0) {
+ advise (tfile, "write");
+ }
+ charstring_free (scanl);
+ if (bell) {
+ if (write (fd, "\007", 1) < 0) {
+ advise (tfile, "write BEL");
+ }
+ }
return fd;
}
+#if HAVE_GETUTXENT
static void
alert (char *tty, int md)
{
if (stat (ttyspec, &st) == NOTOK || (st.st_mode & mask) == 0)
return;
- if (!setjmp (myctx)) {
- SIGNAL (SIGALRM, alrmser);
- alarm (2);
- td = open (ttyspec, O_WRONLY);
- alarm (0);
- if (td == NOTOK)
- return;
- } else {
+ if (setjmp (myctx)) {
alarm (0);
return;
}
+ SIGNAL (SIGALRM, alrmser);
+ alarm (2);
+ td = open (ttyspec, O_WRONLY);
+ alarm (0);
+ if (td == NOTOK)
+ return;
- lseek (md, (off_t) 0, SEEK_SET);
+ lseek(md, 0, SEEK_SET);
while ((i = read (md, buffer, sizeof(buffer))) > 0)
if (write (td, buffer, i) != i)
close (td);
}
-
+#endif /* HAVE_GETUTXENT */
projects / nmh / blobdiff