fix: add tas vector param count validation

p2sr · Jul 6, 2023 · 64137af · 64137af
1 parent 6b3fc75
commit 64137af
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/Features/Tas/TasParser.cpp b/src/Features/Tas/TasParser.cpp
@@ -384,6 +384,19 @@ static Vector parseVector(const Line &line, size_t idx) {
 	else if (t1.type == TasToken::FLOAT) vec.y = t1.f;
 	else throw TasParserException(Utils::ssprintf("expected vector B %d '%s'", (int)t1.type, t1.tok.c_str()));
 
+	if (idx + 2 < line.tokens.size()) {
+		auto &t3 = line.tokens[idx + 2];
+		switch (t3.type) {
+		case TasToken::PIPE:
+			break;
+		case TasToken::INTEGER:
+		case TasToken::FLOAT:
+			throw TasParserException("invalid vector param count");
+		default:
+			throw TasParserException(Utils::ssprintf("expected end of vector, found '%s'", t3.tok.c_str()));
+		}
+	}
+
 	return vec;
 }