Organize code.
This commit is contained in:
@@ -12,6 +12,7 @@ const struct option opts[] =
|
|||||||
{ "verbose", required_argument, NULL, 'v' },
|
{ "verbose", required_argument, NULL, 'v' },
|
||||||
{ "log-file", required_argument, NULL, 0 },
|
{ "log-file", required_argument, NULL, 0 },
|
||||||
{ "interface", required_argument, NULL, 'i' },
|
{ "interface", required_argument, NULL, 'i' },
|
||||||
|
{ "pin-maps", required_argument, NULL, 'p' },
|
||||||
{ "update-time", required_argument, NULL, 'u' },
|
{ "update-time", required_argument, NULL, 'u' },
|
||||||
{ "no-stats", required_argument, NULL, 'n' },
|
{ "no-stats", required_argument, NULL, 'n' },
|
||||||
{ "stats-ps", required_argument, NULL, 1 },
|
{ "stats-ps", required_argument, NULL, 1 },
|
||||||
@@ -31,7 +32,7 @@ void ParseCommandLine(cmdline_t *cmd, int argc, char *argv[])
|
|||||||
{
|
{
|
||||||
int c;
|
int c;
|
||||||
|
|
||||||
while ((c = getopt_long(argc, argv, "c:ost:lhv:i:u:n:", opts, NULL)) != -1)
|
while ((c = getopt_long(argc, argv, "c:ost:lhv:i:p:u:n:", opts, NULL)) != -1)
|
||||||
{
|
{
|
||||||
switch (c)
|
switch (c)
|
||||||
{
|
{
|
||||||
@@ -80,6 +81,11 @@ void ParseCommandLine(cmdline_t *cmd, int argc, char *argv[])
|
|||||||
|
|
||||||
break;
|
break;
|
||||||
|
|
||||||
|
case 'p':
|
||||||
|
cmd->pin_maps = atoi(optarg);
|
||||||
|
|
||||||
|
break;
|
||||||
|
|
||||||
case 'u':
|
case 'u':
|
||||||
cmd->update_time = atoi(optarg);
|
cmd->update_time = atoi(optarg);
|
||||||
|
|
||||||
|
|||||||
@@ -113,7 +113,7 @@ void PrintToolInfo()
|
|||||||
/**
|
/**
|
||||||
* Retrieves nanoseconds since system boot.
|
* Retrieves nanoseconds since system boot.
|
||||||
*
|
*
|
||||||
* @return The current nanoseconds since the system started.
|
* @return The current nanoseconds since the system last booted.
|
||||||
*/
|
*/
|
||||||
u64 GetBootNanoTime()
|
u64 GetBootNanoTime()
|
||||||
{
|
{
|
||||||
|
|||||||
@@ -157,6 +157,8 @@ int main(int argc, char *argv[])
|
|||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
printf("Using 'map_range_drop' FD => %d.\n", map_range_drop);
|
||||||
|
|
||||||
// Parse IP range.
|
// Parse IP range.
|
||||||
ip_range_t range = ParseIpCidr(cmd.ip);
|
ip_range_t range = ParseIpCidr(cmd.ip);
|
||||||
|
|
||||||
@@ -168,7 +170,7 @@ int main(int argc, char *argv[])
|
|||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
printf("Removed IP range '%s' from BPF map.\n", cmd.ip);
|
printf("Removed IP range '%s'...\n", cmd.ip);
|
||||||
|
|
||||||
if (cmd.save)
|
if (cmd.save)
|
||||||
{
|
{
|
||||||
@@ -216,6 +218,8 @@ int main(int argc, char *argv[])
|
|||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
printf("Using 'map_block6' FD => %d.\n", map_block6);
|
||||||
|
|
||||||
struct in6_addr addr;
|
struct in6_addr addr;
|
||||||
|
|
||||||
if ((ret = inet_pton(AF_INET6, cmd.ip, &addr)) != 1)
|
if ((ret = inet_pton(AF_INET6, cmd.ip, &addr)) != 1)
|
||||||
@@ -248,6 +252,8 @@ int main(int argc, char *argv[])
|
|||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
printf("Using 'map_block' FD => %d.\n", map_block);
|
||||||
|
|
||||||
struct in_addr addr;
|
struct in_addr addr;
|
||||||
|
|
||||||
if ((ret = inet_pton(AF_INET, cmd.ip, &addr)) != 1)
|
if ((ret = inet_pton(AF_INET, cmd.ip, &addr)) != 1)
|
||||||
@@ -264,7 +270,7 @@ int main(int argc, char *argv[])
|
|||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
printf("Deleted '%s' from block map...\n", cmd.ip);
|
printf("Deleted IP '%s'...\n", cmd.ip);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user