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