From 43ea5e861be8ce4914c53cc78a7c4e6c9deec28e Mon Sep 17 00:00:00 2001 From: rsiddharth Date: Thu, 21 Nov 2019 21:14:39 -0500 Subject: nserve: Update nserve. * nserver/src/nserve.c (nserve): Add command size check. --- nserver/src/nserve.c | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'nserver') 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. -- cgit v1.2.3