bf_compile_and_go.h

// Copyright 2014 Brian Quinlan
// See "LICENSE" file for details.
//
// Implements a BrainfuckRunner that compiles the complete source into amd64
// machine code. For a description of the team "compile and go", see:
// http://en.wikipedia.org/wiki/Compile_and_go_system

#ifndef BF_COMPILE_AND_GO_H_
#define BF_COMPILE_AND_GO_H_

#include <map>
#include <string>

#include "bf_runner.h"

using std::string;
using std::map;

class BrainfuckCompileAndGo : public BrainfuckRunner {
 public:
  BrainfuckCompileAndGo();
  virtual bool init(string::const_iterator start,
                    string::const_iterator end);
  virtual void* run(BrainfuckReader reader,
                    void* reader_arg,
                    BrainfuckWriter writer,
                    void* writer_arg,
                    void* memory);

  virtual ~BrainfuckCompileAndGo();

 private:
  int executable_size_;
  void* executable_;
  int exit_offset_;

  void add_jne_to_exit(string* code);
  void add_jl_to_exit(string* code);
  void add_jmp_to_offset(int offset, string* code);
  void add_jmp_to_exit(string* code);
  void emit_offset_table(map<int8_t, uint8_t>* offset_to_change,
                         int8_t* offset,
                         string *code);
  bool generate_sequence_code(string::const_iterator start,
                              string::const_iterator end,
                              string* code);
  bool generate_loop_code(string::const_iterator start,
                          string::const_iterator end,
                          string* code);
  void generate_read_code(string* code);
  void generate_write_code(string* code);
};

#endif  // BF_COMPILE_AND_GO_H_