// Server0327.cpp: 定義應用程序的入口點。
//

#include "Server0327.h"
#include <WinSock2.h>
#include <vector>

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

using namespace std;
class TcpServer
{
public:
?? ?bool Listen(const char* ip, int port)
?? ?{
?? ??? ?//創建socket
?? ??? ?_serverSock = socket(AF_INET, SOCK_STREAM, 0);
?? ??? ?if (INVALID_SOCKET == _serverSock)
?? ??? ??? ?goto Exit;
?? ??? ?//綁定IP和端口號
?? ??? ?SOCKADDR_IN addr;
?? ??? ?addr.sin_addr.s_addr = inet_addr(ip);
?? ??? ?addr.sin_family = AF_INET;
?? ??? ?addr.sin_port = htons(port);
?? ??? ?if (INVALID_SOCKET == bind(_serverSock, (sockaddr*)&addr, sizeof(SOCKADDR_IN)))
?? ??? ??? ?goto Exit;

?? ??? ?//監聽端口
?? ??? ?if (INVALID_SOCKET == listen(_serverSock, 255))
?? ??? ??? ?goto Exit;

?? ??? ?return true;
?? ?Exit:
?? ??? ?closesocket(_serverSock);
?? ??? ?_serverSock = INVALID_SOCKET;
?? ??? ?return false;
?? ?}


?? ?void Update()
?? ?{
?? ??? ?FD_SET reads;
?? ??? ?FD_ZERO(&reads);
?? ??? ?FD_SET(_serverSock, &reads);

?? ??? ?for (auto&& clientSock : _clientSocks)
?? ??? ??? ?FD_SET(clientSock, &reads);

?? ??? ?int nRet = select(0, &reads, nullptr, nullptr, nullptr);
?? ??? ?if (nRet > 0)
?? ??? ?{
?? ??? ??? ?if (FD_ISSET(_serverSock, &reads))
?? ??? ??? ??? ?Accept();
?? ??? ??? ?else
?? ??? ??? ?{
?? ??? ??? ??? ?for (int i = 0; i < _clientSocks.size(); i++)
?? ??? ??? ??? ?{
?? ??? ??? ??? ??? ?if (FD_ISSET(_clientSocks[i], &reads))
?? ??? ??? ??? ??? ?{
?? ??? ??? ??? ??? ??? ?char buf[1024]{ 0 };
?? ??? ??? ??? ??? ??? ?int nRecv = recv(_clientSocks[i], buf, 1024, 0);
?? ??? ??? ??? ??? ??? ?if (nRecv > 0)
?? ??? ??? ??? ??? ??? ??? ?OnNetMsg(_clientSocks[i], buf, sizeof(buf));
?? ??? ??? ??? ??? ??? ?if (nRecv <= 0)
?? ??? ??? ??? ??? ??? ?{
?? ??? ??? ??? ??? ??? ??? ?closesocket(_clientSocks[i]);
?? ??? ??? ??? ??? ??? ??? ?OnDisConnect(_clientSocks[i]);
?? ??? ??? ??? ??? ??? ??? ?_clientSocks.erase(_clientSocks.begin() + i);
?? ??? ??? ??? ??? ??? ??? ?continue;
?? ??? ??? ??? ??? ??? ?}
?? ??? ??? ??? ??? ?}
?? ??? ??? ??? ?}
?? ??? ??? ?}
?? ??? ?}
?? ?}

?? ?void Accept()
?? ?{
?? ??? ?SOCKADDR_IN clientAddr;
?? ??? ?int clientAddrLen = sizeof(clientAddr);
?? ??? ?SOCKET clientSock = accept(_serverSock, (sockaddr*)&clientAddr, &clientAddrLen);
?? ??? ?if (INVALID_SOCKET != clientSock)
?? ??? ?{
?? ??? ??? ?_clientSocks.push_back(clientSock);
?? ??? ??? ?OnConnect(clientSock);
?? ??? ?}

?? ?}

?? ?void OnConnect(SOCKET clientSock)
?? ?{
?? ??? ?printf("%d 客戶端連接\n", clientSock);
?? ?}

?? ?int OnNetMsg(SOCKET clientSock, const char* buff, int len)
?? ?{
?? ??? ?printf("%d == %s\n", clientSock, buff);
?? ??? ?for (auto&& clientSock : _clientSocks)
?? ??? ??? ?send(clientSock, buff, len, 0);
?? ??? ?return len;
?? ?}


?? ?void OnDisConnect(SOCKET sock)
?? ?{
?? ??? ?printf("%d==客戶端斷開連接\n", sock);
?? ?}

private:
?? ?SOCKET _serverSock;
?? ?vector<SOCKET> _clientSocks;
};

int main()
{
?? ?WSADATA wsa;
?? ?WSAStartup(MAKEWORD(2, 2), &wsa);

?? ?TcpServer server;
?? ?server.Listen("0.0.0.0", 7890);
?? ?while (true)
?? ??? ?server.Update();

?? ?WSACleanup();
?? ?return 0;
}

// CMakeClient0331.cpp: 定義應用程序的入口點。
//

#include "CMakeClient0331.h"
#include<iostream>
#include<WinSock2.h>
#include<thread>

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


int main()
{
?? ?WSADATA wsaData;
?? ?WSAStartup(MAKEWORD(2, 2), &wsaData);

?? ?SOCKET clientSock = socket(AF_INET, SOCK_STREAM, 0);
?? ?if (INVALID_SOCKET == clientSock)
?? ?{
?? ??? ?closesocket(clientSock);
?? ??? ?WSACleanup();
?? ??? ?return 0;
?? ?}

?? ?SOCKADDR_IN serverAddr;
?? ?serverAddr.sin_family = AF_INET;
?? ?serverAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
?? ?serverAddr.sin_port = htons(7890);
?? ?if (INVALID_SOCKET == connect(clientSock, (sockaddr*)&serverAddr, sizeof(SOCKADDR_IN)))
?? ?{
?? ??? ?closesocket(clientSock);
?? ??? ?WSACleanup();
?? ??? ?return 0;
?? ?}
?? ?std::cout << "已連接服務端" << std::endl;

?? ?std::thread t([&clientSock] {
?? ??? ?while (true)
?? ??? ?{
?? ??? ??? ?char buf[1024]{ 0 };
?? ??? ??? ?int nRecvLen = recv(clientSock, buf, 1024, 0);
?? ??? ??? ?if (nRecvLen <= 0)
?? ??? ??? ??? ?break;
?? ??? ??? ?std::cout << "接收數據長度：" << nRecvLen << " 接受數據：" << buf << std::endl;
?? ??? ?}
?? ??? ?});
?? ?t.detach();

?? ?while (true)
?? ?{
?? ??? ?char buf[1024]{ 0 };
?? ??? ?std::cin >> buf;
?? ??? ?if (buf[0] == 'q')break;
?? ??? ?int nSendLen = send(clientSock, buf, sizeof(buf), 0);
?? ?}

?? ?closesocket(clientSock);
?? ?WSACleanup();
?? ?return 0;
}