#include <winsock2.h>
#pragma comment(lib,"WS2_32.lib")
以下实现简单的基于TCP协议的C/S程序
1. 编写服务端
主要流程:
加载ws2_32.dll ==》 创建套字节 ==》 绑定端口 ==》监听 ==》 接收/发送 ==》 关闭套字节 ==》 释放ws2_32.dll
SocketServer.cpp :
- #include "StdAfx.h"
- #include "SocketServer.h"
- using namespace std;
- /**
- * 功能描述:服务端类
- * 作者:herong
- * 日期:2011/12/06
- */
- SocketServer::SocketServer(const char* ip,const u_short port)
- {
- //用来接收winsocket实现的细节
- WSADATA wsaData;
- //初始化ws2_32.dll动态链接库
- //提供给调用方可使用的最高版本号,高位为副版本号,低位为主版本号
- WSAStartup(MAKEWORD(2,2),&wsaData);
- //创建套字节
- SOCKET s = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
- sockaddr_in sockaddr;
- sockaddr.sin_family = PF_INET;
- sockaddr.sin_addr.S_un.S_addr = inet_addr(ip);
- sockaddr.sin_port = htons(port);
- //绑定IP地址和端口
- int iResult = bind(s,(SOCKADDR*)&sockaddr,sizeof(SOCKADDR));
- if (iResult == SOCKET_ERROR) {
- //关闭套字节
- closesocket(s);
- s = INVALID_SOCKET;
- //释放ws2_32.dll动态链接库
- WSACleanup();
- throw "绑定IP地址和端口失败!";
- }
- //监听
- iResult = listen(s,1);
- if (iResult == SOCKET_ERROR) {
- //关闭套字节
- closesocket(s);
- s = INVALID_SOCKET;
- //释放ws2_32.dll动态链接库
- WSACleanup();
- throw "监听失败!";
- }
- SOCKADDR clientAddr;
- int nSize = sizeof(SOCKADDR);
- clientSock = accept(s,(SOCKADDR*)&clientAddr,&nSize);
- if (clientSock == INVALID_SOCKET) {
- //关闭套字节
- closesocket(s);
- s = INVALID_SOCKET;
- //释放ws2_32.dll动态链接库
- WSACleanup();
- throw "接受连接失败!";
- }
- cout<< "服务器已启动"<<std::endl;
- }
- SocketServer::~SocketServer(void)
- {
- //关闭客户端套字节
- closesocket(clientSock);
- //关闭套字节
- closesocket(s);
- //释放ws2_32.dll动态链接库
- WSACleanup();
- }
- int SocketServer::sendMsg(char* msg){
- return send(clientSock,msg,strlen(msg)+sizeof(char),NULL);
- }
- int SocketServer::recvMsg(string& msg) {
- int recvByte = 0;
- char recvMsg[BUF_SIZE] = {0};
- recvByte = recv(clientSock,recvMsg,BUF_SIZE,0);
- msg.append(recvMsg);
- cout<<"收到信息("<<recvByte<<"):"<< msg <<std::endl;
- return recvByte;
- }
- void SocketServer::start() {
- try {
- string msg;
- while (TRUE) {
- recvMsg(msg);
- dispacth(msg.c_str());
- msg.clear();
- }
- } catch (...) {
- cout << "出现异常";
- exit(-1);
- }
- }
- void SocketServer::dispacth(const char * msg) {
- cout << "CMD:"<<msg<<std::endl;
- if (!strcmp(msg,"exit")) {
- cout<<"客户端退出!";
- exit(0);
- }
- if (!strcmp(msg,"getInfo")) {
- char* getInfo ="查询结果!";
- sendMsg(getInfo);
- return;
- }
- char* noneInfo ="无信息!";
- sendMsg(noneInfo);
- }
2. 编写客户端程序
主要流程:
加载ws2_32.dll ==》 创建套字节 ==》 连接服务端 ==》 接收/发送 ==》 关闭套字节 ==》 释放ws2_32.dll
ClientSocket.cpp :
- #include "StdAfx.h"
- #include "ClientSocket.h"
- using namespace std;
- /**
- * 功能描述:客户端类
- * 作者:herong
- * 日期:2011/12/06
- */
- ClientSocket::ClientSocket(const char* ip,const u_short port)
- {
- //用来接收winsocket实现的细节
- WSADATA wsaData;
- //初始化ws2_32.dll动态链接库
- //提供给调用方可使用的最高版本号,高位为副版本号,低位为主版本号
- WSAStartup(MAKEWORD(2,2),&wsaData);
- //创建套字节
- server = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
- sockaddr_in sockaddr;
- sockaddr.sin_family = PF_INET;
- sockaddr.sin_addr.S_un.S_addr = inet_addr(ip);
- sockaddr.sin_port = htons(port);
- //连接服务器
- int iResult = connect(server,(SOCKADDR*)&sockaddr,sizeof(SOCKADDR));
- if (iResult == SOCKET_ERROR) {
- //关闭套字节
- closesocket(server);
- server = INVALID_SOCKET;
- //释放ws2_32.dll动态链接库
- WSACleanup();
- throw "连接服务器失败!";
- }
- }
- ClientSocket::~ClientSocket(void)
- {
- //关闭套字节
- closesocket(server);
- server = INVALID_SOCKET;
- //释放ws2_32.dll动态链接库
- WSACleanup();
- }
- int ClientSocket::sendMsg(char* msg,int len) {
- cout<<"发送信息:"<<msg<<std::endl;
- return send(server,msg,strlen(msg)+sizeof(char),0);
- //return send(server,msg,len,0);
- }
- int ClientSocket::recvMsg(string& msg) {
- int recvByte = 0;
- char recvMsg[BUF_SIZE] = {0};
- recvByte = recv(server,recvMsg,BUF_SIZE,0);
- msg.append(recvMsg);
- cout<<"收到信息("<<recvByte<< "):" << msg <<std::endl;
- return recvByte;
- }
- void ClientSocket::start(){
- try {
- char cmd[512] = {0};
- string msg;
- while (TRUE) {
- cin >> cmd;
- sendMsg(cmd,1024);
- if (!strcmp(cmd,"exit")){
- cout << "退出系统!";
- break;
- }
- recvMsg(msg);
- msg.clear();
- msg.clear();
- }
- } catch (...) {
- cout << "出现异常";
- exit(-1);
- }
- }
3. 编写服务端主程序
- // server.cpp : 定义控制台应用程序的入口点。
- //
- #include "stdafx.h"
- #include "SocketServer.h"
- int _tmain(int argc, _TCHAR* argv[])
- {
- SocketServer server("127.0.0.1",8080);
- server.start();
- return 0;
- }
4. 编写客户端主程序
- // client.cpp : 定义控制台应用程序的入口点。
- //
- #include "stdafx.h"
- #include "ClientSocket.h"
- int _tmain(int argc, _TCHAR* argv[])
- {
- ClientSocket client("127.0.0.1",8080);
- client.start();
- return 0;
- }
演示: