📄 socksupport.h
字号:
/**
@file SockSupport.h
Implementation of the winsock2
@author dream2fly
@date 20060921
@note 创建文件
**/
#ifndef SOCKSUPPORT_H_
#define SOCKSUPPORT_H_
#include <vector>
#include <string>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <mstcpip.h>
#pragma comment(lib, "ws2_32.lib")
#define STATUS_FAILED 0xFFFF //定义异常出错代码
#define MAX_PACK_LEN 65535 // The max IP packet to receive.
#define MAX_ADDR_LEN 16 // The dotted addres's length.
#define MAX_PROTO_TEXT_LEN 16 // The length of sub protocol name(like "TCP").
#define MAX_PROTO_NUM 12 // The count of sub protocols.
#define MAX_HOSTNAME_LAN 255 // The max length of the host name.
/*
// The IP packet is like this. Took from RFC791.
0 1 2 3
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|Version| IHL |Type of Service| Total Length |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Identification |Flags| Fragment Offset |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Time to Live | Protocol | Header Checksum |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Source Address |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Destination Address |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Options | Padding |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
*/
//定义IP首部
typedef struct ip_hdr // 20 Bytes
{
unsigned char h_verlen; //4位首部长度,4位IP版本号
unsigned char tos; //8位服务类型TOS
unsigned short total_len; //16位总长度（字节）
unsigned short ident; //16位标识
unsigned short frag_and_flags; //3位标志位
unsigned char ttl; //8位生存时间 TTL
unsigned char proto; //8位协议 (TCP, UDP 或其他)
unsigned short checksum; //16位IP首部校验和
unsigned int sourceIP; //32位源IP地址
unsigned int destIP; //32位目的IP地址
}IP_HEADER;
/*
// The TCP packet is like this. Took from RFC793.
0 1 2 3
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Source Port | Destination Port |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Sequence Number |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Acknowledgment Number |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Data | |U|A|P|R|S|F| |
| Offset| Reserved |R|C|S|S|Y|I| Window |
| | |G|K|H|T|N|N| |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Checksum | Urgent Pointer |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| Options | Padding |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
| data |
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
*/
//定义TCP首部
typedef struct tcp_hdr // 20 Bytes
{
unsigned short th_sport; //16位源端口
unsigned short th_dport; //16位目的端口
unsigned int th_seq; //32位序列号
unsigned int th_ack; //32位确认号
unsigned char th_lenres; //4位首部长度/6位保留字
unsigned char th_flag; //6位标志位
unsigned short th_win; //16位窗口大小
unsigned short th_sum; //16位校验和
unsigned short th_urp; //16位紧急数据偏移量
}TCP_HEADER;
/*
// The TCP's pseudo header is like this. Took from RFC793.
+--------+--------+--------+--------+
| Source Address |
+--------+--------+--------+--------+
| Destination Address |
+--------+--------+--------+--------+
| zero | PTCL | TCP Length |
+--------+--------+--------+--------+
*/
//定义TCP伪首部 pseudo[ 'psju:dou ] 假的,冒充的
typedef struct psd_hdr // 16 Bytes
{
unsigned long saddr; //源地址
unsigned long daddr; //目的地址
char mbz; //置空
char ptcl; //协议类型
unsigned short tcpl; //TCP长度
}PSD_HEADER;
/*
// The UDP packet is lick this. Took from RFC768.
0 7 8 15 16 23 24 31
+--------+--------+--------+--------+
| Source | Destination |
| Port | Port |
+--------+--------+--------+--------+
| | |
| Length | Checksum |
+--------+--------+--------+--------+
|
| data octets ...
+---------------- ...
*/
typedef struct udp_hdr // 8 Bytes
{
unsigned short uh_sport; //16位源端口
unsigned short uh_dport; //16位目的端口
unsigned short uh_len; //16位长度
unsigned short uh_sum; //16位校验和
} UDP_HEADER;
typedef struct icmp_hdr // 12 Bytes
{
unsigned char ih_type; //8位类型
unsigned char ih_code; //8位代码
unsigned short ih_cksum; //16位校验和
unsigned short ih_id; //识别号（一般用进程号作为识别号）
unsigned short ih_seq; //报文序列号
unsigned long ih_timestamp; //时间戳
}ICMP_HEADER;
// The protocol's map.
typedef struct proto_map
{
int ProtoNum;
char ProtoText[MAX_PROTO_TEXT_LEN];
}PROTOMAP;
static PROTOMAP ProtoMap[MAX_PROTO_NUM]=
{
{ IPPROTO_IP , "IP " },
{ IPPROTO_ICMP , "ICMP" },
{ IPPROTO_IGMP , "IGMP" },
{ IPPROTO_GGP , "GGP " },
{ IPPROTO_TCP , "TCP " },
{ IPPROTO_PUP , "PUP " },
{ IPPROTO_UDP , "UDP " },
{ IPPROTO_IDP , "IDP " },
{ IPPROTO_ND , "NP " },
{ IPPROTO_RAW , "RAW " },
{ IPPROTO_MAX , "MAX " },
{ NULL , "" }
};
//计算检验和函数
static u_short checksum(u_short *buffer, int size)
{
u_long cksum=0;
while(size >1)
{
cksum+=*buffer++;
size-=sizeof(u_short);
}
if(size==1) cksum+=*(u_char*)buffer;
cksum=(cksum >> 16)+(cksum & 0xffff);
cksum+=(cksum >> 16);
return (u_short)(~cksum);
}
static char* CheckProtocol(int iProtocol)
{
for(int i=0; i<MAX_PROTO_NUM; ++i)
{
if(ProtoMap[i].ProtoNum==iProtocol)
return ProtoMap[i].ProtoText;
}
return "";
}
//判断操作系统涵数及变量
typedef enum _OS_LEVEL{
WIN_UNSUP,
WIN_NT_3_5,
WIN_NT_3_51,
WIN_NT_4,
WIN_NT_4_SP2,
WIN_NT_4_SP3,
WIN_NT_4_SP4,
WIN_NT_4_SP5,
WIN_NT_4_SP6,
WIN_2000,
WIN_2000_SP1,
WIN_2000_SP2,
WIN_XP,
WIN_XP_SP1,
WIN_XP_SP2,
WIN_2003
}OS_LEVEL;
static OS_LEVEL get_win32_type()
{
static OS_LEVEL os_level;
static OSVERSIONINFO oslev;
oslev.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
GetVersionEx(&oslev);
if (oslev.dwPlatformId == VER_PLATFORM_WIN32_NT)
{
static unsigned int servpack = 0;
char *pservpack;
if (pservpack = oslev.szCSDVersion) {
while (*pservpack && !isdigit(*pservpack)) {
pservpack++;
}
if (*pservpack)
servpack = atoi(pservpack);
}
if (oslev.dwMajorVersion == 3) {
if (oslev.dwMajorVersion < 50) {
os_level = WIN_UNSUP;
}
else if (oslev.dwMajorVersion == 50) {
os_level = WIN_NT_3_5;
}
else {
os_level = WIN_NT_3_51;
}
}
else if (oslev.dwMajorVersion == 4) {
if (servpack < 2)
os_level = WIN_NT_4;
else if (servpack <= 2)
os_level = WIN_NT_4_SP2;
else if (servpack <= 3)
os_level = WIN_NT_4_SP3;
else if (servpack <= 4)
os_level = WIN_NT_4_SP4;
else if (servpack <= 5)
os_level = WIN_NT_4_SP5;
else
os_level = WIN_NT_4_SP6;
}
else if (oslev.dwMajorVersion == 5) {
if (oslev.dwMinorVersion == 0) {
if (servpack == 0)
os_level = WIN_2000;
else if (servpack == 1)
os_level = WIN_2000_SP1;
else
os_level = WIN_2000_SP2;
}
else if (oslev.dwMinorVersion == 2) {
os_level = WIN_2003;
}
else if (oslev.dwMinorVersion == 1){
if (servpack < 1)
os_level = WIN_XP;
else if (servpack == 1)
os_level = WIN_XP_SP1;
else
os_level = WIN_XP_SP2;
}
}
else {
os_level = WIN_UNSUP;
}
}
return os_level;
}
//------------------------------------------------------------
// 函 数：TransformIp(char* addr,DWORD &dwIP,bool flag)
// 参 数：char* addr,DWORD &dwIP,bool flag
// 返回值：如果成功返回true，不成功，返回false
// 描 述：字符串的IP与DWORD的IP转化.当为真时，把DWORD的IP转换为字符串，当为假时转化为DWORD型
//------------------------------------------------------------
static bool TransformIp(char* addr,DWORD &dwIP,bool flag) //注意：传入的dwIP参数必须为网络字节循序！
{
//char 转化为 DWORD
if (flag)
{
char* ipaddr = NULL;
in_addr inaddr;
inaddr.s_addr=dwIP;
ipaddr= inet_ntoa(inaddr);
strcpy(addr,ipaddr);
return true;
}
//DWORD 转化为 char
else
{
dwIP = inet_addr(addr);
return true;
}
}
static char* GetLocalIP()
{
char szLocname[MAX_HOSTNAME_LAN];
LPHOSTENT pHostent;
struct in_addr in_addrIP;
int iRc = gethostname(szLocname, MAX_HOSTNAME_LAN);
if (iRc == SOCKET_ERROR)
{
return NULL;
}
pHostent = gethostbyname(szLocname);
for (int nAdapter=0; pHostent->h_addr_list[nAdapter]; nAdapter++)
{
in_addrIP = *(struct in_addr *) (pHostent->h_addr_list[nAdapter]);
break;//just 暂时解决方案
}
return inet_ntoa(in_addrIP);
}
static unsigned long MakeRand32(int i)
{
unsigned long j1,j2,s;
i = i << 15;
//srand( (unsigned int)time(NULL) + i );
j1 = rand()+i;
j2 = rand()+i;
j1 = j1 << 16;
s = j1+j2;
return s;
}
static unsigned short MakeRand16(int i)
{
unsigned short s;
i = i << 15;
//srand( (unsigned int)time(NULL) +i );
s = rand()+i;
return s;
}
static int PrintError(const char* strErr)
{
LPTSTR lpErr=0;
DWORD dwErr;
CString CStrErr;
dwErr=WSAGetLastError();
FormatMessage(
FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
NULL, dwErr,LANG_NEUTRAL,
lpErr, 0, NULL);
if(lpErr==NULL)
{
switch(dwErr)
{
case 10004: lpErr = "Interrupted system call"; break;
case 10009: lpErr = "Bad file number"; break;
case 10013: lpErr = "Permission denied"; break;
case 10014: lpErr = "Bad address"; break;
case 10022: lpErr = "Invalid argument (not bind)"; break;
case 10024: lpErr = "Too many open files"; break;
case 10035: lpErr = "Operation would block"; break;
case 10036: lpErr = "Operation now in progress"; break;
case 10037: lpErr = "Operation already in progress"; break;
case 10038: lpErr = "Socket operation on non-socket"; break;
case 10039: lpErr = "Destination address required"; break;
case 10040: lpErr = "Message too long"; break;
case 10041: lpErr = "Protocol wrong type for socket"; break;
case 10042: lpErr = "Bad protocol option"; break;
case 10043: lpErr = "Protocol not supported"; break;
case 10044: lpErr = "Socket type not supported"; break;
case 10045: lpErr = "Operation not supported on socket"; break;
case 10046: lpErr = "Protocol family not supported"; break;
case 10047: lpErr = "Address family not supported by protocol family"; break;
case 10048: lpErr = "Address already in use"; break;
case 10049: lpErr = "Can't assign requested address"; break;
case 10050: lpErr = "Network is down"; break;
case 10051: lpErr = "Network is unreachable"; break;
case 10052: lpErr = "Net dropped connection or reset"; break;
case 10053: lpErr = "Software caused connection abort"; break;
case 10054: lpErr = "Connection reset by peer"; break;
case 10055: lpErr = "No buffer space available"; break;
case 10056: lpErr = "Socket is already connected"; break;
case 10057: lpErr = "Socket is not connected"; break;
case 10058: lpErr = "Can't send after socket shutdown"; break;
case 10059: lpErr = "Too many references, can't splice"; break;
case 10060: lpErr = "Connection timed out"; break;
case 10061: lpErr = "Connection refused"; break;
case 10062: lpErr = "Too many levels of symbolic links"; break;
case 10063: lpErr = "File name too long"; break;
case 10064: lpErr = "Host is down"; break;
case 10065: lpErr = "No Route to Host"; break;
case 10066: lpErr = "Directory not empty"; break;
case 10067: lpErr = "Too many processes"; break;
case 10068: lpErr = "Too many users"; break;
case 10069: lpErr = "Disc Quota Exceeded"; break;
case 10070: lpErr = "Stale NFS file handle"; break;
case 10091: lpErr = "Network SubSystem is unavailable"; break;
case 10092: lpErr = "WINSOCK DLL Version out of range"; break;
case 10093: lpErr = "Successful WSASTARTUP not yet performed"; break;
case 10071: lpErr = "Too many levels of remote in path"; break;
case 11001: lpErr = "Host not found"; break;
case 11002: lpErr = "Non-Authoritative Host not found"; break;
case 11003: lpErr = "Non-Recoverable errors: FORMERR, REFUSED, NOTIMP"; break;
case 11004: lpErr = "Valid name, no data record of requested type"; break;
default: lpErr = "unknown"; break;
}
}
CStrErr.Format("\n\tFail:%s, Reason:%d/%s , at line %d in %s\n\n",strErr,dwErr,lpErr,__LINE__,__FILE__);
TRACE("\n\tFail:%s, Reason:%d/%s , at line %d in %s\n\n",strErr,dwErr,lpErr,__LINE__,__FILE__);
AfxMessageBox(CStrErr);
exit(EXIT_FAILURE);
return 0;
}
enum link_type
{
_CONNECT,
_SYN,
_ICMP,
_TCP,
_UDP
};
///////////////////////
class CSockSupport
{
public:
CSockSupport(bool bAuto = true);
~CSockSupport(void);
int Init();
int Uninit();
// Check whether the winsock is initialized(supported).
bool IsSupported();
private:
bool m_bSupported;
};
#endif
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -