87 lines
2.8 KiB
Diff
87 lines
2.8 KiB
Diff
From 5e4b2366417b5f951ffd5ee45c0917ae67bd502e Mon Sep 17 00:00:00 2001
|
|
From: hantwofish <hankangkang5@huawei.com>
|
|
Date: Sun, 3 Mar 2024 21:40:23 +0800
|
|
Subject: [PATCH] add limit with level for sockopt
|
|
|
|
---
|
|
src/lstack/api/lstack_wrap.c | 49 +++++++++++++++++++++++++-----------
|
|
2 files changed, 35 insertions(+), 16 deletions(-)
|
|
|
|
diff --git a/src/lstack/api/lstack_wrap.c b/src/lstack/api/lstack_wrap.c
|
|
index 1e36d3d..49bbf91 100644
|
|
--- a/src/lstack/api/lstack_wrap.c
|
|
+++ b/src/lstack/api/lstack_wrap.c
|
|
@@ -364,27 +364,46 @@ static inline int32_t do_getsockname(int32_t s, struct sockaddr *name, socklen_t
|
|
return posix_api->getsockname_fn(s, name, namelen);
|
|
}
|
|
|
|
-static bool unsupport_optname(int32_t optname)
|
|
-{
|
|
- if (optname == SO_BROADCAST ||
|
|
- optname == SO_PROTOCOL ||
|
|
- optname == TCP_QUICKACK ||
|
|
- optname == SO_SNDTIMEO ||
|
|
- optname == SO_RCVTIMEO ||
|
|
- optname == SO_SNDBUF ||
|
|
- optname == SO_RCVBUF ||
|
|
- optname == TCP_INFO ||
|
|
- optname == TCP_MAXSEG ||
|
|
- optname == SO_DONTROUTE ||
|
|
- optname == TCP_CONGESTION) {
|
|
+static bool unsupport_tcp_optname(int32_t optname)
|
|
+{
|
|
+ if ((optname == TCP_QUICKACK) ||
|
|
+ (optname == TCP_INFO) ||
|
|
+ (optname == TCP_MAXSEG) ||
|
|
+ (optname == TCP_CONGESTION)) {
|
|
+ return true;
|
|
+ }
|
|
+ return false;
|
|
+}
|
|
+
|
|
+static bool unsupport_socket_optname(int32_t optname)
|
|
+{
|
|
+ if ((optname == SO_BROADCAST) ||
|
|
+ (optname == SO_PROTOCOL) ||
|
|
+ (optname == SO_SNDTIMEO) ||
|
|
+ (optname == SO_RCVTIMEO) ||
|
|
+ (optname == SO_SNDBUF) ||
|
|
+ (optname == SO_RCVBUF) ||
|
|
+ (optname == SO_DONTROUTE)) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
+static bool unsupport_optname(int32_t level, int32_t optname)
|
|
+{
|
|
+ if (level == SOL_TCP) {
|
|
+ return unsupport_tcp_optname(optname);
|
|
+ }
|
|
+
|
|
+ if (level == SOL_SOCKET) {
|
|
+ return unsupport_socket_optname(optname);
|
|
+ }
|
|
+ return false;
|
|
+}
|
|
+
|
|
static inline int32_t do_getsockopt(int32_t s, int32_t level, int32_t optname, void *optval, socklen_t *optlen)
|
|
{
|
|
- if (select_fd_posix_path(s, NULL) == PATH_LWIP && !unsupport_optname(optname)) {
|
|
+ if (select_fd_posix_path(s, NULL) == PATH_LWIP && !unsupport_optname(level, optname)) {
|
|
return g_wrap_api->getsockopt_fn(s, level, optname, optval, optlen);
|
|
}
|
|
|
|
@@ -393,7 +412,7 @@ static inline int32_t do_getsockopt(int32_t s, int32_t level, int32_t optname, v
|
|
|
|
static inline int32_t do_setsockopt(int32_t s, int32_t level, int32_t optname, const void *optval, socklen_t optlen)
|
|
{
|
|
- if (select_fd_posix_path(s, NULL) == PATH_KERNEL || unsupport_optname(optname)) {
|
|
+ if (select_fd_posix_path(s, NULL) == PATH_KERNEL || unsupport_optname(level, optname)) {
|
|
return posix_api->setsockopt_fn(s, level, optname, optval, optlen);
|
|
}
|
|
|
|
--
|
|
2.27.0
|
|
|