Merge pull request #2802 from BrzVlad/feature-evacuation-opt2

[mono.git] / mono / utils / mono-filemap.c

diff --git a/mono/utils/mono-filemap.c b/mono/utils/mono-filemap.c
index 0068607f827fdf238032ee5a7a4cecf5207d322b..cd93eaff39fc9aacdee597c3a2d69c2c5b9fded6 100644 (file)
--- a/mono/utils/mono-filemap.c
+++ b/mono/utils/mono-filemap.c
@@ -5,6 +5,7 @@
  *   Paolo Molaro (lupus@ximian.com)
  *
  * Copyright 2008-2008 Novell, Inc.
+ * Licensed under the MIT license. See LICENSE file in the project root for full license information.
  */
 
 #include "config.h"
@@ -14,7 +15,9 @@
 #endif
 #include <fcntl.h>
 #include <string.h>
+#ifdef HAVE_UNISTD_H
 #include <unistd.h>
+#endif
 #include <stdlib.h>
 #include <stdio.h>
 
@@ -23,14 +26,28 @@
 MonoFileMap *
 mono_file_map_open (const char* name)
 {
-       return (MonoFileMap *)fopen (name, "rb");
+#ifdef WIN32
+       gunichar2 *wname = g_utf8_to_utf16 (name, -1, 0, 0, 0);
+       MonoFileMap *result;
+
+       if (wname == NULL)
+               return NULL;
+       result = (MonoFileMap *) _wfopen ((wchar_t *) wname, L"rb");
+       g_free (wname);
+       return result;
+#else
+       int fd = open (name, O_RDONLY);
+       if (fd < 0)
+               return NULL;
+       return (MonoFileMap *)(size_t)fd;
+#endif
 }
 
 guint64 
 mono_file_map_size (MonoFileMap *fmap)
 {
        struct stat stat_buf;
-       if (fstat (fileno ((FILE*)fmap), &stat_buf) < 0)
+       if (fstat (mono_file_map_fd (fmap), &stat_buf) < 0)
                return 0;
        return stat_buf.st_size;
 }
@@ -38,32 +55,73 @@ mono_file_map_size (MonoFileMap *fmap)
 int
 mono_file_map_fd (MonoFileMap *fmap)
 {
+#ifdef WIN32
        return fileno ((FILE*)fmap);
+#else
+       return (int)(size_t)fmap;
+#endif
 }
 
 int 
 mono_file_map_close (MonoFileMap *fmap)
 {
+#ifdef WIN32
        return fclose ((FILE*)fmap);
+#else
+       return close (mono_file_map_fd (fmap));
+#endif
+}
+
+#if !defined (HOST_WIN32)
+
+static mono_file_map_alloc_fn alloc_fn = (mono_file_map_alloc_fn) malloc;
+static mono_file_map_release_fn release_fn = (mono_file_map_release_fn) free;
+
+void
+mono_file_map_set_allocator (mono_file_map_alloc_fn alloc, mono_file_map_release_fn release)
+{
+       alloc_fn = alloc == NULL     ? (mono_file_map_alloc_fn) malloc : alloc;
+       release_fn = release == NULL ? (mono_file_map_release_fn) free : release;
 }
 
-#ifndef HAVE_MMAP
 void *
-mono_file_map (size_t length, int flags, int fd, guint64 offset, void **ret_handle)
+mono_file_map_fileio (size_t length, int flags, int fd, guint64 offset, void **ret_handle)
 {
        guint64 cur_offset;
        size_t bytes_read;
-       void *ptr = malloc (length);
+       void *ptr = (*alloc_fn) (length);
        if (!ptr)
                return NULL;
        cur_offset = lseek (fd, 0, SEEK_CUR);
        if (lseek (fd, offset, SEEK_SET) != offset) {
-               free (ptr);
+               (*release_fn) (ptr);
                return NULL;
        }
        bytes_read = read (fd, ptr, length);
+       if (bytes_read != length)
+               return NULL;
        lseek (fd, cur_offset, SEEK_SET);
        *ret_handle = NULL;
        return ptr;
 }
+
+int
+mono_file_unmap_fileio (void *addr, void *handle)
+{
+       (*release_fn) (addr);
+       return 0;
+}
+#if !defined(HAVE_MMAP)
+void *
+mono_file_map (size_t length, int flags, int fd, guint64 offset, void **ret_handle)
+{
+       return mono_file_map_fileio (length, flags, fd, offset, ret_handle);
+}
+
+int
+mono_file_unmap (void *addr, void *handle)
+{
+       return mono_file_unmap_fileio(addr, handle);
+}
+#endif
 #endif