summaryrefslogtreecommitdiff
path: root/source/compiler-core/slang-spirv-dis-compiler.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'source/compiler-core/slang-spirv-dis-compiler.cpp')
-rw-r--r--source/compiler-core/slang-spirv-dis-compiler.cpp42
1 files changed, 29 insertions, 13 deletions
diff --git a/source/compiler-core/slang-spirv-dis-compiler.cpp b/source/compiler-core/slang-spirv-dis-compiler.cpp
index 04e5c8e4a..0e484c7c5 100644
--- a/source/compiler-core/slang-spirv-dis-compiler.cpp
+++ b/source/compiler-core/slang-spirv-dis-compiler.cpp
@@ -1,8 +1,12 @@
#include "slang-spirv-dis-compiler.h"
#include "../core/slang-common.h"
+#include "../core/slang-string-util.h"
+#include "../core/slang-string.h"
+#include "slang-artifact-desc-util.h"
#include "slang-artifact-representation.h"
#include "slang-artifact-util.h"
+#include "slang-artifact-representation-impl.h"
namespace Slang
{
@@ -39,19 +43,23 @@ SlangResult SLANG_MCALL SPIRVDisDownstreamCompiler::convert(
ISlangBlob* fromBlob;
SLANG_RETURN_ON_FAIL(from->loadBlob(ArtifactKeep::No, &fromBlob));
+ ComPtr<IOSFileArtifactRepresentation> fromFile;
+ SLANG_RETURN_ON_FAIL(from->requireFile(ArtifactKeep::No, fromFile.writeRef()));
+
+ String toFile;
+ File::generateTemporary(UnownedStringSlice("spv-asm"), toFile);
+
// Set up our process
CommandLine commandLine;
commandLine.m_executableLocation.setName("spirv-dis");
+ commandLine.addArg("--comment");
+ commandLine.addArg(fromFile->getPath());
+ commandLine.addArg("-o");
+ commandLine.addArg(toFile);
RefPtr<Process> p;
SLANG_RETURN_ON_FAIL(Process::create(commandLine, 0, p));
- const auto in = p->getStream(StdStreamType::In);
- const auto out = p->getStream(StdStreamType::Out);
const auto err = p->getStream(StdStreamType::ErrorOut);
- // Write the assembly
- SLANG_RETURN_ON_FAIL(in->write(fromBlob->getBufferPointer(), fromBlob->getBufferSize()));
- in->close();
-
// Wait for it to finish
if(!p->waitForTermination(1000))
return SLANG_FAIL;
@@ -61,18 +69,26 @@ SlangResult SLANG_MCALL SPIRVDisDownstreamCompiler::convert(
SLANG_RETURN_ON_FAIL(StreamUtil::readAll(err, 0, errData));
fwrite(errData.getBuffer(), errData.getCount(), 1, stderr);
+ // If spirv-dis failed, we fail
const auto ret = p->getReturnValue();
if(ret != 0)
return SLANG_FAIL;
- // Read the disassembly
- List<Byte> outData;
- SLANG_RETURN_ON_FAIL(StreamUtil::readAll(out, 0, outData));
-
- // Wobble it into an artifact
- ComPtr<ISlangBlob> outBlob = RawBlob::create(outData.getBuffer(), outData.getCount());
+ // Normalize line endings
+ String outContents;
+ SLANG_RETURN_ON_FAIL(File::readAllText(toFile, outContents));
+ StringBuilder outBuilder;
+ StringUtil::appendStandardLines(outContents.getUnownedSlice(), outBuilder);
+ SLANG_RETURN_ON_FAIL(File::writeAllBytes(toFile, outBuilder.getBuffer(), outBuilder.getLength()));
+
+ // Return as a file artifact
+ auto fileRep = OSFileArtifactRepresentation::create(
+ IOSFileArtifactRepresentation::Kind::Owned,
+ toFile.getUnownedSlice(),
+ nullptr
+ );
auto artifact = ArtifactUtil::createArtifact(to);
- artifact->addRepresentationUnknown(outBlob.detach());
+ artifact->addRepresentation(fileRep.detach());
*outArtifact = artifact.detach();
return SLANG_OK;