C++ 实现简单Tcp服务器端 -- Select方式

最新推荐文章于 2024-07-08 09:58:19 发布

WuZePei

最新推荐文章于 2024-07-08 09:58:19 发布

阅读量514

点赞数

分类专栏： c++学习

本文链接：https://blog.csdn.net/weixin_38425390/article/details/107714802

版权

本文档包含四个部分：test_server.h、test_server.cpp、test_client.h和test_client.cpp，详细介绍了使用C++编程语言实现基于Select模型的简单TCP服务器端的代码实现。

摘要由CSDN通过智能技术生成

test_server.h

#ifndef INCLUDE_TEST_SERVER_H
#define INCLUDE_TEST_SERVER_H

#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <string.h>
#include <vector>

#include <winsock2.h>
#include <Ws2tcpip.h>
#pragma comment (lib, "ws2_32.lib") 

#include <fstream>
#include <iostream>
#include <time.h>

using namespace std;

#define SIZE 60
#define FILENAME "setting.txt"
#define OUTPUTFILENAME "output.txt"

class testServer
{
   
public:
	testServer(void);
	~testServer(void);

	bool loadSocketLib(void);
	bool readSocketParam(std::string fileName);
	bool listenClient(void);
	bool selectSocket(void);
	bool disconnectListen(void);
	bool receiveData(void);
	bool outputDataProcessing(void);
	bool writeFile(std::string outPutFileName);


protected:
	SOCKET  s_socket_fd;
	SOCKET new_socket_fd;
	std::string ipAddr;
	uint16_t portNum;
	char r_buf[SIZE];
	std::string oustr;
	fd_set readSet, tmpSet; 

};

#endif /*INCLUDE_TEST_SERVER_H*/

test_server.cpp

#include "test_server.h"

using namespace std;

/*
 *  @brief Contructor
 */
testServer::testServer(void)
{
   
	s_socket_fd = INVALID_SOCKET;
	new_socket_fd = INVALID_SOCKET;
	ipAddr = "";
	portNum = 0;
	memset(r_buf, 0, (size_t)SIZE);
	oustr = "";
	FD_ZERO(&tmpSet);
	FD_ZERO(&readSet);
}

/*
 *  @brief Destructor
 */
testServer::~testServer(void) {
   }

/*
 *  @brief Start SOCKET library
 *  @return[in] result (ture:success  false:failure)
 */
bool testServer::loadSocketLib(void)
{
   
	WSADATA wsaData;
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) 
	{
   
		cout << "startSOCKETlib error : error = start Socket lib failure" << endl;
		return false;
	}
	return true;
}

/*
 *  @brief read setting file(load socket param)
 *  @param [in] FileName
 *  @return[in] result (ture:success  false:failure)
 */
bool testServer::readSocketParam(std::string FileName)
{
   
	ifstream input(FileName.c_str());
	std::string line;
	std::vector<string> IPinfo;

	if (!input)
	{
   
		ipAddr = "";
		portNum = 0;
		cout << "load file error : error = no such file" << endl;
		return false;
	}
	else
	{
   
		input.seekg(std::ios::beg);
		while (getline(input, line))
		{
   
			IPinfo.push_back(line);
		}
		
		if (IPinfo[0].empty()|| IPinfo[1].empty())
		{
   
			cout << "read parameter error : error =  Parameter setting error" << endl;
			return false;
		}
		
		ipAddr = IPinfo[0];
		portNum = static_cast<uint16_t>(std::stol(IPinfo[1]));
		if (portNum > 1023 && portNum < 65536)
		{
   
			cout << "[load IPaddrs]:" << ipAddr << endl;
			cout << "[load portNum]:" << portNum << endl;
			return true;
		}
		else
		{
   
			cout << "read parameter error : error = invalid port number" << endl;
			return false;
		}

	}
}

/*
 *  @brief Listen for client requests
 *  @return [in] result (ture:success  false:failure)
 */
bool testServer::listenClient(void)
{
   
	if ((s_socket_fd = socket(AF_INET, SOCK_STREAM, 0