1/*
2 * Copyright (c) 2008 - 2010 Apple Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. Please obtain a copy of the License at
10 * http://www.opensource.apple.com/apsl/ and read it before using this
11 * file.
12 *
13 * The Original Code and all software distributed under the License are
14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
18 * Please see the License for the specific language governing rights and
19 * limitations under the License.
20 *
21 * @APPLE_LICENSE_HEADER_END@
22 */
23
24#include "memory.hpp"
25#include "compiler.h"
26
27#include <cstdlib>
28#include <new>
29
30namespace platform {
31
32void
33invoke_new_handler(void)
34{
35    std::new_handler handler;
36
37    // Get the curent new_handler by double-swapping. If multiple threads
38    // race over this, then we will end up aborting. Ce la vie.
39    handler = std::set_new_handler(::abort);
40    std::set_new_handler(handler);
41
42    handler();
43}
44
45void *
46allocate(
47        void * buf,
48        std::size_t nbytes)
49{
50retry:
51    if (buf == NULL) {
52        if (nbytes % platform::pagesize()) {
53            buf = ::malloc(nbytes);
54        } else {
55            buf = ::valloc(nbytes);
56        }
57    } else {
58        buf = ::realloc(buf, nbytes);
59    }
60
61    if (UNLIKELY(buf == NULL)) {
62        invoke_new_handler();
63        goto retry;
64    }
65
66    return buf;
67}
68
69} // namespace platform
70/* vim: set ts=4 sw=4 tw=79 et cindent : */
71