4 files changed, 32 insertions, 0 deletions

diff --git a/src/options.c b/src/options.c
index 90e6d2d..bbadec3 100644
--- a/src/options.c
+++ b/src/options.c
@@ -151,6 +151,26 @@ int opt_set_broadcast(lua_State *L, p_socket ps)
     return opt_setboolean(L, ps, SOL_SOCKET, SO_BROADCAST);
 }
 
+int opt_set_recv_buf_size(lua_State *L, p_socket ps)
+{
+        return opt_setint(L, ps, SOL_SOCKET, SO_RCVBUF);
+}
+
+int opt_get_recv_buf_size(lua_State *L, p_socket ps)
+{
+        return opt_getint(L, ps, SOL_SOCKET, SO_RCVBUF);
+}
+
+int opt_get_send_buf_size(lua_State *L, p_socket ps)
+{
+        return opt_getint(L, ps, SOL_SOCKET, SO_SNDBUF);
+}
+
+int opt_set_send_buf_size(lua_State *L, p_socket ps)
+{
+        return opt_setint(L, ps, SOL_SOCKET, SO_SNDBUF);
+}
+
 int opt_get_broadcast(lua_State *L, p_socket ps)
 {
     return opt_getboolean(L, ps, SOL_SOCKET, SO_BROADCAST);
diff --git a/src/options.h b/src/options.h
index 2ecc696..fb82c4d 100644
--- a/src/options.h
+++ b/src/options.h
@@ -46,6 +46,8 @@ int opt_set_ip6_multicast_loop(lua_State *L, p_socket ps);
 int opt_set_ip6_add_membership(lua_State *L, p_socket ps);
 int opt_set_ip6_drop_membersip(lua_State *L, p_socket ps);
 int opt_set_ip6_v6only(lua_State *L, p_socket ps);
+int opt_set_recv_buf_size(lua_State *L, p_socket ps);
+int opt_set_send_buf_size(lua_State *L, p_socket ps);
 
 /* supported options for getoption */
 int opt_get_dontroute(lua_State *L, p_socket ps);
@@ -72,6 +74,8 @@ int opt_get_ip6_multicast_hops(lua_State *L, p_socket ps);
 int opt_get_ip6_unicast_hops(lua_State *L, p_socket ps);
 int opt_get_ip6_v6only(lua_State *L, p_socket ps);
 int opt_get_reuseport(lua_State *L, p_socket ps);
+int opt_get_recv_buf_size(lua_State *L, p_socket ps);
+int opt_get_send_buf_size(lua_State *L, p_socket ps);
 
 /* invokes the appropriate option handler */
 int opt_meth_setoption(lua_State *L, p_opt opt, p_socket ps);
diff --git a/src/tcp.c b/src/tcp.c
index 79a680b..c7384b4 100644
--- a/src/tcp.c
+++ b/src/tcp.c
@@ -88,6 +88,8 @@ static t_opt optget[] = {
 #endif
     {"linger",      opt_get_linger},
     {"error",       opt_get_error},
+        {"recv-buffer-size",     opt_get_recv_buf_size},
+        {"send-buffer-size",     opt_get_send_buf_size},
     {NULL,          NULL}
 };
 
@@ -107,6 +109,8 @@ static t_opt optset[] = {
 #endif
     {"ipv6-v6only", opt_set_ip6_v6only},
     {"linger",      opt_set_linger},
+        {"recv-buffer-size",     opt_set_recv_buf_size},
+        {"send-buffer-size",     opt_set_send_buf_size},
     {NULL,          NULL}
 };
 
diff --git a/src/udp.c b/src/udp.c
index 605c195..037f9a4 100644
--- a/src/udp.c
+++ b/src/udp.c
@@ -88,6 +88,8 @@ static t_opt optset[] = {
     {"ipv6-add-membership",  opt_set_ip6_add_membership},
     {"ipv6-drop-membership", opt_set_ip6_drop_membersip},
     {"ipv6-v6only",          opt_set_ip6_v6only},
+        {"recv-buffer-size",     opt_set_recv_buf_size},
+        {"send-buffer-size",     opt_set_send_buf_size},
     {NULL,                   NULL}
 };
 
@@ -104,6 +106,8 @@ static t_opt optget[] = {
     {"ipv6-multicast-hops",  opt_get_ip6_unicast_hops},
     {"ipv6-multicast-loop",  opt_get_ip6_multicast_loop},
     {"ipv6-v6only",          opt_get_ip6_v6only},
+        {"recv-buffer-size",     opt_get_recv_buf_size},
+        {"send-buffer-size",     opt_get_send_buf_size},
     {NULL,                   NULL}
 };