1 /**************************************************************************
3 * Copyright 2011 Jose Fonseca
4 * Copyright 2008-2009 VMware, Inc.
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
25 **************************************************************************/
30 #include "d3dshader.hpp"
34 struct ID3DXBuffer : public IUnknown {
35 virtual LPVOID STDMETHODCALLTYPE GetBufferPointer(void) = 0;
36 virtual DWORD STDMETHODCALLTYPE GetBufferSize(void) = 0;
40 (WINAPI *PD3DXDISASSEMBLESHADER)(
44 ID3DXBuffer **ppDisassembly
49 DisassembleShader(const DWORD *tokens, IDisassemblyBuffer **ppDisassembly)
51 static BOOL firsttime = TRUE;
54 * TODO: Consider using d3dcompile_xx.dll per
55 * http://msdn.microsoft.com/en-us/library/windows/desktop/ee663275.aspx
58 static HMODULE hD3DXModule = NULL;
59 static PD3DXDISASSEMBLESHADER pfnD3DXDisassembleShader = NULL;
65 for (release = 0; release <= 1; ++release) {
66 /* Version 41 corresponds to Mar 2009 version of DirectX Runtime / SDK */
67 for (version = 41; version >= 0; --version) {
69 _snprintf(filename, sizeof(filename),
70 "d3dx9%s%s%u.dll", release ? "" : "d", version ? "_" : "", version);
71 hD3DXModule = LoadLibraryA(filename);
81 if (!pfnD3DXDisassembleShader) {
82 pfnD3DXDisassembleShader = (PD3DXDISASSEMBLESHADER)GetProcAddress(hD3DXModule, "D3DXDisassembleShader");
90 if (pfnD3DXDisassembleShader) {
91 hr = pfnD3DXDisassembleShader(tokens, FALSE, NULL,
92 reinterpret_cast<ID3DXBuffer **>(ppDisassembly));
98 struct ID3D10Blob : public IUnknown {
99 virtual LPVOID STDMETHODCALLTYPE GetBufferPointer(void) = 0;
100 virtual SIZE_T STDMETHODCALLTYPE GetBufferSize(void) = 0;
103 typedef ID3D10Blob ID3DBlob;
105 #define D3D_DISASM_ENABLE_COLOR_CODE 0x00000001
106 #define D3D_DISASM_ENABLE_DEFAULT_VALUE_PRINTS 0x00000002
107 #define D3D_DISASM_ENABLE_INSTRUCTION_NUMBERING 0x00000004
108 #define D3D_DISASM_ENABLE_INSTRUCTION_CYCLE 0x00000008
109 #define D3D_DISASM_DISABLE_DEBUG_INFO 0x00000010
110 #define D3D_DISASM_ENABLE_INSTRUCTION_OFFSET 0x00000020
111 #define D3D_DISASM_INSTRUCTION_ONLY 0x00000040
114 (WINAPI *PFND3DDISASSEMBLE)(
119 ID3DBlob **ppDisassembly
122 static PFND3DDISASSEMBLE pfnD3DDisassemble = NULL;
125 (WINAPI *PFND3D10DISASSEMBLESHADER)(
127 SIZE_T BytecodeLength,
128 BOOL EnableColorCode,
130 ID3D10Blob **ppDisassembly
133 static PFND3D10DISASSEMBLESHADER pfnD3D10DisassembleShader = NULL;
137 DisassembleShader(const void *pShaderBytecode, SIZE_T BytecodeLength, IDisassemblyBuffer **ppDisassembly)
139 static bool firsttime = true;
142 char szFilename[MAX_PATH];
143 HMODULE hModule = NULL;
145 for (version = 44; version >= 33; --version) {
146 _snprintf(szFilename, sizeof(szFilename), "d3dcompiler_%i.dll", version);
147 hModule = LoadLibraryA(szFilename);
149 pfnD3DDisassemble = (PFND3DDISASSEMBLE)
150 GetProcAddress(hModule, "D3DDisassemble");
151 if (pfnD3DDisassemble) {
156 if (!pfnD3DDisassemble) {
158 * Fallback to D3D10DisassembleShader, which should be always present.
160 if (GetSystemDirectoryA(szFilename, MAX_PATH)) {
161 strcat(szFilename, "\\d3d10.dll");
162 hModule = LoadLibraryA(szFilename);
164 pfnD3D10DisassembleShader = (PFND3D10DISASSEMBLESHADER)
165 GetProcAddress(hModule, "D3D10DisassembleShader");
175 if (pfnD3DDisassemble) {
176 hr = pfnD3DDisassemble(pShaderBytecode, BytecodeLength, 0, NULL,
177 reinterpret_cast<ID3DBlob **>(ppDisassembly));
178 } else if (pfnD3D10DisassembleShader) {
179 hr = pfnD3D10DisassembleShader(pShaderBytecode, BytecodeLength, 0, NULL,
180 reinterpret_cast<ID3D10Blob **>(ppDisassembly));