#pragma once
#include "Cookie.h"
#include "CookieJar.h"
#include "SQLiteDatabase.h"
#include "SQLiteStatement.h"
#include <wtf/HashMap.h>
#include <wtf/Noncopyable.h>
#include <wtf/text/StringHash.h>
namespace WebCore {
enum class CookieAcceptPolicy {
Always,
Never,
OnlyFromMainDocumentDomain,
ExclusivelyFromMainDocumentDomain
};
class CookieJarDB {
WTF_MAKE_FAST_ALLOCATED;
WTF_MAKE_NONCOPYABLE(CookieJarDB);
public:
enum class Source : uint8_t {
Network,
Script
};
void open();
bool isEnabled() const;
void setAcceptPolicy(CookieAcceptPolicy policy) { m_acceptPolicy = policy; }
CookieAcceptPolicy acceptPolicy() const { return m_acceptPolicy; }
HashSet<String> allDomains();
Optional<Vector<Cookie>> searchCookies(const URL& firstParty, const URL& requestUrl, const Optional<bool>& httpOnly, const Optional<bool>& secure, const Optional<bool>& session);
Vector<Cookie> getAllCookies();
bool setCookie(const URL& firstParty, const URL&, const String& cookie, Source, Optional<Seconds> cappedLifetime = WTF::nullopt);
bool setCookie(const Cookie&);
bool deleteCookie(const String& url, const String& name);
bool deleteCookies(const String& url);
bool deleteCookiesForHostname(const String& hostname, IncludeHttpOnlyCookies);
bool deleteAllCookies();
WEBCORE_EXPORT CookieJarDB(const String& databasePath);
WEBCORE_EXPORT ~CookieJarDB();
private:
CookieAcceptPolicy m_acceptPolicy { CookieAcceptPolicy::Always };
String m_databasePath;
bool m_detectedDatabaseCorruption { false };
bool isOnMemory() const { return (m_databasePath == ":memory:"); };
bool openDatabase();
void closeDatabase();
bool checkSQLiteReturnCode(int);
void flagDatabaseCorruption();
bool checkDatabaseCorruptionAndRemoveIfNeeded();
String getCorruptionMarkerPath() const;
bool checkDatabaseValidity();
void deleteAllDatabaseFiles();
void verifySchemaVersion();
void deleteAllTables();
void createPrepareStatement(const String&);
SQLiteStatement& preparedStatement(const String&);
bool executeSql(const String&);
bool deleteCookieInternal(const String& name, const String& domain, const String& path);
bool hasHttpOnlyCookie(const String& name, const String& domain, const String& path);
bool canAcceptCookie(const Cookie&, const URL& firstParty, const URL&, CookieJarDB::Source);
bool checkCookieAcceptPolicy(const URL& firstParty, const URL&);
bool hasCookies(const URL&);
SQLiteDatabase m_database;
HashMap<String, std::unique_ptr<SQLiteStatement>> m_statements;
};
}