木马程序C&＃43;&＃43;类
#pragma once是一个比较常用的C/C&＃43;&＃43;杂注&＃xff0c;只要在头文件的最开始加入这条杂注&＃xff0c;就能够保证头文件只被编译一次。
头文件
#if !defined(_MYTCPTRAN_H_)
#define _MYTCPTRAN_H_
#endif
#if _MSC_VER> 1000
#pragma once
#endif
#include
#include
#pragma comment(lib,"ws2_32.lib")
using namespacestd;
//.........................
#define SOCKETBIND1
#define SOCKETNOBIND2
#define SOCKET_TIMEOUT-100
//自己的代码
class CMyTcpTran
{
public:
CMyTcpTran();
virtual~CMyTcpTran();
public:
staticBOOL InitSocketLibray(intlower,int higver);//初始化winsocke api链接库文件
public:
//初始化socket操作
SOCKET InitSocket(intSocketType,string strBindIp,u_shortBindPort,int opt);
//针对本地监听的程序
SOCKET myaccept(SOCKET s,structsockaddr* addr,int*addrlen);
//接受
intmysend(SOCKET sock,constchar*buf,int len,intflag,int overtime);
//发送
intmyrecv(SOCKET sock,char*buf,int len,intflag,int overtime,char*EndMark,BOOL soonflag&＃61;FALSE);
private:
SOCKET m_Socket;
};
文件本身
// MyTcpTran.cp______________p: implementation ofthe CMyTcpTran class.
//
//
#include
#include
#include "MyTcpTran.h"
#ifdef _DEBUG
#undef THIS_FILE
static charTHIS_FILE[]&＃61;__FILE__;
#define newDEBUG_NEW
#endif
//
// Construction/Destruction
//
CMyTcpTran::CMyTcpTran()
{
m_Socket&＃61;INVALID_SOCKET;//初始化套接字为无效套接字
}
CMyTcpTran::~CMyTcpTran()
{
//暂时不需要
}
BOOL CMyTcpTran::InitSocketLibray(intlower,inthigver)
{
WORD wVersion&＃61;0;
interret&＃61;-1;
WSADATA wsaData;
//wVersion&＃61;MAKEWORD(2,1); //makeword 将两个8位的字节。。合成一个16位的字so..又理解了
wVersion&＃61;MAKEWORD(lower,higver);
erret&＃61;WSAStartup(wVersion,&wsaData);
if(LOBYTE(wsaData.wVersion)!&＃61;2 || HIBYTE(wsaData.wVersion)!&＃61;2) //makeword lobyte hibyte
{
MessageBox(NULL,_T("版本太低了"),_T("提示"),MB_OK);
returnFALSE;
}
returnTRUE;
}
SOCKET CMyTcpTran::InitSocket(intSocketType,string strBindIp,u_short BindPort,intopt)
{
SOCKET socketid &＃61; INVALID_SOCKET;
socketid&＃61;socket(PF_INET,SOCK_STREAM,0);
SOCKADDR_IN sockStruct;
sockStruct.sin_family&＃61;AF_INET;
if(strBindIp.empty())//绑定任意IP
{
sockStruct.sin_addr.S_un.S_addr&＃61;INADDR_ANY;
}
else
{
sockStruct.sin_addr.S_un.S_addr&＃61;inet_addr(strBindIp.c_str());//又学到一个函数c_str
}
sockStruct.sin_port&＃61;htons(BindPort);//htons 以及u_short(16bit)正好65535fuck....!!!
if(SocketType&＃61;&＃61;SOCKETBIND)
{
//主动连
if(connect(socketid,(sockaddr*)&sockStruct,sizeof(sockStruct))&＃61;&＃61;SOCKET_ERROR)
{
OutputDebugString(_T("链接错误"));
closesocket(socketid);
shutdown(socketid,2);
socketid&＃61;INVALID_SOCKET;
}
m_Socket&＃61;socketid;
}
//监听端口 等待链接
elseif(SocketType&＃61;&＃61;SOCKETNOBIND)
{
if(bind(socketid,(sockaddr*)&sockStruct,sizeof(sockStruct))&＃61;&＃61;INVALID_SOCKET) //绑定
{
closesocket(socketid);
socketid&＃61;INVALID_SOCKET;
}
else
{
if(listen(socketid,SOMAXCONN)&＃61;&＃61;SOCKET_ERROR) //监听
{
closesocket(socketid);
socketid&＃61;INVALID_SOCKET;
}
}
m_Socket&＃61;socketid;
}
returnsocketid; //完美封装
}
SOCKET CMyTcpTran::myaccept(SOCKET s,structsockaddr* addr,int*addrlen)
{
SOCKET acpsocket&＃61;INVALID_SOCKET;
acpsocket&＃61;accept(s,addr,addrlen);
returnacpsocket; //创建一个新的套接字,处理链接
}
int CMyTcpTran::myrecv(SOCKET sock,char*buf,intlen,intflag,intovertime,char*EndMark,BOOL soonflag)
{
intret;
intnleft&＃61;len;
intidx&＃61;0;
intnCount&＃61;0;
fd_set readfds; //文件描述符集合
structtimeval timeout; //
timeout.tv_sec&＃61;0;
timeout.tv_usec&＃61;500;//
DWORD s_stime&＃61;GetTickCount(); //开机后的秒数
while(nleft>0)
{
MSG msg;
PeekMessage(&msg,NULL,0,0,PM_REMOVE); //压力泵。。然后压一个删一个。。嘎嘎。。
if(msg.message&＃61;&＃61;WM_QUIT)
return0;
FD_ZERO(&readfds);//清0
FD_SET(sock,&readfds);//加入。。。虽然以前不知道FD_SET 不过今天知道了
if(select(0,&readfds,NULL,NULL,&timeout)&＃61;&＃61;SOCKET_ERROR)
{
returnSOCKET_ERROR;
}
DWORD e_time&＃61;GetTickCount();
if(!FD_ISSET(sock,&readfds))
{
if(e_time-s_stime>overtime*1000) //超时
returnSOCKET_ERROR;
else
continue;
}
ret&＃61;recv(sock,&buf[idx],nleft,flag);//nleft&＃xff1a;Length of buf. 接受缓存区
if(soonflag&＃61;&＃61;TRUE)
{
returnret;
//立即返回接受到的字节
}
s_stime&＃61;e_time;
if(ret<&＃61;0) //错误处理
{
intLastError&＃61;GetLastError();
if((-1 &＃61;&＃61; ret) && (WSAETIMEDOUT &＃61;&＃61; LastError))
continue;
if((-1 &＃61;&＃61; ret) && (WSAEWOULDBLOCK&＃61;&＃61; LastError)) //没找到这个错误类型的定义。。
{
if(nCount<2000)
{
Sleep(10);
nCount&＃43;&＃43;;
continue;
}
}
returnret;
}
nCount&＃61;0;
nleft-&＃61;ret;
idx&＃43;&＃61;ret;
if(EndMark!&＃61;NULL && idx>5)
{
if(strstr(buf&＃43;(idx-5),EndMark)!&＃61;NULL) //应该是查找EndMark
{
break;
}
}
}
returnidx;
}
int CMyTcpTran::mysend(SOCKET sock,constchar*buf,intlen,intflag,intovertime) //发送了
{
intret;
intnLeft&＃61;len;
intidx;
fd_set readfds;
structtimeval timeout;
timeout.tv_sec&＃61;0;
timeout.tv_usec&＃61;500;
DWORD s_time &＃61; GetTickCount();
while(nLeft>0)
{
MSG msg;
PeekMessage(&msg,NULL,0,0,PM_REMOVE);
if(msg.message&＃61;&＃61;WM_QUIT)
return0;
FD_ZERO(&readfds);
FD_SET(sock,&readfds);
interrorret&＃61;select(0,NULL,&readfds,NULL,&timeout);
if(errorret&＃61;&＃61;SOCKET_ERROR)
{
OutputDebugString(_T("mysendEx socket 错误"));
returnSOCKET_ERROR;
}
DWORD e_time&＃61;GetTickCount();
if(!FD_ISSET(sock,&readfds))
{
if(e_time-s_time>overtime*1000)
{
OutputDebugString(_T("超时"));
return0;
}
else
{
continue;
}
}
ret&＃61;send(sock,&buf[idx],nLeft,flag);
if(ret<&＃61;0)
{
returnret;
}
nLeft-&＃61;ret;
idx&＃43;&＃61;ret;
}
returnlen;
}
类说明
头文件定义了一个类&＃xff0c;包括构造函数&＃xff0c;析构函数。和5个主要函数。
InitSocketLibray返回BOOL
InitSocket返回SOCKET
Myaccept返回SOCKET
Mysend返回INT
Myrecv返回INT
此外还有私有变量m_Socket;
InitSocketLibray函数
比较简单入参为两个整型&＃xff0c;表示winsock版本号。
版本低于2.2就弹出报错。返回FALSE
InitSocket函数
支持4个参数&＃xff1a;int SocketType,string strBindIp,u_short BindPort,int opt
第1个参数是连接类型&＃xff0c;确定是服务端还是客户端。
如果是SOCKETBIND说明是服务端&＃xff0c;如果是SOCKETNOBIND,那么就是客户端。
第2个参数strBindIp是绑定的IP地址。
第3个参数BindPort是绑定的端口
第4个参数Opt是保留参数&＃xff0c;暂时没有用处。
如果是服务端则返回一个服务端的SOCKET&＃xff0c;如果是客户端则返回一个客户端的SOCKET。
Myaccept函数
输入3个参数&＃xff0c;套接字&＃xff0c;sockaddr指针和,addrlen.
返回一个新的套接字用户处理链接。
Mysend函数
接受5个参数&＃xff0c;分别是套接字&＃xff0c;接受缓冲区&＃xff0c;长度&＃xff0c;flag和overtime.
最后返回发送的字节数&＃xff0c;如果发送失败返回负数。
超时或退出返回0.
Myrecv函数
输入参数7个&＃xff0c;有服务端套接字&＃xff0c;接受缓存区&＃xff0c;长度&＃xff0c;flag,overtime, EndMark&＃xff0c;soonflag
其中soonflag表示立即返回接受到的字节。
如果soonflag为false,返回接受的字节数idx。
如果soonflag为true&＃xff0c;则返回指定长度的字节数。
如果出错返回负数。
此外&＃xff0c;如果EndMark不会NULL&＃xff0c;则会判断是否有EndMark存在&＃xff0c;存在则也函数结束。
TIPS
fd_set
select()机制中提供一fd_set的数据结构&＃xff0c;实际上是一long类型的数组&＃xff0c;每一个数组元素都能与一打开的文件句柄&＃xff08;不管是socket句柄&＃xff0c;还是其他文件或命名管道或设备句柄&＃xff09;建立联系&＃xff0c;建立联系的工作由程序员完成&＃xff0c;当调用select()时&＃xff0c;由内核根据IO状态修改fd_set的内容&＃xff0c;由此来通知执行了select()的进程哪一socket或文件发生了可读或可写事件。
FD_ZERO(&set); /*将set清零使集合中不含任何fd*/
FD_SET(fd, &set); /*将fd加入set集合*/
FD_CLR(fd, &set); /*将fd从set集合中清除*/
FD_ISSET(fd, &set); /*在调用select()函数后&＃xff0c;用FD_ISSET来检测fd是否在set集合中&＃xff0c;当检测到fd在set中则返回真&＃xff0c;否则&＃xff0c;返回假&＃xff08;0&＃xff09;*/
select
select()的机制中提供一fd_set的数据结构&＃xff0c;实际上是一long类型的数组&＃xff0c; 每一个数组元素都能与一打开的文件句柄&＃xff08;不管是Socket句柄&＃xff0c;还是其他 文件或命名管道或设备句柄&＃xff09;建立联系&＃xff0c;建立联系的工作由程序员完成&＃xff0c; 当调用select()时&＃xff0c;由内核根据IO状态修改fd_set的内容&＃xff0c;由此来通知执 行了select()的进程哪一Socket或文件可读或可写。主要用于Socket通信当中!
PeekMessage
PeekMessage是一个Windows API函数。该函数为一个消息检查线程消息队列&＃xff0c;并将该消息&＃xff08;如果存在&＃xff09;放于指定的结构。
BOOL PeekMessage(
LPMSG IpMsg,
HWND hWnd,
UINT wMSGfilterMin,
UINT wMsgFilterMax,
UINT wRemoveMsg
);
lpMsg
接收消息信息的MSG结构指针。
hWnd
其消息被检查的窗口句柄。
wMsgFilterMin
指定被检查的消息范围里的第一个消息。
wMsgFilterMax
指定被检查的消息范围里的最后一个消息。
wRemoveMsg
确定消息如何被处理。此参数可取下列值之一&＃xff1a;
京公网安备 11010802041100号