luamem.c 1020 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. /*
  2. ** mem.c
  3. ** TecCGraf - PUC-Rio
  4. */
  5. char *rcs_mem = "$Id: mem.c,v 1.7 1996/02/04 16:59:12 roberto Exp roberto $";
  6. #include <stdlib.h>
  7. #include <string.h>
  8. #include <stdio.h>
  9. #include "mem.h"
  10. #include "lua.h"
  11. #include "table.h"
  12. static void mem_error (void)
  13. {
  14. Long recovered = luaI_collectgarbage(); /* try to collect garbage */
  15. if (recovered)
  16. lua_error("not enough memory");
  17. else
  18. { /* if there is no garbage then must exit */
  19. fprintf(stderr, "lua error: memory overflow - unable to recover\n");
  20. exit(1);
  21. }
  22. }
  23. void luaI_free (void *block)
  24. {
  25. if (block)
  26. {
  27. *((int *)block) = -1; /* to catch errors */
  28. free(block);
  29. }
  30. }
  31. void *luaI_malloc (unsigned long size)
  32. {
  33. void *block = malloc((size_t)size);
  34. if (block == NULL)
  35. mem_error();
  36. return block;
  37. }
  38. void *luaI_realloc (void *oldblock, unsigned long size)
  39. {
  40. void *block = oldblock ? realloc(oldblock, (size_t)size) :
  41. malloc((size_t)size);
  42. if (block == NULL)
  43. mem_error();
  44. return block;
  45. }