server.c

一个简单的linux下的网络传输程序

/********************************************************************

   Copyright 2004 by YAAN all rights reserved.
   Filename:    server.c
   Author:      Edgar Liu
   Date:        2004.8.12
   Description: 
   Others:
   Main Function Lists:
	1.
   History:

********************************************************************/
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <dirent.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include "public.h"

int       g_nSwitch = 1;
char      g_szPath[MAX_PATH_LENTH];
short int g_nPort = DEFAULT_PORT;
unsigned long int g_nSequence = 1;
unsigned long int g_nFileCount = 1;
unsigned long int g_nTotalBytes = 0;

/********************************************************************

   Function:   void Usage(int argc,char** argv,int nCode)
   Description: Show Usage of Software;
   Arguments:   int argc  ----The Number of CommandLine Paramiter
		char** argv --
   Return:	NULL
   Others:

********************************************************************/
void Usage(int argc,char** argv,int nCode)
{
	switch (nCode)
	{
	case 1:
		break;
	case 2:
		fprintf(stderr,"Please Input your Directory!\n");
		break;
	case 3:
		fprintf(stderr,"Port Error!\n");
		break;
	}
	fprintf(stderr,"Usage:%s [-d Directory][-p Port]\n",argv[0]);
	exit(1);
}
/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/
void parseCmdLine(int argc,char** argv)
{
	int i,nErrCode = 0;
	if(argc == 1)
	{
		nErrCode = 1;
		Usage(argc,argv,nErrCode);
	}

	for(i=0;i<argc;i++)
	{
		if(argv[i][0] == '-')
		{
			switch (argv[i][1])
			{
			case 'd':
				if(argv[i+1] == NULL)
				{
					nErrCode = 2;
					Usage(argc,argv,nErrCode);
				}
				memset(g_szPath,0,MAX_PATH_LENTH);
				strcpy(g_szPath,argv[i+1]);
				int nLen = strlen(g_szPath);
				if(g_szPath[nLen-1] != '/')
					g_szPath[nLen] = '/';
				break;
			case 'p':
				if(argv[i+1] == NULL || IsNumber(argv[i+1]))
				{
					nErrCode = 3;
					Usage(argc,argv,nErrCode);
				}
				g_nPort = atoi(argv[i+1]);
				break;		
			default:
				break;
			}
			
		}
	}
}

/*******************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Other:

*******************************************************************/
void verifyUser(char* pClient,char* pPasswd)
{

}

/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/
void  parseCommand(char* pCommand)
{
	if(!strcmp(pCommand,"q") || !strcmp(pCommand,"Q") || !strcmp(pCommand,"quit"))
		g_nSwitch = 0;
}

/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/
void  parseClientMsg(PCLIENT_INFO pCI)
{
	printf("Logo : %s  ",pCI->szLogo);
	printf("User: %s Passwd: %s Command:%s\n\n",pCI->szClient,pCI->szPasswd,pCI->szCommand);
	verifyUser(pCI->szClient,pCI->szPasswd);
	parseCommand(pCI->szCommand);
}

/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/

int  parsePath(char* pPath)
{
	char szTemp[MAX_PATH_LENTH];
	char* pPos;
	int nLen,nLen1; 
	memset(szTemp,0,MAX_PATH_LENTH);
	strcpy(szTemp,pPath);
	nLen = strlen(szTemp);
	szTemp[nLen-1] = 0;
	pPos = strrchr(szTemp,'/');
	nLen1 = strlen(pPos);

	return nLen-1-nLen1;	
}
/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/
void sendData(int sockfd,struct sockaddr_in* addr,socklen_t len,char* pPath,int nLen)
{
	FILE* fp;
	struct ServerData sd;
	struct stat fs;
	int nSize;

	memset(&sd,0,sizeof(struct ServerData));

	if(stat(pPath,&fs) != 0)
	{
		perror("stat");
	}

	fp = fopen(pPath,"r");
	if(fp == NULL)
	{
		fprintf(stderr,"%s no exist!\n",pPath);
		fclose(fp);
	}

	strcpy(sd.Header.szLogo,"YAAN");
	sd.Header.uFileCount = g_nFileCount++;
	strcpy(sd.Header.szFileName,pPath+nLen+1);
	sd.Header.uFileSize = fs.st_size;
	sd.Header.nMode     = fs.st_mode;
	
	g_nTotalBytes += fs.st_size;

	if(fs.st_size < MAX_DATA_SIZE)
	{
 		fread(sd.szData,1,fs.st_size,fp);
		sd.Header.uSequence  = g_nSequence++;
		sendto(sockfd,
			(char*)&sd,
			SERVER_HEADER_SIZE+sd.Header.uFileSize+1,
			0,
			(struct sockaddr*)addr,len);
		printf("%ld\tSend %s   successfully\n",g_nSequence-1,pPath);	
	}
	else
	{
		while((nSize = fread(sd.szData,1,MAX_DATA_SIZE,fp)) > 0)	
		{
			sd.Header.uSequence  = g_nSequence++;
			sd.Header.uFileSize = nSize;
			sendto(sockfd,
				(char*)&sd,
				SERVER_HEADER_SIZE+nSize,
				0,
				(struct sockaddr*)addr,len);
			memset(sd.szData,0,MAX_DATA_SIZE);
			usleep(20);		
		}
		printf("%ld\t Send %s   successfully\n",g_nSequence-1,pPath);
	
	}
	fclose(fp);		
}

