diff options
| author | kaizhangNV <149626564+kaizhangNV@users.noreply.github.com> | 2024-07-23 10:45:26 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-23 08:45:26 -0700 |
| commit | 986256ffb92ab7c8fc7cf9f2c424919a439a824f (patch) | |
| tree | 260e37bd439275e3398d16fe238b20cd00d08cb7 /source/slang-capture-replay/slang-module.cpp | |
| parent | c28d8b6aec721fa3350fc52647f1572a353f6151 (diff) | |
Feature/capture (#4625)
* Add decoder
* Add a replay executable to consume the decoded content
Add file-processor.cpp/h where we implement the logic to process
the captured file block by block. Each block is:
function header + parameter buffer + function tailer + function
output[optional].
After reading one block, the block of data is sent to decoder module
to dispatch the corresponding API.
Add slang-decoder.cpp/h where we implement the logic to dispatch
the slang API according to the input block data.
- Rename api_callId.h to capture-format.h
- Renmae capture_utility.cpp to capture-utility.cpp
- Renmae capture_utility.h to capture-utility.h
- Change the #include file name accordingly.
* Reorganize source files structure
Move all the capture logic code into `capture` directory.
- the capture code will be build with slang dll.
Move all the replay logic code into `relay` directoy.
- the replay code is not part of slang dll, it will be built
as a stand alone binary and link against slang dll.
Change the #include file names accordingly.
Add tools/slang-replay/main.cpp for the slang-replay stand alone
binary place holder. Will implement it later.
Update premake5.lua accordingly.
* Update cmake files
Update cmake files to change the build process for
capture and relay system.
- capture component should be build with slang dll, so we
should not include replay component.
- replay component should be a separate executable tool, which
should not include capture component.
- In order to easy use our current cmake infrastructure, move
the shared files to a `util` folder
- change the header include path
* Redesgin the interfaces of consumers
Fix some issues in capture
Finish implementing all slang-decoder functions
* Fix the AppleClang build issue
* Address few comments
- Fix the weird indent issues.
- Correct the function name for CreateGlobalSession()
- Rename file-processor to captureFile-processor to be more specific.
- Use Slang::List instead of std::vector
* record/replay: name refactor change
Refactor the naming.
Change the name "encoder/capture" to "record".
Diffstat (limited to 'source/slang-capture-replay/slang-module.cpp')
| -rw-r--r-- | source/slang-capture-replay/slang-module.cpp | 506 |
1 files changed, 0 insertions, 506 deletions
diff --git a/source/slang-capture-replay/slang-module.cpp b/source/slang-capture-replay/slang-module.cpp deleted file mode 100644 index 273faa59d..000000000 --- a/source/slang-capture-replay/slang-module.cpp +++ /dev/null @@ -1,506 +0,0 @@ -#include "capture_utility.h" -#include "slang-module.h" - -namespace SlangCapture -{ - ModuleCapture::ModuleCapture(slang::IModule* module, CaptureManager* captureManager) - : m_actualModule(module), - m_captureManager(captureManager) - { - SLANG_CAPTURE_ASSERT(m_actualModule != nullptr); - SLANG_CAPTURE_ASSERT(m_captureManager != nullptr); - - m_moduleHandle = reinterpret_cast<uint64_t>(m_actualModule.get()); - slangCaptureLog(LogLevel::Verbose, "%s: %p\n", __PRETTY_FUNCTION__, module); - } - - ModuleCapture::~ModuleCapture() - { - m_actualModule->release(); - } - - ISlangUnknown* ModuleCapture::getInterface(const Guid& guid) - { - if(guid == ModuleCapture::getTypeGuid()) - return static_cast<ISlangUnknown*>(this); - else - return nullptr; - } - - SLANG_NO_THROW slang::DeclReflection* ModuleCapture::getModuleReflection() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - slang::DeclReflection* res = (slang::DeclReflection*)m_actualModule->getModuleReflection(); - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::findEntryPointByName( - char const* name, - slang::IEntryPoint** outEntryPoint) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_findEntryPointByName, m_moduleHandle); - encoder->encodeString(name); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->findEntryPointByName(name, outEntryPoint); - - { - encoder->encodeAddress(*outEntryPoint); - m_captureManager->endMethodCaptureAppendOutput(); - } - - if (SLANG_OK == res) - { - EntryPointCapture* entryPointCapture = getEntryPointCapture(*outEntryPoint); - *outEntryPoint = static_cast<slang::IEntryPoint*>(entryPointCapture); - } - return res; - } - - SLANG_NO_THROW SlangInt32 ModuleCapture::getDefinedEntryPointCount() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - SlangInt32 res = m_actualModule->getDefinedEntryPointCount(); - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::getDefinedEntryPoint(SlangInt32 index, slang::IEntryPoint** outEntryPoint) - { - // This call is to find the existing entry point, so it has been created already. Therefore, we don't create a new one - // and assert the error if it is not found in our map. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getDefinedEntryPoint, m_moduleHandle); - encoder->encodeInt32(index); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->getDefinedEntryPoint(index, outEntryPoint); - - { - encoder->encodeAddress(*outEntryPoint); - m_captureManager->endMethodCaptureAppendOutput(); - } - - if (*outEntryPoint) - { - EntryPointCapture* entryPointCapture = m_mapEntryPointToCapture.tryGetValue(*outEntryPoint); - if (!entryPointCapture) - { - SLANG_CAPTURE_ASSERT(!"Entrypoint not found in mapEntryPointToCapture"); - } - *outEntryPoint = static_cast<slang::IEntryPoint*>(entryPointCapture); - } - else - *outEntryPoint = nullptr; - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::serialize(ISlangBlob** outSerializedBlob) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_serialize, m_moduleHandle); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->serialize(outSerializedBlob); - - { - encoder->encodeAddress(*outSerializedBlob); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::writeToFile(char const* fileName) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_writeToFile, m_moduleHandle); - encoder->encodeString(fileName); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->writeToFile(fileName); - return res; - } - - SLANG_NO_THROW const char* ModuleCapture::getName() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - const char* res = m_actualModule->getName(); - return res; - } - - SLANG_NO_THROW const char* ModuleCapture::getFilePath() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - const char* res = m_actualModule->getFilePath(); - return res; - } - - SLANG_NO_THROW const char* ModuleCapture::getUniqueIdentity() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - const char* res = m_actualModule->getUniqueIdentity(); - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::findAndCheckEntryPoint( - char const* name, - SlangStage stage, - slang::IEntryPoint** outEntryPoint, - ISlangBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_findAndCheckEntryPoint, m_moduleHandle); - encoder->encodeString(name); - encoder->encodeEnumValue(stage); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->findAndCheckEntryPoint(name, stage, outEntryPoint, outDiagnostics); - - { - encoder->encodeAddress(*outEntryPoint); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - if (SLANG_OK == res) - { - EntryPointCapture* entryPointCapture = getEntryPointCapture(*outEntryPoint); - *outEntryPoint = static_cast<slang::IEntryPoint*>(entryPointCapture); - } - return res; - } - - SLANG_NO_THROW SlangInt32 ModuleCapture::getDependencyFileCount() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - SlangInt32 res = m_actualModule->getDependencyFileCount(); - return res; - } - - SLANG_NO_THROW char const* ModuleCapture::getDependencyFilePath(SlangInt32 index) - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - const char* res = m_actualModule->getDependencyFilePath(index); - return res; - } - - SLANG_NO_THROW slang::ISession* ModuleCapture::getSession() - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getSession, m_moduleHandle); - encoder = m_captureManager->endMethodCapture(); - } - - slang::ISession* session = m_actualModule->getSession(); - - { - encoder->encodeAddress(session); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return session; - } - - SLANG_NO_THROW slang::ProgramLayout* ModuleCapture::getLayout( - SlangInt targetIndex, - slang::IBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getLayout, m_moduleHandle); - encoder->encodeInt64(targetIndex); - encoder = m_captureManager->endMethodCapture(); - } - - slang::ProgramLayout* programLayout = m_actualModule->getLayout(targetIndex, outDiagnostics); - - { - encoder->encodeAddress(*outDiagnostics); - encoder->encodeAddress(programLayout); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return programLayout; - } - - SLANG_NO_THROW SlangInt ModuleCapture::getSpecializationParamCount() - { - // No need to capture this call as it is just a query. - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - SlangInt res = m_actualModule->getSpecializationParamCount(); - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::getEntryPointCode( - SlangInt entryPointIndex, - SlangInt targetIndex, - slang::IBlob** outCode, - slang::IBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getEntryPointCode, m_moduleHandle); - encoder->encodeInt64(entryPointIndex); - encoder->encodeInt64(targetIndex); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->getEntryPointCode(entryPointIndex, targetIndex, outCode, outDiagnostics); - - { - encoder->encodeAddress(*outCode); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::getTargetCode( - SlangInt targetIndex, - slang::IBlob** outCode, - slang::IBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getTargetCode, m_moduleHandle); - encoder->encodeInt64(targetIndex); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->getTargetCode(targetIndex, outCode, outDiagnostics); - - { - encoder->encodeAddress(*outCode); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::getResultAsFileSystem( - SlangInt entryPointIndex, - SlangInt targetIndex, - ISlangMutableFileSystem** outFileSystem) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getResultAsFileSystem, m_moduleHandle); - encoder->encodeInt64(entryPointIndex); - encoder->encodeInt64(targetIndex); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->getResultAsFileSystem(entryPointIndex, targetIndex, outFileSystem); - - { - encoder->encodeAddress(*outFileSystem); - } - - // TODO: We might need to wrap the file system object. - return res; - } - - SLANG_NO_THROW void ModuleCapture::getEntryPointHash( - SlangInt entryPointIndex, - SlangInt targetIndex, - slang::IBlob** outHash) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getEntryPointHash, m_moduleHandle); - encoder->encodeInt64(entryPointIndex); - encoder->encodeInt64(targetIndex); - encoder = m_captureManager->endMethodCapture(); - } - - m_actualModule->getEntryPointHash(entryPointIndex, targetIndex, outHash); - - { - encoder->encodeAddress(*outHash); - m_captureManager->endMethodCaptureAppendOutput(); - } - } - - SLANG_NO_THROW SlangResult ModuleCapture::specialize( - slang::SpecializationArg const* specializationArgs, - SlangInt specializationArgCount, - slang::IComponentType** outSpecializedComponentType, - ISlangBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_specialize, m_moduleHandle); - encoder->encodeInt64(specializationArgCount); - encoder->encodeStructArray(specializationArgs, specializationArgCount); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->specialize(specializationArgs, specializationArgCount, outSpecializedComponentType, outDiagnostics); - - { - encoder->encodeAddress(*outSpecializedComponentType); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::link( - IComponentType** outLinkedComponentType, - ISlangBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_link, m_moduleHandle); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->link(outLinkedComponentType, outDiagnostics); - - { - encoder->encodeAddress(*outLinkedComponentType); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::getEntryPointHostCallable( - int entryPointIndex, - int targetIndex, - ISlangSharedLibrary** outSharedLibrary, - slang::IBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_getEntryPointHostCallable, m_moduleHandle); - encoder->encodeInt32(entryPointIndex); - encoder->encodeInt32(targetIndex); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->getEntryPointHostCallable(entryPointIndex, targetIndex, outSharedLibrary, outDiagnostics); - - { - encoder->encodeAddress(*outSharedLibrary); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::renameEntryPoint( - const char* newName, IComponentType** outEntryPoint) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_renameEntryPoint, m_moduleHandle); - encoder->encodeString(newName); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->renameEntryPoint(newName, outEntryPoint); - - { - encoder->encodeAddress(*outEntryPoint); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - SLANG_NO_THROW SlangResult ModuleCapture::linkWithOptions( - IComponentType** outLinkedComponentType, - uint32_t compilerOptionEntryCount, - slang::CompilerOptionEntry* compilerOptionEntries, - ISlangBlob** outDiagnostics) - { - slangCaptureLog(LogLevel::Verbose, "%s\n", __PRETTY_FUNCTION__); - - ParameterEncoder* encoder {}; - { - encoder = m_captureManager->beginMethodCapture(ApiCallId::IModule_linkWithOptions, m_moduleHandle); - encoder->encodeUint32(compilerOptionEntryCount); - encoder->encodeStructArray(compilerOptionEntries, compilerOptionEntryCount); - encoder = m_captureManager->endMethodCapture(); - } - - SlangResult res = m_actualModule->linkWithOptions(outLinkedComponentType, compilerOptionEntryCount, compilerOptionEntries, outDiagnostics); - - { - encoder->encodeAddress(*outLinkedComponentType); - encoder->encodeAddress(*outDiagnostics); - m_captureManager->endMethodCaptureAppendOutput(); - } - - return res; - } - - EntryPointCapture* ModuleCapture::getEntryPointCapture(slang::IEntryPoint* entryPoint) - { - EntryPointCapture* entryPointCapture = nullptr; - entryPointCapture = m_mapEntryPointToCapture.tryGetValue(entryPoint); - if (!entryPointCapture) - { - entryPointCapture = new EntryPointCapture(entryPoint, m_captureManager); - Slang::ComPtr<EntryPointCapture> result(entryPointCapture); - m_mapEntryPointToCapture.add(entryPoint, *result.detach()); - } - return entryPointCapture; - } -} |
