Return to test_loadext.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sqlite3 / src

sqlite3

    1: /*
    2: ** 2006 June 14
    3: **
    4: ** The author disclaims copyright to this source code.  In place of
    5: ** a legal notice, here is a blessing:
    6: **
    7: **    May you do good and not evil.
    8: **    May you find forgiveness for yourself and forgive others.
    9: **    May you share freely, never taking more than you give.
   10: **
   11: *************************************************************************
   12: ** Test extension for testing the sqlite3_load_extension() function.
   13: */
   14: #include <string.h>
   15: #include "sqlite3ext.h"
   16: SQLITE_EXTENSION_INIT1
   17: 
   18: /*
   19: ** The half() SQL function returns half of its input value.
   20: */
   21: static void halfFunc(
   22:   sqlite3_context *context,
   23:   int argc,
   24:   sqlite3_value **argv
   25: ){
   26:   sqlite3_result_double(context, 0.5*sqlite3_value_double(argv[0]));
   27: }
   28: 
   29: /*
   30: ** SQL functions to call the sqlite3_status function and return results.
   31: */
   32: static void statusFunc(
   33:   sqlite3_context *context,
   34:   int argc,
   35:   sqlite3_value **argv
   36: ){
   37:   int op, mx, cur, resetFlag, rc;
   38:   if( sqlite3_value_type(argv[0])==SQLITE_INTEGER ){
   39:     op = sqlite3_value_int(argv[0]);
   40:   }else if( sqlite3_value_type(argv[0])==SQLITE_TEXT ){
   41:     int i;
   42:     const char *zName;
   43:     static const struct {
   44:       const char *zName;
   45:       int op;
   46:     } aOp[] = {
   47:       { "MEMORY_USED",         SQLITE_STATUS_MEMORY_USED         },
   48:       { "PAGECACHE_USED",      SQLITE_STATUS_PAGECACHE_USED      },
   49:       { "PAGECACHE_OVERFLOW",  SQLITE_STATUS_PAGECACHE_OVERFLOW  },
   50:       { "SCRATCH_USED",        SQLITE_STATUS_SCRATCH_USED        },
   51:       { "SCRATCH_OVERFLOW",    SQLITE_STATUS_SCRATCH_OVERFLOW    },
   52:       { "MALLOC_SIZE",         SQLITE_STATUS_MALLOC_SIZE         },
   53:     };
   54:     int nOp = sizeof(aOp)/sizeof(aOp[0]);
   55:     zName = (const char*)sqlite3_value_text(argv[0]);
   56:     for(i=0; i<nOp; i++){
   57:       if( strcmp(aOp[i].zName, zName)==0 ){
   58:         op = aOp[i].op;
   59:         break;
   60:       }
   61:     }
   62:     if( i>=nOp ){
   63:       char *zMsg = sqlite3_mprintf("unknown status property: %s", zName);
   64:       sqlite3_result_error(context, zMsg, -1);
   65:       sqlite3_free(zMsg);
   66:       return;
   67:     }
   68:   }else{
   69:     sqlite3_result_error(context, "unknown status type", -1);
   70:     return;
   71:   }
   72:   if( argc==2 ){
   73:     resetFlag = sqlite3_value_int(argv[1]);
   74:   }else{
   75:     resetFlag = 0;
   76:   }
   77:   rc = sqlite3_status(op, &cur, &mx, resetFlag);
   78:   if( rc!=SQLITE_OK ){
   79:     char *zMsg = sqlite3_mprintf("sqlite3_status(%d,...) returns %d", op, rc);
   80:     sqlite3_result_error(context, zMsg, -1);
   81:     sqlite3_free(zMsg);
   82:     return;
   83:   } 
   84:   if( argc==2 ){
   85:     sqlite3_result_int(context, mx);
   86:   }else{
   87:     sqlite3_result_int(context, cur);
   88:   }
   89: }
   90: 
   91: /*
   92: ** Extension load function.
   93: */
   94: int testloadext_init(
   95:   sqlite3 *db, 
   96:   char **pzErrMsg, 
   97:   const sqlite3_api_routines *pApi
   98: ){
   99:   int nErr = 0;
  100:   SQLITE_EXTENSION_INIT2(pApi);
  101:   nErr |= sqlite3_create_function(db, "half", 1, SQLITE_ANY, 0, halfFunc, 0, 0);
  102:   nErr |= sqlite3_create_function(db, "sqlite3_status", 1, SQLITE_ANY, 0,
  103:                           statusFunc, 0, 0);
  104:   nErr |= sqlite3_create_function(db, "sqlite3_status", 2, SQLITE_ANY, 0,
  105:                           statusFunc, 0, 0);
  106:   return nErr ? SQLITE_ERROR : SQLITE_OK;
  107: }
  108: 
  109: /*
  110: ** Another extension entry point. This one always fails.
  111: */
  112: int testbrokenext_init(
  113:   sqlite3 *db, 
  114:   char **pzErrMsg, 
  115:   const sqlite3_api_routines *pApi
  116: ){
  117:   char *zErr;
  118:   SQLITE_EXTENSION_INIT2(pApi);
  119:   zErr = sqlite3_mprintf("broken!");
  120:   *pzErrMsg = zErr;
  121:   return 1;
  122: }