mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-25 19:56:03 +00:00
87 lines
No EOL
2.8 KiB
C++
87 lines
No EOL
2.8 KiB
C++
/*
|
|
* Copyright (c) 2018 Atmosphère-NX
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope 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, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#pragma once
|
|
#include <switch.h>
|
|
|
|
#include "iwaitable.hpp"
|
|
#include "ipc.hpp"
|
|
|
|
template<typename T>
|
|
class IServer : public IWaitable {
|
|
static_assert(std::is_base_of<IServiceObject, T>::value, "Service Objects must derive from IServiceObject");
|
|
protected:
|
|
Handle port_handle;
|
|
unsigned int max_sessions;
|
|
|
|
public:
|
|
IServer(unsigned int max_s) : port_handle(0), max_sessions(max_s) { }
|
|
|
|
virtual ~IServer() {
|
|
if (port_handle) {
|
|
svcCloseHandle(port_handle);
|
|
}
|
|
}
|
|
|
|
SessionManagerBase *GetSessionManager() {
|
|
return static_cast<SessionManagerBase *>(this->GetManager());
|
|
}
|
|
|
|
/* IWaitable */
|
|
virtual Handle GetHandle() override {
|
|
return this->port_handle;
|
|
}
|
|
|
|
virtual Result HandleSignaled(u64 timeout) override {
|
|
/* If this server's port was signaled, accept a new session. */
|
|
Handle session_h;
|
|
Result rc = svcAcceptSession(&session_h, this->port_handle);
|
|
if (R_FAILED(rc)) {
|
|
return rc;
|
|
}
|
|
|
|
this->GetSessionManager()->AddSession(session_h, std::move(ServiceObjectHolder(std::move(std::make_shared<T>()))));
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
class ServiceServer : public IServer<T> {
|
|
public:
|
|
ServiceServer(const char *service_name, unsigned int max_s) : IServer<T>(max_s) {
|
|
if (R_FAILED(smRegisterService(&this->port_handle, service_name, false, this->max_sessions))) {
|
|
/* TODO: Panic. */
|
|
}
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
class ExistingPortServer : public IServer<T> {
|
|
public:
|
|
ExistingPortServer(Handle port_h, unsigned int max_s) : IServer<T>(max_s) {
|
|
this->port_handle = port_h;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
class ManagedPortServer : public IServer<T> {
|
|
public:
|
|
ManagedPortServer(const char *service_name, unsigned int max_s) : IServer<T>(max_s) {
|
|
if (R_FAILED(svcManageNamedPort(&this->port_handle, service_name, this->max_sessions))) {
|
|
/* TODO: panic */
|
|
}
|
|
}
|
|
}; |