Fix issue with writing setting field presence mask for empty arrays

This commit is contained in:
2017-04-30 20:25:37 -05:00
parent 5328f9004e
commit c3388c58f9
2 changed files with 30 additions and 21 deletions
+8 -2
View File
@@ -10,6 +10,9 @@
#include "err.hpp" #include "err.hpp"
#include "presencemask.hpp" #include "presencemask.hpp"
#include <iostream>
using namespace std;
namespace ox { namespace ox {
FieldPresenseMask::FieldPresenseMask(uint8_t *mask, size_t maxLen) { FieldPresenseMask::FieldPresenseMask(uint8_t *mask, size_t maxLen) {
@@ -26,9 +29,12 @@ bool FieldPresenseMask::get(int i) {
} }
int FieldPresenseMask::set(int i, bool on) { int FieldPresenseMask::set(int i, bool on) {
uint8_t val = on ? 1 : 0; // normalize to 0 or 1
if (i / 8 < m_maxLen) { if (i / 8 < m_maxLen) {
m_mask[i / 8] |= val << (i % 8); if (on) {
m_mask[i / 8] |= 1 << (i % 8);
} else {
m_mask[i / 8] &= ~(1 << (i % 8));
}
return 0; return 0;
} else { } else {
return MC_PRESENCEMASKOUTBOUNDS; return MC_PRESENCEMASKOUTBOUNDS;
+22 -19
View File
@@ -70,7 +70,7 @@ int MetalClawWriter::op(const char*, ox::bstring<L> *val) {
m_buffIt += val->size(); m_buffIt += val->size();
fieldSet = true; fieldSet = true;
} else { } else {
err = 1; err = MC_BUFFENDED;
} }
} }
err |= m_fieldPresence.set(m_field, fieldSet); err |= m_fieldPresence.set(m_field, fieldSet);
@@ -84,6 +84,7 @@ int MetalClawWriter::op(const char*, T *val) {
MetalClawWriter writer(m_buff + m_buffIt, m_buffLen - m_buffIt); MetalClawWriter writer(m_buff + m_buffIt, m_buffLen - m_buffIt);
err |= ioOp(&writer, val); err |= ioOp(&writer, val);
m_buffIt += writer.m_buffIt; m_buffIt += writer.m_buffIt;
m_field++;
return err; return err;
}; };
@@ -110,26 +111,28 @@ int MetalClawWriter::op(const char*, T *val, size_t len) {
int err = 0; int err = 0;
bool fieldSet = false; bool fieldSet = false;
// write the length if (len) {
typedef uint32_t ArrayLength; // write the length
if (m_buffIt + sizeof(ArrayLength) < m_buffLen) { typedef uint32_t ArrayLength;
*((T*) &m_buff[m_buffIt]) = ox::std::bigEndianAdapt((ArrayLength) len); if (m_buffIt + sizeof(ArrayLength) < m_buffLen) {
m_buffIt += sizeof(ArrayLength); *((T*) &m_buff[m_buffIt]) = ox::std::bigEndianAdapt((ArrayLength) len);
} else { m_buffIt += sizeof(ArrayLength);
err = MC_BUFFENDED; } else {
err = MC_BUFFENDED;
}
MetalClawWriter writer(m_buff + m_buffIt, m_buffLen - m_buffIt);
writer.setFields(len);
// write the array
for (size_t i = 0; i < len; i++) {
err |= writer.op("", &val[i]);
}
m_buffIt += writer.m_buffIt;
fieldSet = true;
} }
MetalClawWriter writer(m_buff + m_buffIt, m_buffLen - m_buffIt);
writer.setFields(len);
// write the string
for (size_t i = 0; i < len; i++) {
err |= writer.op("", &val[i]);
}
m_buffIt += writer.m_buffIt;
fieldSet = true;
err |= m_fieldPresence.set(m_field, fieldSet); err |= m_fieldPresence.set(m_field, fieldSet);
m_field++; m_field++;
return err; return err;