网络编程第3天作业

     1	头文件部分
        #ifndef __HEAD_H__
     2	#define __HEAD_H__
     3	
     4	#include 
     5	#include 
     6	#include 
     7	#include 
     8	#include 
     9	#include 
    10	#include 
    11	#include 
    12	#include 
    13	#include 
    14	#include 
    15	#include 
    16	#include 
    17	#include 
    18	#include 
    19	#include 
    20	#include 
    21	#include 
    22	#include 
    23	#include 
    24	#include 
    25	#define ERR_MSG(msg) do{\
    26		fprintf(stderr,"__%d__",__LINE__);\
    27		perror(msg);\
    28	}while(0)
    29	#define LINE do{\
    30		printf("__%d__\n",__LINE__);\
    31	}while(0)
    32	int do_download(int cfd,struct sockaddr_in sin);
    33	int do_upload(int cfd,struct sockaddr_in sin);
    34	#endif
    35	
    36   函数部分
    37	#include "head.h"
    38	int do_download(int cfd,struct sockaddr_in sin)
    39	{
    40		char buf[516] = "";
    41		short *ptr1 = (short*)buf;
    42		*ptr1 = htons(1);
    43		char arr[20] = "";
    44		printf("请输入文件名");
    45		scanf("%s",arr);
    46		getchar();
    47		char *ptr2 = buf+2;
    48		strcpy(ptr2,arr);
    49		char* ptr3 = ptr2+strlen(ptr2)+1;
    50		strcpy(ptr3,"octet");
    51		int size = 2+strlen(ptr2)+1+strlen(ptr3)+1;
    52		if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
    53		{
    54			ERR_MSG("sendto");
    55			return -1;
    56		}
    57		printf("发送下载请求成功");
    58		LINE;
    59		struct sockaddr_in cin;
    60		socklen_t addrlen = sizeof(cin);
    61		ssize_t res = 0;
    62		int fdr = open(arr,O_WRONLY|O_CREAT|O_APPEND,0664);
    63		if(fdr < 0)
    64		{
    65			ERR_MSG("fdr");
    66			return -1;
    67		}
    68		bzero(buf,sizeof(buf));
    69		int i = 1;
    70		char adf[4] = "";
    71		short *pt1 = (short*)adf;
    72		short *pt2 = pt1+1;
    73		while(1)
    74		{
    75			bzero(buf,sizeof(buf));
    76			res = recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&addrlen);
    77			if(res < 0);
    78			{
    79				ERR_MSG("recvfrom");
    80			}
    81	
    82			*pt1 = htons(4);
    83			*pt2 = htons(i);
    84			i++;
    85			if(sendto(cfd,adf,4,0,(struct sockaddr*)&cin,sizeof(cin)) < 0)
    86			{
    87				ERR_MSG("sendto");
    88				return -1;
    89			}
    90			if(res-4 == 512)
    91			{
    92				write(fdr,&buf[4],512);
    93			}
    94			else
    95			{
    96				write(fdr,&buf[4],res-4);
    97				break;
    98			}
    99		}
   100		close(fdr);
   101	
   102	
   103		return 0;
   104	}
   105	int do_upload(int cfd,struct sockaddr_in sin)
   106	{
   107		char buf[516] = "";
   108		short* p1 = (short*)buf;
   109		*p1 = htons(2);
   110		char arr[20] = "";
   111		printf("请输入文件名");
   112		scanf("%s",arr);
   113		getchar();
   114		char *p2 = buf+2;
   115		strcpy(p2,arr);
   116		char* p3 = p2+strlen(p2)+1;
   117		strcpy(p3,"octet");
   118		int size = 2+strlen(p2)+1+strlen(p3)+1;
   119		if(sendto(cfd,buf,size,0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
   120		{
   121			ERR_MSG("sendto");
   122			return -1;
   123		}
   124		printf("发送上传请求成功");
   125		int fdr = open(arr,O_RDONLY);
   126		if(fdr < 0)
   127		{
   128			ERR_MSG("open");
   129			return -1;
   130		}
   131		ssize_t res1 = 0;
   132		int sum = 0;
   133		off_t offset = lseek(fdr,0,SEEK_END);
   134		lseek(fdr,0,SEEK_SET);
   135		sum = (int)offset/512;
   136		int flag = 0;
   137		bzero(buf,sizeof(buf));
   138		short *ptr1 = (short*)buf;
   139		short* ptr2 = ptr1+1;
   140		struct sockaddr_in cin;
   141		socklen_t addrlen = sizeof(cin);
   142		int num = 1;
   143		ssize_t res = 0;
   144		char adf[4] = "";
   145		short* pd1 = (short*)adf;
   146		short* pd2 = pd1+1;
   147		while(1)
   148		{
   149			bzero(buf,sizeof(buf));
   150			bzero(adf,sizeof(adf));
   151			*ptr1 = htons(3);
   152			*ptr2 = htons(num);
   153			*pd1 = htons(4);
   154			*pd2 = htons(num);
   155			printf("%d %d",buf[1],buf[2]);
   156			LINE;
   157			res = recvfrom(cfd,adf,4,0,(struct sockaddr*)&cin,&addrlen);
   158			printf("%d %d %d\n",adf[0],adf[1],(int)res);
   159			if(res < 0)
   160			{
   161				ERR_MSG("recvfrom");
   162				return -1;
   163			}
   164			if(flag < sum)
   165			{
   166				res1 = read(fdr,&buf[4],512);
   167				printf("%d\n",(int)res1);
   168				LINE;
   169				if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,addrlen) < 0)
   170				{
   171					ERR_MSG("sendto");
   172					return -1;
   173				}
   174			}
   175			else
   176			{
   177				res1 = read(fdr,&buf[4],offset-sum*512);
   178				LINE;
   179				if(sendto(cfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,addrlen) < 0)
   180				{
   181					ERR_MSG("sendto");
   182					return -1;
   183				}
   184				break;
   185			}
   186			flag++;
   187			num++;
   188			LINE;
   189		}
   190	
   191	
   192	
   193	
   194	}
        主函数部分
   195	#include "head.h"
   196	#define PORT 69
   197	#define IP "192.168.0.100"
   198	int main(int argc, const char *argv[])
   199	{
   200		int cfd = socket(AF_INET,SOCK_DGRAM,0);
   201		if(cfd < 0)
   202		{
   203			ERR_MSG("socket");
   204			return -1;
   205		}
   206		printf("socket create success cfd=%d",cfd);
   207		LINE;
   208		struct sockaddr_in sin;
   209		sin.sin_family 		= AF_INET;
   210		sin.sin_port 		= htons(PORT);
   211		sin.sin_addr.s_addr = inet_addr(IP);
   212		char choose = 0;
   213		while(1)
   214		{
   215			printf("------------------------\n");
   216			printf("---------1. 下载--------\n");
   217			printf("---------2. 上传--------\n");
   218			printf("---------3. 退出--------\n");
   219			printf("------------------------\n");
   220			printf("请输入>>>>");
   221			choose = getchar();
   222			while(getchar()!='\n');
   223			switch(choose)
   224			{
   225			case '1':
   226				do_download(cfd,sin);
   227				break;
   228			case '2':
   229				do_upload(cfd,sin);
   230				break;
   231			case '3':
   232				goto FOUNCE;
   233			default:
   234				printf("输入错误,请重新输入\n");
   235				break;
   236	
   237			}
   238		}
   239	FOUNCE:
   240		close(cfd);
   241		return 0;
   242	}

你可能感兴趣的:(网络)