操作系统与网络 2019-4-15

csdn[-1]

于 2019-04-18 23:50:09 发布

阅读量184

点赞数

分类专栏： C/C++ 文章标签：直播项目

本文链接：https://blog.csdn.net/weixin_42896619/article/details/89390904

版权

C/C++ 专栏收录该内容

140 篇文章 1 订阅

订阅专栏

1.在数据库中存储信息

1.1 添加三个表

1.author：用来存放主播的id、登陆密码、用户名；
2.audience：用来存放观众的id、登陆密码、用户名；
3.room：用来存放房间的id、房间名name、对应的主播；

1.2 在MySQL中创建上面三个表

1.新建author表，添加 a_id 属性，类型为BIGINT(11)，选择前两个选项 PK NN ；添加 a_name 属性，选择 NN ；添加 a_password 属性，选择 NN ；
2.新建audience表，添加 a_id 属性，类型为BIGINT(11)，选择前两个选项 PK NN ；添加 a_name 属性，选择 NN ；添加 a_password 属性，选择 NN ；
3.新建room表，添加 r_id 属性，选择选项 PK NN AI ；添加 r_name 属性，选择 NN ；添加 a_id 属性，类型为BIGINT(11)，选择 NN ；

1.3 在PackDef.h中定义协议

1.定义基础约定： _DEF_PROTOCOL_BASE 100 ；
2.定义注册请求协议与注册回复协议： _DEF_PROTOCOL_REGISTER_RQ _DEF_PROTOCOL_REGISTER_RS 分别为 _DEF_PROTOCOL_BASE+1 _DEF_PROTOCOL_BASE+2 ；
3.定义登陆请求协议与登陆回复协议： _DEF_PROTOCOL_LOGIN_RQ _DEF_PROTOCOL_LOGIN_RS 分别为 _DEF_PROTOCOL_BASE+3 _DEF_PROTOCOL_BASE+4 ；
4.定义获取房间信息请求协议与获取房间信息回复协议： _DEF_PROTOCOL_GETROOMINFO_RQ _DEF_PROTOCOL_GETROOMINFO_RS 分别为 _DEF_PROTOCOL_BASE+5 _DEF_PROTOCOL_BASE+6 ；
5.定义设置房间信息请求协议与设置房间信息回复协议： _DEF_PROTOCOL_SETROOMINFO_RQ _DEF_PROTOCOL_SETROOMINFO_RS 分别为 _DEF_PROTOCOL_BASE+7 _DEF_PROTOCOL_BASE+8 ；
6.定义开始推送视频流请求协议与开始推送视频流回复协议： _DEF_PROTOCOL_STARTTRANSFER_RQ _DEF_PROTOCOL_STARTTRANSFER_RS 分别为 _DEF_PROTOCOL_BASE+9 _DEF_PROTOCOL_BASE+10 ；
7.定义停止推送视频流请求协议与停止推送视频流回复协议： _DEF_PROTOCOL_STOPTRANSFER_RQ _DEF_PROTOCOL_STOPTRANSFER_RS 分别为 _DEF_PROTOCOL_BASE+11 _DEF_PROTOCOL_BASE+12 ；
8.定义数据流信息请求协议与数据流信息回复协议： _DEF_PROTOCOL_STREAMINFO_RQ _DEF_PROTOCOL_STREAMINFO_RS 分别为 _DEF_PROTOCOL_BASE+13 _DEF_PROTOCOL_BASE+14 ；
9.定义数据流链接请求协议与数据流连接回复协议： _DEF_PROTOCOL_STREAMCONNECT_RQ _DEF_PROTOCOL_STREAMCONNECT_RS 分别为 _DEF_PROTOCOL_BASE+15 _DEF_PROTOCOL_BASE+16 ；
10.定义观众获取主播列表请求协议与获取主播回复协议： _DEF_PROTOCOL_GETAUTHORLIST_RQ _DEF_PROTOCOL_GETAUTHORLIST_RS 分别为 _DEF_PROTOCOL_BASE+17 _DEF_PROTOCOL_BASE+18 ；
11.定义观众选择主播请求协议与选择主播回复协议： _DEF_PROTOCOL_SELECTAUTHOR_RQ _DEF_PROTOCOL_SELECTAUTHOR_RS 分别为 _DEF_PROTOCOL_BASE+19 _DEF_PROTOCOL_BASE+20 ；
12.定义退出房间请求协议与退出房间回复协议： _DEF_PROTOCOL_QUITAUTHOR_RQ _DEF_PROTOCOL_QUITAUTHOR_RS 分别为 _DEF_PROTOCOL_BASE+21 _DEF_PROTOCOL_BASE+22 ；

