#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;
}