Include full-write.h.

[chaz/tar] / src / rmt.c

/* Remote connection server.
   Copyright 1994,95,96,97,99,2000, 2001 Free Software Foundation, Inc.

   This program is free software; you can redistribute it and/or modify it
   under the terms of the GNU General Public License as published by the
   Free Software Foundation; either version 2, or (at your option) any later
   version.

   This program is distributed in the hope that it will be useful, but
   WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
   Public License for more details.

   You should have received a copy of the GNU General Public License along
   with this program; if not, write to the Free Software Foundation, Inc.,
   59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */

/* Copyright (C) 1983 Regents of the University of California.
   All rights reserved.

   Redistribution and use in source and binary forms are permitted provided
   that the above copyright notice and this paragraph are duplicated in all
   such forms and that any documentation, advertising materials, and other
   materials related to such distribution and use acknowledge that the
   software was developed by the University of California, Berkeley.  The
   name of the University may not be used to endorse or promote products
   derived from this software without specific prior written permission.
   THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
   WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
   MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.  */

#include "system.h"
#include <localedir.h>
#include <safe-read.h>
#include <full-write.h>

#include <getopt.h>
#include <sys/socket.h>

#ifndef EXIT_FAILURE
# define EXIT_FAILURE 1
#endif
#ifndef EXIT_SUCCESS
# define EXIT_SUCCESS 0
#endif

/* Maximum size of a string from the requesting program.  */
#define STRING_SIZE 64

/* Name of executing program.  */
const char *program_name;

/* File descriptor of the tape device, or negative if none open.  */
static int tape = -1;

/* Buffer containing transferred data, and its allocated size.  */
static char *record_buffer;
static size_t allocated_size;

/* Buffer for constructing the reply.  */
static char reply_buffer[BUFSIZ];

/* Debugging tools.  */

static FILE *debug_file;

#define DEBUG(File) \
  if (debug_file) fprintf(debug_file, File)

#define DEBUG1(File, Arg) \
  if (debug_file) fprintf(debug_file, File, Arg)

#define DEBUG2(File, Arg1, Arg2) \
  if (debug_file) fprintf(debug_file, File, Arg1, Arg2)

/* Return an error string, given an error number.  */
#if HAVE_STRERROR
# ifndef strerror
char *strerror ();
# endif
#else
static char *
private_strerror (int errnum)
{
  extern char *sys_errlist[];
  extern int sys_nerr;

  if (errnum > 0 && errnum <= sys_nerr)
    return _(sys_errlist[errnum]);
  return _("Unknown system error");
}
# define strerror private_strerror
#endif

static void
report_error_message (const char *string)
{
  DEBUG1 ("rmtd: E 0 (%s)\n", string);

  sprintf (reply_buffer, "E0\n%s\n", string);
  full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
}

static void
report_numbered_error (int num)
{
  DEBUG2 ("rmtd: E %d (%s)\n", num, strerror (num));

  sprintf (reply_buffer, "E%d\n%s\n", num, strerror (num));
  full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
}

static void
get_string (char *string)
{
  int counter;

  for (counter = 0; counter < STRING_SIZE; counter++)
    {
      if (safe_read (STDIN_FILENO, string + counter, 1) != 1)
        exit (EXIT_SUCCESS);

      if (string[counter] == '\n')
        break;
    }
  string[counter] = '\0';
}

static void
prepare_record_buffer (size_t size)
{
  if (size <= allocated_size)
    return;

  if (record_buffer)
    free (record_buffer);

  record_buffer = malloc (size);

  if (! record_buffer)
    {
      DEBUG (_("rmtd: Cannot allocate buffer space\n"));

      report_error_message (N_("Cannot allocate buffer space"));
      exit (EXIT_FAILURE);      /* exit status used to be 4 */
    }

  allocated_size = size;

#ifdef SO_RCVBUF
  while (size > 1024 &&
         (setsockopt (STDIN_FILENO, SOL_SOCKET, SO_RCVBUF,
                      (char *) &size, sizeof size)
          < 0))
    size -= 1024;
#else
  /* FIXME: I do not see any purpose to the following line...  Sigh! */
  size = 1 + ((size - 1) % 1024);
#endif
}

/* Decode OFLAG_STRING, which represents the 2nd argument to `open'.
   OFLAG_STRING should contain an optional integer, followed by an optional
   symbolic representation of an open flag using only '|' to separate its
   components (e.g. "O_WRONLY|O_CREAT|O_TRUNC").  Prefer the symbolic
   representation if available, falling back on the numeric
   representation, or to zero if both formats are absent.

   This function should be the inverse of encode_oflag.  The numeric
   representation is not portable from one host to another, but it is
   for backward compatibility with old-fashioned clients that do not
   emit symbolic open flags.  */

static int
decode_oflag (char const *oflag_string)
{
  char *oflag_num_end;
  int numeric_oflag = strtol (oflag_string, &oflag_num_end, 10);
  int symbolic_oflag = 0;
  
  oflag_string = oflag_num_end;
  while (ISSPACE ((unsigned char) *oflag_string))
    oflag_string++;
    
  do
    {
      struct name_value_pair { char const *name; int value; };
      static struct name_value_pair const table[] =
      {
#ifdef O_APPEND
        {"APPEND", O_APPEND},
#endif
        {"CREAT", O_CREAT},
#ifdef O_DSYNC
        {"DSYNC", O_DSYNC},
#endif
        {"EXCL", O_EXCL},
#ifdef O_LARGEFILE
        {"LARGEFILE", O_LARGEFILE}, /* LFS extension for opening large files */
#endif
#ifdef O_NOCTTY
        {"NOCTTY", O_NOCTTY},
#endif
#ifdef O_NONBLOCK
        {"NONBLOCK", O_NONBLOCK},
#endif
        {"RDONLY", O_RDONLY},
        {"RDWR", O_RDWR},
#ifdef O_RSYNC
        {"RSYNC", O_RSYNC},
#endif
#ifdef O_SYNC
        {"SYNC", O_SYNC},
#endif
        {"TRUNC", O_TRUNC},
        {"WRONLY", O_WRONLY}
      };
      struct name_value_pair const *t;
      size_t s;

      if (*oflag_string++ != 'O' || *oflag_string++ != '_')
        return numeric_oflag;

      for (t = table;
           (strncmp (oflag_string, t->name, s = strlen (t->name)) != 0
            || (oflag_string[s]
                && strchr ("ABCDEFGHIJKLMNOPQRSTUVWXYZ_0123456789",
                           oflag_string[s])));
           t++)
        if (t == table + sizeof table / sizeof *table - 1)
          return numeric_oflag;

      symbolic_oflag |= t->value;
      oflag_string += s;
    }
  while (*oflag_string++ == '|');

  return symbolic_oflag;
}

static struct option const long_opts[] =
{
  {"help", no_argument, 0, 'h'},
  {"version", no_argument, 0, 'v'},
  {0, 0, 0, 0}
};

static void
usage (int status)
{
  if (status != EXIT_SUCCESS)
    fprintf (stderr, _("Try `%s --help' for more information.\n"),
             program_name);
  else
    {
      printf (_("\
Usage: %s [OPTION]\n\
Manipulate a tape drive, accepting commands from a remote process.\n\
\n\
  --version  Output version info.\n\
  --help  Output this help.\n"),
              program_name);
      fputs (_("\nReport bugs to <bug-tar@gnu.org>.\n"), stdout);
    }

  exit (status);
}

int
main (int argc, char *const *argv)
{
  char command;
  ssize_t status;

  /* FIXME: Localization is meaningless, unless --help and --version are
     locally used.  Localization would be best accomplished by the calling
     tar, on messages found within error packets.  */

  program_name = argv[0];
  setlocale (LC_ALL, "");
  bindtextdomain (PACKAGE, LOCALEDIR);
  textdomain (PACKAGE);

  switch (getopt_long (argc, argv, "", long_opts, NULL))
    {
    default:
      usage (EXIT_FAILURE);
      
    case 'h':
      usage (EXIT_SUCCESS);
      
    case 'v':
      printf ("rmt (GNU %s) %s\n%s\n%s\n", PACKAGE, VERSION,
              "Copyright 2001 Free Software Foundation, Inc.",
              _("\
This program comes with NO WARRANTY, to the extent permitted by law.\n\
You may redistribute it under the terms of the GNU General Public License;\n\
see the file named COPYING for details."));
      return EXIT_SUCCESS;

    case -1:
      break;
    }

  if (optind < argc)
    {
      if (optind != argc - 1)
        usage (EXIT_FAILURE);
      debug_file = fopen (argv[optind], "w");
      if (debug_file == 0)
        {
          report_numbered_error (errno);
          exit (EXIT_FAILURE);
        }
      setbuf (debug_file, 0);
    }

top:
  errno = 0;
  status = 0;
  if (safe_read (STDIN_FILENO, &command, 1) != 1)
    return EXIT_SUCCESS;

  switch (command)
    {
      /* FIXME: Maybe 'H' and 'V' for --help and --version output?  */

    case 'O':
      {
        char device_string[STRING_SIZE];
        char oflag_string[STRING_SIZE];

        get_string (device_string);
        get_string (oflag_string);
        DEBUG2 ("rmtd: O %s %s\n", device_string, oflag_string);

        if (tape >= 0)
          close (tape);

        tape = open (device_string, decode_oflag (oflag_string), MODE_RW);
        if (tape < 0)
          goto ioerror;
        goto respond;
      }

    case 'C':
      {
        char device_string[STRING_SIZE];

        get_string (device_string); /* discard */
        DEBUG ("rmtd: C\n");

        if (close (tape) < 0)
          goto ioerror;
        tape = -1;
        goto respond;
      }

    case 'L':
      {
        char count_string[STRING_SIZE];
        char position_string[STRING_SIZE];
        off_t count = 0;
        int negative;
        int whence;
        char *p;

        get_string (count_string);
        get_string (position_string);
        DEBUG2 ("rmtd: L %s %s\n", count_string, position_string);

        /* Parse count_string, taking care to check for overflow.
           We can't use standard functions,
           since off_t might be longer than long.  */

        for (p = count_string;  *p == ' ' || *p == '\t';  p++)
          continue;

        negative = *p == '-';
        p += negative || *p == '+';

        for (;;)
          {
            int digit = *p++ - '0';
            if (9 < (unsigned) digit)
              break;
            else
              {
                off_t c10 = 10 * count;
                off_t nc = negative ? c10 - digit : c10 + digit;
                if (c10 / 10 != count || (negative ? c10 < nc : nc < c10))
                  {
                    report_error_message (N_("Seek offset out of range"));
                    exit (EXIT_FAILURE);
                  }
                count = nc;
              }
          }

        switch (atoi (position_string))
          {
          case 0: whence = SEEK_SET; break;
          case 1: whence = SEEK_CUR; break;
          case 2: whence = SEEK_END; break;
          default:
            report_error_message (N_("Seek direction out of range"));
            exit (EXIT_FAILURE);
          }
        count = lseek (tape, count, whence);
        if (count < 0)
          goto ioerror;

        /* Convert count back to string for reply.
           We can't use sprintf, since off_t might be longer than long.  */
        p = count_string + sizeof count_string;
        *--p = '\0';
        do
          *--p = '0' + (int) (count % 10);
        while ((count /= 10) != 0);
        
        DEBUG1 ("rmtd: A %s\n", p);

        sprintf (reply_buffer, "A%s\n", p);
        full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
        goto top;
      }

    case 'W':
      {
        char count_string[STRING_SIZE];
        size_t size;
        size_t counter;

        get_string (count_string);
        size = atol (count_string);
        DEBUG1 ("rmtd: W %s\n", count_string);

        prepare_record_buffer (size);
        for (counter = 0; counter < size; counter += status)
          {
            status = safe_read (STDIN_FILENO, &record_buffer[counter],
                                size - counter);
            if (status <= 0)
              {
                DEBUG (_("rmtd: Premature eof\n"));

                report_error_message (N_("Premature end of file"));
                exit (EXIT_FAILURE); /* exit status used to be 2 */
              }
          }
        status = full_write (tape, record_buffer, size);
        if (status < 0)
          goto ioerror;
        goto respond;
      }

    case 'R':
      {
        char count_string[STRING_SIZE];
        size_t size;

        get_string (count_string);
        DEBUG1 ("rmtd: R %s\n", count_string);

        size = atol (count_string);
        prepare_record_buffer (size);
        status = safe_read (tape, record_buffer, size);
        if (status < 0)
          goto ioerror;
        sprintf (reply_buffer, "A%ld\n", (long) status);
        full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
        full_write (STDOUT_FILENO, record_buffer, status);
        goto top;
      }

    case 'I':
      {
        char operation_string[STRING_SIZE];
        char count_string[STRING_SIZE];

        get_string (operation_string);
        get_string  (count_string);
        DEBUG2 ("rmtd: I %s %s\n", operation_string, count_string);

#ifdef MTIOCTOP
        {
          struct mtop mtop;
          const char *p;
          off_t count = 0;
          int negative;

          /* Parse count_string, taking care to check for overflow.
             We can't use standard functions,
             since off_t might be longer than long.  */
          
          for (p = count_string;  *p == ' ' || *p == '\t';  p++)
            continue;
          
          negative = *p == '-';
          p += negative || *p == '+';
          
          for (;;)
            {
              int digit = *p++ - '0';
              if (9 < (unsigned) digit)
                break;
              else
                {
                  off_t c10 = 10 * count;
                  off_t nc = negative ? c10 - digit : c10 + digit;
                  if (c10 / 10 != count || (negative ? c10 < nc : nc < c10))
                    {
                      report_error_message (N_("Seek offset out of range"));
                      exit (EXIT_FAILURE);
                    }
                  count = nc;
                }
            }

          mtop.mt_count = count;
          if (mtop.mt_count != count)
            {
              report_error_message (N_("Seek offset out of range"));
              exit (EXIT_FAILURE);
            }
          mtop.mt_op = atoi (operation_string);

          if (ioctl (tape, MTIOCTOP, (char *) &mtop) < 0)
            goto ioerror;
        }
#endif
        goto respond;
      }

    case 'S':                   /* status */
      {
        DEBUG ("rmtd: S\n");

#ifdef MTIOCGET
        {
          struct mtget operation;

          if (ioctl (tape, MTIOCGET, (char *) &operation) < 0)
            goto ioerror;
          status = sizeof operation;
          sprintf (reply_buffer, "A%ld\n", (long) status);
          full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
          full_write (STDOUT_FILENO, (char *) &operation, sizeof operation);
        }
#endif
        goto top;
      }

    default:
      DEBUG1 (_("rmtd: Garbage command %c\n"), command);

      report_error_message (N_("Garbage command"));
      exit (EXIT_FAILURE);      /* exit status used to be 3 */
    }

respond:
  DEBUG1 ("rmtd: A %ld\n", (long) status);

  sprintf (reply_buffer, "A%ld\n", (long) status);
  full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
  goto top;

ioerror:
  report_numbered_error (errno);
  goto top;
}