|
@@ -512,9 +512,34 @@ void statistics_db::readPortServicesFromNmap()
|
|
|
*/
|
|
|
std::string statistics_db::getNmapPath()
|
|
|
{
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ std::string filename = "nmap-services-tcp.csv";
|
|
|
+ std::string resourcesDir = "/resources/";
|
|
|
+ std::string codeDir = "/code";
|
|
|
+ std::string testDir = "/code/Test";
|
|
|
char buff[FILENAME_MAX];
|
|
|
+
|
|
|
std::string dir(getcwd(buff, FILENAME_MAX));
|
|
|
- dir = dir.substr(0, dir.rfind("/ID2T-toolkit")) + "/ID2T-toolkit/resources/nmap-services-tcp.csv";
|
|
|
+
|
|
|
+
|
|
|
+ if(dir.rfind(testDir) == (dir.size()-testDir.size()))
|
|
|
+ {
|
|
|
+
|
|
|
+ dir = dir.substr(0, (dir.size()-testDir.size()));
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ if(dir.rfind(codeDir) == (dir.size()-codeDir.size()))
|
|
|
+ {
|
|
|
+
|
|
|
+ dir = dir.substr(0, (dir.size()-codeDir.size()));
|
|
|
+ }
|
|
|
+
|
|
|
+ dir = dir + resourcesDir + filename;
|
|
|
|
|
|
return dir;
|
|
|
}
|