基于UDP的TFTP文件传输 + 基于UDP的网络聊天室

一、基于UDP的TFTP文件传输

1）tftp协议概述

简单文件传输协议，适用于在网络上进行文件传输的一套标准协议，使用UDP传输

特点：

1.是应用层协议

2.基于UDP协议实现

数据传输模式

octet：二进制模式（常用）

mail：已经不再支持

2）tftp下载模型

TFTP通信过程总结

1. 服务器在69号端口等待客户端的请求
2. 服务器若批准此请求，则使用 临时端口 与客户端进行通信。
3. 每个数据包的编号都有变化（从1开始）
4. 每个数据包都要得到ACK的确认，如果出现超时，则需要重新发送最后的数据包或ACK包
5. 数据长度以512Byte传输的，小于512Byte的数据意味着数据传输结束。

3）tftp协议分析

差错码：

0 未定义,差错错误信息

1 File not found.

2 Access violation.

3 Disk full or allocation exceeded.

4 illegal TFTP operation.

5 Unknown transfer ID.

6 File already exists.

7 No such user.

8 Unsupported option(s) requested.

4）读写请求原理图

5）代码+成果图

#include <myhead.h>
 
#define PORT 69              //服务器绑定的端口号
#define IP "192.168.123.174" //服务器的IP地址
 
int do_download(int cfd,struct sockaddr_in sin);
int do_upload(int cfd, struct sockaddr_in sin);
 
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(cfd ==-1)
	{
		perror("socket");
		return -1;
	}
	printf("cfd = %d\n",cfd);
 
	//绑定客户端的地址信息结构体到套接字上--->非必须绑定
	//若不绑定，则操作系统会给客户端绑定运行主机的IP和随机的端口号
 
	//填充要连接的服务器地址信息结构体，真实的地址信息结构体根据地址族制定
	//要发给谁，就填谁的地址信息
	//AF_INET : man 7 ip
	struct sockaddr_in sin;
	socklen_t addrlen=sizeof(sin);
	sin.sin_family = AF_INET;       //必须填AF_INET
	sin.sin_port = htons(PORT);   //端口号：服务器绑定的端口号
	sin.sin_addr.s_addr = inet_addr(IP); //服务器绑定的IP
 
	char abc =0;
	while(1)
	{
		printf("------------------------\n");
		printf("---------1. 下载--------\n");
		printf("---------2. 上传--------\n");
		printf("---------3. 退出--------\n");
		printf("------------------------\n");
		printf("------------------------\n");
		printf("请输入>>> ");
 
		abc = getchar();
		while(getchar() != 10);   //循环获取字符，直到遇到\n结束循环
 
		switch(abc)
		{
		case '1':
			do_download(cfd,sin);
			break;
		case '2':
			do_upload(cfd,sin);
			break;
		case '3':
			goto END;
			break;
		default:
			printf("输入错误！请重新输入\n");
 
		}
	}
END:
	//关闭文件描述符
	close(cfd);
 
	return 0;
}
 
