make sure table arg is considered

This commit is contained in:
prx 2022-10-12 22:01:34 +02:00
parent e364ce9780
commit 0265627230
1 changed files with 9 additions and 8 deletions

17
main.c
View File

@ -67,6 +67,15 @@ main(int argc, char *argv[])
int kq = 0;
socklen_t sin_size = 0;
int s[MAXSOCK] = {0};
struct kevent ev[MAXSOCK] = {0};
struct addrinfo hints, *servinfo, *res;
struct sockaddr_storage client_addr;
if (argc > 2)
usage();
else if (argc == 2)
table = argv[1];
const char *bancmd[] = { "/usr/bin/doas", "-n",
"/sbin/pfctl", "-t", table,
"-T", "add", ip,
@ -75,9 +84,6 @@ main(int argc, char *argv[])
"/sbin/pfctl",
"-k", ip,
NULL };
struct kevent ev[MAXSOCK] = {0};
struct addrinfo hints, *servinfo, *res;
struct sockaddr_storage client_addr;
/* safety first */
if (unveil("/usr/bin/doas", "rx") != 0)
@ -88,11 +94,6 @@ main(int argc, char *argv[])
if (pledge("stdio inet exec proc rpath", NULL) != 0)
err(1, "pledge");
if (argc > 2)
usage();
else if (argc == 2)
table = argv[1];
/* initialize structures */
memset(&client_addr, 0, sizeof(client_addr));
memset(&hints, 0, sizeof(hints));