use $(LIBENSCVECTOR) instead of libensc_vector.a
[util-vserver.git] / util-vserver / lib_internal / unify-copy.c
index 0264723..6b93089 100644 (file)
 #define ENSC_WRAPPERS_IO       1
 #include <wrappers.h>
 
-  //#define MMAP_BLOCKSIZE             0x10000000
-#define MMAP_BLOCKSIZE         0x4000
+#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)
@@ -66,26 +69,54 @@ copyLnk(char const *src, char const *dst)
   }
 }
 
-static jmp_buf                 bus_error_restore;
+static sigjmp_buf              bus_error_restore;
 static volatile sig_atomic_t   bus_error;
 
 static void
 handlerSIGBUS(int UNUSED num)
 {
   bus_error = 1;
-  longjmp(bus_error_restore, 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 volatile const  *in_buf  = 0;
-  void volatile                *out_buf = 0;
+  void const * volatile in_buf   = 0;
+  void       * volatile        out_buf  = 0;
   
-  volatile loff_t      in_size  = 0;
-  volatile loff_t      out_size = 0;
-  volatile bool                res      = false;
+  loff_t volatile      buf_size = 0;
+  bool   volatile      res      = false;
 
   if (in_len==-1) return false;
   if (in_len>0 &&
@@ -94,36 +125,36 @@ copyMMap(int in_fd, int out_fd)
     return false;
   
   bus_error = 0;
-  if (setjmp(bus_error_restore)==0) {
-    off_t              offset = 0;
+  if (sigsetjmp(bus_error_restore, 1)==0) {
+    off_t              offset   = 0;
 
     while (offset < in_len) {
-      in_size = in_len - offset;
-      if (in_size > MMAP_BLOCKSIZE) in_size = MMAP_BLOCKSIZE;
+      buf_size = in_len - offset;
+      if (buf_size > MMAP_BLOCKSIZE) buf_size = MMAP_BLOCKSIZE;
       
-      in_buf   = mmap(0, in_size,  PROT_READ,  MAP_SHARED, in_fd,  offset);
-      if (in_buf==0)  goto out;
-
-      out_size = in_size;
-      out_buf  = mmap(0, out_size, PROT_WRITE, MAP_SHARED, out_fd, offset);
-      if (out_buf==0) goto out;
+      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  += in_size;
-      madvise(const_cast(void *)(in_buf),  in_size,  MADV_SEQUENTIAL);
-      madvise(out_buf,                     out_size, MADV_SEQUENTIAL);
+      offset  += buf_size;
+      madvise(const_cast(void *)(in_buf),  buf_size, MADV_SEQUENTIAL);
+      madvise(out_buf,                     buf_size, MADV_SEQUENTIAL);
 
-      memcpy(out_buf, in_buf, in_size);
+      TESTSUITE_COPY_CODE;
+      copyMem(out_buf, in_buf, buf_size);
 
-      munmap(const_cast(void *)(in_buf),  in_size);  in_buf  = 0;
-      munmap(out_buf,                     out_size); out_buf = 0;
+      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),  in_size);
-  if (out_buf!=0) munmap(out_buf,                     out_size);
+  if (in_buf !=0) munmap(const_cast(void *)(in_buf),  buf_size);
+  if (out_buf!=0) munmap(out_buf,                     buf_size);
 
   return res;
 }