BOOL CServer::ProcessCommand(CAdminSocket *pAdminSocket, int nID, unsigned char *pData, int nDataLength) { switch (nID) { case 2: if (!nDataLength) { unsigned char buffer[2]; buffer[0] = m_nServerState / 256; buffer[1] = m_nServerState % 256; pAdminSocket->SendCommand(1, 2, buffer, 2); } else if (nDataLength == 2) { ToggleActive(*pData * 256 + pData[1]); unsigned char buffer[2]; buffer[0] = m_nServerState / 256; buffer[1] = m_nServerState % 256; pAdminSocket->SendCommand(1, 2, buffer, 2); } else pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length") + 1); break; case 3: if (!nDataLength) { pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length") + 1); } else if (*pData == USERCONTROL_GETLIST) { int len = 4; std::map<int, t_connectiondata>::iterator iter; for (iter = m_UsersList.begin(); iter != m_UsersList.end(); ++iter) { const t_connectiondata& data = iter->second; auto ip = ConvToNetwork(data.ip); auto user = ConvToNetwork(data.user); len += 4 + ip.size() + 2 + 4 + user.size() + 2 + 1; if (data.transferMode) { auto physicalFile = ConvToNetwork(data.physicalFile); auto logicalFile = ConvToNetwork(data.logicalFile); len += 2 + physicalFile.size() + 2 + logicalFile.size(); if (data.currentOffset != 0) len += 8; if (data.totalSize != -1) len += 8; } } unsigned char *buffer = new unsigned char[len]; buffer[0] = USERCONTROL_GETLIST; buffer[1] = ((m_UsersList.size() / 256) / 256) & 0xff; buffer[2] = (m_UsersList.size() / 256) & 0xff; buffer[3] = m_UsersList.size() % 256; unsigned char *p = buffer + 4; for (iter = m_UsersList.begin(); iter != m_UsersList.end(); ++iter) { const t_connectiondata& data = iter->second; auto ip = ConvToNetwork(data.ip); auto user = ConvToNetwork(data.user); memcpy(p, &data.userid, 4); p += 4; *p++ = (ip.size() / 256) & 0xff; *p++ = ip.size() % 256; memcpy(p, ip.c_str(), ip.size()); p += ip.size(); memcpy(p, &data.port, 4); p += 4; *p++ = (user.size() / 256) & 0xff; *p++ = user.size() % 256; memcpy(p, user.c_str(), user.size()); p += user.size(); *p = data.transferMode; if (data.transferMode) { // Bit 5 and 6 indicate presence of currentOffset and totalSize. if (data.currentOffset != 0) { *p |= 0x20; } if (data.totalSize != -1) { *p |= 0x40; } p++; auto physicalFile = ConvToNetwork(data.physicalFile); *p++ = (physicalFile.size() / 256) & 0xff; *p++ = physicalFile.size() % 256; memcpy(p, physicalFile.c_str(), physicalFile.size()); p += physicalFile.size(); auto logicalFile = ConvToNetwork(data.logicalFile); *p++ = (logicalFile.size() / 256) & 0xff; *p++ = logicalFile.size() % 256; memcpy(p, logicalFile.c_str(), logicalFile.size()); p += logicalFile.size(); if (data.currentOffset != 0) { memcpy(p, &data.currentOffset, 8); p += 8; } if (data.totalSize != -1) { memcpy(p, &data.totalSize, 8); p += 8; } } else { p++; } } m_pAdminInterface->SendCommand(1, 3, buffer, len); delete [] buffer; } else if (*pData == USERCONTROL_KICK || *pData == USERCONTROL_BAN) { if (nDataLength != 5) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else { int nUserID; memcpy(&nUserID, pData+1, 4); std::map<int, t_connectiondata>::iterator iter = m_UsersList.find(nUserID); if (iter!=m_UsersList.end()) { if (*pData == USERCONTROL_BAN) { // Get the list of IP filter rules. CStdString ips = m_pOptions->GetOption(OPTION_IPFILTER_DISALLOWED); if (ips != _T("")) ips += _T(" "); int pos = ips.Find(' '); while (pos != -1) { CStdString blockedIP = ips.Left(pos); ips = ips.Mid(pos + 1); pos = ips.Find(' '); if (MatchesFilter(blockedIP, iter->second.ip)) break; } if (pos == -1) { ips = m_pOptions->GetOption(OPTION_IPFILTER_DISALLOWED); if (ips != _T("")) ips += _T(" "); ips += iter->second.ip; m_pOptions->SetOption(OPTION_IPFILTER_DISALLOWED, ips); } } t_controlmessage *msg = new t_controlmessage; msg->command = USERCONTROL_KICK; msg->socketid = nUserID; iter->second.pThread->PostThreadMessage(WM_FILEZILLA_THREADMSG, FTM_CONTROL, (LPARAM)msg); char buffer[2]; buffer[0] = *pData; buffer[1] = 0; pAdminSocket->SendCommand(1, 3, &buffer, 2); } else { char buffer[2]; buffer[0] = USERCONTROL_KICK; buffer[1] = 1; pAdminSocket->SendCommand(1, 3, &buffer, 2); } } } else pAdminSocket->SendCommand(1, 1, "\001Protocol error: Invalid data", strlen("\001Protocol error: Invalid data")+1); break; case 5: if (!nDataLength) { unsigned char *pBuffer = NULL; DWORD nBufferLength = 0; if (m_pOptions && m_pOptions->GetAsCommand(&pBuffer, &nBufferLength)) { pAdminSocket->SendCommand(1, 5, pBuffer, nBufferLength); delete [] pBuffer; } } else if (m_pOptions) { if (nDataLength < 2) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else { CStdString const listenPorts = m_pOptions->GetOption(OPTION_SERVERPORT); CStdString const listenPortsSsl = m_pOptions->GetOption(OPTION_TLSPORTS); bool const enableSsl = m_pOptions->GetOptionVal(OPTION_ENABLETLS) != 0; CStdString const ipBindings = m_pOptions->GetOption(OPTION_IPBINDINGS); int const nAdminListenPort = (int)m_pOptions->GetOptionVal(OPTION_ADMINPORT); CStdString const adminIpBindings = m_pOptions->GetOption(OPTION_ADMINIPBINDINGS); CStdString peerIP; UINT port = 0; bool bLocal = false; if (!pAdminSocket->GetPeerName(peerIP, port)) return FALSE; else bLocal = IsLocalhost(peerIP); if (!m_pOptions->ParseOptionsCommand(pData, nDataLength, bLocal)) { pAdminSocket->SendCommand(1, 1, "\001Protocol error: Invalid data, could not import settings.", strlen("\001Protocol error: Invalid data, could not import settings.")+1); char buffer = 1; pAdminSocket->SendCommand(1, 5, &buffer, 1); break; } char buffer = 0; pAdminSocket->SendCommand(1, 5, &buffer, 1); unsigned int threadnum = (int)m_pOptions->GetOptionVal(OPTION_THREADNUM); if (m_nServerState & STATE_ONLINE) { if (threadnum > m_ThreadArray.size()) { while (threadnum > m_ThreadArray.size()) { int index = GetNextThreadNotificationID(); CServerThread *pThread = new CServerThread(WM_FILEZILLA_SERVERMSG + index); m_ThreadNotificationIDs[index] = pThread; if (pThread->Create(THREAD_PRIORITY_NORMAL, CREATE_SUSPENDED)) { pThread->ResumeThread(); m_ThreadArray.push_back(pThread); } } CStdString str; str.Format(_T("Number of threads increased to %d."), threadnum); ShowStatus(str, 0); } else if (threadnum < m_ThreadArray.size()) { CStdString str; str.Format(_T("Decreasing number of threads to %d."), threadnum); ShowStatus(str, 0); unsigned int i = 0; std::vector<CServerThread *> newList; for (auto iter = m_ThreadArray.begin(); iter != m_ThreadArray.end(); iter++, i++) { if (i >= threadnum) { (*iter)->PostThreadMessage(WM_FILEZILLA_THREADMSG, FTM_GOOFFLINE, 2); m_ClosedThreads.push_back(*iter); } else newList.push_back(*iter); } m_ThreadArray.swap(newList); } } if (listenPorts != m_pOptions->GetOption(OPTION_SERVERPORT) || enableSsl != (m_pOptions->GetOptionVal(OPTION_ENABLETLS) != 0) || (m_pOptions->GetOptionVal(OPTION_ENABLETLS) && listenPortsSsl != m_pOptions->GetOption(OPTION_TLSPORTS)) || ipBindings != m_pOptions->GetOption(OPTION_IPBINDINGS)) { if (!m_ListenSocketList.empty()) { ShowStatus(_T("Closing all listening sockets"), 0); for (std::list<CListenSocket*>::iterator listIter = m_ListenSocketList.begin(); listIter != m_ListenSocketList.end(); ++listIter) { (*listIter)->Close(); delete *listIter; } m_ListenSocketList.clear(); if (!CreateListenSocket()) { ShowStatus(_T("Failed to create a listen socket on any of the specified ports. Server is not online!"), 1); m_nServerState &= ~STATE_ONLINE; } else { ShowStatus(_T("Listen socket port changed"), 0); if (!(m_nServerState & STATE_MASK_GOOFFLINE)) m_nServerState |= STATE_ONLINE; } SendState(); } } if (nAdminListenPort != m_pOptions->GetOptionVal(OPTION_ADMINPORT) || adminIpBindings != m_pOptions->GetOption(OPTION_ADMINIPBINDINGS)) { CreateAdminListenSocket(); } VerifyTlsSettings(pAdminSocket); VerifyPassiveModeSettings(pAdminSocket); } } break; case 6: if (!nDataLength) { unsigned char *pBuffer = NULL; DWORD nBufferLength = 0; CPermissions permissions = CPermissions(std::function<void()>()); permissions.GetAsCommand(&pBuffer, &nBufferLength); pAdminSocket->SendCommand(1, 6, pBuffer, nBufferLength); delete [] pBuffer; } else { if (nDataLength < 2) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else { CPermissions permissions = CPermissions(std::function<void()>()); if (!permissions.ParseUsersCommand(pData, nDataLength)) { pAdminSocket->SendCommand(1, 1, "\001Protocol error: Invalid data, could not import account settings.", strlen("\001Protocol error: Invalid data, could not import account settings.")+1); char buffer = 1; pAdminSocket->SendCommand(1, 6, &buffer, 1); break; } char buffer = 0; pAdminSocket->SendCommand(1, 6, &buffer, 1); } } break; case 8: pAdminSocket->SendCommand(1, 8, NULL, 0); break; default: { CStdStringA str; str.Format("\001Protocol error: Unknown command (%d).", nID); pAdminSocket->SendCommand(1, 1, str.c_str(), str.GetLength()); } break; } return TRUE; }
BOOL CServer::ProcessCommand(CAdminSocket *pAdminSocket, int nID, unsigned char *pData, int nDataLength) { switch (nID) { case 2: if (!nDataLength) { unsigned char buffer[2]; buffer[0]=m_nServerState/256; buffer[1]=m_nServerState%256; pAdminSocket->SendCommand(1, 2, buffer, 2); } else if (nDataLength == 2) { ToggleActive(*pData*256+pData[1]); unsigned char buffer[2]; buffer[0]=m_nServerState/256; buffer[1]=m_nServerState%256; pAdminSocket->SendCommand(1, 2, buffer, 2); } else pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); break; case 3: if (!nDataLength) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else if (*pData == USERCONTROL_GETLIST) { int len = 3; std::map<int, t_connectiondata>::iterator iter; for (iter=m_UsersList.begin(); iter!=m_UsersList.end(); iter++) len += 4 + strlen(iter->second.ip)+2 + 4 + strlen(iter->second.user)+2; unsigned char *buffer = new unsigned char[len]; buffer[0] = USERCONTROL_GETLIST; buffer[1] = m_UsersList.size() / 256; buffer[2] = m_UsersList.size() % 256; unsigned char *p = buffer + 3; for (iter=m_UsersList.begin(); iter!=m_UsersList.end(); iter++) { USES_CONVERSION; memcpy(p, &iter->second.userid, 4); p+=4; *p++ = strlen(iter->second.ip) / 256; *p++ = strlen(iter->second.ip) % 256; memcpy(p, T2CA(iter->second.ip), strlen(T2CA(iter->second.ip))); p+=strlen(T2CA(iter->second.ip)); memcpy(p, &iter->second.port, 4); p+=4; if (iter->second.user) { *p++ = strlen(iter->second.user) / 256; *p++ = strlen(iter->second.user) % 256; memcpy(p, T2CA(iter->second.user), strlen(T2CA(iter->second.user))); p+=strlen(T2CA(iter->second.user)); } else { *p++ = 0; *p++ = 0; } } m_pAdminInterface->SendCommand(1, 3, buffer, len); delete [] buffer; } else if (*pData == USERCONTROL_KICK) { if (nDataLength != 5) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else { int nUserID; memcpy(&nUserID, pData+1, 4); std::map<int, t_connectiondata>::iterator iter = m_UsersList.find(nUserID); if (iter!=m_UsersList.end()) { t_controlmessage *msg=new t_controlmessage; msg->command=USERCONTROL_KICK; msg->socketid=nUserID; iter->second.pThread->PostThreadMessage(WM_FILEZILLA_THREADMSG, FTM_CONTROL, (LPARAM)msg); char buffer[2]; buffer[0] = USERCONTROL_KICK; buffer[1] = 0; pAdminSocket->SendCommand(1, 3, &buffer, 2); } else { char buffer[2]; buffer[0] = USERCONTROL_KICK; buffer[1] = 1; pAdminSocket->SendCommand(1, 3, &buffer, 2); } } } else pAdminSocket->SendCommand(1, 1, "\001Protocol error: Invalid data", strlen("\001Protocol error: Invalid data")+1); break; case 5: if (!nDataLength) { char *pBuffer = NULL; DWORD nBufferLength = 0; if (m_pOptions && m_pOptions->GetAsCommand(&pBuffer, &nBufferLength)) { pAdminSocket->SendCommand(1, 5, pBuffer, nBufferLength); delete [] pBuffer; } } else if (m_pOptions) { if (nDataLength < 2) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else { int nListenPort = (int)m_pOptions->GetOptionVal(OPTION_SERVERPORT); int nAdminListenPort = (int)m_pOptions->GetOptionVal(OPTION_ADMINPORT); CStdString adminIpBindings = m_pOptions->GetOption(OPTION_ADMINIPBINDINGS); SOCKADDR_IN sockAddr; memset(&sockAddr, 0, sizeof(sockAddr)); int nSockAddrLen = sizeof(sockAddr); BOOL bLocal = pAdminSocket->GetPeerName((SOCKADDR*)&sockAddr, &nSockAddrLen) && sockAddr.sin_addr.S_un.S_addr == 0x0100007f; if (!m_pOptions->ParseOptionsCommand(pData, nDataLength, bLocal)) { pAdminSocket->SendCommand(1, 1, "\001Protocol error: Invalid data, could not import settings.", strlen("\001Protocol error: Invalid data, could not import settings.")+1); char buffer = 1; pAdminSocket->SendCommand(1, 5, &buffer, 1); break; } char buffer = 0; pAdminSocket->SendCommand(1, 5, &buffer, 1); unsigned int threadnum = (int)m_pOptions->GetOptionVal(OPTION_THREADNUM); if (threadnum>m_ThreadArray.size()) { int newthreads=threadnum-m_ThreadArray.size(); for (int i=0;i<newthreads;i++) { CServerThread *pThread = new CServerThread; if (pThread->Create(THREAD_PRIORITY_NORMAL, CREATE_SUSPENDED)) { pThread->ResumeThread(); m_ThreadArray.push_back(pThread); } } CStdString str; str.Format("Number of threads increased to %d.", threadnum); ShowStatus(str, 0); } else if (threadnum<m_ThreadArray.size()) { CStdString str; str.Format("Decreasing number of threads to %d.", threadnum); ShowStatus(str, 0); unsigned int i=0; for (std::list<CServerThread *>::iterator iter=m_ThreadArray.begin(); iter!=m_ThreadArray.end(); iter++,i++) if (i>=threadnum) (*iter)->PostThreadMessage(WM_FILEZILLA_THREADMSG, FTM_GOOFFLINE, 2); } if (nListenPort != m_pOptions->GetOptionVal(OPTION_SERVERPORT)) { if (m_pListenSocket) { CStdString str; str.Format("Closing listen socket on port %d", nListenPort); ShowStatus(str, 0); m_pListenSocket->Close(); str.Format("Creating listen socket on port %I64d...", m_pOptions->GetOptionVal(OPTION_SERVERPORT)); ShowStatus(str, 0); if (!m_pListenSocket->Create((int)m_pOptions->GetOptionVal(OPTION_SERVERPORT), SOCK_STREAM,FD_ACCEPT,0) || !m_pListenSocket->Listen()) { delete m_pListenSocket; m_pListenSocket = NULL; str.Format("Failed to create listen socket on port %I64d. Server is not online!", m_pOptions->GetOptionVal(OPTION_SERVERPORT)); ShowStatus(str,1); m_nServerState = 0; } else ShowStatus("Listen socket port changed",0); } } if (nAdminListenPort != m_pOptions->GetOptionVal(OPTION_ADMINPORT) || adminIpBindings!=m_pOptions->GetOption(OPTION_ADMINIPBINDINGS)) { if (nAdminListenPort == m_pOptions->GetOptionVal(OPTION_ADMINPORT)) { for (std::list<CAdminListenSocket*>::iterator iter = m_AdminListenSocketList.begin(); iter!=m_AdminListenSocketList.end(); iter++) { (*iter)->Close(); delete *iter; } m_AdminListenSocketList.clear(); } CAdminListenSocket *pSocket = new CAdminListenSocket(m_pAdminInterface); if (!pSocket->Create((int)m_pOptions->GetOptionVal(OPTION_ADMINPORT), SOCK_STREAM, FD_ACCEPT, (m_pOptions->GetOption(OPTION_ADMINIPBINDINGS)!="*") ? _T("127.0.0.1") : NULL)) { delete pSocket; CStdString str; str.Format(_T("Failed to change admin listen port to %I64d."), m_pOptions->GetOptionVal(OPTION_ADMINPORT)); m_pOptions->SetOption(OPTION_ADMINPORT, nAdminListenPort); ShowStatus(str, 1); } else { pSocket->Listen(); for (std::list<CAdminListenSocket*>::iterator iter = m_AdminListenSocketList.begin(); iter!=m_AdminListenSocketList.end(); iter++) { (*iter)->Close(); delete *iter; } m_AdminListenSocketList.clear(); m_AdminListenSocketList.push_back(pSocket); if (nAdminListenPort != m_pOptions->GetOptionVal(OPTION_ADMINPORT)) { CStdString str; str.Format(_T("Admin listen port changed to %I64d."), m_pOptions->GetOptionVal(OPTION_ADMINPORT)); ShowStatus(str, 0); } if (m_pOptions->GetOption(OPTION_ADMINIPBINDINGS) != "*") { BOOL bError = FALSE; CStdString str = _T("Failed to bind the admin interface to the following IPs:"); CStdString ipBindings = m_pOptions->GetOption(OPTION_ADMINIPBINDINGS); if (ipBindings != "") ipBindings += " "; while (ipBindings != "") { int pos = ipBindings.Find(" "); if (pos == -1) break; CStdString ip = ipBindings.Left(pos); ipBindings = ipBindings.Mid(pos+1); CAdminListenSocket *pAdminListenSocket = new CAdminListenSocket(m_pAdminInterface); if (!pAdminListenSocket->Create((int)m_pOptions->GetOptionVal(OPTION_ADMINPORT), SOCK_STREAM, FD_ACCEPT, ip) || !pAdminListenSocket->Listen()) { bError = TRUE; str += _T(" ") + ip; delete pAdminListenSocket; } else m_AdminListenSocketList.push_back(pAdminListenSocket); } if (bError) ShowStatus(str, 1); } if (adminIpBindings!=m_pOptions->GetOption(OPTION_ADMINIPBINDINGS)) ShowStatus(_T("Admin interface IP bindings changed"), 0); } } } } break; case 6: if (!nDataLength) { char *pBuffer = NULL; DWORD nBufferLength = 0; CPermissions permissions; permissions.GetAsCommand(&pBuffer, &nBufferLength); pAdminSocket->SendCommand(1, 6, pBuffer, nBufferLength); delete [] pBuffer; } else { if (nDataLength < 2) pAdminSocket->SendCommand(1, 1, "\001Protocol error: Unexpected data length", strlen("\001Protocol error: Unexpected data length")+1); else { CPermissions permissions; if (!permissions.ParseUsersCommand(pData, nDataLength)) { pAdminSocket->SendCommand(1, 1, "\001Protocol error: Invalid data, could not import account settings.", strlen("\001Protocol error: Invalid data, could not import account settings.")+1); char buffer = 1; pAdminSocket->SendCommand(1, 6, &buffer, 1); break; } char buffer = 0; pAdminSocket->SendCommand(1, 6, &buffer, 1); } } break; case 8: pAdminSocket->SendCommand(1, 8, NULL, 0); break; default: { CStdString str; str.Format("\001Protocol error: Unknown command (%d).", nID); pAdminSocket->SendCommand(1, 1, str.c_str(), str.GetLength()); } break; } return TRUE; }
LRESULT CALLBACK CServer::WindowProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { CServer *pServer=(CServer *)GetWindowLongPtr(hWnd, GWLP_USERDATA); if (message == WM_CLOSE) { pServer->OnClose(); return 0; } else if (hWnd && message == WM_DESTROY) { ASSERT (hWnd == pServer->m_hWnd); HANDLE *handle = new HANDLE[pServer->m_ThreadArray.size()]; unsigned int i = 0; for (auto iter = pServer->m_ThreadArray.begin(); iter != pServer->m_ThreadArray.end(); ++iter, ++i) { handle[i] = (*iter)->m_hThread; (*iter)->PostThreadMessage(WM_QUIT, 0, 0); } for (i = 0; i < pServer->m_ThreadArray.size(); ++i) { int res = WaitForSingleObject(handle[i], INFINITE); if (res == WAIT_FAILED) res = GetLastError(); } delete [] handle; handle = new HANDLE[pServer->m_ClosedThreads.size()]; i = 0; for (auto iter = pServer->m_ClosedThreads.begin(); iter != pServer->m_ClosedThreads.end(); ++iter, ++i) { handle[i] = (*iter)->m_hThread; (*iter)->PostThreadMessage(WM_QUIT, 0, 0); } for (i = 0; i < pServer->m_ClosedThreads.size(); ++i) { int res = WaitForSingleObject(handle[i], INFINITE); if (res == WAIT_FAILED) res = GetLastError(); } delete [] handle; pServer->m_AdminListenSocketList.clear(); delete pServer->m_pAdminInterface; pServer->m_pAdminInterface = NULL; delete pServer->m_pOptions; pServer->m_pOptions = NULL; if (pServer->m_nTimerID) { KillTimer(pServer->m_hWnd, pServer->m_nTimerID); pServer->m_nTimerID = 0; } PostQuitMessage(0); return 0; } else if (message == WM_TIMER) pServer->OnTimer(wParam); else if (message == WM_FILEZILLA_RELOADCONFIG) { COptions options; options.ReloadConfig(); CPermissions perm = CPermissions(std::function<void()>()); perm.ReloadConfig(); } else if (message >= WM_FILEZILLA_SERVERMSG) { UINT index = message - WM_FILEZILLA_SERVERMSG; if (index >= pServer->m_ThreadNotificationIDs.size()) return 0; CServerThread *pThread = pServer->m_ThreadNotificationIDs[index]; if (pThread) { std::list<CServerThread::t_Notification> notifications; pThread->GetNotifications(notifications); for (std::list<CServerThread::t_Notification>::const_iterator iter = notifications.begin(); iter != notifications.end(); iter++) if (pServer->OnServerMessage(pThread, iter->wParam, iter->lParam) != 0) break; } return 0; } return ::DefWindowProc(hWnd, message, wParam, lParam); }
LRESULT CServer::OnServerMessage(WPARAM wParam, LPARAM lParam) { if (wParam==FSM_STATUSMESSAGE) { t_statusmsg *msg=reinterpret_cast<t_statusmsg *>(lParam); CStdString str; FILETIME fFileTime; SystemTimeToFileTime(&msg->time, &fFileTime); _int64 time = ((_int64)fFileTime.dwHighDateTime<<32) + fFileTime.dwLowDateTime; str.Format("(%06d)- %s (%s)> %s", msg->userid, (LPCTSTR)msg->user, (LPCTSTR)msg->ip, (LPCTSTR)msg->status); ShowStatus(time, str, msg->type); delete [] msg->user; delete [] msg->status; delete msg; } else if (wParam==FSM_CONNECTIONDATA) { t_connop *pConnOp = reinterpret_cast<t_connop*>(lParam); if (pConnOp->op != USERCONTROL_CONNOP_REMOVE) m_UsersList[pConnOp->data->userid] = *pConnOp->data; else { std::map<int, t_connectiondata>::iterator iter=m_UsersList.find(pConnOp->data->userid); if (iter!=m_UsersList.end()) m_UsersList.erase(iter); } USES_CONVERSION; int userlen = pConnOp->data->user ? strlen(pConnOp->data->user) : 0; int len = 2 + 4 + strlen(pConnOp->data->ip)+2 + 4 + userlen+2; unsigned char *buffer = new unsigned char[len]; buffer[0] = USERCONTROL_CONNOP; buffer[1] = pConnOp->op; memcpy(buffer+2, &pConnOp->data->userid, 4); buffer[2 + 4] = strlen(pConnOp->data->ip)/256; buffer[2 + 4 + 1] = strlen(pConnOp->data->ip)%256; memcpy(buffer + 2 + 4 + 2, T2CA(pConnOp->data->ip), strlen(T2CA(pConnOp->data->ip))); memcpy(buffer + 2 + 4 +2 + strlen(pConnOp->data->ip)+2, &pConnOp->data->port, 4); buffer[2 + 4 + strlen(pConnOp->data->ip)+2 + 4]= userlen/256; buffer[2 + 4 + strlen(pConnOp->data->ip)+2 + 4 + 1]= userlen%256; if (pConnOp->data->user) memcpy(buffer + 2 + 4 + strlen(pConnOp->data->ip)+2 + 4 + 2, T2CA(pConnOp->data->user), userlen); m_pAdminInterface->SendCommand(2, 3, buffer, len); delete [] buffer; delete pConnOp->data; delete pConnOp; } else if (wParam==FSM_THREADCANQUIT) { CServerThread *pThread=(CServerThread *)lParam; for (std::list<CServerThread *>::iterator iter=m_ThreadArray.begin(); iter!=m_ThreadArray.end(); iter++) { if (*iter==pThread) { HANDLE handle=pThread->m_hThread; pThread->PostThreadMessage(WM_QUIT, 0, 0); int res=WaitForSingleObject(handle, INFINITE); if (res==WAIT_FAILED) res=GetLastError(); m_ThreadArray.erase(iter); if (!m_ThreadArray.size()) { if (!m_bQuit) ShowStatus(_T("Server offline."), 1); else DestroyWindow(m_hWnd); } break; } } } else if (wParam==FSM_SEND) { char buffer[5]; buffer[0] = 1; memcpy(buffer+1, &lParam, 4); m_pAdminInterface->SendCommand(2, 7, buffer, 5); m_nSendCount += lParam; } else if (wParam==FSM_RECV) { char buffer[5]; buffer[0] = 0; memcpy(buffer+1, &lParam, 4); m_pAdminInterface->SendCommand(2, 7, buffer, 5); m_nRecvCount += lParam; } else if (wParam == FSM_RELOADCONFIG) { COptions options; options.ReloadConfig(); CPermissions perm; perm.ReloadConfig(); } return 0; }