1N/A/*
1N/A** Copyright (c) 1999-2002 Sendmail, Inc. and its suppliers.
1N/A** All rights reserved.
1N/A**
1N/A** By using this file, you agree to the terms and conditions set
1N/A** forth in the LICENSE file which can be found at the top level of
1N/A** the sendmail distribution.
1N/A*/
1N/A
1N/A#pragma ident "%Z%%M% %I% %E% SMI"
1N/A
1N/A#include <sm/gen.h>
1N/ASM_RCSID("@(#)$Id: smndbm.c,v 8.52 2002/05/21 22:30:30 gshapiro Exp $")
1N/A
1N/A#include <fcntl.h>
1N/A#include <stdlib.h>
1N/A#include <unistd.h>
1N/A
1N/A#include <sendmail/sendmail.h>
1N/A#include <libsmdb/smdb.h>
1N/A
1N/A#ifdef NDBM
1N/A
1N/A# define SMNDB_DIR_FILE_EXTENSION "dir"
1N/A# define SMNDB_PAG_FILE_EXTENSION "pag"
1N/A
1N/Astruct smdb_dbm_database_struct
1N/A{
1N/A DBM *smndbm_dbm;
1N/A int smndbm_lock_fd;
1N/A bool smndbm_cursor_in_use;
1N/A};
1N/Atypedef struct smdb_dbm_database_struct SMDB_DBM_DATABASE;
1N/A
1N/Astruct smdb_dbm_cursor_struct
1N/A{
1N/A SMDB_DBM_DATABASE *smndbmc_db;
1N/A datum smndbmc_current_key;
1N/A};
1N/Atypedef struct smdb_dbm_cursor_struct SMDB_DBM_CURSOR;
1N/A
1N/A/*
1N/A** SMDB_PUT_FLAGS_TO_NDBM_FLAGS -- Translates smdb put flags to ndbm put flags.
1N/A**
1N/A** Parameters:
1N/A** flags -- The flags to translate.
1N/A**
1N/A** Returns:
1N/A** The ndbm flags that are equivalent to the smdb flags.
1N/A**
1N/A** Notes:
1N/A** Any invalid flags are ignored.
1N/A**
1N/A*/
1N/A
1N/Aint
1N/Asmdb_put_flags_to_ndbm_flags(flags)
1N/A SMDB_FLAG flags;
1N/A{
1N/A int return_flags;
1N/A
1N/A return_flags = 0;
1N/A if (bitset(SMDBF_NO_OVERWRITE, flags))
1N/A return_flags = DBM_INSERT;
1N/A else
1N/A return_flags = DBM_REPLACE;
1N/A
1N/A return return_flags;
1N/A}
1N/A
1N/A/*
1N/A** Except for smdb_ndbm_open, the rest of these function correspond to the
1N/A** interface laid out in smdb.h.
1N/A*/
1N/A
1N/ASMDB_DBM_DATABASE *
1N/Asmdbm_malloc_database()
1N/A{
1N/A SMDB_DBM_DATABASE *db;
1N/A
1N/A db = (SMDB_DBM_DATABASE *) malloc(sizeof(SMDB_DBM_DATABASE));
1N/A if (db != NULL)
1N/A {
1N/A db->smndbm_dbm = NULL;
1N/A db->smndbm_lock_fd = -1;
1N/A db->smndbm_cursor_in_use = false;
1N/A }
1N/A
1N/A return db;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_close(database)
1N/A SMDB_DATABASE *database;
1N/A{
1N/A SMDB_DBM_DATABASE *db = (SMDB_DBM_DATABASE *) database->smdb_impl;
1N/A DBM *dbm = ((SMDB_DBM_DATABASE *) database->smdb_impl)->smndbm_dbm;
1N/A
1N/A dbm_close(dbm);
1N/A if (db->smndbm_lock_fd != -1)
1N/A close(db->smndbm_lock_fd);
1N/A
1N/A free(db);
1N/A database->smdb_impl = NULL;
1N/A
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_del(database, key, flags)
1N/A SMDB_DATABASE *database;
1N/A SMDB_DBENT *key;
1N/A unsigned int flags;
1N/A{
1N/A int result;
1N/A DBM *dbm = ((SMDB_DBM_DATABASE *) database->smdb_impl)->smndbm_dbm;
1N/A datum dbkey;
1N/A
1N/A (void) memset(&dbkey, '\0', sizeof dbkey);
1N/A dbkey.dptr = key->data;
1N/A dbkey.dsize = key->size;
1N/A
1N/A errno = 0;
1N/A result = dbm_delete(dbm, dbkey);
1N/A if (result != 0)
1N/A {
1N/A int save_errno = errno;
1N/A
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A
1N/A if (save_errno != 0)
1N/A return save_errno;
1N/A
1N/A return SMDBE_NOT_FOUND;
1N/A }
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_fd(database, fd)
1N/A SMDB_DATABASE *database;
1N/A int *fd;
1N/A{
1N/A DBM *dbm = ((SMDB_DBM_DATABASE *) database->smdb_impl)->smndbm_dbm;
1N/A
1N/A *fd = dbm_dirfno(dbm);
1N/A if (*fd <= 0)
1N/A return EINVAL;
1N/A
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_lockfd(database)
1N/A SMDB_DATABASE *database;
1N/A{
1N/A SMDB_DBM_DATABASE *db = (SMDB_DBM_DATABASE *) database->smdb_impl;
1N/A
1N/A return db->smndbm_lock_fd;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_get(database, key, data, flags)
1N/A SMDB_DATABASE *database;
1N/A SMDB_DBENT *key;
1N/A SMDB_DBENT *data;
1N/A unsigned int flags;
1N/A{
1N/A DBM *dbm = ((SMDB_DBM_DATABASE *) database->smdb_impl)->smndbm_dbm;
1N/A datum dbkey, dbdata;
1N/A
1N/A (void) memset(&dbkey, '\0', sizeof dbkey);
1N/A (void) memset(&dbdata, '\0', sizeof dbdata);
1N/A dbkey.dptr = key->data;
1N/A dbkey.dsize = key->size;
1N/A
1N/A errno = 0;
1N/A dbdata = dbm_fetch(dbm, dbkey);
1N/A if (dbdata.dptr == NULL)
1N/A {
1N/A int save_errno = errno;
1N/A
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A
1N/A if (save_errno != 0)
1N/A return save_errno;
1N/A
1N/A return SMDBE_NOT_FOUND;
1N/A }
1N/A data->data = dbdata.dptr;
1N/A data->size = dbdata.dsize;
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_put(database, key, data, flags)
1N/A SMDB_DATABASE *database;
1N/A SMDB_DBENT *key;
1N/A SMDB_DBENT *data;
1N/A unsigned int flags;
1N/A{
1N/A int result;
1N/A int save_errno;
1N/A DBM *dbm = ((SMDB_DBM_DATABASE *) database->smdb_impl)->smndbm_dbm;
1N/A datum dbkey, dbdata;
1N/A
1N/A (void) memset(&dbkey, '\0', sizeof dbkey);
1N/A (void) memset(&dbdata, '\0', sizeof dbdata);
1N/A dbkey.dptr = key->data;
1N/A dbkey.dsize = key->size;
1N/A dbdata.dptr = data->data;
1N/A dbdata.dsize = data->size;
1N/A
1N/A errno = 0;
1N/A result = dbm_store(dbm, dbkey, dbdata,
1N/A smdb_put_flags_to_ndbm_flags(flags));
1N/A switch (result)
1N/A {
1N/A case 1:
1N/A return SMDBE_DUPLICATE;
1N/A
1N/A case 0:
1N/A return SMDBE_OK;
1N/A
1N/A default:
1N/A save_errno = errno;
1N/A
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A
1N/A if (save_errno != 0)
1N/A return save_errno;
1N/A
1N/A return SMDBE_IO_ERROR;
1N/A }
1N/A /* NOTREACHED */
1N/A}
1N/A
1N/Aint
1N/Asmndbm_set_owner(database, uid, gid)
1N/A SMDB_DATABASE *database;
1N/A uid_t uid;
1N/A gid_t gid;
1N/A{
1N/A# if HASFCHOWN
1N/A int fd;
1N/A int result;
1N/A DBM *dbm = ((SMDB_DBM_DATABASE *) database->smdb_impl)->smndbm_dbm;
1N/A
1N/A fd = dbm_dirfno(dbm);
1N/A if (fd <= 0)
1N/A return EINVAL;
1N/A
1N/A result = fchown(fd, uid, gid);
1N/A if (result < 0)
1N/A return errno;
1N/A
1N/A fd = dbm_pagfno(dbm);
1N/A if (fd <= 0)
1N/A return EINVAL;
1N/A
1N/A result = fchown(fd, uid, gid);
1N/A if (result < 0)
1N/A return errno;
1N/A# endif /* HASFCHOWN */
1N/A
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_sync(database, flags)
1N/A SMDB_DATABASE *database;
1N/A unsigned int flags;
1N/A{
1N/A return SMDBE_UNSUPPORTED;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_cursor_close(cursor)
1N/A SMDB_CURSOR *cursor;
1N/A{
1N/A SMDB_DBM_CURSOR *dbm_cursor = (SMDB_DBM_CURSOR *) cursor->smdbc_impl;
1N/A SMDB_DBM_DATABASE *db = dbm_cursor->smndbmc_db;
1N/A
1N/A if (!db->smndbm_cursor_in_use)
1N/A return SMDBE_NOT_A_VALID_CURSOR;
1N/A
1N/A db->smndbm_cursor_in_use = false;
1N/A free(dbm_cursor);
1N/A free(cursor);
1N/A
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_cursor_del(cursor, flags)
1N/A SMDB_CURSOR *cursor;
1N/A unsigned int flags;
1N/A{
1N/A int result;
1N/A SMDB_DBM_CURSOR *dbm_cursor = (SMDB_DBM_CURSOR *) cursor->smdbc_impl;
1N/A SMDB_DBM_DATABASE *db = dbm_cursor->smndbmc_db;
1N/A DBM *dbm = db->smndbm_dbm;
1N/A
1N/A errno = 0;
1N/A result = dbm_delete(dbm, dbm_cursor->smndbmc_current_key);
1N/A if (result != 0)
1N/A {
1N/A int save_errno = errno;
1N/A
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A
1N/A if (save_errno != 0)
1N/A return save_errno;
1N/A
1N/A return SMDBE_NOT_FOUND;
1N/A }
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_cursor_get(cursor, key, value, flags)
1N/A SMDB_CURSOR *cursor;
1N/A SMDB_DBENT *key;
1N/A SMDB_DBENT *value;
1N/A SMDB_FLAG flags;
1N/A{
1N/A SMDB_DBM_CURSOR *dbm_cursor = (SMDB_DBM_CURSOR *) cursor->smdbc_impl;
1N/A SMDB_DBM_DATABASE *db = dbm_cursor->smndbmc_db;
1N/A DBM *dbm = db->smndbm_dbm;
1N/A datum dbkey, dbdata;
1N/A
1N/A (void) memset(&dbkey, '\0', sizeof dbkey);
1N/A (void) memset(&dbdata, '\0', sizeof dbdata);
1N/A
1N/A if (flags == SMDB_CURSOR_GET_RANGE)
1N/A return SMDBE_UNSUPPORTED;
1N/A
1N/A if (dbm_cursor->smndbmc_current_key.dptr == NULL)
1N/A {
1N/A dbm_cursor->smndbmc_current_key = dbm_firstkey(dbm);
1N/A if (dbm_cursor->smndbmc_current_key.dptr == NULL)
1N/A {
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A return SMDBE_LAST_ENTRY;
1N/A }
1N/A }
1N/A else
1N/A {
1N/A dbm_cursor->smndbmc_current_key = dbm_nextkey(dbm);
1N/A if (dbm_cursor->smndbmc_current_key.dptr == NULL)
1N/A {
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A return SMDBE_LAST_ENTRY;
1N/A }
1N/A }
1N/A
1N/A errno = 0;
1N/A dbdata = dbm_fetch(dbm, dbm_cursor->smndbmc_current_key);
1N/A if (dbdata.dptr == NULL)
1N/A {
1N/A int save_errno = errno;
1N/A
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A
1N/A if (save_errno != 0)
1N/A return save_errno;
1N/A
1N/A return SMDBE_NOT_FOUND;
1N/A }
1N/A value->data = dbdata.dptr;
1N/A value->size = dbdata.dsize;
1N/A key->data = dbm_cursor->smndbmc_current_key.dptr;
1N/A key->size = dbm_cursor->smndbmc_current_key.dsize;
1N/A
1N/A return SMDBE_OK;
1N/A}
1N/A
1N/Aint
1N/Asmdbm_cursor_put(cursor, key, value, flags)
1N/A SMDB_CURSOR *cursor;
1N/A SMDB_DBENT *key;
1N/A SMDB_DBENT *value;
1N/A SMDB_FLAG flags;
1N/A{
1N/A int result;
1N/A int save_errno;
1N/A SMDB_DBM_CURSOR *dbm_cursor = (SMDB_DBM_CURSOR *) cursor->smdbc_impl;
1N/A SMDB_DBM_DATABASE *db = dbm_cursor->smndbmc_db;
1N/A DBM *dbm = db->smndbm_dbm;
1N/A datum dbdata;
1N/A
1N/A (void) memset(&dbdata, '\0', sizeof dbdata);
1N/A dbdata.dptr = value->data;
1N/A dbdata.dsize = value->size;
1N/A
1N/A errno = 0;
1N/A result = dbm_store(dbm, dbm_cursor->smndbmc_current_key, dbdata,
1N/A smdb_put_flags_to_ndbm_flags(flags));
1N/A switch (result)
1N/A {
1N/A case 1:
1N/A return SMDBE_DUPLICATE;
1N/A
1N/A case 0:
1N/A return SMDBE_OK;
1N/A
1N/A default:
1N/A save_errno = errno;
1N/A
1N/A if (dbm_error(dbm))
1N/A return SMDBE_IO_ERROR;
1N/A
1N/A if (save_errno != 0)
1N/A return save_errno;
1N/A
1N/A return SMDBE_IO_ERROR;
1N/A }
1N/A /* NOTREACHED */
1N/A}
1N/A
1N/Aint
1N/Asmdbm_cursor(database, cursor, flags)
1N/A SMDB_DATABASE *database;
1N/A SMDB_CURSOR **cursor;
1N/A SMDB_FLAG flags;
1N/A{
1N/A SMDB_DBM_DATABASE *db = (SMDB_DBM_DATABASE *) database->smdb_impl;
1N/A SMDB_CURSOR *cur;
1N/A SMDB_DBM_CURSOR *dbm_cursor;
1N/A
1N/A if (db->smndbm_cursor_in_use)
1N/A return SMDBE_ONLY_SUPPORTS_ONE_CURSOR;
1N/A
1N/A db->smndbm_cursor_in_use = true;
1N/A dbm_cursor = (SMDB_DBM_CURSOR *) malloc(sizeof(SMDB_DBM_CURSOR));
1N/A dbm_cursor->smndbmc_db = db;
1N/A dbm_cursor->smndbmc_current_key.dptr = NULL;
1N/A dbm_cursor->smndbmc_current_key.dsize = 0;
1N/A
1N/A cur = (SMDB_CURSOR*) malloc(sizeof(SMDB_CURSOR));
1N/A if (cur == NULL)
1N/A return SMDBE_MALLOC;
1N/A
1N/A cur->smdbc_impl = dbm_cursor;
1N/A cur->smdbc_close = smdbm_cursor_close;
1N/A cur->smdbc_del = smdbm_cursor_del;
1N/A cur->smdbc_get = smdbm_cursor_get;
1N/A cur->smdbc_put = smdbm_cursor_put;
1N/A *cursor = cur;
1N/A
1N/A return SMDBE_OK;
1N/A}
1N/A/*
1N/A** SMDB_NDBM_OPEN -- Opens a ndbm database.
1N/A**
1N/A** Parameters:
1N/A** database -- An unallocated database pointer to a pointer.
1N/A** db_name -- The name of the database without extension.
1N/A** mode -- File permisions on a created database.
1N/A** mode_mask -- Mode bits that much match on an opened database.
1N/A** sff -- Flags to safefile.
1N/A** type -- The type of database to open.
1N/A** Only SMDB_NDBM is supported.
1N/A** user_info -- Information on the user to use for file
1N/A** permissions.
1N/A** db_params -- No params are supported.
1N/A**
1N/A** Returns:
1N/A** SMDBE_OK -- Success, otherwise errno:
1N/A** SMDBE_MALLOC -- Cannot allocate memory.
1N/A** SMDBE_UNSUPPORTED -- The type is not supported.
1N/A** SMDBE_GDBM_IS_BAD -- We have detected GDBM and we don't
1N/A** like it.
1N/A** SMDBE_BAD_OPEN -- dbm_open failed and errno was not set.
1N/A** Anything else: errno
1N/A*/
1N/A
1N/Aint
1N/Asmdb_ndbm_open(database, db_name, mode, mode_mask, sff, type, user_info,
1N/A db_params)
1N/A SMDB_DATABASE **database;
1N/A char *db_name;
1N/A int mode;
1N/A int mode_mask;
1N/A long sff;
1N/A SMDB_DBTYPE type;
1N/A SMDB_USER_INFO *user_info;
1N/A SMDB_DBPARAMS *db_params;
1N/A{
1N/A bool lockcreated = false;
1N/A int result;
1N/A int lock_fd;
1N/A SMDB_DATABASE *smdb_db;
1N/A SMDB_DBM_DATABASE *db;
1N/A DBM *dbm = NULL;
1N/A struct stat dir_stat_info;
1N/A struct stat pag_stat_info;
1N/A
1N/A result = SMDBE_OK;
1N/A *database = NULL;
1N/A
1N/A if (type == NULL)
1N/A return SMDBE_UNKNOWN_DB_TYPE;
1N/A
1N/A result = smdb_setup_file(db_name, SMNDB_DIR_FILE_EXTENSION, mode_mask,
1N/A sff, user_info, &dir_stat_info);
1N/A if (result != SMDBE_OK)
1N/A return result;
1N/A
1N/A result = smdb_setup_file(db_name, SMNDB_PAG_FILE_EXTENSION, mode_mask,
1N/A sff, user_info, &pag_stat_info);
1N/A if (result != SMDBE_OK)
1N/A return result;
1N/A
1N/A if ((dir_stat_info.st_mode == ST_MODE_NOFILE ||
1N/A pag_stat_info.st_mode == ST_MODE_NOFILE) &&
1N/A bitset(mode, O_CREAT))
1N/A lockcreated = true;
1N/A
1N/A lock_fd = -1;
1N/A result = smdb_lock_file(&lock_fd, db_name, mode, sff,
1N/A SMNDB_DIR_FILE_EXTENSION);
1N/A if (result != SMDBE_OK)
1N/A return result;
1N/A
1N/A if (lockcreated)
1N/A {
1N/A int pag_fd;
1N/A
1N/A /* Need to pre-open the .pag file as well with O_EXCL */
1N/A result = smdb_lock_file(&pag_fd, db_name, mode, sff,
1N/A SMNDB_PAG_FILE_EXTENSION);
1N/A if (result != SMDBE_OK)
1N/A {
1N/A (void) close(lock_fd);
1N/A return result;
1N/A }
1N/A (void) close(pag_fd);
1N/A
1N/A mode |= O_TRUNC;
1N/A mode &= ~(O_CREAT|O_EXCL);
1N/A }
1N/A
1N/A smdb_db = smdb_malloc_database();
1N/A if (smdb_db == NULL)
1N/A result = SMDBE_MALLOC;
1N/A
1N/A db = smdbm_malloc_database();
1N/A if (db == NULL)
1N/A result = SMDBE_MALLOC;
1N/A
1N/A /* Try to open database */
1N/A if (result == SMDBE_OK)
1N/A {
1N/A db->smndbm_lock_fd = lock_fd;
1N/A
1N/A errno = 0;
1N/A dbm = dbm_open(db_name, mode, DBMMODE);
1N/A if (dbm == NULL)
1N/A {
1N/A if (errno == 0)
1N/A result = SMDBE_BAD_OPEN;
1N/A else
1N/A result = errno;
1N/A }
1N/A db->smndbm_dbm = dbm;
1N/A }
1N/A
1N/A /* Check for GDBM */
1N/A if (result == SMDBE_OK)
1N/A {
1N/A if (dbm_dirfno(dbm) == dbm_pagfno(dbm))
1N/A result = SMDBE_GDBM_IS_BAD;
1N/A }
1N/A
1N/A /* Check for filechanged */
1N/A if (result == SMDBE_OK)
1N/A {
1N/A result = smdb_filechanged(db_name, SMNDB_DIR_FILE_EXTENSION,
1N/A dbm_dirfno(dbm), &dir_stat_info);
1N/A if (result == SMDBE_OK)
1N/A {
1N/A result = smdb_filechanged(db_name,
1N/A SMNDB_PAG_FILE_EXTENSION,
1N/A dbm_pagfno(dbm),
1N/A &pag_stat_info);
1N/A }
1N/A }
1N/A
1N/A /* XXX Got to get fchown stuff in here */
1N/A
1N/A /* Setup driver if everything is ok */
1N/A if (result == SMDBE_OK)
1N/A {
1N/A *database = smdb_db;
1N/A
1N/A smdb_db->smdb_close = smdbm_close;
1N/A smdb_db->smdb_del = smdbm_del;
1N/A smdb_db->smdb_fd = smdbm_fd;
1N/A smdb_db->smdb_lockfd = smdbm_lockfd;
1N/A smdb_db->smdb_get = smdbm_get;
1N/A smdb_db->smdb_put = smdbm_put;
1N/A smdb_db->smdb_set_owner = smndbm_set_owner;
1N/A smdb_db->smdb_sync = smdbm_sync;
1N/A smdb_db->smdb_cursor = smdbm_cursor;
1N/A
1N/A smdb_db->smdb_impl = db;
1N/A
1N/A return SMDBE_OK;
1N/A }
1N/A
1N/A /* If we're here, something bad happened, clean up */
1N/A if (dbm != NULL)
1N/A dbm_close(dbm);
1N/A
1N/A smdb_unlock_file(db->smndbm_lock_fd);
1N/A free(db);
1N/A smdb_free_database(smdb_db);
1N/A
1N/A return result;
1N/A}
1N/A#endif /* NDBM */