Ftp/FtpServer.h

Go to the documentation of this file.

/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/SmingHub/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * FtpServer.h
 *
 ****/
 
#pragma once
 
#include "../TcpServer.h"
#include <WHashMap.h>
#include <FileSystem.h>
 
class FtpServerConnection;
 
class CustomFtpServer : public TcpServer
{
    friend class FtpServerConnection;
 
public:
    CustomFtpServer(IFS::FileSystem* fileSystem = nullptr) : fileSystem(fileSystem)
    {
        setTimeOut(900);
    }
 
    virtual IFS::UserRole validateUser(const char* login, const char* pass) = 0;
 
protected:
    TcpConnection* createClient(tcp_pcb* clientTcp) override;
 
    virtual bool onCommand([[maybe_unused]] String cmd, [[maybe_unused]] String data,
                           [[maybe_unused]] FtpServerConnection& connection)
    {
        return false;
    }
 
    IFS::FileSystem* getFileSystem() const
    {
        return fileSystem ?: ::getFileSystem();
    }
 
private:
    IFS::FileSystem* fileSystem;
};
 
class FtpServer : public CustomFtpServer
{
public:
    void addUser(const String& login, const String& pass, IFS::UserRole userRole = IFS::UserRole::Admin);
    IFS::UserRole validateUser(const char* login, const char* pass) override;
 
protected:
    bool onCommand(String cmd, String data, FtpServerConnection& connection) override;
 
private:
    struct User {
        String password;
        IFS::UserRole role;
    };
    using UserList = HashMap<String, User>;
    UserList users;
};