aom_mem.c 2.63 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
John Koleszar's avatar
John Koleszar committed
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
John Koleszar's avatar
John Koleszar committed
10 11
 */

Yaowu Xu's avatar
Yaowu Xu committed
12 13 14
#define __AOM_MEM_C__

#include "aom_mem.h"
John Koleszar's avatar
John Koleszar committed
15 16 17
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
Yaowu Xu's avatar
Yaowu Xu committed
18 19
#include "include/aom_mem_intrnl.h"
#include "aom/aom_integer.h"
John Koleszar's avatar
John Koleszar committed
20

Yaowu Xu's avatar
Yaowu Xu committed
21
void *aom_memalign(size_t align, size_t size) {
clang-format's avatar
clang-format committed
22
  void *addr, *x = NULL;
John Koleszar's avatar
John Koleszar committed
23

24
  addr = malloc(size + align - 1 + ADDRESS_STORAGE_SIZE);
John Koleszar's avatar
John Koleszar committed
25

John Koleszar's avatar
John Koleszar committed
26 27 28 29 30
  if (addr) {
    x = align_addr((unsigned char *)addr + ADDRESS_STORAGE_SIZE, (int)align);
    /* save the actual malloc address */
    ((size_t *)x)[-1] = (size_t)addr;
  }
John Koleszar's avatar
John Koleszar committed
31

John Koleszar's avatar
John Koleszar committed
32
  return x;
John Koleszar's avatar
John Koleszar committed
33 34
}

Yaowu Xu's avatar
Yaowu Xu committed
35
void *aom_malloc(size_t size) { return aom_memalign(DEFAULT_ALIGNMENT, size); }
John Koleszar's avatar
John Koleszar committed
36

Yaowu Xu's avatar
Yaowu Xu committed
37
void *aom_calloc(size_t num, size_t size) {
John Koleszar's avatar
John Koleszar committed
38
  void *x;
John Koleszar's avatar
John Koleszar committed
39

Yaowu Xu's avatar
Yaowu Xu committed
40
  x = aom_memalign(DEFAULT_ALIGNMENT, num * size);
John Koleszar's avatar
John Koleszar committed
41

clang-format's avatar
clang-format committed
42
  if (x) memset(x, 0, num * size);
John Koleszar's avatar
John Koleszar committed
43

John Koleszar's avatar
John Koleszar committed
44
  return x;
John Koleszar's avatar
John Koleszar committed
45 46
}

Yaowu Xu's avatar
Yaowu Xu committed
47
void *aom_realloc(void *memblk, size_t size) {
clang-format's avatar
clang-format committed
48
  void *addr, *new_addr = NULL;
John Koleszar's avatar
John Koleszar committed
49 50 51 52 53 54 55 56 57 58 59
  int align = DEFAULT_ALIGNMENT;

  /*
  The realloc() function changes the size of the object pointed to by
  ptr to the size specified by size, and returns a pointer to the
  possibly moved block. The contents are unchanged up to the lesser
  of the new and old sizes. If ptr is null, realloc() behaves like
  malloc() for the specified size. If size is zero (0) and ptr is
  not a null pointer, the object pointed to is freed.
  */
  if (!memblk)
Yaowu Xu's avatar
Yaowu Xu committed
60
    new_addr = aom_malloc(size);
John Koleszar's avatar
John Koleszar committed
61
  else if (!size)
Yaowu Xu's avatar
Yaowu Xu committed
62
    aom_free(memblk);
John Koleszar's avatar
John Koleszar committed
63
  else {
clang-format's avatar
clang-format committed
64
    addr = (void *)(((size_t *)memblk)[-1]);
John Koleszar's avatar
John Koleszar committed
65
    memblk = NULL;
John Koleszar's avatar
John Koleszar committed
66

67
    new_addr = realloc(addr, size + align + ADDRESS_STORAGE_SIZE);
John Koleszar's avatar
John Koleszar committed
68

John Koleszar's avatar
John Koleszar committed
69 70
    if (new_addr) {
      addr = new_addr;
clang-format's avatar
clang-format committed
71 72 73 74
      new_addr =
          (void *)(((size_t)((unsigned char *)new_addr + ADDRESS_STORAGE_SIZE) +
                    (align - 1)) &
                   (size_t)-align);
John Koleszar's avatar
John Koleszar committed
75 76
      /* save the actual malloc address */
      ((size_t *)new_addr)[-1] = (size_t)addr;
John Koleszar's avatar
John Koleszar committed
77
    }
John Koleszar's avatar
John Koleszar committed
78
  }
John Koleszar's avatar
John Koleszar committed
79

John Koleszar's avatar
John Koleszar committed
80
  return new_addr;
John Koleszar's avatar
John Koleszar committed
81 82
}

Yaowu Xu's avatar
Yaowu Xu committed
83
void aom_free(void *memblk) {
John Koleszar's avatar
John Koleszar committed
84 85
  if (memblk) {
    void *addr = (void *)(((size_t *)memblk)[-1]);
86
    free(addr);
John Koleszar's avatar
John Koleszar committed
87
  }
John Koleszar's avatar
John Koleszar committed
88 89
}

Yaowu Xu's avatar
Yaowu Xu committed
90 91
#if CONFIG_AOM_HIGHBITDEPTH
void *aom_memset16(void *dest, int val, size_t length) {
92
  size_t i;
93
  uint16_t *dest16 = (uint16_t *)dest;
clang-format's avatar
clang-format committed
94
  for (i = 0; i < length; i++) *dest16++ = val;
95
  return dest;
96
}
Yaowu Xu's avatar
Yaowu Xu committed
97
#endif  // CONFIG_AOM_HIGHBITDEPTH