leave.c

Go to the documentation of this file.

/* Usage:       leave [ [+] hh[:]mm ]
 *
 * Author:      Terrence W. Holm
 *
 * Revision:
 *              Fred van Kempen, MINIX User Group Holland
 *               -adapted to MSS
 *               -adapted to new utmp database
 *               -adapted to POSIX (MINIX 1.5)
 *              Michael Temari, <temari@ix.netcom.com>
 *               -use localtime/mktime to fix bug with DST
 */

#include <sys/types.h>
#include <signal.h>
#include <time.h>
#include <utmp.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <stdio.h>


#define Min(a,b)  ((a<b) ? a : b)

#define STRING     80           /* lots of room for an argument */
#define MIN        60L          /* seconds per minute */
#define HOUR      (60L*60L)     /* seconds per hour */

/* Set the following to your personal preferences for the
 * time and contents of warnings.
 */
#define INTERVALS 13            /* size of intervals[] */
#define WARNINGS  4             /* size of warnings[] */


static char *Version = "@(#) LEAVE 1.4 (01/09/90)";
static int intervals[INTERVALS] = {
  -5 * MIN,
  -1 * MIN,
  0,
  MIN,
  2 * MIN,
  3 * MIN,
  4 * MIN,
  5 * MIN,
  6 * MIN,
  7 * MIN,
  8 * MIN,
  9 * MIN,
  10 * MIN
};
static char *warnings[WARNINGS] = {
  "You have to leave within 5 minutes",
  "Just one more minute!",
  "Time to leave!",
  "You're going to be late!"    /* for all subsequent warnings */
};


_PROTOTYPE(int main, (int argc, char **argv));
_PROTOTYPE(void Usage, (void));
_PROTOTYPE(void Get_Hour_Min, (char *when, int *hour, int *min));
_PROTOTYPE(int Still_Logged_On, (char *user, char *tty));

void Usage()
{
   fprintf(stderr, "Usage: leave [[+]hh[:]mm]\n");
  exit(1);
}


void Get_Hour_Min(when, hour, min)
char *when;
int *hour;
int *min;
{
  int hour_min;
  int just_min = 0;

  switch (sscanf(when, "%d:%d", &hour_min, &just_min)) {
      case 1:
        *hour = hour_min / 100;
        *min = hour_min % 100;
        break;
      case 2:
        *hour = hour_min;
        *min = just_min;
        break;
      default:
        Usage();
  }

  if (hour_min < 0 || just_min < 0 || *min > 59) Usage();
}


int Still_Logged_On(user, tty)
char *user;
char *tty;
{
  FILE *f;
  struct utmp login;

  if ((f = fopen(UTMP, "r")) == NULL)
        /* no login/logout records kept */
        return(1);

  while (fread(&login, sizeof(struct utmp), (size_t)1, f) == 1) {
        if (!strncmp(login.ut_line, tty, (size_t)8))
                if (!strncmp(login.ut_name, user, (size_t)8)) {
                        fclose(f);
                        return(1);
                } else {
                        fclose(f);
                        return(0);
                }
  }
  fclose(f);
  return(0);
}


int main(argc, argv)
int argc;
char *argv[];
{
  char when[STRING];
  time_t now = time((time_t *)0);
  time_t leave, delta;
  struct tm *tm;
  int hour, min;
  int pid, i;
  char *user = cuserid( (char *)NULL);
  char *tty = ttyname(0) + 5;

  /* get the argument string "when" either from stdin, or argv */
  if (argc <= 1) {
        printf("When do you have to leave? ");
        fflush(stdout);
        if (fgets(when, STRING, stdin) == NULL || when[0] == '\n') exit(0);
  } else {
        strcpy(when, argv[1]);
        if (argc > 2) strcat(when, argv[2]);
  }

  /* determine the leave time from the current time and "when" */
  tm = localtime(&now);
  if (when[0] == '+') {
        Get_Hour_Min(&when[1], &hour, &min);
        tm->tm_hour += hour;
        tm->tm_min += min;
        leave = mktime(tm);
  } else {
        /* user entered an absolute time */
        Get_Hour_Min(&when[0], &hour, &min);
        tm->tm_hour = hour;
        tm->tm_min = min;
        leave = mktime(tm);
        if (leave < now) {
                printf("That time has already passed!\n");
                exit(1);
        }
  }

  printf("Alarm set for %s", ctime(&leave));

  if ((pid = fork()) == -1) {
        fprintf(stderr, "leave: can not fork\n");
        exit(1);
  }
  if (pid != 0) exit(0);

  /* only the child continues on */
  if (user == NULL || tty == NULL) {
        fprintf(stderr, "leave: Can not determine user and terminal name\n");
        exit(1);
  }
  signal(SIGINT, SIG_IGN);
  signal(SIGQUIT, SIG_IGN);
  signal(SIGTERM, SIG_IGN);

  for (;;) {
        if (!Still_Logged_On(user, tty)) exit(0);

        /* how much longer until the leave time? */
        /* XXX - use difftime all over. */
        delta = leave - time((time_t *)0);

        /* which interval are we currently in? */
        for (i = 0; i < INTERVALS; ++i)
                if (delta + intervals[i] > 0) break;

        /* if we are within intervals[0] then print a warning If
         * there are more intervals than messages, then use/
         * warnings[WARNINGS-1] for all subsequent messages. */
        if (i > 0)
                printf("\007\r%s\r\n",
                        warnings[i > WARNINGS ? WARNINGS - 1 : i - 1]);

        if (i == INTERVALS) {
                printf("That was the last time I'll tell you. Bye.\r\n");
                exit(0);
        }
        /* Sleep until the next interval. For long periods, wake up
         * every hour to check if the user is still on (also required
         * because 16 bit ints don't allow long waits). */
        sleep((unsigned) Min(delta + intervals[i], HOUR));
  }
}

---