TCP 实现

最新推荐文章于 2023-07-12 21:35:58 发布

tlxddup

最新推荐文章于 2023-07-12 21:35:58 发布

阅读量284

点赞数

分类专栏：网络文章标签： java 数据库服务器

本文链接：https://blog.csdn.net/tlxddup/article/details/126194772

版权

网络专栏收录该内容

9 篇文章 0 订阅

订阅专栏

SeqNum:代表的是字节数量，初始值是个随机值

ACKNum:也是字节数量,服务器回ACK是这个包的seqnum加上这个包的长度，比如这个包大小为512,seqnum为1000，则回ACK时就会回1512

客户端到服务器

1:发送1m的文件

2:senfbuffer = 2k

3:mss =512

4:mtu =1500

比如客户端发送1M的数据，但是sendbuffer缓存区为2K，此时sendbuffer如果满了；用户态调用send函数就会报-1的错误，要等sendbuf把数据发送出去，才能调用；mss=512就会将sendbuf中的数据分成四次进行发送，每次发送512次；mtu是作用在数据链路层，如果mss大于mtu，就会将mtu进行拆分成多个1500，如果没有超过就会按mss的大小去发包

慢启动与拥塞控制

前面发的包的数量称指数型增长；当达到某个条件时，乘线性增长，在此过程中如果出现三次连续包丢失需要重传就会进行拥塞控制，减少包的发送数量

滑动窗口

第一个指针之前的数据都已经属于被确认的数据，此时可以调用recv来拿数据，而1与2之间还没有被确认，还不能被recv给获取

超时重传

就是启用一个RTT就是发送包到接受到ACK之间的时间，如果超过这个RTT就启用超时重传机制



#include <stdio.h>

#include <sys/poll.h>
#include <arpa/inet.h>

#define NETMAP_WITH_LIBS

#include <net/netmap_user.h>

#pragma pack(1)


#define ETH_ADDR_LENGTH		6

#define PROTO_IP			0x0800
#define PROTO_ARP			0x0806
#define PROTO_UDP			17
#define PROTO_TCP			6
#define PROTO_ICMP			1


struct ethhdr {
	unsigned char h_dst[ETH_ADDR_LENGTH];
	unsigned char h_src[ETH_ADDR_LENGTH];
	unsigned short h_proto;
	
}; // 14


struct iphdr {

	unsigned char hdrlen:4,
				  version:4; // 0x45
				  

	unsigned char tos;

	unsigned short totlen;
	
	unsigned short id;

	unsigned short flag_offset; //

	unsigned char ttl; //time to live
	// 0x1234// htons

	unsigned char type;

	unsigned short check;

	unsigned int sip;
	unsigned int dip;

}; // 20



struct udphdr {

	unsigned short sport;
	unsigned short dport;

	unsigned short length;
	unsigned short check;

}; // 8


struct udppkt {

	struct ethhdr eh; // 14
	struct iphdr ip;  // 20 
	struct udphdr udp; // 8

	unsigned char data[0];

}; // sizeof(struct udppkt) == 


struct arphdr {

	unsigned short h_type;
	unsigned short h_proto;

	unsigned char h_addrlen;
	unsigned char h_protolen;

	unsigned short oper;

	unsigned char smac[ETH_ADDR_LENGTH];
	unsigned int sip;

	unsigned char dmac[ETH_ADDR_LENGTH];
	unsigned int dip;
};

struct arppkt {

	struct ethhdr eh;
	struct arphdr arp;

};


struct tcphdr {

	unsigned short sport;//源端口
	unsigned short dport;//目的端口

	unsigned int seqnum;//发方法的数据开始位置,初始值是个随机值，传的字节数量
	unsigned int acknum;//

	unsigned char hdrlen_resv; //头部长度

	unsigned char flag; 

	unsigned short window; // 1460 窗口大小

	unsigned short checksum;//校验值
	unsigned short urgent_pointer;

	unsigned int options[0];
				  

};


struct tcppkt {

	struct ethhdr eh; // 14
	struct iphdr ip;  // 20 
	struct tcphdr tcp; // 8

	unsigned char data[0];

};


typedef enum _tcp_status {

	TCP_STATUS_CLOSED,
	TCP_STATUS_LISTEN,
	TCP_STATUS_SYN_REVD,
	TCP_STATUS_SYN_SENT,
	TCP_STATUS_ESTABLISHED,
	TCP_STATUS_FIN_WAIT_1,
	TCP_STATUS_FIN_WAIT_2,
	TCP_STATUS_CLOSING,
	TCP_STATUS_TIME_WAIT,

	TCP_STATUS_CLOSE_WAIT,
	TCP_STATUS_LAST_ACK,

};


#define TCP_CWR_FLAG		0x80
#define TCP_ECE_FLAG		0x40
#define TCP_URG_FLAG		0x20
#define TCP_ACK_FLAG		0x10//确认
#define TCP_PSH_FLAG		0x08//赶紧通知应用程序处理该包
#define TCP_RST_FLAG		0x04//发送的数据不合法或者超时，服务端就会发送RST重置
#define TCP_SYN_FLAG		0x02//同步头，就是从哪个位置发送的
#define TCP_FIN_FLAG		0x01//终止位

// arp table
struct ntcb {//就是tcb

	unsigned int sip;
	unsigned int dip;
	unsigned short sport;
	unsigned short dport;

	unsigned char smac[ETH_ADDR_LENGTH];
	unsigned char dmac[ETH_ADDR_LENGTH];


	unsigned char status;
	

};



int str2mac(char *mac, char *str) {

	char *p = str;
	unsigned char value = 0x0;
	int i = 0;

	while (p != '\0') {
		
		if (*p == ':') {
			mac[i++] = value;
			value = 0x0;
		} else {
			
			unsigned char temp = *p;
			if (temp <= '9' && temp >= '0') {
				temp -= '0';
			} else if (temp <= 'f' && temp >= 'a') {
				temp -= 'a';
				temp += 10;
			} else if (temp <= 'F' && temp >= 'A') {
				temp -= 'A';
				temp += 10;
			} else {	
				break;
			}
			value <<= 4;
			value |= temp;
		}
		p ++;
	}

	mac[i] = value;

	return 0;
}



void echo_arp_pkt(struct arppkt *arp, struct arppkt *arp_rt, char *mac) {

	memcpy(arp_rt, arp, sizeof(struct arppkt));

	memcpy(arp_rt->eh.h_dst, arp->eh.h_src, ETH_ADDR_LENGTH);
	str2mac(arp_rt->eh.h_src, mac);
	arp_rt->eh.h_proto = arp->eh.h_proto;

	arp_rt->arp.h_addrlen = 6;
	arp_rt->arp.h_protolen = 4;
	arp_rt->arp.oper = htons(2);
	
	str2mac(arp_rt->arp.smac, mac);
	arp_rt->arp.sip = arp->arp.dip;
	
	memcpy(arp_rt->arp.dmac, arp->arp.smac, ETH_ADDR_LENGTH);
	arp_rt->arp.dip = arp->arp.sip;

}

//
int main() {

	struct nm_pkthdr h;
	struct nm_desc *nmr = nm_open("netmap:eth0", NULL, 0, NULL);
	if (nmr == NULL) return -1;

	struct pollfd pfd = {0};
	pfd.fd = nmr->fd;
	pfd.events = POLLIN;

	struct ntcb tcb;

	while (1) {

		int ret = poll(&pfd, 1, -1);
		if (ret < 0) continue;

		if (pfd.revents & POLLIN) {

			unsigned char *stream = nm_nextpkt(nmr, &h);

			struct ethhdr *eh = (struct ethhdr *)stream;
			if (ntohs(eh->h_proto) ==  PROTO_IP) {

				struct udppkt *udp = (struct udppkt *)stream;

				if (udp->ip.type == PROTO_UDP) { //

					int udplength = ntohs(udp->udp.length);

					udp->data[udplength-8] = '\0';

					printf("udp --> %s\n", udp->data);

				} else if (udp->ip.type == PROTO_ICMP) { //

					

				} else if (udp->ip.type == PROTO_TCP) {

					struct tcppkt *tcp = (struct tcppkt *)stream;
/*
                   //tcb五元组
					unsigned int sip = tcp->ip.sip;
					unsigned int dip = tcp->ip.dip;

					unsigned short sport = tcp->tcp.sport;
					unsigned short dport = tcp->tcp.dport;

					tcb = search_tcb();
				*/	
					if (tcb->status == TCP_STATUS_LISTEN) { //listen状态
						
						if (tcp->tcp.flag & TCP_SYN_FLAG) {//在判断其是否是SYN

							tcb->status = TCP_STATUS_SYN_REVD;//将状态置为recv

							// send syn, ack pkt
							// seqnum, ack 

							

						} 
						
					} else if (tcb->status == TCP_STATUS_SYN_REVD) {

						if (tcp->tcp.flag & TCP_ACK_FLAG) {

							tcb->status = TCP_STATUS_ESTABLISHED;

						}
						
					}

				}
				

			} else if (ntohs(eh->h_proto) ==  PROTO_ARP) {

				struct arppkt *arp = (struct arppkt *)stream;

				struct arppkt arp_rt;

				if (arp->arp.dip == inet_addr("192.168.0.123")) { //

					echo_arp_pkt(arp, &arp_rt, "00:50:56:33:1c:ca");

					nm_inject(nmr, &arp_rt, sizeof(arp_rt));

					printf("arp ret\n");
				
				}

			}

		}

	}
	
	

}