xref: /llvm-project/clang/test/SemaCXX/warn-unsequenced-coro.cpp (revision 0b8daee028a87ab8a6f8fe54d2eb2d5b5c2babd4)

// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -fsyntax-only -verify -std=c++20 -I%S/Inputs -Wno-unused -Wno-uninitialized -Wunsequenced %s

// expected-no-diagnostics

#include "std-coroutine.h"

typedef __PTRDIFF_TYPE__ ptrdiff_t;

using namespace std;

template<class T>
struct Task {
    struct promise_type {
        Task<T> get_return_object() noexcept;
        suspend_always initial_suspend() noexcept;
        suspend_always final_suspend() noexcept;
        void return_value(T);
        void unhandled_exception();
        auto yield_value(Task<T>) noexcept { return final_suspend(); }
    };
    bool await_ready() noexcept { return false; }
    void await_suspend(coroutine_handle<>) noexcept {}
    T await_resume();
};

template<>
struct Task<void> {
    struct promise_type {
        Task<void> get_return_object() noexcept;
        suspend_always initial_suspend() noexcept;
        suspend_always final_suspend() noexcept;
        void return_void() noexcept;
        void unhandled_exception() noexcept;
        auto yield_value(Task<void>) noexcept { return final_suspend(); }
    };
    bool await_ready() noexcept { return false; }
    void await_suspend(coroutine_handle<>) noexcept {}
    void await_resume() noexcept {}
};

template <typename T>
class generator
{
  struct Promise
  {
    auto get_return_object() { return generator{*this}; }
    auto initial_suspend() { return suspend_never{}; }
    auto final_suspend() noexcept { return suspend_always{}; }
    void unhandled_exception() {}
    void return_void() {}

    auto yield_value(T value)
    {
      value_ = std::move(value);
      return suspend_always{};
    }

    T value_;
  };

  using Handle = coroutine_handle<Promise>;

  struct sentinel{};
  struct iterator
  {
    using iterator_category = input_iterator_tag;
    using value_type = T;
    using difference_type = ptrdiff_t;
    using reference = T &;
    using const_reference = const T &;
    using pointer = T *;

    iterator &operator++()
    {
      h_.resume();
      return *this;
    }
    const_reference &operator*() const { return h_.promise().value_; }
    bool operator!=(sentinel) { return !h_.done(); }

    Handle h_;
  };

  explicit generator(Promise &p) : h_(Handle::from_promise(p)) {}
  Handle h_;
public:
  using promise_type = Promise;
  auto begin() { return iterator{h_}; }
  auto end() { return sentinel{}; }
};

Task<void> c(int i) {
  co_await (i = 0, std::suspend_always{});
}

generator<int> range(int start, int end)
{
  while (start < end)
    co_yield start++;
}

Task<int> go(int const& val);
Task<int> go1(int x) {
  co_return co_await go(++x);
}