mirror of
https://github.com/ChronosX88/nesca.git
synced 2024-11-23 18:52:19 +00:00
271 lines
7.1 KiB
C++
271 lines
7.1 KiB
C++
|
#pragma once
|
||
|
|
#include "CheckKey_Th.h"
|
||
|
|
#include "CheckProxy_Th.h"
|
||
|
|
#include "STh.h"
|
||
|
|
#include "externData.h"
|
||
|
|
#include "externFunctions.h"
|
||
|
|
|
||
|
|
void getSubStrEx(char *src, char *startStr, char *endStr, char *dest, int szDest)
|
||
|
|
{
|
||
|
|
ZeroMemory(dest, szDest);
|
||
|
|
char *ptr1 = strstri(src, startStr);
|
||
|
|
if(ptr1 != NULL)
|
||
|
|
{
|
||
|
|
char *ptr2 = strstri(ptr1, endStr);
|
||
|
|
if(ptr2 != NULL)
|
||
|
|
{
|
||
|
|
int szStartStr = strlen(startStr);
|
||
|
|
int sz = ptr2 - ptr1 - szStartStr;
|
||
|
|
strncpy(dest, ptr1 + szStartStr, sz < szDest ? sz : szDest);
|
||
|
|
};
|
||
|
|
};
|
||
|
|
}
|
||
|
|
void getSubStr(char *src, char *startStr, char *endStr, char *dest, int szDest)
|
||
|
|
{
|
||
|
|
ZeroMemory(dest, szDest);
|
||
|
|
char *ptr1 = strstri(src, startStr);
|
||
|
|
if(ptr1 != NULL)
|
||
|
|
{
|
||
|
|
char *ptr2 = strstri(ptr1, endStr);
|
||
|
|
if(ptr2 != NULL)
|
||
|
|
{
|
||
|
|
int sz = ptr2 - ptr1;
|
||
|
|
strncpy(dest, ptr1, sz < szDest ? sz : szDest);
|
||
|
|
};
|
||
|
|
};
|
||
|
|
}
|
||
|
|
int emitIfOK = -1;
|
||
|
|
int KeyCheckerMain()
|
||
|
|
{
|
||
|
|
int kLen = strlen(trcPersKey);
|
||
|
|
if(kLen == 0)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] Key field is empty.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
else if(kLen < 32)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] Key length is not valid.");
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
char msg[1024] = {0};
|
||
|
|
char ndbServer[64] = {0};
|
||
|
|
char ndbScript[64] = {0};
|
||
|
|
|
||
|
|
sockaddr_in sockAddr;
|
||
|
|
sockAddr.sin_family = AF_INET;
|
||
|
|
sockAddr.sin_port = htons(atoi(trcSrvPortLine));
|
||
|
|
|
||
|
|
strcpy(msg, "GET ");
|
||
|
|
strcat(msg, "/");
|
||
|
|
strcat(msg, trcScr);
|
||
|
|
strcat(msg, " HTTP/1.0\r\nHost: ");
|
||
|
|
strcat(msg, trcSrv);
|
||
|
|
strcat(msg, "\r\nX-Nescav3: True");
|
||
|
|
strcat(msg, "\r\nConnection: close");
|
||
|
|
strcat(msg, "\r\n\r\n");
|
||
|
|
|
||
|
|
HOSTENT *host;
|
||
|
|
|
||
|
|
#if defined(WIN32) || defined(_WIN32) || defined(__WIN32) && !defined(__CYGWIN__)
|
||
|
|
if(inet_addr(trcSrv) != INADDR_NONE) sockAddr.sin_addr.S_un.S_addr = inet_addr(trcSrv);
|
||
|
|
else if(host=gethostbyname (trcSrv)) ((unsigned long*) &sockAddr.sin_addr)[0] = ((unsigned long**)host->h_addr_list)[0][0];
|
||
|
|
#else
|
||
|
|
if(inet_addr(trcSrv) != INADDR_NONE) sockAddr.sin_addr.s_addr = inet_addr(trcSrv);
|
||
|
|
else if(host=gethostbyname (trcSrv)) ((unsigned long*) &sockAddr.sin_addr)[0] = ((unsigned long**)host->h_addr_list)[0][0];
|
||
|
|
#endif
|
||
|
|
SOCKET sock = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
|
||
|
|
|
||
|
|
stt->doEmitionYellowFoundData("[Key check] Requesting server ip...");
|
||
|
|
int test = connect(sock, (sockaddr*)&sockAddr, sizeof(sockAddr));
|
||
|
|
if(test == -1)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -connect() returned. Cannot connect to balancer! " + QString::number(WSAGetLastError()) + ".");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
|
||
|
|
test = send(sock, msg, strlen(msg), 0);
|
||
|
|
|
||
|
|
if(test == -1)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -send() returned. Cannot send to balancer! " + QString::number(WSAGetLastError()) + ".");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
|
||
|
|
ZeroMemory(msg, sizeof(msg));
|
||
|
|
int ror = sizeof(msg);
|
||
|
|
|
||
|
|
test = recv(sock, msg, sizeof(msg), 0);
|
||
|
|
char buff[512] = {0};
|
||
|
|
while((test = recv(sock, msg, sizeof(msg), 0)) != 0)
|
||
|
|
{
|
||
|
|
strcat(msg, buff);
|
||
|
|
};
|
||
|
|
|
||
|
|
if(test == -1)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -recv() returned. Cannot recv from balancer! " + QString::number(WSAGetLastError()) + ".");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
|
||
|
|
char *t1;
|
||
|
|
char *t2;
|
||
|
|
if(strstr(msg, "http://") != NULL)
|
||
|
|
{
|
||
|
|
t1 = strstr(msg, "http://");
|
||
|
|
if(strstr((char*)(t1 + strlen("http://")), "/") != NULL)
|
||
|
|
{
|
||
|
|
t2 = strstr((char*)(t1 + strlen("http://")), "/");
|
||
|
|
int ln = t2 - t1 - strlen("http://");
|
||
|
|
if(ln > 64)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Received server string is not valid!");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
else strncpy(ndbServer, (char*)(t1 + strlen("http://")), ln);
|
||
|
|
|
||
|
|
|
||
|
|
if(strlen(t2) > 64)
|
||
|
|
{
|
||
|
|
stt->doEmitionYellowFoundData("[Key check] -Fragmentation detected!");
|
||
|
|
if(strstr(t2, "\r\n") != NULL)
|
||
|
|
{
|
||
|
|
char *t3 = strstr(t2, "\r\n");
|
||
|
|
int y = (int)(t3 - t2);
|
||
|
|
|
||
|
|
if(y > 64)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Received server string is not valid!");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
strncpy(ndbScript, t2, y);
|
||
|
|
};
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Received server string is not valid!");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
} else strcpy(ndbScript, t2);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Cannot receive script value!");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
|
||
|
|
ZeroMemory(msg, sizeof(msg));
|
||
|
|
|
||
|
|
stt->doEmitionGreenFoundData("[Key check] -OK. -Server string aquired! Checking key...");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
sockAddr.sin_family = AF_INET;
|
||
|
|
sockAddr.sin_port = htons(atoi(trcSrvPortLine));
|
||
|
|
strcpy(msg, "GET ");
|
||
|
|
strcat(msg, "/api/checkaccount?key=");
|
||
|
|
strncat(msg, trcPersKey, 32);
|
||
|
|
strcat(msg, " HTTP/1.0\r\nHost: ");
|
||
|
|
strcat(msg, ndbServer);
|
||
|
|
strcat(msg, "\r\nConnection: close");
|
||
|
|
strcat(msg, "\r\n\r\n");
|
||
|
|
|
||
|
|
#if defined(WIN32) || defined(_WIN32) || defined(__WIN32) && !defined(__CYGWIN__)
|
||
|
|
if(inet_addr(ndbServer) != INADDR_NONE) sockAddr.sin_addr.S_un.S_addr = inet_addr(ndbServer);
|
||
|
|
else if(host=gethostbyname (ndbServer)) ((unsigned long*) &sockAddr.sin_addr)[0] = ((unsigned long**)host->h_addr_list)[0][0];
|
||
|
|
#else
|
||
|
|
if(inet_addr(ndbServer) != INADDR_NONE) sockAddr.sin_addr.s_addr = inet_addr(ndbServer);
|
||
|
|
else if(host=gethostbyname (ndbServer)) ((unsigned long*) &sockAddr.sin_addr)[0] = ((unsigned long**)host->h_addr_list)[0][0];
|
||
|
|
#endif
|
||
|
|
sock = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
|
||
|
|
|
||
|
|
int c = connect(sock, (sockaddr*)&sockAddr, sizeof(sockAddr));
|
||
|
|
if(c == SOCKET_ERROR)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Connection timeout.");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
c = send(sock, msg, strlen(msg), 0);
|
||
|
|
if(c == SOCKET_ERROR)
|
||
|
|
{
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Send error.");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
ZeroMemory(msg, sizeof(msg));
|
||
|
|
test = recv(sock, msg, 512, 0);
|
||
|
|
|
||
|
|
if(strstr(msg, "202 Accepted") != NULL)
|
||
|
|
{
|
||
|
|
|
||
|
|
stt->doEmitionGreenFoundData("[Key check] -OK. Key is valid!");
|
||
|
|
|
||
|
|
CSSOCKET(sock);
|
||
|
|
|
||
|
|
if(emitIfOK == 0) stt->doEmitionStartScanIP();
|
||
|
|
else if(emitIfOK == 1) stt->doEmitionStartScanDNS();
|
||
|
|
else if(emitIfOK == 2) stt->doEmitionStartScanImport();
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
else if(strstr(msg, "400 Bad Request") != NULL)
|
||
|
|
{
|
||
|
|
|
||
|
|
QString errorDef = GetNSErrorDefinition(msg, "notify");
|
||
|
|
if(errorDef == "Invalid access key") stt->doEmitionYellowFoundData("[NS-Track] [Key is unauthorized] A valid key is required.");
|
||
|
|
else stt->doEmitionYellowFoundData("[NS-Track] -FAIL! [400 Bad Request : " + GetNSErrorDefinition(msg, "notify") + "]");
|
||
|
|
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
else if(strstr(msg, "503 Bad Gateway") != NULL)
|
||
|
|
{
|
||
|
|
stt->doEmitionYellowFoundData("[NS-Track] 503 Backend not responding!");
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
|
||
|
|
char header[64] = {0};
|
||
|
|
getSubStrEx(msg, "http/1.1 ", "\r\n", header, 64);
|
||
|
|
stt->doEmitionYellowFoundData("[Key check] -FAIL! An error occured. (" + QString::number(WSAGetLastError()) + ") Header: <u>" + QString::fromLocal8Bit(header) + "</u>");
|
||
|
|
if(gDebugMode) stt->doEmitionDebugFoundData(QString(msg));
|
||
|
|
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
};
|
||
|
|
|
||
|
|
ZeroMemory(msg, sizeof(msg));
|
||
|
|
CSSOCKET(sock);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
|
||
|
|
stt->doEmitionRedFoundData("[Key check] -Balancer replied with invalid string.");
|
||
|
|
if(gDebugMode) stt->doEmitionDebugFoundData(QString(msg));
|
||
|
|
CSSOCKET(sock);
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
void CheckKey_Th::run()
|
||
|
|
{
|
||
|
|
KeyCheckerMain();
|
||
|
|
};
|