src/analysis/cfg.h

/*
 * Copyright 2023 WebAssembly Community Group participants
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

// A generic CFG / basic block utility. Unlike the utilities in src/cfg/, this
// is a generic representation of a CFG rather than a generic builder of
// CFG-like objects. It lives here in src/analysis/ because it is primarily
// meant for use in the static analysis framework. Other Binaryen code will find
// it more idiomatic to use the utilities in src/cfg/.

#ifndef wasm_analysis_cfg_h
#define wasm_analysis_cfg_h

#include <iostream>
#include <vector>

#include "wasm.h"

namespace wasm::analysis {

struct CFG;

struct BasicBlock {
  using iterator = std::vector<Expression*>::const_iterator;

  // Iterate through instructions.
  iterator begin() const { return insts.cbegin(); }
  iterator end() const { return insts.cend(); }
  size_t size() const { return insts.size(); }

  // Iterables for predecessor and successor blocks.
  struct Predecessors;
  struct Successors;
  Predecessors preds() const;
  Successors succs() const;

  void print(std::ostream& os, Module* wasm = nullptr, size_t start = 0) const;

private:
  Index index;
  std::vector<Expression*> insts;
  std::vector<BasicBlock*> predecessors;
  std::vector<BasicBlock*> successors;
  friend CFG;
};

struct CFG {
  // Iterate through basic blocks.
  using iterator = std::vector<BasicBlock>::const_iterator;
  iterator begin() const { return blocks.cbegin(); }
  iterator end() const { return blocks.cend(); }
  size_t size() const { return blocks.size(); }

  static CFG fromFunction(Function* func);

  void print(std::ostream& os, Module* wasm = nullptr) const;

private:
  std::vector<BasicBlock> blocks;
  friend BasicBlock;
};

} // namespace wasm::analysis

#include "cfg-impl.h"

#endif // wasm_analysis_cfg_h