/**
* This function is called
**/
void traceInst(INS ins, VOID*)
{
ADDRINT address = INS_Address(ins);
std::string mod_name = getModule( address );
RegList regs;
for ( UINT32 i = 0; i < INS_OperandCount(ins); i++ )
{
if ( INS_OperandIsReg(ins, i) )
{
REG x = INS_OperandReg(ins, i);
if ( x != REG_INVALID() )
regs.push_back( x );
}
}
if (isUnknownAddress(address))
{
// The address is an address that does not belong to any loaded module.
// This is potential shellcode. For these instructions a callback
// function is inserted that dumps information to the trace file when
// the instruction is actually executed.
INS_InsertCall(ins, IPOINT_BEFORE, AFUNPTR(dump_shellcode),
IARG_PTR, new std::string(dumpInstruction(ins)),
IARG_PTR, ®s,
IARG_CONTEXT, IARG_END
);
}
else
{
if ( !modlist.empty() && (modlist.find(mod_name) == modlist.end()) ) // not concerned
return;
// The address is a legit address, meaning it is probably not part of
// any shellcode. In this case we just log the instruction to dump it
// later to show when control flow was transfered from legit code to
// shellcode.
legitInstructions.push_back(dumpInstruction(ins));
if (legitInstructions.size() > KnobMaxLegitInsLogSize.Value())
{
// Log only up to KnobMaxLegitInsLogSize.Value() instructions or the whole
// program before the shellcode will be dumped.
legitInstructions.pop_front();
}
}
}
void IsisMain() {
// Get user interface
UserInterface &ui = Application::GetUserInterface();
// Open the shift definitions file
Pvl shiftdef;
if (ui.WasEntered("SHIFTDEF")) {
shiftdef.Read(ui.GetFilename("SHIFTDEF"));
}
else {
shiftdef.AddObject(PvlObject("Hiccdstitch"));
}
PvlObject &stitch = shiftdef.FindObject("Hiccdstitch", Pvl::Traverse);
// Open the first cube. It will be matched to the second input cube.
HiJitCube trans;
CubeAttributeInput &attTrans = ui.GetInputAttribute("FROM");
vector<string> bandTrans = attTrans.Bands();
trans.SetVirtualBands(bandTrans);
trans.OpenCube(ui.GetFilename("FROM"), stitch);
// Open the second cube, it is held in place. We will be matching the
// first to this one by attempting to compute a sample/line translation
HiJitCube match;
CubeAttributeInput &attMatch = ui.GetInputAttribute("MATCH");
vector<string> bandMatch = attMatch.Bands();
match.SetVirtualBands(bandMatch);
match.OpenCube(ui.GetFilename("MATCH"), stitch);
// Ensure only one band
if ((trans.Bands() != 1) || (match.Bands() != 1)) {
string msg = "Input Cubes must have only one band!";
throw Isis::iException::Message(Isis::iException::User,msg,_FILEINFO_);
}
// Now test compatability (basically summing)
trans.Compatable(match);
// Determine intersection
if (!trans.intersects(match)) {
string msg = "Input Cubes do not overlap!";
throw Isis::iException::Message(Isis::iException::User,msg,_FILEINFO_);
}
// Get overlapping regions of each cube
HiJitCube::Corners fcorns, mcorns;
trans.overlap(match, fcorns);
match.overlap(trans, mcorns);
#if defined(ISIS_DEBUG)
cout << "FROM Poly: " << trans.PolyToString() << std::endl;
cout << "MATCH Poly: " << match.PolyToString() << std::endl;
cout << "From Overlap: (" << fcorns.topLeft.sample << ","
<< fcorns.topLeft.line << "), ("
<< fcorns.lowerRight.sample << ","
<< fcorns.lowerRight.line << ")\n" ;
cout << "Match Overlap: (" << mcorns.topLeft.sample << ","
<< mcorns.topLeft.line << "), ("
<< mcorns.lowerRight.sample << ","
<< mcorns.lowerRight.line << ")\n" ;
#endif
// We need to get a user definition of how to auto correlate around each
// of the grid points.
Pvl regdef;
Filename regFile(ui.GetFilename("REGDEF"));
regdef.Read(regFile.Expanded());
AutoReg *ar = AutoRegFactory::Create(regdef);
double flines(fcorns.lowerRight.line - fcorns.topLeft.line + 1.0);
double fsamps(fcorns.lowerRight.sample - fcorns.topLeft.sample + 1.0);
// We want to create a grid of control points that is N rows by M columns.
// Get row and column variables, if not entered, default to 1% of the input
// image size
int rows(1), cols(1);
if (ui.WasEntered("ROWS")) {
rows = ui.GetInteger("ROWS");
}
else {
rows = (int)(((flines - 1.0) / ar->SearchChip()->Lines()) + 1);
}
cols = ui.GetInteger("COLUMNS");
if (cols == 0) {
cols = (int)(((fsamps - 1.0) / ar->SearchChip()->Samples()) + 1);
}
// Calculate spacing for the grid of points
double lSpacing = floor(flines / rows);
double sSpacing = floor(fsamps / cols);
#if defined(ISIS_DEBUG)
cout << "# Samples in Overlap: " << fsamps << endl;
cout << "# Lines in Overlap : " << flines << endl;
cout << "# Rows: " << rows << endl;
cout << "# Columns: " << cols << endl;
cout << "Line Spacing: " << lSpacing << endl;
cout << "Sample Spacing: " << sSpacing << endl;
#endif
// Display the progress...10% 20% etc.
Progress prog;
prog.SetMaximumSteps(rows * cols);
prog.CheckStatus();
// Initialize control point network
ControlNet cn;
cn.SetType(ControlNet::ImageToImage);
cn.SetUserName(Application::UserName());
cn.SetCreatedDate(iTime::CurrentLocalTime());
// Get serial numbers for input cubes
string transSN = SerialNumber::Compose(trans, true);
string matchSN = SerialNumber::Compose(match, true);
cn.SetTarget(transSN);
cn.SetDescription("Records s/c jitter between two adjacent HiRISE images");
// Set up results parameter saves
JitterParms jparms;
jparms.fromCorns = fcorns;
jparms.fromJit = trans.GetInfo();
jparms.matchCorns = mcorns;
jparms.matchJit = match.GetInfo();
jparms.regFile = regFile.Expanded();
jparms.cols = cols;
jparms.rows = rows;
jparms.lSpacing = lSpacing;
jparms.sSpacing = sSpacing;
jparms.nSuspects = 0;
// Loop through grid of points and get statistics to compute
// translation values
RegList reglist;
double fline0(fcorns.topLeft.line-1.0), fsamp0(fcorns.topLeft.sample-1.0);
double mline0(mcorns.topLeft.line-1.0), msamp0(mcorns.topLeft.sample-1.0);
for (int r=0; r<rows; r++) {
int line = (int)(lSpacing / 2.0 + lSpacing * r + 0.5);
for (int c=0; c<cols; c++) {
int samp = (int)(sSpacing / 2.0 + sSpacing * c + 0.5);
ar->PatternChip()->TackCube(msamp0+samp, mline0+line);
ar->PatternChip()->Load(match);
ar->SearchChip()->TackCube(fsamp0+samp, fline0+line);
ar->SearchChip()->Load(trans);
// Set up ControlMeasure for cube to translate
ControlMeasure cmTrans;
cmTrans.SetCubeSerialNumber(transSN);
cmTrans.SetCoordinate(msamp0+samp, mline0+line,
ControlMeasure::Unmeasured);
cmTrans.SetChooserName("hijitreg");
cmTrans.SetReference(false);
// Set up ControlMeasure for the pattern/Match cube
ControlMeasure cmMatch;
cmMatch.SetCubeSerialNumber(matchSN);
cmMatch.SetCoordinate(fsamp0+samp, fline0+line,
ControlMeasure::Automatic);
cmMatch.SetChooserName("hijitreg");
cmMatch.SetReference(true);
// Match found
if (ar->Register()==AutoReg::Success) {
RegData reg;
reg.fLine = fline0 + line;
reg.fSamp = fsamp0 + samp;
reg.fLTime = trans.getLineTime(reg.fLine);
reg.mLine = mline0 + line;
reg.mSamp = msamp0 + samp;
reg.mLTime = match.getLineTime(reg.mLine);
reg.regLine = ar->CubeLine();
reg.regSamp = ar->CubeSample();
reg.regCorr = ar->GoodnessOfFit();
if (fabs(reg.regCorr) > 1.0) jparms.nSuspects++;
double sDiff = reg.fSamp - reg.regSamp;
double lDiff = reg.fLine - reg.regLine;
jparms.sStats.AddData(&sDiff,(unsigned int)1);
jparms.lStats.AddData(&lDiff,(unsigned int)1);
// Record the translation in the control point
cmTrans.SetCoordinate(ar->CubeSample(), ar->CubeLine(),
ControlMeasure::Automatic);
cmTrans.SetError(sDiff, lDiff);
cmTrans.SetGoodnessOfFit(ar->GoodnessOfFit());
// Reread the chip location centering the offset and compute
// linear regression statistics
try {
Chip &pchip(*ar->PatternChip());
Chip fchip(pchip.Samples(), pchip.Lines());
fchip.TackCube(ar->CubeSample(), ar->CubeLine());
fchip.Load(trans);
// Writes correlated chips to files for visual inspection
#if defined(ISIS_DEBUG)
ostringstream tstr;
tstr << "R" << r << "C" << c << "_chip.cub";
string fcname("from" + tstr.str());
string mcname("match" + tstr.str());
pchip.Write(mcname);
fchip.Write(fcname);
#endif
MultivariateStatistics mstats;
for (int line = 1 ; line <= fchip.Lines() ; line++) {
for(int sample = 1; sample < fchip.Samples(); sample++) {
double fchipValue = fchip.GetValue(sample,line);
double pchipValue = pchip.GetValue(sample,line);
mstats.AddData(&fchipValue, &pchipValue, 1);
}
}
// Get regression and correlation values
mstats.LinearRegression(reg.B0, reg.B1);
reg.Bcorr = mstats.Correlation();
if (IsSpecial(reg.B0)) throw 1;
if (IsSpecial(reg.B1)) throw 2;
if (IsSpecial(reg.Bcorr)) throw 3;
}
catch (...) {
// If fails, flag this condition
reg.B0 = 0.0;
reg.B1= 0.0;
reg.Bcorr = 0.0;
}
reglist.push_back(reg);
}
// Add the measures to a control point
string str = "Row " + iString(r) + " Column " + iString(c);
ControlPoint cp(str);
cp.SetType(ControlPoint::Tie);
cp.Add(cmTrans);
cp.Add(cmMatch);
if (!cmTrans.IsMeasured()) cp.SetIgnore(true);
cn.Add(cp);
prog.CheckStatus();
}
}
// If flatfile was entered, create the flatfile
// The flatfile is comma seperated and can be imported into an excel
// spreadsheet
if (ui.WasEntered("FLATFILE")) {
string fFile = ui.GetFilename("FLATFILE");
ofstream os;
string fFileExpanded = Filename(fFile).Expanded();
os.open(fFileExpanded.c_str(),ios::out);
dumpResults(os, reglist, jparms, *ar);
}
// If a cnet file was entered, write the ControlNet pvl to the file
if (ui.WasEntered("CNETFILE")) {
cn.Write(ui.GetFilename("CNETFILE"));
}
// Don't need the cubes opened anymore
trans.Close();
match.Close();
// Write translation to log
PvlGroup results("AverageTranslation");
if (jparms.sStats.ValidPixels() > 0) {
double sTrans = (int)(jparms.sStats.Average() * 100.0) / 100.0;
double lTrans = (int)(jparms.lStats.Average() * 100.0) / 100.0;
results += PvlKeyword ("Sample",sTrans);
results += PvlKeyword ("Line",lTrans);
results += PvlKeyword ("NSuspects",jparms.nSuspects);
}
else {
results += PvlKeyword ("Sample","NULL");
results += PvlKeyword ("Line","NULL");
}
Application::Log(results);
// add the auto registration information to print.prt
PvlGroup autoRegTemplate = ar->RegTemplate();
Application::Log(autoRegTemplate);
return;
}
