src/utils.c

   1 /*
   2  * Copyright © 2013 Ran Benita <ran234@gmail.com>
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  */
  23
  24 #include "utils.h"
  25
  26 #ifdef HAVE_MMAP
  27
  28 #include <fcntl.h>
  29 #include <unistd.h>
  30 #include <sys/mman.h>
  31 #include <sys/stat.h>
  32 #include <sys/types.h>
  33
  34 bool
  35 map_file(FILE *file, char **string_out, size_t *size_out)
  36 {
  37     struct stat stat_buf;
  38     int fd;
  39     char *string;
  40
  41     /* Make sure to keep the errno on failure! */
  42     fd = fileno(file);
  43     if (fd < 0)
  44         return false;
  45
  46     if (fstat(fd, &stat_buf) != 0)
  47         return false;
  48
  49     string = mmap(NULL, stat_buf.st_size, PROT_READ, MAP_SHARED, fd, 0);
  50     if (string == MAP_FAILED)
  51         return false;
  52
  53     *string_out = string;
  54     *size_out = stat_buf.st_size;
  55     return true;
  56 }
  57
  58 void
  59 unmap_file(char *str, size_t size)
  60 {
  61     munmap(str, size);
  62 }
  63
  64 #else
  65
  66 bool
  67 map_file(FILE *file, char **string_out, size_t *size_out)
  68 {
  69     long ret;
  70     size_t ret_s;
  71     char *string;
  72     size_t size;
  73
  74     /* Make sure to keep the errno on failure! */
  75
  76     ret = fseek(file, 0, SEEK_END);
  77     if (ret != 0)
  78         return false;
  79
  80     ret = ftell(file);
  81     if (ret < 0)
  82         return false;
  83     size = (size_t) ret;
  84
  85     ret = fseek(file, 0, SEEK_SET);
  86     if (ret < 0)
  87         return false;
  88
  89     string = malloc(size);
  90     if (!string)
  91         return false;
  92
  93     ret_s = fread(string, 1, size, file);
  94     if (ret_s < size) {
  95         free(string);
  96         return false;
  97     }
  98
  99     *string_out = string;
 100     *size_out = size;
 101     return true;
 102 }
 103
 104 void
 105 unmap_file(char *str, size_t size)
 106 {
 107     free(str);
 108 }
 109
 110 #endif