1 /* 2 ** 2006 August 23 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_auto_extension() function. 13 */ 14 #include "tcl.h" 15 #include "sqlite3ext.h" 16 17 #ifndef SQLITE_OMIT_LOAD_EXTENSION 18 SQLITE_EXTENSION_INIT1 19 20 /* 21 ** The sqr() SQL function returns the square of its input value. 22 */ 23 static void sqrFunc( 24 sqlite3_context *context, 25 int argc, 26 sqlite3_value **argv 27 ){ 28 double r = sqlite3_value_double(argv[0]); 29 sqlite3_result_double(context, r*r); 30 } 31 32 /* 33 ** This is the entry point to register the extension for the sqr() function. 34 */ 35 static int sqr_init( 36 sqlite3 *db, 37 char **pzErrMsg, 38 const sqlite3_api_routines *pApi 39 ){ 40 SQLITE_EXTENSION_INIT2(pApi); 41 sqlite3_create_function(db, "sqr", 1, SQLITE_ANY, 0, sqrFunc, 0, 0); 42 return 0; 43 } 44 45 /* 46 ** The cube() SQL function returns the cube of its input value. 47 */ 48 static void cubeFunc( 49 sqlite3_context *context, 50 int argc, 51 sqlite3_value **argv 52 ){ 53 double r = sqlite3_value_double(argv[0]); 54 sqlite3_result_double(context, r*r*r); 55 } 56 57 /* 58 ** This is the entry point to register the extension for the cube() function. 59 */ 60 static int cube_init( 61 sqlite3 *db, 62 char **pzErrMsg, 63 const sqlite3_api_routines *pApi 64 ){ 65 SQLITE_EXTENSION_INIT2(pApi); 66 sqlite3_create_function(db, "cube", 1, SQLITE_ANY, 0, cubeFunc, 0, 0); 67 return 0; 68 } 69 70 /* 71 ** This is a broken extension entry point 72 */ 73 static int broken_init( 74 sqlite3 *db, 75 char **pzErrMsg, 76 const sqlite3_api_routines *pApi 77 ){ 78 char *zErr; 79 SQLITE_EXTENSION_INIT2(pApi); 80 zErr = sqlite3_mprintf("broken autoext!"); 81 *pzErrMsg = zErr; 82 return 1; 83 } 84 85 /* 86 ** tclcmd: sqlite3_auto_extension_sqr 87 ** 88 ** Register the "sqr" extension to be loaded automatically. 89 */ 90 static int autoExtSqrObjCmd( 91 void * clientData, 92 Tcl_Interp *interp, 93 int objc, 94 Tcl_Obj *CONST objv[] 95 ){ 96 int rc = sqlite3_auto_extension((void*)sqr_init); 97 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 98 return SQLITE_OK; 99 } 100 101 /* 102 ** tclcmd: sqlite3_cancel_auto_extension_sqr 103 ** 104 ** Unregister the "sqr" extension. 105 */ 106 static int cancelAutoExtSqrObjCmd( 107 void * clientData, 108 Tcl_Interp *interp, 109 int objc, 110 Tcl_Obj *CONST objv[] 111 ){ 112 int rc = sqlite3_cancel_auto_extension((void*)sqr_init); 113 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 114 return SQLITE_OK; 115 } 116 117 /* 118 ** tclcmd: sqlite3_auto_extension_cube 119 ** 120 ** Register the "cube" extension to be loaded automatically. 121 */ 122 static int autoExtCubeObjCmd( 123 void * clientData, 124 Tcl_Interp *interp, 125 int objc, 126 Tcl_Obj *CONST objv[] 127 ){ 128 int rc = sqlite3_auto_extension((void*)cube_init); 129 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 130 return SQLITE_OK; 131 } 132 133 /* 134 ** tclcmd: sqlite3_cancel_auto_extension_cube 135 ** 136 ** Unregister the "cube" extension. 137 */ 138 static int cancelAutoExtCubeObjCmd( 139 void * clientData, 140 Tcl_Interp *interp, 141 int objc, 142 Tcl_Obj *CONST objv[] 143 ){ 144 int rc = sqlite3_cancel_auto_extension((void*)cube_init); 145 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 146 return SQLITE_OK; 147 } 148 149 /* 150 ** tclcmd: sqlite3_auto_extension_broken 151 ** 152 ** Register the broken extension to be loaded automatically. 153 */ 154 static int autoExtBrokenObjCmd( 155 void * clientData, 156 Tcl_Interp *interp, 157 int objc, 158 Tcl_Obj *CONST objv[] 159 ){ 160 int rc = sqlite3_auto_extension((void*)broken_init); 161 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 162 return SQLITE_OK; 163 } 164 165 /* 166 ** tclcmd: sqlite3_cancel_auto_extension_broken 167 ** 168 ** Unregister the broken extension. 169 */ 170 static int cancelAutoExtBrokenObjCmd( 171 void * clientData, 172 Tcl_Interp *interp, 173 int objc, 174 Tcl_Obj *CONST objv[] 175 ){ 176 int rc = sqlite3_cancel_auto_extension((void*)broken_init); 177 Tcl_SetObjResult(interp, Tcl_NewIntObj(rc)); 178 return SQLITE_OK; 179 } 180 181 #endif /* SQLITE_OMIT_LOAD_EXTENSION */ 182 183 184 /* 185 ** tclcmd: sqlite3_reset_auto_extension 186 ** 187 ** Reset all auto-extensions 188 */ 189 static int resetAutoExtObjCmd( 190 void * clientData, 191 Tcl_Interp *interp, 192 int objc, 193 Tcl_Obj *CONST objv[] 194 ){ 195 sqlite3_reset_auto_extension(); 196 return SQLITE_OK; 197 } 198 199 200 /* 201 ** This procedure registers the TCL procs defined in this file. 202 */ 203 int Sqlitetest_autoext_Init(Tcl_Interp *interp){ 204 #ifndef SQLITE_OMIT_LOAD_EXTENSION 205 Tcl_CreateObjCommand(interp, "sqlite3_auto_extension_sqr", 206 autoExtSqrObjCmd, 0, 0); 207 Tcl_CreateObjCommand(interp, "sqlite3_auto_extension_cube", 208 autoExtCubeObjCmd, 0, 0); 209 Tcl_CreateObjCommand(interp, "sqlite3_auto_extension_broken", 210 autoExtBrokenObjCmd, 0, 0); 211 Tcl_CreateObjCommand(interp, "sqlite3_cancel_auto_extension_sqr", 212 cancelAutoExtSqrObjCmd, 0, 0); 213 Tcl_CreateObjCommand(interp, "sqlite3_cancel_auto_extension_cube", 214 cancelAutoExtCubeObjCmd, 0, 0); 215 Tcl_CreateObjCommand(interp, "sqlite3_cancel_auto_extension_broken", 216 cancelAutoExtBrokenObjCmd, 0, 0); 217 #endif 218 Tcl_CreateObjCommand(interp, "sqlite3_reset_auto_extension", 219 resetAutoExtObjCmd, 0, 0); 220 return TCL_OK; 221 } 222