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
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include "ifw_listen_dbus.h"
static int init(plugin_t *plugin, DBusConnection *connection);
static void deinit(plugin_t *plugin, DBusConnection *connection);
static void process_listen_message(plugin_t *plugin, DBusConnection *connection, int seq, msg_listen_t *listen);
static DBusHandlerResult handle_message(DBusConnection *connection, DBusMessage *message, plugin_t *plugin);
#ifdef IFW_LISTEN_FAKE
static void generate_fake_listen(msg_listen_t *listen);
static void handle_fake(plugin_t *plugin, DBusConnection *connection);
#else
static void handle_incoming(plugin_t *plugin, DBusConnection *connection);
#endif
static int init(plugin_t *plugin, DBusConnection *connection) {
ifw_listen_t *ifw_listen;
ifw_listen = malloc(sizeof(ifw_listen_t));
if (!ifw_listen) {
fprintf(stderr, "unable to malloc ifw\n");
return -1;
}
#ifdef IFW_LISTEN_FAKE
plugin->fd = 0;
#else
plugin->fd = nl_ifw_create_socket();
if (plugin->fd < 0) {
fprintf(stderr, "unable to init netlink\n");
return -1;
}
#endif
plugin->priv = (void *) ifw_listen;
return 0;
}
static void deinit(plugin_t *plugin, DBusConnection *connection) {
ifw_listen_t *ifw_listen = (ifw_listen_t *) plugin->priv;
close(plugin->fd);
}
#ifdef IFW_LISTEN_FAKE
#include <time.h>
static void generate_fake_listen(msg_listen_t *listen) {
static int port = 22;
strcpy(listen->program, "sshd");
listen->port = port++;
}
static void handle_fake(plugin_t *plugin, DBusConnection *connection) {
static int seq = 0;
msg_listen_t fake_listen;
read(0, NULL, 1);
generate_fake_listen(&fake_listen);
printf("seq : %d\n", seq);
process_listen_message(plugin, connection, seq++, &fake_listen);
}
#else
static void handle_incoming(plugin_t *plugin, DBusConnection *connection) {
struct nl_msg msg;
static int seq = 0;
msg_listen_t listen;
if (nl_ifw_read_msg(plugin->fd, NULL, &msg) <= 0) {
fprintf(stderr, "unable to read packet from netlink\n");
return;
}
process_listen_message(plugin, connection, seq++, &listen);
}
#endif
static void process_listen_message(plugin_t *plugin, DBusConnection *connection, int seq, msg_listen_t *listen) {
ifw_listen_t *ifw_listen = (ifw_listen_t *) plugin->priv;
ifw_listen_dbus_notify_listen(connection, listen);
}
static DBusHandlerResult handle_message(DBusConnection *connection, DBusMessage *message, plugin_t *plugin) {
ifw_listen_t *ifw_listen = (ifw_listen_t *) plugin->priv;
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
plugin_t ifw_listen_plugin = {
.name = "Interactive Firewall",
.path = IFW_LISTEN_DBUS_PATH,
.init = init,
#ifdef IFW_LISTEN_FAKE
.handle_incoming = handle_fake,
#else
.handle_incoming = handle_incoming,
#endif
.handle_message = handle_message,
.deinit = deinit,
};
|