]> Dogcows Code - chaz/tar/blobdiff - src/buffer.c
(options): Add --restrict option. Use macros for option grouping.
[chaz/tar] / src / buffer.c
index f5ebfb7eb5ebb1e571aaad8bddc615f0ab6c8ebf..586b1f9536fc444605e9c8ac6897aa65a0b0cd1c 100644 (file)
@@ -84,6 +84,9 @@ extern bool time_to_start_writing;
 
 bool write_archive_to_stdout;
 
+void (*flush_write_ptr) (size_t);
+void (*flush_read_ptr) (void);
+
 \f
 char *volume_label;
 char *continued_file_name;
@@ -674,6 +677,7 @@ _flush_read (void)
 void
 flush_archive (void)
 {
+  size_t buffer_level = current_block->buffer - record_start->buffer;
   record_start_block += record_end - record_start;
   current_block = record_start;
   record_end = record_start + blocking_factor;
@@ -692,7 +696,7 @@ flush_archive (void)
       break;
 
     case ACCESS_WRITE:
-      flush_write ();
+      flush_write_ptr (buffer_level);
       break;
 
     case ACCESS_UPDATE:
@@ -784,7 +788,11 @@ void
 close_archive (void)
 {
   if (time_to_start_writing || access_mode == ACCESS_WRITE)
-    flush_archive ();
+    {
+      flush_archive ();
+      if (current_block > record_start)
+       flush_archive ();
+    }
 
   sys_drain_input_pipe ();
 
@@ -855,6 +863,99 @@ increase_volume_number ()
   volno++;
 }
 
+void
+change_tape_menu (FILE *read_file)
+{
+  char *input_buffer = NULL;
+  size_t size = 0;
+
+  while (1)
+    {
+      fputc ('\007', stderr);
+      fprintf (stderr,
+              _("Prepare volume #%d for %s and hit return: "),
+              global_volno + 1, quote (*archive_name_cursor));
+      fflush (stderr);
+      
+      if (getline (&input_buffer, &size, read_file) <= 0)
+       {
+         WARN ((0, 0, _("EOF where user reply was expected")));
+         
+         if (subcommand_option != EXTRACT_SUBCOMMAND
+             && subcommand_option != LIST_SUBCOMMAND
+             && subcommand_option != DIFF_SUBCOMMAND)
+           WARN ((0, 0, _("WARNING: Archive is incomplete")));
+         
+         fatal_exit ();
+       }
+      
+      if (input_buffer[0] == '\n'
+         || input_buffer[0] == 'y'
+         || input_buffer[0] == 'Y')
+       break;
+
+      switch (input_buffer[0])
+       {
+       case '?':
+         {
+           fprintf (stderr, _("\
+ n [name]      Give a new file name for the next (and subsequent) volume(s)\n\
+ q             Abort tar\n\
+ y or newline  Continue operation\n"));
+            if (!restrict_option)
+              fprintf (stderr, _(" !             Spawn a subshell\n"));
+           fprintf (stderr, _(" ?             Print this list\n"));
+         }
+         break;
+
+       case 'q':
+         /* Quit.  */
+         
+         WARN ((0, 0, _("No new volume; exiting.\n")));
+         
+         if (subcommand_option != EXTRACT_SUBCOMMAND
+             && subcommand_option != LIST_SUBCOMMAND
+             && subcommand_option != DIFF_SUBCOMMAND)
+           WARN ((0, 0, _("WARNING: Archive is incomplete")));
+         
+         fatal_exit ();
+         
+       case 'n':
+         /* Get new file name.  */
+         
+         {
+           char *name;
+           char *cursor;
+           
+           for (name = input_buffer + 1;
+                *name == ' ' || *name == '\t';
+                name++)
+             ;
+           
+           for (cursor = name; *cursor && *cursor != '\n'; cursor++)
+             ;
+           *cursor = '\0';
+           
+           /* FIXME: the following allocation is never reclaimed.  */
+           *archive_name_cursor = xstrdup (name);
+         }
+         break;
+         
+       case '!':
+         if (!restrict_option)
+           {
+             sys_spawn_shell ();
+             break;
+           }
+         /* FALL THROUGH */
+
+       default:
+         fprintf (stderr, _("Invalid input. Type ? for help.\n"));
+       }
+    }
+  free (input_buffer);
+}
+
 /* We've hit the end of the old volume.  Close it and open the next one.
    Return nonzero on success.
 */
@@ -896,88 +997,12 @@ new_volume (enum access_mode mode)
        {
          if (volno_file_option)
            closeout_volume_number ();
-         if (system (info_script_option) != 0)
+         if (sys_exec_info_script (archive_name_cursor, global_volno+1))
            FATAL_ERROR ((0, 0, _("%s command failed"),
                          quote (info_script_option)));
        }
       else
-       while (1)
-         {
-           char input_buffer[80];
-
-           fputc ('\007', stderr);
-           fprintf (stderr,
-                    _("Prepare volume #%d for %s and hit return: "),
-                    global_volno + 1, quote (*archive_name_cursor));
-           fflush (stderr);
-
-           if (fgets (input_buffer, sizeof input_buffer, read_file) == 0)
-             {
-               WARN ((0, 0, _("EOF where user reply was expected")));
-
-               if (subcommand_option != EXTRACT_SUBCOMMAND
-                   && subcommand_option != LIST_SUBCOMMAND
-                   && subcommand_option != DIFF_SUBCOMMAND)
-                 WARN ((0, 0, _("WARNING: Archive is incomplete")));
-
-               fatal_exit ();
-             }
-           if (input_buffer[0] == '\n'
-               || input_buffer[0] == 'y'
-               || input_buffer[0] == 'Y')
-             break;
-
-           switch (input_buffer[0])
-             {
-             case '?':
-               {
-                 /* FIXME: Might it be useful to disable the '!' command? */
-                 fprintf (stderr, _("\
- n [name]   Give a new file name for the next (and subsequent) volume(s)\n\
- q          Abort tar\n\
- !          Spawn a subshell\n\
- ?          Print this list\n"));
-               }
-               break;
-
-             case 'q':
-               /* Quit.  */
-
-               WARN ((0, 0, _("No new volume; exiting.\n")));
-
-               if (subcommand_option != EXTRACT_SUBCOMMAND
-                   && subcommand_option != LIST_SUBCOMMAND
-                   && subcommand_option != DIFF_SUBCOMMAND)
-                 WARN ((0, 0, _("WARNING: Archive is incomplete")));
-
-               fatal_exit ();
-
-             case 'n':
-               /* Get new file name.  */
-
-               {
-                 char *name = &input_buffer[1];
-                 char *cursor;
-
-                 for (name = input_buffer + 1;
-                      *name == ' ' || *name == '\t';
-                      name++)
-                   ;
-
-                 for (cursor = name; *cursor && *cursor != '\n'; cursor++)
-                   ;
-                 *cursor = '\0';
-
-                 /* FIXME: the following allocation is never reclaimed.  */
-                 *archive_name_cursor = xstrdup (name);
-               }
-               break;
-
-             case '!':
-               sys_spawn_shell ();
-               break;
-             }
-         }
+       change_tape_menu (read_file);
     }
 
   if (strcmp (archive_name_cursor[0], "-") == 0)
@@ -1021,13 +1046,26 @@ new_volume (enum access_mode mode)
 
   return true;
 }
-      
+
+static bool
+read_header0 ()
+{
+  enum read_header rc = read_header (false);
+
+  if (rc == HEADER_SUCCESS)
+    {
+      set_next_block_after (current_header);
+      return true;
+    }
+  ERROR ((0, 0, _("This does not look like a tar archive")));
+  return false;
+}
+
 bool
 try_new_volume ()
 {
   size_t status;
-  enum read_header rc;
-  union block *block;
+  union block *header;
 
   switch (subcommand_option)
     {
@@ -1051,49 +1089,51 @@ try_new_volume ()
   if (status != record_size)
     short_read (status);
 
- again:
-  block = current_block;
-  rc = read_header (true);
-  switch (rc)
+  header = find_next_block ();
+  if (!header)
+    return false;
+  switch (header->header.typeflag)
     {
-    case HEADER_SUCCESS_EXTENDED:
-      if (current_header->header.typeflag == XGLTYPE)
-       {
-         struct tar_stat_info dummy;
-         xheader_read (current_header,
-                       OFF_FROM_HEADER (current_header->header.size));
-         xheader_decode_global ();
-         xheader_destroy (&extended_header);
-         tar_stat_init (&dummy);
-         xheader_decode (&dummy); /* decodes values from the global header */
-         tar_stat_destroy (&dummy);
-       }
-      break;
+    case XGLTYPE:
+      {
+       struct tar_stat_info dummy;
+       if (!read_header0 ())
+         return false;
+       tar_stat_init (&dummy);
+       xheader_decode (&dummy); /* decodes values from the global header */
+       tar_stat_destroy (&dummy);
+       if (!real_s_name)
+         {
+           /* We have read the extended header of the first member in
+              this volume. Put it back, so next read_header works as
+              expected. */
+           current_block = record_start;
+         }
+       break;
+      }
       
-    case HEADER_SUCCESS:
-      if (current_header->header.typeflag == GNUTYPE_VOLHDR)
-       assign_string (&volume_label, current_header->header.name);
-      else if (current_header->header.typeflag == GNUTYPE_MULTIVOL)
-       {
-         assign_string (&continued_file_name, current_header->header.name);
-         continued_file_size =
-              UINTMAX_FROM_HEADER (current_header->header.size);
-         continued_file_offset =
-              UINTMAX_FROM_HEADER (current_header->oldgnu_header.offset);
-       }
-      else
+    case GNUTYPE_VOLHDR:
+      if (!read_header0 ())
+       return false;
+      assign_string (&volume_label, current_header->header.name);
+      set_next_block_after (header);
+      header = find_next_block ();
+      if (header->header.typeflag != GNUTYPE_MULTIVOL)
        break;
-      set_next_block_after (current_header);
-      goto again;
-
-    case HEADER_ZERO_BLOCK:
-    case HEADER_END_OF_FILE:
-    case HEADER_FAILURE:
-      current_block = block;
+      /* FALL THROUGH */
+      
+    case GNUTYPE_MULTIVOL:
+      if (!read_header0 ())
+       return false;
+      assign_string (&continued_file_name, current_header->header.name);
+      continued_file_size =
+       UINTMAX_FROM_HEADER (current_header->header.size);
+      continued_file_offset =
+       UINTMAX_FROM_HEADER (current_header->oldgnu_header.offset);
       break;
-
+      
     default:
-      abort ();
+      break;
     }
 
   if (real_s_name)
@@ -1219,6 +1259,38 @@ add_volume_label (void)
   free (s);
 }
 
+static void
+add_chunk_header ()
+{
+  if (archive_format == POSIX_FORMAT)
+    {
+      off_t block_ordinal;
+      union block *blk;
+      struct tar_stat_info st;
+      static size_t real_s_part_no; /* FIXME */
+
+      real_s_part_no++;
+      memset (&st, 0, sizeof st);
+      st.orig_file_name = st.file_name = real_s_name;
+      st.stat.st_mode = S_IFREG|S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH;
+      st.stat.st_uid = getuid ();
+      st.stat.st_gid = getgid ();
+      st.orig_file_name = xheader_format_name (&st,
+                                              "%d/GNUFileParts.%p/%f.%n",
+                                              real_s_part_no);
+      st.file_name = st.orig_file_name;
+      st.archive_file_size = st.stat.st_size = real_s_sizeleft;
+      
+      block_ordinal = current_block_ordinal ();
+      blk = start_header (&st);
+      free (st.orig_file_name);
+      if (!blk)
+       abort (); /* FIXME */
+      finish_header (&st, blk, block_ordinal);
+    }
+}
+
+
 /* Add a volume label to the current archive */
 static void
 write_volume_label (void)
@@ -1350,7 +1422,7 @@ simple_flush_read (void)
 
 /* Simple flush write (no multi-volume or label extensions) */
 static void
-simple_flush_write (void)
+simple_flush_write (size_t level __attribute__((unused)))
 {
   ssize_t status;
 
@@ -1431,20 +1503,20 @@ _gnu_flush_read (void)
 static void
 gnu_flush_read (void)
 {
-  flush_read = simple_flush_read; /* Avoid recursion */
+  flush_read_ptr = simple_flush_read; /* Avoid recursion */
   _gnu_flush_read ();
-  flush_read = gnu_flush_read; 
+  flush_read_ptr = gnu_flush_read; 
 }
 
 static void
-_gnu_flush_write (void)
+_gnu_flush_write (size_t buffer_level)
 {
   ssize_t status;
   union block *header;
   char *copy_ptr;
   size_t copy_size;
   size_t bufsize;
-  
+    
   status = _flush_write ();
   if (status != record_size && !multi_volume_option)
     archive_write_error (status);
@@ -1469,14 +1541,14 @@ _gnu_flush_write (void)
     return;
 
   xheader_destroy (&extended_header);
-  
+
   increase_volume_number ();
   if (totals_option)
     prev_written += bytes_written;
   bytes_written = 0;
 
   copy_ptr = record_start->buffer + status;
-  copy_size = record_size - status;
+  copy_size = buffer_level - status;
   /* Switch to the next buffer */
   record_index = !record_index;
   init_buffer ();
@@ -1487,7 +1559,10 @@ _gnu_flush_write (void)
   if (real_s_name)
     add_multi_volume_header ();
 
-  header = write_extended (XGLTYPE, NULL, find_next_block ());
+  write_extended (true, NULL, find_next_block ());
+  if (real_s_name)
+    add_chunk_header ();
+  header = find_next_block ();
   bufsize = available_space_after (header);
   while (bufsize < copy_size)
     {
@@ -1505,18 +1580,30 @@ _gnu_flush_write (void)
 }
 
 static void
-gnu_flush_write (void)
+gnu_flush_write (size_t buffer_level)
 {
-  flush_write = simple_flush_write; /* Avoid recursion */
-  _gnu_flush_write ();
-  flush_write = gnu_flush_write; 
+  flush_write_ptr = simple_flush_write; /* Avoid recursion */
+  _gnu_flush_write (buffer_level);
+  flush_write_ptr = gnu_flush_write; 
 }
-  
+
+void
+flush_read ()
+{
+  flush_read_ptr ();
+}
+
+void
+flush_write ()
+{
+  flush_write_ptr (record_size);
+}    
+
 void
 open_archive (enum access_mode wanted_access)
 {
-  flush_read = gnu_flush_read;
-  flush_write = gnu_flush_write;
+  flush_read_ptr = gnu_flush_read;
+  flush_write_ptr = gnu_flush_write;
 
   _open_archive (wanted_access);
   switch (wanted_access)
@@ -1533,3 +1620,4 @@ open_archive (enum access_mode wanted_access)
       break;
     }
 }
+
This page took 0.0316 seconds and 4 git commands to generate.