home *** CD-ROM | disk | FTP | other *** search
/ Geek Gadgets 1 / ADE-1.bin / ade-dist / gdbm-1.7.3-base.tgz / gdbm-1.7.3-base.tar / fsf / gdbm / gdbmopen.c < prev    next >
C/C++ Source or Header  |  1993-12-26  |  13KB  |  444 lines
  1. /* gdbmopen.c - Open the dbm file and initialize data structures for use. */
  2.  
  3. /*  This file is part of GDBM, the GNU data base manager, by Philip A. Nelson.
  4.     Copyright (C) 1990, 1991, 1993  Free Software Foundation, Inc.
  5.  
  6.     GDBM is free software; you can redistribute it and/or modify
  7.     it under the terms of the GNU General Public License as published by
  8.     the Free Software Foundation; either version 2, or (at your option)
  9.     any later version.
  10.  
  11.     GDBM is distributed in the hope that it will be useful,
  12.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.     GNU General Public License for more details.
  15.  
  16.     You should have received a copy of the GNU General Public License
  17.     along with GDBM; see the file COPYING.  If not, write to
  18.     the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  19.  
  20.     You may contact the author by:
  21.        e-mail:  phil@cs.wwu.edu
  22.       us-mail:  Philip A. Nelson
  23.                 Computer Science Department
  24.                 Western Washington University
  25.                 Bellingham, WA 98226
  26.        
  27. *************************************************************************/
  28.  
  29.  
  30. /* AIX demands this be the very first thing in the file. */
  31. #if !defined(__GNUC__) && defined(_AIX)
  32.  #pragma alloca
  33. #endif
  34.  
  35. /* include system configuration before all else. */
  36. #include "autoconf.h"
  37.  
  38. #include "gdbmdefs.h"
  39. #include "gdbmerrno.h"
  40.  
  41. /* Initialize dbm system.  FILE is a pointer to the file name.  If the file
  42.    has a size of zero bytes, a file initialization procedure is performed,
  43.    setting up the initial structure in the file.  BLOCK_SIZE is used during
  44.    initialization to determine the size of various constructs.  If the value
  45.    is less than 512, the file system blocksize is used, otherwise the value
  46.    of BLOCK_SIZE is used.  BLOCK_SIZE is ignored if the file has previously
  47.    initialized.  If FLAGS is set to GDBM_READ the user wants to just
  48.    read the database and any call to dbm_store or dbm_delete will fail. Many
  49.    readers can access the database at the same time.  If FLAGS is set to
  50.    GDBM_WRITE, the user wants both read and write access to the database and
  51.    requires exclusive access.  If FLAGS is GDBM_WRCREAT, the user wants
  52.    both read and write access to the database and if the database does not
  53.    exist, create a new one.  If FLAGS is GDBM_NEWDB, the user want a
  54.    new database created, regardless of whether one existed, and wants read
  55.    and write access to the new database.  Any error detected will cause a 
  56.    return value of null and an approprate value will be in gdbm_errno.  If
  57.    no errors occur, a pointer to the "gdbm file descriptor" will be
  58.    returned. */
  59.    
  60.  
  61. gdbm_file_info *
  62. gdbm_open (file, block_size, flags, mode, fatal_func)
  63.      char *file;
  64.      int  block_size;
  65.      int  flags;
  66.      int  mode;
  67.      void (*fatal_func) ();
  68. {
  69.   gdbm_file_info *dbf;        /* The record to return. */
  70.   struct stat file_stat;    /* Space for the stat information. */
  71.   int         len;        /* Length of the file name. */
  72.   int         num_bytes;    /* Used in reading and writing. */
  73.   off_t       file_pos;        /* Used with seeks. */
  74.   int          lock_val;         /* Returned by the flock call. */
  75.   int          file_block_size;    /* Block size to use for a new file. */
  76.   int           index;        /* Used as a loop index. */
  77.   char        need_trunc;    /* Used with GDBM_NEWDB and locking to avoid
  78.                    truncating a file from under a reader. */
  79.  
  80.   /* Initialize the gdbm_errno variable. */
  81.   gdbm_errno = GDBM_NO_ERROR;
  82.  
  83.   /* Allocate new info structure. */
  84.   dbf = (gdbm_file_info *) malloc (sizeof (gdbm_file_info));
  85.   if (dbf == NULL)
  86.     {
  87.       gdbm_errno = GDBM_MALLOC_ERROR;
  88.       return NULL;
  89.     }
  90.  
  91.   /* Initialize some fields for known values.  This is done so gdbm_close
  92.      will work if called before allocating some structures. */
  93.   dbf->dir  = NULL;
  94.   dbf->bucket = NULL;
  95.   dbf->header = NULL;
  96.   dbf->bucket_cache = NULL;
  97.   dbf->cache_size = 0;
  98.   
  99.   /* Save name of file. */
  100.   len = strlen (file);
  101.   dbf->name = (char *) malloc (len + 1);
  102.   if (dbf->name == NULL)
  103.     {
  104.       free (dbf);
  105.       gdbm_errno = GDBM_MALLOC_ERROR;
  106.       return NULL;
  107.     }
  108.   strcpy (dbf->name, file);
  109.  
  110.   /* Initialize the fatal error routine. */
  111.   dbf->fatal_err = fatal_func;
  112.  
  113.   /* Check for fast writers. */
  114.   if (flags & GDBM_FAST)
  115.     {
  116.       dbf->fast_write = TRUE;
  117.       flags -= GDBM_FAST;
  118.     }
  119.   else
  120.     {
  121.       dbf->fast_write = FALSE;
  122.     }
  123.   
  124.   /* Open the file. */
  125.   need_trunc = FALSE;
  126.   if (flags == GDBM_READER)
  127.     {
  128.       dbf->desc = open (dbf->name, O_RDONLY, 0);
  129.     }
  130.   else if (flags == GDBM_WRITER)
  131.     {
  132.       dbf->desc = open (dbf->name, O_RDWR, 0);
  133.     }
  134.   else if (flags == GDBM_NEWDB)
  135.     {
  136.       dbf->desc = open (dbf->name, O_RDWR|O_CREAT, mode);
  137.       flags = GDBM_WRITER;
  138.       need_trunc = TRUE;
  139.     }
  140.   else
  141.     {
  142.       dbf->desc = open (dbf->name, O_RDWR|O_CREAT, mode);
  143.       flags = GDBM_WRITER;
  144.     }
  145.   if (dbf->desc < 0)
  146.     {
  147.       free (dbf->name);
  148.       free (dbf);
  149.       gdbm_errno = GDBM_FILE_OPEN_ERROR;
  150.       return NULL;
  151.     }
  152.  
  153.   /* Get the status of the file. */
  154.   fstat (dbf->desc, &file_stat);
  155.  
  156.   /* Lock the file in the approprate way. */
  157.   if (flags == GDBM_READER)
  158.     {
  159.       if (file_stat.st_size == 0)
  160.     {
  161.       close (dbf->desc);
  162.       free (dbf->name);
  163.       free (dbf);
  164.       gdbm_errno = GDBM_EMPTY_DATABASE;
  165.       return NULL;
  166.     }
  167.       /* Sets lock_val to 0 for success.  See systems.h. */
  168.       READLOCK_FILE(dbf);
  169.     }
  170.   else
  171.     {
  172.       /* Sets lock_val to 0 for success.  See systems.h. */
  173.       WRITELOCK_FILE(dbf);
  174.     }
  175.   if (lock_val != 0)
  176.     {
  177.       close (dbf->desc);
  178.       free (dbf->name);
  179.       free (dbf);
  180.       if (flags == GDBM_READER)
  181.     gdbm_errno = GDBM_CANT_BE_READER;
  182.       else
  183.     gdbm_errno = GDBM_CANT_BE_WRITER;
  184.       return NULL;
  185.     }
  186.  
  187.   /* Record the kind of user. */
  188.   dbf->read_write = flags;
  189.  
  190.   /* If we do have a write lock and it was a GDBM_NEWDB, it is 
  191.      now time to truncate the file. */
  192.   if (need_trunc && file_stat.st_size != 0)
  193.     {
  194.       TRUNCATE (dbf);
  195.       fstat (dbf->desc, &file_stat);
  196.     }
  197.  
  198.   /* Decide if this is a new file or an old file. */
  199.   if (file_stat.st_size == 0)
  200.     {
  201.  
  202.       /* This is a new file.  Create an empty database.  */
  203.  
  204.       /* Start with the blocksize. */
  205.       if (block_size < 512)
  206.     file_block_size = STATBLKSIZE;
  207.       else
  208.     file_block_size = block_size;
  209.  
  210.       /* Get space for the file header. */
  211.       dbf->header = (gdbm_file_header *) malloc (file_block_size);
  212.       if (dbf->header == NULL)
  213.     {
  214.       gdbm_close (dbf);
  215.       gdbm_errno = GDBM_MALLOC_ERROR;
  216.       return NULL;
  217.     }
  218.  
  219.       /* Set the magic number and the block_size. */
  220.       dbf->header->header_magic = 0x13579ace;
  221.       dbf->header->block_size = file_block_size;
  222.      
  223.       /* Create the initial hash table directory.  */
  224.       dbf->header->dir_size = 8 * sizeof (off_t);
  225.       dbf->header->dir_bits = 3;
  226.       while (dbf->header->dir_size < dbf->header->block_size)
  227.     {
  228.       dbf->header->dir_size <<= 1;
  229.       dbf->header->dir_bits += 1;
  230.     }
  231.  
  232.       /* Check for correct block_size. */
  233.       if (dbf->header->dir_size != dbf->header->block_size)
  234.     {
  235.       gdbm_close (dbf);
  236.       gdbm_errno = GDBM_BLOCK_SIZE_ERROR;
  237.       return NULL;
  238.     }
  239.  
  240.       /* Allocate the space for the directory. */
  241.       dbf->dir = (off_t *) malloc (dbf->header->dir_size);
  242.       if (dbf->dir == NULL)
  243.     {
  244.       gdbm_close (dbf);
  245.       gdbm_errno = GDBM_MALLOC_ERROR;
  246.       return NULL;
  247.     }
  248.       dbf->header->dir = dbf->header->block_size;
  249.  
  250.       /* Create the first and only hash bucket. */
  251.       dbf->header->bucket_elems =
  252.     (dbf->header->block_size - sizeof (hash_bucket))
  253.     / sizeof (bucket_element) + 1;
  254.       dbf->header->bucket_size  = dbf->header->block_size;
  255. #if !defined(sgi)
  256.       dbf->bucket = (hash_bucket *) (alloca (dbf->header->bucket_size));
  257. #else    /* sgi */
  258.       /* The SGI C compiler doesn't accept the previous form. */
  259.       {
  260.         hash_bucket *ptr;
  261.     ptr = (hash_bucket *) (alloca (dbf->header->bucket_size));
  262.     dbf->bucket = ptr;
  263.       }
  264. #endif    /* sgi */
  265.       if (dbf->bucket == NULL)
  266.     {
  267.       gdbm_close (dbf);
  268.       gdbm_errno = GDBM_MALLOC_ERROR;
  269.       return NULL;
  270.     }
  271.       _gdbm_new_bucket (dbf, dbf->bucket, 0);
  272.       dbf->bucket->av_count = 1;
  273.       dbf->bucket->bucket_avail[0].av_adr = 3*dbf->header->block_size;
  274.       dbf->bucket->bucket_avail[0].av_size = dbf->header->block_size;
  275.  
  276.       /* Set table entries to point to hash buckets. */
  277.       for (index = 0; index < dbf->header->dir_size / sizeof (off_t); index++)
  278.     dbf->dir[index] = 2*dbf->header->block_size;
  279.  
  280.       /* Initialize the active avail block. */
  281.       dbf->header->avail.size
  282.     = ( (dbf->header->block_size - sizeof (gdbm_file_header))
  283.      / sizeof (avail_elem)) + 1;
  284.       dbf->header->avail.count = 0;
  285.       dbf->header->avail.next_block = 0;
  286.       dbf->header->next_block  = 4*dbf->header->block_size;
  287.  
  288.       /* Write initial configuration to the file. */
  289.       /* Block 0 is the file header and active avail block. */
  290.       num_bytes = write (dbf->desc, dbf->header, dbf->header->block_size);
  291.       if (num_bytes != dbf->header->block_size)
  292.     {
  293.       gdbm_close (dbf);
  294.       gdbm_errno = GDBM_FILE_WRITE_ERROR;
  295.       return NULL;
  296.     }
  297.  
  298.       /* Block 1 is the initial bucket directory. */
  299.       num_bytes = write (dbf->desc, dbf->dir, dbf->header->dir_size);
  300.       if (num_bytes != dbf->header->dir_size)
  301.     {
  302.       gdbm_close (dbf);
  303.       gdbm_errno = GDBM_FILE_WRITE_ERROR;
  304.       return NULL;
  305.     }
  306.  
  307.       /* Block 2 is the only bucket. */
  308.       num_bytes = write (dbf->desc, dbf->bucket, dbf->header->bucket_size);
  309.       if (num_bytes != dbf->header->bucket_size)
  310.     {
  311.       gdbm_close (dbf);
  312.       gdbm_errno = GDBM_FILE_WRITE_ERROR;
  313.       return NULL;
  314.     }
  315.  
  316.       /* Wait for initial configuration to be written to disk. */
  317.       fsync (dbf->desc);
  318.  
  319.     }
  320.   else
  321.     {
  322.       /* This is an old database.  Read in the information from the file
  323.      header and initialize the hash directory. */
  324.  
  325.       gdbm_file_header partial_header;  /* For the first part of it. */
  326.  
  327.       /* Read the partial file header. */
  328.       num_bytes = read (dbf->desc, &partial_header, sizeof (gdbm_file_header));
  329.       if (num_bytes != sizeof (gdbm_file_header))
  330.     {
  331.       gdbm_close (dbf);
  332.       gdbm_errno = GDBM_FILE_READ_ERROR;
  333.       return NULL;
  334.     }
  335.  
  336.       /* Is the magic number good? */
  337.       if (partial_header.header_magic != 0x13579ace)
  338.     {
  339.       gdbm_close (dbf);
  340.       gdbm_errno = GDBM_BAD_MAGIC_NUMBER;
  341.       return NULL;
  342.     }
  343.  
  344.       /* It is a good database, read the entire header. */
  345.       dbf->header = (gdbm_file_header *) malloc (partial_header.block_size);
  346.       if (dbf->header == NULL)
  347.     {
  348.       gdbm_close (dbf);
  349.       gdbm_errno = GDBM_MALLOC_ERROR;
  350.       return NULL;
  351.     }
  352.       bcopy (&partial_header, dbf->header, sizeof (gdbm_file_header));
  353.       num_bytes = read (dbf->desc, &dbf->header->avail.av_table[1],
  354.             dbf->header->block_size-sizeof (gdbm_file_header));
  355.       if (num_bytes != dbf->header->block_size-sizeof (gdbm_file_header))
  356.     {
  357.       gdbm_close (dbf);
  358.       gdbm_errno = GDBM_FILE_READ_ERROR;
  359.       return NULL;
  360.     }
  361.     
  362.       /* Allocate space for the hash table directory.  */
  363.       dbf->dir = (off_t *) malloc (dbf->header->dir_size);
  364.       if (dbf->dir == NULL)
  365.     {
  366.       gdbm_close (dbf);
  367.       gdbm_errno = GDBM_MALLOC_ERROR;
  368.       return NULL;
  369.     }
  370.  
  371.       /* Read the hash table directory. */
  372.       file_pos = lseek (dbf->desc, dbf->header->dir, L_SET);
  373.       if (file_pos != dbf->header->dir)
  374.     {
  375.       gdbm_close (dbf);
  376.       gdbm_errno = GDBM_FILE_SEEK_ERROR;
  377.       return NULL;
  378.     }
  379.  
  380.       num_bytes = read (dbf->desc, dbf->dir, dbf->header->dir_size);
  381.       if (num_bytes != dbf->header->dir_size)
  382.     {
  383.       gdbm_close (dbf);
  384.       gdbm_errno = GDBM_FILE_READ_ERROR;
  385.       return NULL;
  386.     }
  387.  
  388.     }
  389.       
  390.   /* Finish initializing dbf. */
  391.   dbf->last_read = -1;
  392.   dbf->bucket = NULL;
  393.   dbf->bucket_dir = 0;
  394.   dbf->cache_entry = NULL;
  395.   dbf->header_changed = FALSE;
  396.   dbf->directory_changed = FALSE;
  397.   dbf->bucket_changed = FALSE;
  398.   dbf->second_changed = FALSE;
  399.   
  400.   /* Everything is fine, return the pointer to the file
  401.      information structure.  */
  402.   return dbf;
  403.  
  404. }
  405.  
  406. /* initialize the bucket cache. */
  407. int
  408. _gdbm_init_cache(dbf, size)
  409.     gdbm_file_info *dbf;
  410.     int size;
  411. {
  412. register int index;
  413.  
  414.   if (dbf->bucket_cache == NULL)
  415.     {
  416.       dbf->bucket_cache = (cache_elem *) malloc(sizeof(cache_elem) * size);
  417.       if(dbf->bucket_cache == NULL)
  418.         {
  419.           gdbm_errno = GDBM_MALLOC_ERROR;
  420.           return(-1);
  421.         }
  422.       dbf->cache_size = size;
  423.  
  424.       for(index = 0; index < size; index++)
  425.         {
  426.           (dbf->bucket_cache[index]).ca_bucket
  427.             = (hash_bucket *) malloc (dbf->header->bucket_size);
  428.           if ((dbf->bucket_cache[index]).ca_bucket == NULL)
  429.         {
  430.               gdbm_errno = GDBM_MALLOC_ERROR;
  431.           return(-1);
  432.             }
  433.           (dbf->bucket_cache[index]).ca_adr = 0;
  434.           (dbf->bucket_cache[index]).ca_changed = FALSE;
  435.           (dbf->bucket_cache[index]).ca_data.hash_val = -1;
  436.           (dbf->bucket_cache[index]).ca_data.elem_loc = -1;
  437.           (dbf->bucket_cache[index]).ca_data.dptr = NULL;
  438.         }
  439.       dbf->bucket = dbf->bucket_cache[0].ca_bucket;
  440.       dbf->cache_entry = &dbf->bucket_cache[0];
  441.     }
  442.   return(0);
  443. }
  444.