aom_mem.c 2.52 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
7
 *  in the file PATENTS.  All contributing project authors may
8
 *  be found in the AUTHORS file in the root of the source tree.
John Koleszar's avatar
John Koleszar committed
9 10
 */

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

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

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

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

John Koleszar's avatar
John Koleszar committed
25 26 27 28 29
  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
30

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

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

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

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

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

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

Yaowu Xu's avatar
Yaowu Xu committed
46
void *aom_realloc(void *memblk, size_t size) {
clang-format's avatar
clang-format committed
47
  void *addr, *new_addr = NULL;
John Koleszar's avatar
John Koleszar committed
48 49 50 51 52 53 54 55 56 57 58
  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
59
    new_addr = aom_malloc(size);
John Koleszar's avatar
John Koleszar committed
60
  else if (!size)
Yaowu Xu's avatar
Yaowu Xu committed
61
    aom_free(memblk);
John Koleszar's avatar
John Koleszar committed
62
  else {
clang-format's avatar
clang-format committed
63
    addr = (void *)(((size_t *)memblk)[-1]);
John Koleszar's avatar
John Koleszar committed
64
    memblk = NULL;
John Koleszar's avatar
John Koleszar committed
65

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

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

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

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

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