diff --git a/src/script/C4AulParse.cpp b/src/script/C4AulParse.cpp index a7f5c4f7a..42a22138c 100644 --- a/src/script/C4AulParse.cpp +++ b/src/script/C4AulParse.cpp @@ -778,12 +778,9 @@ void C4AulParse::UnexpectedToken(const char * Expected) void C4AulParse::Parse_WarningPragma() { assert(SEqual2(TokenSPos, C4AUL_Warning)); - assert(std::isspace(TokenSPos[sizeof(C4AUL_Warning) - 1])); - - - // Read parameters in to string buffer. The sizeof() includes the terminating \0, but - // that's okay because we need to skip (at least) one whitespace character anyway. - std::string line(TokenSPos + sizeof(C4AUL_Warning), SPos); + + // Read parameters in to string buffer. + std::string line(TokenSPos + sizeof(C4AUL_Warning) - 1, SPos); auto end = line.end(); auto cursor = std::find_if_not(begin(line), end, IsWhiteSpace); diff --git a/tests/aul/AulSyntaxTest.cpp b/tests/aul/AulSyntaxTest.cpp index 52d7d445c..7dab56a92 100644 --- a/tests/aul/AulSyntaxTest.cpp +++ b/tests/aul/AulSyntaxTest.cpp @@ -88,6 +88,14 @@ TEST(AulSyntaxTest, ParseSyntaxErrors) EXPECT_THROW(ParseStatement("func Main() {}"), C4AulParseError); } +TEST(AulSyntaxTest, PragmaWarningWithInsufficientData) +{ + EXPECT_THROW(ParseScript("#warning"), C4AulParseError); + EXPECT_THROW(ParseScript("#warning\n"), C4AulParseError); + EXPECT_THROW(ParseScript("#warning "), C4AulParseError); + EXPECT_THROW(ParseScript("#warning \n"), C4AulParseError); +} + using namespace ::aul::ast; TEST(AulSyntaxTest, ParseLiterals) {