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
|
#include <session.h>
#include <server.h>
struct list_head sessions;
int next_id = 1;
int init_session(int socket) {
struct session *ses = malloc(sizeof(struct session));
if (ses == NULL)
return -1;
ses->id = next_id;
next_id++;
ses->socket = socket;
ses->alive = 1;
list_add(&ses->list, &sessions);
return ses->id;
}
struct session* find_session(int id) {
struct list_head *temp = sessions.next;
struct session *ses = NULL;
while (temp != NULL) {
ses = (struct session*)container_of(temp, struct session, list);
if (ses->id == id)
return ses;
temp = temp->next;
}
return NULL;
}
ssize_t write_session(int id, char *buffer, size_t sz) {
struct session *ses = find_session(id);
if (ses == NULL)
return -1;
return send(ses->socket, buffer, sz, 0);
}
ssize_t read_session(int id, char *buffer, size_t sz) {
struct session *ses = find_session(id);
if (ses == NULL)
return -1;
return recv(ses->socket, buffer, sz, 0);
}
void kill_session(int id) {
struct session *s = find_session(id);
s->alive = 0;
list_del(&s->list);
close(s->socket);
free(s);
}
int num_alive_sessions(void) {
int ret = 0;
struct list_head *temp = sessions.next;
while (temp != NULL) {
if (container_of(temp, struct session, list)->alive == 1)
ret++;
temp = temp->next;
}
return ret;
}
|
