-
Notifications
You must be signed in to change notification settings - Fork 248
/
TMainGreenlet.cpp
153 lines (126 loc) · 3.2 KB
/
TMainGreenlet.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
/* -*- indent-tabs-mode: nil; tab-width: 4; -*- */
/**
* Implementation of greenlet::MainGreenlet.
*
* Format with:
* clang-format -i --style=file src/greenlet/greenlet.c
*
*
* Fix missing braces with:
* clang-tidy src/greenlet/greenlet.c -fix -checks="readability-braces-around-statements"
*/
#ifndef T_MAIN_GREENLET_CPP
#define T_MAIN_GREENLET_CPP
#include "TGreenlet.hpp"
// Protected by the GIL. Incremented when we create a main greenlet,
// in a new thread, decremented when it is destroyed.
static Py_ssize_t G_TOTAL_MAIN_GREENLETS;
namespace greenlet {
greenlet::PythonAllocator<MainGreenlet> MainGreenlet::allocator;
void* MainGreenlet::operator new(size_t UNUSED(count))
{
return allocator.allocate(1);
}
void MainGreenlet::operator delete(void* ptr)
{
return allocator.deallocate(static_cast<MainGreenlet*>(ptr),
1);
}
MainGreenlet::MainGreenlet(PyGreenlet* p, ThreadState* state)
: Greenlet(p, StackState::make_main()),
_self(p),
_thread_state(state)
{
G_TOTAL_MAIN_GREENLETS++;
}
MainGreenlet::~MainGreenlet()
{
G_TOTAL_MAIN_GREENLETS--;
this->tp_clear();
}
ThreadState*
MainGreenlet::thread_state() const noexcept
{
return this->_thread_state;
}
void
MainGreenlet::thread_state(ThreadState* t) noexcept
{
assert(!t);
this->_thread_state = t;
}
const BorrowedMainGreenlet
MainGreenlet::main_greenlet() const
{
return this->_self;
}
BorrowedMainGreenlet
MainGreenlet::find_main_greenlet_in_lineage() const
{
return BorrowedMainGreenlet(this->_self);
}
bool
MainGreenlet::was_running_in_dead_thread() const noexcept
{
return !this->_thread_state;
}
OwnedObject
MainGreenlet::g_switch()
{
try {
this->check_switch_allowed();
}
catch (const PyErrOccurred&) {
this->release_args();
throw;
}
switchstack_result_t err = this->g_switchstack();
if (err.status < 0) {
// XXX: This code path is untested, but it is shared
// with the UserGreenlet path that is tested.
return this->on_switchstack_or_initialstub_failure(
this,
err,
true, // target was me
false // was initial stub
);
}
return err.the_new_current_greenlet->g_switch_finish(err);
}
int
MainGreenlet::tp_traverse(visitproc visit, void* arg)
{
if (this->_thread_state) {
// we've already traversed main, (self), don't do it again.
int result = this->_thread_state->tp_traverse(visit, arg, false);
if (result) {
return result;
}
}
return Greenlet::tp_traverse(visit, arg);
}
const OwnedObject&
MainGreenlet::run() const
{
throw AttributeError("Main greenlets do not have a run attribute.");
}
void
MainGreenlet::run(const BorrowedObject UNUSED(nrun))
{
throw AttributeError("Main greenlets do not have a run attribute.");
}
void
MainGreenlet::parent(const BorrowedObject raw_new_parent)
{
if (!raw_new_parent) {
throw AttributeError("can't delete attribute");
}
throw AttributeError("cannot set the parent of a main greenlet");
}
const OwnedGreenlet
MainGreenlet::parent() const
{
return OwnedGreenlet(); // null becomes None
}
}; // namespace greenlet
#endif