forked from jbenc/plotnetcfg
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.c
126 lines (113 loc) · 3.07 KB
/
main.c
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
/*
* This file is a part of plotnetcfg, a tool to visualize network config.
* Copyright (C) 2014 Red Hat, Inc. -- Jiri Benc <jbenc@redhat.com>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*/
#include <linux/capability.h>
#include <net/if.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/syscall.h>
#include <unistd.h>
#include "args.h"
#include "netns.h"
#include "utils.h"
#include "version.h"
#include "frontend.h"
#include "frontends/dot.h"
#include "frontends/json.h"
#include "handler.h"
#include "handlers/bridge.h"
#include "handlers/master.h"
#include "handlers/openvswitch.h"
#include "handlers/veth.h"
#include "handlers/vlan.h"
static void register_frontends(void)
{
frontend_init();
frontend_dot_register();
frontend_json_register();
}
static void register_handlers(void)
{
handler_master_register();
handler_ovs_register();
handler_veth_register();
handler_vlan_register();
handler_bridge_register();
}
static int print_help(_unused char *arg)
{
printf("Usage: plotnetcfg [OPTION]...\n\n");
arg_get_help((arg_help_handler_t)puts);
return 1;
}
static int print_version(_unused char *arg)
{
printf("%s\n", VERSION);
return 1;
}
static struct arg_option options[] = {
{ .long_name = "help", .short_name = 'h',
.type = ARG_CALLBACK, .action.callback = print_help,
.help = "print help and exit",
},
{ .long_name = "version", .short_name = '\0',
.type = ARG_CALLBACK, .action.callback = print_version,
.help = "print version and exit",
},
};
static int check_caps(void)
{
struct __user_cap_header_struct caps_hdr;
struct __user_cap_data_struct caps;
caps_hdr.version = _LINUX_CAPABILITY_VERSION_1;
caps_hdr.pid = 0;
if (syscall(__NR_capget, &caps_hdr, &caps) < 0)
return 0;
if (!(caps.effective & (1U << CAP_SYS_ADMIN)) ||
!(caps.effective & (1U << CAP_NET_ADMIN)))
return 0;
return 1;
}
int main(int argc, char **argv)
{
struct netns_entry *root;
int netns_ok, err;
arg_register_batch(options, ARRAY_SIZE(options));
register_frontends();
register_handlers();
if ((err = arg_parse(argc, argv)))
exit(err);
if (!check_caps()) {
fprintf(stderr, "Must be run under root (or with enough capabilities).\n");
exit(1);
}
netns_ok = netns_switch_root();
if (netns_ok > 0) {
fprintf(stderr, "Cannot change to the root name space: %s\n", strerror(netns_ok));
exit(1);
}
global_handler_init();
if ((err = netns_list(&root, netns_ok == 0))) {
fprintf(stderr, "ERROR: %s\n", strerror(err));
exit(1);
}
if ((err = frontend_output(root))) {
fprintf(stderr, "Invalid output format specified.\n");
exit(1);
}
global_handler_cleanup(root);
netns_list_free(root);
return 0;
}