socket: Implement checked socket functions from LSB

Signed-off-by: Samuel Holland <samuel@sholland.org>

2 files changed, 32 insertions, 0 deletions

diff --git a/Makefile b/Makefile
index e377355..98f87ad 100644
--- a/Makefile
+++ b/Makefile
@@ -16,6 +16,7 @@ LIBGCOMPAT_SRC = \
 	libgcompat/resource.c		\
 	libgcompat/setjmp.c		\
 	libgcompat/signal.c		\
+	libgcompat/socket.c		\
 	libgcompat/stdio.c		\
 	libgcompat/stdlib.c		\
 	libgcompat/string.c		\
diff --git a/libgcompat/socket.c b/libgcompat/socket.c
new file mode 100644
index 0000000..927ebe4
--- /dev/null
+++ b/libgcompat/socket.c
@@ -0,0 +1,31 @@
+#include <assert.h>     /* assert */
+#include <stddef.h>     /* NULL */
+#include <sys/socket.h> /* recv, recvfrom */
+
+/**
+ * Receive a message from a connected socket, with buffer overflow checking.
+ *
+ * LSB 5.0: LSB-Core-generic/baselib---recv-chk-1.html
+ */
+ssize_t __recv_chk(int fd, void *buf, size_t len, size_t buflen, int flags)
+{
+	assert(buf != NULL);
+	assert(buflen >= len);
+
+	return recv(fd, buf, len, flags);
+}
+
+/**
+ * Receive a message from a socket, with buffer overflow checking.
+ *
+ * LSB 5.0: LSB-Core-generic/baselib---recvfrom-chk-1.html
+ */
+ssize_t __recvfrom_chk(int fd, void *buf, size_t len, size_t buflen, int flags,
+                       struct sockaddr *address, socklen_t *address_len)
+{
+	assert(buf != NULL);
+	assert(buflen >= len);
+	assert(address != NULL ? address_len != NULL : address_len == NULL);
+
+	return recvfrom(fd, buf, len, flags, address, address_len);
+}