p2pclientdemo.cpp

P2PClientDemo P2PServer show the p2p test.

//////////////////////////////////////////////////////////
// P2PClient.cpp文件



#include <winsock2.h>
#include <stdio.h>
#include "p2pclient.h"

class CMyP2P : public CP2PClient
{
public:

	void OnRecv(char *pszUserName, char *pszData, int nDataLen)
	{
		pszData[nDataLen] = '\0';
		printf(" Recv a Message from %s :  %s \n", pszUserName, pszData);
	}
};

void main()
{	
	CMyP2P client;
	if(!client.Init(0))
	{
		printf(" CP2PClient::Init() failed \n");
		return ;
	}

	// 获取服务器IP地址和用户名
	char szServerIp[20];
	char szUserName[MAX_USERNAME];
	printf(" Please input server ip: ");
	gets(szServerIp);
	printf(" Please input your name: ");
	gets(szUserName);
	// 登陆服务器
	if(!client.Login(szUserName, szServerIp))
	{
		printf(" CP2PClient::Login() failed \n");
		return ;
	}
	// 第一次登陆，首先更新用户列表
	client.GetUserList();
	// 将当前状态和本程序的用法输出给用户
	printf(" %s has successfully logined server \n", szUserName);
	printf("\n Commands are: \"getu\", \"send\", \"exit\" \n");
	// 循环处理用户命令
	char szCommandLine[256];
	while(TRUE)
	{
		gets(szCommandLine);
		if(strlen(szCommandLine) < 4)
			continue;

		// 解析出命令
		char szCommand[10];
		strncpy(szCommand, szCommandLine, 4);
		szCommand[4] = '\0';
		if(stricmp(szCommand, "getu") == 0)
		{
			// 获取用户列表
			if(client.GetUserList())
			{
				printf(" Have %d users logined server: \n", client.m_PeerList.m_nCurrentSize);
				for(int i=0; i<client.m_PeerList.m_nCurrentSize; i++)
				{
					PEER_INFO *pInfo = &client.m_PeerList.m_pPeer[i];
					printf(" Username: %s(%s:%ld) \n", pInfo->szUserName, 
						::inet_ntoa(*((in_addr*)&pInfo->addr[pInfo->AddrNum -1].dwIp)), pInfo->addr[pInfo->AddrNum - 1].nPort);
				}
			}
			else
			{
				printf(" Get User List Failure !\n");
			}
		}

		else if(stricmp(szCommand, "send") == 0)
		{
			// 解析出对方用户名
			char szPeer[MAX_USERNAME];
			for(int i=5;;i++)
			{
				if(szCommandLine[i] != ' ')
					szPeer[i-5] = szCommandLine[i];
				else
				{
					szPeer[i-5] = '\0';
					break;
				}	
			}
			
			// 解析出要发送的消息
			char szMsg[56];
			strcpy(szMsg, &szCommandLine[i+1]);

			// 发送消息
			if(client.SendText(szPeer, szMsg, strlen(szMsg)))
				printf(" Send OK! \n");
			else
				printf(" Send Failure! \n");
			
		}
		else if(stricmp(szCommand, "exit") == 0)
		{
			break;
		}
	}
}