Ensure the database attached as part of VACUUM can be detached successfully after a malloc() failure. (CVS 2918)

FossilOrigin-Name: 8c26893c65574b0667bb84bde3ca49751079cc8d

4 files changed, 18 insertions, 30 deletions

diff --git a/manifest b/manifest
index 59399a039..73f57727b 100644
--- a/manifest
+++ b/manifest
@@ -1,5 +1,5 @@
-C Fix\sbugs\scaused\sby\sassuming\sthat\sshared-schemas\sare\sinitialized.\s(CVS\s2917)
-D 2006-01-11T14:09:31
+C Ensure\sthe\sdatabase\sattached\sas\spart\sof\sVACUUM\scan\sbe\sdetached\ssuccessfully\safter\sa\smalloc()\sfailure.\s(CVS\s2918)
+D 2006-01-11T16:10:20
 F Makefile.in ab3ffd8d469cef4477257169b82810030a6bb967
 F Makefile.linux-gcc aee18d8a05546dcf1888bd4547e442008a49a092
 F README 9c4e2d6706bdcc3efdd773ce752a8cdab4f90028
@@ -85,8 +85,8 @@ F src/tokenize.c 196486012c871cdcad6cc84a820cc988603f1b9d
 F src/trigger.c 883b5f3b97137fbe417e3337c3fa20ac8e9c1ae5
 F src/update.c cd8ad5bb1a29f2056347481308fca4a59f2f4764
 F src/utf.c b7bffac4260177ae7f83c01d025fe0f5ed70ce71
-F src/util.c 5d5792d4a4dda20d70fdfb973ed8a5ed71fea98c
-F src/vacuum.c f5a068096b22fad438bf1f1cf69ccb7f9e8cc7fb
+F src/util.c 3616a0a1f1b89cfb376240c75434431b8028681e
+F src/vacuum.c cd56995ecea281b3ac306ef88128ebc8a2117f84
 F src/vdbe.c aa4360d28c4476e435c84e92ad3fbd65a8944156
 F src/vdbe.h 8729a4ee16ff9aeab2af9667df3cf300ff978e13
 F src/vdbeInt.h 5451cf71f229e366ac543607c0a17f36e5737ea9
@@ -340,7 +340,7 @@ F www/tclsqlite.tcl bb0d1357328a42b1993d78573e587c6dcbc964b9
 F www/vdbe.tcl 87a31ace769f20d3627a64fa1fade7fed47b90d0
 F www/version3.tcl a99cf5f6d8bd4d5537584a2b342f0fb9fa601d8b
 F www/whentouse.tcl 97e2b5cd296f7d8057e11f44427dea8a4c2db513
-P 1b368c7c5ca7974e0975dc4e3c931680c9e8df1f
-R 1449ef4e2e811b71b71db6703a8a24ab
+P 3970eb875d1830d35b3a70a7583a8ab6b238cad6
+R 5d634fd31ace904b5b4a4cd1fc441144
 U danielk1977
-Z 8596c20274e6f745d3adfacf59742397
+Z 11d0cd52f201ef748e551ed33dbe55f7
diff --git a/manifest.uuid b/manifest.uuid
index 8356830bf..9ed2e2044 100644
--- a/manifest.uuid
+++ b/manifest.uuid
@@ -1 +1 @@
-3970eb875d1830d35b3a70a7583a8ab6b238cad6
\ No newline at end of file
+8c26893c65574b0667bb84bde3ca49751079cc8d
\ No newline at end of file
diff --git a/src/util.c b/src/util.c
index 1a8287b07..4064dadc9 100644
--- a/src/util.c
+++ b/src/util.c
@@ -14,7 +14,7 @@
 ** This file contains functions for allocating memory, comparing
 ** strings, and stuff like that.
 **
-** $Id: util.c,v 1.166 2006/01/10 15:18:28 drh Exp $
+** $Id: util.c,v 1.167 2006/01/11 16:10:20 danielk1977 Exp $
 */
 #include "sqliteInt.h"
 #include "os.h"
