feat: support set TOS in curl

alibaba · Feb 26, 2025 · 7fa477b · 7fa477b
1 parent e7d829c
commit 7fa477b
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/core/sdk/CurlImp.cpp b/core/sdk/CurlImp.cpp
@@ -13,11 +13,19 @@
 // limitations under the License.
 
 #include "CurlImp.h"
+
+#include <curl/curl.h>
+#include <netinet/in.h>
+#include <netinet/ip.h>
+#include <sys/socket.h>
+
 #include "Exception.h"
 #include "CurlAsynInstance.h"
 #include "DNSCache.h"
 #include "app_config/AppConfig.h"
-#include <curl/curl.h>
+#include "common/Flags.h"
+
+DEFINE_FLAG_INT32(sls_request_dscp, "set dscp for sls request, from 0 to 63", -1);
 
 using namespace std;
 
@@ -73,6 +81,13 @@ namespace sdk {
         return sizes;
     }
 
+    static size_t socket_write_callback(void* socketData, curl_socket_t fd, curlsocktype purpose) {
+        // TOS 8 bits: first 6 bits are DSCP (user customized), last 2 bits are ECN (auto set by OS)
+        int32_t tos = *static_cast<int32_t*>(socketData) << 2;
+        setsockopt(fd, IPPROTO_IP, IP_TOS, &tos, sizeof(tos));
+        return 0;
+    }
+
     CURL* PackCurlRequest(const std::string& httpMethod,
                           const std::string& host,
                           const int32_t port,
@@ -134,6 +149,13 @@ namespace sdk {
         curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, data_write_callback);
         curl_easy_setopt(curl, CURLOPT_HEADERDATA, &(httpMessage.header));
         curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, header_write_callback);
+
+        static int sDSCP
+            = INT32_FLAG(sls_request_dscp); // the lifestyle of sDSCP pointer should be longer than local variable
+        if (sDSCP >= 0 && sDSCP <= 63) {
+            curl_easy_setopt(curl, CURLOPT_SOCKOPTDATA, &sDSCP);
+            curl_easy_setopt(curl, CURLOPT_SOCKOPTFUNCTION, socket_write_callback);
+        }
         return curl;
     }