espressif/Unity
1
0
Fork 0
mirror of https://github.com/ThrowTheSwitch/Unity.git synced 2025-06-26 03:17:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove atoi() dependency, only need stdlib.h in Fixture for malloc

Browse Source 
For redefinition of UNITY_FIXTURE_MALLOC/...FREE use both or replace both.
 Clean up whitespace, remaining void*, and comment.
This commit is contained in:
jsalling
2016-01-30 22:35:59 -06:00
parent 34a30f8e98
commit 63a0b98faf
2 changed files with 20 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
extras/fixture/src/unity_fixture.c
View File

@ -149,21 +149,12 @@ void UnityMalloc_MakeMallocFailAfterCount(int countdown)
malloc_fail_countdown = countdown; malloc_fail_countdown = countdown;
} }
#ifdef malloc // These definitions are always included from unity_fixture_malloc_overrides.h
// We undef to use them or avoid conflict with <stdlib.h> per the C standard
#undef malloc #undef malloc
#endif
#ifdef free
#undef free #undef free
#endif
#ifdef calloc
#undef calloc #undef calloc
#endif
#ifdef realloc
#undef realloc #undef realloc
#endif
#include <stdlib.h> #include <stdlib.h>
@ -312,7 +303,6 @@ void UnityPointer_UndoAllSets(void)
pointer_index--; pointer_index--;
*(pointer_store[pointer_index].pointer) = *(pointer_store[pointer_index].pointer) =
pointer_store[pointer_index].old_value; pointer_store[pointer_index].old_value;
} }
} }
@ -373,7 +363,13 @@ int UnityGetCommandLineOptions(int argc, const char* argv[])
{ {
if (*(argv[i]) >= '0' && *(argv[i]) <= '9') if (*(argv[i]) >= '0' && *(argv[i]) <= '9')
{ {
UnityFixture.RepeatCount = atoi(argv[i]); unsigned int digit = 0;
UnityFixture.RepeatCount = 0;
while (argv[i][digit] >= '0' && argv[i][digit] <= '9')
{
UnityFixture.RepeatCount *= 10;
UnityFixture.RepeatCount += (unsigned int)argv[i][digit++] - '0';
}
i++; i++;
} }
} }

23
extras/fixture/src/unity_fixture_malloc_overrides.h
View File

@ -10,25 +10,16 @@
#include <stddef.h> #include <stddef.h>
// This function is used by the Unity Fixture to allocate memory on // These functions are used by the Unity Fixture to allocate and release memory
// the heap and can be overridden with platform-specific heap // on the heap and can be overridden with platform-specific implementations.
// implementations. For example, when using FreeRTOS // For example, when using FreeRTOS UNITY_FIXTURE_MALLOC becomes pvPortMalloc()
// UNITY_FIXTURE_MALLOC becomes pvPortMalloc(). // and UNITY_FIXTURE_FREE becomes vPortFree().
#ifndef UNITY_FIXTURE_MALLOC #if !defined(UNITY_FIXTURE_MALLOC) || !defined(UNITY_FIXTURE_FREE)
#define UNITY_FIXTURE_MALLOC( SIZE ) malloc( ( SIZE ) ) #define UNITY_FIXTURE_MALLOC(size) malloc(size)
#define UNITY_FIXTURE_FREE(ptr) free(ptr)
#else #else
extern void* UNITY_FIXTURE_MALLOC(size_t size); extern void* UNITY_FIXTURE_MALLOC(size_t size);
#endif
// This function is used by the Unity Fixture to release memory in the
// heap and can be overridden with platform-specific heap
// implementations. For example, when using FreeRTOS
// UNITY_FIXTURE_FREE becomes vPortFree().
#ifndef UNITY_FIXTURE_FREE
#define UNITY_FIXTURE_FREE( PTR ) free( ( PTR ) )
#else
extern void UNITY_FIXTURE_FREE(void* ptr); extern void UNITY_FIXTURE_FREE(void* ptr);
#endif #endif