1 files changed, 95 insertions, 0 deletions

diff --git a/ot_accesslist.c b/ot_accesslist.c
new file mode 100644
index 0000000..e63cff8
--- /dev/null
+++ b/ot_accesslist.c
@@ -0,0 +1,95 @@
+/* This software was written by Dirk Engling <erdgeist@erdgeist.org>
+   It is considered beerware. Prost. Skol. Cheers or whatever. */
+
+/* System */
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+
+/* Libowfat */
+#include "byte.h"
+#include "scan.h"
+
+/* Opentracker */
+#include "ot_accesslist.h"
+
+/* GLOBAL VARIABLES */
+#ifdef WANT_ACCESS_CONTROL
+static char *accesslist_filename = NULL;
+static ot_vector accesslist;
+
+static void accesslist_reset( void ) {
+  free( accesslist.data );
+  byte_zero( &accesslist, sizeof( accesslist ) );
+}
+
+static int accesslist_addentry( ot_hash *infohash ) {
+  int em;
+  void *insert = vector_find_or_insert( &accesslist, infohash, OT_HASH_COMPARE_SIZE, OT_HASH_COMPARE_SIZE, &em );
+
+  if( !insert )
+    return -1;
+
+  memmove( insert, infohash, OT_HASH_COMPARE_SIZE );
+
+  return 0;
+}
+
+/* Read initial access list */
+static void accesslist_readfile( int foo ) {
+  FILE *  accesslist_filehandle;
+  ot_hash infohash;
+  foo = foo;
+
+  accesslist_filehandle = fopen( accesslist_filename, "r" );
+
+  /* Free accesslist vector in trackerlogic.c*/
+  accesslist_reset();
+
+  if( accesslist_filehandle == NULL ) {
+    fprintf( stderr, "Warning: Can't open accesslist file: %s (but will try to create it later, if necessary and possible).", accesslist_filename );
+    return;
+  }
+
+  /* We do ignore anything that is not of the form "^[:xdigit:]{40}[^:xdigit:].*" */
+  while( fgets( static_inbuf, sizeof(static_inbuf), accesslist_filehandle ) ) {
+    int i;
+    for( i=0; i<20; ++i ) {
+      int eger = 16 * scan_fromhex( static_inbuf[ 2*i ] ) + scan_fromhex( static_inbuf[ 1 + 2*i ] );
+      if( eger < 0 )
+        continue;
+      infohash[i] = eger;
+    }
+    if( scan_fromhex( static_inbuf[ 40 ] ) >= 0 )
+      continue;
+
+    /* Append accesslist to accesslist vector */
+    accesslist_addentry( &infohash );
+  }
+
+  fclose( accesslist_filehandle );
+}
+
+int accesslist_hashisvalid( ot_hash *hash ) {
+  int exactmatch;
+  binary_search( hash, accesslist.data, accesslist.size, OT_HASH_COMPARE_SIZE, OT_HASH_COMPARE_SIZE, &exactmatch );
+
+#ifdef WANT_BLACKLISTING
+  exactmatch = !exactmatch;
+#endif
+
+  return exactmatch;
+}
+
+void accesslist_init( char *accesslist_filename_in ) {
+  byte_zero( &accesslist, sizeof( accesslist ) );
+
+  /* Passing "0" since read_blacklist_file also is SIGHUP handler */
+  if( accesslist_filename_in ) {
+    accesslist_filename = accesslist_filename_in;
+    accesslist_readfile( 0 );
+    signal( SIGHUP,  accesslist_readfile );
+  }
+}
+
+#endif