/* Remote connection server.
- Copyright 1994, 1995, 1996, 1997, 1999 Free Software Foundation, Inc.
+
+ 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
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. */
+ MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. */
#include "system.h"
-#include "safe-read.h"
+#include <localedir.h>
+#include <safe-read.h>
+#include <full-write.h>
+#include <getopt.h>
#include <sys/socket.h>
#ifndef EXIT_FAILURE
static int tape = -1;
/* Buffer containing transferred data, and its allocated size. */
-static char *record_buffer = NULL;
-static size_t allocated_size = 0;
+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 = NULL;
+static FILE *debug_file;
#define DEBUG(File) \
if (debug_file) fprintf(debug_file, File)
#define DEBUG2(File, Arg1, Arg2) \
if (debug_file) fprintf(debug_file, File, Arg1, Arg2)
-/*------------------------------------------------.
-| Return an error string, given an error number. |
-`------------------------------------------------*/
-
+/* Return an error string, given an error number. */
#if HAVE_STRERROR
# ifndef strerror
char *strerror ();
# define strerror private_strerror
#endif
-/*---.
-| ? |
-`---*/
-
static void
report_error_message (const char *string)
{
full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
}
-/*---.
-| ? |
-`---*/
-
static void
report_numbered_error (int 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++)
+ for (counter = 0; ; counter++)
{
if (safe_read (STDIN_FILENO, string + counter, 1) != 1)
exit (EXIT_SUCCESS);
- if (string[counter] == '\n')
+ if (string[counter] == '\n' || counter == STRING_SIZE - 1)
break;
}
string[counter] = '\0';
}
-/*---.
-| ? |
-`---*/
-
static void
-prepare_record_buffer (size_t size)
+prepare_input_buffer (int fd, size_t size)
{
if (size <= allocated_size)
return;
record_buffer = malloc (size);
- if (record_buffer == NULL)
+ if (! record_buffer)
{
DEBUG (_("rmtd: Cannot allocate buffer space\n"));
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);
+ 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
}
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},
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 (int argc, char *const *argv)
{
char command;
- long status;
+ size_t status;
- /* FIXME: Localisation is meaningless, unless --help and --version are
- locally used. Localisation would be best accomplished by the calling
+ /* 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];
bindtextdomain (PACKAGE, LOCALEDIR);
textdomain (PACKAGE);
- /* FIXME: Implement --help and --version as for any other GNU program. */
+ switch (getopt_long (argc, argv, "", long_opts, NULL))
+ {
+ 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;
+ }
- argc--, argv++;
- if (argc > 0)
+ if (optind < argc)
{
- debug_file = fopen (*argv, "w");
+ if (optind != argc - 1)
+ usage (EXIT_FAILURE);
+ debug_file = fopen (argv[optind], "w");
if (debug_file == 0)
{
report_numbered_error (errno);
- exit (EXIT_FAILURE);
+ return EXIT_FAILURE;
}
- setbuf (debug_file, NULL);
+ setbuf (debug_file, 0);
}
top:
- errno = 0; /* FIXME: errno should be read-only */
+ errno = 0;
status = 0;
if (safe_read (STDIN_FILENO, &command, 1) != 1)
- exit (EXIT_SUCCESS);
+ return EXIT_SUCCESS;
switch (command)
{
if (c10 / 10 != count || (negative ? c10 < nc : nc < c10))
{
report_error_message (N_("Seek offset out of range"));
- exit (EXIT_FAILURE);
+ return EXIT_FAILURE;
}
count = nc;
}
case 2: whence = SEEK_END; break;
default:
report_error_message (N_("Seek direction out of range"));
- exit (EXIT_FAILURE);
+ return EXIT_FAILURE;
}
count = lseek (tape, count, whence);
if (count < 0)
do
*--p = '0' + (int) (count % 10);
while ((count /= 10) != 0);
-
+
DEBUG1 ("rmtd: A %s\n", p);
sprintf (reply_buffer, "A%s\n", p);
size = atol (count_string);
DEBUG1 ("rmtd: W %s\n", count_string);
- prepare_record_buffer (size);
+ 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 <= 0)
+ if (status == SAFE_READ_ERROR || status == 0)
{
DEBUG (_("rmtd: Premature eof\n"));
report_error_message (N_("Premature end of file"));
- exit (EXIT_FAILURE); /* exit status used to be 2 */
+ return EXIT_FAILURE; /* exit status used to be 2 */
}
}
status = full_write (tape, record_buffer, size);
- if (status < 0)
+ if (status != size)
goto ioerror;
goto respond;
}
DEBUG1 ("rmtd: R %s\n", count_string);
size = atol (count_string);
- prepare_record_buffer (size);
+ prepare_input_buffer (-1, size);
status = safe_read (tape, record_buffer, size);
- if (status < 0)
+ if (status == SAFE_READ_ERROR)
goto ioerror;
- sprintf (reply_buffer, "A%ld\n", status);
+ 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, (size_t) status);
+ full_write (STDOUT_FILENO, record_buffer, status);
goto top;
}
/* 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 (c10 / 10 != count || (negative ? c10 < nc : nc < c10))
{
report_error_message (N_("Seek offset out of range"));
- exit (EXIT_FAILURE);
+ return EXIT_FAILURE;
}
count = nc;
}
if (mtop.mt_count != count)
{
report_error_message (N_("Seek offset out of range"));
- exit (EXIT_FAILURE);
+ return EXIT_FAILURE;
}
mtop.mt_op = atoi (operation_string);
if (ioctl (tape, MTIOCGET, (char *) &operation) < 0)
goto ioerror;
- status = sizeof (operation);
- sprintf (reply_buffer, "A%ld\n", status);
+ 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));
+ full_write (STDOUT_FILENO, (char *) &operation, sizeof operation);
}
#endif
goto top;
DEBUG1 (_("rmtd: Garbage command %c\n"), command);
report_error_message (N_("Garbage command"));
- exit (EXIT_FAILURE); /* exit status used to be 3 */
+ return EXIT_FAILURE; /* exit status used to be 3 */
}
respond:
- DEBUG1 ("rmtd: A %ld\n", status);
+ DEBUG1 ("rmtd: A %ld\n", (long) status);
- sprintf (reply_buffer, "A%ld\n", status);
+ sprintf (reply_buffer, "A%ld\n", (long) status);
full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
goto top;