网络编程-tpfp的上传与下载

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>  
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <sys/stat.h>
#include <fcntl.h>

#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define IP "192.168.10.116"
#define PORT 69
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\t__%d__\n",__LINE__);

	//绑定服务器的IP和端口
	struct sockaddr_in sin;
	sin.sin_family  	=AF_INET;
	sin.sin_port 		=htons(PORT);
	sin.sin_addr.s_addr =inet_addr(IP);

	char buf[516]="";
	int flag;
	printf("请选择下载1或上传2：");
	scanf("%d",&flag);
	if(flag==1)
	{
		short *ptr1=(short *)buf;
		*ptr1=htons(1);             //操作码，下载

		char* ptr2=buf+2;
		//strcpy(ptr2,"1_udpSer.c");//要下载的文件名
		strcpy(ptr2,"5.png");//要下载的文件名

		char *ptr3=ptr2+strlen(ptr2);
		*ptr3=0;

		char *ptr4=ptr3+1;
		strcpy(ptr4,"octet");       //模式

		int size=2+strlen(ptr2)+1+strlen(ptr4)+1;

		printf(">>>>request download<<<<\n");
		//发送下载请求包给服务器
		if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		printf(">>>>send success<<<<\n");

		//接收数据包
		socklen_t len=sizeof(sin);
		ssize_t res=0;
		int fd;
		short num=1;
		//fd=open("./receive.c",O_WRONLY|O_CREAT|O_TRUNC,0664);
		fd=open("./receive.png",O_WRONLY|O_CREAT|O_TRUNC,0664);
		//循环接收数据包
		while(1)
		{
			bzero(buf,sizeof(buf));
			res=recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&len);
			if(res<0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			if(3==buf[1])
			{
				if(fd<0)
				{
					ERR_MSG("open");
					return -1;
				}
				if(write(fd,buf+4,res-4)<0)
				{
					ERR_MSG("write");
					return -1;
				}

				buf[1]=4;
				if(sendto(cfd,buf,4,0,(struct sockaddr*)&sin,len)<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
				if(res<516)
				{
					printf("download success\n");
					break;
				}
				num++;
			}

			else if(5==buf[1])
			{
				write(stdout->_fileno,buf+2,1);
				write(stdout->_fileno,buf+4,res-4);
			}

		}

		//关闭套接字文件描述符
		if(close(cfd)<0)
		{
			ERR_MSG("close");
			return -1;
		}
	}
	else if(flag==2)
	{
		short *ptr1=(short *)buf;
		*ptr1=htons(2);             //操作码

		char* ptr2=buf+2;
		strcpy(ptr2,"receive.png");

		char *ptr3=ptr2+strlen(ptr2);
		*ptr3=0;

		char *ptr4=ptr3+1;
		strcpy(ptr4,"octet");       //模式

		int size=2+strlen(ptr2)+1+strlen(ptr4)+1;

		printf(">>>>request upload<<<<\n");
		//发送上传请求包给服务器
		if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin))<0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		printf(">>>>send success<<<<\n");

		//接收数据包
		ssize_t res=0;
		int fd;
		short num=0;
		fd=open("./receive.png",O_RDONLY);
		struct sockaddr recvaddr;
		socklen_t len=sizeof(recvaddr);
		//循环发送数据包
		while(1)
		{
			bzero(buf, sizeof(buf));
			res=recvfrom(cfd, buf, sizeof(buf),0,(struct sockaddr*)&recvaddr,&len);
			if(buf[1]==4)
			{
				bzero(buf,sizeof(buf));
				res = read(fd,buf+4,sizeof(buf)-4);
				buf[1]=3;
				buf[3]=num;
				/**(short*)buf = htons(3);
				*(short*)(buf+2) = htons(++num);*/
				sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&recvaddr,len);
				printf("loading\n");
				if(res<512)
				{
					break;
				}
				num++;
			}

		}
	}

	//关闭套接字文件描述符
	if(close(cfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}