1/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */
2/* If you are missing that file, acquire a complete release at teeworlds.com. */
3
4#include <base/io.h>
5#include <base/logger.h>
6#include <base/net.h>
7#include <base/os.h>
8#include <base/str.h>
9#include <base/time.h>
10
11#include <engine/console.h>
12#include <engine/engine.h>
13#include <engine/shared/config.h>
14#include <engine/shared/jobs.h>
15#include <engine/shared/network.h>
16#include <engine/storage.h>
17
18#include <thread>
19
20class CEngine : public IEngine
21{
22 IConsole *m_pConsole;
23 IStorage *m_pStorage;
24
25 bool m_Logging;
26 std::shared_ptr<CFutureLogger> m_pFutureLogger;
27
28 char m_aAppName[256];
29
30 CJobPool m_JobPool;
31
32 static void Con_DbgLognetwork(IConsole::IResult *pResult, void *pUserData)
33 {
34 CEngine *pEngine = static_cast<CEngine *>(pUserData);
35
36 if(pEngine->m_Logging)
37 {
38 CNetBase::CloseLog();
39 pEngine->m_Logging = false;
40 }
41 else
42 {
43 char aBuf[32];
44 str_timestamp(buffer: aBuf, buffer_size: sizeof(aBuf));
45 char aFilenameSent[IO_MAX_PATH_LENGTH], aFilenameRecv[IO_MAX_PATH_LENGTH];
46 str_format(buffer: aFilenameSent, buffer_size: sizeof(aFilenameSent), format: "dumps/network_sent_%s.txt", aBuf);
47 str_format(buffer: aFilenameRecv, buffer_size: sizeof(aFilenameRecv), format: "dumps/network_recv_%s.txt", aBuf);
48 CNetBase::OpenLog(DataLogSent: pEngine->m_pStorage->OpenFile(pFilename: aFilenameSent, Flags: IOFLAG_WRITE, Type: IStorage::TYPE_SAVE),
49 DataLogRecv: pEngine->m_pStorage->OpenFile(pFilename: aFilenameRecv, Flags: IOFLAG_WRITE, Type: IStorage::TYPE_SAVE));
50 pEngine->m_Logging = true;
51 }
52 }
53
54public:
55 CEngine(bool Test, const char *pAppname, std::shared_ptr<CFutureLogger> pFutureLogger) :
56 m_pFutureLogger(std::move(pFutureLogger))
57 {
58 str_copy(dst&: m_aAppName, src: pAppname);
59 if(!Test)
60 {
61 log_info("engine", "running on %s-%s-%s", CONF_FAMILY_STRING, CONF_PLATFORM_STRING, CONF_ARCH_STRING);
62 log_info("engine", "arch is %s", CONF_ARCH_ENDIAN_STRING);
63
64 char aVersionStr[128];
65 if(os_version_str(version: aVersionStr, length: sizeof(aVersionStr)))
66 {
67 log_info("engine", "operating system version: %s", aVersionStr);
68 }
69
70 // init the network
71 net_init();
72 CNetBase::Init();
73 }
74
75#if defined(CONF_PLATFORM_EMSCRIPTEN)
76 // Make sure we don't use more threads than allowed in total (see PTHREAD_POOL_SIZE in Emscripten.toolchain)
77 // otherwise starting more threads may lead to deadlocks as the threads will simply not start.
78 const size_t ThreadCount = 4;
79#else
80 const size_t ThreadCount = std::max(a: 4, b: (int)std::thread::hardware_concurrency()) - 2;
81#endif
82 m_JobPool.Init(NumThreads: ThreadCount);
83
84 m_Logging = false;
85 }
86
87 ~CEngine() override
88 {
89 CNetBase::CloseLog();
90 }
91
92 void Init() override
93 {
94 m_pConsole = Kernel()->RequestInterface<IConsole>();
95 m_pStorage = Kernel()->RequestInterface<IStorage>();
96
97 if(!m_pConsole || !m_pStorage)
98 return;
99
100 m_pConsole->Register(pName: "dbg_lognetwork", pParams: "", Flags: CFGFLAG_SERVER | CFGFLAG_CLIENT, pfnFunc: Con_DbgLognetwork, pUser: this, pHelp: "Log the network");
101 }
102
103 void AddJob(std::shared_ptr<IJob> pJob) override
104 {
105 m_JobPool.Add(pJob: std::move(pJob));
106 }
107
108 void ShutdownJobs() override
109 {
110 m_JobPool.Shutdown();
111 }
112
113 void SetAdditionalLogger(std::shared_ptr<ILogger> &&pLogger) override
114 {
115 m_pFutureLogger->Set(pLogger);
116 }
117};
118
119IEngine *CreateEngine(const char *pAppname, std::shared_ptr<CFutureLogger> pFutureLogger) { return new CEngine(false, pAppname, std::move(pFutureLogger)); }
120IEngine *CreateTestEngine(const char *pAppname) { return new CEngine(true, pAppname, nullptr); }
121