Input parser implementation requires a newline at end of input,
otherwise the last pattern may not be recognized correctly.

If input comes from a file, the culprit was YY_INPUT macro not expecting
the last line not ending with a newline, so the last word wasn't
accepted. This is easily fixed by checking for feof(yyin) in there. A
simple test case for that is:

| echo -en "table ip t {\nchain c {\n}\n}" >/tmp/foo
| nft -f /tmp/foo

Input from a string buffer is a bit more tricky: The culprit here is
that detection of classid pattern is done by checking the character
following it which makes it impossible to sit right at end of input and
I haven't found an alternative to that. After dropping the manual
newline appending when combining argv into a single buffer in main(),
a rule like this won't be recognized anymore:

| nft add rule ip t c meta priority feed:babe

Since a direct call to run_cmd_from_buffer() via libnftables bypasses
the sanitizing done in main() entirely, it has to happen in libnftables
instead which means creating a newline-terminated duplicate of the input
buffer.

Note that main() created a buffer one byte longer than needed since it
accounts for whitespace at end of each argv but doesn't add it to the
buffer for the last one, so buffer length is reduced by two bytes
instead of just one although only one less character is printed into it.

Signed-off-by: Phil Sutter <p...@nwl.cc>
---
 src/libnftables.c | 9 ++++++++-
 src/main.c        | 5 ++---
 src/scanner.l     | 2 +-
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/libnftables.c b/src/libnftables.c
index 6e271209d87ec..8bf989b08cc54 100644
--- a/src/libnftables.c
+++ b/src/libnftables.c
@@ -280,13 +280,19 @@ int nft_run_cmd_from_buffer(struct nft_ctx *nft, char 
*buf, size_t buflen)
        int rc = 0;
        struct parser_state state;
        LIST_HEAD(msgs);
+       size_t nlbuflen;
        void *scanner;
        FILE *fp;
+       char *nlbuf;
+
+       nlbuflen = max(buflen + 1, strlen(buf) + 2);
+       nlbuf = xzalloc(nlbuflen);
+       snprintf(nlbuf, nlbuflen, "%s\n", buf);
 
        parser_init(nft->nf_sock, &nft->cache, &state,
                    &msgs, nft->debug_mask, &nft->output);
        scanner = scanner_init(&state);
-       scanner_push_buffer(scanner, &indesc_cmdline, buf);
+       scanner_push_buffer(scanner, &indesc_cmdline, nlbuf);
 
        if (nft_run(nft, nft->nf_sock, scanner, &state, &msgs) != 0)
                rc = -1;
@@ -296,6 +302,7 @@ int nft_run_cmd_from_buffer(struct nft_ctx *nft, char *buf, 
size_t buflen)
        nft_ctx_set_output(nft, fp);
        scanner_destroy(scanner);
        iface_cache_release();
+       free(nlbuf);
 
        return rc;
 }
diff --git a/src/main.c b/src/main.c
index 353b87bc66631..1f08dfecc2606 100644
--- a/src/main.c
+++ b/src/main.c
@@ -264,14 +264,13 @@ int main(int argc, char * const *argv)
                for (len = 0, i = optind; i < argc; i++)
                        len += strlen(argv[i]) + strlen(" ");
 
-               buf = xzalloc(len + 2);
+               buf = xzalloc(len);
                for (i = optind; i < argc; i++) {
                        strcat(buf, argv[i]);
                        if (i + 1 < argc)
                                strcat(buf, " ");
                }
-               strcat(buf, "\n");
-               rc = !!nft_run_cmd_from_buffer(nft, buf, len + 2);
+               rc = !!nft_run_cmd_from_buffer(nft, buf, len);
        } else if (filename != NULL) {
                rc = !!nft_run_cmd_from_filename(nft, filename);
        } else if (interactive) {
diff --git a/src/scanner.l b/src/scanner.l
index 1d8e8ba0ecf42..d908a8fefc4f4 100644
--- a/src/scanner.l
+++ b/src/scanner.l
@@ -47,7 +47,7 @@
                errno = 0;                                                      
\
                clearerr(yyin);                                                 
\
        }                                                                       
\
-       if (result > 1) {                                                       
\
+       if (result > 1 && !feof(yyin)) {                                        
\
                while (result > 1 &&                                            
\
                       (buf[result - 1] != '\n' &&  buf[result - 1] != ' '))    
\
                        result--, n++;                                          
\
-- 
2.16.1

--
To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to