[已解決]在關掉thread後，無法使用getline讀取console上的資料 - 伊莉討論區

#ifndef _SERVER_H_
#define _SERVER_H_
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <functional>
#include <atomic>
#include <thread>
class TcpServer
{
public:
TcpServer();
~TcpServer();
void StopListen();
void StartListen();
bool SendData(char* data, int len);
std::function<void(char*)> ReplyClientCmd = nullptr;
private:
char mServerIP[20];
int mPort = 0;
sockaddr_in mMasterAddr;
int mMasterFd = 0;
sockaddr_in mClient;
int mClientFd = 0;
// bool mIsListening = false;
std::atomic<bool> mIsListening;
bool Initialize();
void ListenFunction();
std::thread mListenThread;
};
#endif //_SERVER_H_

複製代碼

#include "TcpServer.h"
#include <iostream>
TcpServer::TcpServer()
{
strcpy(this->mServerIP, "192.168.0.120"); // it would be read from configure file, future.
this->mPort = 8080; // it would be read from configure file, future.
if(!Initialize())
{
std::cout << __func__ << " initialize failded!" << std::endl;
return;
}
}
TcpServer::~TcpServer()
{
StopListen();
}
void TcpServer::StopListen()
{
this->mIsListening = false;
close(this->mMasterFd);
close(this->mClientFd);
}
bool TcpServer::Initialize()
{
this->mMasterFd = socket(AF_INET, SOCK_STREAM, 0);
if(this->mMasterFd == -1)
{
perror("Create socket failed"); // log
return false;
}
int opt = 1;
if(setsockopt(this->mMasterFd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt)))
{
perror("set socket error!"); // log
return false;
}
this->mMasterAddr.sin_family = AF_INET;
//this->mMasterAddr.sin_addr.s_addr = inet_addr(this->mServerIP);
this->mMasterAddr.sin_addr.s_addr = INADDR_ANY;
this->mMasterAddr.sin_port = htons(this->mPort);
return true;
}
void TcpServer::StartListen()
{
if(!this->mIsListening) {
this->mIsListening = true;
mListenThread = std::thread(&TcpServer::ListenFunction, this); //開thread等待連線及監聽
}
else{
perror("Have a thread listening! Please STOP it!");
return;
}
}
void TcpServer::ListenFunction()
{
int sockLen = sizeof(this->mMasterAddr);
if(bind(this->mMasterFd, (sockaddr *)&this->mMasterAddr, sockLen) < 0){
perror("Occuring ERROR in bind!");
this->mIsListening = false;
return;
}
if(listen(this->mMasterFd, 5) < 0){
perror("Occuring ERROR in listen!");
this->mIsListening = false;
return;
}
while(this->mIsListening){ //mIsListening是用來決定要不要繼續監聽的flag
std::cout << "waitting connection.........." << std::endl;
this->mClientFd = accept(this->mMasterFd, (sockaddr *)&this->mClient, (socklen_t *)&sockLen);
if(this->mClientFd < 0){
perror("accept connection failed!");
this->mIsListening = false;
return;
}
std::cout << "accept from " << inet_ntoa(this->mClient.sin_addr) << " connection" << std::endl;
while (this->mIsListening) { // 接收client的訊息
char buffer[1024];
memset((void *)buffer, 0, 1024);
int lenRecv = recv(this->mClientFd, buffer, 1024, 0);
if(lenRecv > 0){
std::cout << "recieve package len = " << lenRecv << ", message = " << buffer << std::endl;
// if connection success, here do some decoding or reply command.
if(ReplyClientCmd != nullptr){
ReplyClientCmd(buffer);
}
const char *retStr = "server receive!";
send(this->mClientFd, retStr, strlen(retStr), 0);
}
else{
std::cout << "connection break!\n" << std::endl;
close(this->mClientFd);
break;
}
}
}
}
bool TcpServer::SendData(char* data, int len)
{
int ret = send(this->mClientFd, data, len, 0);
if(ret > 0)
return true;
else
return false;
}

複製代碼

#include <iostream>
#include "TcpServer.h"
#include <memory>
#include <string>
int main(int argc, const char* argv[])
{
TcpServer *server = new TcpServer();
server->StartListen();
std::string command;
for(;;){ //等待使用者輸入訊息並做相對應的動作
getline(std::cin, command);
if(command == std::string("CloseConnection"))
server->StopListen();
else{
std::cout << "type: " << command << std::endl;
// server->SendData((char*)command.c_str(), command.length());
}
}
return EXIT_SUCCESS;
}

複製代碼

WSAData wsaData;
WORD version = MAKEWORD(2, 2); // 版本
int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData); // 成功回傳 0
if (iResult != 0) {
// 初始化Winsock 失敗
return false;
}

複製代碼