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
|
/*
* Copyright (c) 2023-2024 Ian Marco Moffett and the Osmora Team.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of Hyra nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include <ostp/session.h>
#include <ostp/server.h>
#include <stdio.h>
#include <string.h>
#define OMOC_PORT 5352
static struct ostp_listener l;
static int
send_to(void *buf, size_t len, struct ostp_client *c)
{
struct ostp_session *s;
s = &c->session;
return send_frame(s->sockfd, buf, len, s->session_key);
}
static int
relay_others(struct ostp_client *sender, const void *buf, size_t len)
{
struct ostp_client *clients[MAX_CLIENTS];
char send_buf[4096];
int client_count;
client_count = listener_clients(&l, clients);
if (client_count < 0) {
return client_count;
}
memcpy(send_buf, buf, len);
printf("MESSAGE FROM SOCKFD: %d\n", sender->sockfd);
printf("SERVER SOCKFD: %d\n", l.serv_sockfd);
/* Relay to all *other* clients */
for (int i = 0; i < client_count; ++i) {
/* Do not relay to server or self */
if (clients[i]->sockfd == l.serv_sockfd) {
printf("SKIPPED SERVER SOCKET (%d)\n", i);
continue;
} else if (clients[i]->sockfd == sender->sockfd) {
printf("SKIPPED SENDER SOCKET (%d)\n", i);
continue;
}
printf("SEND TO CLIENT: %d\n", i);
session_send(send_buf, len, &clients[i]->session);
printf("SEND TO CLIENT DONE\n");
}
return 0;
}
static int
recv_msg(struct ostp_client *c, const void *buf, size_t len)
{
return relay_others(c, buf, len);
}
static int
on_conn(struct ostp_client *c)
{
struct ostp_session *s;
char buf[4096];
s = &c->session;
snprintf(buf, sizeof(buf), "[system]: %s has joined", s->username);
return relay_others(c, buf, sizeof(buf));
}
int
main(void)
{
int error;
listener_init(&l);
l.port = OMOC_PORT;
l.on_recv = recv_msg;
l.on_connect = on_conn;
/* Bind and poll the listener */
if ((error = listener_bind(&l)) < 0) {
return error;
}
if ((error = listener_poll(&l)) < 0) {
return error;
}
return 0;
}
|
