Init
This commit is contained in:
13
.clang-format
Normal file
13
.clang-format
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
---
|
||||||
|
BasedOnStyle: Microsoft
|
||||||
|
IndentWidth: 4
|
||||||
|
SortIncludes: false
|
||||||
|
---
|
||||||
|
Language: Cpp
|
||||||
|
IndentPPDirectives: AfterHash
|
||||||
|
FixNamespaceComments: true
|
||||||
|
NamespaceIndentation: All
|
||||||
|
SeparateDefinitionBlocks: Always
|
||||||
|
SpaceAfterCStyleCast: true
|
||||||
|
SpaceAfterLogicalNot: false
|
||||||
|
SpaceAfterTemplateKeyword: false
|
||||||
50
.gitignore
vendored
Normal file
50
.gitignore
vendored
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
# ---> C++
|
||||||
|
# Prerequisites
|
||||||
|
*.d
|
||||||
|
|
||||||
|
# Compiled Object files
|
||||||
|
*.slo
|
||||||
|
*.lo
|
||||||
|
*.o
|
||||||
|
*.obj
|
||||||
|
|
||||||
|
# Precompiled Headers
|
||||||
|
*.gch
|
||||||
|
*.pch
|
||||||
|
|
||||||
|
# Compiled Dynamic libraries
|
||||||
|
*.so
|
||||||
|
*.dylib
|
||||||
|
*.dll
|
||||||
|
|
||||||
|
# Fortran module files
|
||||||
|
*.mod
|
||||||
|
*.smod
|
||||||
|
|
||||||
|
# Compiled Static libraries
|
||||||
|
*.lai
|
||||||
|
*.la
|
||||||
|
*.a
|
||||||
|
*.lib
|
||||||
|
|
||||||
|
# Executables
|
||||||
|
*.exe
|
||||||
|
*.out
|
||||||
|
*.app
|
||||||
|
|
||||||
|
# ---> VisualStudioCode
|
||||||
|
.vscode/*
|
||||||
|
!.vscode/settings.json
|
||||||
|
!.vscode/tasks.json
|
||||||
|
!.vscode/extensions.json
|
||||||
|
!.vscode/*.code-snippets
|
||||||
|
|
||||||
|
# Local History for Visual Studio Code
|
||||||
|
.history/
|
||||||
|
|
||||||
|
# Built Visual Studio Code Extensions
|
||||||
|
*.vsix
|
||||||
|
|
||||||
|
.cache/
|
||||||
|
.local/
|
||||||
|
Build/
|
||||||
13
.vscode/tasks.json
vendored
Normal file
13
.vscode/tasks.json
vendored
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
{
|
||||||
|
"tasks": [
|
||||||
|
{
|
||||||
|
"label": "build",
|
||||||
|
"type": "shell",
|
||||||
|
"command": "python3 Tools/Builder/Build.py",
|
||||||
|
"group": {
|
||||||
|
"kind": "build",
|
||||||
|
"isDefault": true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
}
|
||||||
97
CMake/FindDeps.cmake
Normal file
97
CMake/FindDeps.cmake
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
include(FetchContent)
|
||||||
|
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Wno-error=int-conversion")
|
||||||
|
add_compile_definitions(-D_ITERATOR_DEBUG_LEVEL=0)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
httplib
|
||||||
|
GIT_REPOSITORY https://github.com/yhirose/cpp-httplib.git
|
||||||
|
GIT_TAG v0.28.0
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
nlohmann_json
|
||||||
|
GIT_REPOSITORY https://github.com/nlohmann/json.git
|
||||||
|
GIT_TAG v3.12.0
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
glaze
|
||||||
|
GIT_REPOSITORY https://github.com/stephenberry/glaze.git
|
||||||
|
GIT_TAG v6.1.0
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
simdjson
|
||||||
|
GIT_REPOSITORY https://github.com/simdjson/simdjson.git
|
||||||
|
GIT_TAG v4.2.2
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
ZLIB
|
||||||
|
GIT_REPOSITORY https://github.com/madler/zlib.git
|
||||||
|
GIT_TAG v1.3.1
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
zstd
|
||||||
|
GIT_REPOSITORY https://github.com/facebook/zstd.git
|
||||||
|
GIT_TAG v1.5.7
|
||||||
|
SOURCE_SUBDIR build/cmake
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
mimalloc
|
||||||
|
GIT_REPOSITORY https://github.com/microsoft/mimalloc.git
|
||||||
|
GIT_TAG v3.0.10
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
PATCH_COMMAND ${CMAKE_COMMAND}
|
||||||
|
-DSOURCE_DIR=<SOURCE_DIR>
|
||||||
|
-P ${CMAKE_CURRENT_SOURCE_DIR}/CMake/PatchMimalloc.cmake
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
tl-expected
|
||||||
|
GIT_REPOSITORY https://github.com/TartanLlama/expected.git
|
||||||
|
GIT_TAG v1.3.1
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
FetchContent_Declare(
|
||||||
|
unordered_dense
|
||||||
|
GIT_REPOSITORY https://github.com/martinus/unordered_dense.git
|
||||||
|
GIT_TAG v4.8.1
|
||||||
|
SYSTEM
|
||||||
|
EXCLUDE_FROM_ALL
|
||||||
|
)
|
||||||
|
|
||||||
|
set(MI_BUILD_SHARED ON CACHE BOOL "" FORCE)
|
||||||
|
set(MI_BUILD_STATIC ON CACHE BOOL "" FORCE)
|
||||||
|
set(MI_BUILD_TESTS OFF CACHE BOOL "" FORCE)
|
||||||
|
|
||||||
|
set(EXPECTED_BUILD_TESTS OFF CACHE BOOL "" FORCE)
|
||||||
|
|
||||||
|
set(HTTPLIB_REQUIRE_OPENSSL OFF CACHE BOOL "" FORCE)
|
||||||
|
set(HTTPLIB_REQUIRE_ZLIB OFF CACHE BOOL "" FORCE)
|
||||||
|
|
||||||
|
FetchContent_MakeAvailable(httplib nlohmann_json glaze simdjson ZLIB zstd mimalloc tl-expected unordered_dense)
|
||||||
|
|
||||||
|
if(NOT TARGET ZLIB::ZLIB)
|
||||||
|
add_library(ZLIB::ZLIB ALIAS zlibstatic)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
find_package(OpenSSL REQUIRED)
|
||||||
25
CMake/PatchMimalloc.cmake
Normal file
25
CMake/PatchMimalloc.cmake
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
# Get the file path from the command line argument
|
||||||
|
set(TARGET_FILE "${SOURCE_DIR}/src/prim/windows/prim.c")
|
||||||
|
|
||||||
|
# Read the file
|
||||||
|
file(READ "${TARGET_FILE}" FILE_CONTENT)
|
||||||
|
|
||||||
|
# Check if the patch is already applied
|
||||||
|
string(FIND "${FILE_CONTENT}" "struct _TEB* const teb" ALREADY_PATCHED)
|
||||||
|
|
||||||
|
if(ALREADY_PATCHED EQUAL -1)
|
||||||
|
message(STATUS "Patching mimalloc source: ${TARGET_FILE}")
|
||||||
|
|
||||||
|
# Perform the replacement
|
||||||
|
string(REPLACE
|
||||||
|
"_TEB* const teb = NtCurrentTeb()"
|
||||||
|
"struct _TEB* const teb = NtCurrentTeb()"
|
||||||
|
FILE_CONTENT
|
||||||
|
"${FILE_CONTENT}"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Write the file back only if changes were made
|
||||||
|
file(WRITE "${TARGET_FILE}" "${FILE_CONTENT}")
|
||||||
|
else()
|
||||||
|
message(STATUS "mimalloc source is already patched. Skipping.")
|
||||||
|
endif()
|
||||||
72
CMakeLists.txt
Normal file
72
CMakeLists.txt
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
cmake_minimum_required(VERSION 3.28 FATAL_ERROR)
|
||||||
|
|
||||||
|
set(CMAKE_CXX_STANDARD 20)
|
||||||
|
set(CMAKE_CXX_EXTENSIONS OFF)
|
||||||
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
||||||
|
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
|
||||||
|
|
||||||
|
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
|
||||||
|
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
|
||||||
|
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
|
||||||
|
|
||||||
|
project(IACore)
|
||||||
|
|
||||||
|
enable_language(C)
|
||||||
|
|
||||||
|
include(CMake/FindDeps.cmake)
|
||||||
|
|
||||||
|
# Default to ON if root, OFF if dependency
|
||||||
|
option(IACore_BUILD_TESTS "Build unit tests" ${PROJECT_IS_TOP_LEVEL})
|
||||||
|
|
||||||
|
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
||||||
|
message(STATUS "Configuring IACore for Debug..")
|
||||||
|
add_compile_options(-O0 -g)
|
||||||
|
add_compile_definitions("__IA_DEBUG=1")
|
||||||
|
elseif(CMAKE_BUILD_TYPE STREQUAL "Release")
|
||||||
|
message(STATUS "Configuring IACore for Release..")
|
||||||
|
add_compile_options(-O3 -g0)
|
||||||
|
add_compile_definitions("__IA_DEBUG=0")
|
||||||
|
else()
|
||||||
|
message(FATAL_ERROR "Unknown CMAKE_BUILD_TYPE \"${CMAKE_BUILD_TYPE}\"")
|
||||||
|
endif()
|
||||||
|
|
||||||
|
|
||||||
|
message(STATUS "Detected Compiler ID: ${CMAKE_CXX_COMPILER_ID}")
|
||||||
|
# Check if the compiler is MSVC (cl.exe), but allow Clang acting like MSVC (clang-cl)
|
||||||
|
if (CMAKE_CXX_COMPILER_ID STREQUAL "MSVC")
|
||||||
|
# Note: clang-cl usually reports itself as "Clang" in newer CMake versions,
|
||||||
|
# but if it reports MSVC with a Clang simulation, we want to allow it.
|
||||||
|
if (NOT CMAKE_CXX_COMPILER_ID MATCHES "Clang")
|
||||||
|
message(FATAL_ERROR
|
||||||
|
"\n\n"
|
||||||
|
"-------------------------------------------------------------\n"
|
||||||
|
"CRITICAL ERROR: Unsupported Compiler Detected (MSVC/cl.exe)\n"
|
||||||
|
"-------------------------------------------------------------\n"
|
||||||
|
"IACore requires GCC or Clang to compile.\n"
|
||||||
|
"For compiling with IACore on Windows, please use Clang Tools for MSVC.\n"
|
||||||
|
"-------------------------------------------------------------\n"
|
||||||
|
)
|
||||||
|
endif()
|
||||||
|
endif()
|
||||||
|
if (CMAKE_CXX_COMPILER_ID MATCHES "Clang" OR CMAKE_CXX_COMPILER_ID MATCHES "GNU")
|
||||||
|
add_compile_options(
|
||||||
|
-Wall -Wextra -Wpedantic
|
||||||
|
# Suppress warning for the statement expression macro if -pedantic is on
|
||||||
|
-Wno-language-extension-token
|
||||||
|
)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
add_compile_definitions(-D_ITERATOR_DEBUG_LEVEL=0)
|
||||||
|
|
||||||
|
add_subdirectory(Src/)
|
||||||
|
|
||||||
|
if(IACore_BUILD_TESTS)
|
||||||
|
add_subdirectory(Tests)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
# -------------------------------------------------
|
||||||
|
# Local Development Sandboxes (not included in source control)
|
||||||
|
# -------------------------------------------------
|
||||||
|
if (EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/.local")
|
||||||
|
add_subdirectory("${CMAKE_CURRENT_SOURCE_DIR}/.local")
|
||||||
|
endif()
|
||||||
232
LICENSE
Normal file
232
LICENSE
Normal file
@ -0,0 +1,232 @@
|
|||||||
|
GNU GENERAL PUBLIC LICENSE
|
||||||
|
Version 3, 29 June 2007
|
||||||
|
|
||||||
|
Copyright © 2007 Free Software Foundation, Inc. <https://fsf.org/>
|
||||||
|
|
||||||
|
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.
|
||||||
|
|
||||||
|
IACore
|
||||||
|
Copyright (C) 2025 dev0
|
||||||
|
|
||||||
|
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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
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:
|
||||||
|
|
||||||
|
IACore Copyright (C) 2025 dev0
|
||||||
|
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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
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 <https://www.gnu.org/philosophy/why-not-lgpl.html>.
|
||||||
2
Src/CMakeLists.txt
Normal file
2
Src/CMakeLists.txt
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
|
||||||
|
add_subdirectory(IACore/)
|
||||||
66
Src/IACore/CMakeLists.txt
Normal file
66
Src/IACore/CMakeLists.txt
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
set(SRC_FILES
|
||||||
|
"imp/cpp/IPC.cpp"
|
||||||
|
"imp/cpp/JSON.cpp"
|
||||||
|
"imp/cpp/IACore.cpp"
|
||||||
|
"imp/cpp/Logger.cpp"
|
||||||
|
"imp/cpp/FileOps.cpp"
|
||||||
|
"imp/cpp/AsyncOps.cpp"
|
||||||
|
"imp/cpp/DataOps.cpp"
|
||||||
|
"imp/cpp/SocketOps.cpp"
|
||||||
|
"imp/cpp/StringOps.cpp"
|
||||||
|
"imp/cpp/ProcessOps.cpp"
|
||||||
|
"imp/cpp/HttpClient.cpp"
|
||||||
|
"imp/cpp/StreamReader.cpp"
|
||||||
|
"imp/cpp/StreamWriter.cpp"
|
||||||
|
)
|
||||||
|
|
||||||
|
add_library(IACore STATIC ${SRC_FILES})
|
||||||
|
|
||||||
|
target_include_directories(IACore PUBLIC inc/)
|
||||||
|
target_include_directories(IACore PRIVATE imp/hpp/)
|
||||||
|
|
||||||
|
target_link_libraries(IACore PUBLIC
|
||||||
|
libzstd_static
|
||||||
|
ZLIB::ZLIB
|
||||||
|
tl::expected
|
||||||
|
glaze::glaze
|
||||||
|
simdjson::simdjson
|
||||||
|
nlohmann_json::nlohmann_json
|
||||||
|
unordered_dense::unordered_dense
|
||||||
|
)
|
||||||
|
|
||||||
|
target_link_libraries(IACore PRIVATE
|
||||||
|
httplib::httplib
|
||||||
|
OpenSSL::SSL
|
||||||
|
OpenSSL::Crypto
|
||||||
|
)
|
||||||
|
|
||||||
|
if(WIN32)
|
||||||
|
target_link_libraries(IACore PUBLIC mimalloc-static)
|
||||||
|
|
||||||
|
if(MSVC)
|
||||||
|
target_link_options(IACore PUBLIC "/INCLUDE:mi_version")
|
||||||
|
else()
|
||||||
|
target_link_options(IACore PUBLIC "")
|
||||||
|
endif()
|
||||||
|
else()
|
||||||
|
target_link_libraries(IACore PUBLIC mimalloc)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
target_precompile_headers(IACore PUBLIC inc/IACore/PCH.hpp)
|
||||||
|
|
||||||
|
target_compile_options(IACore PRIVATE -fno-exceptions)
|
||||||
|
target_compile_options(IACore INTERFACE
|
||||||
|
$<$<NOT:$<BOOL:$<TARGET_PROPERTY:USE_EXCEPTIONS>>>:-fno-exceptions>
|
||||||
|
)
|
||||||
|
|
||||||
|
define_property(TARGET PROPERTY USE_EXCEPTIONS
|
||||||
|
BRIEF_DOCS "If ON, this target is allowed to use C++ exceptions."
|
||||||
|
FULL_DOCS "Prevents IACore from propagating -fno-exceptions to this target."
|
||||||
|
)
|
||||||
|
|
||||||
|
target_compile_definitions(IACore PRIVATE
|
||||||
|
CPPHTTPLIB_OPENSSL_SUPPORT
|
||||||
|
CPPHTTPLIB_ZLIB_SUPPORT
|
||||||
|
NOMINMAX
|
||||||
|
)
|
||||||
179
Src/IACore/imp/cpp/AsyncOps.cpp
Normal file
179
Src/IACore/imp/cpp/AsyncOps.cpp
Normal file
@ -0,0 +1,179 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/AsyncOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
Mutex AsyncOps::s_queueMutex;
|
||||||
|
ConditionVariable AsyncOps::s_wakeCondition;
|
||||||
|
Vector<JoiningThread> AsyncOps::s_scheduleWorkers;
|
||||||
|
Deque<AsyncOps::ScheduledTask> AsyncOps::s_highPriorityQueue;
|
||||||
|
Deque<AsyncOps::ScheduledTask> AsyncOps::s_normalPriorityQueue;
|
||||||
|
|
||||||
|
VOID AsyncOps::RunTask(IN Function<VOID()> task)
|
||||||
|
{
|
||||||
|
JoiningThread(task).detach();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID AsyncOps::InitializeScheduler(IN UINT8 workerCount)
|
||||||
|
{
|
||||||
|
if (!workerCount)
|
||||||
|
workerCount = std::max((UINT32) 2, std::thread::hardware_concurrency() - 2);
|
||||||
|
for (UINT32 i = 0; i < workerCount; i++)
|
||||||
|
s_scheduleWorkers.emplace_back(AsyncOps::ScheduleWorkerLoop, i + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID AsyncOps::TerminateScheduler()
|
||||||
|
{
|
||||||
|
for (auto &w : s_scheduleWorkers)
|
||||||
|
{
|
||||||
|
w.request_stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
s_wakeCondition.notify_all();
|
||||||
|
|
||||||
|
for (auto &w : s_scheduleWorkers)
|
||||||
|
{
|
||||||
|
if (w.joinable())
|
||||||
|
{
|
||||||
|
w.join();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
s_scheduleWorkers.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID AsyncOps::ScheduleTask(IN Function<VOID(IN WorkerID workerID)> task, IN TaskTag tag, IN Schedule *schedule,
|
||||||
|
IN Priority priority)
|
||||||
|
{
|
||||||
|
IA_ASSERT(s_scheduleWorkers.size() && "Scheduler must be initialized before calling this function");
|
||||||
|
|
||||||
|
schedule->Counter.fetch_add(1);
|
||||||
|
{
|
||||||
|
ScopedLock lock(s_queueMutex);
|
||||||
|
if (priority == Priority::High)
|
||||||
|
s_highPriorityQueue.emplace_back(ScheduledTask{tag, schedule, IA_MOVE(task)});
|
||||||
|
else
|
||||||
|
s_normalPriorityQueue.emplace_back(ScheduledTask{tag, schedule, IA_MOVE(task)});
|
||||||
|
}
|
||||||
|
s_wakeCondition.notify_one();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID AsyncOps::CancelTasksOfTag(IN TaskTag tag)
|
||||||
|
{
|
||||||
|
ScopedLock lock(s_queueMutex);
|
||||||
|
|
||||||
|
auto cancelFromQueue = [&](Deque<ScheduledTask> &queue) {
|
||||||
|
for (auto it = queue.begin(); it != queue.end(); /* no increment here */)
|
||||||
|
{
|
||||||
|
if (it->Tag == tag)
|
||||||
|
{
|
||||||
|
if (it->ScheduleHandle->Counter.fetch_sub(1) == 1)
|
||||||
|
it->ScheduleHandle->Counter.notify_all();
|
||||||
|
|
||||||
|
it = queue.erase(it);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
++it;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
cancelFromQueue(s_highPriorityQueue);
|
||||||
|
cancelFromQueue(s_normalPriorityQueue);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID AsyncOps::WaitForScheduleCompletion(IN Schedule *schedule)
|
||||||
|
{
|
||||||
|
IA_ASSERT(s_scheduleWorkers.size() && "Scheduler must be initialized before calling this function");
|
||||||
|
|
||||||
|
while (schedule->Counter.load() > 0)
|
||||||
|
{
|
||||||
|
ScheduledTask task;
|
||||||
|
BOOL foundTask{FALSE};
|
||||||
|
{
|
||||||
|
UniqueLock lock(s_queueMutex);
|
||||||
|
if (!s_highPriorityQueue.empty())
|
||||||
|
{
|
||||||
|
task = IA_MOVE(s_highPriorityQueue.front());
|
||||||
|
s_highPriorityQueue.pop_front();
|
||||||
|
foundTask = TRUE;
|
||||||
|
}
|
||||||
|
else if (!s_normalPriorityQueue.empty())
|
||||||
|
{
|
||||||
|
task = IA_MOVE(s_normalPriorityQueue.front());
|
||||||
|
s_normalPriorityQueue.pop_front();
|
||||||
|
foundTask = TRUE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (foundTask)
|
||||||
|
{
|
||||||
|
task.Task(MainThreadWorkerID);
|
||||||
|
if (task.ScheduleHandle->Counter.fetch_sub(1) == 1)
|
||||||
|
task.ScheduleHandle->Counter.notify_all();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
auto currentVal = schedule->Counter.load();
|
||||||
|
if (currentVal > 0)
|
||||||
|
schedule->Counter.wait(currentVal);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
AsyncOps::WorkerID AsyncOps::GetWorkerCount()
|
||||||
|
{
|
||||||
|
return static_cast<WorkerID>(s_scheduleWorkers.size() + 1); // +1 for MainThread (Work Stealing)
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID AsyncOps::ScheduleWorkerLoop(IN StopToken stopToken, IN WorkerID workerID)
|
||||||
|
{
|
||||||
|
while (!stopToken.stop_requested())
|
||||||
|
{
|
||||||
|
ScheduledTask task;
|
||||||
|
BOOL foundTask{FALSE};
|
||||||
|
{
|
||||||
|
UniqueLock lock(s_queueMutex);
|
||||||
|
|
||||||
|
s_wakeCondition.wait(lock, [&stopToken] {
|
||||||
|
return !s_highPriorityQueue.empty() || !s_normalPriorityQueue.empty() || stopToken.stop_requested();
|
||||||
|
});
|
||||||
|
|
||||||
|
if (stopToken.stop_requested() && s_highPriorityQueue.empty() && s_normalPriorityQueue.empty())
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (!s_highPriorityQueue.empty())
|
||||||
|
{
|
||||||
|
task = IA_MOVE(s_highPriorityQueue.front());
|
||||||
|
s_highPriorityQueue.pop_front();
|
||||||
|
foundTask = TRUE;
|
||||||
|
}
|
||||||
|
else if (!s_normalPriorityQueue.empty())
|
||||||
|
{
|
||||||
|
task = IA_MOVE(s_normalPriorityQueue.front());
|
||||||
|
s_normalPriorityQueue.pop_front();
|
||||||
|
foundTask = TRUE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (foundTask)
|
||||||
|
{
|
||||||
|
task.Task(workerID);
|
||||||
|
if (task.ScheduleHandle->Counter.fetch_sub(1) == 1)
|
||||||
|
task.ScheduleHandle->Counter.notify_all();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
319
Src/IACore/imp/cpp/DataOps.cpp
Normal file
319
Src/IACore/imp/cpp/DataOps.cpp
Normal file
@ -0,0 +1,319 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/DataOps.hpp>
|
||||||
|
|
||||||
|
#include <zlib.h>
|
||||||
|
#include <zstd.h>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
STATIC CONSTEXPR UINT32 CRC32_TABLE[] = {
|
||||||
|
/* CRC polynomial 0xedb88320 */
|
||||||
|
0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f, 0xe963a535, 0x9e6495a3, 0x0edb8832,
|
||||||
|
0x79dcb8a4, 0xe0d5e91e, 0x97d2d988, 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,
|
||||||
|
0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7, 0x136c9856, 0x646ba8c0, 0xfd62f97a,
|
||||||
|
0x8a65c9ec, 0x14015c4f, 0x63066cd9, 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
|
||||||
|
0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3,
|
||||||
|
0x45df5c75, 0xdcd60dcf, 0xabd13d59, 0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,
|
||||||
|
0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924, 0x2f6f7c87, 0x58684c11, 0xc1611dab,
|
||||||
|
0xb6662d3d, 0x76dc4190, 0x01db7106, 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
|
||||||
|
0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01, 0x6b6b51f4,
|
||||||
|
0x1c6c6162, 0x856530d8, 0xf262004e, 0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
|
||||||
|
0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65, 0x4db26158, 0x3ab551ce, 0xa3bc0074,
|
||||||
|
0xd4bb30e2, 0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
|
||||||
|
0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa, 0xbe0b1010, 0xc90c2086, 0x5768b525,
|
||||||
|
0x206f85b3, 0xb966d409, 0xce61e49f, 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,
|
||||||
|
0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a, 0xead54739, 0x9dd277af, 0x04db2615,
|
||||||
|
0x73dc1683, 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
|
||||||
|
0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb, 0x196c3671, 0x6e6b06e7, 0xfed41b76,
|
||||||
|
0x89d32be0, 0x10da7a5a, 0x67dd4acc, 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,
|
||||||
|
0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b, 0xd80d2bda, 0xaf0a1b4c, 0x36034af6,
|
||||||
|
0x41047a60, 0xdf60efc3, 0xa867df55, 0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
|
||||||
|
0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7,
|
||||||
|
0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d, 0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,
|
||||||
|
0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38, 0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7,
|
||||||
|
0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242, 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
|
||||||
|
0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45, 0xa00ae278,
|
||||||
|
0xd70dd2ee, 0x4e048354, 0x3903b3c2, 0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,
|
||||||
|
0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9, 0xbdbdf21c, 0xcabac28a, 0x53b39330,
|
||||||
|
0x24b4a3a6, 0xbad03605, 0xcdd70693, 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
|
||||||
|
0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d,
|
||||||
|
};
|
||||||
|
|
||||||
|
// FNV-1a 32-bit Constants
|
||||||
|
constexpr uint32_t FNV1A_32_PRIME = 0x01000193; // 16777619
|
||||||
|
constexpr uint32_t FNV1A_32_OFFSET = 0x811c9dc5; // 2166136261
|
||||||
|
|
||||||
|
UINT32 DataOps::Hash(IN CONST String &string)
|
||||||
|
{
|
||||||
|
uint32_t hash = FNV1A_32_OFFSET;
|
||||||
|
for (char c : string)
|
||||||
|
{
|
||||||
|
hash ^= static_cast<uint8_t>(c);
|
||||||
|
hash *= FNV1A_32_PRIME;
|
||||||
|
}
|
||||||
|
return hash;
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT32 DataOps::Hash(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
uint32_t hash = FNV1A_32_OFFSET;
|
||||||
|
const uint8_t *ptr = static_cast<const uint8_t *>(data.data());
|
||||||
|
|
||||||
|
for (size_t i = 0; i < data.size(); ++i)
|
||||||
|
{
|
||||||
|
hash ^= ptr[i];
|
||||||
|
hash *= FNV1A_32_PRIME;
|
||||||
|
}
|
||||||
|
return hash;
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT32 DataOps::CRC32(IN Span<CONST UINT8> _data)
|
||||||
|
{
|
||||||
|
UINT32 crc32 = 0xFFFFFFFF;
|
||||||
|
|
||||||
|
#define UPDC32(octet, crc) (CRC32_TABLE[((crc) ^ ((UINT8) octet)) & 0xff] ^ ((crc) >> 8))
|
||||||
|
|
||||||
|
auto data = _data.data();
|
||||||
|
auto size = _data.size();
|
||||||
|
for (; size; --size, ++data)
|
||||||
|
{
|
||||||
|
crc32 = UPDC32(*data, crc32);
|
||||||
|
}
|
||||||
|
|
||||||
|
#undef UPDC32
|
||||||
|
|
||||||
|
return ~crc32;
|
||||||
|
}
|
||||||
|
|
||||||
|
DataOps::CompressionType DataOps::DetectCompression(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
if (data.size() < 2)
|
||||||
|
return CompressionType::None;
|
||||||
|
|
||||||
|
// Check for GZIP Magic Number (0x1F 0x8B)
|
||||||
|
if (data[0] == 0x1F && data[1] == 0x8B)
|
||||||
|
return CompressionType::Gzip;
|
||||||
|
|
||||||
|
// Check for ZLIB Magic Number (starts with 0x78)
|
||||||
|
// 0x78 = Deflate compression with 32k window size
|
||||||
|
// Valid second bytes: 0x01 (Fastest), 0x9C (Default), 0xDA (Best)
|
||||||
|
if (data[0] == 0x78 && (data[1] == 0x01 || data[1] == 0x9C || data[1] == 0xDA))
|
||||||
|
return CompressionType::Zlib;
|
||||||
|
|
||||||
|
return CompressionType::None;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> DataOps::ZlibInflate(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
z_stream zs{};
|
||||||
|
zs.zalloc = Z_NULL;
|
||||||
|
zs.zfree = Z_NULL;
|
||||||
|
zs.opaque = Z_NULL;
|
||||||
|
|
||||||
|
// 15 + 32 = Auto-detect Gzip or Zlib
|
||||||
|
if (inflateInit2(&zs, 15 + 32) != Z_OK)
|
||||||
|
return MakeUnexpected("Failed to initialize zlib inflate");
|
||||||
|
|
||||||
|
zs.next_in = const_cast<Bytef *>(data.data());
|
||||||
|
zs.avail_in = static_cast<uInt>(data.size());
|
||||||
|
|
||||||
|
Vector<UINT8> outBuffer;
|
||||||
|
// Start with 2x input size.
|
||||||
|
// Small packets compress well, so maybe 4x for very small inputs.
|
||||||
|
size_t guessSize = data.size() < 1024 ? data.size() * 4 : data.size() * 2;
|
||||||
|
outBuffer.resize(guessSize);
|
||||||
|
|
||||||
|
zs.next_out = reinterpret_cast<Bytef *>(outBuffer.data());
|
||||||
|
zs.avail_out = static_cast<uInt>(outBuffer.size());
|
||||||
|
|
||||||
|
int ret;
|
||||||
|
do
|
||||||
|
{
|
||||||
|
if (zs.avail_out == 0)
|
||||||
|
{
|
||||||
|
size_t currentPos = zs.total_out;
|
||||||
|
|
||||||
|
size_t newSize = outBuffer.size() * 2;
|
||||||
|
outBuffer.resize(newSize);
|
||||||
|
|
||||||
|
zs.next_out = reinterpret_cast<Bytef *>(outBuffer.data() + currentPos);
|
||||||
|
|
||||||
|
zs.avail_out = static_cast<uInt>(newSize - currentPos);
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = inflate(&zs, Z_NO_FLUSH);
|
||||||
|
|
||||||
|
} while (ret == Z_OK);
|
||||||
|
|
||||||
|
inflateEnd(&zs);
|
||||||
|
|
||||||
|
if (ret != Z_STREAM_END)
|
||||||
|
return MakeUnexpected("Failed to inflate: corrupt data or stream error");
|
||||||
|
|
||||||
|
outBuffer.resize(zs.total_out);
|
||||||
|
|
||||||
|
return outBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> DataOps::ZlibDeflate(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
z_stream zs{};
|
||||||
|
zs.zalloc = Z_NULL;
|
||||||
|
zs.zfree = Z_NULL;
|
||||||
|
zs.opaque = Z_NULL;
|
||||||
|
|
||||||
|
if (deflateInit(&zs, Z_DEFAULT_COMPRESSION) != Z_OK)
|
||||||
|
return MakeUnexpected("Failed to initialize zlib deflate");
|
||||||
|
|
||||||
|
zs.next_in = const_cast<Bytef *>(data.data());
|
||||||
|
zs.avail_in = static_cast<uInt>(data.size());
|
||||||
|
|
||||||
|
Vector<UINT8> outBuffer;
|
||||||
|
|
||||||
|
outBuffer.resize(deflateBound(&zs, data.size()));
|
||||||
|
|
||||||
|
zs.next_out = reinterpret_cast<Bytef *>(outBuffer.data());
|
||||||
|
zs.avail_out = static_cast<uInt>(outBuffer.size());
|
||||||
|
|
||||||
|
int ret = deflate(&zs, Z_FINISH);
|
||||||
|
|
||||||
|
if (ret != Z_STREAM_END)
|
||||||
|
{
|
||||||
|
deflateEnd(&zs);
|
||||||
|
return MakeUnexpected("Failed to deflate, ran out of buffer memory");
|
||||||
|
}
|
||||||
|
|
||||||
|
outBuffer.resize(zs.total_out);
|
||||||
|
|
||||||
|
deflateEnd(&zs);
|
||||||
|
return outBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> DataOps::ZstdInflate(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
unsigned long long const contentSize = ZSTD_getFrameContentSize(data.data(), data.size());
|
||||||
|
|
||||||
|
if (contentSize == ZSTD_CONTENTSIZE_ERROR)
|
||||||
|
return MakeUnexpected("Failed to inflate: Not valid ZSTD compressed data");
|
||||||
|
|
||||||
|
if (contentSize != ZSTD_CONTENTSIZE_UNKNOWN)
|
||||||
|
{
|
||||||
|
// FAST PATH: We know the size
|
||||||
|
Vector<UINT8> outBuffer;
|
||||||
|
outBuffer.resize(static_cast<size_t>(contentSize));
|
||||||
|
|
||||||
|
size_t const dSize = ZSTD_decompress(outBuffer.data(), outBuffer.size(), data.data(), data.size());
|
||||||
|
|
||||||
|
if (ZSTD_isError(dSize))
|
||||||
|
return MakeUnexpected(std::format("Failed to inflate: {}", ZSTD_getErrorName(dSize)));
|
||||||
|
|
||||||
|
return outBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
ZSTD_DCtx *dctx = ZSTD_createDCtx();
|
||||||
|
Vector<UINT8> outBuffer;
|
||||||
|
outBuffer.resize(data.size() * 2);
|
||||||
|
|
||||||
|
ZSTD_inBuffer input = {data.data(), data.size(), 0};
|
||||||
|
ZSTD_outBuffer output = {outBuffer.data(), outBuffer.size(), 0};
|
||||||
|
|
||||||
|
size_t ret;
|
||||||
|
do
|
||||||
|
{
|
||||||
|
ret = ZSTD_decompressStream(dctx, &output, &input);
|
||||||
|
|
||||||
|
if (ZSTD_isError(ret))
|
||||||
|
{
|
||||||
|
ZSTD_freeDCtx(dctx);
|
||||||
|
return MakeUnexpected(std::format("Failed to inflate: {}", ZSTD_getErrorName(ret)));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (output.pos == output.size)
|
||||||
|
{
|
||||||
|
size_t newSize = outBuffer.size() * 2;
|
||||||
|
outBuffer.resize(newSize);
|
||||||
|
output.dst = outBuffer.data();
|
||||||
|
output.size = newSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
} while (ret != 0);
|
||||||
|
|
||||||
|
outBuffer.resize(output.pos);
|
||||||
|
ZSTD_freeDCtx(dctx);
|
||||||
|
|
||||||
|
return outBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> DataOps::ZstdDeflate(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
size_t const maxDstSize = ZSTD_compressBound(data.size());
|
||||||
|
|
||||||
|
Vector<UINT8> outBuffer;
|
||||||
|
outBuffer.resize(maxDstSize);
|
||||||
|
|
||||||
|
size_t const compressedSize = ZSTD_compress(outBuffer.data(), maxDstSize, data.data(), data.size(), 3);
|
||||||
|
|
||||||
|
if (ZSTD_isError(compressedSize))
|
||||||
|
return MakeUnexpected(std::format("Failed to deflate: {}", ZSTD_getErrorName(compressedSize)));
|
||||||
|
|
||||||
|
outBuffer.resize(compressedSize);
|
||||||
|
return outBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> DataOps::GZipDeflate(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
z_stream zs{};
|
||||||
|
zs.zalloc = Z_NULL;
|
||||||
|
zs.zfree = Z_NULL;
|
||||||
|
zs.opaque = Z_NULL;
|
||||||
|
|
||||||
|
// WindowBits = 15 + 16 (31) -> This forces GZIP encoding
|
||||||
|
// MemLevel = 8 (default)
|
||||||
|
// Strategy = Z_DEFAULT_STRATEGY
|
||||||
|
if (deflateInit2(&zs, Z_DEFAULT_COMPRESSION, Z_DEFLATED, 15 + 16, 8, Z_DEFAULT_STRATEGY) != Z_OK)
|
||||||
|
return MakeUnexpected("Failed to initialize gzip deflate");
|
||||||
|
|
||||||
|
zs.next_in = const_cast<Bytef *>(data.data());
|
||||||
|
zs.avail_in = static_cast<uInt>(data.size());
|
||||||
|
|
||||||
|
Vector<UINT8> outBuffer;
|
||||||
|
|
||||||
|
outBuffer.resize(deflateBound(&zs, data.size()) + 1024); // Additional 1KB buffer for safety
|
||||||
|
|
||||||
|
zs.next_out = reinterpret_cast<Bytef *>(outBuffer.data());
|
||||||
|
zs.avail_out = static_cast<uInt>(outBuffer.size());
|
||||||
|
|
||||||
|
int ret = deflate(&zs, Z_FINISH);
|
||||||
|
|
||||||
|
if (ret != Z_STREAM_END)
|
||||||
|
{
|
||||||
|
deflateEnd(&zs);
|
||||||
|
return MakeUnexpected("Failed to deflate");
|
||||||
|
}
|
||||||
|
|
||||||
|
outBuffer.resize(zs.total_out);
|
||||||
|
|
||||||
|
deflateEnd(&zs);
|
||||||
|
return outBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> DataOps::GZipInflate(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
return ZlibInflate(data);
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
277
Src/IACore/imp/cpp/FileOps.cpp
Normal file
277
Src/IACore/imp/cpp/FileOps.cpp
Normal file
@ -0,0 +1,277 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/FileOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
UnorderedMap<PCUINT8, Tuple<PVOID, PVOID, PVOID>> FileOps::s_mappedFiles;
|
||||||
|
|
||||||
|
VOID FileOps::UnmapFile(IN PCUINT8 mappedPtr)
|
||||||
|
{
|
||||||
|
if (!s_mappedFiles.contains(mappedPtr))
|
||||||
|
return;
|
||||||
|
const auto handles = s_mappedFiles.extract(mappedPtr)->second;
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
::UnmapViewOfFile(std::get<1>(handles));
|
||||||
|
::CloseHandle(std::get<2>(handles));
|
||||||
|
|
||||||
|
if (std::get<0>(handles) != INVALID_HANDLE_VALUE)
|
||||||
|
::CloseHandle(std::get<0>(handles));
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
::munmap(std::get<1>(handles), (SIZE_T) std::get<2>(handles));
|
||||||
|
const auto fd = (INT32) ((UINT64) std::get<0>(handles));
|
||||||
|
if (fd != -1)
|
||||||
|
::close(fd);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<PUINT8, String> FileOps::MapSharedMemory(IN CONST String &name, IN SIZE_T size, IN BOOL isOwner)
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
int wchars_num = MultiByteToWideChar(CP_UTF8, 0, name.c_str(), -1, NULL, 0);
|
||||||
|
std::wstring wName(wchars_num, 0);
|
||||||
|
MultiByteToWideChar(CP_UTF8, 0, name.c_str(), -1, &wName[0], wchars_num);
|
||||||
|
|
||||||
|
HANDLE hMap = NULL;
|
||||||
|
if (isOwner)
|
||||||
|
hMap = CreateFileMappingW(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, (DWORD) (size >> 32),
|
||||||
|
(DWORD) (size & 0xFFFFFFFF), wName.c_str());
|
||||||
|
else
|
||||||
|
hMap = OpenFileMappingW(FILE_MAP_ALL_ACCESS, FALSE, wName.c_str());
|
||||||
|
|
||||||
|
if (hMap == NULL)
|
||||||
|
return MakeUnexpected(
|
||||||
|
std::format("Failed to {} shared memory '{}'", isOwner ? "owner" : "consumer", name.c_str()));
|
||||||
|
|
||||||
|
const auto result = static_cast<PUINT8>(MapViewOfFile(hMap, FILE_MAP_ALL_ACCESS, 0, 0, size));
|
||||||
|
if (result == NULL)
|
||||||
|
{
|
||||||
|
CloseHandle(hMap);
|
||||||
|
return MakeUnexpected(std::format("Failed to map view of shared memory '{}'", name.c_str()));
|
||||||
|
}
|
||||||
|
|
||||||
|
s_mappedFiles[result] = std::make_tuple((PVOID) INVALID_HANDLE_VALUE, (PVOID) result, (PVOID) hMap);
|
||||||
|
return result;
|
||||||
|
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
int fd = -1;
|
||||||
|
if (isOwner)
|
||||||
|
{
|
||||||
|
fd = shm_open(name.c_str(), O_RDWR | O_CREAT | O_TRUNC, 0666);
|
||||||
|
if (fd != -1)
|
||||||
|
{
|
||||||
|
if (ftruncate(fd, size) == -1)
|
||||||
|
{
|
||||||
|
close(fd);
|
||||||
|
shm_unlink(name.c_str());
|
||||||
|
return MakeUnexpected(std::format("Failed to truncate shared memory '{}'", name.c_str()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
fd = shm_open(name.c_str(), O_RDWR, 0666);
|
||||||
|
|
||||||
|
if (fd == -1)
|
||||||
|
return MakeUnexpected(
|
||||||
|
std::format("Failed to {} shared memory '{}'", isOwner ? "owner" : "consumer", name.c_str()));
|
||||||
|
|
||||||
|
void *addr = mmap(nullptr, size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
|
||||||
|
if (addr == MAP_FAILED)
|
||||||
|
{
|
||||||
|
close(fd);
|
||||||
|
return MakeUnexpected(std::format("Failed to mmap shared memory '{}'", name.c_str()));
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto result = static_cast<PUINT8>(addr);
|
||||||
|
|
||||||
|
s_mappedFiles[result] = std::make_tuple((PVOID) ((UINT64) fd), (PVOID) addr, (PVOID) size);
|
||||||
|
return result;
|
||||||
|
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID FileOps::UnlinkSharedMemory(IN CONST String &name)
|
||||||
|
{
|
||||||
|
if (name.empty())
|
||||||
|
return;
|
||||||
|
#if IA_PLATFORM_UNIX
|
||||||
|
shm_unlink(name.c_str());
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<PCUINT8, String> FileOps::MapFile(IN CONST FilePath &path, OUT SIZE_T &size)
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
|
||||||
|
const auto handle = CreateFileA(path.string().c_str(), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
|
||||||
|
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
|
||||||
|
|
||||||
|
if (handle == INVALID_HANDLE_VALUE)
|
||||||
|
return MakeUnexpected(std::format("Failed to open {} for memory mapping", path.string().c_str()));
|
||||||
|
|
||||||
|
LARGE_INTEGER fileSize;
|
||||||
|
if (!GetFileSizeEx(handle, &fileSize))
|
||||||
|
{
|
||||||
|
CloseHandle(handle);
|
||||||
|
return MakeUnexpected(std::format("Failed to get size of {} for memory mapping", path.string().c_str()));
|
||||||
|
}
|
||||||
|
size = static_cast<size_t>(fileSize.QuadPart);
|
||||||
|
if (size == 0)
|
||||||
|
{
|
||||||
|
CloseHandle(handle);
|
||||||
|
return MakeUnexpected(std::format("Failed to get size of {} for memory mapping", path.string().c_str()));
|
||||||
|
}
|
||||||
|
|
||||||
|
auto hmap = CreateFileMappingW(handle, NULL, PAGE_READONLY, 0, 0, NULL);
|
||||||
|
if (hmap == NULL)
|
||||||
|
{
|
||||||
|
CloseHandle(handle);
|
||||||
|
return MakeUnexpected(std::format("Failed to memory map {}", path.string().c_str()));
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto result = static_cast<PCUINT8>(MapViewOfFile(hmap, FILE_MAP_READ, 0, 0, 0));
|
||||||
|
if (result == NULL)
|
||||||
|
{
|
||||||
|
CloseHandle(handle);
|
||||||
|
CloseHandle(hmap);
|
||||||
|
return MakeUnexpected(std::format("Failed to memory map {}", path.string().c_str()));
|
||||||
|
}
|
||||||
|
s_mappedFiles[result] = std::make_tuple((PVOID) handle, (PVOID) result, (PVOID) hmap);
|
||||||
|
return result;
|
||||||
|
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
|
||||||
|
const auto handle = open(path.string().c_str(), O_RDONLY);
|
||||||
|
if (handle == -1)
|
||||||
|
return MakeUnexpected(std::format("Failed to open {} for memory mapping", path.string().c_str()));
|
||||||
|
struct stat sb;
|
||||||
|
if (fstat(handle, &sb) == -1)
|
||||||
|
{
|
||||||
|
close(handle);
|
||||||
|
return MakeUnexpected(std::format("Failed to get stats of {} for memory mapping", path.string().c_str()));
|
||||||
|
}
|
||||||
|
size = static_cast<size_t>(sb.st_size);
|
||||||
|
if (size == 0)
|
||||||
|
{
|
||||||
|
close(handle);
|
||||||
|
return MakeUnexpected(std::format("Failed to get size of {} for memory mapping", path.string().c_str()));
|
||||||
|
}
|
||||||
|
void *addr = mmap(nullptr, size, PROT_READ, MAP_PRIVATE, handle, 0);
|
||||||
|
if (addr == MAP_FAILED)
|
||||||
|
{
|
||||||
|
close(handle);
|
||||||
|
return MakeUnexpected(std::format("Failed to memory map {}", path.string().c_str()));
|
||||||
|
}
|
||||||
|
const auto result = static_cast<PCUINT8>(addr);
|
||||||
|
madvise(addr, size, MADV_SEQUENTIAL);
|
||||||
|
s_mappedFiles[result] = std::make_tuple((PVOID) ((UINT64) handle), (PVOID) addr, (PVOID) size);
|
||||||
|
return result;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<StreamWriter, String> FileOps::StreamToFile(IN CONST FilePath &path, IN BOOL overwrite)
|
||||||
|
{
|
||||||
|
if (!overwrite && FileSystem::exists(path))
|
||||||
|
return MakeUnexpected(std::format("File aready exists: {}", path.string().c_str()));
|
||||||
|
return StreamWriter(path);
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<StreamReader, String> FileOps::StreamFromFile(IN CONST FilePath &path)
|
||||||
|
{
|
||||||
|
if (!FileSystem::exists(path))
|
||||||
|
return MakeUnexpected(std::format("File does not exist: {}", path.string().c_str()));
|
||||||
|
return StreamReader(path);
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<String, String> FileOps::ReadTextFile(IN CONST FilePath &path)
|
||||||
|
{
|
||||||
|
const auto f = fopen(path.string().c_str(), "r");
|
||||||
|
if (!f)
|
||||||
|
return MakeUnexpected(std::format("Failed to open file: {}", path.string().c_str()));
|
||||||
|
String result;
|
||||||
|
fseek(f, 0, SEEK_END);
|
||||||
|
result.resize(ftell(f));
|
||||||
|
fseek(f, 0, SEEK_SET);
|
||||||
|
fread(result.data(), 1, result.size(), f);
|
||||||
|
fclose(f);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Vector<UINT8>, String> FileOps::ReadBinaryFile(IN CONST FilePath &path)
|
||||||
|
{
|
||||||
|
const auto f = fopen(path.string().c_str(), "rb");
|
||||||
|
if (!f)
|
||||||
|
return MakeUnexpected(std::format("Failed to open file: {}", path.string().c_str()));
|
||||||
|
Vector<UINT8> result;
|
||||||
|
fseek(f, 0, SEEK_END);
|
||||||
|
result.resize(ftell(f));
|
||||||
|
fseek(f, 0, SEEK_SET);
|
||||||
|
fread(result.data(), 1, result.size(), f);
|
||||||
|
fclose(f);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<SIZE_T, String> FileOps::WriteTextFile(IN CONST FilePath &path, IN CONST String &contents,
|
||||||
|
IN BOOL overwrite)
|
||||||
|
{
|
||||||
|
if (!overwrite && FileSystem::exists(path))
|
||||||
|
return MakeUnexpected(std::format("File aready exists: {}", path.string().c_str()));
|
||||||
|
const auto f = fopen(path.string().c_str(), "w");
|
||||||
|
if (!f)
|
||||||
|
return MakeUnexpected(std::format("Failed to write to file: {}", path.string().c_str()));
|
||||||
|
const auto result = fwrite(contents.data(), 1, contents.size(), f);
|
||||||
|
fputc(0, f);
|
||||||
|
fclose(f);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<SIZE_T, String> FileOps::WriteBinaryFile(IN CONST FilePath &path, IN Span<UINT8> contents,
|
||||||
|
IN BOOL overwrite)
|
||||||
|
{
|
||||||
|
if (!overwrite && FileSystem::exists(path))
|
||||||
|
return MakeUnexpected(std::format("File aready exists: {}", path.string().c_str()));
|
||||||
|
|
||||||
|
const auto f = fopen(path.string().c_str(), "wb");
|
||||||
|
if (!f)
|
||||||
|
return MakeUnexpected(std::format("Failed to write to file: {}", path.string().c_str()));
|
||||||
|
const auto result = fwrite(contents.data(), 1, contents.size(), f);
|
||||||
|
fclose(f);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
FilePath FileOps::NormalizeExecutablePath(IN CONST FilePath &path)
|
||||||
|
{
|
||||||
|
FilePath result = path;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
if (!result.has_extension())
|
||||||
|
result.replace_extension(".exe");
|
||||||
|
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
if (result.extension() == ".exe")
|
||||||
|
result.replace_extension("");
|
||||||
|
|
||||||
|
if (result.is_relative())
|
||||||
|
{
|
||||||
|
String pathStr = result.string();
|
||||||
|
if (!pathStr.starts_with("./") && !pathStr.starts_with("../"))
|
||||||
|
result = "./" + pathStr;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
244
Src/IACore/imp/cpp/HttpClient.cpp
Normal file
244
Src/IACore/imp/cpp/HttpClient.cpp
Normal file
@ -0,0 +1,244 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/HttpClient.hpp>
|
||||||
|
#include <IACore/DataOps.hpp>
|
||||||
|
|
||||||
|
#include <httplib.h>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
httplib::Headers BuildHeaders(IN Span<CONST HttpClient::Header> headers, IN PCCHAR defaultContentType)
|
||||||
|
{
|
||||||
|
httplib::Headers out;
|
||||||
|
bool hasContentType = false;
|
||||||
|
|
||||||
|
for (const auto &h : headers)
|
||||||
|
{
|
||||||
|
std::string key = HttpClient::HeaderTypeToString(h.first);
|
||||||
|
out.emplace(key, h.second);
|
||||||
|
|
||||||
|
if (h.first == HttpClient::EHeaderType::CONTENT_TYPE)
|
||||||
|
hasContentType = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!hasContentType && defaultContentType)
|
||||||
|
out.emplace("Content-Type", defaultContentType);
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
HttpClient::HttpClient(IN CONST String &host)
|
||||||
|
: m_client(new httplib::Client(host)), m_lastResponseCode(EResponseCode::INTERNAL_SERVER_ERROR)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
HttpClient::~HttpClient()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
String HttpClient::PreprocessResponse(IN CONST String &response)
|
||||||
|
{
|
||||||
|
const auto responseBytes = Span<CONST UINT8>{(PCUINT8) response.data(), response.size()};
|
||||||
|
const auto compression = DataOps::DetectCompression(responseBytes);
|
||||||
|
switch (compression)
|
||||||
|
{
|
||||||
|
case DataOps::CompressionType::Gzip: {
|
||||||
|
const auto data = DataOps::GZipInflate(responseBytes);
|
||||||
|
if (!data)
|
||||||
|
return response;
|
||||||
|
return String((PCCHAR) data->data(), data->size());
|
||||||
|
}
|
||||||
|
|
||||||
|
case DataOps::CompressionType::Zlib: {
|
||||||
|
const auto data = DataOps::ZlibInflate(responseBytes);
|
||||||
|
if (!data)
|
||||||
|
return response;
|
||||||
|
return String((PCCHAR) data->data(), data->size());
|
||||||
|
}
|
||||||
|
|
||||||
|
case DataOps::CompressionType::None:
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return response;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<String, String> HttpClient::RawGet(IN CONST String &path, IN Span<CONST Header> headers,
|
||||||
|
IN PCCHAR defaultContentType)
|
||||||
|
{
|
||||||
|
auto httpHeaders = BuildHeaders(headers, defaultContentType);
|
||||||
|
|
||||||
|
static_cast<httplib::Client *>(m_client)->enable_server_certificate_verification(false);
|
||||||
|
auto res = static_cast<httplib::Client *>(m_client)->Get(
|
||||||
|
(!path.empty() && path[0] != '/') ? ('/' + path).c_str() : path.c_str(), httpHeaders);
|
||||||
|
|
||||||
|
if (res)
|
||||||
|
{
|
||||||
|
m_lastResponseCode = static_cast<EResponseCode>(res->status);
|
||||||
|
if (res->status >= 200 && res->status < 300)
|
||||||
|
return PreprocessResponse(res->body);
|
||||||
|
else
|
||||||
|
return MakeUnexpected(std::format("HTTP Error {} : {}", res->status, res->body));
|
||||||
|
}
|
||||||
|
|
||||||
|
return MakeUnexpected(std::format("Network Error: {}", httplib::to_string(res.error())));
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<String, String> HttpClient::RawPost(IN CONST String &path, IN Span<CONST Header> headers,
|
||||||
|
IN CONST String &body, IN PCCHAR defaultContentType)
|
||||||
|
{
|
||||||
|
auto httpHeaders = BuildHeaders(headers, defaultContentType);
|
||||||
|
|
||||||
|
String contentType = defaultContentType;
|
||||||
|
if (httpHeaders.count("Content-Type"))
|
||||||
|
{
|
||||||
|
const auto t = httpHeaders.find("Content-Type");
|
||||||
|
contentType = t->second;
|
||||||
|
httpHeaders.erase(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
static_cast<httplib::Client *>(m_client)->set_keep_alive(true);
|
||||||
|
static_cast<httplib::Client *>(m_client)->enable_server_certificate_verification(false);
|
||||||
|
auto res = static_cast<httplib::Client *>(m_client)->Post(
|
||||||
|
(!path.empty() && path[0] != '/') ? ('/' + path).c_str() : path.c_str(), httpHeaders, body,
|
||||||
|
contentType.c_str());
|
||||||
|
|
||||||
|
if (res)
|
||||||
|
{
|
||||||
|
m_lastResponseCode = static_cast<EResponseCode>(res->status);
|
||||||
|
if (res->status >= 200 && res->status < 300)
|
||||||
|
return PreprocessResponse(res->body);
|
||||||
|
else
|
||||||
|
return MakeUnexpected(std::format("HTTP Error {} : {}", res->status, res->body));
|
||||||
|
}
|
||||||
|
|
||||||
|
return MakeUnexpected(std::format("Network Error: {}", httplib::to_string(res.error())));
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
HttpClient::Header HttpClient::CreateHeader(IN EHeaderType key, IN CONST String &value)
|
||||||
|
{
|
||||||
|
return std::make_pair(key, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
String HttpClient::UrlEncode(IN CONST String &value)
|
||||||
|
{
|
||||||
|
std::stringstream escaped;
|
||||||
|
escaped.fill('0');
|
||||||
|
escaped << std::hex << std::uppercase;
|
||||||
|
|
||||||
|
for (char c : value)
|
||||||
|
{
|
||||||
|
if (std::isalnum(static_cast<unsigned char>(c)) || c == '-' || c == '_' || c == '.' || c == '~')
|
||||||
|
escaped << c;
|
||||||
|
else
|
||||||
|
escaped << '%' << std::setw(2) << static_cast<int>(static_cast<unsigned char>(c));
|
||||||
|
}
|
||||||
|
|
||||||
|
return escaped.str();
|
||||||
|
}
|
||||||
|
|
||||||
|
String HttpClient::UrlDecode(IN CONST String &value)
|
||||||
|
{
|
||||||
|
String result;
|
||||||
|
result.reserve(value.length());
|
||||||
|
|
||||||
|
for (size_t i = 0; i < value.length(); ++i)
|
||||||
|
{
|
||||||
|
if (value[i] == '%' && i + 2 < value.length())
|
||||||
|
{
|
||||||
|
std::string hexStr = value.substr(i + 1, 2);
|
||||||
|
char decodedChar = static_cast<char>(std::strtol(hexStr.c_str(), nullptr, 16));
|
||||||
|
result += decodedChar;
|
||||||
|
i += 2;
|
||||||
|
}
|
||||||
|
else if (value[i] == '+')
|
||||||
|
result += ' ';
|
||||||
|
else
|
||||||
|
result += value[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
String HttpClient::HeaderTypeToString(IN EHeaderType type)
|
||||||
|
{
|
||||||
|
switch (type)
|
||||||
|
{
|
||||||
|
case EHeaderType::ACCEPT:
|
||||||
|
return "Accept";
|
||||||
|
case EHeaderType::ACCEPT_CHARSET:
|
||||||
|
return "Accept-Charset";
|
||||||
|
case EHeaderType::ACCEPT_ENCODING:
|
||||||
|
return "Accept-Encoding";
|
||||||
|
case EHeaderType::ACCEPT_LANGUAGE:
|
||||||
|
return "Accept-Language";
|
||||||
|
case EHeaderType::AUTHORIZATION:
|
||||||
|
return "Authorization";
|
||||||
|
case EHeaderType::CACHE_CONTROL:
|
||||||
|
return "Cache-Control";
|
||||||
|
case EHeaderType::CONNECTION:
|
||||||
|
return "Connection";
|
||||||
|
case EHeaderType::CONTENT_LENGTH:
|
||||||
|
return "Content-Length";
|
||||||
|
case EHeaderType::CONTENT_TYPE:
|
||||||
|
return "Content-Type";
|
||||||
|
case EHeaderType::COOKIE:
|
||||||
|
return "Cookie";
|
||||||
|
case EHeaderType::DATE:
|
||||||
|
return "Date";
|
||||||
|
case EHeaderType::EXPECT:
|
||||||
|
return "Expect";
|
||||||
|
case EHeaderType::HOST:
|
||||||
|
return "Host";
|
||||||
|
case EHeaderType::IF_MATCH:
|
||||||
|
return "If-Match";
|
||||||
|
case EHeaderType::IF_MODIFIED_SINCE:
|
||||||
|
return "If-Modified-Since";
|
||||||
|
case EHeaderType::IF_NONE_MATCH:
|
||||||
|
return "If-None-Match";
|
||||||
|
case EHeaderType::ORIGIN:
|
||||||
|
return "Origin";
|
||||||
|
case EHeaderType::PRAGMA:
|
||||||
|
return "Pragma";
|
||||||
|
case EHeaderType::PROXY_AUTHORIZATION:
|
||||||
|
return "Proxy-Authorization";
|
||||||
|
case EHeaderType::RANGE:
|
||||||
|
return "Range";
|
||||||
|
case EHeaderType::REFERER:
|
||||||
|
return "Referer";
|
||||||
|
case EHeaderType::TE:
|
||||||
|
return "TE";
|
||||||
|
case EHeaderType::UPGRADE:
|
||||||
|
return "Upgrade";
|
||||||
|
case EHeaderType::USER_AGENT:
|
||||||
|
return "User-Agent";
|
||||||
|
case EHeaderType::VIA:
|
||||||
|
return "Via";
|
||||||
|
case EHeaderType::WARNING:
|
||||||
|
return "Warning";
|
||||||
|
default:
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL HttpClient::IsSuccessResponseCode(IN EResponseCode code)
|
||||||
|
{
|
||||||
|
return (INT32) code >= 200 && (INT32) code < 300;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
81
Src/IACore/imp/cpp/IACore.cpp
Normal file
81
Src/IACore/imp/cpp/IACore.cpp
Normal file
@ -0,0 +1,81 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/IACore.hpp>
|
||||||
|
#include <IACore/Logger.hpp>
|
||||||
|
|
||||||
|
#include <mimalloc-new-delete.h>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
HighResTimePoint g_startTime{};
|
||||||
|
std::thread::id g_mainThreadID{};
|
||||||
|
|
||||||
|
VOID Initialize()
|
||||||
|
{
|
||||||
|
g_mainThreadID = std::this_thread::get_id();
|
||||||
|
g_startTime = HighResClock::now();
|
||||||
|
Logger::Initialize();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Terminate()
|
||||||
|
{
|
||||||
|
Logger::Terminate();
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT64 GetUnixTime()
|
||||||
|
{
|
||||||
|
auto now = std::chrono::system_clock::now();
|
||||||
|
return std::chrono::duration_cast<std::chrono::seconds>(now.time_since_epoch()).count();
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT64 GetTicksCount()
|
||||||
|
{
|
||||||
|
return std::chrono::duration_cast<std::chrono::milliseconds>(HighResClock::now() - g_startTime).count();
|
||||||
|
}
|
||||||
|
|
||||||
|
FLOAT64 GetSecondsCount()
|
||||||
|
{
|
||||||
|
return std::chrono::duration_cast<std::chrono::seconds>(HighResClock::now() - g_startTime).count();
|
||||||
|
}
|
||||||
|
|
||||||
|
FLOAT32 GetRandom()
|
||||||
|
{
|
||||||
|
return static_cast<FLOAT32>(rand()) / static_cast<FLOAT32>(RAND_MAX);
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT32 GetRandom(IN UINT32 seed)
|
||||||
|
{
|
||||||
|
srand(seed);
|
||||||
|
return (UINT32) GetRandom(0, UINT32_MAX);
|
||||||
|
}
|
||||||
|
|
||||||
|
INT64 GetRandom(IN INT64 min, IN INT64 max)
|
||||||
|
{
|
||||||
|
const auto t = static_cast<FLOAT32>(rand()) / static_cast<FLOAT32>(RAND_MAX);
|
||||||
|
return min + (max - min) * t;
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Sleep(IN UINT64 milliseconds)
|
||||||
|
{
|
||||||
|
std::this_thread::sleep_for(std::chrono::milliseconds(milliseconds));
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL IsMainThread()
|
||||||
|
{
|
||||||
|
return std::this_thread::get_id() == g_mainThreadID;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
451
Src/IACore/imp/cpp/IPC.cpp
Normal file
451
Src/IACore/imp/cpp/IPC.cpp
Normal file
@ -0,0 +1,451 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/IPC.hpp>
|
||||||
|
|
||||||
|
#include <IACore/FileOps.hpp>
|
||||||
|
#include <IACore/StringOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
struct IPC_ConnectionDescriptor
|
||||||
|
{
|
||||||
|
String SocketPath;
|
||||||
|
String SharedMemPath;
|
||||||
|
UINT32 SharedMemSize;
|
||||||
|
|
||||||
|
String Serialize() CONST
|
||||||
|
{
|
||||||
|
return std::format("{}|{}|{}|", SocketPath, SharedMemPath, SharedMemSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC IPC_ConnectionDescriptor Deserialize(IN CONST String &data)
|
||||||
|
{
|
||||||
|
enum class EParseState
|
||||||
|
{
|
||||||
|
SocketPath,
|
||||||
|
SharedMemPath,
|
||||||
|
SharedMemSize
|
||||||
|
};
|
||||||
|
|
||||||
|
IPC_ConnectionDescriptor result{};
|
||||||
|
|
||||||
|
SIZE_T t{};
|
||||||
|
EParseState state{EParseState::SocketPath};
|
||||||
|
for (SIZE_T i = 0; i < data.size(); i++)
|
||||||
|
{
|
||||||
|
if (data[i] != '|')
|
||||||
|
continue;
|
||||||
|
|
||||||
|
switch (state)
|
||||||
|
{
|
||||||
|
case EParseState::SocketPath:
|
||||||
|
result.SocketPath = data.substr(t, i - t);
|
||||||
|
state = EParseState::SharedMemPath;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case EParseState::SharedMemPath:
|
||||||
|
result.SharedMemPath = data.substr(t, i - t);
|
||||||
|
state = EParseState::SharedMemSize;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case EParseState::SharedMemSize: {
|
||||||
|
if (std::from_chars(&data[t], &data[i], result.SharedMemSize).ec != std::errc{})
|
||||||
|
return {};
|
||||||
|
goto done_parsing;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
t = i + 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
done_parsing:
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
IPC_Node::~IPC_Node()
|
||||||
|
{
|
||||||
|
SocketOps::Close(m_socket); // SocketOps gracefully handles INVALID_SOCKET
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<VOID, String> IPC_Node::Connect(IN PCCHAR connectionString)
|
||||||
|
{
|
||||||
|
auto desc = IPC_ConnectionDescriptor::Deserialize(connectionString);
|
||||||
|
m_shmName = desc.SharedMemPath;
|
||||||
|
|
||||||
|
m_socket = SocketOps::CreateUnixSocket();
|
||||||
|
if (!SocketOps::ConnectUnixSocket(m_socket, desc.SocketPath.c_str()))
|
||||||
|
return MakeUnexpected("Failed to create an unix socket");
|
||||||
|
|
||||||
|
auto mapRes = FileOps::MapSharedMemory(desc.SharedMemPath, desc.SharedMemSize, FALSE);
|
||||||
|
if (!mapRes.has_value())
|
||||||
|
return MakeUnexpected("Failed to map the shared memory");
|
||||||
|
|
||||||
|
m_sharedMemory = mapRes.value();
|
||||||
|
|
||||||
|
auto *layout = reinterpret_cast<IPC_SharedMemoryLayout *>(m_sharedMemory);
|
||||||
|
|
||||||
|
if (layout->Meta.Magic != 0x49414950) // "IAIP"
|
||||||
|
return MakeUnexpected("Invalid shared memory header signature");
|
||||||
|
|
||||||
|
if (layout->Meta.Version != 1)
|
||||||
|
return MakeUnexpected("IPC version mismatch");
|
||||||
|
|
||||||
|
PUINT8 moniDataPtr = m_sharedMemory + layout->MONI_DataOffset;
|
||||||
|
PUINT8 minoDataPtr = m_sharedMemory + layout->MINO_DataOffset;
|
||||||
|
|
||||||
|
MONI = std::make_unique<RingBufferView>(
|
||||||
|
&layout->MONI_Control, Span<UINT8>(moniDataPtr, static_cast<size_t>(layout->MONI_DataSize)), FALSE);
|
||||||
|
|
||||||
|
MINO = std::make_unique<RingBufferView>(
|
||||||
|
&layout->MINO_Control, Span<UINT8>(minoDataPtr, static_cast<size_t>(layout->MINO_DataSize)), FALSE);
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
u_long mode = 1;
|
||||||
|
ioctlsocket(m_socket, FIONBIO, &mode);
|
||||||
|
#else
|
||||||
|
fcntl(m_socket, F_SETFL, O_NONBLOCK);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
m_recieveBuffer.resize(UINT16_MAX + 1);
|
||||||
|
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Node::Update()
|
||||||
|
{
|
||||||
|
if (!MONI)
|
||||||
|
return;
|
||||||
|
|
||||||
|
RingBufferView::PacketHeader header;
|
||||||
|
|
||||||
|
// Process all available messages from Manager
|
||||||
|
while (MONI->Pop(header, Span<UINT8>(m_recieveBuffer.data(), m_recieveBuffer.size())))
|
||||||
|
OnPacket(header.ID, {m_recieveBuffer.data(), header.PayloadSize});
|
||||||
|
|
||||||
|
UINT8 signal;
|
||||||
|
const auto res = recv(m_socket, (CHAR *) &signal, 1, 0);
|
||||||
|
if (res == 1)
|
||||||
|
OnSignal(signal);
|
||||||
|
else if (res == 0 || (res < 0 && !SocketOps::IsWouldBlock()))
|
||||||
|
{
|
||||||
|
SocketOps::Close(m_socket);
|
||||||
|
FileOps::UnlinkSharedMemory(m_shmName);
|
||||||
|
|
||||||
|
// Manager disconnected, exit immediately
|
||||||
|
exit(-1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Node::SendSignal(IN UINT8 signal)
|
||||||
|
{
|
||||||
|
if (IS_VALID_SOCKET(m_socket))
|
||||||
|
send(m_socket, (const char *) &signal, sizeof(signal), 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Node::SendPacket(IN UINT16 packetID, IN Span<CONST UINT8> payload)
|
||||||
|
{
|
||||||
|
MINO->Push(packetID, payload);
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
VOID IPC_Manager::NodeSession::SendSignal(IN UINT8 signal)
|
||||||
|
{
|
||||||
|
if (IS_VALID_SOCKET(DataSocket))
|
||||||
|
send(DataSocket, (const char *) &signal, sizeof(signal), 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Manager::NodeSession::SendPacket(IN UINT16 packetID, IN Span<CONST UINT8> payload)
|
||||||
|
{
|
||||||
|
MONI->Push(packetID, payload);
|
||||||
|
}
|
||||||
|
|
||||||
|
IPC_Manager::IPC_Manager()
|
||||||
|
{
|
||||||
|
// SocketOps is smart enough to handle multiple inits
|
||||||
|
SocketOps::Initialize();
|
||||||
|
|
||||||
|
m_recieveBuffer.resize(UINT16_MAX + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
IPC_Manager::~IPC_Manager()
|
||||||
|
{
|
||||||
|
for (auto &session : m_activeSessions)
|
||||||
|
{
|
||||||
|
ProcessOps::TerminateProcess(session->ProcessHandle);
|
||||||
|
FileOps::UnmapFile(session->MappedPtr);
|
||||||
|
FileOps::UnlinkSharedMemory(session->SharedMemName);
|
||||||
|
SocketOps::Close(session->DataSocket);
|
||||||
|
}
|
||||||
|
m_activeSessions.clear();
|
||||||
|
|
||||||
|
for (auto &session : m_pendingSessions)
|
||||||
|
{
|
||||||
|
ProcessOps::TerminateProcess(session->ProcessHandle);
|
||||||
|
FileOps::UnmapFile(session->MappedPtr);
|
||||||
|
FileOps::UnlinkSharedMemory(session->SharedMemName);
|
||||||
|
SocketOps::Close(session->ListenerSocket);
|
||||||
|
}
|
||||||
|
m_pendingSessions.clear();
|
||||||
|
|
||||||
|
// SocketOps is smart enough to handle multiple terminates
|
||||||
|
SocketOps::Terminate();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Manager::Update()
|
||||||
|
{
|
||||||
|
const auto now = SteadyClock::now();
|
||||||
|
|
||||||
|
for (INT32 i = m_pendingSessions.size() - 1; i >= 0; i--)
|
||||||
|
{
|
||||||
|
auto &session = m_pendingSessions[i];
|
||||||
|
|
||||||
|
if (now - session->CreationTime > std::chrono::seconds(5))
|
||||||
|
{
|
||||||
|
ProcessOps::TerminateProcess(session->ProcessHandle);
|
||||||
|
|
||||||
|
FileOps::UnmapFile(session->MappedPtr);
|
||||||
|
FileOps::UnlinkSharedMemory(session->SharedMemName);
|
||||||
|
SocketOps::Close(session->DataSocket);
|
||||||
|
|
||||||
|
m_pendingSessions.erase(m_pendingSessions.begin() + i);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
SocketHandle newSock = accept(session->ListenerSocket, NULL, NULL);
|
||||||
|
|
||||||
|
if (IS_VALID_SOCKET(newSock))
|
||||||
|
{
|
||||||
|
session->DataSocket = newSock;
|
||||||
|
session->IsReady = TRUE;
|
||||||
|
|
||||||
|
// Set Data Socket to Non-Blocking
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
u_long mode = 1;
|
||||||
|
ioctlsocket(session->DataSocket, FIONBIO, &mode);
|
||||||
|
#else
|
||||||
|
fcntl(session->DataSocket, F_SETFL, O_NONBLOCK);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
SocketOps::Close(session->ListenerSocket);
|
||||||
|
session->ListenerSocket = INVALID_SOCKET;
|
||||||
|
|
||||||
|
const auto sessionID = session->ProcessHandle->ID.load();
|
||||||
|
const auto sessionPtr = session.get();
|
||||||
|
m_activeSessions.push_back(std::move(session));
|
||||||
|
m_pendingSessions.erase(m_pendingSessions.begin() + i);
|
||||||
|
m_activeSessionMap[sessionID] = sessionPtr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (INT32 i = m_activeSessions.size() - 1; i >= 0; i--)
|
||||||
|
{
|
||||||
|
auto &node = m_activeSessions[i];
|
||||||
|
|
||||||
|
auto nodeID = node->ProcessHandle->ID.load();
|
||||||
|
|
||||||
|
RingBufferView::PacketHeader header;
|
||||||
|
|
||||||
|
while (node->MINO->Pop(header, Span<UINT8>(m_recieveBuffer.data(), m_recieveBuffer.size())))
|
||||||
|
OnPacket(nodeID, header.ID, {m_recieveBuffer.data(), header.PayloadSize});
|
||||||
|
|
||||||
|
UINT8 signal;
|
||||||
|
const auto res = recv(node->DataSocket, (CHAR *) &signal, 1, 0);
|
||||||
|
|
||||||
|
if (res == 1)
|
||||||
|
OnSignal(nodeID, signal);
|
||||||
|
else if (res == 0 || (res < 0 && !SocketOps::IsWouldBlock()))
|
||||||
|
{
|
||||||
|
ProcessOps::TerminateProcess(node->ProcessHandle);
|
||||||
|
|
||||||
|
FileOps::UnmapFile(node->MappedPtr);
|
||||||
|
FileOps::UnlinkSharedMemory(node->SharedMemName);
|
||||||
|
SocketOps::Close(node->DataSocket);
|
||||||
|
|
||||||
|
m_activeSessions.erase(m_activeSessions.begin() + i);
|
||||||
|
m_activeSessionMap.erase(nodeID);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<NativeProcessID, String> IPC_Manager::SpawnNode(IN CONST FilePath &executablePath,
|
||||||
|
IN UINT32 sharedMemorySize)
|
||||||
|
{
|
||||||
|
auto session = std::make_unique<NodeSession>();
|
||||||
|
|
||||||
|
static Atomic<UINT32> s_idGen{0};
|
||||||
|
UINT32 sid = ++s_idGen;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
char tempPath[MAX_PATH];
|
||||||
|
GetTempPathA(MAX_PATH, tempPath);
|
||||||
|
String sockPath = std::format("{}\\ia_sess_{}.sock", tempPath, sid);
|
||||||
|
#else
|
||||||
|
String sockPath = std::format("/tmp/ia_sess_{}.sock", sid);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
session->ListenerSocket = SocketOps::CreateUnixSocket();
|
||||||
|
if (!SocketOps::BindUnixSocket(session->ListenerSocket, sockPath.c_str()))
|
||||||
|
return MakeUnexpected("Failed to bind unique socket");
|
||||||
|
|
||||||
|
if (listen(session->ListenerSocket, 1) != 0)
|
||||||
|
return MakeUnexpected("Failed to listen on unqiue socket");
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
u_long mode = 1;
|
||||||
|
ioctlsocket(session->ListenerSocket, FIONBIO, &mode);
|
||||||
|
#else
|
||||||
|
fcntl(session->ListenerSocket, F_SETFL, O_NONBLOCK);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
String shmName = std::format("ia_shm_{}", sid);
|
||||||
|
auto mapRes = FileOps::MapSharedMemory(shmName, sharedMemorySize, TRUE);
|
||||||
|
if (!mapRes.has_value())
|
||||||
|
return MakeUnexpected("Failed to map shared memory");
|
||||||
|
|
||||||
|
PUINT8 mappedPtr = mapRes.value();
|
||||||
|
|
||||||
|
auto *layout = reinterpret_cast<IPC_SharedMemoryLayout *>(mappedPtr);
|
||||||
|
|
||||||
|
layout->Meta.Magic = 0x49414950;
|
||||||
|
layout->Meta.Version = 1;
|
||||||
|
layout->Meta.TotalSize = sharedMemorySize;
|
||||||
|
|
||||||
|
UINT64 headerSize = IPC_SharedMemoryLayout::GetHeaderSize();
|
||||||
|
UINT64 usableBytes = sharedMemorySize - headerSize;
|
||||||
|
|
||||||
|
UINT64 halfSize = (usableBytes / 2);
|
||||||
|
halfSize -= (halfSize % 64);
|
||||||
|
|
||||||
|
layout->MONI_DataOffset = headerSize;
|
||||||
|
layout->MONI_DataSize = halfSize;
|
||||||
|
|
||||||
|
layout->MINO_DataOffset = headerSize + halfSize;
|
||||||
|
layout->MINO_DataSize = halfSize;
|
||||||
|
|
||||||
|
session->MONI = std::make_unique<RingBufferView>(
|
||||||
|
&layout->MONI_Control, Span<UINT8>(mappedPtr + layout->MONI_DataOffset, layout->MONI_DataSize), TRUE);
|
||||||
|
|
||||||
|
session->MINO = std::make_unique<RingBufferView>(
|
||||||
|
&layout->MINO_Control, Span<UINT8>(mappedPtr + layout->MINO_DataOffset, layout->MINO_DataSize), TRUE);
|
||||||
|
|
||||||
|
IPC_ConnectionDescriptor desc;
|
||||||
|
desc.SocketPath = sockPath;
|
||||||
|
desc.SharedMemPath = shmName;
|
||||||
|
desc.SharedMemSize = sharedMemorySize;
|
||||||
|
|
||||||
|
String args = std::format("\"{}\"", desc.Serialize());
|
||||||
|
|
||||||
|
session->ProcessHandle = ProcessOps::SpawnProcessAsync(
|
||||||
|
FileOps::NormalizeExecutablePath(executablePath).string(), args,
|
||||||
|
[sid](IN StringView line) {
|
||||||
|
UNUSED(sid);
|
||||||
|
UNUSED(line);
|
||||||
|
#if __IA_DEBUG
|
||||||
|
puts(std::format(__CC_MAGENTA "[Node:{}:STDOUT|STDERR]: {}" __CC_DEFAULT, sid, line).c_str());
|
||||||
|
#endif
|
||||||
|
},
|
||||||
|
[sid](IN Expected<INT32, String> result) {
|
||||||
|
UNUSED(sid);
|
||||||
|
UNUSED(result);
|
||||||
|
#if __IA_DEBUG
|
||||||
|
if (!result)
|
||||||
|
puts(std::format(__CC_RED "Failed to spawn Node: {} with error '{}'" __CC_DEFAULT, sid,
|
||||||
|
result.error())
|
||||||
|
.c_str());
|
||||||
|
else
|
||||||
|
puts(std::format(__CC_RED "[Node: {}]: Exited with code {}" __CC_DEFAULT, sid, *result).c_str());
|
||||||
|
#endif
|
||||||
|
});
|
||||||
|
|
||||||
|
// Give some time for child node to stablize
|
||||||
|
std::this_thread::sleep_for(std::chrono::seconds(1));
|
||||||
|
if (!session->ProcessHandle->IsActive())
|
||||||
|
return MakeUnexpected(std::format("Failed to spawn the child process \"{}\"", executablePath.string()));
|
||||||
|
|
||||||
|
auto processID = session->ProcessHandle->ID.load();
|
||||||
|
|
||||||
|
session->CreationTime = SteadyClock::now();
|
||||||
|
m_pendingSessions.push_back(std::move(session));
|
||||||
|
|
||||||
|
return processID;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL IPC_Manager::WaitTillNodeIsOnline(IN NativeProcessID nodeID)
|
||||||
|
{
|
||||||
|
BOOL isPending = true;
|
||||||
|
while (isPending)
|
||||||
|
{
|
||||||
|
isPending = false;
|
||||||
|
for (auto it = m_pendingSessions.begin(); it != m_pendingSessions.end(); it++)
|
||||||
|
{
|
||||||
|
if (it->get()->ProcessHandle->ID.load() == nodeID)
|
||||||
|
{
|
||||||
|
isPending = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Update();
|
||||||
|
}
|
||||||
|
return m_activeSessionMap.contains(nodeID);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Manager::ShutdownNode(IN NativeProcessID nodeID)
|
||||||
|
{
|
||||||
|
const auto itNode = m_activeSessionMap.find(nodeID);
|
||||||
|
if (itNode == m_activeSessionMap.end())
|
||||||
|
return;
|
||||||
|
|
||||||
|
auto &node = itNode->second;
|
||||||
|
|
||||||
|
ProcessOps::TerminateProcess(node->ProcessHandle);
|
||||||
|
FileOps::UnmapFile(node->MappedPtr);
|
||||||
|
FileOps::UnlinkSharedMemory(node->SharedMemName);
|
||||||
|
SocketOps::Close(node->DataSocket);
|
||||||
|
|
||||||
|
for (auto it = m_activeSessions.begin(); it != m_activeSessions.end(); it++)
|
||||||
|
{
|
||||||
|
if (it->get() == node)
|
||||||
|
{
|
||||||
|
m_activeSessions.erase(it);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
m_activeSessionMap.erase(itNode);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Manager::SendSignal(IN NativeProcessID node, IN UINT8 signal)
|
||||||
|
{
|
||||||
|
const auto itNode = m_activeSessionMap.find(node);
|
||||||
|
if (itNode == m_activeSessionMap.end())
|
||||||
|
return;
|
||||||
|
itNode->second->SendSignal(signal);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID IPC_Manager::SendPacket(IN NativeProcessID node, IN UINT16 packetID, IN Span<CONST UINT8> payload)
|
||||||
|
{
|
||||||
|
const auto itNode = m_activeSessionMap.find(node);
|
||||||
|
if (itNode == m_activeSessionMap.end())
|
||||||
|
return;
|
||||||
|
itNode->second->SendPacket(packetID, payload);
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
43
Src/IACore/imp/cpp/JSON.cpp
Normal file
43
Src/IACore/imp/cpp/JSON.cpp
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/JSON.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
Expected<nlohmann::json, String> JSON::Parse(IN CONST String &json)
|
||||||
|
{
|
||||||
|
const auto parseResult = nlohmann::json::parse(json, nullptr, false, true);
|
||||||
|
if (parseResult.is_discarded())
|
||||||
|
return MakeUnexpected("Failed to parse JSON");
|
||||||
|
return parseResult;
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<Pair<simdjson::dom::parser, simdjson::dom::object>, String> JSON::ParseReadOnly(IN CONST String &json)
|
||||||
|
{
|
||||||
|
simdjson::error_code error{};
|
||||||
|
simdjson::dom::parser parser;
|
||||||
|
simdjson::dom::object object;
|
||||||
|
if ((error = parser.parse(json).get(object)))
|
||||||
|
return MakeUnexpected(std::format("Failed to parse JSON : {}", simdjson::error_message(error)));
|
||||||
|
return std::make_pair(IA_MOVE(parser), IA_MOVE(object));
|
||||||
|
}
|
||||||
|
|
||||||
|
String JSON::Encode(IN nlohmann::json data)
|
||||||
|
{
|
||||||
|
return data.dump();
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
73
Src/IACore/imp/cpp/Logger.cpp
Normal file
73
Src/IACore/imp/cpp/Logger.cpp
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/Logger.hpp>
|
||||||
|
#include <IACore/IACore.hpp>
|
||||||
|
#include <IACore/FileOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
Logger::ELogLevel Logger::s_logLevel{Logger::ELogLevel::WARN};
|
||||||
|
std::ofstream Logger::s_logFile{};
|
||||||
|
|
||||||
|
VOID Logger::Initialize()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Logger::Terminate()
|
||||||
|
{
|
||||||
|
if (s_logFile.is_open())
|
||||||
|
{
|
||||||
|
s_logFile.flush();
|
||||||
|
s_logFile.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL Logger::EnableLoggingToDisk(IN PCCHAR filePath)
|
||||||
|
{
|
||||||
|
if (s_logFile.is_open())
|
||||||
|
{
|
||||||
|
s_logFile.flush();
|
||||||
|
s_logFile.close();
|
||||||
|
}
|
||||||
|
s_logFile.open(filePath);
|
||||||
|
return s_logFile.is_open();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Logger::SetLogLevel(IN ELogLevel logLevel)
|
||||||
|
{
|
||||||
|
s_logLevel = logLevel;
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Logger::FlushLogs()
|
||||||
|
{
|
||||||
|
std::cout.flush();
|
||||||
|
if (s_logFile)
|
||||||
|
s_logFile.flush();
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Logger::LogInternal(IN PCCHAR prefix, IN PCCHAR tag, IN String &&msg)
|
||||||
|
{
|
||||||
|
const auto outLine = std::format("[{:>8.3f}]: [{}]: {}", GetSecondsCount(), tag, msg);
|
||||||
|
std::cout << prefix << outLine << "\033[39m\n";
|
||||||
|
if (s_logFile)
|
||||||
|
{
|
||||||
|
s_logFile.write(outLine.data(), outLine.size());
|
||||||
|
s_logFile.put('\n');
|
||||||
|
s_logFile.flush();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
324
Src/IACore/imp/cpp/ProcessOps.cpp
Normal file
324
Src/IACore/imp/cpp/ProcessOps.cpp
Normal file
@ -0,0 +1,324 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/ProcessOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Output Buffering Helper
|
||||||
|
// Splits raw chunks into lines, preserving partial lines across chunks
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
struct LineBuffer
|
||||||
|
{
|
||||||
|
String Accumulator;
|
||||||
|
Function<VOID(StringView)> &Callback;
|
||||||
|
|
||||||
|
VOID Append(IN PCCHAR data, IN SIZE_T size);
|
||||||
|
VOID Flush();
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
NativeProcessID ProcessOps::GetCurrentProcessID()
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
return ::GetCurrentProcessId();
|
||||||
|
#else
|
||||||
|
return getpid();
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
Expected<INT32, String> ProcessOps::SpawnProcessSync(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(IN StringView line)> onOutputLineCallback)
|
||||||
|
{
|
||||||
|
Atomic<NativeProcessID> id;
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
return SpawnProcessWindows(command, args, onOutputLineCallback, id);
|
||||||
|
#else
|
||||||
|
return SpawnProcessPosix(command, args, onOutputLineCallback, id);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
SharedPtr<ProcessHandle> ProcessOps::SpawnProcessAsync(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(IN StringView line)> onOutputLineCallback,
|
||||||
|
IN Function<VOID(Expected<INT32, String>)> onFinishCallback)
|
||||||
|
{
|
||||||
|
SharedPtr<ProcessHandle> handle = std::make_shared<ProcessHandle>();
|
||||||
|
handle->IsRunning = true;
|
||||||
|
|
||||||
|
handle->ThreadHandle =
|
||||||
|
JoiningThread([=, h = handle.get(), cmd = IA_MOVE(command), args = std::move(args)]() mutable {
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
auto result = SpawnProcessWindows(cmd, args, onOutputLineCallback, h->ID);
|
||||||
|
#else
|
||||||
|
auto result = SpawnProcessPosix(cmd, args, onOutputLineCallback, h->ID);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
h->IsRunning = false;
|
||||||
|
|
||||||
|
if (!onFinishCallback)
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (!result)
|
||||||
|
onFinishCallback(MakeUnexpected(result.error()));
|
||||||
|
else
|
||||||
|
onFinishCallback(*result);
|
||||||
|
});
|
||||||
|
|
||||||
|
return handle;
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID ProcessOps::TerminateProcess(IN CONST SharedPtr<ProcessHandle> &handle)
|
||||||
|
{
|
||||||
|
if (!handle || !handle->IsActive())
|
||||||
|
return;
|
||||||
|
|
||||||
|
NativeProcessID pid = handle->ID.load();
|
||||||
|
if (pid == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
|
||||||
|
if (hProcess != NULL)
|
||||||
|
{
|
||||||
|
::TerminateProcess(hProcess, 9);
|
||||||
|
CloseHandle(hProcess);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
kill(pid, SIGKILL);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
Expected<INT32, String> ProcessOps::SpawnProcessWindows(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(StringView)> onOutputLineCallback,
|
||||||
|
OUT Atomic<NativeProcessID> &id)
|
||||||
|
{
|
||||||
|
SECURITY_ATTRIBUTES saAttr = {sizeof(SECURITY_ATTRIBUTES), NULL, TRUE}; // Allow inheritance
|
||||||
|
HANDLE hRead = NULL, hWrite = NULL;
|
||||||
|
|
||||||
|
if (!CreatePipe(&hRead, &hWrite, &saAttr, 0))
|
||||||
|
return tl::make_unexpected("Failed to create pipe");
|
||||||
|
|
||||||
|
// Ensure the read handle to the pipe for STDOUT is NOT inherited
|
||||||
|
if (!SetHandleInformation(hRead, HANDLE_FLAG_INHERIT, 0))
|
||||||
|
return tl::make_unexpected("Failed to secure pipe handles");
|
||||||
|
|
||||||
|
STARTUPINFOA si = {sizeof(STARTUPINFOA)};
|
||||||
|
si.dwFlags |= STARTF_USESTDHANDLES;
|
||||||
|
si.hStdOutput = hWrite;
|
||||||
|
si.hStdError = hWrite; // Merge stderr
|
||||||
|
si.hStdInput = NULL; // No input
|
||||||
|
|
||||||
|
PROCESS_INFORMATION pi = {0};
|
||||||
|
|
||||||
|
// Windows command line needs to be mutable and concatenated
|
||||||
|
String commandLine = std::format("\"{}\" {}", command, args);
|
||||||
|
|
||||||
|
BOOL success = CreateProcessA(NULL, commandLine.data(), NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi);
|
||||||
|
|
||||||
|
// Important: Close write end in parent, otherwise ReadFile never returns EOF!
|
||||||
|
CloseHandle(hWrite);
|
||||||
|
|
||||||
|
if (!success)
|
||||||
|
{
|
||||||
|
CloseHandle(hRead);
|
||||||
|
return tl::make_unexpected(String("CreateProcess failed: ") + std::to_string(GetLastError()));
|
||||||
|
}
|
||||||
|
|
||||||
|
id.store(pi.dwProcessId);
|
||||||
|
|
||||||
|
// Read Loop
|
||||||
|
LineBuffer lineBuf{"", onOutputLineCallback};
|
||||||
|
DWORD bytesRead;
|
||||||
|
CHAR buffer[4096];
|
||||||
|
|
||||||
|
while (ReadFile(hRead, buffer, sizeof(buffer), &bytesRead, NULL) && bytesRead != 0)
|
||||||
|
{
|
||||||
|
lineBuf.Append(buffer, bytesRead);
|
||||||
|
}
|
||||||
|
lineBuf.Flush();
|
||||||
|
|
||||||
|
// NOW we wait for exit code
|
||||||
|
DWORD exitCode = 0;
|
||||||
|
WaitForSingleObject(pi.hProcess, INFINITE);
|
||||||
|
GetExitCodeProcess(pi.hProcess, &exitCode);
|
||||||
|
|
||||||
|
CloseHandle(pi.hProcess);
|
||||||
|
CloseHandle(pi.hThread);
|
||||||
|
CloseHandle(hRead);
|
||||||
|
id.store(0);
|
||||||
|
|
||||||
|
return static_cast<INT32>(exitCode);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if IA_PLATFORM_UNIX
|
||||||
|
Expected<INT32, String> ProcessOps::SpawnProcessPosix(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(StringView)> onOutputLineCallback,
|
||||||
|
OUT Atomic<NativeProcessID> &id)
|
||||||
|
{
|
||||||
|
int pipefd[2];
|
||||||
|
if (pipe(pipefd) == -1)
|
||||||
|
return tl::make_unexpected("Failed to create pipe");
|
||||||
|
|
||||||
|
pid_t pid = fork();
|
||||||
|
|
||||||
|
if (pid == -1)
|
||||||
|
{
|
||||||
|
return tl::make_unexpected("Failed to fork process");
|
||||||
|
}
|
||||||
|
else if (pid == 0)
|
||||||
|
{
|
||||||
|
// --- Child Process ---
|
||||||
|
close(pipefd[0]);
|
||||||
|
|
||||||
|
dup2(pipefd[1], STDOUT_FILENO);
|
||||||
|
dup2(pipefd[1], STDERR_FILENO);
|
||||||
|
close(pipefd[1]);
|
||||||
|
|
||||||
|
// --- ARGUMENT PARSING START ---
|
||||||
|
std::vector<std::string> argStorage; // To keep strings alive
|
||||||
|
std::vector<char *> argv;
|
||||||
|
|
||||||
|
std::string cmdStr = command;
|
||||||
|
argv.push_back(cmdStr.data());
|
||||||
|
|
||||||
|
// Manual Quote-Aware Splitter
|
||||||
|
std::string currentToken;
|
||||||
|
bool inQuotes = false;
|
||||||
|
|
||||||
|
for (char c : args)
|
||||||
|
{
|
||||||
|
if (c == '\"')
|
||||||
|
{
|
||||||
|
inQuotes = !inQuotes;
|
||||||
|
// Determine if you want to keep the quotes or strip them.
|
||||||
|
// Usually for execvp, you strip them so the shell receives the raw content.
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (c == ' ' && !inQuotes)
|
||||||
|
{
|
||||||
|
if (!currentToken.empty())
|
||||||
|
{
|
||||||
|
argStorage.push_back(currentToken);
|
||||||
|
currentToken.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
currentToken += c;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!currentToken.empty())
|
||||||
|
{
|
||||||
|
argStorage.push_back(currentToken);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build char* array from the std::string storage
|
||||||
|
for (auto &s : argStorage)
|
||||||
|
{
|
||||||
|
argv.push_back(s.data());
|
||||||
|
}
|
||||||
|
argv.push_back(nullptr);
|
||||||
|
// --- ARGUMENT PARSING END ---
|
||||||
|
|
||||||
|
execvp(argv[0], argv.data());
|
||||||
|
_exit(127);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// --- Parent Process ---
|
||||||
|
id.store(pid);
|
||||||
|
|
||||||
|
close(pipefd[1]);
|
||||||
|
|
||||||
|
LineBuffer lineBuf{"", onOutputLineCallback};
|
||||||
|
char buffer[4096];
|
||||||
|
ssize_t count;
|
||||||
|
|
||||||
|
while ((count = read(pipefd[0], buffer, sizeof(buffer))) > 0)
|
||||||
|
{
|
||||||
|
lineBuf.Append(buffer, count);
|
||||||
|
}
|
||||||
|
lineBuf.Flush();
|
||||||
|
close(pipefd[0]);
|
||||||
|
|
||||||
|
int status;
|
||||||
|
waitpid(pid, &status, 0);
|
||||||
|
|
||||||
|
id.store(0);
|
||||||
|
if (WIFEXITED(status))
|
||||||
|
return WEXITSTATUS(status);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
VOID LineBuffer::Append(IN PCCHAR data, IN SIZE_T size)
|
||||||
|
{
|
||||||
|
SIZE_T start = 0;
|
||||||
|
for (SIZE_T i = 0; i < size; ++i)
|
||||||
|
{
|
||||||
|
if (data[i] == '\n' || data[i] == '\r')
|
||||||
|
{
|
||||||
|
// Flush Accumulator + current chunk
|
||||||
|
if (!Accumulator.empty())
|
||||||
|
{
|
||||||
|
Accumulator.append(data + start, i - start);
|
||||||
|
if (!Accumulator.empty())
|
||||||
|
Callback(Accumulator);
|
||||||
|
Accumulator.clear();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// Zero copy optimization for pure lines in one chunk
|
||||||
|
if (i > start)
|
||||||
|
Callback(StringView(data + start, i - start));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Skip \r\n sequence if needed, or just start next
|
||||||
|
if (data[i] == '\r' && i + 1 < size && data[i + 1] == '\n')
|
||||||
|
i++;
|
||||||
|
start = i + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Save remaining partial line
|
||||||
|
if (start < size)
|
||||||
|
{
|
||||||
|
Accumulator.append(data + start, size - start);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID LineBuffer::Flush()
|
||||||
|
{
|
||||||
|
if (!Accumulator.empty())
|
||||||
|
{
|
||||||
|
Callback(Accumulator);
|
||||||
|
Accumulator.clear();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
106
Src/IACore/imp/cpp/SocketOps.cpp
Normal file
106
Src/IACore/imp/cpp/SocketOps.cpp
Normal file
@ -0,0 +1,106 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/SocketOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
INT32 SocketOps::s_initCount{0};
|
||||||
|
|
||||||
|
VOID SocketOps::Close(IN SocketHandle sock)
|
||||||
|
{
|
||||||
|
if (sock == INVALID_SOCKET)
|
||||||
|
return;
|
||||||
|
CLOSE_SOCKET(sock);
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL SocketOps::Listen(IN SocketHandle sock, IN INT32 queueSize)
|
||||||
|
{
|
||||||
|
return listen(sock, queueSize) == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
SocketHandle SocketOps::CreateUnixSocket()
|
||||||
|
{
|
||||||
|
return socket(AF_UNIX, SOCK_STREAM, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL SocketOps::BindUnixSocket(IN SocketHandle sock, IN PCCHAR path)
|
||||||
|
{
|
||||||
|
if (!IS_VALID_SOCKET(sock))
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
UNLINK_FILE(path);
|
||||||
|
|
||||||
|
sockaddr_un addr{};
|
||||||
|
addr.sun_family = AF_UNIX;
|
||||||
|
|
||||||
|
size_t maxLen = sizeof(addr.sun_path) - 1;
|
||||||
|
|
||||||
|
strncpy(addr.sun_path, path, maxLen);
|
||||||
|
|
||||||
|
if (bind(sock, (struct sockaddr *) &addr, sizeof(addr)) == -1)
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL SocketOps::ConnectUnixSocket(IN SocketHandle sock, IN PCCHAR path)
|
||||||
|
{
|
||||||
|
if (!IS_VALID_SOCKET(sock))
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
sockaddr_un addr{};
|
||||||
|
addr.sun_family = AF_UNIX;
|
||||||
|
|
||||||
|
size_t maxLen = sizeof(addr.sun_path) - 1;
|
||||||
|
|
||||||
|
strncpy(addr.sun_path, path, maxLen);
|
||||||
|
|
||||||
|
if (connect(sock, (struct sockaddr *) &addr, sizeof(addr)) == -1)
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL SocketOps::IsPortAvailable(IN UINT16 port, IN INT32 type)
|
||||||
|
{
|
||||||
|
SocketHandle sock = socket(AF_INET, type, IPPROTO_UDP);
|
||||||
|
if (!IS_VALID_SOCKET(sock))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
sockaddr_in addr{};
|
||||||
|
addr.sin_family = AF_INET;
|
||||||
|
addr.sin_port = htons(port);
|
||||||
|
addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||||||
|
|
||||||
|
bool isFree = false;
|
||||||
|
if (bind(sock, (struct sockaddr *) &addr, sizeof(addr)) == 0)
|
||||||
|
isFree = true;
|
||||||
|
|
||||||
|
CLOSE_SOCKET(sock);
|
||||||
|
|
||||||
|
return isFree;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL SocketOps::IsWouldBlock()
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
return WSAGetLastError() == WSAEWOULDBLOCK;
|
||||||
|
#else
|
||||||
|
return errno == EWOULDBLOCK || errno == EAGAIN;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
59
Src/IACore/imp/cpp/StreamReader.cpp
Normal file
59
Src/IACore/imp/cpp/StreamReader.cpp
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/StreamReader.hpp>
|
||||||
|
#include <IACore/FileOps.hpp>
|
||||||
|
#include <IACore/Logger.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
StreamReader::StreamReader(IN CONST FilePath &path) : m_storageType(EStorageType::OWNING_MMAP)
|
||||||
|
{
|
||||||
|
const auto t = FileOps::MapFile(path, m_dataSize);
|
||||||
|
if (!t)
|
||||||
|
{
|
||||||
|
Logger::Error("Failed to memory map file {}", path.string());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
m_data = *t;
|
||||||
|
}
|
||||||
|
|
||||||
|
StreamReader::StreamReader(IN Vector<UINT8> &&data)
|
||||||
|
: m_owningVector(IA_MOVE(data)), m_storageType(EStorageType::OWNING_VECTOR)
|
||||||
|
{
|
||||||
|
m_data = m_owningVector.data();
|
||||||
|
m_dataSize = m_owningVector.size();
|
||||||
|
}
|
||||||
|
|
||||||
|
StreamReader::StreamReader(IN Span<CONST UINT8> data)
|
||||||
|
: m_data(data.data()), m_dataSize(data.size()), m_storageType(EStorageType::NON_OWNING)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
StreamReader::~StreamReader()
|
||||||
|
{
|
||||||
|
switch (m_storageType)
|
||||||
|
{
|
||||||
|
case EStorageType::OWNING_MMAP:
|
||||||
|
FileOps::UnmapFile(m_data);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case EStorageType::NON_OWNING:
|
||||||
|
case EStorageType::OWNING_VECTOR:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
97
Src/IACore/imp/cpp/StreamWriter.cpp
Normal file
97
Src/IACore/imp/cpp/StreamWriter.cpp
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/StreamWriter.hpp>
|
||||||
|
#include <IACore/Logger.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
StreamWriter::StreamWriter() : m_storageType(EStorageType::OWNING_VECTOR)
|
||||||
|
{
|
||||||
|
m_owningVector.resize(m_capacity = 256);
|
||||||
|
m_buffer = m_owningVector.data();
|
||||||
|
}
|
||||||
|
|
||||||
|
StreamWriter::StreamWriter(IN Span<UINT8> data)
|
||||||
|
: m_buffer(data.data()), m_capacity(data.size()), m_storageType(EStorageType::NON_OWNING)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
StreamWriter::StreamWriter(IN CONST FilePath &path) : m_filePath(path), m_storageType(EStorageType::OWNING_FILE)
|
||||||
|
{
|
||||||
|
IA_RELEASE_ASSERT(!path.empty());
|
||||||
|
const auto f = fopen(m_filePath.string().c_str(), "wb");
|
||||||
|
if (!f)
|
||||||
|
{
|
||||||
|
Logger::Error("Failed to open file for writing {}", m_filePath.string().c_str());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
fputc(0, f);
|
||||||
|
fclose(f);
|
||||||
|
|
||||||
|
m_owningVector.resize(m_capacity = 256);
|
||||||
|
m_buffer = m_owningVector.data();
|
||||||
|
}
|
||||||
|
|
||||||
|
StreamWriter::~StreamWriter()
|
||||||
|
{
|
||||||
|
switch (m_storageType)
|
||||||
|
{
|
||||||
|
case EStorageType::OWNING_FILE: {
|
||||||
|
IA_RELEASE_ASSERT(!m_filePath.empty());
|
||||||
|
const auto f = fopen(m_filePath.string().c_str(), "wb");
|
||||||
|
if (!f)
|
||||||
|
{
|
||||||
|
Logger::Error("Failed to open file for writing {}", m_filePath.string().c_str());
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
fwrite(m_owningVector.data(), 1, m_owningVector.size(), f);
|
||||||
|
fclose(f);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case EStorageType::OWNING_VECTOR:
|
||||||
|
case EStorageType::NON_OWNING:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#define HANDLE_OUT_OF_CAPACITY(_size) \
|
||||||
|
if B_UNLIKELY ((m_cursor + _size) > m_capacity) \
|
||||||
|
{ \
|
||||||
|
if (m_storageType == EStorageType::NON_OWNING) \
|
||||||
|
return false; \
|
||||||
|
m_owningVector.resize(m_capacity + (_size << 1)); \
|
||||||
|
m_capacity = m_owningVector.size(); \
|
||||||
|
m_buffer = m_owningVector.data(); \
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL StreamWriter::Write(IN UINT8 byte, IN SIZE_T count)
|
||||||
|
{
|
||||||
|
HANDLE_OUT_OF_CAPACITY(count);
|
||||||
|
memset(&m_buffer[m_cursor], byte, count);
|
||||||
|
m_cursor += count;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL StreamWriter::Write(IN PCVOID buffer, IN SIZE_T size)
|
||||||
|
{
|
||||||
|
HANDLE_OUT_OF_CAPACITY(size);
|
||||||
|
memcpy(&m_buffer[m_cursor], buffer, size);
|
||||||
|
m_cursor += size;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
101
Src/IACore/imp/cpp/StringOps.cpp
Normal file
101
Src/IACore/imp/cpp/StringOps.cpp
Normal file
@ -0,0 +1,101 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/StringOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
CONST String BASE64_CHAR_TABLE = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
||||||
|
|
||||||
|
String StringOps::EncodeBase64(IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
String result;
|
||||||
|
result.reserve(((data.size() + 2) / 3) * 4);
|
||||||
|
for (size_t i = 0; i < data.size(); i += 3)
|
||||||
|
{
|
||||||
|
uint32_t value = 0;
|
||||||
|
INT32 num_bytes = 0;
|
||||||
|
for (INT32 j = 0; j < 3 && (i + j) < data.size(); ++j)
|
||||||
|
{
|
||||||
|
value = (value << 8) | data[i + j];
|
||||||
|
num_bytes++;
|
||||||
|
}
|
||||||
|
for (INT32 j = 0; j < num_bytes + 1; ++j)
|
||||||
|
{
|
||||||
|
if (j < 4)
|
||||||
|
{
|
||||||
|
result += BASE64_CHAR_TABLE[(value >> (6 * (3 - j))) & 0x3F];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (num_bytes < 3)
|
||||||
|
{
|
||||||
|
for (INT32 j = 0; j < (3 - num_bytes); ++j)
|
||||||
|
{
|
||||||
|
result += '=';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
Vector<UINT8> StringOps::DecodeBase64(IN CONST String &data)
|
||||||
|
{
|
||||||
|
Vector<UINT8> result;
|
||||||
|
|
||||||
|
CONST AUTO isBase64 = [](UINT8 c) { return (isalnum(c) || (c == '+') || (c == '/')); };
|
||||||
|
|
||||||
|
INT32 in_len = data.size();
|
||||||
|
INT32 i = 0, j = 0, in_ = 0;
|
||||||
|
UINT8 tmpBuf0[4], tmpBuf1[3];
|
||||||
|
|
||||||
|
while (in_len-- && (data[in_] != '=') && isBase64(data[in_]))
|
||||||
|
{
|
||||||
|
tmpBuf0[i++] = data[in_];
|
||||||
|
in_++;
|
||||||
|
if (i == 4)
|
||||||
|
{
|
||||||
|
for (i = 0; i < 4; i++)
|
||||||
|
tmpBuf0[i] = BASE64_CHAR_TABLE.find(tmpBuf0[i]);
|
||||||
|
|
||||||
|
tmpBuf1[0] = (tmpBuf0[0] << 2) + ((tmpBuf0[1] & 0x30) >> 4);
|
||||||
|
tmpBuf1[1] = ((tmpBuf0[1] & 0xf) << 4) + ((tmpBuf0[2] & 0x3c) >> 2);
|
||||||
|
tmpBuf1[2] = ((tmpBuf0[2] & 0x3) << 6) + tmpBuf0[3];
|
||||||
|
|
||||||
|
for (i = 0; (i < 3); i++)
|
||||||
|
result.push_back(tmpBuf1[i]);
|
||||||
|
i = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (i)
|
||||||
|
{
|
||||||
|
for (j = i; j < 4; j++)
|
||||||
|
tmpBuf0[j] = 0;
|
||||||
|
|
||||||
|
for (j = 0; j < 4; j++)
|
||||||
|
tmpBuf0[j] = BASE64_CHAR_TABLE.find(tmpBuf0[j]);
|
||||||
|
|
||||||
|
tmpBuf1[0] = (tmpBuf0[0] << 2) + ((tmpBuf0[1] & 0x30) >> 4);
|
||||||
|
tmpBuf1[1] = ((tmpBuf0[1] & 0xf) << 4) + ((tmpBuf0[2] & 0x3c) >> 2);
|
||||||
|
tmpBuf1[2] = ((tmpBuf0[2] & 0x3) << 6) + tmpBuf0[3];
|
||||||
|
|
||||||
|
for (j = 0; (j < i - 1); j++)
|
||||||
|
result.push_back(tmpBuf1[j]);
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
226
Src/IACore/inc/IACore/ADT/RingBuffer.hpp
Normal file
226
Src/IACore/inc/IACore/ADT/RingBuffer.hpp
Normal file
@ -0,0 +1,226 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class RingBufferView
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
STATIC CONSTEXPR UINT16 PACKET_ID_SKIP = 0;
|
||||||
|
|
||||||
|
struct ControlBlock
|
||||||
|
{
|
||||||
|
struct alignas(64)
|
||||||
|
{
|
||||||
|
Atomic<UINT32> WriteOffset{0};
|
||||||
|
} Producer;
|
||||||
|
|
||||||
|
struct alignas(64)
|
||||||
|
{
|
||||||
|
Atomic<UINT32> ReadOffset{0};
|
||||||
|
// Capacity is effectively constant after init,
|
||||||
|
// so it doesn't cause false sharing invalidations.
|
||||||
|
UINT32 Capacity{0};
|
||||||
|
} Consumer;
|
||||||
|
};
|
||||||
|
|
||||||
|
static_assert(offsetof(ControlBlock, Consumer) == 64, "False sharing detected in ControlBlock");
|
||||||
|
|
||||||
|
// All of the data in ring buffer will be stored as packets
|
||||||
|
struct PacketHeader
|
||||||
|
{
|
||||||
|
PacketHeader() : ID(0), PayloadSize(0)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
PacketHeader(IN UINT16 id) : ID(id), PayloadSize(0)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
PacketHeader(IN UINT16 id, IN UINT16 payloadSize) : ID(id), PayloadSize(payloadSize)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT16 ID{};
|
||||||
|
UINT16 PayloadSize{};
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
INLINE RingBufferView(IN Span<UINT8> buffer, IN BOOL isOwner);
|
||||||
|
INLINE RingBufferView(IN ControlBlock *controlBlock, IN Span<UINT8> buffer, IN BOOL isOwner);
|
||||||
|
|
||||||
|
INLINE INT32 Pop(OUT PacketHeader &outHeader, OUT Span<UINT8> outBuffer);
|
||||||
|
INLINE BOOL Push(IN UINT16 packetID, IN Span<CONST UINT8> data);
|
||||||
|
|
||||||
|
INLINE ControlBlock *GetControlBlock();
|
||||||
|
|
||||||
|
private:
|
||||||
|
PUINT8 m_dataPtr{};
|
||||||
|
UINT32 m_capacity{};
|
||||||
|
ControlBlock *m_controlBlock{};
|
||||||
|
|
||||||
|
private:
|
||||||
|
INLINE VOID WriteWrapped(IN UINT32 offset, IN PCVOID data, IN UINT32 size);
|
||||||
|
INLINE VOID ReadWrapped(IN UINT32 offset, OUT PVOID outData, IN UINT32 size);
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
RingBufferView::RingBufferView(IN Span<UINT8> buffer, IN BOOL isOwner)
|
||||||
|
{
|
||||||
|
IA_ASSERT(buffer.size() > sizeof(ControlBlock));
|
||||||
|
|
||||||
|
m_controlBlock = reinterpret_cast<ControlBlock *>(buffer.data());
|
||||||
|
m_dataPtr = buffer.data() + sizeof(ControlBlock);
|
||||||
|
|
||||||
|
m_capacity = static_cast<UINT32>(buffer.size()) - sizeof(ControlBlock);
|
||||||
|
|
||||||
|
if (isOwner)
|
||||||
|
{
|
||||||
|
m_controlBlock->Consumer.Capacity = m_capacity;
|
||||||
|
m_controlBlock->Producer.WriteOffset.store(0, std::memory_order_release);
|
||||||
|
m_controlBlock->Consumer.ReadOffset.store(0, std::memory_order_release);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
IA_ASSERT(m_controlBlock->Consumer.Capacity == m_capacity);
|
||||||
|
}
|
||||||
|
|
||||||
|
RingBufferView::RingBufferView(IN ControlBlock *controlBlock, IN Span<UINT8> buffer, IN BOOL isOwner)
|
||||||
|
{
|
||||||
|
IA_ASSERT(controlBlock != nullptr);
|
||||||
|
IA_ASSERT(buffer.size() > 0);
|
||||||
|
|
||||||
|
m_controlBlock = controlBlock;
|
||||||
|
m_dataPtr = buffer.data();
|
||||||
|
m_capacity = static_cast<UINT32>(buffer.size());
|
||||||
|
|
||||||
|
if (isOwner)
|
||||||
|
{
|
||||||
|
m_controlBlock->Consumer.Capacity = m_capacity;
|
||||||
|
m_controlBlock->Producer.WriteOffset.store(0, std::memory_order_release);
|
||||||
|
m_controlBlock->Consumer.ReadOffset.store(0, std::memory_order_release);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
INT32 RingBufferView::Pop(OUT PacketHeader &outHeader, OUT Span<UINT8> outBuffer)
|
||||||
|
{
|
||||||
|
UINT32 write = m_controlBlock->Producer.WriteOffset.load(std::memory_order_acquire);
|
||||||
|
UINT32 read = m_controlBlock->Consumer.ReadOffset.load(std::memory_order_relaxed);
|
||||||
|
UINT32 cap = m_capacity;
|
||||||
|
|
||||||
|
if (read == write)
|
||||||
|
return 0; // Empty
|
||||||
|
|
||||||
|
ReadWrapped(read, &outHeader, sizeof(PacketHeader));
|
||||||
|
|
||||||
|
if (outHeader.PayloadSize > outBuffer.size())
|
||||||
|
return -static_cast<INT32>(outHeader.PayloadSize);
|
||||||
|
|
||||||
|
if (outHeader.PayloadSize > 0)
|
||||||
|
{
|
||||||
|
UINT32 dataReadOffset = (read + sizeof(PacketHeader)) % cap;
|
||||||
|
ReadWrapped(dataReadOffset, outBuffer.data(), outHeader.PayloadSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Move read pointer forward
|
||||||
|
UINT32 newReadOffset = (read + sizeof(PacketHeader) + outHeader.PayloadSize) % cap;
|
||||||
|
m_controlBlock->Consumer.ReadOffset.store(newReadOffset, std::memory_order_release);
|
||||||
|
|
||||||
|
return outHeader.PayloadSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL RingBufferView::Push(IN UINT16 packetID, IN Span<CONST UINT8> data)
|
||||||
|
{
|
||||||
|
IA_ASSERT(data.size() <= UINT16_MAX);
|
||||||
|
|
||||||
|
const UINT32 totalSize = sizeof(PacketHeader) + static_cast<UINT32>(data.size());
|
||||||
|
|
||||||
|
UINT32 read = m_controlBlock->Consumer.ReadOffset.load(std::memory_order_acquire);
|
||||||
|
UINT32 write = m_controlBlock->Producer.WriteOffset.load(std::memory_order_relaxed);
|
||||||
|
UINT32 cap = m_capacity;
|
||||||
|
|
||||||
|
UINT32 freeSpace = (read <= write) ? (m_capacity - write) + read : (read - write);
|
||||||
|
|
||||||
|
// Ensure to always leave 1 byte empty to prevent Read == Write ambiguity (Wait-Free Ring Buffer standard)
|
||||||
|
if (freeSpace <= totalSize)
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
PacketHeader header{packetID, static_cast<UINT16>(data.size())};
|
||||||
|
WriteWrapped(write, &header, sizeof(PacketHeader));
|
||||||
|
|
||||||
|
UINT32 dataWriteOffset = (write + sizeof(PacketHeader)) % cap;
|
||||||
|
|
||||||
|
if (data.size() > 0)
|
||||||
|
{
|
||||||
|
WriteWrapped(dataWriteOffset, data.data(), static_cast<UINT32>(data.size()));
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT32 newWriteOffset = (dataWriteOffset + data.size()) % cap;
|
||||||
|
m_controlBlock->Producer.WriteOffset.store(newWriteOffset, std::memory_order_release);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
RingBufferView::ControlBlock *RingBufferView::GetControlBlock()
|
||||||
|
{
|
||||||
|
return m_controlBlock;
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID RingBufferView::WriteWrapped(IN UINT32 offset, IN PCVOID data, IN UINT32 size)
|
||||||
|
{
|
||||||
|
if (offset + size <= m_capacity)
|
||||||
|
{
|
||||||
|
// Contiguous write
|
||||||
|
memcpy(m_dataPtr + offset, data, size);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// Split write
|
||||||
|
UINT32 firstChunk = m_capacity - offset;
|
||||||
|
UINT32 secondChunk = size - firstChunk;
|
||||||
|
|
||||||
|
const UINT8 *src = static_cast<const UINT8 *>(data);
|
||||||
|
|
||||||
|
memcpy(m_dataPtr + offset, src, firstChunk);
|
||||||
|
memcpy(m_dataPtr, src + firstChunk, secondChunk);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID RingBufferView::ReadWrapped(IN UINT32 offset, OUT PVOID outData, IN UINT32 size)
|
||||||
|
{
|
||||||
|
if (offset + size <= m_capacity)
|
||||||
|
{
|
||||||
|
// Contiguous read
|
||||||
|
memcpy(outData, m_dataPtr + offset, size);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// Split read
|
||||||
|
UINT32 firstChunk = m_capacity - offset;
|
||||||
|
UINT32 secondChunk = size - firstChunk;
|
||||||
|
|
||||||
|
UINT8 *dst = static_cast<UINT8 *>(outData);
|
||||||
|
|
||||||
|
memcpy(dst, m_dataPtr + offset, firstChunk);
|
||||||
|
memcpy(dst + firstChunk, m_dataPtr, secondChunk);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
74
Src/IACore/inc/IACore/AsyncOps.hpp
Normal file
74
Src/IACore/inc/IACore/AsyncOps.hpp
Normal file
@ -0,0 +1,74 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class AsyncOps
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
using TaskTag = UINT64;
|
||||||
|
using WorkerID = UINT16;
|
||||||
|
|
||||||
|
STATIC CONSTEXPR WorkerID MainThreadWorkerID = 0;
|
||||||
|
|
||||||
|
enum class Priority : UINT8
|
||||||
|
{
|
||||||
|
High,
|
||||||
|
Normal
|
||||||
|
};
|
||||||
|
|
||||||
|
struct Schedule
|
||||||
|
{
|
||||||
|
Atomic<INT32> Counter{0};
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC VOID InitializeScheduler(IN UINT8 workerCount = 0);
|
||||||
|
STATIC VOID TerminateScheduler();
|
||||||
|
|
||||||
|
STATIC VOID ScheduleTask(IN Function<VOID(IN WorkerID workerID)> task, IN TaskTag tag, IN Schedule *schedule,
|
||||||
|
IN Priority priority = Priority::Normal);
|
||||||
|
|
||||||
|
STATIC VOID CancelTasksOfTag(IN TaskTag tag);
|
||||||
|
|
||||||
|
STATIC VOID WaitForScheduleCompletion(IN Schedule *schedule);
|
||||||
|
|
||||||
|
STATIC VOID RunTask(IN Function<VOID()> task);
|
||||||
|
|
||||||
|
STATIC WorkerID GetWorkerCount();
|
||||||
|
|
||||||
|
private:
|
||||||
|
struct ScheduledTask
|
||||||
|
{
|
||||||
|
TaskTag Tag{};
|
||||||
|
Schedule *ScheduleHandle{};
|
||||||
|
Function<VOID(IN WorkerID workerID)> Task{};
|
||||||
|
};
|
||||||
|
|
||||||
|
STATIC VOID ScheduleWorkerLoop(IN StopToken stopToken, IN WorkerID workerID);
|
||||||
|
|
||||||
|
private:
|
||||||
|
STATIC Mutex s_queueMutex;
|
||||||
|
STATIC ConditionVariable s_wakeCondition;
|
||||||
|
STATIC Vector<JoiningThread> s_scheduleWorkers;
|
||||||
|
STATIC Deque<ScheduledTask> s_highPriorityQueue;
|
||||||
|
STATIC Deque<ScheduledTask> s_normalPriorityQueue;
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
50
Src/IACore/inc/IACore/DataOps.hpp
Normal file
50
Src/IACore/inc/IACore/DataOps.hpp
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class DataOps
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
enum class CompressionType
|
||||||
|
{
|
||||||
|
None,
|
||||||
|
Gzip,
|
||||||
|
Zlib
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC UINT32 Hash(IN CONST String &string);
|
||||||
|
STATIC UINT32 Hash(IN Span<CONST UINT8> data);
|
||||||
|
|
||||||
|
STATIC UINT32 CRC32(IN Span<CONST UINT8> data);
|
||||||
|
|
||||||
|
STATIC CompressionType DetectCompression(IN Span<CONST UINT8> data);
|
||||||
|
|
||||||
|
STATIC Expected<Vector<UINT8>, String> GZipInflate(IN Span<CONST UINT8> data);
|
||||||
|
STATIC Expected<Vector<UINT8>, String> GZipDeflate(IN Span<CONST UINT8> data);
|
||||||
|
|
||||||
|
STATIC Expected<Vector<UINT8>, String> ZlibInflate(IN Span<CONST UINT8> data);
|
||||||
|
STATIC Expected<Vector<UINT8>, String> ZlibDeflate(IN Span<CONST UINT8> data);
|
||||||
|
|
||||||
|
STATIC Expected<Vector<UINT8>, String> ZstdInflate(IN Span<CONST UINT8> data);
|
||||||
|
STATIC Expected<Vector<UINT8>, String> ZstdDeflate(IN Span<CONST UINT8> data);
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
184
Src/IACore/inc/IACore/DynamicLib.hpp
Normal file
184
Src/IACore/inc/IACore/DynamicLib.hpp
Normal file
@ -0,0 +1,184 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
#include <libloaderapi.h>
|
||||||
|
#include <errhandlingapi.h>
|
||||||
|
#else
|
||||||
|
#include <dlfcn.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
namespace IACore {
|
||||||
|
|
||||||
|
class DynamicLib {
|
||||||
|
public:
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Constructors / Destructors (Move Only)
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
DynamicLib() : m_handle(nullptr) {}
|
||||||
|
|
||||||
|
// Move Constructor: Steal ownership, nullify source
|
||||||
|
DynamicLib(DynamicLib&& other) NOEXCEPT : m_handle(other.m_handle) {
|
||||||
|
other.m_handle = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Move Assignment
|
||||||
|
DynamicLib& operator=(DynamicLib&& other) NOEXCEPT {
|
||||||
|
if (this != &other) {
|
||||||
|
Unload(); // Free current if exists
|
||||||
|
m_handle = other.m_handle;
|
||||||
|
other.m_handle = nullptr;
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
// No Copying allowed (Library handles are unique resources)
|
||||||
|
DynamicLib(CONST DynamicLib&) = delete;
|
||||||
|
DynamicLib& operator=(CONST DynamicLib&) = delete;
|
||||||
|
|
||||||
|
~DynamicLib() {
|
||||||
|
Unload();
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Static Loader
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
// Automatically detects extension (.dll/.so) if not provided
|
||||||
|
NO_DISCARD("Check for load errors")
|
||||||
|
STATIC tl::expected<DynamicLib, String> Load(CONST String& searchPath, CONST String& name) {
|
||||||
|
namespace fs = std::filesystem;
|
||||||
|
|
||||||
|
// 1. Build Path safely
|
||||||
|
fs::path fullPath = fs::path(searchPath) / name;
|
||||||
|
|
||||||
|
// 2. Auto-append extension if missing
|
||||||
|
if (!fullPath.has_extension()) {
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
fullPath += ".dll";
|
||||||
|
#elif IA_PLATFORM_MAC
|
||||||
|
fullPath += ".dylib";
|
||||||
|
#else
|
||||||
|
fullPath += ".so";
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
DynamicLib lib;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
// Use LoadLibraryA (ANSI/UTF-8) assuming manifest is set for UTF-8
|
||||||
|
HMODULE h = LoadLibraryA(fullPath.string().c_str());
|
||||||
|
if (!h) {
|
||||||
|
return tl::make_unexpected(GetWindowsError());
|
||||||
|
}
|
||||||
|
lib.m_handle = CAST(h, PVOID);
|
||||||
|
#else
|
||||||
|
// RTLD_LAZY: Resolve symbols only as code executes (Standard for plugins)
|
||||||
|
// RTLD_NOW: Resolve all immediately (Safer for strict engines)
|
||||||
|
void* h = dlopen(fullPath.c_str(), RTLD_LAZY | RTLD_LOCAL);
|
||||||
|
if (!h) {
|
||||||
|
// dlerror returns a string describing the last error
|
||||||
|
const char* err = dlerror();
|
||||||
|
return tl::make_unexpected(String(err ? err : "Unknown dlopen error"));
|
||||||
|
}
|
||||||
|
lib.m_handle = h;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return IA_MOVE(lib);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Symbol Access
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
NO_DISCARD("Check if symbol exists")
|
||||||
|
tl::expected<PVOID, String> GetSymbol(CONST String& name) CONST {
|
||||||
|
if (!m_handle) return tl::make_unexpected(String("Library not loaded"));
|
||||||
|
|
||||||
|
PVOID sym = nullptr;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
sym = CAST(GetProcAddress(CAST(m_handle, HMODULE), name.c_str()), PVOID);
|
||||||
|
if (!sym) return tl::make_unexpected(GetWindowsError());
|
||||||
|
#else
|
||||||
|
// Clear any previous error
|
||||||
|
dlerror();
|
||||||
|
sym = dlsym(m_handle, name.c_str());
|
||||||
|
const char* err = dlerror();
|
||||||
|
if (err) return tl::make_unexpected(String(err));
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return sym;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Template helper for casting
|
||||||
|
template<typename FuncT>
|
||||||
|
tl::expected<FuncT, String> GetFunction(CONST String& name) CONST {
|
||||||
|
auto res = GetSymbol(name);
|
||||||
|
if (!res) return tl::make_unexpected(res.error());
|
||||||
|
return REINTERPRET(*res, FuncT);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// State Management
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
VOID Unload() {
|
||||||
|
if (m_handle) {
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
FreeLibrary(CAST(m_handle, HMODULE));
|
||||||
|
#else
|
||||||
|
dlclose(m_handle);
|
||||||
|
#endif
|
||||||
|
m_handle = nullptr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL IsLoaded() CONST {
|
||||||
|
return m_handle != nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
PVOID m_handle;
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Private Helpers
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
STATIC String GetWindowsError() {
|
||||||
|
DWORD errorID = ::GetLastError();
|
||||||
|
if(errorID == 0) return String();
|
||||||
|
|
||||||
|
LPSTR messageBuffer = nullptr;
|
||||||
|
size_t size = FormatMessageA(
|
||||||
|
FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
|
||||||
|
NULL, errorID, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
||||||
|
(LPSTR)&messageBuffer, 0, NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
String message(messageBuffer, size);
|
||||||
|
LocalFree(messageBuffer);
|
||||||
|
return String("Win32 Error: ") + message;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
};
|
||||||
|
}
|
||||||
123
Src/IACore/inc/IACore/Environment.hpp
Normal file
123
Src/IACore/inc/IACore/Environment.hpp
Normal file
@ -0,0 +1,123 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class Environment
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Getters
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
// Modern approach: Returns nullopt if variable doesn't exist
|
||||||
|
STATIC std::optional<String> Find(CONST String &name)
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
// 1. Get required size (result includes null terminator if buffer is null/too small)
|
||||||
|
DWORD bufferSize = GetEnvironmentVariableA(name.c_str(), nullptr, 0);
|
||||||
|
|
||||||
|
if (bufferSize == 0)
|
||||||
|
{
|
||||||
|
// DWORD 0 means failed (usually ERROR_ENVVAR_NOT_FOUND)
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Allocate (bufferSize includes the null terminator request)
|
||||||
|
String result;
|
||||||
|
result.resize(bufferSize);
|
||||||
|
|
||||||
|
// 3. Fetch
|
||||||
|
// Returns num chars written EXCLUDING null terminator
|
||||||
|
DWORD actualSize = GetEnvironmentVariableA(name.c_str(), result.data(), bufferSize);
|
||||||
|
|
||||||
|
if (actualSize == 0 || actualSize > bufferSize)
|
||||||
|
{
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Resize down to exclude the null terminator and any slack
|
||||||
|
result.resize(actualSize);
|
||||||
|
return result;
|
||||||
|
|
||||||
|
#else
|
||||||
|
// POSIX (Linux/Mac)
|
||||||
|
// getenv returns a pointer to the environment area. Do NOT free it.
|
||||||
|
const char *val = std::getenv(name.c_str());
|
||||||
|
if (val == nullptr)
|
||||||
|
{
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
return String(val);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// Classic approach: Returns value or default (defaulting to "")
|
||||||
|
// Matches your old API behavior but is safer
|
||||||
|
STATIC String Get(CONST String &name, CONST String &defaultValue = "")
|
||||||
|
{
|
||||||
|
return Find(name).value_or(defaultValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Setters
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
STATIC BOOL Set(CONST String &name, CONST String &value)
|
||||||
|
{
|
||||||
|
if (name.empty())
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
return SetEnvironmentVariableA(name.c_str(), value.c_str()) != 0;
|
||||||
|
#else
|
||||||
|
// setenv(name, value, overwrite)
|
||||||
|
// Returns 0 on success, -1 on error
|
||||||
|
return setenv(name.c_str(), value.c_str(), 1) == 0;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC BOOL Unset(CONST String &name)
|
||||||
|
{
|
||||||
|
if (name.empty())
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
// Windows unsets a variable by setting it to NULL
|
||||||
|
return SetEnvironmentVariableA(name.c_str(), nullptr) != 0;
|
||||||
|
#else
|
||||||
|
return unsetenv(name.c_str()) == 0;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
// Utilities
|
||||||
|
// ---------------------------------------------------------------------
|
||||||
|
|
||||||
|
STATIC BOOL Exists(CONST String &name)
|
||||||
|
{
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
return GetEnvironmentVariableA(name.c_str(), nullptr, 0) > 0;
|
||||||
|
#else
|
||||||
|
return std::getenv(name.c_str()) != nullptr;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
50
Src/IACore/inc/IACore/FileOps.hpp
Normal file
50
Src/IACore/inc/IACore/FileOps.hpp
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/StreamReader.hpp>
|
||||||
|
#include <IACore/StreamWriter.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class FileOps
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
STATIC FilePath NormalizeExecutablePath(IN CONST FilePath &path);
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC VOID UnmapFile(IN PCUINT8 mappedPtr);
|
||||||
|
STATIC Expected<PCUINT8, String> MapFile(IN CONST FilePath &path, OUT SIZE_T &size);
|
||||||
|
|
||||||
|
// @param `isOwner` TRUE to allocate/truncate. FALSE to just open.
|
||||||
|
STATIC Expected<PUINT8, String> MapSharedMemory(IN CONST String &name, IN SIZE_T size, IN BOOL isOwner);
|
||||||
|
STATIC VOID UnlinkSharedMemory(IN CONST String &name);
|
||||||
|
|
||||||
|
STATIC Expected<StreamReader, String> StreamFromFile(IN CONST FilePath &path);
|
||||||
|
STATIC Expected<StreamWriter, String> StreamToFile(IN CONST FilePath &path, IN BOOL overwrite = false);
|
||||||
|
|
||||||
|
STATIC Expected<String, String> ReadTextFile(IN CONST FilePath &path);
|
||||||
|
STATIC Expected<Vector<UINT8>, String> ReadBinaryFile(IN CONST FilePath &path);
|
||||||
|
STATIC Expected<SIZE_T, String> WriteTextFile(IN CONST FilePath &path, IN CONST String &contents,
|
||||||
|
IN BOOL overwrite = false);
|
||||||
|
STATIC Expected<SIZE_T, String> WriteBinaryFile(IN CONST FilePath &path, IN Span<UINT8> contents,
|
||||||
|
IN BOOL overwrite = false);
|
||||||
|
|
||||||
|
private:
|
||||||
|
STATIC UnorderedMap<PCUINT8, Tuple<PVOID, PVOID, PVOID>> s_mappedFiles;
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
196
Src/IACore/inc/IACore/HttpClient.hpp
Normal file
196
Src/IACore/inc/IACore/HttpClient.hpp
Normal file
@ -0,0 +1,196 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/JSON.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class HttpClient
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
enum class EHeaderType
|
||||||
|
{
|
||||||
|
ACCEPT,
|
||||||
|
ACCEPT_CHARSET,
|
||||||
|
ACCEPT_ENCODING,
|
||||||
|
ACCEPT_LANGUAGE,
|
||||||
|
AUTHORIZATION,
|
||||||
|
CACHE_CONTROL,
|
||||||
|
CONNECTION,
|
||||||
|
CONTENT_LENGTH,
|
||||||
|
CONTENT_TYPE,
|
||||||
|
COOKIE,
|
||||||
|
DATE,
|
||||||
|
EXPECT,
|
||||||
|
HOST,
|
||||||
|
IF_MATCH,
|
||||||
|
IF_MODIFIED_SINCE,
|
||||||
|
IF_NONE_MATCH,
|
||||||
|
ORIGIN,
|
||||||
|
PRAGMA,
|
||||||
|
PROXY_AUTHORIZATION,
|
||||||
|
RANGE,
|
||||||
|
REFERER,
|
||||||
|
TE,
|
||||||
|
UPGRADE,
|
||||||
|
USER_AGENT,
|
||||||
|
VIA,
|
||||||
|
WARNING
|
||||||
|
};
|
||||||
|
|
||||||
|
enum class EResponseCode : INT32
|
||||||
|
{
|
||||||
|
// 1xx Informational
|
||||||
|
CONTINUE = 100,
|
||||||
|
SWITCHING_PROTOCOLS = 101,
|
||||||
|
PROCESSING = 102,
|
||||||
|
EARLY_HINTS = 103,
|
||||||
|
|
||||||
|
// 2xx Success
|
||||||
|
OK = 200,
|
||||||
|
CREATED = 201,
|
||||||
|
ACCEPTED = 202,
|
||||||
|
NON_AUTHORITATIVE_INFORMATION = 203,
|
||||||
|
NO_CONTENT = 204,
|
||||||
|
RESET_CONTENT = 205,
|
||||||
|
PARTIAL_CONTENT = 206,
|
||||||
|
MULTI_STATUS = 207,
|
||||||
|
ALREADY_REPORTED = 208,
|
||||||
|
IM_USED = 226,
|
||||||
|
|
||||||
|
// 3xx Redirection
|
||||||
|
MULTIPLE_CHOICES = 300,
|
||||||
|
MOVED_PERMANENTLY = 301,
|
||||||
|
FOUND = 302,
|
||||||
|
SEE_OTHER = 303,
|
||||||
|
NOT_MODIFIED = 304,
|
||||||
|
USE_PROXY = 305,
|
||||||
|
TEMPORARY_REDIRECT = 307,
|
||||||
|
PERMANENT_REDIRECT = 308,
|
||||||
|
|
||||||
|
// 4xx Client Error
|
||||||
|
BAD_REQUEST = 400,
|
||||||
|
UNAUTHORIZED = 401,
|
||||||
|
PAYMENT_REQUIRED = 402,
|
||||||
|
FORBIDDEN = 403,
|
||||||
|
NOT_FOUND = 404,
|
||||||
|
METHOD_NOT_ALLOWED = 405,
|
||||||
|
NOT_ACCEPTABLE = 406,
|
||||||
|
PROXY_AUTHENTICATION_REQUIRED = 407,
|
||||||
|
REQUEST_TIMEOUT = 408,
|
||||||
|
CONFLICT = 409,
|
||||||
|
GONE = 410,
|
||||||
|
LENGTH_REQUIRED = 411,
|
||||||
|
PRECONDITION_FAILED = 412,
|
||||||
|
PAYLOAD_TOO_LARGE = 413,
|
||||||
|
URI_TOO_LONG = 414,
|
||||||
|
UNSUPPORTED_MEDIA_TYPE = 415,
|
||||||
|
RANGE_NOT_SATISFIABLE = 416,
|
||||||
|
EXPECTATION_FAILED = 417,
|
||||||
|
IM_A_TEAPOT = 418,
|
||||||
|
MISDIRECTED_REQUEST = 421,
|
||||||
|
UNPROCESSABLE_ENTITY = 422,
|
||||||
|
LOCKED = 423,
|
||||||
|
FAILED_DEPENDENCY = 424,
|
||||||
|
TOO_EARLY = 425,
|
||||||
|
UPGRADE_REQUIRED = 426,
|
||||||
|
PRECONDITION_REQUIRED = 428,
|
||||||
|
TOO_MANY_REQUESTS = 429,
|
||||||
|
REQUEST_HEADER_FIELDS_TOO_LARGE = 431,
|
||||||
|
UNAVAILABLE_FOR_LEGAL_REASONS = 451,
|
||||||
|
|
||||||
|
// 5xx Server Error
|
||||||
|
INTERNAL_SERVER_ERROR = 500,
|
||||||
|
NOT_IMPLEMENTED = 501,
|
||||||
|
BAD_GATEWAY = 502,
|
||||||
|
SERVICE_UNAVAILABLE = 503,
|
||||||
|
GATEWAY_TIMEOUT = 504,
|
||||||
|
HTTP_VERSION_NOT_SUPPORTED = 505,
|
||||||
|
VARIANT_ALSO_NEGOTIATES = 506,
|
||||||
|
INSUFFICIENT_STORAGE = 507,
|
||||||
|
LOOP_DETECTED = 508,
|
||||||
|
NOT_EXTENDED = 510,
|
||||||
|
NETWORK_AUTHENTICATION_REQUIRED = 511
|
||||||
|
};
|
||||||
|
|
||||||
|
using Header = KeyValuePair<EHeaderType, String>;
|
||||||
|
|
||||||
|
public:
|
||||||
|
HttpClient(IN CONST String &host);
|
||||||
|
~HttpClient();
|
||||||
|
|
||||||
|
public:
|
||||||
|
Expected<String, String> RawGet(IN CONST String &path, IN Span<CONST Header> headers,
|
||||||
|
IN PCCHAR defaultContentType = "application/x-www-form-urlencoded");
|
||||||
|
Expected<String, String> RawPost(IN CONST String &path, IN Span<CONST Header> headers, IN CONST String &body,
|
||||||
|
IN PCCHAR defaultContentType = "application/x-www-form-urlencoded");
|
||||||
|
|
||||||
|
template<typename _response_type>
|
||||||
|
Expected<_response_type, String> JsonGet(IN CONST String &path, IN Span<CONST Header> headers);
|
||||||
|
|
||||||
|
template<typename _payload_type, typename _response_type>
|
||||||
|
Expected<_response_type, String> JsonPost(IN CONST String &path, IN Span<CONST Header> headers,
|
||||||
|
IN CONST _payload_type &body);
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC String UrlEncode(IN CONST String &value);
|
||||||
|
STATIC String UrlDecode(IN CONST String &value);
|
||||||
|
|
||||||
|
STATIC String HeaderTypeToString(IN EHeaderType type);
|
||||||
|
STATIC Header CreateHeader(IN EHeaderType key, IN CONST String &value);
|
||||||
|
|
||||||
|
STATIC BOOL IsSuccessResponseCode(IN EResponseCode code);
|
||||||
|
|
||||||
|
public:
|
||||||
|
EResponseCode LastResponseCode()
|
||||||
|
{
|
||||||
|
return m_lastResponseCode;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
PVOID m_client{};
|
||||||
|
EResponseCode m_lastResponseCode;
|
||||||
|
|
||||||
|
private:
|
||||||
|
String PreprocessResponse(IN CONST String& response);
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename _response_type>
|
||||||
|
Expected<_response_type, String> HttpClient::JsonGet(IN CONST String &path, IN Span<CONST Header> headers)
|
||||||
|
{
|
||||||
|
const auto rawResponse = RawGet(path, headers, "application/json");
|
||||||
|
if (!rawResponse)
|
||||||
|
return MakeUnexpected(rawResponse.error());
|
||||||
|
if (LastResponseCode() != EResponseCode::OK)
|
||||||
|
return MakeUnexpected(std::format("Server responded with code {}", (INT32) LastResponseCode()));
|
||||||
|
return JSON::ParseToStruct<_response_type>(*rawResponse);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename _payload_type, typename _response_type>
|
||||||
|
Expected<_response_type, String> HttpClient::JsonPost(IN CONST String &path, IN Span<CONST Header> headers,
|
||||||
|
IN CONST _payload_type &body)
|
||||||
|
{
|
||||||
|
const auto encodedBody = IA_TRY(JSON::EncodeStruct(body));
|
||||||
|
const auto rawResponse = RawPost(path, headers, encodedBody, "application/json");
|
||||||
|
if (!rawResponse)
|
||||||
|
return MakeUnexpected(rawResponse.error());
|
||||||
|
if (LastResponseCode() != EResponseCode::OK)
|
||||||
|
return MakeUnexpected(std::format("Server responded with code {}", (INT32) LastResponseCode()));
|
||||||
|
return JSON::ParseToStruct<_response_type>(*rawResponse);
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
42
Src/IACore/inc/IACore/IACore.hpp
Normal file
42
Src/IACore/inc/IACore/IACore.hpp
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
// Must be called from main thread
|
||||||
|
VOID Initialize();
|
||||||
|
// Must be called from same thread as Initialize
|
||||||
|
VOID Terminate();
|
||||||
|
|
||||||
|
UINT64 GetUnixTime();
|
||||||
|
UINT64 GetTicksCount();
|
||||||
|
FLOAT64 GetSecondsCount();
|
||||||
|
|
||||||
|
FLOAT32 GetRandom();
|
||||||
|
UINT32 GetRandom(IN UINT32 seed);
|
||||||
|
INT64 GetRandom(IN INT64 min, IN INT64 max);
|
||||||
|
|
||||||
|
BOOL IsMainThread();
|
||||||
|
VOID Sleep(IN UINT64 milliseconds);
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
#endif
|
||||||
261
Src/IACore/inc/IACore/IATest.hpp
Normal file
261
Src/IACore/inc/IACore/IATest.hpp
Normal file
@ -0,0 +1,261 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
|
||||||
|
#include <exception>
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Macros
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#define valid_iatest_runner(type) iatest::_valid_iatest_runner<type>::value_type
|
||||||
|
|
||||||
|
// Internal macro to handle the return logic
|
||||||
|
#define __iat_micro_test(call) \
|
||||||
|
if(!(call)) return FALSE
|
||||||
|
|
||||||
|
#define IAT_CHECK(v) __iat_micro_test(_test((v), #v))
|
||||||
|
#define IAT_CHECK_NOT(v) __iat_micro_test(_test_not((v), "NOT " #v))
|
||||||
|
#define IAT_CHECK_EQ(lhs, rhs) __iat_micro_test(_test_eq((lhs), (rhs), #lhs " == " #rhs))
|
||||||
|
#define IAT_CHECK_NEQ(lhs, rhs) __iat_micro_test(_test_neq((lhs), (rhs), #lhs " != " #rhs))
|
||||||
|
|
||||||
|
// Float specific checks (Game dev essential)
|
||||||
|
#define IAT_CHECK_APPROX(lhs, rhs) __iat_micro_test(_test_approx((lhs), (rhs), #lhs " ~= " #rhs))
|
||||||
|
|
||||||
|
#define IAT_UNIT(func) _test_unit([this](){ return this->func(); }, # func)
|
||||||
|
#define IAT_NAMED_UNIT(n, func) _test_unit([this](){ return this->func(); }, n)
|
||||||
|
|
||||||
|
#define IAT_BLOCK(name) class name: public ia::iatest::block
|
||||||
|
|
||||||
|
// Concatenation fix for macros
|
||||||
|
#define IAT_BEGIN_BLOCK(_group, _name) class _group##_##_name : public ia::iatest::block { \
|
||||||
|
public: PCCHAR name() CONST OVERRIDE { return #_group "::" #_name; } private:
|
||||||
|
|
||||||
|
#define IAT_END_BLOCK() };
|
||||||
|
|
||||||
|
#define IAT_BEGIN_TEST_LIST() public: VOID declareTests() OVERRIDE {
|
||||||
|
#define IAT_ADD_TEST(name) IAT_UNIT(name)
|
||||||
|
#define IAT_END_TEST_LIST() } private:
|
||||||
|
|
||||||
|
namespace ia::iatest
|
||||||
|
{
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Type Printing Helper (To show WHAT failed)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
template<typename T>
|
||||||
|
std::string ToString(CONST T& value) {
|
||||||
|
if constexpr (std::is_arithmetic_v<T>) {
|
||||||
|
return std::to_string(value);
|
||||||
|
} else if constexpr (std::is_same_v<T, std::string> || std::is_same_v<T, const char*>) {
|
||||||
|
return std::string("\"") + value + "\"";
|
||||||
|
} else {
|
||||||
|
return "{Object}"; // Fallback for complex types
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specialization for pointers
|
||||||
|
template<typename T>
|
||||||
|
std::string ToString(T* value) {
|
||||||
|
if (value == NULLPTR) return "nullptr";
|
||||||
|
std::stringstream ss;
|
||||||
|
ss << "ptr(" << (void*)value << ")";
|
||||||
|
return ss.str();
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Core Structures
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
|
||||||
|
DEFINE_TYPE(functor_t, std::function<BOOL()>);
|
||||||
|
|
||||||
|
struct unit_t {
|
||||||
|
std::string Name;
|
||||||
|
functor_t Functor;
|
||||||
|
};
|
||||||
|
|
||||||
|
class block
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
virtual ~block() = default;
|
||||||
|
PURE_VIRTUAL(PCCHAR name() CONST);
|
||||||
|
PURE_VIRTUAL(VOID declareTests());
|
||||||
|
|
||||||
|
std::vector<unit_t>& units() { return m_units; }
|
||||||
|
|
||||||
|
protected:
|
||||||
|
// Generic Equality
|
||||||
|
template<typename T1, typename T2>
|
||||||
|
BOOL _test_eq(IN CONST T1& lhs, IN CONST T2& rhs, IN PCCHAR description)
|
||||||
|
{
|
||||||
|
if(lhs != rhs) {
|
||||||
|
print_fail(description, ToString(lhs), ToString(rhs));
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generic Inequality
|
||||||
|
template<typename T1, typename T2>
|
||||||
|
BOOL _test_neq(IN CONST T1& lhs, IN CONST T2& rhs, IN PCCHAR description)
|
||||||
|
{
|
||||||
|
if(lhs == rhs) {
|
||||||
|
print_fail(description, ToString(lhs), "NOT " + ToString(rhs));
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Floating Point Approximation (Epsilon check)
|
||||||
|
template<typename T>
|
||||||
|
BOOL _test_approx(IN T lhs, IN T rhs, IN PCCHAR description)
|
||||||
|
{
|
||||||
|
static_assert(std::is_floating_point_v<T>, "Approx only works for floats/doubles");
|
||||||
|
T diff = std::abs(lhs - rhs);
|
||||||
|
if (diff > static_cast<T>(0.0001)) { // Default epsilon
|
||||||
|
print_fail(description, ToString(lhs), ToString(rhs));
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL _test(IN BOOL value, IN PCCHAR description)
|
||||||
|
{
|
||||||
|
if(!value) {
|
||||||
|
printf(__CC_BLUE " %s... " __CC_RED "FAILED" __CC_DEFAULT "\n", description);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL _test_not(IN BOOL value, IN PCCHAR description)
|
||||||
|
{
|
||||||
|
if(value) {
|
||||||
|
printf(__CC_BLUE " %s... " __CC_RED "FAILED" __CC_DEFAULT "\n", description);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID _test_unit(IN functor_t functor, IN PCCHAR name)
|
||||||
|
{
|
||||||
|
m_units.push_back({name, functor});
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
VOID print_fail(PCCHAR desc, std::string v1, std::string v2) {
|
||||||
|
printf(__CC_BLUE " %s... " __CC_RED "FAILED" __CC_DEFAULT "\n", desc);
|
||||||
|
printf(__CC_RED " Expected: %s" __CC_DEFAULT "\n", v2.c_str());
|
||||||
|
printf(__CC_RED " Actual: %s" __CC_DEFAULT "\n", v1.c_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<unit_t> m_units;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename block_class>
|
||||||
|
concept valid_block_class = std::derived_from<block_class, block>;
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Runner
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
|
||||||
|
template<BOOL stopOnFail = false, BOOL isVerbose = false>
|
||||||
|
class runner
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
runner(){}
|
||||||
|
~runner() { summarize(); }
|
||||||
|
|
||||||
|
template<typename block_class>
|
||||||
|
requires valid_block_class<block_class>
|
||||||
|
VOID testBlock();
|
||||||
|
|
||||||
|
private:
|
||||||
|
VOID summarize();
|
||||||
|
|
||||||
|
private:
|
||||||
|
SIZE_T m_testCount { 0 };
|
||||||
|
SIZE_T m_failCount { 0 };
|
||||||
|
SIZE_T m_blockCount { 0 };
|
||||||
|
};
|
||||||
|
|
||||||
|
template<BOOL stopOnFail, BOOL isVerbose>
|
||||||
|
template<typename block_class>
|
||||||
|
requires valid_block_class<block_class>
|
||||||
|
VOID runner<stopOnFail, isVerbose>::testBlock()
|
||||||
|
{
|
||||||
|
m_blockCount++;
|
||||||
|
block_class b;
|
||||||
|
b.declareTests();
|
||||||
|
|
||||||
|
printf(__CC_MAGENTA "Testing [%s]..." __CC_DEFAULT "\n", b.name());
|
||||||
|
|
||||||
|
for(auto& v: b.units())
|
||||||
|
{
|
||||||
|
m_testCount++;
|
||||||
|
if constexpr(isVerbose) {
|
||||||
|
printf(__CC_YELLOW " Testing %s...\n" __CC_DEFAULT, v.Name.c_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL result = FALSE;
|
||||||
|
try {
|
||||||
|
// Execute the test function
|
||||||
|
result = v.Functor();
|
||||||
|
}
|
||||||
|
catch (const std::exception& e) {
|
||||||
|
printf(__CC_RED " CRITICAL EXCEPTION in %s: %s\n" __CC_DEFAULT, v.Name.c_str(), e.what());
|
||||||
|
result = FALSE;
|
||||||
|
}
|
||||||
|
catch (...) {
|
||||||
|
printf(__CC_RED " UNKNOWN CRITICAL EXCEPTION in %s\n" __CC_DEFAULT, v.Name.c_str());
|
||||||
|
result = FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(!result)
|
||||||
|
{
|
||||||
|
m_failCount++;
|
||||||
|
if constexpr(stopOnFail) { summarize(); exit(-1); }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fputs("\n", stdout);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<BOOL stopOnFail, BOOL isVerbose>
|
||||||
|
VOID runner<stopOnFail, isVerbose>::summarize()
|
||||||
|
{
|
||||||
|
printf(__CC_GREEN "\n-----------------------------------\n\t SUMMARY\n-----------------------------------\n");
|
||||||
|
|
||||||
|
if(!m_failCount) {
|
||||||
|
printf("\n\tALL TESTS PASSED!\n\n");
|
||||||
|
} else {
|
||||||
|
FLOAT64 successRate = (100.0 * static_cast<FLOAT64>(m_testCount - m_failCount)/static_cast<FLOAT64>(m_testCount));
|
||||||
|
printf(__CC_RED "%zu OUT OF %zu TESTS FAILED\n" __CC_YELLOW "Success Rate: %.2f%%\n", m_failCount, m_testCount, successRate);
|
||||||
|
}
|
||||||
|
|
||||||
|
printf(__CC_MAGENTA "Ran %zu test(s) across %zu block(s)\n" __CC_GREEN "-----------------------------------" __CC_DEFAULT "\n", m_testCount, m_blockCount);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename>
|
||||||
|
struct _valid_iatest_runner : std::false_type {};
|
||||||
|
template<BOOL stopOnFail, BOOL isVerbose>
|
||||||
|
struct _valid_iatest_runner<runner<stopOnFail,isVerbose>> : std::true_type {};
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // __cplusplus
|
||||||
150
Src/IACore/inc/IACore/IPC.hpp
Normal file
150
Src/IACore/inc/IACore/IPC.hpp
Normal file
@ -0,0 +1,150 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/ADT/RingBuffer.hpp>
|
||||||
|
#include <IACore/ProcessOps.hpp>
|
||||||
|
#include <IACore/SocketOps.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
using IPC_PacketHeader = RingBufferView::PacketHeader;
|
||||||
|
|
||||||
|
struct alignas(64) IPC_SharedMemoryLayout
|
||||||
|
{
|
||||||
|
// =========================================================
|
||||||
|
// SECTION 1: METADATA & HANDSHAKE
|
||||||
|
// =========================================================
|
||||||
|
struct Header
|
||||||
|
{
|
||||||
|
UINT32 Magic; // 0x49414950 ("IAIP")
|
||||||
|
UINT32 Version; // 1
|
||||||
|
UINT64 TotalSize; // Total size of SHM block
|
||||||
|
} Meta;
|
||||||
|
|
||||||
|
// Pad to ensure MONI starts on a fresh cache line (64 bytes)
|
||||||
|
UINT8 _pad0[64 - sizeof(Header)];
|
||||||
|
|
||||||
|
// =========================================================
|
||||||
|
// SECTION 2: RING BUFFER CONTROL BLOCKS
|
||||||
|
// =========================================================
|
||||||
|
|
||||||
|
// RingBufferView::ControlBlock is already 64-byte aligned internally.
|
||||||
|
RingBufferView::ControlBlock MONI_Control;
|
||||||
|
RingBufferView::ControlBlock MINO_Control;
|
||||||
|
|
||||||
|
// =========================================================
|
||||||
|
// SECTION 3: DATA BUFFER OFFSETS
|
||||||
|
// =========================================================
|
||||||
|
|
||||||
|
UINT64 MONI_DataOffset;
|
||||||
|
UINT64 MONI_DataSize;
|
||||||
|
|
||||||
|
UINT64 MINO_DataOffset;
|
||||||
|
UINT64 MINO_DataSize;
|
||||||
|
|
||||||
|
// Pad to ensure the actual Data Buffer starts on a fresh cache line
|
||||||
|
UINT8 _pad1[64 - (sizeof(UINT64) * 4)];
|
||||||
|
|
||||||
|
static constexpr size_t GetHeaderSize()
|
||||||
|
{
|
||||||
|
return sizeof(IPC_SharedMemoryLayout);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Static assert to ensure manual padding logic is correct
|
||||||
|
static_assert(sizeof(IPC_SharedMemoryLayout) % 64 == 0, "IPC Layout is not cache-line aligned!");
|
||||||
|
|
||||||
|
class IPC_Node
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
virtual ~IPC_Node();
|
||||||
|
|
||||||
|
// When Manager spawns a node, `connectionString` is passed
|
||||||
|
// as the first command line argument
|
||||||
|
Expected<VOID, String> Connect(IN PCCHAR connectionString);
|
||||||
|
|
||||||
|
VOID Update();
|
||||||
|
|
||||||
|
VOID SendSignal(IN UINT8 signal);
|
||||||
|
VOID SendPacket(IN UINT16 packetID, IN Span<CONST UINT8> payload);
|
||||||
|
|
||||||
|
protected:
|
||||||
|
PURE_VIRTUAL(VOID OnSignal(IN UINT8 signal));
|
||||||
|
PURE_VIRTUAL(VOID OnPacket(IN UINT16 packetID, IN Span<CONST UINT8> payload));
|
||||||
|
|
||||||
|
private:
|
||||||
|
String m_shmName;
|
||||||
|
PUINT8 m_sharedMemory{};
|
||||||
|
Vector<UINT8> m_recieveBuffer;
|
||||||
|
SocketHandle m_socket{INVALID_SOCKET};
|
||||||
|
|
||||||
|
UniquePtr<RingBufferView> MONI; // Manager Out, Node In
|
||||||
|
UniquePtr<RingBufferView> MINO; // Manager In, Node Out
|
||||||
|
};
|
||||||
|
|
||||||
|
class IPC_Manager
|
||||||
|
{
|
||||||
|
struct NodeSession
|
||||||
|
{
|
||||||
|
SteadyTimePoint CreationTime{};
|
||||||
|
SharedPtr<ProcessHandle> ProcessHandle;
|
||||||
|
|
||||||
|
String SharedMemName;
|
||||||
|
PUINT8 MappedPtr{};
|
||||||
|
|
||||||
|
SocketHandle ListenerSocket{INVALID_SOCKET};
|
||||||
|
SocketHandle DataSocket{INVALID_SOCKET};
|
||||||
|
|
||||||
|
UniquePtr<RingBufferView> MONI; // Manager Out, Node In
|
||||||
|
UniquePtr<RingBufferView> MINO; // Manager In, Node Out
|
||||||
|
|
||||||
|
BOOL IsReady{FALSE};
|
||||||
|
|
||||||
|
VOID SendSignal(IN UINT8 signal);
|
||||||
|
VOID SendPacket(IN UINT16 packetID, IN Span<CONST UINT8> payload);
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC CONSTEXPR UINT32 DEFAULT_NODE_SHARED_MEMORY_SIZE = SIZE_MB(4);
|
||||||
|
|
||||||
|
public:
|
||||||
|
IPC_Manager();
|
||||||
|
virtual ~IPC_Manager();
|
||||||
|
|
||||||
|
VOID Update();
|
||||||
|
|
||||||
|
Expected<NativeProcessID, String> SpawnNode(IN CONST FilePath &executablePath,
|
||||||
|
IN UINT32 sharedMemorySize = DEFAULT_NODE_SHARED_MEMORY_SIZE);
|
||||||
|
BOOL WaitTillNodeIsOnline(IN NativeProcessID node);
|
||||||
|
|
||||||
|
VOID ShutdownNode(IN NativeProcessID node);
|
||||||
|
|
||||||
|
VOID SendSignal(IN NativeProcessID node, IN UINT8 signal);
|
||||||
|
VOID SendPacket(IN NativeProcessID node, IN UINT16 packetID, IN Span<CONST UINT8> payload);
|
||||||
|
|
||||||
|
protected:
|
||||||
|
PURE_VIRTUAL(VOID OnSignal(IN NativeProcessID node, IN UINT8 signal));
|
||||||
|
PURE_VIRTUAL(VOID OnPacket(IN NativeProcessID node, IN UINT16 packetID, IN Span<CONST UINT8> payload));
|
||||||
|
|
||||||
|
private:
|
||||||
|
Vector<UINT8> m_recieveBuffer;
|
||||||
|
Vector<UniquePtr<NodeSession>> m_activeSessions;
|
||||||
|
Vector<UniquePtr<NodeSession>> m_pendingSessions;
|
||||||
|
UnorderedMap<NativeProcessID, NodeSession *> m_activeSessionMap;
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
58
Src/IACore/inc/IACore/JSON.hpp
Normal file
58
Src/IACore/inc/IACore/JSON.hpp
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#include <simdjson.h>
|
||||||
|
#include <glaze/glaze.hpp>
|
||||||
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class JSON
|
||||||
|
{
|
||||||
|
private:
|
||||||
|
STATIC CONSTEXPR AUTO GLAZE_JSON_OPTS = glz::opts{.error_on_unknown_keys = false};
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC Expected<nlohmann::json, String> Parse(IN CONST String &json);
|
||||||
|
STATIC Expected<Pair<simdjson::dom::parser, simdjson::dom::object>, String> ParseReadOnly(IN CONST String &json);
|
||||||
|
template<typename _object_type> STATIC Expected<_object_type, String> ParseToStruct(IN CONST String &json);
|
||||||
|
|
||||||
|
STATIC String Encode(IN nlohmann::json data);
|
||||||
|
template<typename _object_type> STATIC Expected<String, String> EncodeStruct(IN CONST _object_type &data);
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename _object_type> Expected<_object_type, String> JSON::ParseToStruct(IN CONST String &json)
|
||||||
|
{
|
||||||
|
_object_type result{};
|
||||||
|
const auto parseError = glz::read<GLAZE_JSON_OPTS>(result, json);
|
||||||
|
if (parseError)
|
||||||
|
return MakeUnexpected(std::format("JSON Error: {}", glz::format_error(parseError, json)));
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename _object_type> Expected<String, String> JSON::EncodeStruct(IN CONST _object_type &data)
|
||||||
|
{
|
||||||
|
String result;
|
||||||
|
const auto encodeError = glz::write_json(data, result);
|
||||||
|
if (encodeError)
|
||||||
|
return MakeUnexpected(std::format("JSON Error: {}", glz::format_error(encodeError)));
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
119
Src/IACore/inc/IACore/Logger.hpp
Normal file
119
Src/IACore/inc/IACore/Logger.hpp
Normal file
@ -0,0 +1,119 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class Logger
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
enum class ELogLevel
|
||||||
|
{
|
||||||
|
VERBOSE,
|
||||||
|
INFO,
|
||||||
|
WARN,
|
||||||
|
ERROR
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
STATIC BOOL EnableLoggingToDisk(IN PCCHAR filePath);
|
||||||
|
STATIC VOID SetLogLevel(IN ELogLevel logLevel);
|
||||||
|
|
||||||
|
template<typename... Args> STATIC VOID Status(FormatterString<Args...> fmt, Args &&...args)
|
||||||
|
{
|
||||||
|
LogStatus(std::vformat(fmt.get(), std::make_format_args(args...)));
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename... Args> STATIC VOID Info(FormatterString<Args...> fmt, Args &&...args)
|
||||||
|
{
|
||||||
|
LogInfo(std::vformat(fmt.get(), std::make_format_args(args...)));
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename... Args> STATIC VOID Warn(FormatterString<Args...> fmt, Args &&...args)
|
||||||
|
{
|
||||||
|
LogWarn(std::vformat(fmt.get(), std::make_format_args(args...)));
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename... Args> STATIC VOID Error(FormatterString<Args...> fmt, Args &&...args)
|
||||||
|
{
|
||||||
|
LogError(std::vformat(fmt.get(), std::make_format_args(args...)));
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID FlushLogs();
|
||||||
|
|
||||||
|
private:
|
||||||
|
#if IA_DISABLE_LOGGING > 0
|
||||||
|
STATIC VOID LogStatus(IN String &&msg)
|
||||||
|
{
|
||||||
|
UNUSED(msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID LogInfo(IN String &&msg)
|
||||||
|
{
|
||||||
|
UNUSED(msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID LogWarn(IN String &&msg)
|
||||||
|
{
|
||||||
|
UNUSED(msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID LogError(IN String &&msg)
|
||||||
|
{
|
||||||
|
UNUSED(msg);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
STATIC VOID LogStatus(IN String &&msg)
|
||||||
|
{
|
||||||
|
if (s_logLevel <= ELogLevel::VERBOSE)
|
||||||
|
LogInternal(__CC_WHITE, "STATUS", IA_MOVE(msg));
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID LogInfo(IN String &&msg)
|
||||||
|
{
|
||||||
|
if (s_logLevel <= ELogLevel::INFO)
|
||||||
|
LogInternal(__CC_GREEN, "INFO", IA_MOVE(msg));
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID LogWarn(IN String &&msg)
|
||||||
|
{
|
||||||
|
if (s_logLevel <= ELogLevel::WARN)
|
||||||
|
LogInternal(__CC_YELLOW, "WARN", IA_MOVE(msg));
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC VOID LogError(IN String &&msg)
|
||||||
|
{
|
||||||
|
if (s_logLevel <= ELogLevel::ERROR)
|
||||||
|
LogInternal(__CC_RED, "ERROR", IA_MOVE(msg));
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
STATIC VOID LogInternal(IN PCCHAR prefix, IN PCCHAR tag, IN String &&msg);
|
||||||
|
|
||||||
|
private:
|
||||||
|
STATIC ELogLevel s_logLevel;
|
||||||
|
STATIC std::ofstream s_logFile;
|
||||||
|
|
||||||
|
STATIC VOID Initialize();
|
||||||
|
STATIC VOID Terminate();
|
||||||
|
|
||||||
|
friend VOID Initialize();
|
||||||
|
friend VOID Terminate();
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
588
Src/IACore/inc/IACore/PCH.hpp
Normal file
588
Src/IACore/inc/IACore/PCH.hpp
Normal file
@ -0,0 +1,588 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Platform Detection
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
#if defined(WIN32) || defined(_WIN32) || defined(__WIN32__) || defined(__NT__)
|
||||||
|
# ifdef _WIN64
|
||||||
|
# define IA_PLATFORM_WIN64 1
|
||||||
|
# define IA_PLATFORM_WINDOWS 1
|
||||||
|
# else
|
||||||
|
# define IA_PLATFORM_WIN32 1
|
||||||
|
# define IA_PLATFORM_WINDOWS 1
|
||||||
|
# endif
|
||||||
|
#elif __APPLE__
|
||||||
|
# include <TargetConditionals.h>
|
||||||
|
# if TARGET_OS_IPHONE || TARGET_IPHONE_SIMULATOR || TARGET_OS_MACCATALYST
|
||||||
|
# define IA_PLATFORM_IOS 1
|
||||||
|
# elif TARGET_OS_MAC
|
||||||
|
# define IA_PLATFORM_MAC 1
|
||||||
|
# endif
|
||||||
|
# define IA_PLATFORM_UNIX 1
|
||||||
|
#elif __ANDROID__
|
||||||
|
# define IA_PLATFORM_ANDROID 1
|
||||||
|
# define IA_PLATFORM_LINUX 1
|
||||||
|
# define IA_PLATFORM_UNIX 1
|
||||||
|
#elif __linux__
|
||||||
|
# define IA_PLATFORM_LINUX 1
|
||||||
|
# define IA_PLATFORM_UNIX 1
|
||||||
|
#elif __unix__
|
||||||
|
# define IA_PLATFORM_UNIX 1
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WIN32 || IA_PLATFORM_WIN64
|
||||||
|
# ifndef WIN32_LEAN_AND_MEAN
|
||||||
|
# define WIN32_LEAN_AND_MEAN
|
||||||
|
# endif
|
||||||
|
# ifndef NOMINMAX
|
||||||
|
# define NOMINMAX
|
||||||
|
# endif
|
||||||
|
# include <windows.h>
|
||||||
|
#undef VOID
|
||||||
|
#undef ERROR
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
# include <unistd.h>
|
||||||
|
# include <sys/wait.h>
|
||||||
|
# include <sys/mman.h>
|
||||||
|
# include <sys/stat.h>
|
||||||
|
# include <fcntl.h>
|
||||||
|
# include <spawn.h>
|
||||||
|
# include <signal.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Configuration Macros
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#define IA_CHECK(o) (o > 0)
|
||||||
|
|
||||||
|
#if defined(__IA_DEBUG) && __IA_DEBUG
|
||||||
|
# define __DEBUG_MODE__
|
||||||
|
# define __BUILD_MODE_NAME "debug"
|
||||||
|
# define DEBUG_ONLY(v) v
|
||||||
|
# ifndef _DEBUG
|
||||||
|
# define _DEBUG
|
||||||
|
# endif
|
||||||
|
#else
|
||||||
|
# define __RELEASE_MODE__
|
||||||
|
# define __BUILD_MODE_NAME "release"
|
||||||
|
# ifndef NDEBUG
|
||||||
|
# define NDEBUG
|
||||||
|
# endif
|
||||||
|
# ifndef __OPTIMIZE__
|
||||||
|
# define __OPTIMIZE__
|
||||||
|
# endif
|
||||||
|
# define DEBUG_ONLY(f)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if IA_CHECK(IA_PLATFORM_WIN64) || IA_CHECK(IA_PLATFORM_UNIX)
|
||||||
|
# define IA_CORE_PLATFORM_FEATURES 1
|
||||||
|
#else
|
||||||
|
# define IA_CORE_PLATFORM_FEATURES 0
|
||||||
|
# warning "IACore Unsupported Platform: Platform specific features will be disabled"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <assert.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <memory.h>
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# include <bit>
|
||||||
|
# include <new>
|
||||||
|
# include <span>
|
||||||
|
# include <atomic>
|
||||||
|
# include <mutex>
|
||||||
|
# include <thread>
|
||||||
|
# include <limits>
|
||||||
|
# include <cstring>
|
||||||
|
# include <cstddef>
|
||||||
|
# include <chrono>
|
||||||
|
# include <iomanip>
|
||||||
|
# include <fstream>
|
||||||
|
# include <iostream>
|
||||||
|
# include <concepts>
|
||||||
|
# include <filesystem>
|
||||||
|
# include <functional>
|
||||||
|
# include <type_traits>
|
||||||
|
# include <initializer_list>
|
||||||
|
# include <condition_variable>
|
||||||
|
|
||||||
|
# include <tuple>
|
||||||
|
# include <array>
|
||||||
|
# include <deque>
|
||||||
|
# include <string>
|
||||||
|
# include <vector>
|
||||||
|
# include <format>
|
||||||
|
# include <sstream>
|
||||||
|
# include <optional>
|
||||||
|
# include <string_view>
|
||||||
|
|
||||||
|
# include <tl/expected.hpp>
|
||||||
|
# include <ankerl/unordered_dense.h>
|
||||||
|
|
||||||
|
#else
|
||||||
|
# include <float.h>
|
||||||
|
# include <stdbool.h>
|
||||||
|
# include <stddef.h>
|
||||||
|
# include <string.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Security Macros
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#define IA_PANIC(msg) \
|
||||||
|
{ \
|
||||||
|
fprintf(stderr, "PANIC: %s\n", msg); \
|
||||||
|
__builtin_trap(); \
|
||||||
|
}
|
||||||
|
|
||||||
|
// Advanced Security features are not included in OSS builds
|
||||||
|
// (OSS version does not implement 'IAC_CHECK_*'s)
|
||||||
|
#define IACORE_SECURITY_LEVEL 0
|
||||||
|
|
||||||
|
#define IAC_SEC_LEVEL(v) (IACORE_SECURITY_LEVEL >= v)
|
||||||
|
|
||||||
|
#if __IA_DEBUG || IAC_SEC_LEVEL(1)
|
||||||
|
# define __IAC_OVERFLOW_CHECKS 1
|
||||||
|
#else
|
||||||
|
# define __IAC_OVERFLOW_CHECKS 0
|
||||||
|
#endif
|
||||||
|
#if __IA_DEBUG || IAC_SEC_LEVEL(2)
|
||||||
|
# define __IAC_SANITY_CHECKS 1
|
||||||
|
#else
|
||||||
|
# define __IAC_SANITY_CHECKS 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Language Abstraction Macros
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#define AUTO auto
|
||||||
|
#define CONST const
|
||||||
|
#define STATIC static
|
||||||
|
#define EXTERN extern
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# define VIRTUAL virtual
|
||||||
|
# define OVERRIDE override
|
||||||
|
# define CONSTEXPR constexpr
|
||||||
|
# define NOEXCEPT noexcept
|
||||||
|
# define NULLPTR nullptr
|
||||||
|
# define IA_MOVE(...) std::move(__VA_ARGS__)
|
||||||
|
# define DECONST(t, v) const_cast<t>(v)
|
||||||
|
# define NORETURN [[noreturn]]
|
||||||
|
#else
|
||||||
|
# define VIRTUAL
|
||||||
|
# define OVERRIDE
|
||||||
|
# define CONSTEXPR const
|
||||||
|
# define NOEXCEPT
|
||||||
|
# define NULLPTR NULL
|
||||||
|
# define IA_MOVE(...) (__VA_ARGS__)
|
||||||
|
# define DECONST(t, v) ((t) (v))
|
||||||
|
# define NORETURN
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define DEFINE_TYPE(t, v) typedef v t
|
||||||
|
#define FORWARD_DECLARE(t, i) t i
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# define PURE_VIRTUAL(f) VIRTUAL f = 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Attributes & Compiler Intrinsics
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#define INLINE inline
|
||||||
|
#if defined(__clang__) || defined(__GNUC__) || defined(__GNUG__)
|
||||||
|
# define ALWAYS_INLINE __attribute__((always_inline)) inline
|
||||||
|
#elif defined(_MSC_VER)
|
||||||
|
# define ALWAYS_INLINE __forceinline
|
||||||
|
#else
|
||||||
|
# define ALWAYS_INLINE inline
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define UNUSED(v) ((void) v);
|
||||||
|
|
||||||
|
#if defined(__cplusplus)
|
||||||
|
# define NO_DISCARD(s) [[nodiscard(s)]]
|
||||||
|
# define B_LIKELY(cond) (cond) [[likely]]
|
||||||
|
# define B_UNLIKELY(cond) (cond) [[unlikely]]
|
||||||
|
#else
|
||||||
|
# define NO_DISCARD(s)
|
||||||
|
# define B_LIKELY(cond) (cond)
|
||||||
|
# define B_UNLIKELY(cond) (cond)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define __INTERNAL_IA_STRINGIFY(value) #value
|
||||||
|
#define IA_STRINGIFY(value) __INTERNAL_IA_STRINGIFY(value)
|
||||||
|
|
||||||
|
#define ALIGN(a) __attribute__((aligned(a)))
|
||||||
|
|
||||||
|
// Mark every explicit assembly instruction as volatile
|
||||||
|
#define ASM(...) __asm__ volatile(__VA_ARGS__)
|
||||||
|
|
||||||
|
#ifndef NULL
|
||||||
|
# define NULL 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifndef _WIN32
|
||||||
|
# undef TRUE
|
||||||
|
# undef FALSE
|
||||||
|
# ifdef __cplusplus
|
||||||
|
# define FALSE false
|
||||||
|
# define TRUE true
|
||||||
|
# else
|
||||||
|
# define FALSE 0
|
||||||
|
# define TRUE 1
|
||||||
|
# endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Parameter Annotations
|
||||||
|
#define IN
|
||||||
|
#define OUT
|
||||||
|
#define INOUT
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Extern C Handling
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# define IA_EXTERN_C_BEGIN \
|
||||||
|
extern "C" \
|
||||||
|
{
|
||||||
|
# define IA_EXTERN_C_END }
|
||||||
|
# define C_DECL(f) extern "C" f
|
||||||
|
#else
|
||||||
|
# define IA_EXTERN_C_BEGIN
|
||||||
|
# define IA_EXTERN_C_END
|
||||||
|
# define C_DECL(f) f
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Utilities
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# define CAST(v, t) (static_cast<t>(v))
|
||||||
|
# define REINTERPRET(v, t) (reinterpret_cast<t>(v))
|
||||||
|
#else
|
||||||
|
# define CAST(v, t) ((t) (v))
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Templates and Aliases
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# define ALIAS_FUNCTION(alias, function) \
|
||||||
|
template<typename... Args> auto alias(Args &&...args) -> decltype(function(std::forward<Args>(args)...)) \
|
||||||
|
{ \
|
||||||
|
return function(std::forward<Args>(args)...); \
|
||||||
|
}
|
||||||
|
|
||||||
|
# define ALIAS_TEMPLATE_FUNCTION(t, alias, function) \
|
||||||
|
template<typename t, typename... Args> \
|
||||||
|
auto alias(Args &&...args) -> decltype(function<t>(std::forward<Args>(args)...)) \
|
||||||
|
{ \
|
||||||
|
return function<t>(std::forward<Args>(args)...); \
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Assertions
|
||||||
|
#define IA_RELEASE_ASSERT(v) assert((v))
|
||||||
|
#define IA_RELEASE_ASSERT_MSG(v, m) assert((v) && m)
|
||||||
|
|
||||||
|
#if defined(__DEBUG_MODE__)
|
||||||
|
# define IA_ASSERT(v) IA_RELEASE_ASSERT(v)
|
||||||
|
# define IA_ASSERT_MSG(v, m) IA_RELEASE_ASSERT_MSG(v, m)
|
||||||
|
#else
|
||||||
|
# define IA_ASSERT(v)
|
||||||
|
# define IA_ASSERT_MSG(v, m)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define IA_ASSERT_EQ(a, b) IA_ASSERT((a) == (b))
|
||||||
|
#define IA_ASSERT_GE(a, b) IA_ASSERT((a) >= (b))
|
||||||
|
#define IA_ASSERT_LE(a, b) IA_ASSERT(a <= b)
|
||||||
|
#define IA_ASSERT_LT(a, b) IA_ASSERT(a < b)
|
||||||
|
#define IA_ASSERT_GT(a, b) IA_ASSERT(a > b)
|
||||||
|
#define IA_ASSERT_IMPLIES(a, b) IA_ASSERT(!(a) || (b))
|
||||||
|
#define IA_ASSERT_NOT_NULL(v) IA_ASSERT(((v) != NULLPTR))
|
||||||
|
|
||||||
|
#define IA_UNREACHABLE(msg) IA_RELEASE_ASSERT_MSG(FALSE, "Unreachable code: " msg)
|
||||||
|
|
||||||
|
#define IA_TRY_PURE(expr) \
|
||||||
|
{ \
|
||||||
|
auto _ia_res = (expr); \
|
||||||
|
if (!_ia_res) \
|
||||||
|
{ \
|
||||||
|
return tl::make_unexpected(std::move(_ia_res.error())); \
|
||||||
|
} \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define IA_TRY(expr) \
|
||||||
|
__extension__({ \
|
||||||
|
auto _ia_res = (expr); \
|
||||||
|
if (!_ia_res) \
|
||||||
|
{ \
|
||||||
|
return tl::make_unexpected(std::move(_ia_res.error())); \
|
||||||
|
} \
|
||||||
|
std::move(*_ia_res); \
|
||||||
|
})
|
||||||
|
|
||||||
|
#define IA_CONCAT_IMPL(x, y) x##y
|
||||||
|
#define IA_CONCAT(x, y) IA_CONCAT_IMPL(x, y)
|
||||||
|
#define IA_UNIQUE_NAME(prefix) IA_CONCAT(prefix, __LINE__)
|
||||||
|
|
||||||
|
#define SIZE_KB(v) (v * 1024)
|
||||||
|
#define SIZE_MB(v) (v * 1024 * 1024)
|
||||||
|
#define SIZE_GB(v) (v * 1024 * 1024 * 1024)
|
||||||
|
|
||||||
|
#define ENSURE_BINARY_COMPATIBILITY(A, B) \
|
||||||
|
static_assert(sizeof(A) == sizeof(B), \
|
||||||
|
#A ", " #B " size mismatch! Do not add virtual functions or new member variables.");
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Limits & Versioning
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
# define IA_MAX_POSSIBLE_SIZE (static_cast<SIZE_T>(0x7FFFFFFFFFFFF))
|
||||||
|
#else
|
||||||
|
# define IA_MAX_POSSIBLE_SIZE ((SIZE_T) (0x7FFFFFFFFFFFF))
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define IA_MAX_PATH_LENGTH 4096
|
||||||
|
#define IA_MAX_STRING_LENGTH (IA_MAX_POSSIBLE_SIZE >> 8)
|
||||||
|
|
||||||
|
#define IA_VERSION_TYPE uint64_t
|
||||||
|
#define IA_MAKE_VERSION(major, minor, patch) \
|
||||||
|
((((uint64_t) (major) & 0xFFFFFF) << 40) | (((uint64_t) (minor) & 0xFFFFFF) << 16) | ((uint64_t) (patch) & 0xFFFF))
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// DLL Export/Import
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#if defined(_MSC_VER)
|
||||||
|
# define IA_DLL_EXPORT __declspec(dllexport)
|
||||||
|
# define IA_DLL_IMPORT __declspec(dllimport)
|
||||||
|
#elif defined(__GNUC__)
|
||||||
|
# define IA_DLL_EXPORT __attribute__((visibility("default")))
|
||||||
|
# define IA_DLL_IMPORT
|
||||||
|
#else
|
||||||
|
# define IA_DLL_EXPORT
|
||||||
|
# define IA_DLL_IMPORT
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Console Colors (ANSI Escape Codes)
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
#define __CC_BLACK "\033[30m"
|
||||||
|
#define __CC_RED "\033[31m"
|
||||||
|
#define __CC_GREEN "\033[32m"
|
||||||
|
#define __CC_YELLOW "\033[33m"
|
||||||
|
#define __CC_BLUE "\033[34m"
|
||||||
|
#define __CC_MAGENTA "\033[35m"
|
||||||
|
#define __CC_CYAN "\033[36m"
|
||||||
|
#define __CC_WHITE "\033[37m"
|
||||||
|
#define __CC_DEFAULT "\033[39m"
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Base Types
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
typedef void VOID;
|
||||||
|
|
||||||
|
#ifndef _WIN32
|
||||||
|
# ifdef __cplusplus
|
||||||
|
typedef bool BOOL;
|
||||||
|
# else
|
||||||
|
typedef _Bool BOOL;
|
||||||
|
# endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
typedef char CHAR;
|
||||||
|
typedef uint16_t CHAR16;
|
||||||
|
|
||||||
|
typedef int8_t INT8;
|
||||||
|
typedef int16_t INT16;
|
||||||
|
typedef int32_t INT32;
|
||||||
|
typedef int64_t INT64;
|
||||||
|
|
||||||
|
typedef uint8_t UINT8;
|
||||||
|
typedef uint16_t UINT16;
|
||||||
|
typedef uint32_t UINT32;
|
||||||
|
typedef uint64_t UINT64;
|
||||||
|
|
||||||
|
typedef float FLOAT32;
|
||||||
|
typedef double FLOAT64;
|
||||||
|
|
||||||
|
typedef INT32 INT;
|
||||||
|
typedef UINT32 UINT;
|
||||||
|
typedef size_t SIZE_T;
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
typedef std::make_signed_t<size_t> SSIZE_T;
|
||||||
|
typedef std::align_val_t ALIGN_T;
|
||||||
|
#else
|
||||||
|
typedef ptrdiff_t SSIZE_T;
|
||||||
|
typedef size_t ALIGN_T;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Pointer Types
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
typedef VOID *PVOID;
|
||||||
|
typedef BOOL *PBOOL;
|
||||||
|
typedef CHAR *PCHAR;
|
||||||
|
typedef CHAR16 *PCHAR16;
|
||||||
|
|
||||||
|
typedef INT8 *PINT8;
|
||||||
|
typedef INT16 *PINT16;
|
||||||
|
typedef INT32 *PINT32;
|
||||||
|
typedef INT64 *PINT64;
|
||||||
|
|
||||||
|
typedef UINT8 *PUINT8;
|
||||||
|
typedef UINT16 *PUINT16;
|
||||||
|
typedef UINT32 *PUINT32;
|
||||||
|
typedef UINT64 *PUINT64;
|
||||||
|
|
||||||
|
typedef INT *PINT;
|
||||||
|
typedef UINT *PUINT;
|
||||||
|
|
||||||
|
typedef FLOAT32 *PFLOAT32;
|
||||||
|
typedef FLOAT64 *PFLOAT64;
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Const Pointer Types
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
typedef CONST VOID *PCVOID;
|
||||||
|
typedef CONST BOOL *PCBOOL;
|
||||||
|
typedef CONST CHAR *PCCHAR;
|
||||||
|
typedef CONST CHAR16 *PCCHAR16;
|
||||||
|
|
||||||
|
typedef CONST INT8 *PCINT8;
|
||||||
|
typedef CONST INT16 *PCINT16;
|
||||||
|
typedef CONST INT32 *PCINT32;
|
||||||
|
typedef CONST INT64 *PCINT64;
|
||||||
|
|
||||||
|
typedef CONST UINT8 *PCUINT8;
|
||||||
|
typedef CONST UINT16 *PCUINT16;
|
||||||
|
typedef CONST UINT32 *PCUINT32;
|
||||||
|
typedef CONST UINT64 *PCUINT64;
|
||||||
|
|
||||||
|
typedef CONST INT *PCINT;
|
||||||
|
typedef CONST UINT *PCUINT;
|
||||||
|
typedef CONST SIZE_T *PCSIZE;
|
||||||
|
typedef CONST SSIZE_T *PCSSIZE;
|
||||||
|
|
||||||
|
typedef CONST FLOAT32 *PCFLOAT32;
|
||||||
|
typedef CONST FLOAT64 *PCFLOAT64;
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// GUID Structure
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
#ifndef _WIN32
|
||||||
|
typedef struct _IA_GUID
|
||||||
|
{
|
||||||
|
UINT32 Data1;
|
||||||
|
UINT16 Data2;
|
||||||
|
UINT16 Data3;
|
||||||
|
UINT8 Data4[8];
|
||||||
|
|
||||||
|
# ifdef __cplusplus
|
||||||
|
bool operator==(const _IA_GUID &other) const
|
||||||
|
{
|
||||||
|
return __builtin_memcmp(this, &other, sizeof(_IA_GUID)) == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool operator!=(const _IA_GUID &other) const
|
||||||
|
{
|
||||||
|
return !(*this == other);
|
||||||
|
}
|
||||||
|
# endif
|
||||||
|
} GUID;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
STATIC INLINE BOOL IA_GUID_Equals(CONST GUID *a, CONST GUID *b)
|
||||||
|
{
|
||||||
|
if (a == NULLPTR || b == NULLPTR)
|
||||||
|
return FALSE;
|
||||||
|
return memcmp(a, b, sizeof(GUID)) == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Numeric Constants
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
#ifdef __cplusplus
|
||||||
|
STATIC CONSTEXPR FLOAT32 FLOAT32_EPSILON = std::numeric_limits<FLOAT32>::epsilon();
|
||||||
|
STATIC CONSTEXPR FLOAT64 FLOAT64_EPSILON = std::numeric_limits<FLOAT64>::epsilon();
|
||||||
|
#else
|
||||||
|
STATIC CONST FLOAT32 FLOAT32_EPSILON = FLT_EPSILON;
|
||||||
|
STATIC CONST FLOAT64 FLOAT64_EPSILON = DBL_EPSILON;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// C++ Containers and Helpers
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
#ifdef __cplusplus
|
||||||
|
|
||||||
|
template<typename _function_type> using Function = std::function<_function_type>;
|
||||||
|
template<typename _value_type> using InitializerList = std::initializer_list<_value_type>;
|
||||||
|
template<typename _value_type, SIZE_T count> using Array = std::array<_value_type, count>;
|
||||||
|
template<typename _value_type> using Vector = std::vector<_value_type>;
|
||||||
|
template<typename _value_type> using Optional = std::optional<_value_type>;
|
||||||
|
template<typename _key_type> using UnorderedSet = ankerl::unordered_dense::set<_key_type>;
|
||||||
|
template<typename _value_type> using Span = std::span<_value_type>;
|
||||||
|
template<typename _key_type, typename _value_type>
|
||||||
|
using UnorderedMap = ankerl::unordered_dense::map<_key_type, _value_type>;
|
||||||
|
template<typename _value_type> using Atomic = std::atomic<_value_type>;
|
||||||
|
template<typename _value_type> using SharedPtr = std::shared_ptr<_value_type>;
|
||||||
|
template<typename _value_type> using UniquePtr = std::unique_ptr<_value_type>;
|
||||||
|
template<typename _value_type> using Deque = std::deque<_value_type>;
|
||||||
|
template<typename _type_a, typename _type_b> using Pair = std::pair<_type_a, _type_b>;
|
||||||
|
template<typename... types> using Tuple = std::tuple<types...>;
|
||||||
|
template<typename _key_type, typename _value_type> using KeyValuePair = std::pair<_key_type, _value_type>;
|
||||||
|
|
||||||
|
template<typename _expected_type, typename _unexpected_type>
|
||||||
|
using Expected = tl::expected<_expected_type, _unexpected_type>;
|
||||||
|
ALIAS_FUNCTION(MakeUnexpected, tl::make_unexpected);
|
||||||
|
|
||||||
|
using String = std::string;
|
||||||
|
using StringView = std::string_view;
|
||||||
|
using StringStream = std::stringstream;
|
||||||
|
|
||||||
|
using SteadyClock = std::chrono::steady_clock;
|
||||||
|
using SteadyTimePoint = std::chrono::time_point<SteadyClock>;
|
||||||
|
using HighResClock = std::chrono::high_resolution_clock;
|
||||||
|
using HighResTimePoint = std::chrono::time_point<HighResClock>;
|
||||||
|
|
||||||
|
using Mutex = std::mutex;
|
||||||
|
using StopToken = std::stop_token;
|
||||||
|
using ScopedLock = std::scoped_lock<Mutex>;
|
||||||
|
using UniqueLock = std::unique_lock<Mutex>;
|
||||||
|
using JoiningThread = std::jthread;
|
||||||
|
using ConditionVariable = std::condition_variable;
|
||||||
|
|
||||||
|
namespace FileSystem = std::filesystem;
|
||||||
|
using FilePath = FileSystem::path;
|
||||||
|
|
||||||
|
template<typename... Args> using FormatterString = std::format_string<Args...>;
|
||||||
|
|
||||||
|
#endif
|
||||||
68
Src/IACore/inc/IACore/ProcessOps.hpp
Normal file
68
Src/IACore/inc/IACore/ProcessOps.hpp
Normal file
@ -0,0 +1,68 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
using NativeProcessID = DWORD;
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
using NativeProcessID = pid_t;
|
||||||
|
#else
|
||||||
|
# error "This platform does not support IACore ProcessOps"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
struct ProcessHandle
|
||||||
|
{
|
||||||
|
Atomic<NativeProcessID> ID{0};
|
||||||
|
Atomic<BOOL> IsRunning{false};
|
||||||
|
|
||||||
|
BOOL IsActive() CONST
|
||||||
|
{
|
||||||
|
return IsRunning && ID != 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
JoiningThread ThreadHandle;
|
||||||
|
|
||||||
|
friend class ProcessOps;
|
||||||
|
};
|
||||||
|
|
||||||
|
class ProcessOps
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
STATIC NativeProcessID GetCurrentProcessID();
|
||||||
|
|
||||||
|
STATIC Expected<INT32, String> SpawnProcessSync(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(IN StringView line)> onOutputLineCallback);
|
||||||
|
STATIC SharedPtr<ProcessHandle> SpawnProcessAsync(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(IN StringView line)> onOutputLineCallback,
|
||||||
|
IN Function<VOID(Expected<INT32, String>)> onFinishCallback);
|
||||||
|
|
||||||
|
STATIC VOID TerminateProcess(IN CONST SharedPtr<ProcessHandle> &handle);
|
||||||
|
|
||||||
|
private:
|
||||||
|
STATIC Expected<INT32, String> SpawnProcessWindows(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(StringView)> onOutputLineCallback,
|
||||||
|
OUT Atomic<NativeProcessID> &id);
|
||||||
|
STATIC Expected<INT32, String> SpawnProcessPosix(IN CONST String &command, IN CONST String &args,
|
||||||
|
IN Function<VOID(StringView)> onOutputLineCallback,
|
||||||
|
OUT Atomic<NativeProcessID> &id);
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
107
Src/IACore/inc/IACore/SocketOps.hpp
Normal file
107
Src/IACore/inc/IACore/SocketOps.hpp
Normal file
@ -0,0 +1,107 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
|
||||||
|
# include <winsock2.h>
|
||||||
|
# include <ws2tcpip.h>
|
||||||
|
# include <afunix.h>
|
||||||
|
# pragma comment(lib, "ws2_32.lib")
|
||||||
|
# define CLOSE_SOCKET(s) closesocket(s)
|
||||||
|
# define IS_VALID_SOCKET(s) (s != INVALID_SOCKET)
|
||||||
|
# define UNLINK_FILE(p) DeleteFileA(p)
|
||||||
|
using SocketHandle = SOCKET;
|
||||||
|
|
||||||
|
#elif IA_PLATFORM_UNIX
|
||||||
|
|
||||||
|
# include <sys/un.h>
|
||||||
|
# include <sys/types.h>
|
||||||
|
# include <sys/socket.h>
|
||||||
|
# include <netinet/in.h>
|
||||||
|
# define CLOSE_SOCKET(s) close(s)
|
||||||
|
# define IS_VALID_SOCKET(s) (s >= 0)
|
||||||
|
# define INVALID_SOCKET -1
|
||||||
|
# define UNLINK_FILE(p) unlink(p)
|
||||||
|
using SocketHandle = int;
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
# warning "IACore SocketOps is not supported on this platform."
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class SocketOps
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
// SocketOps correctly handles multiple calls to Initialize and Terminate. Make sure
|
||||||
|
// every Initialize call is paired with a corresponding Terminate call
|
||||||
|
STATIC VOID Initialize()
|
||||||
|
{
|
||||||
|
s_initCount++;
|
||||||
|
if (s_initCount > 1)
|
||||||
|
return;
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
WSADATA wsaData;
|
||||||
|
WSAStartup(MAKEWORD(2, 2), &wsaData);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// SocketOps correctly handles multiple calls to Initialize and Terminate. Make sure
|
||||||
|
// every Initialize call is paired with a corresponding Terminate call
|
||||||
|
STATIC VOID Terminate()
|
||||||
|
{
|
||||||
|
s_initCount--;
|
||||||
|
if (s_initCount > 0)
|
||||||
|
return;
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
WSACleanup();
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC BOOL IsPortAvailableTCP(IN UINT16 port)
|
||||||
|
{
|
||||||
|
return IsPortAvailable(port, SOCK_STREAM);
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC BOOL IsPortAvailableUDP(IN UINT16 port)
|
||||||
|
{
|
||||||
|
return IsPortAvailable(port, SOCK_DGRAM);
|
||||||
|
}
|
||||||
|
|
||||||
|
STATIC BOOL IsWouldBlock();
|
||||||
|
|
||||||
|
STATIC VOID Close(IN SocketHandle sock);
|
||||||
|
|
||||||
|
STATIC BOOL Listen(IN SocketHandle sock, IN INT32 queueSize = 5);
|
||||||
|
|
||||||
|
STATIC SocketHandle CreateUnixSocket();
|
||||||
|
|
||||||
|
STATIC BOOL BindUnixSocket(IN SocketHandle sock, IN PCCHAR path);
|
||||||
|
STATIC BOOL ConnectUnixSocket(IN SocketHandle sock, IN PCCHAR path);
|
||||||
|
|
||||||
|
private:
|
||||||
|
STATIC BOOL IsPortAvailable(IN UINT16 port, IN INT32 type);
|
||||||
|
|
||||||
|
private:
|
||||||
|
STATIC INT32 s_initCount;
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
104
Src/IACore/inc/IACore/StreamReader.hpp
Normal file
104
Src/IACore/inc/IACore/StreamReader.hpp
Normal file
@ -0,0 +1,104 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class StreamReader
|
||||||
|
{
|
||||||
|
enum class EStorageType
|
||||||
|
{
|
||||||
|
NON_OWNING,
|
||||||
|
OWNING_MMAP,
|
||||||
|
OWNING_VECTOR,
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
INLINE Expected<VOID, String> Read(IN PVOID buffer, IN SIZE_T size);
|
||||||
|
template<typename T> NO_DISCARD("Check for EOF") Expected<T, String> Read();
|
||||||
|
|
||||||
|
VOID Skip(SIZE_T amount)
|
||||||
|
{
|
||||||
|
m_cursor = std::min(m_cursor + amount, m_dataSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
VOID Seek(SIZE_T pos)
|
||||||
|
{
|
||||||
|
m_cursor = (pos > m_dataSize) ? m_dataSize : pos;
|
||||||
|
}
|
||||||
|
|
||||||
|
SIZE_T Cursor() CONST
|
||||||
|
{
|
||||||
|
return m_cursor;
|
||||||
|
}
|
||||||
|
|
||||||
|
SIZE_T Size() CONST
|
||||||
|
{
|
||||||
|
return m_dataSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
SIZE_T Remaining() CONST
|
||||||
|
{
|
||||||
|
return m_dataSize - m_cursor;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOL IsEOF() CONST
|
||||||
|
{
|
||||||
|
return m_cursor >= m_dataSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
public:
|
||||||
|
StreamReader(IN CONST FilePath &path);
|
||||||
|
explicit StreamReader(IN Vector<UINT8> &&data);
|
||||||
|
explicit StreamReader(IN Span<CONST UINT8> data);
|
||||||
|
~StreamReader();
|
||||||
|
|
||||||
|
private:
|
||||||
|
PCUINT8 m_data{};
|
||||||
|
SIZE_T m_cursor{};
|
||||||
|
SIZE_T m_dataSize{};
|
||||||
|
Vector<UINT8> m_owningVector;
|
||||||
|
CONST EStorageType m_storageType;
|
||||||
|
};
|
||||||
|
|
||||||
|
Expected<VOID, String> StreamReader::Read(IN PVOID buffer, IN SIZE_T size)
|
||||||
|
{
|
||||||
|
if B_UNLIKELY ((m_cursor + size > m_dataSize))
|
||||||
|
return MakeUnexpected(String("Unexpected EOF while reading"));
|
||||||
|
|
||||||
|
std::memcpy(buffer, &m_data[m_cursor], size);
|
||||||
|
m_cursor += size;
|
||||||
|
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T> NO_DISCARD("Check for EOF") Expected<T, String> StreamReader::Read()
|
||||||
|
{
|
||||||
|
constexpr SIZE_T size = sizeof(T);
|
||||||
|
|
||||||
|
if B_UNLIKELY ((m_cursor + size > m_dataSize))
|
||||||
|
return MakeUnexpected(String("Unexpected EOF while reading"));
|
||||||
|
|
||||||
|
T value;
|
||||||
|
std::memcpy(&value, &m_data[m_cursor], size);
|
||||||
|
m_cursor += size;
|
||||||
|
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
66
Src/IACore/inc/IACore/StreamWriter.hpp
Normal file
66
Src/IACore/inc/IACore/StreamWriter.hpp
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class StreamWriter
|
||||||
|
{
|
||||||
|
enum class EStorageType
|
||||||
|
{
|
||||||
|
NON_OWNING,
|
||||||
|
OWNING_FILE,
|
||||||
|
OWNING_VECTOR,
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
BOOL Write(IN UINT8 byte, IN SIZE_T count);
|
||||||
|
BOOL Write(IN PCVOID buffer, IN SIZE_T size);
|
||||||
|
template<typename T> BOOL Write(IN CONST T &value);
|
||||||
|
|
||||||
|
PCUINT8 Data() CONST
|
||||||
|
{
|
||||||
|
return m_buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
SIZE_T Cursor() CONST
|
||||||
|
{
|
||||||
|
return m_cursor;
|
||||||
|
}
|
||||||
|
|
||||||
|
public:
|
||||||
|
StreamWriter();
|
||||||
|
explicit StreamWriter(IN Span<UINT8> data);
|
||||||
|
explicit StreamWriter(IN CONST FilePath &path);
|
||||||
|
~StreamWriter();
|
||||||
|
|
||||||
|
private:
|
||||||
|
PUINT8 m_buffer{};
|
||||||
|
SIZE_T m_cursor{};
|
||||||
|
SIZE_T m_capacity{};
|
||||||
|
FilePath m_filePath{};
|
||||||
|
Vector<UINT8> m_owningVector;
|
||||||
|
CONST EStorageType m_storageType;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T> BOOL StreamWriter::Write(IN CONST T &value)
|
||||||
|
{
|
||||||
|
return Write(&value, sizeof(T));
|
||||||
|
}
|
||||||
|
} // namespace IACore
|
||||||
29
Src/IACore/inc/IACore/StringOps.hpp
Normal file
29
Src/IACore/inc/IACore/StringOps.hpp
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
class StringOps
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
STATIC String EncodeBase64(IN Span<CONST UINT8> data);
|
||||||
|
STATIC Vector<UINT8> DecodeBase64(IN CONST String& data);
|
||||||
|
};
|
||||||
|
}
|
||||||
170
Src/IACore/inc/IACore/Utils.hpp
Normal file
170
Src/IACore/inc/IACore/Utils.hpp
Normal file
@ -0,0 +1,170 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <IACore/PCH.hpp>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
|
||||||
|
namespace IACore
|
||||||
|
{
|
||||||
|
|
||||||
|
class Utils
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
INLINE STATIC String BinaryToHexString(std::span<const UINT8> data)
|
||||||
|
{
|
||||||
|
STATIC CONSTEXPR char LUT[] = "0123456789ABCDEF";
|
||||||
|
String res;
|
||||||
|
res.reserve(data.size() * 2);
|
||||||
|
|
||||||
|
for (UINT8 b : data)
|
||||||
|
{
|
||||||
|
res.push_back(LUT[(b >> 4) & 0x0F]);
|
||||||
|
res.push_back(LUT[b & 0x0F]);
|
||||||
|
}
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
INLINE STATIC tl::expected<Vector<UINT8>, String> HexStringToBinary(CONST StringView &hex)
|
||||||
|
{
|
||||||
|
if (hex.size() % 2 != 0)
|
||||||
|
{
|
||||||
|
return tl::make_unexpected(String("Hex string must have even length"));
|
||||||
|
}
|
||||||
|
|
||||||
|
Vector<UINT8> out;
|
||||||
|
out.reserve(hex.size() / 2);
|
||||||
|
|
||||||
|
for (SIZE_T i = 0; i < hex.size(); i += 2)
|
||||||
|
{
|
||||||
|
char high = hex[i];
|
||||||
|
char low = hex[i + 1];
|
||||||
|
|
||||||
|
// Quick helper to decode nibble
|
||||||
|
auto fromHexChar = [](char c) -> int {
|
||||||
|
if (c >= '0' && c <= '9')
|
||||||
|
return c - '0';
|
||||||
|
if (c >= 'A' && c <= 'F')
|
||||||
|
return c - 'A' + 10;
|
||||||
|
if (c >= 'a' && c <= 'f')
|
||||||
|
return c - 'a' + 10;
|
||||||
|
return -1;
|
||||||
|
};
|
||||||
|
|
||||||
|
int h = fromHexChar(high);
|
||||||
|
int l = fromHexChar(low);
|
||||||
|
|
||||||
|
if (h == -1 || l == -1)
|
||||||
|
{
|
||||||
|
return tl::make_unexpected(String("Invalid hex character found"));
|
||||||
|
}
|
||||||
|
|
||||||
|
out.push_back(CAST((h << 4) | l, UINT8));
|
||||||
|
}
|
||||||
|
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename Range> INLINE STATIC VOID Sort(Range &&range)
|
||||||
|
{
|
||||||
|
std::ranges::sort(range);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename Range, typename T> INLINE STATIC auto BinarySearchLeft(Range &&range, CONST T &value)
|
||||||
|
{
|
||||||
|
return std::ranges::lower_bound(range, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename Range, typename T> INLINE STATIC auto BinarySearchRight(Range &&range, CONST T &value)
|
||||||
|
{
|
||||||
|
return std::ranges::upper_bound(range, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Hash Combination Logic
|
||||||
|
// Uses the "golden ratio" mix (similar to Boost) to preserve entropy.
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
template<typename T> INLINE STATIC void HashCombine(UINT64 &seed, CONST T &v)
|
||||||
|
{
|
||||||
|
UINT64 h;
|
||||||
|
|
||||||
|
// 1. Compile-Time check: Can this be treated as a string view?
|
||||||
|
// This catches "Literal Strings" (char[N]), const char*, and std::string
|
||||||
|
if constexpr (std::is_constructible_v<std::string_view, T>)
|
||||||
|
{
|
||||||
|
std::string_view sv(v);
|
||||||
|
auto hasher = ankerl::unordered_dense::hash<std::string_view>();
|
||||||
|
h = hasher(sv);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// 2. Standard types (int, float, custom structs)
|
||||||
|
auto hasher = ankerl::unordered_dense::hash<T>();
|
||||||
|
h = hasher(v);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 0x9e3779b97f4a7c15 is the 64-bit golden ratio (phi) approximation
|
||||||
|
seed ^= h + 0x9e3779b97f4a7c15 + (seed << 6) + (seed >> 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Variadic Hasher
|
||||||
|
// Allows: IACore::ComputeHash(x, y, z);
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
template<typename... Args> INLINE STATIC UINT64 ComputeHash(CONST Args &...args)
|
||||||
|
{
|
||||||
|
UINT64 seed = 0;
|
||||||
|
(HashCombine(seed, args), ...); // C++17/20 Fold Expression
|
||||||
|
return seed;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Flat Hasher
|
||||||
|
// Allows: IACore::ComputeHashFlat(x, y, z);
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
template<typename T, typename... MemberPtrs> INLINE STATIC UINT64 ComputeHashFlat(CONST T &obj, MemberPtrs... members)
|
||||||
|
{
|
||||||
|
UINT64 seed = 0;
|
||||||
|
// C++17 Fold Expression: applies (seed, obj.*ptr) for every ptr in members
|
||||||
|
(HashCombine(seed, obj.*members), ...);
|
||||||
|
return seed;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
} // namespace IACore
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// MACRO: IA_MAKE_HASHABLE
|
||||||
|
//
|
||||||
|
// Injects the specialization for ankerl::unordered_dense::hash.
|
||||||
|
//
|
||||||
|
// Usage:
|
||||||
|
// struct Vector3 { float x, y, z; };
|
||||||
|
// IA_MAKE_HASHABLE(Vector3, &Vector3::x, &Vector3::y, &Vector3::z)
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
#define IA_MAKE_HASHABLE(Type, ...) \
|
||||||
|
template<> struct ankerl::unordered_dense::hash<Type> \
|
||||||
|
{ \
|
||||||
|
using is_avalanching = void; \
|
||||||
|
NO_DISCARD("Hash value should be used") \
|
||||||
|
UINT64 operator()(CONST Type &v) const NOEXCEPT \
|
||||||
|
{ \
|
||||||
|
/* Pass the object and the list of member pointers */ \
|
||||||
|
return IACore::Utils::ComputeHashFlat(v, __VA_ARGS__); \
|
||||||
|
} \
|
||||||
|
};
|
||||||
|
|
||||||
4
Tests/CMakeLists.txt
Normal file
4
Tests/CMakeLists.txt
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
|
||||||
|
add_subdirectory(Subjects/)
|
||||||
|
add_subdirectory(Unit/)
|
||||||
|
add_subdirectory(Regression/)
|
||||||
0
Tests/Regression/CMakeLists.txt
Normal file
0
Tests/Regression/CMakeLists.txt
Normal file
1
Tests/Subjects/CMakeLists.txt
Normal file
1
Tests/Subjects/CMakeLists.txt
Normal file
@ -0,0 +1 @@
|
|||||||
|
add_executable(LongProcess LongProcess/Main.cpp)
|
||||||
12
Tests/Subjects/LongProcess/Main.cpp
Normal file
12
Tests/Subjects/LongProcess/Main.cpp
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
#include <thread>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
int main(int, char **)
|
||||||
|
{
|
||||||
|
std::cout << "Started!\n";
|
||||||
|
std::cout.flush();
|
||||||
|
std::this_thread::sleep_for(std::chrono::seconds(5));
|
||||||
|
std::cout << "Ended!\n";
|
||||||
|
std::cout.flush();
|
||||||
|
return 100;
|
||||||
|
}
|
||||||
220
Tests/Unit/BinaryStreamReader.cpp
Normal file
220
Tests/Unit/BinaryStreamReader.cpp
Normal file
@ -0,0 +1,220 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/BinaryStreamReader.hpp>
|
||||||
|
|
||||||
|
#include <IACore/IATest.hpp>
|
||||||
|
|
||||||
|
using namespace IACore;
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Test Block Definition
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
IAT_BEGIN_BLOCK(Core, BinaryStreamReader)
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 1. Basic Primitive Reading (UINT8)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestReadUint8()
|
||||||
|
{
|
||||||
|
UINT8 data[] = { 0xAA, 0xBB, 0xCC };
|
||||||
|
BinaryStreamReader reader(data);
|
||||||
|
|
||||||
|
// Read First Byte
|
||||||
|
auto val1 = reader.Read<UINT8>();
|
||||||
|
IAT_CHECK(val1.has_value());
|
||||||
|
IAT_CHECK_EQ(*val1, 0xAA);
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)1);
|
||||||
|
|
||||||
|
// Read Second Byte
|
||||||
|
auto val2 = reader.Read<UINT8>();
|
||||||
|
IAT_CHECK_EQ(*val2, 0xBB);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 2. Multi-byte Reading (Endianness check)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestReadMultiByte()
|
||||||
|
{
|
||||||
|
// 0x04030201 in Little Endian memory layout
|
||||||
|
UINT8 data[] = { 0x01, 0x02, 0x03, 0x04 };
|
||||||
|
BinaryStreamReader reader(data);
|
||||||
|
|
||||||
|
auto val = reader.Read<UINT32>();
|
||||||
|
IAT_CHECK(val.has_value());
|
||||||
|
|
||||||
|
// Assuming standard x86/ARM Little Endian for this test
|
||||||
|
// If your engine supports Big Endian, you'd check architecture here
|
||||||
|
IAT_CHECK_EQ(*val, (UINT32)0x04030201);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)4);
|
||||||
|
IAT_CHECK(reader.IsEOF());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 3. Floating Point (Approx check)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestReadFloat()
|
||||||
|
{
|
||||||
|
FLOAT32 pi = 3.14159f;
|
||||||
|
// Bit-cast float to bytes for setup
|
||||||
|
UINT8 data[4];
|
||||||
|
std::memcpy(data, &pi, 4);
|
||||||
|
|
||||||
|
BinaryStreamReader reader(data);
|
||||||
|
auto val = reader.Read<FLOAT32>();
|
||||||
|
|
||||||
|
IAT_CHECK(val.has_value());
|
||||||
|
IAT_CHECK_APPROX(*val, pi);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 4. String Reading
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestReadString()
|
||||||
|
{
|
||||||
|
const char* raw = "HelloIA";
|
||||||
|
BinaryStreamReader reader(std::span<const UINT8>((const UINT8*)raw, 7));
|
||||||
|
|
||||||
|
// Read "Hello" (5 bytes)
|
||||||
|
auto str = reader.ReadString(5);
|
||||||
|
IAT_CHECK(str.has_value());
|
||||||
|
IAT_CHECK_EQ(*str, String("Hello"));
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)5);
|
||||||
|
|
||||||
|
// Read remaining "IA"
|
||||||
|
auto str2 = reader.ReadString(2);
|
||||||
|
IAT_CHECK_EQ(*str2, String("IA"));
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 5. Batch Buffer Reading
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestReadBuffer()
|
||||||
|
{
|
||||||
|
UINT8 src[] = { 1, 2, 3, 4, 5 };
|
||||||
|
UINT8 dst[3] = { 0 };
|
||||||
|
BinaryStreamReader reader(src);
|
||||||
|
|
||||||
|
// Read 3 bytes into dst
|
||||||
|
auto res = reader.Read(dst, 3);
|
||||||
|
IAT_CHECK(res.has_value());
|
||||||
|
|
||||||
|
// Verify dst content
|
||||||
|
IAT_CHECK_EQ(dst[0], 1);
|
||||||
|
IAT_CHECK_EQ(dst[1], 2);
|
||||||
|
IAT_CHECK_EQ(dst[2], 3);
|
||||||
|
|
||||||
|
// Verify cursor
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)3);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 6. Navigation (Seek, Skip, Remaining)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestNavigation()
|
||||||
|
{
|
||||||
|
UINT8 data[10] = { 0 }; // Zero init
|
||||||
|
BinaryStreamReader reader(data);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(reader.Remaining(), (SIZE_T)10);
|
||||||
|
|
||||||
|
// Skip
|
||||||
|
reader.Skip(5);
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)5);
|
||||||
|
IAT_CHECK_EQ(reader.Remaining(), (SIZE_T)5);
|
||||||
|
|
||||||
|
// Skip clamping
|
||||||
|
reader.Skip(100); // Should clamp to 10
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)10);
|
||||||
|
IAT_CHECK(reader.IsEOF());
|
||||||
|
|
||||||
|
// Seek
|
||||||
|
reader.Seek(2);
|
||||||
|
IAT_CHECK_EQ(reader.Cursor(), (SIZE_T)2);
|
||||||
|
IAT_CHECK_EQ(reader.Remaining(), (SIZE_T)8);
|
||||||
|
IAT_CHECK_NOT(reader.IsEOF());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 7. Error Handling (EOF Protection)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestBoundaryChecks()
|
||||||
|
{
|
||||||
|
UINT8 data[] = { 0x00, 0x00 };
|
||||||
|
BinaryStreamReader reader(data);
|
||||||
|
|
||||||
|
// Valid read
|
||||||
|
UNUSED(reader.Read<UINT16>());
|
||||||
|
IAT_CHECK(reader.IsEOF());
|
||||||
|
|
||||||
|
// Invalid Read Primitive
|
||||||
|
auto val = reader.Read<UINT8>();
|
||||||
|
IAT_CHECK_NOT(val.has_value()); // Should be unexpected
|
||||||
|
|
||||||
|
// Invalid Read String
|
||||||
|
auto str = reader.ReadString(1);
|
||||||
|
IAT_CHECK_NOT(str.has_value());
|
||||||
|
|
||||||
|
// Invalid Batch Read
|
||||||
|
UINT8 buf[1];
|
||||||
|
auto batch = reader.Read(buf, 1);
|
||||||
|
IAT_CHECK_NOT(batch.has_value());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Registration
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
IAT_BEGIN_TEST_LIST()
|
||||||
|
IAT_ADD_TEST(TestReadUint8);
|
||||||
|
IAT_ADD_TEST(TestReadMultiByte);
|
||||||
|
IAT_ADD_TEST(TestReadFloat);
|
||||||
|
IAT_ADD_TEST(TestReadString);
|
||||||
|
IAT_ADD_TEST(TestReadBuffer);
|
||||||
|
IAT_ADD_TEST(TestNavigation);
|
||||||
|
IAT_ADD_TEST(TestBoundaryChecks);
|
||||||
|
IAT_END_TEST_LIST()
|
||||||
|
|
||||||
|
IAT_END_BLOCK()
|
||||||
|
|
||||||
|
int main(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
UNUSED(argc);
|
||||||
|
UNUSED(argv);
|
||||||
|
|
||||||
|
// Define runner (StopOnFail=false, Verbose=true)
|
||||||
|
ia::iatest::runner<false, true> testRunner;
|
||||||
|
|
||||||
|
// Run the BinaryStreamReader block
|
||||||
|
testRunner.testBlock<Core_BinaryReader>();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
222
Tests/Unit/BinaryStreamWriter.cpp
Normal file
222
Tests/Unit/BinaryStreamWriter.cpp
Normal file
@ -0,0 +1,222 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/BinaryStreamWriter.hpp>
|
||||||
|
|
||||||
|
#include <IACore/IATest.hpp>
|
||||||
|
|
||||||
|
using namespace IACore;
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Test Block Definition
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
IAT_BEGIN_BLOCK(Core, BinaryStreamWriter)
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 1. Vector Mode (Dynamic Growth)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestVectorGrowth()
|
||||||
|
{
|
||||||
|
std::vector<UINT8> buffer;
|
||||||
|
// Start empty
|
||||||
|
BinaryStreamWriter writer(buffer);
|
||||||
|
|
||||||
|
// Write 1 Byte
|
||||||
|
writer.Write<UINT8>(0xAA);
|
||||||
|
IAT_CHECK_EQ(buffer.size(), (SIZE_T)1);
|
||||||
|
IAT_CHECK_EQ(buffer[0], 0xAA);
|
||||||
|
|
||||||
|
// Write 4 Bytes (UINT32)
|
||||||
|
// 0xDDCCBBAA -> Little Endian: AA BB CC DD
|
||||||
|
writer.Write<UINT32>(0xDDCCBBAA);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(buffer.size(), (SIZE_T)5);
|
||||||
|
|
||||||
|
// Verify Memory Layout
|
||||||
|
IAT_CHECK_EQ(buffer[1], 0xAA);
|
||||||
|
IAT_CHECK_EQ(buffer[2], 0xBB);
|
||||||
|
IAT_CHECK_EQ(buffer[3], 0xCC);
|
||||||
|
IAT_CHECK_EQ(buffer[4], 0xDD);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 2. Vector Mode (Appending to Existing Data)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestVectorAppend()
|
||||||
|
{
|
||||||
|
// Vector starts with existing data
|
||||||
|
std::vector<UINT8> buffer = { 0x01, 0x02 };
|
||||||
|
BinaryStreamWriter writer(buffer);
|
||||||
|
|
||||||
|
// Should append to end, not overwrite 0x01
|
||||||
|
writer.Write<UINT8>(0x03);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(buffer.size(), (SIZE_T)3);
|
||||||
|
IAT_CHECK_EQ(buffer[0], 0x01);
|
||||||
|
IAT_CHECK_EQ(buffer[2], 0x03);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 3. Fixed Mode (Span / No Allocation)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestFixedBuffer()
|
||||||
|
{
|
||||||
|
UINT8 rawData[10];
|
||||||
|
// Initialize with zeros
|
||||||
|
std::memset(rawData, 0, 10);
|
||||||
|
|
||||||
|
BinaryStreamWriter writer(rawData); // Implicit conversion to span
|
||||||
|
|
||||||
|
// Write UINT8
|
||||||
|
writer.Write<UINT8>(0xFF);
|
||||||
|
IAT_CHECK_EQ(rawData[0], 0xFF);
|
||||||
|
|
||||||
|
// Write UINT16
|
||||||
|
writer.Write<UINT16>(0x2211);
|
||||||
|
IAT_CHECK_EQ(rawData[1], 0x11);
|
||||||
|
IAT_CHECK_EQ(rawData[2], 0x22);
|
||||||
|
|
||||||
|
// Verify we haven't touched the rest
|
||||||
|
IAT_CHECK_EQ(rawData[3], 0x00);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 4. WriteBytes (Bulk Write)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestWriteBytes()
|
||||||
|
{
|
||||||
|
std::vector<UINT8> buffer;
|
||||||
|
BinaryStreamWriter writer(buffer);
|
||||||
|
|
||||||
|
const char* msg = "IA";
|
||||||
|
writer.WriteBytes((PVOID)msg, 2);
|
||||||
|
writer.Write<UINT8>(0x00); // Null term
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(buffer.size(), (SIZE_T)3);
|
||||||
|
IAT_CHECK_EQ(buffer[0], 'I');
|
||||||
|
IAT_CHECK_EQ(buffer[1], 'A');
|
||||||
|
IAT_CHECK_EQ(buffer[2], 0x00);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 5. Random Access (WriteAt)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestRandomAccess()
|
||||||
|
{
|
||||||
|
std::vector<UINT8> buffer;
|
||||||
|
BinaryStreamWriter writer(buffer);
|
||||||
|
|
||||||
|
// Fill with placeholders
|
||||||
|
writer.Write<UINT32>(0xFFFFFFFF);
|
||||||
|
writer.Write<UINT32>(0xFFFFFFFF);
|
||||||
|
|
||||||
|
// Overwrite the first integer with 0x00000000
|
||||||
|
writer.WriteAt<UINT32>(0, 0);
|
||||||
|
|
||||||
|
// Overwrite the byte at index 4 (start of second int)
|
||||||
|
writer.WriteAt<UINT8>(4, 0xAA);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(buffer[0], 0x00);
|
||||||
|
IAT_CHECK_EQ(buffer[3], 0x00);
|
||||||
|
IAT_CHECK_EQ(buffer[4], 0xAA); // Modified
|
||||||
|
IAT_CHECK_EQ(buffer[5], 0xFF); // Unmodified
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 6. Floating Point Writing
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestWriteFloat()
|
||||||
|
{
|
||||||
|
std::vector<UINT8> buffer;
|
||||||
|
BinaryStreamWriter writer(buffer);
|
||||||
|
|
||||||
|
FLOAT32 val = 1.234f;
|
||||||
|
writer.Write<FLOAT32>(val);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(buffer.size(), (SIZE_T)4);
|
||||||
|
|
||||||
|
// Read it back via memcpy to verify
|
||||||
|
FLOAT32 readBack;
|
||||||
|
std::memcpy(&readBack, buffer.data(), 4);
|
||||||
|
|
||||||
|
IAT_CHECK_APPROX(readBack, val);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 7. GetPtrAt (Pointer Arithmetic Check)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestGetPtr()
|
||||||
|
{
|
||||||
|
std::vector<UINT8> buffer = { 0x10, 0x20, 0x30 };
|
||||||
|
BinaryStreamWriter writer(buffer);
|
||||||
|
|
||||||
|
PUINT8 p1 = writer.GetPtrAt(1);
|
||||||
|
IAT_CHECK(p1 != nullptr);
|
||||||
|
IAT_CHECK_EQ(*p1, 0x20);
|
||||||
|
|
||||||
|
// Bounds check (Valid boundary)
|
||||||
|
PUINT8 pLast = writer.GetPtrAt(2);
|
||||||
|
IAT_CHECK(pLast != nullptr);
|
||||||
|
|
||||||
|
// Bounds check (Invalid)
|
||||||
|
// Note: We don't test Panic here, but we check if it returns valid ptrs
|
||||||
|
PUINT8 pInvalid = writer.GetPtrAt(3);
|
||||||
|
IAT_CHECK(pInvalid == nullptr);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Registration
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
IAT_BEGIN_TEST_LIST()
|
||||||
|
IAT_ADD_TEST(TestVectorGrowth);
|
||||||
|
IAT_ADD_TEST(TestVectorAppend);
|
||||||
|
IAT_ADD_TEST(TestFixedBuffer);
|
||||||
|
IAT_ADD_TEST(TestWriteBytes);
|
||||||
|
IAT_ADD_TEST(TestRandomAccess);
|
||||||
|
IAT_ADD_TEST(TestWriteFloat);
|
||||||
|
IAT_ADD_TEST(TestGetPtr);
|
||||||
|
IAT_END_TEST_LIST()
|
||||||
|
|
||||||
|
IAT_END_BLOCK()
|
||||||
|
|
||||||
|
int main(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
UNUSED(argc);
|
||||||
|
UNUSED(argv);
|
||||||
|
|
||||||
|
// Define runner (StopOnFail=false, Verbose=true)
|
||||||
|
ia::iatest::runner<false, true> testRunner;
|
||||||
|
|
||||||
|
// Run the BinaryReader block
|
||||||
|
testRunner.testBlock<Core_BinaryStreamWriter>();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
39
Tests/Unit/CCompile.c
Normal file
39
Tests/Unit/CCompile.c
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#if defined(__cplusplus)
|
||||||
|
#error "CRITICAL: This file MUST be compiled as C to test ABI compatibility."
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <IACore/IACore.hpp>
|
||||||
|
|
||||||
|
#if TRUE != 1
|
||||||
|
#error "TRUE macro is broken in C mode"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int main(void) {
|
||||||
|
IA_VERSION_TYPE version = IA_MAKE_VERSION(1, 0, 0);
|
||||||
|
|
||||||
|
IA_ASSERT(version > 0);
|
||||||
|
|
||||||
|
UNUSED(version);
|
||||||
|
|
||||||
|
int32_t myNumber = 10;
|
||||||
|
|
||||||
|
(void)myNumber;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
62
Tests/Unit/CMakeLists.txt
Normal file
62
Tests/Unit/CMakeLists.txt
Normal file
@ -0,0 +1,62 @@
|
|||||||
|
set(TEST_NAME_PREFIX "IACore_Test_Unit_")
|
||||||
|
|
||||||
|
# ------------------------------------------------
|
||||||
|
# C Compile Test
|
||||||
|
# ------------------------------------------------
|
||||||
|
add_executable(${TEST_NAME_PREFIX}CCompile "CCompile.c")
|
||||||
|
|
||||||
|
set_target_properties(${TEST_NAME_PREFIX}CCompile PROPERTIES
|
||||||
|
C_STANDARD 99
|
||||||
|
C_STANDARD_REQUIRED ON
|
||||||
|
LINKER_LANGUAGE C
|
||||||
|
)
|
||||||
|
|
||||||
|
target_link_libraries(${TEST_NAME_PREFIX}CCompile PRIVATE IACore)
|
||||||
|
|
||||||
|
## ------------------------------------------------
|
||||||
|
## Unit: BinaryReader
|
||||||
|
## ------------------------------------------------
|
||||||
|
#add_executable(${TEST_NAME_PREFIX}BinaryReader "BinaryReader.cpp")
|
||||||
|
#target_link_libraries(${TEST_NAME_PREFIX}BinaryReader PRIVATE IACore)
|
||||||
|
#target_compile_options(${TEST_NAME_PREFIX}BinaryReader PRIVATE -fexceptions)
|
||||||
|
#set_target_properties(${TEST_NAME_PREFIX}BinaryReader PROPERTIES USE_EXCEPTIONS ON)
|
||||||
|
#
|
||||||
|
## ------------------------------------------------
|
||||||
|
## Unit: BinaryWriter
|
||||||
|
## ------------------------------------------------
|
||||||
|
#add_executable(${TEST_NAME_PREFIX}BinaryWriter "BinaryWriter.cpp")
|
||||||
|
#target_link_libraries(${TEST_NAME_PREFIX}BinaryWriter PRIVATE IACore)
|
||||||
|
#target_compile_options(${TEST_NAME_PREFIX}BinaryWriter PRIVATE -fexceptions)
|
||||||
|
#set_target_properties(${TEST_NAME_PREFIX}BinaryWriter PROPERTIES USE_EXCEPTIONS ON)
|
||||||
|
|
||||||
|
# ------------------------------------------------
|
||||||
|
# Unit: Environment
|
||||||
|
# ------------------------------------------------
|
||||||
|
add_executable(${TEST_NAME_PREFIX}Environment "Environment.cpp")
|
||||||
|
target_link_libraries(${TEST_NAME_PREFIX}Environment PRIVATE IACore)
|
||||||
|
target_compile_options(${TEST_NAME_PREFIX}Environment PRIVATE -fexceptions)
|
||||||
|
set_target_properties(${TEST_NAME_PREFIX}Environment PROPERTIES USE_EXCEPTIONS ON)
|
||||||
|
|
||||||
|
# ------------------------------------------------
|
||||||
|
# Unit: FileOps
|
||||||
|
# ------------------------------------------------
|
||||||
|
#add_executable(${TEST_NAME_PREFIX}FileOps "FileOps.cpp")
|
||||||
|
#target_link_libraries(${TEST_NAME_PREFIX}FileOps PRIVATE IACore)
|
||||||
|
#target_compile_options(${TEST_NAME_PREFIX}FileOps PRIVATE -fexceptions)
|
||||||
|
#set_target_properties(${TEST_NAME_PREFIX}FileOps PROPERTIES USE_EXCEPTIONS ON)
|
||||||
|
|
||||||
|
# ------------------------------------------------
|
||||||
|
# Unit: ProcessOps
|
||||||
|
# ------------------------------------------------
|
||||||
|
add_executable(${TEST_NAME_PREFIX}ProcessOps "ProcessOps.cpp")
|
||||||
|
target_link_libraries(${TEST_NAME_PREFIX}ProcessOps PRIVATE IACore)
|
||||||
|
target_compile_options(${TEST_NAME_PREFIX}ProcessOps PRIVATE -fexceptions)
|
||||||
|
set_target_properties(${TEST_NAME_PREFIX}ProcessOps PROPERTIES USE_EXCEPTIONS ON)
|
||||||
|
|
||||||
|
# ------------------------------------------------
|
||||||
|
# Unit: Utils
|
||||||
|
# ------------------------------------------------
|
||||||
|
add_executable(${TEST_NAME_PREFIX}Utils "Utils.cpp")
|
||||||
|
target_link_libraries(${TEST_NAME_PREFIX}Utils PRIVATE IACore)
|
||||||
|
target_compile_options(${TEST_NAME_PREFIX}Utils PRIVATE -fexceptions)
|
||||||
|
set_target_properties(${TEST_NAME_PREFIX}Utils PROPERTIES USE_EXCEPTIONS ON)
|
||||||
190
Tests/Unit/Environment.cpp
Normal file
190
Tests/Unit/Environment.cpp
Normal file
@ -0,0 +1,190 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/Environment.hpp>
|
||||||
|
|
||||||
|
#include <IACore/IATest.hpp>
|
||||||
|
|
||||||
|
using namespace IACore;
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Constants
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// We use a unique prefix to ensure we don't accidentally mess with real
|
||||||
|
// system variables like PATH or HOME.
|
||||||
|
static const char* TEST_KEY = "IA_TEST_ENV_VAR_12345";
|
||||||
|
static const char* TEST_VAL = "Hello World";
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Test Block Definition
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
IAT_BEGIN_BLOCK(Core, Environment)
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 1. Basic Set and Get (The Happy Path)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestBasicCycle()
|
||||||
|
{
|
||||||
|
// 1. Ensure clean slate
|
||||||
|
Environment::Unset(TEST_KEY);
|
||||||
|
IAT_CHECK_NOT(Environment::Exists(TEST_KEY));
|
||||||
|
|
||||||
|
// 2. Set
|
||||||
|
BOOL setRes = Environment::Set(TEST_KEY, TEST_VAL);
|
||||||
|
IAT_CHECK(setRes);
|
||||||
|
IAT_CHECK(Environment::Exists(TEST_KEY));
|
||||||
|
|
||||||
|
// 3. Find (Optional)
|
||||||
|
auto opt = Environment::Find(TEST_KEY);
|
||||||
|
IAT_CHECK(opt.has_value());
|
||||||
|
IAT_CHECK_EQ(*opt, String(TEST_VAL));
|
||||||
|
|
||||||
|
// 4. Get (Direct)
|
||||||
|
String val = Environment::Get(TEST_KEY);
|
||||||
|
IAT_CHECK_EQ(val, String(TEST_VAL));
|
||||||
|
|
||||||
|
// Cleanup
|
||||||
|
Environment::Unset(TEST_KEY);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 2. Overwriting Values
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestOverwrite()
|
||||||
|
{
|
||||||
|
Environment::Set(TEST_KEY, "ValueA");
|
||||||
|
IAT_CHECK_EQ(Environment::Get(TEST_KEY), String("ValueA"));
|
||||||
|
|
||||||
|
// Overwrite
|
||||||
|
Environment::Set(TEST_KEY, "ValueB");
|
||||||
|
IAT_CHECK_EQ(Environment::Get(TEST_KEY), String("ValueB"));
|
||||||
|
|
||||||
|
Environment::Unset(TEST_KEY);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 3. Unset Logic
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestUnset()
|
||||||
|
{
|
||||||
|
Environment::Set(TEST_KEY, "To Be Deleted");
|
||||||
|
IAT_CHECK(Environment::Exists(TEST_KEY));
|
||||||
|
|
||||||
|
BOOL unsetRes = Environment::Unset(TEST_KEY);
|
||||||
|
IAT_CHECK(unsetRes);
|
||||||
|
|
||||||
|
// Verify it is actually gone
|
||||||
|
IAT_CHECK_NOT(Environment::Exists(TEST_KEY));
|
||||||
|
|
||||||
|
// Find should return nullopt
|
||||||
|
auto opt = Environment::Find(TEST_KEY);
|
||||||
|
IAT_CHECK_NOT(opt.has_value());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 4. Default Value Fallbacks
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestDefaults()
|
||||||
|
{
|
||||||
|
const char* ghostKey = "IA_THIS_KEY_DOES_NOT_EXIST";
|
||||||
|
|
||||||
|
// Ensure it really doesn't exist
|
||||||
|
Environment::Unset(ghostKey);
|
||||||
|
|
||||||
|
// Test Get with implicit default ("")
|
||||||
|
String empty = Environment::Get(ghostKey);
|
||||||
|
IAT_CHECK(empty.empty());
|
||||||
|
|
||||||
|
// Test Get with explicit default
|
||||||
|
String fallback = Environment::Get(ghostKey, "MyDefault");
|
||||||
|
IAT_CHECK_EQ(fallback, String("MyDefault"));
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 5. Empty Strings vs Null/Unset
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// This is a critical edge case.
|
||||||
|
// Does Set(Key, "") create an existing empty variable, or unset it?
|
||||||
|
// Standard POSIX/Windows API behavior is that it EXISTS, but is empty.
|
||||||
|
BOOL TestEmptyValue()
|
||||||
|
{
|
||||||
|
Environment::Set(TEST_KEY, "");
|
||||||
|
|
||||||
|
// It should exist
|
||||||
|
IAT_CHECK(Environment::Exists(TEST_KEY));
|
||||||
|
|
||||||
|
// It should be retrievable as empty string
|
||||||
|
auto opt = Environment::Find(TEST_KEY);
|
||||||
|
IAT_CHECK(opt.has_value());
|
||||||
|
IAT_CHECK(opt->empty());
|
||||||
|
|
||||||
|
// Cleanup
|
||||||
|
Environment::Unset(TEST_KEY);
|
||||||
|
IAT_CHECK_NOT(Environment::Exists(TEST_KEY));
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 6. Validation / Bad Input
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestBadInput()
|
||||||
|
{
|
||||||
|
// Setting an empty key should fail gracefully
|
||||||
|
BOOL res = Environment::Set("", "Value");
|
||||||
|
IAT_CHECK_NOT(res);
|
||||||
|
|
||||||
|
// Unsetting an empty key should fail
|
||||||
|
BOOL resUnset = Environment::Unset("");
|
||||||
|
IAT_CHECK_NOT(resUnset);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Registration
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
IAT_BEGIN_TEST_LIST()
|
||||||
|
IAT_ADD_TEST(TestBasicCycle);
|
||||||
|
IAT_ADD_TEST(TestOverwrite);
|
||||||
|
IAT_ADD_TEST(TestUnset);
|
||||||
|
IAT_ADD_TEST(TestDefaults);
|
||||||
|
IAT_ADD_TEST(TestEmptyValue);
|
||||||
|
IAT_ADD_TEST(TestBadInput);
|
||||||
|
IAT_END_TEST_LIST()
|
||||||
|
|
||||||
|
IAT_END_BLOCK()
|
||||||
|
|
||||||
|
int main(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
UNUSED(argc);
|
||||||
|
UNUSED(argv);
|
||||||
|
|
||||||
|
// Define runner (StopOnFail=false, Verbose=true)
|
||||||
|
ia::iatest::runner<false, true> testRunner;
|
||||||
|
|
||||||
|
// Run the BinaryReader block
|
||||||
|
testRunner.testBlock<Core_Environment>();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
246
Tests/Unit/FileOps.cpp
Normal file
246
Tests/Unit/FileOps.cpp
Normal file
@ -0,0 +1,246 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/FileOps.hpp>
|
||||||
|
|
||||||
|
#include <IACore/IATest.hpp>
|
||||||
|
|
||||||
|
using namespace IACore;
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Test Block Definition
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
IAT_BEGIN_BLOCK(Core, File)
|
||||||
|
|
||||||
|
// Helper to generate a temp path for testing
|
||||||
|
String GetTempPath(const String& filename) {
|
||||||
|
auto path = std::filesystem::temp_directory_path() / filename;
|
||||||
|
return path.string();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper to write raw data for setup (bypassing the class we are testing)
|
||||||
|
void CreateDummyFile(const String& path, const String& content) {
|
||||||
|
std::ofstream out(path);
|
||||||
|
out << content;
|
||||||
|
out.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 1. Path Utilities (Pure Logic)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestPathUtils()
|
||||||
|
{
|
||||||
|
// We use forward slashes as std::filesystem handles them cross-platform
|
||||||
|
String complexPath = "assets/textures/hero_diffuse.png";
|
||||||
|
|
||||||
|
// Test ExtractDirectory
|
||||||
|
String dir = File::ExtractDirectory(complexPath);
|
||||||
|
// Note: On Windows this might return "assets\textures", check specific impl if strict
|
||||||
|
// But std::filesystem usually normalizes based on the input string format.
|
||||||
|
IAT_CHECK(dir.find("textures") != String::npos);
|
||||||
|
|
||||||
|
// Test ExtractFilename (With Extension)
|
||||||
|
String nameWithExt = File::ExtractFilename<true>(complexPath);
|
||||||
|
IAT_CHECK_EQ(nameWithExt, String("hero_diffuse.png"));
|
||||||
|
|
||||||
|
// Test ExtractFilename (No Extension / Stem)
|
||||||
|
String nameNoExt = File::ExtractFilename<false>(complexPath);
|
||||||
|
IAT_CHECK_EQ(nameNoExt, String("hero_diffuse"));
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 2. Static Read String
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestStaticReadString()
|
||||||
|
{
|
||||||
|
String path = GetTempPath("ia_test_text.txt");
|
||||||
|
String content = "Hello IA Engine";
|
||||||
|
|
||||||
|
// Arrange
|
||||||
|
CreateDummyFile(path, content);
|
||||||
|
|
||||||
|
// Act
|
||||||
|
auto result = File::ReadToString(path);
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
IAT_CHECK(result.has_value());
|
||||||
|
IAT_CHECK_EQ(*result, content);
|
||||||
|
|
||||||
|
// Cleanup
|
||||||
|
std::filesystem::remove(path);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 3. Static Read Binary (Vector)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestStaticReadVector()
|
||||||
|
{
|
||||||
|
String path = GetTempPath("ia_test_bin.dat");
|
||||||
|
|
||||||
|
// Arrange: Create a binary file manually
|
||||||
|
std::ofstream out(path, std::ios::binary);
|
||||||
|
UINT8 rawBytes[] = { 0xDE, 0xAD, 0xBE, 0xEF };
|
||||||
|
out.write((char*)rawBytes, 4);
|
||||||
|
out.close();
|
||||||
|
|
||||||
|
// Act
|
||||||
|
auto result = File::ReadToVector(path);
|
||||||
|
|
||||||
|
// Assert
|
||||||
|
IAT_CHECK(result.has_value());
|
||||||
|
Vector<UINT8>& vec = *result;
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(vec.size(), (SIZE_T)4);
|
||||||
|
IAT_CHECK_EQ(vec[0], 0xDE);
|
||||||
|
IAT_CHECK_EQ(vec[3], 0xEF);
|
||||||
|
|
||||||
|
// Cleanup
|
||||||
|
std::filesystem::remove(path);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 4. Instance Write & Read Loop
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestInstanceWriteRead()
|
||||||
|
{
|
||||||
|
String path = GetTempPath("ia_instance_io.bin");
|
||||||
|
UINT32 flagsWrite = (UINT32)File::EOpenFlags::Write |
|
||||||
|
(UINT32)File::EOpenFlags::Binary |
|
||||||
|
(UINT32)File::EOpenFlags::Trunc;
|
||||||
|
|
||||||
|
// 1. Write
|
||||||
|
{
|
||||||
|
File f;
|
||||||
|
auto res = f.Open(path, (File::EOpenFlags)flagsWrite);
|
||||||
|
IAT_CHECK(res.has_value());
|
||||||
|
IAT_CHECK(f.IsOpen());
|
||||||
|
|
||||||
|
UINT32 magic = 12345;
|
||||||
|
f.Write(&magic, sizeof(magic));
|
||||||
|
|
||||||
|
f.Close();
|
||||||
|
IAT_CHECK_NOT(f.IsOpen());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Read Back
|
||||||
|
{
|
||||||
|
UINT32 flagsRead = (UINT32)File::EOpenFlags::Read |
|
||||||
|
(UINT32)File::EOpenFlags::Binary;
|
||||||
|
|
||||||
|
File f(path, (File::EOpenFlags)flagsRead); // Test RAII constructor
|
||||||
|
IAT_CHECK(f.IsOpen());
|
||||||
|
|
||||||
|
UINT32 magicRead = 0;
|
||||||
|
SIZE_T bytesRead = f.Read(&magicRead, sizeof(magicRead));
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(bytesRead, sizeof(UINT32));
|
||||||
|
IAT_CHECK_EQ(magicRead, (UINT32)12345);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::filesystem::remove(path);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 5. Append Mode
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestAppendMode()
|
||||||
|
{
|
||||||
|
String path = GetTempPath("ia_append.txt");
|
||||||
|
UINT32 flagsAppend = (UINT32)File::EOpenFlags::Write | (UINT32)File::EOpenFlags::Append;
|
||||||
|
|
||||||
|
// Create initial file
|
||||||
|
CreateDummyFile(path, "A");
|
||||||
|
|
||||||
|
// Open in append mode
|
||||||
|
File f;
|
||||||
|
const auto openResult = f.Open(path, (File::EOpenFlags)flagsAppend);
|
||||||
|
if(!openResult)
|
||||||
|
{
|
||||||
|
IA_PANIC(openResult.error().c_str())
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
char c = 'B';
|
||||||
|
f.Write(&c, 1);
|
||||||
|
f.Close();
|
||||||
|
|
||||||
|
// Verify content is "AB"
|
||||||
|
auto content = File::ReadToString(path);
|
||||||
|
IAT_CHECK(content.has_value());
|
||||||
|
IAT_CHECK_EQ(*content, String("AB"));
|
||||||
|
|
||||||
|
std::filesystem::remove(path);
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 6. Error Handling
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestNonExistentFile()
|
||||||
|
{
|
||||||
|
String ghostPath = GetTempPath("this_does_not_exist.ghost");
|
||||||
|
|
||||||
|
// Ensure it really doesn't exist
|
||||||
|
if(File::Exists(ghostPath)) std::filesystem::remove(ghostPath);
|
||||||
|
|
||||||
|
// Test Static
|
||||||
|
auto resStatic = File::ReadToString(ghostPath);
|
||||||
|
IAT_CHECK_NOT(resStatic.has_value());
|
||||||
|
// Optional: Check error message content
|
||||||
|
// IAT_CHECK(resStatic.error().find("not found") != String::npos);
|
||||||
|
|
||||||
|
// Test Instance
|
||||||
|
File f;
|
||||||
|
auto resInstance = f.Open(ghostPath, File::EOpenFlags::Read);
|
||||||
|
IAT_CHECK_NOT(resInstance.has_value());
|
||||||
|
IAT_CHECK_NOT(f.IsOpen());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Registration
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
IAT_BEGIN_TEST_LIST()
|
||||||
|
IAT_ADD_TEST(TestPathUtils);
|
||||||
|
IAT_ADD_TEST(TestStaticReadString);
|
||||||
|
IAT_ADD_TEST(TestStaticReadVector);
|
||||||
|
IAT_ADD_TEST(TestInstanceWriteRead);
|
||||||
|
IAT_ADD_TEST(TestAppendMode);
|
||||||
|
IAT_ADD_TEST(TestNonExistentFile);
|
||||||
|
IAT_END_TEST_LIST()
|
||||||
|
|
||||||
|
IAT_END_BLOCK()
|
||||||
|
|
||||||
|
|
||||||
|
int main(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
UNUSED(argc);
|
||||||
|
UNUSED(argv);
|
||||||
|
|
||||||
|
// Define runner (StopOnFail=false, Verbose=true)
|
||||||
|
ia::iatest::runner<false, true> testRunner;
|
||||||
|
|
||||||
|
// Run the BinaryReader block
|
||||||
|
testRunner.testBlock<Core_File>();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
247
Tests/Unit/ProcessOps.cpp
Normal file
247
Tests/Unit/ProcessOps.cpp
Normal file
@ -0,0 +1,247 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/ProcessOps.hpp>
|
||||||
|
|
||||||
|
#include <IACore/IATest.hpp>
|
||||||
|
|
||||||
|
using namespace IACore;
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Platform Abstraction for Test Commands
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
#define CMD_ECHO_EXE "cmd.exe"
|
||||||
|
// Windows requires /c to run a command string
|
||||||
|
#define CMD_ARG_PREFIX "/c"
|
||||||
|
#define NULL_DEVICE "NUL"
|
||||||
|
#else
|
||||||
|
#define CMD_ECHO_EXE "/bin/echo"
|
||||||
|
#define CMD_ARG_PREFIX ""
|
||||||
|
#define NULL_DEVICE "/dev/null"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
IAT_BEGIN_BLOCK(Core, Process)
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 1. Basic Execution (Exit Code 0)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestBasicRun()
|
||||||
|
{
|
||||||
|
// Simple "echo hello"
|
||||||
|
String captured;
|
||||||
|
|
||||||
|
auto result = ProcessOps::SpawnProcessSync(CMD_ECHO_EXE, CMD_ARG_PREFIX " HelloIA",
|
||||||
|
[&](StringView line) {
|
||||||
|
captured = line;
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
IAT_CHECK(result.has_value());
|
||||||
|
IAT_CHECK_EQ(*result, 0); // Exit code 0
|
||||||
|
|
||||||
|
// Note: Echo might add newline, but your LineBuffer trims/handles it.
|
||||||
|
// We check if "HelloIA" is contained or equal.
|
||||||
|
IAT_CHECK(captured.find("HelloIA") != String::npos);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 2. Argument Parsing
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestArguments()
|
||||||
|
{
|
||||||
|
Vector<String> lines;
|
||||||
|
|
||||||
|
// Echo two distinct words.
|
||||||
|
// Windows: cmd.exe /c echo one two
|
||||||
|
// Linux: /bin/echo one two
|
||||||
|
String args = String(CMD_ARG_PREFIX) + " one two";
|
||||||
|
if(args[0] == ' ') args.erase(0, 1); // cleanup space if prefix empty
|
||||||
|
|
||||||
|
auto result = ProcessOps::SpawnProcessSync(CMD_ECHO_EXE, args,
|
||||||
|
[&](StringView line) {
|
||||||
|
lines.push_back(String(line));
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(*result, 0);
|
||||||
|
IAT_CHECK(lines.size() > 0);
|
||||||
|
|
||||||
|
// Output should contain "one two"
|
||||||
|
IAT_CHECK(lines[0].find("one two") != String::npos);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 3. Error / Non-Zero Exit Codes
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestExitCodes()
|
||||||
|
{
|
||||||
|
// We need a command that returns non-zero.
|
||||||
|
// Windows: cmd /c exit 1
|
||||||
|
// Linux: /bin/sh -c "exit 1"
|
||||||
|
|
||||||
|
String cmd, arg;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
cmd = "cmd.exe";
|
||||||
|
arg = "/c exit 42";
|
||||||
|
#else
|
||||||
|
cmd = "/bin/sh";
|
||||||
|
arg = "-c \"exit 42\""; // quotes needed for sh -c
|
||||||
|
#endif
|
||||||
|
|
||||||
|
auto result = ProcessOps::SpawnProcessSync(cmd, arg, [](StringView){});
|
||||||
|
|
||||||
|
IAT_CHECK(result.has_value());
|
||||||
|
IAT_CHECK_EQ(*result, 42);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 4. Missing Executable Handling
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestMissingExe()
|
||||||
|
{
|
||||||
|
// Try to run a random string
|
||||||
|
auto result = ProcessOps::SpawnProcessSync("sdflkjghsdflkjg", "", [](StringView){});
|
||||||
|
|
||||||
|
// Windows: CreateProcess usually fails -> returns unexpected
|
||||||
|
// Linux: execvp fails inside child, returns 127 via waitpid
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
IAT_CHECK_NOT(result.has_value()); // Should be an error string
|
||||||
|
#else
|
||||||
|
// Linux fork succeeds, but execvp fails, returning 127
|
||||||
|
IAT_CHECK(result.has_value());
|
||||||
|
IAT_CHECK_EQ(*result, 127);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 5. Line Buffer Logic (The 4096 split test)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestLargeOutput()
|
||||||
|
{
|
||||||
|
// We need to generate output larger than the internal 4096 buffer
|
||||||
|
// to ensure the "partial line" logic works when a line crosses a buffer boundary.
|
||||||
|
|
||||||
|
// We will construct a python script or shell command to print a massive line.
|
||||||
|
// Cross platform approach: Use Python if available, or just a long echo.
|
||||||
|
// Let's assume 'python3' or 'python' is in path, otherwise skip?
|
||||||
|
// Safer: Use pure shell loop if possible, or just a massive command line arg.
|
||||||
|
|
||||||
|
String massiveString;
|
||||||
|
massiveString.reserve(5000);
|
||||||
|
for(int i=0; i<500; ++i) massiveString += "1234567890"; // 5000 chars
|
||||||
|
|
||||||
|
String cmd, arg;
|
||||||
|
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
cmd = "cmd.exe";
|
||||||
|
// Windows has command line length limits (~8k), 5k is safe.
|
||||||
|
arg = "/c echo " + massiveString;
|
||||||
|
#else
|
||||||
|
cmd = "/bin/echo";
|
||||||
|
arg = massiveString;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
String captured;
|
||||||
|
auto result = ProcessOps::SpawnProcessSync(cmd, arg,
|
||||||
|
[&](StringView line) {
|
||||||
|
captured += line;
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
IAT_CHECK(result.has_value());
|
||||||
|
IAT_CHECK_EQ(*result, 0);
|
||||||
|
|
||||||
|
// If the LineBuffer failed to stitch chunks, the length wouldn't match
|
||||||
|
// or we would get multiple callbacks if we expected 1 line.
|
||||||
|
IAT_CHECK_EQ(captured.length(), massiveString.length());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 6. Multi-Line Handling
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestMultiLine()
|
||||||
|
{
|
||||||
|
// Windows: cmd /c "echo A && echo B"
|
||||||
|
// Linux: /bin/sh -c "echo A; echo B"
|
||||||
|
|
||||||
|
String cmd, arg;
|
||||||
|
#if IA_PLATFORM_WINDOWS
|
||||||
|
cmd = "cmd.exe";
|
||||||
|
arg = "/c \"echo LineA && echo LineB\"";
|
||||||
|
#else
|
||||||
|
cmd = "/bin/sh";
|
||||||
|
arg = "-c \"echo LineA; echo LineB\"";
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int lineCount = 0;
|
||||||
|
bool foundA = false;
|
||||||
|
bool foundB = false;
|
||||||
|
|
||||||
|
UNUSED(ProcessOps::SpawnProcessSync(cmd, arg, [&](StringView line) {
|
||||||
|
lineCount++;
|
||||||
|
if (line.find("LineA") != String::npos) foundA = true;
|
||||||
|
if (line.find("LineB") != String::npos) foundB = true;
|
||||||
|
}));
|
||||||
|
|
||||||
|
IAT_CHECK(foundA);
|
||||||
|
IAT_CHECK(foundB);
|
||||||
|
// We expect at least 2 lines.
|
||||||
|
// (Windows sometimes echoes the command itself depending on echo settings, but we check contents)
|
||||||
|
IAT_CHECK(lineCount >= 2);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Registration
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
IAT_BEGIN_TEST_LIST()
|
||||||
|
IAT_ADD_TEST(TestBasicRun);
|
||||||
|
IAT_ADD_TEST(TestArguments);
|
||||||
|
IAT_ADD_TEST(TestExitCodes);
|
||||||
|
IAT_ADD_TEST(TestMissingExe);
|
||||||
|
IAT_ADD_TEST(TestLargeOutput);
|
||||||
|
IAT_ADD_TEST(TestMultiLine);
|
||||||
|
IAT_END_TEST_LIST()
|
||||||
|
|
||||||
|
IAT_END_BLOCK()
|
||||||
|
|
||||||
|
int main(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
UNUSED(argc);
|
||||||
|
UNUSED(argv);
|
||||||
|
|
||||||
|
// Define runner (StopOnFail=false, Verbose=true)
|
||||||
|
ia::iatest::runner<false, true> testRunner;
|
||||||
|
|
||||||
|
// Run the BinaryReader block
|
||||||
|
testRunner.testBlock<Core_Process>();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
240
Tests/Unit/Utils.cpp
Normal file
240
Tests/Unit/Utils.cpp
Normal file
@ -0,0 +1,240 @@
|
|||||||
|
// IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
// Copyright (C) 2025 IAS (ias@iasoft.dev)
|
||||||
|
//
|
||||||
|
// 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
#include <IACore/Utils.hpp>
|
||||||
|
|
||||||
|
#include <IACore/IATest.hpp>
|
||||||
|
|
||||||
|
using namespace IACore;
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Test Structs for Hashing (Must be defined at Global Scope)
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
struct TestVec3 {
|
||||||
|
FLOAT32 x, y, z;
|
||||||
|
|
||||||
|
// Equality operator required for hash maps, though strictly
|
||||||
|
// the hash function itself doesn't need it, it's good practice to test both.
|
||||||
|
bool operator==(const TestVec3& other) const {
|
||||||
|
return x == other.x && y == other.y && z == other.z;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Inject the hash specialization into the ankerl namespace
|
||||||
|
// This proves the macro works structurally
|
||||||
|
IA_MAKE_HASHABLE(TestVec3, &TestVec3::x, &TestVec3::y, &TestVec3::z);
|
||||||
|
|
||||||
|
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
// Test Block Definition
|
||||||
|
// -----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
IAT_BEGIN_BLOCK(Core, Utils)
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 1. Binary <-> Hex String Conversion
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestHexConversion()
|
||||||
|
{
|
||||||
|
// A. Binary To Hex
|
||||||
|
UINT8 bin[] = { 0xDE, 0xAD, 0xBE, 0xEF, 0x00, 0xFF };
|
||||||
|
String hex = IACore::Utils::BinaryToHexString(bin);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(hex, String("DEADBEEF00FF"));
|
||||||
|
|
||||||
|
// B. Hex To Binary (Valid Upper)
|
||||||
|
auto resUpper = IACore::Utils::HexStringToBinary("DEADBEEF00FF");
|
||||||
|
IAT_CHECK(resUpper.has_value());
|
||||||
|
IAT_CHECK_EQ(resUpper->size(), (SIZE_T)6);
|
||||||
|
IAT_CHECK_EQ((*resUpper)[0], 0xDE);
|
||||||
|
IAT_CHECK_EQ((*resUpper)[5], 0xFF);
|
||||||
|
|
||||||
|
// C. Hex To Binary (Valid Lower/Mixed)
|
||||||
|
auto resLower = IACore::Utils::HexStringToBinary("deadbeef00ff");
|
||||||
|
IAT_CHECK(resLower.has_value());
|
||||||
|
IAT_CHECK_EQ((*resLower)[0], 0xDE);
|
||||||
|
|
||||||
|
// D. Round Trip Integrity
|
||||||
|
Vector<UINT8> original = { 1, 2, 3, 4, 5 };
|
||||||
|
String s = IACore::Utils::BinaryToHexString(original);
|
||||||
|
auto back = IACore::Utils::HexStringToBinary(s);
|
||||||
|
IAT_CHECK(back.has_value());
|
||||||
|
IAT_CHECK_EQ(original.size(), back->size());
|
||||||
|
IAT_CHECK_EQ(original[2], (*back)[2]);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 2. Hex Error Handling
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestHexErrors()
|
||||||
|
{
|
||||||
|
// Odd Length
|
||||||
|
auto odd = IACore::Utils::HexStringToBinary("ABC");
|
||||||
|
IAT_CHECK_NOT(odd.has_value());
|
||||||
|
// Optional: IAT_CHECK_EQ(odd.error(), "Hex string must have even length");
|
||||||
|
|
||||||
|
// Invalid Characters
|
||||||
|
auto invalid = IACore::Utils::HexStringToBinary("ZZTOP");
|
||||||
|
IAT_CHECK_NOT(invalid.has_value());
|
||||||
|
|
||||||
|
// Empty string is valid (empty vector)
|
||||||
|
auto empty = IACore::Utils::HexStringToBinary("");
|
||||||
|
IAT_CHECK(empty.has_value());
|
||||||
|
IAT_CHECK_EQ(empty->size(), (SIZE_T)0);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 3. Algorithms: Sorting
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestSort()
|
||||||
|
{
|
||||||
|
Vector<int> nums = { 5, 1, 4, 2, 3 };
|
||||||
|
|
||||||
|
IACore::Utils::Sort(nums);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(nums[0], 1);
|
||||||
|
IAT_CHECK_EQ(nums[1], 2);
|
||||||
|
IAT_CHECK_EQ(nums[2], 3);
|
||||||
|
IAT_CHECK_EQ(nums[3], 4);
|
||||||
|
IAT_CHECK_EQ(nums[4], 5);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 4. Algorithms: Binary Search (Left/Right)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestBinarySearch()
|
||||||
|
{
|
||||||
|
// Must be sorted for Binary Search
|
||||||
|
Vector<int> nums = { 10, 20, 20, 20, 30 };
|
||||||
|
|
||||||
|
// Search Left (Lower Bound) -> First element >= value
|
||||||
|
auto itLeft = IACore::Utils::BinarySearchLeft(nums, 20);
|
||||||
|
IAT_CHECK(itLeft != nums.end());
|
||||||
|
IAT_CHECK_EQ(*itLeft, 20);
|
||||||
|
IAT_CHECK_EQ(std::distance(nums.begin(), itLeft), 1); // Index 1 is first 20
|
||||||
|
|
||||||
|
// Search Right (Upper Bound) -> First element > value
|
||||||
|
auto itRight = IACore::Utils::BinarySearchRight(nums, 20);
|
||||||
|
IAT_CHECK(itRight != nums.end());
|
||||||
|
IAT_CHECK_EQ(*itRight, 30); // Points to 30
|
||||||
|
IAT_CHECK_EQ(std::distance(nums.begin(), itRight), 4); // Index 4
|
||||||
|
|
||||||
|
// Search for non-existent
|
||||||
|
auto itFail = IACore::Utils::BinarySearchLeft(nums, 99);
|
||||||
|
IAT_CHECK(itFail == nums.end());
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 5. Hashing Basics
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestHashBasics()
|
||||||
|
{
|
||||||
|
UINT64 h1 = IACore::Utils::ComputeHash(10, 20.5f, "Hello");
|
||||||
|
UINT64 h2 = IACore::Utils::ComputeHash(10, 20.5f, "Hello");
|
||||||
|
UINT64 h3 = IACore::Utils::ComputeHash(10, 20.5f, "World");
|
||||||
|
|
||||||
|
// Determinism
|
||||||
|
IAT_CHECK_EQ(h1, h2);
|
||||||
|
|
||||||
|
// Differentiation
|
||||||
|
IAT_CHECK_NEQ(h1, h3);
|
||||||
|
|
||||||
|
// Order sensitivity (Golden ratio combine should care about order)
|
||||||
|
// Hash(A, B) != Hash(B, A)
|
||||||
|
UINT64 orderA = IACore::Utils::ComputeHash(1, 2);
|
||||||
|
UINT64 orderB = IACore::Utils::ComputeHash(2, 1);
|
||||||
|
IAT_CHECK_NEQ(orderA, orderB);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// 6. Macro Verification (IA_MAKE_HASHABLE)
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
BOOL TestHashMacro()
|
||||||
|
{
|
||||||
|
TestVec3 v1 { 1.0f, 2.0f, 3.0f };
|
||||||
|
TestVec3 v2 { 1.0f, 2.0f, 3.0f };
|
||||||
|
TestVec3 v3 { 1.0f, 2.0f, 4.0f }; // Slight change
|
||||||
|
|
||||||
|
// Instantiate the hasher manually to verify the struct specialization exists
|
||||||
|
ankerl::unordered_dense::hash<TestVec3> hasher;
|
||||||
|
|
||||||
|
UINT64 h1 = hasher(v1);
|
||||||
|
UINT64 h2 = hasher(v2);
|
||||||
|
UINT64 h3 = hasher(v3);
|
||||||
|
|
||||||
|
IAT_CHECK_EQ(h1, h2); // Same content = same hash
|
||||||
|
IAT_CHECK_NEQ(h1, h3); // Different content = different hash
|
||||||
|
|
||||||
|
// -------------------------------------------------------------
|
||||||
|
// Verify ComputeHash integration
|
||||||
|
// -------------------------------------------------------------
|
||||||
|
|
||||||
|
// We cannot check EQ(h1, ComputeHash(v1)) because ComputeHash applies
|
||||||
|
// one extra layer of "Golden Ratio Mixing" on top of the object's hash.
|
||||||
|
// Instead, we verify that ComputeHash behaves exactly like a manual HashCombine.
|
||||||
|
|
||||||
|
UINT64 hManual = 0;
|
||||||
|
IACore::Utils::HashCombine(hManual, v1);
|
||||||
|
|
||||||
|
UINT64 hWrapper = IACore::Utils::ComputeHash(v1);
|
||||||
|
|
||||||
|
// This proves ComputeHash found the specialization and mixed it correctly
|
||||||
|
IAT_CHECK_EQ(hManual, hWrapper);
|
||||||
|
|
||||||
|
// Optional: Verify the avalanche effect took place (hWrapper should NOT be h1)
|
||||||
|
IAT_CHECK_NEQ(h1, hWrapper);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
// Registration
|
||||||
|
// -------------------------------------------------------------------------
|
||||||
|
IAT_BEGIN_TEST_LIST()
|
||||||
|
IAT_ADD_TEST(TestHexConversion);
|
||||||
|
IAT_ADD_TEST(TestHexErrors);
|
||||||
|
IAT_ADD_TEST(TestSort);
|
||||||
|
IAT_ADD_TEST(TestBinarySearch);
|
||||||
|
IAT_ADD_TEST(TestHashBasics);
|
||||||
|
IAT_ADD_TEST(TestHashMacro);
|
||||||
|
IAT_END_TEST_LIST()
|
||||||
|
|
||||||
|
IAT_END_BLOCK()
|
||||||
|
|
||||||
|
int main(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
UNUSED(argc);
|
||||||
|
UNUSED(argv);
|
||||||
|
|
||||||
|
// Define runner (StopOnFail=false, Verbose=true)
|
||||||
|
ia::iatest::runner<false, true> testRunner;
|
||||||
|
|
||||||
|
// Run the BinaryReader block
|
||||||
|
testRunner.testBlock<Core_Utils>();
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
26
Tools/Builder/Build.py
Normal file
26
Tools/Builder/Build.py
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# IACore-OSS; The Core Library for All IA Open Source Projects
|
||||||
|
# Copyright (C) 20245IAS (ias@iasoft.dev)
|
||||||
|
#
|
||||||
|
# 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 <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
def main(args: list[str]):
|
||||||
|
os.system("cmake -S. -B./Build -G Ninja -DCMAKE_C_COMPILER=clang -DCMAKE_CXX_COMPILER=clang++ -DCMAKE_EXPORT_COMPILE_COMMANDS=ON -DCMAKE_BUILD_TYPE=Debug")
|
||||||
|
os.system("cmake --build ./Build")
|
||||||
|
|
||||||
|
main(sys.argv)
|
||||||
Reference in New Issue
Block a user