Initial Commit
This commit is contained in:
Riley Schneider
494
database/FileZillaFTP/source/interface/AdminSocket.cpp
Normal file
494
database/FileZillaFTP/source/interface/AdminSocket.cpp
Normal file
@@ -0,0 +1,494 @@
|
||||
// FileZilla Server - a Windows ftp server
|
||||
|
||||
// Copyright (C) 2002-2004 - Tim Kosse <tim.kosse@gmx.de>
|
||||
|
||||
// This program is free software; you can redistribute it and/or
|
||||
// modify it under the terms of the GNU General Public License
|
||||
// as published by the Free Software Foundation; either version 2
|
||||
// of the License, or (at your option) any later version.
|
||||
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program; if not, write to the Free Software
|
||||
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
||||
|
||||
// AdminSocket.cpp: Implementierung der Klasse CAdminSocket.
|
||||
//
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
|
||||
#include "stdafx.h"
|
||||
#include "AdminSocket.h"
|
||||
#include "MainFrm.h"
|
||||
#include "../iputils.h"
|
||||
#include "../OptionTypes.h"
|
||||
#include "../platform.h"
|
||||
#include "../misc\md5.h"
|
||||
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
// Konstruktion/Destruktion
|
||||
//////////////////////////////////////////////////////////////////////
|
||||
|
||||
#define BUFSIZE 4096
|
||||
|
||||
CAdminSocket::CAdminSocket(CMainFrame *pMainFrame)
|
||||
{
|
||||
ASSERT(pMainFrame);
|
||||
m_pMainFrame = pMainFrame;
|
||||
m_pRecvBuffer = new unsigned char[BUFSIZE];
|
||||
m_nRecvBufferLen = BUFSIZE;
|
||||
m_nRecvBufferPos = 0;
|
||||
m_nConnectionState = 0;
|
||||
m_bClosed = FALSE;
|
||||
}
|
||||
|
||||
CAdminSocket::~CAdminSocket()
|
||||
{
|
||||
delete [] m_pRecvBuffer;
|
||||
|
||||
for (std::list<t_data>::iterator iter=m_SendBuffer.begin(); iter!=m_SendBuffer.end(); iter++)
|
||||
delete [] iter->pData;
|
||||
}
|
||||
|
||||
void CAdminSocket::OnConnect(int nErrorCode)
|
||||
{
|
||||
if (!nErrorCode)
|
||||
{
|
||||
if (!m_nConnectionState)
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Connected, waiting for authentication"), 0);
|
||||
m_nConnectionState = 1;
|
||||
}
|
||||
m_pMainFrame->OnAdminInterfaceConnected();
|
||||
}
|
||||
else
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Error, could not connect to server"), 1);
|
||||
Close();
|
||||
}
|
||||
}
|
||||
|
||||
void CAdminSocket::OnReceive(int nErrorCode)
|
||||
{
|
||||
if (nErrorCode)
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("OnReceive failed, closing connection"), 1);
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
|
||||
if (!m_nConnectionState)
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Connected, waiting for authentication"), 0);
|
||||
m_nConnectionState = 1;
|
||||
}
|
||||
|
||||
int numread = Receive(m_pRecvBuffer + m_nRecvBufferPos, m_nRecvBufferLen - m_nRecvBufferPos);
|
||||
if (numread > 0)
|
||||
{
|
||||
m_nRecvBufferPos += numread;
|
||||
if (m_nRecvBufferLen-m_nRecvBufferPos < (BUFSIZE/4))
|
||||
{
|
||||
unsigned char *tmp = m_pRecvBuffer;
|
||||
m_nRecvBufferLen += BUFSIZE;
|
||||
m_pRecvBuffer = new unsigned char[m_nRecvBufferLen];
|
||||
memcpy(m_pRecvBuffer, tmp, m_nRecvBufferPos);
|
||||
delete [] tmp;
|
||||
}
|
||||
}
|
||||
if (!numread)
|
||||
{
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
else if (numread == SOCKET_ERROR)
|
||||
{
|
||||
if (WSAGetLastError() != WSAEWOULDBLOCK)
|
||||
{
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
}
|
||||
while (ParseRecvBuffer());
|
||||
}
|
||||
|
||||
void CAdminSocket::OnSend(int nErrorCode)
|
||||
{
|
||||
if (nErrorCode)
|
||||
{
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
if (!m_nConnectionState)
|
||||
return;
|
||||
|
||||
while (!m_SendBuffer.empty())
|
||||
{
|
||||
t_data data = m_SendBuffer.front();
|
||||
int nSent = Send(data.pData + data.dwOffset, data.dwLength - data.dwOffset);
|
||||
if (!nSent)
|
||||
{
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
if (nSent == SOCKET_ERROR)
|
||||
{
|
||||
if (WSAGetLastError()!=WSAEWOULDBLOCK)
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
|
||||
if ((DWORD)nSent < (data.dwLength - data.dwOffset))
|
||||
data.dwOffset += nSent;
|
||||
else
|
||||
{
|
||||
m_SendBuffer.pop_front();
|
||||
delete [] data.pData;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void CAdminSocket::Close()
|
||||
{
|
||||
if (m_nConnectionState)
|
||||
m_pMainFrame->ShowStatus(_T("Connection to server closed."), 1);
|
||||
m_nConnectionState = 0;
|
||||
if (!m_bClosed)
|
||||
{
|
||||
m_bClosed = TRUE;
|
||||
m_pMainFrame->PostMessage(WM_APP + 1, 0, 0);
|
||||
}
|
||||
}
|
||||
|
||||
BOOL CAdminSocket::ParseRecvBuffer()
|
||||
{
|
||||
DWORD len;
|
||||
switch (m_nConnectionState)
|
||||
{
|
||||
case 1:
|
||||
{
|
||||
if (m_nRecvBufferPos<3)
|
||||
return FALSE;
|
||||
if (m_pRecvBuffer[0] != 'F' || m_pRecvBuffer[1] != 'Z' || m_pRecvBuffer[2] != 'S')
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Unknown protocol identifier (0x%d 0x%d 0x%d). Most likely connected to the wrong port."), (int)m_pRecvBuffer[0], (int)m_pRecvBuffer[1], (int)m_pRecvBuffer[2]);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
if (m_nRecvBufferPos < 5)
|
||||
return FALSE;
|
||||
len = m_pRecvBuffer[3] * 256 + m_pRecvBuffer[4];
|
||||
if (len != 4)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Invalid server version length (%d)."), len);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
if (m_nRecvBufferPos < 9)
|
||||
return FALSE;
|
||||
|
||||
int version = (int)GET32(m_pRecvBuffer + 5);
|
||||
if (version != SERVER_VERSION)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol warning: Server version mismatch: Server version is %d.%d.%d.%d, interface version is %d.%d.%d.%d"),
|
||||
(version >> 24) & 0xFF,
|
||||
(version >> 16) & 0xFF,
|
||||
(version >> 8) & 0xFF,
|
||||
(version >> 0) & 0xFF,
|
||||
(SERVER_VERSION >> 24) & 0xFF,
|
||||
(SERVER_VERSION >> 16) & 0xFF,
|
||||
(SERVER_VERSION >> 8) & 0xFF,
|
||||
(SERVER_VERSION >> 0) & 0xFF);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
}
|
||||
|
||||
if (m_nRecvBufferPos<11)
|
||||
return FALSE;
|
||||
len = m_pRecvBuffer[9] * 256 + m_pRecvBuffer[10];
|
||||
if (len != 4)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Invalid protocol version length (%d)."), len);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
if (m_nRecvBufferPos<15)
|
||||
return FALSE;
|
||||
version = (int)GET32(m_pRecvBuffer + 11);
|
||||
if (version != PROTOCOL_VERSION)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Protocol version mismatch: Server protocol version is %d.%d.%d.%d, interface protocol version is %d.%d.%d.%d"),
|
||||
(version >> 24) & 0xFF,
|
||||
(version >> 16) & 0xFF,
|
||||
(version >> 8) & 0xFF,
|
||||
(version >> 0) & 0xFF,
|
||||
(PROTOCOL_VERSION >> 24) & 0xFF,
|
||||
(PROTOCOL_VERSION >> 16) & 0xFF,
|
||||
(PROTOCOL_VERSION >> 8) & 0xFF,
|
||||
(PROTOCOL_VERSION >> 0) & 0xFF);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
memmove(m_pRecvBuffer, m_pRecvBuffer+15, m_nRecvBufferPos-15);
|
||||
m_nRecvBufferPos-=15;
|
||||
m_nConnectionState = 2;
|
||||
}
|
||||
break;
|
||||
case 2:
|
||||
if (m_nRecvBufferPos<5)
|
||||
return FALSE;
|
||||
if ((m_pRecvBuffer[0]&0x03) > 2)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Unknown command type (%d), closing connection."), (int)(m_pRecvBuffer[0]&0x03));
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
len = (int)GET32(m_pRecvBuffer + 1);
|
||||
if (len + 5 <= m_nRecvBufferPos)
|
||||
{
|
||||
if ((m_pRecvBuffer[0]&0x03) == 0 && (m_pRecvBuffer[0]&0x7C)>>2 == 0)
|
||||
{
|
||||
if (len<4)
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Invalid auth data"), 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
unsigned char *p = m_pRecvBuffer + 5;
|
||||
|
||||
unsigned int noncelen1 = *p*256 + p[1];
|
||||
if ((noncelen1+2) > (len-2))
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Invalid auth data"), 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
unsigned int noncelen2 = p[2 + noncelen1]*256 + p[2 + noncelen1 +1];
|
||||
if ((noncelen1+noncelen2+4) > len)
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Invalid auth data"), 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
MD5 md5;
|
||||
if (noncelen1)
|
||||
md5.update(p+2, noncelen1);
|
||||
char* utf8 = ConvToNetwork(m_Password);
|
||||
if (!utf8)
|
||||
{
|
||||
m_pMainFrame->ShowStatus(_T("Can't convert password to UTF-8"), 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
md5.update((const unsigned char *)utf8, strlen(utf8));
|
||||
delete [] utf8;
|
||||
if (noncelen2)
|
||||
md5.update(p+noncelen1+4, noncelen2);
|
||||
md5.finalize();
|
||||
|
||||
memmove(m_pRecvBuffer, m_pRecvBuffer+len+5, m_nRecvBufferPos-len-5);
|
||||
m_nRecvBufferPos-=len+5;
|
||||
|
||||
unsigned char *digest = md5.raw_digest();
|
||||
SendCommand(0, digest, 16);
|
||||
delete [] digest;
|
||||
m_nConnectionState=3;
|
||||
return TRUE;
|
||||
}
|
||||
else if ((m_pRecvBuffer[0]&0x03) == 1 && (m_pRecvBuffer[0]&0x7C)>>2 == 0)
|
||||
{
|
||||
m_nConnectionState=3;
|
||||
m_pMainFrame->ParseReply((m_pRecvBuffer[0]&0x7C)>>2, m_pRecvBuffer+5, len);
|
||||
}
|
||||
else
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Unknown command ID (%d), closing connection."), (int)(m_pRecvBuffer[0]&0x7C)>>2);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
memmove(m_pRecvBuffer, m_pRecvBuffer+len+5, m_nRecvBufferPos-len-5);
|
||||
m_nRecvBufferPos-=len+5;
|
||||
}
|
||||
break;
|
||||
case 3:
|
||||
if (m_nRecvBufferPos < 5)
|
||||
return FALSE;
|
||||
int nType = *m_pRecvBuffer & 0x03;
|
||||
int nID = (*m_pRecvBuffer & 0x7C) >> 2;
|
||||
if (nType > 2 || nType < 1)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Unknown command type (%d), closing connection."), nType);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
else
|
||||
{
|
||||
len = (unsigned int)GET32(m_pRecvBuffer + 1);
|
||||
if (len > 0xFFFFFF)
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol error: Invalid data length (%u) for command (%d:%d)"), len, nType, nID);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
if (m_nRecvBufferPos < len+5)
|
||||
return FALSE;
|
||||
else
|
||||
{
|
||||
if (nType == 1)
|
||||
m_pMainFrame->ParseReply(nID, m_pRecvBuffer + 5, len);
|
||||
else if (nType == 2)
|
||||
m_pMainFrame->ParseStatus(nID, m_pRecvBuffer + 5, len);
|
||||
else
|
||||
{
|
||||
CString str;
|
||||
str.Format(_T("Protocol warning: Command type %d not implemented."), nType);
|
||||
m_pMainFrame->ShowStatus(str, 1);
|
||||
}
|
||||
|
||||
memmove(m_pRecvBuffer, m_pRecvBuffer+len+5, m_nRecvBufferPos-len-5);
|
||||
m_nRecvBufferPos-=len+5;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL CAdminSocket::SendCommand(int nType)
|
||||
{
|
||||
t_data data;
|
||||
data.pData = new unsigned char[5];
|
||||
data.pData[0] = nType << 2;
|
||||
data.dwOffset = 0;
|
||||
DWORD dwDataLength = 0;
|
||||
memcpy(data.pData + 1, &dwDataLength, 4);
|
||||
|
||||
data.dwLength = 5;
|
||||
|
||||
m_SendBuffer.push_back(data);
|
||||
|
||||
do
|
||||
{
|
||||
data=m_SendBuffer.front();
|
||||
int nSent = Send(data.pData+data.dwOffset, data.dwLength-data.dwOffset);
|
||||
if (!nSent)
|
||||
{
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
if (nSent == SOCKET_ERROR)
|
||||
{
|
||||
if (WSAGetLastError()!=WSAEWOULDBLOCK)
|
||||
{
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
if ((DWORD)nSent < (data.dwLength-data.dwOffset))
|
||||
data.dwOffset+=nSent;
|
||||
else
|
||||
{
|
||||
m_SendBuffer.pop_front();
|
||||
delete [] data.pData;
|
||||
}
|
||||
} while (!m_SendBuffer.empty());
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL CAdminSocket::SendCommand(int nType, void *pData, int nDataLength)
|
||||
{
|
||||
ASSERT((pData && nDataLength) || (!pData && !nDataLength));
|
||||
|
||||
t_data data;
|
||||
data.pData = new unsigned char[nDataLength+5];
|
||||
data.pData[0] = nType << 2;
|
||||
data.dwOffset = 0;
|
||||
memcpy(data.pData + 1, &nDataLength, 4);
|
||||
if (pData)
|
||||
memcpy(data.pData+5, pData, nDataLength);
|
||||
|
||||
data.dwLength = nDataLength + 5;
|
||||
|
||||
m_SendBuffer.push_back(data);
|
||||
|
||||
do
|
||||
{
|
||||
data=m_SendBuffer.front();
|
||||
int nSent = Send(data.pData+data.dwOffset, data.dwLength-data.dwOffset);
|
||||
if (!nSent)
|
||||
{
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
if (nSent == SOCKET_ERROR)
|
||||
{
|
||||
if (WSAGetLastError()!=WSAEWOULDBLOCK)
|
||||
{
|
||||
Close();
|
||||
return FALSE;
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
if ((DWORD)nSent < (data.dwLength-data.dwOffset))
|
||||
data.dwOffset += nSent;
|
||||
else
|
||||
{
|
||||
m_SendBuffer.pop_front();
|
||||
delete [] data.pData;
|
||||
}
|
||||
} while (!m_SendBuffer.empty());
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
BOOL CAdminSocket::IsConnected()
|
||||
{
|
||||
return m_nConnectionState == 3;
|
||||
}
|
||||
|
||||
void CAdminSocket::OnClose(int nErrorCode)
|
||||
{
|
||||
Close();
|
||||
}
|
||||
|
||||
void CAdminSocket::DoClose()
|
||||
{
|
||||
m_bClosed = true;
|
||||
Close();
|
||||
}
|
||||
|
||||
bool CAdminSocket::IsLocal()
|
||||
{
|
||||
CString ip;
|
||||
UINT port;
|
||||
if (!GetPeerName(ip, port))
|
||||
return false;
|
||||
|
||||
return IsLocalhost(ip);
|
||||
}
|
||||
Reference in New Issue
Block a user