Mirror-bot/AnyPS5
1
0 Fork 0
mirror of https://github.com/boykopovar/AnyPS5.git synced 2026-10-28 12:05:06 +00:00
Code Issues Projects Releases Activity
Files
main
AnyPS5/core/libs/tests/GuestHttp2Cookies.cpp
T

196 lines
10 KiB
C++
Raw Permalink Blame History

#include "prx/libSceHttp2/include/Http2.hpp"
#include <cstddef>
#include <cstdint>
#include <cstdlib>
#include <cstring>
#include <string>
static void Require(bool value) { if (!value) std::abort(); }
static constexpr int NotFound = static_cast<int>(0x817B1025);
static constexpr int InvalidId = static_cast<int>(0x817B1100);
static constexpr int OutOfSize = static_cast<int>(0x817B1104);
static constexpr int InvalidValue = static_cast<int>(0x817B11FE);
static constexpr int PoolTooSmall = static_cast<int>(0x817B121D);
static constexpr int Rejected = static_cast<int>(0x817B1219);
static int Add(int box, const char* url, const std::string& cookie) {
return sceHttp2AddCookie(box, url, cookie.c_str(), cookie.size());
}
static std::string Get(int box, const char* url) {
std::size_t required = 0;
Require(sceHttp2GetCookie(box, url, nullptr, &required, 0) == 0 && required >= 3);
std::string header(required, '\x7f');
std::size_t again = 0;
Require(sceHttp2GetCookie(box, url, header.data(), &again, required) == 0 && again == required);
Require(header.compare(required - 3, 3, std::string("\0 \0", 3)) == 0);
header.resize(required - 3);
return header;
}
static bool NoMatch(int box, const char* url) {
std::size_t required = 7;
char buffer[16] = {'x'};
return sceHttp2GetCookie(box, url, nullptr, &required, 0) == NotFound && required == 7 &&
sceHttp2GetCookie(box, url, buffer, &required, sizeof(buffer)) == NotFound && required == 7 && buffer[0] == 'x';
}
static Http2CookieStats Stats(int box) {
alignas(8) unsigned char raw[40];
std::memset(raw, 0xff, sizeof(raw));
Require(sceHttp2GetCookieStats(box, reinterpret_cast<Http2CookieStats*>(raw)) == 0);
for (std::size_t i = 12; i < 16; ++i) Require(raw[i] == 0xff);
for (std::size_t i = 32; i < 40; ++i) Require(raw[i] == 0xff);
Http2CookieStats stats{};
std::memcpy(&stats, raw, sizeof(stats));
return stats;
}
static int FillDomain(int box, int count) {
for (int i = 0; i < count; ++i) {
if (Add(box, "http://example.com/", "c" + std::to_string(i) + "=1") != 0) return -1;
}
return 0;
}
int main() {
static_assert(sizeof(Http2CookieStats) == 32);
static_assert(offsetof(Http2CookieStats, currentInuseNum) == 8 && offsetof(Http2CookieStats, maxInuseSize) == 16);
static_assert(offsetof(Http2CookieStats, maxInuseNum) == 24 && offsetof(Http2CookieStats, removedNum) == 28);
constexpr std::size_t maxSize = static_cast<std::size_t>(-1);
Require(sceHttp2CreateCookieBox(maxSize) < 0);
Require(sceHttp2CreateCookieBox(maxSize - 0x3fff) < 0);
Require(sceHttp2CreateCookieBox(maxSize / 2) < 0);
Require(sceHttp2CreateCookieBox(0xffff) == PoolTooSmall);
const int box = sceHttp2CreateCookieBox(0x10000);
Require(box > 0);
Require(NoMatch(box, "http://example.com/"));
Require(Add(box, "http://example.com/a/b", "sid=1") == 0);
Require(Add(box, "http://Example.COM:8080/", "lang=fr; Path=/; Domain=.example.com") == 0);
Require(Add(box, "https://example.com/", "token=x; Secure; HttpOnly; Path=/") == 0);
Require(Add(box, "http://example.com/", "wide=1; Path=/a; Max-Age=3600") == 0);
Require(Get(box, "http://example.com/a/c") == "sid=1; wide=1; lang=fr");
Require(Get(box, "https://example.com/a/b?q#f") == "sid=1; wide=1; token=x; lang=fr");
Require(Get(box, "HTTPS://example.com/") == "token=x; lang=fr");
Require(Get(box, "http://example.com/") == "lang=fr");
Require(Get(box, "http://example.com/a") == "wide=1; lang=fr");
Require(Get(box, "http://example.com/ab") == "wide=1; lang=fr");
Require(Get(box, "http://www.example.com/a/") == "sid=1; wide=1; lang=fr");
Require(Get(box, "ftp://example.com/a/b") == "sid=1; wide=1; lang=fr");
Require(Get(box, "example.com/a/b") == "sid=1; wide=1; lang=fr");
Require(Get(box, "http://user:[email protected]/a/x") == "sid=1; wide=1; lang=fr");
Require(NoMatch(box, "http://badexample.com/a"));
Require(NoMatch(box, "http:///a"));
Require(NoMatch(box, ""));
Require(Add(box, "http://example.com/a/b", "sid=2") == 0);
Require(Get(box, "http://example.com/a/") == "sid=2; wide=1; lang=fr");
Require(Add(box, "http://example.com/", "wide=0; Path=/a; Expires=Thu, 01 Jan 1970 00:00:00 GMT") == 0);
Require(Add(box, "http://example.com/", "lang=; Domain=example.com; Max-Age=0") == 0);
Require(Get(box, "http://example.com/a/") == "sid=2");
Require(Add(box, "http://example.com/", "future=1; Expires=Wed, 09 Jun 2121 10:18:14 GMT") == 0);
Require(Add(box, "http://example.com/", "invalid=1; Expires=Feb 30 2121 10:18:14") == 0);
Require(Get(box, "http://example.com/") == "invalid=1; future=1");
Require(Add(box, "http://example.com/", "noequals") == 0);
Require(Add(box, "http://example.com/", "=value") == 0);
Require(Add(box, "http://example.com/", "evil=1; Domain=other.com") == Rejected);
Require(Add(box, "http://example.com/", "evil=1; Domain=ample.com") == Rejected);
Require(NoMatch(box, "http://other.com/"));
Require(Get(box, "http://example.com/") == "invalid=1; noequals; value; future=1");
Require(sceHttp2AddCookie(box, "http://example.com/", "cut=12345", 5) == 0);
Require(Get(box, "http://example.com/") == "invalid=1; noequals; value; cut=1; future=1");
Http2CookieStats stats = Stats(box);
Require(stats.currentInuseNum == 7 && stats.currentInuseSize == 23 + 23 + 24 + 25 + 25 + 22 + 21);
Require(stats.maxInuseNum == 7 && stats.maxInuseSize == stats.currentInuseSize && stats.removedNum == 0);
const std::string header = "invalid=1; noequals; value; cut=1; future=1";
std::size_t required = 0;
char small[8] = {'x', 'x'};
Require(sceHttp2GetCookie(box, "http://example.com/", small, &required, sizeof(small)) == OutOfSize && required == 0 && small[0] == '\0' && small[1] == 'x');
std::string large(64, 'x');
Require(sceHttp2GetCookie(box, "http://example.com/", large.data(), nullptr, large.size()) == 0 && large.compare(0, header.size() + 1, header.c_str(), header.size() + 1) == 0);
Require(sceHttp2GetCookie(box, "http://example.com/", nullptr, nullptr, 0) == InvalidValue);
Require(sceHttp2GetCookie(box, nullptr, nullptr, &required, 0) == InvalidValue);
Require(sceHttp2AddCookie(box, nullptr, "a=1", 3) == InvalidValue);
Require(sceHttp2AddCookie(box, "http://example.com/", nullptr, 3) == InvalidValue);
Require(sceHttp2AddCookie(box, "http://example.com/", "a=1", 0) == InvalidValue);
Require(sceHttp2AddCookie(box, "http:///path", "a=1", 3) == Rejected);
Require(sceHttp2AddCookie(box, "", "a=1", 3) == Rejected);
Require(sceHttp2GetCookieStats(box, nullptr) == InvalidValue);
Require(sceHttp2GetCookieStats(0, nullptr) == InvalidValue);
Require(sceHttp2CookieFlush(box) == 0);
Require(NoMatch(box, "http://example.com/"));
stats = Stats(box);
Require(stats.currentInuseNum == 0 && stats.currentInuseSize == 0 && stats.maxInuseNum == 7);
Require(sceHttp2SetCookieMaxSize(box, 4) == 0);
Require(Add(box, "http://example.com/", "abc=defg") == OutOfSize);
Require(Add(box, "http://example.com/", "ab=cd") == 0);
Require(Get(box, "http://example.com/") == "ab=cd");
Require(Stats(box).currentInuseSize == 21);
Require(sceHttp2SetCookieMaxSize(box, 0) == 0);
Require(Add(box, "http://example.com/", "x=1") == OutOfSize);
Require(sceHttp2SetCookieMaxSize(box, 4096) == 0);
Require(sceHttp2SetCookieMaxNumPerDomain(box, 2) == 0);
Require(Add(box, "http://example.com/", "c=1") == 0);
Require(Get(box, "http://example.com/") == "c=1; ab=cd");
Require(Add(box, "http://example.com/", "d=1") == 0);
Require(Get(box, "http://example.com/") == "d=1; c=1");
Require(Add(box, "http://example.com/", "e=1") == 0);
Require(Get(box, "http://example.com/") == "e=1; d=1");
Require(Add(box, "http://other.com/", "o=1") == 0);
Require(sceHttp2SetCookieMaxNum(box, 2) == 0);
Require(Add(box, "http://third.com/", "t=1") == 0);
Require(NoMatch(box, "http://example.com/"));
Require(NoMatch(box, "http://other.com/"));
Require(Get(box, "http://third.com/") == "t=1");
stats = Stats(box);
Require(stats.currentInuseNum == 1 && stats.removedNum == 5);
const int defaults = sceHttp2CreateCookieBox(0x10000);
Require(defaults > 0);
Require(Add(defaults, "http://big.test/", "n=" + std::string(4096, 'v')) == OutOfSize);
Require(Add(defaults, "http://big.test/", "n=" + std::string(4095, 'v')) == 0);
Require(FillDomain(defaults, 50) == 0);
Require(Get(defaults, "http://example.com/").compare(0, 6, "c1=1; ") == 0);
for (int i = 50; i < 60; ++i) Require(Add(defaults, "http://example.com/", "c" + std::to_string(i) + "=1") == 0);
const std::string kept = Get(defaults, "http://example.com/");
Require(kept.compare(0, 7, "c11=1; ") == 0 && kept.compare(kept.size() - 5, 5, "c10=1") == 0 && kept.find("c0=") == std::string::npos);
stats = Stats(defaults);
Require(stats.currentInuseNum == 51 && stats.removedNum == 10);
const int noTotal = sceHttp2CreateCookieBox(0x10000);
Require(noTotal > 0 && sceHttp2SetCookieMaxNum(noTotal, 0) == 0 && FillDomain(noTotal, 60) == 0);
stats = Stats(noTotal);
Require(stats.currentInuseNum == 1 && stats.removedNum == 59);
const int noPerDomain = sceHttp2CreateCookieBox(0x10000);
Require(noPerDomain > 0 && sceHttp2SetCookieMaxNumPerDomain(noPerDomain, 0) == 0 && FillDomain(noPerDomain, 60) == 0);
stats = Stats(noPerDomain);
Require(stats.currentInuseNum == 1 && stats.removedNum == 59);
Require(Get(noPerDomain, "http://example.com/") == "c59=1");
Require(sceHttp2AddCookie(0, "http://example.com/", "a=1", 3) == InvalidId);
Require(sceHttp2SetCookieMaxNum(0x7fffffff, 1) == InvalidId);
Require(sceHttp2SetCookieMaxNumPerDomain(-1, 1) == InvalidId);
Require(sceHttp2SetCookieMaxSize(0, 1) == InvalidId);
Require(sceHttp2GetCookie(0, "http://example.com/", nullptr, &required, 0) == InvalidId);
Http2CookieStats ignored{};
Require(sceHttp2GetCookieStats(0, &ignored) == InvalidId);
const int other = sceHttp2CreateCookieBox(0x10000);
Require(other > 0 && NoMatch(other, "http://example.com/"));
Require(sceHttp2DeleteCookieBox(box) == 0);
Require(sceHttp2GetCookieStats(box, &ignored) == InvalidId);
Require(sceHttp2AddCookie(box, "http://example.com/", "a=1", 3) == InvalidId);
Require(sceHttp2DeleteCookieBox(other) == 0);
Require(sceHttp2DeleteCookieBox(defaults) == 0);
Require(sceHttp2DeleteCookieBox(noTotal) == 0);
Require(sceHttp2DeleteCookieBox(noPerDomain) == 0);
}
Reference in new issue View Git Blame Copy Permalink