PTHLexer.cpp revision e013d685c6689ac7ae103ee88acf573422d1ed6a
1274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//===--- PTHLexer.cpp - Lex from a token stream ---------------------------===//
2274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//
3274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//                     The LLVM Compiler Infrastructure
4274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//
5274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek// This file is distributed under the University of Illinois Open Source
6274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek// License. See LICENSE.TXT for details.
7274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//
8274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//===----------------------------------------------------------------------===//
9274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//
10274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek// This file implements the PTHLexer interface.
11274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//
12274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek//===----------------------------------------------------------------------===//
13274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek
140c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "clang/Basic/TokenKinds.h"
150c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "clang/Basic/FileManager.h"
160c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "clang/Basic/IdentifierTable.h"
179378ba44b3f46d697653003c784be87746e138d2Douglas Gregor#include "clang/Basic/OnDiskHashTable.h"
18274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek#include "clang/Lex/PTHLexer.h"
19274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek#include "clang/Lex/Preprocessor.h"
200c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "clang/Lex/PTHManager.h"
210c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "clang/Lex/Token.h"
220c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "clang/Lex/Preprocessor.h"
230c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek#include "llvm/ADT/OwningPtr.h"
242596e429a61602312bdd149786045b8a90cd2d10Daniel Dunbar#include "llvm/ADT/StringExtras.h"
252596e429a61602312bdd149786045b8a90cd2d10Daniel Dunbar#include "llvm/ADT/StringMap.h"
266f78c3b8b9343e7e9fbf2d457cccf00df6da5d47Chris Lattner#include "llvm/Support/MemoryBuffer.h"
27337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek#include <sys/stat.h>
28274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenekusing namespace clang;
299378ba44b3f46d697653003c784be87746e138d2Douglas Gregorusing namespace clang::io;
30274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek
317b78b7c6d96deb1e63f8d0655ee6fa53de0b65efTed Kremenek#define DISK_TOKEN_SIZE (1+1+2+4+4)
32268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
33e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek//===----------------------------------------------------------------------===//
34e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek// PTHLexer methods.
35e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek//===----------------------------------------------------------------------===//
36e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek
37da9d61c96c412f6babc7f824152609562f302388Chris LattnerPTHLexer::PTHLexer(Preprocessor &PP, FileID FID, const unsigned char *D,
38277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek                   const unsigned char *ppcond, PTHManager &PM)
392b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner  : PreprocessorLexer(&PP, FID), TokBuf(D), CurPtr(D), LastHashTokPtr(0),
40277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek    PPCond(ppcond), CurPPCondPtr(ppcond), PTHMgr(PM) {
411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
422b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner  FileStartLoc = PP.getSourceManager().getLocForStartOfFile(FID);
435f074266cc59563036c40516c814d63825723e20Ted Kremenek}
44274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek
45e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenekvoid PTHLexer::Lex(Token& Tok) {
46e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted KremenekLexNextToken:
47866bdf74547efe32c320554837ffce00fcc084feTed Kremenek
48866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  //===--------------------------------------==//
49866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  // Read the raw token data.
50866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  //===--------------------------------------==//
511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
52866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  // Shadow CurPtr into an automatic variable.
531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const unsigned char *CurPtrShadow = CurPtr;
54866bdf74547efe32c320554837ffce00fcc084feTed Kremenek
551b5285e1ba31975864da356b2ed927e87670e654Chris Lattner  // Read in the data for the token.
565ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  unsigned Word0 = ReadLE32(CurPtrShadow);
575ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  uint32_t IdentifierID = ReadLE32(CurPtrShadow);
585ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  uint32_t FileOffset = ReadLE32(CurPtrShadow);
591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
607b78b7c6d96deb1e63f8d0655ee6fa53de0b65efTed Kremenek  tok::TokenKind TKind = (tok::TokenKind) (Word0 & 0xFF);
617b78b7c6d96deb1e63f8d0655ee6fa53de0b65efTed Kremenek  Token::TokenFlags TFlags = (Token::TokenFlags) ((Word0 >> 8) & 0xFF);
62aff6ef8e7bc3c3739f984c390e0af693e60be064Chris Lattner  uint32_t Len = Word0 >> 16;
637b78b7c6d96deb1e63f8d0655ee6fa53de0b65efTed Kremenek
64aff6ef8e7bc3c3739f984c390e0af693e60be064Chris Lattner  CurPtr = CurPtrShadow;
651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
66866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  //===--------------------------------------==//
67866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  // Construct the token itself.
68866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  //===--------------------------------------==//
691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
70866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  Tok.startToken();
71898a0bb1972efb6e03cb1151412ec7392cef07deChris Lattner  Tok.setKind(TKind);
72898a0bb1972efb6e03cb1151412ec7392cef07deChris Lattner  Tok.setFlag(TFlags);
7359d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek  assert(!LexingRawMode);
742b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner  Tok.setLocation(FileStartLoc.getFileLocWithOffset(FileOffset));
75866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  Tok.setLength(Len);
7689d7ee9619d2dbdfa8d956a695c612a104a92cadTed Kremenek
77d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner  // Handle identifiers.
78277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  if (Tok.isLiteral()) {
79277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek    Tok.setLiteralData((const char*) (PTHMgr.SpellingBase + IdentifierID));
80277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  }
81277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  else if (IdentifierID) {
82d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner    MIOpt.ReadToken();
83d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner    IdentifierInfo *II = PTHMgr.GetIdentifierInfo(IdentifierID-1);
841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
85d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner    Tok.setIdentifierInfo(II);
861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
87863c486fcb6162495a94fddf7ac8409de2638995Chris Lattner    // Change the kind of this identifier to the appropriate token kind, e.g.
88863c486fcb6162495a94fddf7ac8409de2638995Chris Lattner    // turning "for" into a keyword.
89863c486fcb6162495a94fddf7ac8409de2638995Chris Lattner    Tok.setKind(II->getTokenID());
901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
91d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner    if (II->isHandleIdentifierCase())
92d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner      PP->HandleIdentifier(Tok);
93d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner    return;
94d0a69696acca62798dfc8b98f97c92bfa7fa0490Chris Lattner  }
951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
96866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  //===--------------------------------------==//
97866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  // Process the token.
98866bdf74547efe32c320554837ffce00fcc084feTed Kremenek  //===--------------------------------------==//
99898a0bb1972efb6e03cb1151412ec7392cef07deChris Lattner  if (TKind == tok::eof) {
100e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek    // Save the end-of-file token.
101e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek    EofToken = Tok;
1021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
103cd4e2aecde5bb7810715d5d5a88ac63ce7946f34Ted Kremenek    Preprocessor *PPCache = PP;
1041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10559d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    assert(!ParsingPreprocessorDirective);
10659d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    assert(!LexingRawMode);
1071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10859d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    // FIXME: Issue diagnostics similar to Lexer.
10959d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    if (PP->HandleEndOfFile(Tok, false))
110d6f53dc4951aace69014619761760addac9e59ecTed Kremenek      return;
1111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
112cd4e2aecde5bb7810715d5d5a88ac63ce7946f34Ted Kremenek    assert(PPCache && "Raw buffer::LexEndOfFile should return a token");
113cd4e2aecde5bb7810715d5d5a88ac63ce7946f34Ted Kremenek    return PPCache->Lex(Tok);
114cd4e2aecde5bb7810715d5d5a88ac63ce7946f34Ted Kremenek  }
1151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
116898a0bb1972efb6e03cb1151412ec7392cef07deChris Lattner  if (TKind == tok::hash && Tok.isAtStartOfLine()) {
11759d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    LastHashTokPtr = CurPtr - DISK_TOKEN_SIZE;
11859d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    assert(!LexingRawMode);
11959d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    PP->HandleDirective(Tok);
1201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12159d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    if (PP->isCurrentLexer(this))
12259d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek      goto LexNextToken;
1231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12459d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    return PP->Lex(Tok);
125e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  }
1261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
127898a0bb1972efb6e03cb1151412ec7392cef07deChris Lattner  if (TKind == tok::eom) {
12859d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    assert(ParsingPreprocessorDirective);
12959d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    ParsingPreprocessorDirective = false;
13059d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek    return;
13159d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek  }
132274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek
13359d08cb672136322375e5400578ee1fbd0947de2Ted Kremenek  MIOpt.ReadToken();
134cd4e2aecde5bb7810715d5d5a88ac63ce7946f34Ted Kremenek}
135cd4e2aecde5bb7810715d5d5a88ac63ce7946f34Ted Kremenek
136e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek// FIXME: We can just grab the last token instead of storing a copy
137e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek// into EofToken.
13859d08cb672136322375e5400578ee1fbd0947de2Ted Kremenekvoid PTHLexer::getEOF(Token& Tok) {
139defb7094c835998bb821e894253287625ce8c74dTed Kremenek  assert(EofToken.is(tok::eof));
1400c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  Tok = EofToken;
141274b20863a728cc6a31ee75c670e3733600c1531Ted Kremenek}
14217ff58a63197b398ae52697b088dc0fb8b255519Ted Kremenek
14317ff58a63197b398ae52697b088dc0fb8b255519Ted Kremenekvoid PTHLexer::DiscardToEndOfLine() {
14417ff58a63197b398ae52697b088dc0fb8b255519Ted Kremenek  assert(ParsingPreprocessorDirective && ParsingFilename == false &&
14517ff58a63197b398ae52697b088dc0fb8b255519Ted Kremenek         "Must be in a preprocessing directive!");
1464d35da2e41941965bbee8ed7e8c30e7c21000d71Ted Kremenek
147e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  // We assume that if the preprocessor wishes to discard to the end of
148e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  // the line that it also means to end the current preprocessor directive.
149e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  ParsingPreprocessorDirective = false;
1501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
15174c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek  // Skip tokens by only peeking at their token kind and the flags.
15274c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek  // We don't need to actually reconstruct full tokens from the token buffer.
15374c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek  // This saves some copies and it also reduces IdentifierInfo* lookup.
154da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* p = CurPtr;
15574c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek  while (1) {
15674c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    // Read the token kind.  Are we at the end of the file?
15774c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    tok::TokenKind x = (tok::TokenKind) (uint8_t) *p;
15874c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    if (x == tok::eof) break;
1591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
16074c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    // Read the token flags.  Are we at the start of the next line?
16174c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    Token::TokenFlags y = (Token::TokenFlags) (uint8_t) p[1];
16280d2f3059326f99ebf7c867db1c7f106ec9485f5Ted Kremenek    if (y & Token::StartOfLine) break;
16374c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek
16474c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    // Skip to the next token.
16574c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek    p += DISK_TOKEN_SIZE;
16674c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek  }
1671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
16874c3e6e5e95af08096aab415d1ce15f15ffff02aTed Kremenek  CurPtr = p;
16917ff58a63197b398ae52697b088dc0fb8b255519Ted Kremenek}
1700c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
171268ee7016a2811803989487c0ad3799486092c63Ted Kremenek/// SkipBlock - Used by Preprocessor to skip the current conditional block.
172268ee7016a2811803989487c0ad3799486092c63Ted Kremenekbool PTHLexer::SkipBlock() {
173268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  assert(CurPPCondPtr && "No cached PP conditional information.");
174268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  assert(LastHashTokPtr && "No known '#' token.");
1751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
176da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* HashEntryI = 0;
1771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  uint32_t Offset;
178268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  uint32_t TableIdx;
1791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
180268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  do {
18141a2660377d215d004fe413c03874bd066b5384cTed Kremenek    // Read the token offset from the side-table.
1825ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner    Offset = ReadLE32(CurPPCondPtr);
1831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // Read the target table index from the side-table.
1855ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner    TableIdx = ReadLE32(CurPPCondPtr);
1861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18741a2660377d215d004fe413c03874bd066b5384cTed Kremenek    // Compute the actual memory address of the '#' token data for this entry.
18841a2660377d215d004fe413c03874bd066b5384cTed Kremenek    HashEntryI = TokBuf + Offset;
18941a2660377d215d004fe413c03874bd066b5384cTed Kremenek
19041a2660377d215d004fe413c03874bd066b5384cTed Kremenek    // Optmization: "Sibling jumping".  #if...#else...#endif blocks can
19141a2660377d215d004fe413c03874bd066b5384cTed Kremenek    //  contain nested blocks.  In the side-table we can jump over these
19241a2660377d215d004fe413c03874bd066b5384cTed Kremenek    //  nested blocks instead of doing a linear search if the next "sibling"
19341a2660377d215d004fe413c03874bd066b5384cTed Kremenek    //  entry is not at a location greater than LastHashTokPtr.
19441a2660377d215d004fe413c03874bd066b5384cTed Kremenek    if (HashEntryI < LastHashTokPtr && TableIdx) {
19541a2660377d215d004fe413c03874bd066b5384cTed Kremenek      // In the side-table we are still at an entry for a '#' token that
19641a2660377d215d004fe413c03874bd066b5384cTed Kremenek      // is earlier than the last one we saw.  Check if the location we would
19741a2660377d215d004fe413c03874bd066b5384cTed Kremenek      // stride gets us closer.
198da9d61c96c412f6babc7f824152609562f302388Chris Lattner      const unsigned char* NextPPCondPtr =
199da9d61c96c412f6babc7f824152609562f302388Chris Lattner        PPCond + TableIdx*(sizeof(uint32_t)*2);
20041a2660377d215d004fe413c03874bd066b5384cTed Kremenek      assert(NextPPCondPtr >= CurPPCondPtr);
20141a2660377d215d004fe413c03874bd066b5384cTed Kremenek      // Read where we should jump to.
2025ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner      uint32_t TmpOffset = ReadLE32(NextPPCondPtr);
203da9d61c96c412f6babc7f824152609562f302388Chris Lattner      const unsigned char* HashEntryJ = TokBuf + TmpOffset;
2041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20541a2660377d215d004fe413c03874bd066b5384cTed Kremenek      if (HashEntryJ <= LastHashTokPtr) {
20641a2660377d215d004fe413c03874bd066b5384cTed Kremenek        // Jump directly to the next entry in the side table.
20741a2660377d215d004fe413c03874bd066b5384cTed Kremenek        HashEntryI = HashEntryJ;
20841a2660377d215d004fe413c03874bd066b5384cTed Kremenek        Offset = TmpOffset;
2095ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner        TableIdx = ReadLE32(NextPPCondPtr);
21041a2660377d215d004fe413c03874bd066b5384cTed Kremenek        CurPPCondPtr = NextPPCondPtr;
21141a2660377d215d004fe413c03874bd066b5384cTed Kremenek      }
21241a2660377d215d004fe413c03874bd066b5384cTed Kremenek    }
213268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  }
2141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  while (HashEntryI < LastHashTokPtr);
21541a2660377d215d004fe413c03874bd066b5384cTed Kremenek  assert(HashEntryI == LastHashTokPtr && "No PP-cond entry found for '#'");
216268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  assert(TableIdx && "No jumping from #endifs.");
2171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
218268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // Update our side-table iterator.
219da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* NextPPCondPtr = PPCond + TableIdx*(sizeof(uint32_t)*2);
220268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  assert(NextPPCondPtr >= CurPPCondPtr);
221268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  CurPPCondPtr = NextPPCondPtr;
2221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
223268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // Read where we should jump to.
2245ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  HashEntryI = TokBuf + ReadLE32(NextPPCondPtr);
2255ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  uint32_t NextIdx = ReadLE32(NextPPCondPtr);
2261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
227268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // By construction NextIdx will be zero if this is a #endif.  This is useful
228268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // to know to obviate lexing another token.
229268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  bool isEndif = NextIdx == 0;
2301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
231268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // This case can occur when we see something like this:
232268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  //
233268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  //  #if ...
234268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  //   /* a comment or nothing */
235268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  //  #elif
236268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  //
237268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // If we are skipping the first #if block it will be the case that CurPtr
238268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // already points 'elif'.  Just return.
2391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
24041a2660377d215d004fe413c03874bd066b5384cTed Kremenek  if (CurPtr > HashEntryI) {
24141a2660377d215d004fe413c03874bd066b5384cTed Kremenek    assert(CurPtr == HashEntryI + DISK_TOKEN_SIZE);
242268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // Did we reach a #endif?  If so, go ahead and consume that token as well.
243268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    if (isEndif)
244e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek      CurPtr += DISK_TOKEN_SIZE*2;
245268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    else
24641a2660377d215d004fe413c03874bd066b5384cTed Kremenek      LastHashTokPtr = HashEntryI;
2471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
248268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    return isEndif;
249268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  }
250268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
251268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // Otherwise, we need to advance.  Update CurPtr to point to the '#' token.
25241a2660377d215d004fe413c03874bd066b5384cTed Kremenek  CurPtr = HashEntryI;
2531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
254268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // Update the location of the last observed '#'.  This is useful if we
255268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // are skipping multiple blocks.
256268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  LastHashTokPtr = CurPtr;
257268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
258e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  // Skip the '#' token.
259da9d61c96c412f6babc7f824152609562f302388Chris Lattner  assert(((tok::TokenKind)*CurPtr) == tok::hash);
260e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  CurPtr += DISK_TOKEN_SIZE;
2611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
262268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // Did we reach a #endif?  If so, go ahead and consume that token as well.
263e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek  if (isEndif) { CurPtr += DISK_TOKEN_SIZE*2; }
264268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
265268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  return isEndif;
266268ee7016a2811803989487c0ad3799486092c63Ted Kremenek}
267268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
26830a12ec2a7f331d9e08acabe7cda853aaa7ba54bTed KremenekSourceLocation PTHLexer::getSourceLocation() {
2691b5285e1ba31975864da356b2ed927e87670e654Chris Lattner  // getSourceLocation is not on the hot path.  It is used to get the location
2701b5285e1ba31975864da356b2ed927e87670e654Chris Lattner  // of the next token when transitioning back to this lexer when done
27130a12ec2a7f331d9e08acabe7cda853aaa7ba54bTed Kremenek  // handling a #included file.  Just read the necessary data from the token
27230a12ec2a7f331d9e08acabe7cda853aaa7ba54bTed Kremenek  // data buffer to construct the SourceLocation object.
27330a12ec2a7f331d9e08acabe7cda853aaa7ba54bTed Kremenek  // NOTE: This is a virtual function; hence it is defined out-of-line.
274b248d53f2599d8e7b53b144b713e163ca521ffcaTed Kremenek  const unsigned char *OffsetPtr = CurPtr + (DISK_TOKEN_SIZE - 4);
2755ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  uint32_t Offset = ReadLE32(OffsetPtr);
2761b5285e1ba31975864da356b2ed927e87670e654Chris Lattner  return FileStartLoc.getFileLocWithOffset(Offset);
27730a12ec2a7f331d9e08acabe7cda853aaa7ba54bTed Kremenek}
27830a12ec2a7f331d9e08acabe7cda853aaa7ba54bTed Kremenek
2795f074266cc59563036c40516c814d63825723e20Ted Kremenek//===----------------------------------------------------------------------===//
280d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek// PTH file lookup: map from strings to file data.
281d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek//===----------------------------------------------------------------------===//
2820c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
2830c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek/// PTHFileLookup - This internal data structure is used by the PTHManager
2840c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek///  to map from FileEntry objects managed by FileManager to offsets within
2850c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek///  the PTH file.
2860c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremeneknamespace {
287d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenekclass VISIBILITY_HIDDEN PTHFileData {
288d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  const uint32_t TokenOff;
289d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  const uint32_t PPCondOff;
2900c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenekpublic:
291d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  PTHFileData(uint32_t tokenOff, uint32_t ppCondOff)
292d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek    : TokenOff(tokenOff), PPCondOff(ppCondOff) {}
2931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  uint32_t getTokenOffset() const { return TokenOff; }
2951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  uint32_t getPPCondOffset() const { return PPCondOff; }
296d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek};
2971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
299337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekclass VISIBILITY_HIDDEN PTHFileLookupCommonTrait {
3000c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenekpublic:
301a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  typedef std::pair<unsigned char, const char*> internal_key_type;
302a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek
303a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static unsigned ComputeHash(internal_key_type x) {
3042596e429a61602312bdd149786045b8a90cd2d10Daniel Dunbar    return llvm::HashString(x.second);
305d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  }
3061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
307d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  static std::pair<unsigned, unsigned>
308d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  ReadKeyDataLength(const unsigned char*& d) {
309a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    unsigned keyLen = (unsigned) ReadUnalignedLE16(d);
310a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    unsigned dataLen = (unsigned) *(d++);
311a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return std::make_pair(keyLen, dataLen);
3120c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  }
3131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
314a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static internal_key_type ReadKey(const unsigned char* d, unsigned) {
315a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    unsigned char k = *(d++); // Read the entry kind.
316a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return std::make_pair(k, (const char*) d);
317d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  }
318337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek};
3191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
320337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekclass VISIBILITY_HIDDEN PTHFileLookupTrait : public PTHFileLookupCommonTrait {
321337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekpublic:
322337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  typedef const FileEntry* external_key_type;
323337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  typedef PTHFileData      data_type;
3241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
325a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static internal_key_type GetInternalKey(const FileEntry* FE) {
326a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return std::make_pair((unsigned char) 0x1, FE->getName());
327337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  }
328a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek
329a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static bool EqualKey(internal_key_type a, internal_key_type b) {
330a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return a.first == b.first && strcmp(a.second, b.second) == 0;
3311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
3321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static PTHFileData ReadData(const internal_key_type& k,
3341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                              const unsigned char* d, unsigned) {
335a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    assert(k.first == 0x1 && "Only file lookups can match!");
336d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek    uint32_t x = ::ReadUnalignedLE32(d);
337d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek    uint32_t y = ::ReadUnalignedLE32(d);
3381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    return PTHFileData(x, y);
3390c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  }
3400c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek};
3417e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek
3427e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenekclass VISIBILITY_HIDDEN PTHStringLookupTrait {
3437e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenekpublic:
3441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  typedef uint32_t
3457e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek          data_type;
3467e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek
3477e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  typedef const std::pair<const char*, unsigned>
3487e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek          external_key_type;
3497e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek
3507e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  typedef external_key_type internal_key_type;
3511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3527e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  static bool EqualKey(const internal_key_type& a,
3537e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek                       const internal_key_type& b) {
3547e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    return (a.second == b.second) ? memcmp(a.first, b.first, a.second) == 0
3557e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek                                  : false;
3567e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  }
3571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3587e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  static unsigned ComputeHash(const internal_key_type& a) {
3592596e429a61602312bdd149786045b8a90cd2d10Daniel Dunbar    return llvm::HashString(llvm::StringRef(a.first, a.second));
3607e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  }
3611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3627e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  // This hopefully will just get inlined and removed by the optimizer.
3637e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  static const internal_key_type&
3647e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  GetInternalKey(const external_key_type& x) { return x; }
3651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3667e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  static std::pair<unsigned, unsigned>
3677e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  ReadKeyDataLength(const unsigned char*& d) {
3687e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    return std::make_pair((unsigned) ReadUnalignedLE16(d), sizeof(uint32_t));
3697e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  }
3701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3717e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  static std::pair<const char*, unsigned>
3727e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  ReadKey(const unsigned char* d, unsigned n) {
3737e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek      assert(n >= 2 && d[n-1] == '\0');
3747e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek      return std::make_pair((const char*) d, n-1);
3757e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    }
3761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
377a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static uint32_t ReadData(const internal_key_type& k, const unsigned char* d,
378a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek                           unsigned) {
3797e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    return ::ReadUnalignedLE32(d);
3807e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  }
3817e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek};
3821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump} // end anonymous namespace
384d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek
3857e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenektypedef OnDiskChainedHashTable<PTHFileLookupTrait>   PTHFileLookup;
3867e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenektypedef OnDiskChainedHashTable<PTHStringLookupTrait> PTHStringIdLookup;
3870c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
3880c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek//===----------------------------------------------------------------------===//
3890c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek// PTHManager methods.
3900c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek//===----------------------------------------------------------------------===//
3910c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
3920c6a77bc1f52f282a969538f139ebde429076ed3Ted KremenekPTHManager::PTHManager(const llvm::MemoryBuffer* buf, void* fileLookup,
393da9d61c96c412f6babc7f824152609562f302388Chris Lattner                       const unsigned char* idDataTable,
3941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                       IdentifierInfo** perIDCache,
3957e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek                       void* stringIdLookup, unsigned numIds,
39668228634016f644a1164fa1f024a9ce2093656bfTed Kremenek                       const unsigned char* spellingBase,
39768228634016f644a1164fa1f024a9ce2093656bfTed Kremenek                       const char* originalSourceFile)
3986183e4815a4019e97ad01bd880f12355599b75fdTed Kremenek: Buf(buf), PerIDCache(perIDCache), FileLookup(fileLookup),
3997e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  IdDataTable(idDataTable), StringIdLookup(stringIdLookup),
40068228634016f644a1164fa1f024a9ce2093656bfTed Kremenek  NumIds(numIds), PP(0), SpellingBase(spellingBase),
40168228634016f644a1164fa1f024a9ce2093656bfTed Kremenek  OriginalSourceFile(originalSourceFile) {}
4020c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
4030c6a77bc1f52f282a969538f139ebde429076ed3Ted KremenekPTHManager::~PTHManager() {
4040c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  delete Buf;
4050c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  delete (PTHFileLookup*) FileLookup;
4067e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  delete (PTHStringIdLookup*) StringIdLookup;
4070e50b6e7c104d00614baa3d80df62f1630a94d9cTed Kremenek  free(PerIDCache);
4080c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek}
4090c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
41022f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenekstatic void InvalidPTH(Diagnostic *Diags, Diagnostic::Level level,
41122f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek                       const char* Msg = 0) {
4121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  if (!Diags) return;
41326555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek  if (!Msg) Msg = "Invalid or corrupted PTH file";
41422f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek  unsigned DiagID = Diags->getCustomDiagID(level, Msg);
41526555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek  Diags->Report(FullSourceLoc(), DiagID);
41626555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek}
41726555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek
41822f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted KremenekPTHManager* PTHManager::Create(const std::string& file, Diagnostic* Diags,
41922f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek                               Diagnostic::Level level) {
4200c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Memory map the PTH file.
4210c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  llvm::OwningPtr<llvm::MemoryBuffer>
4220c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  File(llvm::MemoryBuffer::getFile(file.c_str()));
4231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4248a6aec620dbec1f292fe4116c0373ac81ab90234Ted Kremenek  if (!File) {
4258a6aec620dbec1f292fe4116c0373ac81ab90234Ted Kremenek    if (Diags) {
42622f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek      unsigned DiagID = Diags->getCustomDiagID(level,
4278a6aec620dbec1f292fe4116c0373ac81ab90234Ted Kremenek                                               "PTH file %0 could not be read");
4288a6aec620dbec1f292fe4116c0373ac81ab90234Ted Kremenek      Diags->Report(FullSourceLoc(), DiagID) << file;
4298a6aec620dbec1f292fe4116c0373ac81ab90234Ted Kremenek    }
4307e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek
4310c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek    return 0;
4328a6aec620dbec1f292fe4116c0373ac81ab90234Ted Kremenek  }
4331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4340c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Get the buffer ranges and check if there are at least three 32-bit
4350c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // words at the end of the file.
436da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* BufBeg = (unsigned char*)File->getBufferStart();
437da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* BufEnd = (unsigned char*)File->getBufferEnd();
438e1b6498c41b94c3bc5cede17b0702282543385efTed Kremenek
439e1b6498c41b94c3bc5cede17b0702282543385efTed Kremenek  // Check the prologue of the file.
4404adc71ae2cfc190f8d2cf58876e2a7893aa74ee0Ted Kremenek  if ((BufEnd - BufBeg) < (signed) (sizeof("cfe-pth") + 3 + 4) ||
44126555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek      memcmp(BufBeg, "cfe-pth", sizeof("cfe-pth") - 1) != 0) {
44222f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level);
443e1b6498c41b94c3bc5cede17b0702282543385efTed Kremenek    return 0;
44426555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek  }
4451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
44667d15050bbea16ae256e204ecd464f2e454c3c99Ted Kremenek  // Read the PTH version.
447e1b6498c41b94c3bc5cede17b0702282543385efTed Kremenek  const unsigned char *p = BufBeg + (sizeof("cfe-pth") - 1);
44867d15050bbea16ae256e204ecd464f2e454c3c99Ted Kremenek  unsigned Version = ReadLE32(p);
4491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
45026555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek  if (Version != PTHManager::Version) {
45122f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level,
4521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        Version < PTHManager::Version
45326555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek        ? "PTH file uses an older PTH format that is no longer supported"
45426555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek        : "PTH file uses a newer PTH format that cannot be read");
45567d15050bbea16ae256e204ecd464f2e454c3c99Ted Kremenek    return 0;
45626555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek  }
45767d15050bbea16ae256e204ecd464f2e454c3c99Ted Kremenek
4581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Compute the address of the index table at the end of the PTH file.
459a4bd8eb4d6d4b625f6bbb62fc180b02eab6433edTed Kremenek  const unsigned char *PrologueOffset = p;
4601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
461a4bd8eb4d6d4b625f6bbb62fc180b02eab6433edTed Kremenek  if (PrologueOffset >= BufEnd) {
46222f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level);
463e1b6498c41b94c3bc5cede17b0702282543385efTed Kremenek    return 0;
46426555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek  }
4651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4660c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Construct the file lookup table.  This will be used for mapping from
4670c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // FileEntry*'s to cached tokens.
468a4bd8eb4d6d4b625f6bbb62fc180b02eab6433edTed Kremenek  const unsigned char* FileTableOffset = PrologueOffset + sizeof(uint32_t)*2;
4695ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  const unsigned char* FileTable = BufBeg + ReadLE32(FileTableOffset);
4701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4710c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  if (!(FileTable > BufBeg && FileTable < BufEnd)) {
47222f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level);
4730c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek    return 0; // FIXME: Proper error diagnostic?
4740c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  }
4751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
476d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  llvm::OwningPtr<PTHFileLookup> FL(PTHFileLookup::Create(FileTable, BufBeg));
4771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4781d20197cb9478f3db17cbcfc57a1f8182843fc6aTed Kremenek  // Warn if the PTH file is empty.  We still want to create a PTHManager
4791d20197cb9478f3db17cbcfc57a1f8182843fc6aTed Kremenek  // as the PTH could be used with -include-pth.
4801d20197cb9478f3db17cbcfc57a1f8182843fc6aTed Kremenek  if (FL->isEmpty())
48122f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level, "PTH file contains no cached source data");
4821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4830c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Get the location of the table mapping from persistent ids to the
4840c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // data needed to reconstruct identifiers.
485a4bd8eb4d6d4b625f6bbb62fc180b02eab6433edTed Kremenek  const unsigned char* IDTableOffset = PrologueOffset + sizeof(uint32_t)*0;
4865ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  const unsigned char* IData = BufBeg + ReadLE32(IDTableOffset);
4871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
488cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek  if (!(IData >= BufBeg && IData < BufEnd)) {
48922f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level);
49026555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek    return 0;
4910c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  }
4921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4937e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  // Get the location of the hashtable mapping between strings and
4947e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  // persistent IDs.
495a4bd8eb4d6d4b625f6bbb62fc180b02eab6433edTed Kremenek  const unsigned char* StringIdTableOffset = PrologueOffset + sizeof(uint32_t)*1;
4967e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  const unsigned char* StringIdTable = BufBeg + ReadLE32(StringIdTableOffset);
4977e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  if (!(StringIdTable >= BufBeg && StringIdTable < BufEnd)) {
49822f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level);
49926555b18aa2c3b78744e77927acd3faa53ae7369Ted Kremenek    return 0;
50072b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  }
5017e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek
5027e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  llvm::OwningPtr<PTHStringIdLookup> SL(PTHStringIdLookup::Create(StringIdTable,
5037e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek                                                                  BufBeg));
5041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
505277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  // Get the location of the spelling cache.
506a4bd8eb4d6d4b625f6bbb62fc180b02eab6433edTed Kremenek  const unsigned char* spellingBaseOffset = PrologueOffset + sizeof(uint32_t)*3;
507277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  const unsigned char* spellingBase = BufBeg + ReadLE32(spellingBaseOffset);
508277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  if (!(spellingBase >= BufBeg && spellingBase < BufEnd)) {
50922f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek    InvalidPTH(Diags, level);
510277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek    return 0;
511277faca30c9f8f72b79f55695cbe3395ec246e7cTed Kremenek  }
5121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5136183e4815a4019e97ad01bd880f12355599b75fdTed Kremenek  // Get the number of IdentifierInfos and pre-allocate the identifier cache.
5145ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  uint32_t NumIds = ReadLE32(IData);
5151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5166183e4815a4019e97ad01bd880f12355599b75fdTed Kremenek  // Pre-allocate the peristent ID -> IdentifierInfo* cache.  We use calloc()
5176183e4815a4019e97ad01bd880f12355599b75fdTed Kremenek  // so that we in the best case only zero out memory once when the OS returns
5186183e4815a4019e97ad01bd880f12355599b75fdTed Kremenek  // us new pages.
519cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek  IdentifierInfo** PerIDCache = 0;
5201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
521cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek  if (NumIds) {
5221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    PerIDCache = (IdentifierInfo**)calloc(NumIds, sizeof(*PerIDCache));
523cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek    if (!PerIDCache) {
5241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      InvalidPTH(Diags, level,
52522f0d09298ef05136f1ef6ebc4469b482cbd73e7Ted Kremenek                 "Could not allocate memory for processing PTH file");
526cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek      return 0;
527cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek    }
5286183e4815a4019e97ad01bd880f12355599b75fdTed Kremenek  }
529cdd8f2153e18796e9e2a126ebcbd4f3e1bd7135bTed Kremenek
53068228634016f644a1164fa1f024a9ce2093656bfTed Kremenek  // Compute the address of the original source file.
53168228634016f644a1164fa1f024a9ce2093656bfTed Kremenek  const unsigned char* originalSourceBase = PrologueOffset + sizeof(uint32_t)*4;
53268228634016f644a1164fa1f024a9ce2093656bfTed Kremenek  unsigned len = ReadUnalignedLE16(originalSourceBase);
5331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  if (!len) originalSourceBase = 0;
5341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
53572b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  // Create the new PTHManager.
53672b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  return new PTHManager(File.take(), FL.take(), IData, PerIDCache,
53768228634016f644a1164fa1f024a9ce2093656bfTed Kremenek                        SL.take(), NumIds, spellingBase,
53868228634016f644a1164fa1f024a9ce2093656bfTed Kremenek                        (const char*) originalSourceBase);
5390c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek}
54068228634016f644a1164fa1f024a9ce2093656bfTed Kremenek
54177ecb3a28f21496ecfdbb3d5f5b66b0d2abf48c9Chris LattnerIdentifierInfo* PTHManager::LazilyCreateIdentifierInfo(unsigned PersistentID) {
5420c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Look in the PTH file for the string data for the IdentifierInfo object.
54377ecb3a28f21496ecfdbb3d5f5b66b0d2abf48c9Chris Lattner  const unsigned char* TableEntry = IdDataTable + sizeof(uint32_t)*PersistentID;
544da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* IDData =
5455ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner    (const unsigned char*)Buf->getBufferStart() + ReadLE32(TableEntry);
546da9d61c96c412f6babc7f824152609562f302388Chris Lattner  assert(IDData < (const unsigned char*)Buf->getBufferEnd());
5471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
54872b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  // Allocate the object.
549da9d61c96c412f6babc7f824152609562f302388Chris Lattner  std::pair<IdentifierInfo,const unsigned char*> *Mem =
550da9d61c96c412f6babc7f824152609562f302388Chris Lattner    Alloc.Allocate<std::pair<IdentifierInfo,const unsigned char*> >();
55172b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek
55272b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  Mem->second = IDData;
5537e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  assert(IDData[0] != '\0');
554ea9c26b3dbd74a1497f5609ae6e19a85f42b6073Ted Kremenek  IdentifierInfo *II = new ((void*) Mem) IdentifierInfo();
5551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
55672b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  // Store the new IdentifierInfo in the cache.
55777ecb3a28f21496ecfdbb3d5f5b66b0d2abf48c9Chris Lattner  PerIDCache[PersistentID] = II;
558e013d685c6689ac7ae103ee88acf573422d1ed6aDaniel Dunbar  assert(II->getNameStart() && II->getNameStart()[0] != '\0');
5590c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  return II;
5600c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek}
5610c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek
56272b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted KremenekIdentifierInfo* PTHManager::get(const char *NameStart, const char *NameEnd) {
5637e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  PTHStringIdLookup& SL = *((PTHStringIdLookup*)StringIdLookup);
5647e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  // Double check our assumption that the last character isn't '\0'.
56582320e94606cfb0f68c6049c070c3600f8df9081Daniel Dunbar  assert(NameEnd==NameStart || NameStart[NameEnd-NameStart-1] != '\0');
5667e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  PTHStringIdLookup::iterator I = SL.find(std::make_pair(NameStart,
5677e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek                                                         NameEnd - NameStart));
5687e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  if (I == SL.end()) // No identifier found?
5697e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek    return 0;
57072b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek
5717e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  // Match found.  Return the identifier!
5727e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  assert(*I > 0);
5737e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek  return GetIdentifierInfo(*I-1);
5747e3a004c6ed1fe87912203b9c5a113f8da89d261Ted Kremenek}
57572b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek
576f056d92e182cbe4f62c8d14102544dc38066dabcChris LattnerPTHLexer *PTHManager::CreateLexer(FileID FID) {
577f056d92e182cbe4f62c8d14102544dc38066dabcChris Lattner  const FileEntry *FE = PP->getSourceManager().getFileEntryForID(FID);
5780c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  if (!FE)
5790c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek    return 0;
5801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5810c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Lookup the FileEntry object in our file lookup data structure.  It will
5820c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // return a variant that indicates whether or not there is an offset within
5830c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // the PTH file that contains cached tokens.
584d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  PTHFileLookup& PFL = *((PTHFileLookup*)FileLookup);
585d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  PTHFileLookup::iterator I = PFL.find(FE);
5861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
587d8c02929fe70f03111be73e7b8c402c724238ee9Ted Kremenek  if (I == PFL.end()) // No tokens available?
5880c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek    return 0;
5891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const PTHFileData& FileData = *I;
5911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
592da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char *BufStart = (const unsigned char *)Buf->getBufferStart();
5930c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek  // Compute the offset of the token data within the buffer.
594da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* data = BufStart + FileData.getTokenOffset();
595268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
596268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  // Get the location of pp-conditional table.
597da9d61c96c412f6babc7f824152609562f302388Chris Lattner  const unsigned char* ppcond = BufStart + FileData.getPPCondOffset();
5985ff4317536dbd7f03332bb250c8b35ec04a6f5dbChris Lattner  uint32_t Len = ReadLE32(ppcond);
5991b5285e1ba31975864da356b2ed927e87670e654Chris Lattner  if (Len == 0) ppcond = 0;
6001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
60172b1b15ee88aac0a63e2c1dc53fe22f5ab297b20Ted Kremenek  assert(PP && "No preprocessor set yet!");
6021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  return new PTHLexer(*PP, FID, data, ppcond, *this);
6030c6a77bc1f52f282a969538f139ebde429076ed3Ted Kremenek}
604337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
605337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek//===----------------------------------------------------------------------===//
606337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek// 'stat' caching.
607337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek//===----------------------------------------------------------------------===//
608337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
609337edcdbec05316b407d0d64865c88ff8597d910Ted Kremeneknamespace {
610337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekclass VISIBILITY_HIDDEN PTHStatData {
611337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekpublic:
612a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  const bool hasStat;
613337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  const ino_t ino;
614337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  const dev_t dev;
615337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  const mode_t mode;
616337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  const time_t mtime;
617337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  const off_t size;
6181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
619337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  PTHStatData(ino_t i, dev_t d, mode_t mo, time_t m, off_t s)
6201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  : hasStat(true), ino(i), dev(d), mode(mo), mtime(m), size(s) {}
6211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
622a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  PTHStatData()
623a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    : hasStat(false), ino(0), dev(0), mode(0), mtime(0), size(0) {}
624337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek};
6251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
626337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekclass VISIBILITY_HIDDEN PTHStatLookupTrait : public PTHFileLookupCommonTrait {
627337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekpublic:
628a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  typedef const char* external_key_type;  // const char*
629337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  typedef PTHStatData data_type;
6301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
631a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static internal_key_type GetInternalKey(const char *path) {
632a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    // The key 'kind' doesn't matter here because it is ignored in EqualKey.
633a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return std::make_pair((unsigned char) 0x0, path);
634a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  }
635a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek
636a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static bool EqualKey(internal_key_type a, internal_key_type b) {
637a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    // When doing 'stat' lookups we don't care about the kind of 'a' and 'b',
638a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    // just the paths.
639a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return strcmp(a.second, b.second) == 0;
6401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  }
6411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
642a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek  static data_type ReadData(const internal_key_type& k, const unsigned char* d,
6431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                            unsigned) {
6441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
645ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek    if (k.first /* File or Directory */) {
646ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek      if (k.first == 0x1 /* File */) d += 4 * 2; // Skip the first 2 words.
647a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek      ino_t ino = (ino_t) ReadUnalignedLE32(d);
648a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek      dev_t dev = (dev_t) ReadUnalignedLE32(d);
649a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek      mode_t mode = (mode_t) ReadUnalignedLE16(d);
6501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      time_t mtime = (time_t) ReadUnalignedLE64(d);
651a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek      return data_type(ino, dev, mode, mtime, (off_t) ReadUnalignedLE64(d));
652a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    }
653ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek
654ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek    // Negative stat.  Don't read anything.
655a4b44dd9d30929a35c44d85102e5241ee847b2f2Ted Kremenek    return data_type();
656337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  }
657337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek};
658337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
659337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenekclass VISIBILITY_HIDDEN PTHStatCache : public StatSysCallCache {
660337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  typedef OnDiskChainedHashTable<PTHStatLookupTrait> CacheTy;
661337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  CacheTy Cache;
662337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
6631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumppublic:
664337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  PTHStatCache(PTHFileLookup &FL) :
665337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    Cache(FL.getNumBuckets(), FL.getNumEntries(), FL.getBuckets(),
666337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek          FL.getBase()) {}
667337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
668337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  ~PTHStatCache() {}
6691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
670337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  int stat(const char *path, struct stat *buf) {
671337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    // Do the lookup for the file's data in the PTH file.
672337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    CacheTy::iterator I = Cache.find(path);
673337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
674337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    // If we don't get a hit in the PTH file just forward to 'stat'.
67552e7108f51a4a9f4d6e84f33fb594d06e1d79560Douglas Gregor    if (I == Cache.end())
67652e7108f51a4a9f4d6e84f33fb594d06e1d79560Douglas Gregor      return StatSysCallCache::stat(path, buf);
6771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
678337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    const PTHStatData& Data = *I;
6791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
680ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek    if (!Data.hasStat)
681ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek      return 1;
682ad6ce5cb0e3b5dd853d2b1e337797a6fc1091e21Ted Kremenek
683337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    buf->st_ino = Data.ino;
684337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    buf->st_dev = Data.dev;
685337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    buf->st_mtime = Data.mtime;
686337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    buf->st_mode = Data.mode;
687337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    buf->st_size = Data.size;
688337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek    return 0;
689337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek  }
690337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek};
691d578569d50adc6e9b1a2f91931add6b471b2378eTed Kremenek} // end anonymous namespace
692337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek
693337edcdbec05316b407d0d64865c88ff8597d910Ted KremenekStatSysCallCache *PTHManager::createStatCache() {
6945f747d187bf8dc779a4acc180f9c21f1dbf83f6aTed Kremenek  return new PTHStatCache(*((PTHFileLookup*) FileLookup));
695337edcdbec05316b407d0d64865c88ff8597d910Ted Kremenek}
696