nesca/Connector.cpp

483 lines
14 KiB
C++
Raw Normal View History

2015-03-22 00:43:15 +00:00
#include "Connector.h"
2015-03-23 17:11:00 +00:00
#include "SSHAuth.h"
2015-04-18 13:05:35 +00:00
#include "Filter.h"
2015-03-05 14:29:05 +00:00
2015-04-25 19:45:01 +00:00
2015-03-06 14:32:36 +00:00
#if defined(WIN32) || defined(_WIN32) || defined(__WIN32) && !defined(__CYGWIN__)
2015-03-22 00:43:15 +00:00
int _pingMyTarget(const char *ip)
2015-03-06 14:32:36 +00:00
{
HANDLE hIcmpFile;
unsigned long ipaddr = INADDR_NONE;
DWORD dwRetVal = 0;
char SendData[32] = "Data Buffer";
LPVOID ReplyBuffer = NULL;
DWORD ReplySize = 0;
ipaddr = inet_addr(ip);
if (ipaddr == INADDR_NONE)
{
stt->doEmitionRedFoundData("[Pinger] INADDR_NONE! [" + QString(ip) + "]");
return 0;
}
hIcmpFile = IcmpCreateFile();
if (hIcmpFile == INVALID_HANDLE_VALUE)
{
stt->doEmitionRedFoundData("[Pinger] Unable to open handle. [" + QString::number(GetLastError()) + "]");
return 0;
}
ReplySize = sizeof(ICMP_ECHO_REPLY) + sizeof(SendData);
ReplyBuffer = (VOID*) malloc(ReplySize);
if (ReplyBuffer == NULL)
{
stt->doEmitionRedFoundData("[Pinger] Unable to allocate memory.");
return 0;
}
dwRetVal = IcmpSendEcho(hIcmpFile, ipaddr, SendData, sizeof(SendData),
NULL, ReplyBuffer, ReplySize, gPingTimeout*1000);
if (dwRetVal != 0) {
PICMP_ECHO_REPLY pEchoReply = (PICMP_ECHO_REPLY)ReplyBuffer;
struct in_addr ReplyAddr;
ReplyAddr.S_un.S_addr = pEchoReply->Address;
printf("\tSent icmp message to %s\n", "127.0.0.1");
if (dwRetVal > 1)
{
if(gDebugMode) stt->doEmitionYellowFoundData("[Pinger] Received " + QString::number(dwRetVal) + " icmp message responses.");
}
else
{
if(gDebugMode) stt->doEmitionYellowFoundData("[Pinger] Received " + QString::number(dwRetVal) + " icmp message responses.");
}
if(gDebugMode) stt->doEmitionYellowFoundData("[Pinger] Received from: " + QString(inet_ntoa( ReplyAddr )) + "; Status = " + QString::number(pEchoReply->Status) + "; Roundtrip time = " + QString::number(pEchoReply->RoundTripTime) + "ms.");
return 1;
}
else
{
printf("\tCall to IcmpSendEcho failed.\n");
printf("\tIcmpSendEcho returned error: %ld\n", GetLastError() );
if(gDebugMode) stt->doEmitionRedFoundData("[Pinger] Call to IcmpSendEcho failed. IcmpSendEcho returned error: " + QString::number(GetLastError()));
return 0;
};
}
#else
2015-03-20 14:28:51 +00:00
int _pingMyTarget(const char *ip)
2015-03-06 14:32:36 +00:00
{
FILE *pipe = popen(("ping -w " + std::to_string(gPingTimeout) + " " + ip).c_str(), "r");
if(!pipe) {
stt->doEmitionRedFoundData("Ping pipe failed: cannot open pipe.");
perror("pipe");
return 0;
}
char buffer[128] = {0};
std::string result;
2015-03-05 14:29:05 +00:00
2015-03-06 14:32:36 +00:00
while(!feof(pipe)) {
if(fgets(buffer, 128, pipe) != NULL){
result += buffer;
}
}
pclose(pipe);
2015-03-05 14:29:05 +00:00
2015-03-06 14:32:36 +00:00
if(strstr((char*)result.c_str(), "100% packet loss") != NULL) return 0;
return 1;
}
#endif
struct data {
char trace_ascii; /* 1 or 0 */
};
static
int my_trace(CURL *handle, curl_infotype type,
char *data, size_t size,
void *userp)
{
2015-03-17 14:30:53 +00:00
if (type == CURLINFO_HEADER_OUT) {
2015-03-22 00:43:15 +00:00
data[size] = '\0';
Activity += strlen(data);
stt->doEmitionAddOutData(QString(data));
2015-03-06 14:32:36 +00:00
}
return 0;
}
2015-04-25 19:45:01 +00:00
//struct MemoryStruct {
// char *memory;
// size_t size;
//};
2015-04-01 12:39:14 +00:00
size_t nWriteCallback(void *contents, size_t size, size_t nmemb, void *userp)
2015-03-05 14:29:05 +00:00
{
2015-04-25 19:45:01 +00:00
size_t realsize = size * nmemb;
if (((std::string*)userp)->size() > 180000) return -1;
((std::string*)userp)->append((char*)contents, realsize);
Activity += realsize;
return realsize;
//struct MemoryStruct *mem = (struct MemoryStruct *)userp;
//if (mem->size > 180000) return -1;
//size_t realsize = size * nmemb;
//mem->memory = (char*)realloc(mem->memory, mem->size + realsize + 1);
//if (mem->memory == NULL) {
// stt->doEmitionRedFoundData("not enough memory (realloc returned NULL)\n");
// return 0;
//}
//memcpy(&(mem->memory[mem->size]), contents, realsize);
//mem->size += realsize;
//mem->memory[mem->size] = 0;
//Activity += realsize;
//return realsize;
2015-03-05 14:29:05 +00:00
}
2015-03-06 14:32:36 +00:00
2015-03-20 14:28:51 +00:00
int Connector::nConnect(const char* ip, const int port, std::string *buffer,
2015-03-06 14:32:36 +00:00
const char *postData,
2015-03-16 14:29:34 +00:00
const std::vector<std::string> *customHeaders,
2015-04-18 23:00:40 +00:00
const std::string *lpString,
bool digestMode){
2015-04-25 19:45:01 +00:00
buffer->clear();
//buffer->reserve(100000);
int res = 0;
2015-04-20 19:27:06 +00:00
CURL *curl = curl_easy_init();
2015-03-05 14:29:05 +00:00
2015-04-25 19:45:01 +00:00
//struct MemoryStruct chunk;
//chunk.memory = (char*)malloc(1); /* will be grown as needed by the realloc above */
//chunk.size = 0; /* no data at this point */
//std::string buffer2;
2015-04-23 05:23:02 +00:00
if (curl != NULL)
{
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1L);
if (MapWidgetOpened) {
struct data config;
config.trace_ascii = 1; /* enable ascii tracing */
curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION, my_trace);
curl_easy_setopt(curl, CURLOPT_DEBUGDATA, &config);
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
}
curl_easy_setopt(curl, CURLOPT_URL, ip);
curl_easy_setopt(curl, CURLOPT_PORT, port);
curl_easy_setopt(curl, CURLOPT_USERAGENT,
"Mozilla/5.0 (X11; Linux x86_64; rv:35.0) Gecko/20100101 Firefox/35.0");
curl_easy_setopt(curl, CURLOPT_HEADER, 1L);
curl_easy_setopt(curl, CURLOPT_AUTOREFERER, 1L);
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L);
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, nWriteCallback);
2015-04-25 19:45:01 +00:00
//curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);
2015-04-23 05:23:02 +00:00
curl_easy_setopt(curl, CURLOPT_WRITEDATA, buffer);
int proxyPort = std::atoi(gProxyPort);
if (strlen(gProxyIP) != 0 && (proxyPort > 0 && proxyPort < 65535)) {
curl_easy_setopt(curl, CURLOPT_PROXY, gProxyIP);
curl_easy_setopt(curl, CURLOPT_PROXYPORT, proxyPort);
}
2015-04-25 19:45:01 +00:00
else curl_easy_setopt(curl, CURLOPT_PROXY, "");
2015-04-23 05:23:02 +00:00
curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, gTimeOut);
curl_easy_setopt(curl, CURLOPT_TIMEOUT, gTimeOut + 3);
2015-03-05 14:29:05 +00:00
2015-04-23 05:23:02 +00:00
if (postData != NULL) curl_easy_setopt(curl, CURLOPT_POSTFIELDS, postData);
2015-03-05 14:29:05 +00:00
2015-04-23 05:23:02 +00:00
if (customHeaders != NULL) {
2015-03-05 14:29:05 +00:00
2015-04-23 05:23:02 +00:00
struct curl_slist *chunk = NULL;
for (auto &ch : *customHeaders) chunk = curl_slist_append(chunk, ch.c_str());
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, chunk);
}
2015-04-25 19:45:01 +00:00
2015-04-18 23:24:44 +00:00
if (lpString != NULL) {
curl_easy_setopt(curl, CURLOPT_UNRESTRICTED_AUTH, 1L);
curl_easy_setopt(curl, CURLOPT_FTPLISTONLY, 1L);
2015-03-22 00:43:15 +00:00
curl_easy_setopt(curl, CURLOPT_USERPWD, lpString->c_str());
2015-04-18 23:24:44 +00:00
if (digestMode)
{
curl_easy_setopt(curl, CURLOPT_HTTPAUTH, (long)CURLAUTH_DIGEST);
res = curl_easy_perform(curl);
2015-04-25 19:45:01 +00:00
//if (chunk.size > 0){
// //buffer2 = std::string(chunk.memory);
// buffer->append(chunk.memory, chunk.size);
//}
2015-04-18 23:24:44 +00:00
if (port != 21 && lpString != NULL) {
int pos = Utils::ustrstr(*buffer, "\r\n\r\n");
if (pos != -1) {
*buffer = buffer->substr(pos + 4);
}
}
}
else res = curl_easy_perform(curl);
}
else res = curl_easy_perform(curl);
2015-04-25 19:45:01 +00:00
//if (chunk.size > 0){
// //buffer2 = std::string(chunk.memory);
// buffer->append(chunk.memory, chunk.size);
//}
2015-03-22 00:43:15 +00:00
if (res == CURLE_OK ||
(port == 21 && buffer->size() > 0)) {
if (MapWidgetOpened) stt->doEmitionAddIncData(QString(ip), QString(buffer->c_str()));
Activity += buffer->size();
2015-04-25 19:45:01 +00:00
curl_easy_cleanup(curl);
2015-04-01 12:39:14 +00:00
return buffer->size();
2015-03-22 00:43:15 +00:00
} else {
2015-04-25 19:45:01 +00:00
curl_easy_cleanup(curl);
2015-04-04 07:24:31 +00:00
if (res == 6) return -2;
else if (res != 28 &&
2015-03-22 00:43:15 +00:00
res != 7 &&
2015-04-25 19:45:01 +00:00
res != 13 &&
2015-03-22 00:43:15 +00:00
res != 67 &&
res != 52 &&
res != 55 &&
2015-04-01 16:58:12 +00:00
res != 56 &&
res != 35 &&
res != 19 &&
2015-04-01 12:39:14 +00:00
res != 23) {
2015-03-22 00:43:15 +00:00
if (res == 5) {
stt->doEmitionRedFoundData("Couldn't resolve proxy. The given proxy host could not be resolved. ");
return -2;
} else if (res == 8) {
2015-03-23 16:32:33 +00:00
stt->doEmitionFoundData("Strange ftp reply. (" +
2015-03-22 00:43:15 +00:00
QString::number(res) + ") " + QString(ip) +
":" + QString::number(port));
return -2;
}
2015-03-22 11:05:58 +00:00
else if (res == 18) {
stt->doEmitionFoundData("Inappropriate file size. (" +
QString::number(res) + ") " + QString(ip) +
":" + QString::number(port));
return -2;
2015-03-22 00:43:15 +00:00
}
else stt->doEmitionRedFoundData("CURL error: (" + QString::number(res) + ") " +
QString(ip) + ":" + QString::number(port));
}
2015-04-01 12:39:14 +00:00
if(res == 23 && buffer->size() > 0) {
if (MapWidgetOpened) stt->doEmitionAddIncData(QString(ip), QString("[OVERFLOW]"));
return buffer->size();
2015-04-04 07:24:31 +00:00
} else return -1;
2015-03-22 00:43:15 +00:00
}
2015-04-25 19:45:01 +00:00
if (MapWidgetOpened) stt->doEmitionAddIncData(QString(ip), QString(buffer->c_str()));
return buffer->size();
2015-03-05 14:29:05 +00:00
} else {
stt->doEmitionRedFoundData("Curl error.");
return -1;
2015-03-22 00:43:15 +00:00
};
2015-03-05 14:29:05 +00:00
}
2015-04-25 19:45:01 +00:00
//
//int Connector::nConnect2(const char* ip, const int port, std::string *buffer,
// const char *postData,
// const std::vector<std::string> *customHeaders,
// const std::string *lpString,
// bool digestMode)
//{
// buffer->clear();
//
// int still_running;
// struct timeval timeout;
// int res = 0;
// CURL *curl = curl_easy_init();
//
// if (curl != NULL)
// {
// curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1L);
// if (MapWidgetOpened) {
// struct data config;
// config.trace_ascii = 1; /* enable ascii tracing */
// curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION, my_trace);
// curl_easy_setopt(curl, CURLOPT_DEBUGDATA, &config);
// curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
// }
// curl_easy_setopt(curl, CURLOPT_URL, ip);
// curl_easy_setopt(curl, CURLOPT_PORT, port);
// curl_easy_setopt(curl, CURLOPT_USERAGENT,
// "Mozilla/5.0 (X11; Linux x86_64; rv:35.0) Gecko/20100101 Firefox/35.0");
// curl_easy_setopt(curl, CURLOPT_HEADER, 1L);
// curl_easy_setopt(curl, CURLOPT_AUTOREFERER, 1L);
// curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
// curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L);
// curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, nWriteCallback);
// curl_easy_setopt(curl, CURLOPT_WRITEDATA, buffer);
// int proxyPort = std::atoi(gProxyPort);
// if (strlen(gProxyIP) != 0 && (proxyPort > 0 && proxyPort < 65535)) {
// curl_easy_setopt(curl, CURLOPT_PROXY, gProxyIP);
// curl_easy_setopt(curl, CURLOPT_PROXYPORT, proxyPort);
// }
// else {
// curl_easy_setopt(curl, CURLOPT_PROXY, "");
// }
// curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
// curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, gTimeOut);
// curl_easy_setopt(curl, CURLOPT_TIMEOUT, gTimeOut + 3);
//
// if (postData != NULL) curl_easy_setopt(curl, CURLOPT_POSTFIELDS, postData);
//
// if (customHeaders != NULL) {
// struct curl_slist *chunk = NULL;
// for (auto &ch : *customHeaders) chunk = curl_slist_append(chunk, ch.c_str());
// curl_easy_setopt(curl, CURLOPT_HTTPHEADER, chunk);
// }
//
// if (lpString != NULL) {
// curl_easy_setopt(curl, CURLOPT_UNRESTRICTED_AUTH, 1L);
// curl_easy_setopt(curl, CURLOPT_FTPLISTONLY, 1L);
// curl_easy_setopt(curl, CURLOPT_USERPWD, lpString->c_str());
// if (digestMode) curl_easy_setopt(curl, CURLOPT_HTTPAUTH, (long)CURLAUTH_DIGEST);
// }
//
//
// CURLM *multi_handle = curl_multi_init();
// if (multi_handle == NULL) stt->doEmitionRedFoundData("curl_multi_init == NULL!");
// curl_multi_add_handle(multi_handle, curl);
//
// do { res = curl_multi_perform(multi_handle, &still_running); }
// while (res == CURLM_CALL_MULTI_PERFORM);
//
// while (still_running) {
// fd_set fdread;
// fd_set fdwrite;
// fd_set fdexcep;
// int maxfd = -1;
//
// FD_ZERO(&fdread);
// FD_ZERO(&fdwrite);
// FD_ZERO(&fdexcep);
//
// timeout.tv_sec = gTimeOut;
// timeout.tv_usec = 0;
//
// curl_multi_fdset(multi_handle, &fdread, &fdwrite, &fdexcep, &maxfd);
//
//#ifdef _WIN32
// Sleep(100);
//#else
// struct timeval wait = { 0, 100 * 1000 }; /* 100ms */
// rc = select(0, NULL, NULL, NULL, &wait);
//#endif
//
// /* get file descriptors from the transfers */
// switch (select(maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout)) {
// case -1:
// /* select error */
// still_running = 0;
// stt->doEmitionRedFoundData("select() returns error, this is badness");
// break;
// case 0:
// default:
// /* timeout or readable/writable sockets */
// do
// {
// res = curl_multi_perform(multi_handle, &still_running);
// } while (res == CURLM_CALL_MULTI_PERFORM);
// break;
// }
// }
//
// curl_multi_remove_handle(multi_handle, curl);
// curl_multi_cleanup(multi_handle);
// curl_easy_cleanup(curl);
//
// if (res == CURLE_OK ||
// (port == 21 && buffer->size() > 0)) {
//
// if (digestMode)
// {
// if (port != 21 && lpString != NULL) {
// int pos = Utils::ustrstr(*buffer, "\r\n\r\n");
// if (pos != -1) {
// *buffer = buffer->substr(pos + 4);
// }
// }
// }
//
// if (MapWidgetOpened) stt->doEmitionAddIncData(QString(ip), QString(buffer->c_str()));
// Activity += buffer->size();
// return buffer->size();
// }
// else {
// if (res == 6) return -2;
// else if (res != 28 &&
// res != 7 &&
// res != 67 &&
// res != 52 &&
// res != 55 &&
// res != 56 &&
// res != 35 &&
// res != 19 &&
// res != 23) {
// if (res == 5) {
// stt->doEmitionRedFoundData("Couldn't resolve proxy. The given proxy host could not be resolved. ");
// return -2;
// }
// else if (res == 13) {
// stt->doEmitionFoundData("Unknown ftp. (" + QString::number(res) + ") " +
// QString(ip) + ":" + QString::number(port));
// return -2;
// }
// else if (res == 8) {
// stt->doEmitionFoundData("Strange ftp reply. (" +
// QString::number(res) + ") " + QString(ip) +
// ":" + QString::number(port));
// return -2;
// }
// else if (res == 18) {
// stt->doEmitionFoundData("Inappropriate file size. (" +
// QString::number(res) + ") " + QString(ip) +
// ":" + QString::number(port));
// return -2;
// }
// else stt->doEmitionRedFoundData("CURL error: (" + QString::number(res) + ") " +
// QString(ip) + ":" + QString::number(port));
// }
//
// if (res == 23 && buffer->size() > 0) {
// if (MapWidgetOpened) stt->doEmitionAddIncData(QString(ip), QString("[OVERFLOW]"));
// return buffer->size();
// }
// else return -1;
// }
//
// if (MapWidgetOpened) stt->doEmitionAddIncData(QString(ip), QString(buffer->c_str()));
// return buffer->size();
// }
// else {
// stt->doEmitionRedFoundData("Curl error.");
// return -1;
// };
//}
2015-03-17 14:30:53 +00:00
2015-04-18 13:05:35 +00:00
int Connector::connectToPort(char* ip, int port)
2015-03-05 14:29:05 +00:00
{
if(gPingNScan)
{
if(_pingMyTarget(ip) == 0) return -2;
2015-03-05 14:29:05 +00:00
};
std::string buffer;
int size = 0;
2015-04-01 12:39:14 +00:00
if (port == 22) size = SSHAuth::SSHLobby(ip, port, &buffer);
else size = nConnect(ip, port, &buffer);
2015-03-05 14:29:05 +00:00
2015-03-22 00:43:15 +00:00
if(size > 0)
{
2015-04-04 08:47:27 +00:00
++Alive;//ME2
++found;//PieStat
2015-03-22 00:43:15 +00:00
Lexems lx;
2015-04-18 13:05:35 +00:00
lx.filler(ip, port, &buffer, size, &lx);
}
else if (size == -2) return -2;
2015-03-05 14:29:05 +00:00
return 0;
}