home *** CD-ROM | disk | FTP | other *** search
/ Media Share 13 / mediashare_13.zip / mediashare_13 / ZIPPED / PROGRAM / SNIP9404.ZIP / STRUCFIL.C < prev    next >
C/C++ Source or Header  |  1994-04-03  |  5KB  |  191 lines
  1. /*******************************************************************
  2.  * Generic structure <> disk file manipulations. These functions
  3.  * form a basic template for reading and writing structures to a 
  4.  * sequential file. This template is probably most useful for files
  5.  * with 500 or less records and eliminates the need for a more 
  6.  * elaborate file handler such as C-Tree, DB-Vista, Mix etc.
  7.  * Routines to put data in the struct is out of scope here.
  8.  * Written by Lynn Nash 8/28/91 and donated to the public domain.
  9.  */
  10. #include <io.h>
  11. #include <string.h>
  12. #include <stdlib.h>
  13. #include <stdio.h>
  14.  
  15. #ifndef ERROR
  16. #define ERROR -1
  17. #define OK     0
  18. #endif
  19.  
  20. #ifndef TRUE
  21. #define TRUE 1
  22. #define FALSE !TRUE
  23. #endif
  24.  
  25. /* make sure the record structure is byte aligned */
  26.  
  27. #if defined(__TURBOC__)
  28.  #pragma option -a-
  29. #elif defined(__ZTC__)
  30.  #pragma ZTC align 1
  31. #else /* MSC/QC/WATCOM/METAWARE */
  32.  #pragma pack(1)
  33. #endif
  34.  
  35. static struct blackbook {
  36.     int delete_flag;        /* 0 = active  -1 = deleted */
  37.     int recordnum;          /* a sequential number in the file */
  38.    /* The data fields in asciiz.  */
  39.     char firstname[11];         
  40.     char lastname[16];         
  41.     char addr[26];          
  42.     char city[16];
  43.     char state[3];
  44.     char zip[10];
  45.     char phone[11];
  46. } rec, oldrec;             /* 97 byte record  * 2 */
  47.  
  48. #pragma pack()
  49.  
  50. /*-------------------- general globals ---------------------*/
  51.  
  52. static long cur_rec = 0;      /* the current record number */
  53. static FILE *fsptr = NULL;    /* fixed record data file pointer */
  54.  
  55. /* if file exists open in read/write mode else create file */
  56.  
  57. FILE * open_file(char *filename)
  58. {
  59.       if (access(filename, 0) == 0)
  60.             fsptr = fopen(filename, "rb+"); 
  61.       else  fsptr = fopen(filename, "wb+"); 
  62.       return fsptr;                       /* return the file pointer */
  63. } 
  64.  
  65. /* add new records to the data file */
  66.  
  67. int datadd(void)
  68. {
  69.       if (fsptr)
  70.       {
  71.             if (fseek(fsptr, 0L, SEEK_END) != 0)
  72.                   return ERROR;  /* seek failure */
  73.             rec.delete_flag = 0; /* active record tag */
  74.             rec.recordnum = (int) (ftell(fsptr) / 
  75.                   (long) sizeof(struct blackbook));
  76.             if (fwrite(&rec, sizeof(struct blackbook), 1, fsptr) != 1)
  77.             {
  78.                   return ERROR; /* write error */
  79.             }
  80.             else
  81.             {
  82.                   /* put your clean up code here */ 
  83.                   return OK;
  84.             }
  85.       }
  86.       return ERROR;
  87. }
  88.  
  89. /* tag the last record read in the file as deleted */
  90.  
  91. int data_delete(void)
  92. {
  93.       if (fsptr)
  94.       {
  95.             if (fseek(fsptr, (long) sizeof(struct blackbook) * -1L,
  96.                   SEEK_CUR) != 0)
  97.             {
  98.                   return ERROR;
  99.             }
  100.             rec.delete_flag = -1;   /* tag the record as deleted */
  101.             if (fwrite(&rec, sizeof(struct blackbook), 1, fsptr) != 1)
  102.                   return ERROR;
  103.             else  return OK;
  104.       }
  105.       return ERROR;
  106. }
  107.  
  108. /* read a random structure. If successful the global cur_rec will 
  109.  * contain the number of the last record read & it can be compared
  110.  * to the number in the struct as a double check (belt & suspenders)
  111.  */
  112.  
  113. int data_read(long recnum)
  114. {
  115.       if (fseek(fsptr, (long) sizeof(struct blackbook) * recnum,
  116.             SEEK_SET) != 0)
  117.       {
  118.             return ERROR;
  119.       }
  120.       cur_rec = recnum; /* keep tabs on record pointer in global */  
  121.  
  122.       /* now read the record into save struct*/
  123.  
  124.       if (fread(&oldrec, sizeof(struct blackbook), 1, fsptr) != 1)
  125.       {
  126.             return ERROR;
  127.       }
  128.       else                          /* copy save struct to edit struct */
  129.       {
  130.             memcpy(&rec, &oldrec, sizeof(struct blackbook));
  131.             return OK;
  132.       }
  133. }
  134.  
  135. /* rewrite the last read record back to disk */
  136.  
  137. int data_update(void)
  138. {
  139.       if (memcmp(&rec, &oldrec, sizeof(struct blackbook)) == 0)
  140.             return TRUE;  /* no update needed */
  141.  
  142.       /* back up one record before writing */
  143.  
  144.       if (fseek(fsptr, (long) sizeof(struct blackbook) * -1L, 
  145.             SEEK_CUR) != 0)
  146.       {
  147.             return ERROR; /* seek error */ 
  148.       }
  149.  
  150.       /* now write the record */
  151.  
  152.       if (fwrite(&rec, sizeof(struct blackbook), 1, fsptr) != 1)
  153.             return ERROR; /* write error */
  154.       return OK;
  155. }
  156.  
  157. /* get the next valid record in the file */
  158.  
  159. int read_forward(void)
  160. {
  161.       do
  162.       {
  163.             cur_rec++; /* upcount the record number */
  164.             if (data_read(cur_rec) != 0)
  165.             {
  166.                   cur_rec--; /* decrement the record number */
  167.                   return ERROR;
  168.             }
  169.       } while (oldrec.delete_flag != 0); /* record read was deleted */
  170.       return OK;
  171. }
  172.  
  173. /* get the previous valid record in the file */
  174.  
  175. int read_backward(void)
  176. {
  177.       do
  178.       {
  179.             cur_rec--;  /* decrement the record number */
  180.             if (cur_rec >= 0)
  181.             {
  182.                   if ( data_read(cur_rec) != 0 )
  183.                   {
  184.                         cur_rec++; /* increment the record number */
  185.                         return ERROR;
  186.                   }
  187.             }  
  188.       } while (oldrec.delete_flag != 0); /* record read was deleted */
  189.       return OK;
  190. }
  191.