请教linux上的一个C传送文件的错误

0 0

请教linux上的一个C传送文件的错误10

本人C新手,在linux上面试写了一个文件传送的客户端跟服务器.客户端跟服务器建立连接后,客户发送文件,服务端接收文件.在服务器接收文件fwrite的时候出现了一个段错误,请各位指教.
服务端代码:

           
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/resource.h>
#include <sys/epoll.h>

#include "msghead.h"

#define MAXBUF 1024
#define MAXEPOLLSIZE 10000
#define RECEIVE_FILE "/home/jason/c++/me1.jpg"

int setnonblock(int sockfd){
	int t;
	t=fcntl(sockfd,F_SETFL,fcntl(sockfd,F_GETFD,0)|O_NONBLOCK);
	if(t<0){
		perror("fcntl()");
		return -1;
	}else{
		return 0;
	}
}

int receiveFile(int sockfd){
	char buf[MAXBUF];
	bzero(buf,MAXBUF);
	FILE * stream;
	if(fopen(RECEIVE_FILE,"w+t")==NULL){
		perror("fopen()");
		exit(1);
	}
	int len,totallen=0;
	while(len=recv(sockfd,buf,MAXBUF,0)){
		if(len<0){
			printf("receive file faile\n");
			break;
		}
		printf("len=%d\n",len);
		int write_len=fwrite(buf,sizeof(char),len,stream);
		if(write_len<len){
			printf("file write faile\n");
			break;
		}
		totallen=totallen+write_len;
		bzero(buf,MAXBUF);
	}
	printf("receive file finish!\n");
	fclose(stream);
	return totallen;
}

int handleMsg(int sockfd){
	int filelen=receiveFile(sockfd);
	printf("filelen=%d\n");
}


int main(){
	int sockfd,newfd,handlefd;
	struct sockaddr_in my_addr,des_addr;
	int port=1984,lisnum;
	
	int epollfd,wait_num;
	
	struct rlimit r;
	struct epoll_event e;
	struct epoll_event events[MAXEPOLLSIZE];

	//set rlimit.
	r.rlim_cur=r.rlim_max=MAXEPOLLSIZE;
	if(setrlimit(RLIMIT_NOFILE,&r)<0){
		perror("setrlimit()");
		exit(1);
	}else{
		printf("setrlimit have done!\n");
	}
	
	//socket.
	sockfd=socket(PF_INET,SOCK_STREAM,0);
	if(sockfd<0){
		perror("socket()");
		exit(1);
	}else{
		printf("socket have done!\n");
	}

	setnonblock(sockfd);

	//bind.
	bzero(&my_addr,sizeof(my_addr));
	my_addr.sin_family=PF_INET;
	my_addr.sin_port=htons(port);
	my_addr.sin_addr.s_addr=INADDR_ANY;
	if(bind(sockfd,(struct sockaddr*)&my_addr,sizeof(struct sockaddr))<0){
		perror("bind()");
		exit(1);
	}else{
		printf("bind have done!\n");	
	}	

	//listen.
	if(listen(sockfd,lisnum)<0){
		perror("listen()");
		exit(1);
	}else
		printf("listen have done!\n");

	//epoll_create.
	epollfd=epoll_create(MAXEPOLLSIZE);
	e.events=EPOLLIN|EPOLLET;
	e.data.fd=sockfd;

	//epoll_ctl.
	if(epoll_ctl(epollfd,EPOLL_CTL_ADD,sockfd,&e)<0){
		perror("epoll_ctl()");
		exit(1);
	}else
		printf("epoll_ctl have done!\n");

	int cur=1;
	while(1){
		wait_num=epoll_wait(epollfd,events,cur,-1);
		if(wait_num<0){
			perror("epoll_wait()");
			exit(1);
		}else
			printf("epoll_wait have done!\n");

		int i=0;
		for(i;i<wait_num;++i){
			if(events[i].data.fd==sockfd){
				int len=sizeof(struct sockaddr);
				newfd=accept(sockfd,(struct sockaddr*)&des_addr,&len);
				if(newfd<0){
					perror("accept()");
					printf("aaaa\n");
					continue;
				}else
					printf("have a connection  from ip:%s\n",inet_ntoa(des_addr.sin_addr));
				printf("bbbb\n");
				setnonblock(newfd);
				e.events=EPOLLIN|EPOLLET;
				e.data.fd=newfd;
				
				if(epoll_ctl(epollfd,EPOLL_CTL_ADD,newfd,&e)<0){
					perror("newfd epoll_ctl()");
					exit(1);
				}
				cur++;
			}else{
				handlefd=handleMsg(events[i].data.fd);
				if(handlefd<1&&errno!=11){
					epoll_ctl(epollfd,EPOLL_CTL_DEL,events[i].data.fd,&e);
					cur--;
				}
				
			}
		}
	}
	close(sockfd);

	
}

客户端代码:

        

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <arpa/inet.h>

#define MAXBUF 1024

struct msgpak{
	char msg_from[6];
	char msg_to[7];
	int  msg_len;
};

int main(){

	int sockfd;
	struct sockaddr_in their_addr;
	char buf[MAXBUF+1];

	int desport=1984;
	char * desip="127.0.0.1";
	char * sendFile="/home/jason/c++/me.jpg";
	
	//socket().
	if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1){
		perror("socket!");
		exit(1);
	}else
		printf("socketed!");
	
	//connect().
	bzero(&their_addr,sizeof(their_addr));
	their_addr.sin_family=AF_INET;
	their_addr.sin_port=htons(desport);
	if(inet_aton(desip,(struct in_addr *)&their_addr.sin_addr.s_addr)==0){
		perror(desip);
		exit(1);
	}
	
	if(connect(sockfd,(struct sockaddr*)&their_addr,sizeof(struct sockaddr))!=0){
		perror("connect");
		exit(1);
	}

	//recv().
	//bzero(&buf,MAXBUF+1);
	//int len=recv(sockfd,buf,MAXBUF,0);
	//if(len>0){
		//printf("receive msg:%s\n",buf);
	//}else{
		//printf("receive msg error!\n");
	//}

	//send().
	FILE * stream;
	if((stream=fopen(sendFile,"r"))==NULL){
		perror("fopen()");
		exit(1);
	}else
		printf("the file:%s is opened!\n",sendFile);
	
	bzero(buf,MAXBUF);
	int longsize=0;
	while((longsize=fread(buf,1,1024,stream))>0){
		printf("longsize=%d\n",longsize);
		if(send(sockfd,buf,longsize,0)<0){
			printf("send file is faile:\n");
			break;
		}
		bzero(buf,MAXBUF);
	}
	if(fclose(stream)){
		printf("the file %s was not close\n",sendFile);
		exit(1);
	}

	close(sockfd);
	return 0;
}