release/src/router/libsodium/src/libsodium/crypto_pwhash/scryptsalsa208sha256/scrypt_platform.c

   1 /*-
   2  * Copyright 2013 Alexander Peslyak
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted.
   7  *
   8  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
   9  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  10  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  11  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  12  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  13  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  14  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  15  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  16  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  17  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  18  * SUCH DAMAGE.
  19  */
  20
  21 #ifdef HAVE_SYS_MMAN_H
  22 # include <sys/mman.h>
  23 #endif
  24 #include <errno.h>
  25 #include <stdlib.h>
  26
  27 #include "crypto_scrypt.h"
  28 #include "runtime.h"
  29
  30 #if !defined(MAP_ANON) && defined(MAP_ANONYMOUS)
  31 # define MAP_ANON MAP_ANONYMOUS
  32 #endif
  33
  34 void *
  35 alloc_region(escrypt_region_t * region, size_t size)
  36 {
  37     uint8_t * base, * aligned;
  38 #if defined(MAP_ANON) && defined(HAVE_MMAP)
  39     if ((base = (uint8_t *) mmap(NULL, size, PROT_READ | PROT_WRITE,
  40 #ifdef MAP_NOCORE
  41         MAP_ANON | MAP_PRIVATE | MAP_NOCORE,
  42 #else
  43         MAP_ANON | MAP_PRIVATE,
  44 #endif
  45         -1, 0)) == MAP_FAILED)
  46         base = NULL; /* LCOV_EXCL_LINE */
  47     aligned = base;
  48 #elif defined(HAVE_POSIX_MEMALIGN)
  49     if ((errno = posix_memalign((void **) &base, 64, size)) != 0)
  50         base = NULL;
  51     aligned = base;
  52 #else
  53     base = aligned = NULL;
  54     if (size + 63 < size)
  55         errno = ENOMEM;
  56     else if ((base = (uint8_t *) malloc(size + 63)) != NULL) {
  57         aligned = base + 63;
  58         aligned -= (uintptr_t)aligned & 63;
  59     }
  60 #endif
  61     region->base = base;
  62     region->aligned = aligned;
  63     region->size = base ? size : 0;
  64     return aligned;
  65 }
  66
  67 static inline void
  68 init_region(escrypt_region_t * region)
  69 {
  70     region->base = region->aligned = NULL;
  71     region->size = 0;
  72 }
  73
  74 int
  75 free_region(escrypt_region_t * region)
  76 {
  77     if (region->base) {
  78 #if defined(MAP_ANON) && defined(HAVE_MMAP)
  79         if (munmap(region->base, region->size))
  80             return -1; /* LCOV_EXCL_LINE */
  81 #else
  82         free(region->base);
  83 #endif
  84     }
  85     init_region(region);
  86     return 0;
  87 }
  88
  89 int
  90 escrypt_init_local(escrypt_local_t * local)
  91 {
  92     init_region(local);
  93     return 0;
  94 }
  95
  96 int
  97 escrypt_free_local(escrypt_local_t * local)
  98 {
  99     return free_region(local);
 100 }