summaryrefslogtreecommitdiff
path: root/source/slang/slang-spirv-val.cpp
blob: e62564cc43d53b914bf2d10b811b798a9da68fd8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include "slang-spirv-val.h"

namespace Slang
{

static SlangResult disassembleSPIRV(const List<uint8_t>& spirv, String& outErr, String& outDis)
{
    // Set up our process
    CommandLine commandLine;
    commandLine.m_executableLocation.setName("spirv-dis");
    commandLine.addArg("--comment");
    commandLine.addArg("--color");
    RefPtr<Process> p;

    // If we failed to even start the process, then validation isn't available
    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);

    List<Byte> outData;
    List<Byte> outErrData;
    SLANG_RETURN_ON_FAIL(StreamUtil::readAndWrite(in, spirv.getArrayView(), out, outData, err, outErrData));

    SLANG_RETURN_ON_FAIL(p->waitForTermination(10));

    outDis = String(
        reinterpret_cast<const char*>(outData.begin()),
        reinterpret_cast<const char*>(outData.end())
    );

    outErr = String(
        reinterpret_cast<const char*>(outErrData.begin()),
        reinterpret_cast<const char*>(outErrData.end())
    );

    const auto ret = p->getReturnValue();
    return ret == 0 ? SLANG_OK : SLANG_FAIL;
}

SlangResult debugValidateSPIRV(const List<uint8_t>& spirv)
{
    // Set up our process
    CommandLine commandLine;
    commandLine.m_executableLocation.setName("spirv-val");
    commandLine.addArg("--target-env");
    commandLine.addArg("vulkan1.2");
    commandLine.addArg("--scalar-block-layout");
    RefPtr<Process> p;
    const auto createResult = Process::create(commandLine, 0, p);
    // If we failed to even start the process, then validation isn't available
    if(SLANG_FAILED(createResult))
        return SLANG_E_NOT_AVAILABLE;
    const auto in = p->getStream(StdStreamType::In);
    const auto out = p->getStream(StdStreamType::Out);
    const auto err = p->getStream(StdStreamType::ErrorOut);

    List<Byte> outData;
    List<Byte> errData;
    SLANG_RETURN_ON_FAIL(StreamUtil::readAndWrite(in, spirv.getArrayView(), out, outData, err, errData));

    // Wait for it to finish
    if(!p->waitForTermination(1000))
        return SLANG_FAIL;

    // If we failed, dump the spirv first.
    const auto ret = p->getReturnValue();
    if(ret != 0)
    {
        String spirvDisErr;
        String spirvDis;
        disassembleSPIRV(spirv, spirvDisErr, spirvDis);
        fwrite(spirvDisErr.getBuffer(), spirvDisErr.getLength(), 1, stderr);
        fwrite(spirvDis.getBuffer(), spirvDis.getLength(), 1, stderr);
    }

    fwrite(outData.getBuffer(), outData.getCount(), 1, stderr);
    fwrite(errData.getBuffer(), errData.getCount(), 1, stderr);

    return ret == 0 ? SLANG_OK : SLANG_FAIL;
}

}