mirror of https://github.com/probonopd/MiniDexed
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
2.7 KiB
111 lines
2.7 KiB
3 weeks ago
|
//
|
||
|
|
// ftpdaemon.cpp
|
||
|
|
//
|
||
|
|
// mt32-pi - A baremetal MIDI synthesizer for Raspberry Pi
|
||
|
|
// Copyright (C) 2020-2023 Dale Whinham <daleyo@gmail.com>
|
||
|
|
//
|
||
|
|
// This file is part of mt32-pi.
|
||
|
|
//
|
||
|
|
// mt32-pi 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 3 of the License, or (at your option) any later
|
||
|
|
// version.
|
||
|
|
//
|
||
|
|
// mt32-pi 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
|
||
|
|
// mt32-pi. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
//
|
||
|
|
|
||
|
|
#include <circle/logger.h>
|
||
|
|
#include <circle/net/in.h>
|
||
|
|
#include <circle/net/ipaddress.h>
|
||
|
|
#include <circle/net/netsubsystem.h>
|
||
|
|
#include <circle/string.h>
|
||
|
|
|
||
|
|
#include "ftpdaemon.h"
|
||
|
|
#include "ftpworker.h"
|
||
|
|
|
||
|
|
LOGMODULE("ftpd");
|
||
|
|
|
||
|
|
constexpr u16 ListenPort = 21;
|
||
|
|
constexpr u8 MaxConnections = 1;
|
||
|
|
|
||
|
|
CFTPDaemon::CFTPDaemon(const char* pUser, const char* pPassword)
|
||
|
|
: CTask(TASK_STACK_SIZE, true),
|
||
|
|
m_pListenSocket(nullptr),
|
||
|
|
m_pUser(pUser),
|
||
|
|
m_pPassword(pPassword)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
CFTPDaemon::~CFTPDaemon()
|
||
|
|
{
|
||
|
|
if (m_pListenSocket)
|
||
|
|
delete m_pListenSocket;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool CFTPDaemon::Initialize()
|
||
|
|
{
|
||
|
|
CNetSubSystem* const pNet = CNetSubSystem::Get();
|
||
|
|
|
||
|
|
if ((m_pListenSocket = new CSocket(pNet, IPPROTO_TCP)) == nullptr)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
if (m_pListenSocket->Bind(ListenPort) != 0)
|
||
|
|
{
|
||
|
|
LOGERR("Couldn't bind to port %d", ListenPort);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (m_pListenSocket->Listen() != 0)
|
||
|
|
{
|
||
|
|
LOGERR("Failed to listen on control socket");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// We started as a suspended task; run now that initialization is successful
|
||
|
|
Start();
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
void CFTPDaemon::Run()
|
||
|
|
{
|
||
|
|
assert(m_pListenSocket != nullptr);
|
||
|
|
|
||
|
|
LOGNOTE("Listener task spawned");
|
||
|
|
|
||
|
|
while (true)
|
||
|
|
{
|
||
|
|
CIPAddress ClientIPAddress;
|
||
|
|
u16 nClientPort;
|
||
|
|
|
||
|
|
LOGDBG("Listener: waiting for connection");
|
||
|
|
CSocket* pConnection = m_pListenSocket->Accept(&ClientIPAddress, &nClientPort);
|
||
|
|
|
||
|
|
if (pConnection == nullptr)
|
||
|
|
{
|
||
|
|
LOGERR("Unable to accept connection");
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
CString IPAddressString;
|
||
|
|
ClientIPAddress.Format(&IPAddressString);
|
||
|
|
LOGNOTE("Incoming connection from %s:%d", static_cast<const char*>(IPAddressString), nClientPort);
|
||
|
|
|
||
|
|
if (CFTPWorker::GetInstanceCount() >= MaxConnections)
|
||
|
|
{
|
||
|
|
pConnection->Send("421 Maximum number of connections reached.\r\n", 45, 0);
|
||
|
|
delete pConnection;
|
||
|
|
LOGWARN("Maximum number of connections reached");
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Spawn new worker
|
||
|
|
new CFTPWorker(pConnection, m_pUser, m_pPassword);
|
||
|
|
}
|
||
|
|
}
|