1.4 定义协议包

1.首先定义包的类型： typedef char PackType
2.定义边界值： DEF_SIZE 45
3.定义注册请求协议包：由于注册协议报与请求协议包都需要协议类型、id、用户名、密码以及是主播还是观众的标志，因此将二者合为一个结构体；

typedef struct STRU_RESIGER_RQ
{
	PackType	m_n_type;
	long long	m_id;
	char		m_sz_name[DEF_SIZE];
	char		m_sz_password[DEF_SIZE];
	char		m_sz_role;		// 用来标记是主播还是观众
}STRU_LOGIN_RQ;

4.定义注册回复协议包：由于注册回复包与登陆回复包都需要协议类型、是否成功的标志，因此将二者合为一个结构体；

typedef struct STRU_RESISTER_RS
{
	PackType	m_n_type;
	char		m_sz_result;	// 是否成功(注册或登陆是否成功)
}STRU_LOGIN_RS;

5.定义获取房间信息请求包：需要包的协议、房间名；

struct STRU_GETROOMINFO_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
};

6.定义获取房间信息回复包：需要包的协议类型、房间号、房间名；

struct STRU_GETROOMINFO_RS
{
	PackType	m_nType;
	long long	m_roomid;
	char		m_szRoomName[DEF_SIZE];
};

7.定义设置房间请求包：需要协议的类型、主播名、房间名；

struct STRU_SETROOMINFO_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
	char		m_szRoomName[DEF_SIZE];
};

8.定义设置房间回复包：需要协议的类型、是否设置成功的标志；

struct STRU_SETROOMINFO_RS
{
	PackType	m_nType;
	char		m_szResult;
};

9.定义开始发送流请求包：需要协议的类型、主播名；

struct STRU_STARTTRANSFER_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
};

10.定义开始发送流回复包：需要协议的类型、是否成功的标志；

struct STRU_STARTTRANSFER_RS
{
	PackType	m_nType;
	char		m_szResult;
};

11.定义获取主播列表的请求包：需要协议的类型、主播名；

struct STRU_GETAUHTORLIST_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
};

12.定义包含主播信息的结构体：需要主播名、房间名、房间id；

struct AuthorInfo
{
	char		m_szName[DEF_SIZE];
	char		m_szRoomName[DEF_SIZE];
	long long	m_roomid;
};

13.定义获取主播列表的回复包：需要协议类型、主播信息数组(默认一次刷新10个)、主播的个数；

struct STRU_GETAUHTORLIST_RS
{
	PackType	m_nType;
	AuthorInfo	m_aryinfo[DEF_NUM];
	int			m_Num;
};

14.定义选择主播请求包：需要协议类型、主播名、房间id；

struct STRU_SELECTAUHTORLIST_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
	long long	m_roomid;
};

15.定义主播列表回复包：需要协议类型、是否成功的标志；

struct STRU_SELECTAUHTORLIST_RS
{
	PackType	m_nType;
	char		m_szResult;
};

16.完整代码

#ifndef _PACKDEF_H
#define _PACKDEF_H

#define _DEFPORT		1234
#define _DEFSIZE		1024

// 协议
#define _DEF_PROTOCOL_BASE						100
// 注册
#define _DEF_PROTOCOL_REGISTER_RQ				_DEF_PROTOCOL_BASE + 1
#define _DEF_PROTOCOL_REGISTER_RS				_DEF_PROTOCOL_BASE + 2
// 登陆
#define _DEF_PROTOCOL_LOGIN_RQ					_DEF_PROTOCOL_BASE + 3
#define _DEF_PROTOCOL_LOGIN_RS					_DEF_PROTOCOL_BASE + 4
// 获取房间信息
#define _DEF_PROTOCOL_GETROOMINFO_RQ			_DEF_PROTOCOL_BASE + 5
#define _DEF_PROTOCOL_GETROOMINFO_RS			_DEF_PROTOCOL_BASE + 6
// 设置房间信息
#define _DEF_PROTOCOL_SETROOMINFO_RQ			_DEF_PROTOCOL_BASE + 7
#define _DEF_PROTOCOL_SETROOMINFO_RS			_DEF_PROTOCOL_BASE + 8
// 推视频流
#define _DEF_PROTOCOL_STARTTRANSFER_RQ			_DEF_PROTOCOL_BASE + 9
#define _DEF_PROTOCOL_STARTTRANSFER_RS			_DEF_PROTOCOL_BASE + 10
// 停止推送视频流
#define _DEF_PROTOCOL_STOPTRANSFER_RQ			_DEF_PROTOCOL_BASE + 11
#define _DEF_PROTOCOL_STOPTRANSFER_RS			_DEF_PROTOCOL_BASE + 12
// 流信息
#define _DEF_PROTOCOL_STREAMINFO_RQ				_DEF_PROTOCOL_BASE + 13
#define _DEF_PROTOCOL_STREAMINFO_RS				_DEF_PROTOCOL_BASE + 14
// 流连接
#define _DEF_PROTOCOL_STREAMCONNECT_RQ			_DEF_PROTOCOL_BASE + 15
#define _DEF_PROTOCOL_STREAMCONNECT_RS			_DEF_PROTOCOL_BASE + 16

// 观众
#define _DEF_PROTOCOL_GETAUTHORLIST_RQ			_DEF_PROTOCOL_BASE + 17
#define _DEF_PROTOCOL_GETAUTHORLIST_RS			_DEF_PROTOCOL_BASE + 18
// 选择主播
#define _DEF_PROTOCOL_SELECTAUTHOR_RQ			_DEF_PROTOCOL_BASE + 19
#define _DEF_PROTOCOL_SELECTAUTHOR_RS			_DEF_PROTOCOL_BASE + 20
// 退出房间
#define _DEF_PROTOCOL_QUITAUTHOR_RQ				_DEF_PROTOCOL_BASE + 21
#define _DEF_PROTOCOL_QUITAUTHOR_RS				_DEF_PROTOCOL_BASE + 22

// 定义协议类型
typedef char PackType;

// 定义边界值
#define DEF_SIZE	45
#define DEF_NUM		10

// 定义协议包
// 注册(登陆)请求包
typedef struct STRU_RESIGER_RQ
{
	PackType	m_n_type;
	long long	m_id;
	char		m_sz_name[DEF_SIZE];
	char		m_sz_password[DEF_SIZE];
	char		m_sz_role;		// 用来标记是主播还是观众
}STRU_LOGIN_RQ;

// 注册(登陆)回复包
typedef struct STRU_RESISTER_RS
{
	PackType	m_n_type;
	char		m_sz_result;	// 是否成功(注册或登陆是否成功)
}STRU_LOGIN_RS;

// 获取房间信息请求包
struct STRU_GETROOMINFO_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
};

// 获取房间信息回复包
struct STRU_GETROOMINFO_RS
{
	PackType	m_nType;
	long long	m_roomid;
	char		m_szRoomName[DEF_SIZE];
};

// 设置房间信息请求包
struct STRU_SETROOMINFO_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
	char		m_szRoomName[DEF_SIZE];
};

// 设置房间信息回复包
struct STRU_SETROOMINFO_RS
{
	PackType	m_nType;
	char		m_szResult;
};

// 开始发送流请求包
struct STRU_STARTTRANSFER_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
};

// 开始发送流回复包
struct STRU_STARTTRANSFER_RS
{
	PackType	m_nType;
	char		m_szResult;
};

// 观众
// 获取主播列表请求包
struct STRU_GETAUHTORLIST_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
};

// 主播信息的结构体
struct AuthorInfo
{
	char		m_szName[DEF_SIZE];
	char		m_szRoomName[DEF_SIZE];
	long long	m_roomid;
};

// 获取主播列表回复包
struct STRU_GETAUHTORLIST_RS
{
	PackType	m_nType;
	AuthorInfo	m_aryinfo[DEF_NUM];
	int			m_Num;
};

// 选择主播列表请求包
struct STRU_SELECTAUHTORLIST_RQ
{
	PackType	m_nType;
	char		m_szName[DEF_SIZE];
	long long	m_roomid;
};

// 选择主播列表回复包
struct STRU_SELECTAUHTORLIST_RS
{
	PackType	m_nType;
	char		m_szResult;
};

#endif // _PACKDEF_H

1.5 在TCPNet.cpp文件中的线程接收数据函数 ThreadRecv 中接收数据包

