From ea33cadebf93995c86559f233d4c79d65ff1f860 Mon Sep 17 00:00:00 2001
From: moneromooo-monero <moneromooo-monero@users.noreply.github.com>
Date: Sun, 21 Jun 2015 11:28:16 +0100
Subject: [PATCH] mnemonics: fix prefix extraction with non ASCII text

---
 src/mnemonics/electrum-words.cpp |  8 ++++----
 src/mnemonics/language_base.h    | 22 +++++++++++++++++++++-
 2 files changed, 25 insertions(+), 5 deletions(-)

diff --git a/src/mnemonics/electrum-words.cpp b/src/mnemonics/electrum-words.cpp
index 59032055..9c42c579 100644
--- a/src/mnemonics/electrum-words.cpp
+++ b/src/mnemonics/electrum-words.cpp
@@ -99,7 +99,7 @@ namespace
       {
         if (has_checksum)
         {
-          trimmed_word = it2->substr(0, (*it1)->get_unique_prefix_length());
+          trimmed_word = Language::utf8prefix(*it2, (*it1)->get_unique_prefix_length());
           // Use the trimmed words and map
           if (trimmed_word_map.count(trimmed_word) == 0)
           {
@@ -144,7 +144,7 @@ namespace
     {
       if (it->length() > unique_prefix_length)
       {
-        trimmed_words += it->substr(0, unique_prefix_length);
+        trimmed_words += Language::utf8prefix(*it, unique_prefix_length);
       }
       else
       {
@@ -170,9 +170,9 @@ namespace
 
     std::string checksum = seed[create_checksum_index(seed, unique_prefix_length)];
 
-    std::string trimmed_checksum = checksum.length() > unique_prefix_length ? checksum.substr(0, unique_prefix_length) :
+    std::string trimmed_checksum = checksum.length() > unique_prefix_length ? Language::utf8prefix(checksum, unique_prefix_length) :
       checksum;
-    std::string trimmed_last_word = last_word.length() > unique_prefix_length ? last_word.substr(0, unique_prefix_length) :
+    std::string trimmed_last_word = last_word.length() > unique_prefix_length ? Language::utf8prefix(last_word, unique_prefix_length) :
       last_word;
     return trimmed_checksum == trimmed_last_word;
   }
diff --git a/src/mnemonics/language_base.h b/src/mnemonics/language_base.h
index 0e7d60a0..4d7cce72 100644
--- a/src/mnemonics/language_base.h
+++ b/src/mnemonics/language_base.h
@@ -45,6 +45,26 @@
  */
 namespace Language
 {
+  /*!
+   * \brief Returns a string made of (at most) the first count characters in s.
+   *        Assumes well formedness. No check is made for this.
+   * \param  s               The string from which to return the first count characters.
+   * \param  count           How many characters to return.
+   * \return                 A string consisting of the first count characters in s.
+   */
+  std::string utf8prefix(const std::string &s, size_t count)
+  {
+    std::string prefix = "";
+    const char *ptr = s.c_str();
+    while (count-- && *ptr)
+    {
+      prefix += *ptr++;
+      while (((*ptr) & 0xc0) == 0x80)
+        prefix += *ptr++;
+    }
+    return prefix;
+  }
+
   /*!
    * \class Base
    * \brief A base language class which all languages have to inherit from for
@@ -70,7 +90,7 @@ namespace Language
         (*word_map)[*it] = ii;
         if (it->length() > unique_prefix_length)
         {
-          (*trimmed_word_map)[it->substr(0, unique_prefix_length)] = ii;
+          (*trimmed_word_map)[utf8prefix(*it, unique_prefix_length)] = ii;
         }
         else
         {