KonstantinSeurer
/
mesa
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

radv: make use of radv_sc_read() 

Reviewed-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>
This commit is contained in:
Timothy Arceri 2019-10-29 17:46:57 +11:00
parent 28fff3efbc
commit cf25664686
3 changed files with 76 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
src/amd/vulkan/radv_device.c
View File

@ -2051,10 +2051,11 @@ static void run_secure_compile_device(struct radv_device *device, unsigned proce
goto secure_compile_exit;
while (true) {
read(fd_secure_input[0], &sc_type, sizeof(sc_type));
radv_sc_read(fd_secure_input[0], &sc_type, sizeof(sc_type), false);
if (sc_type == RADV_SC_TYPE_COMPILE_PIPELINE) {
struct radv_pipeline *pipeline;
bool sc_read = true;
pipeline = vk_zalloc2(&device->alloc, NULL, sizeof(*pipeline), 8,
VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
@ -2063,69 +2064,96 @@ static void run_secure_compile_device(struct radv_device *device, unsigned proce
/* Read pipeline layout */
struct radv_pipeline_layout layout;
read(fd_secure_input[0], &layout, sizeof(struct radv_pipeline_layout));
read(fd_secure_input[0], &layout.num_sets, sizeof(uint32_t));
sc_read = radv_sc_read(fd_secure_input[0], &layout, sizeof(struct radv_pipeline_layout), true);
sc_read &= radv_sc_read(fd_secure_input[0], &layout.num_sets, sizeof(uint32_t), true);
if (!sc_read)
goto secure_compile_exit;
for (uint32_t set = 0; set < layout.num_sets; set++) {
uint32_t layout_size;
read(fd_secure_input[0], &layout_size, sizeof(uint32_t));
sc_read &= radv_sc_read(fd_secure_input[0], &layout_size, sizeof(uint32_t), true);
if (!sc_read)
goto secure_compile_exit;
layout.set[set].layout = malloc(layout_size);
layout.set[set].layout->layout_size = layout_size;
read(fd_secure_input[0], layout.set[set].layout, layout.set[set].layout->layout_size);
sc_read &= radv_sc_read(fd_secure_input[0], layout.set[set].layout,
layout.set[set].layout->layout_size, true);
}
pipeline->layout = &layout;
/* Read pipeline key */
struct radv_pipeline_key key;
read(fd_secure_input[0], &key, sizeof(struct radv_pipeline_key));
sc_read &= radv_sc_read(fd_secure_input[0], &key, sizeof(struct radv_pipeline_key), true);
/* Read pipeline create flags */
VkPipelineCreateFlags flags;
read(fd_secure_input[0], &flags, sizeof(VkPipelineCreateFlags));
sc_read &= radv_sc_read(fd_secure_input[0], &flags, sizeof(VkPipelineCreateFlags), true);
/* Read stage and shader information */
uint32_t num_stages;
const VkPipelineShaderStageCreateInfo *pStages[MESA_SHADER_STAGES] = { 0, };
read(fd_secure_input[0], &num_stages, sizeof(uint32_t));
sc_read &= radv_sc_read(fd_secure_input[0], &num_stages, sizeof(uint32_t), true);
if (!sc_read)
goto secure_compile_exit;
for (uint32_t i = 0; i < num_stages; i++) {
/* Read stage */
gl_shader_stage stage;
read(fd_secure_input[0], &stage, sizeof(gl_shader_stage));
sc_read &= radv_sc_read(fd_secure_input[0], &stage, sizeof(gl_shader_stage), true);
VkPipelineShaderStageCreateInfo *pStage = calloc(1, sizeof(VkPipelineShaderStageCreateInfo));
/* Read entry point name */
size_t name_size;
read(fd_secure_input[0], &name_size, sizeof(size_t));
sc_read &= radv_sc_read(fd_secure_input[0], &name_size, sizeof(size_t), true);
if (!sc_read)
goto secure_compile_exit;
char *ep_name = malloc(name_size);
read(fd_secure_input[0], ep_name, name_size);
sc_read &= radv_sc_read(fd_secure_input[0], ep_name, name_size, true);
pStage->pName = ep_name;
/* Read shader module */
size_t module_size;
read(fd_secure_input[0], &module_size, sizeof(size_t));
sc_read &= radv_sc_read(fd_secure_input[0], &module_size, sizeof(size_t), true);
if (!sc_read)
goto secure_compile_exit;
struct radv_shader_module *module = malloc(module_size);
read(fd_secure_input[0], module, module_size);
sc_read &= radv_sc_read(fd_secure_input[0], module, module_size, true);
pStage->module = radv_shader_module_to_handle(module);
/* Read specialization info */
bool has_spec_info;
read(fd_secure_input[0], &has_spec_info, sizeof(bool));
sc_read &= radv_sc_read(fd_secure_input[0], &has_spec_info, sizeof(bool), true);
if (!sc_read)
goto secure_compile_exit;
if (has_spec_info) {
VkSpecializationInfo *specInfo = malloc(sizeof(VkSpecializationInfo));
pStage->pSpecializationInfo = specInfo;
read(fd_secure_input[0], &specInfo->dataSize, sizeof(size_t));
sc_read &= radv_sc_read(fd_secure_input[0], &specInfo->dataSize, sizeof(size_t), true);
if (!sc_read)
goto secure_compile_exit;
void *si_data = malloc(specInfo->dataSize);
read(fd_secure_input[0], si_data, specInfo->dataSize);
sc_read &= radv_sc_read(fd_secure_input[0], si_data, specInfo->dataSize, true);
specInfo->pData = si_data;
read(fd_secure_input[0], &specInfo->mapEntryCount, sizeof(uint32_t));
sc_read &= radv_sc_read(fd_secure_input[0], &specInfo->mapEntryCount, sizeof(uint32_t), true);
if (!sc_read)
goto secure_compile_exit;
VkSpecializationMapEntry *mapEntries = malloc(sizeof(VkSpecializationMapEntry) * specInfo->mapEntryCount);
for (uint32_t j = 0; j < specInfo->mapEntryCount; j++)
read(fd_secure_input[0], &mapEntries[j], sizeof(VkSpecializationMapEntry));
for (uint32_t j = 0; j < specInfo->mapEntryCount; j++) {
sc_read &= radv_sc_read(fd_secure_input[0], &mapEntries[j], sizeof(VkSpecializationMapEntry), true);
if (!sc_read)
goto secure_compile_exit;
}
specInfo->pMapEntries = mapEntries;
}
@ -2221,9 +2249,9 @@ static VkResult fork_secure_compile_device(struct radv_device *device)
/* Read the init result returned from the secure process */
enum radv_secure_compile_type sc_type;
read(fd_secure_output[process][0], &sc_type, sizeof(sc_type));
bool sc_read = radv_sc_read(fd_secure_output[process][0], &sc_type, sizeof(sc_type), true);
if (sc_type == RADV_SC_TYPE_INIT_FAILURE) {
if (sc_type == RADV_SC_TYPE_INIT_FAILURE || !sc_read) {
close(fd_secure_input[process][0]);
close(fd_secure_input[process][1]);
close(fd_secure_output[process][1]);

29
src/amd/vulkan/radv_pipeline.c
View File

@ -4621,7 +4621,7 @@ radv_pipeline_get_streamout_shader(struct radv_pipeline *pipeline)
return NULL;
}
static void
static VkResult
radv_secure_compile(struct radv_pipeline *pipeline,
struct radv_device *device,
const struct radv_pipeline_key *key,
@ -4708,19 +4708,23 @@ radv_secure_compile(struct radv_pipeline *pipeline,
/* Read the data returned from the secure process */
while (sc_type != RADV_SC_TYPE_COMPILE_PIPELINE_FINISHED) {
read(fd_secure_output, &sc_type, sizeof(sc_type));
if (!radv_sc_read(fd_secure_output, &sc_type, sizeof(sc_type), true))
return VK_ERROR_DEVICE_LOST;
if (sc_type == RADV_SC_TYPE_WRITE_DISK_CACHE) {
assert(device->physical_device->disk_cache);
uint8_t disk_sha1[20];
read(fd_secure_output, disk_sha1, sizeof(uint8_t) * 20);
if (!radv_sc_read(fd_secure_output, disk_sha1, sizeof(uint8_t) * 20, true))
return VK_ERROR_DEVICE_LOST;
uint32_t entry_size;
read(fd_secure_output, &entry_size, sizeof(uint32_t));
if (!radv_sc_read(fd_secure_output, &entry_size, sizeof(uint32_t), true))
return VK_ERROR_DEVICE_LOST;
struct cache_entry *entry = malloc(entry_size);
read(fd_secure_output, entry, entry_size);
if (!radv_sc_read(fd_secure_output, entry, entry_size, true))
return VK_ERROR_DEVICE_LOST;
disk_cache_put(device->physical_device->disk_cache,
disk_sha1, entry, entry_size,
@ -4729,7 +4733,8 @@ radv_secure_compile(struct radv_pipeline *pipeline,
free(entry);
} else if (sc_type == RADV_SC_TYPE_READ_DISK_CACHE) {
uint8_t disk_sha1[20];
read(fd_secure_output, disk_sha1, sizeof(uint8_t) * 20);
if (!radv_sc_read(fd_secure_output, disk_sha1, sizeof(uint8_t) * 20, true))
return VK_ERROR_DEVICE_LOST;
size_t size;
struct cache_entry *entry = (struct cache_entry *)
@ -4752,6 +4757,8 @@ radv_secure_compile(struct radv_pipeline *pipeline,
device->sc_state->secure_compile_thread_counter--;
device->sc_state->secure_compile_processes[process].in_use = false;
mtx_unlock(&device->sc_state->secure_compile_mutex);
return VK_SUCCESS;
}
static VkResult
@ -4792,9 +4799,7 @@ radv_pipeline_init(struct radv_pipeline *pipeline,
struct radv_pipeline_key key = radv_generate_graphics_pipeline_key(pipeline, pCreateInfo, &blend, has_view_index);
if (radv_device_use_secure_compile(device->instance)) {
radv_secure_compile(pipeline, device, &key, pStages, pCreateInfo->flags, pCreateInfo->stageCount);
/* TODO: should we actualy return failure ??? */
return VK_SUCCESS;
return radv_secure_compile(pipeline, device, &key, pStages, pCreateInfo->flags, pCreateInfo->stageCount);
} else {
radv_create_shaders(pipeline, device, cache, &key, pStages, pCreateInfo->flags, pipeline_feedback, stage_feedbacks);
}
@ -5057,10 +5062,10 @@ static VkResult radv_compute_pipeline_create(
pStages[MESA_SHADER_COMPUTE] = &pCreateInfo->stage;
if (radv_device_use_secure_compile(device->instance)) {
radv_secure_compile(pipeline, device, &(struct radv_pipeline_key) {0}, pStages, pCreateInfo->flags, 1);
result = radv_secure_compile(pipeline, device, &(struct radv_pipeline_key) {0}, pStages, pCreateInfo->flags, 1);
*pPipeline = radv_pipeline_to_handle(pipeline);
/* TODO: should we actualy return failure ??? */
return VK_SUCCESS;
return result;
} else {
radv_create_shaders(pipeline, device, cache, &(struct radv_pipeline_key) {0}, pStages, pCreateInfo->flags, pipeline_feedback, stage_feedbacks);
}

16
src/amd/vulkan/radv_pipeline_cache.c
View File

@ -260,17 +260,21 @@ radv_sc_read_from_disk_cache(struct radv_device *device, uint8_t *disk_sha1)
disk_sha1, sizeof(uint8_t) * 20);
uint8_t found_cache_entry;
read(device->sc_state->secure_compile_processes[process].fd_secure_input,
&found_cache_entry, sizeof(uint8_t));
if (!radv_sc_read(device->sc_state->secure_compile_processes[process].fd_secure_input,
&found_cache_entry, sizeof(uint8_t), true))
return NULL;
if (found_cache_entry) {
size_t entry_size;
read(device->sc_state->secure_compile_processes[process].fd_secure_input,
&entry_size, sizeof(size_t));
if (!radv_sc_read(device->sc_state->secure_compile_processes[process].fd_secure_input,
&entry_size, sizeof(size_t), true))
return NULL;
entry = malloc(entry_size);
read(device->sc_state->secure_compile_processes[process].fd_secure_input,
entry, entry_size);
if (!radv_sc_read(device->sc_state->secure_compile_processes[process].fd_secure_input,
entry, entry_size, true))
return NULL;
return entry;
}