Merge branch 'edge' into track_allocs

[clinton/Smoothieware.git] / gcc4mbed / src / gcc4mbed.cpp

/* Copyright 2013 Adam Green (http://mbed.org/users/AdamGreen/)

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/
/* Provide routines which hook the MRI debug monitor into GCC4MBED projects. */
#include <string.h>
#include <sys/types.h>
#include <errno.h>
#include <mri.h>
#include <cmsis.h>
#include "mpu.h"

static void configureHighestMpuRegionToAccessAllMemoryWithNoCaching(void);

extern unsigned int __bss_start__;
extern unsigned int __bss_end__;
extern "C" int  main(void);
extern "C" void __libc_init_array(void);
extern "C" void exit(int ErrorCode);
extern "C" void _start(void)
{
    int bssSize = (int)&__bss_end__ - (int)&__bss_start__;
    int mainReturnValue;
    
    memset(&__bss_start__, 0, bssSize);
    
    if (WRITE_BUFFER_DISABLE)
    {
        disableMPU();
        configureHighestMpuRegionToAccessAllMemoryWithNoCaching();    
        enableMPU();
    }
    if (MRI_ENABLE)
    {
        __mriInit(MRI_INIT_PARAMETERS);
        if (MRI_BREAK_ON_INIT)
            __debugbreak();
    }
    
    __libc_init_array();
    mainReturnValue = main();
    exit(mainReturnValue);
}

static void configureHighestMpuRegionToAccessAllMemoryWithNoCaching(void)
{
    static const uint32_t regionToStartAtAddress0 = 0U;
    static const uint32_t regionReadWrite = 1  << MPU_RASR_AP_SHIFT;
    static const uint32_t regionSizeAt4GB = 31 << MPU_RASR_SIZE_SHIFT; /* 4GB = 2^(31+1) */
    static const uint32_t regionEnable    = MPU_RASR_ENABLE;
    static const uint32_t regionSizeAndAttributes = regionReadWrite | regionSizeAt4GB | regionEnable;
    
    prepareToAccessMPURegion(getHighestMPUDataRegionIndex());
    setMPURegionAddress(regionToStartAtAddress0);
    setMPURegionAttributeAndSize(regionSizeAndAttributes);
}



extern "C" int __real__read(int file, char *ptr, int len);
extern "C" int __wrap__read(int file, char *ptr, int len)
{
    if (MRI_SEMIHOST_STDIO && file < 3)
        return __mriNewlib_SemihostRead(file, ptr, len);
     return __real__read(file, ptr, len);
}


extern "C" int __real__write(int file, char *ptr, int len);
extern "C" int __wrap__write(int file, char *ptr, int len)
{
    if (MRI_SEMIHOST_STDIO && file < 3)
        return __mriNewlib_SemihostWrite(file, ptr, len);
    return __real__write(file, ptr, len);
}


extern "C" int __real__isatty(int file);
extern "C" int __wrap__isatty(int file)
{
    /* Hardcoding the stdin/stdout/stderr handles to be interactive tty devices, unlike mbed.ar */
    if (file < 3)
        return 1;
    return __real__isatty(file);
}


extern "C" int __wrap_semihost_connected(void)
{
    /* MRI makes it look like there is no mbed interface attached since it disables the JTAG portion but MRI does
       support some of the mbed semihost calls when it is running so force it to return -1, indicating that the
       interface is attached. */
    return -1;
}



extern "C" void abort(void)
{
    if (MRI_ENABLE)
        __debugbreak();
        
    exit(1);
}


extern "C" void __cxa_pure_virtual(void)
{
    abort();
}


extern "C" int __aeabi_unwind_cpp_pr0(int state, void* controlBlock, void* context)
{
    abort();
}


extern "C" int __aeabi_unwind_cpp_pr1(int state, void* controlBlock, void* context)
{
    abort();
}


extern "C" int __aeabi_unwind_cpp_pr2(int state, void* controlBlock, void* context)
{
    abort();
}

