clang/test/CodeGenCoroutines/pr59181.cpp

   1 // Test for PR59181. Tests that no conditional cleanup is created around await_suspend.
   2 //
   3 // REQUIRES: x86-registered-target
   4 //
   5 // RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -emit-llvm %s -o - -std=c++20 -disable-llvm-passes -fsanitize-address-use-after-scope | FileCheck %s
   6
   7 #include "Inputs/coroutine.h"
   8
   9 struct Task {
  10   int value_;
  11   struct promise_type {
  12     Task get_return_object() {
  13       return Task{0};
  14     }
  15
  16     std::suspend_never initial_suspend() noexcept {
  17       return {};
  18     }
  19
  20     std::suspend_never final_suspend() noexcept {
  21       return {};
  22     }
  23
  24     void return_value(Task t) noexcept {}
  25     void unhandled_exception() noexcept {}
  26
  27     auto await_transform(Task t) {
  28       struct Suspension {
  29         auto await_ready() noexcept { return false;}
  30         auto await_suspend(std::coroutine_handle<> coro) {
  31           coro.destroy();
  32         }
  33
  34         auto await_resume() noexcept {
  35           return 0;
  36         }
  37       };
  38       return Suspension{};
  39     }
  40   };
  41 };
  42
  43 Task bar(bool cond) {
  44   co_return cond ? Task{ co_await Task{}}: Task{};
  45 }
  46
  47 void foo() {
  48   bar(true);
  49 }
  50
  51 // CHECK: cleanup.cont:{{.*}}
  52 // CHECK-NEXT: load i8
  53 // CHECK-NEXT: trunc
  54 // CHECK-NEXT: store i1 false
  55 // CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 8, ptr [[REF:%ref.tmp[0-9]+]])
  56
  57 // CHECK: await.suspend:{{.*}}
  58 // CHECK-NOT: call void @llvm.lifetime.start.p0(i64 8, ptr [[REF]])
  59 // CHECK: call void @_ZZN4Task12promise_type15await_transformES_EN10Suspension13await_suspendESt16coroutine_handleIvE
  60 // CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 8, ptr [[REF]])