网络编程0608

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <pthread.h>
#include <semaphore.h>
#include <signal.h>
#include <sys/wait.h>
#include <sys/shm.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <sys/msg.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define ERR_MSG(msg) do{\
		fprintf(stderr,"line;%d;%s %s",__LINE__,__FILE__,__func__);\
		perror(msg);\
}while(0)
#define IP "192.168.2.84"
#define PORT 69
int xia_zai(int cfd,struct sockaddr_in sin)
{
	char name[20]="";
	char buf[516]={0};
	printf("请输入要下载的文件名>");
	scanf("%s",name);
	while(getchar()!=10);
	//printf("%s",name);
//===============================
	short*p1=(short*)buf;
	*p1 = htons(1);
//==============================
	char *p2 = buf+2;
	strcpy(p2,name);
//=============================
	char *p4 = p2+strlen(p2)+1;
	strcpy(p4,"octet");
//===============================
int len = 4+strlen(p2)+strlen(p4);
printf( "%c%c%s%c%s%c\n", 0, 1, name,0, "octet", 0);
if(sendto(cfd, buf, len, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
{
	ERR_MSG("sendto");
	return -1;
}
printf("发送成功\n");

//变量最好初始化，且初始化为无意义的数据
int fd = -1;
socklen_t addrlen = sizeof(sin);
ssize_t res = 0;
int ret = 0;
unsigned short num = 0;
while (1)
{
	bzero(buf, sizeof(buf));
	//接收数据包
	res = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
	printf("%d\n",buf[1]);
	if (res < 0)
	{
		ERR_MSG("recvfrom");
		ret = -1;
		break;
	}
	//下列测试可以得到结果，有效操作码存储在buf[1]的位置，buf[0]的位置是0
	//printf("操作码：%#x | %d %d\n", ntohs(*(short*)buf), buf[0], buf[1]);
	//printf("    快编号：%d | %d %d\n", ntohs(*(short*)(buf+2)), buf[2], buf[3]);

	if (3 == buf[1])     //数据包
	{
		//判断数据包中的块编号是否是正确的块编号
		//需要与本地记录的块编号进行对比
		if (*(unsigned short*)(buf + 2) == htons((num + 1)))
		{
			num++;
			if (1 == ntohs(*(unsigned short*)(buf + 2)))
			{
				fd = open(name, O_WRONLY | O_CREAT | O_TRUNC, 0664);
				if (fd < 0)
				{
					ERR_MSG("open");
					return -1;
				}
			}

			//将接收到的数据另存
			if (write(fd, buf + 4, res - 4) < 0)
			{
				ERR_MSG("write");
				ret = -1;
				break;
			}


			//回复ACK，由于ACK包和数据包前四个字节只有操作码不一致
			//所以可以直接修改数据包的操作码，然后将前4个字节回复给服务器即可
			buf[1] = 4;
			if (sendto(cfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				ret = -1;
				break;
			}

			//直到数据小于512个字节，实际获取到的数据包大小为res
			//数据大小= res - 操作码 - 块编号
			if (res - 4 < 512)
			{
				printf("%s 文件拷贝完毕\n", name);
				break;
			}
		}
	}
	else if (5 == buf[1])    //错误包
	{
		fprintf(stderr, "DOWNLOAD_ERROR:%d : %s\n", ntohs(*(short*)(buf + 2)), buf + 4);
		break;
	}
}

close(fd);
return ret;



}
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int cfd = socket(AF_INET,SOCK_DGRAM,0);
	if(cfd < 0)
	{		
		ERR_MSG("socket");
		return -1;
	}
	printf("create soccess sfd=%d __%d++\n",cfd,__LINE__);
	//填充服务器的地址信息结构体，给bind函数绑定
	
	//真实的地址信息结构体柑橘地址族定制
	struct sockaddr_in sin;
	
		sin.sin_family  =AF_INET;//必须填AF_TNET;
		sin.sin_port    = htons(PORT);//1024-49151,网络字节序
		sin.sin_addr.s_addr = inet_addr(IP);//ifconfig出来本机ip
		

		char c=0;
		while(1)
		{
			printf("---------------------------\n");
			printf("----------1. 下载----------\n");
			printf("----------2. 上传----------\n");
			printf("----------3. 退出----------\n");
			printf("---------------------------\n");	
			c = getchar();
			while(getchar() != 10);
			switch(c)
			{
			case '1':
				xia_zai(cfd, sin);
				break;
			case '2':
				
				break;
		//	case '3':
		//		goto END;
		//
		//break;
			default:
				printf("输入错误，请重新输入\n");
			}

			printf("输入任意字符清屏>>>");
			while(getchar()!=10);

		}

		//关闭套接字文件描述符


		close(cfd);
		return 0;
}