This commit is contained in:
Akash Mozumdar 2021-01-31 07:58:23 -07:00
parent 4494691117
commit ea46617ce9

View File

@ -17,12 +17,12 @@ extern "C" __declspec(dllexport) wchar_t* OnNewSentence(wchar_t* sentence, const
{ {
try try
{ {
std::wstring sentenceStr(sentence); std::wstring sentenceCopy(sentence);
int origLength = sentenceStr.size(); int oldSize = sentenceCopy.size();
if (ProcessSentence(sentenceStr, SentenceInfo{ sentenceInfo })) if (ProcessSentence(sentenceCopy, SentenceInfo{ sentenceInfo }))
{ {
if (sentenceStr.size() > origLength) sentence = (wchar_t*)HeapReAlloc(GetProcessHeap(), HEAP_GENERATE_EXCEPTIONS, sentence, (sentenceStr.size() + 1) * sizeof(wchar_t)); if (sentenceCopy.size() > oldSize) sentence = (wchar_t*)HeapReAlloc(GetProcessHeap(), HEAP_GENERATE_EXCEPTIONS, sentence, (sentenceCopy.size() + 1) * sizeof(wchar_t));
wcscpy_s(sentence, sentenceStr.size() + 1, sentenceStr.c_str()); wcscpy_s(sentence, sentenceCopy.size() + 1, sentenceCopy.c_str());
} }
} }
catch (SKIP) catch (SKIP)