scummvm/backends/networking/curl/curljsonrequest.cpp

/* ScummVM - Graphic Adventure Engine
*
* ScummVM is the legal property of its developers, whose names
* are too numerous to list here. Please refer to the COPYRIGHT
* file distributed with this source distribution.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
*/

#define FORBIDDEN_SYMBOL_ALLOW_ALL

#include "backends/networking/curl/curljsonrequest.h"
#include "backends/networking/curl/connectionmanager.h"
#include "backends/networking/curl/networkreadstream.h"
#include "common/debug.h"
#include "common/json.h"
#include <curl/curl.h>

namespace Networking {

CurlJsonRequest::CurlJsonRequest(JsonCallback cb, Common::String url):
	CurlRequest(0, url), _jsonCallback(cb), _contentsStream(DisposeAfterUse::YES) {}

CurlJsonRequest::~CurlJsonRequest() { delete _jsonCallback; }

char *CurlJsonRequest::getPreparedContents() {
	//write one more byte in the end
	byte zero[1] = { 0 };
	_contentsStream.write(zero, 1);

	//replace all "bad" bytes with '.' character
	byte *result = _contentsStream.getData();
	uint32 size = _contentsStream.size();
	for (uint32 i = 0; i < size; ++i)
		if (result[i] < 0x20 || result[i] > 0x7f)
			result[i] = '.';

	//make it zero-terminated string
	result[size - 1] = '\0';

	return (char *)result;
}

void CurlJsonRequest::handle() {
	if (!_stream) _stream = new NetworkReadStream(_url.c_str(), _headersList, _postFields);

	if (_stream) {
		const int kBufSize = 16*1024;
		char buf[kBufSize+1];
		uint32 readBytes = _stream->read(buf, kBufSize);
		if (readBytes != 0)
			if (_contentsStream.write(buf, readBytes) != readBytes)
				warning("MemoryWriteStreamDynamic was unable to write all the bytes");

		if (_stream->eos()) {			
			if (_stream->httpResponseCode() != 200)
				warning("HTTP response code is not 200 OK (it's %ld)", _stream->httpResponseCode());

			char *contents = getPreparedContents();
			if (_stream->httpResponseCode() != 200)
				debug("%s", contents);
			Common::JSONValue *json = Common::JSON::parse(contents);
			finishJson(json);
		}
	}
}

void CurlJsonRequest::restart() {
	if (_stream) delete _stream;
	_stream = 0;
	_contentsStream = Common::MemoryWriteStreamDynamic(DisposeAfterUse::YES);
	//with no stream available next handle() will create another one
}

void CurlJsonRequest::finishJson(Common::JSONValue *json) {
	Request::finish();
	if (_jsonCallback) (*_jsonCallback)(JsonResponse(this, json)); //potential memory leak, free it in your callbacks!
	else delete json;
}

void CurlJsonRequest::finish() {
	finishJson(0);
}

} //end of namespace Networking