Compare commits

...

5 Commits

Author SHA1 Message Date
0c63b66544 Special case not needed 2025-01-07 09:34:45 +01:00
30fee38b25 Empty messages are valid 2025-01-07 09:32:30 +01:00
00c52d5ab7 Allow zero size lengths 2025-01-07 09:16:55 +01:00
57af4fca5b Added missing defaults 2025-01-07 09:11:36 +01:00
3c125c9a7a Added missing length size checks
+ code cleanup/refactoring
2025-01-07 08:52:40 +01:00
2 changed files with 80 additions and 70 deletions

View File

@ -40,13 +40,22 @@ static bool varint(CTX &ctx, T &val)
return false; return false;
} }
static bool length(CTX &ctx, qint32 &val)
{
if (TYPE(ctx.tag) != LEN)
return false;
if (!varint(ctx, val))
return false;
return (val >= 0);
}
static bool str(CTX &ctx, QByteArray &val) static bool str(CTX &ctx, QByteArray &val)
{ {
qint32 len; qint32 len;
if (TYPE(ctx.tag) != LEN) if (!length(ctx, len))
return false;
if (!varint(ctx, len))
return false; return false;
if (ctx.bp + len > ctx.be) if (ctx.bp + len > ctx.be)
return false; return false;
@ -89,7 +98,7 @@ static bool packed(CTX &ctx, QVector<quint32> &vals)
if (TYPE(ctx.tag) == LEN) { if (TYPE(ctx.tag) == LEN) {
qint32 len; qint32 len;
if (!varint(ctx, len)) if (!varint(ctx, len) || len < 0)
return false; return false;
const char *ee = ctx.bp + len; const char *ee = ctx.bp + len;
if (ee > ctx.be) if (ee > ctx.be)
@ -120,7 +129,7 @@ static bool skip(CTX &ctx)
len = 8; len = 8;
break; break;
case LEN: case LEN:
if (!varint(ctx, len)) if (!varint(ctx, len) || len < 0)
return false; return false;
break; break;
case I32: case I32:
@ -139,16 +148,13 @@ static bool skip(CTX &ctx)
static bool value(CTX &ctx, QVariant &val) static bool value(CTX &ctx, QVariant &val)
{ {
if (TYPE(ctx.tag) != LEN)
return false;
qint32 len; qint32 len;
QByteArray ba; QByteArray ba;
quint64 num; quint64 num;
double dnum; double dnum;
float fnum; float fnum;
if (!varint(ctx, len)) if (!length(ctx, len))
return false; return false;
const char *ee = ctx.bp + len; const char *ee = ctx.bp + len;
@ -214,13 +220,10 @@ static bool value(CTX &ctx, QVariant &val)
static bool feature(CTX &ctx, Data::Feature &f) static bool feature(CTX &ctx, Data::Feature &f)
{ {
qint32 len;
quint32 e; quint32 e;
if (TYPE(ctx.tag) != LEN) if (!length(ctx, len))
return false;
qint32 len;
if (!varint(ctx, len))
return false; return false;
const char *ee = ctx.bp + len; const char *ee = ctx.bp + len;
@ -266,12 +269,9 @@ static bool feature(CTX &ctx, Data::Feature &f)
static bool layer(CTX &ctx, Data::Layer &l) static bool layer(CTX &ctx, Data::Layer &l)
{ {
if (FIELD(ctx.tag) == 3) {
if (TYPE(ctx.tag) != LEN)
return false;
qint32 len; qint32 len;
if (!varint(ctx, len))
if (!length(ctx, len))
return false; return false;
const char *ee = ctx.bp + len; const char *ee = ctx.bp + len;
@ -321,8 +321,6 @@ static bool layer(CTX &ctx, Data::Layer &l)
} }
return (ctx.bp == ee); return (ctx.bp == ee);
} else
return skip(ctx);
} }
bool Data::load(const QByteArray &ba) bool Data::load(const QByteArray &ba)
@ -332,9 +330,17 @@ bool Data::load(const QByteArray &ba)
while (ctx.bp < ctx.be) { while (ctx.bp < ctx.be) {
if (!varint(ctx, ctx.tag)) if (!varint(ctx, ctx.tag))
return false; return false;
switch (FIELD(ctx.tag)) {
case 3:
_layers.append(Layer()); _layers.append(Layer());
if (!layer(ctx, _layers.last())) if (!layer(ctx, _layers.last()))
return false; return false;
break;
default:
if (!skip(ctx))
return false;
}
} }
return (ctx.bp == ctx.be); return (ctx.bp == ctx.be);

View File

@ -17,6 +17,8 @@ public:
struct Feature struct Feature
{ {
Feature() : id(0), type(UNKNOWN) {}
quint64 id; quint64 id;
QVector<quint32> tags; QVector<quint32> tags;
GeomType type; GeomType type;
@ -25,6 +27,8 @@ public:
struct Layer struct Layer
{ {
Layer() : version(1), extent(4096) {}
quint32 version; quint32 version;
QByteArray name; QByteArray name;
QVector<Feature> features; QVector<Feature> features;