Files
Upsilon/apps/shared/toolbox_helpers.cpp
Émilie Feral b4a3dd10e1 [apps/shared] ToolBoxHelpers: don't overflow buffer
If the commandLength is > than the buffer size, we have to escape at
some point to avoid overflowing the buffer.
2020-07-29 09:43:51 +02:00

85 lines
2.7 KiB
C++

#include "toolbox_helpers.h"
#include <apps/i18n.h>
#include <ion/unicode/utf8_decoder.h>
#include <string.h>
#include <assert.h>
namespace Shared {
namespace ToolboxHelpers {
int CursorIndexInCommandText(const char * text) {
UTF8Decoder decoder(text);
size_t index = 0;
const char * currentPointer = text;
CodePoint codePoint = decoder.nextCodePoint();
const char * nextPointer = decoder.stringPosition();
while (codePoint != UCodePointNull) {
if (codePoint == '(' || codePoint == '\'') {
return index + 1;
}
if (codePoint == '[') {
return index;
}
index+= nextPointer - currentPointer;
currentPointer = nextPointer;
codePoint = decoder.nextCodePoint();
nextPointer = decoder.stringPosition();
}
return index;
}
void TextToInsertForCommandMessage(I18n::Message message, char * buffer, int bufferSize, bool replaceArgsWithEmptyChar) {
TextToInsertForCommandText(I18n::translate(message), -1, buffer, bufferSize, replaceArgsWithEmptyChar);
}
void TextToInsertForCommandText(const char * command, int commandLength, char * buffer, int bufferSize, bool replaceArgsWithEmptyChar) {
int index = 0;
int numberOfOpenParentheses = 0;
int numberOfOpenBrackets = 0;
bool insideQuote = false;
bool argumentAlreadyReplaced = false;
UTF8Decoder decoder(command);
CodePoint codePoint = decoder.nextCodePoint();
while (codePoint != UCodePointNull && index < bufferSize - 1 && (commandLength < 0 || (decoder.stringPosition() - command <= commandLength))) {
if (codePoint == ')') {
numberOfOpenParentheses--;
} else if (codePoint == ']') {
numberOfOpenBrackets--;
}
if ((!insideQuote || codePoint == '\'')
&& ((numberOfOpenParentheses == 0 && numberOfOpenBrackets == 0)
|| codePoint == ','
|| (numberOfOpenBrackets > 0
&& (codePoint == ','
|| codePoint == '['
|| codePoint == ']'))))
{
assert(index < bufferSize);
if (argumentAlreadyReplaced) {
argumentAlreadyReplaced = false;
}
index += UTF8Decoder::CodePointToChars(codePoint, buffer + index, bufferSize - index);
} else {
if (replaceArgsWithEmptyChar && !argumentAlreadyReplaced) {
assert(index < bufferSize);
index += UTF8Decoder::CodePointToChars(UCodePointEmpty, buffer + index, bufferSize - index);
argumentAlreadyReplaced = true;
}
}
if (codePoint == '(') {
numberOfOpenParentheses++;
} else if (codePoint == '[') {
numberOfOpenBrackets++;
} else if (codePoint == '\'') {
insideQuote = !insideQuote;
}
codePoint = decoder.nextCodePoint();
}
assert(index < bufferSize);
buffer[index] = 0;
}
}
}