- added cmock.c file for handling generic parts

- switched to better memory management technique


git-svn-id: http://cmock.svn.sourceforge.net/svnroot/cmock/trunk@155 bf332499-1b4d-0410-844d-d2d48d5cc64c

src/cmock.c

@@ -0,0 +1,152 @@
+#include "unity.h"
+#include "cmock.h"
+
+//define CMOCK_MEM_DYNAMIC to grab memory as needed with malloc
+//when you do that, CMOCK_MEM_SIZE is used for incremental size instead of total
+#ifdef CMOCK_MEM_STATIC
+#undef CMOCK_MEM_DYNAMIC
+#endif
+
+#ifdef CMOCK_MEM_DYNAMIC
+#include <stdlib.h>
+#endif
+
+//should be big enough to index full range of CMOCK_MEM_MAX
+#ifndef CMOCK_MEM_INDEX_TYPE
+#define CMOCK_MEM_INDEX_TYPE  unsigned int
+#endif
+
+//0 for no alignment, 1 for 16-bit, 2 for 32-bit, 3 for 64-bit
+#ifndef CMOCK_MEM_ALIGN
+#define CMOCK_MEM_ALIGN (2)
+#endif
+
+//amount of memory to allow cmock to use in its internal heap
+#ifndef CMOCK_MEM_SIZE
+#define CMOCK_MEM_SIZE (32768)
+#endif
+
+//automatically calculated defs for easier reading
+#define CMOCK_MEM_INDEX_SIZE  (sizeof(CMOCK_MEM_INDEX_TYPE))
+#define CMOCK_MEM_ALIGN_MASK  ((1u << CMOCK_MEM_ALIGN) - 1)
+
+//private variables
+#ifdef CMOCK_MEM_DYNAMIC
+static unsigned char* CMock_Guts_Buffer = NULL;
+static unsigned int   CMock_Guts_BufferSize = 0;
+static unsigned int   CMock_Guts_FreePtr;
+#else
+static unsigned char  CMock_Guts_Buffer[CMOCK_MEM_SIZE];
+static unsigned int   CMock_Guts_BufferSize = CMOCK_MEM_SIZE;
+static unsigned int   CMock_Guts_FreePtr;
+#endif
+//-------------------------------------------------------
+// CMock_Guts_MemNew
+//-------------------------------------------------------
+void* CMock_Guts_MemNew(unsigned int size)
+{
+  unsigned int index;
+
+  //verify arguments valid (we must be allocating space for at least 1 byte, and the existing chain must be in memory somewhere)
+  if (size < 1)
+    return NULL;
+
+  //verify we have enough room
+  size = size + CMOCK_MEM_INDEX_SIZE;
+  if (size & CMOCK_MEM_ALIGN_MASK)
+    size = (size + CMOCK_MEM_ALIGN_MASK) & ~CMOCK_MEM_ALIGN_MASK;
+  if ((CMock_Guts_BufferSize - CMock_Guts_FreePtr) < size)
+  {
+#ifdef CMOCK_MEM_DYNAMIC
+    CMock_Guts_BufferSize += CMOCK_MEM_SIZE + size;
+    CMock_Guts_Buffer = realloc(CMock_Guts_Buffer, CMock_Guts_BufferSize);
+    if (CMock_Guts_Buffer == NULL)
+#endif //yes that if will continue to the return below if TRUE
+      return NULL;
+  }
+
+  //determine where we're putting this new block, and init its pointer to be the end of the line
+  index = CMock_Guts_FreePtr + CMOCK_MEM_INDEX_SIZE;
+  *(CMOCK_MEM_INDEX_TYPE*)(&CMock_Guts_Buffer[CMock_Guts_FreePtr]) = 0;
+  CMock_Guts_FreePtr += size;
+
+  return (&CMock_Guts_Buffer[index]);
+}
+
+//-------------------------------------------------------
+// CMock_Guts_MemChain
+//-------------------------------------------------------
+void* CMock_Guts_MemChain(void* root, void* obj)
+{
+  unsigned int index;
+  void* next;
+
+  if (root == NULL)
+  {
+    //if there is no root currently, we return this object as the root of the chain
+    return obj;
+  }
+  else
+  {
+    //reject illegal nodes
+    if ((root < (void*)CMock_Guts_Buffer) || (root >= (void*)(&CMock_Guts_Buffer[CMock_Guts_FreePtr])))
+      return NULL;
+    if ((obj  < (void*)CMock_Guts_Buffer) || (obj  >= (void*)(&CMock_Guts_Buffer[CMock_Guts_FreePtr])))
+      return NULL;
+
+    //find the end of the existing chain and add us
+    next = root;
+    do {
+      index = *(CMOCK_MEM_INDEX_TYPE*)((unsigned int)next - CMOCK_MEM_INDEX_SIZE);
+      if (index >= CMock_Guts_FreePtr)
+        return NULL;
+      if (index > 0)
+        next = (void*)(&CMock_Guts_Buffer[index]);
+    } while (index > 0);
+    *(CMOCK_MEM_INDEX_TYPE*)((unsigned int)next - CMOCK_MEM_INDEX_SIZE) = ((unsigned int)obj - (unsigned int)CMock_Guts_Buffer);
+    return root;
+  }
+}
+
+//-------------------------------------------------------
+// CMock_Guts_MemNext
+//-------------------------------------------------------
+void* CMock_Guts_MemNext(void* previous_item)
+{
+  CMOCK_MEM_INDEX_TYPE index;
+
+  //There is nothing "next" if the pointer isn't from our buffer
+  if ((previous_item < (void*)CMock_Guts_Buffer) || (previous_item >= (void*)(&CMock_Guts_Buffer[CMock_Guts_FreePtr])))
+    return NULL;
+
+  //if the pointer is good, then use it to look up the next index (we know the first element always goes in zero, so NEXT must always be > 1)
+  index = *(CMOCK_MEM_INDEX_TYPE*)((unsigned int)previous_item - CMOCK_MEM_INDEX_SIZE);
+  if ((index > 1) && (index < CMock_Guts_FreePtr))
+    return (void*)(&CMock_Guts_Buffer[index]);
+  else
+    return NULL;
+}
+
+//-------------------------------------------------------
+// CMock_Guts_MemBytesFree
+//-------------------------------------------------------
+unsigned int CMock_Guts_MemBytesFree(void)
+{
+  return CMock_Guts_BufferSize - CMock_Guts_FreePtr;
+}
+
+//-------------------------------------------------------
+// CMock_Guts_MemBytesUsed
+//-------------------------------------------------------
+unsigned int CMock_Guts_MemBytesUsed(void)
+{
+  return CMock_Guts_FreePtr;
+}
+
+//-------------------------------------------------------
+// CMock_Guts_MemFreeAll
+//-------------------------------------------------------
+void CMock_Guts_MemFreeAll(void)
+{
+  CMock_Guts_FreePtr = 0;
+}

src/cmock.h

@@ -0,0 +1,14 @@
+#ifndef CMOCK_FRAMEWORK_H
+#define CMOCK_FRAMEWORK_H
+
+//-------------------------------------------------------
+// Memory API
+//-------------------------------------------------------
+void*         CMock_Guts_MemNew(unsigned int size);
+void*         CMock_Guts_MemChain(void* root, void* obj);
+void*         CMock_Guts_MemNext(void* previous_item);
+unsigned int  CMock_Guts_MemBytesFree(void);
+unsigned int  CMock_Guts_MemBytesUsed(void);
+void          CMock_Guts_MemFreeAll(void);
+
+#endif //CMOCK_FRAMEWORK