disk.c File Reference

#include <rz_types.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <rz_util/rz_file.h>
#include <rz_util/rz_sys.h>
#include <rz_util/rz_utf8.h>
#include <rz_windows.h>
#include "sdb.h"

Go to the source code of this file.

Macros
#define	IFRET(x)

Functions
RZ_API bool	sdb_disk_create (Sdb *s)
RZ_API bool	sdb_disk_insert (Sdb *s, const char *key, const char *val)
RZ_API bool	sdb_disk_finish (Sdb *s)
RZ_API bool	sdb_disk_unlink (Sdb *s)

Macro Definition Documentation

IFRET

#define IFRET

(

x

)

Value:

    if (x) \
    ret = 0

Definition at line 76 of file disk.c.

Function Documentation

sdb_disk_create()

RZ_API bool sdb_disk_create

(

Sdb *

s

)

Definition at line 17 of file disk.c.

                                    {
    int nlen;
    char *str;
    const char *dir;
    if (!s || s->fdump >= 0) {
        return false; // cannot re-create
    }
    if (!s->dir && s->name) {
        s->dir = strdup(s->name);
    }
    dir = s->dir ? s->dir : "./";
    RZ_FREE(s->ndump);
    nlen = strlen(dir);
    str = malloc(nlen + 5);
    if (!str) {
        return false;
    }
    memcpy(str, dir, nlen + 1);
    char *dirname = rz_file_dirname(str);
    if (!dirname) {
        free(str);
        return false;
    }
    rz_sys_mkdirp(dirname);
    free(dirname);
    memcpy(str + nlen, ".tmp", 5);
    if (s->fdump != -1) {
        close(s->fdump);
    }
#if __WINDOWS__
    wchar_t *wstr = rz_utf8_to_utf16(str);
    if (wstr) {
        s->fdump = _wopen(wstr, O_BINARY | O_RDWR | O_CREAT | O_TRUNC, SDB_MODE);
        free(wstr);
    } else {
        s->fdump = -1;
    }
#else
    s->fdump = open(str, O_BINARY | O_RDWR | O_CREAT | O_TRUNC, SDB_MODE);
#endif
    if (s->fdump == -1) {
        eprintf("sdb: Cannot open '%s' for writing.\n", str);
        free(str);
        return false;
    }
    cdb_make_start(&s->m, s->fdump);
    s->ndump = str;
    return true;
}

References cdb_make_start(), close, eprintf, free(), malloc(), memcpy(), O_BINARY, O_CREAT, O_RDWR, O_TRUNC, rz_file_dirname(), RZ_FREE, rz_sys_mkdirp(), s, SDB_MODE, cmd_descs_generate::str, and strdup().

Referenced by sdb_sync().

sdb_disk_finish()

RZ_API bool sdb_disk_finish

(

Sdb *

s

)

Definition at line 79 of file disk.c.

                                    {
    bool reopen = false, ret = true;
    IFRET(!cdb_make_finish(&s->m));
#if HAVE_HEADER_SYS_MMAN_H
    IFRET(fsync(s->fdump));
#endif
    IFRET(close(s->fdump));
    s->fdump = -1;
    // close current fd to avoid sharing violations
    if (s->fd != -1) {
        close(s->fd);
        s->fd = -1;
        reopen = true;
    }
#if __WINDOWS__
    wchar_t *ndump_ = rz_utf8_to_utf16(s->ndump);
    wchar_t *dir_ = rz_utf8_to_utf16(s->dir);
 
    if (!MoveFileExW(ndump_, dir_, MOVEFILE_REPLACE_EXISTING)) {
        rz_sys_perror("MoveFileExW SDB file to finale location");
    }
    free(ndump_);
    free(dir_);
#else
    if (s->ndump && s->dir) {
        IFRET(rename(s->ndump, s->dir));
    }
#endif
    free(s->ndump);
    s->ndump = NULL;
    // reopen if was open before
    reopen = true; // always reopen if possible
    if (reopen) {
        int rr = sdb_open(s, s->dir);
        if (ret && rr < 0) {
            ret = false;
        }
        cdb_init(&s->db, s->fd);
    }
    return ret;
}

References cdb_init(), cdb_make_finish(), close, rz_socket_t::fd, free(), fsync, IFRET, NULL, rename, rz_sys_perror, s, and sdb_open().

Referenced by sdb_sync().

sdb_disk_insert()

RZ_API bool sdb_disk_insert	(	Sdb *	s,
		const char *	key,
		const char *	val
	)

Definition at line 67 of file disk.c.

                                                                      {
    struct cdb_make *c = &s->m;
    if (!key || !val) {
        return false;
    }
    // if (!*val) return 0; //undefine variable if no value
    return cdb_make_add(c, key, strlen(key), val, strlen(val));
}

References c, cdb_make_add(), key, s, and val.

Referenced by _insert_into_disk(), insertkeys(), and sdb_sync().

sdb_disk_unlink()

RZ_API bool sdb_disk_unlink

(

Sdb *

s

)

Definition at line 121 of file disk.c.

                                    {
    return (s->dir && *(s->dir) && unlink(s->dir) != -1);
}

References s, and unlink.

Referenced by sdb_unlink().