@@ -438,6 +438,7 @@ static void * OSMALLOC(int n){
   ThreadData *pTsd = sqlite3ThreadData();
   pTsd->nMaxAlloc = MAX(pTsd->nMaxAlloc, pTsd->nAlloc);
 #endif
+  assert( sqlite3ThreadData()->mallocAllowed );
   if( !failMalloc() ){
     u32 *p;
     p = (u32 *)sqlite3OsMalloc(n + TESTALLOC_OVERHEAD);
@@ -479,6 +480,7 @@ static void * OSREALLOC(void *pRealloc, int n){
   ThreadData *pTsd = sqlite3ThreadData();
   pTsd->nMaxAlloc = MAX(pTsd->nMaxAlloc, pTsd->nAlloc);
 #endif
+  assert( sqlite3ThreadData()->mallocAllowed );
   if( !failMalloc() ){
     u32 *p = (u32 *)getOsPointer(pRealloc);
     checkGuards(p);
diff --git a/src/vacuum.c b/src/vacuum.c
index 2d7f7652c..891bc9fcc 100644
--- a/src/vacuum.c
+++ b/src/vacuum.c
@@ -14,7 +14,7 @@
 ** Most of the code in this file may be omitted by defining the
 ** SQLITE_OMIT_VACUUM macro.
 **
-** $Id: vacuum.c,v 1.55 2006/01/09 06:29:49 danielk1977 Exp $
+** $Id: vacuum.c,v 1.56 2006/01/11 16:10:20 danielk1977 Exp $
 */
 #include "sqliteInt.h"
 #include "vdbeInt.h"
@@ -103,7 +103,7 @@ int sqlite3RunVacuum(char **pzErrMsg, sqlite3 *db){
   char *zSql = 0;
   int rc2;  
   int saved_flags;       /* Saved value of the db->flags */
-  sqlite3_stmt *pDetach = 0;
+  Db *pDb = 0;           /* Database to detach at end of vacuum */
 
   /* Save the current value of the write-schema flag before setting it. */
   saved_flags = db->flags;
@@ -150,15 +150,6 @@ int sqlite3RunVacuum(char **pzErrMsg, sqlite3 *db){
     randomName((unsigned char*)&zTemp[nFilename+1]);
   } while( sqlite3OsFileExists(zTemp) );
 
-  /* Before we even attach it, compile a DETACH statement for vacuum_db. This
-  ** way, if malloc() fails we can detach the database without needing to
-  ** dynamically allocate memory.
-  */ 
-  rc = sqlite3_prepare(db, "DETACH vacuum_db", -1, &pDetach, 0);
-  if( rc!=SQLITE_OK ){
-    goto end_of_vacuum;
-  }
-
   /* Attach the temporary database as 'vacuum_db'. The synchronous pragma
   ** can be set to 'off' for this file, as it is not recovered if a crash
   ** occurs anyway. The integrity of the database is maintained by a
@@ -176,6 +167,7 @@ int sqlite3RunVacuum(char **pzErrMsg, sqlite3 *db){
   sqliteFree(zSql);
   zSql = 0;
   if( rc!=SQLITE_OK ) goto end_of_vacuum;
+  pDb = &db->aDb[db->nDb-1];
   assert( strcmp(db->aDb[db->nDb-1].zName,"vacuum_db")==0 );
   pTemp = db->aDb[db->nDb-1].pBt;
   sqlite3BtreeSetPageSize(pTemp, sqlite3BtreeGetPageSize(pMain),
@@ -310,18 +302,12 @@ end_of_vacuum:
   */
   db->autoCommit = 1;
 
-  if( pDetach ){
-    int mf = sqlite3ThreadData()->mallocFailed;
-    sqlite3ThreadData()->mallocFailed = 0;
+  if( pDb ){
     sqlite3MallocDisallow();
-    ((Vdbe *)pDetach)->expired = 0;
-    sqlite3_step(pDetach);
-    rc2 = sqlite3_finalize(pDetach);
-    if( rc==SQLITE_OK ){
-      rc = rc2;
-    }
+    sqlite3BtreeClose(pDb->pBt);
     sqlite3MallocAllow();
-    sqlite3ThreadData()->mallocFailed = mf;
+    pDb->pBt = 0;
+    pDb->pSchema = 0;
   }
 
   /* If one of the execSql() calls above returned SQLITE_NOMEM, then the