mirror of git://erdgeist.org/opentracker
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
2.7 KiB
C
105 lines
2.7 KiB
C
/* This software was written by Dirk Engling <erdgeist@erdgeist.org>
|
|
It is considered beerware. Prost. Skol. Cheers or whatever. */
|
|
|
|
/* System */
|
|
#include <pthread.h>
|
|
#include <stdio.h>
|
|
|
|
/* Libowfat */
|
|
#include "byte.h"
|
|
|
|
/* Opentracker */
|
|
#include "trackerlogic.h"
|
|
#include "ot_mutex.h"
|
|
|
|
/* Our global all torrents list */
|
|
static ot_vector all_torrents[OT_BUCKET_COUNT];
|
|
|
|
static int bucket_locklist[ OT_MAX_THREADS ];
|
|
static int bucket_locklist_count = 0;
|
|
static pthread_mutex_t bucket_mutex;
|
|
static pthread_cond_t bucket_being_unlocked;
|
|
|
|
static int bucket_check( int bucket ) {
|
|
/* C should come with auto-i ;) */
|
|
int i;
|
|
|
|
/* No more space to acquire lock to bucket -- should not happen */
|
|
if( bucket_locklist_count == OT_MAX_THREADS ) {
|
|
fprintf( stderr, "More lock requests than mutexes. Consult source code.\n" );
|
|
return -1;
|
|
}
|
|
|
|
/* See, if bucket is already locked */
|
|
for( i=0; i<bucket_locklist_count; ++i )
|
|
if( bucket_locklist[ i ] == bucket )
|
|
return -1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void bucket_push( int bucket ) {
|
|
bucket_locklist[ bucket_locklist_count++ ] = bucket;
|
|
}
|
|
|
|
static void bucket_remove( int bucket ) {
|
|
int i = 0;
|
|
|
|
while( ( i < bucket_locklist_count ) && ( bucket_locklist[ i ] != bucket ) )
|
|
++i;
|
|
|
|
if( i == bucket_locklist_count ) {
|
|
fprintf( stderr, "Request to unlock bucket that was never lock. Consult source code.\n" );
|
|
return;
|
|
}
|
|
|
|
for( ; i < bucket_locklist_count - 1; ++i )
|
|
bucket_locklist[ i ] = bucket_locklist[ i + 1 ];
|
|
|
|
--bucket_locklist_count;
|
|
}
|
|
|
|
ot_vector *mutex_bucket_lock( int bucket ) {
|
|
pthread_mutex_lock( &bucket_mutex );
|
|
while( bucket_check( bucket ) )
|
|
pthread_cond_wait( &bucket_being_unlocked, &bucket_mutex );
|
|
bucket_push( bucket );
|
|
pthread_mutex_unlock( &bucket_mutex );
|
|
return all_torrents + bucket;
|
|
}
|
|
|
|
ot_vector *mutex_bucket_lock_by_hash( ot_hash *hash ) {
|
|
unsigned char *local_hash = hash[0];
|
|
int bucket = ( local_hash[0] << 2 ) | ( local_hash[1] >> 6 );
|
|
|
|
/* Can block */
|
|
mutex_bucket_lock( bucket );
|
|
|
|
return all_torrents + bucket;
|
|
}
|
|
|
|
void mutex_bucket_unlock( int bucket ) {
|
|
pthread_mutex_lock( &bucket_mutex );
|
|
bucket_remove( bucket );
|
|
pthread_cond_broadcast( &bucket_being_unlocked );
|
|
pthread_mutex_unlock( &bucket_mutex );
|
|
}
|
|
|
|
void mutex_bucket_unlock_by_hash( ot_hash *hash ) {
|
|
unsigned char *local_hash = hash[0];
|
|
int bucket = ( local_hash[0] << 2 ) | ( local_hash[1] >> 6 );
|
|
mutex_bucket_unlock( bucket );
|
|
}
|
|
|
|
void mutex_init( ) {
|
|
pthread_mutex_init(&bucket_mutex, NULL);
|
|
pthread_cond_init (&bucket_being_unlocked, NULL);
|
|
byte_zero( all_torrents, sizeof( all_torrents ) );
|
|
}
|
|
|
|
void mutex_deinit( ) {
|
|
pthread_mutex_destroy(&bucket_mutex);
|
|
pthread_cond_destroy(&bucket_being_unlocked);
|
|
byte_zero( all_torrents, sizeof( all_torrents ) );
|
|
}
|