/*
** The subroutines above are not tested by the usual test suite. To test
** these routines, compile just this one file with a -DENCODER_TEST=1 option
** and run the result.
*/
int main(int argc, char **argv){
int i, j, n, m, nOut, nByteIn, nByteOut;
unsigned char in[30000];
unsigned char out[33000];
nByteIn = nByteOut = 0;
for(i=0; i<sizeof(in); i++){
printf("Test %d: ", i+1);
n = rand() % (i+1);
if( i%100==0 ){
int k;
for(j=k=0; j<n; j++){
/* if( k==0 || k=='\'' ) k++; */
in[j] = k;
k = (k+1)&0xff;
}
}else{
for(j=0; j<n; j++) in[j] = rand() & 0xff;
}
nByteIn += n;
nOut = sqlite_encode_binary(in, n, out);
nByteOut += nOut;
if( nOut!=strlen(out) ){
printf(" ERROR return value is %d instead of %d\n", nOut, strlen(out));
exit(1);
}
if( nOut!=sqlite_encode_binary(in, n, 0) ){
printf(" ERROR actual output size disagrees with predicted size\n");
exit(1);
}
m = (256*n + 1262)/253;
printf("size %d->%d (max %d)", n, strlen(out)+1, m);
if( strlen(out)+1>m ){
printf(" ERROR output too big\n");
exit(1);
}
for(j=0; out[j]; j++){
if( out[j]=='\'' ){
printf(" ERROR contains (')\n");
exit(1);
}
}
j = sqlite_decode_binary(out, out);
if( j!=n ){
printf(" ERROR decode size %d\n", j);
exit(1);
}
if( memcmp(in, out, n)!=0 ){
printf(" ERROR decode mismatch\n");
exit(1);
}
printf(" OK\n");
}
fprintf(stderr,"Finished. Total encoding: %d->%d bytes\n",
nByteIn, nByteOut);
fprintf(stderr,"Avg size increase: %.3f%%\n",
(nByteOut-nByteIn)*100.0/(double)nByteIn);
}
/*
** The subroutines above are not tested by the usual test suite. To test
** these routines, compile just this one file with a -DENCODER_TEST=1 option
** and run the result.
*/
int main(int argc, char **argv){
int i, j, n, m, nOut;
unsigned char in[30000];
unsigned char out[33000];
for(i=0; i<sizeof(in); i++){
printf("Test %d: ", i+1);
n = rand() % (i+1);
if( i%100==0 ){
int k;
for(j=k=0; j<n; j++){
/* if( k==0 || k=='\'' ) k++; */
in[j] = k;
k = (k+1)&0xff;
}
}else{
for(j=0; j<n; j++) in[j] = rand() & 0xff;
}
nOut = sqlite_encode_binary(in, n, out);
if( nOut!=strlen(out) ){
printf(" ERROR return value is %d instead of %d\n", nOut, strlen(out));
exit(1);
}
m = (256*n + 1262)/253;
printf("size %d->%d (max %d)", n, strlen(out)+1, m);
if( strlen(out)+1>m ){
printf(" ERROR output too big\n");
exit(1);
}
for(j=0; out[j]; j++){
if( out[j]=='\'' ){
printf(" ERROR contains (')\n");
exit(1);
}
}
j = sqlite_decode_binary(out, out);
if( j!=n ){
printf(" ERROR decode size %d\n", j);
exit(1);
}
if( memcmp(in, out, n)!=0 ){
printf(" ERROR decode mismatch\n");
exit(1);
}
printf(" OK\n");
}
}
int
_ds_get_signature (DSPAM_CTX * CTX, struct _ds_spam_signature *SIG,
const char *signature)
{
struct _sqlite_drv_storage *s = (struct _sqlite_drv_storage *) CTX->storage;
unsigned long length;
unsigned char *mem;
char query[128];
char *err=NULL, **row;
int nrow, ncolumn;
void *ptr;
if (s->dbh == NULL)
{
LOGDEBUG ("_ds_get_signature: invalid database handle (NULL)");
return EINVAL;
}
snprintf (query, sizeof (query),
"select data, length(data) "
" from dspam_signature_data where signature = \"%s\"",
signature);
if ((sqlite_get_table(s->dbh, query, &row, &nrow, &ncolumn, &err))!=SQLITE_OK) {
_sqlite_drv_query_error (err, query);
return EFAILURE;
}
if (nrow<1)
sqlite_free_table(row);
if (nrow<1 || row == NULL)
return EFAILURE;
length = strlen(row[ncolumn]);
if (length == 0)
{
sqlite_free_table(row);
return EFAILURE;
}
mem = malloc(length+1);
if (mem == NULL) {
LOG(LOG_CRIT, ERR_MEM_ALLOC);
sqlite_free_table(row);
return EUNKNOWN;
}
length = sqlite_decode_binary((unsigned char *) row[ncolumn], mem);
if (length<=0) {
LOG(LOG_ERR, "sqlite_decode_binary() failed with error %d", length);
return EFAILURE;
}
ptr = realloc(mem, length);
if (ptr)
SIG->data = ptr;
else {
LOG(LOG_CRIT, ERR_MEM_ALLOC);
SIG->data = mem;
}
SIG->length = length;
sqlite_free_table(row);
return 0;
}
struct _ds_storage_signature *
_ds_get_nextsignature (DSPAM_CTX * CTX)
{
struct _sqlite_drv_storage *s = (struct _sqlite_drv_storage *) CTX->storage;
struct _ds_storage_signature *st;
unsigned long length;
char query[128];
unsigned char *mem;
char *err=NULL;
const char **row, *query_tail=NULL;
int ncolumn, x;
if (s->dbh == NULL)
{
LOGDEBUG ("_ds_get_nextsignature: invalid database handle (NULL)");
return NULL;
}
st = calloc (1, sizeof (struct _ds_storage_signature));
if (st == NULL)
{
LOG (LOG_CRIT, ERR_MEM_ALLOC);
return NULL;
}
if (s->iter_sig == NULL)
{
snprintf (query, sizeof (query),
"select data, signature, strftime('%%s', created_on), "
"length(data) from dspam_signature_data");
if ((sqlite_compile(s->dbh, query, &query_tail, &s->iter_sig, &err))
!=SQLITE_OK)
{
_sqlite_drv_query_error (err, query);
free(st);
return NULL;
}
}
if ((x = sqlite_step(s->iter_sig, &ncolumn, &row, NULL))
!=SQLITE_ROW) {
if (x != SQLITE_DONE) {
_sqlite_drv_query_error (err, query);
s->iter_sig = NULL;
free(st);
return NULL;
}
sqlite_finalize((struct sqlite_vm *) s->iter_sig, &err);
s->iter_sig = NULL;
free(st);
return NULL;
}
length = strtol(row[3], NULL, 0);
if (length == 0)
{
free(st);
return _ds_get_nextsignature(CTX);
}
mem = malloc (length+1);
if (mem == NULL)
{
LOG (LOG_CRIT, ERR_MEM_ALLOC);
sqlite_finalize(s->iter_sig, &err);
s->iter_sig = NULL;
free(st);
return NULL;
}
length = sqlite_decode_binary((const unsigned char *) &row[ncolumn], mem);
if (length<0) {
LOG(LOG_ERR, "sqlite_decode_binary() failed with error %d", length);
s->iter_sig = NULL;
free(st);
return NULL;
}
st->data = realloc(mem, length);
strlcpy(st->signature, row[1], sizeof(st->signature));
st->length = length;
st->created_on = (time_t) strtol(row[2], NULL, 0);
return st;
}
