1、protobuf 定义的消息文件
enum TestType
{
Enum1 = 1;
Enum2 = 2;
Enum3 = 3;
};
message STValue
{
optional bool bValue =1;
optional double dValue =2;
optional float fValue =3;
optional int32 iValue =4;
optional TestType emValue =5;
};
message STValueABC
{
optional uint32 ulId = 1; //ID??
optional string strTime = 2; //??
repeated uint32 ulParas = 3; //??
optional STValue stValue =4;
optional bytes bsValue =5;
};
2、protobuf打包和解包的类。
struct ProbufMsg
{
int leng;//总长度
int nameleng;
char *messagename;
char *protobufdata;
};
struct SOCKE_PAG
{
int MSGtype;
int bufLeng;
char *buf;
};
class CDealMsg
{
public:
CDealMsg();
virtual ~CDealMsg();
bool RecvMsg(SOCKE_PAG &PagMsg);
bool RealeaseMem();//释放内存
Message *GetMessage(){ return m_pMessage; }
unsigned int GetMsgType(){ return m_ulMsgType; }
SOCKE_PAG &PackMsg(Message *pPara);
Message * _createMsg(const string & typeName);
private:
bool ParseRecvMsg(ProbufMsg *pMsg); //解析响应消息
Message *m_pMessage; //消息内容指针
unsigned int m_ulMsgType; //消息类型
SOCKE_PAG m_PagMsg;
ProbufMsg m_ProMsg;
};
CDealMsg::CDealMsg()
{
}
CDealMsg::~CDealMsg()
{
}
bool CDealMsg::ParseRecvMsg(ProbufMsg *pMsg)
{
string strName(pMsg->messagename, pMsg->nameleng);
m_pMessage = _createMsg(strName);
if (!m_pMessage->ParseFromArray(pMsg->protobufdata, pMsg->leng - pMsg->nameleng - 8))
{
return false;
}
return true;
}
Message * CDealMsg::_createMsg(const string & typeName)
{
Message * pMsg = NULL;
const Descriptor * pDescriptor = DescriptorPool::generated_pool()->FindMessageTypeByName(typeName);
if (pDescriptor)
{
const Message * pPrototype = MessageFactory::generated_factory()->GetPrototype(pDescriptor);
if (pPrototype)
{
pMsg = pPrototype->New();
}
}
return pMsg;
}
SOCKE_PAG & CDealMsg::PackMsg(Message *pPara)
{
ProbufMsg sendMsg = { 0 };
sendMsg.nameleng = pPara->GetTypeName().size();
int probufLen = pPara->ByteSize();
sendMsg.protobufdata = new char[probufLen];
pPara->SerializeToArray(sendMsg.protobufdata, probufLen);
sendMsg.messagename = new char[sendMsg.nameleng];
memcpy(sendMsg.messagename, pPara->GetTypeName().c_str(), sendMsg.nameleng);
sendMsg.leng = probufLen + sendMsg.nameleng + 8;
m_PagMsg.MSGtype = 123456;
m_PagMsg.bufLeng = sendMsg.leng;
m_PagMsg.buf = new char[m_PagMsg.bufLeng];
memcpy(m_PagMsg.buf, &sendMsg, 8);
memcpy(m_PagMsg.buf + 8, sendMsg.messagename, sendMsg.nameleng);
memcpy(m_PagMsg.buf + 8 + sendMsg.nameleng, sendMsg.protobufdata, probufLen);
delete[]sendMsg.protobufdata;
delete[]sendMsg.messagename;
sendMsg.protobufdata = nullptr;
sendMsg.messagename = nullptr;
return m_PagMsg;
}
bool CDealMsg::RecvMsg(SOCKE_PAG &PagMsg)
{
// ProbufMsg ProMsg = { 0 };
memcpy((char*)&m_ProMsg, PagMsg.buf, 8);
m_ProMsg.messagename = new char[m_ProMsg.nameleng];
int protoLen = m_ProMsg.leng - m_ProMsg.nameleng - 8;
m_ProMsg.protobufdata = new char[protoLen];
memcpy(m_ProMsg.messagename, PagMsg.buf + 8, m_ProMsg.nameleng);
memcpy(m_ProMsg.protobufdata, PagMsg.buf + m_ProMsg.nameleng + 8, protoLen);
ParseRecvMsg(&m_ProMsg);
return true;
}
bool CDealMsg::RealeaseMem()
{
delete[] m_ProMsg.messagename;
delete[] m_ProMsg.protobufdata;
return true;
}
3、Server服务端序列化和打包。
int main()
{
int iPort = 66666;
WSADATA wsaData;
SOCKET sListen, sAccept;
int iLen; //客户地址长度
int iSend;//发送数据长度
char buf[] = "I am a server";//要发送给客户的信息
struct sockaddr_in ser, cli;//服务器和客户的地址
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
{
printf("Failed to load Winsock./n");
return -1;
}
sListen = socket(AF_INET, SOCK_STREAM, 0);//创建服务器端套接口
if (sListen == INVALID_SOCKET)
{
printf("socket() Failed: %d/n", WSAGetLastError());
return -1;
}
//以下建立服务器端地址
//使用IP地址族
ser.sin_family = AF_INET;
//使用htons()把双字节主机序端口号转换为网络字节序端口号
ser.sin_port = htons(iPort);
//htonl()把一个四字节主机序IP地址转换为网络字节序主机地址
//使用系统指定的IP地址INADDR_ANY
ser.sin_addr.s_addr = htonl(INADDR_ANY);
//bind()函数进行套接定与地址的绑定
if (bind(sListen, (LPSOCKADDR)&ser, sizeof(ser)) == SOCKET_ERROR)
{
printf("bind() Failed: %d/n", WSAGetLastError());
return -1;
}
//进入监听状态
if (listen(sListen, 5) == SOCKET_ERROR)
{
printf("lisiten() Failed: %d/n", WSAGetLastError());
return -1;
}
//初始化客户地址长度参数
iLen = sizeof(cli);
STValueABC test;
test.set_ulid(122);
test.set_strtime("adfdac");
test.add_ulparas(1);
test.add_ulparas(2);
test.set_bsvalue("我发的是双字节中文字符");
STValue *pst = test.mutable_stvalue();
pst->set_bvalue(false);
pst->set_dvalue(123232.45454);
pst->set_fvalue(32.123);
pst->set_ivalue(11111);
pst->set_emvalue(Urgent);
CDealMsg RecvMsg;
SOCKE_PAG &PagMsg = RecvMsg.PackMsg(&test);
sAccept = accept(sListen, (struct sockaddr *)&cli, &iLen);
if (sAccept == INVALID_SOCKET)
{
printf("accept() Failed: %d/n", WSAGetLastError());
return -1;
}
iSend = send(sAccept, (char*)&PagMsg, 8, 0);
iSend = send(sAccept, (char*)PagMsg.buf, PagMsg.bufLeng, 0);
closesocket(sAccept);
closesocket(sListen);
WSACleanup();
delete[] PagMsg.buf;
return 0;
}
4、client客户端接收解包和反序列化。
int main()
{
WSADATA wsaData;
SOCKET sClient;
int iPort = 66666;
int iLen;//从服务器端接收的数据长度
char buf[1024];//接收数据的缓冲区
struct sockaddr_in ser;//服务器端地址
memset(buf, 0, sizeof(buf));//接收缓冲区初始化
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
{
printf("Failed to load Winsock./n");
return -1;
}
//填写要连接的服务器地址信息
ser.sin_family = AF_INET;
ser.sin_port = htons(iPort);
//inet_addr()将命令行中输入的点分IP地址转换为二进制表示的网络字节序IP地址
ser.sin_addr.s_addr = inet_addr("127.0.0.1");
//建立客户端流式套接口
sClient = socket(AF_INET, SOCK_STREAM, 0);
if (sClient == INVALID_SOCKET)
{
printf("socket() Failed: %d/n", WSAGetLastError());
return -1;
}
SOCKE_PAG PagMsg = { 0 };
CDealMsg reMsg;
//ProbufMsg ProMsg = {0};
//请求与服务器端建立TCP连接
if (connect(sClient, (struct sockaddr *)&ser, sizeof(ser)) == INVALID_SOCKET)
{
printf("connect() Failed: %d/n", WSAGetLastError());
return -1;
}
iLen = recv(sClient, (char *)&PagMsg, 8, 0);
if (iLen == SOCKET_ERROR)
{
printf("recv() Failed: %d/n", WSAGetLastError());
return -1;
}
if (PagMsg.MSGtype == 123456)
{
PagMsg.buf = new char[PagMsg.bufLeng];
iLen = recv(sClient, (char *)PagMsg.buf, PagMsg.bufLeng, 0);
reMsg.RecvMsg(PagMsg);
Message *msg = reMsg.GetMessage();
//-----------------------------------------
STValueABC *rTest = (STValueABC *)msg;
int ulid = rTest->ulid();
const string &str = rTest->strtime();
for (int i = 0; i < rTest->ulparas_size(); i++)
{
int paras = rTest->ulparas(i);
}
const string & cstr = rTest->bsvalue();
STValue *pst = rTest->mutable_stvalue();
bool bvalue = pst->bvalue();
double dvalue = pst->dvalue();
/*
STValueABC test;
test.set_ulid(122);
test.set_strtime("adfdac");
test.add_ulparas(1);
test.add_ulparas(2);
test.set_bsvalue("我发的是双字节中文字符");
STValue *pst = test.mutable_stvalue();
pst->set_bvalue(false);
pst->set_dvalue(123232.45454);
pst->set_fvalue(32.123);
pst->set_ivalue(11111);
pst->set_emvalue(Urgent);
*/
reMsg.RealeaseMem();
delete[] PagMsg.buf;
}
closesocket(sClient);
WSACleanup();
}