-/*
- * 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 MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
- */
-
-#ifndef lint
-char copyright[] =
-"@(#) Copyright (c) 1983 Regents of the University of California.\n\
- All rights reserved.\n";
-#endif /* not lint */
-
-/*
- * rmt
- */
-#include <stdio.h>
-#include <sgtty.h>
-#include <sys/types.h>
+/* Remote connection server.
+
+ Copyright (C) 1994, 1995, 1996, 1997, 1999, 2000, 2001, 2003, 2004
+ 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>
-#ifdef HAVE_SYS_GENTAPE_H /* e.g., ISC UNIX */
-#include <sys/gentape.h>
-#else
-#ifdef M_UNIX
-#include <sys/tape.h>
-#else
-#include <sys/mtio.h>
+
+#ifndef EXIT_FAILURE
+# define EXIT_FAILURE 1
#endif
+#ifndef EXIT_SUCCESS
+# define EXIT_SUCCESS 0
#endif
-#include <errno.h>
-#if defined (_I386) && defined (_AIX)
-#include <fcntl.h>
-#endif
+/* Maximum size of a string from the requesting program. */
+#define STRING_SIZE 64
-#ifdef HAVE_UNISTD_H
-#include <unistd.h>
+/* 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
-long lseek ();
+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
-#ifdef STDC_HEADERS
-#include <string.h>
-#include <stdlib.h>
-#else
-extern char *malloc ();
+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++)
+ {
+ if (safe_read (STDIN_FILENO, string + counter, 1) != 1)
+ exit (EXIT_SUCCESS);
+
+ if (string[counter] == '\n' || counter == STRING_SIZE - 1)
+ break;
+ }
+ string[counter] = '\0';
+}
+
+static void
+prepare_input_buffer (int fd, 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
+ if (0 <= fd)
+ {
+ int isize = size < INT_MAX ? size : INT_MAX;
+ while (setsockopt (fd, SOL_SOCKET, SO_RCVBUF,
+ (char *) &isize, sizeof isize)
+ && 1024 < isize)
+ isize >>= 1;
+ }
#endif
+}
-int tape = -1;
+/* 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.
-char *record;
-int maxrecsize = -1;
-char *checkbuf ();
-void getstring ();
-void error ();
+ 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. */
-#define SSIZE 64
-char device[SSIZE];
-char count[SSIZE], mode[SSIZE], pos[SSIZE], op[SSIZE];
+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;
-extern errno;
-extern char *sys_errlist[];
-char resp[BUFSIZ];
+ oflag_string = oflag_num_end;
+ while (ISSPACE ((unsigned char) *oflag_string))
+ oflag_string++;
-FILE *debug;
-#define DEBUG(f) if (debug) fprintf(debug, f)
-#define DEBUG1(f,a) if (debug) fprintf(debug, f, a)
-#define DEBUG2(f,a1,a2) if (debug) fprintf(debug, f, a1, a2)
+ 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) __attribute__ ((noreturn));
+
+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);
+ printf (_("\nReport bugs to <%s>.\n"), PACKAGE_BUGREPORT);
+ }
+
+ exit (status);
+}
int
-main (argc, argv)
- int argc;
- char **argv;
+main (int argc, char *const *argv)
{
- int rval;
- char c;
- int n, i, cc;
+ char command;
+ size_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. */
- argc--, argv++;
- if (argc > 0)
+ program_name = argv[0];
+ setlocale (LC_ALL, "");
+ bindtextdomain (PACKAGE, LOCALEDIR);
+ textdomain (PACKAGE);
+
+ switch (getopt_long (argc, argv, "", long_opts, NULL))
{
- debug = fopen (*argv, "w");
- if (debug == 0)
- exit (1);
- (void) setbuf (debug, (char *) 0);
+ default:
+ usage (EXIT_FAILURE);
+
+ case 'h':
+ usage (EXIT_SUCCESS);
+
+ case 'v':
+ {
+ printf ("rmt (%s) %s\n%s\n", PACKAGE_NAME, PACKAGE_VERSION,
+ "Copyright (C) 2004 Free Software Foundation, Inc.");
+ puts (_("\
+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);
+ return EXIT_FAILURE;
+ }
+ setbuf (debug_file, 0);
+ }
+
top:
errno = 0;
- rval = 0;
- if (read (0, &c, 1) != 1)
- exit (0);
- switch (c)
+ 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':
- if (tape >= 0)
- (void) close (tape);
- getstring (device);
- getstring (mode);
- DEBUG2 ("rmtd: O %s %s\n", device, mode);
-#if defined (i386) && defined (AIX)
- /* This is alleged to fix a byte ordering problem. */
- /* I'm quite suspicious if it's right. -- mib */
{
- int oflag = atoi (mode);
- int nflag = 0;
- if ((oflag & 3) == 0)
- nflag |= O_RDONLY;
- if (oflag & 1)
- nflag |= O_WRONLY;
- if (oflag & 2)
- nflag |= O_RDWR;
- if (oflag & 0x0008)
- nflag |= O_APPEND;
- if (oflag & 0x0200)
- nflag |= O_CREAT;
- if (oflag & 0x0400)
- nflag |= O_TRUNC;
- if (oflag & 0x0800)
- nflag |= O_EXCL;
- tape = open (device, nflag, 0666);
+ 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;
}
-#else
- tape = open (device, atoi (mode), 0666);
-#endif
- if (tape < 0)
- goto ioerror;
- goto respond;
case 'C':
- DEBUG ("rmtd: C\n");
- getstring (device); /* discard */
- if (close (tape) < 0)
- goto ioerror;
- tape = -1;
- goto respond;
+ {
+ 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':
- getstring (count);
- getstring (pos);
- DEBUG2 ("rmtd: L %s %s\n", count, pos);
- rval = lseek (tape, (long) atoi (count), atoi (pos));
- if (rval < 0)
- goto ioerror;
- goto respond;
+ {
+ 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"));
+ return 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"));
+ return 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':
- getstring (count);
- n = atoi (count);
- DEBUG1 ("rmtd: W %s\n", count);
- record = checkbuf (record, n);
- for (i = 0; i < n; i += cc)
- {
- cc = read (0, &record[i], n - i);
- if (cc <= 0)
- {
- DEBUG ("rmtd: premature eof\n");
- exit (2);
- }
- }
- rval = write (tape, record, n);
- if (rval < 0)
- goto ioerror;
- goto respond;
+ {
+ char count_string[STRING_SIZE];
+ size_t size;
+ size_t counter;
- case 'R':
- getstring (count);
- DEBUG1 ("rmtd: R %s\n", count);
- n = atoi (count);
- record = checkbuf (record, n);
- rval = read (tape, record, n);
- if (rval < 0)
- goto ioerror;
- (void) sprintf (resp, "A%d\n", rval);
- (void) write (1, resp, strlen (resp));
- (void) write (1, record, rval);
- goto top;
+ get_string (count_string);
+ size = atol (count_string);
+ DEBUG1 ("rmtd: W %s\n", count_string);
- case 'I':
- getstring (op);
- getstring (count);
- DEBUG2 ("rmtd: I %s %s\n", op, count);
-#ifdef MTIOCTOP
+ prepare_input_buffer (STDIN_FILENO, size);
+ for (counter = 0; counter < size; counter += status)
+ {
+ status = safe_read (STDIN_FILENO, &record_buffer[counter],
+ size - counter);
+ if (status == SAFE_READ_ERROR || status == 0)
+ {
+ DEBUG (_("rmtd: Premature eof\n"));
+
+ report_error_message (N_("Premature end of file"));
+ return EXIT_FAILURE; /* exit status used to be 2 */
+ }
+ }
+ status = full_write (tape, record_buffer, size);
+ if (status != size)
+ goto ioerror;
+ goto respond;
+ }
+
+ case 'R':
{
- struct mtop mtop;
- mtop.mt_op = atoi (op);
- mtop.mt_count = atoi (count);
- if (ioctl (tape, MTIOCTOP, (char *) &mtop) < 0)
+ char count_string[STRING_SIZE];
+ size_t size;
+
+ get_string (count_string);
+ DEBUG1 ("rmtd: R %s\n", count_string);
+
+ size = atol (count_string);
+ prepare_input_buffer (-1, size);
+ status = safe_read (tape, record_buffer, size);
+ if (status == SAFE_READ_ERROR)
goto ioerror;
- rval = mtop.mt_count;
+ sprintf (reply_buffer, "A%lu\n", (unsigned long int) 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"));
+ return EXIT_FAILURE;
+ }
+ count = nc;
+ }
+ }
+
+ mtop.mt_count = count;
+ if (mtop.mt_count != count)
+ {
+ report_error_message (N_("Seek offset out of range"));
+ return EXIT_FAILURE;
+ }
+ mtop.mt_op = atoi (operation_string);
+
+ if (ioctl (tape, MTIOCTOP, (char *) &mtop) < 0)
+ goto ioerror;
+ }
#endif
- goto respond;
+ goto respond;
+ }
case 'S': /* status */
- DEBUG ("rmtd: S\n");
{
+ DEBUG ("rmtd: S\n");
+
#ifdef MTIOCGET
- struct mtget mtget;
- if (ioctl (tape, MTIOCGET, (char *) &mtget) < 0)
- goto ioerror;
- rval = sizeof (mtget);
- (void) sprintf (resp, "A%d\n", rval);
- (void) write (1, resp, strlen (resp));
- (void) write (1, (char *) &mtget, sizeof (mtget));
+ {
+ 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", c);
- exit (3);
- }
-respond:
- DEBUG1 ("rmtd: A %d\n", rval);
- (void) sprintf (resp, "A%d\n", rval);
- (void) write (1, resp, strlen (resp));
- goto top;
-ioerror:
- error (errno);
- goto top;
-}
+ DEBUG1 (_("rmtd: Garbage command %c\n"), command);
-void
-getstring (bp)
- char *bp;
-{
- int i;
- char *cp = bp;
-
- for (i = 0; i < SSIZE; i++)
- {
- if (read (0, cp + i, 1) != 1)
- exit (0);
- if (cp[i] == '\n')
- break;
+ report_error_message (N_("Garbage command"));
+ return EXIT_FAILURE; /* exit status used to be 3 */
}
- cp[i] = '\0';
-}
-char *
-checkbuf (record, size)
- char *record;
- int size;
-{
- if (size <= maxrecsize)
- return (record);
- if (record != 0)
- free (record);
- record = malloc (size);
- if (record == 0)
- {
- DEBUG ("rmtd: cannot allocate buffer space\n");
- exit (4);
- }
- maxrecsize = size;
-#ifdef SO_RCVBUF
- while (size > 1024 &&
- setsockopt (0, SOL_SOCKET, SO_RCVBUF, (char *) &size, sizeof (size)) < 0)
- size -= 1024;
-#else
- size = 1 + ((size - 1) % 1024);
-#endif
- return (record);
-}
+respond:
+ DEBUG1 ("rmtd: A %ld\n", (long) status);
-void
-error (num)
- int num;
-{
+ sprintf (reply_buffer, "A%ld\n", (long) status);
+ full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
+ goto top;
- DEBUG2 ("rmtd: E %d (%s)\n", num, sys_errlist[num]);
- (void) sprintf (resp, "E%d\n%s\n", num, sys_errlist[num]);
- (void) write (1, resp, strlen (resp));
+ioerror:
+ report_numbered_error (errno);
+ goto top;
}
projects / chaz / tar / blobdiff