Socket编程+Google Protocol搭建简单的客户端服务器方式2

最新推荐文章于 2024-07-08 10:29:00 发布

杭州大擎科技有限公司

最新推荐文章于 2024-07-08 10:29:00 发布

阅读量750

点赞数

CC 4.0 BY-SA版权

分类专栏： Socket编程 google protobuf

本文链接：https://blog.youkuaiyun.com/dugaoda/article/details/50803423

google protobuf 同时被 2 个专栏收录

6 篇文章

订阅专栏

Socket编程

4 篇文章

订阅专栏

客户端模块：

1.新建一个win32控制台应用程序,是空项目,名字为MyCient

2. Google Protocol环境配置(详细配置看我文章http://blog.youkuaiyun.com/dugaoda/article/details/50342449)

2.1

2.2.添加lib文件到项目Debug下面

3.生成协议person. proto

内容为

package tutorial;

message Person {

required string name = 1;

required int32 age = 2;

optional string email = 3;

}

message msg

{

required int32 id = 1; // message ID

required int32 len = 2; //message Length

required bytes info = 3; //message content = 二进制方式.

}

得到person.pb.h和person.pb.cc

4.新建TCPPakDef.hpp。

TCPPakDef.hpp内容。buf的前8个字节的内容，包头的内容封装。

#ifndef __OMP_TCPPAKDEF_HEADER_FILE__

#define __OMP_TCPPAKDEF_HEADER_FILE__

//#include "OmpConfig.hpp"

//#include <boost/static_assert.hpp>

#include <cstdlib>

namespace omp {

namespace net {

// TCP Packet Define

#pragma pack(push,1)

struct tcp_pak_header

{

char flag[4];

int length;

inline void reset()

{

//flag[0] = 'f';

//flag[1] = 'l';

//flag[2] = 'a';

//flag[3] = 'g';

// WARNING:

// only for little-endian processor now !!!

*((int*)flag) = 0x67616c66;

length = 0;

}

bool check()

{

#ifdef _DEBUG

char str[5];

memset( str,'\0',sizeof(str) );

memcpy( str,&length,4 );

if( str[0] == '0' && ( str[1] == 'x' || str[1] == 'X' ) )

{

char *ptr_stop(0);

length = std::strtoul( str,&ptr_stop,16 );

if( ptr_stop != &str[4] )

return false;

}

#endif//#ifdef _DEBUG

// WARNING:

// only for little-endian processor now !!!

return( *((int*)flag) == 0x67616c66 && length );

}

};

#pragma pack(pop)

// BOOST_STATIC_ASSERT( sizeof(char)==1 );

// BOOST_STATIC_ASSERT( sizeof(int)==4 );

// BOOST_STATIC_ASSERT( sizeof(tcp_pak_header)==8 );

} // end of namespace net

} // end of namespace omp

#endif//#define __OMP_TCPPAKDEF_HEADER_FILE__

5.新建文件MyClient.cpp

同时添加文件person.pb.h和person.pb.cc和TCPPakDef.hpp

项目文件结构如下

6.编写MyClient.cpp文件内容

先直接贴上代码

// MyClient.cpp : 定义控制台应用程序的入口点。

#include "winsock2.h"

#include <iostream>

#include <fstream>

#include "person.pb.h"

#include "TCPPakDef.hpp"

#pragma comment(lib, "ws2_32.lib")

using namespace std;

using namespace tutorial;

BOOL RecvLine(SOCKET s, char *buf);

int main()

{

tutorial::Person PERSON;

const int BUF_SIZE = 1024;

WSADATA wsd;

SOCKET sHost;

SOCKADDR_IN servAddr;

char buf[BUF_SIZE]="";//接收数据缓冲区

char bufRecv[BUF_SIZE]="";

//截断后面的乱码输出

cout << endl << "1从服务器接受数据:" << buf;

cout << endl << "2从服务器接受数据:" << bufRecv;

int retVal;

//初始化套结字动态库

if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)

{

cout << "WSAStartup failded!" << endl;

return 1;

}

//创建套接字

sHost = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if (INVALID_SOCKET == sHost)

{

WSACleanup();//释放套接字资源

return -1;

}

//设置服务器地址

servAddr.sin_family = AF_INET;

servAddr.sin_addr.s_addr = inet_addr("127.0.0.1");

servAddr.sin_port = htons((short)123);

int nServAddlen = sizeof(servAddr);

//连接服务器

retVal = connect(sHost, (LPSOCKADDR)&servAddr, sizeof(servAddr));

if (SOCKET_ERROR == retVal)

{

cout << "connect failed" <<endl;

closesocket(sHost);

WSACleanup();

return -1;

}

while (true)

{

//向服务器发送数据

ZeroMemory(buf, BUF_SIZE);

cout << "向服务器发送数据:";

char buf123[BUF_SIZE]="";

cin >> buf123;

//使用google protobuf 将数据发送出去

//其实在每个MSG的头上面加上ID 在服务器上面解析出来就可以用了

//PERSON.set_name("dugoada");

//PERSON.set_age(123);

//string data;

//PERSON.SerializeToString(&data);

//char dst[BUF_SIZE];

//strcpy(dst, data.c_str());

//retVal = send(sHost, dst, sizeof(dst), 0);

////*方式2 发送数据加上包的编号*//

//新的

google::protobuf::Message* user_send_msg;

PERSON.set_name("dugoada");

PERSON.set_age(123);

user_send_msg = &PERSON;

tutorial::msg head;// 包的ID 包的长度为Msg的ByteSize

head.set_id(1);

int nLength = 0;

if ( user_send_msg ) nLength = user_send_msg->ByteSize();

head.set_len(nLength);

if (nLength > 0)

{

//序列化之后的字符串将数据存入到 head.info();

user_send_msg->SerializeToString(head.mutable_info());

}

//将数据组装到head完毕

//设置tcp的内容

omp::net::tcp_pak_header tcp = {0};

int tcplength = sizeof(tcp);//tcplength = 8;

int nLen = head.ByteSize() + sizeof(tcp);//包头的长度+message的size

tcp.length = htonl(nLen);

//设置 tcp.length的值

//将tcp首地址开始到sizeof(tcp)的长度将数据存入到Buf中处理Buff前8个长度

::memcpy(buf, &tcp, sizeof(tcp));

//进行序列化,将head中的数据全部序列化到buf+sizeof(tcp)中处理Buff从8开始到最后

head.SerializeToArray(buf+sizeof(tcp), nLen);

retVal = send(sHost, buf, nLen, 0);

////buf nLen

////*发送数据加上包的编号*//

//retVal = send(sHost, buf, strlen(buf), 0);

if (SOCKET_ERROR == retVal)

{

cout << "send failed!" << endl;

closesocket(sHost);//关闭套接字

WSACleanup();

return -1;

}

// 接收服务器端的数据

ZeroMemory(buf, BUF_SIZE);

recv(sHost, bufRecv, BUF_SIZE, 0);

cout << endl << "从服务器接受数据:" << bufRecv;

}

closesocket(sHost);//关闭套接字

WSACleanup();

return 0;

}

