HttpCookie.cpp

#include "HttpCookie.h"
#include <stdio.h>
#include <string.h>
#include <sstream>

static std::string getStringFromFile(const char* filename) {
    std::string tmp;
    do {
        FILE* cpfile = fopen(filename,"rb");
        size_t filelen = 0;
        char* buf = NULL;
        if(cpfile == NULL)
            break;
        fseek(cpfile,0,SEEK_END);
        filelen = ftell(cpfile);
        buf = new char[filelen + 1];
        fseek(cpfile,0,SEEK_SET);

        buf[filelen] = 0;

        fread(buf,1,filelen,cpfile);

        tmp = buf;

        delete[] buf;
        fclose(cpfile);
    }while(0);

    return tmp;
}

void HttpCookie::readFile()
{
    std::string inString  = getStringFromFile(_cookieFileName.c_str());


    if(inString.length() != 0)
    {
        std::vector<std::string> cookiesVec;
        cookiesVec.clear();

        std::stringstream stream(inString);
        std::string item;
        while(std::getline(stream, item, '\n'))
        {
            cookiesVec.push_back(item);
        }

        if(cookiesVec.empty())
            return;

        _cookies.clear();

        for(std::vector<std::string>::iterator iter = cookiesVec.begin();iter != cookiesVec.end(); iter++)
        {
            std::string cookie = *iter;

            if(cookie.length() == 0)
                continue;

            if(cookie.find("#HttpOnly_") != std::string::npos)
            {
                cookie = cookie.substr(10);
            }

            if(cookie.at(0) == '#')
                continue;

            CookiesInfo co;
            std::stringstream streamInfo(cookie);
            std::vector<std::string> elems;
            std::string elemsItem;

            while (std::getline(streamInfo, elemsItem, '\t'))
            {
                elems.push_back(elemsItem);
            }

            co.domain = elems[0];
            if (co.domain.at(0) == '.')
            {
                co.domain = co.domain.substr(1);
            }
            co.tailmatch = strcmp("TRUE", elems[1].c_str())?true: false;
            co.path   = elems[2];
            co.secure = strcmp("TRUE", elems[3].c_str())?true: false;
            co.expires = elems[4];
            co.name = elems[5];
            co.value = elems[6];
            _cookies.push_back(co);
        }
    }
}

const std::vector<CookiesInfo>* HttpCookie::getCookies() const
{
    return &_cookies;
}

const CookiesInfo* HttpCookie::getMatchCookie(const std::string& url) const
{
    for(std::vector<CookiesInfo>::const_iterator iter = _cookies.begin(); iter != _cookies.end(); iter++)
    {
        if(url.find(iter->domain) != std::string::npos)
            return &(*iter);
    }

    return NULL;
}

void HttpCookie::updateOrAddCookie(CookiesInfo* cookie)
{
    for(std::vector<CookiesInfo>::iterator iter = _cookies.begin(); iter != _cookies.end(); iter++)
    {
        if(cookie->domain == iter->domain)
        {
            *iter = *cookie;
            return;
        }
    }
    _cookies.push_back(*cookie);
}

void HttpCookie::writeFile()
{
    FILE *out;
    out = fopen(_cookieFileName.c_str(), "w");
    fputs("# Netscape HTTP Cookie File\n"
          "# http://curl.haxx.se/docs/http-cookies.html\n"
          "# This file was generated by cocos2d-x! Edit at your own risk.\n"
          "# Test cocos2d-x cookie write.\n\n",
          out);

    std::string line;
    for(std::vector<CookiesInfo>::iterator iter = _cookies.begin(); iter != _cookies.end(); iter++)
    {
        line.clear();
        line.append(iter->domain);
        line.append(1, '\t');
        iter->tailmatch ? line.append("TRUE") : line.append("FALSE");
        line.append(1, '\t');
        line.append(iter->path);
        line.append(1, '\t');
        iter->secure ? line.append("TRUE") : line.append("FALSE");
        line.append(1, '\t');
        line.append(iter->expires);
        line.append(1, '\t');
        line.append(iter->name);
        line.append(1, '\t');
        line.append(iter->value);
        //line.append(1, '\n');

        fprintf(out, "%s\n", line.c_str());
    }

    fclose(out);
}

void HttpCookie::setCookieFileName(std::string filename)
{
    _cookieFileName = filename;
}