paulo@0: /*
paulo@0:  * $Id: download.c,v 1.2 2004/04/17 06:06:46 hipnod Exp $
paulo@0:  *
paulo@0:  * Copyright (C) 2004 giFT project (gift.sourceforge.net)
paulo@0:  *
paulo@0:  * This program is free software; you can redistribute it and/or modify it
paulo@0:  * under the terms of the GNU General Public License as published by the
paulo@0:  * Free Software Foundation; either version 2, or (at your option) any
paulo@0:  * later version.
paulo@0:  *
paulo@0:  * This program is distributed in the hope that it will be useful, but
paulo@0:  * WITHOUT ANY WARRANTY; without even the implied warranty of
paulo@0:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
paulo@0:  * General Public License for more details.
paulo@0:  */
paulo@0: 
paulo@0: #include "gt_gnutella.h"
paulo@0: #include "transfer/download.h"
paulo@0: 
paulo@0: /*****************************************************************************/
paulo@0: 
paulo@0: static Dataset *gt_downloads;
paulo@0: 
paulo@0: /*****************************************************************************/
paulo@0: 
paulo@0: void gt_download_add (Transfer *transfer, Source *source)
paulo@0: {
paulo@0: 	Dataset *d;
paulo@0: 
paulo@0: 	d = dataset_lookup (gt_downloads, &transfer, sizeof(transfer));
paulo@0: 	dataset_insert (&d, &source, sizeof(source), source, 0);
paulo@0: 
paulo@0: 	dataset_insert (&gt_downloads, &transfer, sizeof(transfer), d, 0);
paulo@0: }
paulo@0: 
paulo@0: void gt_download_remove (Transfer *transfer, Source *source)
paulo@0: {
paulo@0: 	Dataset *d;
paulo@0: 
paulo@0: 	d = dataset_lookup (gt_downloads, &transfer, sizeof(transfer));
paulo@0: 	dataset_remove (d, &source, sizeof(source));
paulo@0: 
paulo@0: 	if (dataset_length (d) == 0)
paulo@0: 	{
paulo@0: 		dataset_clear (d);
paulo@0: 		dataset_remove (gt_downloads, &transfer, sizeof(transfer));
paulo@0: 	}
paulo@0: 
paulo@0: 	if (dataset_length (gt_downloads) == 0)
paulo@0: 	{
paulo@0: 		dataset_clear (gt_downloads);
paulo@0: 		gt_downloads = NULL;
paulo@0: 	}
paulo@0: }
paulo@0: 
paulo@0: /*****************************************************************************/
paulo@0: 
paulo@0: static int ds_find_hash (ds_data_t *key, ds_data_t *value, void *udata)
paulo@0: {
paulo@0: 	Array      *a    = udata;
paulo@0: 	char       *sha1;
paulo@0: 	Source    **ret;
paulo@0: 	Source     *src  = value->data;
paulo@0: 	int         n;
paulo@0: 
paulo@0: 	n = array_list (&a, &sha1, &ret, NULL);
paulo@0: 	assert (n == 2);
paulo@0: 
paulo@0: 	if (!src->hash)
paulo@0: 		return DS_CONTINUE;
paulo@0: 
paulo@0: 	/* NOTE: the hash is prefixed with giftd's hash here */
paulo@0: 	if (strcmp (src->hash, sha1) == 0)
paulo@0: 	{
paulo@0: 		*ret = src;
paulo@0: 		return DS_BREAK;
paulo@0: 	}
paulo@0: 
paulo@0: 	return DS_CONTINUE;
paulo@0: }
paulo@0: 
paulo@0: static int ds_traverse_transfer (ds_data_t *key, ds_data_t *value, void *udata)
paulo@0: {
paulo@0: 	Dataset *d = value->data;
paulo@0: 
paulo@0: 	dataset_foreach_ex (d, ds_find_hash, udata);
paulo@0: 	return DS_CONTINUE;
paulo@0: }
paulo@0: 
paulo@0: Source *gt_download_lookup (const char *sha1)
paulo@0: {
paulo@0: 	Array  *a;
paulo@0: 	Source *ret = NULL;
paulo@0: 
paulo@0: 	a = array_new ((void *)sha1, &ret, NULL);
paulo@0: 
paulo@0: 	if (!a)
paulo@0: 		return NULL;
paulo@0: 
paulo@0: 	dataset_foreach_ex (gt_downloads, ds_traverse_transfer, a);
paulo@0: 	array_unset (&a);
paulo@0: 
paulo@0: 	return ret;
paulo@0: }