diff options
Diffstat (limited to 'nserver')
-rw-r--r-- | nserver/src/nserve.c | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/nserver/src/nserve.c b/nserver/src/nserve.c index 18a975f..72a97be 100644 --- a/nserver/src/nserve.c +++ b/nserver/src/nserve.c @@ -54,8 +54,16 @@ void nserve(int sock) ssize_t bytes = slurpsock(buf, buf_sz, sock); check(bytes >= 0, "nserve: slurpsock failed"); + int rc = check_cmd_size(buf, sock); + check(rc != -1, "command size check failed"); + + // Quit immediately if cmd size is too large. + if (rc == 1) { + goto error; + } + // Write response to socket. - int rc = barfsock(buf, bytes, sock); + rc = barfsock(buf, bytes, sock); check(rc != -1, "nserve: echo failed"); // Close socket. |