diff --git a/.idea/.name b/.idea/.name index 66d2284..f083ce8 100644 --- a/.idea/.name +++ b/.idea/.name @@ -1 +1 @@ -BeamMP-Server \ No newline at end of file +Server \ No newline at end of file diff --git a/CMakeLists.txt b/CMakeLists.txt index 0f1f029..0d138f2 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -1,16 +1,8 @@ -cmake_minimum_required(VERSION 3.15) -project(BeamMP-Server) - -set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} /O2") -include_directories(${PROJECT_SOURCE_DIR}/curl) - +cmake_minimum_required(VERSION 3.10) +project(Server) set(CMAKE_CXX_STANDARD 17) - -file(GLOB source_files - "src/*.h" "src/*.hpp" "src/*.cpp" - "src/Network 2.0/*.hpp" "src/Network 2.0/*.cpp" - "src/Lua System/*.hpp" "src/Lua System/*.cpp" - "src/curl/*.h" "src/lua/*.h" "src/lua/*.hpp") - -add_executable(BeamMP-Server ${source_files}) -target_link_libraries(BeamMP-Server libcurl_a ws2_32 lua53) \ No newline at end of file +file(GLOB source_files "src/*.cpp" "src/*/*.cpp" "include/*.h" "include/*/*.h" "include/*.hpp" "include/*/*.hpp") +add_executable(${PROJECT_NAME} ${source_files}) +target_include_directories(${PROJECT_NAME} PUBLIC $) +set_target_properties(${PROJECT_NAME} PROPERTIES OUTPUT_NAME "BeamMP-Server") +target_link_libraries(${PROJECT_NAME} libcurl_a urlmon ws2_32 lua53 zlibstatic) diff --git a/src/Network 2.0/Client.hpp b/include/Client.hpp similarity index 63% rename from src/Network 2.0/Client.hpp rename to include/Client.hpp index 468f025..b59a250 100644 --- a/src/Network 2.0/Client.hpp +++ b/include/Client.hpp @@ -9,20 +9,22 @@ #include #include +struct VData{ + int ID = -1; + std::string Data; +}; + class Client { private: - std::set> VehicleData; //ID and Data; + std::set VehicleData; //ID and Data; std::string Name = "Unknown Client"; - bool Connected = false; sockaddr_in UDPADDR; std::string Role; - std::string DID; //Discord ID + std::string DID; SOCKET TCPSOCK; int Status = 0; - int ID = -1; //PlayerID + int ID = -1; public: - bool isDownloading = true; - std::set> GetAllCars(); void AddNewCar(int ident,const std::string& Data); void SetCarData(int ident,const std::string&Data); void SetName(const std::string& name); @@ -30,21 +32,37 @@ public: void SetDID(const std::string& did); std::string GetCarData(int ident); void SetUDPAddr(sockaddr_in Addr); + std::set GetAllCars(); void SetTCPSock(SOCKET CSock); - void SetConnected(bool state); void SetStatus(int status); void DeleteCar(int ident); sockaddr_in GetUDPAddr(); + bool isConnected = false; std::string GetRole(); std::string GetName(); + bool isSynced = false; std::string GetDID(); SOCKET GetTCPSock(); - bool isConnected(); void SetID(int ID); - int GetCarCount(); int GetOpenCarID(); + int GetCarCount(); + void ClearCars(); int GetStatus(); int GetID(); }; +struct ClientInterface{ + std::set Clients; + void RemoveClient(Client *c){ + Clients.erase(c); + delete c; + c = nullptr; + } + void AddClient(Client *c){ + Clients.insert(c); + } + int Size(){ + return int(Clients.size()); + } +}; -extern std::set Clients; \ No newline at end of file +extern ClientInterface* CI; \ No newline at end of file diff --git a/include/Compressor.h b/include/Compressor.h new file mode 100644 index 0000000..1947ef5 --- /dev/null +++ b/include/Compressor.h @@ -0,0 +1,7 @@ +/// +/// Created by Anonymous275 on 7/24/2020 +/// +#pragma once +#include +std::string Comp(std::string Data); +std::string DeComp(std::string Compressed); diff --git a/include/Curl/Http.h b/include/Curl/Http.h new file mode 100644 index 0000000..c8ecc81 --- /dev/null +++ b/include/Curl/Http.h @@ -0,0 +1,7 @@ +/// +/// Created by Anonymous275 on 7/18/2020 +/// +#pragma once +#include +std::string HttpRequest(const std::string& IP,int port); +std::string PostHTTP(const std::string& IP,const std::string& Fields); \ No newline at end of file diff --git a/src/curl/curl.h b/include/Curl/curl.h similarity index 99% rename from src/curl/curl.h rename to include/Curl/curl.h index e0d5c0f..162b606 100644 --- a/src/curl/curl.h +++ b/include/Curl/curl.h @@ -60,8 +60,8 @@ defined(__LWIP_OPT_H__) || defined(LWIP_HDR_OPT_H)) /* The check above prevents the winsock2 inclusion if winsock.h already was included, since they can't co-exist without problems */ -#include -#include +#include +#include #endif #endif diff --git a/src/curl/curlbuild.h b/include/Curl/curlbuild.h similarity index 100% rename from src/curl/curlbuild.h rename to include/Curl/curlbuild.h diff --git a/src/curl/curlrules.h b/include/Curl/curlrules.h similarity index 100% rename from src/curl/curlrules.h rename to include/Curl/curlrules.h diff --git a/src/curl/curlver.h b/include/Curl/curlver.h similarity index 100% rename from src/curl/curlver.h rename to include/Curl/curlver.h diff --git a/src/curl/easy.h b/include/Curl/easy.h similarity index 100% rename from src/curl/easy.h rename to include/Curl/easy.h diff --git a/src/curl/mprintf.h b/include/Curl/mprintf.h similarity index 100% rename from src/curl/mprintf.h rename to include/Curl/mprintf.h diff --git a/src/curl/multi.h b/include/Curl/multi.h similarity index 100% rename from src/curl/multi.h rename to include/Curl/multi.h diff --git a/src/curl/stdcheaders.h b/include/Curl/stdcheaders.h similarity index 100% rename from src/curl/stdcheaders.h rename to include/Curl/stdcheaders.h diff --git a/src/curl/system.h b/include/Curl/system.h similarity index 100% rename from src/curl/system.h rename to include/Curl/system.h diff --git a/src/curl/typecheck-gcc.h b/include/Curl/typecheck-gcc.h similarity index 99% rename from src/curl/typecheck-gcc.h rename to include/Curl/typecheck-gcc.h index 6ec8bcf..315c264 100644 --- a/src/curl/typecheck-gcc.h +++ b/include/Curl/typecheck-gcc.h @@ -581,7 +581,7 @@ typedef CURLcode (_curl_ssl_ctx_callback2)(CURL *, void *, const void *); typedef CURLcode (_curl_ssl_ctx_callback3)(CURL *, const void *, void *); typedef CURLcode (_curl_ssl_ctx_callback4)(CURL *, const void *, const void *); #ifdef HEADER_SSL_H -/* hack: if we included OpenSSL's ssl.h, we know about SSL_CTX +/* hTRG: if we included OpenSSL's ssl.h, we know about SSL_CTX * this will of course break if we're included before OpenSSL headers... */ typedef CURLcode (_curl_ssl_ctx_callback5)(CURL *, SSL_CTX, void *); diff --git a/src/logger.h b/include/Logger.h similarity index 57% rename from src/logger.h rename to include/Logger.h index dd137c3..5c3a52f 100644 --- a/src/logger.h +++ b/include/Logger.h @@ -1,16 +1,12 @@ /// /// Created by Anonymous275 on 4/2/2020. /// - -#include -#include +#pragma once +#include #include - -extern int loggerlevel; -std::stringstream getDate(); -void setLoggerLevel(int level); +void InitLog(); +void except(const std::string& toPrint); +void debug(const std::string& toPrint); +void error(const std::string& toPrint); void info(const std::string& toPrint); void warn(const std::string& toPrint); -void error(const std::string& toPrint); -void debug(const std::string& toPrint); -void Exception(unsigned long Code,char* Origin); \ No newline at end of file diff --git a/src/Lua System/LuaSystem.hpp b/include/Lua/LuaSystem.hpp similarity index 85% rename from src/Lua System/LuaSystem.hpp rename to include/Lua/LuaSystem.hpp index b00fd9f..bc5729c 100644 --- a/src/Lua System/LuaSystem.hpp +++ b/include/Lua/LuaSystem.hpp @@ -3,14 +3,14 @@ /// #pragma once +#include +#include +#include "lua.hpp" +#include +#include #include #include -#include -#include -#include -#include -#include "../lua/lua.hpp" -namespace fs = std::filesystem; +namespace fs = std::experimental::filesystem; struct LuaArg{ std::vector args; void PushArgs(lua_State *State){ @@ -43,7 +43,6 @@ private: public: void RegisterEvent(const std::string&Event,const std::string&FunctionName); - int CallFunction(const std::string& FuncName,LuaArg* args); std::string GetRegistered(const std::string&Event); void UnRegisterEvent(const std::string&Event); void SetLastWrite(fs::file_time_type time); @@ -51,13 +50,17 @@ public: void SetPluginName(const std::string&Name); void SetFileName(const std::string&Name); fs::file_time_type GetLastWrite(); + bool isThreadExecuting = false; std::string GetPluginName(); std::string GetFileName(); + bool isExecuting = false; + bool StopThread = false; bool HasThread = false; lua_State* GetState(); char* GetOrigin(); void Reload(); void Init(); }; - +int CallFunction(Lua*lua,const std::string& FuncName,LuaArg* args); +int TriggerLuaEvent(const std::string& Event,bool local,Lua*Caller,LuaArg* arg); extern std::set PluginEngine; \ No newline at end of file diff --git a/src/lua/lauxlib.h b/include/Lua/lauxlib.h similarity index 100% rename from src/lua/lauxlib.h rename to include/Lua/lauxlib.h diff --git a/src/lua/lua.h b/include/Lua/lua.h similarity index 100% rename from src/lua/lua.h rename to include/Lua/lua.h diff --git a/src/lua/lua.hpp b/include/Lua/lua.hpp similarity index 100% rename from src/lua/lua.hpp rename to include/Lua/lua.hpp diff --git a/src/lua/luaconf.h b/include/Lua/luaconf.h similarity index 100% rename from src/lua/luaconf.h rename to include/Lua/luaconf.h diff --git a/src/lua/lualib.h b/include/Lua/lualib.h similarity index 100% rename from src/lua/lualib.h rename to include/Lua/lualib.h diff --git a/include/Network.h b/include/Network.h new file mode 100644 index 0000000..b22e8f2 --- /dev/null +++ b/include/Network.h @@ -0,0 +1,19 @@ +/// +/// Created by Anonymous275 on 7/31/2020 +/// +#pragma once +#include "Client.hpp" +#include +void TCPServerMain(); +void UpdatePlayers(); +void OnConnect(Client*c); +void InitClient(Client*c); +void SyncResources(Client*c); +[[noreturn]] void UDPServerMain(); +void OnDisconnect(Client*c,bool kicked); +void UDPSend(Client*c,std::string Data); +void TCPSend(Client*c,const std::string&Data); +void SendLarge(Client*c,const std::string&Data); +void GParser(Client*c, const std::string&Packet); +void Respond(Client*c, const std::string& MSG, bool Rel); +void SendToAll(Client*c, const std::string& Data, bool Self, bool Rel); diff --git a/include/Security/Enc.h b/include/Security/Enc.h new file mode 100644 index 0000000..11a35cf --- /dev/null +++ b/include/Security/Enc.h @@ -0,0 +1,16 @@ +/// +/// Created by Anonymous275 on 7/28/2020 +/// +#pragma once +#include +#include +#include "Xor.h" +struct RSA{ + int n = 0; + int e = 0; + int d = 0; +}; +std::string RSA_E(const std::string& Data, RSA*k); +std::string RSA_D(const std::string& Data, RSA*k); +int Handle(EXCEPTION_POINTERS *ep,char* Origin); +RSA* GenKey(); diff --git a/include/Security/Xor.h b/include/Security/Xor.h new file mode 100644 index 0000000..a7b2874 --- /dev/null +++ b/include/Security/Xor.h @@ -0,0 +1,130 @@ +/// +/// Created by Anonymous275 on 8/11/2020 +/// +#pragma once +#include +#include +#include + +#define BEGIN_NAMESPACE(x) namespace x { +#define END_NAMESPACE } + +BEGIN_NAMESPACE(XorCompileTime) + + constexpr auto time = __TIME__; + constexpr auto seed = static_cast(time[7]) + static_cast(time[6]) * 10 + static_cast(time[4]) * 60 + static_cast(time[3]) * 600 + static_cast< int >(time[1]) * 3600 + static_cast< int >(time[0]) * 36000; + +// 1988, Stephen Park and Keith Miller +// "Random Number Generators: Good Ones Are Hard To Find", considered as "minimal standard" +// Park-Miller 31 bit pseudo-random number generator, implemented with G. Carta's optimisation: +// with 32-bit math and without division + + template + struct RandomGenerator{ + private: + static constexpr unsigned a = 16807; // 7^5 + static constexpr unsigned m = 2147483647; // 2^31 - 1 + + static constexpr unsigned s = RandomGenerator< N - 1 >::value; + static constexpr unsigned lo = a * (s & 0xFFFFu); // Multiply lower 16 bits by 16807 + static constexpr unsigned hi = a * (s >> 16u); // Multiply higher 16 bits by 16807 + static constexpr unsigned lo2 = lo + ((hi & 0x7FFFu) << 16u); // Combine lower 15 bits of hi with lo's upper bits + static constexpr unsigned hi2 = hi >> 15u; // Discard lower 15 bits of hi + static constexpr unsigned lo3 = lo2 + hi; + + public: + static constexpr unsigned max = m; + static constexpr unsigned value = lo3 > m ? lo3 - m : lo3; + }; + + template <> + struct RandomGenerator< 0 >{ + static constexpr unsigned value = seed; + }; + + template + struct RandomInt{ + static constexpr auto value = RandomGenerator< N + 1 >::value % M; + }; + + template + struct RandomChar{ + static const char value = static_cast< char >(1 + RandomInt< N, 0x7F - 1 >::value); + }; + + template + struct XorString{ + private: + const char _key; + std::array< Char, N + 1 > _encrypted; + + constexpr Char enc(Char c) const{ + return c ^ _key; + } + + Char dec(Char c) const{ + return c ^ _key; + } + + public: + template + constexpr __forceinline XorString(const Char* str, std::index_sequence< Is... >) : _key(RandomChar< K >::value), _encrypted{ enc(str[Is])... } + {} + + __forceinline decltype(auto) decrypt(){ + for (size_t i = 0; i < N; ++i) { + _encrypted[i] = dec(_encrypted[i]); + } + _encrypted[N] = '\0'; + return _encrypted.data(); + } + }; + + static auto w_printf = [](const char* fmt, ...) { + va_list args; + va_start(args, fmt); + vprintf_s(fmt, args); + va_end(args); + }; + + static auto w_printf_s = [](const char* fmt, ...) { + va_list args; + va_start(args, fmt); + vprintf_s(fmt, args); + va_end(args); + }; + + /*static auto w_sprintf = [](char* buf, const char* fmt, ...) { + va_list args; + va_start(args, fmt); + vsprintf(buf, fmt, args); + va_end(args); + };*/ + + /*static auto w_sprintf_ret = [](char* buf, const char* fmt, ...) { + int ret; + va_list args; + va_start(args, fmt); + ret = vsprintf(buf, fmt, args); + va_end(args); + return ret; + };*/ + + static auto w_sprintf_s = [](char* buf, size_t buf_size, const char* fmt, ...) { + va_list args; + va_start(args, fmt); + vsprintf_s(buf, buf_size, fmt, args); + va_end(args); + }; + + static auto w_sprintf_s_ret = [](char* buf, size_t buf_size, const char* fmt, ...) { + int ret; + va_list args; + va_start(args, fmt); + ret = vsprintf_s(buf, buf_size, fmt, args); + va_end(args); + return ret; + }; +#define Sec( s ) []{ constexpr XorCompileTime::XorString< sizeof(s)/sizeof(char) - 1, __COUNTER__, char > expr( s, std::make_index_sequence< sizeof(s)/sizeof(char) - 1>() ); return expr; }().decrypt() +#define SecW( s ) []{ constexpr XorCompileTime::XorString< sizeof(s)/sizeof(wchar_t) - 1, __COUNTER__, wchar_t > expr( s, std::make_index_sequence< sizeof(s)/sizeof(wchar_t) - 1>() ); return expr; }().decrypt() +END_NAMESPACE diff --git a/src/Settings.hpp b/include/Settings.h similarity index 67% rename from src/Settings.hpp rename to include/Settings.h index 2e4512e..dc9d717 100644 --- a/src/Settings.hpp +++ b/include/Settings.h @@ -1,8 +1,8 @@ /// -/// Created by Anonymous275 on 4/10/2020 +/// Created by Anonymous275 on 7/28/2020 /// -extern std::string ServerVersion; -extern std::string ClientVersion; +#pragma once +#include extern std::string ServerName; extern std::string ServerDesc; extern std::string StatReport; @@ -11,12 +11,14 @@ extern std::string Resource; extern std::string FileList; extern std::string CustomIP; extern std::string MapName; +extern uint64_t MaxModSize; extern std::string Key; +std::string GetSVer(); +std::string GetCVer(); extern int MaxPlayers; extern int ModsLoaded; -extern int Beat; -extern long MaxModSize; extern bool Private; extern int MaxCars; extern bool Debug; -extern int Port; \ No newline at end of file +extern int Port; +extern int PPS; \ No newline at end of file diff --git a/include/Startup.h b/include/Startup.h new file mode 100644 index 0000000..89a1bf0 --- /dev/null +++ b/include/Startup.h @@ -0,0 +1,11 @@ +/// +/// Created by Anonymous275 on 7/28/2020 +/// +#pragma once +void InitServer(int argc, char* argv[]); +void InitConfig(); +void InitLua(); +void InitRes(); +void HBInit(); +void StatInit(); +void NetMain(); \ No newline at end of file diff --git a/include/Zlib/zconf.h b/include/Zlib/zconf.h new file mode 100644 index 0000000..352f552 --- /dev/null +++ b/include/Zlib/zconf.h @@ -0,0 +1,536 @@ +/* zconf.h -- configuration of the zlib compression library + * Copyright (C) 1995-2016 Jean-loup Gailly, Mark Adler + * For conditions of distribution and use, see copyright notice in zlib.h + */ + +/* @(#) $Id$ */ + +#ifndef ZCONF_H +#define ZCONF_H +/* #undef Z_PREFIX */ +/* #undef Z_HAVE_UNISTD_H */ + +/* + * If you *really* need a unique prefix for all types and library functions, + * compile with -DZ_PREFIX. The "standard" zlib should be compiled without it. + * Even better than compiling with -DZ_PREFIX would be to use configure to set + * this permanently in zconf.h using "./configure --zprefix". + */ +#ifdef Z_PREFIX /* may be set to #if 1 by ./configure */ +# define Z_PREFIX_SET + +/* all linked symbols and init macros */ +# define _dist_code z__dist_code +# define _length_code z__length_code +# define _tr_align z__tr_align +# define _tr_flush_bits z__tr_flush_bits +# define _tr_flush_block z__tr_flush_block +# define _tr_init z__tr_init +# define _tr_stored_block z__tr_stored_block +# define _tr_tally z__tr_tally +# define adler32 z_adler32 +# define adler32_combine z_adler32_combine +# define adler32_combine64 z_adler32_combine64 +# define adler32_z z_adler32_z +# ifndef Z_SOLO +# define compress z_compress +# define compress2 z_compress2 +# define compressBound z_compressBound +# endif +# define crc32 z_crc32 +# define crc32_combine z_crc32_combine +# define crc32_combine64 z_crc32_combine64 +# define crc32_z z_crc32_z +# define deflate z_deflate +# define deflateBound z_deflateBound +# define deflateCopy z_deflateCopy +# define deflateEnd z_deflateEnd +# define deflateGetDictionary z_deflateGetDictionary +# define deflateInit z_deflateInit +# define deflateInit2 z_deflateInit2 +# define deflateInit2_ z_deflateInit2_ +# define deflateInit_ z_deflateInit_ +# define deflateParams z_deflateParams +# define deflatePending z_deflatePending +# define deflatePrime z_deflatePrime +# define deflateReset z_deflateReset +# define deflateResetKeep z_deflateResetKeep +# define deflateSetDictionary z_deflateSetDictionary +# define deflateSetHeader z_deflateSetHeader +# define deflateTune z_deflateTune +# define deflate_copyright z_deflate_copyright +# define get_crc_table z_get_crc_table +# ifndef Z_SOLO +# define gz_error z_gz_error +# define gz_intmax z_gz_intmax +# define gz_strwinerror z_gz_strwinerror +# define gzbuffer z_gzbuffer +# define gzclearerr z_gzclearerr +# define gzclose z_gzclose +# define gzclose_r z_gzclose_r +# define gzclose_w z_gzclose_w +# define gzdirect z_gzdirect +# define gzdopen z_gzdopen +# define gzeof z_gzeof +# define gzerror z_gzerror +# define gzflush z_gzflush +# define gzfread z_gzfread +# define gzfwrite z_gzfwrite +# define gzgetc z_gzgetc +# define gzgetc_ z_gzgetc_ +# define gzgets z_gzgets +# define gzoffset z_gzoffset +# define gzoffset64 z_gzoffset64 +# define gzopen z_gzopen +# define gzopen64 z_gzopen64 +# ifdef _WIN32 +# define gzopen_w z_gzopen_w +# endif +# define gzprintf z_gzprintf +# define gzputc z_gzputc +# define gzputs z_gzputs +# define gzread z_gzread +# define gzrewind z_gzrewind +# define gzseek z_gzseek +# define gzseek64 z_gzseek64 +# define gzsetparams z_gzsetparams +# define gztell z_gztell +# define gztell64 z_gztell64 +# define gzungetc z_gzungetc +# define gzvprintf z_gzvprintf +# define gzwrite z_gzwrite +# endif +# define inflate z_inflate +# define inflateBack z_inflateBack +# define inflateBackEnd z_inflateBackEnd +# define inflateBackInit z_inflateBackInit +# define inflateBackInit_ z_inflateBackInit_ +# define inflateCodesUsed z_inflateCodesUsed +# define inflateCopy z_inflateCopy +# define inflateEnd z_inflateEnd +# define inflateGetDictionary z_inflateGetDictionary +# define inflateGetHeader z_inflateGetHeader +# define inflateInit z_inflateInit +# define inflateInit2 z_inflateInit2 +# define inflateInit2_ z_inflateInit2_ +# define inflateInit_ z_inflateInit_ +# define inflateMark z_inflateMark +# define inflatePrime z_inflatePrime +# define inflateReset z_inflateReset +# define inflateReset2 z_inflateReset2 +# define inflateResetKeep z_inflateResetKeep +# define inflateSetDictionary z_inflateSetDictionary +# define inflateSync z_inflateSync +# define inflateSyncPoint z_inflateSyncPoint +# define inflateUndermine z_inflateUndermine +# define inflateValidate z_inflateValidate +# define inflate_copyright z_inflate_copyright +# define inflate_fast z_inflate_fast +# define inflate_table z_inflate_table +# ifndef Z_SOLO +# define uncompress z_uncompress +# define uncompress2 z_uncompress2 +# endif +# define zError z_zError +# ifndef Z_SOLO +# define zcalloc z_zcalloc +# define zcfree z_zcfree +# endif +# define zlibCompileFlags z_zlibCompileFlags +# define zlibVersion z_zlibVersion + +/* all zlib typedefs in zlib.h and zconf.h */ +# define Byte z_Byte +# define Bytef z_Bytef +# define alloc_func z_alloc_func +# define charf z_charf +# define free_func z_free_func +# ifndef Z_SOLO +# define gzFile z_gzFile +# endif +# define gz_header z_gz_header +# define gz_headerp z_gz_headerp +# define in_func z_in_func +# define intf z_intf +# define out_func z_out_func +# define uInt z_uInt +# define uIntf z_uIntf +# define uLong z_uLong +# define uLongf z_uLongf +# define voidp z_voidp +# define voidpc z_voidpc +# define voidpf z_voidpf + +/* all zlib structs in zlib.h and zconf.h */ +# define gz_header_s z_gz_header_s +# define internal_state z_internal_state + +#endif + +#if defined(__MSDOS__) && !defined(MSDOS) +# define MSDOS +#endif +#if (defined(OS_2) || defined(__OS2__)) && !defined(OS2) +# define OS2 +#endif +#if defined(_WINDOWS) && !defined(WINDOWS) +# define WINDOWS +#endif +#if defined(_WIN32) || defined(_WIN32_WCE) || defined(__WIN32__) +# ifndef WIN32 +# define WIN32 +# endif +#endif +#if (defined(MSDOS) || defined(OS2) || defined(WINDOWS)) && !defined(WIN32) +# if !defined(__GNUC__) && !defined(__FLAT__) && !defined(__386__) +# ifndef SYS16BIT +# define SYS16BIT +# endif +# endif +#endif + +/* + * Compile with -DMAXSEG_64K if the alloc function cannot allocate more + * than 64k bytes at a time (needed on systems with 16-bit int). + */ +#ifdef SYS16BIT +# define MAXSEG_64K +#endif +#ifdef MSDOS +# define UNALIGNED_OK +#endif + +#ifdef __STDC_VERSION__ +# ifndef STDC +# define STDC +# endif +# if __STDC_VERSION__ >= 199901L +# ifndef STDC99 +# define STDC99 +# endif +# endif +#endif +#if !defined(STDC) && (defined(__STDC__) || defined(__cplusplus)) +# define STDC +#endif +#if !defined(STDC) && (defined(__GNUC__) || defined(__BORLANDC__)) +# define STDC +#endif +#if !defined(STDC) && (defined(MSDOS) || defined(WINDOWS) || defined(WIN32)) +# define STDC +#endif +#if !defined(STDC) && (defined(OS2) || defined(__HOS_AIX__)) +# define STDC +#endif + +#if defined(__OS400__) && !defined(STDC) /* iSeries (formerly AS/400). */ +# define STDC +#endif + +#ifndef STDC +# ifndef const /* cannot use !defined(STDC) && !defined(const) on Mac */ +# define const /* note: need a more gentle solution here */ +# endif +#endif + +#if defined(ZLIB_CONST) && !defined(z_const) +# define z_const const +#else +# define z_const +#endif + +#ifdef Z_SOLO + typedef unsigned long z_size_t; +#else +# define z_longlong long long +# if defined(NO_SIZE_T) + typedef unsigned NO_SIZE_T z_size_t; +# elif defined(STDC) +# include + typedef size_t z_size_t; +# else + typedef unsigned long z_size_t; +# endif +# undef z_longlong +#endif + +/* Maximum value for memLevel in deflateInit2 */ +#ifndef MAX_MEM_LEVEL +# ifdef MAXSEG_64K +# define MAX_MEM_LEVEL 8 +# else +# define MAX_MEM_LEVEL 9 +# endif +#endif + +/* Maximum value for windowBits in deflateInit2 and inflateInit2. + * WARNING: reducing MAX_WBITS makes minigzip unable to extract .gz files + * created by gzip. (Files created by minigzip can still be extracted by + * gzip.) + */ +#ifndef MAX_WBITS +# define MAX_WBITS 15 /* 32K LZ77 window */ +#endif + +/* The memory requirements for deflate are (in bytes): + (1 << (windowBits+2)) + (1 << (memLevel+9)) + that is: 128K for windowBits=15 + 128K for memLevel = 8 (default values) + plus a few kilobytes for small objects. For example, if you want to reduce + the default memory requirements from 256K to 128K, compile with + make CFLAGS="-O -DMAX_WBITS=14 -DMAX_MEM_LEVEL=7" + Of course this will generally degrade compression (there's no free lunch). + + The memory requirements for inflate are (in bytes) 1 << windowBits + that is, 32K for windowBits=15 (default value) plus about 7 kilobytes + for small objects. +*/ + + /* Type declarations */ + +#ifndef OF /* function prototypes */ +# ifdef STDC +# define OF(args) args +# else +# define OF(args) () +# endif +#endif + +#ifndef Z_ARG /* function prototypes for stdarg */ +# if defined(STDC) || defined(Z_HAVE_STDARG_H) +# define Z_ARG(args) args +# else +# define Z_ARG(args) () +# endif +#endif + +/* The following definitions for FAR are needed only for MSDOS mixed + * model programming (small or medium model with some far allocations). + * This was tested only with MSC; for other MSDOS compilers you may have + * to define NO_MEMCPY in zutil.h. If you don't need the mixed model, + * just define FAR to be empty. + */ +#ifdef SYS16BIT +# if defined(M_I86SM) || defined(M_I86MM) + /* MSC small or medium model */ +# define SMALL_MEDIUM +# ifdef _MSC_VER +# define FAR _far +# else +# define FAR far +# endif +# endif +# if (defined(__SMALL__) || defined(__MEDIUM__)) + /* Turbo C small or medium model */ +# define SMALL_MEDIUM +# ifdef __BORLANDC__ +# define FAR _far +# else +# define FAR far +# endif +# endif +#endif + +#if defined(WINDOWS) || defined(WIN32) + /* If building or using zlib as a DLL, define ZLIB_DLL. + * This is not mandatory, but it offers a little performance increase. + */ +# ifdef ZLIB_DLL +# if defined(WIN32) && (!defined(__BORLANDC__) || (__BORLANDC__ >= 0x500)) +# ifdef ZLIB_INTERNAL +# define ZEXTERN extern __declspec(dllexport) +# else +# define ZEXTERN extern __declspec(dllimport) +# endif +# endif +# endif /* ZLIB_DLL */ + /* If building or using zlib with the WINAPI/WINAPIV calling convention, + * define ZLIB_WINAPI. + * Caution: the standard ZLIB1.DLL is NOT compiled using ZLIB_WINAPI. + */ +# ifdef ZLIB_WINAPI +# ifdef FAR +# undef FAR +# endif +# include + /* No need for _export, use ZLIB.DEF instead. */ + /* For complete Windows compatibility, use WINAPI, not __stdcall. */ +# define ZEXPORT WINAPI +# ifdef WIN32 +# define ZEXPORTVA WINAPIV +# else +# define ZEXPORTVA FAR CDECL +# endif +# endif +#endif + +#if defined (__BEOS__) +# ifdef ZLIB_DLL +# ifdef ZLIB_INTERNAL +# define ZEXPORT __declspec(dllexport) +# define ZEXPORTVA __declspec(dllexport) +# else +# define ZEXPORT __declspec(dllimport) +# define ZEXPORTVA __declspec(dllimport) +# endif +# endif +#endif + +#ifndef ZEXTERN +# define ZEXTERN extern +#endif +#ifndef ZEXPORT +# define ZEXPORT +#endif +#ifndef ZEXPORTVA +# define ZEXPORTVA +#endif + +#ifndef FAR +# define FAR +#endif + +#if !defined(__MACTYPES__) +typedef unsigned char Byte; /* 8 bits */ +#endif +typedef unsigned int uInt; /* 16 bits or more */ +typedef unsigned long uLong; /* 32 bits or more */ + +#ifdef SMALL_MEDIUM + /* Borland C/C++ and some old MSC versions ignore FAR inside typedef */ +# define Bytef Byte FAR +#else + typedef Byte FAR Bytef; +#endif +typedef char FAR charf; +typedef int FAR intf; +typedef uInt FAR uIntf; +typedef uLong FAR uLongf; + +#ifdef STDC + typedef void const *voidpc; + typedef void FAR *voidpf; + typedef void *voidp; +#else + typedef Byte const *voidpc; + typedef Byte FAR *voidpf; + typedef Byte *voidp; +#endif + +#if !defined(Z_U4) && !defined(Z_SOLO) && defined(STDC) +# include +# if (UINT_MAX == 0xffffffffUL) +# define Z_U4 unsigned +# elif (ULONG_MAX == 0xffffffffUL) +# define Z_U4 unsigned long +# elif (USHRT_MAX == 0xffffffffUL) +# define Z_U4 unsigned short +# endif +#endif + +#ifdef Z_U4 + typedef Z_U4 z_crc_t; +#else + typedef unsigned long z_crc_t; +#endif + +#ifdef HAVE_UNISTD_H /* may be set to #if 1 by ./configure */ +# define Z_HAVE_UNISTD_H +#endif + +#ifdef HAVE_STDARG_H /* may be set to #if 1 by ./configure */ +# define Z_HAVE_STDARG_H +#endif + +#ifdef STDC +# ifndef Z_SOLO +# include /* for off_t */ +# endif +#endif + +#if defined(STDC) || defined(Z_HAVE_STDARG_H) +# ifndef Z_SOLO +# include /* for va_list */ +# endif +#endif + +#ifdef _WIN32 +# ifndef Z_SOLO +# include /* for wchar_t */ +# endif +#endif + +/* a little trick to accommodate both "#define _LARGEFILE64_SOURCE" and + * "#define _LARGEFILE64_SOURCE 1" as requesting 64-bit operations, (even + * though the former does not conform to the LFS document), but considering + * both "#undef _LARGEFILE64_SOURCE" and "#define _LARGEFILE64_SOURCE 0" as + * equivalently requesting no 64-bit operations + */ +#if defined(_LARGEFILE64_SOURCE) && -_LARGEFILE64_SOURCE - -1 == 1 +# undef _LARGEFILE64_SOURCE +#endif + +#if defined(__WATCOMC__) && !defined(Z_HAVE_UNISTD_H) +# define Z_HAVE_UNISTD_H +#endif +#ifndef Z_SOLO +# if defined(Z_HAVE_UNISTD_H) || defined(_LARGEFILE64_SOURCE) +# include /* for SEEK_*, off_t, and _LFS64_LARGEFILE */ +# ifdef VMS +# include /* for off_t */ +# endif +# ifndef z_off_t +# define z_off_t off_t +# endif +# endif +#endif + +#if defined(_LFS64_LARGEFILE) && _LFS64_LARGEFILE-0 +# define Z_LFS64 +#endif + +#if defined(_LARGEFILE64_SOURCE) && defined(Z_LFS64) +# define Z_LARGE64 +#endif + +#if defined(_FILE_OFFSET_BITS) && _FILE_OFFSET_BITS-0 == 64 && defined(Z_LFS64) +# define Z_WANT64 +#endif + +#if !defined(SEEK_SET) && !defined(Z_SOLO) +# define SEEK_SET 0 /* Seek from beginning of file. */ +# define SEEK_CUR 1 /* Seek from current position. */ +# define SEEK_END 2 /* Set file pointer to EOF plus "offset" */ +#endif + +#ifndef z_off_t +# define z_off_t long +#endif + +#if !defined(_WIN32) && defined(Z_LARGE64) +# define z_off64_t off64_t +#else +# if defined(_WIN32) && !defined(__GNUC__) && !defined(Z_SOLO) +# define z_off64_t __int64 +# else +# define z_off64_t z_off_t +# endif +#endif + +/* MVS linker does not support external names larger than 8 bytes */ +#if defined(__MVS__) + #pragma map(deflateInit_,"DEIN") + #pragma map(deflateInit2_,"DEIN2") + #pragma map(deflateEnd,"DEEND") + #pragma map(deflateBound,"DEBND") + #pragma map(inflateInit_,"ININ") + #pragma map(inflateInit2_,"ININ2") + #pragma map(inflateEnd,"INEND") + #pragma map(inflateSync,"INSY") + #pragma map(inflateSetDictionary,"INSEDI") + #pragma map(compressBound,"CMBND") + #pragma map(inflate_table,"INTABL") + #pragma map(inflate_fast,"INFA") + #pragma map(inflate_copyright,"INCOPY") +#endif + +#endif /* ZCONF_H */ diff --git a/include/Zlib/zlib.h b/include/Zlib/zlib.h new file mode 100644 index 0000000..f09cdaf --- /dev/null +++ b/include/Zlib/zlib.h @@ -0,0 +1,1912 @@ +/* zlib.h -- interface of the 'zlib' general purpose compression library + version 1.2.11, January 15th, 2017 + + Copyright (C) 1995-2017 Jean-loup Gailly and Mark Adler + + This software is provided 'as-is', without any express or implied + warranty. In no event will the authors be held liable for any damages + arising from the use of this software. + + Permission is granted to anyone to use this software for any purpose, + including commercial applications, and to alter it and redistribute it + freely, subject to the following restrictions: + + 1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. + 2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. + 3. This notice may not be removed or altered from any source distribution. + + Jean-loup Gailly Mark Adler + jloup@gzip.org madler@alumni.caltech.edu + + + The data format used by the zlib library is described by RFCs (Request for + Comments) 1950 to 1952 in the files http://tools.ietf.org/html/rfc1950 + (zlib format), rfc1951 (deflate format) and rfc1952 (gzip format). +*/ + +#ifndef ZLIB_H +#define ZLIB_H + +#include "zconf.h" + +#ifdef __cplusplus +extern "C" { +#endif + +#define ZLIB_VERSION "1.2.11" +#define ZLIB_VERNUM 0x12b0 +#define ZLIB_VER_MAJOR 1 +#define ZLIB_VER_MINOR 2 +#define ZLIB_VER_REVISION 11 +#define ZLIB_VER_SUBREVISION 0 + +/* + The 'zlib' compression library provides in-memory compression and + decompression functions, including integrity checks of the uncompressed data. + This version of the library supports only one compression method (deflation) + but other algorithms will be added later and will have the same stream + interface. + + Compression can be done in a single step if the buffers are large enough, + or can be done by repeated calls of the compression function. In the latter + case, the application must provide more input and/or consume the output + (providing more output space) before each call. + + The compressed data format used by default by the in-memory functions is + the zlib format, which is a zlib wrapper documented in RFC 1950, wrapped + around a deflate stream, which is itself documented in RFC 1951. + + The library also supports reading and writing files in gzip (.gz) format + with an interface similar to that of stdio using the functions that start + with "gz". The gzip format is different from the zlib format. gzip is a + gzip wrapper, documented in RFC 1952, wrapped around a deflate stream. + + This library can optionally read and write gzip and raw deflate streams in + memory as well. + + The zlib format was designed to be compact and fast for use in memory + and on communications channels. The gzip format was designed for single- + file compression on file systems, has a larger header than zlib to maintain + directory information, and uses a different, slower check method than zlib. + + The library does not install any signal handler. The decoder checks + the consistency of the compressed data, so the library should never crash + even in the case of corrupted input. +*/ + +typedef voidpf (*alloc_func) OF((voidpf opaque, uInt items, uInt size)); +typedef void (*free_func) OF((voidpf opaque, voidpf address)); + +struct internal_state; + +typedef struct z_stream_s { + z_const Bytef *next_in; /* next input byte */ + uInt avail_in; /* number of bytes available at next_in */ + uLong total_in; /* total number of input bytes read so far */ + + Bytef *next_out; /* next output byte will go here */ + uInt avail_out; /* remaining free space at next_out */ + uLong total_out; /* total number of bytes output so far */ + + z_const char *msg; /* last error message, NULL if no error */ + struct internal_state FAR *state; /* not visible by applications */ + + alloc_func zalloc; /* used to allocate the internal state */ + free_func zfree; /* used to free the internal state */ + voidpf opaque; /* private data object passed to zalloc and zfree */ + + int data_type; /* best guess about the data type: binary or text + for deflate, or the decoding state for inflate */ + uLong adler; /* Adler-32 or CRC-32 value of the uncompressed data */ + uLong reserved; /* reserved for future use */ +} z_stream; + +typedef z_stream FAR *z_streamp; + +/* + gzip header information passed to and from zlib routines. See RFC 1952 + for more details on the meanings of these fields. +*/ +typedef struct gz_header_s { + int text; /* true if compressed data believed to be text */ + uLong time; /* modification time */ + int xflags; /* extra flags (not used when writing a gzip file) */ + int os; /* operating system */ + Bytef *extra; /* pointer to extra field or Z_NULL if none */ + uInt extra_len; /* extra field length (valid if extra != Z_NULL) */ + uInt extra_max; /* space at extra (only when reading header) */ + Bytef *name; /* pointer to zero-terminated file name or Z_NULL */ + uInt name_max; /* space at name (only when reading header) */ + Bytef *comment; /* pointer to zero-terminated comment or Z_NULL */ + uInt comm_max; /* space at comment (only when reading header) */ + int hcrc; /* true if there was or will be a header crc */ + int done; /* true when done reading gzip header (not used + when writing a gzip file) */ +} gz_header; + +typedef gz_header FAR *gz_headerp; + +/* + The application must update next_in and avail_in when avail_in has dropped + to zero. It must update next_out and avail_out when avail_out has dropped + to zero. The application must initialize zalloc, zfree and opaque before + calling the init function. All other fields are set by the compression + library and must not be updated by the application. + + The opaque value provided by the application will be passed as the first + parameter for calls of zalloc and zfree. This can be useful for custom + memory management. The compression library attaches no meaning to the + opaque value. + + zalloc must return Z_NULL if there is not enough memory for the object. + If zlib is used in a multi-threaded application, zalloc and zfree must be + thread safe. In that case, zlib is thread-safe. When zalloc and zfree are + Z_NULL on entry to the initialization function, they are set to internal + routines that use the standard library functions malloc() and free(). + + On 16-bit systems, the functions zalloc and zfree must be able to allocate + exactly 65536 bytes, but will not be required to allocate more than this if + the symbol MAXSEG_64K is defined (see zconf.h). WARNING: On MSDOS, pointers + returned by zalloc for objects of exactly 65536 bytes *must* have their + offset normalized to zero. The default allocation function provided by this + library ensures this (see zutil.c). To reduce memory requirements and avoid + any allocation of 64K objects, at the expense of compression ratio, compile + the library with -DMAX_WBITS=14 (see zconf.h). + + The fields total_in and total_out can be used for statistics or progress + reports. After compression, total_in holds the total size of the + uncompressed data and may be saved for use by the decompressor (particularly + if the decompressor wants to decompress everything in a single step). +*/ + + /* constants */ + +#define Z_NO_FLUSH 0 +#define Z_PARTIAL_FLUSH 1 +#define Z_SYNC_FLUSH 2 +#define Z_FULL_FLUSH 3 +#define Z_FINISH 4 +#define Z_BLOCK 5 +#define Z_TREES 6 +/* Allowed flush values; see deflate() and inflate() below for details */ + +#define Z_OK 0 +#define Z_STREAM_END 1 +#define Z_NEED_DICT 2 +#define Z_ERRNO (-1) +#define Z_STREAM_ERROR (-2) +#define Z_DATA_ERROR (-3) +#define Z_MEM_ERROR (-4) +#define Z_BUF_ERROR (-5) +#define Z_VERSION_ERROR (-6) +/* Return codes for the compression/decompression functions. Negative values + * are errors, positive values are used for special but normal events. + */ + +#define Z_NO_COMPRESSION 0 +#define Z_BEST_SPEED 1 +#define Z_BEST_COMPRESSION 9 +#define Z_DEFAULT_COMPRESSION (-1) +/* compression levels */ + +#define Z_FILTERED 1 +#define Z_HUFFMAN_ONLY 2 +#define Z_RLE 3 +#define Z_FIXED 4 +#define Z_DEFAULT_STRATEGY 0 +/* compression strategy; see deflateInit2() below for details */ + +#define Z_BINARY 0 +#define Z_TEXT 1 +#define Z_ASCII Z_TEXT /* for compatibility with 1.2.2 and earlier */ +#define Z_UNKNOWN 2 +/* Possible values of the data_type field for deflate() */ + +#define Z_DEFLATED 8 +/* The deflate compression method (the only one supported in this version) */ + +#define Z_NULL 0 /* for initializing zalloc, zfree, opaque */ + +#define zlib_version zlibVersion() +/* for compatibility with versions < 1.0.2 */ + + + /* basic functions */ + +ZEXTERN const char * ZEXPORT zlibVersion OF((void)); +/* The application can compare zlibVersion and ZLIB_VERSION for consistency. + If the first character differs, the library code actually used is not + compatible with the zlib.h header file used by the application. This check + is automatically made by deflateInit and inflateInit. + */ + +/* +ZEXTERN int ZEXPORT deflateInit OF((z_streamp strm, int level)); + + Initializes the internal stream state for compression. The fields + zalloc, zfree and opaque must be initialized before by the caller. If + zalloc and zfree are set to Z_NULL, deflateInit updates them to use default + allocation functions. + + The compression level must be Z_DEFAULT_COMPRESSION, or between 0 and 9: + 1 gives best speed, 9 gives best compression, 0 gives no compression at all + (the input data is simply copied a block at a time). Z_DEFAULT_COMPRESSION + requests a default compromise between speed and compression (currently + equivalent to level 6). + + deflateInit returns Z_OK if success, Z_MEM_ERROR if there was not enough + memory, Z_STREAM_ERROR if level is not a valid compression level, or + Z_VERSION_ERROR if the zlib library version (zlib_version) is incompatible + with the version assumed by the caller (ZLIB_VERSION). msg is set to null + if there is no error message. deflateInit does not perform any compression: + this will be done by deflate(). +*/ + + +ZEXTERN int ZEXPORT deflate OF((z_streamp strm, int flush)); +/* + deflate compresses as much data as possible, and stops when the input + buffer becomes empty or the output buffer becomes full. It may introduce + some output latency (reading input without producing any output) except when + forced to flush. + + The detailed semantics are as follows. deflate performs one or both of the + following actions: + + - Compress more input starting at next_in and update next_in and avail_in + accordingly. If not all input can be processed (because there is not + enough room in the output buffer), next_in and avail_in are updated and + processing will resume at this point for the next call of deflate(). + + - Generate more output starting at next_out and update next_out and avail_out + accordingly. This action is forced if the parameter flush is non zero. + Forcing flush frequently degrades the compression ratio, so this parameter + should be set only when necessary. Some output may be provided even if + flush is zero. + + Before the call of deflate(), the application should ensure that at least + one of the actions is possible, by providing more input and/or consuming more + output, and updating avail_in or avail_out accordingly; avail_out should + never be zero before the call. The application can consume the compressed + output when it wants, for example when the output buffer is full (avail_out + == 0), or after each call of deflate(). If deflate returns Z_OK and with + zero avail_out, it must be called again after making room in the output + buffer because there might be more output pending. See deflatePending(), + which can be used if desired to determine whether or not there is more ouput + in that case. + + Normally the parameter flush is set to Z_NO_FLUSH, which allows deflate to + decide how much data to accumulate before producing output, in order to + maximize compression. + + If the parameter flush is set to Z_SYNC_FLUSH, all pending output is + flushed to the output buffer and the output is aligned on a byte boundary, so + that the decompressor can get all input data available so far. (In + particular avail_in is zero after the call if enough output space has been + provided before the call.) Flushing may degrade compression for some + compression algorithms and so it should be used only when necessary. This + completes the current deflate block and follows it with an empty stored block + that is three bits plus filler bits to the next byte, followed by four bytes + (00 00 ff ff). + + If flush is set to Z_PARTIAL_FLUSH, all pending output is flushed to the + output buffer, but the output is not aligned to a byte boundary. All of the + input data so far will be available to the decompressor, as for Z_SYNC_FLUSH. + This completes the current deflate block and follows it with an empty fixed + codes block that is 10 bits long. This assures that enough bytes are output + in order for the decompressor to finish the block before the empty fixed + codes block. + + If flush is set to Z_BLOCK, a deflate block is completed and emitted, as + for Z_SYNC_FLUSH, but the output is not aligned on a byte boundary, and up to + seven bits of the current block are held to be written as the next byte after + the next deflate block is completed. In this case, the decompressor may not + be provided enough bits at this point in order to complete decompression of + the data provided so far to the compressor. It may need to wait for the next + block to be emitted. This is for advanced applications that need to control + the emission of deflate blocks. + + If flush is set to Z_FULL_FLUSH, all output is flushed as with + Z_SYNC_FLUSH, and the compression state is reset so that decompression can + restart from this point if previous compressed data has been damaged or if + random access is desired. Using Z_FULL_FLUSH too often can seriously degrade + compression. + + If deflate returns with avail_out == 0, this function must be called again + with the same value of the flush parameter and more output space (updated + avail_out), until the flush is complete (deflate returns with non-zero + avail_out). In the case of a Z_FULL_FLUSH or Z_SYNC_FLUSH, make sure that + avail_out is greater than six to avoid repeated flush markers due to + avail_out == 0 on return. + + If the parameter flush is set to Z_FINISH, pending input is processed, + pending output is flushed and deflate returns with Z_STREAM_END if there was + enough output space. If deflate returns with Z_OK or Z_BUF_ERROR, this + function must be called again with Z_FINISH and more output space (updated + avail_out) but no more input data, until it returns with Z_STREAM_END or an + error. After deflate has returned Z_STREAM_END, the only possible operations + on the stream are deflateReset or deflateEnd. + + Z_FINISH can be used in the first deflate call after deflateInit if all the + compression is to be done in a single step. In order to complete in one + call, avail_out must be at least the value returned by deflateBound (see + below). Then deflate is guaranteed to return Z_STREAM_END. If not enough + output space is provided, deflate will not return Z_STREAM_END, and it must + be called again as described above. + + deflate() sets strm->adler to the Adler-32 checksum of all input read + so far (that is, total_in bytes). If a gzip stream is being generated, then + strm->adler will be the CRC-32 checksum of the input read so far. (See + deflateInit2 below.) + + deflate() may update strm->data_type if it can make a good guess about + the input data type (Z_BINARY or Z_TEXT). If in doubt, the data is + considered binary. This field is only for information purposes and does not + affect the compression algorithm in any manner. + + deflate() returns Z_OK if some progress has been made (more input + processed or more output produced), Z_STREAM_END if all input has been + consumed and all output has been produced (only when flush is set to + Z_FINISH), Z_STREAM_ERROR if the stream state was inconsistent (for example + if next_in or next_out was Z_NULL or the state was inadvertently written over + by the application), or Z_BUF_ERROR if no progress is possible (for example + avail_in or avail_out was zero). Note that Z_BUF_ERROR is not fatal, and + deflate() can be called again with more input and more output space to + continue compressing. +*/ + + +ZEXTERN int ZEXPORT deflateEnd OF((z_streamp strm)); +/* + All dynamically allocated data structures for this stream are freed. + This function discards any unprocessed input and does not flush any pending + output. + + deflateEnd returns Z_OK if success, Z_STREAM_ERROR if the + stream state was inconsistent, Z_DATA_ERROR if the stream was freed + prematurely (some input or output was discarded). In the error case, msg + may be set but then points to a static string (which must not be + deallocated). +*/ + + +/* +ZEXTERN int ZEXPORT inflateInit OF((z_streamp strm)); + + Initializes the internal stream state for decompression. The fields + next_in, avail_in, zalloc, zfree and opaque must be initialized before by + the caller. In the current version of inflate, the provided input is not + read or consumed. The allocation of a sliding window will be deferred to + the first call of inflate (if the decompression does not complete on the + first call). If zalloc and zfree are set to Z_NULL, inflateInit updates + them to use default allocation functions. + + inflateInit returns Z_OK if success, Z_MEM_ERROR if there was not enough + memory, Z_VERSION_ERROR if the zlib library version is incompatible with the + version assumed by the caller, or Z_STREAM_ERROR if the parameters are + invalid, such as a null pointer to the structure. msg is set to null if + there is no error message. inflateInit does not perform any decompression. + Actual decompression will be done by inflate(). So next_in, and avail_in, + next_out, and avail_out are unused and unchanged. The current + implementation of inflateInit() does not process any header information -- + that is deferred until inflate() is called. +*/ + + +ZEXTERN int ZEXPORT inflate OF((z_streamp strm, int flush)); +/* + inflate decompresses as much data as possible, and stops when the input + buffer becomes empty or the output buffer becomes full. It may introduce + some output latency (reading input without producing any output) except when + forced to flush. + + The detailed semantics are as follows. inflate performs one or both of the + following actions: + + - Decompress more input starting at next_in and update next_in and avail_in + accordingly. If not all input can be processed (because there is not + enough room in the output buffer), then next_in and avail_in are updated + accordingly, and processing will resume at this point for the next call of + inflate(). + + - Generate more output starting at next_out and update next_out and avail_out + accordingly. inflate() provides as much output as possible, until there is + no more input data or no more space in the output buffer (see below about + the flush parameter). + + Before the call of inflate(), the application should ensure that at least + one of the actions is possible, by providing more input and/or consuming more + output, and updating the next_* and avail_* values accordingly. If the + caller of inflate() does not provide both available input and available + output space, it is possible that there will be no progress made. The + application can consume the uncompressed output when it wants, for example + when the output buffer is full (avail_out == 0), or after each call of + inflate(). If inflate returns Z_OK and with zero avail_out, it must be + called again after making room in the output buffer because there might be + more output pending. + + The flush parameter of inflate() can be Z_NO_FLUSH, Z_SYNC_FLUSH, Z_FINISH, + Z_BLOCK, or Z_TREES. Z_SYNC_FLUSH requests that inflate() flush as much + output as possible to the output buffer. Z_BLOCK requests that inflate() + stop if and when it gets to the next deflate block boundary. When decoding + the zlib or gzip format, this will cause inflate() to return immediately + after the header and before the first block. When doing a raw inflate, + inflate() will go ahead and process the first block, and will return when it + gets to the end of that block, or when it runs out of data. + + The Z_BLOCK option assists in appending to or combining deflate streams. + To assist in this, on return inflate() always sets strm->data_type to the + number of unused bits in the last byte taken from strm->next_in, plus 64 if + inflate() is currently decoding the last block in the deflate stream, plus + 128 if inflate() returned immediately after decoding an end-of-block code or + decoding the complete header up to just before the first byte of the deflate + stream. The end-of-block will not be indicated until all of the uncompressed + data from that block has been written to strm->next_out. The number of + unused bits may in general be greater than seven, except when bit 7 of + data_type is set, in which case the number of unused bits will be less than + eight. data_type is set as noted here every time inflate() returns for all + flush options, and so can be used to determine the amount of currently + consumed input in bits. + + The Z_TREES option behaves as Z_BLOCK does, but it also returns when the + end of each deflate block header is reached, before any actual data in that + block is decoded. This allows the caller to determine the length of the + deflate block header for later use in random access within a deflate block. + 256 is added to the value of strm->data_type when inflate() returns + immediately after reaching the end of the deflate block header. + + inflate() should normally be called until it returns Z_STREAM_END or an + error. However if all decompression is to be performed in a single step (a + single call of inflate), the parameter flush should be set to Z_FINISH. In + this case all pending input is processed and all pending output is flushed; + avail_out must be large enough to hold all of the uncompressed data for the + operation to complete. (The size of the uncompressed data may have been + saved by the compressor for this purpose.) The use of Z_FINISH is not + required to perform an inflation in one step. However it may be used to + inform inflate that a faster approach can be used for the single inflate() + call. Z_FINISH also informs inflate to not maintain a sliding window if the + stream completes, which reduces inflate's memory footprint. If the stream + does not complete, either because not all of the stream is provided or not + enough output space is provided, then a sliding window will be allocated and + inflate() can be called again to continue the operation as if Z_NO_FLUSH had + been used. + + In this implementation, inflate() always flushes as much output as + possible to the output buffer, and always uses the faster approach on the + first call. So the effects of the flush parameter in this implementation are + on the return value of inflate() as noted below, when inflate() returns early + when Z_BLOCK or Z_TREES is used, and when inflate() avoids the allocation of + memory for a sliding window when Z_FINISH is used. + + If a preset dictionary is needed after this call (see inflateSetDictionary + below), inflate sets strm->adler to the Adler-32 checksum of the dictionary + chosen by the compressor and returns Z_NEED_DICT; otherwise it sets + strm->adler to the Adler-32 checksum of all output produced so far (that is, + total_out bytes) and returns Z_OK, Z_STREAM_END or an error code as described + below. At the end of the stream, inflate() checks that its computed Adler-32 + checksum is equal to that saved by the compressor and returns Z_STREAM_END + only if the checksum is correct. + + inflate() can decompress and check either zlib-wrapped or gzip-wrapped + deflate data. The header type is detected automatically, if requested when + initializing with inflateInit2(). Any information contained in the gzip + header is not retained unless inflateGetHeader() is used. When processing + gzip-wrapped deflate data, strm->adler32 is set to the CRC-32 of the output + produced so far. The CRC-32 is checked against the gzip trailer, as is the + uncompressed length, modulo 2^32. + + inflate() returns Z_OK if some progress has been made (more input processed + or more output produced), Z_STREAM_END if the end of the compressed data has + been reached and all uncompressed output has been produced, Z_NEED_DICT if a + preset dictionary is needed at this point, Z_DATA_ERROR if the input data was + corrupted (input stream not conforming to the zlib format or incorrect check + value, in which case strm->msg points to a string with a more specific + error), Z_STREAM_ERROR if the stream structure was inconsistent (for example + next_in or next_out was Z_NULL, or the state was inadvertently written over + by the application), Z_MEM_ERROR if there was not enough memory, Z_BUF_ERROR + if no progress was possible or if there was not enough room in the output + buffer when Z_FINISH is used. Note that Z_BUF_ERROR is not fatal, and + inflate() can be called again with more input and more output space to + continue decompressing. If Z_DATA_ERROR is returned, the application may + then call inflateSync() to look for a good compression block if a partial + recovery of the data is to be attempted. +*/ + + +ZEXTERN int ZEXPORT inflateEnd OF((z_streamp strm)); +/* + All dynamically allocated data structures for this stream are freed. + This function discards any unprocessed input and does not flush any pending + output. + + inflateEnd returns Z_OK if success, or Z_STREAM_ERROR if the stream state + was inconsistent. +*/ + + + /* Advanced functions */ + +/* + The following functions are needed only in some special applications. +*/ + +/* +ZEXTERN int ZEXPORT deflateInit2 OF((z_streamp strm, + int level, + int method, + int windowBits, + int memLevel, + int strategy)); + + This is another version of deflateInit with more compression options. The + fields next_in, zalloc, zfree and opaque must be initialized before by the + caller. + + The method parameter is the compression method. It must be Z_DEFLATED in + this version of the library. + + The windowBits parameter is the base two logarithm of the window size + (the size of the history buffer). It should be in the range 8..15 for this + version of the library. Larger values of this parameter result in better + compression at the expense of memory usage. The default value is 15 if + deflateInit is used instead. + + For the current implementation of deflate(), a windowBits value of 8 (a + window size of 256 bytes) is not supported. As a result, a request for 8 + will result in 9 (a 512-byte window). In that case, providing 8 to + inflateInit2() will result in an error when the zlib header with 9 is + checked against the initialization of inflate(). The remedy is to not use 8 + with deflateInit2() with this initialization, or at least in that case use 9 + with inflateInit2(). + + windowBits can also be -8..-15 for raw deflate. In this case, -windowBits + determines the window size. deflate() will then generate raw deflate data + with no zlib header or trailer, and will not compute a check value. + + windowBits can also be greater than 15 for optional gzip encoding. Add + 16 to windowBits to write a simple gzip header and trailer around the + compressed data instead of a zlib wrapper. The gzip header will have no + file name, no extra data, no comment, no modification time (set to zero), no + header crc, and the operating system will be set to the appropriate value, + if the operating system was determined at compile time. If a gzip stream is + being written, strm->adler is a CRC-32 instead of an Adler-32. + + For raw deflate or gzip encoding, a request for a 256-byte window is + rejected as invalid, since only the zlib header provides a means of + transmitting the window size to the decompressor. + + The memLevel parameter specifies how much memory should be allocated + for the internal compression state. memLevel=1 uses minimum memory but is + slow and reduces compression ratio; memLevel=9 uses maximum memory for + optimal speed. The default value is 8. See zconf.h for total memory usage + as a function of windowBits and memLevel. + + The strategy parameter is used to tune the compression algorithm. Use the + value Z_DEFAULT_STRATEGY for normal data, Z_FILTERED for data produced by a + filter (or predictor), Z_HUFFMAN_ONLY to force Huffman encoding only (no + string match), or Z_RLE to limit match distances to one (run-length + encoding). Filtered data consists mostly of small values with a somewhat + random distribution. In this case, the compression algorithm is tuned to + compress them better. The effect of Z_FILTERED is to force more Huffman + coding and less string matching; it is somewhat intermediate between + Z_DEFAULT_STRATEGY and Z_HUFFMAN_ONLY. Z_RLE is designed to be almost as + fast as Z_HUFFMAN_ONLY, but give better compression for PNG image data. The + strategy parameter only affects the compression ratio but not the + correctness of the compressed output even if it is not set appropriately. + Z_FIXED prevents the use of dynamic Huffman codes, allowing for a simpler + decoder for special applications. + + deflateInit2 returns Z_OK if success, Z_MEM_ERROR if there was not enough + memory, Z_STREAM_ERROR if any parameter is invalid (such as an invalid + method), or Z_VERSION_ERROR if the zlib library version (zlib_version) is + incompatible with the version assumed by the caller (ZLIB_VERSION). msg is + set to null if there is no error message. deflateInit2 does not perform any + compression: this will be done by deflate(). +*/ + +ZEXTERN int ZEXPORT deflateSetDictionary OF((z_streamp strm, + const Bytef *dictionary, + uInt dictLength)); +/* + Initializes the compression dictionary from the given byte sequence + without producing any compressed output. When using the zlib format, this + function must be called immediately after deflateInit, deflateInit2 or + deflateReset, and before any call of deflate. When doing raw deflate, this + function must be called either before any call of deflate, or immediately + after the completion of a deflate block, i.e. after all input has been + consumed and all output has been delivered when using any of the flush + options Z_BLOCK, Z_PARTIAL_FLUSH, Z_SYNC_FLUSH, or Z_FULL_FLUSH. The + compressor and decompressor must use exactly the same dictionary (see + inflateSetDictionary). + + The dictionary should consist of strings (byte sequences) that are likely + to be encountered later in the data to be compressed, with the most commonly + used strings preferably put towards the end of the dictionary. Using a + dictionary is most useful when the data to be compressed is short and can be + predicted with good accuracy; the data can then be compressed better than + with the default empty dictionary. + + Depending on the size of the compression data structures selected by + deflateInit or deflateInit2, a part of the dictionary may in effect be + discarded, for example if the dictionary is larger than the window size + provided in deflateInit or deflateInit2. Thus the strings most likely to be + useful should be put at the end of the dictionary, not at the front. In + addition, the current implementation of deflate will use at most the window + size minus 262 bytes of the provided dictionary. + + Upon return of this function, strm->adler is set to the Adler-32 value + of the dictionary; the decompressor may later use this value to determine + which dictionary has been used by the compressor. (The Adler-32 value + applies to the whole dictionary even if only a subset of the dictionary is + actually used by the compressor.) If a raw deflate was requested, then the + Adler-32 value is not computed and strm->adler is not set. + + deflateSetDictionary returns Z_OK if success, or Z_STREAM_ERROR if a + parameter is invalid (e.g. dictionary being Z_NULL) or the stream state is + inconsistent (for example if deflate has already been called for this stream + or if not at a block boundary for raw deflate). deflateSetDictionary does + not perform any compression: this will be done by deflate(). +*/ + +ZEXTERN int ZEXPORT deflateGetDictionary OF((z_streamp strm, + Bytef *dictionary, + uInt *dictLength)); +/* + Returns the sliding dictionary being maintained by deflate. dictLength is + set to the number of bytes in the dictionary, and that many bytes are copied + to dictionary. dictionary must have enough space, where 32768 bytes is + always enough. If deflateGetDictionary() is called with dictionary equal to + Z_NULL, then only the dictionary length is returned, and nothing is copied. + Similary, if dictLength is Z_NULL, then it is not set. + + deflateGetDictionary() may return a length less than the window size, even + when more than the window size in input has been provided. It may return up + to 258 bytes less in that case, due to how zlib's implementation of deflate + manages the sliding window and lookahead for matches, where matches can be + up to 258 bytes long. If the application needs the last window-size bytes of + input, then that would need to be saved by the application outside of zlib. + + deflateGetDictionary returns Z_OK on success, or Z_STREAM_ERROR if the + stream state is inconsistent. +*/ + +ZEXTERN int ZEXPORT deflateCopy OF((z_streamp dest, + z_streamp source)); +/* + Sets the destination stream as a complete copy of the source stream. + + This function can be useful when several compression strategies will be + tried, for example when there are several ways of pre-processing the input + data with a filter. The streams that will be discarded should then be freed + by calling deflateEnd. Note that deflateCopy duplicates the internal + compression state which can be quite large, so this strategy is slow and can + consume lots of memory. + + deflateCopy returns Z_OK if success, Z_MEM_ERROR if there was not + enough memory, Z_STREAM_ERROR if the source stream state was inconsistent + (such as zalloc being Z_NULL). msg is left unchanged in both source and + destination. +*/ + +ZEXTERN int ZEXPORT deflateReset OF((z_streamp strm)); +/* + This function is equivalent to deflateEnd followed by deflateInit, but + does not free and reallocate the internal compression state. The stream + will leave the compression level and any other attributes that may have been + set unchanged. + + deflateReset returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent (such as zalloc or state being Z_NULL). +*/ + +ZEXTERN int ZEXPORT deflateParams OF((z_streamp strm, + int level, + int strategy)); +/* + Dynamically update the compression level and compression strategy. The + interpretation of level and strategy is as in deflateInit2(). This can be + used to switch between compression and straight copy of the input data, or + to switch to a different kind of input data requiring a different strategy. + If the compression approach (which is a function of the level) or the + strategy is changed, and if any input has been consumed in a previous + deflate() call, then the input available so far is compressed with the old + level and strategy using deflate(strm, Z_BLOCK). There are three approaches + for the compression levels 0, 1..3, and 4..9 respectively. The new level + and strategy will take effect at the next call of deflate(). + + If a deflate(strm, Z_BLOCK) is performed by deflateParams(), and it does + not have enough output space to complete, then the parameter change will not + take effect. In this case, deflateParams() can be called again with the + same parameters and more output space to try again. + + In order to assure a change in the parameters on the first try, the + deflate stream should be flushed using deflate() with Z_BLOCK or other flush + request until strm.avail_out is not zero, before calling deflateParams(). + Then no more input data should be provided before the deflateParams() call. + If this is done, the old level and strategy will be applied to the data + compressed before deflateParams(), and the new level and strategy will be + applied to the the data compressed after deflateParams(). + + deflateParams returns Z_OK on success, Z_STREAM_ERROR if the source stream + state was inconsistent or if a parameter was invalid, or Z_BUF_ERROR if + there was not enough output space to complete the compression of the + available input data before a change in the strategy or approach. Note that + in the case of a Z_BUF_ERROR, the parameters are not changed. A return + value of Z_BUF_ERROR is not fatal, in which case deflateParams() can be + retried with more output space. +*/ + +ZEXTERN int ZEXPORT deflateTune OF((z_streamp strm, + int good_length, + int max_lazy, + int nice_length, + int max_chain)); +/* + Fine tune deflate's internal compression parameters. This should only be + used by someone who understands the algorithm used by zlib's deflate for + searching for the best matching string, and even then only by the most + fanatic optimizer trying to squeeze out the last compressed bit for their + specific input data. Read the deflate.c source code for the meaning of the + max_lazy, good_length, nice_length, and max_chain parameters. + + deflateTune() can be called after deflateInit() or deflateInit2(), and + returns Z_OK on success, or Z_STREAM_ERROR for an invalid deflate stream. + */ + +ZEXTERN uLong ZEXPORT deflateBound OF((z_streamp strm, + uLong sourceLen)); +/* + deflateBound() returns an upper bound on the compressed size after + deflation of sourceLen bytes. It must be called after deflateInit() or + deflateInit2(), and after deflateSetHeader(), if used. This would be used + to allocate an output buffer for deflation in a single pass, and so would be + called before deflate(). If that first deflate() call is provided the + sourceLen input bytes, an output buffer allocated to the size returned by + deflateBound(), and the flush value Z_FINISH, then deflate() is guaranteed + to return Z_STREAM_END. Note that it is possible for the compressed size to + be larger than the value returned by deflateBound() if flush options other + than Z_FINISH or Z_NO_FLUSH are used. +*/ + +ZEXTERN int ZEXPORT deflatePending OF((z_streamp strm, + unsigned *pending, + int *bits)); +/* + deflatePending() returns the number of bytes and bits of output that have + been generated, but not yet provided in the available output. The bytes not + provided would be due to the available output space having being consumed. + The number of bits of output not provided are between 0 and 7, where they + await more bits to join them in order to fill out a full byte. If pending + or bits are Z_NULL, then those values are not set. + + deflatePending returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent. + */ + +ZEXTERN int ZEXPORT deflatePrime OF((z_streamp strm, + int bits, + int value)); +/* + deflatePrime() inserts bits in the deflate output stream. The intent + is that this function is used to start off the deflate output with the bits + leftover from a previous deflate stream when appending to it. As such, this + function can only be used for raw deflate, and must be used before the first + deflate() call after a deflateInit2() or deflateReset(). bits must be less + than or equal to 16, and that many of the least significant bits of value + will be inserted in the output. + + deflatePrime returns Z_OK if success, Z_BUF_ERROR if there was not enough + room in the internal buffer to insert the bits, or Z_STREAM_ERROR if the + source stream state was inconsistent. +*/ + +ZEXTERN int ZEXPORT deflateSetHeader OF((z_streamp strm, + gz_headerp head)); +/* + deflateSetHeader() provides gzip header information for when a gzip + stream is requested by deflateInit2(). deflateSetHeader() may be called + after deflateInit2() or deflateReset() and before the first call of + deflate(). The text, time, os, extra field, name, and comment information + in the provided gz_header structure are written to the gzip header (xflag is + ignored -- the extra flags are set according to the compression level). The + caller must assure that, if not Z_NULL, name and comment are terminated with + a zero byte, and that if extra is not Z_NULL, that extra_len bytes are + available there. If hcrc is true, a gzip header crc is included. Note that + the current versions of the command-line version of gzip (up through version + 1.3.x) do not support header crc's, and will report that it is a "multi-part + gzip file" and give up. + + If deflateSetHeader is not used, the default gzip header has text false, + the time set to zero, and os set to 255, with no extra, name, or comment + fields. The gzip header is returned to the default state by deflateReset(). + + deflateSetHeader returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent. +*/ + +/* +ZEXTERN int ZEXPORT inflateInit2 OF((z_streamp strm, + int windowBits)); + + This is another version of inflateInit with an extra parameter. The + fields next_in, avail_in, zalloc, zfree and opaque must be initialized + before by the caller. + + The windowBits parameter is the base two logarithm of the maximum window + size (the size of the history buffer). It should be in the range 8..15 for + this version of the library. The default value is 15 if inflateInit is used + instead. windowBits must be greater than or equal to the windowBits value + provided to deflateInit2() while compressing, or it must be equal to 15 if + deflateInit2() was not used. If a compressed stream with a larger window + size is given as input, inflate() will return with the error code + Z_DATA_ERROR instead of trying to allocate a larger window. + + windowBits can also be zero to request that inflate use the window size in + the zlib header of the compressed stream. + + windowBits can also be -8..-15 for raw inflate. In this case, -windowBits + determines the window size. inflate() will then process raw deflate data, + not looking for a zlib or gzip header, not generating a check value, and not + looking for any check values for comparison at the end of the stream. This + is for use with other formats that use the deflate compressed data format + such as zip. Those formats provide their own check values. If a custom + format is developed using the raw deflate format for compressed data, it is + recommended that a check value such as an Adler-32 or a CRC-32 be applied to + the uncompressed data as is done in the zlib, gzip, and zip formats. For + most applications, the zlib format should be used as is. Note that comments + above on the use in deflateInit2() applies to the magnitude of windowBits. + + windowBits can also be greater than 15 for optional gzip decoding. Add + 32 to windowBits to enable zlib and gzip decoding with automatic header + detection, or add 16 to decode only the gzip format (the zlib format will + return a Z_DATA_ERROR). If a gzip stream is being decoded, strm->adler is a + CRC-32 instead of an Adler-32. Unlike the gunzip utility and gzread() (see + below), inflate() will not automatically decode concatenated gzip streams. + inflate() will return Z_STREAM_END at the end of the gzip stream. The state + would need to be reset to continue decoding a subsequent gzip stream. + + inflateInit2 returns Z_OK if success, Z_MEM_ERROR if there was not enough + memory, Z_VERSION_ERROR if the zlib library version is incompatible with the + version assumed by the caller, or Z_STREAM_ERROR if the parameters are + invalid, such as a null pointer to the structure. msg is set to null if + there is no error message. inflateInit2 does not perform any decompression + apart from possibly reading the zlib header if present: actual decompression + will be done by inflate(). (So next_in and avail_in may be modified, but + next_out and avail_out are unused and unchanged.) The current implementation + of inflateInit2() does not process any header information -- that is + deferred until inflate() is called. +*/ + +ZEXTERN int ZEXPORT inflateSetDictionary OF((z_streamp strm, + const Bytef *dictionary, + uInt dictLength)); +/* + Initializes the decompression dictionary from the given uncompressed byte + sequence. This function must be called immediately after a call of inflate, + if that call returned Z_NEED_DICT. The dictionary chosen by the compressor + can be determined from the Adler-32 value returned by that call of inflate. + The compressor and decompressor must use exactly the same dictionary (see + deflateSetDictionary). For raw inflate, this function can be called at any + time to set the dictionary. If the provided dictionary is smaller than the + window and there is already data in the window, then the provided dictionary + will amend what's there. The application must insure that the dictionary + that was used for compression is provided. + + inflateSetDictionary returns Z_OK if success, Z_STREAM_ERROR if a + parameter is invalid (e.g. dictionary being Z_NULL) or the stream state is + inconsistent, Z_DATA_ERROR if the given dictionary doesn't match the + expected one (incorrect Adler-32 value). inflateSetDictionary does not + perform any decompression: this will be done by subsequent calls of + inflate(). +*/ + +ZEXTERN int ZEXPORT inflateGetDictionary OF((z_streamp strm, + Bytef *dictionary, + uInt *dictLength)); +/* + Returns the sliding dictionary being maintained by inflate. dictLength is + set to the number of bytes in the dictionary, and that many bytes are copied + to dictionary. dictionary must have enough space, where 32768 bytes is + always enough. If inflateGetDictionary() is called with dictionary equal to + Z_NULL, then only the dictionary length is returned, and nothing is copied. + Similary, if dictLength is Z_NULL, then it is not set. + + inflateGetDictionary returns Z_OK on success, or Z_STREAM_ERROR if the + stream state is inconsistent. +*/ + +ZEXTERN int ZEXPORT inflateSync OF((z_streamp strm)); +/* + Skips invalid compressed data until a possible full flush point (see above + for the description of deflate with Z_FULL_FLUSH) can be found, or until all + available input is skipped. No output is provided. + + inflateSync searches for a 00 00 FF FF pattern in the compressed data. + All full flush points have this pattern, but not all occurrences of this + pattern are full flush points. + + inflateSync returns Z_OK if a possible full flush point has been found, + Z_BUF_ERROR if no more input was provided, Z_DATA_ERROR if no flush point + has been found, or Z_STREAM_ERROR if the stream structure was inconsistent. + In the success case, the application may save the current current value of + total_in which indicates where valid compressed data was found. In the + error case, the application may repeatedly call inflateSync, providing more + input each time, until success or end of the input data. +*/ + +ZEXTERN int ZEXPORT inflateCopy OF((z_streamp dest, + z_streamp source)); +/* + Sets the destination stream as a complete copy of the source stream. + + This function can be useful when randomly accessing a large stream. The + first pass through the stream can periodically record the inflate state, + allowing restarting inflate at those points when randomly accessing the + stream. + + inflateCopy returns Z_OK if success, Z_MEM_ERROR if there was not + enough memory, Z_STREAM_ERROR if the source stream state was inconsistent + (such as zalloc being Z_NULL). msg is left unchanged in both source and + destination. +*/ + +ZEXTERN int ZEXPORT inflateReset OF((z_streamp strm)); +/* + This function is equivalent to inflateEnd followed by inflateInit, + but does not free and reallocate the internal decompression state. The + stream will keep attributes that may have been set by inflateInit2. + + inflateReset returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent (such as zalloc or state being Z_NULL). +*/ + +ZEXTERN int ZEXPORT inflateReset2 OF((z_streamp strm, + int windowBits)); +/* + This function is the same as inflateReset, but it also permits changing + the wrap and window size requests. The windowBits parameter is interpreted + the same as it is for inflateInit2. If the window size is changed, then the + memory allocated for the window is freed, and the window will be reallocated + by inflate() if needed. + + inflateReset2 returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent (such as zalloc or state being Z_NULL), or if + the windowBits parameter is invalid. +*/ + +ZEXTERN int ZEXPORT inflatePrime OF((z_streamp strm, + int bits, + int value)); +/* + This function inserts bits in the inflate input stream. The intent is + that this function is used to start inflating at a bit position in the + middle of a byte. The provided bits will be used before any bytes are used + from next_in. This function should only be used with raw inflate, and + should be used before the first inflate() call after inflateInit2() or + inflateReset(). bits must be less than or equal to 16, and that many of the + least significant bits of value will be inserted in the input. + + If bits is negative, then the input stream bit buffer is emptied. Then + inflatePrime() can be called again to put bits in the buffer. This is used + to clear out bits leftover after feeding inflate a block description prior + to feeding inflate codes. + + inflatePrime returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent. +*/ + +ZEXTERN long ZEXPORT inflateMark OF((z_streamp strm)); +/* + This function returns two values, one in the lower 16 bits of the return + value, and the other in the remaining upper bits, obtained by shifting the + return value down 16 bits. If the upper value is -1 and the lower value is + zero, then inflate() is currently decoding information outside of a block. + If the upper value is -1 and the lower value is non-zero, then inflate is in + the middle of a stored block, with the lower value equaling the number of + bytes from the input remaining to copy. If the upper value is not -1, then + it is the number of bits back from the current bit position in the input of + the code (literal or length/distance pair) currently being processed. In + that case the lower value is the number of bytes already emitted for that + code. + + A code is being processed if inflate is waiting for more input to complete + decoding of the code, or if it has completed decoding but is waiting for + more output space to write the literal or match data. + + inflateMark() is used to mark locations in the input data for random + access, which may be at bit positions, and to note those cases where the + output of a code may span boundaries of random access blocks. The current + location in the input stream can be determined from avail_in and data_type + as noted in the description for the Z_BLOCK flush parameter for inflate. + + inflateMark returns the value noted above, or -65536 if the provided + source stream state was inconsistent. +*/ + +ZEXTERN int ZEXPORT inflateGetHeader OF((z_streamp strm, + gz_headerp head)); +/* + inflateGetHeader() requests that gzip header information be stored in the + provided gz_header structure. inflateGetHeader() may be called after + inflateInit2() or inflateReset(), and before the first call of inflate(). + As inflate() processes the gzip stream, head->done is zero until the header + is completed, at which time head->done is set to one. If a zlib stream is + being decoded, then head->done is set to -1 to indicate that there will be + no gzip header information forthcoming. Note that Z_BLOCK or Z_TREES can be + used to force inflate() to return immediately after header processing is + complete and before any actual data is decompressed. + + The text, time, xflags, and os fields are filled in with the gzip header + contents. hcrc is set to true if there is a header CRC. (The header CRC + was valid if done is set to one.) If extra is not Z_NULL, then extra_max + contains the maximum number of bytes to write to extra. Once done is true, + extra_len contains the actual extra field length, and extra contains the + extra field, or that field truncated if extra_max is less than extra_len. + If name is not Z_NULL, then up to name_max characters are written there, + terminated with a zero unless the length is greater than name_max. If + comment is not Z_NULL, then up to comm_max characters are written there, + terminated with a zero unless the length is greater than comm_max. When any + of extra, name, or comment are not Z_NULL and the respective field is not + present in the header, then that field is set to Z_NULL to signal its + absence. This allows the use of deflateSetHeader() with the returned + structure to duplicate the header. However if those fields are set to + allocated memory, then the application will need to save those pointers + elsewhere so that they can be eventually freed. + + If inflateGetHeader is not used, then the header information is simply + discarded. The header is always checked for validity, including the header + CRC if present. inflateReset() will reset the process to discard the header + information. The application would need to call inflateGetHeader() again to + retrieve the header from the next gzip stream. + + inflateGetHeader returns Z_OK if success, or Z_STREAM_ERROR if the source + stream state was inconsistent. +*/ + +/* +ZEXTERN int ZEXPORT inflateBackInit OF((z_streamp strm, int windowBits, + unsigned char FAR *window)); + + Initialize the internal stream state for decompression using inflateBack() + calls. The fields zalloc, zfree and opaque in strm must be initialized + before the call. If zalloc and zfree are Z_NULL, then the default library- + derived memory allocation routines are used. windowBits is the base two + logarithm of the window size, in the range 8..15. window is a caller + supplied buffer of that size. Except for special applications where it is + assured that deflate was used with small window sizes, windowBits must be 15 + and a 32K byte window must be supplied to be able to decompress general + deflate streams. + + See inflateBack() for the usage of these routines. + + inflateBackInit will return Z_OK on success, Z_STREAM_ERROR if any of + the parameters are invalid, Z_MEM_ERROR if the internal state could not be + allocated, or Z_VERSION_ERROR if the version of the library does not match + the version of the header file. +*/ + +typedef unsigned (*in_func) OF((void FAR *, + z_const unsigned char FAR * FAR *)); +typedef int (*out_func) OF((void FAR *, unsigned char FAR *, unsigned)); + +ZEXTERN int ZEXPORT inflateBack OF((z_streamp strm, + in_func in, void FAR *in_desc, + out_func out, void FAR *out_desc)); +/* + inflateBack() does a raw inflate with a single call using a call-back + interface for input and output. This is potentially more efficient than + inflate() for file i/o applications, in that it avoids copying between the + output and the sliding window by simply making the window itself the output + buffer. inflate() can be faster on modern CPUs when used with large + buffers. inflateBack() trusts the application to not change the output + buffer passed by the output function, at least until inflateBack() returns. + + inflateBackInit() must be called first to allocate the internal state + and to initialize the state with the user-provided window buffer. + inflateBack() may then be used multiple times to inflate a complete, raw + deflate stream with each call. inflateBackEnd() is then called to free the + allocated state. + + A raw deflate stream is one with no zlib or gzip header or trailer. + This routine would normally be used in a utility that reads zip or gzip + files and writes out uncompressed files. The utility would decode the + header and process the trailer on its own, hence this routine expects only + the raw deflate stream to decompress. This is different from the default + behavior of inflate(), which expects a zlib header and trailer around the + deflate stream. + + inflateBack() uses two subroutines supplied by the caller that are then + called by inflateBack() for input and output. inflateBack() calls those + routines until it reads a complete deflate stream and writes out all of the + uncompressed data, or until it encounters an error. The function's + parameters and return types are defined above in the in_func and out_func + typedefs. inflateBack() will call in(in_desc, &buf) which should return the + number of bytes of provided input, and a pointer to that input in buf. If + there is no input available, in() must return zero -- buf is ignored in that + case -- and inflateBack() will return a buffer error. inflateBack() will + call out(out_desc, buf, len) to write the uncompressed data buf[0..len-1]. + out() should return zero on success, or non-zero on failure. If out() + returns non-zero, inflateBack() will return with an error. Neither in() nor + out() are permitted to change the contents of the window provided to + inflateBackInit(), which is also the buffer that out() uses to write from. + The length written by out() will be at most the window size. Any non-zero + amount of input may be provided by in(). + + For convenience, inflateBack() can be provided input on the first call by + setting strm->next_in and strm->avail_in. If that input is exhausted, then + in() will be called. Therefore strm->next_in must be initialized before + calling inflateBack(). If strm->next_in is Z_NULL, then in() will be called + immediately for input. If strm->next_in is not Z_NULL, then strm->avail_in + must also be initialized, and then if strm->avail_in is not zero, input will + initially be taken from strm->next_in[0 .. strm->avail_in - 1]. + + The in_desc and out_desc parameters of inflateBack() is passed as the + first parameter of in() and out() respectively when they are called. These + descriptors can be optionally used to pass any information that the caller- + supplied in() and out() functions need to do their job. + + On return, inflateBack() will set strm->next_in and strm->avail_in to + pass back any unused input that was provided by the last in() call. The + return values of inflateBack() can be Z_STREAM_END on success, Z_BUF_ERROR + if in() or out() returned an error, Z_DATA_ERROR if there was a format error + in the deflate stream (in which case strm->msg is set to indicate the nature + of the error), or Z_STREAM_ERROR if the stream was not properly initialized. + In the case of Z_BUF_ERROR, an input or output error can be distinguished + using strm->next_in which will be Z_NULL only if in() returned an error. If + strm->next_in is not Z_NULL, then the Z_BUF_ERROR was due to out() returning + non-zero. (in() will always be called before out(), so strm->next_in is + assured to be defined if out() returns non-zero.) Note that inflateBack() + cannot return Z_OK. +*/ + +ZEXTERN int ZEXPORT inflateBackEnd OF((z_streamp strm)); +/* + All memory allocated by inflateBackInit() is freed. + + inflateBackEnd() returns Z_OK on success, or Z_STREAM_ERROR if the stream + state was inconsistent. +*/ + +ZEXTERN uLong ZEXPORT zlibCompileFlags OF((void)); +/* Return flags indicating compile-time options. + + Type sizes, two bits each, 00 = 16 bits, 01 = 32, 10 = 64, 11 = other: + 1.0: size of uInt + 3.2: size of uLong + 5.4: size of voidpf (pointer) + 7.6: size of z_off_t + + Compiler, assembler, and debug options: + 8: ZLIB_DEBUG + 9: ASMV or ASMINF -- use ASM code + 10: ZLIB_WINAPI -- exported functions use the WINAPI calling convention + 11: 0 (reserved) + + One-time table building (smaller code, but not thread-safe if true): + 12: BUILDFIXED -- build static block decoding tables when needed + 13: DYNAMIC_CRC_TABLE -- build CRC calculation tables when needed + 14,15: 0 (reserved) + + Library content (indicates missing functionality): + 16: NO_GZCOMPRESS -- gz* functions cannot compress (to avoid linking + deflate code when not needed) + 17: NO_GZIP -- deflate can't write gzip streams, and inflate can't detect + and decode gzip streams (to avoid linking crc code) + 18-19: 0 (reserved) + + Operation variations (changes in library functionality): + 20: PKZIP_BUG_WORKAROUND -- slightly more permissive inflate + 21: FASTEST -- deflate algorithm with only one, lowest compression level + 22,23: 0 (reserved) + + The sprintf variant used by gzprintf (zero is best): + 24: 0 = vs*, 1 = s* -- 1 means limited to 20 arguments after the format + 25: 0 = *nprintf, 1 = *printf -- 1 means gzprintf() not secure! + 26: 0 = returns value, 1 = void -- 1 means inferred string length returned + + Remainder: + 27-31: 0 (reserved) + */ + +#ifndef Z_SOLO + + /* utility functions */ + +/* + The following utility functions are implemented on top of the basic + stream-oriented functions. To simplify the interface, some default options + are assumed (compression level and memory usage, standard memory allocation + functions). The source code of these utility functions can be modified if + you need special options. +*/ + +ZEXTERN int ZEXPORT compress OF((Bytef *dest, uLongf *destLen, + const Bytef *source, uLong sourceLen)); +/* + Compresses the source buffer into the destination buffer. sourceLen is + the byte length of the source buffer. Upon entry, destLen is the total size + of the destination buffer, which must be at least the value returned by + compressBound(sourceLen). Upon exit, destLen is the actual size of the + compressed data. compress() is equivalent to compress2() with a level + parameter of Z_DEFAULT_COMPRESSION. + + compress returns Z_OK if success, Z_MEM_ERROR if there was not + enough memory, Z_BUF_ERROR if there was not enough room in the output + buffer. +*/ + +ZEXTERN int ZEXPORT compress2 OF((Bytef *dest, uLongf *destLen, + const Bytef *source, uLong sourceLen, + int level)); +/* + Compresses the source buffer into the destination buffer. The level + parameter has the same meaning as in deflateInit. sourceLen is the byte + length of the source buffer. Upon entry, destLen is the total size of the + destination buffer, which must be at least the value returned by + compressBound(sourceLen). Upon exit, destLen is the actual size of the + compressed data. + + compress2 returns Z_OK if success, Z_MEM_ERROR if there was not enough + memory, Z_BUF_ERROR if there was not enough room in the output buffer, + Z_STREAM_ERROR if the level parameter is invalid. +*/ + +ZEXTERN uLong ZEXPORT compressBound OF((uLong sourceLen)); +/* + compressBound() returns an upper bound on the compressed size after + compress() or compress2() on sourceLen bytes. It would be used before a + compress() or compress2() call to allocate the destination buffer. +*/ + +ZEXTERN int ZEXPORT uncompress OF((Bytef *dest, uLongf *destLen, + const Bytef *source, uLong sourceLen)); +/* + Decompresses the source buffer into the destination buffer. sourceLen is + the byte length of the source buffer. Upon entry, destLen is the total size + of the destination buffer, which must be large enough to hold the entire + uncompressed data. (The size of the uncompressed data must have been saved + previously by the compressor and transmitted to the decompressor by some + mechanism outside the scope of this compression library.) Upon exit, destLen + is the actual size of the uncompressed data. + + uncompress returns Z_OK if success, Z_MEM_ERROR if there was not + enough memory, Z_BUF_ERROR if there was not enough room in the output + buffer, or Z_DATA_ERROR if the input data was corrupted or incomplete. In + the case where there is not enough room, uncompress() will fill the output + buffer with the uncompressed data up to that point. +*/ + +ZEXTERN int ZEXPORT uncompress2 OF((Bytef *dest, uLongf *destLen, + const Bytef *source, uLong *sourceLen)); +/* + Same as uncompress, except that sourceLen is a pointer, where the + length of the source is *sourceLen. On return, *sourceLen is the number of + source bytes consumed. +*/ + + /* gzip file access functions */ + +/* + This library supports reading and writing files in gzip (.gz) format with + an interface similar to that of stdio, using the functions that start with + "gz". The gzip format is different from the zlib format. gzip is a gzip + wrapper, documented in RFC 1952, wrapped around a deflate stream. +*/ + +typedef struct gzFile_s *gzFile; /* semi-opaque gzip file descriptor */ + +/* +ZEXTERN gzFile ZEXPORT gzopen OF((const char *path, const char *mode)); + + Opens a gzip (.gz) file for reading or writing. The mode parameter is as + in fopen ("rb" or "wb") but can also include a compression level ("wb9") or + a strategy: 'f' for filtered data as in "wb6f", 'h' for Huffman-only + compression as in "wb1h", 'R' for run-length encoding as in "wb1R", or 'F' + for fixed code compression as in "wb9F". (See the description of + deflateInit2 for more information about the strategy parameter.) 'T' will + request transparent writing or appending with no compression and not using + the gzip format. + + "a" can be used instead of "w" to request that the gzip stream that will + be written be appended to the file. "+" will result in an error, since + reading and writing to the same gzip file is not supported. The addition of + "x" when writing will create the file exclusively, which fails if the file + already exists. On systems that support it, the addition of "e" when + reading or writing will set the flag to close the file on an execve() call. + + These functions, as well as gzip, will read and decode a sequence of gzip + streams in a file. The append function of gzopen() can be used to create + such a file. (Also see gzflush() for another way to do this.) When + appending, gzopen does not test whether the file begins with a gzip stream, + nor does it look for the end of the gzip streams to begin appending. gzopen + will simply append a gzip stream to the existing file. + + gzopen can be used to read a file which is not in gzip format; in this + case gzread will directly read from the file without decompression. When + reading, this will be detected automatically by looking for the magic two- + byte gzip header. + + gzopen returns NULL if the file could not be opened, if there was + insufficient memory to allocate the gzFile state, or if an invalid mode was + specified (an 'r', 'w', or 'a' was not provided, or '+' was provided). + errno can be checked to determine if the reason gzopen failed was that the + file could not be opened. +*/ + +ZEXTERN gzFile ZEXPORT gzdopen OF((int fd, const char *mode)); +/* + gzdopen associates a gzFile with the file descriptor fd. File descriptors + are obtained from calls like open, dup, creat, pipe or fileno (if the file + has been previously opened with fopen). The mode parameter is as in gzopen. + + The next call of gzclose on the returned gzFile will also close the file + descriptor fd, just like fclose(fdopen(fd, mode)) closes the file descriptor + fd. If you want to keep fd open, use fd = dup(fd_keep); gz = gzdopen(fd, + mode);. The duplicated descriptor should be saved to avoid a leak, since + gzdopen does not close fd if it fails. If you are using fileno() to get the + file descriptor from a FILE *, then you will have to use dup() to avoid + double-close()ing the file descriptor. Both gzclose() and fclose() will + close the associated file descriptor, so they need to have different file + descriptors. + + gzdopen returns NULL if there was insufficient memory to allocate the + gzFile state, if an invalid mode was specified (an 'r', 'w', or 'a' was not + provided, or '+' was provided), or if fd is -1. The file descriptor is not + used until the next gz* read, write, seek, or close operation, so gzdopen + will not detect if fd is invalid (unless fd is -1). +*/ + +ZEXTERN int ZEXPORT gzbuffer OF((gzFile file, unsigned size)); +/* + Set the internal buffer size used by this library's functions. The + default buffer size is 8192 bytes. This function must be called after + gzopen() or gzdopen(), and before any other calls that read or write the + file. The buffer memory allocation is always deferred to the first read or + write. Three times that size in buffer space is allocated. A larger buffer + size of, for example, 64K or 128K bytes will noticeably increase the speed + of decompression (reading). + + The new buffer size also affects the maximum length for gzprintf(). + + gzbuffer() returns 0 on success, or -1 on failure, such as being called + too late. +*/ + +ZEXTERN int ZEXPORT gzsetparams OF((gzFile file, int level, int strategy)); +/* + Dynamically update the compression level or strategy. See the description + of deflateInit2 for the meaning of these parameters. Previously provided + data is flushed before the parameter change. + + gzsetparams returns Z_OK if success, Z_STREAM_ERROR if the file was not + opened for writing, Z_ERRNO if there is an error writing the flushed data, + or Z_MEM_ERROR if there is a memory allocation error. +*/ + +ZEXTERN int ZEXPORT gzread OF((gzFile file, voidp buf, unsigned len)); +/* + Reads the given number of uncompressed bytes from the compressed file. If + the input file is not in gzip format, gzread copies the given number of + bytes into the buffer directly from the file. + + After reaching the end of a gzip stream in the input, gzread will continue + to read, looking for another gzip stream. Any number of gzip streams may be + concatenated in the input file, and will all be decompressed by gzread(). + If something other than a gzip stream is encountered after a gzip stream, + that remaining trailing garbage is ignored (and no error is returned). + + gzread can be used to read a gzip file that is being concurrently written. + Upon reaching the end of the input, gzread will return with the available + data. If the error code returned by gzerror is Z_OK or Z_BUF_ERROR, then + gzclearerr can be used to clear the end of file indicator in order to permit + gzread to be tried again. Z_OK indicates that a gzip stream was completed + on the last gzread. Z_BUF_ERROR indicates that the input file ended in the + middle of a gzip stream. Note that gzread does not return -1 in the event + of an incomplete gzip stream. This error is deferred until gzclose(), which + will return Z_BUF_ERROR if the last gzread ended in the middle of a gzip + stream. Alternatively, gzerror can be used before gzclose to detect this + case. + + gzread returns the number of uncompressed bytes actually read, less than + len for end of file, or -1 for error. If len is too large to fit in an int, + then nothing is read, -1 is returned, and the error state is set to + Z_STREAM_ERROR. +*/ + +ZEXTERN z_size_t ZEXPORT gzfread OF((voidp buf, z_size_t size, z_size_t nitems, + gzFile file)); +/* + Read up to nitems items of size size from file to buf, otherwise operating + as gzread() does. This duplicates the interface of stdio's fread(), with + size_t request and return types. If the library defines size_t, then + z_size_t is identical to size_t. If not, then z_size_t is an unsigned + integer type that can contain a pointer. + + gzfread() returns the number of full items read of size size, or zero if + the end of the file was reached and a full item could not be read, or if + there was an error. gzerror() must be consulted if zero is returned in + order to determine if there was an error. If the multiplication of size and + nitems overflows, i.e. the product does not fit in a z_size_t, then nothing + is read, zero is returned, and the error state is set to Z_STREAM_ERROR. + + In the event that the end of file is reached and only a partial item is + available at the end, i.e. the remaining uncompressed data length is not a + multiple of size, then the final partial item is nevetheless read into buf + and the end-of-file flag is set. The length of the partial item read is not + provided, but could be inferred from the result of gztell(). This behavior + is the same as the behavior of fread() implementations in common libraries, + but it prevents the direct use of gzfread() to read a concurrently written + file, reseting and retrying on end-of-file, when size is not 1. +*/ + +ZEXTERN int ZEXPORT gzwrite OF((gzFile file, + voidpc buf, unsigned len)); +/* + Writes the given number of uncompressed bytes into the compressed file. + gzwrite returns the number of uncompressed bytes written or 0 in case of + error. +*/ + +ZEXTERN z_size_t ZEXPORT gzfwrite OF((voidpc buf, z_size_t size, + z_size_t nitems, gzFile file)); +/* + gzfwrite() writes nitems items of size size from buf to file, duplicating + the interface of stdio's fwrite(), with size_t request and return types. If + the library defines size_t, then z_size_t is identical to size_t. If not, + then z_size_t is an unsigned integer type that can contain a pointer. + + gzfwrite() returns the number of full items written of size size, or zero + if there was an error. If the multiplication of size and nitems overflows, + i.e. the product does not fit in a z_size_t, then nothing is written, zero + is returned, and the error state is set to Z_STREAM_ERROR. +*/ + +ZEXTERN int ZEXPORTVA gzprintf Z_ARG((gzFile file, const char *format, ...)); +/* + Converts, formats, and writes the arguments to the compressed file under + control of the format string, as in fprintf. gzprintf returns the number of + uncompressed bytes actually written, or a negative zlib error code in case + of error. The number of uncompressed bytes written is limited to 8191, or + one less than the buffer size given to gzbuffer(). The caller should assure + that this limit is not exceeded. If it is exceeded, then gzprintf() will + return an error (0) with nothing written. In this case, there may also be a + buffer overflow with unpredictable consequences, which is possible only if + zlib was compiled with the insecure functions sprintf() or vsprintf() + because the secure snprintf() or vsnprintf() functions were not available. + This can be determined using zlibCompileFlags(). +*/ + +ZEXTERN int ZEXPORT gzputs OF((gzFile file, const char *s)); +/* + Writes the given null-terminated string to the compressed file, excluding + the terminating null character. + + gzputs returns the number of characters written, or -1 in case of error. +*/ + +ZEXTERN char * ZEXPORT gzgets OF((gzFile file, char *buf, int len)); +/* + Reads bytes from the compressed file until len-1 characters are read, or a + newline character is read and transferred to buf, or an end-of-file + condition is encountered. If any characters are read or if len == 1, the + string is terminated with a null character. If no characters are read due + to an end-of-file or len < 1, then the buffer is left untouched. + + gzgets returns buf which is a null-terminated string, or it returns NULL + for end-of-file or in case of error. If there was an error, the contents at + buf are indeterminate. +*/ + +ZEXTERN int ZEXPORT gzputc OF((gzFile file, int c)); +/* + Writes c, converted to an unsigned char, into the compressed file. gzputc + returns the value that was written, or -1 in case of error. +*/ + +ZEXTERN int ZEXPORT gzgetc OF((gzFile file)); +/* + Reads one byte from the compressed file. gzgetc returns this byte or -1 + in case of end of file or error. This is implemented as a macro for speed. + As such, it does not do all of the checking the other functions do. I.e. + it does not check to see if file is NULL, nor whether the structure file + points to has been clobbered or not. +*/ + +ZEXTERN int ZEXPORT gzungetc OF((int c, gzFile file)); +/* + Push one character back onto the stream to be read as the first character + on the next read. At least one character of push-back is allowed. + gzungetc() returns the character pushed, or -1 on failure. gzungetc() will + fail if c is -1, and may fail if a character has been pushed but not read + yet. If gzungetc is used immediately after gzopen or gzdopen, at least the + output buffer size of pushed characters is allowed. (See gzbuffer above.) + The pushed character will be discarded if the stream is repositioned with + gzseek() or gzrewind(). +*/ + +ZEXTERN int ZEXPORT gzflush OF((gzFile file, int flush)); +/* + Flushes all pending output into the compressed file. The parameter flush + is as in the deflate() function. The return value is the zlib error number + (see function gzerror below). gzflush is only permitted when writing. + + If the flush parameter is Z_FINISH, the remaining data is written and the + gzip stream is completed in the output. If gzwrite() is called again, a new + gzip stream will be started in the output. gzread() is able to read such + concatenated gzip streams. + + gzflush should be called only when strictly necessary because it will + degrade compression if called too often. +*/ + +/* +ZEXTERN z_off_t ZEXPORT gzseek OF((gzFile file, + z_off_t offset, int whence)); + + Sets the starting position for the next gzread or gzwrite on the given + compressed file. The offset represents a number of bytes in the + uncompressed data stream. The whence parameter is defined as in lseek(2); + the value SEEK_END is not supported. + + If the file is opened for reading, this function is emulated but can be + extremely slow. If the file is opened for writing, only forward seeks are + supported; gzseek then compresses a sequence of zeroes up to the new + starting position. + + gzseek returns the resulting offset location as measured in bytes from + the beginning of the uncompressed stream, or -1 in case of error, in + particular if the file is opened for writing and the new starting position + would be before the current position. +*/ + +ZEXTERN int ZEXPORT gzrewind OF((gzFile file)); +/* + Rewinds the given file. This function is supported only for reading. + + gzrewind(file) is equivalent to (int)gzseek(file, 0L, SEEK_SET) +*/ + +/* +ZEXTERN z_off_t ZEXPORT gztell OF((gzFile file)); + + Returns the starting position for the next gzread or gzwrite on the given + compressed file. This position represents a number of bytes in the + uncompressed data stream, and is zero when starting, even if appending or + reading a gzip stream from the middle of a file using gzdopen(). + + gztell(file) is equivalent to gzseek(file, 0L, SEEK_CUR) +*/ + +/* +ZEXTERN z_off_t ZEXPORT gzoffset OF((gzFile file)); + + Returns the current offset in the file being read or written. This offset + includes the count of bytes that precede the gzip stream, for example when + appending or when using gzdopen() for reading. When reading, the offset + does not include as yet unused buffered input. This information can be used + for a progress indicator. On error, gzoffset() returns -1. +*/ + +ZEXTERN int ZEXPORT gzeof OF((gzFile file)); +/* + Returns true (1) if the end-of-file indicator has been set while reading, + false (0) otherwise. Note that the end-of-file indicator is set only if the + read tried to go past the end of the input, but came up short. Therefore, + just like feof(), gzeof() may return false even if there is no more data to + read, in the event that the last read request was for the exact number of + bytes remaining in the input file. This will happen if the input file size + is an exact multiple of the buffer size. + + If gzeof() returns true, then the read functions will return no more data, + unless the end-of-file indicator is reset by gzclearerr() and the input file + has grown since the previous end of file was detected. +*/ + +ZEXTERN int ZEXPORT gzdirect OF((gzFile file)); +/* + Returns true (1) if file is being copied directly while reading, or false + (0) if file is a gzip stream being decompressed. + + If the input file is empty, gzdirect() will return true, since the input + does not contain a gzip stream. + + If gzdirect() is used immediately after gzopen() or gzdopen() it will + cause buffers to be allocated to allow reading the file to determine if it + is a gzip file. Therefore if gzbuffer() is used, it should be called before + gzdirect(). + + When writing, gzdirect() returns true (1) if transparent writing was + requested ("wT" for the gzopen() mode), or false (0) otherwise. (Note: + gzdirect() is not needed when writing. Transparent writing must be + explicitly requested, so the application already knows the answer. When + linking statically, using gzdirect() will include all of the zlib code for + gzip file reading and decompression, which may not be desired.) +*/ + +ZEXTERN int ZEXPORT gzclose OF((gzFile file)); +/* + Flushes all pending output if necessary, closes the compressed file and + deallocates the (de)compression state. Note that once file is closed, you + cannot call gzerror with file, since its structures have been deallocated. + gzclose must not be called more than once on the same file, just as free + must not be called more than once on the same allocation. + + gzclose will return Z_STREAM_ERROR if file is not valid, Z_ERRNO on a + file operation error, Z_MEM_ERROR if out of memory, Z_BUF_ERROR if the + last read ended in the middle of a gzip stream, or Z_OK on success. +*/ + +ZEXTERN int ZEXPORT gzclose_r OF((gzFile file)); +ZEXTERN int ZEXPORT gzclose_w OF((gzFile file)); +/* + Same as gzclose(), but gzclose_r() is only for use when reading, and + gzclose_w() is only for use when writing or appending. The advantage to + using these instead of gzclose() is that they avoid linking in zlib + compression or decompression code that is not used when only reading or only + writing respectively. If gzclose() is used, then both compression and + decompression code will be included the application when linking to a static + zlib library. +*/ + +ZEXTERN const char * ZEXPORT gzerror OF((gzFile file, int *errnum)); +/* + Returns the error message for the last error which occurred on the given + compressed file. errnum is set to zlib error number. If an error occurred + in the file system and not in the compression library, errnum is set to + Z_ERRNO and the application may consult errno to get the exact error code. + + The application must not modify the returned string. Future calls to + this function may invalidate the previously returned string. If file is + closed, then the string previously returned by gzerror will no longer be + available. + + gzerror() should be used to distinguish errors from end-of-file for those + functions above that do not distinguish those cases in their return values. +*/ + +ZEXTERN void ZEXPORT gzclearerr OF((gzFile file)); +/* + Clears the error and end-of-file flags for file. This is analogous to the + clearerr() function in stdio. This is useful for continuing to read a gzip + file that is being written concurrently. +*/ + +#endif /* !Z_SOLO */ + + /* checksum functions */ + +/* + These functions are not related to compression but are exported + anyway because they might be useful in applications using the compression + library. +*/ + +ZEXTERN uLong ZEXPORT adler32 OF((uLong adler, const Bytef *buf, uInt len)); +/* + Update a running Adler-32 checksum with the bytes buf[0..len-1] and + return the updated checksum. If buf is Z_NULL, this function returns the + required initial value for the checksum. + + An Adler-32 checksum is almost as reliable as a CRC-32 but can be computed + much faster. + + Usage example: + + uLong adler = adler32(0L, Z_NULL, 0); + + while (read_buffer(buffer, length) != EOF) { + adler = adler32(adler, buffer, length); + } + if (adler != original_adler) error(); +*/ + +ZEXTERN uLong ZEXPORT adler32_z OF((uLong adler, const Bytef *buf, + z_size_t len)); +/* + Same as adler32(), but with a size_t length. +*/ + +/* +ZEXTERN uLong ZEXPORT adler32_combine OF((uLong adler1, uLong adler2, + z_off_t len2)); + + Combine two Adler-32 checksums into one. For two sequences of bytes, seq1 + and seq2 with lengths len1 and len2, Adler-32 checksums were calculated for + each, adler1 and adler2. adler32_combine() returns the Adler-32 checksum of + seq1 and seq2 concatenated, requiring only adler1, adler2, and len2. Note + that the z_off_t type (like off_t) is a signed integer. If len2 is + negative, the result has no meaning or utility. +*/ + +ZEXTERN uLong ZEXPORT crc32 OF((uLong crc, const Bytef *buf, uInt len)); +/* + Update a running CRC-32 with the bytes buf[0..len-1] and return the + updated CRC-32. If buf is Z_NULL, this function returns the required + initial value for the crc. Pre- and post-conditioning (one's complement) is + performed within this function so it shouldn't be done by the application. + + Usage example: + + uLong crc = crc32(0L, Z_NULL, 0); + + while (read_buffer(buffer, length) != EOF) { + crc = crc32(crc, buffer, length); + } + if (crc != original_crc) error(); +*/ + +ZEXTERN uLong ZEXPORT crc32_z OF((uLong adler, const Bytef *buf, + z_size_t len)); +/* + Same as crc32(), but with a size_t length. +*/ + +/* +ZEXTERN uLong ZEXPORT crc32_combine OF((uLong crc1, uLong crc2, z_off_t len2)); + + Combine two CRC-32 check values into one. For two sequences of bytes, + seq1 and seq2 with lengths len1 and len2, CRC-32 check values were + calculated for each, crc1 and crc2. crc32_combine() returns the CRC-32 + check value of seq1 and seq2 concatenated, requiring only crc1, crc2, and + len2. +*/ + + + /* various hacks, don't look :) */ + +/* deflateInit and inflateInit are macros to allow checking the zlib version + * and the compiler's view of z_stream: + */ +ZEXTERN int ZEXPORT deflateInit_ OF((z_streamp strm, int level, + const char *version, int stream_size)); +ZEXTERN int ZEXPORT inflateInit_ OF((z_streamp strm, + const char *version, int stream_size)); +ZEXTERN int ZEXPORT deflateInit2_ OF((z_streamp strm, int level, int method, + int windowBits, int memLevel, + int strategy, const char *version, + int stream_size)); +ZEXTERN int ZEXPORT inflateInit2_ OF((z_streamp strm, int windowBits, + const char *version, int stream_size)); +ZEXTERN int ZEXPORT inflateBackInit_ OF((z_streamp strm, int windowBits, + unsigned char FAR *window, + const char *version, + int stream_size)); +#ifdef Z_PREFIX_SET +# define z_deflateInit(strm, level) \ + deflateInit_((strm), (level), ZLIB_VERSION, (int)sizeof(z_stream)) +# define z_inflateInit(strm) \ + inflateInit_((strm), ZLIB_VERSION, (int)sizeof(z_stream)) +# define z_deflateInit2(strm, level, method, windowBits, memLevel, strategy) \ + deflateInit2_((strm),(level),(method),(windowBits),(memLevel),\ + (strategy), ZLIB_VERSION, (int)sizeof(z_stream)) +# define z_inflateInit2(strm, windowBits) \ + inflateInit2_((strm), (windowBits), ZLIB_VERSION, \ + (int)sizeof(z_stream)) +# define z_inflateBackInit(strm, windowBits, window) \ + inflateBackInit_((strm), (windowBits), (window), \ + ZLIB_VERSION, (int)sizeof(z_stream)) +#else +# define deflateInit(strm, level) \ + deflateInit_((strm), (level), ZLIB_VERSION, (int)sizeof(z_stream)) +# define inflateInit(strm) \ + inflateInit_((strm), ZLIB_VERSION, (int)sizeof(z_stream)) +# define deflateInit2(strm, level, method, windowBits, memLevel, strategy) \ + deflateInit2_((strm),(level),(method),(windowBits),(memLevel),\ + (strategy), ZLIB_VERSION, (int)sizeof(z_stream)) +# define inflateInit2(strm, windowBits) \ + inflateInit2_((strm), (windowBits), ZLIB_VERSION, \ + (int)sizeof(z_stream)) +# define inflateBackInit(strm, windowBits, window) \ + inflateBackInit_((strm), (windowBits), (window), \ + ZLIB_VERSION, (int)sizeof(z_stream)) +#endif + +#ifndef Z_SOLO + +/* gzgetc() macro and its supporting function and exposed data structure. Note + * that the real internal state is much larger than the exposed structure. + * This abbreviated structure exposes just enough for the gzgetc() macro. The + * user should not mess with these exposed elements, since their names or + * behavior could change in the future, perhaps even capriciously. They can + * only be used by the gzgetc() macro. You have been warned. + */ +struct gzFile_s { + unsigned have; + unsigned char *next; + z_off64_t pos; +}; +ZEXTERN int ZEXPORT gzgetc_ OF((gzFile file)); /* backward compatibility */ +#ifdef Z_PREFIX_SET +# undef z_gzgetc +# define z_gzgetc(g) \ + ((g)->have ? ((g)->have--, (g)->pos++, *((g)->next)++) : (gzgetc)(g)) +#else +# define gzgetc(g) \ + ((g)->have ? ((g)->have--, (g)->pos++, *((g)->next)++) : (gzgetc)(g)) +#endif + +/* provide 64-bit offset functions if _LARGEFILE64_SOURCE defined, and/or + * change the regular functions to 64 bits if _FILE_OFFSET_BITS is 64 (if + * both are true, the application gets the *64 functions, and the regular + * functions are changed to 64 bits) -- in case these are set on systems + * without large file support, _LFS64_LARGEFILE must also be true + */ +#ifdef Z_LARGE64 + ZEXTERN gzFile ZEXPORT gzopen64 OF((const char *, const char *)); + ZEXTERN z_off64_t ZEXPORT gzseek64 OF((gzFile, z_off64_t, int)); + ZEXTERN z_off64_t ZEXPORT gztell64 OF((gzFile)); + ZEXTERN z_off64_t ZEXPORT gzoffset64 OF((gzFile)); + ZEXTERN uLong ZEXPORT adler32_combine64 OF((uLong, uLong, z_off64_t)); + ZEXTERN uLong ZEXPORT crc32_combine64 OF((uLong, uLong, z_off64_t)); +#endif + +#if !defined(ZLIB_INTERNAL) && defined(Z_WANT64) +# ifdef Z_PREFIX_SET +# define z_gzopen z_gzopen64 +# define z_gzseek z_gzseek64 +# define z_gztell z_gztell64 +# define z_gzoffset z_gzoffset64 +# define z_adler32_combine z_adler32_combine64 +# define z_crc32_combine z_crc32_combine64 +# else +# define gzopen gzopen64 +# define gzseek gzseek64 +# define gztell gztell64 +# define gzoffset gzoffset64 +# define adler32_combine adler32_combine64 +# define crc32_combine crc32_combine64 +# endif +# ifndef Z_LARGE64 + ZEXTERN gzFile ZEXPORT gzopen64 OF((const char *, const char *)); + ZEXTERN z_off_t ZEXPORT gzseek64 OF((gzFile, z_off_t, int)); + ZEXTERN z_off_t ZEXPORT gztell64 OF((gzFile)); + ZEXTERN z_off_t ZEXPORT gzoffset64 OF((gzFile)); + ZEXTERN uLong ZEXPORT adler32_combine64 OF((uLong, uLong, z_off_t)); + ZEXTERN uLong ZEXPORT crc32_combine64 OF((uLong, uLong, z_off_t)); +# endif +#else + ZEXTERN gzFile ZEXPORT gzopen OF((const char *, const char *)); + ZEXTERN z_off_t ZEXPORT gzseek OF((gzFile, z_off_t, int)); + ZEXTERN z_off_t ZEXPORT gztell OF((gzFile)); + ZEXTERN z_off_t ZEXPORT gzoffset OF((gzFile)); + ZEXTERN uLong ZEXPORT adler32_combine OF((uLong, uLong, z_off_t)); + ZEXTERN uLong ZEXPORT crc32_combine OF((uLong, uLong, z_off_t)); +#endif + +#else /* Z_SOLO */ + + ZEXTERN uLong ZEXPORT adler32_combine OF((uLong, uLong, z_off_t)); + ZEXTERN uLong ZEXPORT crc32_combine OF((uLong, uLong, z_off_t)); + +#endif /* !Z_SOLO */ + +/* undocumented functions */ +ZEXTERN const char * ZEXPORT zError OF((int)); +ZEXTERN int ZEXPORT inflateSyncPoint OF((z_streamp)); +ZEXTERN const z_crc_t FAR * ZEXPORT get_crc_table OF((void)); +ZEXTERN int ZEXPORT inflateUndermine OF((z_streamp, int)); +ZEXTERN int ZEXPORT inflateValidate OF((z_streamp, int)); +ZEXTERN unsigned long ZEXPORT inflateCodesUsed OF ((z_streamp)); +ZEXTERN int ZEXPORT inflateResetKeep OF((z_streamp)); +ZEXTERN int ZEXPORT deflateResetKeep OF((z_streamp)); +#if (defined(_WIN32) || defined(__CYGWIN__)) && !defined(Z_SOLO) +ZEXTERN gzFile ZEXPORT gzopen_w OF((const wchar_t *path, + const char *mode)); +#endif +#if defined(STDC) || defined(Z_HAVE_STDARG_H) +# ifndef Z_SOLO +ZEXTERN int ZEXPORTVA gzvprintf Z_ARG((gzFile file, + const char *format, + va_list va)); +# endif +#endif + +#ifdef __cplusplus +} +#endif + +#endif /* ZLIB_H */ diff --git a/include/Zlib/zutil.h b/include/Zlib/zutil.h new file mode 100644 index 0000000..b079ea6 --- /dev/null +++ b/include/Zlib/zutil.h @@ -0,0 +1,271 @@ +/* zutil.h -- internal interface and configuration of the compression library + * Copyright (C) 1995-2016 Jean-loup Gailly, Mark Adler + * For conditions of distribution and use, see copyright notice in zlib.h + */ + +/* WARNING: this file should *not* be used by applications. It is + part of the implementation of the compression library and is + subject to change. Applications should only use zlib.h. + */ + +/* @(#) $Id$ */ + +#ifndef ZUTIL_H +#define ZUTIL_H + +#ifdef HAVE_HIDDEN +# define ZLIB_INTERNAL __attribute__((visibility ("hidden"))) +#else +# define ZLIB_INTERNAL +#endif + +#include "zlib.h" + +#if defined(STDC) && !defined(Z_SOLO) +# if !(defined(_WIN32_WCE) && defined(_MSC_VER)) +# include +# endif +# include +# include +#endif + +#ifdef Z_SOLO + typedef long ptrdiff_t; /* guess -- will be caught if guess is wrong */ +#endif + +#ifndef local +# define local static +#endif +/* since "static" is used to mean two completely different things in C, we + define "local" for the non-static meaning of "static", for readability + (compile with -Dlocal if your debugger can't find static symbols) */ + +typedef unsigned char uch; +typedef uch FAR uchf; +typedef unsigned short ush; +typedef ush FAR ushf; +typedef unsigned long ulg; + +extern z_const char * const z_errmsg[10]; /* indexed by 2-zlib_error */ +/* (size given to avoid silly warnings with Visual C++) */ + +#define ERR_MSG(err) z_errmsg[Z_NEED_DICT-(err)] + +#define ERR_RETURN(strm,err) \ + return (strm->msg = ERR_MSG(err), (err)) +/* To be used only when the state is known to be valid */ + + /* common constants */ + +#ifndef DEF_WBITS +# define DEF_WBITS MAX_WBITS +#endif +/* default windowBits for decompression. MAX_WBITS is for compression only */ + +#if MAX_MEM_LEVEL >= 8 +# define DEF_MEM_LEVEL 8 +#else +# define DEF_MEM_LEVEL MAX_MEM_LEVEL +#endif +/* default memLevel */ + +#define STORED_BLOCK 0 +#define STATIC_TREES 1 +#define DYN_TREES 2 +/* The three kinds of block type */ + +#define MIN_MATCH 3 +#define MAX_MATCH 258 +/* The minimum and maximum match lengths */ + +#define PRESET_DICT 0x20 /* preset dictionary flag in zlib header */ + + /* target dependencies */ + +#if defined(MSDOS) || (defined(WINDOWS) && !defined(WIN32)) +# define OS_CODE 0x00 +# ifndef Z_SOLO +# if defined(__TURBOC__) || defined(__BORLANDC__) +# if (__STDC__ == 1) && (defined(__LARGE__) || defined(__COMPACT__)) + /* Allow compilation with ANSI keywords only enabled */ + void _Cdecl farfree( void *block ); + void *_Cdecl farmalloc( unsigned long nbytes ); +# else +# include +# endif +# else /* MSC or DJGPP */ +# include +# endif +# endif +#endif + +#ifdef AMIGA +# define OS_CODE 1 +#endif + +#if defined(VAXC) || defined(VMS) +# define OS_CODE 2 +# define F_OPEN(name, mode) \ + fopen((name), (mode), "mbc=60", "ctx=stm", "rfm=fix", "mrs=512") +#endif + +#ifdef __370__ +# if __TARGET_LIB__ < 0x20000000 +# define OS_CODE 4 +# elif __TARGET_LIB__ < 0x40000000 +# define OS_CODE 11 +# else +# define OS_CODE 8 +# endif +#endif + +#if defined(ATARI) || defined(atarist) +# define OS_CODE 5 +#endif + +#ifdef OS2 +# define OS_CODE 6 +# if defined(M_I86) && !defined(Z_SOLO) +# include +# endif +#endif + +#if defined(MACOS) || defined(TARGET_OS_MAC) +# define OS_CODE 7 +# ifndef Z_SOLO +# if defined(__MWERKS__) && __dest_os != __be_os && __dest_os != __win32_os +# include /* for fdopen */ +# else +# ifndef fdopen +# define fdopen(fd,mode) NULL /* No fdopen() */ +# endif +# endif +# endif +#endif + +#ifdef __acorn +# define OS_CODE 13 +#endif + +#if defined(WIN32) && !defined(__CYGWIN__) +# define OS_CODE 10 +#endif + +#ifdef _BEOS_ +# define OS_CODE 16 +#endif + +#ifdef __TOS_OS400__ +# define OS_CODE 18 +#endif + +#ifdef __APPLE__ +# define OS_CODE 19 +#endif + +#if defined(_BEOS_) || defined(RISCOS) +# define fdopen(fd,mode) NULL /* No fdopen() */ +#endif + +#if (defined(_MSC_VER) && (_MSC_VER > 600)) && !defined __INTERIX +# if defined(_WIN32_WCE) +# define fdopen(fd,mode) NULL /* No fdopen() */ +# ifndef _PTRDIFF_T_DEFINED + typedef int ptrdiff_t; +# define _PTRDIFF_T_DEFINED +# endif +# else +# define fdopen(fd,type) _fdopen(fd,type) +# endif +#endif + +#if defined(__BORLANDC__) && !defined(MSDOS) + #pragma warn -8004 + #pragma warn -8008 + #pragma warn -8066 +#endif + +/* provide prototypes for these when building zlib without LFS */ +#if !defined(_WIN32) && \ + (!defined(_LARGEFILE64_SOURCE) || _LFS64_LARGEFILE-0 == 0) + ZEXTERN uLong ZEXPORT adler32_combine64 OF((uLong, uLong, z_off_t)); + ZEXTERN uLong ZEXPORT crc32_combine64 OF((uLong, uLong, z_off_t)); +#endif + + /* common defaults */ + +#ifndef OS_CODE +# define OS_CODE 3 /* assume Unix */ +#endif + +#ifndef F_OPEN +# define F_OPEN(name, mode) fopen((name), (mode)) +#endif + + /* functions */ + +#if defined(pyr) || defined(Z_SOLO) +# define NO_MEMCPY +#endif +#if defined(SMALL_MEDIUM) && !defined(_MSC_VER) && !defined(__SC__) + /* Use our own functions for small and medium model with MSC <= 5.0. + * You may have to use the same strategy for Borland C (untested). + * The __SC__ check is for Symantec. + */ +# define NO_MEMCPY +#endif +#if defined(STDC) && !defined(HAVE_MEMCPY) && !defined(NO_MEMCPY) +# define HAVE_MEMCPY +#endif +#ifdef HAVE_MEMCPY +# ifdef SMALL_MEDIUM /* MSDOS small or medium model */ +# define zmemcpy _fmemcpy +# define zmemcmp _fmemcmp +# define zmemzero(dest, len) _fmemset(dest, 0, len) +# else +# define zmemcpy memcpy +# define zmemcmp memcmp +# define zmemzero(dest, len) memset(dest, 0, len) +# endif +#else + void ZLIB_INTERNAL zmemcpy OF((Bytef* dest, const Bytef* source, uInt len)); + int ZLIB_INTERNAL zmemcmp OF((const Bytef* s1, const Bytef* s2, uInt len)); + void ZLIB_INTERNAL zmemzero OF((Bytef* dest, uInt len)); +#endif + +/* Diagnostic functions */ +#ifdef ZLIB_DEBUG +# include + extern int ZLIB_INTERNAL z_verbose; + extern void ZLIB_INTERNAL z_error OF((char *m)); +# define Assert(cond,msg) {if(!(cond)) z_error(msg);} +# define Trace(x) {if (z_verbose>=0) fprintf x ;} +# define Tracev(x) {if (z_verbose>0) fprintf x ;} +# define Tracevv(x) {if (z_verbose>1) fprintf x ;} +# define Tracec(c,x) {if (z_verbose>0 && (c)) fprintf x ;} +# define Tracecv(c,x) {if (z_verbose>1 && (c)) fprintf x ;} +#else +# define Assert(cond,msg) +# define Trace(x) +# define Tracev(x) +# define Tracevv(x) +# define Tracec(c,x) +# define Tracecv(c,x) +#endif + +#ifndef Z_SOLO + voidpf ZLIB_INTERNAL zcalloc OF((voidpf opaque, unsigned items, + unsigned size)); + void ZLIB_INTERNAL zcfree OF((voidpf opaque, voidpf ptr)); +#endif + +#define ZALLOC(strm, items, size) \ + (*((strm)->zalloc))((strm)->opaque, (items), (size)) +#define ZFREE(strm, addr) (*((strm)->zfree))((strm)->opaque, (voidpf)(addr)) +#define TRY_FREE(s, p) {if (p) ZFREE(s, p);} + +/* Reverse the bytes in a 32-bit value */ +#define ZSWAP32(q) ((((q) >> 24) & 0xff) + (((q) >> 8) & 0xff00) + \ + (((q) & 0xff00) << 8) + (((q) & 0xff) << 24)) + +#endif /* ZUTIL_H */ diff --git a/src/Compressor.cpp b/src/Compressor.cpp new file mode 100644 index 0000000..9733650 --- /dev/null +++ b/src/Compressor.cpp @@ -0,0 +1,49 @@ +/// +/// Created by Anonymous275 on 7/15/2020 +/// +#include "Zlib/zlib.h" +#include + +#define Biggest 30000 +std::string Comp(std::string Data){ + char*C = new char[Biggest]; + memset(C, 0, Biggest); + z_stream defstream; + defstream.zalloc = Z_NULL; + defstream.zfree = Z_NULL; + defstream.opaque = Z_NULL; + defstream.avail_in = (uInt)Data.length(); + defstream.next_in = (Bytef *)&Data[0]; + defstream.avail_out = Biggest; + defstream.next_out = reinterpret_cast(C); + deflateInit(&defstream, Z_BEST_COMPRESSION); + deflate(&defstream, Z_SYNC_FLUSH); + deflate(&defstream, Z_FINISH); + deflateEnd(&defstream); + int TO = defstream.total_out; + std::string Ret(TO,0); + memcpy_s(&Ret[0],TO,C,TO); + delete [] C; + return Ret; +} +std::string DeComp(std::string Compressed){ + char*C = new char[Biggest]; + memset(C, 0, Biggest); + z_stream infstream; + infstream.zalloc = Z_NULL; + infstream.zfree = Z_NULL; + infstream.opaque = Z_NULL; + infstream.avail_in = Biggest; + infstream.next_in = (Bytef *)(&Compressed[0]); + infstream.avail_out = Biggest; + infstream.next_out = (Bytef *)(C); + inflateInit(&infstream); + inflate(&infstream, Z_SYNC_FLUSH); + inflate(&infstream, Z_FINISH); + inflateEnd(&infstream); + int TO = infstream.total_out; + std::string Ret(TO,0); + memcpy_s(&Ret[0],TO,C,TO); + delete [] C; + return Ret; +} \ No newline at end of file diff --git a/src/Enc.cpp b/src/Enc.cpp new file mode 100644 index 0000000..a01eb73 --- /dev/null +++ b/src/Enc.cpp @@ -0,0 +1,113 @@ +/// +/// Created by Anonymous275 on 7/28/2020 +/// +#include "Security/Enc.h" +#include "Settings.h" +#include +#include "Logger.h" +#include +#include +#include + +int Rand(){ + std::random_device r; + std::default_random_engine e1(r()); + std::uniform_int_distribution uniform_dist(1, 5000); + return uniform_dist(e1); +} + +int log_power(int n,unsigned int p, int mod){ + int result = 1; + for (; p; p >>= 1u){ + if (p & 1u)result = int((1LL * result * n) % mod); + n = int((1LL * n * n) % mod); + } + return result; +} +bool rabin_miller(int n){ + bool ok = true; + for (int i = 1; i <= 5 && ok; i++) { + int a = Rand() + 1; + int result = log_power(a, n - 1, n); + ok &= (result == 1); + } + return ok; +} +int generate_prime(){ + int generated = Rand(); + while (!rabin_miller(generated))generated = Rand(); + return generated; +} +int gcd(int a, int b){ + while (b){ + int r = a % b; + a = b; + b = r; + } + return a; +} + +int generate_coprime(int n){ + int generated = Rand(); + while (gcd(n, generated) != 1)generated = Rand(); + return generated; +} + +std::pair euclid_extended(int a, int b) { + if(!b)return {1, 0}; + auto result = euclid_extended(b, a % b); + return {result.second, result.first - (a / b) * result.second}; +} + +int modular_inverse(int n, int mod){ + int inverse = euclid_extended(n, mod).first; + while(inverse < 0)inverse += mod; + return inverse; +} + +RSA* GenKey(){ + int p, q; + p = generate_prime(); + q = generate_prime(); + int n = p * q; + int phi = (p -1) * (q - 1); + int e = generate_coprime(phi); + int d = modular_inverse(e, phi); + return new RSA{n,e,d}; +} + +int Enc(int value,int e,int n){ + return log_power(value, e, n); +} + +int Dec(int value,int d,int n){ + return log_power(value, d, n); +} + +int Handle(EXCEPTION_POINTERS *ep,char* Origin){ + std::stringstream R; + R << Sec("Code : ") << std::hex + << ep->ExceptionRecord->ExceptionCode + << std::dec << Sec(" Origin : ") << Origin; + except(R.str()); + return 1; +} + +std::string RSA_E(const std::string& Data, RSA*k){ + std::stringstream stream; + for(const char&c : Data){ + stream << std::hex << Enc(uint8_t(c),k->e,k->n) << "g"; + } + return stream.str(); +} + +std::string RSA_D(const std::string& Data, RSA*k){ + std::stringstream ss(Data); + std::string token,ret; + while (std::getline(ss, token, 'g')) { + if(token.find_first_not_of(Sec("0123456789abcdef")) != std::string::npos)return ""; + int c = std::stoi(token, nullptr, 16); + ret += char(Dec(c,k->d,k->n)); + } + return ret; +} \ No newline at end of file diff --git a/src/Init/Config.cpp b/src/Init/Config.cpp new file mode 100644 index 0000000..9789121 --- /dev/null +++ b/src/Init/Config.cpp @@ -0,0 +1,149 @@ +/// +/// Created by Anonymous275 on 7/28/2020 +/// +#include "Security/Enc.h" +#include "Logger.h" +#include +#include +#include +std::string ServerName; +std::string ServerDesc; +std::string Resource; +std::string MapName; +std::string Key; +int MaxPlayers; +bool Private; +int MaxCars; +bool Debug; +int Port; + +void SetValues(const std::string& Line, int Index) { + int state = 0; + std::string Data; + bool Switch = false; + if (Index > 5)Switch = true; + for (char c : Line) { + if (Switch){ + if (c == '\"')state++; + if (state > 0 && state < 2)Data += c; + }else{ + if (c == ' ')state++; + if (state > 1)Data += c; + } + } + Data = Data.substr(1); + std::string::size_type sz; + bool Boolean = std::string(Data).find("true") != -1;//searches for "true" + switch (Index) { + case 1 : + Debug = Boolean;//checks and sets the Debug Value + break; + case 2 : + Private = Boolean;//checks and sets the Private Value + break; + case 3 : + Port = std::stoi(Data, &sz);//sets the Port + break; + case 4 : + MaxCars = std::stoi(Data, &sz);//sets the Max Car amount + break; + case 5 : + MaxPlayers = std::stoi(Data, &sz); //sets the Max Amount of player + break; + case 6 : + MapName = Data; //Map + break; + case 7 : + ServerName = Data; //Name + break; + case 8 : + ServerDesc = Data; //desc + break; + case 9 : + Resource = Data; //File name + break; + case 10 : + Key = Data; //File name + default: + break; + } +} +std::string RemoveComments(const std::string& Line){ + std::string Return; + for(char c : Line) { + if(c == '#')break; + Return += c; + } + return Return; +} +void LoadConfig(std::ifstream& IFS){ + std::string line; + int index = 1; + while (getline(IFS, line)) { + index++; + } + if(index-1 < 11){ + error(Sec("Outdated/Incorrect config please remove it server will close in 5 secs")); + std::this_thread::sleep_for(std::chrono::seconds(3)); + exit(0); + } + IFS.close(); + IFS.open(Sec("Server.cfg")); + info(Sec("Config found updating values")); + index = 1; + while (getline(IFS, line)) { + if(line.rfind('#', 0) != 0 && line.rfind(' ', 0) != 0){ //Checks if it starts as Comment + std::string CleanLine = RemoveComments(line); //Cleans it from the Comments + SetValues(CleanLine,index); //sets the values + index++; + } + } +} +void GenerateConfig(){ + std::ofstream FileStream; + FileStream.open (Sec("Server.cfg")); + FileStream << Sec("# This is the BeamMP Server Configuration File v0.60\n" + "Debug = false # true or false to enable debug console output\n" + "Private = false # Private?\n" + "Port = 30814 # Port to run the server on UDP and TCP\n" + "Cars = 1 # Max cars for every player\n" + "MaxPlayers = 10 # Maximum Amount of Clients\n" + "Map = \"/levels/gridmap/info.json\" # Default Map\n" + "Name = \"BeamMP New Server\" # Server Name\n" + "Desc = \"BeamMP Default Description\" # Server Description\n" + "use = \"Resources\" # Resource file name\n" + "AuthKey = \"\" # Auth Key"); + FileStream.close(); +} +void Default(){ + info(Sec("Config not found generating default")); + GenerateConfig(); + warn(Sec("You are required to input the AuthKey")); + std::this_thread::sleep_for(std::chrono::seconds(3)); + exit(0); +} +void DebugData(){ + debug(std::string(Sec("Debug : ")) + (Debug?"true":"false")); + debug(std::string(Sec("Private : ")) + (Private?"true":"false")); + debug(Sec("Port : ") + std::to_string(Port)); + debug(Sec("Max Cars : ") + std::to_string(MaxCars)); + debug(Sec("MaxPlayers : ") + std::to_string(MaxPlayers)); + debug(Sec("MapName : ") + MapName); + debug(Sec("ServerName : ") + ServerName); + debug(Sec("ServerDesc : ") + ServerDesc); + debug(Sec("File : ") + Resource); + debug(Sec("Key length: ") + std::to_string(Key.length())); +} +void InitConfig(){ + std::ifstream IFS; + IFS.open(Sec("Server.cfg")); + if(IFS.good())LoadConfig(IFS); + else Default(); + if(IFS.is_open())IFS.close(); + if(Key.empty()){ + error(Sec("No AuthKey was found")); + std::this_thread::sleep_for(std::chrono::seconds(3)); + exit(0); + } + if(Debug)DebugData(); +} \ No newline at end of file diff --git a/src/Init/Heartbeat.cpp b/src/Init/Heartbeat.cpp new file mode 100644 index 0000000..fcae4d5 --- /dev/null +++ b/src/Init/Heartbeat.cpp @@ -0,0 +1,59 @@ +/// +/// Created by Anonymous275 on 7/28/2020 +/// +#include "Security/Enc.h" +#include "Curl/Http.h" +#include "Client.hpp" +#include "Settings.h" +#include "Logger.h" +#include +#include + +std::string GetPlayers(){ + std::string Return; + for(Client* c : CI->Clients){ + if(c != nullptr){ + Return += c->GetName() + ";"; + } + } + return Return; +} +std::string GenerateCall(){ + std::string State = Private ? Sec("true") : Sec("false"); + std::string ret = Sec("uuid="); + ret += Key+Sec("&players=")+std::to_string(CI->Size())+Sec("&maxplayers=")+std::to_string(MaxPlayers)+Sec("&port=") + + std::to_string(Port) + Sec("&map=") + MapName + Sec("&private=")+State+Sec("&version=")+GetSVer()+ + Sec("&clientversion=")+GetCVer()+Sec("&name=")+ServerName+Sec("&pps=")+StatReport+Sec("&modlist=")+FileList+ + Sec("&modstotalsize=")+std::to_string(MaxModSize)+Sec("&modstotal=")+std::to_string(ModsLoaded) + +Sec("&playerslist=")+GetPlayers()+Sec("&desc=")+ServerDesc; + return ret; +} +void Heartbeat(){ + std::string R,T; + while(true){ + R = GenerateCall(); + if(!CustomIP.empty())R+=Sec("&ip=")+CustomIP; + //https://beamng-mp.com/heartbeatv2 + std::string link = Sec("https://beamng-mp.com/heartbeatv2"); + T = PostHTTP(link,R); + if(T.find_first_not_of(Sec("20")) != std::string::npos){ + //Backend system refused server startup! + std::this_thread::sleep_for(std::chrono::seconds(10)); + T = PostHTTP(link,R); + if(T.find_first_not_of(Sec("20")) != std::string::npos){ + error(Sec("Backend system refused server! Check your AuthKey")); + std::this_thread::sleep_for(std::chrono::seconds(3)); + exit(-1); + } + } + //Server Authenticated + if(T.length() == 4)info(Sec("Server authenticated")); + R.clear(); + T.clear(); + std::this_thread::sleep_for(std::chrono::seconds(3)); + } +} +void HBInit(){ + std::thread HB(Heartbeat); + HB.detach(); +} \ No newline at end of file diff --git a/src/Resources.cpp b/src/Init/Resources.cpp similarity index 55% rename from src/Resources.cpp rename to src/Init/Resources.cpp index 61d6297..233f192 100644 --- a/src/Resources.cpp +++ b/src/Init/Resources.cpp @@ -1,29 +1,22 @@ /// -/// Created by Anonymous275 on 4/11/2020 +/// Created by Anonymous275 on 7/28/2020 /// - -#include +#include "Security/Enc.h" #include - -namespace fs = std::filesystem; - -std::string FileList; +#include "Settings.h" +#include +#include "Logger.h" +namespace fs = std::experimental::filesystem; +uint64_t MaxModSize = 0; std::string FileSizes; +std::string FileList; int ModsLoaded = 0; -long MaxModSize = 0; -void LuaMain(std::string Path); -void HandleResources(std::string path){ - struct stat info{}; - if(stat(path.c_str(), &info) != 0){ - fs::create_directory(path); - } - LuaMain(path); - path += "/Client"; - if(stat(path.c_str(), &info) != 0) { - fs::create_directory(path); - } - for (const auto & entry : fs::directory_iterator(path)){ - int pos = entry.path().string().find(".zip"); + +void InitRes(){ + std::string Path = Resource + Sec("/Client"); + if(!fs::exists(Path))fs::create_directory(Path); + for (const auto & entry : fs::directory_iterator(Path)){ + auto pos = entry.path().string().find(Sec(".zip")); if(pos != std::string::npos){ if(entry.path().string().length() - pos == 4){ FileList += entry.path().string() + ";"; @@ -34,4 +27,7 @@ void HandleResources(std::string path){ } } std::replace(FileList.begin(),FileList.end(),'\\','/'); + if(ModsLoaded){ + info(Sec("Loaded ")+std::to_string(ModsLoaded)+Sec(" Mods")); + } } \ No newline at end of file diff --git a/src/Init/Startup.cpp b/src/Init/Startup.cpp new file mode 100644 index 0000000..2c17116 --- /dev/null +++ b/src/Init/Startup.cpp @@ -0,0 +1,32 @@ +/// +/// Created by Anonymous275 on 7/28/2020 +/// +#include "Security/Enc.h" +#include "Client.hpp" +#include "Logger.h" +#include + +std::string CustomIP; +std::string GetSVer(){ + return Sec("0.60"); +} +std::string GetCVer(){ + return Sec("1.60"); +} +void Args(int argc, char* argv[]){ + info(Sec("BeamMP Server Running version ") + GetSVer()); + if(argc > 1){ + CustomIP = argv[1]; + size_t n = std::count(CustomIP.begin(), CustomIP.end(), '.'); + auto p = CustomIP.find_first_not_of(Sec(".0123456789")); + if(p != std::string::npos || n != 3 || CustomIP.substr(0,3) == Sec("127")){ + CustomIP.clear(); + warn(Sec("IP Specified is invalid! Ignoring")); + }else info(Sec("Server started with custom IP")); + } +} +void InitServer(int argc, char* argv[]){ + InitLog(); + Args(argc,argv); + CI = new ClientInterface; +} \ No newline at end of file diff --git a/src/Lua System/LuaMain.cpp b/src/Lua/LuaMain.cpp similarity index 74% rename from src/Lua System/LuaMain.cpp rename to src/Lua/LuaMain.cpp index e9da94e..f788c91 100644 --- a/src/Lua System/LuaMain.cpp +++ b/src/Lua/LuaMain.cpp @@ -2,9 +2,12 @@ /// Created by Anonymous275 on 5/20/2020 /// -#include "LuaSystem.hpp" -#include "../logger.h" +#include "Lua/LuaSystem.hpp" +#include "Security/Enc.h" +#include "Settings.h" +#include "Logger.h" #include + std::set PluginEngine; bool NewFile(const std::string&Path){ for(Lua*Script : PluginEngine){ @@ -14,9 +17,9 @@ bool NewFile(const std::string&Path){ } void RegisterFiles(const std::string& Path,bool HotSwap){ std::string Name = Path.substr(Path.find_last_of('\\')+1); - if(!HotSwap)info("Loading plugin : " + Name); + if(!HotSwap)info(Sec("Loading plugin : ") + Name); for (const auto &entry : fs::directory_iterator(Path)){ - int pos = entry.path().string().find(".lua"); + auto pos = entry.path().string().find(Sec(".lua")); if (pos != std::string::npos && entry.path().string().length() - pos == 4) { if(!HotSwap || NewFile(entry.path().string())){ Lua *Script = new Lua(); @@ -25,7 +28,7 @@ void RegisterFiles(const std::string& Path,bool HotSwap){ Script->SetPluginName(Name); Script->SetLastWrite(fs::last_write_time(Script->GetFileName())); Script->Init(); - if(HotSwap)info("[HOTSWAP] Added : " + + if(HotSwap)info(Sec("[HOTSWAP] Added : ") + Script->GetFileName().substr(Script->GetFileName().find('\\'))); } } @@ -33,7 +36,7 @@ void RegisterFiles(const std::string& Path,bool HotSwap){ } void FolderList(const std::string& Path,bool HotSwap){ for (const auto &entry : fs::directory_iterator(Path)) { - int pos = entry.path().filename().string().find('.'); + auto pos = entry.path().filename().string().find('.'); if (pos == std::string::npos) { RegisterFiles(entry.path().string(),HotSwap); } @@ -44,13 +47,15 @@ void FolderList(const std::string& Path,bool HotSwap){ for(Lua*Script : PluginEngine){ struct stat Info{}; if(stat(Script->GetFileName().c_str(), &Info) != 0){ + Script->StopThread = true; PluginEngine.erase(Script); - info("[HOTSWAP] Removed : "+ + info(Sec("[HOTSWAP] Removed : ")+ Script->GetFileName().substr(Script->GetFileName().find('\\'))); break; } if(Script->GetLastWrite() != fs::last_write_time(Script->GetFileName())){ - info("[HOTSWAP] Updated : "+ + Script->StopThread = true; + info(Sec("[HOTSWAP] Updated : ")+ Script->GetFileName().substr(Script->GetFileName().find('\\'))); Script->SetLastWrite(fs::last_write_time(Script->GetFileName())); Script->Reload(); @@ -61,16 +66,16 @@ void FolderList(const std::string& Path,bool HotSwap){ } } - - -void LuaMain(std::string Path){ - Path += "/Server"; - struct stat Info{}; - if(stat( Path.c_str(), &Info) != 0){ +void InitLua(){ + if(!fs::exists(Resource)){ + fs::create_directory(Resource); + } + std::string Path = Resource + Sec("/Server"); + if(!fs::exists(Path)){ fs::create_directory(Path); } FolderList(Path,false); std::thread t1(HotSwaps,Path); t1.detach(); - info("Lua system online"); + info(Sec("Lua system online")); } diff --git a/src/Lua System/LuaSystem.cpp b/src/Lua/LuaSystem.cpp similarity index 53% rename from src/Lua System/LuaSystem.cpp rename to src/Lua/LuaSystem.cpp index 7f4b8a5..467de44 100644 --- a/src/Lua System/LuaSystem.cpp +++ b/src/Lua/LuaSystem.cpp @@ -2,15 +2,18 @@ /// Created by Anonymous275 on 5/19/2020 /// -#include "../Network 2.0/Client.hpp" -#include "LuaSystem.hpp" -#include "../logger.h" +#include "Lua/LuaSystem.hpp" +#include "Security/Enc.h" +#include "Client.hpp" +#include "Network.h" +#include "Logger.h" #include #include -LuaArg* CreateArg(lua_State *L,int T){ - LuaArg* temp = new LuaArg; - for(int C = 2;C <= T;C++){ +LuaArg* CreateArg(lua_State *L,int T,int S){ + if(S > T)return nullptr; + auto* temp = new LuaArg; + for(int C = S;C <= T;C++){ if(lua_isstring(L,C)){ temp->args.emplace_back(std::string(lua_tostring(L,C))); }else if(lua_isinteger(L,C)){ @@ -23,31 +26,6 @@ LuaArg* CreateArg(lua_State *L,int T){ } return temp; } - - -int TriggerLuaEvent(const std::string& Event,bool local,Lua*Caller,LuaArg* arg){ - int R = 0; - for(Lua*Script : PluginEngine){ - if(Script->IsRegistered(Event)){ - if(local){ - if (Script->GetPluginName() == Caller->GetPluginName()){ - R += Script->CallFunction(Script->GetRegistered(Event),arg); - } - }else R += Script->CallFunction(Script->GetRegistered(Event), arg); - } - } - return R; -} -bool CheckLua(lua_State *L, int r) -{ - if (r != LUA_OK) - { - std::string errormsg = lua_tostring(L, -1); - std::cout << errormsg << std::endl; - return false; - } - return true; -} Lua* GetScript(lua_State *L){ for(Lua*Script : PluginEngine){ if (Script->GetState() == L)return Script; @@ -55,121 +33,199 @@ Lua* GetScript(lua_State *L){ return nullptr; } void SendError(lua_State *L,const std::string&msg){ - Lua* Script = GetScript(L); - error(Script->GetFileName() + " | Incorrect Call of " +msg); + Lua* S = GetScript(L); + std::string a = S->GetFileName().substr(S->GetFileName().find('\\')); + warn(a + Sec(" | Incorrect Call of ") +msg); } -int lua_RegisterEvent(lua_State *L) -{ +int Trigger(Lua*lua,const std::string& R, LuaArg*arg){ + std::packaged_task task([lua,R,arg]{return CallFunction(lua,R,arg);}); + std::future f1 = task.get_future(); + std::thread t(std::move(task)); + t.detach(); + auto status = f1.wait_for(std::chrono::seconds(3)); + if(status != std::future_status::timeout)return f1.get(); + SendError(lua->GetState(),R + " took too long to respond"); + return 0; +} +int TriggerLuaEvent(const std::string& Event,bool local,Lua*Caller,LuaArg* arg){ + int R = 0; + for(Lua*Script : PluginEngine){ + if(Script->IsRegistered(Event)){ + if(local){ + if (Script->GetPluginName() == Caller->GetPluginName()){ + R += Trigger(Script,Script->GetRegistered(Event),arg); + } + }else R += Trigger(Script,Script->GetRegistered(Event), arg); + } + } + if(arg != nullptr){ + delete arg; + arg = nullptr; + } + return R; +} + +bool CheckLua(lua_State *L, int r){ + if (r != LUA_OK){ + std::string msg = lua_tostring(L, -1); + Lua * S = GetScript(L); + std::string a = S->GetFileName().substr(S->GetFileName().find('\\')); + warn(a + " | at line " + msg.substr(msg.find(':')+1)); + return false; + } + return true; +} + +int lua_RegisterEvent(lua_State *L){ int Args = lua_gettop(L); Lua* Script = GetScript(L); if(Args == 2 && lua_isstring(L,1) && lua_isstring(L,2)){ Script->RegisterEvent(lua_tostring(L,1),lua_tostring(L,2)); - }else SendError(L,"RegisterEvent invalid argument count expected 2 got " + std::to_string(Args)); + }else SendError(L,Sec("RegisterEvent invalid argument count expected 2 got ") + std::to_string(Args)); return 0; } -int lua_TriggerEventL(lua_State *L) -{ +int lua_TriggerEventL(lua_State *L){ int Args = lua_gettop(L); Lua* Script = GetScript(L); if(Args > 0){ - if(lua_isstring(L,1)) { - TriggerLuaEvent(lua_tostring(L, 1), true, Script, CreateArg(L,Args)); - }else{ - SendError(L,"TriggerLocalEvent wrong argument [1] need string"); - } + if(lua_isstring(L,1)){ + TriggerLuaEvent(lua_tostring(L, 1), true, Script, CreateArg(L,Args,2)); + }else SendError(L,Sec("TriggerLocalEvent wrong argument [1] need string")); }else{ - SendError(L,"TriggerLocalEvent not enough arguments expected 1 got 0"); + SendError(L,Sec("TriggerLocalEvent not enough arguments expected 1 got 0")); } return 0; } -int lua_TriggerEventG(lua_State *L) -{ +int lua_TriggerEventG(lua_State *L){ int Args = lua_gettop(L); Lua* Script = GetScript(L); if(Args > 0){ if(lua_isstring(L,1)) { - TriggerLuaEvent(lua_tostring(L, 1), false, Script, CreateArg(L,Args)); - }else SendError(L,"TriggerGlobalEvent wrong argument [1] need string"); - }else{ - SendError(L,"TriggerGlobalEvent not enough arguments"); - } + TriggerLuaEvent(lua_tostring(L, 1), false, Script, CreateArg(L,Args,2)); + }else SendError(L,Sec("TriggerGlobalEvent wrong argument [1] need string")); + }else SendError(L,Sec("TriggerGlobalEvent not enough arguments")); return 0; } -void CallAsync(Lua* lua,const std::string& FuncName,LuaArg* args){ + +char* ThreadOrigin(Lua*lua){ + std::string T = "Thread in " + lua->GetFileName().substr(lua->GetFileName().find('\\')); + char* Data = new char[T.size()]; + ZeroMemory(Data,T.size()); + memcpy_s(Data,T.size(),T.c_str(),T.size()); + return Data; +} + +void Lock(Lua* lua,bool thread){ + bool Lock; + do{ + if(thread){ + Lock = lua->isExecuting; + }else Lock = lua->isThreadExecuting; + std::this_thread::sleep_for(std::chrono::milliseconds(1)); + }while(Lock); +} +void ExecuteAsync(Lua* lua,const std::string& FuncName){ + Lock(lua,true); + lua->isThreadExecuting = true; + lua_State* luaState = lua->GetState(); + lua_getglobal(luaState, FuncName.c_str()); + if(lua_isfunction(luaState, -1)) { + char* Origin = ThreadOrigin(lua); + __try{ + int R = lua_pcall(luaState, 0, 0, 0); + CheckLua(luaState, R); + }__except(Handle(GetExceptionInformation(),Origin)){} + delete [] Origin; + } + lua->isThreadExecuting = false; +} +void CallAsync(Lua* lua,const std::string& Func,int U){ if(lua->HasThread){ - SendError(lua->GetState(),"CreateThread there is a thread already running"); + SendError(lua->GetState(),Sec("CreateThread : There is already a thread running!")); return; } + lua->StopThread = false; lua->HasThread = true; - std::this_thread::sleep_for(std::chrono::milliseconds(200)); - lua->CallFunction(FuncName,args); + int D = 1000 / U; + while(!lua->StopThread){ + ExecuteAsync(lua,Func); + std::this_thread::sleep_for(std::chrono::milliseconds(D)); + } lua->HasThread = false; } +int lua_StopThread(lua_State *L){ + GetScript(L)->StopThread = true; + return 0; +} int lua_CreateThread(lua_State *L){ int Args = lua_gettop(L); - if(Args > 0){ + if(Args > 1){ if(lua_isstring(L,1)) { std::string STR = lua_tostring(L,1); - std::thread t1(CallAsync,GetScript(L),STR,CreateArg(L,Args)); - t1.detach(); - }else SendError(L,"CreateThread wrong argument [1] need string"); - }else SendError(L,"CreateThread not enough arguments"); + if(lua_isinteger(L,2) || lua_isnumber(L,2)){ + int U = int(lua_tointeger(L,2)); + if(U > 0 && U < 501){ + std::thread t1(CallAsync,GetScript(L),STR,U); + t1.detach(); + }else SendError(L,Sec("CreateThread wrong argument [2] number must be between 1 and 500")); + }else SendError(L,Sec("CreateThread wrong argument [2] need number")); + }else SendError(L,Sec("CreateThread wrong argument [1] need string")); + }else SendError(L,Sec("CreateThread not enough arguments")); return 0; } int lua_Sleep(lua_State *L){ if(lua_isnumber(L,1)){ - int t = lua_tonumber(L, 1); + int t = int(lua_tonumber(L, 1)); std::this_thread::sleep_for(std::chrono::milliseconds(t)); }else{ - SendError(L,"Sleep not enough arguments"); + SendError(L,Sec("Sleep not enough arguments")); return 0; } return 1; } Client* GetClient(int ID){ - for(Client*c:Clients) { - if(c->GetID() == ID)return c; + for(Client*c : CI->Clients) { + if(c != nullptr && c->GetID() == ID)return c; } return nullptr; } -int lua_isConnected(lua_State *L) -{ +int lua_isConnected(lua_State *L){ if(lua_isnumber(L,1)){ - int ID = lua_tonumber(L, 1); + int ID = int(lua_tonumber(L, 1)); Client*c = GetClient(ID); - if(c != nullptr)lua_pushboolean(L, c->isConnected()); + if(c != nullptr)lua_pushboolean(L, c->isConnected); else return 0; }else{ - SendError(L,"CreateThread not enough arguments"); + SendError(L,Sec("isConnected not enough arguments")); return 0; } return 1; } int lua_GetPlayerName(lua_State *L){ if(lua_isnumber(L,1)){ - int ID = lua_tonumber(L, 1); + int ID = int(lua_tonumber(L, 1)); Client*c = GetClient(ID); if(c != nullptr)lua_pushstring(L, c->GetName().c_str()); else return 0; }else{ - SendError(L,"CreateThread not enough arguments"); + SendError(L,Sec("GetPlayerName not enough arguments")); return 0; } return 1; } int lua_GetPlayerCount(lua_State *L){ - lua_pushinteger(L, Clients.size()); + lua_pushinteger(L, CI->Size()); return 1; } int lua_GetDID(lua_State *L){ if(lua_isnumber(L,1)){ - int ID = lua_tonumber(L, 1); + int ID = int(lua_tonumber(L, 1)); Client*c = GetClient(ID); if(c != nullptr)lua_pushstring(L, c->GetDID().c_str()); else return 0; }else{ - SendError(L,"GetDID not enough arguments"); + SendError(L,Sec("GetDID not enough arguments")); return 0; } return 1; @@ -177,85 +233,87 @@ int lua_GetDID(lua_State *L){ int lua_GetAllPlayers(lua_State *L){ lua_newtable(L); int i = 1; - for (Client *c : Clients) { + for (Client *c : CI->Clients) { + if(c == nullptr)continue; lua_pushinteger(L, c->GetID()); lua_pushstring(L, c->GetName().c_str()); lua_settable(L,-3); i++; } - if(Clients.empty())return 0; + if(CI->Clients.empty())return 0; return 1; } int lua_GetCars(lua_State *L){ if(lua_isnumber(L,1)){ - int ID = lua_tonumber(L, 1); + int ID = int(lua_tonumber(L, 1)); Client*c = GetClient(ID); if(c != nullptr){ int i = 1; - for (const std::pair &a : c->GetAllCars()) { - lua_pushinteger(L, a.first); - lua_pushstring(L, a.second.substr(3).c_str()); - lua_settable(L,-3); - i++; + for (VData*v : c->GetAllCars()) { + if(v != nullptr) { + lua_pushinteger(L, v->ID); + lua_pushstring(L, v->Data.substr(3).c_str()); + lua_settable(L, -3); + i++; + } } if(c->GetAllCars().empty())return 0; }else return 0; }else{ - SendError(L,"GetPlayerVehicles not enough arguments"); + SendError(L,Sec("GetPlayerVehicles not enough arguments")); return 0; } return 1; } -void Respond(Client*c, const std::string& MSG, bool Rel); int lua_dropPlayer(lua_State *L){ int Args = lua_gettop(L); if(lua_isnumber(L,1)){ - int ID = lua_tonumber(L, 1); + int ID = int(lua_tonumber(L, 1)); Client*c = GetClient(ID); if(c == nullptr)return 0; - if(c->GetRole() == "MDEV")return 0; + if(c->GetRole() == Sec("MDEV"))return 0; std::string Reason; if(Args > 1 && lua_isstring(L,2)){ - Reason = std::string(" Reason : ")+lua_tostring(L,2); + Reason = std::string(Sec(" Reason : "))+lua_tostring(L,2); } - Respond(c,"C:Server:You have been Kicked from the server!" + Reason,true); + Respond(c,"C:Server:You have been Kicked from the server! " + Reason,true); c->SetStatus(-2); closesocket(c->GetTCPSock()); - }else SendError(L,"DropPlayer not enough arguments"); + }else SendError(L,Sec("DropPlayer not enough arguments")); return 0; } -void SendToAll(Client*c, const std::string& Data, bool Self, bool Rel); int lua_sendChat(lua_State *L){ if(lua_isinteger(L,1) || lua_isnumber(L,1)){ if(lua_isstring(L,2)){ - int ID = lua_tointeger(L,1); + int ID = int(lua_tointeger(L,1)); if(ID == -1){ std::string Packet = "C:Server: " + std::string(lua_tostring(L, 2)); SendToAll(nullptr,Packet,true,true); }else{ Client*c = GetClient(ID); if(c != nullptr) { - std::string Packet = "C:Server: " + std::string(lua_tostring(L, 2)); + if(!c->isSynced)return 0; + std::string Packet ="C:Server: " + std::string(lua_tostring(L, 2)); Respond(c, Packet, true); - }else SendError(L,"SendChatMessage invalid argument [1] invalid ID"); + }else SendError(L,Sec("SendChatMessage invalid argument [1] invalid ID")); } - }else SendError(L,"SendChatMessage invalid argument [2] expected string"); - }else SendError(L,"SendChatMessage invalid argument [1] expected number"); + }else SendError(L,Sec("SendChatMessage invalid argument [2] expected string")); + }else SendError(L,Sec("SendChatMessage invalid argument [1] expected number")); return 0; } int lua_RemoveVehicle(lua_State *L){ int Args = lua_gettop(L); if(Args != 2){ - SendError(L,"RemoveVehicle invalid argument count expected 2 got " + std::to_string(Args)); + SendError(L,Sec("RemoveVehicle invalid argument count expected 2 got ") + std::to_string(Args)); return 0; } if((lua_isinteger(L,1) || lua_isnumber(L,1)) && (lua_isinteger(L,2) || lua_isnumber(L,2))){ - int PID = lua_tointeger(L,1); - int VID = lua_tointeger(L,2); + int PID = int(lua_tointeger(L,1)); + int VID = int(lua_tointeger(L,2)); Client *c = GetClient(PID); if(c == nullptr){ - SendError(L,"RemoveVehicle invalid Player ID"); + SendError(L,Sec("RemoveVehicle invalid Player ID")); return 0; } if(c->GetRole() == "MDEV")return 0; @@ -264,7 +322,7 @@ int lua_RemoveVehicle(lua_State *L){ SendToAll(nullptr,Destroy,true,true); c->DeleteCar(VID); } - }else SendError(L,"RemoveVehicle invalid argument expected number"); + }else SendError(L,Sec("RemoveVehicle invalid argument expected number")); return 0; } int lua_HWID(lua_State *L){ @@ -285,6 +343,7 @@ void Lua::Init(){ lua_register(luaState,"CreateThread",lua_CreateThread); lua_register(luaState,"SendChatMessage",lua_sendChat); lua_register(luaState,"GetPlayers",lua_GetAllPlayers); + lua_register(luaState,"StopThread",lua_StopThread); lua_register(luaState,"DropPlayer",lua_dropPlayer); lua_register(luaState,"GetPlayerHWID",lua_HWID); lua_register(luaState,"Sleep",lua_Sleep); @@ -293,10 +352,9 @@ void Lua::Init(){ void Lua::Reload(){ if(CheckLua(luaState,luaL_dofile(luaState,FileName.c_str()))){ - CallFunction("onInit",{}); + CallFunction(this,Sec("onInit"),{}); } } -int Handle(EXCEPTION_POINTERS *ep,char* Origin); char* Lua::GetOrigin(){ std::string T = GetFileName().substr(GetFileName().find('\\')); char* Data = new char[T.size()]; @@ -304,26 +362,30 @@ char* Lua::GetOrigin(){ memcpy_s(Data,T.size(),T.c_str(),T.size()); return Data; } -int Lua::CallFunction(const std::string& FuncName,LuaArg* Arg){ +int CallFunction(Lua*lua,const std::string& FuncName,LuaArg* Arg){ + Lock(lua,false); + lua->isExecuting = true; + lua_State*luaState = lua->GetState(); lua_getglobal(luaState, FuncName.c_str()); if(lua_isfunction(luaState, -1)) { int Size = 0; if(Arg != nullptr){ - Size = Arg->args.size(); + Size = int(Arg->args.size()); Arg->PushArgs(luaState); } int R = 0; - char* Origin = GetOrigin(); + char* Origin = lua->GetOrigin(); __try{ R = lua_pcall(luaState, Size, 1, 0); + if (CheckLua(luaState, R)){ + if (lua_isnumber(luaState, -1)) { + return int(lua_tointeger(luaState, -1)); + } + } }__except(Handle(GetExceptionInformation(),Origin)){} delete [] Origin; - if (CheckLua(luaState, R)){ - if (lua_isnumber(luaState, -1)) { - return lua_tointeger(luaState, -1); - } - } } + lua->isExecuting = false; return 0; } void Lua::SetPluginName(const std::string&Name){ diff --git a/src/Network 2.0/ClientInterface.cpp b/src/Network 2.0/ClientInterface.cpp deleted file mode 100644 index 5d8f000..0000000 --- a/src/Network 2.0/ClientInterface.cpp +++ /dev/null @@ -1,85 +0,0 @@ -/// -/// Created by Anonymous275 on 2/4/2020. -/// -#include "Client.hpp" -#include "../logger.h" -#include "../Settings.hpp" -#include "../Lua System/LuaSystem.hpp" -void UDPSend(Client*c,const std::string&Data); -void TCPSend(Client*c,const std::string&Data); - -int OpenID(){ - int ID = 0; - bool found; - do { - found = true; - for (Client *c : Clients) { - if(c->GetID() == ID){ - found = false; - ID++; - } - } - }while (!found); - return ID; -} -void SendLarge(Client*c,const std::string&Data); -void Respond(Client*c, const std::string& MSG, bool Rel){ - char C = MSG.at(0); - if(Rel){ - if(C == 'C' || C == 'O' || C == 'T' || MSG.length() > 1000)SendLarge(c,MSG); - else TCPSend(c,MSG); - }else UDPSend(c,MSG); -} - -void SendToAll(Client*c, const std::string& Data, bool Self, bool Rel){ - char C = Data.at(0); - for(Client*client : Clients){ - if(Self || client != c){ - if(!client->isDownloading){ - if(Rel){ - if(C == 'C' || C == 'O' || C == 'T' || Data.length() > 1000)SendLarge(client,Data); - else TCPSend(client,Data); - } - else UDPSend(client,Data); - } - } - } -} - -void UpdatePlayers(){ - std::string Packet = "Ss" + std::to_string(Clients.size())+"/"+std::to_string(MaxPlayers) + ":"; - for (Client*c : Clients) { - Packet += c->GetName() + ","; - } - Packet = Packet.substr(0,Packet.length()-1); - SendToAll(nullptr, Packet,true,true); -} -int TriggerLuaEvent(const std::string& Event,bool local,Lua*Caller,LuaArg* arg); -void Destroy(Client*c){ - Clients.erase(c); - delete c; -} -void OnDisconnect(Client*c,bool kicked){ - std::string Packet; - for(const std::pair&a : c->GetAllCars()){ - Packet = "Od:" + std::to_string(c->GetID()) + "-" + std::to_string(a.first); - SendToAll(c, Packet,false,true); - } - if(kicked)Packet = "L"+c->GetName()+" was kicked!"; - Packet = "L"+c->GetName()+" Left the server!"; - SendToAll(c, Packet,false,true); - Packet.clear(); - TriggerLuaEvent("onPlayerDisconnect",false,nullptr,new LuaArg{{c->GetID()}}); - Destroy(c); ///Removes the Client from existence -} -void SyncResources(Client*c); -#include -void OnConnect(Client*c){ - c->SetID(OpenID()); - std::cout << "New Client Created! ID : " << c->GetID() << std::endl; - TriggerLuaEvent("onPlayerConnecting",false,nullptr,new LuaArg{{c->GetID()}}); - SyncResources(c); - Respond(c,"M"+MapName,true); //Send the Map on connect - info(c->GetName() + " : Connected"); - TriggerLuaEvent("onPlayerJoining",false,nullptr,new LuaArg{{c->GetID()}}); -} diff --git a/src/Network 2.0/Handler.cpp b/src/Network 2.0/Handler.cpp deleted file mode 100644 index 99ade44..0000000 --- a/src/Network 2.0/Handler.cpp +++ /dev/null @@ -1,58 +0,0 @@ -/// -/// Created by Anonymous275 on 5/9/2020 -/// -///TCP - -#include "Client.hpp" -#include -#include - -void TCPSend(Client*c,const std::string&Data){ - int BytesSent = send(c->GetTCPSock(), Data.c_str(), int(Data.length())+1, 0); - if (BytesSent == 0){ - std::cout << "(TCP) Connection closing..." << std::endl; - if(c->GetStatus() > -1)c->SetStatus(-1); - } - else if (BytesSent < 0) { - std::cout << "(TCP) send failed with error: " << WSAGetLastError() << std::endl; - if(c->GetStatus() > -1)c->SetStatus(-1); - closesocket(c->GetTCPSock()); - } -} - -void GlobalParser(Client*c, const std::string&Packet); -void TCPRcv(Client*c){ - char buf[4096]; - int len = 4096; - ZeroMemory(buf, len); - int BytesRcv = recv(c->GetTCPSock(), buf, len,0); - if (BytesRcv == 0){ - std::cout << "(TCP) Connection closing..." << std::endl; - if(c->GetStatus() > -1)c->SetStatus(-1); - } - else if (BytesRcv < 0) { - std::cout << "(TCP) recv failed with error: " << WSAGetLastError() << std::endl; - if(c->GetStatus() > -1)c->SetStatus(-1); - closesocket(c->GetTCPSock()); - } - GlobalParser(c, std::string(buf)); -} - -void OnConnect(Client*c); -void OnDisconnect(Client*c,bool kicked); -void TCPClient(Client*client){ - if(client->GetTCPSock() == -1)return; - std::cout << "Client connected" << std::endl; - OnConnect(client); - while (client->GetStatus() > -1){ - TCPRcv(client); - } - //OnDisconnect - OnDisconnect(client, client->GetStatus() == -2); - std::cout << "Client Terminated" << std::endl; -} - -void CreateNewThread(Client*client){ - std::thread NewClient(TCPClient,client); - NewClient.detach(); -} diff --git a/src/Network 2.0/NetworkMain.cpp b/src/Network 2.0/NetworkMain.cpp deleted file mode 100644 index 933a012..0000000 --- a/src/Network 2.0/NetworkMain.cpp +++ /dev/null @@ -1,10 +0,0 @@ -#include "Client.hpp" -#include -void TCPServerMain(); -void UDPServerMain(); -std::set Clients; -void NetMain() { - std::thread TCP(TCPServerMain); - TCP.detach(); - UDPServerMain(); -} diff --git a/src/Network 2.0/Security.cpp b/src/Network 2.0/Security.cpp deleted file mode 100644 index cb336c4..0000000 --- a/src/Network 2.0/Security.cpp +++ /dev/null @@ -1,57 +0,0 @@ -/// -/// Created by Anonymous275 on 7/9/2020 -/// -#include -#include -#include "Client.hpp" -#include "../logger.h" -#include "../Settings.hpp" -#include -void VehicleParser(Client*c,const std::string& Pckt); -int Handle(EXCEPTION_POINTERS *ep,char* Origin){ - Exception(ep->ExceptionRecord->ExceptionCode,Origin); - return 1; -} -int Rand(){ - std::random_device r; - std::default_random_engine e1(r()); - std::uniform_int_distribution uniform_dist(1, 200); - return uniform_dist(e1); -} - -std::string Encrypt(std::string msg){ - if(msg.size() < 2)return msg; - int R = (Rand()+Rand())/2,T = R; - for(char&c : msg){ - if(R > 30)c = char(int(c) + (R-=3)); - else c = char(int(c) - (R+=4)); - } - return char(T) + msg; -} - -std::string Decrypt(std::string msg){ - int R = uint8_t(msg.at(0)); - if(msg.size() < 2 || R > 200 || R < 1)return ""; - msg = msg.substr(1); - for(char&c : msg){ - if(R > 30)c = char(int(c) - (R-=3)); - else c = char(int(c) + (R+=4)); - } - return msg; -} -[[noreturn]]void DLoop(){ - while(true) { - if(IsDebuggerPresent())VehicleParser(nullptr, ""); - std::this_thread::sleep_for(std::chrono::seconds(1)); - } -} -[[noreturn]]void SLoop(){ - std::thread D(DLoop); - D.detach(); - int A = -1; - while(true) { - std::this_thread::sleep_for(std::chrono::seconds(20)); - if (A == Beat)VehicleParser(nullptr, ""); - A = Beat; - } -} \ No newline at end of file diff --git a/src/Network 2.0/StatMonitor.cpp b/src/Network 2.0/StatMonitor.cpp deleted file mode 100644 index 92b3dfb..0000000 --- a/src/Network 2.0/StatMonitor.cpp +++ /dev/null @@ -1,38 +0,0 @@ -/// -/// Created by Anonymous275 on 6/18/2020 -/// -#include "Client.hpp" -#include -#include -#include -std::string StatReport = "-"; -int PPS = 0; -[[noreturn]] void Monitor(){ - int R,C,V; - while(true){ - if(Clients.empty()){ - StatReport = "-"; - }else{ - C = 0;V = 0; - for(Client *c : Clients){ - if(c->GetCarCount() > 0){ - C++; - V += c->GetCarCount(); - } - } - if(C == 0 || PPS == 0){ - StatReport = "-"; - }else{ - R = (PPS/C)/V; - StatReport = std::to_string(R); - } - PPS = 0; - } - std::this_thread::sleep_for(std::chrono::seconds(1)); - } -} - -void StatInit(){ - std::thread Init(Monitor); - Init.detach(); -} diff --git a/src/Network 2.0/VehicleEvent.cpp b/src/Network 2.0/VehicleEvent.cpp deleted file mode 100644 index b200bcd..0000000 --- a/src/Network 2.0/VehicleEvent.cpp +++ /dev/null @@ -1,168 +0,0 @@ -/// -/// Created by Anonymous275 on 5/9/2020 -/// -///TCP - -#include "Client.hpp" -#include -#include -#include "../logger.h" -#include "../Settings.hpp" -#include -std::string HTTP_REQUEST(const std::string& IP,int port); -std::string HTA(const std::string& hex); -std::string Decrypt(std::string); -struct Sequence{ - SOCKET TCPSock; - bool Done = false; -}; -void CreateNewThread(Client*client); -void CreateClient(SOCKET TCPSock,const std::string &Name, const std::string &DID,const std::string &Role) { - auto *client = new Client; - client->SetTCPSock(TCPSock); - client->SetName(Name); - client->SetRole(Role); - client->SetDID(DID); - Clients.insert(client); - CreateNewThread(client); -} -std::string TCPRcv(SOCKET TCPSock){ - char buf[4096]; - int len = 4096; - ZeroMemory(buf, len); - int BytesRcv = recv(TCPSock, buf, len,0); - if (BytesRcv == 0){ - return ""; - } - else if (BytesRcv < 0) { - closesocket(TCPSock); - return ""; - } - return std::string(buf); -} -std::string HTTP(const std::string &DID){ - if(!DID.empty()){ - std::string a = HTTP_REQUEST("https://beamng-mp.com/entitlement?did="+DID,443); - if(!a.empty()){ - int pos = a.find('"'); - if(pos != std::string::npos){ - return a.substr(pos+1,a.find('"',pos+1)-2); - }else if(a == "[]")return "Member"; - } - } - return ""; -} -void Check(Sequence* S){ - std::this_thread::sleep_for(std::chrono::seconds(5)); - if(S != nullptr){ - if(!S->Done)closesocket(S->TCPSock); - delete S; - } -} -int Max(){ - int M = MaxPlayers; - for(Client*c : Clients){ - if(c->GetRole() == "MDEV")M++; - } - return M; -} - -void Identification(SOCKET TCPSock){ - auto* S = new Sequence; - S->TCPSock = TCPSock; - std::thread Timeout(Check,S); - Timeout.detach(); - std::string Name,DID,Role,Res = TCPRcv(TCPSock),Ver = TCPRcv(TCPSock); - S->Done = true; - Ver = Decrypt(Ver); - if(Ver.size() > 3 && Ver.substr(0,2) == "VC"){ - Ver = Ver.substr(2); - if(Ver.length() > 4 || Ver != ClientVersion){ - closesocket(TCPSock); - return; - } - }else{ - closesocket(TCPSock); - return; - } - Res = Decrypt(Res); - if(Res.size() < 3 || Res.substr(0,2) != "NR") { - closesocket(TCPSock); - return; - } - if(Res.find(':') == std::string::npos){ - closesocket(TCPSock); - return; - } - Name = Res.substr(2,Res.find(':')-2); - DID = Res.substr(Res.find(':')+1); - Role = HTTP(DID); - if(Role.empty() || Role.find("Error") != std::string::npos){ - closesocket(TCPSock); - return; - } - - for(Client*c: Clients){ - if(c->GetDID() == DID){ - closesocket(c->GetTCPSock()); - c->SetStatus(-2); - break; - } - } - - if(Debug)debug("Name -> " + Name + ", Role -> " + Role + ", ID -> " + DID); - if(Role == "MDEV"){ - CreateClient(TCPSock,Name,DID,Role); - return; - } - if(Clients.size() < Max()){ - CreateClient(TCPSock,Name,DID,Role); - }else closesocket(TCPSock); -} - -void TCPServerMain(){ - WSADATA wsaData; - if (WSAStartup(514, &wsaData)) //2.2 - { - std::cout << "Can't start Winsock!" << std::endl; - return; - } - SOCKET client, Listener = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); - sockaddr_in addr{}; - addr.sin_addr.S_un.S_addr = ADDR_ANY; - addr.sin_family = AF_INET; - addr.sin_port = htons(Port); - - - if (bind(Listener, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR) - { - std::cout << "Can't bind socket! " << WSAGetLastError() << std::endl; - return; - } - - if(Listener == -1) - { - printf("Invalid socket"); - return; - } - - if(listen(Listener,SOMAXCONN)) - { - std::cout << "listener failed " << GetLastError(); - return; - } - info("Vehicle event network online"); - do{ - client = accept(Listener, nullptr, nullptr); - if(client == -1) - { - std::cout << "invalid client socket" << std::endl; - continue; - } - std::thread ID(Identification,client); - ID.detach(); - }while(client); - - closesocket(client); - WSACleanup(); -} \ No newline at end of file diff --git a/src/Network/Auth.cpp b/src/Network/Auth.cpp new file mode 100644 index 0000000..9bd4e5b --- /dev/null +++ b/src/Network/Auth.cpp @@ -0,0 +1,172 @@ +/// +/// Created by Anonymous275 on 7/31/2020 +/// +#include "Security/Enc.h" +#include "Curl/Http.h" +#include "Settings.h" +#include "Network.h" +#include "Logger.h" +#include +#include +struct Hold{ + SOCKET TCPSock{}; + bool Done = false; +}; +bool Send(SOCKET TCPSock,std::string Data){ + int BytesSent; + BytesSent = send(TCPSock, Data.c_str(), int(Data.size()), 0); + Data.clear(); + if (BytesSent <= 0)return false; + return true; +} +std::string Rcv(SOCKET TCPSock){ + char buf[6768]; + int len = 6768; + ZeroMemory(buf, len); + int BytesRcv = recv(TCPSock, buf, len,0); + if (BytesRcv <= 0)return ""; + return std::string(buf); +} +std::string GetRole(const std::string &DID){ + if(!DID.empty()){ + std::string a = HttpRequest(Sec("https://beamng-mp.com/entitlement?did=")+DID,443); + if(!a.empty()){ + auto pos = a.find('"'); + if(pos != std::string::npos){ + return a.substr(pos+1,a.find('"',pos+1)-2); + }else if(a == "[]")return Sec("Member"); + } + } + return ""; +} +void Check(Hold* S){ + std::this_thread::sleep_for(std::chrono::seconds(5)); + if(S != nullptr){ + if(!S->Done)closesocket(S->TCPSock); + } +} +int Max(){ + int M = MaxPlayers; + for(Client*c : CI->Clients){ + if(c != nullptr){ + if(c->GetRole() == Sec("MDEV"))M++; + } + } + return M; +} +void CreateClient(SOCKET TCPSock,const std::string &Name, const std::string &DID,const std::string &Role) { + auto *c = new Client; + c->SetTCPSock(TCPSock); + c->SetName(Name); + c->SetRole(Role); + c->SetDID(DID); + CI->AddClient(c); + InitClient(c); +} + +std::string GenerateM(RSA*key){ + std::stringstream stream; + stream << std::hex << key->n << "g" << key->e << "g" << RSA_E(Sec("IDC"),key); + return stream.str(); +} + +void Identification(SOCKET TCPSock,Hold*S,RSA*key){ + S->TCPSock = TCPSock; + std::thread Timeout(Check,S); + Timeout.detach(); + std::string Name,DID,Role; + if(!Send(TCPSock,GenerateM(key))){ + closesocket(TCPSock); + return; + } + std::string Res = Rcv(TCPSock); + std::string Ver = Rcv(TCPSock); + S->Done = true; + Ver = RSA_D(Ver,key); + if(Ver.size() > 3 && Ver.substr(0,2) == Sec("VC")){ + Ver = Ver.substr(2); + if(Ver.length() > 4 || Ver != GetCVer()){ + closesocket(TCPSock); + return; + } + }else{ + closesocket(TCPSock); + return; + } + Res = RSA_D(Res,key); + if(Res.size() < 3 || Res.substr(0,2) != Sec("NR")) { + closesocket(TCPSock); + return; + } + if(Res.find(':') == std::string::npos){ + closesocket(TCPSock); + return; + } + Name = Res.substr(2,Res.find(':')-2); + DID = Res.substr(Res.find(':')+1); + Role = GetRole(DID); + if(Role.empty() || Role.find(Sec("Error")) != -1){ + closesocket(TCPSock); + return; + } + debug(Sec("Name -> ") + Name + Sec(", Role -> ") + Role + Sec(", ID -> ") + DID); + for(Client*c: CI->Clients){ + if(c != nullptr){ + if(c->GetDID() == DID){ + closesocket(c->GetTCPSock()); + c->SetStatus(-2); + break; + } + } + } + if(Role == Sec("MDEV") || CI->Size() < Max()){ + CreateClient(TCPSock,Name,DID,Role); + }else closesocket(TCPSock); +} +void Identify(SOCKET TCPSock){ + auto* S = new Hold; + RSA*key = GenKey(); + __try{ + Identification(TCPSock,S,key); + }__except(1){} + delete key; + delete S; +} +void TCPServerMain(){ + WSADATA wsaData; + if (WSAStartup(514, &wsaData)){ + error(Sec("Can't start Winsock!")); + return; + } + SOCKET client, Listener = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); + sockaddr_in addr{}; + addr.sin_addr.S_un.S_addr = ADDR_ANY; + addr.sin_family = AF_INET; + addr.sin_port = htons(Port); + if (bind(Listener, (sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR){ + error(Sec("Can't bind socket! ") + std::to_string(WSAGetLastError())); + std::this_thread::sleep_for(std::chrono::seconds(5)); + exit(-1); + } + if(Listener == -1){ + error(Sec("Invalid listening socket")); + return; + } + if(listen(Listener,SOMAXCONN)){ + error(Sec("listener failed ")+ std::to_string(GetLastError())); + return; + } + info(Sec("Vehicle event network online")); + do{ + client = accept(Listener, nullptr, nullptr); + if(client == -1){ + warn(Sec("Got an invalid client socket on connect! Skipping...")); + continue; + } + std::thread ID(Identify,client); + ID.detach(); + }while(client); + + closesocket(client); + WSACleanup(); +} \ No newline at end of file diff --git a/src/Network 2.0/Client.cpp b/src/Network/Client.cpp similarity index 65% rename from src/Network 2.0/Client.cpp rename to src/Network/Client.cpp index 0774f19..8c7959b 100644 --- a/src/Network 2.0/Client.cpp +++ b/src/Network/Client.cpp @@ -12,12 +12,6 @@ void Client::SetName(const std::string& name){ void Client::SetDID(const std::string& did){ DID = did; } -void Client::SetConnected(bool state){ - Connected = state; -} -bool Client::isConnected(){ - return Connected; -} std::string Client::GetDID(){ return DID; } @@ -52,20 +46,31 @@ SOCKET Client::GetTCPSock(){ return TCPSOCK; } void Client::DeleteCar(int ident){ - for(const std::pair& a : VehicleData){ - if(a.first == ident){ - VehicleData.erase(a); + for(VData* v : VehicleData){ + if(v != nullptr && v->ID == ident){ + VehicleData.erase(v); + delete v; + v = nullptr; break; } } } +void Client::ClearCars(){ + for(VData* v : VehicleData){ + if(v != nullptr){ + delete v; + v = nullptr; + } + } + VehicleData.clear(); +} int Client::GetOpenCarID(){ int OpenID = 0; bool found; do { found = true; - for (const std::pair &a : VehicleData) { - if (a.first == OpenID){ + for (VData*v : VehicleData) { + if (v != nullptr && v->ID == OpenID){ OpenID++; found = false; } @@ -74,31 +79,31 @@ int Client::GetOpenCarID(){ return OpenID; } void Client::AddNewCar(int ident,const std::string& Data){ - VehicleData.insert(std::make_pair(ident,Data)); + VehicleData.insert(new VData{ident,Data}); } -std::set> Client::GetAllCars(){ +std::set Client::GetAllCars(){ return VehicleData; } std::string Client::GetCarData(int ident){ - for(const std::pair& a : VehicleData){ - if(a.first == ident){ - return a.second; + for(VData*v : VehicleData){ + if(v != nullptr && v->ID == ident){ + return v->Data; } } DeleteCar(ident); return ""; } void Client::SetCarData(int ident,const std::string&Data){ - for(const std::pair& a : VehicleData){ - if(a.first == ident){ - VehicleData.erase(a); - VehicleData.insert(std::make_pair(ident,Data)); + for(VData*v : VehicleData){ + if(v != nullptr && v->ID == ident){ + v->Data = Data; return; } } DeleteCar(ident); } int Client::GetCarCount(){ - return VehicleData.size(); -} \ No newline at end of file + return int(VehicleData.size()); +} + diff --git a/src/Network 2.0/DataParser.cpp b/src/Network/GParser.cpp similarity index 57% rename from src/Network 2.0/DataParser.cpp rename to src/Network/GParser.cpp index b6cf336..eba2f9a 100644 --- a/src/Network 2.0/DataParser.cpp +++ b/src/Network/GParser.cpp @@ -1,26 +1,22 @@ /// -/// Created by Anonymous275 on 4/2/2020 +/// Created by Anonymous275 on 8/1/2020 /// -#include +#include "Lua/LuaSystem.hpp" +#include "Security/Enc.h" #include "Client.hpp" -#include "../logger.h" -#include "../Settings.hpp" -#include "../Lua System/LuaSystem.hpp" - -void SendToAll(Client*c, const std::string& Data, bool Self, bool Rel); -void Respond(Client*c, const std::string& MSG, bool Rel); -void UpdatePlayers(); -int TriggerLuaEvent(const std::string& Event,bool local,Lua*Caller,LuaArg* arg); +#include "Settings.h" +#include "Network.h" +#include "Logger.h" int FC(const std::string& s,const std::string& p,int n) { - std::string::size_type i = s.find(p); + auto i = s.find(p); int j; - for (j = 1; j < n && i != std::string::npos; ++j)i = s.find(p, i+1); - if (j == n)return(i); - else return(-1); + for (j = 1; j < n && i != std::string::npos; ++j){ + i = s.find(p, i+1); + } + if (j == n)return int(i); + else return -1; } -int Handle(EXCEPTION_POINTERS *ep,char* Origin); - void Apply(Client*c,int VID,const std::string& pckt){ std::string Packet = pckt; std::string VD = c->GetCarData(VID); @@ -30,12 +26,9 @@ void Apply(Client*c,int VID,const std::string& pckt){ VD.substr(FC(VD, ",\"", 7)); c->SetCarData(VID, Packet); } -void UpdateCarData(Client*c,int VID,const std::string& Packet){ - __try{ - Apply(c,VID,Packet); - }__except(Handle(GetExceptionInformation(),(char*)"Car Data Updater")){} -} + void VehicleParser(Client*c,const std::string& Pckt){ + if(c == nullptr || Pckt.length() < 4)return; std::string Packet = Pckt; char Code = Packet.at(1); int PID = -1; @@ -45,20 +38,21 @@ void VehicleParser(Client*c,const std::string& Pckt){ case 's': if(Data.at(0) == '0'){ int CarID = c->GetOpenCarID(); - std::cout << c->GetName() << " CarID : " << CarID << std::endl; + debug(c->GetName() + Sec(" created a car with ID ") + std::to_string(CarID)); Packet = "Os:"+c->GetRole()+":"+c->GetName()+":"+std::to_string(c->GetID())+"-"+std::to_string(CarID)+Packet.substr(4); if(c->GetCarCount() >= MaxCars || - TriggerLuaEvent("onVehicleSpawn",false,nullptr, + TriggerLuaEvent(Sec("onVehicleSpawn"),false,nullptr, new LuaArg{{c->GetID(),CarID,Packet.substr(3)}})){ Respond(c,Packet,true); std::string Destroy = "Od:" + std::to_string(c->GetID())+"-"+std::to_string(CarID); Respond(c,Destroy,true); + debug(c->GetName() + Sec(" (force : car limit/lua) removed ID ") + std::to_string(CarID)); }else{ c->AddNewCar(CarID,Packet); SendToAll(nullptr, Packet,true,true); } } - break; + return; case 'c': pid = Data.substr(0,Data.find('-')); vid = Data.substr(Data.find('-')+1,Data.find(':',1)-Data.find('-')-1); @@ -67,103 +61,111 @@ void VehicleParser(Client*c,const std::string& Pckt){ VID = stoi(vid); } if(PID != -1 && VID != -1 && PID == c->GetID()){ - if(!TriggerLuaEvent("onVehicleEdited",false,nullptr, - new LuaArg{{c->GetID(),VID,Packet.substr(3)}})) { + if(!TriggerLuaEvent(Sec("onVehicleEdited"),false,nullptr, + new LuaArg{{c->GetID(),VID,Packet.substr(3)}})) { SendToAll(c, Packet, false, true); - UpdateCarData(c,VID,Packet); + Apply(c,VID,Packet); }else{ std::string Destroy = "Od:" + std::to_string(c->GetID())+"-"+std::to_string(VID); Respond(c,Destroy,true); c->DeleteCar(VID); } - } - break; + return; case 'd': pid = Data.substr(0,Data.find('-')); vid = Data.substr(Data.find('-')+1); if(pid.find_first_not_of("0123456789") == std::string::npos && vid.find_first_not_of("0123456789") == std::string::npos){ - PID = stoi(pid); - VID = stoi(vid); + PID = stoi(pid); + VID = stoi(vid); } if(PID != -1 && VID != -1 && PID == c->GetID()){ SendToAll(nullptr,Packet,true,true); - TriggerLuaEvent("onVehicleDeleted",false,nullptr, + TriggerLuaEvent(Sec("onVehicleDeleted"),false,nullptr, new LuaArg{{c->GetID(),VID}}); c->DeleteCar(VID); + debug(c->GetName() + Sec(" deleted car with ID ") + std::to_string(VID)); } - break; + return; case 'r': SendToAll(c,Packet,false,true); - break; - //case 'm': - // break; + return; default: - break; + return; } - Data.clear(); - Packet.clear(); } -void SyncVehicles(Client*c){ - Respond(c,"Sn"+c->GetName(),true); - SendToAll(c,"JWelcome "+c->GetName()+"!",false,true); - TriggerLuaEvent("onPlayerJoin",false,nullptr,new LuaArg{{c->GetID()}}); - for (Client*client : Clients) { - if (client != c) { - for(const std::pair&a : client->GetAllCars()){ - Respond(c,a.second,true); +void SyncClient(Client*c){ + if(c->isSynced)return; + Respond(c,Sec("Sn")+c->GetName(),true); + SendToAll(c,Sec("JWelcome ")+c->GetName()+"!",false,true); + TriggerLuaEvent(Sec("onPlayerJoin"),false,nullptr,new LuaArg{{c->GetID()}}); + for (Client*client : CI->Clients) { + if(client != nullptr){ + if (client != c) { + for (VData *v : client->GetAllCars()) { + Respond(c, v->Data, true); + } } } } + c->isSynced = true; + info(c->GetName() + Sec(" is now synced!")); } - -extern int PPS; - void ParseVeh(Client*c, const std::string&Packet){ __try{ VehicleParser(c,Packet); - }__except(Handle(GetExceptionInformation(),(char*)"Vehicle Handler")){} + }__except(Handle(GetExceptionInformation(),Sec("Vehicle Handler"))){} } -void GlobalParser(Client*c, const std::string&Packet){ - if(Packet.empty())return; - if(Packet.find("TEST")!=std::string::npos)SyncVehicles(c); +void GlobalParser(Client*c, const std::string& Packet){ + static int lastRecv = 0; + if(Packet.empty() || c == nullptr)return; std::string pct; char Code = Packet.at(0); + + //V to Z + if(Code <= 90 && Code >= 86){ + PPS++; + SendToAll(c,Packet,false,false); + return; + } + switch (Code) { case 'P': - Respond(c, "P" + std::to_string(c->GetID()),true); + Respond(c, Sec("P") + std::to_string(c->GetID()),true); + SyncClient(c); return; case 'p': - Respond(c,"p",false); + Respond(c,Sec("p"),false); UpdatePlayers(); return; case 'O': if(Packet.length() > 1000) { - std::cout << "Received data from: " << c->GetName() << " Size: " << Packet.length() << std::endl; + debug(Sec("Received data from: ") + c->GetName() + Sec(" Size: ") + std::to_string(Packet.length())); } ParseVeh(c,Packet); return; case 'J': SendToAll(c,Packet,false,true); - break; + return; case 'C': + if(Packet.length() < 4 || Packet.find(':', 3) == -1)break; pct = "C:" + c->GetName() + Packet.substr(Packet.find(':', 3)); - if (TriggerLuaEvent("onChatMessage", false, nullptr, - new LuaArg{{ c->GetID(), c->GetName(), pct.substr(pct.find(':', 3) + 1) }}))break; + if (TriggerLuaEvent(Sec("onChatMessage"), false, nullptr, + new LuaArg{{c->GetID(), c->GetName(), pct.substr(pct.find(':', 3) + 1)}}))break; SendToAll(nullptr, pct, true, true); pct.clear(); - break; + return; case 'E': SendToAll(nullptr,Packet,true,true); - break; + return; default: - break; + return; } - //V to Z - if(Code <= 90 && Code >= 86){ - PPS++; - SendToAll(c,Packet,false,false); - } - if(Debug)debug("Vehicle Data Received from " + c->GetName()); +} + +void GParser(Client*c, const std::string&Packet){ + __try{ + GlobalParser(c, Packet); + }__except(Handle(GetExceptionInformation(),Sec("Global Handler"))){} } \ No newline at end of file diff --git a/src/http.cpp b/src/Network/Http.cpp similarity index 88% rename from src/http.cpp rename to src/Network/Http.cpp index 2b9b5fc..e95e5c7 100644 --- a/src/http.cpp +++ b/src/Network/Http.cpp @@ -1,18 +1,14 @@ /// /// Created by Anonymous275 on 4/9/2020 /// - #define CURL_STATICLIB - -#include "curl/curl.h" +#include "Curl/curl.h" #include - -static size_t WriteCallback(void *contents, size_t size, size_t nmemb, void *userp) -{ +static size_t WriteCallback(void *contents, size_t size, size_t nmemb, void *userp){ ((std::string*)userp)->append((char*)contents, size * nmemb); return size * nmemb; } -std::string HTTP_REQUEST(const std::string& IP,int port){ +std::string HttpRequest(const std::string& IP,int port){ CURL *curl; CURLcode res; std::string readBuffer; @@ -24,6 +20,7 @@ std::string HTTP_REQUEST(const std::string& IP,int port){ curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer); res = curl_easy_perform(curl); curl_easy_cleanup(curl); + if(res != CURLE_OK)return "-1"; } return readBuffer; } @@ -32,7 +29,6 @@ std::string PostHTTP(const std::string& IP,const std::string& Fields){ CURL *curl; CURLcode res; std::string readBuffer; - curl = curl_easy_init(); if(curl) { curl_easy_setopt(curl, CURLOPT_URL, IP.c_str()); @@ -43,6 +39,7 @@ std::string PostHTTP(const std::string& IP,const std::string& Fields){ curl_easy_setopt(curl, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4); res = curl_easy_perform(curl); curl_easy_cleanup(curl); + if(res != CURLE_OK)return "-1"; } return readBuffer; } \ No newline at end of file diff --git a/src/Network/InitClient.cpp b/src/Network/InitClient.cpp new file mode 100644 index 0000000..b37c1fe --- /dev/null +++ b/src/Network/InitClient.cpp @@ -0,0 +1,82 @@ +/// +/// Created by Anonymous275 on 8/1/2020 +/// +#include "Lua/LuaSystem.hpp" +#include "Security/Enc.h" +#include "Client.hpp" +#include "Settings.h" +#include "Network.h" +#include "Logger.h" +int OpenID(){ + int ID = 0; + bool found; + do { + found = true; + for (Client *c : CI->Clients){ + if(c != nullptr){ + if(c->GetID() == ID){ + found = false; + ID++; + } + } + } + }while (!found); + return ID; +} +void Respond(Client*c, const std::string& MSG, bool Rel){ + char C = MSG.at(0); + if(Rel){ + if(C == 'O' || C == 'T' || MSG.length() > 1000)SendLarge(c,MSG); + else TCPSend(c,MSG); + }else UDPSend(c,MSG); +} +void SendToAll(Client*c, const std::string& Data, bool Self, bool Rel){ + char C = Data.at(0); + for(Client*client : CI->Clients){ + if(client != nullptr) { + if (Self || client != c) { + if (client->isSynced) { + if (Rel) { + if (C == 'O' || C == 'T' || Data.length() > 1000)SendLarge(client, Data); + else TCPSend(client, Data); + } else UDPSend(client, Data); + } + } + } + } +} +void UpdatePlayers(){ + std::string Packet = Sec("Ss") + std::to_string(CI->Size())+"/"+std::to_string(MaxPlayers) + ":"; + for (Client*c : CI->Clients) { + if(c != nullptr)Packet += c->GetName() + ","; + } + Packet = Packet.substr(0,Packet.length()-1); + SendToAll(nullptr, Packet,true,true); +} +void OnDisconnect(Client*c,bool kicked){ + if(c == nullptr)return; + std::string Packet; + for(VData*v : c->GetAllCars()){ + if(v != nullptr) { + Packet = "Od:" + std::to_string(c->GetID()) + "-" + std::to_string(v->ID); + SendToAll(c, Packet, false, true); + } + } + if(kicked)Packet = Sec("L")+c->GetName()+Sec(" was kicked!"); + Packet = Sec("L")+c->GetName()+Sec(" Left the server!"); + SendToAll(c, Packet,false,true); + Packet.clear(); + TriggerLuaEvent(Sec("onPlayerDisconnect"),false,nullptr,new LuaArg{{c->GetID()}}); + c->ClearCars(); + CI->RemoveClient(c); ///Removes the Client from existence +} +void OnConnect(Client*c){ + c->SetID(OpenID()); + info(Sec("Assigned ID ") + std::to_string(c->GetID()) + Sec(" to ") + c->GetName()); + TriggerLuaEvent(Sec("onPlayerConnecting"),false,nullptr,new LuaArg{{c->GetID()}}); + SyncResources(c); + if(c->GetStatus() < 0)return; + Respond(c,"M"+MapName,true); //Send the Map on connect + info(c->GetName() + Sec(" : Connected")); + TriggerLuaEvent(Sec("onPlayerJoining"),false,nullptr,new LuaArg{{c->GetID()}}); +} \ No newline at end of file diff --git a/src/Network/NetMain.cpp b/src/Network/NetMain.cpp new file mode 100644 index 0000000..8df428c --- /dev/null +++ b/src/Network/NetMain.cpp @@ -0,0 +1,8 @@ +#include "Network.h" +#include +ClientInterface* CI; +void NetMain(){ + std::thread TCP(TCPServerMain); + TCP.detach(); + UDPServerMain(); +} diff --git a/src/Network/StatMonitor.cpp b/src/Network/StatMonitor.cpp new file mode 100644 index 0000000..b555547 --- /dev/null +++ b/src/Network/StatMonitor.cpp @@ -0,0 +1,43 @@ +/// +/// Created by Anonymous275 on 6/18/2020 +/// +#include "Security/Enc.h" +#include "Client.hpp" +#include +#include +#include +std::string StatReport; +int PPS = 0; +void Monitor() { + int R, C = 0, V = 0; + if (CI->Clients.empty()){ + StatReport = "-"; + return; + } + for (Client *c : CI->Clients) { + if (c != nullptr && c->GetCarCount() > 0) { + C++; + V += c->GetCarCount(); + } + } + if (C == 0 || PPS == 0) { + StatReport = "-"; + } else { + R = (PPS / C) / V; + StatReport = std::to_string(R); + } + PPS = 0; +} + +[[noreturn]]void Stat(){ + while(true){ + Monitor(); + std::this_thread::sleep_for(std::chrono::seconds(1)); + } +} + +void StatInit(){ + StatReport = "-"; + std::thread Init(Stat); + Init.detach(); +} diff --git a/src/Network 2.0/ResourceSync.cpp b/src/Network/Sync.cpp similarity index 50% rename from src/Network 2.0/ResourceSync.cpp rename to src/Network/Sync.cpp index e0dbfa6..0431d78 100644 --- a/src/Network 2.0/ResourceSync.cpp +++ b/src/Network/Sync.cpp @@ -1,70 +1,57 @@ /// -/// Created by Anonymous275 on 6/10/2020 +/// Created by Anonymous275 on 8/1/2020 /// -#include -#include "../Settings.hpp" +#include "Security/Enc.h" #include "Client.hpp" -#include +#include "Settings.h" +#include "Logger.h" #include -#include -std::string Encrypt(std::string msg); -void STCPSend(Client*c,std::any Data,size_t Size){ + +void STCPSend(Client*c,std::string Data){ + if(c == nullptr)return; int BytesSent; - if(std::string(Data.type().name()).find("string") != std::string::npos){ - auto data = std::any_cast(Data); - BytesSent = send(c->GetTCPSock(), data.c_str(), data.size(), 0); - data.clear(); - }else{ - char* D = std::any_cast(Data); - BytesSent = send(c->GetTCPSock(), D, Size, 0); - delete[] D; - } + BytesSent = send(c->GetTCPSock(), Data.c_str(), int(Data.size()), 0); + Data.clear(); if (BytesSent == 0){ - std::cout << "(STCPS) Connection closing..." << std::endl; if(c->GetStatus() > -1)c->SetStatus(-1); - } - else if (BytesSent < 0) { - std::cout << "(STCPS) send failed with error: " << WSAGetLastError() << std::endl; + }else if (BytesSent < 0) { if(c->GetStatus() > -1)c->SetStatus(-1); closesocket(c->GetTCPSock()); } } void SendFile(Client*c,const std::string&Name){ - std::cout << c->GetName() << " requesting : " - << Name.substr(Name.find_last_of('/')) << std::endl; + info(c->GetName()+Sec(" requesting : ")+Name.substr(Name.find_last_of('/'))); struct stat Info{}; if(stat(Name.c_str(), &Info) != 0){ - STCPSend(c,std::string("Cannot Open"),0); + STCPSend(c,Sec("Cannot Open")); return; } std::ifstream f(Name.c_str(), std::ios::binary); f.seekg(0, std::ios_base::end); std::streampos fileSize = f.tellg(); size_t Size = fileSize,Sent = 0,Diff; - char* Data; int Split = 64000; while(c->GetStatus() > -1 && Sent < Size){ Diff = Size - Sent; if(Diff > Split){ - Data = new char[Split]; + std::string Data(Split,0); f.seekg(Sent, std::ios_base::beg); - f.read(Data, Split); - STCPSend(c,Data,Split); + f.read(&Data[0], Split); + STCPSend(c,Data); Sent += Split; }else{ - Data = new char[Diff]; + std::string Data(Diff,0); f.seekg(Sent, std::ios_base::beg); - f.read(Data, Diff); - STCPSend(c,Data,Diff); + f.read(&Data[0], Diff); + STCPSend(c,Data); Sent += Diff; } } f.close(); } -void Parse(Client*c,char*data){ - std::string Packet = data; - if(Packet.empty())return; +void Parse(Client*c,const std::string&Packet){ + if(c == nullptr || Packet.empty())return; char Code = Packet.at(0),SubCode = 0; if(Packet.length() > 1)SubCode = Packet.at(1); switch (Code) { @@ -73,42 +60,44 @@ void Parse(Client*c,char*data){ return; case 'S': if(SubCode == 'R'){ - std::cout << "Sending File Info" << std::endl; + debug(Sec("Sending Mod Info")); std::string ToSend = FileList+FileSizes; if(ToSend.empty())ToSend = "-"; - STCPSend(c,ToSend,0); + STCPSend(c,ToSend); } return; + default: + return; } } bool STCPRecv(Client*c){ + if(c == nullptr)return false; char buf[200]; int len = 200; ZeroMemory(buf, len); int BytesRcv = recv(c->GetTCPSock(), buf, len,0); if (BytesRcv == 0){ - std::cout << "(STCPR) Connection closing..." << std::endl; if(c->GetStatus() > -1)c->SetStatus(-1); closesocket(c->GetTCPSock()); return false; - } - else if (BytesRcv < 0) { - std::cout << "(STCPR) recv failed with error: " << WSAGetLastError() << std::endl; + }else if (BytesRcv < 0) { if(c->GetStatus() > -1)c->SetStatus(-1); closesocket(c->GetTCPSock()); return false; } if(strcmp(buf,"Done") == 0)return false; - char* Ret = new char[BytesRcv]; - memcpy_s(Ret,BytesRcv,buf,BytesRcv); - ZeroMemory(buf, len); + std::string Ret(buf,BytesRcv); Parse(c,Ret); - delete[] Ret; return true; } void SyncResources(Client*c){ - STCPSend(c,Encrypt("WS"),0); - while(c->GetStatus() > -1 && STCPRecv(c)); - c->isDownloading = false; + if(c == nullptr)return; + try{ + STCPSend(c,Sec("WS")); + while(c->GetStatus() > -1 && STCPRecv(c)); + }catch (std::exception& e){ + except(Sec("Exception! : ") + std::string(e.what())); + c->SetStatus(-1); + } } \ No newline at end of file diff --git a/src/Network/TCPHandler.cpp b/src/Network/TCPHandler.cpp new file mode 100644 index 0000000..6a9edcf --- /dev/null +++ b/src/Network/TCPHandler.cpp @@ -0,0 +1,52 @@ +/// +/// Created by Anonymous275 on 8/1/2020 +/// +#include "Security/Enc.h" +#include "Network.h" +#include "Logger.h" +#include + +void TCPSend(Client*c,const std::string&Data){ + if(c == nullptr)return; + int BytesSent = send(c->GetTCPSock(), Data.c_str(), int(Data.length())+1, 0); + if (BytesSent == 0){ + if(c->GetStatus() > -1)c->SetStatus(-1); + }else if (BytesSent < 0) { + if(c->GetStatus() > -1)c->SetStatus(-1); + closesocket(c->GetTCPSock()); + } +} +void TCPRcv(Client*c){ + if(c == nullptr)return; + char buf[4096]; + int len = 4096; + ZeroMemory(buf, len); + int BytesRcv = recv(c->GetTCPSock(), buf, len,0); + if (BytesRcv == 0){ + debug(Sec("(TCP) Connection closing...")); + if(c->GetStatus() > -1)c->SetStatus(-1); + return; + }else if (BytesRcv < 0) { + debug(Sec("(TCP) recv failed with error: ") + std::to_string(WSAGetLastError())); + if(c->GetStatus() > -1)c->SetStatus(-1); + closesocket(c->GetTCPSock()); + return; + } + std::string Buf(buf,BytesRcv); + GParser(c, Buf); +} +void TCPClient(Client*c){ + if(c->GetTCPSock() == -1){ + CI->RemoveClient(c); + return; + } + info(Sec("Client connected")); + OnConnect(c); + while (c->GetStatus() > -1)TCPRcv(c); + info(c->GetName() + Sec(" Connection Terminated")); + OnDisconnect(c, c->GetStatus() == -2); +} +void InitClient(Client*c){ + std::thread NewClient(TCPClient,c); + NewClient.detach(); +} \ No newline at end of file diff --git a/src/Network 2.0/VehicleData.cpp b/src/Network/VehicleData.cpp similarity index 64% rename from src/Network 2.0/VehicleData.cpp rename to src/Network/VehicleData.cpp index ad45868..a6c4823 100644 --- a/src/Network 2.0/VehicleData.cpp +++ b/src/Network/VehicleData.cpp @@ -2,13 +2,14 @@ /// Created by Anonymous275 on 5/8/2020 /// ///UDP - +#include "Security/Enc.h" +#include "Compressor.h" #include "Client.hpp" -#include +#include "Settings.h" +#include "Network.h" +#include "Logger.h" #include #include -#include "../logger.h" -#include "../Settings.hpp" struct PacketData{ int ID; @@ -25,12 +26,20 @@ struct SplitData{ SOCKET UDPSock; std::set DataAcks; std::set SplitPackets; -void UDPSend(Client*c,const std::string&Data){ - if(!c->isConnected())return; +void UDPSend(Client*c,std::string Data){ + if(c == nullptr || !c->isConnected || c->GetStatus() < 0)return; sockaddr_in Addr = c->GetUDPAddr(); int AddrSize = sizeof(c->GetUDPAddr()); - int sendOk = sendto(UDPSock, Data.c_str(), int(Data.length()) + 1, 0, (sockaddr*)&Addr, AddrSize); - if (sendOk == SOCKET_ERROR)error("(UDP) Send Error Code : " + std::to_string(WSAGetLastError()) + " Size : " + std::to_string(AddrSize)); + if(Data.length() > 400){ + std::string CMP(Comp(Data)); + Data = "ABG:" + CMP; + } + + int sendOk = sendto(UDPSock, Data.c_str(), int(Data.size()), 0, (sockaddr *) &Addr, AddrSize); + if (sendOk == SOCKET_ERROR) { + debug(Sec("(UDP) Send Failed Code : ") + std::to_string(WSAGetLastError())); + if(c->GetStatus() > -1)c->SetStatus(-1); + } } void AckID(int ID){ @@ -58,7 +67,7 @@ void SendLarge(Client*c,const std::string&Data){ std::string Packet; if(Data.length() > 1000){ std::string pckt = Data; - int S = 1,Split = ceil(float(pckt.length()) / 1000); + int S = 1,Split = int(ceil(float(pckt.length()) / 1000)); int SID = SplitID(); while(pckt.length() > 1000){ Packet = "SC"+std::to_string(S)+"/"+std::to_string(Split)+":"+std::to_string(ID)+"|"+ @@ -79,8 +88,6 @@ void SendLarge(Client*c,const std::string&Data){ UDPSend(c,Packet); } } - - struct HandledC{ int Pos = 0; Client *c{}; @@ -114,12 +121,12 @@ bool Handled(Client*c,int ID){ } } for(HandledC*h : HandledIDs){ - if(h->c == nullptr || !h->c->isConnected()){ + if(h->c == nullptr || !h->c->isConnected){ HandledIDs.erase(h); break; } } - if(!handle) { + if(!handle){ HandledC *h = GetHandled(c); ResetIDs(h); if (h->Pos > 49)h->Pos = 0; @@ -135,13 +142,14 @@ std::string UDPRcvFromClient(sockaddr_in& client){ int clientLength = sizeof(client); ZeroMemory(&client, clientLength); ZeroMemory(buf, 10240); - int bytesIn = recvfrom(UDPSock, buf, 10240, 0, (sockaddr*)&client, &clientLength); - if (bytesIn == -1) - { - error("(UDP) Error receiving from Client! Code : " + std::to_string(WSAGetLastError())); + int Rcv = recvfrom(UDPSock, buf, 10240, 0, (sockaddr*)&client, &clientLength); + if (Rcv == -1){ + error(Sec("(UDP) Error receiving from Client! Code : ") + std::to_string(WSAGetLastError())); return ""; } - return std::string(buf); + std::string Ret(Rcv,0); + memcpy_s(&Ret[0],Rcv,buf,Rcv); + return Ret; } SplitData*GetSplit(int SplitID){ @@ -152,16 +160,17 @@ SplitData*GetSplit(int SplitID){ SplitPackets.insert(SP); return SP; } -void GlobalParser(Client*c, const std::string&Packet); void HandleChunk(Client*c,const std::string&Data){ - int pos1 = int(Data.find(':'))+1,pos2 = Data.find(':',pos1),pos3 = Data.find('/'); - int pos4 = Data.find('|'); + int pos1 = int(Data.find(':'))+1, + pos2 = int(Data.find(':',pos1)), + pos3 = int(Data.find('/')), + pos4 = int(Data.find('|')); if(pos1 == std::string::npos)return; int Max = stoi(Data.substr(pos3+1,pos1-pos3-2)); int Current = stoi(Data.substr(2,pos3-2)); int ID = stoi(Data.substr(pos1,pos4-pos1)); int SplitID = stoi(Data.substr(pos4+1,pos2-pos4-1)); - std::string ack = "ACK:" + Data.substr(pos1,pos4-pos1); + std::string ack = Sec("TRG:") + Data.substr(pos1,pos4-pos1); UDPSend(c,ack); if(Handled(c,ID))return; SplitData* SData = GetSplit(SplitID); @@ -173,43 +182,46 @@ void HandleChunk(Client*c,const std::string&Data){ for(const std::pair& a : SData->Fragments){ ToHandle += a.second; } - GlobalParser(c,ToHandle); + GParser(c,ToHandle); SplitPackets.erase(SData); } } -void UDPParser(Client*c, const std::string&Packet){ - if(Packet.substr(0,4) == "ACK:"){ - AckID(stoi(Packet.substr(4))); - return; - }else if(Packet.substr(0,3) == "BD:"){ - int pos = Packet.find(':',4); - int ID = stoi(Packet.substr(3,pos-3)); - std::string pckt = "ACK:" + std::to_string(ID); - UDPSend(c,pckt); - if(!Handled(c,ID)) { - pckt = Packet.substr(pos + 1); - GlobalParser(c, pckt); +void UDPParser(Client*c,std::string Packet){ + if(Packet.substr(0,4) == Sec("ABG:")){ + Packet = DeComp(Packet.substr(4)); + } + if(Packet.substr(0,4) == Sec("TRG:")){ + std::string pkt = Packet.substr(4); + if(Packet.find_first_not_of("0123456789") == -1){ + AckID(stoi(Packet)); } return; - }else if(Packet.substr(0,2) == "SC"){ - + }else if(Packet.substr(0,3) == Sec("BD:")){ + auto pos = Packet.find(':',4); + int ID = stoi(Packet.substr(3,pos-3)); + std::string pkt = Sec("TRG:") + std::to_string(ID); + UDPSend(c,pkt); + if(!Handled(c,ID)) { + pkt = Packet.substr(pos + 1); + GParser(c, pkt); + } + return; + }else if(Packet.substr(0,2) == Sec("SC")){ HandleChunk(c,Packet); return; } - GlobalParser(c,Packet); + GParser(c,Packet); } +#include void StartLoop(); - [[noreturn]] void UDPServerMain(){ WSADATA data; - if (WSAStartup(514, &data)) //2.2 - { - std::cout << "Can't start Winsock!" << std::endl; + if (WSAStartup(514, &data)){ + error(Sec("Can't start Winsock!")); //return; } UDPSock = socket(AF_INET, SOCK_DGRAM, 0); - // Create a server hint structure for the server sockaddr_in serverAddr{}; serverAddr.sin_addr.S_un.S_addr = ADDR_ANY; //Any Local @@ -217,53 +229,56 @@ void StartLoop(); serverAddr.sin_port = htons(Port); // Convert from little to big endian // Try and bind the socket to the IP and port - if (bind(UDPSock, (sockaddr*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) - { - std::cout << "Can't bind socket! " << WSAGetLastError() << std::endl; + if (bind(UDPSock, (sockaddr*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR){ + error(Sec("Can't bind socket!") + std::to_string(WSAGetLastError())); + std::this_thread::sleep_for(std::chrono::seconds(5)); + exit(-1); //return; } DataAcks.clear(); StartLoop(); - info("Vehicle data network online on port "+std::to_string(Port)+" with a Max of "+std::to_string(MaxPlayers)+" Clients"); - while (true) - { + info(Sec("Vehicle data network online on port ")+std::to_string(Port)+Sec(" with a Max of ")+std::to_string(MaxPlayers)+Sec(" Clients")); + while (true){ sockaddr_in client{}; std::string Data = UDPRcvFromClient(client); //Receives any data from Socket - int Pos = Data.find(':'); + auto Pos = Data.find(':'); if(Data.empty() || Pos < 0 || Pos > 2)continue; /*char clientIp[256]; ZeroMemory(clientIp, 256); ///Code to get IP we don't need that yet inet_ntop(AF_INET, &client.sin_addr, clientIp, 256);*/ uint8_t ID = Data.at(0)-1; - for(Client*c : Clients){ - if(c->GetID() == ID){ + for(Client*c : CI->Clients){ + if(c != nullptr && c->GetID() == ID){ c->SetUDPAddr(client); - c->SetConnected(true); - UDPParser(c,Data.substr(2)); + c->isConnected = true; + std::thread Parse(UDPParser,c,Data.substr(2)); + Parse.detach(); } } } - ///UDPSendToClient(c->GetUDPAddr(), sizeof(c->GetUDPAddr()), Data); /*closesocket(UDPSock); WSACleanup(); return;*/ } -#include void LOOP(){ while(UDPSock != -1) { for (PacketData* p : DataAcks){ - if (p->Client == nullptr || p->Client->GetTCPSock() == -1) { - DataAcks.erase(p); - break; - } - if(p->Tries < 20){ - UDPSend(p->Client,p->Data); - p->Tries++; - }else{ - DataAcks.erase(p); - break; + if(p != nullptr) { + if (p->Client == nullptr || p->Client->GetTCPSock() == -1) { + DataAcks.erase(p); + break; + } + if (p->Tries < 20) { + UDPSend(p->Client, p->Data); + p->Tries++; + } else { + DataAcks.erase(p); + delete p; + p = nullptr; + break; + } } } std::this_thread::sleep_for(std::chrono::milliseconds(200)); diff --git a/src/config.cpp b/src/config.cpp deleted file mode 100644 index ada9616..0000000 --- a/src/config.cpp +++ /dev/null @@ -1,132 +0,0 @@ -/// -/// Created by Anonymous275 on 1/28/2020 -/// - -#include -#include -#include -#include -#include "logger.h" - -void GenerateConfig(); -std::string RemoveComments(const std::string& Line); -void SetValues(const std::string& Line, int Index); -std::string MapName = "/levels/gridmap/info.json"; -std::string ServerName = "BeamMP New Server"; -std::string ServerDesc = "BeamMP Default Description"; -std::string Resource = "Resources"; -std::string Key; -bool Private = false; -bool Debug = false; -int MaxPlayers = 10; -int Port = 30814; -int MaxCars = 1; - -//Generates or Reads Config -void ParseConfig(){ - std::ifstream InFileStream; - InFileStream.open("Server.cfg"); - if(InFileStream.good()){ //Checks if Config Exists - std::string line; - int index = 1; - while (getline(InFileStream, line)) { - index++; - } - if(index-1 < 11){ - error("Outdated/Incorrect config please remove it server will close in 5 secs"); - std::this_thread::sleep_for(std::chrono::seconds(5)); - exit(3); - } - InFileStream.close(); - InFileStream.open("Server.cfg"); - info("Config Found Updating Values"); - index = 1; - while (getline(InFileStream, line)) { - if(line.rfind('#', 0) != 0 && line.rfind(' ', 0) != 0){ //Checks if it starts as Comment - std::string CleanLine = RemoveComments(line); //Cleans it from the Comments - SetValues(CleanLine,index); //sets the values - index++; - } - } - }else{ - info("Config Not Found Generating A new One"); - GenerateConfig(); - } - InFileStream.close(); -} - - - -void SetValues(const std::string& Line, int Index) { - int state = 0; - std::string Data; - bool Switch = false; - if (Index > 5)Switch = true; - for (char c : Line) { - if (Switch) { - if (c == '\"'){state++;} - if (state > 0 && state < 2) { - Data += c; - } - } else { - if (c == ' ') { state++; } - if (state > 1) { - Data += c; - } - } - } - Data = Data.substr(1); - std::string::size_type sz; - bool Boolean = std::string(Data).find("true") != std::string::npos;//searches for "true" - switch (Index){ - case 1 : Debug = Boolean;//checks and sets the Debug Value - break; - case 2 : Private = Boolean;//checks and sets the Private Value - break; - case 3 : Port = std::stoi(Data, &sz);//sets the Port - break; - case 4 : MaxCars = std::stoi(Data, &sz);//sets the Max Car amount - break; - case 5 : MaxPlayers = std::stoi(Data, &sz); //sets the Max Amount of player - break; - case 6 : MapName = Data; //Map - break; - case 7 : ServerName = Data; //Name - break; - case 8 : ServerDesc = Data; //desc - break; - case 9 : Resource = Data; //File name - break; - case 10 : Key = Data; //File name - } -} - - - -//generates default Config -void GenerateConfig(){ - std::ofstream FileStream; - FileStream.open ("Server.cfg"); - FileStream << "# This is the BeamMP Server Configuration File\n" - "Debug = false # true or false to enable debug console output\n" - "Private = false # Private?\n" - "Port = 30814 # Port to run the server on UDP and TCP\n" - "Cars = 1 # Max cars for every player\n" - "MaxPlayers = 10 # Maximum Amount of Clients\n" - "Map = \"/levels/gridmap/info.json\" # Default Map\n" - "Name = \"BeamMP New Server\" # Server Name\n" - "Desc = \"BeamMP Default Description\" # Server Description\n" - "use = \"Resources\" # Resource file name\n" - "AuthKey = \"\" # Auth Key"; - FileStream.close(); -} - - -std::string RemoveComments(const std::string& Line){ - std::string Return; - for(char c : Line) { - if(c == '#'){break;} //when it finds the # it will stop - Return += c; - } - return Return; //Converts it from a char array to string and returns it -} \ No newline at end of file diff --git a/src/functions.cpp b/src/functions.cpp deleted file mode 100644 index 4497886..0000000 --- a/src/functions.cpp +++ /dev/null @@ -1,17 +0,0 @@ -/// -/// Created by Anonymous275 on 4/1/2020 -/// - -#include -std::vector Split(const std::string& String,const std::string& delimiter){ - std::vector Val; - size_t pos = 0; - std::string token,s = String; - while ((pos = s.find(delimiter)) != std::string::npos) { - token = s.substr(0, pos); - Val.push_back(token); - s.erase(0, pos + delimiter.length()); - } - Val.push_back(s); - return Val; -} \ No newline at end of file diff --git a/src/heartbeat.cpp b/src/heartbeat.cpp deleted file mode 100644 index 0359f8b..0000000 --- a/src/heartbeat.cpp +++ /dev/null @@ -1,76 +0,0 @@ -/// -/// Created by Mitch on 04/02/2020 -/// - -#include -#include -#include -#include "logger.h" -#include "Settings.hpp" -#include "Network 2.0/Client.hpp" -extern std::string StatReport; -std::string HTTP_REQUEST(const std::string&,int); -std::string PostHTTP(const std::string& IP,const std::string& Fields); -int Beat = 0; -std::string HTA(const std::string& hex) -{ - std::string ascii; - for (size_t i = 0; i < hex.length(); i += 2) - { - std::string part = hex.substr(i, 2); - char ch = stoul(part, nullptr, 16); - ascii += ch; - } - return ascii; -} -std::string GetPlayers(){ - std::string Return; - for(Client* c : Clients){ - Return += c->GetName() + ";"; - } - return Return; -} - -void Heartbeat(){ - std::string State,R,T; - while(true){ - State = Private ? "true" : "false"; - R = "uuid="+Key+"&players="+std::to_string(Clients.size())+"&maxplayers="+std::to_string(MaxPlayers)+"&port=" - + std::to_string(Port) + "&map=" + MapName + "&private="+State+"&version="+ServerVersion+ - "&clientversion="+ClientVersion+"&name="+ServerName+"&pps="+StatReport+"&modlist="+FileList+ - "&modstotalsize="+std::to_string(MaxModSize)+"&modstotal="+std::to_string(ModsLoaded) - +"&playerslist="+GetPlayers()+"&desc="+ServerDesc; - if(!CustomIP.empty())R+="&ip="+CustomIP; - //https://beamng-mp.com/heartbeatv2 - T = PostHTTP(HTA("68747470733a2f2f6265616d6e672d6d702e636f6d2f6865617274626561747632"),R); - if(T.find_first_not_of("20") != std::string::npos){ - //Backend system refused server startup! - if(Beat > 50)Beat = 1; - else Beat++; - std::this_thread::sleep_for(std::chrono::seconds(10)); - if(Beat > 50)Beat = 1; - else Beat++; - T = PostHTTP(HTA("68747470733a2f2f6265616d6e672d6d702e636f6d2f6865617274626561747632"),R); - if(T.find_first_not_of("20") != std::string::npos){ - error(HTA("4261636b656e642073797374656d20726566757365642073657276657221")); - std::this_thread::sleep_for(std::chrono::seconds(3)); - exit(-1); - } - } - //Server Authenticated - if(T.length() == 4)info(HTA("5365727665722061757468656e746963617465642077697468206261636b656e64")); - R.clear(); - T.clear(); - State.clear(); - if(Beat > 50)Beat = 1; - else Beat++; - std::this_thread::sleep_for(std::chrono::seconds(3)); - } -} - - -void HeartbeatInit() -{ - std::thread HB(Heartbeat); - HB.detach(); -} \ No newline at end of file diff --git a/src/logger.cpp b/src/logger.cpp index a14d306..61d91a7 100644 --- a/src/logger.cpp +++ b/src/logger.cpp @@ -1,101 +1,80 @@ /// -/// Created by jojos38 on 28/01/2020 +/// Created by Anonymous275 on 7/17/2020 /// - - +#include "Security/Enc.h" +#include "Settings.h" +#include "Logger.h" #include -#include "logger.h" -#include - -void addToLog(const std::string& Data); -int loggerlevel; - -void setLoggerLevel(int level) { - //0 ALL 1 DEBUG 2 INFO 3 WARN 4 ERROR 5 OFF - loggerlevel = level; -} - -std::stringstream getDate() { - // current date/time based on current system - time_t now = time(nullptr); - tm* ltm = localtime(&now); - - int month = 1 + ltm->tm_mon; - int day = ltm->tm_mday; - int hours = ltm->tm_hour; - int minutes = ltm->tm_min; - int seconds = ltm->tm_sec; - - std::string month_string; - if (month < 10) month_string = "0" + std::to_string(month); - else month_string = std::to_string(month); - - std::string day_string; - if (day < 10) day_string = "0" + std::to_string(day); - else day_string = std::to_string(day); - - std::string hours_string; - if (hours < 10) hours_string = "0" + std::to_string(hours); - else hours_string = std::to_string(hours); - - std::string minutes_string; - if (minutes < 10) minutes_string = "0" + std::to_string(minutes); - else minutes_string = std::to_string(minutes); - - std::string seconds_string; - if (seconds < 10) seconds_string = "0" + std::to_string(seconds); - else seconds_string = std::to_string(seconds); - +#include +#include +std::string getDate() { + typedef std::chrono::duration>::type> days; + std::chrono::system_clock::time_point now = std::chrono::system_clock::now(); + std::chrono::system_clock::duration tp = now.time_since_epoch(); + days d = std::chrono::duration_cast(tp);tp -= d; + auto h = std::chrono::duration_cast(tp);tp -= h; + auto m = std::chrono::duration_cast(tp);tp -= m; + auto s = std::chrono::duration_cast(tp);tp -= s; + time_t tt = std::chrono::system_clock::to_time_t(now); + tm local_tm{}; + localtime_s(&local_tm,&tt); std::stringstream date; + int S = local_tm.tm_sec; + int M = local_tm.tm_min; + int H = local_tm.tm_hour; + std::string Secs = (S > 9 ? std::to_string(S) : "0" + std::to_string(S)); + std::string Min = (M > 9 ? std::to_string(M) : "0" + std::to_string(M)); + std::string Hour = (H > 9 ? std::to_string(H) : "0" + std::to_string(H)); date - << "[" - << day_string << "/" - << month_string << "/" - << 1900 + ltm->tm_year << " " - << hours_string << ":" - << minutes_string << ":" - << seconds_string - << "] "; - return date; + << "[" + << local_tm.tm_mday << "/" + << local_tm.tm_mon + 1 << "/" + << local_tm.tm_year + 1900 << " " + << Hour << ":" + << Min << ":" + << Secs + << "] "; + return date.str(); +} +void InitLog(){ + std::ofstream LFS; + LFS.open (Sec("Server.log")); + if(!LFS.is_open()){ + error(Sec("logger file init failed!")); + }else LFS.close(); +} +void addToLog(const std::string& Line){ + std::ofstream LFS; + LFS.open (Sec("Server.log"), std::ios_base::app); + LFS << Line.c_str(); + LFS.close(); } - - void info(const std::string& toPrint) { - if (loggerlevel <= 2){ - std::string Print = getDate().str() + "[INFO] " + toPrint + "\n"; - std::cout << Print; - addToLog(Print); - } -} - -void error(const std::string& toPrint) { - if (loggerlevel <= 4) { - std::string Print = getDate().str() + "[ERROR] " + toPrint + "\n"; - std::cout << Print; - addToLog(Print); - } -} - -void warn(const std::string& toPrint) { - if (loggerlevel <= 3) { - std::string Print = getDate().str() + "[WARN] " + toPrint + "\n"; - std::cout << Print; - addToLog(Print); - } + std::string Print = getDate() + Sec("[INFO] ") + toPrint + "\n"; + std::cout << Print; + addToLog(Print); } void debug(const std::string& toPrint) { - if (loggerlevel <= 1) { - std::string Print = getDate().str() + "[DEBUG] " + toPrint + "\n"; - std::cout << Print; - addToLog(Print); - } + if(!Debug)return; + std::string Print = getDate() + Sec("[DEBUG] ") + toPrint + "\n"; + std::cout << Print; + addToLog(Print); +} +void warn(const std::string& toPrint){ + std::string Print = getDate() + Sec("[WARN] ") + toPrint + "\n"; + std::cout << Print; + addToLog(Print); +} +void error(const std::string& toPrint) { + static int ECounter = 0; + std::string Print = getDate() + Sec("[ERROR] ") + toPrint + "\n"; + std::cout << Print; + addToLog(Print); + if(ECounter > 10)exit(7); + ECounter++; +} +void except(const std::string& toPrint) { + std::string Print = getDate() + Sec("[EXCEP] ") + toPrint + "\n"; + std::cout << Print; + addToLog(Print); } -void Exception(unsigned long Code,char* Origin) { - char* hex_string = new char[100]; - sprintf(hex_string, "%lX", Code); //convert number to hex - if (loggerlevel <= 4) { - std::string Print = getDate().str() + "[EXCEP] code " + hex_string + " Origin: "+ std::string(Origin) +"\n"; - std::cout << Print; - addToLog(Print); - } -} \ No newline at end of file diff --git a/src/main.cpp b/src/main.cpp index 4a9108b..e1bb2a5 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -1,74 +1,21 @@ -/// -/// Created by Anonymous275 on 28/01/2020 -/// -#include -#include -#include -#include "logger.h" -#include -#include "Settings.hpp" - -void DebugData(); -void LogInit(); -void ParseConfig(); -void addToLog(const std::string& Data); -//void ServerMain(int Port, int MaxClients); -void HeartbeatInit(); -std::string ServerVersion = "0.51"; -std::string ClientVersion = "1.50"; -std::string CustomIP; -void HandleResources(std::string path); -void StatInit(); -void NetMain(); - +#include "Startup.h" +#include +#include +[[noreturn]] void loop(){ + while(true){ + std::cout.flush(); + std::this_thread::sleep_for(std::chrono::milliseconds(600)); + } +} int main(int argc, char* argv[]) { - if(argc > 1){ - CustomIP = argv[1]; - size_t n = std::count(CustomIP.begin(), CustomIP.end(), '.'); - int p = CustomIP.find_first_not_of(".0123456789"); - if(p != std::string::npos || n != 3 || CustomIP.substr(0,3) == "127"){ - CustomIP.clear(); - warn("IP Specified is invalid!"); - }else info("Started with custom ip : " + CustomIP); - } - info("BeamMP Server Running version " + ServerVersion); - LogInit(); - ParseConfig(); - HandleResources(Resource); - HeartbeatInit(); - if(Debug)DebugData(); - setLoggerLevel(0); //0 for all - if(ModsLoaded){ - info("Loaded "+std::to_string(ModsLoaded)+" Mods"); - } + std::thread t1(loop); + t1.detach(); + InitServer(argc,argv); + InitConfig(); + InitLua(); + InitRes(); + HBInit(); StatInit(); NetMain(); + return 0; } - - -void DebugData(){ - debug(std::string("Debug : ") + (Debug?"true":"false")); - debug(std::string("Private : ") + (Private?"true":"false")); - debug("Port : " + std::to_string(Port)); - debug("Max Cars : " + std::to_string(MaxCars)); - debug("MaxPlayers : " + std::to_string(MaxPlayers)); - debug("MapName : " + MapName); - debug("ServerName : " + ServerName); - debug("ServerDesc : " + ServerDesc); - debug("File : " + Resource); - debug("Auth Key : " + Key); -} - - -void LogInit(){ - std::ofstream LFS; - LFS.open ("Server.log"); - LFS.close(); -} - -void addToLog(const std::string& Data){ - std::ofstream LFS; - LFS.open ("Server.log", std::ios_base::app); - LFS << Data.c_str(); - LFS.close(); -} \ No newline at end of file