CF-Connecting-IP

mattn · Aug 25, 2024 · d6a6bbd · d6a6bbd
1 parent bae3e91
commit d6a6bbd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/main.cxx b/main.cxx
@@ -390,14 +390,19 @@ static const std::string realIp(ws28::HTTPRequest &req) {
     value = req.headers.Get("X-Real-Ip");
     if (value.has_value()) {
       ip = value.value();
+    } else {
+      value = req.headers.Get("CF-Connecting-IP");
+      if (value.has_value()) {
+        ip = value.value();
+      }
     }
   }
   return ip;
 }
 
 static void connect_callback(ws28::Client *client, ws28::HTTPRequest &req) {
   auto ip = realIp(req);
-  char *p = new char[ip.length()+1];
+  char *p = new char[ip.length() + 1];
   std::strcpy(p, ip.c_str());
   client->SetUserData(p);
   spdlog::debug("CONNECTED {}", ip);