Skip to content

Commit

Permalink
mg_schematic: fix leak in lua API, and small cleanup
Browse files Browse the repository at this point in the history
* Fix leak like behaviour if you load multiple schematics in a loop.

* Cleanup check in for, fixing theoretical out of bounds read if
	Schematic::deserializeFromMts reduced the number of elements
	in m_nodenames. A != check may need an overflow of the counter
	before it hits, if origsize is larger than m_nodenames.size().

* Fix function name passed to errorstream: it was wrong. Also use
	__FUNCTION__ instead of manually using the method name at other
	places in the function.

* Don't shadow the name member in the loop.
  • Loading branch information
est31 committed Mar 30, 2016
1 parent 0115da1 commit 0aac1b7
Showing 1 changed file with 11 additions and 9 deletions.
20 changes: 11 additions & 9 deletions src/mg_schematic.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -267,15 +267,15 @@ bool Schematic::deserializeFromMts(std::istream *is,
//// Read signature
u32 signature = readU32(ss);
if (signature != MTSCHEM_FILE_SIGNATURE) {
errorstream << "Schematic::deserializeFromMts: invalid schematic "
errorstream << __FUNCTION__ << ": invalid schematic "
"file" << std::endl;
return false;
}

//// Read version
u16 version = readU16(ss);
if (version > MTSCHEM_FILE_VER_HIGHEST_READ) {
errorstream << "Schematic::deserializeFromMts: unsupported schematic "
errorstream << __FUNCTION__ << ": unsupported schematic "
"file version" << std::endl;
return false;
}
Expand Down Expand Up @@ -439,7 +439,7 @@ bool Schematic::loadSchematicFromFile(const std::string &filename,
{
std::ifstream is(filename.c_str(), std::ios_base::binary);
if (!is.good()) {
errorstream << "Schematic::loadSchematicFile: unable to open file '"
errorstream << __FUNCTION__ << ": unable to open file '"
<< filename << "'" << std::endl;
return false;
}
Expand All @@ -448,17 +448,19 @@ bool Schematic::loadSchematicFromFile(const std::string &filename,
if (!deserializeFromMts(&is, &m_nodenames))
return false;

m_nnlistsizes.push_back(m_nodenames.size() - origsize);

name = filename;

if (replace_names) {
for (size_t i = origsize; i != m_nodenames.size(); i++) {
std::string &name = m_nodenames[i];
StringMap::iterator it = replace_names->find(name);
for (size_t i = origsize; i < m_nodenames.size(); i++) {
std::string &node_name = m_nodenames[i];
StringMap::iterator it = replace_names->find(node_name);
if (it != replace_names->end())
name = it->second;
node_name = it->second;
}
}

m_nnlistsizes.push_back(m_nodenames.size() - origsize);

if (ndef)
ndef->pendNodeResolve(this);

Expand Down

0 comments on commit 0aac1b7

Please sign in to comment.