int do_download(int cfd, struct sockaddr_in sin)
{
	//组包准备发送下载请求
	char buf[516]="";
	char name[20]="";
	printf("请输入要下载的文件名>>> ");
	scanf("%s",name);
	while(getchar()!=10);
 
	unsigned short *p1=(short*)buf;   //操作码
	*p1=htons(1);
 
	char *p2=buf+2;          //文件名
	strcpy(p2,name);
 
	char *p3=p2+strlen(p2); //第一个0 
	*p3=0;
 
	char *p4=p3+1;   //模式  
	strcpy(p4,"octet");
 
	size_t size=2+strlen(p2)+1+strlen(p4)+1; //操作码+文件名+0+模式+0
 
	//发送下载请求
	if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		perror("sendto error");
		return -1;
	}
 
	//创建下载文件并清空
	int fd = -1;   //必须初始化成一个无效的文件描述符
	socklen_t addrlen = sizeof(sin);
	ssize_t res = 0;
	unsigned short num = 0;  //记录本地的块编号
 
	//发送下载请求
	while(1)
	{
		//接收数据
		bzero(buf,sizeof(buf));
		res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(res < 0)
		{
			perror("recvfrom error");
			return -1;
		}
 
		if(3 == buf[1])  //数据包
		{
			//判断服务器返回的数据包的块编号与本地记录的块编号是否一致
			if(*(unsigned short*)(buf+2) == htons((num+1)))
			{
				num++;  //更新本地记录的块编号
				if(-1 == fd)
				{
					fd=open(name,O_WRONLY|O_CREAT|O_TRUNC,0664);
					if(fd < 0)
					{
						perror("open error");
						return -1;
					}
				}
 
				//将数据写到文件中
				if(write(fd,buf+4,res-4) < 0 )
				{
					perror("write error");
					return -1;
				}
 
				//发送ACK
				buf[1] = 4;
				//*p1=htons(4);
				if(sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
				{
					perror("sendto error");
					return -1;
				}
				//若接收到的数据小于512跳出循环，结束下载
				if(res-4 < 512)
				{
					printf("%s 文件下载完毕\n",name);
					break;
				}
			}
		}
		else if(5 == buf[1])    //错误包
		{
			printf("错误: %d %s\n",ntohs(*(short*)(buf+2)),buf+4);
			close(fd);
			return -1;
		}
	}
	close(fd);
	return 0;
}
 
 
int do_upload(int cfd, struct sockaddr_in sin)
{
	//组包准备发送上传请求
	char buf[516]="";
	char name[20]="";
	printf("请输入要上传的文件名>>> ");
	scanf("%s",name);
	while(getchar()!=10);
 
	int fd = open(name,O_RDONLY);
	if(fd < 0)
	{
		if( errno == ENOENT)
		{
			printf(">>>文件不存在，请重新输入<<<\n");
			return -2;
		}
		else
		{
			perror("open error");
			return -1;
		}
	}
	//组包准备发送上传请求
	unsigned short *p1=(short*)buf;   //操作码
	*p1=htons(2);
 
	char *p2=buf+2;          //文件名
	strcpy(p2,name);
 
	char *p3=p2+strlen(p2); //第一个0 
	*p3=0;
 
	char *p4=p3+1;   //模式  
	strcpy(p4,"octet");
 
	size_t size=2+strlen(p2)+1+strlen(p4)+1; //操作码+文件名+0+模式+0
 
	//发送上传请求
	if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		perror("sendto error");
		return -1;
	}
 
	//循环接收发送数据包
	ssize_t res;
	unsigned short num = 0;
	socklen_t addrlen = sizeof(sin);
	while(1)
	{
		//将数据从文件中读取到buf中
		bzero(buf,sizeof(buf));
		res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
		if(res < 0)
		{
			perror("recvfrom error");
			return -1;
		}
 
		//操作码的范围是1-5，因为是网络字节序
		//所以有效操作码存储在高位，即buf[1]的位置
		//printf("buf[1] = %d\n",buf[1]);   //4 服务器返回应答包
		if(4 == buf[1])  //数据包
		{
			//判断当前数据包的编号是否等于应答包的编号
			//防止数据包在传送过程丢包或重复收包
			if(num == ntohs(*(unsigned short*)(buf+2)))
			{
				//修改操作码为数据包
				buf[1] = 3;
				//填充块编号
				num++;
				*(unsigned short*)(buf+2) = htons(num);
 
				//读取数据
				res = read(fd,buf+4,sizeof(buf)-4);
				if(res < 0)
				{
					perror("read error");
					return -1;
				}
				else if(0 == res)
				{
					printf("%s 文件上传完毕!\n",name);
					break;
				}
 
				//发送数据包
				//发送的数据包大小为，读取到的字节数res+操作码2byte+块编号2bytes
				if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
				{
					perror("sendto error");
					return -1;
				}
			}
			else
			{
				printf("文件上传失败，请检查网络环境\n");
					break;
			}
		}
		else if(5 == buf[1])    //错误包
		{
			printf("错误: %d %s\n",ntohs(*(short*)(buf+2)),buf+4);
			close(fd);
			return -1;
		}
	}
	close(fd);
	return 0;
}

 

下载：

 

上传：

二、基于UDP的网络聊天室

1）项目需求：

1. 如果有用户登录，其他用户可以收到这个人的登录信息
2. 如果有人发送信息，其他用户可以收到这个人的群聊信息
3. 如果有人下线，其他用户可以收到这个人的下线信息
4. 服务器可以发送系统信息

 

2)代码部分

 服务器端代码：

#include<myhead.h>
struct sockaddr_in serveraddr, caddr;
enum type_t //枚举
{
    Login,
    Chat,
    Quit,
};
typedef struct MSG
{
    char type;    
    char name[32];  
    char text[128]; 
} msg_t;
 
typedef struct NODE //链表
{
    struct sockaddr_in caddr;
    struct NODE *next;
} node_t;
 
