test/CodeGenCoroutines/pr59181.cpp

// Test for PR59181. Tests that no conditional cleanup is created around await_suspend.
//
// REQUIRES: x86-registered-target
//
// 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

#include "Inputs/coroutine.h"

struct Task {
  int value_;
  struct promise_type {
    Task get_return_object() {
      return Task{0};
    }

    std::suspend_never initial_suspend() noexcept {
      return {};
    }

    std::suspend_never final_suspend() noexcept {
      return {};
    }

    void return_value(Task t) noexcept {}
    void unhandled_exception() noexcept {}

    auto await_transform(Task t) {
      struct Suspension {
        auto await_ready() noexcept { return false;}
        auto await_suspend(std::coroutine_handle<> coro) {
          coro.destroy();
        }

        auto await_resume() noexcept {
          return 0;
        }
      };
      return Suspension{};
    }
  };
};

Task bar(bool cond) {
  co_return cond ? Task{ co_await Task{}}: Task{};
}

void foo() {
  bar(true);
}

// CHECK: cleanup.cont:{{.*}}
// CHECK-NEXT: load i8
// CHECK-NEXT: trunc
// CHECK-NEXT: store i1 false
// CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 8, ptr [[REF:%ref.tmp[0-9]+]])

// CHECK: await.suspend:{{.*}}
// CHECK-NOT: call void @llvm.lifetime.start.p0(i64 8, ptr [[REF]])
// CHECK: call void @_ZZN4Task12promise_type15await_transformES_EN10Suspension13await_suspendESt16coroutine_handleIvE
// CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 8, ptr [[REF]])