Prevents PLY from parsing duplicate properties (#5743)
Co-authored-by: Kim Kulling <kimkulling@users.noreply.github.com>pull/5754/head^2
parent
f69e55058d
commit
ff2dc2fb2e
|
@ -48,10 +48,29 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||||
#include <assimp/ByteSwapper.h>
|
#include <assimp/ByteSwapper.h>
|
||||||
#include <assimp/fast_atof.h>
|
#include <assimp/fast_atof.h>
|
||||||
#include <assimp/DefaultLogger.hpp>
|
#include <assimp/DefaultLogger.hpp>
|
||||||
|
#include <unordered_set>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
namespace Assimp {
|
namespace Assimp {
|
||||||
|
|
||||||
|
std::string to_string(EElementSemantic e) {
|
||||||
|
|
||||||
|
switch (e) {
|
||||||
|
case EEST_Vertex:
|
||||||
|
return std::string{ "vertex" };
|
||||||
|
case EEST_TriStrip:
|
||||||
|
return std::string{ "tristrips" };
|
||||||
|
case EEST_Edge:
|
||||||
|
return std::string{ "edge" };
|
||||||
|
case EEST_Material:
|
||||||
|
return std::string{ "material" };
|
||||||
|
case EEST_TextureFile:
|
||||||
|
return std::string{ "TextureFile" };
|
||||||
|
default:
|
||||||
|
return std::string{ "invalid" };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
PLY::EDataType PLY::Property::ParseDataType(std::vector<char> &buffer) {
|
PLY::EDataType PLY::Property::ParseDataType(std::vector<char> &buffer) {
|
||||||
ai_assert(!buffer.empty());
|
ai_assert(!buffer.empty());
|
||||||
|
@ -281,6 +300,8 @@ bool PLY::Element::ParseElement(IOStreamBuffer<char> &streamBuffer, std::vector<
|
||||||
// if the exact semantic can't be determined, just store
|
// if the exact semantic can't be determined, just store
|
||||||
// the original string identifier
|
// the original string identifier
|
||||||
pOut->szName = std::string(&buffer[0], &buffer[0] + strlen(&buffer[0]));
|
pOut->szName = std::string(&buffer[0], &buffer[0] + strlen(&buffer[0]));
|
||||||
|
auto pos = pOut->szName.find_last_of(' ');
|
||||||
|
pOut->szName.erase(pos, pOut->szName.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!PLY::DOM::SkipSpaces(buffer))
|
if (!PLY::DOM::SkipSpaces(buffer))
|
||||||
|
@ -413,6 +434,7 @@ bool PLY::DOM::SkipComments(std::vector<char> buffer) {
|
||||||
bool PLY::DOM::ParseHeader(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, bool isBinary) {
|
bool PLY::DOM::ParseHeader(IOStreamBuffer<char> &streamBuffer, std::vector<char> &buffer, bool isBinary) {
|
||||||
ASSIMP_LOG_VERBOSE_DEBUG("PLY::DOM::ParseHeader() begin");
|
ASSIMP_LOG_VERBOSE_DEBUG("PLY::DOM::ParseHeader() begin");
|
||||||
|
|
||||||
|
std::unordered_set<std::string> definedAlElements;
|
||||||
// parse all elements
|
// parse all elements
|
||||||
while (!buffer.empty()) {
|
while (!buffer.empty()) {
|
||||||
// skip all comments
|
// skip all comments
|
||||||
|
@ -421,6 +443,13 @@ bool PLY::DOM::ParseHeader(IOStreamBuffer<char> &streamBuffer, std::vector<char>
|
||||||
PLY::Element out;
|
PLY::Element out;
|
||||||
if (PLY::Element::ParseElement(streamBuffer, buffer, &out)) {
|
if (PLY::Element::ParseElement(streamBuffer, buffer, &out)) {
|
||||||
// add the element to the list of elements
|
// add the element to the list of elements
|
||||||
|
|
||||||
|
const auto propertyName = (out.szName.empty()) ? to_string(out.eSemantic) : out.szName;
|
||||||
|
auto alreadyDefined = definedAlElements.find(propertyName);
|
||||||
|
if (alreadyDefined != definedAlElements.end()) {
|
||||||
|
throw DeadlyImportError("Property '" + propertyName + "' in header already defined ");
|
||||||
|
}
|
||||||
|
definedAlElements.insert(propertyName);
|
||||||
alElements.push_back(out);
|
alElements.push_back(out);
|
||||||
} else if (TokenMatch(buffer, "end_header", 10)) {
|
} else if (TokenMatch(buffer, "end_header", 10)) {
|
||||||
// we have reached the end of the header
|
// we have reached the end of the header
|
||||||
|
|
|
@ -209,3 +209,55 @@ TEST_F(utPLYImportExport, payload_JVN42386607) {
|
||||||
const aiScene *scene = importer.ReadFile(ASSIMP_TEST_MODELS_DIR "/PLY/payload_JVN42386607", 0);
|
const aiScene *scene = importer.ReadFile(ASSIMP_TEST_MODELS_DIR "/PLY/payload_JVN42386607", 0);
|
||||||
EXPECT_EQ(nullptr, scene);
|
EXPECT_EQ(nullptr, scene);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Tests Issue #5729. Test, if properties defined multiple times. Unclear what to do, better to abort than to crash entirely
|
||||||
|
TEST_F(utPLYImportExport, parseInvalidDoubleProperty) {
|
||||||
|
const char data[] = "ply\n"
|
||||||
|
"format ascii 1.0\n"
|
||||||
|
"element vertex 4\n"
|
||||||
|
"property float x\n"
|
||||||
|
"property float y\n"
|
||||||
|
"property float z\n"
|
||||||
|
"element vertex 8\n"
|
||||||
|
"property float x\n"
|
||||||
|
"property float y\n"
|
||||||
|
"property float z\n"
|
||||||
|
"end_header\n"
|
||||||
|
"0.0 0.0 0.0 0.0 0.0 0.0\n"
|
||||||
|
"0.0 0.0 1.0 0.0 0.0 1.0\n"
|
||||||
|
"0.0 1.0 0.0 0.0 1.0 0.0\n"
|
||||||
|
"0.0 0.0 1.0\n"
|
||||||
|
"0.0 1.0 0.0 0.0 0.0 1.0\n"
|
||||||
|
"0.0 1.0 1.0 0.0 1.0 1.0\n";
|
||||||
|
|
||||||
|
Assimp::Importer importer;
|
||||||
|
const aiScene *scene = importer.ReadFileFromMemory(data, sizeof(data), 0);
|
||||||
|
EXPECT_EQ(nullptr, scene);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tests Issue #5729. Test, if properties defined multiple times. Unclear what to do, better to abort than to crash entirely
|
||||||
|
TEST_F(utPLYImportExport, parseInvalidDoubleCustomProperty) {
|
||||||
|
const char data[] = "ply\n"
|
||||||
|
"format ascii 1.0\n"
|
||||||
|
"element vertex 4\n"
|
||||||
|
"property float x\n"
|
||||||
|
"property float y\n"
|
||||||
|
"property float z\n"
|
||||||
|
"element name 8\n"
|
||||||
|
"property float x\n"
|
||||||
|
"element name 5\n"
|
||||||
|
"property float x\n"
|
||||||
|
"end_header\n"
|
||||||
|
"0.0 0.0 0.0 100.0 10.0\n"
|
||||||
|
"0.0 0.0 1.0 200.0 20.0\n"
|
||||||
|
"0.0 1.0 0.0 300.0 30.0\n"
|
||||||
|
"0.0 1.0 1.0 400.0 40.0\n"
|
||||||
|
"0.0 0.0 0.0 500.0 50.0\n"
|
||||||
|
"0.0 0.0 1.0 600.0 60.0\n"
|
||||||
|
"0.0 1.0 0.0 700.0 70.0\n"
|
||||||
|
"0.0 1.0 1.0 800.0 80.0\n";
|
||||||
|
|
||||||
|
Assimp::Importer importer;
|
||||||
|
const aiScene *scene = importer.ReadFileFromMemory(data, sizeof(data), 0);
|
||||||
|
EXPECT_EQ(nullptr, scene);
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue