|
| 1 | +// Copyright Copyright (c) Kang Lin studio, All Rights Reserved |
| 2 | +// Author: Kang Lin <kl222@126.com> |
| 3 | + |
| 4 | +#include <QSslSocket> |
| 5 | +#include <QLoggingCategory> |
| 6 | +#include "BackendFtpServer.h" |
| 7 | +#include "OperateFtpServer.h" |
| 8 | + |
| 9 | +static Q_LOGGING_CATEGORY(log, "FtpServer.Backend") |
| 10 | +CBackendFtpServer::CBackendFtpServer(COperate *pOperate) : CBackend(pOperate) |
| 11 | + , m_pServer(nullptr) |
| 12 | +{ |
| 13 | + qDebug(log) << Q_FUNC_INFO; |
| 14 | + m_Para = ((COperateFtpServer*)pOperate)->GetParameter(); |
| 15 | +} |
| 16 | + |
| 17 | +CBackendFtpServer::~CBackendFtpServer() |
| 18 | +{ |
| 19 | + qDebug(log) << Q_FUNC_INFO; |
| 20 | +} |
| 21 | + |
| 22 | +CBackend::OnInitReturnValue CBackendFtpServer::OnInit() |
| 23 | +{ |
| 24 | + qDebug(log) << Q_FUNC_INFO; |
| 25 | + if(m_pServer) { |
| 26 | + qCritical(log) << "Server is exist"; |
| 27 | + return OnInitReturnValue::Fail; |
| 28 | + } |
| 29 | + QString szUser; |
| 30 | + QString szPassword; |
| 31 | + if(!m_Para->GetAnonymousLogin()) { |
| 32 | + szUser = m_Para->GetUser(); |
| 33 | + szPassword = m_Para->GetPassword(); |
| 34 | + } |
| 35 | + m_pServer = new FtpServer(this, m_Para->GetRoot(), m_Para->GetPort(), |
| 36 | + szUser, szPassword, |
| 37 | + m_Para->GetReadOnly(), false); |
| 38 | + m_pServer->SetFilter(this); |
| 39 | + if(m_pServer->isListening()) { |
| 40 | + qInfo(log) << "The ftp server listen in" << m_Para->GetPort(); |
| 41 | + } else { |
| 42 | + QString szErr = tr("The ftp server is not listening in %1").arg(m_Para->GetPort()); |
| 43 | + emit sigError(-1, szErr); |
| 44 | + } |
| 45 | + return OnInitReturnValue::NotUseOnProcess; |
| 46 | +} |
| 47 | + |
| 48 | +int CBackendFtpServer::OnClean() |
| 49 | +{ |
| 50 | + qDebug(log) << Q_FUNC_INFO; |
| 51 | + |
| 52 | + if(m_pServer) |
| 53 | + delete m_pServer; |
| 54 | + |
| 55 | + return 0; |
| 56 | +} |
| 57 | + |
| 58 | +bool CBackendFtpServer::onFilter(QSslSocket *socket) |
| 59 | +{ |
| 60 | + if(!socket) return true; |
| 61 | + if(m_Para->GetConnectCount() >= 0 && m_Para->GetConnectCount() <= m_Sockets.size()) { |
| 62 | + qDebug(log) << "Exceeded the allowed number of connections:" << m_Para->GetConnectCount(); |
| 63 | + return true; |
| 64 | + } |
| 65 | + |
| 66 | + QString ip = socket->peerAddress().toString(); |
| 67 | + m_Sockets.append(socket); |
| 68 | + qDebug(log) << "Current connect count:" << m_Sockets.size() |
| 69 | + << "; new connect from:" << ip; |
| 70 | + bool check = false; |
| 71 | + check = connect(socket, &QSslSocket::disconnected, |
| 72 | + this, &CBackendFtpServer::slotDisconnected); |
| 73 | + Q_ASSERT(check); |
| 74 | + return false; |
| 75 | +} |
| 76 | + |
| 77 | +void CBackendFtpServer::slotDisconnected() |
| 78 | +{ |
| 79 | + QSslSocket *socket = qobject_cast<QSslSocket*>(sender()); |
| 80 | + if(!socket) return; |
| 81 | + m_Sockets.removeAll(socket); |
| 82 | + qDebug(log) << "Current connect count:" << m_Sockets.size() |
| 83 | + << "; remove connect:" << socket->peerAddress().toString(); |
| 84 | +} |
0 commit comments