1>TFTP客户端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#define ERR_MSG(msg) do{\
perror(msg);\
fprintf(stderr,"line : __%d__\n",__LINE__);\
}while(0)
#define PORT 69 //服务器端口号
#define IP "192.168.8.194" //服务器IP
int download(int cfd,struct sockaddr_in sin)
{
char buf[516]="";
char filename[20]="";
short* p1=(short*)buf;
*p1=htons(1); //下载的编号
char* p2=buf+2;
printf("请输入要下载的文件名>>>");
scanf("%s",filename);
while(getchar()!=10);
strcpy(p2,filename); //文件名
char* p3=p2+strlen(p2);
*p3=0; //0
char* p4=p3+1;
strcpy(p4,"octet");//模式
char* p5=p4+strlen(p4);
*p5=0; //0
int size=2+strlen(p2)+1+strlen(p4)+1;
//发送下载请求
if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0){
ERR_MSG("sendto");
return -1;
}
printf("send success\n");
int fd; //打开一个文件用于存储
int flag=0; //文件状态 0关闭 1打开
socklen_t addrlen=sizeof(sin); //定义recvfrom参数
ssize_t res=0; //接recvfrom返回值,实际接受到的数据大小
unsigned short num=0; //记录数据包编号
while(1){
bzero(buf,sizeof(buf));
//接受数据包
res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);
if(res<0){
ERR_MSG("recvfrom");
return -1;
}
if(3==buf[1]){
//UDP数据可能重复或者失序
//所以记录数据包编号,存在num中
//如果数据包块编号与本地一致则是正确的
if(htons(num+1)==*(unsigned short*)(buf+2)){
num++;
if(0==flag){
fd=open(filename,O_WRONLY|O_CREAT|O_TRUNC,0664);
if(fd<0){
ERR_MSG("open");
return -1;
}
flag=1;
}
//另存数据,write到文件中
//最后一次写入的数据小于512,所以写入的大小为(buf+4)
//实际获取到的数据包大小为(res-操作码-块编号)
if(write(fd,buf+4,res-4)<0){
ERR_MSG("write");
break;
}
//回复ACK,应答包和数据包的前4个字节只有操作码不一致
//有效操作码存在buf[1]的位置
buf[1]=4;
if(sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0){
ERR_MSG("sendto");
break;
}
//判断数据是否小于512字节
if(res-4<512){
printf("文件传输完毕\n");
break;
}
}
}else if(5==buf[1]){ //错误包
fprintf(stderr,"error=%d errmsg=%s\n",ntohs(*(short*)(buf+2)),buf+4);
break;
}
}
close(fd);
return 0;
}
int upload(int cfd,struct sockaddr_in sin)
{
char filename[20]="";
char buf[516]="";
bzero(buf,sizeof(buf));
START:
printf("请输入要上传的文件名>>>");
fgets(filename,sizeof(filename),stdin);
filename[strlen(filename)-1]=0;
if(access(filename,F_OK)!=0)
{
ERR_MSG("access");
goto START;
}
FILE* fp;
fp=fopen(filename,"r");
//发送下载请求包
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)
{
ERR_MSG("sendto");
return -1;
}
socklen_t addrlen=sizeof(sin);
ssize_t res=0;
unsigned short num=0; //记录数据包的编号
//循环发送数据包,接收ACK
while(1)
{
bzero(buf,sizeof(buf));
res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr *)&sin,&addrlen);
if(res<0){
ERR_MSG("recvfrom");
return -1;
}
if(4==buf[1]){
//构造数据包
*((unsigned short*)buf)=htons(3);
*((unsigned short*)(buf+2))=htons(num+1);
res=fread(buf+4,1,512,fp);
if(res!=512){
if(ferror(fp)){
printf("读取错误\n");
return -1;
}else if(feof(fp)){
printf("文件读取完毕\n");
}
}
//发送数据包
if(res==sendto(cfd,buf,res+4,0,(struct sockaddr*)&sin,sizeof(sin))<0){
perror("sendto");
fclose(fp);
return -1;
}
if(res<512){
printf("send success\n");
break;
}
num++;
}else{
if(num==0){
if(sendto(cfd, buf, size, 0, (struct sockaddr *)&sin, sizeof(sin)) < 0){
ERR_MSG("sendto");
return -1;
}
}else{
if(res==sendto(cfd,buf,res+4,0,(struct sockaddr*)&sin,sizeof(sin))<0){
ERR_MSG("sendto");
fclose(fp);
return -1;
}
}
}
}
fclose(fp);
return 0;
}
int main(int argc,const char *argv[])
{
//创建套接字
int cfd = socket(AF_INET,SOCK_DGRAM,0);
if(cfd < 0){
ERR_MSG("socket");
return -1;
}
//存放服务器地址信息
struct sockaddr_in sin;
sin.sin_family=AF_INET;
sin.sin_port=htons(PORT);
sin.sin_addr.s_addr =inet_addr(IP);
char c=0;
while(1){
puts("---------------TFTP client-------------------");
puts("---------------------------------------------");
puts("---------------1.download--------------------");
puts("---------------2.upload----------------------");
puts("---------------3.exit------------------------");
puts("---------------------------------------------");
puts("============ select your option =============\n");
printf("please input>>>");
c=getchar();
while(getchar()!=10);
switch(c){
case '1':
download(cfd,sin);
break;
case '2':
upload(cfd,sin);
break;
case '3':
goto END;
default:
printf("输入错误请重新输入\n");
}
}
//关闭文件描述符
END:
close(cfd);
return 0;
}