home *** CD-ROM | disk | FTP | other *** search
/ Usenet 1994 January / usenetsourcesnewsgroupsinfomagicjanuary1994.iso / sources / unix / volume13 / rolodex / part01 / toolsdir / mem.c < prev    next >
Encoding:
C/C++ Source or Header  |  1988-01-30  |  2.2 KB  |  94 lines
  1. /**************************************************************************/
  2. /**************************************************************************/
  3.  
  4.  
  5.                   /***** Block Memory Allocator *****/
  6.  
  7.  
  8. /**************************************************************************/
  9. /**************************************************************************/
  10.  
  11. /* Author: JP Massar */
  12.  
  13. #include <stdio.h>
  14.  
  15. #include "sys5.h"
  16.  
  17. #ifdef BSD42
  18. #include <strings.h>
  19. #endif
  20.  
  21. #define NO_MORE_MEMORY -1
  22.  
  23. static int bytes_left;                  /* space left in current block */
  24. static char *ptr_next_byte;             /* next free byte */
  25. static char *ptr_space;                 /* current block */
  26. static unsigned chunk_size;                  /* size of block last allocated */
  27.  
  28. extern char *malloc();
  29.  
  30.  
  31. unsigned allocate_memory_chunk(space) 
  32. unsigned space;
  33.  
  34. /* malloc up a new block of memory.  Set our static variables */
  35. /* returns NO_MORE_MEMORY if can't allocate block. */
  36.  
  37. { 
  38. #ifdef MSDOS
  39.   if ((char *)NULL == (ptr_space = malloc(space))) {
  40. #else
  41.   if (0 == (ptr_space = malloc(space))) {
  42. #endif
  43.         fprintf(stderr,"fatal error, no more memory\n");
  44.         return(NO_MORE_MEMORY);
  45.   }
  46.   ptr_next_byte = ptr_space;
  47.   bytes_left = space;
  48.   chunk_size = space;
  49.   return(0);
  50. }
  51.  
  52.  
  53. char * get_memory_chunk (size) unsigned size;
  54.  
  55. /* allocate a small segment out of our large block of memory.  If we */
  56. /* run out allocate another block.  Adjust our static variables. */
  57. /* returns 0 if no more memory. */
  58.  
  59. { char *rval;
  60.         
  61.   if (size > chunk_size) {
  62.         fprintf(stderr,"attempt to allocate too large a chunk\n");
  63.         return(0);
  64.   }
  65.         
  66.   if (size > bytes_left) {
  67.         if (NO_MORE_MEMORY == allocate_memory_chunk(chunk_size)) {
  68.                 return(0);
  69.         }
  70.         return(get_memory_chunk(size));
  71.   }
  72.  
  73.   rval = ptr_next_byte;
  74.   ptr_next_byte += size;
  75.   bytes_left -= size;
  76.   return(rval);
  77.   
  78. }
  79.  
  80.  
  81. char * store_string (str,len) char *str; unsigned len;
  82.  
  83. /* put copy of a string into storage in a memory block.  Return a pointer to */
  84. /* the copied string.  Returns 0 if out of memory. */
  85.  
  86. { char *ptr_space;
  87.  
  88.   if (0 == (ptr_space = get_memory_chunk(len+1))) {
  89.         return(0);
  90.   }
  91.   strcpy(ptr_space,str);
  92.   return(ptr_space);
  93. }
  94.