1.定义等待的套接字；定义真正读取到的数据的大小；接受到包的大小；存储内容的空间；
2.循环接受数据包中的内容，直到数据包被解析完；

unsigned _stdcall TCPNet::ThreadRecv( void * lpvoid)
{
	SOCKET sockWaiter = (SOCKET)lpvoid;
	char szbuf[_DEFSIZE] = {0};
	
	int nRelReadNum = 0;
	int n_pack_size;
	char *p_sz_buf = NULL;

	while(TCPNet::m_bFlagQuit)
	{
		// 接受包的大小
		nRelReadNum = recv(sockWaiter, (char*)&n_pack_size, sizeof(int),0);
		if(nRelReadNum <=0)
		{
			//是不是客户端下载
			if(WSAECONNRESET  == GetLastError())
			{
				closesocket(sockWaiter);
				sockWaiter = NULL;
				break;
			}
			continue;
		}

		// 接收完整的数据包
		p_sz_buf = new char[n_pack_size];
		int noffset = 0;
		while(n_pack_size)
		{
			nRelReadNum = recv(sockWaiter, p_sz_buf+noffset, n_pack_size, 0);
			noffset += nRelReadNum;
			n_pack_size -= nRelReadNum;
		}

		// 处理
		m_p_kernel->DealData(sockWaiter, p_sz_buf);

		delete []p_sz_buf;
		p_sz_buf = NULL;
	}

	return 0;
}

3.其中需要注意的就是我们将 m_p_kernel 定义为了静态变量，因此我们需要在类外进行初始化；
4.其次，我们将 TCPNet 更改为了带参的构造函数，因此在给 m_p_kernel 赋值时需要注意；

IKernel *TCPNet::m_p_kernel = NULL;
TCPNet::TCPNet(IKernel* p_kernel)
{
	m_sockListen = NULL;
	m_p_kernel = p_kernel;
}

1.6 在 TCPKernel 类中需要处理数据包，但是直接通过switch或if语句来选择对应的处理方式太过麻烦，我们考虑使用映射关系来解决这个问题

1.在 TCPKernel.h 中定义一个函数指针，以及一个用来装协议宏和对应函数指针的结构体；

typedef bool (TCPKernel::*PFUN)(SOCKET, char*);
struct ProtocolMap
{
	PackType	m_n_type;		// 协议
	PFUN		m_p_fun;		// 协议对应的函数地址
};

2.在 TCPKernel.cpp 文件中定义这个结构体，用来作为映射表；

ProtocolMap m_protocol_entries[] = 
{
	{_DEF_PROTOCOL_REGISTER_RQ, &TCPKernel::RegisterRq},
	{0, 0}
};

3.上面的映射表中是注册时的数据协议以及其对应的处理函数的地址；{0, 0}是映射表结束的标志(作为遍历结构体终止的标志)；
4.添加一个新的函数： bool RegisterRq(SOCKET, char*) ；
5.在 DealData 函数中遍历映射表

bool TCPKernel::DealData(SOCKET sock,char* szbuf)
{
	//处理协议 
	char* p_type = (char*)szbuf;
	int i = 0;
	while (1)
	{
		if(m_protocol_entries[i].m_n_type == *p_type)
		{
			(this->*m_protocol_entries[i].m_p_fun)(sock, szbuf);
			break;
		}
		else if(m_protocol_entries[i].m_n_type == 0 && m_protocol_entries[i].m_p_fun == 0)
			break;

		i++;
	}

	return true;
}

6.在添加一个登陆的函数 bool LoginRq(SOCKET sock,char* szbuf) ；

ProtocolMap m_protocol_entries[] = 
{
	{_DEF_PROTOCOL_REGISTER_RQ, &TCPKernel::RegisterRq},
	{_DEF_PROTOCOL_LOGIN_RQ, &TCPKernel::LoginRq},
	{0, 0}
};

bool TCPKernel::LoginRq(SOCKET sock,char* szbuf)
{
	return false;
}

csdn[-1]

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
操作系统与网络 2019-4-15

1.在数据库中存储信息1.1 添加三个表1.author：用来存放主播的id、登陆密码、用户名；2.audience：用来存放观众的id、登陆密码、用户名；3.room：用来存放房间的id、房间名name、对应的主播；1.2 在MySQL中创建上面三个表1.新建author表，添加 a_id 属性，类型为BIGINT(11)，选择前两个选项 PK NN ；添加 a_name 属性...
复制链接

扫一扫