dvander@2562: /**
dvander@2562:  * vim: set ts=4 :
dvander@2562:  * =============================================================================
dvander@2562:  * SourceMod Updater Extension
dvander@2562:  * Copyright (C) 2004-2009 AlliedModders LLC.  All rights reserved.
dvander@2562:  * =============================================================================
dvander@2562:  *
dvander@2562:  * This program is free software; you can redistribute it and/or modify it under
dvander@2562:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@2562:  * Free Software Foundation.
dvander@2562:  * 
dvander@2562:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@2562:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@2562:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@2562:  * details.
dvander@2562:  *
dvander@2562:  * You should have received a copy of the GNU General Public License along with
dvander@2562:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@2562:  *
dvander@2562:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@2562:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@2562:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@2562:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@2562:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@2562:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@2562:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@2562:  * or <http://www.sourcemod.net/license.php>.
dvander@2562:  *
dvander@2562:  * Version: $Id$
dvander@2562:  */
dvander@2562: 
dvander@2562: #include <assert.h>
dvander@2562: #include <string.h>
dvander@2562: #include <stdlib.h>
dvander@2562: #include "MemoryDownloader.h"
dvander@2562: 
dvander@2562: using namespace SourceMod;
dvander@2562: 
dvander@2562: MemoryDownloader::MemoryDownloader() : buffer(NULL), bufsize(0), bufpos(0)
dvander@2562: {
dvander@2562: }
dvander@2562: 
dvander@2562: MemoryDownloader::~MemoryDownloader()
dvander@2562: {
dvander@2562: 	free(buffer);
dvander@2562: }
dvander@2562: 
dvander@2562: DownloadWriteStatus MemoryDownloader::OnDownloadWrite(IWebTransfer *session,
dvander@2562: 													  void *userdata,
dvander@2562: 													  void *ptr,
dvander@2562: 													  size_t size,
dvander@2562: 													  size_t nmemb)
dvander@2562: {
dvander@2562: 	size_t total = size * nmemb;
dvander@2562: 
dvander@2562: 	if (bufpos + total > bufsize)
dvander@2562: 	{
dvander@2562: 		size_t rem = (bufpos + total) - bufsize;
dvander@2562: 		bufsize += rem + (rem / 2);
dvander@2562: 		buffer = (char *)realloc(buffer, bufsize);
dvander@2562: 	}
dvander@2562: 
dvander@2562: 	assert(bufpos + total <= bufsize);
dvander@2562: 
dvander@2562: 	memcpy(&buffer[bufpos], ptr, total);
dvander@2562: 	bufpos += total;
dvander@2562: 
dvander@2562: 	return DownloadWrite_Okay;
dvander@2562: }
dvander@2562: 
dvander@2562: void MemoryDownloader::Reset()
dvander@2562: {
dvander@2562: 	bufpos = 0;
dvander@2562: }
dvander@2562: 
dvander@2562: char *MemoryDownloader::GetBuffer()
dvander@2562: {
dvander@2562: 	return buffer;
dvander@2562: }
dvander@2562: 
dvander@2562: size_t MemoryDownloader::GetSize()
dvander@2562: {
dvander@2562: 	return bufpos;
dvander@2562: }
dvander@2562: