#include <stdio.h> 
#include <winsock2.h> 
#include <ws2tcpip.h> 
#include <time.h>
#pragma comment(lib,"ws2_32.lib")
char TargetIP[20]; // 目標機IP地址
char FakeIP[20]; // 偽IP地址
//定義IP首部格式
typedef struct ip_hdr
{ 
	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地址 
}IPHEADER; 

//定義TCP首部格式
typedef struct tcp_hdr 
{ 
	USHORT th_sport;         //16位源端口 
	USHORT th_dport;         //16位目的端口 
	unsigned int th_seq;     //32位序列號 
	unsigned int th_ack;     //32位確認號 
	unsigned char th_lenres; //4位首部長度/6位保留字 
	unsigned char th_flag;   //6位標志位 
	USHORT th_win;           //16位窗口大小 
	USHORT th_sum;           //16位校驗和 
	USHORT th_urp;           //16位緊急數據偏移量 
}TCPHEADER; 

//定義TCP偽首部格式
typedef struct tsd_hdr  
{ 
	unsigned long saddr; //源地址 
	unsigned long daddr; //目的地址 
	char mbz; 
	char ptcl;           //協議類型 
	unsigned short tcpl; //TCP長度 
}PSDHEADER; 

//CheckSum:計算校驗和的子函數 
USHORT checksum(USHORT *buffer, int size) 
{ 
	unsigned long cksum=0; 
	while(size >1) 
	{ 
		cksum+=*buffer++; 
		size -=sizeof(USHORT); 
	} 
	if(size ) 
	{ 
		cksum += *(UCHAR*)buffer; 
	} 
	
	cksum = (cksum >> 16) + (cksum & 0xffff); 
	cksum += (cksum >>16); 
	return (USHORT)(~cksum); 
} 

// 產生隨機IP地址，用于填充偽源IP
void GenerateRandIP(char *ip)
{

    int ips[4]={0}; // IP地址四個字段
	srand( (unsigned)time(NULL)); // 設置隨機種子
	// 產生隨機數
	for(int i=0; i<4; i++)
	{
		ips[i]=1 + rand() % (255-1+1);
	}
	// 設置隨機IP地址
	sprintf(ip,"%d.%d.%d.%d",ips[0],ips[1],ips[2],ips[3]);
}

// 產生隨機源端口
int GenerateRandSrcPort()
{
	srand( (unsigned)time(NULL)); // 設置隨機種子
	return 1 + rand() % (65535-1+1);

}
// 封裝偽源IP數據包
void PackFakeIP(char *szSendBuf, int dport)
{
	char srcip[20];
    GenerateRandIP(srcip); // 生成隨機IP地址
	strcpy(FakeIP,srcip);
	IPHEADER ipHeader; // IP首部
	TCPHEADER tcpHeader; // TCP首部 
	PSDHEADER psdHeader; // TCP偽首部
	//填充IP首部 
	ipHeader.h_verlen=(4<<4 | sizeof(ipHeader)/sizeof(unsigned long)); 
	// ipHeader.tos=0; 
	ipHeader.total_len=htons(sizeof(ipHeader)+sizeof(tcpHeader)); 
	ipHeader.ident=1; 
	ipHeader.frag_and_flags=0; 
	ipHeader.ttl=128; 
	ipHeader.proto=IPPROTO_TCP; 
	ipHeader.checksum=0; 
	ipHeader.sourceIP=inet_addr(srcip); 
	ipHeader.destIP=inet_addr(TargetIP); 
	
	//填充TCP首部 
	tcpHeader.th_dport=htons(dport); 
	tcpHeader.th_sport=htons(GenerateRandSrcPort()); //源端口號 
	tcpHeader.th_seq=htonl(0x12345678); 
	tcpHeader.th_ack=0; 
	tcpHeader.th_lenres=(sizeof(tcpHeader)/4<<4|0); 
	tcpHeader.th_flag=2; //修改這里來實現不同的標志位探測，2是SYN，1是FIN，16是ACK探測 等等 
	tcpHeader.th_win=htons(512); 
	tcpHeader.th_urp=0; 
	tcpHeader.th_sum=0; 
	// 填充TCP偽首部
	psdHeader.saddr=ipHeader.sourceIP; 
	psdHeader.daddr=ipHeader.destIP; 
	psdHeader.mbz=0; 
	psdHeader.ptcl=IPPROTO_TCP; 
	psdHeader.tcpl=htons(sizeof(tcpHeader)); 
	
	//計算校驗和 
	memcpy(szSendBuf, &psdHeader, sizeof(psdHeader)); 
	memcpy(szSendBuf+sizeof(psdHeader), &tcpHeader, sizeof(tcpHeader)); 
	tcpHeader.th_sum=checksum((USHORT *)szSendBuf,sizeof(psdHeader)+sizeof(tcpHeader)); 
	
	memcpy(szSendBuf, &ipHeader, sizeof(ipHeader)); 
	memcpy(szSendBuf+sizeof(ipHeader), &tcpHeader, sizeof(tcpHeader)); 
	memset(szSendBuf+sizeof(ipHeader)+sizeof(tcpHeader), 0, 4); 
	ipHeader.checksum=checksum((USHORT *)szSendBuf, sizeof(ipHeader)+sizeof(tcpHeader)); 
	
	memcpy(szSendBuf, &ipHeader, sizeof(ipHeader));

}

