#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
int rwpacksend(short rw, char *filename, int clientSocket, struct sockaddr_in sin) // 读写操作函数
{
int len = 2 + strlen(filename) + 1 + strlen("octet") + 1;
char *rwpacket = (char *)malloc(len);
*(rwpacket + 1) = rw & 0xff; // 处理操作码低位
*rwpacket = (rw & (0xff << 8)) >> 8; // 处理操作码高位
char *cur;
sprintf(rwpacket + 2, "%s%c%s%c", filename, '\0', "octet", '\0');
if (sendto(clientSocket, rwpacket, len, 0, (struct sockaddr *)&sin, sizeof(sin)) < 0)
{
perror("sendto error");
return -1;
}
// printf("发送成功:%s\n",buf);
free(rwpacket);
rwpacket = NULL;
}
int copydata(char *path, char *data, long int packlen) // 拷贝数据包操作函数
{
int fq1;
if (fq1 = (open(path, O_WRONLY | O_APPEND | O_CREAT, 0664)))
{
perror("open");
return -1;
}
write(fq1, data, packlen);
close(fq1);
return 0;
}
int recvdata(char *buf,long buflen,int receverSocket, struct sockaddr_in *cin,socklen_t *len)
{
int packlen = recvfrom(receverSocket, buf, buflen, 0, (struct sockaddr *)cin, &len);
if (packlen < 0)
{
perror("recvfrom error");
return -1;
}
return 0;
}
/***************
* 函数名:dealdata
* 功能:处理报错和接收数据
*
****************/
int dealdata(char *buf,long len,char *path,int ACKSocket, struct sockaddr_in *ackin)
{
short contrlnum = *buf|*(buf+1)<<8;
short block = *(buf+2)|*(buf+3)<<8;
char ackpack[4]="0";
ackpack[1]=*(buf+1)+1;
ackpack[3]=*(buf+3);
switch (contrlnum)
{
case 3://拷贝数据处理函数
{
copydata(path, buf+4, len);
if(sendto(ACKSocket,ackpack,sizeof(ackpack),0,(struct sockaddr*)&ackin,sizeof(ackin))<0)
{
perror("ACK error");
return -1;
}
printf("ACKsuccess ACKnum:%d\n",block);
}
break;
case 4://error处理
{
printf("%s",buf+4);
exit(-1);
}
break;
}
}
int main()
{
int sendSocket = socket(AF_INET, SOCK_DGRAM, 0);
if (sendSocket == -1)
{
perror("socket error");
return -1;
}
printf("sfd=%d__%d__\n", sendSocket, __LINE__);
// 【四】接收服务器数据
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(69);
sin.sin_addr.s_addr = inet_addr("192.168.2.51");
socklen_t addrlen = sizeof(sin);
rwpacksend(1, "5.png", sendSocket, sin); // 下载请求
// 接收文件变量
int recvSocket = socket(AF_INET, SOCK_DGRAM, 0);
if (recvSocket == -1)
{
perror("socket error");
return -1;
}
printf("sfd=%d__%d__\n", recvSocket, __LINE__);
char buf[516]="";
struct sockaddr_in cin;
socklen_t cinlen=sizeof(cin);
while (1)
{
memset(buf,0,sizeof(buf));
recvdata(buf,sizeof(buf),recvSocket,&cin,&cinlen);
dealdata(buf,sizeof(buf),"5.png",recvSocket,&cin);
/*bzero(buf,sizeof(buf));
//发送信息给服务器
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
if(sendto(clientSocket,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
perror("sendto error");
return -1;
}
printf("发送成功:%s\n",buf);
//接收数据
if(recvfrom(clientSocket,buf,sizeof(buf),0,NULL,0)<0)
{
perror("recvfrom error");
return -1;
}
printf("[%s:%d]:_%s_\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port),buf);
*/
}
close(recvSocket);
return 0;
}
udp downlond模型
最新推荐文章于 2024-07-19 15:16:23 发布