forked from Public-Mirror/Textractor
60 lines
1.9 KiB
C++
60 lines
1.9 KiB
C++
#include "extensions.h"
|
|
#include <set>
|
|
#include <mutex>
|
|
#include <algorithm>
|
|
|
|
bool RemoveRepeatedChars(std::wstring& sentence)
|
|
{
|
|
unsigned int repeatNumber = 0;
|
|
wchar_t prevChar = sentence[0];
|
|
for (auto i : sentence)
|
|
if (i == prevChar) repeatNumber++;
|
|
else break;
|
|
if (repeatNumber == 1) return false;
|
|
|
|
for (int i = 0; i < sentence.size(); i += repeatNumber)
|
|
for (int j = i; j < sentence.size(); ++j)
|
|
if (sentence[j] != sentence[i])
|
|
if ((j - i) % repeatNumber != 0) return false;
|
|
else break;
|
|
|
|
// Removes every repeatNumber'th character.
|
|
sentence.erase(std::remove_if(sentence.begin(), sentence.end(), [&](const wchar_t& c) {return (&c - &*sentence.begin()) % repeatNumber != 0; }), sentence.end());
|
|
return true;
|
|
}
|
|
|
|
bool RemoveCyclicRepeats(std::wstring& sentence)
|
|
{
|
|
unsigned int realLength = 6; // If the first 6 characters appear later on, there's probably a repetition issue.
|
|
if (sentence.size() < realLength) return false;
|
|
wchar_t realSentence[2000] = {};
|
|
memcpy(realSentence, sentence.c_str(), realLength * sizeof(wchar_t));
|
|
while (wcsstr(sentence.c_str() + realLength, realSentence))
|
|
{
|
|
realSentence[realLength] = sentence[realLength];
|
|
if (++realLength >= 2000) return false;
|
|
}
|
|
if (realLength > 7)
|
|
{
|
|
sentence = std::wstring(realSentence);
|
|
RemoveCyclicRepeats(sentence);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool RemoveRepeatedSentences(std::wstring& sentence, int threadHandle)
|
|
{
|
|
static std::set<std::pair<int, std::wstring>> seenSentences;
|
|
static std::mutex m;
|
|
std::lock_guard<std::mutex> l(m);
|
|
if (seenSentences.count({ threadHandle, sentence }) != 0) throw std::exception();
|
|
seenSentences.insert({ threadHandle, sentence });
|
|
return false;
|
|
}
|
|
|
|
bool ProcessSentence(std::wstring& sentence, const InfoForExtension* miscInfo)
|
|
{
|
|
if (GetProperty("hook address", miscInfo) == -1) return false;
|
|
return RemoveRepeatedChars(sentence) | RemoveCyclicRepeats(sentence) | RemoveRepeatedSentences(sentence, GetProperty("thread handle", miscInfo));
|
|
} |