server.cpp

本程序是网络编程部分

/********************************************************************
created:	2006/07/28
filename: 	Server.cpp
author:		李创
http://www.cppblog.com/converse/

purpose:	
*********************************************************************/

#pragma comment(lib, "WS2_32")

#include <WinSock2.h>
#include <stdio.h>
#include <assert.h>
#include "../CommonDef.h"

char	g_szNewFileName[MAXFILEDIRLENGTH];
char	g_szBuff[MAX_PACKET_SIZE + 1];
long	g_lLength;
char*	g_pBuff = NULL;

// 初始化socket库
bool InitSocket();
// 关闭socket库
bool CloseSocket();
// 解析消息进行相应的处理
bool ProcessMsg(SOCKET sClient);
// 监听Client的消息
void ListenToClient();
// 打开文件
bool OpenFile(CCSDef::TMSG_HEADER* pMsgHeader, SOCKET sClient);
// 传送文件
bool SendFile(SOCKET sClient);
// 读取文件进入缓冲区
bool ReadFile(SOCKET sClient);

int main()
{
	InitSocket();
	ListenToClient();
	CloseSocket();

	return 0;
}

void ListenToClient()
{
	// 创建socket套接字
	SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (SOCKET_ERROR == sListen)
	{
		printf("Init Socket Error!\n");
		return;
	}

	// 绑定socket到一个本地地址
	sockaddr_in sin;
	sin.sin_family	= AF_INET;
	sin.sin_port	= htons(PORT);
	sin.sin_addr.S_un.S_addr = INADDR_ANY;
	if (::bind(sListen, (LPSOCKADDR)&sin, sizeof(sockaddr_in)) == SOCKET_ERROR)
	{
		printf("Bind Error!\n");
		return;
	}

	// 设置socket进入监听状态
	if (::listen(sListen, 10) == SOCKET_ERROR)
	{
		printf("Listen Error!\n");
		return;
	}

	printf("Listening To Client...\n");

	// 循环接收client端的连接请求
	sockaddr_in ClientAddr;
	int nAddrLen = sizeof(sockaddr_in);
	SOCKET sClient;

	while (INVALID_SOCKET == (sClient = ::accept(sListen, (sockaddr*)&ClientAddr, &nAddrLen)))
	{
	}

	while (true == ProcessMsg(sClient))
	{
	}

	// 关闭同客户端的连接
	::closesocket(sClient);

	::closesocket(sListen);
}

bool InitSocket()
{
	// 初始化socket dll
	WSADATA wsaData;
	WORD socketVersion = MAKEWORD(2, 2);
	if (::WSAStartup(socketVersion, &wsaData) != 0)
	{
		printf("Init socket dll error\n");
		return false;
	}

	return true;
}

bool CloseSocket()
{
	// 释放winsock库
	::WSACleanup();

	if (NULL != g_pBuff)
	{
		delete [] g_pBuff;
		g_pBuff = NULL;
	}

	return true;
}

bool ProcessMsg(SOCKET sClient)
{
	int nRecv = ::recv(sClient, g_szBuff, MAX_PACKET_SIZE + 1, 0);
	if (nRecv > 0)
	{
		g_szBuff[nRecv] = '\0';
	}

	// 解析命令
	CCSDef::TMSG_HEADER* pMsgHeader = (CCSDef::TMSG_HEADER*)g_szBuff;
	switch (pMsgHeader->cMsgID)
	{
	case MSG_FILENAME:			// 文件名
		{
			OpenFile(pMsgHeader, sClient);
		}
		break;
	case MSG_CLIENT_READY:		// 客户端准备好了,开始传送文件
		{
			SendFile(sClient);
		}
		break;
	case MSG_SENDFILESUCCESS:	// 传送文件成功
		{
			printf("Send File Success!\n");
			return false;
		}
		break;
	case MSG_FILEALREADYEXIT_ERROR:	// 要保存的文件已经存在了
		{
			printf("The file reay to send already exit!\n");
			return false;
		}
		break;
	}

	return true;
}