7.关键代码说明

////*方式2 发送数据加上包的编号*//

//新的

google::protobuf::Message* user_send_msg;

PERSON.set_name("dugoada");

PERSON.set_age(123);

user_send_msg = &PERSON;

tutorial::msg head;// 包的ID 包的长度为Msg的ByteSize

head.set_id(1);

int nLength = 0;

if ( user_send_msg ) nLength = user_send_msg->ByteSize();

head.set_len(nLength);

if (nLength > 0)

{

//序列化之后的字符串将数据存入到 head.info();

user_send_msg->SerializeToString(head.mutable_info());

}

//将数据组装到head完毕

//设置tcp的内容

omp::net::tcp_pak_header tcp = {0};

int tcplength = sizeof(tcp);//tcplength = 8;

int nLen = head.ByteSize() + sizeof(tcp);//包头的长度+message的size

tcp.length = htonl(nLen);

//设置 tcp.length的值

//将tcp首地址开始到sizeof(tcp)的长度将数据存入到Buf中处理Buff前8个长度

::memcpy(buf, &tcp, sizeof(tcp));

//进行序列化,将head中的数据全部序列化到buf+sizeof(tcp)中处理Buff从8开始到最后

head.SerializeToArray(buf+sizeof(tcp), nLen);

retVal = send(sHost, buf, nLen, 0);

////buf nLen

////*发送数据加上包的编号*//

B 服务器模块：

环境配置部分跟客户端一样，代码不一样（注意）

1.编写MyServer.cpp文件内容

先直接贴上代码

// MyServer.cpp : 定义控制台应用程序的入口点。

//#include "stdafx.h"

#include "winsock2.h"

#pragma comment(lib, "ws2_32.lib")

#include <iostream>

using namespace std;

#include <iostream>

#include<fstream>

#include "TCPPakDef.hpp"

#include "person.pb.h"

using namespace tutorial;

int main()

{

tutorial::Person PERSON;

const int BUF_SIZE = 1024;

WSADATA wsd;

SOCKET sServer;

SOCKET sClient;

SOCKADDR_IN addrServ;

char buf[BUF_SIZE];

char sendBuf[BUF_SIZE];

int retVal;

//初始化套结字动态库

if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)

{

cout << "WSAStartup failded!" << endl;

return 1;

}

//创建套接字

sServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

if (INVALID_SOCKET == sServer)

{

cout << "socket failed!" << endl;

WSACleanup();

return -1;

}

//服务器套接字地址信息设置

addrServ.sin_family = AF_INET;

addrServ.sin_port = htons(123);

addrServ.sin_addr.s_addr = INADDR_ANY;

//绑定

retVal = bind(sServer, (LPSOCKADDR)&addrServ, sizeof(SOCKADDR_IN));

if (SOCKET_ERROR == retVal)

{

cout << "bind failed!" << endl;

closesocket(sServer);

WSACleanup();

return -1;

}

//开始监听

retVal = listen(sServer, 1);

if (SOCKET_ERROR == retVal)

{

cout << "listen failed!" << endl;

closesocket(sServer);

WSACleanup();

return -1;

}

//接受客户端请求

sockaddr_in addrClient;

int addrClientlen = sizeof(addrClient);

sClient = accept(sServer, (sockaddr FAR *)&addrClient, &addrClientlen);

if (INVALID_SOCKET == sClient)

{

cout << "accept failed!" << endl;

closesocket(sServer);

WSACleanup();

return -1;

}

//数据处理

while (true)

{

//接受客户端数据

ZeroMemory(buf, BUF_SIZE);

retVal = recv(sClient, buf, BUF_SIZE, 0);

//方法1 解析google protobuf

//string data = buf;

//PERSON.ParseFromString(data);

//cout << "Name: " << PERSON.name() << endl;

//cout << "Age: " << PERSON.age() << endl;

int nHeadLength = sizeof(omp::net::tcp_pak_header);

//char receveBuf[BUF_SIZE];

omp::net::tcp_pak_header* pTcpMsg = (omp::net::tcp_pak_header*)buf;

unsigned short wAllSize = ntohl(pTcpMsg->length);

int nPacketSize = wAllSize - nHeadLength;//得到 tutorial::msg 的ByteSize()

tutorial::msg head;

head.ParseFromArray(buf+nHeadLength, nPacketSize);

unsigned char cbDataBuffer[1024+20];

//将head的数据放入到buf中

::memcpy(cbDataBuffer, head.info().data(), head.info().length());

//::memmove(buf, buf+nPacketSize, m_wRecvSize);

int sfde = head.id();

cbDataBuffer;

nPacketSize;

tutorial::Person PP;

PP.ParseFromArray(cbDataBuffer, nPacketSize);

string sfe = PP.name();

int asfewe = PP.age();

cout << "名字:" << PP.name() << endl;

cout << "年龄:" << PP.age() << endl;

if (SOCKET_ERROR == retVal)

{

cout << "recv, failed!" << endl;

closesocket(sClient);

closesocket(sServer);

WSACleanup();

return -1;

}

if (buf[0] == '0')

break;

cout << "客户端发送的数据:" << buf << endl;

cout << "向客户端发送数据:";

cin >> sendBuf;

send(sClient, sendBuf, strlen(sendBuf), 0);

}

closesocket(sClient);

closesocket(sServer);

WSACleanup();

return 0;

}

6.关键代码说明

retVal = recv(sClient, buf, BUF_SIZE, 0);

//方法1 解析google protobuf

//string data = buf;

//PERSON.ParseFromString(data);

//cout << "Name: " << PERSON.name() << endl;

//cout << "Age: " << PERSON.age() << endl;

int nHeadLength = sizeof(omp::net::tcp_pak_header);

//char receveBuf[BUF_SIZE];

omp::net::tcp_pak_header* pTcpMsg = (omp::net::tcp_pak_header*)buf;

unsigned short wAllSize = ntohl(pTcpMsg->length);

int nPacketSize = wAllSize - nHeadLength;//得到 tutorial::msg 的ByteSize()

tutorial::msg head;

head.ParseFromArray(buf+nHeadLength, nPacketSize);

unsigned char cbDataBuffer[1024+20];

//将head的数据放入到buf中

::memcpy(cbDataBuffer, head.info().data(), head.info().length());

//::memmove(buf, buf+nPacketSize, m_wRecvSize);

int sfde = head.id();

cbDataBuffer;

nPacketSize;

tutorial::Person PP;

PP.ParseFromArray(cbDataBuffer, nPacketSize);

string sfe = PP.name();

int asfewe = PP.age();

cout << "名字:" << PP.name() << endl;

cout << "年龄:" << PP.age() << endl;