Navigation Menu

Skip to content

Commit

Permalink
EnrichedString: Fix substr segfault caused by non-formatted text
Browse files Browse the repository at this point in the history
  • Loading branch information
SmallJoker committed Jan 24, 2020
1 parent 987b2c5 commit 9cb3219
Showing 1 changed file with 16 additions and 5 deletions.
21 changes: 16 additions & 5 deletions src/util/enriched_string.cpp
Expand Up @@ -19,7 +19,9 @@ with this program; if not, write to the Free Software Foundation, Inc.,

#include "enriched_string.h"
#include "util/string.h"
#include "debug.h"
#include "log.h"

using namespace irr::video;

EnrichedString::EnrichedString()
Expand All @@ -28,10 +30,12 @@ EnrichedString::EnrichedString()
}

EnrichedString::EnrichedString(const std::wstring &string,
const std::vector<SColor> &colors):
m_string(string),
m_colors(colors)
{}
const std::vector<SColor> &colors)
{
clear();
m_string = string;
m_colors = colors;
}

EnrichedString::EnrichedString(const std::wstring &s, const SColor &color)
{
Expand All @@ -52,6 +56,7 @@ void EnrichedString::clear()
m_has_background = false;
m_default_length = 0;
m_default_color = irr::video::SColor(255, 255, 255, 255);
m_background = irr::video::SColor(0, 0, 0, 0);
}

void EnrichedString::operator=(const wchar_t *str)
Expand Down Expand Up @@ -170,8 +175,12 @@ EnrichedString EnrichedString::substr(size_t pos, size_t len) const
m_string.substr(pos, len),
std::vector<SColor>(m_colors.begin() + pos, m_colors.begin() + pos + len)
);

str.m_has_background = m_has_background;
str.m_background = m_background;

if (pos < m_default_length)
str.m_default_length = m_default_length - pos;
str.m_default_length = std::min(m_default_length - pos, str.size());
str.setDefaultColor(m_default_color);
return str;
}
Expand Down Expand Up @@ -199,6 +208,8 @@ void EnrichedString::setDefaultColor(const irr::video::SColor &color)

void EnrichedString::updateDefaultColor()
{
sanity_check(m_default_length <= m_colors.size());

for (size_t i = 0; i < m_default_length; ++i)
m_colors[i] = m_default_color;
}

0 comments on commit 9cb3219

Please sign in to comment.