added yum-2.6.0-chroot.patch
[util-vserver.git] / util-vserver / lib_internal / unify-copy.c
index c67b48e..6b93089 100644 (file)
 #endif
 
 #include "unify.h"
-#include <sys/stat.h>
+#include "util.h"
+
 #include <unistd.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <setjmp.h>
+#include <signal.h>
+#include <sys/stat.h>
+#include <sys/mman.h>
 
 #define ENSC_WRAPPERS_IO       1
 #include <wrappers.h>
 
+#define MMAP_BLOCKSIZE         (16 * 1024*1024)
+
+#ifndef   TESTSUITE_COPY_CODE
+#  define TESTSUITE_COPY_CODE  do { } while (false)
+#endif
+
 static inline bool
 verifySource(int fd, struct stat const *exp_stat)
 {
@@ -58,27 +69,125 @@ copyLnk(char const *src, char const *dst)
   }
 }
 
+static sigjmp_buf              bus_error_restore;
+static volatile sig_atomic_t   bus_error;
+
+static void
+handlerSIGBUS(int UNUSED num)
+{
+  bus_error = 1;
+  siglongjmp(bus_error_restore, 1);
+}
+
+static void
+copyMem(void *dst_v, void const *src_v, size_t len_v)
+{
+#if 1
+  int          *dst = dst_v;
+  int const    *src = src_v;
+  size_t       len  = len_v / sizeof(int);
+  size_t       rest = len_v - sizeof(int)*len;
+  size_t       i=0;
+
+  for (; i<len; ++i) {
+    if (*src != 0) *dst = *src;
+    ++dst;
+    ++src;
+  }
+
+  char         *dst_c = (void *)(dst);
+  char const   *src_c = (void const *)(src);
+
+  for (i=0; i<rest; ++i) {
+    if (*src_c != 0) *dst_c = *src_c;
+    ++dst_c;
+    ++src_c;
+  }
+#else
+  memcpy(dst_v, src_v, len_v);
+#endif  
+}
+
+static UNUSED bool
+copyMMap(int in_fd, int out_fd)
+{
+  off_t                        in_len   = lseek(in_fd, 0, SEEK_END);
+  void const * volatile in_buf   = 0;
+  void       * volatile        out_buf  = 0;
+  
+  loff_t volatile      buf_size = 0;
+  bool   volatile      res      = false;
+
+  if (in_len==-1) return false;
+  if (in_len>0 &&
+      (lseek(out_fd, in_len-1, SEEK_SET)==-1 ||
+       write(out_fd, "\0",     1)!=1))         // create sparse file
+    return false;
+  
+  bus_error = 0;
+  if (sigsetjmp(bus_error_restore, 1)==0) {
+    off_t              offset   = 0;
+
+    while (offset < in_len) {
+      buf_size = in_len - offset;
+      if (buf_size > MMAP_BLOCKSIZE) buf_size = MMAP_BLOCKSIZE;
+      
+      if ((in_buf  = mmap(0, buf_size, PROT_READ,  MAP_SHARED,  in_fd, offset))==0 ||
+         (out_buf = mmap(0, buf_size, PROT_WRITE, MAP_SHARED, out_fd, offset))==0) {
+       perror("mmap()");
+       goto out;
+      }
+
+      offset  += buf_size;
+      madvise(const_cast(void *)(in_buf),  buf_size, MADV_SEQUENTIAL);
+      madvise(out_buf,                     buf_size, MADV_SEQUENTIAL);
+
+      TESTSUITE_COPY_CODE;
+      copyMem(out_buf, in_buf, buf_size);
+
+      munmap(const_cast(void *)(in_buf),  buf_size);  in_buf = 0;
+      munmap(out_buf,                     buf_size); out_buf = 0;
+    }
+
+    res = true;
+  }
+
+  out:
+  if (in_buf !=0) munmap(const_cast(void *)(in_buf),  buf_size);
+  if (out_buf!=0) munmap(out_buf,                     buf_size);
+
+  return res;
+}
+
 static inline bool
 copyReg(char const *src, struct stat const *src_stat,
        char const *dst)
 {
   int          in_fd  = open(src, O_RDONLY|O_NOCTTY|O_NONBLOCK|O_NOFOLLOW|O_LARGEFILE);
-  int          out_fd = in_fd==-1 ? -1 : open(dst, O_CREAT|O_EXCL, 0200);
+  int          out_fd = in_fd==-1 ? -1 : open(dst, O_RDWR|O_CREAT|O_EXCL, 0200);
   bool         res    = false;
   
   if (in_fd==-1 || out_fd==-1 ||
       !verifySource(in_fd, src_stat)) goto err;
 
+#if 0  
   for (;;) {
     char       buf[2048];
     ssize_t    l = read(in_fd, buf, sizeof buf);
     if (l==-1) goto err;
     if (l==0)  break;
-    if (!WwriteAll(out_fd, buf, l, 0)) return false;
+    if (!WwriteAll(out_fd, buf, l, 0)) goto err;
   }
 
   res = true;
-  
+#else
+  void         (*old_handler)(int) = signal(SIGBUS, handlerSIGBUS);
+
+  res = copyMMap(in_fd, out_fd);
+
+  signal(SIGBUS, old_handler);
+#endif
+
   err:
   if (out_fd!=-1 && close(out_fd)==-1) res=false;
   if (in_fd!=-1  && close(in_fd)==-1)  res=false;