#include "stdafx.h"
|
#include "ServerSocket.h"
|
#include "Respond.h"
|
|
#include <mutex>
|
|
//È«¾Ö»¥³âËø
|
mutex g_mtx;//Äã´´½¨ÁËÒ»¸ö»¥³âËø¶ÔÏó
|
|
ServerSocket::ServerSocket()
|
{
|
//ÕâÀï³õʼ»¯·þÎñÆ÷socket
|
if (init_socket())
|
{
|
cout << "init server socket ok..." << endl;
|
}
|
else
|
{
|
cout << "init false !" << endl;
|
}
|
}
|
|
|
ServerSocket::~ServerSocket()
|
{
|
WSACleanup(); // ¹Ø±ÕÍøÂç»·¾³
|
}
|
|
bool ServerSocket::init_socket()
|
{
|
WSAData wsa = { 0 };
|
|
//Æô¶¯ÍøÂç»·¾³
|
WSAStartup(MAKEWORD(2,2), &wsa);
|
|
m_socket = socket(AF_INET, SOCK_STREAM, 0);
|
|
SOCKADDR_IN addr;
|
addr.sin_family = AF_INET;
|
addr.sin_port = htons(16888);
|
addr.sin_addr.S_un.S_addr = 0; // ±¾»úµÄÈÎÒâip
|
|
//Á¬½Ó
|
int ret_bind = ::bind(m_socket, (sockaddr*)&addr, sizeof(addr));
|
if (ret_bind) {
|
cout << "bind fail:" << WSAGetLastError() << endl;
|
return false;
|
}
|
|
//¼àÌý
|
int ret_listen = listen(m_socket, 5); // Èô²¢·¢Á¿´ó ¸ø¸ö´óµãµÄÖµÈç 50£¬500µÈ
|
if (ret_listen) {
|
cout << "listen fail:" << WSAGetLastError() << endl;
|
return false;
|
}
|
|
//³õʼ»¯Íê³É
|
//ʹÓÃÏ߳̽ÓÊÕ¿Í»§¶ËÁ¬½ÓÇëÇó
|
thread accept_thread(&ServerSocket::acceptThread,this);
|
accept_thread.detach(); // ·ÖÀëỊ̈߳¬·ÀÖ¹Òì³£½áÊø
|
|
return true;
|
}
|
|
void ServerSocket::acceptThread()
|
{
|
while (true)
|
{
|
SOCKET client;
|
SOCKADDR_IN addr;
|
int len = sizeof(addr);
|
|
//½ÓÊÕÁ¬½Ó
|
cout << "waiting for client connect..." << endl;
|
client = accept(m_socket, (SOCKADDR *)&addr, &len);
|
cout << "client :" << client <<"success connected."<< endl;
|
|
//¼ì²éһϣ¬clientÊÇ·ñÕýÈ·
|
if (client == SOCKET_ERROR)
|
{
|
cout << "socket error :" << WSAGetLastError() << endl;
|
break;
|
}
|
else if (client == 0) {
|
cout << "socket close:" << WSAGetLastError() << endl;
|
break;
|
}
|
|
//ʹÓÃÒ»¸ö{}Ȧ¶¨´úÂë¿é£¬ÏÞ¶¨±£»¤ËøµÄÉÏËø·¶Î§
|
{
|
lock_guard<mutex> lg(g_mtx);
|
|
//¹²ÏíÊý¾Ýmap<SOCKET, int> m_clients--¡¾¼ü¡¤Öµ¶Ô¡¿
|
//map[client] = ... //ÓÃ×÷ÐÄÌø¼ì²âµÄ¼üÖµ¶Ô
|
}
|
|
//¿ÉÒÔ½øÐÐÊÕ·¢°üµÄ²Ù×÷ÁË£¬Ò²Ê¹ÓÃÒ»¸öÏ̴߳¦Àí ¡¾ÄäÃû¶ÔÏó¡¿[clientÊÇÕâ¸öÏ̵߳÷Óú¯ÊýµÄ²ÎÊý]
|
thread(&ServerSocket::recvandsendThread, this, client).detach();
|
}
|
}
|
|
void ServerSocket::recvandsendThread(SOCKET client)//²ÉÓÃHead·ÖÀ뱨ÎÄÍ·µÄ·½·¨£¬½â¾öÕ³°üÎÊÌâ
|
{
|
while (true)
|
{
|
//ÉèÖÃÁËÓ¦¸Ã½ÓÊܵÄHeadµÄ×Ö½Ú´óС
|
int head_len = sizeof(Head); //±¨ÎÄÍ· Ó¦¸Ã½ÓÊյĸöÊý
|
int recv_len = 0; //±¾´Îʵ¼Ê½ÓÊÕµ½µÄ¸öÊý
|
int gap_len = head_len - recv_len; //ÔÚ±¾´Î½ÓÊպ󣬻¹ÐèÒª½ÓÊյĸöÊý
|
|
//¿ªÊ¼½ÓÊÕ
|
char *head_buf = new char[head_len];
|
recv_len = recv(client, head_buf, head_len, 0);
|
|
//HeadδÂú¶øʣϵĿռä
|
gap_len = head_len - recv_len;
|
|
//Èç¹û£¬½ÓÊÕµÄ×Ö½ÚÊý²»¹»£¬¾ÍÒ»Ö±½ÓÊÕ
|
while (gap_len > 0)
|
{
|
recv_len = recv(client, head_buf + (head_len - gap_len), gap_len, 0);
|
gap_len -= recv_len;
|
}
|
|
//½â°üÍ·
|
Head *head = (Head *)head_buf;
|
|
int bag_len = head->len;//Õâ¸öÊÇ°üµÄ×ܳ¤¶È£¬±¨ÎÄ Ó¦¸Ã½ÓÊյĸöÊý
|
|
//ºÍ°üÍ·µÄÁ÷³ÌÒ»Ñù£¬³ÖÐø½ÓÊÕ£¬Ö±µ½È¡ÂúÊý¾Ý
|
char *all_buf = new char[bag_len];
|
memcpy(all_buf, head_buf, head_len);//°üÍ·¿½±´µ½ÍêÕûµÄ°üÖÐ
|
|
gap_len = bag_len - head_len; //ÔÚ±¾´Î½ÓÊպ󣬻¹ÐèÒª½ÓÊյĸöÊý
|
|
//³ÖÐø£¬Ö±µ½Íê³É
|
while (gap_len > 0) {
|
recv_len = recv(client, all_buf + (bag_len - gap_len), gap_len, 0);
|
gap_len -= recv_len;
|
}
|
|
|
/*
|
ÐèÒª½âÃÜ
|
*/
|
|
|
//¿ªÊ¼½â°ü
|
if (head->type == LOGIN_REQ)
|
{
|
cout << "ÊÕµ½ÇëÇó°ü" << endl;
|
LoginReq *bag = (LoginReq *)all_buf;
|
cout << "ÐÕÃû£º" << bag->user_name << "ÃÜÂ룺" << bag->password << endl;
|
|
//µÃµ½¿Í»§¶ËµÄÊý¾Ý£¬ÎÒÃÇÓÃËüÈ¥²éÕÒÊý¾Ý¿â£¬Íê³ÉµÇ¼ҵÎñµÄÏìÓ¦
|
Respond loginRes;
|
loginRes.business_login(bag, client);
|
}
|
else if (head->type == REGISTER_REQ)
|
{
|
cout << "ÊÕµ½ÇëÇó°ü len:" << head->len << endl;
|
RegisterReq *bag = (RegisterReq *)all_buf;
|
cout << "ÐÕÃû£º" << bag->user_name << endl
|
<< "ÃÜÂ룺" << bag->password << endl
|
<< "ÓÊÏ䣺" << bag->email << endl
|
<< "µç»°£º" << bag->telephone << endl
|
<< "²¿ÃÅ£º" << bag->department << endl;
|
|
//µÃµ½¿Í»§¶ËµÄÊý¾Ý£¬ÎÒÃÇÓÃËüÈ¥²éÕÒÊý¾Ý¿â£¬Íê³ÉµÇ¼ҵÎñµÄÏìÓ¦
|
Respond registerRes;
|
registerRes.business_register(bag, client);
|
}
|
|
|
|
|
// ÓÃÍêÁË£¬¼ÇµÃÊͷŶ¯Ì¬¿ª±ÙµÄÄÚ´æ
|
delete[] head_buf;
|
delete[] all_buf;
|
head_buf = nullptr; // ·ÀÖ¹Ö¸ÕëÐü¿Õ£¬¶ø²úÉúÐü¿ÕÖ¸Õë
|
all_buf = nullptr;
|
}
|
closesocket(client); // ¹Ø±Õ¿Í»§¶ËÌ×½Ó×Ö
|
}
|
