Browse Source
Add sqlite session handler.
Add sqlite session handler.
Modified (quite a bit!) patch from John Coggeshall. It compiles, but it otherwise untested. session.save_path == path to actual database file for the session. eg: session.save_path=/tmp/mysite-session.dbPEAR_1_4DEV
Wez Furlong
23 years ago
3 changed files with 221 additions and 2 deletions
@ -0,0 +1,213 @@ |
|||
/* |
|||
+----------------------------------------------------------------------+ |
|||
| PHP Version 4 | |
|||
+----------------------------------------------------------------------+ |
|||
| Copyright (c) 1997-2003 The PHP Group | |
|||
+----------------------------------------------------------------------+ |
|||
| This source file is subject to version 3.0 of the PHP license, | |
|||
| that is bundled with this package in the file LICENSE, and is | |
|||
| available through the world-wide-web at the following url: | |
|||
| http://www.php.net/license/3_0.txt. | |
|||
| If you did not receive a copy of the PHP license and are unable to | |
|||
| obtain it through the world-wide-web, please send a note to | |
|||
| license@php.net so we can mail you a copy immediately. | |
|||
+----------------------------------------------------------------------+ |
|||
| Authors: John Coggeshall <john@php.net> | |
|||
| Wez Furlong <wez@thebrainroom.com> | |
|||
+----------------------------------------------------------------------+ |
|||
*/ |
|||
|
|||
/* $Id$ */ |
|||
|
|||
#include "php.h" |
|||
#include "ext/session/php_session.h" |
|||
#include <sqlite.h> |
|||
|
|||
#define PS_SQLITE_DATA sqlite *db = (sqlite*)PS_GET_MOD_DATA() |
|||
extern int sqlite_encode_binary(const unsigned char *in, int n, unsigned char *out); |
|||
extern int sqlite_decode_binary(const unsigned char *in, unsigned char *out); |
|||
|
|||
#define CREATE_TBL_QUERY "CREATE TABLE session_data(sess_id TEXT PRIMARY KEY, value TEXT, updated INTEGER)" |
|||
#define INSERT_QUERY "REPLACE INTO session_data VALUES('%q', '%q', %d)" |
|||
#define SELECT_QUERY "SELECT value FROM session_data WHERE sess_id='%q' LIMIT 1" |
|||
#define GC_QUERY "DELETE FROM session_data WHERE (%d - updated) > %d" |
|||
#define DELETE_QUERY "DELETE FROM session_data WHERE sess_id='%q'" |
|||
|
|||
PS_FUNCS(sqlite); |
|||
|
|||
ps_module ps_mod_sqlite = { |
|||
PS_MOD(sqlite) |
|||
}; |
|||
|
|||
/* If you change the logic here, please also update the error message in |
|||
* ps_sqlite_open() appropriately (code taken from ps_files_valid_key()) */ |
|||
|
|||
static int ps_sqlite_valid_key(const char *key) |
|||
{ |
|||
size_t len; |
|||
const char *p; |
|||
char c; |
|||
int ret = 1; |
|||
|
|||
for (p = key; (c = *p); p++) { |
|||
/* valid characters are a..z,A..Z,0..9 */ |
|||
if (!((c >= 'a' && c <= 'z') |
|||
|| (c >= 'A' && c <= 'Z') |
|||
|| (c >= '0' && c <= '9') |
|||
|| c == ',' |
|||
|| c == '-')) { |
|||
ret = 0; |
|||
break; |
|||
} |
|||
} |
|||
|
|||
len = p - key; |
|||
|
|||
if (len == 0) |
|||
ret = 0; |
|||
|
|||
return ret; |
|||
} |
|||
|
|||
PS_OPEN_FUNC(sqlite) |
|||
{ |
|||
char *filepath; |
|||
char *errmsg = NULL; |
|||
int spath_len, sname_len, fp_len; |
|||
sqlite *db; |
|||
|
|||
/* TODO: do we need a safe_mode check here? */ |
|||
db = sqlite_open(save_path, 0666, &errmsg); |
|||
if (db == NULL) { |
|||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "SQLite: failed to open/create session database `%s' - %s", save_path, errmsg); |
|||
sqlite_freemem(errmsg); |
|||
return FAILURE; |
|||
} |
|||
|
|||
/* allow up to 1 minute when busy */ |
|||
sqlite_busy_timeout(db, 60000); |
|||
|
|||
/* This will fail if the table already exists, but that's not a big problem. I'm |
|||
unclear as to how to check for a table's existence in SQLite -- that would be better here. */ |
|||
sqlite_exec(db, CREATE_TBL_QUERY, NULL, NULL, NULL); |
|||
|
|||
PS_SET_MOD_DATA(db); |
|||
|
|||
return SUCCESS; |
|||
} |
|||
|
|||
PS_CLOSE_FUNC(sqlite) |
|||
{ |
|||
PS_SQLITE_DATA; |
|||
|
|||
sqlite_close(db); |
|||
|
|||
return SUCCESS; |
|||
} |
|||
|
|||
PS_READ_FUNC(sqlite) |
|||
{ |
|||
PS_SQLITE_DATA; |
|||
char *query; |
|||
sqlite_vm *vm; |
|||
int colcount, result; |
|||
const char **rowdata, **colnames; |
|||
char *error; |
|||
|
|||
if (!ps_sqlite_valid_key(key)) { |
|||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "SQLite: The session id contains illegal characters, valid characters are a-z, A-Z, 0-9 and '-,'"); |
|||
return FAILURE; |
|||
} |
|||
|
|||
query = sqlite_mprintf(SELECT_QUERY, key); |
|||
if (query == NULL) { |
|||
/* no memory */ |
|||
return FAILURE; |
|||
} |
|||
|
|||
if (sqlite_compile(db, query, NULL, &vm, &error) != SQLITE_OK) { |
|||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "SQLite: Could not compile session read query: %s", error); |
|||
sqlite_freemem(error); |
|||
sqlite_freemem(query); |
|||
return FAILURE; |
|||
} |
|||
|
|||
switch ((result = sqlite_step(vm, &colcount, &rowdata, &colnames))) { |
|||
case SQLITE_ROW: |
|||
if (rowdata[0] == NULL) { |
|||
*vallen = 0; |
|||
*val = NULL; |
|||
} else { |
|||
*vallen = strlen(rowdata[0]); |
|||
*val = emalloc(*vallen); |
|||
*vallen = sqlite_decode_binary(rowdata[0], *val); |
|||
(*val)[*vallen] = '\0'; |
|||
} |
|||
break; |
|||
default: |
|||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "SQLite: session read query failed: %s", error); |
|||
sqlite_freemem(error); |
|||
error = NULL; |
|||
} |
|||
|
|||
if (SQLITE_OK != sqlite_finalize(vm, &error)) { |
|||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "SQLite: session read: error %s", error); |
|||
sqlite_freemem(error); |
|||
error = NULL; |
|||
} |
|||
|
|||
sqlite_freemem(query); |
|||
|
|||
return *val == NULL ? FAILURE : SUCCESS; |
|||
} |
|||
|
|||
PS_WRITE_FUNC(sqlite) |
|||
{ |
|||
PS_SQLITE_DATA; |
|||
char *error; |
|||
int result = SUCCESS; |
|||
time_t t; |
|||
char *binary; |
|||
int binlen; |
|||
|
|||
t = time(NULL); |
|||
|
|||
binary = emalloc((256 * vallen + 1262) / 253); |
|||
binlen = sqlite_encode_binary((const unsigned char*)val, vallen, binary); |
|||
|
|||
if (SQLITE_OK != sqlite_exec_printf(db, INSERT_QUERY, NULL, NULL, &error, key, binary, t)) { |
|||
php_error_docref(NULL TSRMLS_CC, E_WARNING, "SQLite: session write query failed: %s", error); |
|||
sqlite_freemem(error); |
|||
result = FAILURE; |
|||
} |
|||
|
|||
efree(binary); |
|||
|
|||
return result; |
|||
} |
|||
|
|||
PS_DESTROY_FUNC(sqlite) |
|||
{ |
|||
PS_SQLITE_DATA; |
|||
|
|||
return SQLITE_OK == sqlite_exec_printf(db, DELETE_QUERY, NULL, NULL, NULL, key) ? |
|||
SUCCESS : FAILURE; |
|||
} |
|||
|
|||
PS_GC_FUNC(sqlite) |
|||
{ |
|||
PS_SQLITE_DATA; |
|||
time_t t = time(NULL); |
|||
|
|||
return SQLITE_OK == sqlite_exec_printf(db, GC_QUERY, NULL, NULL, NULL, t, maxlifetime) ? |
|||
SUCCESS : FAILURE; |
|||
} |
|||
|
|||
/* |
|||
* Local variables: |
|||
* tab-width: 4 |
|||
* c-basic-offset: 4 |
|||
* End: |
|||
* vim600: sw=4 ts=4 fdm=marker |
|||
* vim<600: sw=4 ts=4 |
|||
*/ |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue