source/fuzz/transformation_add_local_variable.h

   1 // Copyright (c) 2020 Google LLC
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 #ifndef SOURCE_FUZZ_TRANSFORMATION_ADD_LOCAL_VARIABLE_H_
  16 #define SOURCE_FUZZ_TRANSFORMATION_ADD_LOCAL_VARIABLE_H_
  17
  18 #include "source/fuzz/protobufs/spirvfuzz_protobufs.h"
  19 #include "source/fuzz/transformation.h"
  20 #include "source/fuzz/transformation_context.h"
  21 #include "source/opt/ir_context.h"
  22
  23 namespace spvtools {
  24 namespace fuzz {
  25
  26 class TransformationAddLocalVariable : public Transformation {
  27  public:
  28   explicit TransformationAddLocalVariable(
  29       protobufs::TransformationAddLocalVariable message);
  30
  31   TransformationAddLocalVariable(uint32_t fresh_id, uint32_t type_id,
  32                                  uint32_t function_id, uint32_t initializer_id,
  33                                  bool value_is_irrelevant);
  34
  35   // - |message_.fresh_id| must not be used by the module
  36   // - |message_.type_id| must be the id of a pointer type with Function
  37   //   storage class
  38   // - |message_.initializer_id| must be the id of a constant with the same
  39   //   type as the pointer's pointee type
  40   // - |message_.function_id| must be the id of a function
  41   bool IsApplicable(
  42       opt::IRContext* ir_context,
  43       const TransformationContext& transformation_context) const override;
  44
  45   // Adds an instruction to the start of |message_.function_id|, of the form:
  46   //   |message_.fresh_id| = OpVariable |message_.type_id| Function
  47   //                         |message_.initializer_id|
  48   // If |message_.value_is_irrelevant| holds, adds a corresponding fact to the
  49   // fact manager in |transformation_context|.
  50   void Apply(opt::IRContext* ir_context,
  51              TransformationContext* transformation_context) const override;
  52
  53   std::unordered_set<uint32_t> GetFreshIds() const override;
  54
  55   protobufs::Transformation ToMessage() const override;
  56
  57  private:
  58   protobufs::TransformationAddLocalVariable message_;
  59 };
  60
  61 }  // namespace fuzz
  62 }  // namespace spvtools
  63
  64 #endif  // SOURCE_FUZZ_TRANSFORMATION_ADD_LOCAL_VARIABLE_H_