test_alloc.cpp

#include "disasm.h"
#include <iostream>
#include <elfio/elfio.hpp>
#include "cfg.h"
#include "analysis.h"
#include "reg-alloc.h"

int fastrand() { 
	static int g_seed;
	g_seed = (214013*g_seed+2531011); 
	return (g_seed>>16)&0x7FFF; 
}

static void replace_reg(basic_block& bb, std::map<vreg, vreg>& replace_map) {
	if (bb.visited)
		return;

	for (vins& in : bb) {
		for (vreg& reg : in.regs) {
			auto f = replace_map.find(reg);
			if (f != replace_map.end()) {
				reg = f->second;
			}
		}
	}

	bb.visited = true;

	for (basic_block* succ : bb.successors) {
		replace_reg(*succ, replace_map);
	}

	for (basic_block* pred : bb.predecessors) {
		replace_reg(*pred, replace_map);
	}
}

int main(int argc, char *argv[]) {
	lifter lift;
	if (!lift.load(argv[1])) {
		std::cerr << "Cannot open input file " << argv[1] << '\n';
		return 1;
	}

	if (lift.instructions.empty()) {
		try {
			lift.save(argv[argc-1]);
			return 0;
		}
		catch (std::runtime_error& e) {
			std::cout << e.what() << '\n';
			return 1;
		}
	}

	control_flow_graph cfg = get_cfg(lift);

	int bb_from = 0;
	int bb_to = 5000;
	int bb_count = 0;

	for (auto& bb : cfg) {
		if (bb.front().is_data())
			continue;

		if (lift.functions.find(bb.front().label) != lift.functions.end()) {
			vins tmp = vins::ins_mov(vreg(17), 321);
			bb.insert(std::next(bb.begin()), std::move(tmp));
		}

		for (auto in = bb.begin(); in != bb.end();) {
			if (in->is_pseudo() || !in->label.empty()) {
				++in;
				continue;
			}
			auto next = std::next(in);
			if (bb_count >= bb_from && bb_count <= bb_to) {
				vins tmp = vins::ins_add(vreg(17), vreg(17), 99);
				bb.insert(in, std::move(tmp));
			}
			++bb_count;
			in = next;
		}
	}

	allocate_registers(cfg);
	
	lift.instructions = cfg_dump(cfg);
	
	try { lift.save(argv[argc-1]); }
	catch (std::runtime_error& e) {
		std::cout << e.what() << '\n';
		return 1;
	}
}