blob: 70a1cf1ab8e611b915ceec7ed2f027e7d8ff72ee (
plain)
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
|
#include <session.h>
#include <stdlib.h>
#include <unistd.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)
break;
temp = temp->next;
}
return ses;
}
int write_session(int id, const char *data, size_t size) {
struct session *ses = find_session(id);
return write(ses->socket, data, size);
}
size_t read_session(int id, char *data, size_t size) {
struct session *ses = find_session(id);
if (ses == NULL)
return -1;
return read(ses->socket, data, size);
}
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;
}
|