diff options
author | Timothy Pearson <kb9vqf@pearsoncomputing.net> | 2011-11-06 15:56:40 -0600 |
---|---|---|
committer | Timothy Pearson <kb9vqf@pearsoncomputing.net> | 2011-11-06 15:56:40 -0600 |
commit | e16866e072f94410321d70daedbcb855ea878cac (patch) | |
tree | ee3f52eabde7da1a0e6ca845fb9c2813cf1558cf /tdecore/kvmallocator.cpp | |
parent | a58c20c1a7593631a1b50213c805507ebc16adaf (diff) | |
download | tdelibs-e16866e072f94410321d70daedbcb855ea878cac.tar.gz tdelibs-e16866e072f94410321d70daedbcb855ea878cac.zip |
Actually move the kde files that were renamed in the last commit
Diffstat (limited to 'tdecore/kvmallocator.cpp')
-rw-r--r-- | tdecore/kvmallocator.cpp | 274 |
1 files changed, 274 insertions, 0 deletions
diff --git a/tdecore/kvmallocator.cpp b/tdecore/kvmallocator.cpp new file mode 100644 index 000000000..996d55791 --- /dev/null +++ b/tdecore/kvmallocator.cpp @@ -0,0 +1,274 @@ +/* + This file is part of the KDE libraries + + Copyright (C) 1999 Waldo Bastian (bastian@kde.org) + + This library is free software; you can redistribute it and/or + modify it under the terms of the GNU Library General Public + License as published by the Free Software Foundation; either + version 2 of the License, or (at your option) any later version. + + This library 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 + Library General Public License for more details. + + You should have received a copy of the GNU Library General Public License + along with this library; see the file COPYING.LIB. If not, write to + the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + Boston, MA 02110-1301, USA. +*/ +//---------------------------------------------------------------------------- +// +// Virtual Memory Allocator + +// TODO: Add large file support. +// TODO: Error reporting. (e.g. disk-full) + +#include <unistd.h> +#include <sys/mman.h> + +#include <tqintdict.h> +#include <tqmap.h> + +#include <ktempfile.h> +#include <kdebug.h> + +#include "kvmallocator.h" + + +#define KVM_ALIGN 4095 + +struct KVMAllocator::Block +{ + off_t start; + size_t length; // Requested length + size_t size; // Actual size + void *mmap; +}; + + +class KVMAllocatorPrivate +{ +public: + KTempFile *tempfile; + off_t max_length; + TQMap<off_t, KVMAllocator::Block> used_blocks; + TQMap<off_t, KVMAllocator::Block> free_blocks; +}; + +/** + * Create a KVMAllocator + */ +KVMAllocator::KVMAllocator() +{ + d = new KVMAllocatorPrivate; + d->tempfile = 0; + d->max_length = 0; +} + +/** + * Destruct the KVMAllocator and release all memory. + */ +KVMAllocator::~KVMAllocator() +{ + delete d->tempfile; + delete d; +} + +/** + * Allocate a virtual memory block. + * @param _size Size in bytes of the memory block. + */ +KVMAllocator::Block * +KVMAllocator::allocate(size_t _size) +{ + if (!d->tempfile) + { + d->tempfile = new KTempFile(TQString::null, "vmdata"); + d->tempfile->unlink(); + } + // Search in free list + TQMap<off_t,KVMAllocator::Block>::iterator it; + it = d->free_blocks.begin(); + while (it != d->free_blocks.end()) + { + if (it.data().size > _size) + { + Block &free_block = it.data(); + Block block; + kdDebug(180)<<"VM alloc: using block from free list "<<(long)free_block.start<<" size ="<<(long)free_block.size<<" request = "<<_size<< endl; + block.start = free_block.start; + block.length = _size; + block.size = (_size + KVM_ALIGN) & ~KVM_ALIGN; + block.mmap = 0; + free_block.size -= block.size; + free_block.start += block.size; + if (!free_block.size) + d->free_blocks.remove(it); + it = d->used_blocks.replace(block.start, block); + return &(it.data()); + } + ++it; + } + + + // Create new block + Block block; + block.start = d->max_length; + block.length = _size; + block.size = (_size + KVM_ALIGN) & ~KVM_ALIGN; + block.mmap = 0; + kdDebug(180)<<"VM alloc: using new block "<<(long)block.start<<" size ="<<(long)block.size<<" request = "<<_size<< endl; + it = d->used_blocks.replace(block.start, block); + d->max_length += block.size; + return &(it.data()); +} + +/** + * Free a virtual memory block + */ +void +KVMAllocator::free(Block *block_p) +{ + Block block = *block_p; + if (block.mmap) + { + kdDebug(180)<<"VM free: Block "<<(long)block.start<<" is still mmapped!"<<endl; + return; + } + TQMap<off_t,KVMAllocator::Block>::iterator it; + it = d->used_blocks.find(block.start); + if (it == d->used_blocks.end()) + { + kdDebug(180)<<"VM free: Block "<<(long)block.start<<" is not allocated."<<endl; + return; + } + d->used_blocks.remove(it); + it = d->free_blocks.replace(block.start, block); + TQMap<off_t,KVMAllocator::Block>::iterator before = it; + --before; + if (before != d->free_blocks.end()) + { + Block &block_before = before.data(); + if ((block_before.start + off_t(block_before.size)) == block.start) + { + // Merge blocks. + kdDebug(180) << "VM merging: Block "<< (long)block_before.start<< + " with "<< (long)block.start<< " (before)" << endl; + block.size += block_before.size; + block.start = block_before.start; + it.data() = block; + d->free_blocks.remove(before); + } + } + + TQMap<off_t,KVMAllocator::Block>::iterator after = it; + ++after; + if (after != d->free_blocks.end()) + { + Block &block_after = after.data(); + if ((block.start + off_t(block.size)) == block_after.start) + { + // Merge blocks. + kdDebug(180) << "VM merging: Block "<< (long)block.start<< + " with "<< (long)block_after.start<< " (after)" << endl; + block.size += block_after.size; + it.data() = block; + d->free_blocks.remove(after); + } + } +} + +/** + * Copy data from a virtual memory block to normal memory + */ +void +KVMAllocator::copy(void *dest, Block *src, int _offset, size_t length) +{ + (void) copyBlock(dest, src, _offset, length); +} + +bool +KVMAllocator::copyBlock(void *dest, Block *src, int _offset, size_t length) +{ + //kdDebug(180)<<"VM read: seek "<<(long)src->start<<" +"<<_offset<<":"<<length<<endl; + lseek(d->tempfile->handle(), src->start+_offset, SEEK_SET); + if (length == 0) + length = src->length - _offset; + int to_go = length; + int done = 0; + char *buf = (char *) dest; + while(to_go > 0) + { + int n = read(d->tempfile->handle(), buf+done, to_go); + if (n <= 0) + { + if (n < 0) + return false; // Error + else + return true; // End of data + } + done += n; + to_go -= n; + } + // Done. + return true; +} + +/** + * Copy data from normal memory to a virtual memory block + */ +void +KVMAllocator::copy(Block *dest, void *src, int _offset, size_t length) +{ + (void) copyBlock(dest, src, _offset, length); +} + +bool +KVMAllocator::copyBlock(Block *dest, void *src, int _offset, size_t length) +{ + //kdDebug(180)<<"VM write: seek "<<(long)dest->start<<" +"<<_offset<< ":" << length << endl; + lseek(d->tempfile->handle(), dest->start+_offset, SEEK_SET); + if (length == 0) + length = dest->length - _offset; + int to_go = length; + int done = 0; + char *buf = (char *) src; + while(to_go > 0) + { + int n = write(d->tempfile->handle(), buf+done, to_go); + if (n <= 0) return false; // Error + done += n; + to_go -= n; + } + // Done. + return true; +} + +/** + * Map a virtual memory block in memory + */ +void * +KVMAllocator::map(Block *block) +{ + if (block->mmap) + return block->mmap; + + void *result = mmap(0, block->length, PROT_READ| PROT_WRITE, + MAP_SHARED, d->tempfile->handle(), block->start); + block->mmap = result; + return block->mmap; +} + +/** + * Unmap a virtual memory block + */ +void +KVMAllocator::unmap(Block *block) +{ + // The following cast is necassery for Solaris. + // (touch it and die). --Waba + munmap((char *)block->mmap, block->length); + block->mmap = 0; +} |