#include <myhead.h>
#define ERR_LOG(msg)do{perror(msg);printf("%d %s %s\n",__LINE__,__func__,__FILE__);}while(0)
//定义TFTP默认端口号（69）和数据包大小（516字节）
#define PORT    69
#define N     516
int main(int argc, const char *argv[])
{
//检查命令行参数，确保提供了服务器的IP地址
if(argc<2)
{
printf("请输入IP\n");
return -1;
}
//创建UDP套接字文件描述符
int sfd=socket(AF_INET,SOCK_DGRAM,0);
if(sfd<0)
{
perror("socket:");
return -1;
}
//填充服务器地址结构
struct sockaddr_in sin;
sin.sin_family=AF_INET;
sin.sin_addr.s_addr=inet_addr(argv[1]);//IP地址转换为网络字节序
sin.sin_port=htons(PORT);               //端口号转换为网络字节序
//主循环
char choose;//选择的容器变量
while(1)
{
system("clear");//清屏
//显示功能菜单
printf("************菜单************\n");
printf("**********1:下载**********\n");
printf("**********2:上传**********\n");
printf("**********3:退出**********\n");
printf("请输入您要选择的功能：");
//用户选择
scanf("%c",&choose);
//清空输入缓冲区
while(getchar()!='\n');
//根据用户选择执行相应的功能
switch(choose)
{
case '1':
//下载自定义函数
xia(sfd,sin);
break;
case '2':
//上传自定义函数
chuan(sfd,sin);
break;
case '3':
goto END;
break;
default:
printf("输入错误\n");
}
printf("请输入任意键清屏");
while(getchar()!='\n');
}
END:
close(sfd);
return 0;
}
//下载函数
int xia(int sfd,struct sockaddr_in sin)
{
//获取用户要下载的文件名
char name[20]="";
printf("请输入要下载的文件名:");
fgets(name,20,stdin);//从终端输入20字节写入到name里
name[strlen(name)-1]=0;//去掉换行符
//发送下载请求
char buf[N]="";//定义一个请求包大小的储存容器
//构建TFTP读请求包(操作码1)
int size=sprintf(buf,"%c%c%s%c%s%c",0,1,name,0,"octet",0);
//%c%c：前两个字节为操作码，这里传入0, 1表示 "读请求"（下载操作）
//%s：紧跟操作码的是文件名（变量filename）
//%c：文件名后用空字符0作为分隔符
//%s：接下来是传输模式字符串 "octet"（二进制模式）
//最后的%c：模式字符串后也用空字符0结尾
//发送请求到服务器
if(sendto(sfd,buf,size,0,(struct sockaddr *)&sin,sizeof(sin))<0)
{
ERR_LOG("sendto");
return -1;
}
//循环接收数据并处理
int flag=0;//标记文件是否已创建
int fd;//本地文件描述符(用于保存下载的文件)
ssize_t recv_len;       //接收的字节字数
unsigned short num =1;  //期望接收的块编号(从1开始)
socklen_t addrlen =sizeof(sin);
while(1)
{
bzero(buf,N);//清空缓冲区
//接收服务器的相应
recv_len=recvfrom(sfd,buf,N,0,(struct sockaddr *)&sin,&addrlen);
if(recv_len<0)
{
ERR_LOG("recv_len:");
return -1;
}
//处理服务器发送的数据包(操作码3)
if(3==buf[1])//操作码3表示[数据块]
{
if(0==flag)//首次接收数据，就创建本地文件
{
fd=open(name,O_WRONLY|O_CREAT|O_TRUNC,0664);
if(fd<0)
{
ERR_LOG("open:");
return -1;
}
flag=1;//标记文件已创建
}
//验证数据块编号(防丢包/重复)
//检查接收的块编号是否与期望的一致
if(htons(num)==*(unsigned short *)(buf+2))
{
//将数据块中的内容跳过头部写入本地文件
if(write(fd,buf+4,recv_len-4)<0)
{
ERR_LOG("write:");
break;
}
buf[1]=4;//将操作吗改成4(确认包)
if(sendto(sfd,buf,4,0,(struct sockaddr *)&sin,sizeof(sin))<0)
{
ERR_LOG("sendto");
}
//如果数据小于516字节呢就说明是最后一个
if(recv_len<516)
{
printf("下载完毕");
break;
}
num++;//准备接收下一个块
}
}else if(5==buf[1])//收到错误包
{
printf("ERROR:%s",buf+4);//打印错误信息
break;
}
}
return 0
}
int chuan(int sfd, struct sockaddr_in sin)
{
char filename[20] = "";
printf("请输入要上传的文件名>>>");
fgets(filename, 20, stdin);
filename[strlen(filename)-1] = 0;
//判断该文件是否存在
int fd = open(filename, O_RDONLY);
if(fd < 0)
{
if(errno == ENOENT)
{
printf(">>>文件不存在,请重新输入<<<\n");
return -2;
}
else
{
ERR_LOG("open");
return -1;
}
}
//发送上传请求
//上传协议
char buf[N] = "";
int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, name, 0, "octet", 0);
if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
{
ERR_LOG("sendto");
return -1;
}
//循环接收发送数据包
int recv_len;
unsigned short num = 0;
socklen_t addrlen = sizeof(sin);
while(1)
{
bzero(buf, N);
recv_len = recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
if(recv_len < 0)
{
ERR_LOG("recvfrom");
return -1;
}

        //操作码的范围是1-5,因为是网络字节序
//所以有效操作吗存储在高位,即buf[1]的位置.
//printf("buf[1] = %d\n", buf[1]);     //4 服务器返回应答包
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, N-4);
if(res < 0)
{
ERR_LOG("read");
return -1;
}
else if(0 == res)
{
printf("-----文件上传完毕-----\n");
break;
}
//发送数据包
//发送的数据包大小为,读取到的字节数(res)+操作码(2byte)+快编号(2byte)
if(sendto(sfd, buf, res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) <0)
{
ERR_LOG("sendto");
return -1;
}
}
else
{
printf("-----文件上传失败,请检查网络环境-----\n");
break;
}
}
else if(5 == buf[1])
{
printf("-----ERROR:%s-----\n", buf+4);
break;
}
}
return 0;
}

#include <myhead.h>
#define SER_IP "192.168.108.124"//服务器IP地址
#define SER_PORT 8888//服务器端口号
#define CLT_IP "192.168.24.128"//客户端IP地址
#define CLT_PORT 7777//客户端端口号
int main(int argc, const char *argv[])
{
//创建用于连接的套接字文件描述符
int cfd=socket(AF_INET,SOCK_STREAM,0);
if(cfd==-1)
{
perror("scoket error:");
return -1;
}
//打开键盘驱动文件
int fd=open("/dev/input/event1",O_RDONLY);
if(fd==-1)
{
perror("open error:");
return -1;
}
//定义容器接收数据
struct input_event ie;
//绑定IP地址和端口号
//填充地址信息结构体
struct sockaddr_in cin;
cin.sin_family=AF_INET;
cin.sin_addr.s_addr=inet_addr(CLT_IP);
cin.sin_port=htons(CLT_PORT);
//绑定
if(bind(cfd,(struct sockaddr *)&cin,sizeof(cin))==-1)
{
perror("bind error:");
return -1;
}
//连接到服务器
//填充服务器地址信息结构体
struct sockaddr_in sin;
sin.sin_family=AF_INET;
sin.sin_addr.s_addr=inet_addr(SER_IP);
sin.sin_port=htons(SER_PORT);
//连接服务器
if(connect(cfd,(struct sockaddr *)&sin,sizeof(sin))==-1);
{
perror("connect error:");
return -1;
}
printf("连接成功");

    char rbuf[5]={0xff,0x02,0x00,0x00,0xff};
unsigned char bbuf[5]={0xff,0x02,0x01,0x00,0xff};
//发送给服务器，初始化机械臂
send(cfd,rbuf,sizeof(rbuf),0);
sleep(1);
send(cfd,bbuf,sizeof(bbuf),0);
while(1)
{
//从文件中读取数据
read(fd,&ie,sizeof(ie));
//对输入的数据判断
switch(ie.code*ie.value)
{
case 17://W
{
bbuf[3]+=2;
if(bbuf[3]>180)
{
bbuf[3]=180;
}
//将更新过后的数据发送给服务器
send(cfd,bbuf,sizeof(bbuf),0);
}
break;
case 31://S
{
bbuf[3]-=2;
if(bbuf[3]<0)
{
bbuf[3]=0;
}
//将更新过后的数据发送给服务器
send(cfd,bbuf,sizeof(bbuf),0);
}
break;
case 30://A
{
rbuf[3]+=2;               
if(rbuf[3]>90)
{
rbuf[3]=90;
}   
//将更新过后的数据发送给服务器
send(cfd,rbuf,sizeof(rbuf),0);
}     
break;
case 32://D
{
rbuf[3]-=2;               
if(rbuf[3]<0)
{
rbuf[3]=0;
}   
//将更新过后的数据发送给服务器
send(cfd,rbuf,sizeof(rbuf),0);
}     
break;
}
}
close(cfd);
return 0;
}