'功能完善'

admin

2022-06-30 a17738e1545ff7dbef6398b8ec1eab93ab59c9a1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
#include "pch.h"
#include "SocketManager.h"
#define CHANNEL_NUM 9
#include <thread>
 
SOCKET SocketManager::socketServer;
 
ActionCallback SocketManager::actionCallback;
 
void* SocketManager::callbackContext;
 
SocketManager::SocketManager(ActionCallback callback, void* context) {
    init(callback,context);
}
 
SocketManager::~SocketManager() {
 
}
 
void SocketManager::processMsg(SOCKET  client, void* context) {
    while (true)
    {
        //Ò»Ö±¶ÁÈ¡ÏûÏ¢
        char recvBuff[1024];
        memset(recvBuff, 0, sizeof(recvBuff));
        int recv_len = recv(client, recvBuff, sizeof(recvBuff), 0);
        if (recv_len > 0)
        {
            cout << "½ÓÊܵ½²Ù×÷Ö¸Á" << recvBuff << endl;
            try {
                bool success = actionCallback(string(recvBuff), callbackContext);
                //·¢ËͲÙ×÷½á¹û
                if (success)
                {
                    char buff[] = "OK";
                    int iSend = send(client, buff, sizeof(buff), 0);
                    cout << "·µ»Ø¸ø¿Í»§¶ËÊý¾Ý£º" << buff << endl;
                }
                else {
                    char buff[] = "FAIL";
                    int iSend = send(client, buff, sizeof(buff), 0);
                    cout << "·µ»Ø¸ø¿Í»§¶ËÊý¾Ý£º" << buff << endl;
                }
            }
            catch (string st) {
                char buff[] = "FAIL";
                int iSend = send(client, buff, sizeof(buff), 0);
                cout << "ÉèÖòɼ¯´úÂë³ö´í£º" << st << endl;
            }
        }
    }
 
}
 
void SocketManager::runSocketServer(void* context) {
    //´´½¨server
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2, 2), &wsaData);//³õʼ»¯
    socketServer = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    sockaddr_in sockaddr;
    sockaddr.sin_family = AF_INET;
    sockaddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
    sockaddr.sin_port = htons(LOCAL_SERVER_PORT);
    int result;
    if (bind(socketServer, (const struct sockaddr*)&sockaddr, sizeof(sockaddr)) == SOCKET_ERROR) {
        closesocket(socketServer);
        WSACleanup();
        printf("°ó¶¨Ê§°Ü\n");
        return;
    }
    //ÔÚ±¾µØ°ó¶¨¶Ë¿Ú
    result = listen(socketServer, SOMAXCONN);//¿ªÊ¼¼àÌý
 
    SocketManager* sm = (SocketManager*)context;
 
 
    printf("wait for connection...\n");
 
 
    while (true)
    {
        int nsize = sizeof(SOCKADDR);
        SOCKADDR   clientAddr;
        SOCKET  client = accept(sm->socketServer, &clientAddr, &nsize);
        if (client == SOCKET_ERROR)
        {
            cout << "Á¬½Óʧ°Ü" << endl;
            continue;
        }
 
        thread msg(SocketManager::processMsg, client, context);
        msg.detach();
    }
 
    closesocket(socketServer);
    WSACleanup();
 
}
 
void SocketManager::init(ActionCallback callback, void* contex) {
 
    actionCallback = callback;
    callbackContext = contex;
    clientaddr.sin_family = AF_INET;
    clientaddr.sin_port = htons(PORT);
    clientaddr.sin_addr.S_un.S_addr = inet_addr(ADDR);
    //½¨Á¢9¸öÁ¬½Ó,Ç°8¸öΪÊý¾Ý´«ÊäͨµÀ£¬ºóÒ»¸öΪÃüÁî´«ÊäͨµÀ
    sockClients.clear();
    for (int i = 0;i < CHANNEL_NUM;i++)
    {
        addClient(i);
    }
    Connect();
 
    thread server(SocketManager::runSocketServer, this);
    server.detach();
}
 
int SocketManager::getClientsNum() {
 
    return sockClients.size();
}
 
void SocketManager::addClient(int p) {
    WSADATA wsd;
    WSAStartup(MAKEWORD(2, 2), &wsd);
    SOCKET m_SockClient;
    m_SockClient = socket(AF_INET, SOCK_STREAM, 0);
    sockClients.push_back(m_SockClient);
}
 
int SocketManager::Connect(int p) {
    int i = connect(getClient(p), (sockaddr*)&clientaddr, sizeof(clientaddr));
    return i;
}
 
int SocketManager::Connect() {
    int scount = 0;
    list<SOCKET>::iterator ele;
    for (ele = sockClients.begin();ele != sockClients.end();++ele) {
        int i = connect(*ele, (sockaddr*)&clientaddr, sizeof(clientaddr));
        if (i >= 0) {
            scount++;
        }
    }
 
    return scount;
}
 
int SocketManager::disConnect(int p) {
    return closesocket(getClient(p));
}
 
void SocketManager::disConnect() {
    list<SOCKET>::iterator ele;
    for (ele = sockClients.begin();ele != sockClients.end();++ele) {
        closesocket(*ele);
    }
}
 
SOCKET SocketManager::getClient(int p) {
    list<SOCKET>::iterator ele = sockClients.begin();
    if (p > sockClients.size() - 1) {
        return 0;
    }
    std::advance(ele, p);
    return *ele;
}
 
 
BOOL SocketManager::sendMsg(int p, const char* msg) {
    SOCKET socket = getClient(p);
    char buffer[1024];
    send(socket, msg, strlen(msg), 0);
    int num = recv(socket, buffer, 1024, 0);
    if (num >= 0) {
        std::cout << "Receive form server: " << buffer << std::endl;
        return 1;
    }
 
    return 0;
}