Protobuf C++ 简单例子

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();

}