/*

 ** 2006 June 14

 **

 ** The author disclaims copyright to this source code.  In place of

 ** a legal notice, here is a blessing:

 **

 **    May you do good and not evil.

 **    May you find forgiveness for yourself and forgive others.

 **    May you share freely, never taking more than you give.

 **

 *************************************************************************

 ** Test extension for testing the sqlite3_load_extension() function.

 **

 ** $Id: test_loadext.c,v 1.3 2008/08/02 03:50:39 drh Exp $

 */

 #include <string.h>

 #include "sqlite3ext.h"

 SQLITE_EXTENSION_INIT1

 /*

 ** The half() SQL function returns half of its input value.

 */

 static void halfFunc(

   sqlite3_context *context,

   int argc,

   sqlite3_value **argv

 ){

   sqlite3_result_double(context, 0.5*sqlite3_value_double(argv[0]));

 }

 /*

 ** SQL functions to call the sqlite3_status function and return results.

 */

 static void statusFunc(

   sqlite3_context *context,

   int argc,

   sqlite3_value **argv

 ){

   int op, mx, cur, resetFlag, rc;

   if( sqlite3_value_type(argv[0])==SQLITE_INTEGER ){

     op = sqlite3_value_int(argv[0]);

   }else if( sqlite3_value_type(argv[0])==SQLITE_TEXT ){

     int i;

     const char *zName;

     static const struct {

       const char *zName;

       int op;

     } aOp[] = {

       { "MEMORY_USED",         SQLITE_STATUS_MEMORY_USED         },

       { "PAGECACHE_USED",      SQLITE_STATUS_PAGECACHE_USED      },

       { "PAGECACHE_OVERFLOW",  SQLITE_STATUS_PAGECACHE_OVERFLOW  },

       { "SCRATCH_USED",        SQLITE_STATUS_SCRATCH_USED        },

       { "SCRATCH_OVERFLOW",    SQLITE_STATUS_SCRATCH_OVERFLOW    },

       { "MALLOC_SIZE",         SQLITE_STATUS_MALLOC_SIZE         },

     };

     int nOp = sizeof(aOp)/sizeof(aOp[0]);

     zName = (const char*)sqlite3_value_text(argv[0]);

     for(i=0; i<nOp; i++){

       if( strcmp(aOp[i].zName, zName)==0 ){

         op = aOp[i].op;

         break;

       }

     }

     if( i>=nOp ){

       char *zMsg = sqlite3_mprintf("unknown status property: %s", zName);

       sqlite3_result_error(context, zMsg, -1);

       sqlite3_free(zMsg);

       return;

     }

   }else{

     sqlite3_result_error(context, "unknown status type", -1);

     return;

   }

   if( argc==2 ){

     resetFlag = sqlite3_value_int(argv[1]);

   }else{

     resetFlag = 0;

   }

   rc = sqlite3_status(op, &cur, &mx, resetFlag);

   if( rc!=SQLITE_OK ){

     char *zMsg = sqlite3_mprintf("sqlite3_status(%d,...) returns %d", op, rc);

     sqlite3_result_error(context, zMsg, -1);

     sqlite3_free(zMsg);

     return;

   } 

   if( argc==2 ){

     sqlite3_result_int(context, mx);

   }else{

     sqlite3_result_int(context, cur);

   }

 }

 /*

 ** Extension load function.

 */

 int testloadext_init(

   sqlite3 *db, 

   char **pzErrMsg, 

   const sqlite3_api_routines *pApi

 ){

   int nErr = 0;

   SQLITE_EXTENSION_INIT2(pApi);

   nErr |= sqlite3_create_function(db, "half", 1, SQLITE_ANY, 0, halfFunc, 0, 0);

   nErr |= sqlite3_create_function(db, "sqlite3_status", 1, SQLITE_ANY, 0,

                           statusFunc, 0, 0);

   nErr |= sqlite3_create_function(db, "sqlite3_status", 2, SQLITE_ANY, 0,

                           statusFunc, 0, 0);

   return nErr ? SQLITE_ERROR : SQLITE_OK;

 }

 /*

 ** Another extension entry point. This one always fails.

 */

 int testbrokenext_init(

   sqlite3 *db, 

   char **pzErrMsg, 

   const sqlite3_api_routines *pApi

 ){

   char *zErr;

   SQLITE_EXTENSION_INIT2(pApi);

   zErr = sqlite3_mprintf("broken!");

   *pzErrMsg = zErr;

   return 1;

 }