// pdfile.cpp /** * Copyright (C) 2008 10gen Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . */ /* todo: _ table scans must be sequential, not next/prev pointers _ coalesce deleted _ disallow system* manipulations from the database. */ #include "stdafx.h" #include "pdfile.h" #include "db.h" #include "../util/mmap.h" #include "../util/hashtab.h" #include "btree.h" #include #include #include "query.h" #include "repl.h" #include "dbhelpers.h" #include "namespace.h" #include "queryutil.h" namespace mongo { extern bool quota; extern int port; const char *dbpath = "/data/db/"; DataFileMgr theDataFileMgr; map databases; Database *database; const char *curNs = ""; int MAGIC = 0x1000; int curOp = -2; int callDepth = 0; extern int otherTraceLevel; void addNewNamespaceToCatalog(const char *ns, const BSONObj *options = 0); string getDbContext() { stringstream ss; if ( database ) { ss << database->name << ' '; if ( curNs ) ss << curNs << ' '; } ss<< "op:" << curOp << ' ' << callDepth; return ss.str(); } BSONObj::BSONObj(const Record *r) { init(r->data, false); /* _objdata = r->data; _objsize = *((int*) _objdata); if( _objsize > r->netLength() ) { out() << "About to assert fail _objsize <= r->netLength()" << endl; out() << " _objsize: " << _objsize << endl; out() << " netLength(): " << r->netLength() << endl; out() << " extentOfs: " << r->extentOfs << endl; out() << " nextOfs: " << r->nextOfs << endl; out() << " prevOfs: " << r->prevOfs << endl; assert( _objsize <= r->netLength() ); } iFree = false; */ } /*---------------------------------------------------------------------*/ int initialExtentSize(int len) { long long sz = len * 16; if ( len < 1000 ) sz = len * 64; if ( sz > 1000000000 ) sz = 1000000000; int z = ((int)sz) & 0xffffff00; assert( z > len ); DEV log() << "initialExtentSize(" << len << ") returns " << z << endl; return z; } bool _userCreateNS(const char *ns, const BSONObj& j, string& err) { if ( nsdetails(ns) ) { err = "collection already exists"; return false; } log(1) << "create collection " << ns << ' ' << j << '\n'; /* todo: do this only when we have allocated space successfully? or we could insert with a { ok: 0 } field and then go back and set to ok : 1 after we are done. */ if( strstr(ns, ".$freelist") == 0 ) addNewNamespaceToCatalog(ns, j.isEmpty() ? 0 : &j); long long size = initialExtentSize(128); BSONElement e = j.findElement("size"); if ( e.isNumber() ) { size = (long long) e.number(); size += 256; size &= 0xffffffffffffff00LL; } bool newCapped = false; int mx = 0; e = j.findElement("capped"); if ( e.type() == Bool && e.boolean() ) { newCapped = true; e = j.findElement("max"); if ( e.isNumber() ) { mx = (int) e.number(); } } // $nExtents just for debug/testing. We create '$nExtents' extents, // each of size 'size'. e = j.findElement( "$nExtents" ); int nExtents = int( e.number() ); if ( nExtents > 0 ) { assert( size <= 0x7fffffff ); for ( int i = 0; i < nExtents; ++i ) { database->suitableFile((int) size)->allocExtent( ns, (int) size, newCapped ); } } else while ( size > 0 ) { int max = MongoDataFile::maxSize() - MDFHeader::headerSize(); int desiredExtentSize = (int) (size > max ? max : size); Extent *e = database->suitableFile( desiredExtentSize )->allocExtent( ns, desiredExtentSize, newCapped ); size -= e->length; } NamespaceDetails *d = nsdetails(ns); assert(d); if ( mx > 0 ) d->max = mx; return true; } // { ..., capped: true, size: ..., max: ... } // returns true if successful bool userCreateNS(const char *ns, BSONObj j, string& err, bool logForReplication) { bool ok = _userCreateNS(ns, j, err); if ( logForReplication && ok ) logOp("c", ns, j); return ok; } /*---------------------------------------------------------------------*/ int MongoDataFile::maxSize() { if ( sizeof( int* ) == 4 ) return 512 * 1024 * 1024; else return 0x7ff00000; } int MongoDataFile::defaultSize( const char *filename ) const { int size; if ( fileNo <= 4 ) size = (64*1024*1024) << fileNo; else size = 0x7ff00000; if ( strstr(filename, "_hudsonSmall") ) { int mult = 1; if ( fileNo > 1 && fileNo < 1000 ) mult = fileNo; size = 1024 * 512 * mult; log() << "Warning : using small files for _hudsonSmall" << endl; } return size; } void MongoDataFile::open( const char *filename, int minSize ) { { /* check quotas very simple temporary implementation - we will in future look up the quota from the grid database */ if ( quota && fileNo > 8 && !boost::filesystem::exists(filename) ) { /* todo: if we were adding / changing keys in an index did we do some work previously that needs cleaning up? Possible. We should check code like that and have it catch the exception and do something reasonable. */ string s = "db disk space quota exceeded "; if ( database ) s += database->name; uasserted(s); } } int size = defaultSize( filename ); while ( size < minSize ) { if ( size < maxSize() / 2 ) size *= 2; else { size = maxSize(); break; } } if ( size > maxSize() ) size = maxSize(); assert( ( size >= 64*1024*1024 ) || ( strstr( filename, "_hudsonSmall" ) ) ); assert( size % 4096 == 0 ); header = (MDFHeader *) mmf.map(filename, size); if( sizeof(char *) == 4 ) uassert("can't map file memory - mongo requires 64 bit build for larger datasets", header); else uassert("can't map file memory", header); // If opening an existing file, this is a no-op. header->init(fileNo, size); } void addNewExtentToNamespace(const char *ns, Extent *e, DiskLoc eloc, DiskLoc emptyLoc, bool capped) { DiskLoc oldExtentLoc; NamespaceIndex *ni = nsindex(ns); NamespaceDetails *details = ni->details(ns); if ( details ) { assert( !details->lastExtent.isNull() ); assert( !details->firstExtent.isNull() ); e->xprev = details->lastExtent; details->lastExtent.ext()->xnext = eloc; assert( !eloc.isNull() ); details->lastExtent = eloc; } else { ni->add(ns, eloc, capped); details = ni->details(ns); } details->lastExtentSize = e->length; DEBUGGING out() << "temp: newextent adddelrec " << ns << endl; details->addDeletedRec(emptyLoc.drec(), emptyLoc); } Extent* MongoDataFile::createExtent(const char *ns, int approxSize, bool newCapped, int loops) { massert( "bad new extent size", approxSize >= 0 && approxSize <= 0x7ff00000 ); massert( "header==0 on new extent: 32 bit mmap space exceeded?", header ); // null if file open failed int ExtentSize = approxSize <= header->unusedLength ? approxSize : header->unusedLength; DiskLoc loc; if ( ExtentSize <= 0 ) { /* not there could be a lot of looping here is db just started and no files are open yet. we might want to do something about that. */ if ( loops > 8 ) { assert( loops < 10000 ); out() << "warning: loops=" << loops << " fileno:" << fileNo << ' ' << ns << '\n'; } log() << "newExtent: " << ns << " file " << fileNo << " full, adding a new file\n"; return database->addAFile()->createExtent(ns, approxSize, newCapped, loops+1); } int offset = header->unused.getOfs(); header->unused.setOfs( fileNo, offset + ExtentSize ); header->unusedLength -= ExtentSize; loc.setOfs(fileNo, offset); Extent *e = _getExtent(loc); DiskLoc emptyLoc = e->init(ns, ExtentSize, fileNo, offset); addNewExtentToNamespace(ns, e, loc, emptyLoc, newCapped); DEV log() << "new extent " << ns << " size: 0x" << hex << ExtentSize << " loc: 0x" << hex << offset << " emptyLoc:" << hex << emptyLoc.getOfs() << dec << endl; return e; } Extent* MongoDataFile::allocExtent(const char *ns, int approxSize, bool capped) { string s = database->name + ".$freelist"; NamespaceDetails *f = nsdetails(s.c_str()); if( f ) { int low, high; if( capped ) { // be strict about the size low = approxSize; if( low > 2048 ) low -= 256; high = (int) (approxSize * 1.05) + 256; } else { low = (int) (approxSize * 0.8); high = (int) (approxSize * 1.4); } if( high < 0 ) high = approxSize; int n = 0; Extent *best = 0; int bestDiff = 0x7fffffff; { DiskLoc L = f->firstExtent; while( !L.isNull() ) { Extent * e = L.ext(); if( e->length >= low && e->length <= high ) { int diff = abs(e->length - approxSize); if( diff < bestDiff ) { bestDiff = diff; best = e; if( diff == 0 ) break; } } L = e->xnext; ++n; } } OCCASIONALLY if( n > 512 ) log() << "warning: newExtent " << n << " scanned\n"; if( best ) { Extent *e = best; // remove from the free list if( !e->xprev.isNull() ) e->xprev.ext()->xnext = e->xnext; if( !e->xnext.isNull() ) e->xnext.ext()->xprev = e->xprev; if( f->firstExtent == e->myLoc ) f->firstExtent = e->xnext; if( f->lastExtent == e->myLoc ) f->lastExtent = e->xprev; // use it OCCASIONALLY if( n > 512 ) log() << "warning: newExtent " << n << " scanned\n"; DiskLoc emptyLoc = e->reuse(ns); addNewExtentToNamespace(ns, e, e->myLoc, emptyLoc, capped); return e; } } return createExtent(ns, approxSize, capped); } /*---------------------------------------------------------------------*/ DiskLoc Extent::reuse(const char *nsname) { log(3) << "reset extent was:" << ns.buf << " now:" << nsname << '\n'; massert( "Extent::reset bad magic value", magic == 0x41424344 ); xnext.Null(); xprev.Null(); ns = nsname; firstRecord.Null(); lastRecord.Null(); DiskLoc emptyLoc = myLoc; emptyLoc.inc( (extentData-(char*)this) ); int delRecLength = length - (extentData - (char *) this); DeletedRecord *empty1 = (DeletedRecord *) extentData; DeletedRecord *empty = (DeletedRecord *) getRecord(emptyLoc); assert( empty == empty1 ); memset(empty, delRecLength, 1); empty->lengthWithHeaders = delRecLength; empty->extentOfs = myLoc.getOfs(); empty->nextDeleted.Null(); return emptyLoc; } /* assumes already zeroed -- insufficient for block 'reuse' perhaps */ DiskLoc Extent::init(const char *nsname, int _length, int _fileNo, int _offset) { magic = 0x41424344; myLoc.setOfs(_fileNo, _offset); xnext.Null(); xprev.Null(); ns = nsname; length = _length; firstRecord.Null(); lastRecord.Null(); DiskLoc emptyLoc = myLoc; emptyLoc.inc( (extentData-(char*)this) ); DeletedRecord *empty1 = (DeletedRecord *) extentData; DeletedRecord *empty = (DeletedRecord *) getRecord(emptyLoc); assert( empty == empty1 ); empty->lengthWithHeaders = _length - (extentData - (char *) this); empty->extentOfs = myLoc.getOfs(); return emptyLoc; } /* Record* Extent::newRecord(int len) { if( firstEmptyRegion.isNull() ) return 0; assert(len > 0); int newRecSize = len + Record::HeaderSize; DiskLoc newRecordLoc = firstEmptyRegion; Record *r = getRecord(newRecordLoc); int left = r->netLength() - len; if( left < 0 ) { // firstEmptyRegion.Null(); return 0; } DiskLoc nextEmpty = r->next.getNextEmpty(firstEmptyRegion); r->lengthWithHeaders = newRecSize; r->next.markAsFirstOrLastInExtent(this); // we're now last in the extent if( !lastRecord.isNull() ) { assert(getRecord(lastRecord)->next.lastInExtent()); // it was the last one getRecord(lastRecord)->next.set(newRecordLoc); // until now r->prev.set(lastRecord); } else { r->prev.markAsFirstOrLastInExtent(this); // we are the first in the extent assert( firstRecord.isNull() ); firstRecord = newRecordLoc; } lastRecord = newRecordLoc; if( left < Record::HeaderSize + 32 ) { firstEmptyRegion.Null(); } else { firstEmptyRegion.inc(newRecSize); Record *empty = getRecord(firstEmptyRegion); empty->next.set(nextEmpty); // not for empty records, unless in-use records, next and prev can be null. empty->prev.Null(); empty->lengthWithHeaders = left; } return r; } */ /*---------------------------------------------------------------------*/ auto_ptr DataFileMgr::findAll(const char *ns, const DiskLoc &startLoc) { DiskLoc loc; bool found = nsindex(ns)->find(ns, loc); if ( !found ) { // out() << "info: findAll() namespace does not exist: " << ns << endl; return auto_ptr(new BasicCursor(DiskLoc())); } Extent *e = getExtent(loc); DEBUGGING { out() << "listing extents for " << ns << endl; DiskLoc tmp = loc; set extents; while ( 1 ) { Extent *f = getExtent(tmp); out() << "extent: " << tmp.toString() << endl; extents.insert(tmp); tmp = f->xnext; if ( tmp.isNull() ) break; f = f->getNextExtent(); } out() << endl; nsdetails(ns)->dumpDeleted(&extents); } if ( !nsdetails( ns )->capped ) { if ( !startLoc.isNull() ) return auto_ptr(new BasicCursor( startLoc )); while ( e->firstRecord.isNull() && !e->xnext.isNull() ) { /* todo: if extent is empty, free it for reuse elsewhere. that is a bit complicated have to clean up the freelists. */ RARELY out() << "info DFM::findAll(): extent " << loc.toString() << " was empty, skipping ahead " << ns << endl; // find a nonempty extent // it might be nice to free the whole extent here! but have to clean up free recs then. e = e->getNextExtent(); } return auto_ptr(new BasicCursor( e->firstRecord )); } else { return auto_ptr< Cursor >( new ForwardCappedCursor( nsdetails( ns ), startLoc ) ); } } /* get a table scan cursor, but can be forward or reverse direction. order.$natural - if set, > 0 means forward (asc), < 0 backward (desc). */ auto_ptr findTableScan(const char *ns, const BSONObj& order, const DiskLoc &startLoc) { BSONElement el = order.findElement("$natural"); // e.g., { $natural : -1 } if ( el.number() >= 0 ) return DataFileMgr::findAll(ns, startLoc); // "reverse natural order" NamespaceDetails *d = nsdetails(ns); if ( !d ) return auto_ptr(new BasicCursor(DiskLoc())); if ( !d->capped ) { if ( !startLoc.isNull() ) return auto_ptr(new ReverseCursor( startLoc )); Extent *e = d->lastExtent.ext(); while ( e->lastRecord.isNull() && !e->xprev.isNull() ) { OCCASIONALLY out() << " findTableScan: extent empty, skipping ahead" << endl; e = e->getPrevExtent(); } return auto_ptr(new ReverseCursor( e->lastRecord )); } else { return auto_ptr< Cursor >( new ReverseCappedCursor( d, startLoc ) ); } } void aboutToDelete(const DiskLoc& dl); /* drop a collection/namespace */ void dropNS(const string& nsToDrop) { NamespaceDetails* d = nsdetails(nsToDrop.c_str()); uassert( "ns not found", d ); uassert( "can't drop system ns", strstr(nsToDrop.c_str(), ".system.") == 0 ); { // remove from the system catalog BSONObj cond = BSON( "name" << nsToDrop ); // { name: "colltodropname" } string system_namespaces = database->name + ".system.namespaces"; /*int n = */ deleteObjects(system_namespaces.c_str(), cond, false, 0, true); // no check of return code as this ns won't exist for some of the new storage engines } // free extents if( !d->firstExtent.isNull() ) { string s = database->name + ".$freelist"; NamespaceDetails *freeExtents = nsdetails(s.c_str()); if( freeExtents == 0 ) { string err; _userCreateNS(s.c_str(), BSONObj(), err); freeExtents = nsdetails(s.c_str()); massert("can't create .$freelist", freeExtents); } if( freeExtents->firstExtent.isNull() ) { freeExtents->firstExtent = d->firstExtent; freeExtents->lastExtent = d->lastExtent; } else { DiskLoc a = freeExtents->firstExtent; assert( a.ext()->xprev.isNull() ); a.ext()->xprev = d->lastExtent; d->lastExtent.ext()->xnext = a; freeExtents->firstExtent = d->firstExtent; d->firstExtent.setInvalid(); d->lastExtent.setInvalid(); } } // remove from the catalog hashtable database->namespaceIndex.kill(nsToDrop.c_str()); } /* delete this index. does NOT clean up the system catalog (system.indexes or system.namespaces) -- only NamespaceIndex. */ void IndexDetails::kill() { string ns = indexNamespace(); // e.g. foo.coll.$ts_1 // clean up parent namespace index cache NamespaceDetailsTransient::get( parentNS().c_str() ).deletedIndex(); BSONObjBuilder b; b.append("name", indexName().c_str()); b.append("ns", parentNS().c_str()); BSONObj cond = b.done(); // e.g.: { name: "ts_1", ns: "foo.coll" } btreeStore->drop(ns.c_str()); head.setInvalid(); info.setInvalid(); // clean up in system.indexes. we do this last on purpose. note we have // to make the cond object before the drop() above though. string system_indexes = database->name + ".system.indexes"; int n = deleteObjects(system_indexes.c_str(), cond, false, 0, true); wassert( n == 1 ); } /* Pull out the relevant key objects from obj, so we can index them. Note that the set is multiple elements only when it's a "multikey" array. Keys will be left empty if key not found in the object. */ void IndexDetails::getKeysFromObject( const BSONObj& obj, BSONObjSetDefaultOrder& keys) const { BSONObj keyPattern = info.obj().getObjectField("key"); // e.g., keyPattern == { ts : 1 } if ( keyPattern.objsize() == 0 ) { out() << keyPattern.toString() << endl; out() << info.obj().toString() << endl; assert(false); } BSONObjBuilder b; const char *nameWithinArray; BSONObj key = obj.extractFieldsDotted(keyPattern, b, nameWithinArray); massert( "new key empty", !key.isEmpty() ); BSONObjIterator keyIter( key ); BSONElement arrayElt; int arrayPos = -1; for ( int i = 0; keyIter.more(); ++i ) { BSONElement e = keyIter.next(); if ( e.eoo() ) break; if ( e.type() == Array ) { uassert( "Index cannot be created on parallel arrays.", arrayPos == -1 ); arrayPos = i; arrayElt = e; } } if ( arrayPos == -1 ) { assert( strlen( nameWithinArray ) == 0 ); BSONObjBuilder b; BSONObjIterator keyIter( key ); while ( keyIter.more() ) { BSONElement f = keyIter.next(); if ( f.eoo() ) break; b.append( f ); } BSONObj o = b.obj(); assert( !o.isEmpty() ); keys.insert(o); return; } BSONObj arr = arrayElt.embeddedObject(); BSONObjIterator arrIter(arr); while ( arrIter.more() ) { BSONElement e = arrIter.next(); if ( e.eoo() ) break; if ( strlen( nameWithinArray ) != 0 ) { e = e.embeddedObject().getFieldDotted( nameWithinArray ); if ( e.eoo() ) continue; } BSONObjBuilder b; BSONObjIterator keyIter( key ); for ( int i = 0; keyIter.more(); ++i ) { BSONElement f = keyIter.next(); if ( f.eoo() ) break; if ( i != arrayPos ) b.append( f ); else b.appendAs( e, "" ); } BSONObj o = b.obj(); assert( !o.isEmpty() ); keys.insert(o); } } int nUnindexes = 0; void _unindexRecord(const char *ns, IndexDetails& id, BSONObj& obj, const DiskLoc& dl) { BSONObjSetDefaultOrder keys; id.getKeysFromObject(obj, keys); for ( BSONObjSetDefaultOrder::iterator i=keys.begin(); i != keys.end(); i++ ) { BSONObj j = *i; // out() << "UNINDEX: j:" << j.toString() << " head:" << id.head.toString() << dl.toString() << endl; if ( otherTraceLevel >= 5 ) { out() << "_unindexRecord() " << obj.toString(); out() << "\n unindex:" << j.toString() << endl; } nUnindexes++; bool ok = false; try { ok = id.head.btree()->unindex(id.head, id, j, dl); } catch (AssertionException&) { problem() << "Assertion failure: _unindex failed " << id.indexNamespace() << endl; out() << "Assertion failure: _unindex failed" << '\n'; out() << " obj:" << obj.toString() << '\n'; out() << " key:" << j.toString() << '\n'; out() << " dl:" << dl.toString() << endl; sayDbContext(); } if ( !ok ) { out() << "unindex failed (key too big?) " << id.indexNamespace() << '\n'; } } } /* unindex all keys in all indexes for this record. */ void unindexRecord(const char *ns, NamespaceDetails *d, Record *todelete, const DiskLoc& dl) { if ( d->nIndexes == 0 ) return; BSONObj obj(todelete); for ( int i = 0; i < d->nIndexes; i++ ) { _unindexRecord(ns, d->indexes[i], obj, dl); } } /* deletes a record, just the pdfile portion -- no index cleanup, no cursor cleanup, etc. caller must check if capped */ void DataFileMgr::_deleteRecord(NamespaceDetails *d, const char *ns, Record *todelete, const DiskLoc& dl) { /* remove ourself from the record next/prev chain */ { if ( todelete->prevOfs != DiskLoc::NullOfs ) todelete->getPrev(dl).rec()->nextOfs = todelete->nextOfs; if ( todelete->nextOfs != DiskLoc::NullOfs ) todelete->getNext(dl).rec()->prevOfs = todelete->prevOfs; } /* remove ourself from extent pointers */ { Extent *e = todelete->myExtent(dl); if ( e->firstRecord == dl ) { if ( todelete->nextOfs == DiskLoc::NullOfs ) e->firstRecord.Null(); else e->firstRecord.setOfs(dl.a(), todelete->nextOfs); } if ( e->lastRecord == dl ) { if ( todelete->prevOfs == DiskLoc::NullOfs ) e->lastRecord.Null(); else e->lastRecord.setOfs(dl.a(), todelete->prevOfs); } } /* add to the free list */ { d->nrecords--; d->datasize -= todelete->netLength(); /* temp: if in system.indexes, don't reuse, and zero out: we want to be careful until validated more, as IndexDetails has pointers to this disk location. so an incorrectly done remove would cause a lot of problems. */ if ( strstr(ns, ".system.indexes") ) { memset(todelete, 0, todelete->lengthWithHeaders); } else { DEV memset(todelete->data, 0, todelete->netLength()); // attempt to notice invalid reuse. d->addDeletedRec((DeletedRecord*)todelete, dl); } } } void DataFileMgr::deleteRecord(const char *ns, Record *todelete, const DiskLoc& dl, bool cappedOK) { dassert( todelete == dl.rec() ); NamespaceDetails* d = nsdetails(ns); if ( d->capped && !cappedOK ) { out() << "failing remove on a capped ns " << ns << endl; return; } /* check if any cursors point to us. if so, advance them. */ aboutToDelete(dl); unindexRecord(ns, d, todelete, dl); _deleteRecord(d, ns, todelete, dl); NamespaceDetailsTransient::get( ns ).registerWriteOp(); } void setDifference(BSONObjSetDefaultOrder &l, BSONObjSetDefaultOrder &r, vector &diff) { BSONObjSetDefaultOrder::iterator i = l.begin(); BSONObjSetDefaultOrder::iterator j = r.begin(); while ( 1 ) { if ( i == l.end() ) break; while ( j != r.end() && j->woCompare( *i ) < 0 ) j++; if ( j == r.end() || i->woCompare(*j) != 0 ) { const BSONObj *jo = &*i; diff.push_back( (BSONObj *) jo ); } i++; } } /** Note: as written so far, if the object shrinks a lot, we don't free up space. */ void DataFileMgr::update( const char *ns, Record *toupdate, const DiskLoc& dl, const char *buf, int len, stringstream& ss) { dassert( toupdate == dl.rec() ); NamespaceDetails *d = nsdetails(ns); BSONObj objOld(toupdate); BSONElement idOld; int addID = 0; { /* duplicate _id check... */ BSONObj objNew(buf); BSONElement idNew; objOld.getObjectID(idOld); if( objNew.getObjectID(idNew) ) { if( idOld == idNew ) ; else { /* check if idNew would be a duplicate. very unusual that an _id would change, so not worried about the bit of extra work here. */ if( d->findIdIndex() >= 0 ) { BSONObj result; BSONObjBuilder b; b.append(idNew); uassert("duplicate _id key on update", !Helpers::findOne(ns, b.done(), result)); } } } else { if ( !idOld.eoo() ) { addID = len; len += idOld.size(); } } } if ( toupdate->netLength() < len ) { // doesn't fit. must reallocate. if ( d && d->capped ) { ss << " failing a growing update on a capped ns " << ns << endl; return; } d->paddingTooSmall(); if ( database->profile ) ss << " moved "; deleteRecord(ns, toupdate, dl); insert(ns, buf, len, false, idOld); return; } NamespaceDetailsTransient::get( ns ).registerWriteOp(); d->paddingFits(); /* has any index keys changed? */ { NamespaceDetails *d = nsdetails(ns); if ( d->nIndexes ) { BSONObj newObj(buf); BSONObj oldObj = dl.obj(); for ( int i = 0; i < d->nIndexes; i++ ) { IndexDetails& idx = d->indexes[i]; BSONObj idxKey = idx.info.obj().getObjectField("key"); BSONObjSetDefaultOrder oldkeys; BSONObjSetDefaultOrder newkeys; idx.getKeysFromObject(oldObj, oldkeys); idx.getKeysFromObject(newObj, newkeys); vector removed; setDifference(oldkeys, newkeys, removed); string idxns = idx.indexNamespace(); for ( unsigned i = 0; i < removed.size(); i++ ) { try { idx.head.btree()->unindex(idx.head, idx, *removed[i], dl); } catch (AssertionException&) { ss << " exception update unindex "; problem() << " caught assertion update unindex " << idxns.c_str() << endl; } } vector added; setDifference(newkeys, oldkeys, added); assert( !dl.isNull() ); for ( unsigned i = 0; i < added.size(); i++ ) { try { idx.head.btree()->bt_insert( idx.head, dl, *added[i], idxKey, /*dupsAllowed*/true, idx); } catch (AssertionException&) { ss << " exception update index "; out() << " caught assertion update index " << idxns.c_str() << '\n'; problem() << " caught assertion update index " << idxns.c_str() << endl; } } if ( database->profile ) ss << '\n' << added.size() << " key updates "; } } } // update in place if ( addID ) { ((int&)*toupdate->data) = *((int*) buf) + idOld.size(); memcpy(toupdate->data+4, idOld.rawdata(), idOld.size()); memcpy(toupdate->data+4+idOld.size(), ((char *)buf)+4, addID-4); } else { memcpy(toupdate->data, buf, len); } } int followupExtentSize(int len, int lastExtentLen) { int x = initialExtentSize(len); int y = (int) (lastExtentLen < 4000000 ? lastExtentLen * 4.0 : lastExtentLen * 1.2); int sz = y > x ? y : x; sz = ((int)sz) & 0xffffff00; assert( sz > len ); return sz; } int deb=0; /* add keys to indexes for a new record */ inline void _indexRecord(IndexDetails& idx, BSONObj& obj, DiskLoc newRecordLoc, bool dupsAllowed) { BSONObjSetDefaultOrder keys; idx.getKeysFromObject(obj, keys); BSONObj order = idx.keyPattern(); for ( BSONObjSetDefaultOrder::iterator i=keys.begin(); i != keys.end(); i++ ) { assert( !newRecordLoc.isNull() ); try { idx.head.btree()->bt_insert(idx.head, newRecordLoc, (BSONObj&) *i, order, dupsAllowed, idx); } catch (AssertionException& ) { if( !dupsAllowed ) { // dup key exception, presumably. throw; } problem() << " caught assertion _indexRecord " << idx.indexNamespace() << endl; } } } /* note there are faster ways to build an index in bulk, that can be done eventually */ void addExistingToIndex(const char *ns, IndexDetails& idx) { bool dupsAllowed = !idx.isIdIndex(); Timer t; Nullstream& l = log(); l << "building new index on " << idx.keyPattern() << " for " << ns << "..."; l.flush(); int err = 0; int n = 0; auto_ptr c = theDataFileMgr.findAll(ns); while ( c->ok() ) { BSONObj js = c->current(); try { _indexRecord(idx, js, c->currLoc(),dupsAllowed); } catch( AssertionException& ) { err++; } c->advance(); n++; }; l << "done for " << n << " records"; if( err ) l << ' ' << err << " (dupkey) errors during indexing"; l << endl; if( err ) { // if duplicate _id's, report the problem stringstream ss; ss << err << " dupkey errors building index for " << ns; string s = ss.str(); uassert_nothrow(s.c_str()); } } /* add keys to indexes for a new record */ void indexRecord(NamespaceDetails *d, const void *buf, int len, DiskLoc newRecordLoc) { BSONObj obj((const char *)buf); /* we index _id first so that on a dup key error for it we don't have to roll back the other work. */ int id = d->findIdIndex(); if( id >= 0 ) _indexRecord(d->indexes[id], obj, newRecordLoc, /*dupsAllowed*/false); for ( int i = 0; i < d->nIndexes; i++ ) { if( i != id ) _indexRecord(d->indexes[i], obj, newRecordLoc, /*dupsAllowed*/true); } } extern BSONObj id_obj; void ensureHaveIdIndex(const char *ns) { NamespaceDetails *d = nsdetails(ns); if ( d == 0 || (d->flags & NamespaceDetails::Flag_HaveIdIndex) ) return; d->flags |= NamespaceDetails::Flag_HaveIdIndex; string system_indexes = database->name + ".system.indexes"; BSONObjBuilder b; b.append("name", "_id_"); b.append("ns", ns); b.append("key", id_obj); BSONObj o = b.done(); /* edge case: note the insert could fail if we have hit maxindexes already */ theDataFileMgr.insert(system_indexes.c_str(), o.objdata(), o.objsize()); } // should be { : , .keyp.. } bool validKeyPattern(BSONObj kp) { BSONObjIterator i(kp); while( i.more() ) { BSONElement e = i.next(); if( e.type() == Object || e.type() == Array ) return false; } return true; } #pragma pack(1) struct IDToInsert_ { char type; char _id[4]; OID oid; IDToInsert_() { type = (char) jstOID; strcpy(_id, "_id"); assert( sizeof(IDToInsert_) == 17 ); } } idToInsert_; struct IDToInsert : public BSONElement { IDToInsert() : BSONElement( ( char * )( &idToInsert_ ) ) {} } idToInsert; #pragma pack() DiskLoc DataFileMgr::insert(const char *ns, BSONObj &o) { DiskLoc loc = insert( ns, o.objdata(), o.objsize() ); if ( !loc.isNull() ) o = BSONObj( loc.rec() ); return loc; } DiskLoc DataFileMgr::insert(const char *ns, const void *obuf, int len, bool god, const BSONElement &writeId) { bool addIndex = false; const char *sys = strstr(ns, "system."); if ( sys ) { uassert("attempt to insert in reserved database name 'system'", sys != ns); if ( strstr(ns, ".system.") ) { // todo later: check for dba-type permissions here. if ( strstr(ns, ".system.indexes") ) addIndex = true; else if ( strstr(ns, ".system.users") ) ; else if ( !god ) { out() << "ERROR: attempt to insert in system namespace " << ns << endl; return DiskLoc(); } } else sys = 0; } NamespaceDetails *d = nsdetails(ns); if ( d == 0 ) { addNewNamespaceToCatalog(ns); /* todo: shouldn't be in the namespace catalog until after the allocations here work. also if this is an addIndex, those checks should happen before this! */ database->newestFile()->allocExtent(ns, initialExtentSize(len)); d = nsdetails(ns); } d->paddingFits(); NamespaceDetails *tableToIndex = 0; string tabletoidxns; if ( addIndex ) { BSONObj io((const char *) obuf); const char *name = io.getStringField("name"); // name of the index tabletoidxns = io.getStringField("ns"); // table it indexes if ( database->name != nsToClient(tabletoidxns.c_str()) ) { uassert("bad table to index name on add index attempt", false); return DiskLoc(); } BSONObj key = io.getObjectField("key"); if( !validKeyPattern(key) ) { string s = string("bad index key pattern ") + key.toString(); uassert(s.c_str(), false); } if ( *name == 0 || tabletoidxns.empty() || key.isEmpty() || key.objsize() > 2048 ) { out() << "user warning: bad add index attempt name:" << (name?name:"") << "\n ns:" << tabletoidxns << "\n ourns:" << ns; out() << "\n idxobj:" << io.toString() << endl; string s = "bad add index attempt " + tabletoidxns + " key:" + key.toString(); uasserted(s); } tableToIndex = nsdetails(tabletoidxns.c_str()); if ( tableToIndex == 0 ) { // try to create it string err; if ( !userCreateNS(tabletoidxns.c_str(), BSONObj(), err, false) ) { problem() << "ERROR: failed to create collection while adding its index. " << tabletoidxns << endl; return DiskLoc(); } tableToIndex = nsdetails(tabletoidxns.c_str()); log() << "info: creating collection " << tabletoidxns << " on add index\n"; assert( tableToIndex ); } if ( tableToIndex->nIndexes >= MaxIndexes ) { stringstream ss; ss << "add index fails, too many indexes for " << tabletoidxns << " key:" << key.toString(); string s = ss.str(); log() << s << '\n'; uasserted(s); } if ( tableToIndex->findIndexByName(name) >= 0 ) { //out() << "INFO: index:" << name << " already exists for:" << tabletoidxns << endl; return DiskLoc(); } //indexFullNS = tabletoidxns; //indexFullNS += ".$"; //indexFullNS += name; // database.table.$index -- note this doesn't contain jsobjs, it contains BtreeBuckets. } const BSONElement *newId = &writeId; int addID = 0; if( !god ) { /* Check if we have an _id field. If we don't, we'll add it. Note that btree buckets which we insert aren't BSONObj's, but in that case god==true. */ BSONObj io((const char *) obuf); if( !io.hasField("_id") && !addIndex && strstr(ns, ".local.") == 0 ) { addID = len; if ( writeId.eoo() ) { // Very likely we'll add this elt, so little harm in init'ing here. idToInsert_.oid.init(); newId = &idToInsert; } len += newId->size(); } BSONElementManipulator::lookForTimestamps( io ); } DiskLoc extentLoc; int lenWHdr = len + Record::HeaderSize; lenWHdr = (int) (lenWHdr * d->paddingFactor); if ( lenWHdr == 0 ) { // old datafiles, backward compatible here. assert( d->paddingFactor == 0 ); d->paddingFactor = 1.0; lenWHdr = len + Record::HeaderSize; } DiskLoc loc = d->alloc(ns, lenWHdr, extentLoc); if ( loc.isNull() ) { // out of space if ( d->capped == 0 ) { // size capped doesn't grow DEV log() << "allocating new extent for " << ns << " padding:" << d->paddingFactor << endl; database->newestFile()->allocExtent(ns, followupExtentSize(len, d->lastExtentSize)); loc = d->alloc(ns, lenWHdr, extentLoc); } if ( loc.isNull() ) { log() << "out of space in datafile " << ns << " capped:" << d->capped << endl; assert(d->capped); return DiskLoc(); } } Record *r = loc.rec(); assert( r->lengthWithHeaders >= lenWHdr ); if( addID ) { /* a little effort was made here to avoid a double copy when we add an ID */ ((int&)*r->data) = *((int*) obuf) + newId->size(); memcpy(r->data+4, newId->rawdata(), newId->size()); memcpy(r->data+4+newId->size(), ((char *)obuf)+4, addID-4); } else { memcpy(r->data, obuf, len); } Extent *e = r->myExtent(loc); if ( e->lastRecord.isNull() ) { e->firstRecord = e->lastRecord = loc; r->prevOfs = r->nextOfs = DiskLoc::NullOfs; } else { Record *oldlast = e->lastRecord.rec(); r->prevOfs = e->lastRecord.getOfs(); r->nextOfs = DiskLoc::NullOfs; oldlast->nextOfs = loc.getOfs(); e->lastRecord = loc; } d->nrecords++; d->datasize += r->netLength(); if ( !god ) NamespaceDetailsTransient::get( ns ).registerWriteOp(); if ( tableToIndex ) { IndexDetails& idxinfo = tableToIndex->indexes[tableToIndex->nIndexes]; idxinfo.info = loc; idxinfo.head = BtreeBucket::addHead(idxinfo); tableToIndex->addingIndex(tabletoidxns.c_str(), idxinfo); /* todo: index existing records here */ addExistingToIndex(tabletoidxns.c_str(), idxinfo); } /* add this record to our indexes */ if ( d->nIndexes ) { try { indexRecord(d, r->data/*buf*/, len, loc); } catch( AssertionException& e ) { // should be a dup key error on _id index if( tableToIndex || d->capped ) { string s = e.toString(); s += " : on addIndex/capped - collection and its index will not match"; uassert_nothrow(s.c_str()); log() << s << '\n'; } else { // normal case -- we can roll back _deleteRecord(d, ns, r, loc); throw; } } } // out() << " inserted at loc:" << hex << loc.getOfs() << " lenwhdr:" << hex << lenWHdr << dec << ' ' << ns << endl; return loc; } /* special version of insert for transaction logging -- streamlined a bit. assumes ns is capped and no indexes */ Record* DataFileMgr::fast_oplog_insert(NamespaceDetails *d, const char *ns, int len) { RARELY assert( d == nsdetails(ns) ); DiskLoc extentLoc; int lenWHdr = len + Record::HeaderSize; DiskLoc loc = d->alloc(ns, lenWHdr, extentLoc); if ( loc.isNull() ) { assert(false); return 0; } Record *r = loc.rec(); assert( r->lengthWithHeaders >= lenWHdr ); Extent *e = r->myExtent(loc); if ( e->lastRecord.isNull() ) { e->firstRecord = e->lastRecord = loc; r->prevOfs = r->nextOfs = DiskLoc::NullOfs; } else { Record *oldlast = e->lastRecord.rec(); r->prevOfs = e->lastRecord.getOfs(); r->nextOfs = DiskLoc::NullOfs; oldlast->nextOfs = loc.getOfs(); e->lastRecord = loc; } d->nrecords++; return r; } void DataFileMgr::init(const char *dir) { /* boost::filesystem::path path( dir ); path /= "temp.dat"; string pathString = path.string(); temp.open(pathString.c_str(), 64 * 1024 * 1024); */ } void pdfileInit() { // namespaceIndex.init(dbpath); theDataFileMgr.init(dbpath); } } // namespace mongo #include "clientcursor.h" namespace mongo { void dropDatabase(const char *ns) { // ns is of the form ".$cmd" char cl[256]; nsToClient(ns, cl); log(1) << "dropDatabase " << cl << endl; assert( database->name == cl ); closeClient( cl ); _deleteDataFiles(cl); } typedef boost::filesystem::path Path; // back up original database files to 'temp' dir void _renameForBackup( const char *database, const Path &reservedPath ) { class Renamer : public FileOp { public: Renamer( const Path &reservedPath ) : reservedPath_( reservedPath ) {} private: const boost::filesystem::path &reservedPath_; virtual bool apply( const Path &p ) { if ( !boost::filesystem::exists( p ) ) return false; boost::filesystem::rename( p, reservedPath_ / ( p.leaf() + ".bak" ) ); return true; } virtual const char * op() const { return "renaming"; } } renamer( reservedPath ); _applyOpToDataFiles( database, renamer ); } // move temp files to standard data dir void _replaceWithRecovered( const char *database, const char *reservedPathString ) { class : public FileOp { virtual bool apply( const Path &p ) { if ( !boost::filesystem::exists( p ) ) return false; boost::filesystem::rename( p, boost::filesystem::path(dbpath) / p.leaf() ); return true; } virtual const char * op() const { return "renaming"; } } renamer; _applyOpToDataFiles( database, renamer, reservedPathString ); } // generate a directory name for storing temp data files Path uniqueReservedPath( const char *prefix ) { Path dbPath = Path( dbpath ); Path reservedPath; int i = 0; bool exists = false; do { stringstream ss; ss << prefix << "_repairDatabase_" << i++; reservedPath = dbPath / ss.str(); BOOST_CHECK_EXCEPTION( exists = boost::filesystem::exists( reservedPath ) ); } while ( exists ); return reservedPath; } boost::intmax_t dbSize( const char *database ) { class SizeAccumulator : public FileOp { public: SizeAccumulator() : totalSize_( 0 ) {} boost::intmax_t size() const { return totalSize_; } private: virtual bool apply( const boost::filesystem::path &p ) { if ( !boost::filesystem::exists( p ) ) return false; totalSize_ += boost::filesystem::file_size( p ); return true; } virtual const char *op() const { return "checking size"; } boost::intmax_t totalSize_; }; SizeAccumulator sa; _applyOpToDataFiles( database, sa ); return sa.size(); } #if !defined(_WIN32) } // namespace mongo #include namespace mongo { #endif boost::intmax_t freeSpace() { #if !defined(_WIN32) struct statvfs info; assert( !statvfs( dbpath, &info ) ); return boost::intmax_t( info.f_bavail ) * info.f_frsize; #else return -1; #endif } bool repairDatabase( const char *ns, string &errmsg, bool preserveClonedFilesOnFailure, bool backupOriginalFiles ) { stringstream ss; ss << "localhost:" << port; string localhost = ss.str(); // ns is of the form ".$cmd" char dbName[256]; nsToClient(ns, dbName); problem() << "repairDatabase " << dbName << endl; assert( database->name == dbName ); boost::intmax_t totalSize = dbSize( dbName ); boost::intmax_t freeSize = freeSpace(); if ( freeSize > -1 && freeSize < totalSize ) { stringstream ss; ss << "Cannot repair database " << dbName << " having size: " << totalSize << " (bytes) because free disk space is: " << freeSize << " (bytes)"; errmsg = ss.str(); problem() << errmsg << endl; return false; } Path reservedPath = uniqueReservedPath( ( preserveClonedFilesOnFailure || backupOriginalFiles ) ? "backup" : "tmp" ); BOOST_CHECK_EXCEPTION( boost::filesystem::create_directory( reservedPath ) ); string reservedPathString = reservedPath.native_directory_string(); assert( setClient( dbName, reservedPathString.c_str() ) ); bool res = cloneFrom(localhost.c_str(), errmsg, dbName, /*logForReplication=*/false, /*slaveok*/false, /*replauth*/false); closeClient( dbName, reservedPathString.c_str() ); if ( !res ) { problem() << "clone failed for " << dbName << " with error: " << errmsg << endl; if ( !preserveClonedFilesOnFailure ) BOOST_CHECK_EXCEPTION( boost::filesystem::remove_all( reservedPath ) ); return false; } assert( !setClientTempNs( dbName ) ); closeClient( dbName ); if ( backupOriginalFiles ) _renameForBackup( dbName, reservedPath ); else _deleteDataFiles( dbName ); _replaceWithRecovered( dbName, reservedPathString.c_str() ); if ( !backupOriginalFiles ) BOOST_CHECK_EXCEPTION( boost::filesystem::remove_all( reservedPath ) ); return true; } } // namespace mongo