-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy pathlua_tcp_handle.h
More file actions
93 lines (83 loc) · 3.6 KB
/
Copy pathlua_tcp_handle.h
File metadata and controls
93 lines (83 loc) · 3.6 KB
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
#ifndef LUA_TCP_HANDLE_H_
#define LUA_TCP_HANDLE_H_
#include "common.h"
#include "message.h"
#include "lua_handle_base.h"
#include "context_lua.h"
#include <queue>
class context_lua_t;
class uv_handle_base_t;
class uv_tcp_listen_handle_t;
class uv_tcp_socket_handle_t;
class lua_tcp_listen_handle_t : public lua_handle_base_t
{
public:
lua_tcp_listen_handle_t(uv_tcp_listen_handle_t* handle, lua_State* L)
: lua_handle_base_t((uv_handle_base_t*)handle, L) {}
~lua_tcp_listen_handle_t() {}
public:
static int32_t listen(lua_State* L);
static int32_t listen6(lua_State* L);
static int32_t listens(lua_State* L);
static int32_t accept(lua_State* L);
static int32_t close(lua_State* L);
private:
static int32_t _listen(lua_State* L, bool ipv6);
static int32_t listen_yield_finalize(lua_State *root_coro, lua_State *main_coro, void *userdata, uint32_t destination);
static int32_t accept_yield_finalize(lua_State *root_coro, lua_State *main_coro, void *userdata, uint32_t destination);
static int32_t listen_callback_adjust(lua_State* L);
static int32_t accept_callback_adjust(lua_State* L);
static int32_t listen_yield_continue(lua_State* L, int status, lua_KContext ctx);
static int32_t accept_yield_continue(lua_State* L, int status, lua_KContext ctx);
static int32_t accept_timeout(lua_State *L, int32_t session, void* userdata, bool is_repeat);
private:
ref_sessions_t m_accept_ref_sessions;
public:
static lua_tcp_listen_handle_t* create_tcp_listen_socket(uv_tcp_listen_handle_t* handle, lua_State* L);
static int32_t wakeup_listen(lua_State* L, message_t& message);
static int32_t wakeup_accept(lua_State* L, message_t& message);
};
class lua_tcp_socket_handle_t : public lua_handle_base_t
{
public:
lua_tcp_socket_handle_t(uv_tcp_socket_handle_t* handle, lua_State* L)
: lua_handle_base_t((uv_handle_base_t*)handle, L),
m_read_timeout_session_count(0) {}
~lua_tcp_socket_handle_t();
public:
static int32_t connect(lua_State* L);
static int32_t connect6(lua_State* L);
static int32_t connects(lua_State* L);
static int32_t write(lua_State* L);
static int32_t read(lua_State* L);
static int32_t set_rwopt(lua_State* L);
static int32_t get_rwopt(lua_State* L);
static int32_t set_nodelay(lua_State* L);
static int32_t set_wshared(lua_State* L);
static int32_t get_local_addr(lua_State* L);
static int32_t get_remote_addr(lua_State* L);
static int32_t get_local_port(lua_State* L);
static int32_t get_remote_port(lua_State* L);
static int32_t close(lua_State* L);
private:
static int32_t _connect(lua_State* L, bool ipv6);
static int32_t connect_yield_finalize(lua_State *root_coro, lua_State *main_coro, void *userdata, uint32_t destination);
static int32_t write_handle(lua_State* L);
static int32_t write_fd(lua_State* L);
static int32_t write_yield_finalize(lua_State *root_coro, lua_State *main_coro, void *userdata, uint32_t destination);
static int32_t read_yield_finalize(lua_State *root_coro, lua_State *main_coro, void *userdata, uint32_t destination);
static int32_t connect_callback_adjust(lua_State* L);
static int32_t connect_yield_continue(lua_State* L, int status, lua_KContext ctx);
static int32_t read_timeout(lua_State *L, int32_t session, void* userdata, bool is_repeat);
public:
static lua_tcp_socket_handle_t* create_tcp_socket(uv_tcp_socket_handle_t* handle, lua_State* L);
static int32_t wakeup_connect(lua_State* L, message_t& message);
static int32_t wakeup_read(lua_State* L, message_t& message);
private:
void release_read_overflow_buffers();
private:
int32_t m_read_timeout_session_count; //blocking
ref_sessions_t m_read_ref_sessions;
std::queue<buffer_t> m_read_overflow_buffers;
};
#endif