1、tftp客户端的上传和下载
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
int do_download(int sfd,struct sockaddr_in sin);
int do_upload(int sfd,struct sockaddr_in sin);
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0)
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd=socket(AF_INET,SOCK_DGRAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
//绑定ip和端口号--->非必须绑定
//因为要发信息给结构体,所以要填充服务器的地址信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必须填AF_INET
sin.sin_port = htons(69);//服务器端口号69
sin.sin_addr.s_addr = inet_addr("192.168.1.105");
//将点分十进制ip转化为网络字节序
char c=0;
//菜单
while(1)
{
printf("------------------------------------\n");
printf("--------------1.上传----------------\n");
printf("--------------2.下载----------------\n");
printf("--------------3.退出----------------\n");
printf("请输入>>>");
c=getchar();
while(getchar()!=10);
switch(c)
{
case '1':
do_upload(sfd,sin);
break;
case '2':
do_download(sfd,sin);
break;
case '3':
goto END;
break;
default:
printf("输入错误\n");
}
}
//关闭文件描述符
END:
close(sfd);
return 0;
}
int do_download(int sfd,struct sockaddr_in sin)
{
//组下载请求,发送给服务器,服务器在69号端口等待请求
char buf[520]="";
//操作码
unsigned short *pa=(unsigned short*)buf;
*pa=htons(1);
//文件名
printf("请输入文件名>>");
char name[32]="";
scanf("%s",name);
char *pb=buf+2;
strcpy(pb,name);
//0
char *pc=pb+strlen(pb);
//模式
char *pd=pc+1;
strcpy(pd,"octet");
//0
int size=2+strlen(pb)+1+strlen(pd)+1;
//客户端发送下载请求
if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("下载请求成功\n");
//打开文件
int fd=open("./1.c",O_RDWR|O_CREAT|O_TRUNC,0777);
if(fd<0)
{
ERR_MSG("open");
return -1;
}
printf("文件打开成功\n");
socklen_t addrlen=sizeof(sin);
ssize_t res=0;
unsigned short num=0;
while(1)
{
//客户端接收来自服务器的数据包
bzero(buf,sizeof(buf));
res=recvfrom(sfd,buf,516,0,(struct sockaddr*)&sin,&addrlen);
if(res<0)
{
ERR_MSG("recvfrom");
return -1;
}
// printf("%d %d\n",buf[0],buf[1]);
// printf("%d %d\n",*(unsigned short*)buf,ntohs(*(unsigned short*)buf));
// 由于发送回来的操作码是网络字节序,因此只需要判断buf[1]是3还是5即可
if(buf[1]==3)
{
//由于UDP传输 可能丢包或者重复,因此每次收到数据包时都要判断是不是我需要的
if(num+1==ntohs(*(unsigned short*)(buf+2)))
{
/* //将数据输出到终端
printf("%s\n",buf+4);
fflush(stdout);*/
//将下载的数据写到文件中
ssize_t reg=write(fd,buf+4,res-4);
if(reg<0)
{
ERR_MSG("write");
return -1;
}
//客户端回复ACK给服务器
buf[1]=4;
if(sendto(sfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
return -1;
}
if(res-2-2<512)
{
printf("下载完成\n");
break;
}
num++;
}
}
else if(buf[1]==5)
{
//错误包
printf("error=%s\n",buf+4);
return -2;
}
}
close(fd);
return 0;
}
//上传
int do_upload(int sfd,struct sockaddr_in sin)
{
//组上传请求,发送给服务器,服务器在69号端口等待请求
char buf[520]="";
//操作码
unsigned short *pa=(unsigned short*)buf;
*pa=htons(2);
//文件名
printf("请输入文件名>>");
char name[32]="";
scanf("%s",name);
char *pb=buf+2;
strcpy(pb,name);
//0
char *pc=pb+strlen(pb);
//模式
char *pd=pc+1;
strcpy(pd,"octet");
//0
int size=2+strlen(pb)+1+strlen(pd)+1;
socklen_t addrlen=sizeof(sin);
ssize_t ret=0;
unsigned short num=0;
//客户端发送上传请求
if(sendto(sfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("上传请求发送成功\n");
//打开文件
int fd=open(name,O_RDONLY);
if(fd<0)
{
ERR_MSG("open");
return -1;
}
printf("open success\n");
char ack[4];
unsigned short*ack0=(unsigned short*)ack;
unsigned short *ack1=ack0+1;
int acklen=sizeof(ack);
bzero(buf,sizeof(buf));
while(1)
{
//接收服务器回复包
if(recvfrom(sfd,ack,4,0,(struct sockaddr*)&sin,&acklen)<0)
{
ERR_MSG("recvfrom");
return -1;
}
printf("recvfrom succ\n");
if(*ack0==htons(4))
{
//组数据包
//操作码
unsigned short *p1=(unsigned short *)buf;
*p1=htons(3);
//块编号
unsigned short *p2=(unsigned short*)(buf+2);
//数据
char *p3=buf+4;
//读取文件数据
ret=read(fd,p3,512);
if(ret<0)
{
ERR_MSG("read");
return -1;
}
if(sendto(sfd,buf,ret+4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
}
if(ret<512)
{
printf("上传成功\n");
break;
}
bzero(buf,sizeof(buf));
}
}
close(fd);
return 0;
}
ubuntu@ubuntu:02_1020$ gcc 05_tftpcli.c
ubuntu@ubuntu:02_1020$ ./a.out
------------------------------------
--------------1.上传----------------
--------------2.下载----------------
--------------3.退出----------------
请输入>>>2
请输入文件名>>5.png
下载请求成功
文件打开成功
下载完成
请输入>>>3
ubuntu@ubuntu:02_1020$ eog 1.c
ubuntu@ubuntu:02_1020$ gcc 05_tftpcli.c
ubuntu@ubuntu:02_1020$ ./a.out
------------------------------------
--------------1.上传----------------
--------------2.下载----------------
--------------3.退出----------------
请输入>>>1
请输入文件名>>2.c
上传请求发送成功
open success
recvfrom succ
上传成功
请输入>>>3
ubuntu@ubuntu:02_1020$
2、广播
//广播的接收
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <netinet/in.h>
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd=socket(AF_INET,SOCK_DGRAM,0);
if(sfd<0)
{
perror("socket");
return -1;
}
//填充IP和端口
struct sockaddr_in sin;
sin.sin_family=AF_INET;//必填
sin.sin_port=htons(2212);//端口1024-49151
sin.sin_addr.s_addr=inet_addr("192.168.52.255");
//绑定IP和端口
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
perror("bind");
return -1;
}
//存储由谁发送数据包的IP和端口
struct sockaddr_in cin;
socklen_t addrlen=sizeof(cin);
char buf[128]="";
ssize_t ret=0;
while(1)
{
bzero(buf,sizeof(buf));
//接收数据
ret=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&addrlen);
if(ret<0)
{
perror("recvfrom");
return -1;
}
printf("[%s:%d]%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),buf);
}
close(sfd);
return 0;
}
//广播的发送
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <netinet/in.h>
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd=socket(AF_INET,SOCK_DGRAM,0);
if(sfd<0)
{
perror("socket");
return -1;
}
//设置允许广播
int set_board=1;
if(setsockopt(sfd,SOL_SOCKET,SO_BROADCAST,&set_board,sizeof(set_board))<0)
{
perror("setsockopt");
return -1;
}
//填充接收端的IP和端口
struct sockaddr_in sin;
sin.sin_family=AF_INET;//必填
sin.sin_port=htons(2212);//端口1024-49151
sin.sin_addr.s_addr=inet_addr("192.168.52.255");
/* //绑定IP和端口
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
perror("bind");
return -1;
}
*/
char buf[128]="";
ssize_t ret=0;
int i=0;
while(1)
{
bzero(buf,sizeof(buf));
printf("please input:");
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
//发送数据
ret=sendto(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin));
if(ret<0)
{
perror("recvfrom");
return -1;
}
printf("send success\n");
}
close(sfd);
return 0;
}