day3-网络编程

基于UDP的TFTP文件传输

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>

#define ERR_MSG(msg) do {\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
} while(0)

#define IP "192.168.10.130"
#define PORT 69

int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("报式套接字创建成功\n");

	//填充服务器的地址信息结构体
	//给下面的sendto函数使用
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//组下载请求协议
	char buf[516] = "";

	short *ptr1 = (short*)buf;
	*ptr1 = htons(1); //操作码，下载

	char *ptr2 = buf + 2;
	strcpy(ptr2, "1.png"); //要下载的文件名

	char *ptr3 = ptr2 + strlen(ptr2);
	*ptr3 = 0;

	char *ptr4 = ptr3 + 1;
	strcpy(ptr4, "octet");  //指定模式

	int size = 2 + strlen(ptr2) + 1 + strlen(ptr4) + 1;

	//发送下载请求包给服务器
	if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("操作码发送成功\n");

	//下载地址
	int fd = open("./1cpy.png",O_RDWR|O_CREAT|O_TRUNC,0664);
	if(fd < 0)
	{
		ERR_MSG("open");
		return -1;
	}

	//存储服务器发送的地址信息结构体
	struct sockaddr_in recvaddr;
	socklen_t len = sizeof(recvaddr);
	ssize_t res = 0;
	while(1)
	{
		bzero(buf,sizeof(buf));
		//接收数据包
		res = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&recvaddr, &len);
		if(res < 0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}


		//提取数据包的操作码
		short *s1 = (short*)buf; //获取数据包的操作码
		short p1  = htons(*s1);

		short *s2 = s1 + 1;      //获取数据包的块编号
		short p2 = htons(*s2);

		char *s3 = buf + 4;      //偏移到数据包中的数据段

		write(fd, s3, res-4);

		*s1 = htons(4); //4转换成网络字节序

		//发送ack给服务器
		if(sendto(cfd, buf, 4, 0, (struct sockaddr*)&recvaddr, len) < 0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		if(res < 516 )
		{
			printf("下载完成\n");
			break;
		}
	}


	if(close(fd) < 0) 
	{
		ERR_MSG("close");
		return -1;
	}

	//关闭套接字文件描述符
	if(close(cfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}