C++編寫自定義TCP包併傳送

轉載修改自：
https://blog.csdn.net/e_wsq/article/details/12766443

#include "pch.h"
#include <stdio.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <time.h>
#include <windows.h>
#include <string.h>
#include <stdlib.h>

#pragma comment(lib,"ws2_32.lib")
#pragma  warning (disable:4996)  //去除安全方法提示

#define IPVER 4
#define MAX_BUFF_LEN 65500

typedef struct ip_hdr
{
	UCHAR h_verlen;
	UCHAR tos;
	USHORT total_len;
	USHORT ident;
	USHORT frag_and_flags;
	UCHAR ttl;
	UCHAR proto;
	USHORT checksum;
	ULONG sourceIP;
	ULONG destIP;
}IP_HEADER;

typedef struct tsd_hdr
{
	ULONG saddr;
	ULONG daddr;
	UCHAR mbz;
	UCHAR ptcl;
	USHORT tcpl;
}PSD_HEADER;

typedef struct tcp_hdr
{
	USHORT th_sport;
	USHORT th_dport;
	ULONG th_seq;
	ULONG th_ack;
	UCHAR th_lenres;
	UCHAR th_flag;
	USHORT th_win;
	USHORT th_sum;
	USHORT th_urp;
}TCP_HEADER;

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);
}

int main(int argc, char* argv[])
{
	WSADATA WSAData;
	SOCKET sock;

	IP_HEADER ipHeader;
	TCP_HEADER tcpHeader;
	PSD_HEADER psdHeader;

	char Sendto_Buff[MAX_BUFF_LEN];
	unsigned short check_Buff[MAX_BUFF_LEN];
	const char tcp_send_data[] = { "This is TCP data!" };

	BOOL flag;
	int rect, nTimeOver;

	if (argc != 5)
	{
		printf("Useage: SendTcp soruce_ip source_port dest_ip dest_port \n");

		return false;
	}

	if (WSAStartup(MAKEWORD(2, 2), &WSAData) != 0)
	{
		printf("WSAStartup Error!\n");
		return false;
	}

	if ((sock = WSASocket

	(AF_INET, SOCK_RAW, IPPROTO_IP, 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;
	}

	ipHeader.h_verlen = (IPVER << 4 | sizeof(ipHeader) / sizeof(unsigned long));

	ipHeader.tos = (UCHAR)0;
	ipHeader.total_len = htons((unsigned short)sizeof(ipHeader) + sizeof

	(tcpHeader)+sizeof(tcp_send_data));
	ipHeader.ident = 0;
	ipHeader.frag_and_flags = 0;
	ipHeader.ttl = 128;
	ipHeader.proto = IPPROTO_TCP;
	ipHeader.checksum = 0;
	ipHeader.sourceIP = inet_addr(argv[1]);
	ipHeader.destIP = inet_addr(argv[3]);

	memset(check_Buff, 0, MAX_BUFF_LEN);
	memcpy(check_Buff, &ipHeader, sizeof(IP_HEADER));
	ipHeader.checksum = checksum(check_Buff, sizeof(IP_HEADER));

	psdHeader.saddr = ipHeader.sourceIP;
	psdHeader.daddr = ipHeader.destIP;
	psdHeader.mbz = 0;
	psdHeader.ptcl = ipHeader.proto;
	psdHeader.tcpl = htons(sizeof(TCP_HEADER) + sizeof(tcp_send_data));

	tcpHeader.th_dport = htons(atoi(argv[4]));
	tcpHeader.th_sport = htons(atoi(argv[2]));
	tcpHeader.th_seq = 0;
	tcpHeader.th_ack = 0;
	tcpHeader.th_lenres = (sizeof(tcpHeader) / sizeof(unsigned long) << 4 | 0);
	tcpHeader.th_flag = 2;

	tcpHeader.th_win = htons((unsigned short)16384);
	tcpHeader.th_urp = 0;
	tcpHeader.th_sum = 0;

	memset(check_Buff, 0, MAX_BUFF_LEN);
	memcpy(check_Buff, &psdHeader, sizeof(psdHeader));
	memcpy(check_Buff + sizeof(psdHeader), &tcpHeader, sizeof(tcpHeader));
	memcpy(check_Buff + sizeof(PSD_HEADER) + sizeof

	(TCP_HEADER), tcp_send_data, sizeof(tcp_send_data));
	tcpHeader.th_sum = checksum(check_Buff, sizeof(PSD_HEADER) + sizeof

	(TCP_HEADER)+sizeof(tcp_send_data));

	memset(Sendto_Buff, 0, MAX_BUFF_LEN);
	memcpy(Sendto_Buff, &ipHeader, sizeof(IP_HEADER));
	memcpy(Sendto_Buff + sizeof(IP_HEADER), &tcpHeader, sizeof

	(TCP_HEADER));
	memcpy(Sendto_Buff + sizeof(IP_HEADER) + sizeof

	(TCP_HEADER), tcp_send_data, sizeof(tcp_send_data));
	int datasize = sizeof(IP_HEADER) + sizeof(TCP_HEADER) + sizeof

	(tcp_send_data);
	SOCKADDR_IN dest;
	memset(&dest, 0, sizeof(dest));
	dest.sin_family = AF_INET;
	dest.sin_addr.s_addr = inet_addr(argv[3]);

	rect = sendto(sock, Sendto_Buff, datasize, 0, (struct sockaddr*)&dest,

		sizeof(dest));
	if (rect == SOCKET_ERROR)
	{
		printf("send error!:%d\n", WSAGetLastError());
		return false;
	}
	else
		printf("send ok!\n");
	closesocket(sock);
	WSACleanup();

	return 0;
}