]> Dogcows Code - chaz/tar/blobdiff - src/rmt.c
Rename full_read to safe_read.
[chaz/tar] / src / rmt.c
index 27b3f094ca2ca79681a83694c79e7d7e4728524a..f5e036cd6ceddae21e80b1b58aa126b6c6ad3633 100644 (file)
--- a/src/rmt.c
+++ b/src/rmt.c
@@ -1,5 +1,5 @@
 /* Remote connection server.
-   Copyright (C) 1994, 1995, 1996 Free Software Foundation, Inc.
+   Copyright 1994, 1995, 1996, 1997, 1999 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
@@ -13,7 +13,7 @@
 
    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 Place - Suite 330, Boston, MA 02111-1307, USA.  */
+   59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
 
 /* Copyright (C) 1983 Regents of the University of California.
    All rights reserved.
@@ -51,7 +51,7 @@ static int tape = -1;
 
 /* Buffer containing transferred data, and its allocated size.  */
 static char *record_buffer = NULL;
-static int allocated_size = -1;
+static size_t allocated_size = 0;
 
 /* Buffer for constructing the reply.  */
 static char reply_buffer[BUFSIZ];
@@ -81,12 +81,12 @@ char *strerror ();
 static char *
 private_strerror (int errnum)
 {
-  extern const char *const sys_errlist[];
+  extern char *sys_errlist[];
   extern int sys_nerr;
 
   if (errnum > 0 && errnum <= sys_nerr)
-    return sys_errlist[errnum];
-  return N_("Unknown system error");
+    return _(sys_errlist[errnum]);
+  return _("Unknown system error");
 }
 # define strerror private_strerror
 #endif
@@ -101,7 +101,7 @@ report_error_message (const char *string)
   DEBUG1 ("rmtd: E 0 (%s)\n", string);
 
   sprintf (reply_buffer, "E0\n%s\n", string);
-  write (1, reply_buffer, strlen (reply_buffer));
+  full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
 }
 
 /*---.
@@ -114,7 +114,7 @@ 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));
-  write (1, reply_buffer, strlen (reply_buffer));
+  full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
 }
 
 /*---.
@@ -128,7 +128,7 @@ get_string (char *string)
 
   for (counter = 0; counter < STRING_SIZE; counter++)
     {
-      if (read (0, string + counter, 1) != 1)
+      if (safe_read (STDIN_FILENO, string + counter, 1) != 1)
        exit (EXIT_SUCCESS);
 
       if (string[counter] == '\n')
@@ -142,7 +142,7 @@ get_string (char *string)
 `---*/
 
 static void
-prepare_record_buffer (int size)
+prepare_record_buffer (size_t size)
 {
   if (size <= allocated_size)
     return;
@@ -150,7 +150,7 @@ prepare_record_buffer (int size)
   if (record_buffer)
     free (record_buffer);
 
-  record_buffer = malloc ((size_t) size);
+  record_buffer = malloc (size);
 
   if (record_buffer == NULL)
     {
@@ -164,7 +164,9 @@ prepare_record_buffer (int size)
 
 #ifdef SO_RCVBUF
   while (size > 1024 &&
-   setsockopt (0, SOL_SOCKET, SO_RCVBUF, (char *) &size, sizeof (size)) < 0)
+        (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! */
@@ -180,7 +182,7 @@ int
 main (int argc, char *const *argv)
 {
   char command;
-  int status;
+  long status;
 
   /* FIXME: Localisation is meaningless, unless --help and --version are
      locally used.  Localisation would be best accomplished by the calling
@@ -208,7 +210,7 @@ main (int argc, char *const *argv)
 top:
   errno = 0;                   /* FIXME: errno should be read-only */
   status = 0;
-  if (read (0, &command, 1) != 1)
+  if (safe_read (STDIN_FILENO, &command, 1) != 1)
     exit (EXIT_SUCCESS);
 
   switch (command)
@@ -227,34 +229,7 @@ top:
        if (tape >= 0)
          close (tape);
 
-#if defined (i386) && defined (AIX)
-
-       /* This is alleged to fix a byte ordering problem.  I'm quite
-          suspicious if it's right. -- mib.  */
-
-       {
-         int old_mode = atoi (mode_string);
-         int new_mode = 0;
-
-         if ((old_mode & 3) == 0)
-           new_mode |= O_RDONLY;
-         if (old_mode & 1)
-           new_mode |= O_WRONLY;
-         if (old_mode & 2)
-           new_mode |= O_RDWR;
-         if (old_mode & 0x0008)
-           new_mode |= O_APPEND;
-         if (old_mode & 0x0200)
-           new_mode |= O_CREAT;
-         if (old_mode & 0x0400)
-           new_mode |= O_TRUNC;
-         if (old_mode & 0x0800)
-           new_mode |= O_EXCL;
-         tape = open (device_string, new_mode, 0666);
-       }
-#else
        tape = open (device_string, atoi (mode_string), 0666);
-#endif
        if (tape < 0)
          goto ioerror;
        goto respond;
@@ -277,32 +252,86 @@ top:
       {
        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);
 
-       status
-         = lseek (tape, (off_t) atol (count_string), atoi (position_string));
-       if (status < 0)
+       /* 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;
-       goto respond;
+
+       /* 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];
-       int size;
-       int counter;
+       size_t size;
+       size_t counter;
 
        get_string (count_string);
-       size = atoi (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 = read (0, &record_buffer[counter], size - counter);
+           status = safe_read (STDIN_FILENO, &record_buffer[counter],
+                               size - counter);
            if (status <= 0)
              {
                DEBUG (_("rmtd: Premature eof\n"));
@@ -311,7 +340,7 @@ top:
                exit (EXIT_FAILURE); /* exit status used to be 2 */
              }
          }
-       status = write (tape, record_buffer, size);
+       status = full_write (tape, record_buffer, size);
        if (status < 0)
          goto ioerror;
        goto respond;
@@ -320,19 +349,19 @@ top:
     case 'R':
       {
        char count_string[STRING_SIZE];
-       int size;
+       size_t size;
 
        get_string (count_string);
        DEBUG1 ("rmtd: R %s\n", count_string);
 
-       size = atoi (count_string);
+       size = atol (count_string);
        prepare_record_buffer (size);
-       status = read (tape, record_buffer, size);
+       status = safe_read (tape, record_buffer, size);
        if (status < 0)
          goto ioerror;
-       sprintf (reply_buffer, "A%d\n", status);
-       write (1, reply_buffer, strlen (reply_buffer));
-       write (1, record_buffer, status);
+       sprintf (reply_buffer, "A%ld\n", status);
+       full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
+       full_write (STDOUT_FILENO, record_buffer, (size_t) status);
        goto top;
       }
 
@@ -348,12 +377,48 @@ top:
 #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);
-         mtop.mt_count = atoi (count_string);
+
          if (ioctl (tape, MTIOCTOP, (char *) &mtop) < 0)
            goto ioerror;
-         status = mtop.mt_count;
        }
 #endif
        goto respond;
@@ -370,9 +435,9 @@ top:
          if (ioctl (tape, MTIOCGET, (char *) &operation) < 0)
            goto ioerror;
          status = sizeof (operation);
-         sprintf (reply_buffer, "A%d\n", status);
-         write (1, reply_buffer, strlen (reply_buffer));
-         write (1, (char *) &operation, sizeof (operation));
+         sprintf (reply_buffer, "A%ld\n", status);
+         full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
+         full_write (STDOUT_FILENO, (char *) &operation, sizeof (operation));
        }
 #endif
        goto top;
@@ -386,10 +451,10 @@ top:
     }
 
 respond:
-  DEBUG1 ("rmtd: A %d\n", status);
+  DEBUG1 ("rmtd: A %ld\n", status);
 
-  sprintf (reply_buffer, "A%d\n", status);
-  write (1, reply_buffer, strlen (reply_buffer));
+  sprintf (reply_buffer, "A%ld\n", status);
+  full_write (STDOUT_FILENO, reply_buffer, strlen (reply_buffer));
   goto top;
 
 ioerror:
This page took 0.028724 seconds and 4 git commands to generate.