/********************************************************************
   
   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/
void  browseDir(int sockfd,struct sockaddr_in* addr,socklen_t len,char* pDir,int nLen)
{
	DIR* dp;
	struct dirent* dirp;
	char szPath[MAX_PATH_LENTH];
	struct stat buf;
	
	if((dp = opendir(pDir)) == NULL)
		perror("opendir");
	while((dirp = readdir(dp)) != NULL)
	{
		memset(szPath,0,MAX_PATH_LENTH);
		strcpy(szPath,pDir);
		strcat(szPath,dirp->d_name);
		if(stat(szPath,&buf) != 0)
			perror("stat");
		if(S_ISDIR(buf.st_mode))  //is Directory
		{
			if(!strcmp(dirp->d_name,".")||!strcmp(dirp->d_name,".."))
				continue;
			strcat(szPath,"/");
			browseDir(sockfd,addr,len,szPath,nLen);
		}
		else if(S_ISREG(buf.st_mode))  //is General File
		{
			sendData(sockfd,addr,len,szPath,nLen);
				
		} 
	}
	closedir(dp);

}
/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/

int initSocket()
{
	int sockfd;
	struct sockaddr_in servAddr;
	sockfd = socket(AF_INET,SOCK_DGRAM,0);
	if(sockfd < 0)
	{
		fprintf(stderr,"Socket Error:%s\n",strerror(errno));
		exit(1);
	}
//	int on = 1;
//	ioctl(sockfd,FIONBIO,&on);
	bzero(&servAddr,sizeof(servAddr));
	servAddr.sin_family      = AF_INET;
	servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
	servAddr.sin_port        = htons(g_nPort);
	if(bind(sockfd,(struct sockaddr*)&servAddr,sizeof(servAddr)) < 0)
	{
		fprintf(stderr,"Bind Error: %s\n",strerror(errno));
		exit(1);
	}
	return sockfd;
}
/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/

void broadcastData(int sockfd,char* pDir)
{
	int n,nLen;
	CLIENT_INFO clientInfo;
	struct sockaddr_in addr;
	socklen_t socklen;

	while(g_nSwitch)
	{
		memset(&clientInfo,0,sizeof(CLIENT_INFO));
		n = recvfrom(sockfd,(char*)&clientInfo,sizeof(CLIENT_INFO),0,(struct sockaddr*)&addr,&socklen);

		printf("From %s received data:\n",inet_ntoa(addr.sin_addr));
		parseClientMsg(&clientInfo);
		printf("Begin send data to %s ......\n",inet_ntoa(addr.sin_addr));
		nLen = parsePath(pDir);
		browseDir(sockfd,&addr,socklen,pDir,nLen);

		printf("统计: 共发送 %ld个文件, %ld个数据包, %ldK字节数据\n",
			g_nFileCount-1,g_nSequence-1,g_nTotalBytes/1024);
		if(g_nSwitch == 0)
			fprintf(stderr,"Server Quit!!\n");
	}
}
/********************************************************************

   Function:
   Description:
   Arguments:
   Return:
   Others:

********************************************************************/

int main(int argc,char** argv)
{
	//
	int sockfd;

	//init block;
	memset(g_szPath,0,MAX_PATH_LENTH);
	strcpy(g_szPath,"/home/edgar/");

	//parse Command Line
	parseCmdLine(argc,argv);
	printf("Path:%s============\n",g_szPath);
	//main block;
	sockfd = initSocket();	
	broadcastData(sockfd,g_szPath);
	close(sockfd);
}