96 lines
2.8 KiB
C
96 lines
2.8 KiB
C
// cl.exe defines macro _WIN32, but erl_interface.h checks for __WIN32__
|
|
#ifdef _WIN32
|
|
#define __WIN32__
|
|
#endif
|
|
|
|
#include <erl_driver.h>
|
|
#include <erl_interface.h>
|
|
#include <ei.h>
|
|
#include <sqlite3.h>
|
|
#include <stdio.h>
|
|
#include <ctype.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
|
|
#if SQLITE_VERSION_NUMBER < 3006001
|
|
#error "SQLite3 of version 3.6.1 minumum required"
|
|
#endif
|
|
|
|
// Path to file where data will be stored.
|
|
// It will be created if it doesn't exist
|
|
#define DB_PATH "./store.db"
|
|
#define LOG_PATH "/tmp/erlang-sqlite3-drv.log"
|
|
|
|
// Binary commands between Erlang VM and Driver
|
|
#define CMD_SQL_EXEC 2
|
|
// #define CMD_DEL 3
|
|
#define CMD_SQL_BIND_AND_EXEC 4
|
|
#define CMD_PREPARE 5
|
|
#define CMD_PREPARED_BIND 6
|
|
#define CMD_PREPARED_STEP 7
|
|
#define CMD_PREPARED_RESET 8
|
|
#define CMD_PREPARED_CLEAR_BINDINGS 9
|
|
#define CMD_PREPARED_FINALIZE 10
|
|
#define CMD_PREPARED_COLUMNS 11
|
|
|
|
// Number of bytes for each key
|
|
// (160 bits for SHA1 hash)
|
|
#define KEY_SIZE 20
|
|
|
|
typedef struct ptr_list {
|
|
void *head;
|
|
struct ptr_list *tail;
|
|
} ptr_list;
|
|
|
|
// Define struct to hold state across calls
|
|
typedef struct sqlite3_drv_t {
|
|
ErlDrvPort port;
|
|
unsigned int key;
|
|
struct sqlite3 *db;
|
|
long async_handle;
|
|
FILE *log;
|
|
sqlite3_stmt **prepared_stmts;
|
|
unsigned int prepared_count;
|
|
unsigned int prepared_alloc;
|
|
ErlDrvTermData atom_blob;
|
|
ErlDrvTermData atom_error;
|
|
ErlDrvTermData atom_columns;
|
|
ErlDrvTermData atom_rows;
|
|
ErlDrvTermData atom_null;
|
|
ErlDrvTermData atom_rowid;
|
|
ErlDrvTermData atom_ok;
|
|
ErlDrvTermData atom_done;
|
|
ErlDrvTermData atom_unknown_cmd;
|
|
} sqlite3_drv_t;
|
|
|
|
typedef struct async_sqlite3_command {
|
|
sqlite3_drv_t *driver_data;
|
|
sqlite3_stmt *statement;
|
|
ErlDrvTermData *dataset;
|
|
int term_count;
|
|
int term_allocated;
|
|
int row_count;
|
|
ptr_list *ptrs;
|
|
ptr_list *binaries;
|
|
int finalize_statement_on_free;
|
|
} async_sqlite3_command;
|
|
|
|
|
|
static ErlDrvData start(ErlDrvPort port, char* cmd);
|
|
static void stop(ErlDrvData handle);
|
|
static int control(ErlDrvData drv_data, unsigned int command, char *buf,
|
|
int len, char **rbuf, int rlen);
|
|
static int sql_exec(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int sql_bind_and_exec(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepare(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepared_bind(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepared_step(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepared_reset(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepared_clear_bindings(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepared_finalize(sqlite3_drv_t *drv, char *buf, int len);
|
|
static int prepared_columns(sqlite3_drv_t *drv, char *buf, int len);
|
|
static void sql_exec_async(void *async_command);
|
|
static void sql_free_async(void *async_command);
|
|
static void ready_async(ErlDrvData drv_data, ErlDrvThreadData thread_data);
|
|
static int unknown(sqlite3_drv_t *bdb_drv, char *buf, int len);
|