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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
|
#pragma once
// socket.hpp - Network event system.
//
// Copyright © ????-2004 Athena Dev Teams
// Copyright © 2004-2011 The Mana World Development Team
// Copyright © 2011-2014 Ben Longbons <b.r.longbons@gmail.com>
//
// This file is part of The Mana World (Athena server)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "fwd.hpp"
#include <sys/select.h>
#include <algorithm>
#include <memory>
#include "../strings/astring.hpp"
#include "../strings/vstring.hpp"
#include "../strings/xstring.hpp"
#include "../compat/iter.hpp"
#include "../compat/rawmem.hpp"
#include "../compat/time_t.hpp"
#include "../generic/dumb_ptr.hpp"
#include "../io/fd.hpp"
#include "ip.hpp"
#include "timer.t.hpp"
namespace tmwa
{
struct SessionData
{
};
struct SessionDeleter
{
// defined per-server
void operator()(SessionData *sd);
};
struct SessionIO
{
void (*func_recv)(Session *);
void (*func_send)(Session *);
};
struct SessionParsers
{
void (*func_parse)(Session *);
void (*func_delete)(Session *);
};
struct Session
{
Session(SessionIO, SessionParsers);
Session(Session&&) = delete;
Session& operator = (Session&&) = delete;
void set_io(SessionIO);
void set_parsers(SessionParsers);
/// Checks whether a newly-connected socket actually does anything
TimeT created;
TimeT last_tick;
bool connected;
private:
struct {
unsigned char eof : 1; /// Flag needed since structure must be freed in a server-dependent manner
unsigned char server : 1;
} flag;
public:
void set_eof() { flag.eof = 1; }
void set_server() { flag.server = 1; }
/// Currently used by clif_setwaitclose
Timer timed_close;
/// Since this is a single-threaded application, it can't block
/// These are the read/write queues
dumb_ptr<uint8_t[]> rdata, wdata;
size_t max_rdata, max_wdata;
/// How much is actually in the queue
size_t rdata_size, wdata_size;
/// How much has already been read from the queue
/// Note that there is no need for a wdata_pos
size_t rdata_pos;
IP4Address client_ip;
private:
/// Send or recieve
/// Only called when select() indicates the socket is ready
/// If, after that, nothing is read, it sets eof
// These could probably be hard-coded with a little work
void (*func_recv)(Session *);
void (*func_send)(Session *);
/// This is the important one
/// Set to different functions depending on whether the connection
/// is a player or a server/ladmin
void (*func_parse)(Session *);
/// Cleanup function since we're not fully RAII yet
void (*func_delete)(Session *);
public:
// this really ought to be part of session_data, once that gets sane
SessionParsers for_inferior;
/// Server-specific data type
// (this really should include the deleter, but ...)
std::unique_ptr<SessionData, SessionDeleter> session_data;
io::FD fd;
friend bool do_sendrecv(interval_t next);
friend bool do_parsepacket(void);
friend void delete_session(Session *);
};
inline
int convert_for_printf(Session *s)
{
return s->fd.uncast_dammit();
}
// save file descriptors for important stuff
constexpr int SOFT_LIMIT = FD_SETSIZE - 50;
// socket timeout to establish a full connection in seconds
constexpr int CONNECT_TIMEOUT = 15;
// socket timeout (no activity at all) in seconds
// the server will try to contact the session halfway through
constexpr int STALL_TIMEOUT = 300; // 5 minutes
void set_session(io::FD fd, std::unique_ptr<Session> sess);
Session *get_session(io::FD fd);
void reset_session(io::FD fd);
int get_fd_max();
class IncrFD
{
public:
static
io::FD inced(io::FD v)
{
return io::FD::cast_dammit(v.uncast_dammit() + 1);
}
};
IteratorPair<ValueIterator<io::FD, IncrFD>> iter_fds();
/// open a socket, bind, and listen. Return an fd, or -1 if socket() fails,
/// but exit if bind() or listen() fails
Session *make_listen_port(uint16_t port, SessionParsers inferior);
/// Connect to an address, return a connected socket or -1
// FIXME - this is IPv4 only!
Session *make_connection(IP4Address ip, uint16_t port, SessionParsers);
/// free() the structure and close() the fd
void delete_session(Session *);
/// Make a the internal queues bigger
void realloc_fifo(Session *s, size_t rfifo_size, size_t wfifo_size);
/// Update all sockets that can be read/written from the queues
bool do_sendrecv(interval_t next);
/// Call the parser function for every socket that has read data
bool do_parsepacket(void);
} // namespace tmwa
|