SPIRV-Tools/source/fuzz/transformation_add_copy_memory.h

79 lines
3.1 KiB
C++

// Copyright (c) 2020 Vasyl Teliman
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#ifndef SOURCE_FUZZ_TRANSFORMATION_ADD_COPY_MEMORY_H_
#define SOURCE_FUZZ_TRANSFORMATION_ADD_COPY_MEMORY_H_
#include "source/fuzz/protobufs/spirvfuzz_protobufs.h"
#include "source/fuzz/transformation.h"
#include "source/fuzz/transformation_context.h"
#include "source/opt/ir_context.h"
namespace spvtools {
namespace fuzz {
class TransformationAddCopyMemory : public Transformation {
public:
explicit TransformationAddCopyMemory(
const protobufs::TransformationAddCopyMemory& message);
TransformationAddCopyMemory(
const protobufs::InstructionDescriptor& instruction_descriptor,
uint32_t fresh_id, uint32_t source_id, SpvStorageClass storage_class,
uint32_t initializer_id);
// - |instruction_descriptor| must point to a valid instruction in the module.
// - it should be possible to insert OpCopyMemory before
// |instruction_descriptor| (i.e. the module remains valid after the
// insertion).
// - |source_id| must be a result id for some valid instruction in the module.
// - |fresh_id| must be a fresh id to copy memory into.
// - type of |source_id| must be OpTypePointer where pointee can be used with
// OpCopyMemory.
// - If the pointee type of |source_id| is a struct type, it must not have the
// Block or BufferBlock decoration.
// - |storage_class| must be either Private or Function.
// - type ids of instructions with result ids |source_id| and |initialize_id|
// must be the same.
bool IsApplicable(
opt::IRContext* ir_context,
const TransformationContext& transformation_context) const override;
// A global or local variable with id |target_id| and |storage_class| class is
// created. An 'OpCopyMemory %fresh_id %source_id' instruction is inserted
// before the |instruction_descriptor|.
void Apply(opt::IRContext* ir_context,
TransformationContext* transformation_context) const override;
std::unordered_set<uint32_t> GetFreshIds() const override;
protobufs::Transformation ToMessage() const override;
// Returns true if we can copy memory from |instruction| using OpCopyMemory.
static bool IsInstructionSupported(opt::IRContext* ir_context,
opt::Instruction* inst);
private:
// Returns whether the type, pointed to by some OpTypePointer, can be used
// with OpCopyMemory instruction.
static bool CanUsePointeeWithCopyMemory(const opt::analysis::Type& type);
protobufs::TransformationAddCopyMemory message_;
};
} // namespace fuzz
} // namespace spvtools
#endif // SOURCE_FUZZ_TRANSFORMATION_ADD_COPY_MEMORY_H_