sender.cpp

多线程文件的传输实现 需要在linux shell 下操作

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

int l_socket; //The Main Listen_socket
struct sockaddr_in adr_s;
int a_socket[5];
pthread_t thread_id[5];
char *fileBlocks[5];
long fileLens[5];

void EOP(const char* errorCode) { printf("Sender Error: %s;\n",errorCode); exit(1); }

int GetFileLength(char *filename) {
	int fsize = 0,temp;
	char buf[512];
	FILE * file=fopen(filename,"r");	
	if (file==NULL) EOP("No Such File.");
	while( (temp=fread(buf,1,512,file))!=0 ) {
		if(temp==-1) EOP("fread file error.");
		fsize+=temp;
	}
	fclose(file);
	return fsize;
}

void CalculateFileLens(long total) {
	long len=total/5;
	for(int i=0;i<4;i++) fileLens[i]=len;
	fileLens[4]=total-len*4;
}

void FillFileBlocks(char *filename) {
	int temp,byteread;
	char *buf_p;
	FILE *file = fopen(filename,"r");
	if (file==NULL) EOP("No Such File.");
	for(int i=0;i<5;i++) {
		fileBlocks[i] = (char*)malloc(sizeof(char)*fileLens[i]+1);
		fileBlocks[i][sizeof(char)*fileLens[i]] = 0;
		if (fileBlocks[i]==NULL) EOP("malloc failed.");
		temp = fileLens[i];
		buf_p = fileBlocks[i];
		while( (byteread=fread(buf_p,1,temp,file))!=0) {
			if (temp==-1) EOP("fill block failed.");
			buf_p+=byteread;
			temp-=byteread;
		}		
	}
}

void InitListenSocket(char *IPaddress,int port) {
	l_socket=socket(PF_INET,SOCK_STREAM,0); 
	  if(l_socket<0) EOP("socket create failed..");
	
	memset(&adr_s,0,sizeof adr_s);
	adr_s.sin_family=AF_INET;
	adr_s.sin_port=htons(port);
	adr_s.sin_addr.s_addr=inet_addr(IPaddress);
	if (adr_s.sin_addr.s_addr==INADDR_NONE) EOP("wrong address..");
}		

void SendBlock(int id) {
	int temp;
	char *buf_p = fileBlocks[id];
	int len=fileLens[id];

	write(a_socket[id],&fileLens[id],sizeof(long));
	write(a_socket[id],&id,sizeof(int));

	while( (temp=write(a_socket[id],buf_p,len))!=0 ) {
		if(temp==-1) EOP("one sendblock() error.");
		len -= temp;
		buf_p += temp;
	}
	printf("Thread %d, Send OK.. -- %ld bytes --\n",id,fileLens[id]);
	close(a_socket[id]);
}

void Listen() {
    int tmp;
    tmp=bind(l_socket,(struct sockaddr*)&adr_s,sizeof adr_s);
    if(tmp<0) EOP("bind failed..");

    tmp=listen(l_socket,5);
    if(tmp<0) EOP("listen failed");	
	
    struct sockaddr_in adr_a;
    int len = sizeof adr_a;
    int id[5];
    int t_s;
    for(int i=0;i<5;i++) {
        a_socket[i]=accept(l_socket,(struct sockaddr*)&adr_a,(socklen_t*)&len);
	if (a_socket[i]<0)  EOP("one socket accepting failed.");
	id[i] = i;
        t_s=pthread_create(&thread_id[i],NULL,(void*(*)(void*))SendBlock,(void*)id[i]);
	if(t_s!=0) EOP("one thread creating failed.");
    }
    close(l_socket);
}

int main(void) {
	char fileName[128],IP[20];
	
	printf("Input the filename you want to send: ");
	scanf("%s",fileName);
 	
	int flen = GetFileLength ( fileName );	

	printf("Then your IP address: ");
	scanf("%s",IP);

	CalculateFileLens ( flen );
	FillFileBlocks ( fileName );
	InitListenSocket(IP,8083);
	
	printf("Now waiting for connection..\n");
	Listen();
	
	for(int i=0;i<5;i++) pthread_join(thread_id[i],NULL);
	
	printf("+--Server Sending Complete--+\n");
	return 0;
}