diff options
Diffstat (limited to 'debug/recv_chk.c')
-rw-r--r-- | debug/recv_chk.c | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/debug/recv_chk.c b/debug/recv_chk.c new file mode 100644 index 0000000000..7a49d17234 --- /dev/null +++ b/debug/recv_chk.c @@ -0,0 +1,33 @@ +/* Copyright (C) 2005 Free Software Foundation, Inc. + This file is part of the GNU C Library. + + The GNU C Library is free software; you can redistribute it and/or + modify it under the terms of the GNU Lesser General Public + License as published by the Free Software Foundation; either + version 2.1 of the License, or (at your option) any later version. + + The GNU C Library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Lesser General Public License for more details. + + You should have received a copy of the GNU Lesser General Public + License along with the GNU C Library; if not, write to the Free + Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA + 02111-1307 USA. */ + +#include <sys/param.h> +#include <sys/socket.h> + + +ssize_t +__recv_chk (int fd, void *buf, size_t n, size_t buflen, int flags) +{ + /* In case N is greater than BUFLEN, we read BUFLEN+1 bytes. + This might overflow the buffer but the damage is reduced to just + one byte. And the program will terminate right away. */ + ssize_t nrecv = __recv (fd, buf, MIN (n, buflen + 1), flags); + if (nrecv > 0 && (size_t) nrecv > buflen) + __chk_fail (); + return nrecv; +} |