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"
15*fa0ca6cbSZachary Turner #include "llvm/Option/ArgList.h"
16*fa0ca6cbSZachary Turner #include "llvm/Support/FileSystem.h"
17*fa0ca6cbSZachary Turner #include "llvm/Support/Path.h"
18*fa0ca6cbSZachary Turner #include "llvm/Support/Process.h"
195cd3d5c8SMarek Sokolowski 
205cd3d5c8SMarek Sokolowski // Take an expression returning llvm::Error and forward the error if it exists.
215cd3d5c8SMarek Sokolowski #define RETURN_IF_ERROR(Expr)                                                  \
225cd3d5c8SMarek Sokolowski   if (auto Err = (Expr))                                                       \
235cd3d5c8SMarek Sokolowski     return std::move(Err);
245cd3d5c8SMarek Sokolowski 
255cd3d5c8SMarek Sokolowski // Take an expression returning llvm::Expected<T> and assign it to Var or
265cd3d5c8SMarek Sokolowski // forward the error out of the function.
275cd3d5c8SMarek Sokolowski #define ASSIGN_OR_RETURN(Var, Expr)                                            \
285cd3d5c8SMarek Sokolowski   auto Var = (Expr);                                                           \
295cd3d5c8SMarek Sokolowski   if (!Var)                                                                    \
305cd3d5c8SMarek Sokolowski     return Var.takeError();
315cd3d5c8SMarek Sokolowski 
325cd3d5c8SMarek Sokolowski namespace llvm {
335cd3d5c8SMarek Sokolowski namespace rc {
345cd3d5c8SMarek Sokolowski 
35514b7105SZachary Turner RCParser::ParserError::ParserError(const Twine &Expected, const LocIter CurLoc,
365cd3d5c8SMarek Sokolowski                                    const LocIter End)
375cd3d5c8SMarek Sokolowski     : ErrorLoc(CurLoc), FileEnd(End) {
385cd3d5c8SMarek Sokolowski   CurMessage = "Error parsing file: expected " + Expected.str() + ", got " +
395cd3d5c8SMarek Sokolowski                (CurLoc == End ? "<EOF>" : CurLoc->value()).str();
405cd3d5c8SMarek Sokolowski }
415cd3d5c8SMarek Sokolowski 
425cd3d5c8SMarek Sokolowski char RCParser::ParserError::ID = 0;
435cd3d5c8SMarek Sokolowski 
44514b7105SZachary Turner RCParser::RCParser(std::vector<RCToken> TokenList)
455cd3d5c8SMarek Sokolowski     : Tokens(std::move(TokenList)), CurLoc(Tokens.begin()), End(Tokens.end()) {}
465cd3d5c8SMarek Sokolowski 
475cd3d5c8SMarek Sokolowski bool RCParser::isEof() const { return CurLoc == End; }
485cd3d5c8SMarek Sokolowski 
495cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseSingleResource() {
505cd3d5c8SMarek Sokolowski   // The first thing we read is usually a resource's name. However, in some
515cd3d5c8SMarek Sokolowski   // cases (LANGUAGE and STRINGTABLE) the resources don't have their names
525cd3d5c8SMarek Sokolowski   // and the first token to be read is the type.
535cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(NameToken, readTypeOrName());
545cd3d5c8SMarek Sokolowski 
555cd3d5c8SMarek Sokolowski   if (NameToken->equalsLower("LANGUAGE"))
565cd3d5c8SMarek Sokolowski     return parseLanguageResource();
575cd3d5c8SMarek Sokolowski   else if (NameToken->equalsLower("STRINGTABLE"))
585cd3d5c8SMarek Sokolowski     return parseStringTableResource();
595cd3d5c8SMarek Sokolowski 
605cd3d5c8SMarek Sokolowski   // If it's not an unnamed resource, what we've just read is a name. Now,
615cd3d5c8SMarek Sokolowski   // read resource type;
625cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(TypeToken, readTypeOrName());
635cd3d5c8SMarek Sokolowski 
645cd3d5c8SMarek Sokolowski   ParseType Result = std::unique_ptr<RCResource>();
655cd3d5c8SMarek Sokolowski   (void)!Result;
665cd3d5c8SMarek Sokolowski 
677f110527SMarek Sokolowski   if (TypeToken->equalsLower("ACCELERATORS"))
687f110527SMarek Sokolowski     Result = parseAcceleratorsResource();
697f110527SMarek Sokolowski   else if (TypeToken->equalsLower("CURSOR"))
7072aa937eSMarek Sokolowski     Result = parseCursorResource();
714ac54d93SMarek Sokolowski   else if (TypeToken->equalsLower("DIALOG"))
724ac54d93SMarek Sokolowski     Result = parseDialogResource(false);
734ac54d93SMarek Sokolowski   else if (TypeToken->equalsLower("DIALOGEX"))
744ac54d93SMarek Sokolowski     Result = parseDialogResource(true);
7572aa937eSMarek Sokolowski   else if (TypeToken->equalsLower("ICON"))
765cd3d5c8SMarek Sokolowski     Result = parseIconResource();
7772aa937eSMarek Sokolowski   else if (TypeToken->equalsLower("HTML"))
7872aa937eSMarek Sokolowski     Result = parseHTMLResource();
7999ecb0ebSMarek Sokolowski   else if (TypeToken->equalsLower("MENU"))
8099ecb0ebSMarek Sokolowski     Result = parseMenuResource();
81fb74cb1eSMarek Sokolowski   else if (TypeToken->equalsLower("VERSIONINFO"))
82fb74cb1eSMarek Sokolowski     Result = parseVersionInfoResource();
835cd3d5c8SMarek Sokolowski   else
84b5f39a05SMarek Sokolowski     Result = parseUserDefinedResource(*TypeToken);
855cd3d5c8SMarek Sokolowski 
865cd3d5c8SMarek Sokolowski   if (Result)
875cd3d5c8SMarek Sokolowski     (*Result)->setName(*NameToken);
885cd3d5c8SMarek Sokolowski 
895cd3d5c8SMarek Sokolowski   return Result;
905cd3d5c8SMarek Sokolowski }
915cd3d5c8SMarek Sokolowski 
925cd3d5c8SMarek Sokolowski bool RCParser::isNextTokenKind(Kind TokenKind) const {
935cd3d5c8SMarek Sokolowski   return !isEof() && look().kind() == TokenKind;
945cd3d5c8SMarek Sokolowski }
955cd3d5c8SMarek Sokolowski 
965cd3d5c8SMarek Sokolowski const RCToken &RCParser::look() const {
975cd3d5c8SMarek Sokolowski   assert(!isEof());
985cd3d5c8SMarek Sokolowski   return *CurLoc;
995cd3d5c8SMarek Sokolowski }
1005cd3d5c8SMarek Sokolowski 
1015cd3d5c8SMarek Sokolowski const RCToken &RCParser::read() {
1025cd3d5c8SMarek Sokolowski   assert(!isEof());
1035cd3d5c8SMarek Sokolowski   return *CurLoc++;
1045cd3d5c8SMarek Sokolowski }
1055cd3d5c8SMarek Sokolowski 
1065cd3d5c8SMarek Sokolowski void RCParser::consume() {
1075cd3d5c8SMarek Sokolowski   assert(!isEof());
1085cd3d5c8SMarek Sokolowski   CurLoc++;
1095cd3d5c8SMarek Sokolowski }
1105cd3d5c8SMarek Sokolowski 
1117e89ee7fSMarek Sokolowski // An integer description might consist of a single integer or
1127e89ee7fSMarek Sokolowski // an arithmetic expression evaluating to the integer. The expressions
1137e89ee7fSMarek Sokolowski // can contain the following tokens: <int> ( ) + - | & ~. Their meaning
1147e89ee7fSMarek Sokolowski // is the same as in C++.
1157e89ee7fSMarek Sokolowski // The operators in the original RC implementation have the following
1167e89ee7fSMarek Sokolowski // precedence:
1177e89ee7fSMarek Sokolowski //   1) Unary operators (- ~),
1187e89ee7fSMarek Sokolowski //   2) Binary operators (+ - & |), with no precedence.
1197e89ee7fSMarek Sokolowski //
1207e89ee7fSMarek Sokolowski // The following grammar is used to parse the expressions Exp1:
1217e89ee7fSMarek Sokolowski //   Exp1 ::= Exp2 || Exp1 + Exp2 || Exp1 - Exp2 || Exp1 | Exp2 || Exp1 & Exp2
1227e89ee7fSMarek Sokolowski //   Exp2 ::= -Exp2 || ~Exp2 || Int || (Exp1).
1237e89ee7fSMarek Sokolowski // (More conveniently, Exp1 is a non-empty sequence of Exp2 expressions,
1247e89ee7fSMarek Sokolowski // separated by binary operators.)
1257e89ee7fSMarek Sokolowski //
1267e89ee7fSMarek Sokolowski // Expressions of type Exp1 are read by parseIntExpr1(Inner) method, while Exp2
1277e89ee7fSMarek Sokolowski // is read by parseIntExpr2().
1287e89ee7fSMarek Sokolowski //
1297e89ee7fSMarek Sokolowski // The original Microsoft tool handles multiple unary operators incorrectly.
1307e89ee7fSMarek Sokolowski // For example, in 16-bit little-endian integers:
1317e89ee7fSMarek Sokolowski //    1 => 01 00, -1 => ff ff, --1 => ff ff, ---1 => 01 00;
1327e89ee7fSMarek Sokolowski //    1 => 01 00, ~1 => fe ff, ~~1 => fd ff, ~~~1 => fc ff.
1337e89ee7fSMarek Sokolowski // Our implementation differs from the original one and handles these
1347e89ee7fSMarek Sokolowski // operators correctly:
1357e89ee7fSMarek Sokolowski //    1 => 01 00, -1 => ff ff, --1 => 01 00, ---1 => ff ff;
1367e89ee7fSMarek Sokolowski //    1 => 01 00, ~1 => fe ff, ~~1 => 01 00, ~~~1 => fe ff.
1377e89ee7fSMarek Sokolowski 
13807bc04ffSZachary Turner Expected<RCInt> RCParser::readInt() { return parseIntExpr1(); }
1397e89ee7fSMarek Sokolowski 
14007bc04ffSZachary Turner Expected<RCInt> RCParser::parseIntExpr1() {
1417e89ee7fSMarek Sokolowski   // Exp1 ::= Exp2 || Exp1 + Exp2 || Exp1 - Exp2 || Exp1 | Exp2 || Exp1 & Exp2.
1427e89ee7fSMarek Sokolowski   ASSIGN_OR_RETURN(FirstResult, parseIntExpr2());
14307bc04ffSZachary Turner   RCInt Result = *FirstResult;
1447e89ee7fSMarek Sokolowski 
1457e89ee7fSMarek Sokolowski   while (!isEof() && look().isBinaryOp()) {
1467e89ee7fSMarek Sokolowski     auto OpToken = read();
1477e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(NextResult, parseIntExpr2());
1487e89ee7fSMarek Sokolowski 
1497e89ee7fSMarek Sokolowski     switch (OpToken.kind()) {
1507e89ee7fSMarek Sokolowski     case Kind::Plus:
1517e89ee7fSMarek Sokolowski       Result += *NextResult;
1527e89ee7fSMarek Sokolowski       break;
1537e89ee7fSMarek Sokolowski 
1547e89ee7fSMarek Sokolowski     case Kind::Minus:
1557e89ee7fSMarek Sokolowski       Result -= *NextResult;
1567e89ee7fSMarek Sokolowski       break;
1577e89ee7fSMarek Sokolowski 
1587e89ee7fSMarek Sokolowski     case Kind::Pipe:
1597e89ee7fSMarek Sokolowski       Result |= *NextResult;
1607e89ee7fSMarek Sokolowski       break;
1617e89ee7fSMarek Sokolowski 
1627e89ee7fSMarek Sokolowski     case Kind::Amp:
1637e89ee7fSMarek Sokolowski       Result &= *NextResult;
1647e89ee7fSMarek Sokolowski       break;
1657e89ee7fSMarek Sokolowski 
1667e89ee7fSMarek Sokolowski     default:
1677e89ee7fSMarek Sokolowski       llvm_unreachable("Already processed all binary ops.");
1687e89ee7fSMarek Sokolowski     }
1697e89ee7fSMarek Sokolowski   }
1707e89ee7fSMarek Sokolowski 
1717e89ee7fSMarek Sokolowski   return Result;
1727e89ee7fSMarek Sokolowski }
1737e89ee7fSMarek Sokolowski 
17407bc04ffSZachary Turner Expected<RCInt> RCParser::parseIntExpr2() {
1757e89ee7fSMarek Sokolowski   // Exp2 ::= -Exp2 || ~Exp2 || Int || (Exp1).
1767e89ee7fSMarek Sokolowski   static const char ErrorMsg[] = "'-', '~', integer or '('";
1777e89ee7fSMarek Sokolowski 
1787e89ee7fSMarek Sokolowski   if (isEof())
1797e89ee7fSMarek Sokolowski     return getExpectedError(ErrorMsg);
1807e89ee7fSMarek Sokolowski 
1817e89ee7fSMarek Sokolowski   switch (look().kind()) {
1827e89ee7fSMarek Sokolowski   case Kind::Minus: {
1837e89ee7fSMarek Sokolowski     consume();
1847e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(Result, parseIntExpr2());
1857e89ee7fSMarek Sokolowski     return -(*Result);
1867e89ee7fSMarek Sokolowski   }
1877e89ee7fSMarek Sokolowski 
1887e89ee7fSMarek Sokolowski   case Kind::Tilde: {
1897e89ee7fSMarek Sokolowski     consume();
1907e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(Result, parseIntExpr2());
1917e89ee7fSMarek Sokolowski     return ~(*Result);
1927e89ee7fSMarek Sokolowski   }
1937e89ee7fSMarek Sokolowski 
1947e89ee7fSMarek Sokolowski   case Kind::Int:
19507bc04ffSZachary Turner     return RCInt(read());
1967e89ee7fSMarek Sokolowski 
1977e89ee7fSMarek Sokolowski   case Kind::LeftParen: {
1987e89ee7fSMarek Sokolowski     consume();
1997e89ee7fSMarek Sokolowski     ASSIGN_OR_RETURN(Result, parseIntExpr1());
2007e89ee7fSMarek Sokolowski     RETURN_IF_ERROR(consumeType(Kind::RightParen));
2017e89ee7fSMarek Sokolowski     return *Result;
2027e89ee7fSMarek Sokolowski   }
2037e89ee7fSMarek Sokolowski 
2047e89ee7fSMarek Sokolowski   default:
2057e89ee7fSMarek Sokolowski     return getExpectedError(ErrorMsg);
2067e89ee7fSMarek Sokolowski   }
2075cd3d5c8SMarek Sokolowski }
2085cd3d5c8SMarek Sokolowski 
2095cd3d5c8SMarek Sokolowski Expected<StringRef> RCParser::readString() {
2105cd3d5c8SMarek Sokolowski   if (!isNextTokenKind(Kind::String))
2115cd3d5c8SMarek Sokolowski     return getExpectedError("string");
2125cd3d5c8SMarek Sokolowski   return read().value();
2135cd3d5c8SMarek Sokolowski }
2145cd3d5c8SMarek Sokolowski 
2155cd3d5c8SMarek Sokolowski Expected<StringRef> RCParser::readIdentifier() {
2165cd3d5c8SMarek Sokolowski   if (!isNextTokenKind(Kind::Identifier))
2175cd3d5c8SMarek Sokolowski     return getExpectedError("identifier");
2185cd3d5c8SMarek Sokolowski   return read().value();
2195cd3d5c8SMarek Sokolowski }
2205cd3d5c8SMarek Sokolowski 
2217f110527SMarek Sokolowski Expected<IntOrString> RCParser::readIntOrString() {
2227f110527SMarek Sokolowski   if (!isNextTokenKind(Kind::Int) && !isNextTokenKind(Kind::String))
2237f110527SMarek Sokolowski     return getExpectedError("int or string");
2247f110527SMarek Sokolowski   return IntOrString(read());
2257f110527SMarek Sokolowski }
2267f110527SMarek Sokolowski 
2275cd3d5c8SMarek Sokolowski Expected<IntOrString> RCParser::readTypeOrName() {
2285cd3d5c8SMarek Sokolowski   // We suggest that the correct resource name or type should be either an
2295cd3d5c8SMarek Sokolowski   // identifier or an integer. The original RC tool is much more liberal.
2305cd3d5c8SMarek Sokolowski   if (!isNextTokenKind(Kind::Identifier) && !isNextTokenKind(Kind::Int))
2315cd3d5c8SMarek Sokolowski     return getExpectedError("int or identifier");
2327f110527SMarek Sokolowski   return IntOrString(read());
2335cd3d5c8SMarek Sokolowski }
2345cd3d5c8SMarek Sokolowski 
2355cd3d5c8SMarek Sokolowski Error RCParser::consumeType(Kind TokenKind) {
2365cd3d5c8SMarek Sokolowski   if (isNextTokenKind(TokenKind)) {
2375cd3d5c8SMarek Sokolowski     consume();
2385cd3d5c8SMarek Sokolowski     return Error::success();
2395cd3d5c8SMarek Sokolowski   }
2405cd3d5c8SMarek Sokolowski 
2415cd3d5c8SMarek Sokolowski   switch (TokenKind) {
2425cd3d5c8SMarek Sokolowski #define TOKEN(TokenName)                                                       \
2435cd3d5c8SMarek Sokolowski   case Kind::TokenName:                                                        \
2445cd3d5c8SMarek Sokolowski     return getExpectedError(#TokenName);
2455cd3d5c8SMarek Sokolowski #define SHORT_TOKEN(TokenName, TokenCh)                                        \
2465cd3d5c8SMarek Sokolowski   case Kind::TokenName:                                                        \
2475cd3d5c8SMarek Sokolowski     return getExpectedError(#TokenCh);
2485cd3d5c8SMarek Sokolowski #include "ResourceScriptTokenList.h"
2495cd3d5c8SMarek Sokolowski #undef SHORT_TOKEN
2505cd3d5c8SMarek Sokolowski #undef TOKEN
2515cd3d5c8SMarek Sokolowski   }
2525cd3d5c8SMarek Sokolowski 
2535cd3d5c8SMarek Sokolowski   llvm_unreachable("All case options exhausted.");
2545cd3d5c8SMarek Sokolowski }
2555cd3d5c8SMarek Sokolowski 
2565cd3d5c8SMarek Sokolowski bool RCParser::consumeOptionalType(Kind TokenKind) {
2575cd3d5c8SMarek Sokolowski   if (isNextTokenKind(TokenKind)) {
2585cd3d5c8SMarek Sokolowski     consume();
2595cd3d5c8SMarek Sokolowski     return true;
2605cd3d5c8SMarek Sokolowski   }
2615cd3d5c8SMarek Sokolowski 
2625cd3d5c8SMarek Sokolowski   return false;
2635cd3d5c8SMarek Sokolowski }
2645cd3d5c8SMarek Sokolowski 
26507bc04ffSZachary Turner Expected<SmallVector<RCInt, 8>> RCParser::readIntsWithCommas(size_t MinCount,
26607bc04ffSZachary Turner                                                              size_t MaxCount) {
2675cd3d5c8SMarek Sokolowski   assert(MinCount <= MaxCount);
2685cd3d5c8SMarek Sokolowski 
26907bc04ffSZachary Turner   SmallVector<RCInt, 8> Result;
2705cd3d5c8SMarek Sokolowski 
2715cd3d5c8SMarek Sokolowski   auto FailureHandler =
27207bc04ffSZachary Turner       [&](llvm::Error Err) -> Expected<SmallVector<RCInt, 8>> {
2735cd3d5c8SMarek Sokolowski     if (Result.size() < MinCount)
2745cd3d5c8SMarek Sokolowski       return std::move(Err);
2755cd3d5c8SMarek Sokolowski     consumeError(std::move(Err));
2765cd3d5c8SMarek Sokolowski     return Result;
2775cd3d5c8SMarek Sokolowski   };
2785cd3d5c8SMarek Sokolowski 
2795cd3d5c8SMarek Sokolowski   for (size_t i = 0; i < MaxCount; ++i) {
2805cd3d5c8SMarek Sokolowski     // Try to read a comma unless we read the first token.
2815cd3d5c8SMarek Sokolowski     // Sometimes RC tool requires them and sometimes not. We decide to
2825cd3d5c8SMarek Sokolowski     // always require them.
2835cd3d5c8SMarek Sokolowski     if (i >= 1) {
2845cd3d5c8SMarek Sokolowski       if (auto CommaError = consumeType(Kind::Comma))
2855cd3d5c8SMarek Sokolowski         return FailureHandler(std::move(CommaError));
2865cd3d5c8SMarek Sokolowski     }
2875cd3d5c8SMarek Sokolowski 
2885cd3d5c8SMarek Sokolowski     if (auto IntResult = readInt())
2895cd3d5c8SMarek Sokolowski       Result.push_back(*IntResult);
2905cd3d5c8SMarek Sokolowski     else
2915cd3d5c8SMarek Sokolowski       return FailureHandler(IntResult.takeError());
2925cd3d5c8SMarek Sokolowski   }
2935cd3d5c8SMarek Sokolowski 
2945cd3d5c8SMarek Sokolowski   return std::move(Result);
2955cd3d5c8SMarek Sokolowski }
2965cd3d5c8SMarek Sokolowski 
297c75a087cSMarek Sokolowski Expected<uint32_t> RCParser::parseFlags(ArrayRef<StringRef> FlagDesc,
298c75a087cSMarek Sokolowski                                         ArrayRef<uint32_t> FlagValues) {
2997f110527SMarek Sokolowski   assert(!FlagDesc.empty());
300c75a087cSMarek Sokolowski   assert(FlagDesc.size() == FlagValues.size());
3017f110527SMarek Sokolowski 
3027f110527SMarek Sokolowski   uint32_t Result = 0;
3037f110527SMarek Sokolowski   while (isNextTokenKind(Kind::Comma)) {
3047f110527SMarek Sokolowski     consume();
3057f110527SMarek Sokolowski     ASSIGN_OR_RETURN(FlagResult, readIdentifier());
3067f110527SMarek Sokolowski     bool FoundFlag = false;
3077f110527SMarek Sokolowski 
3087f110527SMarek Sokolowski     for (size_t FlagId = 0; FlagId < FlagDesc.size(); ++FlagId) {
3097f110527SMarek Sokolowski       if (!FlagResult->equals_lower(FlagDesc[FlagId]))
3107f110527SMarek Sokolowski         continue;
3117f110527SMarek Sokolowski 
312c75a087cSMarek Sokolowski       Result |= FlagValues[FlagId];
3137f110527SMarek Sokolowski       FoundFlag = true;
3147f110527SMarek Sokolowski       break;
3157f110527SMarek Sokolowski     }
3167f110527SMarek Sokolowski 
3177f110527SMarek Sokolowski     if (!FoundFlag)
3187f110527SMarek Sokolowski       return getExpectedError(join(FlagDesc, "/"), true);
3197f110527SMarek Sokolowski   }
3207f110527SMarek Sokolowski 
3217f110527SMarek Sokolowski   return Result;
3227f110527SMarek Sokolowski }
3237f110527SMarek Sokolowski 
324420090afSZachary Turner Expected<OptionalStmtList>
325420090afSZachary Turner RCParser::parseOptionalStatements(OptStmtType StmtsType) {
3265cd3d5c8SMarek Sokolowski   OptionalStmtList Result;
3275cd3d5c8SMarek Sokolowski 
3285cd3d5c8SMarek Sokolowski   // The last statement is always followed by the start of the block.
3295cd3d5c8SMarek Sokolowski   while (!isNextTokenKind(Kind::BlockBegin)) {
330420090afSZachary Turner     ASSIGN_OR_RETURN(SingleParse, parseSingleOptionalStatement(StmtsType));
3315cd3d5c8SMarek Sokolowski     Result.addStmt(std::move(*SingleParse));
3325cd3d5c8SMarek Sokolowski   }
3335cd3d5c8SMarek Sokolowski 
3345cd3d5c8SMarek Sokolowski   return std::move(Result);
3355cd3d5c8SMarek Sokolowski }
3365cd3d5c8SMarek Sokolowski 
3375cd3d5c8SMarek Sokolowski Expected<std::unique_ptr<OptionalStmt>>
338420090afSZachary Turner RCParser::parseSingleOptionalStatement(OptStmtType StmtsType) {
3395cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(TypeToken, readIdentifier());
3405cd3d5c8SMarek Sokolowski   if (TypeToken->equals_lower("CHARACTERISTICS"))
3415cd3d5c8SMarek Sokolowski     return parseCharacteristicsStmt();
3424ac54d93SMarek Sokolowski   if (TypeToken->equals_lower("LANGUAGE"))
3435cd3d5c8SMarek Sokolowski     return parseLanguageStmt();
3444ac54d93SMarek Sokolowski   if (TypeToken->equals_lower("VERSION"))
3455cd3d5c8SMarek Sokolowski     return parseVersionStmt();
3464ac54d93SMarek Sokolowski 
347420090afSZachary Turner   if (StmtsType != OptStmtType::BasicStmt) {
3484ac54d93SMarek Sokolowski     if (TypeToken->equals_lower("CAPTION"))
3494ac54d93SMarek Sokolowski       return parseCaptionStmt();
3504ac54d93SMarek Sokolowski     if (TypeToken->equals_lower("FONT"))
351420090afSZachary Turner       return parseFontStmt(StmtsType);
3524ac54d93SMarek Sokolowski     if (TypeToken->equals_lower("STYLE"))
3534ac54d93SMarek Sokolowski       return parseStyleStmt();
3544ac54d93SMarek Sokolowski   }
3554ac54d93SMarek Sokolowski 
3565cd3d5c8SMarek Sokolowski   return getExpectedError("optional statement type, BEGIN or '{'",
3575cd3d5c8SMarek Sokolowski                           /* IsAlreadyRead = */ true);
3585cd3d5c8SMarek Sokolowski }
3595cd3d5c8SMarek Sokolowski 
3605cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseLanguageResource() {
3615cd3d5c8SMarek Sokolowski   // Read LANGUAGE as an optional statement. If it's read correctly, we can
3625cd3d5c8SMarek Sokolowski   // upcast it to RCResource.
3635cd3d5c8SMarek Sokolowski   return parseLanguageStmt();
3645cd3d5c8SMarek Sokolowski }
3655cd3d5c8SMarek Sokolowski 
3667f110527SMarek Sokolowski RCParser::ParseType RCParser::parseAcceleratorsResource() {
3677f110527SMarek Sokolowski   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements());
3687f110527SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
3697f110527SMarek Sokolowski 
3704a765da3SMarek Sokolowski   auto Accels =
3714a765da3SMarek Sokolowski       llvm::make_unique<AcceleratorsResource>(std::move(*OptStatements));
3727f110527SMarek Sokolowski 
3737f110527SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
3747f110527SMarek Sokolowski     ASSIGN_OR_RETURN(EventResult, readIntOrString());
3757f110527SMarek Sokolowski     RETURN_IF_ERROR(consumeType(Kind::Comma));
3767f110527SMarek Sokolowski     ASSIGN_OR_RETURN(IDResult, readInt());
377c75a087cSMarek Sokolowski     ASSIGN_OR_RETURN(
378c75a087cSMarek Sokolowski         FlagsResult,
379c75a087cSMarek Sokolowski         parseFlags(AcceleratorsResource::Accelerator::OptionsStr,
380c75a087cSMarek Sokolowski                    AcceleratorsResource::Accelerator::OptionsFlags));
3817f110527SMarek Sokolowski     Accels->addAccelerator(*EventResult, *IDResult, *FlagsResult);
3827f110527SMarek Sokolowski   }
3837f110527SMarek Sokolowski 
3847f110527SMarek Sokolowski   return std::move(Accels);
3857f110527SMarek Sokolowski }
3867f110527SMarek Sokolowski 
38772aa937eSMarek Sokolowski RCParser::ParseType RCParser::parseCursorResource() {
38872aa937eSMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
3894a765da3SMarek Sokolowski   return llvm::make_unique<CursorResource>(*Arg);
39072aa937eSMarek Sokolowski }
39172aa937eSMarek Sokolowski 
3924ac54d93SMarek Sokolowski RCParser::ParseType RCParser::parseDialogResource(bool IsExtended) {
3934ac54d93SMarek Sokolowski   // Dialog resources have the following format of the arguments:
3944ac54d93SMarek Sokolowski   //  DIALOG:   x, y, width, height [opt stmts...] {controls...}
3954ac54d93SMarek Sokolowski   //  DIALOGEX: x, y, width, height [, helpID] [opt stmts...] {controls...}
3964ac54d93SMarek Sokolowski   // These are very similar, so we parse them together.
3974ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(LocResult, readIntsWithCommas(4, 4));
3984ac54d93SMarek Sokolowski 
3994ac54d93SMarek Sokolowski   uint32_t HelpID = 0; // When HelpID is unset, it's assumed to be 0.
4004ac54d93SMarek Sokolowski   if (IsExtended && consumeOptionalType(Kind::Comma)) {
4014ac54d93SMarek Sokolowski     ASSIGN_OR_RETURN(HelpIDResult, readInt());
4024ac54d93SMarek Sokolowski     HelpID = *HelpIDResult;
4034ac54d93SMarek Sokolowski   }
4044ac54d93SMarek Sokolowski 
405420090afSZachary Turner   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements(
406420090afSZachary Turner                                       IsExtended ? OptStmtType::DialogExStmt
407420090afSZachary Turner                                                  : OptStmtType::DialogStmt));
4084ac54d93SMarek Sokolowski 
4094ac54d93SMarek Sokolowski   assert(isNextTokenKind(Kind::BlockBegin) &&
4104ac54d93SMarek Sokolowski          "parseOptionalStatements, when successful, halts on BlockBegin.");
4114ac54d93SMarek Sokolowski   consume();
4124ac54d93SMarek Sokolowski 
4134a765da3SMarek Sokolowski   auto Dialog = llvm::make_unique<DialogResource>(
4144ac54d93SMarek Sokolowski       (*LocResult)[0], (*LocResult)[1], (*LocResult)[2], (*LocResult)[3],
4154ac54d93SMarek Sokolowski       HelpID, std::move(*OptStatements), IsExtended);
4164ac54d93SMarek Sokolowski 
4174ac54d93SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
4184ac54d93SMarek Sokolowski     ASSIGN_OR_RETURN(ControlDefResult, parseControl());
4194ac54d93SMarek Sokolowski     Dialog->addControl(std::move(*ControlDefResult));
4204ac54d93SMarek Sokolowski   }
4214ac54d93SMarek Sokolowski 
4224ac54d93SMarek Sokolowski   return std::move(Dialog);
4234ac54d93SMarek Sokolowski }
4244ac54d93SMarek Sokolowski 
425b5f39a05SMarek Sokolowski RCParser::ParseType RCParser::parseUserDefinedResource(IntOrString Type) {
426b5f39a05SMarek Sokolowski   if (isEof())
427b5f39a05SMarek Sokolowski     return getExpectedError("filename, '{' or BEGIN");
428b5f39a05SMarek Sokolowski 
429b5f39a05SMarek Sokolowski   // Check if this is a file resource.
430b5f39a05SMarek Sokolowski   if (look().kind() == Kind::String)
4314a765da3SMarek Sokolowski     return llvm::make_unique<UserDefinedResource>(Type, read().value());
432b5f39a05SMarek Sokolowski 
433b5f39a05SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
434b5f39a05SMarek Sokolowski   std::vector<IntOrString> Data;
435b5f39a05SMarek Sokolowski 
436b5f39a05SMarek Sokolowski   // Consume comma before each consecutive token except the first one.
437b5f39a05SMarek Sokolowski   bool ConsumeComma = false;
438b5f39a05SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
439b5f39a05SMarek Sokolowski     if (ConsumeComma)
440b5f39a05SMarek Sokolowski       RETURN_IF_ERROR(consumeType(Kind::Comma));
441b5f39a05SMarek Sokolowski     ConsumeComma = true;
442b5f39a05SMarek Sokolowski 
443b5f39a05SMarek Sokolowski     ASSIGN_OR_RETURN(Item, readIntOrString());
444b5f39a05SMarek Sokolowski     Data.push_back(*Item);
445b5f39a05SMarek Sokolowski   }
446b5f39a05SMarek Sokolowski 
4474a765da3SMarek Sokolowski   return llvm::make_unique<UserDefinedResource>(Type, std::move(Data));
448b5f39a05SMarek Sokolowski }
449b5f39a05SMarek Sokolowski 
450fb74cb1eSMarek Sokolowski RCParser::ParseType RCParser::parseVersionInfoResource() {
451fb74cb1eSMarek Sokolowski   ASSIGN_OR_RETURN(FixedResult, parseVersionInfoFixed());
452fb74cb1eSMarek Sokolowski   ASSIGN_OR_RETURN(BlockResult, parseVersionInfoBlockContents(StringRef()));
45399ead70fSMarek Sokolowski   return llvm::make_unique<VersionInfoResource>(std::move(**BlockResult),
454fb74cb1eSMarek Sokolowski                                                 std::move(*FixedResult));
455fb74cb1eSMarek Sokolowski }
456fb74cb1eSMarek Sokolowski 
4574ac54d93SMarek Sokolowski Expected<Control> RCParser::parseControl() {
4584ac54d93SMarek Sokolowski   // Each control definition (except CONTROL) follows one of the schemes below
4594ac54d93SMarek Sokolowski   // depending on the control class:
4604ac54d93SMarek Sokolowski   //  [class] text, id, x, y, width, height [, style] [, exstyle] [, helpID]
4614ac54d93SMarek Sokolowski   //  [class]       id, x, y, width, height [, style] [, exstyle] [, helpID]
4624ac54d93SMarek Sokolowski   // Note that control ids must be integers.
4637f7745c0SMarek Sokolowski   // Text might be either a string or an integer pointing to resource ID.
4644ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(ClassResult, readIdentifier());
46575fa173eSMarek Sokolowski   std::string ClassUpper = ClassResult->upper();
4667f7745c0SMarek Sokolowski   auto CtlInfo = Control::SupportedCtls.find(ClassUpper);
4677f7745c0SMarek Sokolowski   if (CtlInfo == Control::SupportedCtls.end())
4684ac54d93SMarek Sokolowski     return getExpectedError("control type, END or '}'", true);
4694ac54d93SMarek Sokolowski 
4704ac54d93SMarek Sokolowski   // Read caption if necessary.
4717f7745c0SMarek Sokolowski   IntOrString Caption{StringRef()};
4727f7745c0SMarek Sokolowski   if (CtlInfo->getValue().HasTitle) {
4737f7745c0SMarek Sokolowski     ASSIGN_OR_RETURN(CaptionResult, readIntOrString());
4744ac54d93SMarek Sokolowski     RETURN_IF_ERROR(consumeType(Kind::Comma));
4754ac54d93SMarek Sokolowski     Caption = *CaptionResult;
4764ac54d93SMarek Sokolowski   }
4774ac54d93SMarek Sokolowski 
4784ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(Args, readIntsWithCommas(5, 8));
4794ac54d93SMarek Sokolowski 
4804ac54d93SMarek Sokolowski   auto TakeOptArg = [&Args](size_t Id) -> Optional<uint32_t> {
48107bc04ffSZachary Turner     return Args->size() > Id ? (uint32_t)(*Args)[Id] : Optional<uint32_t>();
4824ac54d93SMarek Sokolowski   };
4834ac54d93SMarek Sokolowski 
4844ac54d93SMarek Sokolowski   return Control(*ClassResult, Caption, (*Args)[0], (*Args)[1], (*Args)[2],
4854ac54d93SMarek Sokolowski                  (*Args)[3], (*Args)[4], TakeOptArg(5), TakeOptArg(6),
4864ac54d93SMarek Sokolowski                  TakeOptArg(7));
4874ac54d93SMarek Sokolowski }
4884ac54d93SMarek Sokolowski 
4895cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseIconResource() {
4905cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
4914a765da3SMarek Sokolowski   return llvm::make_unique<IconResource>(*Arg);
4925cd3d5c8SMarek Sokolowski }
4935cd3d5c8SMarek Sokolowski 
49472aa937eSMarek Sokolowski RCParser::ParseType RCParser::parseHTMLResource() {
49572aa937eSMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
4964a765da3SMarek Sokolowski   return llvm::make_unique<HTMLResource>(*Arg);
49772aa937eSMarek Sokolowski }
49872aa937eSMarek Sokolowski 
49999ecb0ebSMarek Sokolowski RCParser::ParseType RCParser::parseMenuResource() {
50099ecb0ebSMarek Sokolowski   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements());
50199ecb0ebSMarek Sokolowski   ASSIGN_OR_RETURN(Items, parseMenuItemsList());
5024a765da3SMarek Sokolowski   return llvm::make_unique<MenuResource>(std::move(*OptStatements),
50399ecb0ebSMarek Sokolowski                                          std::move(*Items));
50499ecb0ebSMarek Sokolowski }
50599ecb0ebSMarek Sokolowski 
50699ecb0ebSMarek Sokolowski Expected<MenuDefinitionList> RCParser::parseMenuItemsList() {
50799ecb0ebSMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
50899ecb0ebSMarek Sokolowski 
50999ecb0ebSMarek Sokolowski   MenuDefinitionList List;
51099ecb0ebSMarek Sokolowski 
51199ecb0ebSMarek Sokolowski   // Read a set of items. Each item is of one of three kinds:
51299ecb0ebSMarek Sokolowski   //   MENUITEM SEPARATOR
51399ecb0ebSMarek Sokolowski   //   MENUITEM caption:String, result:Int [, menu flags]...
51499ecb0ebSMarek Sokolowski   //   POPUP caption:String [, menu flags]... { items... }
51599ecb0ebSMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
51699ecb0ebSMarek Sokolowski     ASSIGN_OR_RETURN(ItemTypeResult, readIdentifier());
51799ecb0ebSMarek Sokolowski 
51899ecb0ebSMarek Sokolowski     bool IsMenuItem = ItemTypeResult->equals_lower("MENUITEM");
51999ecb0ebSMarek Sokolowski     bool IsPopup = ItemTypeResult->equals_lower("POPUP");
52099ecb0ebSMarek Sokolowski     if (!IsMenuItem && !IsPopup)
52199ecb0ebSMarek Sokolowski       return getExpectedError("MENUITEM, POPUP, END or '}'", true);
52299ecb0ebSMarek Sokolowski 
52399ecb0ebSMarek Sokolowski     if (IsMenuItem && isNextTokenKind(Kind::Identifier)) {
52499ecb0ebSMarek Sokolowski       // Now, expecting SEPARATOR.
52599ecb0ebSMarek Sokolowski       ASSIGN_OR_RETURN(SeparatorResult, readIdentifier());
52699ecb0ebSMarek Sokolowski       if (SeparatorResult->equals_lower("SEPARATOR")) {
5274a765da3SMarek Sokolowski         List.addDefinition(llvm::make_unique<MenuSeparator>());
52899ecb0ebSMarek Sokolowski         continue;
52999ecb0ebSMarek Sokolowski       }
53099ecb0ebSMarek Sokolowski 
53199ecb0ebSMarek Sokolowski       return getExpectedError("SEPARATOR or string", true);
53299ecb0ebSMarek Sokolowski     }
53399ecb0ebSMarek Sokolowski 
53499ecb0ebSMarek Sokolowski     // Not a separator. Read the caption.
53599ecb0ebSMarek Sokolowski     ASSIGN_OR_RETURN(CaptionResult, readString());
53699ecb0ebSMarek Sokolowski 
53799ecb0ebSMarek Sokolowski     // If MENUITEM, expect also a comma and an integer.
53899ecb0ebSMarek Sokolowski     uint32_t MenuResult = -1;
53999ecb0ebSMarek Sokolowski 
54099ecb0ebSMarek Sokolowski     if (IsMenuItem) {
54199ecb0ebSMarek Sokolowski       RETURN_IF_ERROR(consumeType(Kind::Comma));
54299ecb0ebSMarek Sokolowski       ASSIGN_OR_RETURN(IntResult, readInt());
54399ecb0ebSMarek Sokolowski       MenuResult = *IntResult;
54499ecb0ebSMarek Sokolowski     }
54599ecb0ebSMarek Sokolowski 
546c75a087cSMarek Sokolowski     ASSIGN_OR_RETURN(FlagsResult, parseFlags(MenuDefinition::OptionsStr,
547c75a087cSMarek Sokolowski                                              MenuDefinition::OptionsFlags));
54899ecb0ebSMarek Sokolowski 
54999ecb0ebSMarek Sokolowski     if (IsPopup) {
55099ecb0ebSMarek Sokolowski       // If POPUP, read submenu items recursively.
55199ecb0ebSMarek Sokolowski       ASSIGN_OR_RETURN(SubMenuResult, parseMenuItemsList());
5524a765da3SMarek Sokolowski       List.addDefinition(llvm::make_unique<PopupItem>(
5534a765da3SMarek Sokolowski           *CaptionResult, *FlagsResult, std::move(*SubMenuResult)));
55499ecb0ebSMarek Sokolowski       continue;
55599ecb0ebSMarek Sokolowski     }
55699ecb0ebSMarek Sokolowski 
55799ecb0ebSMarek Sokolowski     assert(IsMenuItem);
55899ecb0ebSMarek Sokolowski     List.addDefinition(
5594a765da3SMarek Sokolowski         llvm::make_unique<MenuItem>(*CaptionResult, MenuResult, *FlagsResult));
56099ecb0ebSMarek Sokolowski   }
56199ecb0ebSMarek Sokolowski 
56299ecb0ebSMarek Sokolowski   return std::move(List);
56399ecb0ebSMarek Sokolowski }
56499ecb0ebSMarek Sokolowski 
5655cd3d5c8SMarek Sokolowski RCParser::ParseType RCParser::parseStringTableResource() {
5665cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(OptStatements, parseOptionalStatements());
5675cd3d5c8SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
5685cd3d5c8SMarek Sokolowski 
5694a765da3SMarek Sokolowski   auto Table =
5704a765da3SMarek Sokolowski       llvm::make_unique<StringTableResource>(std::move(*OptStatements));
5715cd3d5c8SMarek Sokolowski 
5725cd3d5c8SMarek Sokolowski   // Read strings until we reach the end of the block.
5735cd3d5c8SMarek Sokolowski   while (!consumeOptionalType(Kind::BlockEnd)) {
5745cd3d5c8SMarek Sokolowski     // Each definition consists of string's ID (an integer) and a string.
5755cd3d5c8SMarek Sokolowski     // Some examples in documentation suggest that there might be a comma in
5765cd3d5c8SMarek Sokolowski     // between, however we strictly adhere to the single statement definition.
5775cd3d5c8SMarek Sokolowski     ASSIGN_OR_RETURN(IDResult, readInt());
5785cd3d5c8SMarek Sokolowski     ASSIGN_OR_RETURN(StrResult, readString());
5795cd3d5c8SMarek Sokolowski     Table->addString(*IDResult, *StrResult);
5805cd3d5c8SMarek Sokolowski   }
5815cd3d5c8SMarek Sokolowski 
5825cd3d5c8SMarek Sokolowski   return std::move(Table);
5835cd3d5c8SMarek Sokolowski }
5845cd3d5c8SMarek Sokolowski 
585fb74cb1eSMarek Sokolowski Expected<std::unique_ptr<VersionInfoBlock>>
586fb74cb1eSMarek Sokolowski RCParser::parseVersionInfoBlockContents(StringRef BlockName) {
587fb74cb1eSMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::BlockBegin));
588fb74cb1eSMarek Sokolowski 
58999ead70fSMarek Sokolowski   auto Contents = llvm::make_unique<VersionInfoBlock>(BlockName);
590fb74cb1eSMarek Sokolowski 
591fb74cb1eSMarek Sokolowski   while (!isNextTokenKind(Kind::BlockEnd)) {
592fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(Stmt, parseVersionInfoStmt());
593fb74cb1eSMarek Sokolowski     Contents->addStmt(std::move(*Stmt));
594fb74cb1eSMarek Sokolowski   }
595fb74cb1eSMarek Sokolowski 
596fb74cb1eSMarek Sokolowski   consume(); // Consume BlockEnd.
597fb74cb1eSMarek Sokolowski 
598fb74cb1eSMarek Sokolowski   return std::move(Contents);
599fb74cb1eSMarek Sokolowski }
600fb74cb1eSMarek Sokolowski 
601fb74cb1eSMarek Sokolowski Expected<std::unique_ptr<VersionInfoStmt>> RCParser::parseVersionInfoStmt() {
602fb74cb1eSMarek Sokolowski   // Expect either BLOCK or VALUE, then a name or a key (a string).
603fb74cb1eSMarek Sokolowski   ASSIGN_OR_RETURN(TypeResult, readIdentifier());
604fb74cb1eSMarek Sokolowski 
605fb74cb1eSMarek Sokolowski   if (TypeResult->equals_lower("BLOCK")) {
606fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(NameResult, readString());
607fb74cb1eSMarek Sokolowski     return parseVersionInfoBlockContents(*NameResult);
608fb74cb1eSMarek Sokolowski   }
609fb74cb1eSMarek Sokolowski 
610fb74cb1eSMarek Sokolowski   if (TypeResult->equals_lower("VALUE")) {
611fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(KeyResult, readString());
61207bc04ffSZachary Turner     // Read a non-empty list of strings and/or ints, each
61307bc04ffSZachary Turner     // possibly preceded by a comma. Unfortunately, the tool behavior depends
61407bc04ffSZachary Turner     // on them existing or not, so we need to memorize where we found them.
615fb74cb1eSMarek Sokolowski     std::vector<IntOrString> Values;
61607bc04ffSZachary Turner     std::vector<bool> PrecedingCommas;
61707bc04ffSZachary Turner     RETURN_IF_ERROR(consumeType(Kind::Comma));
61807bc04ffSZachary Turner     while (!isNextTokenKind(Kind::Identifier) &&
61907bc04ffSZachary Turner            !isNextTokenKind(Kind::BlockEnd)) {
62007bc04ffSZachary Turner       // Try to eat a comma if it's not the first statement.
62107bc04ffSZachary Turner       bool HadComma = Values.size() > 0 && consumeOptionalType(Kind::Comma);
622fb74cb1eSMarek Sokolowski       ASSIGN_OR_RETURN(ValueResult, readIntOrString());
623fb74cb1eSMarek Sokolowski       Values.push_back(*ValueResult);
62407bc04ffSZachary Turner       PrecedingCommas.push_back(HadComma);
625fb74cb1eSMarek Sokolowski     }
62607bc04ffSZachary Turner     return llvm::make_unique<VersionInfoValue>(*KeyResult, std::move(Values),
62707bc04ffSZachary Turner                                                std::move(PrecedingCommas));
628fb74cb1eSMarek Sokolowski   }
629fb74cb1eSMarek Sokolowski 
630fb74cb1eSMarek Sokolowski   return getExpectedError("BLOCK or VALUE", true);
631fb74cb1eSMarek Sokolowski }
632fb74cb1eSMarek Sokolowski 
633fb74cb1eSMarek Sokolowski Expected<VersionInfoResource::VersionInfoFixed>
634fb74cb1eSMarek Sokolowski RCParser::parseVersionInfoFixed() {
635fb74cb1eSMarek Sokolowski   using RetType = VersionInfoResource::VersionInfoFixed;
636fb74cb1eSMarek Sokolowski   RetType Result;
637fb74cb1eSMarek Sokolowski 
638fb74cb1eSMarek Sokolowski   // Read until the beginning of the block.
639fb74cb1eSMarek Sokolowski   while (!isNextTokenKind(Kind::BlockBegin)) {
640fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(TypeResult, readIdentifier());
641fb74cb1eSMarek Sokolowski     auto FixedType = RetType::getFixedType(*TypeResult);
642fb74cb1eSMarek Sokolowski 
643fb74cb1eSMarek Sokolowski     if (!RetType::isTypeSupported(FixedType))
644fb74cb1eSMarek Sokolowski       return getExpectedError("fixed VERSIONINFO statement type", true);
645fb74cb1eSMarek Sokolowski     if (Result.IsTypePresent[FixedType])
646fb74cb1eSMarek Sokolowski       return getExpectedError("yet unread fixed VERSIONINFO statement type",
647fb74cb1eSMarek Sokolowski                               true);
648fb74cb1eSMarek Sokolowski 
649fb74cb1eSMarek Sokolowski     // VERSION variations take multiple integers.
650fb74cb1eSMarek Sokolowski     size_t NumInts = RetType::isVersionType(FixedType) ? 4 : 1;
651fb74cb1eSMarek Sokolowski     ASSIGN_OR_RETURN(ArgsResult, readIntsWithCommas(NumInts, NumInts));
65207bc04ffSZachary Turner     SmallVector<uint32_t, 4> ArgInts(ArgsResult->begin(), ArgsResult->end());
65307bc04ffSZachary Turner     Result.setValue(FixedType, ArgInts);
654fb74cb1eSMarek Sokolowski   }
655fb74cb1eSMarek Sokolowski 
656fb74cb1eSMarek Sokolowski   return Result;
657fb74cb1eSMarek Sokolowski }
658fb74cb1eSMarek Sokolowski 
6595cd3d5c8SMarek Sokolowski RCParser::ParseOptionType RCParser::parseLanguageStmt() {
6605cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Args, readIntsWithCommas(/* min = */ 2, /* max = */ 2));
6614a765da3SMarek Sokolowski   return llvm::make_unique<LanguageResource>((*Args)[0], (*Args)[1]);
6625cd3d5c8SMarek Sokolowski }
6635cd3d5c8SMarek Sokolowski 
6645cd3d5c8SMarek Sokolowski RCParser::ParseOptionType RCParser::parseCharacteristicsStmt() {
6655cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readInt());
6664a765da3SMarek Sokolowski   return llvm::make_unique<CharacteristicsStmt>(*Arg);
6675cd3d5c8SMarek Sokolowski }
6685cd3d5c8SMarek Sokolowski 
6695cd3d5c8SMarek Sokolowski RCParser::ParseOptionType RCParser::parseVersionStmt() {
6705cd3d5c8SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readInt());
6714a765da3SMarek Sokolowski   return llvm::make_unique<VersionStmt>(*Arg);
6725cd3d5c8SMarek Sokolowski }
6735cd3d5c8SMarek Sokolowski 
6744ac54d93SMarek Sokolowski RCParser::ParseOptionType RCParser::parseCaptionStmt() {
6754ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readString());
6764a765da3SMarek Sokolowski   return llvm::make_unique<CaptionStmt>(*Arg);
6774ac54d93SMarek Sokolowski }
6784ac54d93SMarek Sokolowski 
679420090afSZachary Turner RCParser::ParseOptionType RCParser::parseFontStmt(OptStmtType DialogType) {
680420090afSZachary Turner   assert(DialogType != OptStmtType::BasicStmt);
681420090afSZachary Turner 
6824ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(SizeResult, readInt());
6834ac54d93SMarek Sokolowski   RETURN_IF_ERROR(consumeType(Kind::Comma));
6844ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(NameResult, readString());
685420090afSZachary Turner 
686420090afSZachary Turner   // Default values for the optional arguments.
687420090afSZachary Turner   uint32_t FontWeight = 0;
688420090afSZachary Turner   bool FontItalic = false;
689420090afSZachary Turner   uint32_t FontCharset = 1;
690420090afSZachary Turner   if (DialogType == OptStmtType::DialogExStmt) {
691420090afSZachary Turner     if (consumeOptionalType(Kind::Comma)) {
692420090afSZachary Turner       ASSIGN_OR_RETURN(Args, readIntsWithCommas(/* min = */ 0, /* max = */ 3));
693420090afSZachary Turner       if (Args->size() >= 1)
694420090afSZachary Turner         FontWeight = (*Args)[0];
695420090afSZachary Turner       if (Args->size() >= 2)
696420090afSZachary Turner         FontItalic = (*Args)[1] != 0;
697420090afSZachary Turner       if (Args->size() >= 3)
698420090afSZachary Turner         FontCharset = (*Args)[2];
699420090afSZachary Turner     }
700420090afSZachary Turner   }
701420090afSZachary Turner   return llvm::make_unique<FontStmt>(*SizeResult, *NameResult, FontWeight,
702420090afSZachary Turner                                      FontItalic, FontCharset);
7034ac54d93SMarek Sokolowski }
7044ac54d93SMarek Sokolowski 
7054ac54d93SMarek Sokolowski RCParser::ParseOptionType RCParser::parseStyleStmt() {
7064ac54d93SMarek Sokolowski   ASSIGN_OR_RETURN(Arg, readInt());
7074a765da3SMarek Sokolowski   return llvm::make_unique<StyleStmt>(*Arg);
7084ac54d93SMarek Sokolowski }
7094ac54d93SMarek Sokolowski 
710514b7105SZachary Turner Error RCParser::getExpectedError(const Twine &Message, bool IsAlreadyRead) {
7115cd3d5c8SMarek Sokolowski   return make_error<ParserError>(
7125cd3d5c8SMarek Sokolowski       Message, IsAlreadyRead ? std::prev(CurLoc) : CurLoc, End);
7135cd3d5c8SMarek Sokolowski }
7145cd3d5c8SMarek Sokolowski 
7155cd3d5c8SMarek Sokolowski } // namespace rc
7165cd3d5c8SMarek Sokolowski } // namespace llvm
717