/* Trap calls to malloc/free/realloc in ISR. */
extern "C" void __malloc_lock(void)
{
    if (__get_IPSR() != 0)
        __debugbreak();
}

extern "C" void __malloc_unlock(void)
{
}


/* Linker defined symbol to be used by sbrk for where dynamically heap should start. */
extern "C" unsigned int __HeapBase;

/* Turn off the errno macro and use actual external global variable instead. */
#undef errno
extern int errno;

/* Dynamic memory allocation related syscalls. */
extern "C" caddr_t _sbrk(int incr) 
{
    static unsigned char* heap = (unsigned char*)&__HeapBase;
    unsigned char*        prev_heap = heap;
    unsigned char*        new_heap = heap + incr;

    if (new_heap >= (unsigned char*)__get_MSP()) {
        errno = ENOMEM;
        return (caddr_t)-1;
    }
    
    heap = new_heap;
    return (caddr_t) prev_heap;
}


/* Optional functionality which will tag each heap allocation with the caller's return address. */
#ifdef HEAP_TAGS

const unsigned int* __smoothieHeapBase = &__HeapBase;

extern "C" void* __real_malloc(size_t size);
extern "C" void* __real_realloc(void* ptr, size_t size);
extern "C" void  __real_free(void* ptr);

static void setTag(void* pv, unsigned int tag);
static unsigned int* footerForChunk(void* pv);
static unsigned int* headerForChunk(void* pv);
static unsigned int sizeOfChunk(unsigned int* pHeader);
static int isChunkInUse(void* pv);

extern "C" __attribute__((naked)) void __wrap_malloc(size_t size)
{
    __asm (
        ".syntax unified\n"
        ".thumb\n"
        "mov r1,lr\n"
        "b mallocWithTag\n"
    );
}

extern "C" void* mallocWithTag(size_t size, unsigned int tag)
{
    void* p = __real_malloc(size + sizeof(tag));
    if (!p && __smoothieHeapBase)
        return p;
    setTag(p, tag);
    return p;
}

static void setTag(void* pv, unsigned int tag)
{
    unsigned int* pFooter = footerForChunk(pv);
    *pFooter = tag;
}

static unsigned int* footerForChunk(void* pv)
{
    unsigned int* pHeader = headerForChunk(pv);
    unsigned int  size = sizeOfChunk(pHeader);
    return (unsigned int*)(void*)((char*)pHeader + size);
}

static unsigned int* headerForChunk(void* pv)
{
    // Header is allocated two words (8 bytes) before the publicly returned allocation chunk address.
    unsigned int* p = (unsigned int*)pv;
    return &p[-2];
}

static unsigned int sizeOfChunk(unsigned int* pHeader)
{
    /* Remove previous chunk in use flag. */
    return pHeader[1] & ~1;
}

extern "C" __attribute__((naked)) void __wrap_realloc(void* ptr, size_t size)
{
    __asm (
        ".syntax unified\n"
        ".thumb\n"
        "mov r2,lr\n"
        "b reallocWithTag\n"
    );
}

extern "C" void* reallocWithTag(void* ptr, size_t size, unsigned int tag)
{
    void* p = __real_realloc(ptr, size + sizeof(tag));
    if (!p)
        return p;
    setTag(p, tag);
    return p;
}

extern "C" void __wrap_free(void* ptr)
{
    if (!isChunkInUse(ptr))
        __debugbreak();
    __real_free(ptr);
}

static int isChunkInUse(void* pv)
{
    unsigned int* pFooter = footerForChunk(pv);
    return pFooter[1] & 1;
}

__attribute__((naked)) void* operator new(size_t size)
{
    __asm (
        ".syntax unified\n"
        ".thumb\n"
        "push {r4,lr}\n"
        "mov r1,lr\n"
        "bl mallocWithTag\n"
        "cbnz r0, 1$\n"
        "bl abort\n"
        "1$:\n"
        "pop {r4,pc}\n"
    );
    // This line never executes but silences no return value warning from compiler.
    return (void*)1;
}

#endif // HEAP_TAGS