15cd3d5c8SMarek Sokolowski //===-- ResourceScriptParser.cpp --------------------------------*- C++-*-===//
25cd3d5c8SMarek Sokolowski //
35cd3d5c8SMarek Sokolowski //                     The LLVM Compiler Infrastructure
45cd3d5c8SMarek Sokolowski //
55cd3d5c8SMarek Sokolowski // This file is distributed under the University of Illinois Open Source
65cd3d5c8SMarek Sokolowski // License. See LICENSE.TXT for details.
75cd3d5c8SMarek Sokolowski //
85cd3d5c8SMarek Sokolowski //===---------------------------------------------------------------------===//
95cd3d5c8SMarek Sokolowski //
105cd3d5c8SMarek Sokolowski // This implements the parser defined in ResourceScriptParser.h.
115cd3d5c8SMarek Sokolowski //
125cd3d5c8SMarek Sokolowski //===---------------------------------------------------------------------===//
135cd3d5c8SMarek Sokolowski 
145cd3d5c8SMarek Sokolowski #include "ResourceScriptParser.h"
155cd3d5c8SMarek Sokolowski 
165cd3d5c8SMarek Sokolowski // Take an expression returning llvm::Error and forward the error if it exists.
175cd3d5c8SMarek Sokolowski #define RETURN_IF_ERROR(Expr)                                                  \
185cd3d5c8SMarek Sokolowski   if (auto Err = (Expr))                                                       \
195cd3d5c8SMarek Sokolowski     return std::move(Err);
205cd3d5c8SMarek Sokolowski 
215cd3d5c8SMarek Sokolowski // Take an expression returning llvm::Expected<T> and assign it to Var or
225cd3d5c8SMarek Sokolowski // forward the error out of the function.
235cd3d5c8SMarek Sokolowski #define ASSIGN_OR_RETURN(Var, Expr)                                            \
245cd3d5c8SMarek Sokolowski   auto Var = (Expr);                                                           \
255cd3d5c8SMarek Sokolowski   if (!Var)                                                                    \
265cd3d5c8SMarek Sokolowski     return Var.takeError();
275cd3d5c8SMarek Sokolowski 
285cd3d5c8SMarek Sokolowski namespace llvm {
295cd3d5c8SMarek Sokolowski namespace rc {
305cd3d5c8SMarek Sokolowski 
31*514b7105SZachary Turner RCParser::ParserError::ParserError(const Twine &Expected, const LocIter CurLoc,
325cd3d5c8SMarek Sokolowski                                    const LocIter End)
335cd3d5c8SMarek Sokolowski     : ErrorLoc(CurLoc), FileEnd(End) {
345cd3d5c8SMarek Sokolowski   CurMessage = "Error parsing file: expected " + Expected.str() + ", got " +
355cd3d5c8SMarek Sokolowski                (CurLoc == End ? "<EOF>" : CurLoc->value()).str();
365cd3d5c8SMarek Sokolowski }
375cd3d5c8SMarek Sokolowski 
385cd3d5c8SMarek Sokolowski char RCParser::ParserError::ID = 0;
395cd3d5c8SMarek Sokolowski 
40*514b7105SZachary Turner RCParser::RCParser(std::vector<RCToken> TokenList)
415cd3d5c8SMarek Sokolowski     : Tokens(std::move(TokenList)), CurLoc(Tokens.begin()), End(Tokens.end()) {}
425cd3d5c8SMarek Sokolowski 
435cd3d5c8SMarek Sokolowski bool RCParser::isEof() const { return CurLoc == End; }
445cd3d5c8SMarek Sokolowski 
455cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseSingleResource() {
465cd3d5c8SMarek Sokolowski   // The first thing we read is usually a resource's name. However, in some
475cd3d5c8SMarek Sokolowski   // cases (LANGUAGE and STRINGTABLE) the resources don't have their names
485cd3d5c8SMarek Sokolowski   // and the first token to be read is the type.
495cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(NameToken, readTypeOrName());
505cd3d5c8SMarek Sokolowski 
515cd3d5c8SMarek Sokolowski   if (NameToken->equalsLower("LANGUAGE"))
525cd3d5c8SMarek Sokolowski     return parseLanguageResource();
535cd3d5c8SMarek Sokolowski   else if (NameToken->equalsLower("STRINGTABLE"))
545cd3d5c8SMarek Sokolowski     return parseStringTableResource();
555cd3d5c8SMarek Sokolowski 
565cd3d5c8SMarek Sokolowski   // If it's not an unnamed resource, what we've just read is a name. Now,
575cd3d5c8SMarek Sokolowski   // read resource type;
585cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(TypeToken, readTypeOrName());
595cd3d5c8SMarek Sokolowski 
605cd3d5c8SMarek Sokolowski   ParseType Result = std::unique_ptr<RCResource>();
615cd3d5c8SMarek Sokolowski   (void)!Result;
625cd3d5c8SMarek Sokolowski 
637f110527SMarek Sokolowski   if (TypeToken->equalsLower("ACCELERATORS"))
647f110527SMarek Sokolowski     Result = parseAcceleratorsResource();
657f110527SMarek Sokolowski   else if (TypeToken->equalsLower("CURSOR"))
6672aa937eSMarek Sokolowski     Result = parseCursorResource();
674ac54d93SMarek Sokolowski   else if (TypeToken->equalsLower("DIALOG"))
684ac54d93SMarek Sokolowski     Result = parseDialogResource(false);
694ac54d93SMarek Sokolowski   else if (TypeToken->equalsLower("DIALOGEX"))
704ac54d93SMarek Sokolowski     Result = parseDialogResource(true);
7172aa937eSMarek Sokolowski   else if (TypeToken->equalsLower("ICON"))
725cd3d5c8SMarek Sokolowski     Result = parseIconResource();
7372aa937eSMarek Sokolowski   else if (TypeToken->equalsLower("HTML"))
7472aa937eSMarek Sokolowski     Result = parseHTMLResource();
7599ecb0ebSMarek Sokolowski   else if (TypeToken->equalsLower("MENU"))
7699ecb0ebSMarek Sokolowski     Result = parseMenuResource();
77fb74cb1eSMarek Sokolowski   else if (TypeToken->equalsLower("VERSIONINFO"))
78fb74cb1eSMarek Sokolowski     Result = parseVersionInfoResource();
795cd3d5c8SMarek Sokolowski   else
80b5f39a05SMarek Sokolowski     Result = parseUserDefinedResource(*TypeToken);
815cd3d5c8SMarek Sokolowski 
825cd3d5c8SMarek Sokolowski   if (Result)
835cd3d5c8SMarek Sokolowski     (*Result)->setName(*NameToken);
845cd3d5c8SMarek Sokolowski 
855cd3d5c8SMarek Sokolowski   return Result;
865cd3d5c8SMarek Sokolowski }
875cd3d5c8SMarek Sokolowski 
885cd3d5c8SMarek Sokolowski bool RCParser::isNextTokenKind(Kind TokenKind) const {
895cd3d5c8SMarek Sokolowski   return !isEof() && look().kind() == TokenKind;
905cd3d5c8SMarek Sokolowski }
915cd3d5c8SMarek Sokolowski 
925cd3d5c8SMarek Sokolowski const RCToken &RCParser::look() const {
935cd3d5c8SMarek Sokolowski   assert(!isEof());
945cd3d5c8SMarek Sokolowski   return *CurLoc;
955cd3d5c8SMarek Sokolowski }
965cd3d5c8SMarek Sokolowski 
975cd3d5c8SMarek Sokolowski const RCToken &RCParser::read() {
985cd3d5c8SMarek Sokolowski   assert(!isEof());
995cd3d5c8SMarek Sokolowski   return *CurLoc++;
1005cd3d5c8SMarek Sokolowski }
1015cd3d5c8SMarek Sokolowski 
1025cd3d5c8SMarek Sokolowski void RCParser::consume() {
1035cd3d5c8SMarek Sokolowski   assert(!isEof());
1045cd3d5c8SMarek Sokolowski   CurLoc++;
1055cd3d5c8SMarek Sokolowski }
1065cd3d5c8SMarek Sokolowski 
1077e89ee7fSMarek Sokolowski // An integer description might consist of a single integer or
1087e89ee7fSMarek Sokolowski // an arithmetic expression evaluating to the integer. The expressions
1097e89ee7fSMarek Sokolowski // can contain the following tokens: <int> ( ) + - | & ~. Their meaning
1107e89ee7fSMarek Sokolowski // is the same as in C++.
1117e89ee7fSMarek Sokolowski // The operators in the original RC implementation have the following
1127e89ee7fSMarek Sokolowski // precedence:
1137e89ee7fSMarek Sokolowski //   1) Unary operators (- ~),
1147e89ee7fSMarek Sokolowski //   2) Binary operators (+ - & |), with no precedence.
1157e89ee7fSMarek Sokolowski //
1167e89ee7fSMarek Sokolowski // The following grammar is used to parse the expressions Exp1:
1177e89ee7fSMarek Sokolowski //   Exp1 ::= Exp2 || Exp1 + Exp2 || Exp1 - Exp2 || Exp1 | Exp2 || Exp1 & Exp2
1187e89ee7fSMarek Sokolowski //   Exp2 ::= -Exp2 || ~Exp2 || Int || (Exp1).
1197e89ee7fSMarek Sokolowski // (More conveniently, Exp1 is a non-empty sequence of Exp2 expressions,
1207e89ee7fSMarek Sokolowski // separated by binary operators.)
1217e89ee7fSMarek Sokolowski //
1227e89ee7fSMarek Sokolowski // Expressions of type Exp1 are read by parseIntExpr1(Inner) method, while Exp2
1237e89ee7fSMarek Sokolowski // is read by parseIntExpr2().
1247e89ee7fSMarek Sokolowski //
1257e89ee7fSMarek Sokolowski // The original Microsoft tool handles multiple unary operators incorrectly.
1267e89ee7fSMarek Sokolowski // For example, in 16-bit little-endian integers:
1277e89ee7fSMarek Sokolowski //    1 => 01 00, -1 => ff ff, --1 => ff ff, ---1 => 01 00;
1287e89ee7fSMarek Sokolowski //    1 => 01 00, ~1 => fe ff, ~~1 => fd ff, ~~~1 => fc ff.
1297e89ee7fSMarek Sokolowski // Our implementation differs from the original one and handles these
1307e89ee7fSMarek Sokolowski // operators correctly:
1317e89ee7fSMarek Sokolowski //    1 => 01 00, -1 => ff ff, --1 => 01 00, ---1 => ff ff;
1327e89ee7fSMarek Sokolowski //    1 => 01 00, ~1 => fe ff, ~~1 => 01 00, ~~~1 => fe ff.
1337e89ee7fSMarek Sokolowski 
13407bc04ffSZachary Turner Expected<RCInt> RCParser::readInt() { return parseIntExpr1(); }
1357e89ee7fSMarek Sokolowski 
13607bc04ffSZachary Turner Expected<RCInt> RCParser::parseIntExpr1() {
1377e89ee7fSMarek Sokolowski   // Exp1 ::= Exp2 || Exp1 + Exp2 || Exp1 - Exp2 || Exp1 | Exp2 || Exp1 & Exp2.
1387e89ee7fSMarek Sokolowski   ASSIGN_OR_RETURN(FirstResult, parseIntExpr2());
13907bc04ffSZachary Turner   RCInt Result = *FirstResult;
1407e89ee7fSMarek Sokolowski 
1417e89ee7fSMarek Sokolowski   while (!isEof() && look().isBinaryOp()) {
1427e89ee7fSMarek Sokolowski     auto OpToken = read();
1437e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(NextResult, parseIntExpr2());
1447e89ee7fSMarek Sokolowski 
1457e89ee7fSMarek Sokolowski     switch (OpToken.kind()) {
1467e89ee7fSMarek Sokolowski     case Kind::Plus:
1477e89ee7fSMarek Sokolowski       Result += *NextResult;
1487e89ee7fSMarek Sokolowski       break;
1497e89ee7fSMarek Sokolowski 
1507e89ee7fSMarek Sokolowski     case Kind::Minus:
1517e89ee7fSMarek Sokolowski       Result -= *NextResult;
1527e89ee7fSMarek Sokolowski       break;
1537e89ee7fSMarek Sokolowski 
1547e89ee7fSMarek Sokolowski     case Kind::Pipe:
1557e89ee7fSMarek Sokolowski       Result |= *NextResult;
1567e89ee7fSMarek Sokolowski       break;
1577e89ee7fSMarek Sokolowski 
1587e89ee7fSMarek Sokolowski     case Kind::Amp:
1597e89ee7fSMarek Sokolowski       Result &= *NextResult;
1607e89ee7fSMarek Sokolowski       break;
1617e89ee7fSMarek Sokolowski 
1627e89ee7fSMarek Sokolowski     default:
1637e89ee7fSMarek Sokolowski       llvm_unreachable("Already processed all binary ops.");
1647e89ee7fSMarek Sokolowski     }
1657e89ee7fSMarek Sokolowski   }
1667e89ee7fSMarek Sokolowski 
1677e89ee7fSMarek Sokolowski   return Result;
1687e89ee7fSMarek Sokolowski }
1697e89ee7fSMarek Sokolowski 
17007bc04ffSZachary Turner Expected<RCInt> RCParser::parseIntExpr2() {
1717e89ee7fSMarek Sokolowski   // Exp2 ::= -Exp2 || ~Exp2 || Int || (Exp1).
1727e89ee7fSMarek Sokolowski   static const char ErrorMsg[] = "'-', '~', integer or '('";
1737e89ee7fSMarek Sokolowski 
1747e89ee7fSMarek Sokolowski   if (isEof())
1757e89ee7fSMarek Sokolowski     return getExpectedError(ErrorMsg);
1767e89ee7fSMarek Sokolowski 
1777e89ee7fSMarek Sokolowski   switch (look().kind()) {
1787e89ee7fSMarek Sokolowski   case Kind::Minus: {
1797e89ee7fSMarek Sokolowski     consume();
1807e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(Result, parseIntExpr2());
1817e89ee7fSMarek Sokolowski     return -(*Result);
1827e89ee7fSMarek Sokolowski   }
1837e89ee7fSMarek Sokolowski 
1847e89ee7fSMarek Sokolowski   case Kind::Tilde: {
1857e89ee7fSMarek Sokolowski     consume();
1867e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(Result, parseIntExpr2());
1877e89ee7fSMarek Sokolowski     return ~(*Result);
1887e89ee7fSMarek Sokolowski   }
1897e89ee7fSMarek Sokolowski 
1907e89ee7fSMarek Sokolowski   case Kind::Int:
19107bc04ffSZachary Turner     return RCInt(read());
1927e89ee7fSMarek Sokolowski 
1937e89ee7fSMarek Sokolowski   case Kind::LeftParen: {
1947e89ee7fSMarek Sokolowski     consume();
1957e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(Result, parseIntExpr1());
1967e89ee7fSMarek Sokolowski     RETURN_IF_ERROR(consumeType(Kind::RightParen));
1977e89ee7fSMarek Sokolowski     return *Result;
1987e89ee7fSMarek Sokolowski   }
1997e89ee7fSMarek Sokolowski 
2007e89ee7fSMarek Sokolowski   default:
2017e89ee7fSMarek Sokolowski     return getExpectedError(ErrorMsg);
2027e89ee7fSMarek Sokolowski   }
2035cd3d5c8SMarek Sokolowski }
2045cd3d5c8SMarek Sokolowski 
2055cd3d5c8SMarek Sokolowski Expected<StringRef> RCParser::readString() {
2065cd3d5c8SMarek Sokolowski   if (!isNextTokenKind(Kind::String))
2075cd3d5c8SMarek Sokolowski     return getExpectedError("string");
2085cd3d5c8SMarek Sokolowski   return read().value();
2095cd3d5c8SMarek Sokolowski }
2105cd3d5c8SMarek Sokolowski 
2115cd3d5c8SMarek Sokolowski Expected<StringRef> RCParser::readIdentifier() {
2125cd3d5c8SMarek Sokolowski   if (!isNextTokenKind(Kind::Identifier))
2135cd3d5c8SMarek Sokolowski     return getExpectedError("identifier");
2145cd3d5c8SMarek Sokolowski   return read().value();
2155cd3d5c8SMarek Sokolowski }
2165cd3d5c8SMarek Sokolowski 
2177f110527SMarek Sokolowski Expected<IntOrString> RCParser::readIntOrString() {
2187f110527SMarek Sokolowski   if (!isNextTokenKind(Kind::Int) && !isNextTokenKind(Kind::String))
2197f110527SMarek Sokolowski     return getExpectedError("int or string");
2207f110527SMarek Sokolowski   return IntOrString(read());
2217f110527SMarek Sokolowski }
2227f110527SMarek Sokolowski 
2235cd3d5c8SMarek Sokolowski Expected<IntOrString> RCParser::readTypeOrName() {
2245cd3d5c8SMarek Sokolowski   // We suggest that the correct resource name or type should be either an
2255cd3d5c8SMarek Sokolowski   // identifier or an integer. The original RC tool is much more liberal.
2265cd3d5c8SMarek Sokolowski   if (!isNextTokenKind(Kind::Identifier) && !isNextTokenKind(Kind::Int))
2275cd3d5c8SMarek Sokolowski     return getExpectedError("int or identifier");
2287f110527SMarek Sokolowski   return IntOrString(read());
2295cd3d5c8SMarek Sokolowski }
2305cd3d5c8SMarek Sokolowski 
2315cd3d5c8SMarek Sokolowski Error RCParser::consumeType(Kind TokenKind) {
2325cd3d5c8SMarek Sokolowski   if (isNextTokenKind(TokenKind)) {
2335cd3d5c8SMarek Sokolowski     consume();
2345cd3d5c8SMarek Sokolowski     return Error::success();
2355cd3d5c8SMarek Sokolowski   }
2365cd3d5c8SMarek Sokolowski 
2375cd3d5c8SMarek Sokolowski   switch (TokenKind) {
2385cd3d5c8SMarek Sokolowski #define TOKEN(TokenName)                                                       \
2395cd3d5c8SMarek Sokolowski   case Kind::TokenName:                                                        \
2405cd3d5c8SMarek Sokolowski     return getExpectedError(#TokenName);
2415cd3d5c8SMarek Sokolowski #define SHORT_TOKEN(TokenName, TokenCh)                                        \
2425cd3d5c8SMarek Sokolowski   case Kind::TokenName:                                                        \
2435cd3d5c8SMarek Sokolowski     return getExpectedError(#TokenCh);
2445cd3d5c8SMarek Sokolowski #include "ResourceScriptTokenList.h"
2455cd3d5c8SMarek Sokolowski #undef SHORT_TOKEN
2465cd3d5c8SMarek Sokolowski #undef TOKEN
2475cd3d5c8SMarek Sokolowski   }
2485cd3d5c8SMarek Sokolowski 
2495cd3d5c8SMarek Sokolowski   llvm_unreachable("All case options exhausted.");
2505cd3d5c8SMarek Sokolowski }
2515cd3d5c8SMarek Sokolowski 
2525cd3d5c8SMarek Sokolowski bool RCParser::consumeOptionalType(Kind TokenKind) {
2535cd3d5c8SMarek Sokolowski   if (isNextTokenKind(TokenKind)) {
2545cd3d5c8SMarek Sokolowski     consume();
2555cd3d5c8SMarek Sokolowski     return true;
2565cd3d5c8SMarek Sokolowski   }
2575cd3d5c8SMarek Sokolowski 
2585cd3d5c8SMarek Sokolowski   return false;
2595cd3d5c8SMarek Sokolowski }
2605cd3d5c8SMarek Sokolowski 
26107bc04ffSZachary Turner Expected<SmallVector<RCInt, 8>> RCParser::readIntsWithCommas(size_t MinCount,
26207bc04ffSZachary Turner                                                              size_t MaxCount) {
2635cd3d5c8SMarek Sokolowski   assert(MinCount <= MaxCount);
2645cd3d5c8SMarek Sokolowski 
26507bc04ffSZachary Turner   SmallVector<RCInt, 8> Result;
2665cd3d5c8SMarek Sokolowski 
2675cd3d5c8SMarek Sokolowski   auto FailureHandler =
26807bc04ffSZachary Turner       [&](llvm::Error Err) -> Expected<SmallVector<RCInt, 8>> {
2695cd3d5c8SMarek Sokolowski     if (Result.size() < MinCount)
2705cd3d5c8SMarek Sokolowski       return std::move(Err);
2715cd3d5c8SMarek Sokolowski     consumeError(std::move(Err));
2725cd3d5c8SMarek Sokolowski     return Result;
2735cd3d5c8SMarek Sokolowski   };
2745cd3d5c8SMarek Sokolowski 
2755cd3d5c8SMarek Sokolowski   for (size_t i = 0; i < MaxCount; ++i) {
2765cd3d5c8SMarek Sokolowski     // Try to read a comma unless we read the first token.
2775cd3d5c8SMarek Sokolowski     // Sometimes RC tool requires them and sometimes not. We decide to
2785cd3d5c8SMarek Sokolowski     // always require them.
2795cd3d5c8SMarek Sokolowski     if (i >= 1) {
2805cd3d5c8SMarek Sokolowski       if (auto CommaError = consumeType(Kind::Comma))
2815cd3d5c8SMarek Sokolowski         return FailureHandler(std::move(CommaError));
2825cd3d5c8SMarek Sokolowski     }
2835cd3d5c8SMarek Sokolowski 
2845cd3d5c8SMarek Sokolowski     if (auto IntResult = readInt())
2855cd3d5c8SMarek Sokolowski       Result.push_back(*IntResult);
2865cd3d5c8SMarek Sokolowski     else
2875cd3d5c8SMarek Sokolowski       return FailureHandler(IntResult.takeError());
2885cd3d5c8SMarek Sokolowski   }
2895cd3d5c8SMarek Sokolowski 
2905cd3d5c8SMarek Sokolowski   return std::move(Result);
2915cd3d5c8SMarek Sokolowski }
2925cd3d5c8SMarek Sokolowski 
293c75a087cSMarek Sokolowski Expected<uint32_t> RCParser::parseFlags(ArrayRef<StringRef> FlagDesc,
294c75a087cSMarek Sokolowski                                         ArrayRef<uint32_t> FlagValues) {
2957f110527SMarek Sokolowski   assert(!FlagDesc.empty());
296c75a087cSMarek Sokolowski   assert(FlagDesc.size() == FlagValues.size());
2977f110527SMarek Sokolowski 
2987f110527SMarek Sokolowski   uint32_t Result = 0;
2997f110527SMarek Sokolowski   while (isNextTokenKind(Kind::Comma)) {
3007f110527SMarek Sokolowski     consume();
3017f110527SMarek Sokolowski     ASSIGN_OR_RETURN(FlagResult, readIdentifier());
3027f110527SMarek Sokolowski     bool FoundFlag = false;
3037f110527SMarek Sokolowski 
3047f110527SMarek Sokolowski     for (size_t FlagId = 0; FlagId < FlagDesc.size(); ++FlagId) {
3057f110527SMarek Sokolowski       if (!FlagResult->equals_lower(FlagDesc[FlagId]))
3067f110527SMarek Sokolowski         continue;
3077f110527SMarek Sokolowski 
308c75a087cSMarek Sokolowski       Result |= FlagValues[FlagId];
3097f110527SMarek Sokolowski       FoundFlag = true;
3107f110527SMarek Sokolowski       break;
3117f110527SMarek Sokolowski     }
3127f110527SMarek Sokolowski 
3137f110527SMarek Sokolowski     if (!FoundFlag)
3147f110527SMarek Sokolowski       return getExpectedError(join(FlagDesc, "/"), true);
3157f110527SMarek Sokolowski   }
3167f110527SMarek Sokolowski 
3177f110527SMarek Sokolowski   return Result;
3187f110527SMarek Sokolowski }
3197f110527SMarek Sokolowski 
320420090afSZachary Turner Expected<OptionalStmtList>
321420090afSZachary Turner RCParser::parseOptionalStatements(OptStmtType StmtsType) {
3225cd3d5c8SMarek Sokolowski   OptionalStmtList Result;
3235cd3d5c8SMarek Sokolowski 
3245cd3d5c8SMarek Sokolowski   // The last statement is always followed by the start of the block.
3255cd3d5c8SMarek Sokolowski   while (!isNextTokenKind(Kind::BlockBegin)) {
326420090afSZachary Turner     ASSIGN_OR_RETURN(SingleParse, parseSingleOptionalStatement(StmtsType));
3275cd3d5c8SMarek Sokolowski     Result.addStmt(std::move(*SingleParse));
3285cd3d5c8SMarek Sokolowski   }
3295cd3d5c8SMarek Sokolowski 
3305cd3d5c8SMarek Sokolowski   return std::move(Result);
3315cd3d5c8SMarek Sokolowski }
3325cd3d5c8SMarek Sokolowski 
3335cd3d5c8SMarek Sokolowski Expected<std::unique_ptr<OptionalStmt>>
334420090afSZachary Turner RCParser::parseSingleOptionalStatement(OptStmtType StmtsType) {
3355cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(TypeToken, readIdentifier());
3365cd3d5c8SMarek Sokolowski   if (TypeToken->equals_lower("CHARACTERISTICS"))
3375cd3d5c8SMarek Sokolowski     return parseCharacteristicsStmt();
3384ac54d93SMarek Sokolowski   if (TypeToken->equals_lower("LANGUAGE"))
3395cd3d5c8SMarek Sokolowski     return parseLanguageStmt();
3404ac54d93SMarek Sokolowski   if (TypeToken->equals_lower("VERSION"))
3415cd3d5c8SMarek Sokolowski     return parseVersionStmt();
3424ac54d93SMarek Sokolowski 
343420090afSZachary Turner   if (StmtsType != OptStmtType::BasicStmt) {
3444ac54d93SMarek Sokolowski     if (TypeToken->equals_lower("CAPTION"))
3454ac54d93SMarek Sokolowski       return parseCaptionStmt();
3464ac54d93SMarek Sokolowski     if (TypeToken->equals_lower("FONT"))
347420090afSZachary Turner       return parseFontStmt(StmtsType);
3484ac54d93SMarek Sokolowski     if (TypeToken->equals_lower("STYLE"))
3494ac54d93SMarek Sokolowski       return parseStyleStmt();
3504ac54d93SMarek Sokolowski   }
3514ac54d93SMarek Sokolowski 
3525cd3d5c8SMarek Sokolowski   return getExpectedError("optional statement type, BEGIN or '{'",
3535cd3d5c8SMarek Sokolowski                           /* IsAlreadyRead = */ true);
3545cd3d5c8SMarek Sokolowski }
3555cd3d5c8SMarek Sokolowski 
3565cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseLanguageResource() {
3575cd3d5c8SMarek Sokolowski   // Read LANGUAGE as an optional statement. If it's read correctly, we can
3585cd3d5c8SMarek Sokolowski   // upcast it to RCResource.
3595cd3d5c8SMarek Sokolowski   return parseLanguageStmt();
3605cd3d5c8SMarek Sokolowski }
3615cd3d5c8SMarek Sokolowski 
3627f110527SMarek Sokolowski RCParser::ParseType RCParser::parseAcceleratorsResource() {
3637f110527SMarek Sokolowski   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements());
3647f110527SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
3657f110527SMarek Sokolowski 
3664a765da3SMarek Sokolowski   auto Accels =
3674a765da3SMarek Sokolowski       llvm::make_unique<AcceleratorsResource>(std::move(*OptStatements));
3687f110527SMarek Sokolowski 
3697f110527SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
3707f110527SMarek Sokolowski     ASSIGN_OR_RETURN(EventResult, readIntOrString());
3717f110527SMarek Sokolowski     RETURN_IF_ERROR(consumeType(Kind::Comma));
3727f110527SMarek Sokolowski     ASSIGN_OR_RETURN(IDResult, readInt());
373c75a087cSMarek Sokolowski     ASSIGN_OR_RETURN(
374c75a087cSMarek Sokolowski         FlagsResult,
375c75a087cSMarek Sokolowski         parseFlags(AcceleratorsResource::Accelerator::OptionsStr,
376c75a087cSMarek Sokolowski                    AcceleratorsResource::Accelerator::OptionsFlags));
3777f110527SMarek Sokolowski     Accels->addAccelerator(*EventResult, *IDResult, *FlagsResult);
3787f110527SMarek Sokolowski   }
3797f110527SMarek Sokolowski 
3807f110527SMarek Sokolowski   return std::move(Accels);
3817f110527SMarek Sokolowski }
3827f110527SMarek Sokolowski 
38372aa937eSMarek Sokolowski RCParser::ParseType RCParser::parseCursorResource() {
38472aa937eSMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
3854a765da3SMarek Sokolowski   return llvm::make_unique<CursorResource>(*Arg);
38672aa937eSMarek Sokolowski }
38772aa937eSMarek Sokolowski 
3884ac54d93SMarek Sokolowski RCParser::ParseType RCParser::parseDialogResource(bool IsExtended) {
3894ac54d93SMarek Sokolowski   // Dialog resources have the following format of the arguments:
3904ac54d93SMarek Sokolowski   //  DIALOG:   x, y, width, height [opt stmts...] {controls...}
3914ac54d93SMarek Sokolowski   //  DIALOGEX: x, y, width, height [, helpID] [opt stmts...] {controls...}
3924ac54d93SMarek Sokolowski   // These are very similar, so we parse them together.
3934ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(LocResult, readIntsWithCommas(4, 4));
3944ac54d93SMarek Sokolowski 
3954ac54d93SMarek Sokolowski   uint32_t HelpID = 0; // When HelpID is unset, it's assumed to be 0.
3964ac54d93SMarek Sokolowski   if (IsExtended && consumeOptionalType(Kind::Comma)) {
3974ac54d93SMarek Sokolowski     ASSIGN_OR_RETURN(HelpIDResult, readInt());
3984ac54d93SMarek Sokolowski     HelpID = *HelpIDResult;
3994ac54d93SMarek Sokolowski   }
4004ac54d93SMarek Sokolowski 
401420090afSZachary Turner   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements(
402420090afSZachary Turner                                       IsExtended ? OptStmtType::DialogExStmt
403420090afSZachary Turner                                                  : OptStmtType::DialogStmt));
4044ac54d93SMarek Sokolowski 
4054ac54d93SMarek Sokolowski   assert(isNextTokenKind(Kind::BlockBegin) &&
4064ac54d93SMarek Sokolowski          "parseOptionalStatements, when successful, halts on BlockBegin.");
4074ac54d93SMarek Sokolowski   consume();
4084ac54d93SMarek Sokolowski 
4094a765da3SMarek Sokolowski   auto Dialog = llvm::make_unique<DialogResource>(
4104ac54d93SMarek Sokolowski       (*LocResult)[0], (*LocResult)[1], (*LocResult)[2], (*LocResult)[3],
4114ac54d93SMarek Sokolowski       HelpID, std::move(*OptStatements), IsExtended);
4124ac54d93SMarek Sokolowski 
4134ac54d93SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
4144ac54d93SMarek Sokolowski     ASSIGN_OR_RETURN(ControlDefResult, parseControl());
4154ac54d93SMarek Sokolowski     Dialog->addControl(std::move(*ControlDefResult));
4164ac54d93SMarek Sokolowski   }
4174ac54d93SMarek Sokolowski 
4184ac54d93SMarek Sokolowski   return std::move(Dialog);
4194ac54d93SMarek Sokolowski }
4204ac54d93SMarek Sokolowski 
421b5f39a05SMarek Sokolowski RCParser::ParseType RCParser::parseUserDefinedResource(IntOrString Type) {
422b5f39a05SMarek Sokolowski   if (isEof())
423b5f39a05SMarek Sokolowski     return getExpectedError("filename, '{' or BEGIN");
424b5f39a05SMarek Sokolowski 
425b5f39a05SMarek Sokolowski   // Check if this is a file resource.
426b5f39a05SMarek Sokolowski   if (look().kind() == Kind::String)
4274a765da3SMarek Sokolowski     return llvm::make_unique<UserDefinedResource>(Type, read().value());
428b5f39a05SMarek Sokolowski 
429b5f39a05SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
430b5f39a05SMarek Sokolowski   std::vector<IntOrString> Data;
431b5f39a05SMarek Sokolowski 
432b5f39a05SMarek Sokolowski   // Consume comma before each consecutive token except the first one.
433b5f39a05SMarek Sokolowski   bool ConsumeComma = false;
434b5f39a05SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
435b5f39a05SMarek Sokolowski     if (ConsumeComma)
436b5f39a05SMarek Sokolowski       RETURN_IF_ERROR(consumeType(Kind::Comma));
437b5f39a05SMarek Sokolowski     ConsumeComma = true;
438b5f39a05SMarek Sokolowski 
439b5f39a05SMarek Sokolowski     ASSIGN_OR_RETURN(Item, readIntOrString());
440b5f39a05SMarek Sokolowski     Data.push_back(*Item);
441b5f39a05SMarek Sokolowski   }
442b5f39a05SMarek Sokolowski 
4434a765da3SMarek Sokolowski   return llvm::make_unique<UserDefinedResource>(Type, std::move(Data));
444b5f39a05SMarek Sokolowski }
445b5f39a05SMarek Sokolowski 
446fb74cb1eSMarek Sokolowski RCParser::ParseType RCParser::parseVersionInfoResource() {
447fb74cb1eSMarek Sokolowski   ASSIGN_OR_RETURN(FixedResult, parseVersionInfoFixed());
448fb74cb1eSMarek Sokolowski   ASSIGN_OR_RETURN(BlockResult, parseVersionInfoBlockContents(StringRef()));
44999ead70fSMarek Sokolowski   return llvm::make_unique<VersionInfoResource>(std::move(**BlockResult),
450fb74cb1eSMarek Sokolowski                                                 std::move(*FixedResult));
451fb74cb1eSMarek Sokolowski }
452fb74cb1eSMarek Sokolowski 
4534ac54d93SMarek Sokolowski Expected<Control> RCParser::parseControl() {
4544ac54d93SMarek Sokolowski   // Each control definition (except CONTROL) follows one of the schemes below
4554ac54d93SMarek Sokolowski   // depending on the control class:
4564ac54d93SMarek Sokolowski   //  [class] text, id, x, y, width, height [, style] [, exstyle] [, helpID]
4574ac54d93SMarek Sokolowski   //  [class]       id, x, y, width, height [, style] [, exstyle] [, helpID]
4584ac54d93SMarek Sokolowski   // Note that control ids must be integers.
4597f7745c0SMarek Sokolowski   // Text might be either a string or an integer pointing to resource ID.
4604ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(ClassResult, readIdentifier());
46175fa173eSMarek Sokolowski   std::string ClassUpper = ClassResult->upper();
4627f7745c0SMarek Sokolowski   auto CtlInfo = Control::SupportedCtls.find(ClassUpper);
4637f7745c0SMarek Sokolowski   if (CtlInfo == Control::SupportedCtls.end())
4644ac54d93SMarek Sokolowski     return getExpectedError("control type, END or '}'", true);
4654ac54d93SMarek Sokolowski 
4664ac54d93SMarek Sokolowski   // Read caption if necessary.
4677f7745c0SMarek Sokolowski   IntOrString Caption{StringRef()};
4687f7745c0SMarek Sokolowski   if (CtlInfo->getValue().HasTitle) {
4697f7745c0SMarek Sokolowski     ASSIGN_OR_RETURN(CaptionResult, readIntOrString());
4704ac54d93SMarek Sokolowski     RETURN_IF_ERROR(consumeType(Kind::Comma));
4714ac54d93SMarek Sokolowski     Caption = *CaptionResult;
4724ac54d93SMarek Sokolowski   }
4734ac54d93SMarek Sokolowski 
4744ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(Args, readIntsWithCommas(5, 8));
4754ac54d93SMarek Sokolowski 
4764ac54d93SMarek Sokolowski   auto TakeOptArg = [&Args](size_t Id) -> Optional<uint32_t> {
47707bc04ffSZachary Turner     return Args->size() > Id ? (uint32_t)(*Args)[Id] : Optional<uint32_t>();
4784ac54d93SMarek Sokolowski   };
4794ac54d93SMarek Sokolowski 
4804ac54d93SMarek Sokolowski   return Control(*ClassResult, Caption, (*Args)[0], (*Args)[1], (*Args)[2],
4814ac54d93SMarek Sokolowski                  (*Args)[3], (*Args)[4], TakeOptArg(5), TakeOptArg(6),
4824ac54d93SMarek Sokolowski                  TakeOptArg(7));
4834ac54d93SMarek Sokolowski }
4844ac54d93SMarek Sokolowski 
4855cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseIconResource() {
4865cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
4874a765da3SMarek Sokolowski   return llvm::make_unique<IconResource>(*Arg);
4885cd3d5c8SMarek Sokolowski }
4895cd3d5c8SMarek Sokolowski 
49072aa937eSMarek Sokolowski RCParser::ParseType RCParser::parseHTMLResource() {
49172aa937eSMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
4924a765da3SMarek Sokolowski   return llvm::make_unique<HTMLResource>(*Arg);
49372aa937eSMarek Sokolowski }
49472aa937eSMarek Sokolowski 
49599ecb0ebSMarek Sokolowski RCParser::ParseType RCParser::parseMenuResource() {
49699ecb0ebSMarek Sokolowski   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements());
49799ecb0ebSMarek Sokolowski   ASSIGN_OR_RETURN(Items, parseMenuItemsList());
4984a765da3SMarek Sokolowski   return llvm::make_unique<MenuResource>(std::move(*OptStatements),
49999ecb0ebSMarek Sokolowski                                          std::move(*Items));
50099ecb0ebSMarek Sokolowski }
50199ecb0ebSMarek Sokolowski 
50299ecb0ebSMarek Sokolowski Expected<MenuDefinitionList> RCParser::parseMenuItemsList() {
50399ecb0ebSMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
50499ecb0ebSMarek Sokolowski 
50599ecb0ebSMarek Sokolowski   MenuDefinitionList List;
50699ecb0ebSMarek Sokolowski 
50799ecb0ebSMarek Sokolowski   // Read a set of items. Each item is of one of three kinds:
50899ecb0ebSMarek Sokolowski   //   MENUITEM SEPARATOR
50999ecb0ebSMarek Sokolowski   //   MENUITEM caption:String, result:Int [, menu flags]...
51099ecb0ebSMarek Sokolowski   //   POPUP caption:String [, menu flags]... { items... }
51199ecb0ebSMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
51299ecb0ebSMarek Sokolowski     ASSIGN_OR_RETURN(ItemTypeResult, readIdentifier());
51399ecb0ebSMarek Sokolowski 
51499ecb0ebSMarek Sokolowski     bool IsMenuItem = ItemTypeResult->equals_lower("MENUITEM");
51599ecb0ebSMarek Sokolowski     bool IsPopup = ItemTypeResult->equals_lower("POPUP");
51699ecb0ebSMarek Sokolowski     if (!IsMenuItem && !IsPopup)
51799ecb0ebSMarek Sokolowski       return getExpectedError("MENUITEM, POPUP, END or '}'", true);
51899ecb0ebSMarek Sokolowski 
51999ecb0ebSMarek Sokolowski     if (IsMenuItem && isNextTokenKind(Kind::Identifier)) {
52099ecb0ebSMarek Sokolowski       // Now, expecting SEPARATOR.
52199ecb0ebSMarek Sokolowski       ASSIGN_OR_RETURN(SeparatorResult, readIdentifier());
52299ecb0ebSMarek Sokolowski       if (SeparatorResult->equals_lower("SEPARATOR")) {
5234a765da3SMarek Sokolowski         List.addDefinition(llvm::make_unique<MenuSeparator>());
52499ecb0ebSMarek Sokolowski         continue;
52599ecb0ebSMarek Sokolowski       }
52699ecb0ebSMarek Sokolowski 
52799ecb0ebSMarek Sokolowski       return getExpectedError("SEPARATOR or string", true);
52899ecb0ebSMarek Sokolowski     }
52999ecb0ebSMarek Sokolowski 
53099ecb0ebSMarek Sokolowski     // Not a separator. Read the caption.
53199ecb0ebSMarek Sokolowski     ASSIGN_OR_RETURN(CaptionResult, readString());
53299ecb0ebSMarek Sokolowski 
53399ecb0ebSMarek Sokolowski     // If MENUITEM, expect also a comma and an integer.
53499ecb0ebSMarek Sokolowski     uint32_t MenuResult = -1;
53599ecb0ebSMarek Sokolowski 
53699ecb0ebSMarek Sokolowski     if (IsMenuItem) {
53799ecb0ebSMarek Sokolowski       RETURN_IF_ERROR(consumeType(Kind::Comma));
53899ecb0ebSMarek Sokolowski       ASSIGN_OR_RETURN(IntResult, readInt());
53999ecb0ebSMarek Sokolowski       MenuResult = *IntResult;
54099ecb0ebSMarek Sokolowski     }
54199ecb0ebSMarek Sokolowski 
542c75a087cSMarek Sokolowski     ASSIGN_OR_RETURN(FlagsResult, parseFlags(MenuDefinition::OptionsStr,
543c75a087cSMarek Sokolowski                                              MenuDefinition::OptionsFlags));
54499ecb0ebSMarek Sokolowski 
54599ecb0ebSMarek Sokolowski     if (IsPopup) {
54699ecb0ebSMarek Sokolowski       // If POPUP, read submenu items recursively.
54799ecb0ebSMarek Sokolowski       ASSIGN_OR_RETURN(SubMenuResult, parseMenuItemsList());
5484a765da3SMarek Sokolowski       List.addDefinition(llvm::make_unique<PopupItem>(
5494a765da3SMarek Sokolowski           *CaptionResult, *FlagsResult, std::move(*SubMenuResult)));
55099ecb0ebSMarek Sokolowski       continue;
55199ecb0ebSMarek Sokolowski     }
55299ecb0ebSMarek Sokolowski 
55399ecb0ebSMarek Sokolowski     assert(IsMenuItem);
55499ecb0ebSMarek Sokolowski     List.addDefinition(
5554a765da3SMarek Sokolowski         llvm::make_unique<MenuItem>(*CaptionResult, MenuResult, *FlagsResult));
55699ecb0ebSMarek Sokolowski   }
55799ecb0ebSMarek Sokolowski 
55899ecb0ebSMarek Sokolowski   return std::move(List);
55999ecb0ebSMarek Sokolowski }
56099ecb0ebSMarek Sokolowski 
5615cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseStringTableResource() {
5625cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements());
5635cd3d5c8SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
5645cd3d5c8SMarek Sokolowski 
5654a765da3SMarek Sokolowski   auto Table =
5664a765da3SMarek Sokolowski       llvm::make_unique<StringTableResource>(std::move(*OptStatements));
5675cd3d5c8SMarek Sokolowski 
5685cd3d5c8SMarek Sokolowski   // Read strings until we reach the end of the block.
5695cd3d5c8SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
5705cd3d5c8SMarek Sokolowski     // Each definition consists of string's ID (an integer) and a string.
5715cd3d5c8SMarek Sokolowski     // Some examples in documentation suggest that there might be a comma in
5725cd3d5c8SMarek Sokolowski     // between, however we strictly adhere to the single statement definition.
5735cd3d5c8SMarek Sokolowski     ASSIGN_OR_RETURN(IDResult, readInt());
5745cd3d5c8SMarek Sokolowski     ASSIGN_OR_RETURN(StrResult, readString());
5755cd3d5c8SMarek Sokolowski     Table->addString(*IDResult, *StrResult);
5765cd3d5c8SMarek Sokolowski   }
5775cd3d5c8SMarek Sokolowski 
5785cd3d5c8SMarek Sokolowski   return std::move(Table);
5795cd3d5c8SMarek Sokolowski }
5805cd3d5c8SMarek Sokolowski 
581fb74cb1eSMarek Sokolowski Expected<std::unique_ptr<VersionInfoBlock>>
582fb74cb1eSMarek Sokolowski RCParser::parseVersionInfoBlockContents(StringRef BlockName) {
583fb74cb1eSMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
584fb74cb1eSMarek Sokolowski 
58599ead70fSMarek Sokolowski   auto Contents = llvm::make_unique<VersionInfoBlock>(BlockName);
586fb74cb1eSMarek Sokolowski 
587fb74cb1eSMarek Sokolowski   while (!isNextTokenKind(Kind::BlockEnd)) {
588fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(Stmt, parseVersionInfoStmt());
589fb74cb1eSMarek Sokolowski     Contents->addStmt(std::move(*Stmt));
590fb74cb1eSMarek Sokolowski   }
591fb74cb1eSMarek Sokolowski 
592fb74cb1eSMarek Sokolowski   consume(); // Consume BlockEnd.
593fb74cb1eSMarek Sokolowski 
594fb74cb1eSMarek Sokolowski   return std::move(Contents);
595fb74cb1eSMarek Sokolowski }
596fb74cb1eSMarek Sokolowski 
597fb74cb1eSMarek Sokolowski Expected<std::unique_ptr<VersionInfoStmt>> RCParser::parseVersionInfoStmt() {
598fb74cb1eSMarek Sokolowski   // Expect either BLOCK or VALUE, then a name or a key (a string).
599fb74cb1eSMarek Sokolowski   ASSIGN_OR_RETURN(TypeResult, readIdentifier());
600fb74cb1eSMarek Sokolowski 
601fb74cb1eSMarek Sokolowski   if (TypeResult->equals_lower("BLOCK")) {
602fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(NameResult, readString());
603fb74cb1eSMarek Sokolowski     return parseVersionInfoBlockContents(*NameResult);
604fb74cb1eSMarek Sokolowski   }
605fb74cb1eSMarek Sokolowski 
606fb74cb1eSMarek Sokolowski   if (TypeResult->equals_lower("VALUE")) {
607fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(KeyResult, readString());
60807bc04ffSZachary Turner     // Read a non-empty list of strings and/or ints, each
60907bc04ffSZachary Turner     // possibly preceded by a comma. Unfortunately, the tool behavior depends
61007bc04ffSZachary Turner     // on them existing or not, so we need to memorize where we found them.
611fb74cb1eSMarek Sokolowski     std::vector<IntOrString> Values;
61207bc04ffSZachary Turner     std::vector<bool> PrecedingCommas;
61307bc04ffSZachary Turner     RETURN_IF_ERROR(consumeType(Kind::Comma));
61407bc04ffSZachary Turner     while (!isNextTokenKind(Kind::Identifier) &&
61507bc04ffSZachary Turner            !isNextTokenKind(Kind::BlockEnd)) {
61607bc04ffSZachary Turner       // Try to eat a comma if it's not the first statement.
61707bc04ffSZachary Turner       bool HadComma = Values.size() > 0 && consumeOptionalType(Kind::Comma);
618fb74cb1eSMarek Sokolowski       ASSIGN_OR_RETURN(ValueResult, readIntOrString());
619fb74cb1eSMarek Sokolowski       Values.push_back(*ValueResult);
62007bc04ffSZachary Turner       PrecedingCommas.push_back(HadComma);
621fb74cb1eSMarek Sokolowski     }
62207bc04ffSZachary Turner     return llvm::make_unique<VersionInfoValue>(*KeyResult, std::move(Values),
62307bc04ffSZachary Turner                                                std::move(PrecedingCommas));
624fb74cb1eSMarek Sokolowski   }
625fb74cb1eSMarek Sokolowski 
626fb74cb1eSMarek Sokolowski   return getExpectedError("BLOCK or VALUE", true);
627fb74cb1eSMarek Sokolowski }
628fb74cb1eSMarek Sokolowski 
629fb74cb1eSMarek Sokolowski Expected<VersionInfoResource::VersionInfoFixed>
630fb74cb1eSMarek Sokolowski RCParser::parseVersionInfoFixed() {
631fb74cb1eSMarek Sokolowski   using RetType = VersionInfoResource::VersionInfoFixed;
632fb74cb1eSMarek Sokolowski   RetType Result;
633fb74cb1eSMarek Sokolowski 
634fb74cb1eSMarek Sokolowski   // Read until the beginning of the block.
635fb74cb1eSMarek Sokolowski   while (!isNextTokenKind(Kind::BlockBegin)) {
636fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(TypeResult, readIdentifier());
637fb74cb1eSMarek Sokolowski     auto FixedType = RetType::getFixedType(*TypeResult);
638fb74cb1eSMarek Sokolowski 
639fb74cb1eSMarek Sokolowski     if (!RetType::isTypeSupported(FixedType))
640fb74cb1eSMarek Sokolowski       return getExpectedError("fixed VERSIONINFO statement type", true);
641fb74cb1eSMarek Sokolowski     if (Result.IsTypePresent[FixedType])
642fb74cb1eSMarek Sokolowski       return getExpectedError("yet unread fixed VERSIONINFO statement type",
643fb74cb1eSMarek Sokolowski                               true);
644fb74cb1eSMarek Sokolowski 
645fb74cb1eSMarek Sokolowski     // VERSION variations take multiple integers.
646fb74cb1eSMarek Sokolowski     size_t NumInts = RetType::isVersionType(FixedType) ? 4 : 1;
647fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(ArgsResult, readIntsWithCommas(NumInts, NumInts));
64807bc04ffSZachary Turner     SmallVector<uint32_t, 4> ArgInts(ArgsResult->begin(), ArgsResult->end());
64907bc04ffSZachary Turner     Result.setValue(FixedType, ArgInts);
650fb74cb1eSMarek Sokolowski   }
651fb74cb1eSMarek Sokolowski 
652fb74cb1eSMarek Sokolowski   return Result;
653fb74cb1eSMarek Sokolowski }
654fb74cb1eSMarek Sokolowski 
6555cd3d5c8SMarek Sokolowski RCParser::ParseOptionType RCParser::parseLanguageStmt() {
6565cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Args, readIntsWithCommas(/* min = */ 2, /* max = */ 2));
6574a765da3SMarek Sokolowski   return llvm::make_unique<LanguageResource>((*Args)[0], (*Args)[1]);
6585cd3d5c8SMarek Sokolowski }
6595cd3d5c8SMarek Sokolowski 
6605cd3d5c8SMarek Sokolowski RCParser::ParseOptionType RCParser::parseCharacteristicsStmt() {
6615cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readInt());
6624a765da3SMarek Sokolowski   return llvm::make_unique<CharacteristicsStmt>(*Arg);
6635cd3d5c8SMarek Sokolowski }
6645cd3d5c8SMarek Sokolowski 
6655cd3d5c8SMarek Sokolowski RCParser::ParseOptionType RCParser::parseVersionStmt() {
6665cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readInt());
6674a765da3SMarek Sokolowski   return llvm::make_unique<VersionStmt>(*Arg);
6685cd3d5c8SMarek Sokolowski }
6695cd3d5c8SMarek Sokolowski 
6704ac54d93SMarek Sokolowski RCParser::ParseOptionType RCParser::parseCaptionStmt() {
6714ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
6724a765da3SMarek Sokolowski   return llvm::make_unique<CaptionStmt>(*Arg);
6734ac54d93SMarek Sokolowski }
6744ac54d93SMarek Sokolowski 
675420090afSZachary Turner RCParser::ParseOptionType RCParser::parseFontStmt(OptStmtType DialogType) {
676420090afSZachary Turner   assert(DialogType != OptStmtType::BasicStmt);
677420090afSZachary Turner 
6784ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(SizeResult, readInt());
6794ac54d93SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::Comma));
6804ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(NameResult, readString());
681420090afSZachary Turner 
682420090afSZachary Turner   // Default values for the optional arguments.
683420090afSZachary Turner   uint32_t FontWeight = 0;
684420090afSZachary Turner   bool FontItalic = false;
685420090afSZachary Turner   uint32_t FontCharset = 1;
686420090afSZachary Turner   if (DialogType == OptStmtType::DialogExStmt) {
687420090afSZachary Turner     if (consumeOptionalType(Kind::Comma)) {
688420090afSZachary Turner       ASSIGN_OR_RETURN(Args, readIntsWithCommas(/* min = */ 0, /* max = */ 3));
689420090afSZachary Turner       if (Args->size() >= 1)
690420090afSZachary Turner         FontWeight = (*Args)[0];
691420090afSZachary Turner       if (Args->size() >= 2)
692420090afSZachary Turner         FontItalic = (*Args)[1] != 0;
693420090afSZachary Turner       if (Args->size() >= 3)
694420090afSZachary Turner         FontCharset = (*Args)[2];
695420090afSZachary Turner     }
696420090afSZachary Turner   }
697420090afSZachary Turner   return llvm::make_unique<FontStmt>(*SizeResult, *NameResult, FontWeight,
698420090afSZachary Turner                                      FontItalic, FontCharset);
6994ac54d93SMarek Sokolowski }
7004ac54d93SMarek Sokolowski 
7014ac54d93SMarek Sokolowski RCParser::ParseOptionType RCParser::parseStyleStmt() {
7024ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readInt());
7034a765da3SMarek Sokolowski   return llvm::make_unique<StyleStmt>(*Arg);
7044ac54d93SMarek Sokolowski }
7054ac54d93SMarek Sokolowski 
706*514b7105SZachary Turner Error RCParser::getExpectedError(const Twine &Message, bool IsAlreadyRead) {
7075cd3d5c8SMarek Sokolowski   return make_error<ParserError>(
7085cd3d5c8SMarek Sokolowski       Message, IsAlreadyRead ? std::prev(CurLoc) : CurLoc, End);
7095cd3d5c8SMarek Sokolowski }
7105cd3d5c8SMarek Sokolowski 
7115cd3d5c8SMarek Sokolowski } // namespace rc
7125cd3d5c8SMarek Sokolowski } // namespace llvm
713