AuroraMiddleware/SPIRV-Cross-Vulnerable
1
0
Fork 0
mirror of https://github.com/KhronosGroup/SPIRV-Cross.git synced 2024-11-10 06:00:07 +00:00
Code Issues Projects Releases Wiki Activity

Check if the location of the original variable is set

Browse Source
This commit is contained in:
Polona Caserman 2016-12-19 19:52:41 +01:00 committed by Robert Konrad
parent 2f48065ec1
commit bafde4e917
3 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spirv_cross.cpp
View File

@ -996,6 +996,12 @@ uint64_t Compiler::get_decoration_mask(uint32_t id) const
return dec.decoration_flags;
}
bool Compiler::is_decoration_set(uint32_t id, spv::Decoration decoration) const
{
auto &dec = meta.at(id).decoration;
return (dec.decoration_flags & (1ull << DecorationLocation));
}
uint32_t Compiler::get_decoration(uint32_t id, Decoration decoration) const
{
auto &dec = meta.at(id).decoration;

2
spirv_cross.hpp
View File

@ -136,6 +136,8 @@ public:
// Gets a bitmask for the decorations which are applied to ID.
// I.e. (1ull << spv::DecorationFoo) | (1ull << spv::DecorationBar)
uint64_t get_decoration_mask(uint32_t id) const;
bool is_decoration_set(uint32_t id, spv::Decoration decoration) const;
// Gets the value for decorations which take arguments.
// If decoration doesn't exist or decoration is not recognized,

4
spirv_msl.cpp
View File

@ -494,7 +494,9 @@ uint32_t CompilerMSL::add_interface_struct(StorageClass storage, uint32_t vtx_bi
// Copy the variable location from the original variable to the member
auto &dec = meta[p_var->self].decoration;
set_member_decoration(ib_type.self, ib_mbr_idx, DecorationLocation, dec.location);
if (is_decoration_set(p_var->self, DecorationLocation)) {
set_member_decoration(ib_type.self, ib_mbr_idx, DecorationLocation, dec.location);
}
// Mark the member as builtin if needed
if (is_builtin_variable(*p_var))