|
|
@@ -5,14 +5,14 @@ |
|
|
|
#include "../CodeDweller/faults.hpp"
|
|
|
|
#include "../SNFMulti/GBUdb.hpp"
|
|
|
|
|
|
|
|
using namespace std;
|
|
|
|
namespace cd = codedweller;
|
|
|
|
|
|
|
|
const string VersionInfo = "V0.1";
|
|
|
|
const std::string VersionInfo = "V0.1";
|
|
|
|
const int ErrorResultCode = 1;
|
|
|
|
const int SuccessResultCode = 0;
|
|
|
|
|
|
|
|
struct Configuration {
|
|
|
|
string GBXFile;
|
|
|
|
std::string GBXFile;
|
|
|
|
double Probability;
|
|
|
|
double Confidence;
|
|
|
|
bool findBlack;
|
|
|
@@ -25,14 +25,14 @@ struct Configuration { |
|
|
|
showDetails(false) {}
|
|
|
|
} myConfig;
|
|
|
|
|
|
|
|
class BadArguments : public RuntimeFault {public: BadArguments(string s):RuntimeFault(s){}};
|
|
|
|
class BadArguments : public cd::RuntimeFault {public: BadArguments(std::string s):RuntimeFault(s){}};
|
|
|
|
const BadArguments BadArgumentCount("Wrong number of arguments");
|
|
|
|
const BadArguments BadSwitch("Bad command line switch");
|
|
|
|
|
|
|
|
void setConfiguration(Configuration &conf, int argc, char* argv[]) {
|
|
|
|
BadArgumentCount(argc > 6 || argc < 1);
|
|
|
|
for(int i = 1; i < argc; i++) {
|
|
|
|
string thisArgument(argv[i]);
|
|
|
|
std::string thisArgument(argv[i]);
|
|
|
|
|
|
|
|
if(0 == thisArgument.find("-black")) {
|
|
|
|
conf.findBlack = true;
|
|
|
@@ -65,27 +65,27 @@ void setConfiguration(Configuration &conf, int argc, char* argv[]) { |
|
|
|
}
|
|
|
|
|
|
|
|
void displayHelp() {
|
|
|
|
cerr << "GBUDBTool " << VersionInfo << endl;
|
|
|
|
cerr << "Use: GBUDBTool [-black | -white] [-p=<value>] [-c=<value>] <gbx file>" << endl;
|
|
|
|
cerr << " -black prints out blacklist IPs based on p & c" << endl;
|
|
|
|
cerr << " -white prints out whitelist IPs based on p & c" << endl;
|
|
|
|
cerr << " -details prints out counts, confidence, and probability" << endl;
|
|
|
|
cerr << " -c=<value> sets the confidence figure for the list" << endl;
|
|
|
|
cerr << " -p=<value> sets the probability figure for the list" << endl;
|
|
|
|
cerr << " <gbx file> is the path to a GBUdb snapshot" << endl;
|
|
|
|
std::cerr << "GBUDBTool " << VersionInfo << std::endl;
|
|
|
|
std::cerr << "Use: GBUDBTool [-black | -white] [-p=<value>] [-c=<value>] <gbx file>" << std::endl;
|
|
|
|
std::cerr << " -black prints out blacklist IPs based on p & c" << std::endl;
|
|
|
|
std::cerr << " -white prints out whitelist IPs based on p & c" << std::endl;
|
|
|
|
std::cerr << " -details prints out counts, confidence, and probability" << std::endl;
|
|
|
|
std::cerr << " -c=<value> sets the confidence figure for the list" << std::endl;
|
|
|
|
std::cerr << " -p=<value> sets the probability figure for the list" << std::endl;
|
|
|
|
std::cerr << " <gbx file> is the path to a GBUdb snapshot" << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
class Reporter : public GBUdbOperator {
|
|
|
|
private:
|
|
|
|
Configuration myConfig;
|
|
|
|
|
|
|
|
string toIP4String(unsigned int IP) {
|
|
|
|
std::string toIP4String(unsigned int IP) {
|
|
|
|
int a, b, c, d;
|
|
|
|
d = IP & 0x000000ff; IP >>= 8;
|
|
|
|
c = IP & 0x000000ff; IP >>= 8;
|
|
|
|
b = IP & 0x000000ff; IP >>= 8;
|
|
|
|
a = IP;
|
|
|
|
ostringstream S;
|
|
|
|
std::ostringstream S;
|
|
|
|
S << a << "." << b << "." << c << "." << d;
|
|
|
|
return S.str();
|
|
|
|
}
|
|
|
@@ -100,15 +100,15 @@ class Reporter : public GBUdbOperator { |
|
|
|
if(myConfig.findBlack) goodProbability = (myConfig.Probability <= R.Probability());
|
|
|
|
else goodProbability = (myConfig.Probability <= R.Probability());
|
|
|
|
if(goodProbability && goodConfidence) {
|
|
|
|
cout << toIP4String(IP);
|
|
|
|
std::cout << toIP4String(IP);
|
|
|
|
if(myConfig.showDetails) {
|
|
|
|
cout << "\t"
|
|
|
|
std::cout << "\t"
|
|
|
|
<< "g=" << R.Good()
|
|
|
|
<< ", b=" << R.Bad()
|
|
|
|
<< ", c=" << R.Confidence()
|
|
|
|
<< ", p=" << R.Probability();
|
|
|
|
}
|
|
|
|
cout << endl;
|
|
|
|
std::cout << std::endl;
|
|
|
|
}
|
|
|
|
return R;
|
|
|
|
}
|
|
|
@@ -123,9 +123,9 @@ int main(int argc, char* argv[]) { |
|
|
|
DB.doForAllRecords(R);
|
|
|
|
}
|
|
|
|
|
|
|
|
catch(exception &e) {
|
|
|
|
catch(const std::exception &e) {
|
|
|
|
displayHelp();
|
|
|
|
cerr << endl << "Exception: " << e.what() << endl;
|
|
|
|
std::cerr << std::endl << "Exception: " << e.what() << std::endl;
|
|
|
|
return ErrorResultCode;
|
|
|
|
}
|
|
|
|
|