commit 32db868ae61e8b51dc58771e5bdab8e128c5cfd8 Author: Diavolo Date: Fri May 26 16:09:29 2023 +0200 init diff --git a/.github/dependabot.yml b/.github/dependabot.yml new file mode 100644 index 0000000..4fcd556 --- /dev/null +++ b/.github/dependabot.yml @@ -0,0 +1,7 @@ +version: 2 +updates: +- package-ecosystem: gitsubmodule + directory: "/" + schedule: + interval: daily + open-pull-requests-limit: 10 diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml new file mode 100644 index 0000000..bd7c938 --- /dev/null +++ b/.github/workflows/build.yml @@ -0,0 +1,158 @@ +name: Build + +on: + push: + branches: + - "*" + pull_request: + branches: + - "*" + types: [opened, synchronize, reopened] + +concurrency: + group: ${{ github.ref }} + cancel-in-progress: true + +jobs: + build-win: + name: Build Windows + runs-on: windows-latest + strategy: + fail-fast: false + matrix: + configuration: + - Debug + - Release + arch: + - x64 + include: + - arch: x64 + platform: x64 + steps: + - name: Check out files + uses: actions/checkout@v3.5.2 + with: + submodules: true + fetch-depth: 0 + # NOTE - If LFS ever starts getting used during builds, switch this to true! + lfs: false + + - name: Add msbuild to PATH + uses: microsoft/setup-msbuild@v1.3.1 + + - name: Generate project files + run: cmd /C call generate.bat + + - name: Set up problem matching + uses: ammaraskar/msvc-problem-matcher@master + + - name: Build ${{matrix.configuration}} ${{matrix.arch}} binaries + run: msbuild /m /v:minimal /p:Configuration=${{matrix.configuration}} /p:Platform=${{matrix.platform}} build/alterware-master.sln + + - name: Upload ${{matrix.configuration}} ${{matrix.arch}} binaries + uses: actions/upload-artifact@v3.1.2 + with: + name: Windows ${{matrix.configuration}} ${{matrix.arch}} binaries + path: | + build/bin/${{matrix.arch}}/${{matrix.configuration}}/alterware-master.exe + build/bin/${{matrix.arch}}/${{matrix.configuration}}/alterware-master.pdb + + build-linux: + name: Build Linux + runs-on: ubuntu-latest + strategy: + fail-fast: false + matrix: + configuration: + - Debug + - Release + arch: + - x64 + include: + - configuration: Debug + config: debug + - configuration: Release + config: release + steps: + - name: Check out files + uses: actions/checkout@v3.5.2 + with: + submodules: true + fetch-depth: 0 + # NOTE - If LFS ever starts getting used during builds, switch this to true! + lfs: false + + - name: Install dependencies (x64) + if: matrix.arch == 'x64' + run: | + sudo apt-get update + sudo apt-get -y install libcurl4-gnutls-dev + + - name: Install Mold + uses: rui314/setup-mold@staging + + - name: Generate project files + run: ./generate.sh + + - name: Set up problem matching + uses: ammaraskar/gcc-problem-matcher@master + + - name: Build ${{matrix.configuration}} ${{matrix.arch}} binaries + run: | + pushd build + ../tools/mold -run make config=${{matrix.config}}_${{matrix.arch}} -j$(nproc) + + - name: Upload ${{matrix.configuration}} ${{matrix.arch}} binaries + uses: actions/upload-artifact@v3.1.2 + with: + name: Linux ${{matrix.configuration}} ${{matrix.arch}} binaries + path: | + build/bin/${{matrix.arch}}/${{matrix.configuration}}/alterware-master + + build-mac: + name: Build macOS + runs-on: macos-13 + strategy: + fail-fast: false + matrix: + configuration: + - Debug + - Release + arch: + - x64 + - ARM64 + include: + - configuration: Debug + config: debug + - configuration: Release + config: release + - arch: x64 + platform: x64 + - arch: ARM64 + platform: arm64 + steps: + - name: Check out files + uses: actions/checkout@v3.5.2 + with: + submodules: true + fetch-depth: 0 + # NOTE - If LFS ever starts getting used during builds, switch this to true! + lfs: false + + - name: Generate project files + run: ./generate.sh + + - name: Set up problem matching + uses: ammaraskar/gcc-problem-matcher@master + + - name: Build ${{matrix.configuration}} ${{matrix.arch}} binaries + run: | + pushd build + make config=${{matrix.config}}_${{matrix.platform}} -j$(sysctl -n hw.logicalcpu) + + - name: Upload ${{matrix.configuration}} ${{matrix.arch}} binaries + uses: actions/upload-artifact@v3.1.2 + with: + name: macOS ${{matrix.configuration}} ${{matrix.arch}} binaries + path: | + build/bin/${{matrix.platform}}/${{matrix.configuration}}/alterware-master diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..130a33a --- /dev/null +++ b/.gitignore @@ -0,0 +1,150 @@ +### Windows + +# Windows image file caches +Thumbs.db +ehthumbs.db + +# Folder config file +Desktop.ini + +# Recycle Bin used on file shares +$RECYCLE.BIN/ + +# Windows Installer files +*.cab +*.msi +*.msm +*.msp + +# Shortcuts +*.lnk + +### OSX + +.DS_Store +.AppleDouble +.LSOverride + +# Icon must end with two \r +Icon + +# Thumbnails +._* + +# Files that might appear on external disk +.Spotlight-V100 +.Trashes + +# Directories potentially created on remote AFP share +.AppleDB +.AppleDesktop +Network Trash Folder +Temporary Items +.apdisk + +### Visual Studio + +# User-specific files +*.suo +*.user +*.userosscache +*.sln.docstates + +# User-specific files (MonoDevelop/Xamarin Studio) +*.userprefs + +# Build results +build + +# Visual Studio 2015 cache/options directory +.vs/ + +# MSTest test Results +[Tt]est[Rr]esult*/ +[Bb]uild[Ll]og.* + +*_i.c +*_p.c +*_i.h +*.ilk +*.meta +*.obj +*.pch +*.pdb +*.pgc +*.pgd +*.rsp +*.sbr +*.tlb +*.tli +*.tlh +*.tmp +*.tmp_proj +*.log +*.vspscc +*.vssscc +.builds +*.pidb +*.svclog +*.scc + +# Visual C++ cache files +ipch/ +*.aps +*.ncb +*.opendb +*.opensdf +*.sdf +*.cachefile + +# Visual Studio profiler +*.psess +*.vsp +*.vspx +*.sap + +# TFS 2012 Local Workspace +$tf/ + +# Guidance Automation Toolkit +*.gpState + +# Visual Studio cache files +# files ending in .cache can be ignored +*.[Cc]ache +# but keep track of directories ending in .cache +!*.[Cc]ache/ + +# Others +~$* +*~ +*.dbmdl +*.dbproj.schemaview +*.pfx +*.publishsettings + +# Backup & report files from converting an old project file +# to a newer Visual Studio version. Backup files are not needed, +# because we have git ;-) +_UpgradeReport_Files/ +Backup*/ +UpgradeLog*.XML +UpgradeLog*.htm + +# SQL Server files +*.mdf +*.ldf + +### IDA +*.id0 +*.id1 +*.id2 +*.nam +*.til + +### Custom user files +# User scripts +user*.bat + +# Premake binary +#premake5.exe \ No newline at end of file diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..ac91bd2 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,22 @@ +[submodule "deps/libtommath"] + path = deps/libtommath + url = https://github.com/libtom/libtommath.git + branch = develop +[submodule "deps/zlib"] + path = deps/zlib + url = https://github.com/madler/zlib.git + branch = develop +[submodule "deps/GSL"] + path = deps/GSL + url = https://github.com/Microsoft/GSL.git +[submodule "deps/rapidjson"] + path = deps/rapidjson + url = https://github.com/Tencent/rapidjson.git +[submodule "deps/libtomcrypt"] + path = deps/libtomcrypt + url = https://github.com/libtom/libtomcrypt.git + branch = develop +[submodule "deps/curl"] + path = deps/curl + url = https://github.com/curl/curl.git + branch = curl-8_0_1 diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..f288702 --- /dev/null +++ b/LICENSE @@ -0,0 +1,674 @@ + GNU GENERAL PUBLIC LICENSE + Version 3, 29 June 2007 + + Copyright (C) 2007 Free Software Foundation, Inc. + Everyone is permitted to copy and distribute verbatim copies + of this license document, but changing it is not allowed. + + Preamble + + The GNU General Public License is a free, copyleft license for +software and other kinds of works. + + The licenses for most software and other practical works are designed +to take away your freedom to share and change the works. By contrast, +the GNU General Public License is intended to guarantee your freedom to +share and change all versions of a program--to make sure it remains free +software for all its users. We, the Free Software Foundation, use the +GNU General Public License for most of our software; it applies also to +any other work released this way by its authors. You can apply it to +your programs, too. + + When we speak of free software, we are referring to freedom, not +price. Our General Public Licenses are designed to make sure that you +have the freedom to distribute copies of free software (and charge for +them if you wish), that you receive source code or can get it if you +want it, that you can change the software or use pieces of it in new +free programs, and that you know you can do these things. + + To protect your rights, we need to prevent others from denying you +these rights or asking you to surrender the rights. Therefore, you have +certain responsibilities if you distribute copies of the software, or if +you modify it: responsibilities to respect the freedom of others. + + For example, if you distribute copies of such a program, whether +gratis or for a fee, you must pass on to the recipients the same +freedoms that you received. You must make sure that they, too, receive +or can get the source code. And you must show them these terms so they +know their rights. + + Developers that use the GNU GPL protect your rights with two steps: +(1) assert copyright on the software, and (2) offer you this License +giving you legal permission to copy, distribute and/or modify it. + + For the developers' and authors' protection, the GPL clearly explains +that there is no warranty for this free software. For both users' and +authors' sake, the GPL requires that modified versions be marked as +changed, so that their problems will not be attributed erroneously to +authors of previous versions. + + Some devices are designed to deny users access to install or run +modified versions of the software inside them, although the manufacturer +can do so. This is fundamentally incompatible with the aim of +protecting users' freedom to change the software. The systematic +pattern of such abuse occurs in the area of products for individuals to +use, which is precisely where it is most unacceptable. Therefore, we +have designed this version of the GPL to prohibit the practice for those +products. If such problems arise substantially in other domains, we +stand ready to extend this provision to those domains in future versions +of the GPL, as needed to protect the freedom of users. + + Finally, every program is threatened constantly by software patents. +States should not allow patents to restrict development and use of +software on general-purpose computers, but in those that do, we wish to +avoid the special danger that patents applied to a free program could +make it effectively proprietary. To prevent this, the GPL assures that +patents cannot be used to render the program non-free. + + The precise terms and conditions for copying, distribution and +modification follow. + + TERMS AND CONDITIONS + + 0. Definitions. + + "This License" refers to version 3 of the GNU General Public License. + + "Copyright" also means copyright-like laws that apply to other kinds of +works, such as semiconductor masks. + + "The Program" refers to any copyrightable work licensed under this +License. Each licensee is addressed as "you". "Licensees" and +"recipients" may be individuals or organizations. + + To "modify" a work means to copy from or adapt all or part of the work +in a fashion requiring copyright permission, other than the making of an +exact copy. The resulting work is called a "modified version" of the +earlier work or a work "based on" the earlier work. + + A "covered work" means either the unmodified Program or a work based +on the Program. + + To "propagate" a work means to do anything with it that, without +permission, would make you directly or secondarily liable for +infringement under applicable copyright law, except executing it on a +computer or modifying a private copy. Propagation includes copying, +distribution (with or without modification), making available to the +public, and in some countries other activities as well. + + To "convey" a work means any kind of propagation that enables other +parties to make or receive copies. Mere interaction with a user through +a computer network, with no transfer of a copy, is not conveying. + + An interactive user interface displays "Appropriate Legal Notices" +to the extent that it includes a convenient and prominently visible +feature that (1) displays an appropriate copyright notice, and (2) +tells the user that there is no warranty for the work (except to the +extent that warranties are provided), that licensees may convey the +work under this License, and how to view a copy of this License. If +the interface presents a list of user commands or options, such as a +menu, a prominent item in the list meets this criterion. + + 1. Source Code. + + The "source code" for a work means the preferred form of the work +for making modifications to it. "Object code" means any non-source +form of a work. + + A "Standard Interface" means an interface that either is an official +standard defined by a recognized standards body, or, in the case of +interfaces specified for a particular programming language, one that +is widely used among developers working in that language. + + The "System Libraries" of an executable work include anything, other +than the work as a whole, that (a) is included in the normal form of +packaging a Major Component, but which is not part of that Major +Component, and (b) serves only to enable use of the work with that +Major Component, or to implement a Standard Interface for which an +implementation is available to the public in source code form. A +"Major Component", in this context, means a major essential component +(kernel, window system, and so on) of the specific operating system +(if any) on which the executable work runs, or a compiler used to +produce the work, or an object code interpreter used to run it. + + The "Corresponding Source" for a work in object code form means all +the source code needed to generate, install, and (for an executable +work) run the object code and to modify the work, including scripts to +control those activities. However, it does not include the work's +System Libraries, or general-purpose tools or generally available free +programs which are used unmodified in performing those activities but +which are not part of the work. For example, Corresponding Source +includes interface definition files associated with source files for +the work, and the source code for shared libraries and dynamically +linked subprograms that the work is specifically designed to require, +such as by intimate data communication or control flow between those +subprograms and other parts of the work. + + The Corresponding Source need not include anything that users +can regenerate automatically from other parts of the Corresponding +Source. + + The Corresponding Source for a work in source code form is that +same work. + + 2. Basic Permissions. + + All rights granted under this License are granted for the term of +copyright on the Program, and are irrevocable provided the stated +conditions are met. This License explicitly affirms your unlimited +permission to run the unmodified Program. The output from running a +covered work is covered by this License only if the output, given its +content, constitutes a covered work. This License acknowledges your +rights of fair use or other equivalent, as provided by copyright law. + + You may make, run and propagate covered works that you do not +convey, without conditions so long as your license otherwise remains +in force. You may convey covered works to others for the sole purpose +of having them make modifications exclusively for you, or provide you +with facilities for running those works, provided that you comply with +the terms of this License in conveying all material for which you do +not control copyright. Those thus making or running the covered works +for you must do so exclusively on your behalf, under your direction +and control, on terms that prohibit them from making any copies of +your copyrighted material outside their relationship with you. + + Conveying under any other circumstances is permitted solely under +the conditions stated below. Sublicensing is not allowed; section 10 +makes it unnecessary. + + 3. Protecting Users' Legal Rights From Anti-Circumvention Law. + + No covered work shall be deemed part of an effective technological +measure under any applicable law fulfilling obligations under article +11 of the WIPO copyright treaty adopted on 20 December 1996, or +similar laws prohibiting or restricting circumvention of such +measures. + + When you convey a covered work, you waive any legal power to forbid +circumvention of technological measures to the extent such circumvention +is effected by exercising rights under this License with respect to +the covered work, and you disclaim any intention to limit operation or +modification of the work as a means of enforcing, against the work's +users, your or third parties' legal rights to forbid circumvention of +technological measures. + + 4. Conveying Verbatim Copies. + + You may convey verbatim copies of the Program's source code as you +receive it, in any medium, provided that you conspicuously and +appropriately publish on each copy an appropriate copyright notice; +keep intact all notices stating that this License and any +non-permissive terms added in accord with section 7 apply to the code; +keep intact all notices of the absence of any warranty; and give all +recipients a copy of this License along with the Program. + + You may charge any price or no price for each copy that you convey, +and you may offer support or warranty protection for a fee. + + 5. Conveying Modified Source Versions. + + You may convey a work based on the Program, or the modifications to +produce it from the Program, in the form of source code under the +terms of section 4, provided that you also meet all of these conditions: + + a) The work must carry prominent notices stating that you modified + it, and giving a relevant date. + + b) The work must carry prominent notices stating that it is + released under this License and any conditions added under section + 7. This requirement modifies the requirement in section 4 to + "keep intact all notices". + + c) You must license the entire work, as a whole, under this + License to anyone who comes into possession of a copy. This + License will therefore apply, along with any applicable section 7 + additional terms, to the whole of the work, and all its parts, + regardless of how they are packaged. This License gives no + permission to license the work in any other way, but it does not + invalidate such permission if you have separately received it. + + d) If the work has interactive user interfaces, each must display + Appropriate Legal Notices; however, if the Program has interactive + interfaces that do not display Appropriate Legal Notices, your + work need not make them do so. + + A compilation of a covered work with other separate and independent +works, which are not by their nature extensions of the covered work, +and which are not combined with it such as to form a larger program, +in or on a volume of a storage or distribution medium, is called an +"aggregate" if the compilation and its resulting copyright are not +used to limit the access or legal rights of the compilation's users +beyond what the individual works permit. Inclusion of a covered work +in an aggregate does not cause this License to apply to the other +parts of the aggregate. + + 6. Conveying Non-Source Forms. + + You may convey a covered work in object code form under the terms +of sections 4 and 5, provided that you also convey the +machine-readable Corresponding Source under the terms of this License, +in one of these ways: + + a) Convey the object code in, or embodied in, a physical product + (including a physical distribution medium), accompanied by the + Corresponding Source fixed on a durable physical medium + customarily used for software interchange. + + b) Convey the object code in, or embodied in, a physical product + (including a physical distribution medium), accompanied by a + written offer, valid for at least three years and valid for as + long as you offer spare parts or customer support for that product + model, to give anyone who possesses the object code either (1) a + copy of the Corresponding Source for all the software in the + product that is covered by this License, on a durable physical + medium customarily used for software interchange, for a price no + more than your reasonable cost of physically performing this + conveying of source, or (2) access to copy the + Corresponding Source from a network server at no charge. + + c) Convey individual copies of the object code with a copy of the + written offer to provide the Corresponding Source. This + alternative is allowed only occasionally and noncommercially, and + only if you received the object code with such an offer, in accord + with subsection 6b. + + d) Convey the object code by offering access from a designated + place (gratis or for a charge), and offer equivalent access to the + Corresponding Source in the same way through the same place at no + further charge. You need not require recipients to copy the + Corresponding Source along with the object code. If the place to + copy the object code is a network server, the Corresponding Source + may be on a different server (operated by you or a third party) + that supports equivalent copying facilities, provided you maintain + clear directions next to the object code saying where to find the + Corresponding Source. Regardless of what server hosts the + Corresponding Source, you remain obligated to ensure that it is + available for as long as needed to satisfy these requirements. + + e) Convey the object code using peer-to-peer transmission, provided + you inform other peers where the object code and Corresponding + Source of the work are being offered to the general public at no + charge under subsection 6d. + + A separable portion of the object code, whose source code is excluded +from the Corresponding Source as a System Library, need not be +included in conveying the object code work. + + A "User Product" is either (1) a "consumer product", which means any +tangible personal property which is normally used for personal, family, +or household purposes, or (2) anything designed or sold for incorporation +into a dwelling. In determining whether a product is a consumer product, +doubtful cases shall be resolved in favor of coverage. For a particular +product received by a particular user, "normally used" refers to a +typical or common use of that class of product, regardless of the status +of the particular user or of the way in which the particular user +actually uses, or expects or is expected to use, the product. A product +is a consumer product regardless of whether the product has substantial +commercial, industrial or non-consumer uses, unless such uses represent +the only significant mode of use of the product. + + "Installation Information" for a User Product means any methods, +procedures, authorization keys, or other information required to install +and execute modified versions of a covered work in that User Product from +a modified version of its Corresponding Source. The information must +suffice to ensure that the continued functioning of the modified object +code is in no case prevented or interfered with solely because +modification has been made. + + If you convey an object code work under this section in, or with, or +specifically for use in, a User Product, and the conveying occurs as +part of a transaction in which the right of possession and use of the +User Product is transferred to the recipient in perpetuity or for a +fixed term (regardless of how the transaction is characterized), the +Corresponding Source conveyed under this section must be accompanied +by the Installation Information. But this requirement does not apply +if neither you nor any third party retains the ability to install +modified object code on the User Product (for example, the work has +been installed in ROM). + + The requirement to provide Installation Information does not include a +requirement to continue to provide support service, warranty, or updates +for a work that has been modified or installed by the recipient, or for +the User Product in which it has been modified or installed. Access to a +network may be denied when the modification itself materially and +adversely affects the operation of the network or violates the rules and +protocols for communication across the network. + + Corresponding Source conveyed, and Installation Information provided, +in accord with this section must be in a format that is publicly +documented (and with an implementation available to the public in +source code form), and must require no special password or key for +unpacking, reading or copying. + + 7. Additional Terms. + + "Additional permissions" are terms that supplement the terms of this +License by making exceptions from one or more of its conditions. +Additional permissions that are applicable to the entire Program shall +be treated as though they were included in this License, to the extent +that they are valid under applicable law. If additional permissions +apply only to part of the Program, that part may be used separately +under those permissions, but the entire Program remains governed by +this License without regard to the additional permissions. + + When you convey a copy of a covered work, you may at your option +remove any additional permissions from that copy, or from any part of +it. (Additional permissions may be written to require their own +removal in certain cases when you modify the work.) You may place +additional permissions on material, added by you to a covered work, +for which you have or can give appropriate copyright permission. + + Notwithstanding any other provision of this License, for material you +add to a covered work, you may (if authorized by the copyright holders of +that material) supplement the terms of this License with terms: + + a) Disclaiming warranty or limiting liability differently from the + terms of sections 15 and 16 of this License; or + + b) Requiring preservation of specified reasonable legal notices or + author attributions in that material or in the Appropriate Legal + Notices displayed by works containing it; or + + c) Prohibiting misrepresentation of the origin of that material, or + requiring that modified versions of such material be marked in + reasonable ways as different from the original version; or + + d) Limiting the use for publicity purposes of names of licensors or + authors of the material; or + + e) Declining to grant rights under trademark law for use of some + trade names, trademarks, or service marks; or + + f) Requiring indemnification of licensors and authors of that + material by anyone who conveys the material (or modified versions of + it) with contractual assumptions of liability to the recipient, for + any liability that these contractual assumptions directly impose on + those licensors and authors. + + All other non-permissive additional terms are considered "further +restrictions" within the meaning of section 10. If the Program as you +received it, or any part of it, contains a notice stating that it is +governed by this License along with a term that is a further +restriction, you may remove that term. If a license document contains +a further restriction but permits relicensing or conveying under this +License, you may add to a covered work material governed by the terms +of that license document, provided that the further restriction does +not survive such relicensing or conveying. + + If you add terms to a covered work in accord with this section, you +must place, in the relevant source files, a statement of the +additional terms that apply to those files, or a notice indicating +where to find the applicable terms. + + Additional terms, permissive or non-permissive, may be stated in the +form of a separately written license, or stated as exceptions; +the above requirements apply either way. + + 8. Termination. + + You may not propagate or modify a covered work except as expressly +provided under this License. Any attempt otherwise to propagate or +modify it is void, and will automatically terminate your rights under +this License (including any patent licenses granted under the third +paragraph of section 11). + + However, if you cease all violation of this License, then your +license from a particular copyright holder is reinstated (a) +provisionally, unless and until the copyright holder explicitly and +finally terminates your license, and (b) permanently, if the copyright +holder fails to notify you of the violation by some reasonable means +prior to 60 days after the cessation. + + Moreover, your license from a particular copyright holder is +reinstated permanently if the copyright holder notifies you of the +violation by some reasonable means, this is the first time you have +received notice of violation of this License (for any work) from that +copyright holder, and you cure the violation prior to 30 days after +your receipt of the notice. + + Termination of your rights under this section does not terminate the +licenses of parties who have received copies or rights from you under +this License. If your rights have been terminated and not permanently +reinstated, you do not qualify to receive new licenses for the same +material under section 10. + + 9. Acceptance Not Required for Having Copies. + + You are not required to accept this License in order to receive or +run a copy of the Program. Ancillary propagation of a covered work +occurring solely as a consequence of using peer-to-peer transmission +to receive a copy likewise does not require acceptance. However, +nothing other than this License grants you permission to propagate or +modify any covered work. These actions infringe copyright if you do +not accept this License. Therefore, by modifying or propagating a +covered work, you indicate your acceptance of this License to do so. + + 10. Automatic Licensing of Downstream Recipients. + + Each time you convey a covered work, the recipient automatically +receives a license from the original licensors, to run, modify and +propagate that work, subject to this License. You are not responsible +for enforcing compliance by third parties with this License. + + An "entity transaction" is a transaction transferring control of an +organization, or substantially all assets of one, or subdividing an +organization, or merging organizations. If propagation of a covered +work results from an entity transaction, each party to that +transaction who receives a copy of the work also receives whatever +licenses to the work the party's predecessor in interest had or could +give under the previous paragraph, plus a right to possession of the +Corresponding Source of the work from the predecessor in interest, if +the predecessor has it or can get it with reasonable efforts. + + You may not impose any further restrictions on the exercise of the +rights granted or affirmed under this License. For example, you may +not impose a license fee, royalty, or other charge for exercise of +rights granted under this License, and you may not initiate litigation +(including a cross-claim or counterclaim in a lawsuit) alleging that +any patent claim is infringed by making, using, selling, offering for +sale, or importing the Program or any portion of it. + + 11. Patents. + + A "contributor" is a copyright holder who authorizes use under this +License of the Program or a work on which the Program is based. The +work thus licensed is called the contributor's "contributor version". + + A contributor's "essential patent claims" are all patent claims +owned or controlled by the contributor, whether already acquired or +hereafter acquired, that would be infringed by some manner, permitted +by this License, of making, using, or selling its contributor version, +but do not include claims that would be infringed only as a +consequence of further modification of the contributor version. For +purposes of this definition, "control" includes the right to grant +patent sublicenses in a manner consistent with the requirements of +this License. + + Each contributor grants you a non-exclusive, worldwide, royalty-free +patent license under the contributor's essential patent claims, to +make, use, sell, offer for sale, import and otherwise run, modify and +propagate the contents of its contributor version. + + In the following three paragraphs, a "patent license" is any express +agreement or commitment, however denominated, not to enforce a patent +(such as an express permission to practice a patent or covenant not to +sue for patent infringement). To "grant" such a patent license to a +party means to make such an agreement or commitment not to enforce a +patent against the party. + + If you convey a covered work, knowingly relying on a patent license, +and the Corresponding Source of the work is not available for anyone +to copy, free of charge and under the terms of this License, through a +publicly available network server or other readily accessible means, +then you must either (1) cause the Corresponding Source to be so +available, or (2) arrange to deprive yourself of the benefit of the +patent license for this particular work, or (3) arrange, in a manner +consistent with the requirements of this License, to extend the patent +license to downstream recipients. "Knowingly relying" means you have +actual knowledge that, but for the patent license, your conveying the +covered work in a country, or your recipient's use of the covered work +in a country, would infringe one or more identifiable patents in that +country that you have reason to believe are valid. + + If, pursuant to or in connection with a single transaction or +arrangement, you convey, or propagate by procuring conveyance of, a +covered work, and grant a patent license to some of the parties +receiving the covered work authorizing them to use, propagate, modify +or convey a specific copy of the covered work, then the patent license +you grant is automatically extended to all recipients of the covered +work and works based on it. + + A patent license is "discriminatory" if it does not include within +the scope of its coverage, prohibits the exercise of, or is +conditioned on the non-exercise of one or more of the rights that are +specifically granted under this License. You may not convey a covered +work if you are a party to an arrangement with a third party that is +in the business of distributing software, under which you make payment +to the third party based on the extent of your activity of conveying +the work, and under which the third party grants, to any of the +parties who would receive the covered work from you, a discriminatory +patent license (a) in connection with copies of the covered work +conveyed by you (or copies made from those copies), or (b) primarily +for and in connection with specific products or compilations that +contain the covered work, unless you entered into that arrangement, +or that patent license was granted, prior to 28 March 2007. + + Nothing in this License shall be construed as excluding or limiting +any implied license or other defenses to infringement that may +otherwise be available to you under applicable patent law. + + 12. No Surrender of Others' Freedom. + + If conditions are imposed on you (whether by court order, agreement or +otherwise) that contradict the conditions of this License, they do not +excuse you from the conditions of this License. If you cannot convey a +covered work so as to satisfy simultaneously your obligations under this +License and any other pertinent obligations, then as a consequence you may +not convey it at all. For example, if you agree to terms that obligate you +to collect a royalty for further conveying from those to whom you convey +the Program, the only way you could satisfy both those terms and this +License would be to refrain entirely from conveying the Program. + + 13. Use with the GNU Affero General Public License. + + Notwithstanding any other provision of this License, you have +permission to link or combine any covered work with a work licensed +under version 3 of the GNU Affero General Public License into a single +combined work, and to convey the resulting work. The terms of this +License will continue to apply to the part which is the covered work, +but the special requirements of the GNU Affero General Public License, +section 13, concerning interaction through a network will apply to the +combination as such. + + 14. Revised Versions of this License. + + The Free Software Foundation may publish revised and/or new versions of +the GNU General Public License from time to time. Such new versions will +be similar in spirit to the present version, but may differ in detail to +address new problems or concerns. + + Each version is given a distinguishing version number. If the +Program specifies that a certain numbered version of the GNU General +Public License "or any later version" applies to it, you have the +option of following the terms and conditions either of that numbered +version or of any later version published by the Free Software +Foundation. If the Program does not specify a version number of the +GNU General Public License, you may choose any version ever published +by the Free Software Foundation. + + If the Program specifies that a proxy can decide which future +versions of the GNU General Public License can be used, that proxy's +public statement of acceptance of a version permanently authorizes you +to choose that version for the Program. + + Later license versions may give you additional or different +permissions. However, no additional obligations are imposed on any +author or copyright holder as a result of your choosing to follow a +later version. + + 15. Disclaimer of Warranty. + + THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY +APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT +HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY +OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, +THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR +PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM +IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF +ALL NECESSARY SERVICING, REPAIR OR CORRECTION. + + 16. Limitation of Liability. + + IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING +WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS +THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY +GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE +USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF +DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD +PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS), +EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF +SUCH DAMAGES. + + 17. Interpretation of Sections 15 and 16. + + If the disclaimer of warranty and limitation of liability provided +above cannot be given local legal effect according to their terms, +reviewing courts shall apply local law that most closely approximates +an absolute waiver of all civil liability in connection with the +Program, unless a warranty or assumption of liability accompanies a +copy of the Program in return for a fee. + + END OF TERMS AND CONDITIONS + + How to Apply These Terms to Your New Programs + + If you develop a new program, and you want it to be of the greatest +possible use to the public, the best way to achieve this is to make it +free software which everyone can redistribute and change under these terms. + + To do so, attach the following notices to the program. It is safest +to attach them to the start of each source file to most effectively +state the exclusion of warranty; and each file should have at least +the "copyright" line and a pointer to where the full notice is found. + + + Copyright (C) + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . + +Also add information on how to contact you by electronic and paper mail. + + If the program does terminal interaction, make it output a short +notice like this when it starts in an interactive mode: + + Copyright (C) + This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'. + This is free software, and you are welcome to redistribute it + under certain conditions; type `show c' for details. + +The hypothetical commands `show w' and `show c' should show the appropriate +parts of the General Public License. Of course, your program's commands +might be different; for a GUI interface, you would use an "about box". + + You should also get your employer (if you work as a programmer) or school, +if any, to sign a "copyright disclaimer" for the program, if necessary. +For more information on this, and how to apply and follow the GNU GPL, see +. + + The GNU General Public License does not permit incorporating your program +into proprietary programs. If your program is a subroutine library, you +may consider it more useful to permit linking proprietary applications with +the library. If this is what you want to do, use the GNU Lesser General +Public License instead of this License. But first, please read +. diff --git a/README.md b/README.md new file mode 100644 index 0000000..b7a88d8 --- /dev/null +++ b/README.md @@ -0,0 +1,4 @@ +[![build](https://github.com/alterware/master-server/workflows/Build/badge.svg)](https://github.com/alterware/master-server/actions) + + +# AlterWare: Master Server diff --git a/deps/GSL b/deps/GSL new file mode 160000 index 0000000..afaaa71 --- /dev/null +++ b/deps/GSL @@ -0,0 +1 @@ +Subproject commit afaaa71bcee45d9c90c21f8bd3ba2b12902242e9 diff --git a/deps/curl b/deps/curl new file mode 160000 index 0000000..b16d1fa --- /dev/null +++ b/deps/curl @@ -0,0 +1 @@ +Subproject commit b16d1fa8ee567b52c09a0f89940b07d8491b881d diff --git a/deps/libtomcrypt b/deps/libtomcrypt new file mode 160000 index 0000000..fae62af --- /dev/null +++ b/deps/libtomcrypt @@ -0,0 +1 @@ +Subproject commit fae62af0ab16f469c2512ec04575dd60ca018657 diff --git a/deps/libtommath b/deps/libtommath new file mode 160000 index 0000000..0df542c --- /dev/null +++ b/deps/libtommath @@ -0,0 +1 @@ +Subproject commit 0df542cb70f621bbeec207be1949832fb1442479 diff --git a/deps/premake/curl.lua b/deps/premake/curl.lua new file mode 100644 index 0000000..4377e31 --- /dev/null +++ b/deps/premake/curl.lua @@ -0,0 +1,73 @@ +curl = { + source = path.join(dependencies.basePath, "curl"), +} + +function curl.import() + links { "curl" } + + filter "toolset:msc*" + links { "Crypt32.lib" } + filter {} + + curl.includes() +end + +function curl.includes() +filter "toolset:msc*" + includedirs { + path.join(curl.source, "include"), + } + + defines { + "CURL_STRICTER", + "CURL_STATICLIB", + "CURL_DISABLE_LDAP", + } +filter {} +end + +function curl.project() + if not os.istarget("windows") then + return + end + + project "curl" + language "C" + + curl.includes() + + includedirs { + path.join(curl.source, "lib"), + } + + files { + path.join(curl.source, "lib/**.c"), + path.join(curl.source, "lib/**.h"), + } + + defines { + "BUILDING_LIBCURL", + } + + filter "toolset:msc*" + + defines { + "USE_SCHANNEL", + "USE_WINDOWS_SSPI", + "USE_THREADS_WIN32", + } + + filter "toolset:not msc*" + + defines { + "USE_GNUTLS", + "USE_THREADS_POSIX", + } + + filter {} + + warnings "Off" + kind "StaticLib" +end + +table.insert(dependencies, curl) diff --git a/deps/premake/gsl.lua b/deps/premake/gsl.lua new file mode 100644 index 0000000..7a2daf6 --- /dev/null +++ b/deps/premake/gsl.lua @@ -0,0 +1,19 @@ +gsl = { + source = path.join(dependencies.basePath, "GSL"), +} + +function gsl.import() + gsl.includes() +end + +function gsl.includes() + includedirs { + path.join(gsl.source, "include") + } +end + +function gsl.project() + +end + +table.insert(dependencies, gsl) diff --git a/deps/premake/libtomcrypt.lua b/deps/premake/libtomcrypt.lua new file mode 100644 index 0000000..7ca3570 --- /dev/null +++ b/deps/premake/libtomcrypt.lua @@ -0,0 +1,64 @@ +libtomcrypt = { + source = path.join(dependencies.basePath, "libtomcrypt"), +} + +function libtomcrypt.import() + links { + "libtomcrypt" + } + + libtomcrypt.includes() +end + +function libtomcrypt.includes() + includedirs { + path.join(libtomcrypt.source, "src/headers") + } + + defines { + "LTC_NO_FAST", + "LTC_NO_PROTOTYPES", + "LTC_NO_RSA_BLINDING", + "LTC_NO_FILE", + "ARGTYPE=4", + } +end + +function libtomcrypt.project() + project "libtomcrypt" + language "C" + + libtomcrypt.includes() + libtommath.import() + + files { + path.join(libtomcrypt.source, "src/**.c"), + } + + removefiles { + path.join(libtomcrypt.source, "src/**/*_test.c"), + path.join(libtomcrypt.source, "src/**/*tab.c"), + path.join(libtomcrypt.source, "src/encauth/ocb3/**.c"), + } + + defines { + "_CRT_SECURE_NO_WARNINGS", + "LTC_SOURCE", + "_LIB", + "USE_LTM" + } + + removedefines { + "_DLL", + "_USRDLL" + } + + linkoptions { + "-IGNORE:4221" + } + + warnings "Off" + kind "StaticLib" +end + +table.insert(dependencies, libtomcrypt) diff --git a/deps/premake/libtommath.lua b/deps/premake/libtommath.lua new file mode 100644 index 0000000..35cdcf6 --- /dev/null +++ b/deps/premake/libtommath.lua @@ -0,0 +1,57 @@ +libtommath = { + source = path.join(dependencies.basePath, "libtommath"), +} + +function libtommath.import() + links { + "libtommath" + } + + libtommath.includes() +end + +function libtommath.includes() + includedirs { + libtommath.source + } + + if os.istarget("windows") then + defines { + "MP_NO_DEV_URANDOM", + } + end + + defines { + "LTM_DESC", + "__STDC_IEC_559__", + } +end + +function libtommath.project() + project "libtommath" + language "C" + + libtommath.includes() + + files { + path.join(libtommath.source, "*.c"), + } + + defines { + "_LIB" + } + + removedefines { + "_DLL", + "_USRDLL" + } + + linkoptions { + "-IGNORE:4221" + } + + warnings "Off" + kind "StaticLib" +end + +table.insert(dependencies, libtommath) diff --git a/deps/premake/minizip.lua b/deps/premake/minizip.lua new file mode 100644 index 0000000..21e30f0 --- /dev/null +++ b/deps/premake/minizip.lua @@ -0,0 +1,49 @@ +minizip = { + source = path.join(dependencies.basePath, "zlib/contrib/minizip"), +} + +function minizip.import() + links { "minizip" } + zlib.import() + minizip.includes() +end + +function minizip.includes() + includedirs { + minizip.source + } + + zlib.includes() +end + +function minizip.project() + project "minizip" + language "C" + + minizip.includes() + + files { + path.join(minizip.source, "*.h"), + path.join(minizip.source, "*.c"), + } + + filter "system:not windows" + removefiles { + path.join(minizip.source, "iowin32.c"), + } + filter {} + + removefiles { + path.join(minizip.source, "miniunz.c"), + path.join(minizip.source, "minizip.c"), + } + + defines { + "_CRT_SECURE_NO_DEPRECATE", + } + + warnings "Off" + kind "StaticLib" +end + +table.insert(dependencies, minizip) diff --git a/deps/premake/rapidjson.lua b/deps/premake/rapidjson.lua new file mode 100644 index 0000000..ee4929d --- /dev/null +++ b/deps/premake/rapidjson.lua @@ -0,0 +1,20 @@ +rapidjson = { + source = path.join(dependencies.basePath, "rapidjson"), +} + +function rapidjson.import() + defines{"RAPIDJSON_HAS_STDSTRING"} + rapidjson.includes() +end + +function rapidjson.includes() + includedirs { + path.join(rapidjson.source, "include"), + } +end + +function rapidjson.project() + +end + +table.insert(dependencies, rapidjson) diff --git a/deps/premake/zlib.lua b/deps/premake/zlib.lua new file mode 100644 index 0000000..566a707 --- /dev/null +++ b/deps/premake/zlib.lua @@ -0,0 +1,39 @@ +zlib = { + source = path.join(dependencies.basePath, "zlib"), +} + +function zlib.import() + links { "zlib" } + zlib.includes() +end + +function zlib.includes() + includedirs { + zlib.source + } + + defines { + "ZLIB_CONST", + } +end + +function zlib.project() + project "zlib" + language "C" + + zlib.includes() + + files { + path.join(zlib.source, "*.h"), + path.join(zlib.source, "*.c"), + } + + defines { + "_CRT_SECURE_NO_DEPRECATE", + } + + warnings "Off" + kind "StaticLib" +end + +table.insert(dependencies, zlib) diff --git a/deps/rapidjson b/deps/rapidjson new file mode 160000 index 0000000..973dc9c --- /dev/null +++ b/deps/rapidjson @@ -0,0 +1 @@ +Subproject commit 973dc9c06dcd3d035ebd039cfb9ea457721ec213 diff --git a/deps/zlib b/deps/zlib new file mode 160000 index 0000000..48c3741 --- /dev/null +++ b/deps/zlib @@ -0,0 +1 @@ +Subproject commit 48c3741002aca9dae84e9f2288ca149af14c9128 diff --git a/generate.bat b/generate.bat new file mode 100644 index 0000000..561de2a --- /dev/null +++ b/generate.bat @@ -0,0 +1,3 @@ +@echo off +git submodule update --init --recursive +tools\premake5 %* vs2022 diff --git a/generate.sh b/generate.sh new file mode 100755 index 0000000..9e96289 --- /dev/null +++ b/generate.sh @@ -0,0 +1,7 @@ +#!/bin/bash +git submodule update --init --recursive +if [ "$(uname)" == "Darwin" ]; then + ./tools/premake5-mac gmake2 +else + ./tools/premake5-linux gmake2 +fi diff --git a/premake5.lua b/premake5.lua new file mode 100644 index 0000000..662f9f5 --- /dev/null +++ b/premake5.lua @@ -0,0 +1,135 @@ +dependencies = { + basePath = "./deps" +} + +function dependencies.load() + dir = path.join(dependencies.basePath, "premake/*.lua") + deps = os.matchfiles(dir) + + for i, dep in pairs(deps) do + dep = dep:gsub(".lua", "") + require(dep) + end +end + +function dependencies.imports() + for i, proj in pairs(dependencies) do + if type(i) == 'number' then + proj.import() + end + end +end + +function dependencies.projects() + for i, proj in pairs(dependencies) do + if type(i) == 'number' then + proj.project() + end + end +end + +dependencies.load() + +workspace "alterware-master" +startproject "alterware-master" +location "./build" +objdir "%{wks.location}/obj" +targetdir "%{wks.location}/bin/%{cfg.platform}/%{cfg.buildcfg}" + +configurations {"Debug", "Release"} + +if os.istarget("darwin") then + platforms {"x64", "arm64"} +else + platforms {"x86", "x64"} +end + +filter "platforms:x86" +architecture "x86" +filter {} + +filter "platforms:x64" +architecture "x86_64" +filter {} + +filter "platforms:arm64" +architecture "ARM64" +filter {} + +filter {"language:C++", "toolset:not msc*"} + buildoptions { + "-std=c++17" + } +filter "toolset:msc*" + buildoptions { + "/std:c++17" + } +filter {} + +filter {"system:windows"} + systemversion "latest" +filter {} + +symbols "On" +staticruntime "On" +editandcontinue "Off" +warnings "Extra" +characterset "ASCII" + +if os.istarget("linux") or os.istarget("darwin") then + buildoptions "-pthread" + linkoptions "-pthread" +end + +if _OPTIONS["dev-build"] then + defines {"DEV_BUILD"} +end + +if os.getenv("CI") then + defines "CI" +end + +flags {"NoIncrementalLink", "NoMinimalRebuild", "MultiProcessorCompile", "No64BitChecks"} + +filter "configurations:Release" + optimize "Speed" + defines {"NDEBUG"} + flags {"FatalCompileWarnings"} +filter {} + +filter "configurations:Debug" + optimize "Debug" + defines {"DEBUG", "_DEBUG"} +filter {} + +project "alterware-master" +kind "ConsoleApp" +language "C++" + +pchheader "std_include.hpp" +pchsource "src/std_include.cpp" + +files {"./src/**.rc", "./src/**.hpp", "./src/**.cpp"} + +includedirs {"./src", "%{prj.location}/src"} + +filter "system:windows" + files { + "./src/**.rc", + } +filter { "system:windows", "toolset:not msc*" } + resincludedirs { + "%{_MAIN_SCRIPT_DIR}/src" + } +filter { "system:windows", "toolset:msc*" } + linkoptions {"/IGNORE:4254", "/SAFESEH:NO", "/LARGEADDRESSAWARE", "/PDBCompress"} + resincludedirs { + "$(ProjectDir)src" -- fix for VS IDE + } +filter {} + +dependencies.imports() + + +group "Dependencies" +dependencies.projects() diff --git a/src/client.hpp b/src/client.hpp new file mode 100644 index 0000000..f584370 --- /dev/null +++ b/src/client.hpp @@ -0,0 +1,23 @@ +#pragma once + +#include "utils/cryptography.hpp" +#include "game_server.hpp" + +struct client +{ + enum class state + { + can_authenticate = 0, + key_received, + challenge_sent, + }; + + uint64_t guid{0}; + bool registered{false}; + game_type game{game_type::unknown}; + state state{state::can_authenticate}; + utils::cryptography::ecc::key key{}; + std::string challenge{}; + std::string aes_key{}; + std::chrono::high_resolution_clock::time_point heartbeat{}; +}; diff --git a/src/client_list.cpp b/src/client_list.cpp new file mode 100644 index 0000000..804c67c --- /dev/null +++ b/src/client_list.cpp @@ -0,0 +1,38 @@ +#include "std_include.hpp" +#include "client_list.hpp" + +bool client_list::find_client(const uint64_t guid, const access_func& accessor) +{ + auto found = false; + + this->iterate([&](iteration_context& context) + { + auto& client = context.get(); + if (client.guid == guid) + { + accessor(client, context.get_address()); + context.stop_iterating(); + found = true; + } + }); + + return found; +} + +bool client_list::find_client(const uint64_t guid, const const_access_func& accessor) const +{ + auto found = false; + + this->iterate([&](const iteration_context& context) + { + const auto& client = context.get(); + if (client.guid == guid) + { + accessor(client, context.get_address()); + context.stop_iterating(); + found = true; + } + }); + + return found; +} diff --git a/src/client_list.hpp b/src/client_list.hpp new file mode 100644 index 0000000..d3d898f --- /dev/null +++ b/src/client_list.hpp @@ -0,0 +1,13 @@ +#pragma once + +#include "client.hpp" +#include "network_list.hpp" + +class client_list : public network_list +{ +public: + bool find_client(uint64_t guid, const access_func& accessor); + bool find_client(uint64_t guid, const const_access_func& accessor) const; + + using network_list::insert; +}; \ No newline at end of file diff --git a/src/console.cpp b/src/console.cpp new file mode 100644 index 0000000..f92acc2 --- /dev/null +++ b/src/console.cpp @@ -0,0 +1,248 @@ +#include "std_include.hpp" +#include "console.hpp" + +#ifdef _WIN32 +#define COLOR_LOG_INFO 11//15 +#define COLOR_LOG_WARN 14 +#define COLOR_LOG_ERROR 12 +#define COLOR_LOG_DEBUG 15//7 +#else +#define COLOR_LOG_INFO "\033[0;36m" +#define COLOR_LOG_WARN "\033[0;33m" +#define COLOR_LOG_ERROR "\033[0;31m" +#define COLOR_LOG_DEBUG "\033[0m" +#endif + +namespace console +{ + namespace + { + std::mutex signal_mutex; + std::function signal_callback; + +#ifdef _WIN32 +#define COLOR(win, posix) win + using color_type = WORD; +#else +#define COLOR(win, posix) posix + using color_type = const char*; +#endif + + const color_type color_array[] = + { + COLOR(0x8, "\033[0;90m"), // 0 - black + COLOR(0xC, "\033[0;91m"), // 1 - red + COLOR(0xA, "\033[0;92m"), // 2 - green + COLOR(0xE, "\033[0;93m"), // 3 - yellow + COLOR(0x9, "\033[0;94m"), // 4 - blue + COLOR(0xB, "\033[0;96m"), // 5 - cyan + COLOR(0xD, "\033[0;95m"), // 6 - pink + COLOR(0xF, "\033[0;97m"), // 7 - white + }; + +#ifdef _WIN32 + BOOL WINAPI handler(const DWORD signal) + { + if (signal == CTRL_C_EVENT && signal_callback) + { + signal_callback(); + } + + return TRUE; + } + +#else + void handler(int signal) + { + if (signal == SIGINT && signal_callback) + { + signal_callback(); + } + } +#endif + + std::string format(va_list* ap, const char* message) + { + static thread_local char buffer[0x1000]; + +#ifdef _WIN32 + const int count = _vsnprintf_s(buffer, sizeof(buffer), sizeof(buffer), message, *ap); +#else + const int count = vsnprintf(buffer, sizeof(buffer), message, *ap); +#endif + + if (count < 0) return {}; + return {buffer, static_cast(count)}; + } + +#ifdef _WIN32 + HANDLE get_console_handle() + { + return GetStdHandle(STD_OUTPUT_HANDLE); + } +#endif + + void set_color(const color_type color) + { +#ifdef _WIN32 + SetConsoleTextAttribute(get_console_handle(), color); +#else + printf("%s", color); +#endif + } + + bool apply_color(const std::string& data, const size_t index, const color_type base_color) + { + if (data[index] != '^' || (index + 1) >= data.size()) + { + return false; + } + + auto code = data[index + 1] - '0'; + if (code < 0 || code > 11) + { + return false; + } + + code = std::min(code, 7); // Everything above white is white + if (code == 7) + { + set_color(base_color); + } + else + { + set_color(color_array[code]); + } + + return true; + } + + void print_colored(const std::string& line, const color_type base_color) + { + lock _{}; + set_color(base_color); + + for (size_t i = 0; i < line.size(); ++i) + { + if (apply_color(line, i, base_color)) + { + ++i; + continue; + } + + putchar(line[i]); + } + + reset_color(); + } + } + + lock::lock() + { +#ifdef _WIN32 + _lock_file(stdout); +#else + flockfile(stdout); +#endif + } + + lock::~lock() + { +#ifdef _WIN32 + _unlock_file(stdout); +#else + funlockfile(stdout); +#endif + } + + void reset_color() + { + lock _{}; +#ifdef _WIN32 + SetConsoleTextAttribute(get_console_handle(), 7); +#else + printf("\033[0m"); +#endif + + fflush(stdout); + } + + void info(const char* message, ...) + { + va_list ap; + va_start(ap, message); + + const auto data = format(&ap, message); + print_colored("[+] " + data + "\n", COLOR_LOG_INFO); + + va_end(ap); + } + + void warn(const char* message, ...) + { + va_list ap; + va_start(ap, message); + + const auto data = format(&ap, message); + print_colored("[!] " + data + "\n", COLOR_LOG_WARN); + + va_end(ap); + } + + void error(const char* message, ...) + { + va_list ap; + va_start(ap, message); + + const auto data = format(&ap, message); + print_colored("[-] " + data + "\n", COLOR_LOG_ERROR); + + va_end(ap); + } + + void log(const char* message, ...) + { + va_list ap; + va_start(ap, message); + + const auto data = format(&ap, message); + print_colored("[*] " + data + "\n", COLOR_LOG_DEBUG); + + va_end(ap); + } + + void set_title(const std::string& title) + { + lock _{}; + +#ifdef _WIN32 + SetConsoleTitleA(title.data()); +#else + printf("\033]0;%s\007", title.data()); + fflush(stdout); +#endif + } + + signal_handler::signal_handler(std::function callback) + : std::lock_guard(signal_mutex) + { + signal_callback = std::move(callback); + +#ifdef _WIN32 + SetConsoleCtrlHandler(handler, TRUE); +#else + signal(SIGINT, handler); +#endif + } + + signal_handler::~signal_handler() + { +#ifdef _WIN32 + SetConsoleCtrlHandler(handler, FALSE); +#else + signal(SIGINT, SIG_DFL); +#endif + + signal_callback = {}; + } +} diff --git a/src/console.hpp b/src/console.hpp new file mode 100644 index 0000000..f655dd0 --- /dev/null +++ b/src/console.hpp @@ -0,0 +1,32 @@ +#pragma once + +namespace console +{ + class lock + { + public: + lock(); + ~lock(); + + lock(lock&&) = delete; + lock(const lock&) = delete; + lock& operator=(lock&&) = delete; + lock& operator=(const lock&) = delete; + }; + + void reset_color(); + + void info(const char* message, ...); + void warn(const char* message, ...); + void error(const char* message, ...); + void log(const char* message, ...); + + void set_title(const std::string& title); + + class signal_handler : std::lock_guard + { + public: + signal_handler(std::function callback); + ~signal_handler(); + }; +} diff --git a/src/crypto_key.cpp b/src/crypto_key.cpp new file mode 100644 index 0000000..d2fe2b1 --- /dev/null +++ b/src/crypto_key.cpp @@ -0,0 +1,74 @@ +#include "std_include.hpp" +#include "crypto_key.hpp" +#include "console.hpp" + +#include "utils/io.hpp" + +namespace crypto_key +{ + namespace + { + bool load_key(utils::cryptography::ecc::key& key) + { + std::string data{}; + if (!utils::io::read_file("./private.key", &data)) + { + return false; + } + + key.deserialize(data); + if (!key.is_valid()) + { + console::warn("Loaded key is invalid!"); + return false; + } + + return true; + } + + utils::cryptography::ecc::key generate_key() + { + auto key = utils::cryptography::ecc::generate_key(512); + if (!key.is_valid()) + { + throw std::runtime_error("Failed to generate server key!"); + } + + if (!utils::io::write_file("./private.key", key.serialize())) + { + throw std::runtime_error("Failed to write server key!"); + } + + console::info("Generated cryptographic key: %llX", key.get_hash()); + return key; + } + + utils::cryptography::ecc::key load_or_generate_key() + { + utils::cryptography::ecc::key key{}; + if (load_key(key)) + { + console::log("Loaded cryptographic key: %llX", key.get_hash()); + return key; + } + + return generate_key(); + } + + utils::cryptography::ecc::key get_key_internal() + { + auto key = load_or_generate_key(); + if (!utils::io::write_file("./public.key", key.get_public_key())) + { + console::error("Failed to write public key!"); + } + return key; + } + } + + const utils::cryptography::ecc::key& get() + { + static auto key = get_key_internal(); + return key; + } +} diff --git a/src/crypto_key.hpp b/src/crypto_key.hpp new file mode 100644 index 0000000..6f9f9c2 --- /dev/null +++ b/src/crypto_key.hpp @@ -0,0 +1,8 @@ +#pragma once + +#include + +namespace crypto_key +{ + const utils::cryptography::ecc::key& get(); +} diff --git a/src/game_server.hpp b/src/game_server.hpp new file mode 100644 index 0000000..7fc76bc --- /dev/null +++ b/src/game_server.hpp @@ -0,0 +1,73 @@ +#pragma once + +#include "utils/info_string.hpp" + +enum class game_type +{ + unknown = 0, + iw4, + iw6, + s1, + t7, +}; + +inline const std::string& resolve_game_type_name(const game_type game) +{ + static const std::unordered_map names = + { + {game_type::unknown, "Unknown"}, + {game_type::iw4, "IW4"}, + {game_type::iw6, "IW6"}, + {game_type::s1, "S1"}, + {game_type::t7, "T7"}, + }; + + return names.at(game); +} + +inline game_type resolve_game_type(const std::string& game_name) +{ + if (game_name == "IW4") + { + return game_type::iw4; + } + + if (game_name == "IW6") + { + return game_type::iw6; + } + + if (game_name == "S1") + { + return game_type::s1; + } + + if (game_name == "T7") + { + return game_type::t7; + } + + return game_type::unknown; +} + +struct game_server +{ + enum class state + { + can_ping = 0, + needs_ping, + pinged, + dead, + }; + + state state{state::can_ping}; + bool registered{false}; + + game_type game{game_type::unknown}; + int protocol{}; + uint32_t clients{}; + std::string name{}; + std::string challenge{}; + utils::info_string info_string{}; + std::chrono::high_resolution_clock::time_point heartbeat{}; +}; diff --git a/src/main.cpp b/src/main.cpp new file mode 100644 index 0000000..d04f2e6 --- /dev/null +++ b/src/main.cpp @@ -0,0 +1,47 @@ +#include + +#include "console.hpp" +#include "server.hpp" +#include "crypto_key.hpp" + +namespace +{ + void unsafe_main(const uint16_t port) + { + crypto_key::get(); + + console::log("Creating socket on port %hu", port); + + network::address a{"0.0.0.0"}; + a.set_port(port); + server s{a}; + + console::signal_handler handler([&s]() + { + s.stop(); + }); + + s.run(); + + console::log("Terminating server..."); + } +} + + +int main(const int argc, const char** argv) +{ + console::set_title("X Labs Master"); + console::log("Starting X Labs Master"); + + try + { + unsafe_main(argc > 1 ? static_cast(atoi(argv[1])) : 20810); + } + catch (std::exception& e) + { + console::error("Fatal error: %s\n", e.what()); + return -1; + } + + return 0; +} diff --git a/src/network/address.cpp b/src/network/address.cpp new file mode 100644 index 0000000..6eccc62 --- /dev/null +++ b/src/network/address.cpp @@ -0,0 +1,160 @@ +#include "std_include.hpp" + +#include "network/address.hpp" + +namespace network +{ + address::address() + { + ZeroMemory(&this->address_, sizeof(this->address_)); + } + + address::address(const std::string& addr) + : address() + { + this->parse(addr); + } + + address::address(sockaddr_in& addr) + { + this->address_ = addr; + } + + bool address::operator==(const address& obj) const + { + //return !std::memcmp(&this->address_, &obj.address_, sizeof(this->address_)); + return this->address_.sin_family == obj.address_.sin_family // + && this->address_.sin_addr.s_addr == obj.address_.sin_addr.s_addr // + && this->address_.sin_port == obj.address_.sin_port; + } + + void address::set_ipv4(const in_addr addr) + { + this->address_.sin_family = AF_INET; + this->address_.sin_addr = addr; + } + + void address::set_port(const unsigned short port) + { + this->address_.sin_port = htons(port); + } + + unsigned short address::get_port() const + { + return ntohs(this->address_.sin_port); + } + + std::string address::to_string(bool with_port) const + { + char buffer[1000] = {0}; + inet_ntop(this->address_.sin_family, &this->address_.sin_addr, buffer, sizeof(buffer)); + + auto address = std::string(buffer); + if (with_port) + { + address += ":"s + std::to_string(this->get_port()); + } + + return address; + } + + bool address::is_local() const + { + // According to: https://en.wikipedia.org/wiki/Private_network + + uint8_t bytes[4]; + *reinterpret_cast(&bytes) = this->address_.sin_addr.s_addr; + + // 10.X.X.X + if (bytes[0] == 10) + { + return true; + } + + // 192.168.X.X + if (bytes[0] == 192 + && bytes[1] == 168) + { + return true; + } + + // 172.16.X.X - 172.31.X.X + if (bytes[0] == 172 + && bytes[1] >= 16 + && bytes[1] < 32) + { + return true; + } + + // 127.0.0.1 + if (this->address_.sin_addr.s_addr == 0x0100007F) + { + return true; + } + + return false; + } + + sockaddr& address::get_addr() + { + return reinterpret_cast(this->get_in_addr()); + } + + const sockaddr& address::get_addr() const + { + return reinterpret_cast(this->get_in_addr()); + } + + sockaddr_in& address::get_in_addr() + { + return this->address_; + } + + const sockaddr_in& address::get_in_addr() const + { + return this->address_; + } + + void address::parse(std::string addr) + { + const auto pos = addr.find_last_of(':'); + if (pos != std::string::npos) + { + auto port = addr.substr(pos + 1); + this->set_port(uint16_t(atoi(port.data()))); + + addr = addr.substr(0, pos); + } + + this->resolve(addr); + } + + void address::resolve(const std::string& hostname) + { + addrinfo* result = nullptr; + if (!getaddrinfo(hostname.data(), nullptr, nullptr, &result)) + { + for (auto* i = result; i; i = i->ai_next) + { + if (i->ai_addr->sa_family == AF_INET) + { + const auto port = this->get_port(); + std::memcpy(&this->address_, i->ai_addr, sizeof(this->address_)); + this->set_port(port); + + break; + } + } + + freeaddrinfo(result); + } + } +} + +std::size_t std::hash::operator()(const network::address& a) const noexcept +{ + const auto h1 = std::hash{}(a.get_in_addr().sin_family); + const auto h2 = std::hash{}(a.get_in_addr().sin_addr.s_addr); + const auto h3 = std::hash{}(a.get_in_addr().sin_port); + return h1 ^ (h2 << 1) ^ (h3 << 2); +} diff --git a/src/network/address.hpp b/src/network/address.hpp new file mode 100644 index 0000000..bc2b579 --- /dev/null +++ b/src/network/address.hpp @@ -0,0 +1,46 @@ +#pragma once + +namespace network +{ + class address + { + public: + address(); + address(const std::string& addr); + address(sockaddr_in& addr); + + void set_ipv4(in_addr addr); + void set_port(unsigned short port); + [[nodiscard]] unsigned short get_port() const; + + sockaddr& get_addr(); + const sockaddr& get_addr() const; + sockaddr_in& get_in_addr(); + const sockaddr_in& get_in_addr() const; + + [[nodiscard]] bool is_local() const; + [[nodiscard]] std::string to_string(bool with_port = true) const; + + bool operator==(const address& obj) const; + + bool operator!=(const address& obj) const + { + return !(*this == obj); + } + + private: + sockaddr_in address_{}; + + void parse(std::string addr); + void resolve(const std::string& hostname); + }; +} + +namespace std +{ + template <> + struct hash + { + std::size_t operator()(const network::address& a) const noexcept; + }; +} diff --git a/src/network/socket.cpp b/src/network/socket.cpp new file mode 100644 index 0000000..8b7bc56 --- /dev/null +++ b/src/network/socket.cpp @@ -0,0 +1,129 @@ +#include "std_include.hpp" + +#include "network/socket.hpp" + +namespace network +{ + namespace + { +#ifdef _WIN32 + [[maybe_unused]] class wsa_initializer + { + public: + wsa_initializer() + { + WSADATA wsa_data; + if (WSAStartup(MAKEWORD(2, 2), &wsa_data)) + { + throw std::runtime_error("Unable to initialize WSA"); + } + } + + ~wsa_initializer() + { + WSACleanup(); + } + } _; +#endif + } + + socket::socket() + { + this->socket_ = ::socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); + } + + socket::~socket() + { + if (this->socket_ != INVALID_SOCKET) + { +#ifdef _WIN32 + closesocket(this->socket_); +#else + close(this->socket_); +#endif + } + } + + socket::socket(socket&& obj) noexcept + { + this->operator=(std::move(obj)); + } + + socket& socket::operator=(socket&& obj) noexcept + { + if (this != &obj) + { + this->~socket(); + this->socket_ = obj.socket_; + obj.socket_ = INVALID_SOCKET; + } + + return *this; + } + + bool socket::bind(const address& target) + { + return ::bind(this->socket_, &target.get_addr(), sizeof(target.get_addr())) == 0; + } + + void socket::send(const address& target, const std::string& data) const + { + sendto(this->socket_, data.data(), static_cast(data.size()), 0, &target.get_addr(), + sizeof(target.get_addr())); + } + + bool socket::receive(address& source, std::string& data) const + { + char buffer[0x2000]; + socklen_t len = sizeof(source.get_in_addr()); + + const auto result = recvfrom(this->socket_, buffer, sizeof(buffer), 0, &source.get_addr(), &len); + if (result == SOCKET_ERROR) // Probably WSAEWOULDBLOCK + { + return false; + } + + data.assign(buffer, buffer + result); + return true; + } + + bool socket::set_blocking(const bool blocking) + { +#ifdef _WIN32 + unsigned long mode = blocking ? 0 : 1; + return ioctlsocket(this->socket_, FIONBIO, &mode) == 0; +#else + int flags = fcntl(this->socket_, F_GETFL, 0); + if (flags == -1) return false; + flags = blocking ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK); + return fcntl(this->socket_, F_SETFL, flags) == 0; +#endif + } + + bool socket::sleep(const std::chrono::milliseconds timeout) const + { + fd_set fdr; + FD_ZERO(&fdr); + FD_SET(this->socket_, &fdr); + + const auto msec = timeout.count(); + + timeval tv{}; + tv.tv_sec = static_cast(msec / 1000ll); + tv.tv_usec = static_cast((msec % 1000) * 1000); + + const auto retval = select(static_cast(this->socket_) + 1, &fdr, nullptr, nullptr, &tv); + if (retval == SOCKET_ERROR) + { + std::this_thread::sleep_for(1ms); + return socket_is_ready; + } + + if (retval > 0) + { + return socket_is_ready; + } + + return !socket_is_ready; + } +} diff --git a/src/network/socket.hpp b/src/network/socket.hpp new file mode 100644 index 0000000..d1959de --- /dev/null +++ b/src/network/socket.hpp @@ -0,0 +1,40 @@ +#pragma once + +#include "network/address.hpp" + +namespace network +{ + class socket + { + public: + socket(); + ~socket(); + + socket(const socket& obj) = delete; + socket& operator=(const socket& obj) = delete; + + socket(socket&& obj) noexcept; + socket& operator=(socket&& obj) noexcept; + + bool bind(const address& target); + + void send(const address& target, const std::string& data) const; + bool receive(address& source, std::string& data) const; + + bool set_blocking(bool blocking); + + static const bool socket_is_ready = true; + bool sleep(std::chrono::milliseconds timeout) const; + + private: +#ifdef _WIN32 + using socklen_t = int; +#else + using SOCKET = int; +#define INVALID_SOCKET (SOCKET)(~0) +#define SOCKET_ERROR (-1) +#endif + + SOCKET socket_ = INVALID_SOCKET; + }; +} diff --git a/src/network_list.hpp b/src/network_list.hpp new file mode 100644 index 0000000..970bb17 --- /dev/null +++ b/src/network_list.hpp @@ -0,0 +1,172 @@ +#pragma once + +#include + +#include "console.hpp" +#include "network/address.hpp" +#include "utils/concurrency.hpp" + +template +class network_list +{ +public: + class iteration_context + { + public: + friend network_list; + + const network::address& get_address() const { return *this->address_; } + T& get() { return *this->element_; } + const T& get() const { return *this->element_; } + void remove() { this->remove_ = true; } + void stop_iterating() const { this->stop_iterating_ = true; } + + private: + T* element_{}; + const network::address* address_{}; + bool remove_{false}; + mutable bool stop_iterating_{false}; + }; + + using iterate_func = std::function; + using const_iterate_func = std::function; + + using access_func = std::function; + using const_access_func = std::function; + + using insert_func = std::function; + + bool find(const network::address& address, const access_func& accessor) + { + return this->elements_.template access([&](list_type& list) + { + const auto i = list.find(address); + if (i == list.end()) + { + return false; + } + + accessor(i->second, i->first); + return true; + }); + } + + bool find(const network::address& address, const const_access_func& accessor) const + { + return this->elements_.template access([&](const list_type& list) + { + const auto i = list.find(address); + if (i == list.end()) + { + return false; + } + + accessor(i->second, i->first); + return true; + }); + } + + void iterate(const iterate_func& iterator) + { + this->elements_.access([&](list_type& list) + { + iteration_context context{}; + + for (auto i = list.begin(); i != list.end();) + { + context.element_ = &i->second; + context.address_ = &i->first; + context.remove_ = false; + + iterator(context); + + if (context.remove_) + { + i = list.erase(i); + } + else + { + ++i; + } + + if (context.stop_iterating_) + { + break; + } + } + }); + } + + void iterate(const const_iterate_func& iterator) const + { + this->elements_.access([&](const list_type& list) + { + iteration_context context{}; + + for (const auto& server : list) + { + // const_cast is ok here + context.element_ = const_cast(&server.second); + context.address_ = &server.first; + + iterator(context); + + if (context.stop_iterating_) + { + break; + } + } + }); + } + +protected: + void insert(const network::address& address, const insert_func& callback) + { + this->elements_.access([&](list_type& list) + { + auto entry = list.find(address); + if (entry != list.end()) + { + callback(entry->second); + return; + } + + if (!is_insertion_allowed(list, address)) + { + console::log("Insertion rejected for target:\t%s", address.to_string().data()); + return; + } + + callback(list[address]); + }); + } + +private: + using list_type = std::unordered_map; + utils::concurrency::container elements_; + + static bool is_insertion_allowed(const list_type& list, const network::address& address) + { + if constexpr (IPLimit == 0) + { + return true; + } + + const auto target_ip = address.get_in_addr().sin_addr.s_addr; + + size_t occurrences = 0; + for (const auto& entry : list) + { + const auto entry_ip = entry.first.get_in_addr().sin_addr.s_addr; + if (entry_ip == target_ip) + { + if (++occurrences >= IPLimit) + { + return false; + } + } + } + + return true; + } +}; diff --git a/src/server.cpp b/src/server.cpp new file mode 100644 index 0000000..77f293a --- /dev/null +++ b/src/server.cpp @@ -0,0 +1,105 @@ +#include "std_include.hpp" +#include "server.hpp" +#include "service.hpp" +#include "console.hpp" + +#include "services/getbots_command.hpp" +#include "services/getservers_command.hpp" +#include "services/heartbeat_command.hpp" +#include "services/info_response_command.hpp" +#include "services/patch_kill_list_command.hpp" +#include "services/ping_handler.hpp" +#include "services/elimination_handler.hpp" +#include "services/statistics_handler.hpp" +#include "services/kill_list.hpp" + +server::server(const network::address& bind_addr) + : server_base(bind_addr) +{ + this->register_service(); + this->register_service(); + this->register_service(); + this->register_service(); + this->register_service(); + this->register_service(); + this->register_service(); + this->register_service(); + this->register_service(); +} + +server_list& server::get_server_list() +{ + return this->server_list_; +} + +const server_list& server::get_server_list() const +{ + return this->server_list_; +} + +client_list& server::get_client_list() +{ + return this->client_list_; +} + +const client_list& server::get_client_list() const +{ + return this->client_list_; +} + +void server::run_frame() +{ + for (const auto& service : services_) + { + try + { + service->run_frame(); + } + catch (const service::execution_exception& e) + { + console::warn("Execption in service: %s", e.what()); + } + catch (const std::exception& e) + { + console::error("Fatal execption in service: %s", e.what()); + } + } +} + +void server::handle_command(const network::address& target, const std::string_view& command, + const std::string_view& data) +{ + const auto handler = this->command_services_.find(std::string{command}); + if (handler == this->command_services_.end()) + { + console::warn("Unhandled command (%s): %.*s", target.to_string().data(), command.size(), command.data()); + return; + } + + std::string ban_reason; + if (this->get_service()->contains(target, ban_reason)) + { + console::log("Refused command from server %s - target is on the kill list (%s)", target.to_string().data(), ban_reason.data()); + return; + } + +#ifdef DEBUG + console::log("Handling command (%s): %.*s - %.*s", target.to_string().data(), command.size(), command.data(), + data.size(), data.data()); +#endif + + try + { + handler->second->handle_command(target, data); + } + catch (const service::execution_exception& e) + { + console::warn("Execption in command %.*s (%s): %s", command.size(), command.data(), target.to_string().data(), + e.what()); + } + catch (const std::exception& e) + { + console::error("Fatal execption in command %.*s (%s): %s", command.size(), command.data(), + target.to_string().data(), e.what()); + } +} diff --git a/src/server.hpp b/src/server.hpp new file mode 100644 index 0000000..71d8ea5 --- /dev/null +++ b/src/server.hpp @@ -0,0 +1,61 @@ +#pragma once + +#include "server_base.hpp" +#include "server_list.hpp" +#include "client_list.hpp" +#include "service.hpp" + +class server : public server_base +{ +public: + server(const network::address& bind_addr); + + server_list& get_server_list(); + const server_list& get_server_list() const; + + client_list& get_client_list(); + const client_list& get_client_list() const; + + template + T* get_service() + { + static_assert(std::is_base_of_v, "Type must be a service!"); + + for (auto& service : this->services_) + { + const auto& service_ref = *service; + if (typeid(service_ref) == typeid(T)) + { + return reinterpret_cast(service.get()); + } + } + + return nullptr; + } + +private: + server_list server_list_; + client_list client_list_; + + std::vector> services_; + std::unordered_map command_services_; + + template + void register_service(Args&&... args) + { + static_assert(std::is_base_of_v, "Type must be a service!"); + + auto service = std::make_unique(*this, std::forward(args)...); + auto* const command = service->get_command(); + if (command) + { + command_services_[command] = service.get(); + } + + services_.emplace_back(std::move(service)); + } + + void run_frame() override; + void handle_command(const network::address& target, const std::string_view& command, + const std::string_view& data) override; +}; diff --git a/src/server_base.cpp b/src/server_base.cpp new file mode 100644 index 0000000..cd63e3d --- /dev/null +++ b/src/server_base.cpp @@ -0,0 +1,114 @@ +#include "std_include.hpp" +#include "server_base.hpp" +#include "console.hpp" + +namespace +{ + bool is_command(const std::string_view& data) + { + return data.size() > 4 && *reinterpret_cast(data.data()) == -1; + } + + int find_separator(const std::string_view& data) + { + for (size_t i = 4; i < data.size(); ++i) + { + const auto& chr = data[i]; + + if (chr == ' ' || chr == '\n' || chr == '\0') + { + return static_cast(i); + } + } + + return -1; + } +} + +server_base::server_base(const network::address& bind_addr) +{ + if (!this->socket_.bind(bind_addr)) + { + throw std::runtime_error("Failed to bind socket!"); + } + + this->socket_.set_blocking(false); +} + +void server_base::run() +{ + this->stopped_ = false; + std::thread thread = std::thread([this] + { + std::this_thread::sleep_for(30ms); + this->run_socket(); + }); + + while (!this->stopped_) + { + this->run_frame(); + std::this_thread::sleep_for(100ms); + } + + if (thread.joinable()) + { + thread.join(); + } +} + +void server_base::stop() +{ + stopped_ = true; +} + +void server_base::send(const network::address& target, const std::string& command, const std::string& data, + const std::string& separator) const +{ + this->socket_.send(target, "\xFF\xFF\xFF\xFF" + command + separator + data); +} + +void server_base::run_socket() +{ + while (!this->stopped_) + { + if (!this->receive_data()) + { + (void)this->socket_.sleep(100ms); + } + } +} + +bool server_base::receive_data() +{ + std::string data{}; + network::address address{}; + + if (!this->socket_.receive(address, data)) + { + return false; + } + + if (!is_command(data)) + { + console::warn("Received invalid data from: %s", address.to_string().data()); + return false; + } + + this->parse_data(address, std::string_view{data.data() + 4, data.size() - 4}); + + return true; +} + +void server_base::parse_data(const network::address& target, const std::string_view& data) +{ + const auto separator = find_separator(data); + if (separator <= 0) + { + this->handle_command(target, data, {}); + } + else + { + this->handle_command(target, std::string_view{data.data(), static_cast(separator)}, + std::string_view{data.data() + (separator + 1), data.size() - (separator + 1)}); + } +} diff --git a/src/server_base.hpp b/src/server_base.hpp new file mode 100644 index 0000000..9d4c3fc --- /dev/null +++ b/src/server_base.hpp @@ -0,0 +1,30 @@ +#pragma once + +#include "network/socket.hpp" + + +class server_base +{ +public: + server_base(const network::address& bind_addr); + virtual ~server_base() = default; + + void run(); + void stop(); + + void send(const network::address& target, const std::string& command, const std::string& data, + const std::string& separator = " ") const; + +protected: + virtual void run_frame() = 0; + virtual void handle_command(const network::address& target, const std::string_view& command, + const std::string_view& data) = 0; + +private: + network::socket socket_{}; + volatile bool stopped_ = false; + + void run_socket(); + [[nodiscard]] bool receive_data(); + void parse_data(const network::address& target, const std::string_view& data); +}; diff --git a/src/server_list.cpp b/src/server_list.cpp new file mode 100644 index 0000000..19d35e2 --- /dev/null +++ b/src/server_list.cpp @@ -0,0 +1,39 @@ +#include "std_include.hpp" +#include "server_list.hpp" + +void server_list::find_registered_servers(const game_type game, const int protocol, const access_func& accessor) +{ + this->iterate([&](iteration_context& context) + { + auto& server = context.get(); + if (server.registered && server.game == game && server.protocol == protocol) + { + accessor(server, context.get_address()); + } + }); +} + +void server_list::find_registered_servers(const game_type game, const int protocol, + const const_access_func& accessor) const +{ + this->iterate([&](const iteration_context& context) + { + const auto& server = context.get(); + if (server.registered && server.game == game && server.protocol == protocol) + { + accessor(server, context.get_address()); + } + }); +} + +void server_list::heartbeat(const network::address& address) +{ + this->insert(address, [](game_server& server) + { + if (server.state == game_server::state::can_ping) + { + server.heartbeat = std::chrono::high_resolution_clock::now(); + server.state = game_server::state::needs_ping; + } + }); +} diff --git a/src/server_list.hpp b/src/server_list.hpp new file mode 100644 index 0000000..37de262 --- /dev/null +++ b/src/server_list.hpp @@ -0,0 +1,13 @@ +#pragma once + +#include "game_server.hpp" +#include "network_list.hpp" + +class server_list : public network_list +{ +public: + void find_registered_servers(game_type game, int protocol, const access_func& accessor); + void find_registered_servers(game_type game, int protocol, const const_access_func& accessor) const; + + void heartbeat(const network::address& address); +}; diff --git a/src/service.hpp b/src/service.hpp new file mode 100644 index 0000000..b7bcc7c --- /dev/null +++ b/src/service.hpp @@ -0,0 +1,56 @@ +#pragma once + +#include "network/address.hpp" + +class server; + +class service +{ +public: + class execution_exception : public std::runtime_error + { + public: + using std::runtime_error::runtime_error; + }; + + service(server& server) + : server_(server) + { + } + + virtual ~service() = default; + + service(service&&) = delete; + service(const service&) = delete; + service& operator=(service&&) = delete; + service& operator=(const service&) = delete; + + [[nodiscard]] virtual const char* get_command() const { return nullptr; } + + virtual void handle_command([[maybe_unused]] const network::address& target, + [[maybe_unused]] const std::string_view& data) + { + } + + virtual void run_frame() + { + } + +protected: + [[nodiscard]] const server& get_server() const + { + return server_; + } + + server& get_server() + { + return server_; + } + +private: + server& server_; +}; + +// server and service have a cycle, but fuck, it's easier that way +// include guards should handle that +#include "server.hpp" diff --git a/src/services/elimination_handler.cpp b/src/services/elimination_handler.cpp new file mode 100644 index 0000000..4e6c75b --- /dev/null +++ b/src/services/elimination_handler.cpp @@ -0,0 +1,60 @@ +#include +#include "elimination_handler.hpp" + +constexpr auto T7_PROTOCOL = 7; + +constexpr size_t MAX_SERVERS_PER_GAME = 15; + +void elimination_handler::run_frame() +{ + std::unordered_map> server_count; + + auto now = std::chrono::high_resolution_clock::now(); + this->get_server().get_server_list().iterate([&](server_list::iteration_context& context) + { + auto& server = context.get(); + const auto diff = now - server.heartbeat; + + if ((server.state == game_server::state::pinged && diff > 2min) || + (server.state == game_server::state::can_ping && diff > 15min)) + { + context.remove(); + } + + if (server.game == game_type::unknown) + { + return; + } + + if (server.game == game_type::t7 && server.protocol < T7_PROTOCOL) + { +#ifdef _DEBUG + console::info("Removing T7 server '%s' because protocol %i is less than %i\n", + context.get_address().to_string().data(), server.protocol, T7_PROTOCOL); +#endif + context.remove(); + } + + ++server_count[server.game][context.get_address()]; + if (server_count[server.game][context.get_address()] >= MAX_SERVERS_PER_GAME) + { +#ifdef _DEBUG + console::info("Removing server '%s' because it exceeds MAX_SERVERS_PER_GAME\n", + context.get_address().to_string().data()); +#endif + context.remove(); + } + }); + + now = std::chrono::high_resolution_clock::now(); + this->get_server().get_client_list().iterate([&](client_list::iteration_context& context) + { + auto& client = context.get(); + const auto diff = now - client.heartbeat; + + if (diff > 5min || (!client.registered && diff > 20s)) + { + context.remove(); + } + }); +} diff --git a/src/services/elimination_handler.hpp b/src/services/elimination_handler.hpp new file mode 100644 index 0000000..c10c16d --- /dev/null +++ b/src/services/elimination_handler.hpp @@ -0,0 +1,11 @@ +#pragma once + +#include "../service.hpp" + +class elimination_handler : public service +{ +public: + using service::service; + + void run_frame() override; +}; diff --git a/src/services/getbots_command.cpp b/src/services/getbots_command.cpp new file mode 100644 index 0000000..e12be95 --- /dev/null +++ b/src/services/getbots_command.cpp @@ -0,0 +1,56 @@ +#include +#include "getbots_command.hpp" + +#include "../console.hpp" + +const char* getbots_command::get_command() const +{ + return "getbots"; +} + +void getbots_command::handle_command(const network::address& target, const std::string_view&) +{ + static const std::vector bot_names + { + "aerosoul", + "Eldor", + "FutureRave", + "Girl", + "INeedBots", + "INeedGames", + "Infamous", + "Jebus3211", + "Joel", + "JTAG", + "Laupetin", + "Louvenarde", + "OneFourOne", + "PeterG", + "quaK", + "RezTech", + "sass", + "Slykuiper", + "st0rm", + "xensik", + "xoxor4d", + "Diamante", + "Dsso", + "Evan", + "FragsAreUs", + "FryTechTip", + "H3X1C", + "homura", + "Jimbo", + "RektInator", + "Squirrel", + }; + + std::stringstream stream{}; + for (const auto& bot : bot_names) + { + stream << bot << std::endl; + } + + this->get_server().send(target, "getbotsResponse", stream.str()); + console::log("Sent bot names: %s", target.to_string().data()); +} diff --git a/src/services/getbots_command.hpp b/src/services/getbots_command.hpp new file mode 100644 index 0000000..6da6466 --- /dev/null +++ b/src/services/getbots_command.hpp @@ -0,0 +1,12 @@ +#pragma once + +#include "../service.hpp" + +class getbots_command : public service +{ +public: + using service::service; + + const char* get_command() const override; + void handle_command(const network::address& target, const std::string_view& data) override; +}; diff --git a/src/services/getservers_command.cpp b/src/services/getservers_command.cpp new file mode 100644 index 0000000..ad7d331 --- /dev/null +++ b/src/services/getservers_command.cpp @@ -0,0 +1,96 @@ +#include +#include "getservers_command.hpp" + +#include "../console.hpp" + +#include + +namespace +{ + struct prepared_server + { + uint32_t address; + uint16_t port; + }; +} + +constexpr auto MTU = 1400; // Real UDP MTU is more like 1500 bytes, but we keep a little wiggle room just in case +constexpr auto DPM_PROTOCOL_ADDRESS_LENGTH = sizeof prepared_server::address; +constexpr auto DPM_PROTOCOL_PORT_LENGTH = sizeof prepared_server::port; + +static_assert(DPM_PROTOCOL_ADDRESS_LENGTH == 4); +static_assert(DPM_PROTOCOL_PORT_LENGTH == 2); + +const char* getservers_command::get_command() const +{ + return "getservers"; +} + +void getservers_command::handle_command(const network::address& target, const std::string_view& data) +{ + const utils::parameters params(data); + if (params.size() < 2) + { + throw execution_exception("Invalid parameter count"); + } + + const auto& game = params[0]; + + const auto* p = params[1].data(); + char* end; + const auto protocol = strtol(params[1].data(), &end, 10); + if (p == end) + { + throw execution_exception("Invalid protocol"); + } + + const auto game_type = resolve_game_type(game); + if (game_type == game_type::unknown) + { + throw execution_exception("Invalid game type: " + game); + } + + std::queue prepared_servers{}; + + this->get_server().get_server_list() // + .find_registered_servers(game_type, protocol, + [&prepared_servers](const game_server&, const network::address& address) + { + const auto addr = address.get_in_addr().sin_addr.s_addr; + const auto port = htons(address.get_port()); + + prepared_servers.push({ addr, port }); + }); + + size_t packet_count = 0; + std::string response{}; + + while (!prepared_servers.empty()) + { + const auto& server = prepared_servers.front(); + response.push_back('\\'); + response.append(reinterpret_cast(&server.address), DPM_PROTOCOL_ADDRESS_LENGTH); + response.append(reinterpret_cast(&server.port), DPM_PROTOCOL_PORT_LENGTH); + prepared_servers.pop(); + + if (response.size() >= MTU || prepared_servers.empty()) + { + // Only send EOT if the queue is empty (last packet) + if (prepared_servers.empty()) + { + response.push_back('\\'); + response.append("EOT"); + response.push_back('\0'); + response.push_back('\0'); + response.push_back('\0'); + } + + this->get_server().send(target, "getserversResponse", response); + packet_count++; + + response.clear(); + } + } + + console::log("Sent %zu servers in %zu parts for game %s:\t%s", prepared_servers.size(), packet_count, game.data(), target.to_string().data()); +} diff --git a/src/services/getservers_command.hpp b/src/services/getservers_command.hpp new file mode 100644 index 0000000..09f8dd0 --- /dev/null +++ b/src/services/getservers_command.hpp @@ -0,0 +1,12 @@ +#pragma once + +#include "../service.hpp" + +class getservers_command : public service +{ +public: + using service::service; + + const char* get_command() const override; + void handle_command(const network::address& target, const std::string_view& data) override; +}; diff --git a/src/services/heartbeat_command.cpp b/src/services/heartbeat_command.cpp new file mode 100644 index 0000000..adabbda --- /dev/null +++ b/src/services/heartbeat_command.cpp @@ -0,0 +1,12 @@ +#include +#include "heartbeat_command.hpp" + +const char* heartbeat_command::get_command() const +{ + return "heartbeat"; +} + +void heartbeat_command::handle_command(const network::address& target, [[maybe_unused]] const std::string_view& data) +{ + this->get_server().get_server_list().heartbeat(target); +} diff --git a/src/services/heartbeat_command.hpp b/src/services/heartbeat_command.hpp new file mode 100644 index 0000000..870dd6b --- /dev/null +++ b/src/services/heartbeat_command.hpp @@ -0,0 +1,12 @@ +#pragma once + +#include "../service.hpp" + +class heartbeat_command : public service +{ +public: + using service::service; + + const char* get_command() const override; + void handle_command(const network::address& target, const std::string_view& data) override; +}; diff --git a/src/services/info_response_command.cpp b/src/services/info_response_command.cpp new file mode 100644 index 0000000..c633745 --- /dev/null +++ b/src/services/info_response_command.cpp @@ -0,0 +1,60 @@ +#include +#include "info_response_command.hpp" + +#include "../console.hpp" + +const char* info_response_command::get_command() const +{ + return "infoResponse"; +} + +void info_response_command::handle_command(const network::address& target, const std::string_view& data) +{ + const auto found = this->get_server().get_server_list().find( + target, [&data](game_server& server, const network::address& address) + { + utils::info_string info{data}; + const auto game = info.get("gamename"); + const auto challenge = info.get("challenge"); + + const auto game_type = resolve_game_type(game); + + if (game_type == game_type::unknown) + { + server.state = game_server::state::dead; + throw execution_exception{"Invalid game type: " + game}; + } + + if (server.state != game_server::state::pinged) + { + throw execution_exception{"Stray info response"}; + } + + if (challenge != server.challenge) + { + throw execution_exception{"Invalid challenge"}; + } + + const auto player_count = atoi(info.get("clients").data()); + const auto bot_count = atoi(info.get("bots").data()); + auto real_player_count = player_count - bot_count; + real_player_count = std::clamp(real_player_count, 0, 18); + + server.registered = true; + server.game = game_type; + server.state = game_server::state::can_ping; + server.protocol = atoi(info.get("protocol").data()); + server.clients = static_cast(real_player_count); + server.name = info.get("hostname"); + server.heartbeat = std::chrono::high_resolution_clock::now(); + server.info_string = std::move(info); + + console::log("Server registered for game %s (%i):\t%s\t- %s", game.data(), server.protocol, + address.to_string().data(), server.name.data()); + }); + + if (!found) + { + throw execution_exception{"infoResponse without server!"}; + } +} diff --git a/src/services/info_response_command.hpp b/src/services/info_response_command.hpp new file mode 100644 index 0000000..ea20650 --- /dev/null +++ b/src/services/info_response_command.hpp @@ -0,0 +1,12 @@ +#pragma once + +#include "../service.hpp" + +class info_response_command : public service +{ +public: + using service::service; + + const char* get_command() const override; + void handle_command(const network::address& target, const std::string_view& data) override; +}; diff --git a/src/services/kill_list.cpp b/src/services/kill_list.cpp new file mode 100644 index 0000000..a349d95 --- /dev/null +++ b/src/services/kill_list.cpp @@ -0,0 +1,153 @@ +#include "std_include.hpp" +#include "kill_list.hpp" + +#include + +constexpr auto* kill_file = "./kill.txt"; + +kill_list::kill_list_entry::kill_list_entry(std::string ip_address, std::string reason) + : ip_address_(std::move(ip_address)), reason_(std::move(reason)) +{ +} + +bool kill_list::contains(const network::address& address, std::string& reason) +{ + auto str_address = address.to_string(false); + + return this->entries_container_.access([&str_address, &reason](const kill_list_entries& entries) + { + if (const auto itr = entries.find(str_address); itr != entries.end()) + { + reason = itr->second.reason_; + return true; + } + + return false; + }); +} + +void kill_list::add_to_kill_list(const kill_list_entry& add) +{ + const auto any_change = this->entries_container_.access([&add](kill_list_entries& entries) + { + const auto existing_entry = entries.find(add.ip_address_); + if (existing_entry == entries.end() || existing_entry->second.reason_ != add.reason_) + { + console::info("Added %s to kill list (reason: %s)", add.ip_address_.data(), add.reason_.data()); + entries[add.ip_address_] = add; + return true; + } + + return false; + }); + + if (!any_change) + { + console::info("%s already in kill list, doing nothing", add.ip_address_.data()); + return; + } + + this->write_to_disk(); +} + +void kill_list::remove_from_kill_list(const network::address& remove) +{ + this->remove_from_kill_list(remove.to_string()); +} + +void kill_list::remove_from_kill_list(const std::string& remove) +{ + const auto any_change = this->entries_container_.access([&remove](kill_list_entries& entries) + { + if (entries.erase(remove)) + { + console::info("Removed %s from kill list", remove.data()); + return true; + } + + return false; + }); + + if (!any_change) + { + console::info("%s not in kill list, doing nothing", remove.data()); + return; + } + + this->write_to_disk(); +} + +void kill_list::reload_from_disk() +{ + std::string contents; + if (!utils::io::read_file(kill_file, &contents)) + { + console::info("Could not find %s, kill list will not be loaded.", kill_file); + return; + } + + std::istringstream string_stream(contents); + std::string line; + + this->entries_container_.access([&string_stream, &line](kill_list_entries& entries) + { + entries.clear(); + while (std::getline(string_stream, line)) + { + if (line[0] == '#') + { + // comments or ignored line + continue; + } + + std::string ip; + std::string comment; + + const auto index = line.find(' '); + if (index != std::string::npos) + { + ip = line.substr(0, index); + comment = line.substr(index + 1); + } + else + { + ip = line; + } + + if (ip.empty()) + { + continue; + } + + // Double line breaks from windows' \r\n + if (ip[ip.size() - 1] == '\r') + { + ip.pop_back(); + } + + entries.emplace(ip, kill_list_entry(ip, comment)); + } + + console::info("Loaded %zu kill list entries from %s", entries.size(), kill_file); + }); +} + +void kill_list::write_to_disk() +{ + std::ostringstream stream; + this->entries_container_.access([&stream](const kill_list_entries& entries) + { + for (const auto& [ip, entry] : entries) + { + stream << entry.ip_address_ << " " << entry.reason_ << "\n"; + } + + utils::io::write_file(kill_file, stream.str(), false); + console::info("Wrote %s to disk (%zu entries)", kill_file, entries.size()); + }); +} + +kill_list::kill_list(server& server) : service(server) +{ + this->reload_from_disk(); +} diff --git a/src/services/kill_list.hpp b/src/services/kill_list.hpp new file mode 100644 index 0000000..97308ae --- /dev/null +++ b/src/services/kill_list.hpp @@ -0,0 +1,32 @@ +#pragma once +#include + +#include "../service.hpp" + +class kill_list : public service +{ +public: + class kill_list_entry + { + public: + kill_list_entry() = default; + kill_list_entry(std::string ip_address, std::string reason); + + std::string ip_address_; + std::string reason_; + }; + + kill_list(server& server); + + bool contains(const network::address& address, std::string& reason); + void add_to_kill_list(const kill_list_entry& add); + void remove_from_kill_list(const network::address& remove); + void remove_from_kill_list(const std::string& remove); + +private: + using kill_list_entries = std::unordered_map; + utils::concurrency::container entries_container_; + + void reload_from_disk(); + void write_to_disk(); +}; diff --git a/src/services/patch_kill_list_command.cpp b/src/services/patch_kill_list_command.cpp new file mode 100644 index 0000000..e8ed862 --- /dev/null +++ b/src/services/patch_kill_list_command.cpp @@ -0,0 +1,65 @@ +#include +#include "patch_kill_list_command.hpp" + +#include "crypto_key.hpp" +#include "services/kill_list.hpp" + +#include +#include +#include + +const char* patch_kill_list_command::get_command() const +{ + return "patchkill"; +} + +// patchkill timestamp signature add/remove target_ip (ban_reason) +void patch_kill_list_command::handle_command([[maybe_unused]] const network::address& target, const std::string_view& data) +{ + const utils::parameters params(data); + if (params.size() < 3) + { + throw execution_exception("Invalid parameter count"); + } + + const auto supplied_timestamp = std::chrono::seconds(std::stoul(params[0])); + const auto current_timestamp = std::chrono::duration_cast(std::chrono::system_clock::now().time_since_epoch()); + + // Abs the duration so that the client can be ahead or behind + const auto time_stretch = std::chrono::abs(current_timestamp - supplied_timestamp); + + // not offset by more than 5 minutes in either direction + if (time_stretch > 5min) + { + throw execution_exception(utils::string::va("Invalid timestamp supplied - expected %llu, got %llu, which is more than 5 minutes apart", current_timestamp.count(), supplied_timestamp.count())); + } + + const auto& signature = utils::cryptography::base64::decode(params[1]); + const auto should_remove = params[2] == "remove"s; + + if (!should_remove && params[2] != "add"s) + { + throw execution_exception("Invalid parameter #2: should be 'add' or 'remove'"); + } + + const auto supplied_reason = params.join(4); + const auto& crypto_key = crypto_key::get(); + const auto signature_candidate = std::to_string(supplied_timestamp.count()); + + if (!utils::cryptography::ecc::verify_message(crypto_key, signature_candidate, signature)) + { + throw execution_exception("Signature verification of the kill list patch key failed"); + } + + const auto kill_list_service = this->get_server().get_service(); + const auto& supplied_address = params[3]; + + if (should_remove) + { + kill_list_service->remove_from_kill_list(supplied_address); + } + else + { + kill_list_service->add_to_kill_list(kill_list::kill_list_entry(supplied_address, supplied_reason)); + } +} diff --git a/src/services/patch_kill_list_command.hpp b/src/services/patch_kill_list_command.hpp new file mode 100644 index 0000000..d7c5eb9 --- /dev/null +++ b/src/services/patch_kill_list_command.hpp @@ -0,0 +1,12 @@ +#pragma once + +#include "../service.hpp" + +class patch_kill_list_command : public service +{ +public: + using service::service; + + const char* get_command() const override; + void handle_command(const network::address& target, const std::string_view& data) override; +}; diff --git a/src/services/ping_handler.cpp b/src/services/ping_handler.cpp new file mode 100644 index 0000000..df4dfea --- /dev/null +++ b/src/services/ping_handler.cpp @@ -0,0 +1,26 @@ +#include +#include "ping_handler.hpp" + +#include + +void ping_handler::run_frame() +{ + auto count = 0; + this->get_server().get_server_list().iterate([&](server_list::iteration_context& context) + { + auto& server = context.get(); + if (server.state == game_server::state::needs_ping) + { + server.state = game_server::state::pinged; + server.challenge = utils::cryptography::random::get_challenge(); + server.heartbeat = std::chrono::high_resolution_clock::now(); + + this->get_server().send(context.get_address(), "getinfo", server.challenge); + + if (20 >= ++count) + { + context.stop_iterating(); + } + } + }); +} diff --git a/src/services/ping_handler.hpp b/src/services/ping_handler.hpp new file mode 100644 index 0000000..b8c8efd --- /dev/null +++ b/src/services/ping_handler.hpp @@ -0,0 +1,11 @@ +#pragma once + +#include "../service.hpp" + +class ping_handler : public service +{ +public: + using service::service; + + void run_frame() override; +}; diff --git a/src/services/statistics_handler.cpp b/src/services/statistics_handler.cpp new file mode 100644 index 0000000..0eda204 --- /dev/null +++ b/src/services/statistics_handler.cpp @@ -0,0 +1,86 @@ +#include +#include "statistics_handler.hpp" +#include "../console.hpp" + +#include +#include + +namespace +{ + void write_stats(const std::map>>& servers, + std::map& players) + { + rapidjson::Document root{}; + root.SetObject(); + + for (const auto& game_servers : servers) + { + const auto server_count = static_cast(game_servers.second.size()); + const auto player_count = players[game_servers.first]; + + rapidjson::Value game{}; + game.SetObject(); + + game.AddMember("servers", server_count, root.GetAllocator()); + game.AddMember("players", player_count, root.GetAllocator()); + + auto game_name = resolve_game_type_name(game_servers.first); + game_name = utils::string::to_lower(game_name); + + rapidjson::Value game_name_object(game_name, root.GetAllocator()); + root.AddMember(game_name_object, game, root.GetAllocator()); + } + + rapidjson::StringBuffer root_buffer{}; + rapidjson::Writer> + root_writer(root_buffer); + root.Accept(root_writer); + + std::string root_data(root_buffer.GetString(), root_buffer.GetLength()); + utils::io::write_file("/var/www/master.xlabs.dev/stats.json", root_data); + } +} + +statistics_handler::statistics_handler(server& server) + : service(server) + , last_print(std::chrono::high_resolution_clock::now()) +{ +} + +void statistics_handler::run_frame() +{ + const auto now = std::chrono::high_resolution_clock::now(); + if (now - this->last_print < 5min) + { + return; + } + + std::map players{}; + std::map>> servers; + + this->last_print = std::chrono::high_resolution_clock::now(); + this->get_server().get_server_list().iterate([&servers, &players](const server_list::iteration_context& context) + { + const auto& server = context.get(); + if (server.registered) + { + servers[server.game].emplace_back(server.name, context.get_address()); + players[server.game] += server.clients; + } + }); + + console::lock _{}; + + for (const auto& game_servers : servers) + { + console::log("%s (%d):", resolve_game_type_name(game_servers.first).data(), + static_cast(game_servers.second.size())); + + for (const auto& server : game_servers.second) + { + console::log("\t%s\t%s", server.second.to_string().data(), server.first.data()); + } + } + + write_stats(servers, players); +} diff --git a/src/services/statistics_handler.hpp b/src/services/statistics_handler.hpp new file mode 100644 index 0000000..bfa98b9 --- /dev/null +++ b/src/services/statistics_handler.hpp @@ -0,0 +1,14 @@ +#pragma once + +#include "../service.hpp" + +class statistics_handler : public service +{ +public: + statistics_handler(server& server); + + void run_frame() override; + +private: + std::chrono::high_resolution_clock::time_point last_print; +}; diff --git a/src/std_include.cpp b/src/std_include.cpp new file mode 100644 index 0000000..f601ac4 --- /dev/null +++ b/src/std_include.cpp @@ -0,0 +1,33 @@ +#include + +extern "C" +{ + int s_read_arc4random(void*, size_t) + { + return -1; + } + +#if !defined(__linux__) + int s_read_getrandom(void*, size_t) + { + return -1; + } +#endif + +#ifdef _WIN32 + int s_read_urandom(void*, size_t) + { + return -1; + } + + int s_read_ltm_rng(void*, size_t) + { + return -1; + } +#else + int s_read_wincsp(void*, size_t) + { + return -1; + } +#endif +} diff --git a/src/std_include.hpp b/src/std_include.hpp new file mode 100644 index 0000000..e610487 --- /dev/null +++ b/src/std_include.hpp @@ -0,0 +1,71 @@ +#ifdef _WIN32 +#pragma once + +#define WIN32_LEAN_AND_MEAN + +#include +#include +#include + +#else + +#include +#include +#include +#include +#include +#include +#include + +#define ZeroMemory(x, y) std::memset(x, 0, y) + +#endif + +// min and max is required by gdi, therefore NOMINMAX won't work +#ifdef max +#undef max +#endif + +#ifdef min +#undef min +#endif + +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include + +#ifdef _WIN32 + +#pragma comment(lib, "ntdll.lib") +#pragma comment(lib, "ws2_32.lib") +#pragma comment(lib, "urlmon.lib" ) +#pragma comment(lib, "iphlpapi.lib") + +#endif + +using namespace std::literals; diff --git a/src/utils/compression.cpp b/src/utils/compression.cpp new file mode 100644 index 0000000..f103eb0 --- /dev/null +++ b/src/utils/compression.cpp @@ -0,0 +1,170 @@ +#include + +#include "compression.hpp" + +#include +#include + +#include + +#include "io.hpp" + +namespace utils::compression +{ + namespace zlib + { + namespace + { + class zlib_stream + { + public: + zlib_stream() + { + memset(&stream_, 0, sizeof(stream_)); + valid_ = inflateInit(&stream_) == Z_OK; + } + + zlib_stream(zlib_stream&&) = delete; + zlib_stream(const zlib_stream&) = delete; + zlib_stream& operator=(zlib_stream&&) = delete; + zlib_stream& operator=(const zlib_stream&) = delete; + + ~zlib_stream() + { + if (valid_) + { + inflateEnd(&stream_); + } + } + + z_stream& get() + { + return stream_; // + } + + bool is_valid() const + { + return valid_; + } + + private: + bool valid_{false}; + z_stream stream_{}; + }; + } + + std::string decompress(const std::string& data) + { + std::string buffer{}; + zlib_stream stream_container{}; + if (!stream_container.is_valid()) + { + return {}; + } + + int ret{}; + size_t offset = 0; + static thread_local uint8_t dest[CHUNK] = {0}; + auto& stream = stream_container.get(); + + do + { + const auto input_size = std::min(sizeof(dest), data.size() - offset); + stream.avail_in = static_cast(input_size); + stream.next_in = reinterpret_cast(data.data()) + offset; + offset += stream.avail_in; + + do + { + stream.avail_out = sizeof(dest); + stream.next_out = dest; + + ret = inflate(&stream, Z_NO_FLUSH); + if (ret != Z_OK && ret != Z_STREAM_END) + { + return {}; + } + + buffer.insert(buffer.end(), dest, dest + sizeof(dest) - stream.avail_out); + } + while (stream.avail_out == 0); + } + while (ret != Z_STREAM_END); + + return buffer; + } + + std::string compress(const std::string& data) + { + std::string result{}; + auto length = compressBound(static_cast(data.size())); + result.resize(length); + + if (compress2(reinterpret_cast(result.data()), &length, + reinterpret_cast(data.data()), static_cast(data.size()), + Z_BEST_COMPRESSION) != Z_OK) + { + return {}; + } + + result.resize(length); + return result; + } + } + + namespace zip + { + namespace + { + bool add_file(zipFile& zip_file, const std::string& filename, const std::string& data) + { + const auto zip_64 = data.size() > 0xffffffff ? 1 : 0; + if (ZIP_OK != zipOpenNewFileInZip64(zip_file, filename.data(), nullptr, nullptr, 0, nullptr, 0, nullptr, + Z_DEFLATED, Z_BEST_COMPRESSION, zip_64)) + { + return false; + } + + const auto _ = gsl::finally([&zip_file]() + { + zipCloseFileInZip(zip_file); + }); + + return ZIP_OK == zipWriteInFileInZip(zip_file, data.data(), static_cast(data.size())); + } + } + + void archive::add(std::string filename, std::string data) + { + this->files_[std::move(filename)] = std::move(data); + } + + bool archive::write(const std::string& filename, const std::string& comment) + { + // Hack to create the directory :3 + io::write_file(filename, {}); + io::remove_file(filename); + + auto* zip_file = zipOpen64(filename.data(), 0); + if (!zip_file) + { + return false; + } + + const auto _ = gsl::finally([&zip_file, &comment]() + { + zipClose(zip_file, comment.empty() ? nullptr : comment.data()); + }); + + for (const auto& file : this->files_) + { + if (!add_file(zip_file, file.first, file.second)) + { + return false; + } + } + + return true; + } + } +} diff --git a/src/utils/compression.hpp b/src/utils/compression.hpp new file mode 100644 index 0000000..dfe36ad --- /dev/null +++ b/src/utils/compression.hpp @@ -0,0 +1,28 @@ +#pragma once + +#include +#include + +#define CHUNK 16384u + +namespace utils::compression +{ + namespace zlib + { + std::string compress(const std::string& data); + std::string decompress(const std::string& data); + } + + namespace zip + { + class archive + { + public: + void add(std::string filename, std::string data); + bool write(const std::string& filename, const std::string& comment = {}); + + private: + std::unordered_map files_; + }; + } +}; diff --git a/src/utils/concurrency.hpp b/src/utils/concurrency.hpp new file mode 100644 index 0000000..05c5d3a --- /dev/null +++ b/src/utils/concurrency.hpp @@ -0,0 +1,46 @@ +#pragma once + +#include + +namespace utils::concurrency +{ + template + class container + { + public: + template + R access(F&& accessor) const + { + std::lock_guard _{mutex_}; + return accessor(object_); + } + + template + R access(F&& accessor) + { + std::lock_guard _{mutex_}; + return accessor(object_); + } + + template + R access_with_lock(F&& accessor) const + { + std::unique_lock lock{mutex_}; + return accessor(object_, lock); + } + + template + R access_with_lock(F&& accessor) + { + std::unique_lock lock{mutex_}; + return accessor(object_, lock); + } + + T& get_raw() { return object_; } + const T& get_raw() const { return object_; } + + private: + mutable MutexType mutex_{}; + T object_{}; + }; +} diff --git a/src/utils/cryptography.cpp b/src/utils/cryptography.cpp new file mode 100644 index 0000000..b669c46 --- /dev/null +++ b/src/utils/cryptography.cpp @@ -0,0 +1,649 @@ +#include +#include + +#include "string.hpp" +#include "cryptography.hpp" + +/// http://www.opensource.apple.com/source/CommonCrypto/CommonCrypto-55010/Source/libtomcrypt/doc/libTomCryptDoc.pdf + +namespace utils::cryptography +{ + namespace + { + struct __ + { + __() + { + ltc_mp = ltm_desc; + + register_cipher(&aes_desc); + register_cipher(&des3_desc); + + register_prng(&sprng_desc); + register_prng(&fortuna_desc); + register_prng(&yarrow_desc); + + register_hash(&sha1_desc); + register_hash(&sha256_desc); + register_hash(&sha512_desc); + } + } ___; + + [[maybe_unused]] const char* cs(const uint8_t* data) + { + return reinterpret_cast(data); + } + + [[maybe_unused]] char* cs(uint8_t* data) + { + return reinterpret_cast(data); + } + + [[maybe_unused]] const uint8_t* cs(const char* data) + { + return reinterpret_cast(data); + } + + [[maybe_unused]] uint8_t* cs(char* data) + { + return reinterpret_cast(data); + } + + [[maybe_unused]] unsigned long ul(const size_t value) + { + return static_cast(value); + } + + class prng + { + public: + prng(const ltc_prng_descriptor& descriptor, const bool autoseed = true) + : state_(std::make_unique()) + , descriptor_(descriptor) + { + this->id_ = register_prng(&descriptor); + if (this->id_ == -1) + { + throw std::runtime_error("PRNG "s + this->descriptor_.name + " could not be registered!"); + } + + if (autoseed) + { + this->auto_seed(); + } + else + { + this->descriptor_.start(this->state_.get()); + } + } + + ~prng() + { + this->descriptor_.done(this->state_.get()); + } + + prng_state* get_state() const + { + this->descriptor_.ready(this->state_.get()); + return this->state_.get(); + } + + int get_id() const + { + return this->id_; + } + + void add_entropy(const void* data, const size_t length) const + { + this->descriptor_.add_entropy(static_cast(data), ul(length), this->state_.get()); + } + + void read(void* data, const size_t length) const + { + this->descriptor_.read(static_cast(data), ul(length), this->get_state()); + } + + private: + int id_; + std::unique_ptr state_; + const ltc_prng_descriptor& descriptor_; + + void auto_seed() const + { + rng_make_prng(128, this->id_, this->state_.get(), nullptr); + + int i[4]; // uninitialized data + auto* i_ptr = &i; + this->add_entropy(&i, sizeof(i)); + this->add_entropy(&i_ptr, sizeof(i_ptr)); + + auto t = std::time(nullptr); + this->add_entropy(&t, sizeof(t)); + + std::random_device rd{}; + for (auto j = 0; j < 4; ++j) + { + const auto x = rd(); + this->add_entropy(&x, sizeof(x)); + } + } + }; + + const prng prng_(fortuna_desc); + } + + ecc::key::key() + { + ZeroMemory(&this->key_storage_, sizeof(this->key_storage_)); + } + + ecc::key::~key() + { + this->free(); + } + + ecc::key::key(key&& obj) noexcept + : key() + { + this->operator=(std::move(obj)); + } + + ecc::key::key(const key& obj) + : key() + { + this->operator=(obj); + } + + ecc::key& ecc::key::operator=(key&& obj) noexcept + { + if (this != &obj) + { + std::memmove(&this->key_storage_, &obj.key_storage_, sizeof(this->key_storage_)); + ZeroMemory(&obj.key_storage_, sizeof(obj.key_storage_)); + } + + return *this; + } + + ecc::key& ecc::key::operator=(const key& obj) + { + if (this != &obj && obj.is_valid()) + { + this->deserialize(obj.serialize(obj.key_storage_.type)); + } + + return *this; + } + + bool ecc::key::is_valid() const + { + return (!memory::is_set(&this->key_storage_, 0, sizeof(this->key_storage_))); + } + + ecc_key& ecc::key::get() + { + return this->key_storage_; + } + + const ecc_key& ecc::key::get() const + { + return this->key_storage_; + } + + std::string ecc::key::get_public_key() const + { + uint8_t buffer[512] = {0}; + unsigned long length = sizeof(buffer); + + if (ecc_ansi_x963_export(&this->key_storage_, buffer, &length) == CRYPT_OK) + { + return std::string(cs(buffer), length); + } + + return {}; + } + + void ecc::key::set(const std::string& pub_key_buffer) + { + this->free(); + + if (ecc_ansi_x963_import(cs(pub_key_buffer.data()), + ul(pub_key_buffer.size()), + &this->key_storage_) != CRYPT_OK) + { + ZeroMemory(&this->key_storage_, sizeof(this->key_storage_)); + } + } + + void ecc::key::deserialize(const std::string& key) + { + this->free(); + + if (ecc_import(cs(key.data()), ul(key.size()), + &this->key_storage_) != CRYPT_OK + ) + { + ZeroMemory(&this->key_storage_, sizeof(this->key_storage_)); + } + } + + std::string ecc::key::serialize(const int type) const + { + uint8_t buffer[4096] = {0}; + unsigned long length = sizeof(buffer); + + if (ecc_export(buffer, &length, type, &this->key_storage_) == CRYPT_OK) + { + return std::string(cs(buffer), length); + } + + return {}; + } + + void ecc::key::free() + { + if (this->is_valid()) + { + ecc_free(&this->key_storage_); + } + + ZeroMemory(&this->key_storage_, sizeof(this->key_storage_)); + } + + bool ecc::key::operator==(key& key) const + { + return (this->is_valid() && key.is_valid() && this->serialize(PK_PUBLIC) == key.serialize(PK_PUBLIC)); + } + + uint64_t ecc::key::get_hash() const + { + const auto hash = sha1::compute(this->get_public_key()); + if (hash.size() >= 8) + { + return *reinterpret_cast(hash.data()); + } + + return 0; + } + + ecc::key ecc::generate_key(const int bits) + { + key key; + ecc_make_key(prng_.get_state(), prng_.get_id(), bits / 8, &key.get()); + + return key; + } + + ecc::key ecc::generate_key(const int bits, const std::string& entropy) + { + key key{}; + const prng yarrow(yarrow_desc, false); + yarrow.add_entropy(entropy.data(), entropy.size()); + + ecc_make_key(yarrow.get_state(), yarrow.get_id(), bits / 8, &key.get()); + + return key; + } + + std::string ecc::sign_message(const key& key, const std::string& message) + { + if (!key.is_valid()) return {}; + + uint8_t buffer[512]; + unsigned long length = sizeof(buffer); + + const auto hash = sha512::compute(message); + + ecc_sign_hash(cs(hash.data()), ul(hash.size()), buffer, &length, prng_.get_state(), prng_.get_id(), + &key.get()); + + return std::string(cs(buffer), length); + } + + bool ecc::verify_message(const key& key, const std::string& message, const std::string& signature) + { + if (!key.is_valid()) return false; + + const auto hash = sha512::compute(message); + + auto result = 0; + return (ecc_verify_hash(cs(signature.data()), + ul(signature.size()), + cs(hash.data()), + ul(hash.size()), &result, + &key.get()) == CRYPT_OK && result != 0); + } + + bool ecc::encrypt(const key& key, std::string& data) + { + std::string out_data{}; + out_data.resize(std::max(ul(data.size() * 3), ul(0x100))); + + auto out_len = ul(out_data.size()); + auto crypt = [&]() + { + return ecc_encrypt_key(cs(data.data()), ul(data.size()), cs(out_data.data()), &out_len, + prng_.get_state(), prng_.get_id(), find_hash("sha512"), &key.get()); + }; + + auto res = crypt(); + + if (res == CRYPT_BUFFER_OVERFLOW) + { + out_data.resize(out_len); + res = crypt(); + } + + if (res != CRYPT_OK) + { + return false; + } + + out_data.resize(out_len); + data = std::move(out_data); + return true; + } + + bool ecc::decrypt(const key& key, std::string& data) + { + std::string out_data{}; + out_data.resize(std::max(ul(data.size() * 3), ul(0x100))); + + auto out_len = ul(out_data.size()); + auto crypt = [&]() + { + return ecc_decrypt_key(cs(data.data()), ul(data.size()), cs(out_data.data()), &out_len, &key.get()); + }; + + auto res = crypt(); + + if (res == CRYPT_BUFFER_OVERFLOW) + { + out_data.resize(out_len); + res = crypt(); + } + + if (res != CRYPT_OK) + { + return false; + } + + out_data.resize(out_len); + data = std::move(out_data); + return true; + } + + std::string rsa::encrypt(const std::string& data, const std::string& hash, const std::string& key) + { + rsa_key new_key; + rsa_import(cs(key.data()), ul(key.size()), &new_key); + const auto _ = gsl::finally([&]() + { + rsa_free(&new_key); + }); + + + std::string out_data{}; + out_data.resize(std::max(ul(data.size() * 3), ul(0x100))); + + auto out_len = ul(out_data.size()); + auto crypt = [&]() + { + return rsa_encrypt_key(cs(data.data()), ul(data.size()), cs(out_data.data()), &out_len, cs(hash.data()), + ul(hash.size()), prng_.get_state(), prng_.get_id(), find_hash("sha512"), &new_key); + }; + + auto res = crypt(); + + if (res == CRYPT_BUFFER_OVERFLOW) + { + out_data.resize(out_len); + res = crypt(); + } + + if (res == CRYPT_OK) + { + out_data.resize(out_len); + return out_data; + } + + return {}; + } + + std::string des3::encrypt(const std::string& data, const std::string& iv, const std::string& key) + { + std::string enc_data; + enc_data.resize(data.size()); + + symmetric_CBC cbc; + const auto des3 = find_cipher("3des"); + + cbc_start(des3, cs(iv.data()), cs(key.data()), static_cast(key.size()), 0, &cbc); + cbc_encrypt(cs(data.data()), cs(enc_data.data()), ul(data.size()), &cbc); + cbc_done(&cbc); + + return enc_data; + } + + std::string des3::decrypt(const std::string& data, const std::string& iv, const std::string& key) + { + std::string dec_data; + dec_data.resize(data.size()); + + symmetric_CBC cbc; + const auto des3 = find_cipher("3des"); + + cbc_start(des3, cs(iv.data()), cs(key.data()), static_cast(key.size()), 0, &cbc); + cbc_decrypt(cs(data.data()), cs(dec_data.data()), ul(data.size()), &cbc); + cbc_done(&cbc); + + return dec_data; + } + + std::string tiger::compute(const std::string& data, const bool hex) + { + return compute(cs(data.data()), data.size(), hex); + } + + std::string tiger::compute(const uint8_t* data, const size_t length, const bool hex) + { + uint8_t buffer[24] = {0}; + + hash_state state; + tiger_init(&state); + tiger_process(&state, data, ul(length)); + tiger_done(&state, buffer); + + std::string hash(cs(buffer), sizeof(buffer)); + if (!hex) return hash; + + return string::dump_hex(hash, {}); + } + + std::string aes::encrypt(const std::string& data, const std::string& iv, const std::string& key) + { + std::string enc_data; + enc_data.resize(data.size()); + + symmetric_CBC cbc; + const auto aes = find_cipher("aes"); + + cbc_start(aes, cs(iv.data()), cs(key.data()), + static_cast(key.size()), 0, &cbc); + cbc_encrypt(cs(data.data()), + cs(enc_data.data()), + ul(data.size()), &cbc); + cbc_done(&cbc); + + return enc_data; + } + + std::string aes::decrypt(const std::string& data, const std::string& iv, const std::string& key) + { + std::string dec_data; + dec_data.resize(data.size()); + + symmetric_CBC cbc; + const auto aes = find_cipher("aes"); + + cbc_start(aes, cs(iv.data()), cs(key.data()), + static_cast(key.size()), 0, &cbc); + cbc_decrypt(cs(data.data()), + cs(dec_data.data()), + ul(data.size()), &cbc); + cbc_done(&cbc); + + return dec_data; + } + + std::string hmac_sha1::compute(const std::string& data, const std::string& key) + { + std::string buffer; + buffer.resize(20); + + hmac_state state; + hmac_init(&state, find_hash("sha1"), cs(key.data()), ul(key.size())); + hmac_process(&state, cs(data.data()), static_cast(data.size())); + + auto out_len = ul(buffer.size()); + hmac_done(&state, cs(buffer.data()), &out_len); + + buffer.resize(out_len); + return buffer; + } + + std::string sha1::compute(const std::string& data, const bool hex) + { + return compute(cs(data.data()), data.size(), hex); + } + + std::string sha1::compute(const uint8_t* data, const size_t length, const bool hex) + { + uint8_t buffer[20] = {0}; + + hash_state state; + sha1_init(&state); + sha1_process(&state, data, ul(length)); + sha1_done(&state, buffer); + + std::string hash(cs(buffer), sizeof(buffer)); + if (!hex) return hash; + + return string::dump_hex(hash, {}); + } + + std::string sha256::compute(const std::string& data, const bool hex) + { + return compute(cs(data.data()), data.size(), hex); + } + + std::string sha256::compute(const uint8_t* data, const size_t length, const bool hex) + { + uint8_t buffer[32] = {0}; + + hash_state state; + sha256_init(&state); + sha256_process(&state, data, ul(length)); + sha256_done(&state, buffer); + + std::string hash(cs(buffer), sizeof(buffer)); + if (!hex) return hash; + + return string::dump_hex(hash, {}); + } + + std::string sha512::compute(const std::string& data, const bool hex) + { + return compute(cs(data.data()), data.size(), hex); + } + + std::string sha512::compute(const uint8_t* data, const size_t length, const bool hex) + { + uint8_t buffer[64] = {0}; + + hash_state state; + sha512_init(&state); + sha512_process(&state, data, ul(length)); + sha512_done(&state, buffer); + + std::string hash(cs(buffer), sizeof(buffer)); + if (!hex) return hash; + + return string::dump_hex(hash, {}); + } + + std::string base64::encode(const uint8_t* data, const size_t len) + { + std::string result; + result.resize((len + 2) * 2); + + auto out_len = ul(result.size()); + if (base64_encode(data, ul(len), result.data(), &out_len) != CRYPT_OK) + { + return {}; + } + + result.resize(out_len); + return result; + } + + std::string base64::encode(const std::string& data) + { + return base64::encode(cs(data.data()), static_cast(data.size())); + } + + std::string base64::decode(const std::string& data) + { + std::string result; + result.resize((data.size() + 2) * 2); + + auto out_len = ul(result.size()); + if (base64_decode(data.data(), ul(data.size()), cs(result.data()), &out_len) != CRYPT_OK) + { + return {}; + } + + result.resize(out_len); + return result; + } + + unsigned int jenkins_one_at_a_time::compute(const std::string& data) + { + return compute(data.data(), data.size()); + } + + unsigned int jenkins_one_at_a_time::compute(const char* key, const size_t len) + { + unsigned int hash, i; + for (hash = i = 0; i < len; ++i) + { + hash += key[i]; + hash += (hash << 10); + hash ^= (hash >> 6); + } + hash += (hash << 3); + hash ^= (hash >> 11); + hash += (hash << 15); + return hash; + } + + uint32_t random::get_integer() + { + uint32_t result; + random::get_data(&result, sizeof(result)); + return result; + } + + std::string random::get_challenge() + { + std::string result; + result.resize(sizeof(uint32_t)); + random::get_data(result.data(), result.size()); + return string::dump_hex(result, {}); + } + + void random::get_data(void* data, const size_t size) + { + prng_.read(data, size); + } +} diff --git a/src/utils/cryptography.hpp b/src/utils/cryptography.hpp new file mode 100644 index 0000000..9538c5e --- /dev/null +++ b/src/utils/cryptography.hpp @@ -0,0 +1,118 @@ +#pragma once + +#include +#include + +namespace utils::cryptography +{ + namespace ecc + { + class key final + { + public: + key(); + ~key(); + + key(key&& obj) noexcept; + key(const key& obj); + key& operator=(key&& obj) noexcept; + key& operator=(const key& obj); + + bool is_valid() const; + + ecc_key& get(); + const ecc_key& get() const; + + std::string get_public_key() const; + + void set(const std::string& pub_key_buffer); + + void deserialize(const std::string& key); + + std::string serialize(int type = PK_PRIVATE) const; + + void free(); + + bool operator==(key& key) const; + + uint64_t get_hash() const; + + private: + ecc_key key_storage_{}; + }; + + key generate_key(int bits); + key generate_key(int bits, const std::string& entropy); + std::string sign_message(const key& key, const std::string& message); + bool verify_message(const key& key, const std::string& message, const std::string& signature); + + bool encrypt(const key& key, std::string& data); + bool decrypt(const key& key, std::string& data); + } + + namespace rsa + { + std::string encrypt(const std::string& data, const std::string& hash, const std::string& key); + } + + namespace des3 + { + std::string encrypt(const std::string& data, const std::string& iv, const std::string& key); + std::string decrypt(const std::string& data, const std::string& iv, const std::string& key); + } + + namespace tiger + { + std::string compute(const std::string& data, bool hex = false); + std::string compute(const uint8_t* data, size_t length, bool hex = false); + } + + namespace aes + { + std::string encrypt(const std::string& data, const std::string& iv, const std::string& key); + std::string decrypt(const std::string& data, const std::string& iv, const std::string& key); + } + + namespace hmac_sha1 + { + std::string compute(const std::string& data, const std::string& key); + } + + namespace sha1 + { + std::string compute(const std::string& data, bool hex = false); + std::string compute(const uint8_t* data, size_t length, bool hex = false); + } + + namespace sha256 + { + std::string compute(const std::string& data, bool hex = false); + std::string compute(const uint8_t* data, size_t length, bool hex = false); + } + + namespace sha512 + { + std::string compute(const std::string& data, bool hex = false); + std::string compute(const uint8_t* data, size_t length, bool hex = false); + } + + namespace base64 + { + std::string encode(const uint8_t* data, size_t len); + std::string encode(const std::string& data); + std::string decode(const std::string& data); + } + + namespace jenkins_one_at_a_time + { + unsigned int compute(const std::string& data); + unsigned int compute(const char* key, size_t len); + }; + + namespace random + { + uint32_t get_integer(); + std::string get_challenge(); + void get_data(void* data, size_t size); + } +} diff --git a/src/utils/http.cpp b/src/utils/http.cpp new file mode 100644 index 0000000..c6343e5 --- /dev/null +++ b/src/utils/http.cpp @@ -0,0 +1,59 @@ +#include + +#include "http.hpp" +#include + +namespace utils::http +{ + namespace + { + size_t write_callback(void* contents, const size_t size, const size_t nmemb, void* userp) + { + static_cast(userp)->append(static_cast(contents), size * nmemb); + return size * nmemb; + } + } + + std::optional get_data(const std::string& url, const headers& headers) + { + curl_slist* header_list = nullptr; + auto* curl = curl_easy_init(); + if (!curl) + { + return {}; + } + + auto _ = gsl::finally([&]() + { + curl_slist_free_all(header_list); + curl_easy_cleanup(curl); + }); + + + for(const auto& header : headers) + { + auto data = header.first + ": "s + header.second; + header_list = curl_slist_append(header_list, data.data()); + } + + std::string buffer{}; + curl_easy_setopt(curl, CURLOPT_HTTPHEADER, header_list); + curl_easy_setopt(curl, CURLOPT_URL, url.data()); + curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_callback); + curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer); + if (curl_easy_perform(curl) == CURLE_OK) + { + return {std::move(buffer)}; + } + + return {}; + } + + std::future> get_data_async(const std::string& url, const headers& headers) + { + return std::async(std::launch::async, [url, headers]() + { + return get_data(url, headers); + }); + } +} diff --git a/src/utils/http.hpp b/src/utils/http.hpp new file mode 100644 index 0000000..773d6fa --- /dev/null +++ b/src/utils/http.hpp @@ -0,0 +1,14 @@ +#pragma once + +#include +#include +#include +#include + +namespace utils::http +{ + using headers = std::unordered_map; + + std::optional get_data(const std::string& url, const headers& headers = {}); + std::future> get_data_async(const std::string& url, const headers& headers = {}); +} diff --git a/src/utils/info_string.cpp b/src/utils/info_string.cpp new file mode 100644 index 0000000..7f373f2 --- /dev/null +++ b/src/utils/info_string.cpp @@ -0,0 +1,67 @@ +#include + +#include "info_string.hpp" +#include "string.hpp" + +namespace utils +{ + info_string::info_string(const std::string& buffer) + { + this->parse(buffer); + } + + info_string::info_string(const std::string_view& buffer) + : info_string(std::string{buffer}) + { + } + + void info_string::set(const std::string& key, const std::string& value) + { + this->key_value_pairs_[key] = value; + } + + std::string info_string::get(const std::string& key) const + { + const auto value = this->key_value_pairs_.find(key); + if (value != this->key_value_pairs_.end()) + { + return value->second; + } + + return {}; + } + + void info_string::parse(std::string buffer) + { + if (buffer[0] == '\\') + { + buffer = buffer.substr(1); + } + + auto key_values = string::split(buffer, '\\'); + for (size_t i = 0; !key_values.empty() && i < (key_values.size() - 1); i += 2) + { + const auto& key = key_values[i]; + const auto& value = key_values[i + 1]; + this->key_value_pairs_[key] = value; + } + } + + std::string info_string::build() const + { + //auto first = true; + std::string info_string; + for (auto i = this->key_value_pairs_.begin(); i != this->key_value_pairs_.end(); ++i) + { + //if (first) first = false; + /*else*/ + info_string.append("\\"); + + info_string.append(i->first); // Key + info_string.append("\\"); + info_string.append(i->second); // Value + } + + return info_string; + } +} diff --git a/src/utils/info_string.hpp b/src/utils/info_string.hpp new file mode 100644 index 0000000..7391041 --- /dev/null +++ b/src/utils/info_string.hpp @@ -0,0 +1,24 @@ +#pragma once + +#include +#include + +namespace utils +{ + class info_string + { + public: + info_string() = default; + info_string(const std::string& buffer); + info_string(const std::string_view& buffer); + + void set(const std::string& key, const std::string& value); + std::string get(const std::string& key) const; + std::string build() const; + + private: + std::unordered_map key_value_pairs_{}; + + void parse(std::string buffer); + }; +} diff --git a/src/utils/io.cpp b/src/utils/io.cpp new file mode 100644 index 0000000..67ff246 --- /dev/null +++ b/src/utils/io.cpp @@ -0,0 +1,132 @@ +#include + +#include "io.hpp" +#include +#include + +namespace utils::io +{ + bool remove_file(const std::string& file) + { + return remove(file.data()) == 0; + } + + bool move_file(const std::string& src, const std::string& target) + { + return rename(src.data(), target.data()) == 0; + } + + bool file_exists(const std::string& file) + { + return std::ifstream(file).good(); + } + + bool write_file(const std::string& file, const std::string& data, const bool append) + { + const auto pos = file.find_last_of("/\\"); + if (pos != std::string::npos) + { + create_directory(file.substr(0, pos)); + } + + auto mode = std::ios::binary | std::ofstream::out; + if (append) + { + mode |= std::ofstream::app; + } + + std::ofstream stream(file, mode); + + if (stream.is_open()) + { + stream.write(data.data(), data.size()); + stream.close(); + return true; + } + + return false; + } + + std::string read_file(const std::string& file) + { + std::string data; + read_file(file, &data); + return data; + } + + bool read_file(const std::string& file, std::string* data) + { + if (!data) return false; + data->clear(); + + if (file_exists(file)) + { + std::ifstream stream(file, std::ios::binary); + if (!stream.is_open()) return false; + + stream.seekg(0, std::ios::end); + const std::streamsize size = stream.tellg(); + stream.seekg(0, std::ios::beg); + + if (size > -1) + { + data->resize(static_cast(size)); + stream.read(const_cast(data->data()), size); + stream.close(); + return true; + } + } + + return false; + } + + size_t file_size(const std::string& file) + { + if (file_exists(file)) + { + std::ifstream stream(file, std::ios::binary); + + if (stream.good()) + { + stream.seekg(0, std::ios::end); + return static_cast(stream.tellg()); + } + } + + return 0; + } + + bool create_directory(const std::string& directory) + { + return std::filesystem::create_directories(directory); + } + + bool directory_exists(const std::string& directory) + { + return std::filesystem::is_directory(directory); + } + + bool directory_is_empty(const std::string& directory) + { + return std::filesystem::is_empty(directory); + } + + std::vector list_files(const std::string& directory) + { + std::vector files; + + for (auto& file : std::filesystem::directory_iterator(directory)) + { + files.push_back(file.path().generic_string()); + } + + return files; + } + + void copy_folder(const std::filesystem::path& src, const std::filesystem::path& target) + { + std::filesystem::copy(src, target, + std::filesystem::copy_options::overwrite_existing | + std::filesystem::copy_options::recursive); + } +} diff --git a/src/utils/io.hpp b/src/utils/io.hpp new file mode 100644 index 0000000..ab4ebaa --- /dev/null +++ b/src/utils/io.hpp @@ -0,0 +1,21 @@ +#pragma once + +#include +#include +#include + +namespace utils::io +{ + bool remove_file(const std::string& file); + bool move_file(const std::string& src, const std::string& target); + bool file_exists(const std::string& file); + bool write_file(const std::string& file, const std::string& data, bool append = false); + bool read_file(const std::string& file, std::string* data); + std::string read_file(const std::string& file); + size_t file_size(const std::string& file); + bool create_directory(const std::string& directory); + bool directory_exists(const std::string& directory); + bool directory_is_empty(const std::string& directory); + std::vector list_files(const std::string& directory); + void copy_folder(const std::filesystem::path& src, const std::filesystem::path& target); +} diff --git a/src/utils/memory.cpp b/src/utils/memory.cpp new file mode 100644 index 0000000..5cf465a --- /dev/null +++ b/src/utils/memory.cpp @@ -0,0 +1,107 @@ +#include + +#include "memory.hpp" + +namespace utils +{ + memory::allocator memory::mem_allocator_; + + memory::allocator::~allocator() + { + this->clear(); + } + + void memory::allocator::clear() + { + std::lock_guard _(this->mutex_); + + for (auto& data : this->pool_) + { + memory::free(data); + } + + this->pool_.clear(); + } + + void memory::allocator::free(void* data) + { + std::lock_guard _(this->mutex_); + + const auto j = std::find(this->pool_.begin(), this->pool_.end(), data); + if (j != this->pool_.end()) + { + memory::free(data); + this->pool_.erase(j); + } + } + + void memory::allocator::free(const void* data) + { + this->free(const_cast(data)); + } + + void* memory::allocator::allocate(const size_t length) + { + std::lock_guard _(this->mutex_); + + const auto data = memory::allocate(length); + this->pool_.push_back(data); + return data; + } + + bool memory::allocator::empty() const + { + return this->pool_.empty(); + } + + char* memory::allocator::duplicate_string(const std::string& string) + { + std::lock_guard _(this->mutex_); + + const auto data = memory::duplicate_string(string); + this->pool_.push_back(data); + return data; + } + + void* memory::allocate(const size_t length) + { + return ::calloc(length, 1); + } + + char* memory::duplicate_string(const std::string& string) + { + const auto new_string = allocate_array(string.size() + 1); + std::memcpy(new_string, string.data(), string.size()); + return new_string; + } + + void memory::free(void* data) + { + ::free(data); + } + + void memory::free(const void* data) + { + free(const_cast(data)); + } + + bool memory::is_set(const void* mem, const char chr, const size_t length) + { + auto* const mem_arr = static_cast(mem); + + for (size_t i = 0; i < length; ++i) + { + if (mem_arr[i] != chr) + { + return false; + } + } + + return true; + } + + memory::allocator* memory::get_allocator() + { + return &memory::mem_allocator_; + } +} diff --git a/src/utils/memory.hpp b/src/utils/memory.hpp new file mode 100644 index 0000000..a6f29e1 --- /dev/null +++ b/src/utils/memory.hpp @@ -0,0 +1,71 @@ +#pragma once + +#include +#include + +namespace utils +{ + class memory final + { + public: + class allocator final + { + public: + ~allocator(); + + void clear(); + + void free(void* data); + + void free(const void* data); + + void* allocate(size_t length); + + template + inline T* allocate() + { + return this->allocate_array(1); + } + + template + inline T* allocate_array(const size_t count = 1) + { + return static_cast(this->allocate(count * sizeof(T))); + } + + bool empty() const; + + char* duplicate_string(const std::string& string); + + private: + std::mutex mutex_; + std::vector pool_; + }; + + static void* allocate(size_t length); + + template + static inline T* allocate() + { + return allocate_array(1); + } + + template + static inline T* allocate_array(const size_t count = 1) + { + return static_cast(allocate(count * sizeof(T))); + } + + static char* duplicate_string(const std::string& string); + + static void free(void* data); + static void free(const void* data); + + static bool is_set(const void* mem, char chr, size_t length); + + static allocator* get_allocator(); + + private: + static allocator mem_allocator_; + }; +} diff --git a/src/utils/parameters.cpp b/src/utils/parameters.cpp new file mode 100644 index 0000000..654a128 --- /dev/null +++ b/src/utils/parameters.cpp @@ -0,0 +1,92 @@ +#include + +#include "parameters.hpp" +#include "string.hpp" + +namespace utils +{ + parameters::parameters(std::string buffer) + { + while (!buffer.empty() && (buffer.back() == '\0' || isspace(buffer.back()))) + { + buffer.pop_back(); + } + + this->arguments_ = string::split(buffer, ' '); + } + + parameters::parameters(const std::string_view& buffer) + : parameters(std::string{buffer.data(), buffer.size()}) + { + } + + void parameters::add(std::string value) + { + this->arguments_.emplace_back(std::move(value)); + } + + size_t parameters::size() const + { + return this->arguments_.size(); + } + + const std::string& parameters::get(const size_t index) const + { + return this->arguments_.at(index); + } + + std::string parameters::join(const size_t index, const std::string& separator) const + { + std::string buffer{}; + + for (auto i = index; i < this->size(); ++i) + { + if (i != 0) + { + buffer.append(separator); + } + + buffer.append(this->get(i)); + } + + return buffer; + } + + const std::string& parameters::operator[](const size_t index) const + { + return this->get(index); + } + + parameters::list_type::iterator parameters::begin() + { + return this->arguments_.begin(); + } + + parameters::list_type::const_iterator parameters::begin() const + { + return this->arguments_.begin(); + } + + parameters::list_type::iterator parameters::end() + { + return this->arguments_.end(); + } + + parameters::list_type::const_iterator parameters::end() const + { + return this->arguments_.end(); + } + + bool parameters::has(const std::string& value) const + { + for (const auto& val : this->arguments_) + { + if (val == value) + { + return true; + } + } + + return false; + } +} diff --git a/src/utils/parameters.hpp b/src/utils/parameters.hpp new file mode 100644 index 0000000..e0fe093 --- /dev/null +++ b/src/utils/parameters.hpp @@ -0,0 +1,35 @@ +#pragma once + +#include +#include + +namespace utils +{ + class parameters + { + public: + using list_type = std::vector; + + parameters() = default; + parameters(std::string buffer); + parameters(const std::string_view& buffer); + + void add(std::string value); + + size_t size() const; + std::string join(size_t index = 0, const std::string& separator = " ") const; + + const std::string& operator [](size_t index) const; + const std::string& get(size_t index) const; + + list_type::iterator begin(); + list_type::const_iterator begin() const; + list_type::iterator end(); + list_type::const_iterator end() const; + + bool has(const std::string& value) const; + + private: + list_type arguments_{}; + }; +} diff --git a/src/utils/string.cpp b/src/utils/string.cpp new file mode 100644 index 0000000..d227766 --- /dev/null +++ b/src/utils/string.cpp @@ -0,0 +1,161 @@ +#include + +#include "string.hpp" +#include +#include +#include + +namespace utils::string +{ + const char* va(const char* fmt, ...) + { + static thread_local va_provider<8, 256> provider; + + va_list ap; + va_start(ap, fmt); + + const char* result = provider.get(fmt, ap); + + va_end(ap); + return result; + } + + std::vector split(const std::string& s, const char delim) + { + std::stringstream ss(s); + std::string item; + std::vector elems; + + while (std::getline(ss, item, delim)) + { + elems.push_back(std::move(item)); + item = std::string{}; + } + + return elems; + } + + std::string to_lower(std::string text) + { + std::transform(text.begin(), text.end(), text.begin(), [](const unsigned char input) + { + return static_cast(tolower(input)); + }); + + return text; + } + + std::string to_upper(std::string text) + { + std::transform(text.begin(), text.end(), text.begin(), [](const unsigned char input) + { + return static_cast(toupper(input)); + }); + + return text; + } + + bool starts_with(const std::string& text, const std::string& substring) + { + return text.find(substring) == 0; + } + + bool ends_with(const std::string& text, const std::string& substring) + { + if (substring.size() > text.size()) return false; + return std::equal(substring.rbegin(), substring.rend(), text.rbegin()); + } + + std::string dump_hex(const std::string& data, const std::string& separator) + { + std::string result; + + for (unsigned int i = 0; i < data.size(); ++i) + { + if (i > 0) + { + result.append(separator); + } + + result.append(va("%02X", data[i] & 0xFF)); + } + + return result; + } + + void strip(const char* in, char* out, int max) + { + if (!in || !out) return; + + max--; + auto current = 0; + while (*in != 0 && current < max) + { + const auto color_index = (*(in + 1) - 48) >= 0xC ? 7 : (*(in + 1) - 48); + + if (*in == '^' && (color_index != 7 || *(in + 1) == '7')) + { + ++in; + } + else + { + *out = *in; + ++out; + ++current; + } + + ++in; + } + *out = '\0'; + } + +#ifdef _WIN32 +#pragma warning(push) +#pragma warning(disable: 4100) +#endif + std::string convert(const std::wstring& wstr) + { + std::string result; + result.reserve(wstr.size()); + + for (const auto& chr : wstr) + { + result.push_back(static_cast(chr)); + } + + return result; + } + + std::wstring convert(const std::string& str) + { + std::wstring result; + result.reserve(str.size()); + + for (const auto& chr : str) + { + result.push_back(static_cast(chr)); + } + + return result; + } +#ifdef _WIN32 +#pragma warning(pop) +#endif + + std::string replace(std::string str, const std::string& from, const std::string& to) + { + if (from.empty()) + { + return str; + } + + size_t start_pos = 0; + while ((start_pos = str.find(from, start_pos)) != std::string::npos) + { + str.replace(start_pos, from.length(), to); + start_pos += to.length(); + } + + return str; + } +} diff --git a/src/utils/string.hpp b/src/utils/string.hpp new file mode 100644 index 0000000..b9355ab --- /dev/null +++ b/src/utils/string.hpp @@ -0,0 +1,105 @@ +#pragma once +#include "memory.hpp" +#include + +#ifndef ARRAYSIZE +template +size_t ARRAYSIZE(Type (&)[n]) { return n; } +#endif + +namespace utils::string +{ + template + class va_provider final + { + public: + static_assert(Buffers != 0 && MinBufferSize != 0, "Buffers and MinBufferSize mustn't be 0"); + + va_provider() : current_buffer_(0) + { + } + + char* get(const char* format, va_list ap) + { + ++this->current_buffer_ %= ARRAYSIZE(this->string_pool_); + auto entry = &this->string_pool_[this->current_buffer_]; + + if (!entry->size || !entry->buffer) + { + throw std::runtime_error("String pool not initialized"); + } + + while (true) + { +#ifdef _WIN32 + const int res = vsnprintf_s(entry->buffer, entry->size, _TRUNCATE, format, ap); +#else + const int res = vsnprintf(entry->buffer, entry->size, format, ap); +#endif + + if (res > 0) break; // Success + if (res == 0) return nullptr; // Error + + entry->double_size(); + } + + return entry->buffer; + } + + private: + class entry final + { + public: + explicit entry(const size_t _size = MinBufferSize) : size(_size), buffer(nullptr) + { + if (this->size < MinBufferSize) this->size = MinBufferSize; + this->allocate(); + } + + ~entry() + { + if (this->buffer) memory::get_allocator()->free(this->buffer); + this->size = 0; + this->buffer = nullptr; + } + + void allocate() + { + if (this->buffer) memory::get_allocator()->free(this->buffer); + this->buffer = memory::get_allocator()->allocate_array(this->size + 1); + } + + void double_size() + { + this->size *= 2; + this->allocate(); + } + + size_t size; + char* buffer; + }; + + size_t current_buffer_; + entry string_pool_[Buffers]; + }; + + const char* va(const char* fmt, ...); + + std::vector split(const std::string& s, char delim); + std::vector split(const std::string_view& s, char delim); + + std::string to_lower(std::string text); + std::string to_upper(std::string text); + + bool starts_with(const std::string& text, const std::string& substring); + bool ends_with(const std::string& text, const std::string& substring); + + std::string dump_hex(const std::string& data, const std::string& separator = " "); + + void strip(const char* in, char* out, int max); + + std::string convert(const std::wstring& wstr); + std::wstring convert(const std::string& str); + + std::string replace(std::string str, const std::string& from, const std::string& to); +} diff --git a/tools/premake5-linux b/tools/premake5-linux new file mode 100755 index 0000000..bcb51ce Binary files /dev/null and b/tools/premake5-linux differ diff --git a/tools/premake5-mac b/tools/premake5-mac new file mode 100755 index 0000000..48daa61 Binary files /dev/null and b/tools/premake5-mac differ diff --git a/tools/premake5.exe b/tools/premake5.exe new file mode 100644 index 0000000..1a637aa Binary files /dev/null and b/tools/premake5.exe differ