// 主函數
int main(int argc, char* argv[]) 
{ 

	WSADATA WSAData; 
	SOCKET sock; 
	SOCKADDR_IN addr_in; // 目標機地址
	
	char szSendBuf[60]={0}; // 發送數據緩存區
	BOOL flag; 
	int rect,nTimeOver; 
	
	// 校驗輸入命令個數
	if (argc != 2)
	{
		printf("usage : FakeIPSend.exe [TargetIP]\n");
		exit(0);
	}
	strcpy(TargetIP,argv[1]); // 設置目標機IP地址
	
	// 初始化Winsock庫
	if (WSAStartup(MAKEWORD(2,2), &WSAData)!=0) 
	{ 
		printf("WSAStartup Error!\n"); 
		return false; 
	} 
	
	// 創建原始套接字
	if ((sock=WSASocket(AF_INET,
		SOCK_RAW,
		IPPROTO_RAW,
		NULL,0,
		WSA_FLAG_OVERLAPPED))==INVALID_SOCKET) 
	{ 
		printf("Socket Setup Error!\n"); 
		return false; 
	} 
	flag=true; 
	// 設置原始套接字參數
	if (setsockopt(sock,IPPROTO_IP, IP_HDRINCL,(char *)&flag,sizeof(flag))==SOCKET_ERROR) 
	{ 
		printf("setsockopt IP_HDRINCL error!\n"); 
		return false; 
	} 
	
	nTimeOver=1000; // 超時時間
	// 設置原始套接字發送超時時間
	if (setsockopt(sock, 
		SOL_SOCKET, 
		SO_SNDTIMEO, 
		(char*)&nTimeOver, 
		sizeof(nTimeOver))==SOCKET_ERROR) 
	{ 
		printf("setsockopt SO_SNDTIMEO error!\n"); 
		return false; 
	} 
	// 設置目標機地址結構
	addr_in.sin_family=AF_INET; 
	addr_in.sin_addr.S_un.S_addr=inet_addr(TargetIP); 
	for(int i=1;i<65535;i++)
	{
		
		addr_in.sin_port=htons(i); // 設置目標端口
		PackFakeIP(szSendBuf,i); // 封裝發送數據包
		// 發送偽IP數據包
		rect=sendto(sock, szSendBuf, sizeof(IPHEADER)+sizeof(TCPHEADER), 
			0, (struct sockaddr*)&addr_in, sizeof(addr_in)); 
		if (rect==SOCKET_ERROR) 
		{ 
			printf("Send error!:%d\n",WSAGetLastError()); 
			return false; 
		} 
		else 
			printf("干擾目標機 %s 成功！偽IP ：%s \n",TargetIP,FakeIP);
		Sleep(500);
	}
	// 釋放套接字及內存資源
	closesocket(sock); 
	WSACleanup(); 
	
	return 0; 
}