Merge branch 'development' into LTS_development

This commit is contained in:
Thulinma 2018-03-20 14:30:42 +01:00
commit 0385daabe8
7 changed files with 262 additions and 124 deletions

View file

@ -17,7 +17,8 @@
#if DEBUG > -1 #if DEBUG > -1
#include <stdio.h> #include <stdio.h>
#include <unistd.h> #include <inttypes.h>
#include <stdint.h>
#include "config.h" #include "config.h"
static const char * DBG_LVL_LIST[] = {"NONE", "FAIL", "ERROR", "WARN", "INFO", "MEDIUM", "HIGH", "VERYHIGH", "EXTREME", "INSANE", "DONTEVEN"}; static const char * DBG_LVL_LIST[] = {"NONE", "FAIL", "ERROR", "WARN", "INFO", "MEDIUM", "HIGH", "VERYHIGH", "EXTREME", "INSANE", "DONTEVEN"};

View file

@ -349,6 +349,18 @@ namespace IPC {
close(); close();
} }
/// Returns true if the open file still exists.
/// Not implemented under Windows.
bool sharedPage::exists(){
#ifdef SHM_ENABLED
struct stat sb;
if (fstat(handle, &sb)){return false;}
return (sb.st_nlink > 0);
#else
return true;
#endif
}
#ifdef SHM_ENABLED #ifdef SHM_ENABLED
///\brief Unmaps a shared page if allowed ///\brief Unmaps a shared page if allowed
void sharedPage::unmap() { void sharedPage::unmap() {

View file

@ -110,6 +110,7 @@ namespace IPC {
bool operator < (const sharedFile & rhs) const { bool operator < (const sharedFile & rhs) const {
return name < rhs.name; return name < rhs.name;
} }
bool exists(){return true;}
void close(); void close();
void unmap(); void unmap();
///\brief The fd handle of the opened shared file ///\brief The fd handle of the opened shared file
@ -144,6 +145,7 @@ namespace IPC {
} }
void unmap(); void unmap();
void close(); void close();
bool exists();
#if defined(__CYGWIN__) || defined(_WIN32) #if defined(__CYGWIN__) || defined(_WIN32)
///\brief The handle of the opened shared memory page ///\brief The handle of the opened shared memory page
HANDLE handle; HANDLE handle;

View file

@ -8,6 +8,7 @@
#include "procs.h" #include "procs.h"
#include <errno.h> // errno, ENOENT, EEXIST #include <errno.h> // errno, ENOENT, EEXIST
#include <iostream> #include <iostream>
#include <iomanip>
#include <stdio.h> #include <stdio.h>
#include <sys/stat.h> // stat #include <sys/stat.h> // stat
#if defined(_WIN32) #if defined(_WIN32)
@ -23,7 +24,8 @@
#define RAXHDR_DELETED *(uint64_t *)(p + 14) #define RAXHDR_DELETED *(uint64_t *)(p + 14)
#define RAXHDR_PRESENT *(uint32_t *)(p + 22) #define RAXHDR_PRESENT *(uint32_t *)(p + 22)
#define RAXHDR_OFFSET *(uint16_t *)(p + 26) #define RAXHDR_OFFSET *(uint16_t *)(p + 26)
#define RAX_REQDFIELDS_LEN 28 #define RAXHDR_ENDPOS *(uint64_t*)(p + 28)
#define RAX_REQDFIELDS_LEN 36
namespace Util{ namespace Util{
/// Helper function that cross-platform checks if a given directory exists. /// Helper function that cross-platform checks if a given directory exists.
@ -155,6 +157,7 @@ namespace Util{
/// 64-bits version of fseek /// 64-bits version of fseek
uint64_t fseek(FILE *stream, uint64_t offset, int whence){ uint64_t fseek(FILE *stream, uint64_t offset, int whence){
/// \TODO Windows implementation (e.g. _fseeki64 ?) /// \TODO Windows implementation (e.g. _fseeki64 ?)
clearerr(stream);
return fseeko(stream, offset, whence); return fseeko(stream, offset, whence);
} }
@ -171,14 +174,14 @@ namespace Util{
maxSize = 0; maxSize = 0;
} }
bool ResizeablePointer::assign(void * p, uint32_t l){ bool ResizeablePointer::assign(const void * p, uint32_t l){
if (!allocate(l)){return false;} if (!allocate(l)){return false;}
memcpy(ptr, p, l); memcpy(ptr, p, l);
currSize = l; currSize = l;
return true; return true;
} }
bool ResizeablePointer::append(void * p, uint32_t l){ bool ResizeablePointer::append(const void * p, uint32_t l){
if (!allocate(l+currSize)){return false;} if (!allocate(l+currSize)){return false;}
memcpy(((char*)ptr)+currSize, p, l); memcpy(((char*)ptr)+currSize, p, l);
currSize += l; currSize += l;
@ -189,7 +192,7 @@ namespace Util{
if (l > maxSize){ if (l > maxSize){
void *tmp = realloc(ptr, l); void *tmp = realloc(ptr, l);
if (!tmp){ if (!tmp){
FAIL_MSG("Could not allocate %lu bytes of memory", l); FAIL_MSG("Could not allocate %" PRIu32 " bytes of memory", l);
return false; return false;
} }
ptr = tmp; ptr = tmp;
@ -319,7 +322,7 @@ namespace Util{
if (lineno && strlen(lineno)){ if (lineno && strlen(lineno)){
dprintf(out, " (%s) ", lineno); dprintf(out, " (%s) ", lineno);
} }
dprintf(out, "\n", lineno); dprintf(out, "\n");
}else{ }else{
//could not be parsed as log string - print the whole thing //could not be parsed as log string - print the whole thing
dprintf(out, "%s\n", buf); dprintf(out, "%s\n", buf);
@ -329,8 +332,27 @@ namespace Util{
close(in); close(in);
} }
FieldAccX::FieldAccX(RelAccX * _src, RelAccXFieldData _field, char * _data) : src(_src), field(_field), data(_data) {}
uint64_t FieldAccX::uint(size_t recordNo) const {
return src->getInt(field, recordNo);
}
std::string FieldAccX::string(size_t recordNo) const {
return std::string(src->getPointer(field, recordNo), field.size) ;
}
void FieldAccX::set(uint64_t val, size_t recordNo){
src->setInt(field, val, recordNo);
}
void FieldAccX::set(const std::string & val, size_t recordNo){
char * place = src->getPointer(field, recordNo);
memcpy(place, val.data(), std::min((size_t)field.size, val.size()));
}
/// If waitReady is true (default), waits for isReady() to return true in 50ms sleep increments. /// If waitReady is true (default), waits for isReady() to return true in 50ms sleep increments.
RelAccX::RelAccX(char *data, bool waitReady){ RelAccX::RelAccX(char * data, bool waitReady){
if (!data){ if (!data){
p = 0; p = 0;
return; return;
@ -346,7 +368,7 @@ namespace Util{
p = 0; p = 0;
return; return;
} }
uint32_t dataOffset = 0; uint64_t dataOffset = 0;
while (offset < getOffset()){ while (offset < getOffset()){
const uint8_t sizeByte = p[offset]; const uint8_t sizeByte = p[offset];
const uint8_t nameLen = sizeByte >> 3; const uint8_t nameLen = sizeByte >> 3;
@ -376,8 +398,7 @@ namespace Util{
default: WARN_MSG("Unhandled field data size!"); break; default: WARN_MSG("Unhandled field data size!"); break;
} }
fields[fieldName] = RelAccXFieldData(fieldType, size, dataOffset); fields[fieldName] = RelAccXFieldData(fieldType, size, dataOffset);
DONTEVEN_MSG("Field %s: type %u, size %lu, offset %lu", fieldName.c_str(), fieldType, size, DONTEVEN_MSG("Field %s: type %u, size %" PRIu32 ", offset %" PRIu64, fieldName.c_str(), fieldType, size, dataOffset);
dataOffset);
dataOffset += size; dataOffset += size;
offset += nameLen + typeLen + 1; offset += nameLen + typeLen + 1;
} }
@ -390,17 +411,14 @@ namespace Util{
/// Gets the size in bytes of a single record in the structure. /// Gets the size in bytes of a single record in the structure.
uint32_t RelAccX::getRSize() const{return RAXHDR_RECORDSIZE;} uint32_t RelAccX::getRSize() const{return RAXHDR_RECORDSIZE;}
/// Gets the position in the records where the entries start
uint32_t RelAccX::getStartPos() const{return RAXHDR_STARTPOS;}
/// Gets the number of deleted records /// Gets the number of deleted records
uint64_t RelAccX::getDeleted() const{return RAXHDR_DELETED;} uint64_t RelAccX::getDeleted() const{return RAXHDR_DELETED;}
/// Gets the number of records present /// Gets the number of the last valid index
/// Defaults to the record count if set to zero. uint64_t RelAccX::getEndPos() const{return RAXHDR_ENDPOS;}
uint32_t RelAccX::getPresent() const{
return (RAXHDR_PRESENT ? RAXHDR_PRESENT : RAXHDR_RECORDCNT); ///Gets the number of fields per recrd
} uint32_t RelAccX::getFieldCount() const{return fields.size();}
/// Gets the offset from the structure start where records begin. /// Gets the offset from the structure start where records begin.
uint16_t RelAccX::getOffset() const{return *(uint16_t *)(p + 26);} uint16_t RelAccX::getOffset() const{return *(uint16_t *)(p + 26);}
@ -419,7 +437,7 @@ namespace Util{
// Check if the record has been deleted // Check if the record has been deleted
if (getDeleted() > recordNo){return false;} if (getDeleted() > recordNo){return false;}
// Check if the record hasn't been created yet // Check if the record hasn't been created yet
if (recordNo - getDeleted() >= getPresent()){return false;} if (recordNo >= getEndPos()){return false;}
return true; return true;
} }
@ -455,7 +473,10 @@ namespace Util{
/// Returns a null pointer if the field does not exist. /// Returns a null pointer if the field does not exist.
char *RelAccX::getPointer(const std::string &name, uint64_t recordNo) const{ char *RelAccX::getPointer(const std::string &name, uint64_t recordNo) const{
if (!fields.count(name)){return 0;} if (!fields.count(name)){return 0;}
const RelAccXFieldData &fd = fields.at(name); return getPointer(fields.at(name), recordNo);
}
char * RelAccX::getPointer(const RelAccXFieldData & fd, uint64_t recordNo) const{
return RECORD_POINTER; return RECORD_POINTER;
} }
@ -463,9 +484,12 @@ namespace Util{
/// Returns 0 if the field does not exist or is not an integer type. /// Returns 0 if the field does not exist or is not an integer type.
uint64_t RelAccX::getInt(const std::string &name, uint64_t recordNo) const{ uint64_t RelAccX::getInt(const std::string &name, uint64_t recordNo) const{
if (!fields.count(name)){return 0;} if (!fields.count(name)){return 0;}
const RelAccXFieldData &fd = fields.at(name); return getInt(fields.at(name), recordNo);
char *ptr = RECORD_POINTER; }
if ((fd.type & 0xF0) == RAX_UINT){// unsigned int
uint64_t RelAccX::getInt(const RelAccXFieldData & fd, uint64_t recordNo) const{
char * ptr = RECORD_POINTER;
if ((fd.type & 0xF0) == RAX_UINT){//unsigned int
switch (fd.size){ switch (fd.size){
case 1: return *(uint8_t *)ptr; case 1: return *(uint8_t *)ptr;
case 2: return *(uint16_t *)ptr; case 2: return *(uint16_t *)ptr;
@ -488,28 +512,82 @@ namespace Util{
return 0; // Not an integer type, or not implemented return 0; // Not an integer type, or not implemented
} }
std::string RelAccX::toPrettyString() const{
std::string RelAccX::toPrettyString(size_t indent) const{
std::stringstream r; std::stringstream r;
uint64_t delled = getDeleted(); uint64_t delled = getDeleted();
uint64_t max = delled + getRCount(); uint64_t max = getEndPos();
r << "RelAccX: " << getRCount() << " x " << getRSize() << "b @" << getOffset() << " (#" r << std::string(indent, ' ') << "RelAccX: " << getRCount() << " x " << getRSize() << "b @" << getOffset() << " (#" << getDeleted() << " - #" << getEndPos()-1 << ")" << std::endl;
<< getDeleted() << " - #" << (getDeleted() + getPresent() - 1) << ")" << std::endl;
for (uint64_t i = delled; i < max; ++i){ for (uint64_t i = delled; i < max; ++i){
r << " #" << i << ":" << std::endl; r << std::string(indent + 2, ' ') << "#" << i << ":" << std::endl;
for (std::map<std::string, RelAccXFieldData>::const_iterator it = fields.begin(); for (std::map<std::string, RelAccXFieldData>::const_iterator it = fields.begin(); it != fields.end(); ++it){
it != fields.end(); ++it){ r << std::string(indent + 4, ' ') << it->first << ": ";
r << " " << it->first << ": ";
switch (it->second.type & 0xF0){ switch (it->second.type & 0xF0){
case RAX_INT: r << (int64_t)getInt(it->first, i) << std::endl; break; case RAX_INT: r << (int64_t)getInt(it->first, i) << std::endl; break;
case RAX_UINT: r << getInt(it->first, i) << std::endl; break; case RAX_UINT: r << getInt(it->first, i) << std::endl; break;
case RAX_STRING: r << getPointer(it->first, i) << std::endl; break; case RAX_STRING: r << getPointer(it->first, i) << std::endl; break;
default: r << "[UNIMPLEMENTED]" << std::endl; break; case 0: { //RAX_NESTED
RelAccX n(getPointer(it->first, i), false);
if (n.isReady()){
r << "Nested RelAccX:" << std::endl;
r << (n.getFieldCount() > 6 ? n.toPrettyString(indent + 6) : n.toCompactString(indent + 6));
}else{
r << "Nested RelAccX: not ready" << std::endl;
}
break;
}
case RAX_RAW: {
char * ptr = getPointer(it->first, i);
size_t sz = getSize(it->first, i);
size_t zeroCount = 0;
for (size_t j = 0; j < sz && j < 100 && zeroCount < 10; ++j){
r << "0x" << std::hex << std::setw(2) << std::setfill('0') << (int)ptr[j] << std::dec << " ";
if (ptr[j] == 0x00){
zeroCount++;
}else{
zeroCount = 0;
}
}
r << std::endl;
break;
}
default: r << "[UNIMPLEMENTED]" << std::endl; break;
} }
} }
} }
return r.str(); return r.str();
} }
std::string RelAccX::toCompactString(size_t indent) const{
std::stringstream r;
uint64_t delled = getDeleted();
uint64_t max = getEndPos();
r << std::string(indent, ' ') << "RelAccX: " << getRCount() << " x " << getRSize() << "b @" << getOffset() << " (#" << getDeleted() << " - #" << getEndPos()-1 << ")" << std::endl;
for (uint64_t i = delled; i < max; ++i){
r << std::string(indent + 2, ' ') << "#" << i << ": ";
for (std::map<std::string, RelAccXFieldData>::const_iterator it = fields.begin(); it != fields.end(); ++it){
r << it->first << ": ";
switch (it->second.type & 0xF0){
case RAX_INT: r << (int64_t)getInt(it->first, i) << ", "; break;
case RAX_UINT: r << getInt(it->first, i) << ", "; break;
case RAX_STRING: r << getPointer(it->first, i) << ", "; break;
case 0: { //RAX_NESTED
RelAccX n(getPointer(it->first, i), false);
if (n.isReady()){
r << (n.getFieldCount() > 6 ? n.toPrettyString(indent + 2) : n.toCompactString(indent + 2));
}else{
r << "Nested RelAccX not ready" << std::endl;
}
break;
}
default: r << "[UNIMPLEMENTED], "; break;
}
}
r << std::endl;
}
return r.str();
}
/// Returns the default size in bytes of the data component of a field type number. /// Returns the default size in bytes of the data component of a field type number.
/// Returns zero if not implemented, unknown or the type has no default. /// Returns zero if not implemented, unknown or the type has no default.
uint32_t RelAccX::getDefaultSize(uint8_t fType){ uint32_t RelAccX::getDefaultSize(uint8_t fType){
@ -532,8 +610,7 @@ namespace Util{
return; return;
} }
if (!name.size() || name.size() > 31){ if (!name.size() || name.size() > 31){
WARN_MSG("Attempting to add a field with illegal name: %s (%u chars)", name.c_str(), WARN_MSG("Attempting to add a field with illegal name: %s (%zu chars)", name.c_str(), name.size());
name.size());
return; return;
} }
// calculate fLen if missing // calculate fLen if missing
@ -582,15 +659,11 @@ namespace Util{
/// Sets the record counter to the given value. /// Sets the record counter to the given value.
void RelAccX::setRCount(uint32_t count){RAXHDR_RECORDCNT = count;} void RelAccX::setRCount(uint32_t count){RAXHDR_RECORDCNT = count;}
/// Sets the position in the records where the entries start
void RelAccX::setStartPos(uint32_t n){RAXHDR_STARTPOS = n;}
/// Sets the number of deleted records /// Sets the number of deleted records
void RelAccX::setDeleted(uint64_t n){RAXHDR_DELETED = n;} void RelAccX::setDeleted(uint64_t n){RAXHDR_DELETED = n;}
/// Sets the number of records present /// Sets the number of the last valid index
/// Defaults to the record count if set to zero. void RelAccX::setEndPos(uint64_t n){RAXHDR_ENDPOS = n;}
void RelAccX::setPresent(uint32_t n){RAXHDR_PRESENT = n;}
/// Sets the ready flag. /// Sets the ready flag.
/// After calling this function, addField() may no longer be called. /// After calling this function, addField() may no longer be called.
@ -636,9 +709,12 @@ namespace Util{
WARN_MSG("Setting non-existent integer %s", name.c_str()); WARN_MSG("Setting non-existent integer %s", name.c_str());
return; return;
} }
const RelAccXFieldData &fd = fields.at(name); return setInt(fields.at(name), val, recordNo);
char *ptr = RECORD_POINTER; }
if ((fd.type & 0xF0) == RAX_UINT){// unsigned int
void RelAccX::setInt(const RelAccXFieldData & fd, uint64_t val, uint64_t recordNo){
char * ptr = RECORD_POINTER;
if ((fd.type & 0xF0) == RAX_UINT){//unsigned int
switch (fd.size){ switch (fd.size){
case 1: *(uint8_t *)ptr = val; return; case 1: *(uint8_t *)ptr = val; return;
case 2: *(uint16_t *)ptr = val; return; case 2: *(uint16_t *)ptr = val; return;
@ -658,38 +734,34 @@ namespace Util{
default: WARN_MSG("Unimplemented integer size %u", fd.size); return; default: WARN_MSG("Unimplemented integer size %u", fd.size); return;
} }
} }
WARN_MSG("Setting non-integer %s", name.c_str()); WARN_MSG("Setting non-integer field (%u) to integer value!", fd.type);
}
void RelAccX::setInts(const std::string & name, uint64_t * values, size_t len){
if (!fields.count(name)){
WARN_MSG("Setting non-existent integer %s", name.c_str());
return;
}
const RelAccXFieldData & fd = fields.at(name);
for (uint64_t recordNo = 0; recordNo < len; recordNo++){
setInt(fd, values[recordNo], recordNo);
}
} }
/// Updates the deleted record counter, the start position and the present record counter, /// Updates the deleted record counter, the start position and the present record counter,
/// shifting the ring buffer start position forward without moving the ring buffer end position. /// shifting the ring buffer start position forward without moving the ring buffer end position.
/// If the records present counter would be pushed into the negative by this function, sets it to
/// zero, defaulting it to the record count for all relevant purposes.
void RelAccX::deleteRecords(uint32_t amount){ void RelAccX::deleteRecords(uint32_t amount){
uint32_t &startPos = RAXHDR_STARTPOS; RAXHDR_DELETED += amount; // update deleted record counter
uint64_t &deletedRecs = RAXHDR_DELETED;
uint32_t &recsPresent = RAXHDR_PRESENT;
startPos += amount; // update start position
deletedRecs += amount; // update deleted record counter
if (recsPresent >= amount){
recsPresent -= amount; // decrease records present
}else{
recsPresent = 0;
}
} }
/// Updates the present record counter, shifting the ring buffer end position forward without /// Updates the present record counter, shifting the ring buffer end position forward without
/// moving the ring buffer start position. /// moving the ring buffer start position.
/// If the records present counter would be pushed past the record counter by this function, sets
/// it to zero, defaulting it to the record count for all relevant purposes.
void RelAccX::addRecords(uint32_t amount){ void RelAccX::addRecords(uint32_t amount){
uint32_t &recsPresent = RAXHDR_PRESENT; RAXHDR_ENDPOS += amount;
uint32_t &recordsCount = RAXHDR_RECORDCNT; }
if (recsPresent + amount > recordsCount){
recsPresent = 0; FieldAccX RelAccX::getFieldAccX(const std::string & fName){
}else{ return FieldAccX(this, fields.at(fName), p + getOffset());
recsPresent += amount;
}
} }
} }

View file

@ -1,8 +1,11 @@
#pragma once #pragma once
#include <deque>
#include <map>
#include <stdint.h> #include <stdint.h>
#include <string> #include <string>
#include <deque>
#include <vector>
#include <map>
#include <stdint.h>
#include <stdlib.h>
namespace Util{ namespace Util{
bool isDirectory(const std::string &path); bool isDirectory(const std::string &path);
@ -15,14 +18,17 @@ namespace Util{
uint64_t ftell(FILE *stream); uint64_t ftell(FILE *stream);
uint64_t fseek(FILE *stream, uint64_t offset, int whence); uint64_t fseek(FILE *stream, uint64_t offset, int whence);
//Forward declaration
class FieldAccX;
/// Helper class that maintains a resizeable pointer and will free it upon deletion of the class. /// Helper class that maintains a resizeable pointer and will free it upon deletion of the class.
class ResizeablePointer{ class ResizeablePointer{
public: public:
ResizeablePointer(); ResizeablePointer();
~ResizeablePointer(); ~ResizeablePointer();
inline uint32_t& size(){return currSize;} inline uint32_t& size(){return currSize;}
bool assign(void * p, uint32_t l); bool assign(const void * p, uint32_t l);
bool append(void * p, uint32_t l); bool append(const void * p, uint32_t l);
bool allocate(uint32_t l); bool allocate(uint32_t l);
inline operator char*(){return (char*)ptr;} inline operator char*(){return (char*)ptr;}
inline operator void*(){return ptr;} inline operator void*(){return ptr;}
@ -64,6 +70,7 @@ namespace Util{
#define RAX_64STRING 0x32 #define RAX_64STRING 0x32
#define RAX_128STRING 0x33 #define RAX_128STRING 0x33
#define RAX_256STRING 0x34 #define RAX_256STRING 0x34
#define RAX_512STRING 0x35
#define RAX_RAW 0x40 #define RAX_RAW 0x40
#define RAX_256RAW 0x44 #define RAX_256RAW 0x44
#define RAX_512RAW 0x45 #define RAX_512RAW 0x45
@ -82,6 +89,7 @@ namespace Util{
/// 8 bytes records deleted - amount of records no longer present /// 8 bytes records deleted - amount of records no longer present
/// 4 bytes records present - amount of record currently present /// 4 bytes records present - amount of record currently present
/// 2 bytes record offset /// 2 bytes record offset
/// 8 bytes record endpos - index after the last valid record in the buffer
/// @field_offset: offset-field_offset bytes fields: /// @field_offset: offset-field_offset bytes fields:
/// 5 bits field name len (< 32), 3 bits type len (1-5) /// 5 bits field name len (< 32), 3 bits type len (1-5)
/// len bytes field name string (< 32 bytes) /// len bytes field name string (< 32 bytes)
@ -102,44 +110,67 @@ namespace Util{
/// Setting reload means the writer needed to change fields, and the pointer should be closed and /// Setting reload means the writer needed to change fields, and the pointer should be closed and
/// re-opened through outside means (e.g. closing and re-opening the containing shm page). /// re-opened through outside means (e.g. closing and re-opening the containing shm page).
class RelAccX{ class RelAccX{
public: public:
RelAccX(char *data, bool waitReady = true); RelAccX(char * data = NULL, bool waitReady = true);
// Read-only functions: //Read-only functions:
uint32_t getRCount() const; uint32_t getRCount() const;
uint32_t getRSize() const; uint32_t getRSize() const;
uint16_t getOffset() const; uint16_t getOffset() const;
uint32_t getStartPos() const; uint64_t getDeleted() const;
uint64_t getDeleted() const; uint64_t getEndPos() const;
uint32_t getPresent() const; uint32_t getFieldCount() const;
bool isReady() const; bool isReady() const;
bool isExit() const; bool isExit() const;
bool isReload() const; bool isReload() const;
bool isRecordAvailable(uint64_t recordNo) const; bool isRecordAvailable(uint64_t recordNo) const;
uint32_t getRecordPosition(uint64_t recordNo) const; uint32_t getRecordPosition(uint64_t recordNo) const;
uint32_t getSize(const std::string &name, uint64_t recordNo = 0) const; uint32_t getSize(const std::string & name, uint64_t recordNo=0) const;
char *getPointer(const std::string &name, uint64_t recordNo = 0) const;
uint64_t getInt(const std::string &name, uint64_t recordNo = 0) const;
std::string toPrettyString() const;
// Read-write functions:
void addField(const std::string &name, uint8_t fType, uint32_t fLen = 0);
void setRCount(uint32_t count);
void setStartPos(uint32_t n);
void setDeleted(uint64_t n);
void setPresent(uint32_t n);
void setReady();
void setExit();
void setReload();
void setString(const std::string &name, const std::string &val, uint64_t recordNo = 0);
void setInt(const std::string &name, uint64_t val, uint64_t recordNo = 0);
void deleteRecords(uint32_t amount);
void addRecords(uint32_t amount);
protected: char * getPointer(const std::string & name, uint64_t recordNo=0) const;
static uint32_t getDefaultSize(uint8_t fType); char * getPointer(const RelAccXFieldData & fd, uint64_t recordNo=0) const;
private: uint64_t getInt(const std::string & name, uint64_t recordNo=0) const;
char *p; uint64_t getInt(const RelAccXFieldData & fd, uint64_t recordNo=0) const;
std::map<std::string, RelAccXFieldData> fields;
std::string toPrettyString(size_t indent = 0) const;
std::string toCompactString(size_t indent = 0) const;
//Read-write functions:
void addField(const std::string & name, uint8_t fType, uint32_t fLen=0);
void setRCount(uint32_t count);
void setDeleted(uint64_t n);
void setEndPos(uint64_t n);
void setReady();
void setExit();
void setReload();
void setString(const std::string & name, const std::string & val, uint64_t recordNo=0);
void setInt(const std::string & name, uint64_t val, uint64_t recordNo=0);
void setInt(const RelAccXFieldData & fd, uint64_t val, uint64_t recordNo=0);
void setInts(const std::string & name, uint64_t * values, size_t len);
void deleteRecords(uint32_t amount);
void addRecords(uint32_t amount);
void minimalFrom(const RelAccX & src);
void copyFieldsFrom(const RelAccX & src, bool minimal = false);
void flowFrom(const RelAccX & src);
FieldAccX getFieldAccX(const std::string & fName);
protected:
static uint32_t getDefaultSize(uint8_t fType);
std::map<std::string, RelAccXFieldData> fields;
private:
char * p;
};
class FieldAccX {
public:
FieldAccX(RelAccX * _src = NULL, RelAccXFieldData _field = RelAccXFieldData(), char * _data = NULL);
uint64_t uint(size_t recordNo) const;
std::string string(size_t recordNo) const;
void set(uint64_t val, size_t recordNo = 0);
void set(const std::string & val, size_t recordNo = 0);
private:
RelAccX * src;
RelAccXFieldData field;
char * data;
}; };
} }

View file

@ -164,6 +164,7 @@ namespace Mist{
/// The standard implementation will set isInitialized to false and close the client connection, /// The standard implementation will set isInitialized to false and close the client connection,
/// thus causing the process to exit cleanly. /// thus causing the process to exit cleanly.
void Output::onFail(){ void Output::onFail(){
MEDIUM_MSG("onFail");
isInitialized = false; isInitialized = false;
wantRequest = true; wantRequest = true;
parseData= false; parseData= false;
@ -172,6 +173,7 @@ namespace Mist{
} }
void Output::initialize(){ void Output::initialize(){
MEDIUM_MSG("initialize");
if (isInitialized){ if (isInitialized){
return; return;
} }
@ -181,7 +183,6 @@ namespace Mist{
if (streamName.size() < 1){ if (streamName.size() < 1){
return; //abort - no stream to initialize... return; //abort - no stream to initialize...
} }
isInitialized = true;
reconnect(); reconnect();
//if the connection failed, fail //if the connection failed, fail
if (streamName.size() < 1){ if (streamName.size() < 1){
@ -322,6 +323,21 @@ namespace Mist{
return false; return false;
} }
/// Disconnects from all stat/user-related shared structures.
void Output::disconnect(){
MEDIUM_MSG("disconnect");
if (statsPage.getData()){
statsPage.finish();
myConn.resetCounter();
}
if (nProxy.userClient.getData()){
nProxy.userClient.finish();
}
isInitialized = false;
myMeta.reset();
nProxy.metaPages.clear();
}
/// Connects or reconnects to the stream. /// Connects or reconnects to the stream.
/// Assumes streamName class member has been set already. /// Assumes streamName class member has been set already.
/// Will start input if not currently active, calls onFail() if this does not succeed. /// Will start input if not currently active, calls onFail() if this does not succeed.
@ -343,13 +359,7 @@ namespace Mist{
return; return;
} }
} }
if (statsPage.getData()){ disconnect();
statsPage.finish();
myConn.resetCounter();
}
if (nProxy.userClient.getData()){
nProxy.userClient.finish();
}
nProxy.streamName = streamName; nProxy.streamName = streamName;
char userPageName[NAME_BUFFER_SIZE]; char userPageName[NAME_BUFFER_SIZE];
snprintf(userPageName, NAME_BUFFER_SIZE, SHM_USERS, streamName.c_str()); snprintf(userPageName, NAME_BUFFER_SIZE, SHM_USERS, streamName.c_str());
@ -364,13 +374,13 @@ namespace Mist{
} }
char pageId[NAME_BUFFER_SIZE]; char pageId[NAME_BUFFER_SIZE];
snprintf(pageId, NAME_BUFFER_SIZE, SHM_STREAM_INDEX, streamName.c_str()); snprintf(pageId, NAME_BUFFER_SIZE, SHM_STREAM_INDEX, streamName.c_str());
nProxy.metaPages.clear();
nProxy.metaPages[0].init(pageId, DEFAULT_STRM_PAGE_SIZE); nProxy.metaPages[0].init(pageId, DEFAULT_STRM_PAGE_SIZE);
if (!nProxy.metaPages[0].mapped){ if (!nProxy.metaPages[0].mapped){
FAIL_MSG("Could not connect to data for %s", streamName.c_str()); FAIL_MSG("Could not connect to data for %s", streamName.c_str());
onFail(); onFail();
return; return;
} }
isInitialized = true;
statsPage = IPC::sharedClient(SHM_STATISTICS, STAT_EX_SIZE, true); statsPage = IPC::sharedClient(SHM_STATISTICS, STAT_EX_SIZE, true);
stats(true); stats(true);
updateMeta(); updateMeta();
@ -1562,8 +1572,11 @@ namespace Mist{
nProxy.userClient = IPC::sharedClient(userPageName, PLAY_EX_SIZE, true); nProxy.userClient = IPC::sharedClient(userPageName, PLAY_EX_SIZE, true);
if (!nProxy.userClient.getData()){ if (!nProxy.userClient.getData()){
WARN_MSG("Player connection failure - aborting output"); WARN_MSG("Player connection failure - aborting output");
onFinish(); if (!onFinish()){
myConn.close(); myConn.close();
}else{
disconnect();
}
return; return;
} }
} }
@ -1572,14 +1585,20 @@ namespace Mist{
waitForStreamPushReady(); waitForStreamPushReady();
if (!nProxy.userClient.isAlive()){ if (!nProxy.userClient.isAlive()){
WARN_MSG("Failed to wait for buffer, aborting incoming push"); WARN_MSG("Failed to wait for buffer, aborting incoming push");
onFinish(); if (!onFinish()){
myConn.close(); myConn.close();
}else{
disconnect();
}
return; return;
} }
}else{ }else{
INFO_MSG("Received disconnect request from input"); INFO_MSG("Received disconnect request from input");
onFinish(); if (!onFinish()){
myConn.close(); myConn.close();
}else{
disconnect();
}
return; return;
} }
} }

View file

@ -75,6 +75,7 @@ namespace Mist {
return false; return false;
} }
void reconnect(); void reconnect();
void disconnect();
virtual void initialize(); virtual void initialize();
virtual void sendHeader(); virtual void sendHeader();
virtual void onFail(); virtual void onFail();