bool ReadFile(SOCKET sClient)
{
	if (NULL != g_pBuff)
	{
		return true;
	}

	// 打开文件
	FILE *pFile;
	if (NULL == (pFile = fopen(g_szNewFileName, "rb")))		// 打开文件失败
	{
		printf("Cannot find the file, request the client input file name again\n");
		CCSDef::TMSG_ERROR_MSG tMsgErrorMsg(MSG_OPENFILE_ERROR);
		::send(sClient, (char*)(&tMsgErrorMsg), sizeof(CCSDef::TMSG_ERROR_MSG), 0);
		return false;
	}

	// 把文件的长度传回到client去
	fseek(pFile, 0, SEEK_END);
	g_lLength = ftell(pFile);
	printf("File Length = %d\n", g_lLength);
	CCSDef::TMSG_FILELENGTH tMsgFileLength(g_lLength);
	::send(sClient, (char*)(&tMsgFileLength), sizeof(CCSDef::TMSG_FILELENGTH), 0);

	// 处理文件全路径名,把文件名分解出来
	char szDrive[_MAX_DRIVE], szDir[_MAX_DIR], szFname[_MAX_FNAME], szExt[_MAX_EXT];
	_splitpath(g_szNewFileName, szDrive, szDir, szFname, szExt);
	strcat(szFname,szExt);
	CCSDef::TMSG_FILENAME tMsgFileName;
	strcpy(tMsgFileName.szFileName, szFname);
	printf("Send File Name: %s\n", tMsgFileName.szFileName);
	::send(sClient, (char*)(&tMsgFileName), sizeof(CCSDef::TMSG_FILENAME), 0);

	// 分配缓冲区读取文件内容
	g_pBuff = new char[g_lLength + 1];
	if (NULL == g_pBuff)
	{
		return false;
	}

	fseek(pFile, 0, SEEK_SET);
	fread(g_pBuff, sizeof(char), g_lLength, pFile);
	g_pBuff[g_lLength] = '\0';
	fclose(pFile);

	return true;
}

// 打开文件
bool OpenFile(CCSDef::TMSG_HEADER* pMsgHeader, SOCKET sClient)
{
	CCSDef::TMSG_FILENAME* pRequestFilenameMsg = (CCSDef::TMSG_FILENAME*)pMsgHeader;

	// 对文件路径名进行一些处理
	char *p1, *p2;
	for (p1 = pRequestFilenameMsg->szFileName, p2 = g_szNewFileName;
		'\0' != *p1;
		++p1, ++p2)
	{
		if ('\n' != *p1)
		{
			*p2 = *p1;
		}
		if ('\\' == *p2)
		{
			*(++p2) = '\\';
		}
	}
	*p2 = '\0';

	ReadFile(sClient);

	return true;
}

// 传送文件
bool SendFile(SOCKET sClient)
{
	if (NULL == g_pBuff)
	{
		ReadFile(sClient);
	}

	int nPacketBufferSize = MAX_PACKET_SIZE - 2 * sizeof(int);	// 每个数据包存放文件的buffer大小
	// 如果文件的长度大于每个数据包所能传送的buffer长度那么就分块传送
	for (int i = 0; i < g_lLength; i += nPacketBufferSize)
	{		
		CCSDef::TMSG_FILE tMsgFile;
		tMsgFile.tFile.nStart = i;
		if (i + nPacketBufferSize + 1> g_lLength)
		{
			tMsgFile.tFile.nSize  = g_lLength - i;
		}
		else
		{
			tMsgFile.tFile.nSize  = nPacketBufferSize;
		}
		//printf("start = %d, size = %d\n", tMsgFile.tFile.nStart, tMsgFile.tFile.nSize);
		memcpy(tMsgFile.tFile.szBuff, g_pBuff + tMsgFile.tFile.nStart, tMsgFile.tFile.nSize);
		::send(sClient, (char*)(&tMsgFile), sizeof(CCSDef::TMSG_FILE), 0);
		Sleep(0.5);
	}

	delete [] g_pBuff;
	g_pBuff = NULL;

	return true;
}