复现已有算法

2026-07-14 15:43:18 +08:00
parent abebd2a683
commit 50b8111fd9
860 changed files with 182250 additions and 18 deletions
@@ -0,0 +1,78 @@
+#include <drogon/Cookie.h>
+#include <drogon/drogon_test.h>
+#include <drogon/HttpAppFramework.h>
+#include <drogon/HttpClient.h>
+#include <drogon/HttpRequest.h>
+#include <drogon/HttpResponse.h>
+#include <drogon/HttpTypes.h>
+
+using namespace drogon;
+
+struct CookieSameSiteSequence
+{
+    CookieSameSiteSequence()
+    {
+        i = valid_sameSite_values.begin();
+        sessionCookie.setKey("JSESSIONID");
+    }
+
+    std::vector<std::string> valid_sameSite_values{"Null",
+                                                   "Lax",
+                                                   "None",
+                                                   "Strict"};
+
+    std::vector<std::string>::const_iterator i;
+    Cookie sessionCookie;
+};
+
+DROGON_TEST(CookieSameSite)
+{
+    auto client =
+        HttpClient::newHttpClient("https://127.0.0.1:8855",
+                                  HttpAppFramework::instance().getLoop(),
+                                  false,
+                                  false);
+    auto req = HttpRequest::newHttpRequest();
+
+    CookieSameSiteSequence seq;
+
+    while (seq.i != seq.valid_sameSite_values.end())
+    {
+        std::promise<void> p1;
+        std::future<void> f1 = p1.get_future();
+        req->setPath(std::string("/CookieSameSiteController/set/") + *seq.i);
+        if (seq.sessionCookie.getValue() != "")
+        {
+            // add session cookie
+            req->addCookie(seq.sessionCookie.key(), seq.sessionCookie.value());
+        }  // endif
+
+        client->sendRequest(
+            req,
+            [TEST_CTX, &seq, &p1](ReqResult res, const HttpResponsePtr &resp) {
+                REQUIRE(res == ReqResult::Ok);
+                REQUIRE(resp != nullptr);
+
+                CHECK(resp->getStatusCode() == HttpStatusCode::k200OK);
+                CHECK(resp->contentType() == CT_APPLICATION_JSON);
+
+                auto json = resp->getJsonObject();
+                auto cookie = resp->getCookie("JSESSIONID");
+
+                LOG_INFO << "Client: cookie-value == " << cookie.value()
+                         << ", requested value == " << (*seq.i)
+                         << ", new  value == "
+                         << (*json)["new value"].asString()
+                         << ", received value == "
+                         << Cookie::convertSameSite2String(
+                                cookie.getSameSite());
+                seq.sessionCookie = resp->getCookie("JSESSIONID");
+                CHECK(*seq.i ==
+                      Cookie::convertSameSite2String(cookie.getSameSite()));
+
+                p1.set_value();
+            });
+        f1.get();
+        seq.i++;
+    }
+}