node_t *create_node(void) //建头节点
{
    node_t *p = (node_t *)malloc(sizeof(node_t));
    if (p == NULL)
    {
        perror("malloc err");
        return NULL;
    }
    p->next = NULL;
    return p;
}
void do_login(int, msg_t, node_t *, struct sockaddr_in); //登录的函数
void do_chat(int, msg_t, node_t *, struct sockaddr_in);  //群聊的函数
void do_quit(int, msg_t, node_t *, struct sockaddr_in);  //退出函数
int main(int argc, char const *argv[])
{
    if (argc != 3)
    {
        printf("Usage:./b.out <port>\n");
        return -1;
    }
    //创建UDP套接字
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        exit(-1);
    }
    //填充服务器网络信息结构体
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
    socklen_t len = sizeof(caddr);
    //定义保存客户端网络信息的结构体
    //绑定套接字和服务器网络信息的结构体
    bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
    printf("bind success!\n");
    msg_t msg;
    node_t *p = create_node();
    while (1)
    {
        if (recvfrom(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&caddr, &len) < 0)
        {
            perror("recvfrom err");
            return -1;
        }
        if (msg.type == Login)
        {
            strcpy(msg.text, "已上线");
            printf("ip:%s pord:%d name:%s\n", inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port), msg.name);
            printf("状态:%s\n", msg.text);
            do_login(sockfd, msg, p, caddr);
        }
        else if (msg.type == Chat)
        {
            do_chat(sockfd, msg, p, caddr);
        }
        else if (msg.type == Quit)
        {
            strcpy(msg.text, "已下线");
            printf("ip:%s pord:%d name:%s\n", inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port), msg.name);
            printf("状态:%s\n", msg.text);
            do_quit(sockfd, msg, p, caddr);
        }
    }
    close(sockfd);
    return 0;
}
//登录的函数
void do_login(int sockfd, msg_t msg, node_t *p, struct sockaddr_in caddr)
{
 
    sprintf(msg.text, "%s 已上线", msg.name);
    while (p->next != NULL)
    {
        p = p->next;
        sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(p->caddr), sizeof(p->caddr));
        //printf("%s\n",msg.text);
    }
    node_t *new = (node_t *)malloc(sizeof(node_t));
    //初始化
    new->caddr = caddr;
    new->next = NULL;
    //链接到链表尾
    p->next = new;
    return;
}
//群聊的函数
void do_chat(int sockfd, msg_t msg, node_t *p, struct sockaddr_in caddr)
{
    //遍历链表
    while (p->next != NULL)
    {
        p = p->next;
 
        if (memcmp(&(p->caddr), &caddr, sizeof(caddr)) != 0)
        {
            sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(p->caddr), sizeof(p->caddr));
        }
    }
    return;
}
//退出函数
 
void do_quit(int sockfd, msg_t msg, node_t *p, struct sockaddr_in caddr)
{
    sprintf(msg.text, "%s 已下线", msg.name);
    while (p->next != NULL)
    {
        if ((memcmp(&(p->next->caddr), &caddr, sizeof(caddr))) == 0)
        {
            node_t *dele = NULL;
            dele = p->next;
            p->next = dele->next;
            free(dele);
            dele = NULL;
        }
        else
        {
            p = p->next;
            sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(p->caddr), sizeof(p->caddr));
        }
    }
    return;
}

客户端代码：

#include<myhead.h>
enum type_t
{
    Login,
    Chat,
    Quit,
};
typedef struct
{
    char type;      
    char name[32];  
    char text[128]; 
} msg_t;
 
int main(int argc, char const *argv[])
{
    if (argc != 3)
    {
        printf("Usage ./c.out <ip> <port>\n");
        return -1;
    }
 
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        exit(-1);
    }
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
    socklen_t len = sizeof(serveraddr);
    msg_t msg;
    //先执行登录操作
    printf("请登录:\n");
    msg.type = Login;
    printf("请输入用户名:");
    fgets(msg.name, 32, stdin);
    if (msg.name[strlen(msg.name) - 1] == '\n')
        msg.name[strlen(msg.name) - 1] = '\0';
    //发送登录消息
    if (sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, len) < 0)
    {
        perror("sendto err");
        exit(-1);
    }
    pid_t pid = fork();
    if (pid < 0)
    {
        perror("fork err");
        exit(-1);
    }
    else if (pid == 0)
    {
        while (1)
        {
            if (recvfrom(sockfd, &msg, sizeof(msg), 0, NULL, NULL) < 0)
            {
                perror("recvfrom err");
                return -1;
            }
            printf("[%s]:%s\n", msg.name, msg.text);
        }
    }
    else
    {
        while (1)
        {
            fgets(msg.text, sizeof(msg.text), stdin);
            if (msg.text[strlen(msg.text) - 1] == '\n')
                msg.text[strlen(msg.text) - 1] = '\0';
            if (strcmp(msg.text, "quit") == 0)
            {
                msg.type = Quit;
                sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, len);
                kill(pid, SIGKILL);
                wait(NULL);
                exit(-1);
            }
            else
            {
                msg.type = Chat;
            }
            //发送消息
            sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, len);
        }
    }
    close(sockfd);
    return 0;
}

运行结果