CEF3开发者系列之Cookies管理和共享

涉及网页登录相关的技术，Cookies肯定是忽略不了的。由于项目的需要，要做一个双核的产品。双核间切换会涉及到登录状态的问题，共享Cookies是一个很好的方案。既然涉及到共享cookies，那么读取完整的cookies和设置cookies就是问题的关键。由于应用本身只需要加载自家的平台，不用考虑共享所有网站cookie的问题，所以需要获取的和设置的cookies相对比较简单。
IE浏览器内核Cookies的获取和设置相对难一点，但是好在网络上资料比较多。Chromium内核基于CEF3框架，获取和设置cookies有现成的接口。在这里主要总结的是IE浏览器内核Cookies的获取和基于CEF3框架的Cookies设置以及获取。

IE浏览器内核的Cookies获取的三种方式：
1、通过读取存在硬盘上的cookies文件，来获取指定网站的cookies；
2、通过用MSHTMLI的Dispatch接口IWebBrowser2和IHTMLDocument2来获取cookies；
3、通过API函数InternetGetCookieEx和InternetGetCookie来获取cookies；
4、通过Hook WINNET 相关的函数获取Cookies，HOOK住发送和接收包API(send、recv、WSASend、WSARecv等)。然后分析HTTP header。此处不讲。
第一种方式，实现起来比较笨重，一般作为后边两种方式的补充。实现代码如下：

std::wstring GetCookiesUnderIE7( wstring strDomain )
{
    wstring strCookiePath;
    TCHAR dataPath[MAX_PATH];
    ::SHGetSpecialFolderPath(NULL, dataPath, CSIDL_COOKIES, FALSE);
    strCookiePath = dataPath;
    
    TCHAR        szUserName[32];
    DWORD        cbUserName;
    cbUserName = 32;
    GetUserName(szUserName, &cbUserName);
    WIN32_FIND_DATA  FindData;
    HANDLE hfile;  
    DWORD errorcode = 0;
    hfile = FindFirstFile((strCookiePath + L"\\*.txt").c_str() ,&FindData);
    wstring wsUserName = szUserName;
    wstring wsCookieName = wsUserName + L"@xx"; //xx为网站域名，如百度的baidu
    wstring wsFileName;
    wstring wsCookies = L"";
    hile(hfile!= INVALID_HANDLE_VALUE && errorcode != ERROR_NO_MORE_FILES)  
    {
        wsFileName = FindData.cFileName;
        //判断是否存在【user】@xx为名称的cookie脚本，提高过滤效率
        if (wsFileName.find(wsCookieName.c_str()) != wstring::npos)
        {
            wsFileName = strCookiePath + L"\\"+ FindData.cFileName;
            //http://www.cnblogs.com/guolixiucai/;
            std::ifstream ifsCookie(wsFileName.c_str(), std::ios::in | std::ios::binary);  
            if (ifsCookie)  
            {
                std::string contents;  
                ifsCookie.seekg(0, std::ios::end);
                contents.resize(ifsCookie.tellg());
                ifsCookie.seekg(0, std::ios::beg);
                ifsCookie.read(&contents[0], contents.size());
                if(contents.find("ispass_37wan_com") != string::npos)
                {
                    wsCookies = s2ws(contents);
                    ifsCookie.close();
                    break;
                }
                ifsCookie.close();
            }
        }

        BOOL flag = TRUE;
        BOOL isNextFile = FindNextFile(hfile,&FindData);//判断该目录下是否还有文件
        if(flag == TRUE && isNextFile == TRUE)//如果还有文件，则调用SetLastError，设为NO_ERROR，这样才能继续遍历后面的文件
            SetLastError(NO_ERROR);
        else
            errorcode=GetLastError();
    }
    return wsCookies;
}

第二种方式获取的Cookies并不完全，只能获取到部分Cookies
IHTMLDocument2::get_cookie()
IHTMLDocument2::put_cookie()
用MSHTML对Document对象的Cookie属性进行操作。也可以实现类似InternetGetCookie和InternetSetCookie的效果。实际上它等同于用javascript来读写文档中的Cookie。它同样可以获取、添加、覆盖、修改、删除持久Cookie和会话级Cookie，但不可读写HTTPOnly的Cookie。这种方式同样也要十分注意设置的Cookie的每个属性都要和目标Cookie一一对应才能正确操作Cookie。特别是Path和Domain属性,否则会导致添加了一个同名Cookie而不能覆盖或清除目标Cookie的结果。
代码如下：

