网络编程基础

#include <myhead.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)

#define SER_PORT 69     //指定要将数据发送给谁,填谁到端口号
#define SER_IP "192.168.1.105" 

int do_download(int cfd,struct sockaddr_in sin);
int main(int argc, const char *argv[])
{
	//创建报式套接字 socket
	int cfd =socket(AF_INET,SOCK_DGRAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success cfd=%d\n",cfd);

	//填充服务器的地址信息结构体,给sendto函数使用
	struct sockaddr_in sin;
	sin.sin_family     =AF_INET;
	sin.sin_port       =htons(SER_PORT);
	sin.sin_addr.s_addr=inet_addr(SER_IP);

	char choose=0;
	while(1)
	{
		system("clear");
		printf("************************\n");
		printf("******1.下载************\n");
		printf("******2.上传************\n");
		printf("******3,退出************\n");
		printf("************************\n");

		printf("请输入>>>");
		scanf("%c",&choose);
		while(getchar()!=10);

		switch(choose)
		{
		case '1':
			do_download(cfd,sin);
			break;
		case '2':
			//do_upload();
			break;
		case '3':
			goto END;
			break;
		default:
			printf("输入错误,请重新输入\n");
			break;
		}		}
	printf("请输入任意字符清");
	while(getchar()!=10);


#if 0
	+-- 35 lines: char buf[128]="";
#endif

END:
	//关闭文件描述符
	close(cfd);
	return 0;
}
int do_download(int cfd,struct sockaddr_in sin)
{
	char filename[20]="";
	printf("请输入要下载的文件名>>");
	scanf("%s",filename);
	while(getchar()!=10);

	//组下载请求包
	char buf[516]="";
	unsigned short *ptr1=(unsigned short*)buf;
	*ptr1=htons(1);    //组操作码

	char *ptr2=buf+2;
	strcpy(ptr2,filename); //组名字

	char *ptr3=ptr2+strlen(ptr2);//第一个0

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

	int size =2+strlen(ptr2)+1+strlen(ptr4)+1; // 组操作码+文件名+0+模式+0

	//创建下载文件并清空
	int fd=-1;   //必须初始化一个无效的文件描述符
	socklen_t addrlen =sizeof(sin);
	ssize_t res=0;	
	unsigned short num= 0;
	//发送下载请求
	if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sento");
		return -1;
	}
	printf("sento download  success\n");


	while(1)
	{

		bzero(buf,sizeof(buf));
		//接收数据
		if(recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen)<0)
		{
			ERR_MSG("recfrom");
			return -1;
		}
		printf("recfrom recvfrom success\n");


		if(3==fd)   //数据包
		{
			//判断服务器返回的数据包到块编号和本地记录到是否一致
			if(*(unsigned short*)(buf+2)==htons((num+1)))
			{
				num++;
				if(-1==fd)
				{
					fd=open(filename,O_WRONLY|O_CREAT,0664);
					if(fd<0)
					{
						ERR_MSG("open");
						return -1;
					}
				}
				//将数据写到文件中
				if(write(fd,buf+4,res-4)<0)
				{
					ERR_MSG("write");
					return -1;
				}
				//发送ACK
				buf[1]=4;
				if(sendto(cfd,buf,4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
					ERR_MSG("sendto");
					return -1;
				}
				if(res-4<512)
				{
					printf("%s 文件下载完毕\n",filename);
					break;
				}


			}

		}
		else if(5==buf[1])  //错误包
		{
			printf("错误:%d %s\n",ntohs(*(unsigned short*)(buf+2)),buf+4);
			close(fd);
			return -1;
		}
	}
	//发送ACK   sendto
	//判断数据是否小于512个字节,若小于则下载完成
	close(fd);
	return 0;

}