jit/src/cc/parser.h

/*
 * =====================================================================
 * Copyright (c) 2012, PLUMgrid, http://plumgrid.com
 *
 * This source is subject to the PLUMgrid License.
 * All rights reserved.
 *
 * THIS CODE AND INFORMATION ARE PROVIDED "AS IS" WITHOUT WARRANTY OF
 * ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
 * PARTICULAR PURPOSE.
 *
 * PLUMgrid confidential information, delete if you are not the
 * intended recipient.
 *
 * =====================================================================
 */

#pragma once

#include <fstream> // NOLINT
#include "cc/node.h"
#include "cc/lexer.h"
#include "cc/scope.h"

namespace ebpf {
namespace cc {

using std::string;
using std::pair;

class Parser {
 public:
  explicit Parser(const string& infile)
      : root_node_(NULL), scopes_(new Scopes), in_(infile), lexer(&in_), parser(lexer, *this) {
    // parser.set_debug_level(1);
  }
  ~Parser() { delete root_node_; }
  int parse() {
    return parser.parse();
  }

  VariableDeclStmtNode* variable_add(VariableDeclStmtNode* decl);
  VariableDeclStmtNode* variable_add(VariableDeclStmtNode* decl, ExprNode* init_expr);
  StructVariableDeclStmtNode* variable_add(StructVariableDeclStmtNode* decl, ExprNodeList* args, bool is_kv);
  StmtNode* state_add(Scopes::StateScope* scope, IdentExprNode* id1, BlockStmtNode* body);
  StmtNode* timer_add(Scopes::StateScope* scope, BlockStmtNode* body);
  StmtNode* state_add(Scopes::StateScope* scope, IdentExprNode* id1, IdentExprNode* id2, BlockStmtNode* body);
  StmtNode* table_add(IdentExprNode* type, IdentExprNodeList* templates, IdentExprNode* id, string* size);
  StmtNode* struct_add(IdentExprNode* type, FormalList* formals);
  StmtNode* result_add(int token, IdentExprNode* id, FormalList* formals, BlockStmtNode* body);
  bool variable_exists(VariableDeclStmtNode* decl, bool search_local = true);
  bool table_exists(TableDeclStmtNode* decl, bool search_local = true);
  void add_pragma(const std::string& pr, const std::string& v) { pragmas_[pr] = v; }
  void set_loc(Node *n, const BisonParser::location_type &loc) const;
  std::string pragma(const std::string &name) const;

  Node* root_node_;
  Scopes::Ptr scopes_;
  std::map<std::string, std::string> pragmas_;
 private:
  std::ifstream in_;
  Lexer lexer;
  BisonParser parser;
};

}  // namespace cc
}  // namespace ebpf