std::wstring BrowserView::GetCookies()
{
    CComPtr webBrowser;
    HRESULT hr = QueryControl(IID_IWebBrowser2,
        reinterpret_cast<void**>(&webBrowser));
    if (FAILED(hr) || (!webBrowser))
        return L"";

    CComPtr disp;
    hr = webBrowser->get_Document(&disp);
    if (FAILED(hr) || (!disp))
        return L"";


    CComPtr document;  
    hr = disp->QueryInterface(IID_IHTMLDocument2,
        reinterpret_cast<void**>(&document));
    if (FAILED(hr) || (!document))
        return L"";

    BSTR bstrCookie;
    hr = document->get_cookie(&bstrCookie);
    BSTR bstrDomain;
    document->get_domain(&bstrDomain);
    if (FAILED(hr))
    {
        return L"";
    }
    wstring strCookie(bstrCookie);
     char* lpszText2 = _com_util::ConvertBSTRToString(bstrCookie);
    SysFreeString(bstrCookie);
    return strCookie;
}

第三种方式：
InternetGetCookie
InternetSetCookie
InternetGetCookieEx
InternetSetCookieEx
这些API属于Wininet，可以获取、添加、覆盖、修改、删除持久Cookie和会话级Cookie（会话级Cookie需要在同进程中操作）。加Ex后缀的API可以对HTTPOnly的Cookie进行操作。但这组API对IE7及以下的IE浏览器内核无效，取不到Cookies，这时候就需要使用第一种方法补充。当然也可以使用其他比较复杂的方法，例如Hook WINNET 相关的函数。
std::wstring GetCookies( wstring strDomain )
{
    LPDWORD lpdwSize = new DWORD;
    wchar_t strCookie[2048] = {0};
    int size = 0;
    //InternetGetCookie 取不到httponly的cookie，留在这里仅供后来者参考
//         InternetGetCookie(strDomain.c_str(), NULL, NULL, lpdwSize);
//         InternetGetCookie(strDomain.c_str(), NULL, strCookie, lpdwSize/*, INTERNET_COOKIE_HTTPONLY, NULL*/);

    //InternetGetCookieEx 在IE7和IE6里也是取不全cookie。
    InternetGetCookieEx(strDomain.c_str(), NULL, strCookie, lpdwSize, 0x2000, NULL);
    wstring wsCookies = strCookie;
    return wsCookies;
}

CEF3中Cookies的管理。
如果不想听我啰嗦，移步http://magpcss.org/ceforum/apidocs3/projects/(default)/CefCookieManager.html
下边的内容基本上是翻译这个文档，在加上两个实例。
CEF3中，CefCookieManager这个类就是用来管理cookies的。在头文件cef_cookies中，在cef_cookies_capi.h里，有详细的注释，和上边链接里的文档说明一样。
Cookies的管理无外乎Cookies的设置、获取、删除、查找，外加一个存储位置的设置。

class CefCookieManager
extends CefBase
Class used for managing cookies. The methods of this class may be called on any thread unless otherwise indicated.
该类用来管理cookies。除非另有说明，该类的方法可以在任何线程中调用

  
Method Detail
CreateManager
public static CefRefPtr< CefCookieManager > CreateManager( const CefString& path, bool persist_session_cookies, CefRefPtr< CefCompletionCallback > callback );
创建一个新的cookie管理器，当|path|没有被设置时，数据存储在内存中。否则，数据保存在指定的|path|（如果在cef初始化时，指定了缓存路径，那么cookies会保存在此路径）。持久化会话cookies（没有有效期或有效区域），需要设置|persist_session_cookies| 为true。会话cookies一般是暂时性的并且大多数浏览器并不支持。如果|callback| 为 non-NULL ，那么在管理器存储空间初始化后，会在IO线程中异步执行。

DeleteCookies
public virtual bool DeleteCookies( const CefString& url, const CefString& cookie_name, CefRefPtr< CefDeleteCookiesCallback > callback )= 0;
删除与指定参数的相匹配的所有cookies。如果 |url| 和 |cookie_name| 都被指定，那么 host和domian与这两个参数匹配的都被删除。如果 |url|为空，那么所有的hosts 和domains的cookies都没清空。如果|callback| 为 non-NULL ，那么在cookies被删除后，该回调函数会在IO线程中异步执行。如果不存在指定的url或者cookies不能被访问，那么返回falsh。cookies可以通过交替使用Visit*Cookies()方法被删除。

//向指定的域名删除cookies

