xref: /netbsd-src/external/apache2/llvm/dist/clang/lib/AST/FormatString.cpp (revision e038c9c4676b0f19b1b7dd08a940c6ed64a6d5ae)
17330f729Sjoerg // FormatString.cpp - Common stuff for handling printf/scanf formats -*- C++ -*-
27330f729Sjoerg //
37330f729Sjoerg // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
47330f729Sjoerg // See https://llvm.org/LICENSE.txt for license information.
57330f729Sjoerg // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
67330f729Sjoerg //
77330f729Sjoerg //===----------------------------------------------------------------------===//
87330f729Sjoerg //
97330f729Sjoerg // Shared details for processing format strings of printf and scanf
107330f729Sjoerg // (and friends).
117330f729Sjoerg //
127330f729Sjoerg //===----------------------------------------------------------------------===//
137330f729Sjoerg 
147330f729Sjoerg #include "FormatStringParsing.h"
157330f729Sjoerg #include "clang/Basic/LangOptions.h"
167330f729Sjoerg #include "clang/Basic/TargetInfo.h"
177330f729Sjoerg #include "llvm/Support/ConvertUTF.h"
187330f729Sjoerg 
197330f729Sjoerg using clang::analyze_format_string::ArgType;
207330f729Sjoerg using clang::analyze_format_string::FormatStringHandler;
217330f729Sjoerg using clang::analyze_format_string::FormatSpecifier;
227330f729Sjoerg using clang::analyze_format_string::LengthModifier;
237330f729Sjoerg using clang::analyze_format_string::OptionalAmount;
247330f729Sjoerg using clang::analyze_format_string::PositionContext;
257330f729Sjoerg using clang::analyze_format_string::ConversionSpecifier;
267330f729Sjoerg using namespace clang;
277330f729Sjoerg 
287330f729Sjoerg // Key function to FormatStringHandler.
~FormatStringHandler()297330f729Sjoerg FormatStringHandler::~FormatStringHandler() {}
307330f729Sjoerg 
317330f729Sjoerg //===----------------------------------------------------------------------===//
327330f729Sjoerg // Functions for parsing format strings components in both printf and
337330f729Sjoerg // scanf format strings.
347330f729Sjoerg //===----------------------------------------------------------------------===//
357330f729Sjoerg 
367330f729Sjoerg OptionalAmount
ParseAmount(const char * & Beg,const char * E)377330f729Sjoerg clang::analyze_format_string::ParseAmount(const char *&Beg, const char *E) {
387330f729Sjoerg   const char *I = Beg;
397330f729Sjoerg   UpdateOnReturn <const char*> UpdateBeg(Beg, I);
407330f729Sjoerg 
417330f729Sjoerg   unsigned accumulator = 0;
427330f729Sjoerg   bool hasDigits = false;
437330f729Sjoerg 
447330f729Sjoerg   for ( ; I != E; ++I) {
457330f729Sjoerg     char c = *I;
467330f729Sjoerg     if (c >= '0' && c <= '9') {
477330f729Sjoerg       hasDigits = true;
487330f729Sjoerg       accumulator = (accumulator * 10) + (c - '0');
497330f729Sjoerg       continue;
507330f729Sjoerg     }
517330f729Sjoerg 
527330f729Sjoerg     if (hasDigits)
537330f729Sjoerg       return OptionalAmount(OptionalAmount::Constant, accumulator, Beg, I - Beg,
547330f729Sjoerg           false);
557330f729Sjoerg 
567330f729Sjoerg     break;
577330f729Sjoerg   }
587330f729Sjoerg 
597330f729Sjoerg   return OptionalAmount();
607330f729Sjoerg }
617330f729Sjoerg 
627330f729Sjoerg OptionalAmount
ParseNonPositionAmount(const char * & Beg,const char * E,unsigned & argIndex)637330f729Sjoerg clang::analyze_format_string::ParseNonPositionAmount(const char *&Beg,
647330f729Sjoerg                                                      const char *E,
657330f729Sjoerg                                                      unsigned &argIndex) {
667330f729Sjoerg   if (*Beg == '*') {
677330f729Sjoerg     ++Beg;
687330f729Sjoerg     return OptionalAmount(OptionalAmount::Arg, argIndex++, Beg, 0, false);
697330f729Sjoerg   }
707330f729Sjoerg 
717330f729Sjoerg   return ParseAmount(Beg, E);
727330f729Sjoerg }
737330f729Sjoerg 
747330f729Sjoerg OptionalAmount
ParsePositionAmount(FormatStringHandler & H,const char * Start,const char * & Beg,const char * E,PositionContext p)757330f729Sjoerg clang::analyze_format_string::ParsePositionAmount(FormatStringHandler &H,
767330f729Sjoerg                                                   const char *Start,
777330f729Sjoerg                                                   const char *&Beg,
787330f729Sjoerg                                                   const char *E,
797330f729Sjoerg                                                   PositionContext p) {
807330f729Sjoerg   if (*Beg == '*') {
817330f729Sjoerg     const char *I = Beg + 1;
827330f729Sjoerg     const OptionalAmount &Amt = ParseAmount(I, E);
837330f729Sjoerg 
847330f729Sjoerg     if (Amt.getHowSpecified() == OptionalAmount::NotSpecified) {
857330f729Sjoerg       H.HandleInvalidPosition(Beg, I - Beg, p);
867330f729Sjoerg       return OptionalAmount(false);
877330f729Sjoerg     }
887330f729Sjoerg 
897330f729Sjoerg     if (I == E) {
907330f729Sjoerg       // No more characters left?
917330f729Sjoerg       H.HandleIncompleteSpecifier(Start, E - Start);
927330f729Sjoerg       return OptionalAmount(false);
937330f729Sjoerg     }
947330f729Sjoerg 
957330f729Sjoerg     assert(Amt.getHowSpecified() == OptionalAmount::Constant);
967330f729Sjoerg 
977330f729Sjoerg     if (*I == '$') {
987330f729Sjoerg       // Handle positional arguments
997330f729Sjoerg 
1007330f729Sjoerg       // Special case: '*0$', since this is an easy mistake.
1017330f729Sjoerg       if (Amt.getConstantAmount() == 0) {
1027330f729Sjoerg         H.HandleZeroPosition(Beg, I - Beg + 1);
1037330f729Sjoerg         return OptionalAmount(false);
1047330f729Sjoerg       }
1057330f729Sjoerg 
1067330f729Sjoerg       const char *Tmp = Beg;
1077330f729Sjoerg       Beg = ++I;
1087330f729Sjoerg 
1097330f729Sjoerg       return OptionalAmount(OptionalAmount::Arg, Amt.getConstantAmount() - 1,
1107330f729Sjoerg                             Tmp, 0, true);
1117330f729Sjoerg     }
1127330f729Sjoerg 
1137330f729Sjoerg     H.HandleInvalidPosition(Beg, I - Beg, p);
1147330f729Sjoerg     return OptionalAmount(false);
1157330f729Sjoerg   }
1167330f729Sjoerg 
1177330f729Sjoerg   return ParseAmount(Beg, E);
1187330f729Sjoerg }
1197330f729Sjoerg 
1207330f729Sjoerg 
1217330f729Sjoerg bool
ParseFieldWidth(FormatStringHandler & H,FormatSpecifier & CS,const char * Start,const char * & Beg,const char * E,unsigned * argIndex)1227330f729Sjoerg clang::analyze_format_string::ParseFieldWidth(FormatStringHandler &H,
1237330f729Sjoerg                                               FormatSpecifier &CS,
1247330f729Sjoerg                                               const char *Start,
1257330f729Sjoerg                                               const char *&Beg, const char *E,
1267330f729Sjoerg                                               unsigned *argIndex) {
1277330f729Sjoerg   // FIXME: Support negative field widths.
1287330f729Sjoerg   if (argIndex) {
1297330f729Sjoerg     CS.setFieldWidth(ParseNonPositionAmount(Beg, E, *argIndex));
1307330f729Sjoerg   }
1317330f729Sjoerg   else {
1327330f729Sjoerg     const OptionalAmount Amt =
1337330f729Sjoerg       ParsePositionAmount(H, Start, Beg, E,
1347330f729Sjoerg                           analyze_format_string::FieldWidthPos);
1357330f729Sjoerg 
1367330f729Sjoerg     if (Amt.isInvalid())
1377330f729Sjoerg       return true;
1387330f729Sjoerg     CS.setFieldWidth(Amt);
1397330f729Sjoerg   }
1407330f729Sjoerg   return false;
1417330f729Sjoerg }
1427330f729Sjoerg 
1437330f729Sjoerg bool
ParseArgPosition(FormatStringHandler & H,FormatSpecifier & FS,const char * Start,const char * & Beg,const char * E)1447330f729Sjoerg clang::analyze_format_string::ParseArgPosition(FormatStringHandler &H,
1457330f729Sjoerg                                                FormatSpecifier &FS,
1467330f729Sjoerg                                                const char *Start,
1477330f729Sjoerg                                                const char *&Beg,
1487330f729Sjoerg                                                const char *E) {
1497330f729Sjoerg   const char *I = Beg;
1507330f729Sjoerg 
1517330f729Sjoerg   const OptionalAmount &Amt = ParseAmount(I, E);
1527330f729Sjoerg 
1537330f729Sjoerg   if (I == E) {
1547330f729Sjoerg     // No more characters left?
1557330f729Sjoerg     H.HandleIncompleteSpecifier(Start, E - Start);
1567330f729Sjoerg     return true;
1577330f729Sjoerg   }
1587330f729Sjoerg 
1597330f729Sjoerg   if (Amt.getHowSpecified() == OptionalAmount::Constant && *(I++) == '$') {
1607330f729Sjoerg     // Warn that positional arguments are non-standard.
1617330f729Sjoerg     H.HandlePosition(Start, I - Start);
1627330f729Sjoerg 
1637330f729Sjoerg     // Special case: '%0$', since this is an easy mistake.
1647330f729Sjoerg     if (Amt.getConstantAmount() == 0) {
1657330f729Sjoerg       H.HandleZeroPosition(Start, I - Start);
1667330f729Sjoerg       return true;
1677330f729Sjoerg     }
1687330f729Sjoerg 
1697330f729Sjoerg     FS.setArgIndex(Amt.getConstantAmount() - 1);
1707330f729Sjoerg     FS.setUsesPositionalArg();
1717330f729Sjoerg     // Update the caller's pointer if we decided to consume
1727330f729Sjoerg     // these characters.
1737330f729Sjoerg     Beg = I;
1747330f729Sjoerg     return false;
1757330f729Sjoerg   }
1767330f729Sjoerg 
1777330f729Sjoerg   return false;
1787330f729Sjoerg }
1797330f729Sjoerg 
1807330f729Sjoerg bool
ParseVectorModifier(FormatStringHandler & H,FormatSpecifier & FS,const char * & I,const char * E,const LangOptions & LO)1817330f729Sjoerg clang::analyze_format_string::ParseVectorModifier(FormatStringHandler &H,
1827330f729Sjoerg                                                   FormatSpecifier &FS,
1837330f729Sjoerg                                                   const char *&I,
1847330f729Sjoerg                                                   const char *E,
1857330f729Sjoerg                                                   const LangOptions &LO) {
1867330f729Sjoerg   if (!LO.OpenCL)
1877330f729Sjoerg     return false;
1887330f729Sjoerg 
1897330f729Sjoerg   const char *Start = I;
1907330f729Sjoerg   if (*I == 'v') {
1917330f729Sjoerg     ++I;
1927330f729Sjoerg 
1937330f729Sjoerg     if (I == E) {
1947330f729Sjoerg       H.HandleIncompleteSpecifier(Start, E - Start);
1957330f729Sjoerg       return true;
1967330f729Sjoerg     }
1977330f729Sjoerg 
1987330f729Sjoerg     OptionalAmount NumElts = ParseAmount(I, E);
1997330f729Sjoerg     if (NumElts.getHowSpecified() != OptionalAmount::Constant) {
2007330f729Sjoerg       H.HandleIncompleteSpecifier(Start, E - Start);
2017330f729Sjoerg       return true;
2027330f729Sjoerg     }
2037330f729Sjoerg 
2047330f729Sjoerg     FS.setVectorNumElts(NumElts);
2057330f729Sjoerg   }
2067330f729Sjoerg 
2077330f729Sjoerg   return false;
2087330f729Sjoerg }
2097330f729Sjoerg 
2107330f729Sjoerg bool
ParseLengthModifier(FormatSpecifier & FS,const char * & I,const char * E,const LangOptions & LO,bool IsScanf)2117330f729Sjoerg clang::analyze_format_string::ParseLengthModifier(FormatSpecifier &FS,
2127330f729Sjoerg                                                   const char *&I,
2137330f729Sjoerg                                                   const char *E,
2147330f729Sjoerg                                                   const LangOptions &LO,
2157330f729Sjoerg                                                   bool IsScanf) {
2167330f729Sjoerg   LengthModifier::Kind lmKind = LengthModifier::None;
2177330f729Sjoerg   const char *lmPosition = I;
2187330f729Sjoerg   switch (*I) {
2197330f729Sjoerg     default:
2207330f729Sjoerg       return false;
2217330f729Sjoerg     case 'h':
2227330f729Sjoerg       ++I;
2237330f729Sjoerg       if (I != E && *I == 'h') {
2247330f729Sjoerg         ++I;
2257330f729Sjoerg         lmKind = LengthModifier::AsChar;
2267330f729Sjoerg       } else if (I != E && *I == 'l' && LO.OpenCL) {
2277330f729Sjoerg         ++I;
2287330f729Sjoerg         lmKind = LengthModifier::AsShortLong;
2297330f729Sjoerg       } else {
2307330f729Sjoerg         lmKind = LengthModifier::AsShort;
2317330f729Sjoerg       }
2327330f729Sjoerg       break;
2337330f729Sjoerg     case 'l':
2347330f729Sjoerg       ++I;
2357330f729Sjoerg       if (I != E && *I == 'l') {
2367330f729Sjoerg         ++I;
2377330f729Sjoerg         lmKind = LengthModifier::AsLongLong;
2387330f729Sjoerg       } else {
2397330f729Sjoerg         lmKind = LengthModifier::AsLong;
2407330f729Sjoerg       }
2417330f729Sjoerg       break;
2427330f729Sjoerg     case 'j': lmKind = LengthModifier::AsIntMax;     ++I; break;
2437330f729Sjoerg     case 'z': lmKind = LengthModifier::AsSizeT;      ++I; break;
2447330f729Sjoerg     case 't': lmKind = LengthModifier::AsPtrDiff;    ++I; break;
2457330f729Sjoerg     case 'L': lmKind = LengthModifier::AsLongDouble; ++I; break;
2467330f729Sjoerg     case 'q': lmKind = LengthModifier::AsQuad;       ++I; break;
2477330f729Sjoerg     case 'a':
2487330f729Sjoerg       if (IsScanf && !LO.C99 && !LO.CPlusPlus11) {
2497330f729Sjoerg         // For scanf in C90, look at the next character to see if this should
2507330f729Sjoerg         // be parsed as the GNU extension 'a' length modifier. If not, this
2517330f729Sjoerg         // will be parsed as a conversion specifier.
2527330f729Sjoerg         ++I;
2537330f729Sjoerg         if (I != E && (*I == 's' || *I == 'S' || *I == '[')) {
2547330f729Sjoerg           lmKind = LengthModifier::AsAllocate;
2557330f729Sjoerg           break;
2567330f729Sjoerg         }
2577330f729Sjoerg         --I;
2587330f729Sjoerg       }
2597330f729Sjoerg       return false;
2607330f729Sjoerg     case 'm':
2617330f729Sjoerg       if (IsScanf) {
2627330f729Sjoerg         lmKind = LengthModifier::AsMAllocate;
2637330f729Sjoerg         ++I;
2647330f729Sjoerg         break;
2657330f729Sjoerg       }
2667330f729Sjoerg       return false;
2677330f729Sjoerg     // printf: AsInt64, AsInt32, AsInt3264
2687330f729Sjoerg     // scanf:  AsInt64
2697330f729Sjoerg     case 'I':
2707330f729Sjoerg       if (I + 1 != E && I + 2 != E) {
2717330f729Sjoerg         if (I[1] == '6' && I[2] == '4') {
2727330f729Sjoerg           I += 3;
2737330f729Sjoerg           lmKind = LengthModifier::AsInt64;
2747330f729Sjoerg           break;
2757330f729Sjoerg         }
2767330f729Sjoerg         if (IsScanf)
2777330f729Sjoerg           return false;
2787330f729Sjoerg 
2797330f729Sjoerg         if (I[1] == '3' && I[2] == '2') {
2807330f729Sjoerg           I += 3;
2817330f729Sjoerg           lmKind = LengthModifier::AsInt32;
2827330f729Sjoerg           break;
2837330f729Sjoerg         }
2847330f729Sjoerg       }
2857330f729Sjoerg       ++I;
2867330f729Sjoerg       lmKind = LengthModifier::AsInt3264;
2877330f729Sjoerg       break;
2887330f729Sjoerg     case 'w':
2897330f729Sjoerg       lmKind = LengthModifier::AsWide; ++I; break;
2907330f729Sjoerg   }
2917330f729Sjoerg   LengthModifier lm(lmPosition, lmKind);
2927330f729Sjoerg   FS.setLengthModifier(lm);
2937330f729Sjoerg   return true;
2947330f729Sjoerg }
2957330f729Sjoerg 
ParseUTF8InvalidSpecifier(const char * SpecifierBegin,const char * FmtStrEnd,unsigned & Len)2967330f729Sjoerg bool clang::analyze_format_string::ParseUTF8InvalidSpecifier(
2977330f729Sjoerg     const char *SpecifierBegin, const char *FmtStrEnd, unsigned &Len) {
2987330f729Sjoerg   if (SpecifierBegin + 1 >= FmtStrEnd)
2997330f729Sjoerg     return false;
3007330f729Sjoerg 
3017330f729Sjoerg   const llvm::UTF8 *SB =
3027330f729Sjoerg       reinterpret_cast<const llvm::UTF8 *>(SpecifierBegin + 1);
3037330f729Sjoerg   const llvm::UTF8 *SE = reinterpret_cast<const llvm::UTF8 *>(FmtStrEnd);
3047330f729Sjoerg   const char FirstByte = *SB;
3057330f729Sjoerg 
3067330f729Sjoerg   // If the invalid specifier is a multibyte UTF-8 string, return the
3077330f729Sjoerg   // total length accordingly so that the conversion specifier can be
3087330f729Sjoerg   // properly updated to reflect a complete UTF-8 specifier.
3097330f729Sjoerg   unsigned NumBytes = llvm::getNumBytesForUTF8(FirstByte);
3107330f729Sjoerg   if (NumBytes == 1)
3117330f729Sjoerg     return false;
3127330f729Sjoerg   if (SB + NumBytes > SE)
3137330f729Sjoerg     return false;
3147330f729Sjoerg 
3157330f729Sjoerg   Len = NumBytes + 1;
3167330f729Sjoerg   return true;
3177330f729Sjoerg }
3187330f729Sjoerg 
3197330f729Sjoerg //===----------------------------------------------------------------------===//
3207330f729Sjoerg // Methods on ArgType.
3217330f729Sjoerg //===----------------------------------------------------------------------===//
3227330f729Sjoerg 
3237330f729Sjoerg clang::analyze_format_string::ArgType::MatchKind
matchesType(ASTContext & C,QualType argTy) const3247330f729Sjoerg ArgType::matchesType(ASTContext &C, QualType argTy) const {
3257330f729Sjoerg   if (Ptr) {
3267330f729Sjoerg     // It has to be a pointer.
3277330f729Sjoerg     const PointerType *PT = argTy->getAs<PointerType>();
3287330f729Sjoerg     if (!PT)
3297330f729Sjoerg       return NoMatch;
3307330f729Sjoerg 
3317330f729Sjoerg     // We cannot write through a const qualified pointer.
3327330f729Sjoerg     if (PT->getPointeeType().isConstQualified())
3337330f729Sjoerg       return NoMatch;
3347330f729Sjoerg 
3357330f729Sjoerg     argTy = PT->getPointeeType();
3367330f729Sjoerg   }
3377330f729Sjoerg 
3387330f729Sjoerg   switch (K) {
3397330f729Sjoerg     case InvalidTy:
3407330f729Sjoerg       llvm_unreachable("ArgType must be valid");
3417330f729Sjoerg 
3427330f729Sjoerg     case UnknownTy:
3437330f729Sjoerg       return Match;
3447330f729Sjoerg 
3457330f729Sjoerg     case AnyCharTy: {
3467330f729Sjoerg       if (const EnumType *ETy = argTy->getAs<EnumType>()) {
3477330f729Sjoerg         // If the enum is incomplete we know nothing about the underlying type.
3487330f729Sjoerg         // Assume that it's 'int'.
3497330f729Sjoerg         if (!ETy->getDecl()->isComplete())
3507330f729Sjoerg           return NoMatch;
3517330f729Sjoerg         argTy = ETy->getDecl()->getIntegerType();
3527330f729Sjoerg       }
3537330f729Sjoerg 
3547330f729Sjoerg       if (const BuiltinType *BT = argTy->getAs<BuiltinType>())
3557330f729Sjoerg         switch (BT->getKind()) {
3567330f729Sjoerg           default:
3577330f729Sjoerg             break;
3587330f729Sjoerg           case BuiltinType::Char_S:
3597330f729Sjoerg           case BuiltinType::SChar:
3607330f729Sjoerg           case BuiltinType::UChar:
3617330f729Sjoerg           case BuiltinType::Char_U:
3627330f729Sjoerg           case BuiltinType::Bool:
3637330f729Sjoerg             return Match;
3647330f729Sjoerg         }
3657330f729Sjoerg       return NoMatch;
3667330f729Sjoerg     }
3677330f729Sjoerg 
3687330f729Sjoerg     case SpecificTy: {
3697330f729Sjoerg       if (const EnumType *ETy = argTy->getAs<EnumType>()) {
3707330f729Sjoerg         // If the enum is incomplete we know nothing about the underlying type.
3717330f729Sjoerg         // Assume that it's 'int'.
3727330f729Sjoerg         if (!ETy->getDecl()->isComplete())
3737330f729Sjoerg           argTy = C.IntTy;
3747330f729Sjoerg         else
3757330f729Sjoerg           argTy = ETy->getDecl()->getIntegerType();
3767330f729Sjoerg       }
3777330f729Sjoerg       argTy = C.getCanonicalType(argTy).getUnqualifiedType();
3787330f729Sjoerg 
3797330f729Sjoerg       if (T == argTy)
3807330f729Sjoerg         return Match;
3817330f729Sjoerg       // Check for "compatible types".
3827330f729Sjoerg       if (const BuiltinType *BT = argTy->getAs<BuiltinType>())
3837330f729Sjoerg         switch (BT->getKind()) {
3847330f729Sjoerg           default:
3857330f729Sjoerg             break;
3867330f729Sjoerg           case BuiltinType::Char_S:
3877330f729Sjoerg           case BuiltinType::SChar:
3887330f729Sjoerg           case BuiltinType::Char_U:
3897330f729Sjoerg           case BuiltinType::UChar:
3907330f729Sjoerg           case BuiltinType::Bool:
3917330f729Sjoerg             if (T == C.UnsignedShortTy || T == C.ShortTy)
3927330f729Sjoerg               return NoMatchTypeConfusion;
3937330f729Sjoerg             return T == C.UnsignedCharTy || T == C.SignedCharTy ? Match
3947330f729Sjoerg                                                                 : NoMatch;
3957330f729Sjoerg           case BuiltinType::Short:
3967330f729Sjoerg             return T == C.UnsignedShortTy ? Match : NoMatch;
3977330f729Sjoerg           case BuiltinType::UShort:
3987330f729Sjoerg             return T == C.ShortTy ? Match : NoMatch;
3997330f729Sjoerg           case BuiltinType::Int:
4007330f729Sjoerg             return T == C.UnsignedIntTy ? Match : NoMatch;
4017330f729Sjoerg           case BuiltinType::UInt:
4027330f729Sjoerg             return T == C.IntTy ? Match : NoMatch;
4037330f729Sjoerg           case BuiltinType::Long:
4047330f729Sjoerg             return T == C.UnsignedLongTy ? Match : NoMatch;
4057330f729Sjoerg           case BuiltinType::ULong:
4067330f729Sjoerg             return T == C.LongTy ? Match : NoMatch;
4077330f729Sjoerg           case BuiltinType::LongLong:
4087330f729Sjoerg             return T == C.UnsignedLongLongTy ? Match : NoMatch;
4097330f729Sjoerg           case BuiltinType::ULongLong:
4107330f729Sjoerg             return T == C.LongLongTy ? Match : NoMatch;
4117330f729Sjoerg         }
4127330f729Sjoerg       return NoMatch;
4137330f729Sjoerg     }
4147330f729Sjoerg 
4157330f729Sjoerg     case CStrTy: {
4167330f729Sjoerg       const PointerType *PT = argTy->getAs<PointerType>();
4177330f729Sjoerg       if (!PT)
4187330f729Sjoerg         return NoMatch;
4197330f729Sjoerg       QualType pointeeTy = PT->getPointeeType();
4207330f729Sjoerg       if (const BuiltinType *BT = pointeeTy->getAs<BuiltinType>())
4217330f729Sjoerg         switch (BT->getKind()) {
4227330f729Sjoerg           case BuiltinType::Char_U:
4237330f729Sjoerg           case BuiltinType::UChar:
4247330f729Sjoerg           case BuiltinType::Char_S:
4257330f729Sjoerg           case BuiltinType::SChar:
4267330f729Sjoerg             return Match;
4277330f729Sjoerg           default:
4287330f729Sjoerg             break;
4297330f729Sjoerg         }
4307330f729Sjoerg 
4317330f729Sjoerg       return NoMatch;
4327330f729Sjoerg     }
4337330f729Sjoerg 
4347330f729Sjoerg     case WCStrTy: {
4357330f729Sjoerg       const PointerType *PT = argTy->getAs<PointerType>();
4367330f729Sjoerg       if (!PT)
4377330f729Sjoerg         return NoMatch;
4387330f729Sjoerg       QualType pointeeTy =
4397330f729Sjoerg         C.getCanonicalType(PT->getPointeeType()).getUnqualifiedType();
4407330f729Sjoerg       return pointeeTy == C.getWideCharType() ? Match : NoMatch;
4417330f729Sjoerg     }
4427330f729Sjoerg 
4437330f729Sjoerg     case WIntTy: {
4447330f729Sjoerg       QualType WInt = C.getCanonicalType(C.getWIntType()).getUnqualifiedType();
4457330f729Sjoerg 
4467330f729Sjoerg       if (C.getCanonicalType(argTy).getUnqualifiedType() == WInt)
4477330f729Sjoerg         return Match;
4487330f729Sjoerg 
4497330f729Sjoerg       QualType PromoArg = argTy->isPromotableIntegerType()
4507330f729Sjoerg                               ? C.getPromotedIntegerType(argTy)
4517330f729Sjoerg                               : argTy;
4527330f729Sjoerg       PromoArg = C.getCanonicalType(PromoArg).getUnqualifiedType();
4537330f729Sjoerg 
4547330f729Sjoerg       // If the promoted argument is the corresponding signed type of the
4557330f729Sjoerg       // wint_t type, then it should match.
4567330f729Sjoerg       if (PromoArg->hasSignedIntegerRepresentation() &&
4577330f729Sjoerg           C.getCorrespondingUnsignedType(PromoArg) == WInt)
4587330f729Sjoerg         return Match;
4597330f729Sjoerg 
4607330f729Sjoerg       return WInt == PromoArg ? Match : NoMatch;
4617330f729Sjoerg     }
4627330f729Sjoerg 
4637330f729Sjoerg     case CPointerTy:
4647330f729Sjoerg       if (argTy->isVoidPointerType()) {
4657330f729Sjoerg         return Match;
4667330f729Sjoerg       } if (argTy->isPointerType() || argTy->isObjCObjectPointerType() ||
4677330f729Sjoerg             argTy->isBlockPointerType() || argTy->isNullPtrType()) {
4687330f729Sjoerg         return NoMatchPedantic;
4697330f729Sjoerg       } else {
4707330f729Sjoerg         return NoMatch;
4717330f729Sjoerg       }
4727330f729Sjoerg 
4737330f729Sjoerg     case ObjCPointerTy: {
4747330f729Sjoerg       if (argTy->getAs<ObjCObjectPointerType>() ||
4757330f729Sjoerg           argTy->getAs<BlockPointerType>())
4767330f729Sjoerg         return Match;
4777330f729Sjoerg 
4787330f729Sjoerg       // Handle implicit toll-free bridging.
4797330f729Sjoerg       if (const PointerType *PT = argTy->getAs<PointerType>()) {
4807330f729Sjoerg         // Things such as CFTypeRef are really just opaque pointers
4817330f729Sjoerg         // to C structs representing CF types that can often be bridged
4827330f729Sjoerg         // to Objective-C objects.  Since the compiler doesn't know which
4837330f729Sjoerg         // structs can be toll-free bridged, we just accept them all.
4847330f729Sjoerg         QualType pointee = PT->getPointeeType();
4857330f729Sjoerg         if (pointee->getAsStructureType() || pointee->isVoidType())
4867330f729Sjoerg           return Match;
4877330f729Sjoerg       }
4887330f729Sjoerg       return NoMatch;
4897330f729Sjoerg     }
4907330f729Sjoerg   }
4917330f729Sjoerg 
4927330f729Sjoerg   llvm_unreachable("Invalid ArgType Kind!");
4937330f729Sjoerg }
4947330f729Sjoerg 
makeVectorType(ASTContext & C,unsigned NumElts) const4957330f729Sjoerg ArgType ArgType::makeVectorType(ASTContext &C, unsigned NumElts) const {
4967330f729Sjoerg   // Check for valid vector element types.
4977330f729Sjoerg   if (T.isNull())
4987330f729Sjoerg     return ArgType::Invalid();
4997330f729Sjoerg 
5007330f729Sjoerg   QualType Vec = C.getExtVectorType(T, NumElts);
5017330f729Sjoerg   return ArgType(Vec, Name);
5027330f729Sjoerg }
5037330f729Sjoerg 
getRepresentativeType(ASTContext & C) const5047330f729Sjoerg QualType ArgType::getRepresentativeType(ASTContext &C) const {
5057330f729Sjoerg   QualType Res;
5067330f729Sjoerg   switch (K) {
5077330f729Sjoerg     case InvalidTy:
5087330f729Sjoerg       llvm_unreachable("No representative type for Invalid ArgType");
5097330f729Sjoerg     case UnknownTy:
5107330f729Sjoerg       llvm_unreachable("No representative type for Unknown ArgType");
5117330f729Sjoerg     case AnyCharTy:
5127330f729Sjoerg       Res = C.CharTy;
5137330f729Sjoerg       break;
5147330f729Sjoerg     case SpecificTy:
5157330f729Sjoerg       Res = T;
5167330f729Sjoerg       break;
5177330f729Sjoerg     case CStrTy:
5187330f729Sjoerg       Res = C.getPointerType(C.CharTy);
5197330f729Sjoerg       break;
5207330f729Sjoerg     case WCStrTy:
5217330f729Sjoerg       Res = C.getPointerType(C.getWideCharType());
5227330f729Sjoerg       break;
5237330f729Sjoerg     case ObjCPointerTy:
5247330f729Sjoerg       Res = C.ObjCBuiltinIdTy;
5257330f729Sjoerg       break;
5267330f729Sjoerg     case CPointerTy:
5277330f729Sjoerg       Res = C.VoidPtrTy;
5287330f729Sjoerg       break;
5297330f729Sjoerg     case WIntTy: {
5307330f729Sjoerg       Res = C.getWIntType();
5317330f729Sjoerg       break;
5327330f729Sjoerg     }
5337330f729Sjoerg   }
5347330f729Sjoerg 
5357330f729Sjoerg   if (Ptr)
5367330f729Sjoerg     Res = C.getPointerType(Res);
5377330f729Sjoerg   return Res;
5387330f729Sjoerg }
5397330f729Sjoerg 
getRepresentativeTypeName(ASTContext & C) const5407330f729Sjoerg std::string ArgType::getRepresentativeTypeName(ASTContext &C) const {
541*e038c9c4Sjoerg   std::string S = getRepresentativeType(C).getAsString(C.getPrintingPolicy());
5427330f729Sjoerg 
5437330f729Sjoerg   std::string Alias;
5447330f729Sjoerg   if (Name) {
5457330f729Sjoerg     // Use a specific name for this type, e.g. "size_t".
5467330f729Sjoerg     Alias = Name;
5477330f729Sjoerg     if (Ptr) {
5487330f729Sjoerg       // If ArgType is actually a pointer to T, append an asterisk.
5497330f729Sjoerg       Alias += (Alias[Alias.size()-1] == '*') ? "*" : " *";
5507330f729Sjoerg     }
5517330f729Sjoerg     // If Alias is the same as the underlying type, e.g. wchar_t, then drop it.
5527330f729Sjoerg     if (S == Alias)
5537330f729Sjoerg       Alias.clear();
5547330f729Sjoerg   }
5557330f729Sjoerg 
5567330f729Sjoerg   if (!Alias.empty())
5577330f729Sjoerg     return std::string("'") + Alias + "' (aka '" + S + "')";
5587330f729Sjoerg   return std::string("'") + S + "'";
5597330f729Sjoerg }
5607330f729Sjoerg 
5617330f729Sjoerg 
5627330f729Sjoerg //===----------------------------------------------------------------------===//
5637330f729Sjoerg // Methods on OptionalAmount.
5647330f729Sjoerg //===----------------------------------------------------------------------===//
5657330f729Sjoerg 
5667330f729Sjoerg ArgType
getArgType(ASTContext & Ctx) const5677330f729Sjoerg analyze_format_string::OptionalAmount::getArgType(ASTContext &Ctx) const {
5687330f729Sjoerg   return Ctx.IntTy;
5697330f729Sjoerg }
5707330f729Sjoerg 
5717330f729Sjoerg //===----------------------------------------------------------------------===//
5727330f729Sjoerg // Methods on LengthModifier.
5737330f729Sjoerg //===----------------------------------------------------------------------===//
5747330f729Sjoerg 
5757330f729Sjoerg const char *
toString() const5767330f729Sjoerg analyze_format_string::LengthModifier::toString() const {
5777330f729Sjoerg   switch (kind) {
5787330f729Sjoerg   case AsChar:
5797330f729Sjoerg     return "hh";
5807330f729Sjoerg   case AsShort:
5817330f729Sjoerg     return "h";
5827330f729Sjoerg   case AsShortLong:
5837330f729Sjoerg     return "hl";
5847330f729Sjoerg   case AsLong: // or AsWideChar
5857330f729Sjoerg     return "l";
5867330f729Sjoerg   case AsLongLong:
5877330f729Sjoerg     return "ll";
5887330f729Sjoerg   case AsQuad:
5897330f729Sjoerg     return "q";
5907330f729Sjoerg   case AsIntMax:
5917330f729Sjoerg     return "j";
5927330f729Sjoerg   case AsSizeT:
5937330f729Sjoerg     return "z";
5947330f729Sjoerg   case AsPtrDiff:
5957330f729Sjoerg     return "t";
5967330f729Sjoerg   case AsInt32:
5977330f729Sjoerg     return "I32";
5987330f729Sjoerg   case AsInt3264:
5997330f729Sjoerg     return "I";
6007330f729Sjoerg   case AsInt64:
6017330f729Sjoerg     return "I64";
6027330f729Sjoerg   case AsLongDouble:
6037330f729Sjoerg     return "L";
6047330f729Sjoerg   case AsAllocate:
6057330f729Sjoerg     return "a";
6067330f729Sjoerg   case AsMAllocate:
6077330f729Sjoerg     return "m";
6087330f729Sjoerg   case AsWide:
6097330f729Sjoerg     return "w";
6107330f729Sjoerg   case None:
6117330f729Sjoerg     return "";
6127330f729Sjoerg   }
6137330f729Sjoerg   return nullptr;
6147330f729Sjoerg }
6157330f729Sjoerg 
6167330f729Sjoerg //===----------------------------------------------------------------------===//
6177330f729Sjoerg // Methods on ConversionSpecifier.
6187330f729Sjoerg //===----------------------------------------------------------------------===//
6197330f729Sjoerg 
toString() const6207330f729Sjoerg const char *ConversionSpecifier::toString() const {
6217330f729Sjoerg   switch (kind) {
6227330f729Sjoerg   case dArg: return "d";
6237330f729Sjoerg   case DArg: return "D";
6247330f729Sjoerg   case iArg: return "i";
6257330f729Sjoerg   case oArg: return "o";
6267330f729Sjoerg   case OArg: return "O";
6277330f729Sjoerg   case uArg: return "u";
6287330f729Sjoerg   case UArg: return "U";
6297330f729Sjoerg   case xArg: return "x";
6307330f729Sjoerg   case XArg: return "X";
6317330f729Sjoerg   case fArg: return "f";
6327330f729Sjoerg   case FArg: return "F";
6337330f729Sjoerg   case eArg: return "e";
6347330f729Sjoerg   case EArg: return "E";
6357330f729Sjoerg   case gArg: return "g";
6367330f729Sjoerg   case GArg: return "G";
6377330f729Sjoerg   case aArg: return "a";
6387330f729Sjoerg   case AArg: return "A";
6397330f729Sjoerg   case cArg: return "c";
6407330f729Sjoerg   case sArg: return "s";
6417330f729Sjoerg   case pArg: return "p";
6427330f729Sjoerg   case PArg:
6437330f729Sjoerg     return "P";
6447330f729Sjoerg   case nArg: return "n";
6457330f729Sjoerg   case PercentArg:  return "%";
6467330f729Sjoerg   case ScanListArg: return "[";
6477330f729Sjoerg   case InvalidSpecifier: return nullptr;
6487330f729Sjoerg 
6497330f729Sjoerg   // POSIX unicode extensions.
6507330f729Sjoerg   case CArg: return "C";
6517330f729Sjoerg   case SArg: return "S";
6527330f729Sjoerg 
6537330f729Sjoerg   // Objective-C specific specifiers.
6547330f729Sjoerg   case ObjCObjArg: return "@";
6557330f729Sjoerg 
6567330f729Sjoerg   // FreeBSD kernel specific specifiers.
6577330f729Sjoerg   case FreeBSDbArg: return "b";
6587330f729Sjoerg   case FreeBSDDArg: return "D";
6597330f729Sjoerg   case FreeBSDrArg: return "r";
6607330f729Sjoerg   case FreeBSDyArg: return "y";
6617330f729Sjoerg 
6627330f729Sjoerg   // GlibC specific specifiers.
6637330f729Sjoerg   case PrintErrno: return "m";
6647330f729Sjoerg 
6657330f729Sjoerg   // MS specific specifiers.
6667330f729Sjoerg   case ZArg: return "Z";
6677330f729Sjoerg   }
6687330f729Sjoerg   return nullptr;
6697330f729Sjoerg }
6707330f729Sjoerg 
6717330f729Sjoerg Optional<ConversionSpecifier>
getStandardSpecifier() const6727330f729Sjoerg ConversionSpecifier::getStandardSpecifier() const {
6737330f729Sjoerg   ConversionSpecifier::Kind NewKind;
6747330f729Sjoerg 
6757330f729Sjoerg   switch (getKind()) {
6767330f729Sjoerg   default:
6777330f729Sjoerg     return None;
6787330f729Sjoerg   case DArg:
6797330f729Sjoerg     NewKind = dArg;
6807330f729Sjoerg     break;
6817330f729Sjoerg   case UArg:
6827330f729Sjoerg     NewKind = uArg;
6837330f729Sjoerg     break;
6847330f729Sjoerg   case OArg:
6857330f729Sjoerg     NewKind = oArg;
6867330f729Sjoerg     break;
6877330f729Sjoerg   }
6887330f729Sjoerg 
6897330f729Sjoerg   ConversionSpecifier FixedCS(*this);
6907330f729Sjoerg   FixedCS.setKind(NewKind);
6917330f729Sjoerg   return FixedCS;
6927330f729Sjoerg }
6937330f729Sjoerg 
6947330f729Sjoerg //===----------------------------------------------------------------------===//
6957330f729Sjoerg // Methods on OptionalAmount.
6967330f729Sjoerg //===----------------------------------------------------------------------===//
6977330f729Sjoerg 
toString(raw_ostream & os) const6987330f729Sjoerg void OptionalAmount::toString(raw_ostream &os) const {
6997330f729Sjoerg   switch (hs) {
7007330f729Sjoerg   case Invalid:
7017330f729Sjoerg   case NotSpecified:
7027330f729Sjoerg     return;
7037330f729Sjoerg   case Arg:
7047330f729Sjoerg     if (UsesDotPrefix)
7057330f729Sjoerg         os << ".";
7067330f729Sjoerg     if (usesPositionalArg())
7077330f729Sjoerg       os << "*" << getPositionalArgIndex() << "$";
7087330f729Sjoerg     else
7097330f729Sjoerg       os << "*";
7107330f729Sjoerg     break;
7117330f729Sjoerg   case Constant:
7127330f729Sjoerg     if (UsesDotPrefix)
7137330f729Sjoerg         os << ".";
7147330f729Sjoerg     os << amt;
7157330f729Sjoerg     break;
7167330f729Sjoerg   }
7177330f729Sjoerg }
7187330f729Sjoerg 
hasValidLengthModifier(const TargetInfo & Target,const LangOptions & LO) const7197330f729Sjoerg bool FormatSpecifier::hasValidLengthModifier(const TargetInfo &Target,
7207330f729Sjoerg                                              const LangOptions &LO) const {
7217330f729Sjoerg   switch (LM.getKind()) {
7227330f729Sjoerg     case LengthModifier::None:
7237330f729Sjoerg       return true;
7247330f729Sjoerg 
7257330f729Sjoerg     // Handle most integer flags
7267330f729Sjoerg     case LengthModifier::AsShort:
7277330f729Sjoerg       // Length modifier only applies to FP vectors.
7287330f729Sjoerg       if (LO.OpenCL && CS.isDoubleArg())
7297330f729Sjoerg         return !VectorNumElts.isInvalid();
7307330f729Sjoerg 
7317330f729Sjoerg       if (Target.getTriple().isOSMSVCRT()) {
7327330f729Sjoerg         switch (CS.getKind()) {
7337330f729Sjoerg           case ConversionSpecifier::cArg:
7347330f729Sjoerg           case ConversionSpecifier::CArg:
7357330f729Sjoerg           case ConversionSpecifier::sArg:
7367330f729Sjoerg           case ConversionSpecifier::SArg:
7377330f729Sjoerg           case ConversionSpecifier::ZArg:
7387330f729Sjoerg             return true;
7397330f729Sjoerg           default:
7407330f729Sjoerg             break;
7417330f729Sjoerg         }
7427330f729Sjoerg       }
7437330f729Sjoerg       LLVM_FALLTHROUGH;
7447330f729Sjoerg     case LengthModifier::AsChar:
7457330f729Sjoerg     case LengthModifier::AsLongLong:
7467330f729Sjoerg     case LengthModifier::AsQuad:
7477330f729Sjoerg     case LengthModifier::AsIntMax:
7487330f729Sjoerg     case LengthModifier::AsSizeT:
7497330f729Sjoerg     case LengthModifier::AsPtrDiff:
7507330f729Sjoerg       switch (CS.getKind()) {
7517330f729Sjoerg         case ConversionSpecifier::dArg:
7527330f729Sjoerg         case ConversionSpecifier::DArg:
7537330f729Sjoerg         case ConversionSpecifier::iArg:
7547330f729Sjoerg         case ConversionSpecifier::oArg:
7557330f729Sjoerg         case ConversionSpecifier::OArg:
7567330f729Sjoerg         case ConversionSpecifier::uArg:
7577330f729Sjoerg         case ConversionSpecifier::UArg:
7587330f729Sjoerg         case ConversionSpecifier::xArg:
7597330f729Sjoerg         case ConversionSpecifier::XArg:
7607330f729Sjoerg         case ConversionSpecifier::nArg:
7617330f729Sjoerg           return true;
7627330f729Sjoerg         case ConversionSpecifier::FreeBSDrArg:
7637330f729Sjoerg         case ConversionSpecifier::FreeBSDyArg:
7647330f729Sjoerg           return Target.getTriple().isOSFreeBSD() || Target.getTriple().isPS4();
7657330f729Sjoerg         default:
7667330f729Sjoerg           return false;
7677330f729Sjoerg       }
7687330f729Sjoerg 
7697330f729Sjoerg     case LengthModifier::AsShortLong:
7707330f729Sjoerg       return LO.OpenCL && !VectorNumElts.isInvalid();
7717330f729Sjoerg 
7727330f729Sjoerg     // Handle 'l' flag
7737330f729Sjoerg     case LengthModifier::AsLong: // or AsWideChar
7747330f729Sjoerg       if (CS.isDoubleArg()) {
7757330f729Sjoerg         // Invalid for OpenCL FP scalars.
7767330f729Sjoerg         if (LO.OpenCL && VectorNumElts.isInvalid())
7777330f729Sjoerg           return false;
7787330f729Sjoerg         return true;
7797330f729Sjoerg       }
7807330f729Sjoerg 
7817330f729Sjoerg       switch (CS.getKind()) {
7827330f729Sjoerg         case ConversionSpecifier::dArg:
7837330f729Sjoerg         case ConversionSpecifier::DArg:
7847330f729Sjoerg         case ConversionSpecifier::iArg:
7857330f729Sjoerg         case ConversionSpecifier::oArg:
7867330f729Sjoerg         case ConversionSpecifier::OArg:
7877330f729Sjoerg         case ConversionSpecifier::uArg:
7887330f729Sjoerg         case ConversionSpecifier::UArg:
7897330f729Sjoerg         case ConversionSpecifier::xArg:
7907330f729Sjoerg         case ConversionSpecifier::XArg:
7917330f729Sjoerg         case ConversionSpecifier::nArg:
7927330f729Sjoerg         case ConversionSpecifier::cArg:
7937330f729Sjoerg         case ConversionSpecifier::sArg:
7947330f729Sjoerg         case ConversionSpecifier::ScanListArg:
7957330f729Sjoerg         case ConversionSpecifier::ZArg:
7967330f729Sjoerg           return true;
7977330f729Sjoerg         case ConversionSpecifier::FreeBSDrArg:
7987330f729Sjoerg         case ConversionSpecifier::FreeBSDyArg:
7997330f729Sjoerg           return Target.getTriple().isOSFreeBSD() || Target.getTriple().isPS4();
8007330f729Sjoerg         default:
8017330f729Sjoerg           return false;
8027330f729Sjoerg       }
8037330f729Sjoerg 
8047330f729Sjoerg     case LengthModifier::AsLongDouble:
8057330f729Sjoerg       switch (CS.getKind()) {
8067330f729Sjoerg         case ConversionSpecifier::aArg:
8077330f729Sjoerg         case ConversionSpecifier::AArg:
8087330f729Sjoerg         case ConversionSpecifier::fArg:
8097330f729Sjoerg         case ConversionSpecifier::FArg:
8107330f729Sjoerg         case ConversionSpecifier::eArg:
8117330f729Sjoerg         case ConversionSpecifier::EArg:
8127330f729Sjoerg         case ConversionSpecifier::gArg:
8137330f729Sjoerg         case ConversionSpecifier::GArg:
8147330f729Sjoerg           return true;
8157330f729Sjoerg         // GNU libc extension.
8167330f729Sjoerg         case ConversionSpecifier::dArg:
8177330f729Sjoerg         case ConversionSpecifier::iArg:
8187330f729Sjoerg         case ConversionSpecifier::oArg:
8197330f729Sjoerg         case ConversionSpecifier::uArg:
8207330f729Sjoerg         case ConversionSpecifier::xArg:
8217330f729Sjoerg         case ConversionSpecifier::XArg:
8227330f729Sjoerg           return !Target.getTriple().isOSDarwin() &&
8237330f729Sjoerg                  !Target.getTriple().isOSWindows();
8247330f729Sjoerg         default:
8257330f729Sjoerg           return false;
8267330f729Sjoerg       }
8277330f729Sjoerg 
8287330f729Sjoerg     case LengthModifier::AsAllocate:
8297330f729Sjoerg       switch (CS.getKind()) {
8307330f729Sjoerg         case ConversionSpecifier::sArg:
8317330f729Sjoerg         case ConversionSpecifier::SArg:
8327330f729Sjoerg         case ConversionSpecifier::ScanListArg:
8337330f729Sjoerg           return true;
8347330f729Sjoerg         default:
8357330f729Sjoerg           return false;
8367330f729Sjoerg       }
8377330f729Sjoerg 
8387330f729Sjoerg     case LengthModifier::AsMAllocate:
8397330f729Sjoerg       switch (CS.getKind()) {
8407330f729Sjoerg         case ConversionSpecifier::cArg:
8417330f729Sjoerg         case ConversionSpecifier::CArg:
8427330f729Sjoerg         case ConversionSpecifier::sArg:
8437330f729Sjoerg         case ConversionSpecifier::SArg:
8447330f729Sjoerg         case ConversionSpecifier::ScanListArg:
8457330f729Sjoerg           return true;
8467330f729Sjoerg         default:
8477330f729Sjoerg           return false;
8487330f729Sjoerg       }
8497330f729Sjoerg     case LengthModifier::AsInt32:
8507330f729Sjoerg     case LengthModifier::AsInt3264:
8517330f729Sjoerg     case LengthModifier::AsInt64:
8527330f729Sjoerg       switch (CS.getKind()) {
8537330f729Sjoerg         case ConversionSpecifier::dArg:
8547330f729Sjoerg         case ConversionSpecifier::iArg:
8557330f729Sjoerg         case ConversionSpecifier::oArg:
8567330f729Sjoerg         case ConversionSpecifier::uArg:
8577330f729Sjoerg         case ConversionSpecifier::xArg:
8587330f729Sjoerg         case ConversionSpecifier::XArg:
8597330f729Sjoerg           return Target.getTriple().isOSMSVCRT();
8607330f729Sjoerg         default:
8617330f729Sjoerg           return false;
8627330f729Sjoerg       }
8637330f729Sjoerg     case LengthModifier::AsWide:
8647330f729Sjoerg       switch (CS.getKind()) {
8657330f729Sjoerg         case ConversionSpecifier::cArg:
8667330f729Sjoerg         case ConversionSpecifier::CArg:
8677330f729Sjoerg         case ConversionSpecifier::sArg:
8687330f729Sjoerg         case ConversionSpecifier::SArg:
8697330f729Sjoerg         case ConversionSpecifier::ZArg:
8707330f729Sjoerg           return Target.getTriple().isOSMSVCRT();
8717330f729Sjoerg         default:
8727330f729Sjoerg           return false;
8737330f729Sjoerg       }
8747330f729Sjoerg   }
8757330f729Sjoerg   llvm_unreachable("Invalid LengthModifier Kind!");
8767330f729Sjoerg }
8777330f729Sjoerg 
hasStandardLengthModifier() const8787330f729Sjoerg bool FormatSpecifier::hasStandardLengthModifier() const {
8797330f729Sjoerg   switch (LM.getKind()) {
8807330f729Sjoerg     case LengthModifier::None:
8817330f729Sjoerg     case LengthModifier::AsChar:
8827330f729Sjoerg     case LengthModifier::AsShort:
8837330f729Sjoerg     case LengthModifier::AsLong:
8847330f729Sjoerg     case LengthModifier::AsLongLong:
8857330f729Sjoerg     case LengthModifier::AsIntMax:
8867330f729Sjoerg     case LengthModifier::AsSizeT:
8877330f729Sjoerg     case LengthModifier::AsPtrDiff:
8887330f729Sjoerg     case LengthModifier::AsLongDouble:
8897330f729Sjoerg       return true;
8907330f729Sjoerg     case LengthModifier::AsAllocate:
8917330f729Sjoerg     case LengthModifier::AsMAllocate:
8927330f729Sjoerg     case LengthModifier::AsQuad:
8937330f729Sjoerg     case LengthModifier::AsInt32:
8947330f729Sjoerg     case LengthModifier::AsInt3264:
8957330f729Sjoerg     case LengthModifier::AsInt64:
8967330f729Sjoerg     case LengthModifier::AsWide:
8977330f729Sjoerg     case LengthModifier::AsShortLong: // ???
8987330f729Sjoerg       return false;
8997330f729Sjoerg   }
9007330f729Sjoerg   llvm_unreachable("Invalid LengthModifier Kind!");
9017330f729Sjoerg }
9027330f729Sjoerg 
hasStandardConversionSpecifier(const LangOptions & LangOpt) const9037330f729Sjoerg bool FormatSpecifier::hasStandardConversionSpecifier(
9047330f729Sjoerg     const LangOptions &LangOpt) const {
9057330f729Sjoerg   switch (CS.getKind()) {
9067330f729Sjoerg     case ConversionSpecifier::cArg:
9077330f729Sjoerg     case ConversionSpecifier::dArg:
9087330f729Sjoerg     case ConversionSpecifier::iArg:
9097330f729Sjoerg     case ConversionSpecifier::oArg:
9107330f729Sjoerg     case ConversionSpecifier::uArg:
9117330f729Sjoerg     case ConversionSpecifier::xArg:
9127330f729Sjoerg     case ConversionSpecifier::XArg:
9137330f729Sjoerg     case ConversionSpecifier::fArg:
9147330f729Sjoerg     case ConversionSpecifier::FArg:
9157330f729Sjoerg     case ConversionSpecifier::eArg:
9167330f729Sjoerg     case ConversionSpecifier::EArg:
9177330f729Sjoerg     case ConversionSpecifier::gArg:
9187330f729Sjoerg     case ConversionSpecifier::GArg:
9197330f729Sjoerg     case ConversionSpecifier::aArg:
9207330f729Sjoerg     case ConversionSpecifier::AArg:
9217330f729Sjoerg     case ConversionSpecifier::sArg:
9227330f729Sjoerg     case ConversionSpecifier::pArg:
9237330f729Sjoerg     case ConversionSpecifier::nArg:
9247330f729Sjoerg     case ConversionSpecifier::ObjCObjArg:
9257330f729Sjoerg     case ConversionSpecifier::ScanListArg:
9267330f729Sjoerg     case ConversionSpecifier::PercentArg:
9277330f729Sjoerg     case ConversionSpecifier::PArg:
9287330f729Sjoerg       return true;
9297330f729Sjoerg     case ConversionSpecifier::CArg:
9307330f729Sjoerg     case ConversionSpecifier::SArg:
9317330f729Sjoerg       return LangOpt.ObjC;
9327330f729Sjoerg     case ConversionSpecifier::InvalidSpecifier:
9337330f729Sjoerg     case ConversionSpecifier::FreeBSDbArg:
9347330f729Sjoerg     case ConversionSpecifier::FreeBSDDArg:
9357330f729Sjoerg     case ConversionSpecifier::FreeBSDrArg:
9367330f729Sjoerg     case ConversionSpecifier::FreeBSDyArg:
9377330f729Sjoerg     case ConversionSpecifier::PrintErrno:
9387330f729Sjoerg     case ConversionSpecifier::DArg:
9397330f729Sjoerg     case ConversionSpecifier::OArg:
9407330f729Sjoerg     case ConversionSpecifier::UArg:
9417330f729Sjoerg     case ConversionSpecifier::ZArg:
9427330f729Sjoerg       return false;
9437330f729Sjoerg   }
9447330f729Sjoerg   llvm_unreachable("Invalid ConversionSpecifier Kind!");
9457330f729Sjoerg }
9467330f729Sjoerg 
hasStandardLengthConversionCombination() const9477330f729Sjoerg bool FormatSpecifier::hasStandardLengthConversionCombination() const {
9487330f729Sjoerg   if (LM.getKind() == LengthModifier::AsLongDouble) {
9497330f729Sjoerg     switch(CS.getKind()) {
9507330f729Sjoerg         case ConversionSpecifier::dArg:
9517330f729Sjoerg         case ConversionSpecifier::iArg:
9527330f729Sjoerg         case ConversionSpecifier::oArg:
9537330f729Sjoerg         case ConversionSpecifier::uArg:
9547330f729Sjoerg         case ConversionSpecifier::xArg:
9557330f729Sjoerg         case ConversionSpecifier::XArg:
9567330f729Sjoerg           return false;
9577330f729Sjoerg         default:
9587330f729Sjoerg           return true;
9597330f729Sjoerg     }
9607330f729Sjoerg   }
9617330f729Sjoerg   return true;
9627330f729Sjoerg }
9637330f729Sjoerg 
getCorrectedLengthModifier() const9647330f729Sjoerg Optional<LengthModifier> FormatSpecifier::getCorrectedLengthModifier() const {
9657330f729Sjoerg   if (CS.isAnyIntArg() || CS.getKind() == ConversionSpecifier::nArg) {
9667330f729Sjoerg     if (LM.getKind() == LengthModifier::AsLongDouble ||
9677330f729Sjoerg         LM.getKind() == LengthModifier::AsQuad) {
9687330f729Sjoerg       LengthModifier FixedLM(LM);
9697330f729Sjoerg       FixedLM.setKind(LengthModifier::AsLongLong);
9707330f729Sjoerg       return FixedLM;
9717330f729Sjoerg     }
9727330f729Sjoerg   }
9737330f729Sjoerg 
9747330f729Sjoerg   return None;
9757330f729Sjoerg }
9767330f729Sjoerg 
namedTypeToLengthModifier(QualType QT,LengthModifier & LM)9777330f729Sjoerg bool FormatSpecifier::namedTypeToLengthModifier(QualType QT,
9787330f729Sjoerg                                                 LengthModifier &LM) {
9797330f729Sjoerg   assert(isa<TypedefType>(QT) && "Expected a TypedefType");
9807330f729Sjoerg   const TypedefNameDecl *Typedef = cast<TypedefType>(QT)->getDecl();
9817330f729Sjoerg 
9827330f729Sjoerg   for (;;) {
9837330f729Sjoerg     const IdentifierInfo *Identifier = Typedef->getIdentifier();
9847330f729Sjoerg     if (Identifier->getName() == "size_t") {
9857330f729Sjoerg       LM.setKind(LengthModifier::AsSizeT);
9867330f729Sjoerg       return true;
9877330f729Sjoerg     } else if (Identifier->getName() == "ssize_t") {
9887330f729Sjoerg       // Not C99, but common in Unix.
9897330f729Sjoerg       LM.setKind(LengthModifier::AsSizeT);
9907330f729Sjoerg       return true;
9917330f729Sjoerg     } else if (Identifier->getName() == "intmax_t") {
9927330f729Sjoerg       LM.setKind(LengthModifier::AsIntMax);
9937330f729Sjoerg       return true;
9947330f729Sjoerg     } else if (Identifier->getName() == "uintmax_t") {
9957330f729Sjoerg       LM.setKind(LengthModifier::AsIntMax);
9967330f729Sjoerg       return true;
9977330f729Sjoerg     } else if (Identifier->getName() == "ptrdiff_t") {
9987330f729Sjoerg       LM.setKind(LengthModifier::AsPtrDiff);
9997330f729Sjoerg       return true;
10007330f729Sjoerg     }
10017330f729Sjoerg 
10027330f729Sjoerg     QualType T = Typedef->getUnderlyingType();
10037330f729Sjoerg     if (!isa<TypedefType>(T))
10047330f729Sjoerg       break;
10057330f729Sjoerg 
10067330f729Sjoerg     Typedef = cast<TypedefType>(T)->getDecl();
10077330f729Sjoerg   }
10087330f729Sjoerg   return false;
10097330f729Sjoerg }
1010