sikkienl/dash
1
0
Fork 0
mirror of https://github.com/dashpay/dash.git synced 2024-12-26 12:32:48 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix OOM bug: UTXO entries with invalid script length

Browse Source
This commit is contained in:
Pieter Wuille 2016-04-24 16:21:44 +02:00
parent 4bf631e5e4
commit 5d0434d13d
2 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/compressor.h
View File

@ -86,8 +86,14 @@ public:
return; return;
} }
nSize -= nSpecialScripts; nSize -= nSpecialScripts;
script.resize(nSize); if (nSize > MAX_SCRIPT_SIZE) {
s >> REF(CFlatData(script)); // Overly long script, replace with a short invalid one
script << OP_RETURN;
s.ignore(nSize);
} else {
script.resize(nSize);
s >> REF(CFlatData(script));
}
} }
}; };

14
src/streams.h
View File

@ -406,6 +406,20 @@ public:
return (*this); return (*this);
} }
CAutoFile& ignore(size_t nSize)
{
if (!file)
throw std::ios_base::failure("CAutoFile::ignore: file handle is NULL");
unsigned char data[4096];
while (nSize > 0) {
size_t nNow = std::min<size_t>(nSize, sizeof(data));
if (fread(data, 1, nNow, file) != nNow)
throw std::ios_base::failure(feof(file) ? "CAutoFile::ignore: end of file" : "CAutoFile::read: fread failed");
nSize -= nNow;
}
return (*this);
}
CAutoFile& write(const char* pch, size_t nSize) CAutoFile& write(const char* pch, size_t nSize)
{ {
if (!file) if (!file)