src/mmdb.c File Reference

#include "copyright.h"
#include "config.h"
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <dirent.h>
#include <sys/time.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <errno.h>
#include <event.h>
#include <string.h>
#include <sys/file.h>
#include <stdint.h>
#include "debug.h"

Include dependency graph for mmdb.c:

Go to the source code of this file.

Data Structures
struct	mmdb_t
Functions
mmdb_t *	mmdb_open_read (char *filename)
mmdb_t *	mmdb_open_write (char *filename)
void	mmdb_resize (struct mmdb_t *mmdb, int length)
void	mmdb_close (struct mmdb_t *mmdb)
void	mmdb_write (struct mmdb_t *mmdb, void *data, int length)
void	mmdb_write_uint32 (struct mmdb_t *mmdb, uint32_t data)
uint32_t	mmdb_read_uint32 (struct mmdb_t *mmdb)
unsigned int	mmdb_read_uint (struct mmdb_t *mmdb)
void	mmdb_write_uint (struct mmdb_t *mmdb, unsigned int val)
void	mmdb_write_uint64 (struct mmdb_t *mmdb, uint64_t data)
uint64_t	mmdb_read_uint64 (struct mmdb_t *mmdb)
void	mmdb_write_single (struct mmdb_t *mmdb, float data)
float	mmdb_read_single (struct mmdb_t *mmdb)
void	mmdb_write_double (struct mmdb_t *mmdb, double data)
double	mmdb_read_double (struct mmdb_t *mmdb)
void *	mmdb_read (struct mmdb_t *mmdb, void *dest, int length)
void	mmdb_write_opaque (struct mmdb_t *mmdb, void *data, int length)
void	mmdb_write_string (struct mmdb_t *mmdb, void *data)
char *	mmdb_read_string (struct mmdb_t *mmdb)
char	mmdb_read_opaque (struct mmdb_t *mmdb, void *dest, int maxlength)

---

Function Documentation

void mmdb_close

(

struct mmdb_t *

mmdb

)

Definition at line 112 of file mmdb.c.

References mmdb_t::base, dprintk, mmdb_t::fd, mmdb_t::length, and mmdb_t::ppos.

Referenced by dump_restart_db_xdr(), and load_restart_db_xdr().

{
        msync(mmdb->base, mmdb->length, MS_SYNC);
        munmap(mmdb->base, mmdb->length);
        ftruncate(mmdb->fd, mmdb->ppos - mmdb->base);
        dprintk("truncating to %d bytes.", mmdb->ppos - mmdb->base);
        close(mmdb->fd);
        mmdb->fd = 0;
        memset(mmdb, 0, sizeof(struct mmdb_t));
        free(mmdb);
}

struct mmdb_t* mmdb_open_read

(

char *

filename

)

Definition at line 32 of file mmdb.c.

References mmdb_t::base, mmdb_t::end, mmdb_t::fd, mmdb_t::length, and mmdb_t::ppos.

Referenced by load_restart_db_xdr(), and mmdb_db_read().

{
        struct stat statbuf;
        struct mmdb_t *mmdb;
        int fd, length;

        if(stat(filename, &statbuf) < 0) {
        return NULL;
    }
        mmdb = malloc(sizeof(struct mmdb_t));
    if(!mmdb) return NULL;

        fd = open(filename, O_RDONLY);
    if(!fd)  {
        free(mmdb);
        return NULL;
    }
    
        mmdb->fd = fd;
        mmdb->length = (statbuf.st_size + 0x3FF) & ~(0x3FF);
        mmdb->base =
                mmap(NULL, mmdb->length, PROT_READ, MAP_SHARED,
                         mmdb->fd, 0);
        if(!mmdb->base) {
        close(mmdb->fd);
        free(mmdb);
        return NULL;
    }
    mmdb->end = mmdb->base + mmdb->length;

        mmdb->ppos = mmdb->base;
        return mmdb;

}

struct mmdb_t* mmdb_open_write

(

char *

filename

)

Definition at line 67 of file mmdb.c.

References mmdb_t::base, dperror, dprintk, mmdb_t::end, mmdb_t::fd, mmdb_t::length, and mmdb_t::ppos.

Referenced by dump_restart_db_xdr(), and mmdb_db_write().

