void
_tr_blocklistFree( tr_blocklist * b )
{
blocklistClose( b );
tr_free( b->filename );
tr_free( b );
}
void
tr_blocklistFileFree (tr_blocklistFile * b)
{
blocklistClose (b);
tr_free (b->filename);
tr_free (b);
}
static void blocklistLoad(tr_blocklistFile* b)
{
tr_sys_file_t fd;
uint64_t byteCount;
tr_sys_path_info info;
char* base;
tr_error* error = NULL;
char const* err_fmt = _("Couldn't read \"%1$s\": %2$s");
blocklistClose(b);
if (!tr_sys_path_get_info(b->filename, 0, &info, NULL))
{
return;
}
byteCount = info.size;
if (byteCount == 0)
{
return;
}
fd = tr_sys_file_open(b->filename, TR_SYS_FILE_READ, 0, &error);
if (fd == TR_BAD_SYS_FILE)
{
tr_logAddError(err_fmt, b->filename, error->message);
tr_error_free(error);
return;
}
b->rules = tr_sys_file_map_for_reading(fd, 0, byteCount, &error);
if (b->rules == NULL)
{
tr_logAddError(err_fmt, b->filename, error->message);
tr_sys_file_close(fd, NULL);
tr_error_free(error);
return;
}
b->fd = fd;
b->byteCount = byteCount;
b->ruleCount = byteCount / sizeof(struct tr_ipv4_range);
base = tr_sys_path_basename(b->filename, NULL);
tr_logAddInfo(_("Blocklist \"%s\" contains %zu entries"), base, b->ruleCount);
tr_free(base);
}
static void
blocklistLoad( tr_blocklist * b )
{
int fd;
struct stat st;
const char * err_fmt = _( "Couldn't read \"%1$s\": %2$s" );
blocklistClose( b );
if( stat( b->filename, &st ) == -1 )
return;
fd = open( b->filename, O_RDONLY );
if( fd == -1 )
{
tr_err( err_fmt, b->filename, tr_strerror( errno ) );
return;
}
#ifndef WIN32
b->rules = mmap( NULL, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0 );
#else
b->rules = mmap( NULL, st.st_size, 0, 0, fd, 0 );
#endif
if( !b->rules )
{
tr_err( err_fmt, b->filename, tr_strerror( errno ) );
close( fd );
return;
}
b->byteCount = st.st_size;
b->ruleCount = st.st_size / sizeof( struct tr_ip_range );
b->fd = fd;
{
char * base = tr_basename( b->filename );
tr_inf( _( "Blocklist \"%s\" contains %'zu entries" ), base, b->ruleCount );
tr_free( base );
}
}
static void
blocklistLoad (tr_blocklist * b)
{
int fd;
size_t byteCount;
struct stat st;
const char * err_fmt = _("Couldn't read \"%1$s\": %2$s");
blocklistClose (b);
if (stat (b->filename, &st) == -1)
return;
fd = open (b->filename, O_RDONLY | O_BINARY);
if (fd == -1)
{
tr_err (err_fmt, b->filename, tr_strerror (errno));
return;
}
byteCount = (size_t) st.st_size;
b->rules = mmap (NULL, byteCount, PROT_READ, MAP_PRIVATE, fd, 0);
if (!b->rules)
{
tr_err (err_fmt, b->filename, tr_strerror (errno));
close (fd);
return;
}
b->fd = fd;
b->byteCount = byteCount;
b->ruleCount = byteCount / sizeof (struct tr_ipv4_range);
{
char * base = tr_basename (b->filename);
tr_inf (_("Blocklist \"%s\" contains %zu entries"), base, b->ruleCount);
tr_free (base);
}
}
int
tr_blocklistFileSetContent (tr_blocklistFile * b, const char * filename)
{
tr_sys_file_t in;
tr_sys_file_t out;
int inCount = 0;
char line[2048];
const char * err_fmt = _("Couldn't read \"%1$s\": %2$s");
struct tr_ipv4_range * ranges = NULL;
size_t ranges_alloc = 0;
size_t ranges_count = 0;
tr_error * error = NULL;
if (!filename)
{
blocklistDelete (b);
return 0;
}
in = tr_sys_file_open (filename, TR_SYS_FILE_READ, 0, &error);
if (in == TR_BAD_SYS_FILE)
{
tr_logAddError (err_fmt, filename, error->message);
tr_error_free (error);
return 0;
}
blocklistClose (b);
out = tr_sys_file_open (b->filename,
TR_SYS_FILE_WRITE | TR_SYS_FILE_CREATE | TR_SYS_FILE_TRUNCATE,
0666, &error);
if (out == TR_BAD_SYS_FILE)
{
tr_logAddError (err_fmt, b->filename, error->message);
tr_error_free (error);
tr_sys_file_close (in, NULL);
return 0;
}
/* load the rules into memory */
while (tr_sys_file_read_line (in, line, sizeof (line), NULL))
{
struct tr_ipv4_range range;
++inCount;
if (!parseLine (line, &range))
{
/* don't try to display the actual lines - it causes issues */
tr_logAddError (_("blocklist skipped invalid address at line %d"), inCount);
continue;
}
if (ranges_alloc == ranges_count)
{
ranges_alloc += 4096; /* arbitrary */
ranges = tr_renew (struct tr_ipv4_range, ranges, ranges_alloc);
}
ranges[ranges_count++] = range;
}
static void
blocklistDelete (tr_blocklistFile * b)
{
blocklistClose (b);
tr_sys_path_remove (b->filename, NULL);
}
int
_tr_blocklistSetContent( tr_blocklist * b,
const char * filename )
{
FILE * in;
FILE * out;
int inCount = 0;
int outCount = 0;
char line[2048];
const char * err_fmt = _( "Couldn't read \"%1$s\": %2$s" );
if( !filename )
{
blocklistDelete( b );
return 0;
}
in = fopen( filename, "rb" );
if( !in )
{
tr_err( err_fmt, filename, tr_strerror( errno ) );
return 0;
}
blocklistClose( b );
out = fopen( b->filename, "wb+" );
if( !out )
{
tr_err( err_fmt, b->filename, tr_strerror( errno ) );
fclose( in );
return 0;
}
while( fgets( line, sizeof( line ), in ) != NULL )
{
char * walk;
struct tr_ip_range range;
++inCount;
/* zap the linefeed */
if(( walk = strchr( line, '\r' ))) *walk = '\0';
if(( walk = strchr( line, '\n' ))) *walk = '\0';
if( !parseLine( line, &range ) )
{
/* don't try to display the actual lines - it causes issues */
tr_err( _( "blocklist skipped invalid address at line %d" ), inCount );
continue;
}
if( fwrite( &range, sizeof( struct tr_ip_range ), 1, out ) != 1 )
{
tr_err( _( "Couldn't save file \"%1$s\": %2$s" ), b->filename,
tr_strerror( errno ) );
break;
}
++outCount;
}
{
char * base = tr_basename( b->filename );
tr_inf( _( "Blocklist \"%s\" updated with %d entries" ), base, outCount );
tr_free( base );
}
fclose( out );
fclose( in );
blocklistLoad( b );
return outCount;
}
static void
blocklistDelete( tr_blocklist * b )
{
blocklistClose( b );
unlink( b->filename );
}
int
tr_blocklistFileSetContent (tr_blocklistFile * b, const char * filename)
{
FILE * in;
FILE * out;
int inCount = 0;
char line[2048];
const char * err_fmt = _("Couldn't read \"%1$s\": %2$s");
struct tr_ipv4_range * ranges = NULL;
size_t ranges_alloc = 0;
size_t ranges_count = 0;
if (!filename)
{
blocklistDelete (b);
return 0;
}
in = fopen (filename, "rb");
if (in == NULL)
{
tr_logAddError (err_fmt, filename, tr_strerror (errno));
return 0;
}
blocklistClose (b);
out = fopen (b->filename, "wb+");
if (out == NULL)
{
tr_logAddError (err_fmt, b->filename, tr_strerror (errno));
fclose (in);
return 0;
}
/* load the rules into memory */
while (fgets (line, sizeof (line), in) != NULL)
{
char * walk;
struct tr_ipv4_range range;
++inCount;
/* zap the linefeed */
if ((walk = strchr (line, '\r'))) *walk = '\0';
if ((walk = strchr (line, '\n'))) *walk = '\0';
if (!parseLine (line, &range))
{
/* don't try to display the actual lines - it causes issues */
tr_logAddError (_("blocklist skipped invalid address at line %d"), inCount);
continue;
}
if (ranges_alloc == ranges_count)
{
ranges_alloc += 4096; /* arbitrary */
ranges = tr_renew (struct tr_ipv4_range, ranges, ranges_alloc);
}
ranges[ranges_count++] = range;
}
static void
blocklistDelete (tr_blocklistFile * b)
{
blocklistClose (b);
tr_remove (b->filename);
}
int
_tr_blocklistSetContent( tr_blocklist * b,
const char * filename )
{
FILE * in;
FILE * out;
char * line;
int lineCount = 0;
const char * err_fmt = _( "Couldn't read \"%1$s\": %2$s" );
if( !filename )
{
blocklistDelete( b );
return 0;
}
in = fopen( filename, "r" );
if( !in )
{
tr_err( err_fmt, filename, tr_strerror( errno ) );
return 0;
}
blocklistClose( b );
out = fopen( b->filename, "wb+" );
if( !out )
{
tr_err( err_fmt, b->filename, tr_strerror( errno ) );
fclose( in );
return 0;
}
while( !fggets( &line, in ) )
{
char * rangeBegin;
char * rangeEnd;
char * crpos;
tr_address addr;
struct tr_ip_range range;
rangeBegin = strrchr( line, ':' );
if( !rangeBegin ) {
free( line );
continue;
}
++rangeBegin;
rangeEnd = strchr( rangeBegin, '-' );
if( !rangeEnd ) {
free( line );
continue;
}
*rangeEnd++ = '\0';
if(( crpos = strchr( rangeEnd, '\r' )))
*crpos = '\0';
if( !tr_pton( rangeBegin, &addr ) )
tr_err( "blocklist skipped invalid address [%s]\n", rangeBegin );
range.begin = ntohl( addr.addr.addr4.s_addr );
if( !tr_pton( rangeEnd, &addr ) )
tr_err( "blocklist skipped invalid address [%s]\n", rangeEnd );
range.end = ntohl( addr.addr.addr4.s_addr );
free( line );
if( fwrite( &range, sizeof( struct tr_ip_range ), 1, out ) != 1 )
{
tr_err( _(
"Couldn't save file \"%1$s\": %2$s" ), b->filename,
tr_strerror( errno ) );
break;
}
++lineCount;
}
{
char * base = tr_basename( b->filename );
tr_inf( _( "Blocklist \"%1$s\" updated with %2$'d entries" ), base, lineCount );
tr_free( base );
}
fclose( out );
fclose( in );
blocklistLoad( b );
return lineCount;
}