基于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;
}