8.18网络编程——基于UDP的TFTP文件传输客户端

一、思维导图

二、基于UDP的TFTP文件传输

1、myhead.h

#ifndef __MYHEAD_H__
#define __MYHEAD_H__


#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>

#define ERR_MSG(msg) do{perror(msg);printf("%d\n",__LINE__);return -1;}while(0)
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <time.h>
#include <pwd.h>
#include <grp.h>
#include <sys/wait.h>
#include <pthread.h>
#include <semaphore.h>
#include <signal.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <sys/msg.h>
#endif

2、客户端代码

#include<myhead.h>

#define SER_IP "192.168.109.125"       //服务器ip地址
#define SER_PORT 69                  //服务器端口号
#define CLI_IP  "192.168.117.113"      //客户端ip地址
#define CLI_PORT 9999                  //客户端端口号
void dowload(int cfd,struct sockaddr_in sin,socklen_t addrlen)
{
	//定义协议包数组
	char msgBuf[516] = ""; 

	//提示并输入文件名
	char fileName[21] = "";           //记录文件名
	printf("请输入您要下载的文件名：");
	scanf("%s", fileName);

	//封装请求协议
	short *p1 =(short *)msgBuf;           //封装请求
	*p1 = htons(1);               //表示下载请求包

	char *p2 = msgBuf + 2;         //文件名部分
	strcpy(p2, fileName);

	char *p4 = p2 + strlen(p2) + 1;         //传输模式
	strcpy(p4, "octet");

	int msgLen = 2 + strlen(p2) + strlen(p4) +2;    //请求包的长度

	//将消息发送给服务器
	sendto(cfd, msgBuf, msgLen, 0, (struct sockaddr*)&sin, sizeof(sin));
	

	ssize_t len;
	//以只写的形式打开一个文件，文件名为  fileName
	umask(0);
	int fd=open(fileName,O_WRONLY | O_CREAT | O_TRUNC,0664);
	while(1)
	{
		//接收服务器发来的数据包
		//判断是错误包还是数据包
		len=recvfrom(cfd,msgBuf,516,0,(struct sockaddr*)&sin,&addrlen);
		//如果是数据包，读取数据包4字节之后的数据
		//如果数据包长度为512，则将消息写入文件后
		//向服务器回复一个应打包


		//如果数据包长度小于512，则接收完数据写入文件后，回复一个应打包后，就可以
		//结束循环了
		if(msgBuf[1]==3)
		{	
			write(fd,msgBuf+4,len-4);
			msgBuf[1]=4;
			sendto(cfd, msgBuf, 4, 0, (struct sockaddr*)&sin, sizeof(sin));
			if(len-4<512)
			{
				break;
			}
		}
		else if(msgBuf[1]==5)
		{
			if(msgBuf[3]==1)
			{
				remove(fileName);
			}
			printf("%s\n",msgBuf+4);
			break;
		}
	}
}
void updoading(int cfd,struct sockaddr_in sin,socklen_t addrlen)
{
	//定义协议包数组
	char msgBuf[516] = ""; 

	//提示并输入文件名
	char fileName[21] = "";           //记录文件名
	printf("请输入您要上传的文件名：");
	scanf("%s", fileName);	
	//以只读的形式打开一个文件，文件名为  fileName
	int fd=open(fileName,O_RDONLY);
	if(fd==-1)
	{
		perror("open error");
		return;
	}

	//封装请求协议
	short *p1 =(short *)msgBuf;           //封装请求
	*p1 = htons(2);               //表示上传请求包

	char *p2 = msgBuf + 2;         //文件名部分
	strcpy(p2, fileName);

	char *p4 = p2 + strlen(p2) + 1;         //传输模式
	strcpy(p4, "octet");

	int msgLen = 2 + strlen(p2) + strlen(p4) +2;    //请求包的长度

	//将消息发送给服务器
	sendto(cfd, msgBuf, msgLen, 0, (struct sockaddr*)&sin, sizeof(sin));
	
	int len;

	while(1)
	{
		//接收服务器发来的ACK
		//判断是错误包还是ACK
		recvfrom(cfd,msgBuf,4,0,(struct sockaddr*)&sin,&addrlen);
		//如果是ACK，写入数据在数据包4字节之后
		//如果数据包长度为512，则将消息写入文件后，向服务器回复一个数据包
		//如果数据包长度小于512，则发送数据包后，结束循环
		if(msgBuf[1]==4)
		{	
			len=read(fd,msgBuf+4,512);
			printf("len=%d\n",len);
			msgBuf[1]=3;
			sendto(cfd, msgBuf,len+4, 0, (struct sockaddr*)&sin, sizeof(sin));
			if(len<512)
			{
				break;
			}
		}
		else if(msgBuf[1]==5)
		{
			printf("%s\n",msgBuf+4);
			break;
		}
	}

	
}

int main(int argc, const char *argv[])
{
	//1、创建用于通信的套接字文件描述符
	int cfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(-1 == cfd)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success  cfd = %d\n", cfd);           //3
	

	//2、绑定ip地址和端口号（可选）
	//2.1 填充客户端地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family = AF_INET;                //通信域
	cin.sin_port = htons(CLI_PORT);          //客户端端口号
	cin.sin_addr.s_addr = inet_addr(CLI_IP);  //客户端ip地址
	//2.2 进行绑定工作
	if(bind(cfd, (struct sockaddr*)&cin, sizeof(cin)) == -1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind  success\n");
	

	//3、数据收发
	//封装服务器的地址信息结构体
	struct sockaddr_in sin;             //服务器地址信息结构体
	sin.sin_family = AF_INET;
	sin.sin_port = htons(SER_PORT);         //服务器端口号
	sin.sin_addr.s_addr = inet_addr(SER_IP);  //服务器ip地址
	socklen_t addrlen = sizeof(sin);          //地址信息结构体的大小
	
	int num;
	while(1)
	{
		printf("1.下载\n");
		printf("2.上传\n");
		printf("3.退出\n");
		scanf("%d",&num);
		switch(num)
		{
		case 1:
			dowload(cfd,sin,addrlen);
			break;
		case 2:
			updoading(cfd,sin,addrlen);
			break;
		case 3:

			break;
		}
		if(num==3)
			break;
	}



	//4、关闭客户端
	close(cfd);
	
	return 0;
}

三、牛客网刷题