mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-30 17:10:48 +01:00
100 lines
3.0 KiB
C++
100 lines
3.0 KiB
C++
// @file alignedbuilder.cpp
|
|
|
|
/**
|
|
* Copyright (C) 2009 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 <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "pch.h"
|
|
#include "alignedbuilder.h"
|
|
|
|
namespace mongo {
|
|
|
|
AlignedBuilder::AlignedBuilder(unsigned initSize) {
|
|
_len = 0;
|
|
_malloc(initSize);
|
|
uassert(13584, "out of memory AlignedBuilder", _p._allocationAddress);
|
|
}
|
|
|
|
void AlignedBuilder::mallocSelfAligned(unsigned sz) {
|
|
assert( sz == _p._size );
|
|
void *p = malloc(sz + Alignment - 1);
|
|
_p._allocationAddress = p;
|
|
size_t s = (size_t) p;
|
|
size_t sold = s;
|
|
s += Alignment - 1;
|
|
s = (s/Alignment)*Alignment;
|
|
DEV assert( s >= sold );
|
|
_p._data = (char *) s;
|
|
}
|
|
|
|
/* "slow"/infrequent portion of 'grow()' */
|
|
void NOINLINE_DECL AlignedBuilder::growReallocate(unsigned oldLen) {
|
|
unsigned a = _p._size;
|
|
assert( a );
|
|
while( 1 ) {
|
|
a *= 2;
|
|
wassert( a <= 128*1024*1024 );
|
|
assert( a <= 512*1024*1024 );
|
|
if( _len < a )
|
|
break;
|
|
}
|
|
_realloc(a, oldLen);
|
|
}
|
|
|
|
void AlignedBuilder::_malloc(unsigned sz) {
|
|
_p._size = sz;
|
|
#if defined(_WIN32)
|
|
void *p = VirtualAlloc(0, sz, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
|
|
_p._allocationAddress = p;
|
|
_p._data = (char *) p;
|
|
#elif defined(__linux__)
|
|
// in theory #ifdef _POSIX_VERSION should work, but it doesn't on OS X 10.4, and needs to be testeed on solaris.
|
|
// so for now, linux only for this.
|
|
void *p = 0;
|
|
int res = posix_memalign(&p, Alignment, sz);
|
|
massert(13524, "out of memory AlignedBuilder", res == 0);
|
|
_p._allocationAddress = p;
|
|
_p._data = (char *) p;
|
|
#else
|
|
mallocSelfAligned(sz);
|
|
assert( ((size_t) _p._data) % Alignment == 0 );
|
|
#endif
|
|
}
|
|
|
|
void AlignedBuilder::_realloc(unsigned newSize, unsigned oldLen) {
|
|
// posix_memalign alignment is not maintained on reallocs, so we can't use realloc().
|
|
AllocationInfo old = _p;
|
|
_malloc(newSize);
|
|
assert( oldLen <= _len );
|
|
memcpy(_p._data, old._data, oldLen);
|
|
_free(old._allocationAddress);
|
|
}
|
|
|
|
void AlignedBuilder::_free(void *p) {
|
|
#if defined(_WIN32)
|
|
VirtualFree(p, 0, MEM_RELEASE);
|
|
#else
|
|
free(p);
|
|
#endif
|
|
}
|
|
|
|
void AlignedBuilder::kill() {
|
|
_free(_p._allocationAddress);
|
|
_p._allocationAddress = 0;
|
|
_p._data = 0;
|
|
}
|
|
|
|
}
|