3.15网络编程

本文主要是介绍3.15网络编程，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

基于UDP的TFTP文件传输

#include<myhead.h>
#define SER_IP "192.168.117.36"
#define SER_PORT 69//客户端
void menu()
{printf("*************************\n");printf("*\t 1.下载  \t*\n");printf("*\t 2.上传  \t*\n");printf("*\t 3.退出  \t*\n");printf("*************************\n");
}
int rev()
{//下载//1、创建用于通信的套接字文件描述符int cfd = socket(AF_INET, SOCK_DGRAM, 0);if(cfd == -1){perror("socket error");return -1;}printf("cfd = %d\n", cfd);          ////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地址//3、收发数据//校验收到的块编号char filename[128]="";printf("请输入要下载的文件名>>>");scanf("%s",filename);//下载请求char buf[516] = "";int size = sprintf(buf, "%c%c%s%c%s%c", 0, 1, filename, 0, "octet", 0);if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0){perror("sendto error");return -1;}int flag = 0;int fd ;//循环接收发送应答包ssize_t recv_len;unsigned short num = 1;socklen_t addrlen = sizeof(sin);while(1){bzero(buf, 516);recv_len = recvfrom(cfd, buf, 516, 0, (struct sockaddr*)&sin, &addrlen);if(recv_len < 0){perror("recvfrom error");return -1;}if(3 == buf[1]) //如果是数据包{if(0 == flag) //防止文件重复打开{//存储从服务器下载下来的文件,接收数据成功后再创建数据；//创建并打开filename文件fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0664);if(fd < 0){perror("open error");return -1;}flag = 1;}//判断当前的快编号，是否与我期望的快编号一致if(htons(num) == *(unsigned short*)(buf+2)) 	//防止数据包重复到达{//提取数据，写入到文件中if(write(fd, buf+4, recv_len-4) < 0){printf("fd:%d recv_len=%ld\n", fd, recv_len);perror("write error");break;}//回复ack包//由于数据包前四个字节与ack包除了操作码不一致外，其余全部一致。//所以我们直接将数据包的操作码修改成4即可，发送数据包的前四个字节。buf[1] = 4; 	if(sendto(cfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0){perror("sendto error");}//判断数据包的大小是否小于 512+2+2;if(recv_len < 512+2+2){printf("-----文件下载完毕!!-----\n");break;}num++;}}else if(5 == buf[1]) 	//错误包{//打印错误信息printf("-----ERROR:%s-----\n", buf+4);break;}}//5、关闭套接字close(cfd);
}int snd()
{//上传//1、创建用于通信的套接字文件描述符int cfd = socket(AF_INET, SOCK_DGRAM, 0);if(cfd == -1){perror("socket error");return -1;}printf("cfd = %d\n", cfd);          //3//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地址char filename[128] = "";printf("请输入要上传的文件名>>>");scanf("%s",filename);//判断该文件是否存在int fd = open(filename, O_RDONLY);if(fd < 0){perror("open error");return -1;}//发送上传请求//上传协议char buf[516] = "";int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, "octet", 0);if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0){perror("sendto error");return -1;}//循环接收发送数据包int recv_len;unsigned short num = 0;socklen_t addrlen = sizeof(sin);while(1){bzero(buf, 516);recv_len = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);if(recv_len < 0){perror("recvfrom error");return -1;}//操作码的范围是1-5,因为是网络字节序if(4 == buf[1]){//判断当前数据包的编号是否等于应答包的编号if(num == ntohs(*(unsigned short*)(buf+2))){//修改操作码为数据包buf[1] = 3;//填充块编号num++;*(unsigned short*)(buf+2) = htons(num);//读取数据//发送数据int res = read(fd, buf+4, 516-4);if(res < 0){perror("read error");return -1;}else if(0 == res){printf("-----文件上传完毕-----\n");break;}//发送数据包//发送的数据包大小为,读取到的字节数(res)+操作码(2byte)+快编号(2byte)if(sendto(cfd, buf, res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) <0){perror("sendto error");return -1;}}else{printf("-----文件上传失败,请检查网络环境-----\n");break;}}else if(5 == buf[1]){printf("-----ERROR:%s-----\n", buf+4);break;}}//5、关闭套接字close(cfd);
}int main(int argc, const char *argv[])
{int choice = 0;while (1){menu();printf("请输入操作:");scanf("%d",&choice);switch (choice){case 1: rev();break;case 2: snd();break;case 3: exit(0);break;default:printf("输入错误,请重新输入\n");break;}getchar();getchar();system("clear");}return 0;
}