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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
/*
* (C) 2014 by Pablo Neira Ayuso <pablo@netfilter.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* Thanks to the NLnet Foundation <http://nlnet.nl> for making the bootstrap
* of this project possible!
*/
#include <stdio.h>
#include <stdlib.h>
#include <fd.h>
#include <tcp.h>
#include <unistd.h>
#include <config.h>
#include <string.h>
#include <getopt.h>
#include <errno.h>
#include "init.h"
#include "logging.h"
#include "msg_buff.h"
#include "proto.h"
struct nft_sync_inst nfts_inst;
static void print_usage(const char *prog_name)
{
fprintf(stderr,
"%s (c) 2014 by Pablo Neira Ayuso <pablo@netfilter.org>\n"
"Usage: %s [-h] [-c]\n"
" [ --help ]\n"
" [ --config=<FILE> ]\n"
" [ --fetch ]\n", prog_name, prog_name);
}
static const struct option options[] = {
{ .name = "help", .has_arg = false, .val = 'h' },
{ .name = "config", .has_arg = false, .val = 'c' },
{ .name = "fetch", .has_arg = false, .val = 'f' },
{ NULL },
};
#define NFT_SYNC_CONF_DEFAULT "/etc/nft-sync.conf"
static int set_cmd(int cmd)
{
if (nfts_inst.cmd) {
fprintf(stderr,
"Cannot specify multiple commands at the same time\n");
return -1;
}
nfts_inst.cmd = cmd;
return 0;
}
int main(int argc, char *argv[])
{
int ret = EXIT_FAILURE, c;
const char *config = NFT_SYNC_CONF_DEFAULT;
while ((c = getopt_long(argc, argv, "hc:f", options, NULL)) != -1) {
switch (c) {
case 'h':
print_usage(argv[0]);
return EXIT_SUCCESS;
case 'c':
config = optarg;
break;
case 'f':
set_cmd(NFTS_CMD_FETCH);
break;
default:
fprintf(stderr, "Unknown option -%c\n", c);
return EXIT_FAILURE;
}
}
if (nft_sync_config_parse(config) < 0)
return EXIT_FAILURE;
if (nft_sync_event_init() < 0) {
fprintf(stderr, "Cannot start libev: %s\n", strerror(errno));
goto err;
}
if (nft_sync_log_init(&nfts_inst) < 0) {
fprintf(stderr, "Cannot start logging: %s\n", strerror(errno));
goto err;
}
if (nfts_inst.mode & NFTS_MODE_SERVER) {
if (tcp_server_start(&nfts_inst) < 0) {
nfts_log(NFTS_LOG_FATAL,
"Cannot start TCP server: %s\n",
strerror(errno));
goto err;
}
nfts_log(NFTS_LOG_INFO, "listening at %s",
inet_ntoa(nfts_inst.tcp.server.ipv4.inet_addr));
}
if (nfts_inst.mode & NFTS_MODE_CLIENT) {
if (!nfts_inst.cmd) {
nfts_log(NFTS_LOG_FATAL,
"Client needs some command, eg. --fetch",
strerror(errno));
goto err;
}
if (tcp_client_start(&nfts_inst) < 0) {
nfts_log(NFTS_LOG_FATAL,
"Cannot start TCP client: %s",
strerror(errno));
goto err;
}
nfts_log(NFTS_LOG_INFO, "connecting to %s",
inet_ntoa(nfts_inst.tcp.client.inet_addr));
}
/* TODO: add switch to allow to daemonize this process */
nft_sync_event_loop();
nft_sync_event_fini();
ret = EXIT_SUCCESS;
err:
nft_sync_log_fini(&nfts_inst);
return ret;
}
|
