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