src/test_backup.c

   1 /*
   2 ** 2009 January 28
   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 ** This file contains test logic for the sqlite3_backup() interface.
  13 **
  14 */
  15
  16 #include "tclsqlite.h"
  17 #include "sqlite3.h"
  18 #include <assert.h>
  19
  20 /* These functions are implemented in main.c. */
  21 extern const char *sqlite3ErrName(int);
  22
  23 /* These functions are implemented in test1.c. */
  24 extern int getDbPointer(Tcl_Interp *, const char *, sqlite3 **);
  25
  26 static int SQLITE_TCLAPI backupTestCmd(
  27   ClientData clientData,
  28   Tcl_Interp *interp,
  29   int objc,
  30   Tcl_Obj *const*objv
  31 ){
  32   enum BackupSubCommandEnum {
  33     BACKUP_STEP, BACKUP_FINISH, BACKUP_REMAINING, BACKUP_PAGECOUNT
  34   };
  35   struct BackupSubCommand {
  36     const char *zCmd;
  37     enum BackupSubCommandEnum eCmd;
  38     int nArg;
  39     const char *zArg;
  40   } aSub[] = {
  41     {"step",      BACKUP_STEP      , 1, "npage" },
  42     {"finish",    BACKUP_FINISH    , 0, ""      },
  43     {"remaining", BACKUP_REMAINING , 0, ""      },
  44     {"pagecount", BACKUP_PAGECOUNT , 0, ""      },
  45     {0, 0, 0, 0}
  46   };
  47
  48   sqlite3_backup *p = (sqlite3_backup *)clientData;
  49   int iCmd;
  50   int rc;
  51
  52   rc = Tcl_GetIndexFromObjStruct(
  53       interp, objv[1], aSub, sizeof(aSub[0]), "option", 0, &iCmd
  54   );
  55   if( rc!=TCL_OK ){
  56     return rc;
  57   }
  58   if( objc!=(2 + aSub[iCmd].nArg) ){
  59     Tcl_WrongNumArgs(interp, 2, objv, aSub[iCmd].zArg);
  60     return TCL_ERROR;
  61   }
  62
  63   switch( aSub[iCmd].eCmd ){
  64
  65     case BACKUP_FINISH: {
  66       const char *zCmdName;
  67       Tcl_CmdInfo cmdInfo;
  68       zCmdName = Tcl_GetString(objv[0]);
  69       Tcl_GetCommandInfo(interp, zCmdName, &cmdInfo);
  70       cmdInfo.deleteProc = 0;
  71       Tcl_SetCommandInfo(interp, zCmdName, &cmdInfo);
  72       Tcl_DeleteCommand(interp, zCmdName);
  73
  74       rc = sqlite3_backup_finish(p);
  75       Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_STATIC);
  76       break;
  77     }
  78
  79     case BACKUP_STEP: {
  80       int nPage;
  81       if( TCL_OK!=Tcl_GetIntFromObj(interp, objv[2], &nPage) ){
  82         return TCL_ERROR;
  83       }
  84       rc = sqlite3_backup_step(p, nPage);
  85       Tcl_SetResult(interp, (char *)sqlite3ErrName(rc), TCL_STATIC);
  86       break;
  87     }
  88
  89     case BACKUP_REMAINING:
  90       Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_backup_remaining(p)));
  91       break;
  92
  93     case BACKUP_PAGECOUNT:
  94       Tcl_SetObjResult(interp, Tcl_NewIntObj(sqlite3_backup_pagecount(p)));
  95       break;
  96   }
  97
  98   return TCL_OK;
  99 }
 100
 101 static void SQLITE_TCLAPI backupTestFinish(ClientData clientData){
 102   sqlite3_backup *pBackup = (sqlite3_backup *)clientData;
 103   sqlite3_backup_finish(pBackup);
 104 }
 105
 106 /*
 107 **     sqlite3_backup CMDNAME DESTHANDLE DESTNAME SRCHANDLE SRCNAME
 108 **
 109 */
 110 static int SQLITE_TCLAPI backupTestInit(
 111   ClientData clientData,
 112   Tcl_Interp *interp,
 113   int objc,
 114   Tcl_Obj *const*objv
 115 ){
 116   sqlite3_backup *pBackup;
 117   sqlite3 *pDestDb;
 118   sqlite3 *pSrcDb;
 119   const char *zDestName;
 120   const char *zSrcName;
 121   const char *zCmd;
 122
 123   if( objc!=6 ){
 124     Tcl_WrongNumArgs(
 125       interp, 1, objv, "CMDNAME DESTHANDLE DESTNAME SRCHANDLE SRCNAME"
 126     );
 127     return TCL_ERROR;
 128   }
 129
 130   zCmd = Tcl_GetString(objv[1]);
 131   getDbPointer(interp, Tcl_GetString(objv[2]), &pDestDb);
 132   zDestName = Tcl_GetString(objv[3]);
 133   getDbPointer(interp, Tcl_GetString(objv[4]), &pSrcDb);
 134   zSrcName = Tcl_GetString(objv[5]);
 135
 136   pBackup = sqlite3_backup_init(pDestDb, zDestName, pSrcDb, zSrcName);
 137   if( !pBackup ){
 138     Tcl_AppendResult(interp, "sqlite3_backup_init() failed", 0);
 139     return TCL_ERROR;
 140   }
 141
 142   Tcl_CreateObjCommand(interp, zCmd, backupTestCmd, pBackup, backupTestFinish);
 143   Tcl_SetObjResult(interp, objv[1]);
 144   return TCL_OK;
 145 }
 146
 147 int Sqlitetestbackup_Init(Tcl_Interp *interp){
 148   Tcl_CreateObjCommand(interp, "sqlite3_backup", backupTestInit, 0, 0);
 149   return TCL_OK;
 150 }