32: }
33:
34: ssize_t my_write(int fd,void *buffer,size_t length)
35: {
36: size_t bytes_left; //尚未写的文件大小
37: size_t writesize = 4* 1024;
38: ssize_t written_bytes; //已经写的文件大小
39: char *ptr;
40: ptr=buffer;
41: bytes_left=length;
42: while(bytes_left>0)
43: {
44: //开始写
45: written_bytes=write(fd,ptr,min(bytes_left,writesize));
46: //出现了写错误
47: if(written_bytes<=0)
48: {
49: //中断错误，置零重新写
50: if(errno==EINTR)
51: written_bytes=0;
52: //其他错误，退出不写了
53: else
54: return(-1);
55: }
56: //从剩下的地方继续写
57: bytes_left-=written_bytes;
58: ptr+=written_bytes;
59: }
60: return(0);
61: }
62:
63:
64:
65: int main(int argc, char *argv[])
66: {
67: int sockfd; //通信套接字描述符
68: char *buffer; //缓冲区
69: struct sockaddr_in server_addr; //服务器地址结构
70: struct hostent *host; //主机地址与名称信息结构
71: int nbytes; //端口号、字节数
72: FILE *fp; //文件指针
73: int nfilesize; //文件大小
74: char str[128]; //文件名
75: char yes='Y'; //流程控制
76: struct timeval tpstart,tpend; //用于记录文件传输时间
77: float timeuse; //文件传输所用时间
78: char *hostname="127.0.0.1";//主机名/ip地址
79: int portnumber=4321;//端口号
80:
81: //提示用户输入完整的命令行参数
82: if(argc!=3)
83: {
84: fprintf(stderr,"Usage:%s hostname portnumber\a\n",argv[0]);
85: printf("using defaults:\nhostname: %s\nportnumber: %d\n",hostname,portnumber);
86: }
87:
88: //如果利用用户输入的域名无法获得正确的主机地址信息，则退出
89: if (argc>1)
90: {
91: if((host=gethostbyname(argv[1]))==NULL)
92: {
93: fprintf(stderr,"Gethostname error\n");
94: exit(1);
95: }
96: }
97:
98: else
99: if((host=gethostbyname(hostname))==NULL)
100: {
101: fprintf(stderr,"Gethostname error\n");
102: exit(1);
103: }
104:
105: if(argc>2)
106:
107: //如果用户输入的端口不正确，则提示并退出
108: if((portnumber=atoi(argv[2]))<0)
109: {
110: fprintf(stderr,"Usage:%s hostname portnumber\a\n",argv[0]);
111: exit(1);
112: }
113:
114: //客户程序开始建立 sockfd描述符，创建通信套接字
115: if((sockfd=socket(AF_INET,SOCK_STREAM,6))==-1)
116: {
117: fprintf(stderr,"Socket Error:%s\a\n",strerror(errno));
118: exit(1);
119: }
120:
121:
122: //客户程序填充服务端的地址信息
123: bzero(&server_addr,sizeof(server_addr));
124: server_addr.sin_family=AF_INET;
125: server_addr.sin_port=htons(portnumber);
126: server_addr.sin_addr=*((struct in_addr *)host->h_addr);
127:
128: //客户程序发起连接请求
129: if(connect(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)
130: {
131: fprintf(stderr,"Connect Error:%s\a\n",strerror(errno));
132: exit(1);
133: }
134: printf("Connection Succeed!\n");
135:
136: while (toupper(yes)=='Y')
137: {
138: //提示用户输入文件路径
139: printf("Please input the file location:");
140: scanf("%s",str);
141: while ((fp=fopen(str,"r"))==NULL)
142: {
143: fprintf(stderr,"File open error,Retry!\n");
144: printf("Please input the file location:");
145: scanf("%s",str);
146: //exit(1);
147: }
148: getchar();
149:
150: //获取打开的文件的大小，并将文件整个读入内存中
151: fseek(fp,0L,SEEK_END);
152: nfilesize=ftell(fp);
153: rewind(fp);//most important!!!!!
154: char *p=(char *)malloc(nfilesize);
155: if (fread((void *)p,nfilesize,1,fp)<1) {
156:
157: if (feof(fp))
158: printf("read end of file!\nquit!\n");
159: else
160: printf("read file error!\n");
161: }
162:
163: //将要传输的文件的大小信息发送给客户端
164: if (my_write(sockfd,(void *)&nfilesize,4)==-1)
165: {
166: fprintf(stderr,"Write Error:%s\n",strerror(errno));
167: exit(1);
168: }
169:
170: printf("Begin to transfer the file!\n");
171: getchar();
172:
173: //获取传输初始时间
174: gettimeofday(&tpstart,NULL);
175:
176: //传输文件
177: if (my_write(sockfd,p,nfilesize)==-1)
178: {
179: fprintf(stderr,"Transfer failed!");
180: exit(1);
181: }
182:
183: //获取传输结束时间
184: gettimeofday(&tpend,NULL);
185: //计算整个传输用时
186: timeuse=1000000*(tpend.tv_sec-tpstart.tv_sec)+(tpend.tv_usec-tpstart.tv_usec);
187: timeuse/=1000000;
188:
189: printf("Transfer Succeed!\nFile Name: %s\nFile Size: %d bytes\nTotal Time: %f seconds\nTransfer Speed: %f bytes/second",str,nfilesize,timeuse,((float)nfilesize)/timeuse);
190: free(p); //释放文件内存
191: fclose(fp); //关闭文件
192: // printf("\nTransfer another file?(Y/N): ");
193: //scanf("%c",&yes);
194: // getchar();
195: yes='n';
196: }
197: //结束通讯，关闭套接字，关闭连接
198: close(sockfd);
199: printf("\nClient Exit!~~\n");
200: exit(0);
201: }