{
        struct stat statbuf;
        struct mmdb_t *mmdb;
        int fd, length;

        mmdb = malloc(sizeof(struct mmdb_t));
        dperror((fd = open(filename, O_RDWR | O_CREAT, 0644)) < 0);

        mmdb->fd = fd;
        mmdb->length = 0x400 * 16;
        mmdb->base = NULL;
        mmdb->ppos = NULL;
        dprintk("mmdb->length %d", mmdb->length);
        ftruncate(mmdb->fd, mmdb->length);
        fsync(mmdb->fd);
        mmdb->base =
                mmap(NULL, mmdb->length, PROT_READ | PROT_WRITE, MAP_SHARED, mmdb->fd,
                         0);
        dperror(mmdb->base == MAP_FAILED);
        mmdb->end = mmdb->base + mmdb->length;
        mmdb->ppos = mmdb->base;
        return mmdb;
}

void* mmdb_read	(	struct mmdb_t *	mmdb,
		void *	dest,
		int	length
	)

Definition at line 199 of file mmdb.c.

References mmdb_t::end, and mmdb_t::ppos.

Referenced by mmdb_db_read(), and mmdb_read_string().

{
        if((mmdb->end - mmdb->ppos) < length)
                return NULL;
        memcpy(dest, mmdb->ppos, length);
        mmdb->ppos += length;
        if(length & 3)
                mmdb->ppos += 4 - (length & 3);
        return dest;
}

double mmdb_read_double

(

struct mmdb_t *

mmdb

)

Definition at line 191 of file mmdb.c.

References mmdb_t::ppos.

{
        double data = *((double *) (mmdb->ppos));
        mmdb->ppos += sizeof(double);
        return ntohl(data);
}

char mmdb_read_opaque	(	struct mmdb_t *	mmdb,
		void *	dest,
		int	maxlength
	)

Definition at line 237 of file mmdb.c.

References mmdb_t::end, mmdb_t::length, mmdb_read_uint32(), and mmdb_t::ppos.

Referenced by load_restart_db_xdr().

                                                                      {
    int length = mmdb_read_uint32(mmdb);
    
        if((mmdb->end - mmdb->ppos) < length)
                return NULL;
    if(length > maxlength) {
        memcpy(dest, mmdb->ppos, maxlength);
    } else {
        memcpy(dest, mmdb->ppos, length);
    }
        mmdb->ppos += length;
        if(length & 3)
                mmdb->ppos += 4 - (length & 3);
        return dest;
}

float mmdb_read_single

(

struct mmdb_t *

mmdb

)

Definition at line 178 of file mmdb.c.

References mmdb_t::ppos.

{
        float data = *((float *) (mmdb->ppos));
        mmdb->ppos += sizeof(float);
        return ntohl(data);
}

char* mmdb_read_string

(

struct mmdb_t *

mmdb

)

Definition at line 227 of file mmdb.c.

References mmdb_t::length, mmdb_read(), and mmdb_read_uint32().

Referenced by load_bqe(), and load_restart_db_xdr().

                                            {
    char *tmp;
    int length;
    length = mmdb_read_uint32(mmdb);
    if(length == 0) return NULL;
    tmp = malloc(length);
    mmdb_read(mmdb, tmp, length);
    return tmp;
}

unsigned int mmdb_read_uint

(

struct mmdb_t *

mmdb

)

Definition at line 146 of file mmdb.c.

References mmdb_read_uint32(), and printk.

                                                 {
    printk("This function is deprecated, please use mmdb_read_uint32 instead.\n");
    return (unsigned int)mmdb_read_uint32(mmdb);
}

uint32_t mmdb_read_uint32

(

struct mmdb_t *

mmdb

)

Definition at line 139 of file mmdb.c.

References mmdb_t::ppos.

Referenced by cque_load_restart(), load_bqe(), load_objqe(), load_restart_db_xdr(), mmdb_db_read(), mmdb_read_opaque(), mmdb_read_string(), and mmdb_read_uint().

{
        uint32_t data = *((uint32_t *) (mmdb->ppos));
        mmdb->ppos += sizeof(uint32_t);
        return ntohl(data);
}

uint64_t mmdb_read_uint64

(

struct mmdb_t *

mmdb

)

Definition at line 164 of file mmdb.c.

References dprintk, and mmdb_t::ppos.

{
    dprintk("technically, this is not 64 bit safe yet.");
        uint64_t data = *((uint64_t *) (mmdb->ppos));
        mmdb->ppos += sizeof(uint64_t);
        return ntohl(data);
}

void mmdb_resize	(	struct mmdb_t *	mmdb,
		int	length
	)

Definition at line 92 of file mmdb.c.