void ClientAppBrowser::DeleteCookies( std::wstring domain, std::wstring cookiename )
{
    CefRefPtr manager = CefCookieManager::GetGlobalManager(NULL);
    DCHECK(manager.get());
    CefRefPtr callback = NULL;
    std::wstring httpDomain = domain;  
    CefString cookie_name;
    cookie_name.FromWString(cookiename .c_str());
    base::IgnoreResult(&CefCookieManager::DeleteCookies);
    CefPostTask(TID_IO, NewCefRunnableMethod(manager.get(), &CefCookieManager::DeleteCookies,
        CefString(httpDomain.c_str()), cookie_name, callback));
}

FlushStore
public virtual bool FlushStore( CefRefPtr< CefCompletionCallback > callback )= 0;
将备份存储（如果存在）缓存到硬盘，如果|callback| 为 non-NULL ，在缓存完后，该回调函数会在IO线程中异步执行。cookies不能被访问则返回falsh。

GetGlobalManager
public static CefRefPtr< CefCookieManager > GetGlobalManager( CefRefPtr< CefCompletionCallback > callback );
返回全局cookie管理器，默认存储路径是 CefSettings.cache_path或者其他指定路径，否则存储在内存。如果|callback| 为 non-NULL ，在管理器存储空间初始化后，该回调函数会在IO线程中异步执行。该方法与调用CefRequestContext::GetGlobalContext()->GetDefaultCookieManager()相当

SetCookie
public virtual bool SetCookie( const CefString& url, const CefCookie& cookie, CefRefPtr< CefSetCookieCallback > callback )= 0;
设置一个指定了url和由用户设置了属性的cookie。该方法需要每个属性有符合良好的格式。检查非法字符（例如“;”存在cookie属性值中），存在非法字符会导致cookies设置失败。如果|callback| 为 non-NULL ，那么在cookies被设置后，该回调函数会在IO线程中异步执行。指定的url不存在或者cookies不能被访问，返回falsh。

实例

 1 void ClientAppBrowser::SetCookies( std::wstring domain, std::wstring key, std::wstring svalue )
 2 {
 3     /*
 4     向指定的域名写Cookie   http://www.cnblogs.com/guolixiucai/
 5     */  
 6     CefRefPtr manager = CefCookieManager::GetGlobalManager(NULL);
 7     DCHECK(manager.get());
 8     CefRefPtr callback = NULL;
 9     CefCookie cookie;
10     CefString(&cookie.name).FromWString(key.c_str());
11     CefString(&cookie.value).FromWString(svalue.c_str());
12     CefString(&cookie.domain).FromWString(domain.c_str());
13     CefString(&cookie.path).FromASCII("/");
14     cookie.has_expires = true;
15        //以下过期时间随意设置的。
16     cookie.expires.year = 2200;
17     cookie.expires.month = 4;  
18     cookie.expires.day_of_week = 5;  
19     cookie.expires.day_of_month = 11;  
20     std::wstring httpDomain = L"http://www";  
21     httpDomain.append(domain);
22     base::IgnoreResult(&CefCookieManager::SetCookie);
23     CefPostTask(TID_IO, NewCefRunnableMethod(manager.get(), &CefCookieManager::SetCookie,
24            CefString(httpDomain.c_str()), cookie, callback));
25 }

SetStoragePath
public virtual bool SetStoragePath( const CefString& path, bool persist_session_cookies, CefRefPtr< CefCompletionCallback > callback )= 0;
设置存储cookies数据的路径。如果|path|为空，则存储在内存中。持久化会话cookies（没有有效期或有效区域），需要设置|persist_session_cookies| 为true。会话cookies一般是暂时性的并且大多数浏览器并不支持。如果|callback| 为 non-NULL ，那么在管理器存储空间初始化后，会在IO线程中异步执行。

SetSupportedSchemes
public virtual void SetSupportedSchemes( const std::vector< CefString >& schemes, CefRefPtr< CefCompletionCallback > callback )= 0;
设置管理器支持的协议。支持常见的协议（"http", "https", "ws" and "wss"）。如果|callback| 为 non-NULL ，在协议更改生效后，该回调函数会在IO线程中异步执行。该方法必须在所有cookies被访问前调用

VisitAllCookies
public virtual bool VisitAllCookies( CefRefPtr< CefCookieVisitor > visitor )= 0;
访问IO线程中所有的Cookies。返回的cookies按照最长路径排序，然后再按照最长创建时间排序。如果cookies被占用则返回false。

VisitUrlCookies
public virtual bool VisitUrlCookies( const CefString& url, bool includeHttpOnly, CefRefPtr< CefCookieVisitor > visitor )= 0;
访问IO线程上的一组Cookie。返回结果可按照url、host、domain和路径过滤。如果|includeHttpOnly|为ture，那么结果中还会包含HTTP-only cookies。返回的cookies按照最长路径排序，然后再按照最长创建时间排序。如果cookies被占用则返回false。