mem.c 3.79 KB
Newer Older
1
/*
2
 * default memory allocator for libavutil
3 4
 * Copyright (c) 2002 Fabrice Bellard.
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
13 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21

Michael Niedermayer's avatar
Michael Niedermayer committed
22 23
/**
 * @file mem.c
24
 * default memory allocator for libavutil.
Michael Niedermayer's avatar
Michael Niedermayer committed
25
 */
26

27
#include "common.h"
28

29
/* here we can use OS dependent allocation functions */
30 31 32 33
#undef malloc
#undef free
#undef realloc

34 35 36 37 38 39 40 41
#ifdef HAVE_MALLOC_H
#include <malloc.h>
#endif

/* you can redefine av_malloc and av_free in your project to use your
   memory allocator. You do not need to suppress this file because the
   linker will do it automatically */

42
void *av_malloc(unsigned int size)
43 44
{
    void *ptr;
45
#ifdef CONFIG_MEMALIGN_HACK
46
    long diff;
47
#endif
48

49
    /* let's disallow possible ambiguous cases */
50
    if(size > (INT_MAX-16) )
51
        return NULL;
52

53
#ifdef CONFIG_MEMALIGN_HACK
54 55 56
    ptr = malloc(size+16);
    if(!ptr)
        return ptr;
57
    diff= ((-(long)ptr - 1)&15) + 1;
58 59
    ptr += diff;
    ((char*)ptr)[-1]= diff;
60
#elif defined (HAVE_MEMALIGN)
Michael Niedermayer's avatar
Michael Niedermayer committed
61
    ptr = memalign(16,size);
62
    /* Why 64?
63 64 65
       Indeed, we should align it:
         on 4 for 386
         on 16 for 486
66 67
         on 32 for 586, PPro - k6-III
         on 64 for K7 (maybe for P3 too).
68 69 70
       Because L1 and L2 caches are aligned on those values.
       But I don't want to code such logic here!
     */
Michael Niedermayer's avatar
Michael Niedermayer committed
71 72 73 74 75 76 77
     /* Why 16?
        because some cpus need alignment, for example SSE2 on P4, & most RISC cpus
        it will just trigger an exception and the unaligned load will be done in the
        exception handler or it will just segfault (SSE2 on P4)
        Why not larger? because i didnt see a difference in benchmarks ...
     */
     /* benchmarks with p3
78 79 80 81 82 83 84
        memalign(64)+1          3071,3051,3032
        memalign(64)+2          3051,3032,3041
        memalign(64)+4          2911,2896,2915
        memalign(64)+8          2545,2554,2550
        memalign(64)+16         2543,2572,2563
        memalign(64)+32         2546,2545,2571
        memalign(64)+64         2570,2533,2558
85

Michael Niedermayer's avatar
Michael Niedermayer committed
86 87
        btw, malloc seems to do 8 byte alignment by default here
     */
88 89 90 91 92 93
#else
    ptr = malloc(size);
#endif
    return ptr;
}

94 95
void *av_realloc(void *ptr, unsigned int size)
{
96
#ifdef CONFIG_MEMALIGN_HACK
97 98
    int diff;
#endif
99

100
    /* let's disallow possible ambiguous cases */
101
    if(size > (INT_MAX-16) )
102 103
        return NULL;

104
#ifdef CONFIG_MEMALIGN_HACK
105 106 107 108 109 110
    //FIXME this isn't aligned correctly, though it probably isn't needed
    if(!ptr) return av_malloc(size);
    diff= ((char*)ptr)[-1];
    return realloc(ptr - diff, size + diff) + diff;
#else
    return realloc(ptr, size);
111
#endif
Michael Niedermayer's avatar
cleanup  
Michael Niedermayer committed
112 113
}

114 115 116 117
void av_free(void *ptr)
{
    /* XXX: this test should not be needed on most libcs */
    if (ptr)
118
#ifdef CONFIG_MEMALIGN_HACK
119 120
        free(ptr - ((char*)ptr)[-1]);
#else
121
        free(ptr);
122
#endif
123 124
}

125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
void av_freep(void *arg)
{
    void **ptr= (void**)arg;
    av_free(*ptr);
    *ptr = NULL;
}

void *av_mallocz(unsigned int size)
{
    void *ptr;

    ptr = av_malloc(size);
    if (ptr)
        memset(ptr, 0, size);
    return ptr;
}

char *av_strdup(const char *s)
{
    char *ptr;
    int len;
    len = strlen(s) + 1;
    ptr = av_malloc(len);
    if (ptr)
        memcpy(ptr, s, len);
    return ptr;
}