References mmdb_t::base, dperror, mmdb_t::end, mmdb_t::fd, mmdb_t::length, and mmdb_t::ppos.

Referenced by mmdb_write().

{
        int offset = 0;
        if(mmdb->base) {
                offset = mmdb->ppos - mmdb->base;
                msync(mmdb->base, mmdb->length, MS_ASYNC);
                munmap(mmdb->base, mmdb->length);
                mmdb->base = NULL;
                mmdb->ppos = NULL;
        }
        mmdb->length = (length + 0x3FF) & ~(0x3FF);
        ftruncate(mmdb->fd, mmdb->length);
        mmdb->base =
                mmap(NULL, mmdb->length, PROT_READ | PROT_WRITE, MAP_SHARED, mmdb->fd,
                         0);
        dperror(mmdb->base == NULL);
        mmdb->end = mmdb->base + mmdb->length;
        mmdb->ppos = mmdb->base + offset;
}

void mmdb_write	(	struct mmdb_t *	mmdb,
		void *	data,
		int	length
	)

Definition at line 124 of file mmdb.c.

References mmdb_t::end, mmdb_t::length, mmdb_resize(), and mmdb_t::ppos.

Referenced by mmdb_write_double(), mmdb_write_opaque(), mmdb_write_single(), mmdb_write_uint32(), and mmdb_write_uint64().

{
        if(mmdb->end - mmdb->ppos < length) {
                mmdb_resize(mmdb, mmdb->length + length);
        }
        memcpy(mmdb->ppos, data, length);
        mmdb->ppos += length;
}

void mmdb_write_double	(	struct mmdb_t *	mmdb,
		double	data
	)

Definition at line 185 of file mmdb.c.

References mmdb_write().

{
        data = htonl(data);
        mmdb_write(mmdb, &data, sizeof(double));
}

void mmdb_write_opaque	(	struct mmdb_t *	mmdb,
		void *	data,
		int	length
	)

Definition at line 210 of file mmdb.c.

References mmdb_write(), and mmdb_write_uint32().

Referenced by do_save_com_xdr(), mmdb_db_write(), mmdb_write_object(), mmdb_write_string(), and mmdb_write_vattr().

{
        unsigned char *pad = (unsigned char *)"\x00\x00\x00\x00";
        mmdb_write_uint32(mmdb, length);
        mmdb_write(mmdb, data, length);
        if((length & 3) > 0)
                mmdb_write(mmdb, pad, 4 - (length & 3));
}

void mmdb_write_single	(	struct mmdb_t *	mmdb,
		float	data
	)

Definition at line 172 of file mmdb.c.

References mmdb_write().

{
        data = htonl(data);
        mmdb_write(mmdb, &data, sizeof(float));
}

void mmdb_write_string	(	struct mmdb_t *	mmdb,
		void *	data
	)

Definition at line 219 of file mmdb.c.

References mmdb_write_opaque(), and mmdb_write_uint32().

Referenced by dump_bqe(), and dump_restart_db_xdr().

                                                        {
    if(data == NULL) {
        mmdb_write_uint32(mmdb, 0);
    } else {
        mmdb_write_opaque(mmdb, data, strlen(data)+1);
    }
}

void mmdb_write_uint	(	struct mmdb_t *	mmdb,
		unsigned int	val
	)

Definition at line 151 of file mmdb.c.

References mmdb_write_uint32(), and printk.

Referenced by mmdb_db_write(), mmdb_write_object(), and mmdb_write_vattr().

                                                            {
    printk("This function is deprecated, please use mmdb_write_uint32 instead.\n");
    mmdb_write_uint32(mmdb, (uint32_t)val);
    return;
}

void mmdb_write_uint32	(	struct mmdb_t *	mmdb,
		uint32_t	data
	)

Definition at line 133 of file mmdb.c.

References mmdb_write().

Referenced by cque_dump_restart(), do_save_com_xdr(), dump_bqe(), dump_objqe(), dump_restart_db_xdr(), mmdb_db_write(), mmdb_write_opaque(), mmdb_write_string(), and mmdb_write_uint().

{
        data = htonl(data);
        mmdb_write(mmdb, &data, sizeof(uint32_t));
}

void mmdb_write_uint64	(	struct mmdb_t *	mmdb,
		uint64_t	data
	)

Definition at line 157 of file mmdb.c.

References dprintk, and mmdb_write().

{
    dprintk("technically, this is not 64 bit safe yet.");
        data = htonl(data);
        mmdb_write(mmdb, &data, sizeof(uint64_t));
}

---