Home | History | Annotate | Download | only in slang
      1 /*
      2  * Copyright 2010, The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *     http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #include "slang_pragma_recorder.h"
     18 
     19 #include <string>
     20 
     21 #include "clang/Basic/TokenKinds.h"
     22 
     23 #include "clang/Lex/Preprocessor.h"
     24 #include "clang/Lex/Token.h"
     25 
     26 namespace slang {
     27 
     28 bool PragmaRecorder::GetPragmaNameFromToken(const clang::Token &Token,
     29                                             std::string &PragmaName) {
     30   if (Token.isLiteral())
     31     PragmaName.assign(Token.getLiteralData(), Token.getLength());
     32   else if (Token.is(clang::tok::identifier))
     33     PragmaName.assign(Token.getIdentifierInfo()->getNameStart(),
     34                       Token.getIdentifierInfo()->getLength());
     35   else
     36     return false;
     37   return true;
     38 }
     39 
     40 bool PragmaRecorder::GetPragmaValueFromToken(const clang::Token &Token,
     41                                              std::string &PragmaValue) {
     42   // Same as the GetPragmaName()
     43   if (Token.is(clang::tok::r_paren))
     44     PragmaValue.clear();
     45   else
     46     return GetPragmaNameFromToken(Token, PragmaValue);
     47   return true;
     48 }
     49 
     50 PragmaRecorder::PragmaRecorder(PragmaList *Pragmas)
     51     : PragmaHandler(),
     52       mPragmas(Pragmas) {
     53   return;
     54 }
     55 
     56 void PragmaRecorder::HandlePragma(clang::Preprocessor &PP,
     57                                   clang::PragmaIntroducerKind Introducer,
     58                                   clang::Token &FirstToken) {
     59   clang::Token &CurrentToken = FirstToken;
     60   std::string PragmaName, PragmaValue = "";
     61   // Pragma in ACC should be a name/value pair
     62 
     63   if (GetPragmaNameFromToken(FirstToken, PragmaName)) {
     64     // start parsing the value '(' PragmaValue ')', if we have one.
     65     const clang::Token* NextToken = &PP.LookAhead(0);
     66 
     67     if (NextToken->is(clang::tok::l_paren))
     68       PP.LexUnexpandedToken(CurrentToken);
     69     else
     70       goto end_parsing_pragma_value;
     71 
     72     NextToken = &PP.LookAhead(0);
     73     if (GetPragmaValueFromToken(*NextToken, PragmaValue)) {
     74       PP.Lex(CurrentToken);
     75     } else {
     76       PP.LexUnexpandedToken(CurrentToken);
     77       PP.Diag(NextToken->getLocation(),
     78               PP.getDiagnostics().getCustomDiagID(
     79                   clang::DiagnosticsEngine::Error,
     80                   "expected value after '#pragma %0('")) << PragmaName;
     81       return;
     82     }
     83 
     84     if (!NextToken->is(clang::tok::r_paren)) {
     85       NextToken = &PP.LookAhead(0);
     86       if (NextToken->is(clang::tok::r_paren)) {
     87         PP.LexUnexpandedToken(CurrentToken);
     88       } else {
     89         PP.LexUnexpandedToken(CurrentToken);
     90         PP.Diag(NextToken->getLocation(),
     91                 PP.getDiagnostics().getCustomDiagID(
     92                     clang::DiagnosticsEngine::Error,
     93                     "missing ')' after '#pragma %0(%1'"))
     94             << PragmaName << PragmaValue;
     95         return;
     96       }
     97     }
     98   } else {
     99     PP.Diag(FirstToken.getLocation(),
    100             PP.getDiagnostics().getCustomDiagID(
    101                 clang::DiagnosticsEngine::Error,
    102                 "no pragma name or value"));
    103     return;
    104   }
    105 
    106  end_parsing_pragma_value:
    107 
    108   // PragmaValue may be an empty string.
    109   mPragmas->push_back(make_pair(PragmaName, PragmaValue));
    110 
    111   // Inform lex to eat the token
    112   PP.LexUnexpandedToken(CurrentToken);
    113 
    114   return;
    115 }
    116 
    117 }  // namespace slang
    118