mirror of
https://github.com/ksherlock/profuse.git
synced 2026-03-10 20:16:35 +00:00
git-svn-id: https://profuse.googlecode.com/svn/branches/v2@68 aa027e90-d47c-11dd-86d7-074df07e0730
This commit is contained in:
ksherlock
307
MappedFile.cpp
Normal file
307
MappedFile.cpp
Normal file
@@ -0,0 +1,307 @@
|
||||
#include "MappedFile.h"
|
||||
#include "Exception.h"
|
||||
#include <cerrno>
|
||||
#include <cstdlib>
|
||||
#include <cstring>
|
||||
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <sys/mman.h>
|
||||
|
||||
#include "auto.h"
|
||||
|
||||
using namespace ProFUSE;
|
||||
|
||||
|
||||
|
||||
MappedFile::MappedFile(const char *name, bool readOnly) :
|
||||
_fd(-1),
|
||||
_map(MAP_FAILED)
|
||||
{
|
||||
#undef __METHOD__
|
||||
#define __METHOD__ "MappedFile::MappedFile"
|
||||
|
||||
auto_fd fd(::open(name, readOnly ? O_RDONLY : O_RDWR));
|
||||
|
||||
if (fd < 0)
|
||||
{
|
||||
throw Exception(__METHOD__ ": Unable to open file.", errno);
|
||||
}
|
||||
//
|
||||
init(fd.release(), readOnly);
|
||||
}
|
||||
|
||||
MappedFile::MappedFile(int fd, bool readOnly) :
|
||||
_fd(-1),
|
||||
_map(MAP_FAILED)
|
||||
{
|
||||
init(fd, readOnly);
|
||||
}
|
||||
|
||||
// todo -- verify throw calls destructor.
|
||||
MappedFile::MappedFile(const char *name, size_t size) :
|
||||
_fd(-1),
|
||||
_map(MAP_FAILED)
|
||||
{
|
||||
#undef __METHOD__
|
||||
#define __METHOD__ "MappedFile::MappedFile"
|
||||
|
||||
_size = size;
|
||||
_readOnly = false;
|
||||
auto_fd fd(::open(name, O_CREAT | O_TRUNC | O_RDWR, 0644));
|
||||
|
||||
if (fd < 0)
|
||||
throw Exception(__METHOD__ ": Unable to create file.", errno);
|
||||
|
||||
// TODO -- is ftruncate portable?
|
||||
if (::ftruncate(fd, _size) < 0)
|
||||
{
|
||||
throw Exception(__METHOD__ ": Unable to truncate file.", errno);
|
||||
}
|
||||
|
||||
//_map = ::mmap(NULL, _size, PROT_READ | PROT_WRITE, MAP_FILE | MAP_SHARED, _fd, 0);
|
||||
|
||||
auto_map map(
|
||||
fd,
|
||||
_size,
|
||||
PROT_READ | PROT_WRITE,
|
||||
MAP_FILE | MAP_SHARED
|
||||
);
|
||||
|
||||
if (map == MAP_FAILED) throw Exception(__METHOD__ ": Unable to map file.", errno);
|
||||
|
||||
_fd = fd.release();
|
||||
_map = map.release();
|
||||
}
|
||||
|
||||
MappedFile::~MappedFile()
|
||||
{
|
||||
if (_map != MAP_FAILED) ::munmap(_map, _size);
|
||||
if (_fd >= 0) ::close(_fd);
|
||||
}
|
||||
|
||||
void MappedFile::init(int f, bool readOnly)
|
||||
{
|
||||
#undef __METHOD__
|
||||
#define __METHOD__ "MappedFile::init"
|
||||
|
||||
auto_fd fd(f);
|
||||
|
||||
_offset = 0;
|
||||
_blocks = 0;
|
||||
_size = 0;
|
||||
_readOnly = readOnly;
|
||||
_dosOrder = false;
|
||||
|
||||
_size = ::lseek(_fd, 0, SEEK_END);
|
||||
|
||||
if (_size < 0)
|
||||
throw Exception(__METHOD__ ": Unable to determine file size.", errno);
|
||||
|
||||
/*
|
||||
_map = ::mmap(NULL, _size, readOnly ? PROT_READ : PROT_READ | PROT_WRITE,
|
||||
MAP_FILE | MAP_SHARED, fd, 0);
|
||||
*/
|
||||
auto_map map(
|
||||
fd,
|
||||
_size,
|
||||
readOnly ? PROT_READ : PROT_READ | PROT_WRITE,
|
||||
readOnly ? MAP_FILE : MAP_FILE | MAP_SHARED
|
||||
);
|
||||
|
||||
if (map == MAP_FAILED) throw Exception(__METHOD__ ": Unable to map file.", errno);
|
||||
|
||||
_fd = fd.release();
|
||||
_map = map.release();
|
||||
}
|
||||
|
||||
|
||||
const unsigned MappedFile::DOSMap[] = {
|
||||
0x00, 0x0e, 0x0d, 0x0c,
|
||||
0x0b, 0x0a, 0x09, 0x08,
|
||||
0x07, 0x06, 0x05, 0x04,
|
||||
0x03, 0x02, 0x01, 0x0f
|
||||
};
|
||||
|
||||
|
||||
void MappedFile::readBlock(unsigned block, void *bp)
|
||||
{
|
||||
#undef __METHOD__
|
||||
#define __METHOD__ "MappedFile::readBlock"
|
||||
|
||||
|
||||
if (block >= _blocks) throw Exception(__METHOD__ ": Invalid block number.");
|
||||
if (bp == 0) throw Exception(__METHOD__ ": Invalid address.");
|
||||
|
||||
|
||||
if (_dosOrder)
|
||||
{
|
||||
unsigned track = (block & ~0x07) << 9;
|
||||
unsigned sector = (block & 0x07) << 1;
|
||||
|
||||
for (unsigned i = 0; i < 2; ++i)
|
||||
{
|
||||
size_t address = track | (DOSMap[sector+i] << 8);
|
||||
|
||||
std::memcpy(bp, (uint8_t *)_map + _offset + address, 256);
|
||||
|
||||
bp = (uint8_t *)bp + 256;
|
||||
}
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
size_t address = block * 512;
|
||||
std::memcpy(bp, (uint8_t *)_map + _offset + address, 512);
|
||||
}
|
||||
}
|
||||
|
||||
void MappedFile::writeBlock(unsigned block, const void *bp)
|
||||
{
|
||||
#undef __METHOD__
|
||||
#define __METHOD__ "MappedFile::writeBlock"
|
||||
|
||||
if (block > _blocks) throw Exception(__METHOD__ ": Invalid block number.");
|
||||
|
||||
if ( (_readOnly) || (_map == MAP_FAILED))
|
||||
throw Exception(__METHOD__ ": File is readonly.");
|
||||
|
||||
if (_dosOrder)
|
||||
{
|
||||
unsigned track = (block & ~0x07) << 9;
|
||||
unsigned sector = (block & 0x07) << 1;
|
||||
|
||||
for (unsigned i = 0; i < 2; ++i)
|
||||
{
|
||||
size_t address = track | (DOSMap[sector+i] << 8);
|
||||
|
||||
std::memcpy((uint8_t *)_map + _offset + address, bp, 256);
|
||||
bp = (uint8_t *)bp + 256;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
size_t address = block * 512;
|
||||
std::memcpy((uint8_t *)_map + _offset + address , bp, 512);
|
||||
}
|
||||
}
|
||||
|
||||
void MappedFile::sync()
|
||||
{
|
||||
#undef __METHOD__
|
||||
#define __METHOD__ "MappedFile::sync"
|
||||
|
||||
if ( (_readOnly) || (_map == MAP_FAILED))
|
||||
return;
|
||||
|
||||
if (::msync(_map, _size, MS_SYNC) < 0)
|
||||
throw Exception(__METHOD__ ": msync error.", errno);
|
||||
}
|
||||
|
||||
void MappedFile::reset()
|
||||
{
|
||||
_offset = 0;
|
||||
_blocks = 0;
|
||||
_dosOrder = false;
|
||||
}
|
||||
|
||||
/*
|
||||
uint8_t MappedFile::read8(size_t location) const
|
||||
{
|
||||
// check for size?
|
||||
uint8_t *map = (uint8_t *)_map;
|
||||
|
||||
return map[location];
|
||||
}
|
||||
*/
|
||||
|
||||
uint16_t MappedFile::read16(size_t location, int byteOrder) const
|
||||
{
|
||||
// check for size?
|
||||
uint8_t *map = (uint8_t *)_map;
|
||||
|
||||
switch(byteOrder)
|
||||
{
|
||||
case LittleEndian:
|
||||
return (map[location + 1] << 8)
|
||||
| (map[location]);
|
||||
case BigEndian:
|
||||
return (map[location] << 8)
|
||||
| (map[location+1]);
|
||||
default:
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
uint32_t MappedFile::read32(size_t location, int byteOrder) const
|
||||
{
|
||||
// check for size?
|
||||
uint8_t *map = (uint8_t *)_map;
|
||||
|
||||
switch(byteOrder)
|
||||
{
|
||||
case LittleEndian:
|
||||
return (map[location+3] << 24)
|
||||
| (map[location+2] << 16)
|
||||
| (map[location+1] << 8)
|
||||
| (map[location])
|
||||
;
|
||||
case BigEndian:
|
||||
return (map[location] << 24)
|
||||
| (map[location+1] << 16)
|
||||
| (map[location+2] << 8)
|
||||
| (map[location+3])
|
||||
;
|
||||
|
||||
default:
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
void MappedFile::write8(size_t location, uint8_t data)
|
||||
{
|
||||
uint8_t *map = (uint8_t *)_map;
|
||||
|
||||
map[location] = data;
|
||||
}
|
||||
*/
|
||||
|
||||
void MappedFile::write16(size_t location, uint16_t data, int byteOrder)
|
||||
{
|
||||
uint8_t *map = (uint8_t *)_map;
|
||||
|
||||
switch(byteOrder)
|
||||
{
|
||||
case LittleEndian:
|
||||
map[location] = data & 0xff;
|
||||
map[location+1] = (data >> 8) & 0xff;
|
||||
break;
|
||||
case BigEndian:
|
||||
map[location] = (data >> 8) & 0xff;
|
||||
map[location+1] = data & 0xff;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
void MappedFile::write32(size_t location, uint32_t data, int byteOrder)
|
||||
{
|
||||
uint8_t *map = (uint8_t *)_map;
|
||||
|
||||
switch(byteOrder)
|
||||
{
|
||||
case LittleEndian:
|
||||
map[location] = data & 0xff;
|
||||
map[location+1] = (data >> 8) & 0xff;
|
||||
map[location+2] = (data >> 16) & 0xff;
|
||||
map[location+3] = (data >> 24) & 0xff;
|
||||
break;
|
||||
case BigEndian:
|
||||
map[location] = (data >> 24) & 0xff;
|
||||
map[location+1] = (data >> 16) & 0xff;
|
||||
map[location+2] = (data >> 8) & 0xff;
|
||||
map[location+3] = data & 0xff;
|
||||
break;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user