网络编程-day2

实现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.8.90"
#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]="";
	
	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])
		{
			printf("high cjj\n");
			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;
	}
	return 0;
}