80 lines
2.2 KiB
C++
80 lines
2.2 KiB
C++
#include <string>
|
|
#include <sstream>
|
|
|
|
#include "util/Logger.hpp"
|
|
#include "util/Database.hpp"
|
|
#include "FullTraceImporter.hpp"
|
|
|
|
using namespace fail;
|
|
static fail::Logger LOG("FullTraceImporter");
|
|
|
|
Database *db;
|
|
|
|
bool FullTraceImporter::handle_ip_event(simtime_t curtime, instruction_count_t instr,
|
|
Trace_Event &ev) {
|
|
|
|
margin_info_t right_margin;
|
|
right_margin.time = curtime;
|
|
right_margin.dyninstr = instr; // !< The current instruction
|
|
right_margin.ip = ev.ip();
|
|
|
|
// pass through potentially available extended trace information
|
|
if (!add_trace_event(right_margin, right_margin, ev)) {
|
|
LOG << "add_trace_event failed" << std::endl;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool FullTraceImporter::handle_mem_event(simtime_t curtime, instruction_count_t instr,
|
|
Trace_Event &ev) {
|
|
return true;
|
|
}
|
|
|
|
bool FullTraceImporter::create_database() {
|
|
std::stringstream create_statement;
|
|
create_statement << "CREATE TABLE IF NOT EXISTS fulltrace ("
|
|
" variant_id int(11) NOT NULL,"
|
|
" instr int(10) unsigned NOT NULL,"
|
|
" instr_absolute int(10) unsigned DEFAULT NULL,"
|
|
" INDEX(instr),"
|
|
" PRIMARY KEY (variant_id,instr)"
|
|
") engine=MyISAM ";
|
|
return db->query(create_statement.str().c_str());
|
|
}
|
|
|
|
bool FullTraceImporter::clear_database() {
|
|
std::stringstream ss;
|
|
ss << "DELETE FROM fulltrace WHERE variant_id = " << m_variant_id;
|
|
|
|
bool ret = db->query(ss.str().c_str()) == 0 ? false : true;
|
|
LOG << "deleted " << db->affected_rows() << " rows from fulltrace table" << std::endl;
|
|
return ret;
|
|
}
|
|
|
|
bool FullTraceImporter::add_trace_event(margin_info_t &begin, margin_info_t &end,
|
|
Trace_Event &event, bool is_fake) {
|
|
// Is event a fake mem-event?
|
|
if (is_fake) {
|
|
return true;
|
|
}
|
|
|
|
// INSERT group entry
|
|
std::stringstream sql;
|
|
sql << "(" << m_variant_id << "," << end.dyninstr << "," << end.ip << ")";
|
|
|
|
bool ret =
|
|
db->insert_multiple("INSERT INTO fulltrace (variant_id, instr, instr_absolute) VALUES ", sql.str().c_str());
|
|
m_row_count++;
|
|
if (m_row_count % 10000 == 0 && ret) {
|
|
LOG << "Inserted " << std::dec << m_row_count << " trace events into the database" << std::endl;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool FullTraceImporter::trace_end_reached() {
|
|
return db->insert_multiple();
|
|
}
|