网络第四天

udp的下载

#include <head.h>

#define IP "192.168.125.31"



int do_download(int sfd, struct sockaddr_in sin)
{
	//定义变量存储下载请求包
	char buf[516] = "";

	//定义变量存储文件名
	char fileName[40]="";

	printf("请输入文件名: ");
	scanf("%s", fileName);
	getchar();

	//组装请求包
	short *p1 = (short *)buf;
	*p1 = htons(1);

	char *p2 = buf + 2;
	strcpy(p2, fileName);

	char *p4 = p2 + strlen(p2) + 1;  //模式段
	strcpy(p4, "octet");

	int size = 4 + strlen(p2) + strlen(p4);   //要发送数据大小

	//向服务器发送下载请求
	if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) == -1)
	{
		perror("sendto error");
		return -1;
	}
	printf("请求成功\n");

	int fd = -1;
	if((fd=open("./text.txt", O_WRONLY|O_TRUNC|O_CREAT, 0664)) == -1)
	{
		perror("open error\n");
		return -1;
	}

	socklen_t addrlen = sizeof(sin);
	int wr = 0;
	unsigned short num = 0;
	while(1)
	{
		int res = recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
		printf("res = %d\n", res);
		if(res<0)
		{
			perror("recvfrom error\n");
			break;
		}


		//判断收到的是不是错误包
		if(buf[1]==5)
		{
			printf("%s", buf+4);
		}
	
		char *q = buf + 4;

		if(res<516)       //如果接收的字节小于516执行此程序
		{
			write(fd, q, res-4);
			break;
		}

		wr = write(fd, q, res-4);
		printf("wr = %d\n", wr);

	//	*p1=htons(4);      
		buf[1]=4;

		if(sendto(sfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) == -1)
		{
			perror("sendto error");
			return -1;
		}
		
	}
	close(fd);
	return 0;
}

//上传功能
int do_updata(int sfd, struct sockaddr_in sin)
{

}


int main(int argc, const char *argv[])
{
	int sfd = -1;
	if((sfd=socket(AF_INET, SOCK_DGRAM, 0)) == -1)
	{
		perror("socket error\n");
		return -1;
	}

	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(69);
	sin.sin_addr.s_addr = inet_addr(IP);

	int mune = -1;

	while(1)
	{
		system("clear");
		printf("\t\t======1.下载======\n");
		printf("\t\t======2.上传======\n");
		printf("\t\t======0.退出======\n");

		printf("请输执行功能: ");
		scanf("%d", &mune);
		getchar();

		switch(mune)
		{
			case 1:
				{
					do_download(sfd, sin);
				}
				break;
			case 2:
				{
					//do_updata(sfd, sin);
				}
				break;
			case 0:
				goto POS;
			default:
				printf("输入功能有误，请重新输入\n");
		}

		printf("输入任意键清空！");
		while(getchar() != '\n');
	}

POS:

	close(sfd);


	return 0;
}