int CNetConsole::Update()
{
NETSOCKET Socket;
NETADDR Addr;
if(net_tcp_accept(m_Socket, &Socket, &Addr) > 0)
{
// check if we just should drop the packet
char aBuf[128];
if(NetBan() && NetBan()->IsBanned(&Addr, aBuf, sizeof(aBuf)))
{
// banned, reply with a message and drop
net_tcp_send(Socket, aBuf, str_length(aBuf));
net_tcp_close(Socket);
}
else
AcceptClient(Socket, &Addr);
}
for(int i = 0; i < NET_MAX_CONSOLE_CLIENTS; i++)
{
if(m_aSlots[i].m_Connection.State() == NET_CONNSTATE_ONLINE)
m_aSlots[i].m_Connection.Update();
if(m_aSlots[i].m_Connection.State() == NET_CONNSTATE_ERROR)
Drop(i, m_aSlots[i].m_Connection.ErrorString());
}
return 0;
}
static void tcp_accept_handler(event_loop el, int fd, int mask, void *data) {
char cip[128];
int cport, cfd;
client c;
NOT_USED(el);
NOT_USED(mask);
NOT_USED(data);
if ((cfd = net_tcp_accept(fd, cip, &cport, neterr, sizeof neterr)) == -1) {
xcb_log(XCB_LOG_WARNING, "Accepting client connection: %s", neterr);
return;
}
if ((c = client_new(cfd, fd)) == NULL) {
xcb_log(XCB_LOG_WARNING, "Error registering fd '%d' event for the new client: %s",
cfd, strerror(errno));
close(cfd);
} else {
/* heartbeat */
dstr res = dstr_new("HEARTBEAT|");
dstr ip = getipv4();
xcb_log(XCB_LOG_NOTICE, "Accepted %s:%d, client '%p'", cip, cport, c);
res = dstr_cat(res, ip);
res = dstr_cat(res, "\r\n");
pthread_spin_lock(&c->lock);
if (net_try_write(c->fd, res, dstr_length(res), 100, NET_NONBLOCK) == -1)
xcb_log(XCB_LOG_WARNING, "Writing to client '%p': %s", c, strerror(errno));
pthread_spin_unlock(&c->lock);
dstr_free(ip);
dstr_free(res);
}
}
static int accept_tcp_handler(evutil_socket_t fd, short event, void *args)
{
int cport, cfd;
char cip[IP_STR_LEN];
(void) event;
(void) args;
cfd = net_tcp_accept(server.neterr, fd, cip, sizeof(cip), &cport);
if (cfd == -1) {
srv_log(LOG_WARN, "Accepting client connection: %s", server.neterr);
return -1;
}
srv_log(LOG_INFO, "Accepted %s:%d", cip, cport);
return cfd;
}
