cvsroot/tpb/common.pl

#!/opt/bin/perl

use common::sense;
use Compress::LZF;
use EV;
use AnyEvent::HTTP;
use BDB;
use Coro;
use Coro::BDB;
use AnyEvent::DNS;
use Set::IntSpan;
use CBOR::XS;
use Compress::Zlib ();
use Compress::Raw::Zlib ();
use AnyEvent::TLS;

BDB::min_parallel 8;

our @TPB = qw(thepiratebay.la);
our $TPB = $TPB[0];

our $TODAY = int time / 86400;

#our $TPB_SWITCHER = AE::timer 1, 0.02, sub {
#   state $i;
#   $TPB = $TPB[++$i % @TPB];
#};

#############################################################################

$AnyEvent::HTTP::MAX_PER_HOST =
$AnyEvent::HTTP::MAX_PERSISTENT_PER_HOST = 400;

our %stat;

our $statprint = AE::timer 1, 1, sub {
   syswrite STDOUT, "\r" . (join " ", map "$_=$stat{$_}", sort keys %stat) . " "
      if $ENV{VERBOSE};
};

our @DICTIONARY = (do {
   open my $fh, "<:perlio", "dictionary0"
      or die "dictionary0: $!";
   local $/;
   <$fh>
});

#############################################################################

our $db_env = db_env_create;

$db_env->set_flags (BDB::AUTO_COMMIT | BDB::TXN_WRITE_NOSYNC, 1);
$db_env->log_set_config (BDB::LOG_AUTO_REMOVE);
$db_env->set_cachesize (0, 8 * 1024 * 1024);
$db_env->set_lk_max_lockers (20480);
$db_env->set_lk_max_locks (20480);
$db_env->set_lk_max_objects (20480);

mkdir "db", 0777;
db_env_open
  $db_env,
  "db",
  BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL
  | BDB::INIT_TXN | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
  0666;

our %DB;

sub table($) {
   my $db = $DB{$_[0]} = db_create $db_env;
   db_open $db, undef, $_[0], undef, BDB::BTREE, BDB::AUTO_COMMIT | BDB::CREATE, 0666;
   die if $!;
   $db
}

our $db_state = table "state"; # global state
our $db_info  = table "info"; # per torrent state

our $TXN;

sub get($$) {
   db_get $_[0], $TXN, $_[1], my $data, 0;
   $data # will be undef on error
}

sub put($$$) {
   db_put $_[0], $TXN, $_[1], $_[2], 0;
}

sub del($$) {
   db_del $_[0], $TXN, $_[1], 0, sub { };
}

sub sget($) {
   get $db_state, $_[0]
}

sub sput($$) {
   put $db_state, $_[0], $_[1];
}

sub iget($) {
   my $data = get $db_info, $_[0];
   length $data ? decode_cbor $data : []
}

sub iput($$) {
   put $db_info, $_[0], encode_cbor $_[1];
}

our $syncer = AE::timer 5, 5, sub {
   db_env_txn_checkpoint $db_env, 0, 0, 0, sub { };
};

sub dbsync() {
   db_env_txn_checkpoint $db_env;
}

sub dbflush {
   return if $::dbflush_called++;
   db_env_txn_checkpoint $db_env, sub { };
   BDB::set_sync_prepare undef;
   db_close $_ for values %DB;
   db_env_txn_checkpoint $db_env;
   db_env_close $db_env;
}

END {
   dbflush;
}

sub quit {
   EV::unloop;
   #exit 1;
};

our $sigint  = AE::signal INT  => \&quit;
our $sigterm = AE::signal TERM => \&quit;

sub dic_compress($$) {
   my $d = new Compress::Raw::Zlib::Deflate
      -Level        => 9,
      -WindowBits   => -15,
      -Dictionary   => $DICTIONARY[$_[0]],
      -AppendOutput => 1,
   ;

   my $o;

   $d->deflate ($_[1], $o);
   $d->flush ($o);

   $o
}

sub dic_decompress($$) {
   my $i = new Compress::Raw::Zlib::Inflate
      -WindowBits   => -15,
      -Dictionary   => $DICTIONARY[$_[0]],
      -AppendOutput => 1,
      -ConsumeInput => 0,
   ;

   $i->inflate ($_[1], my $o, 1);

   $o
}

#############################################################################

#############################################################################

sub GET {
   my ($data, $hdr);

   for (1..10) {
      http_get $_[0],
         proxy => undef,
         recurse => 0,
         headers => {
#            "user-agent" => "Mozilla/5.0 (X11; Linux x86_64; rv:20.0) Gecko/20100101 Firefox/20.0 Iceweasel/20.0",
            "referer" => undef,
         },
         Coro::rouse_cb;
      ($data, $hdr) = Coro::rouse_wait;

      $hdr->{Status} = 580
         if $data =~ /^(?:Upgrading software|Could not connect to caching server)/;

      $hdr->{Status} = 581
         unless length $data;

      $data = Compress::Zlib::memGunzip $data
         if $hdr->{"content-encoding"} eq "gzip";

      last
         if $hdr->{Status} < 500;

      Coro::AnyEvent::sleep 0.5;
      ++$stat{"retry-$hdr->{Status}"};
   }

   ++$stat{$hdr->{Status}};
   ($data, $hdr)
}

1

Revision:	1.2
Committed:	Sun Sep 27 08:14:50 2015 UTC (11 years ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.1:	+0 -9 lines
Log Message:	* empty log message *
#	User	Rev	Content
1	root	1.1	#!/opt/bin/perl
2
3			use common::sense;
4			use Compress::LZF;
5			use EV;
6			use AnyEvent::HTTP;
7			use BDB;
8			use Coro;
9			use Coro::BDB;
10			use AnyEvent::DNS;
11			use Set::IntSpan;
12			use CBOR::XS;
13			use Compress::Zlib ();
14			use Compress::Raw::Zlib ();
15			use AnyEvent::TLS;
16
17			BDB::min_parallel 8;
18
19			our @TPB = qw(thepiratebay.la);
20			our $TPB = $TPB[0];
21
22			our $TODAY = int time / 86400;
23
24			#our $TPB_SWITCHER = AE::timer 1, 0.02, sub {
25			# state $i;
26			# $TPB = $TPB[++$i % @TPB];
27			#};
28
29			#############################################################################
30
31			$AnyEvent::HTTP::MAX_PER_HOST =
32			$AnyEvent::HTTP::MAX_PERSISTENT_PER_HOST = 400;
33
34			our %stat;
35
36			our $statprint = AE::timer 1, 1, sub {
37			syswrite STDOUT, "\r" . (join " ", map "$_=$stat{$_}", sort keys %stat) . " "
38			if $ENV{VERBOSE};
39			};
40
41			our @DICTIONARY = (do {
42			open my $fh, "<:perlio", "dictionary0"
43			or die "dictionary0: $!";
44			local $/;
45			<$fh>
46			});
47
48			#############################################################################
49
50			our $db_env = db_env_create;
51
52			$db_env->set_flags (BDB::AUTO_COMMIT \| BDB::TXN_WRITE_NOSYNC, 1);
53			$db_env->log_set_config (BDB::LOG_AUTO_REMOVE);
54			$db_env->set_cachesize (0, 8 * 1024 * 1024);
55			$db_env->set_lk_max_lockers (20480);
56			$db_env->set_lk_max_locks (20480);
57			$db_env->set_lk_max_objects (20480);
58
59			mkdir "db", 0777;
60			db_env_open
61			$db_env,
62			"db",
63			BDB::INIT_LOCK \| BDB::INIT_LOG \| BDB::INIT_MPOOL
64			\| BDB::INIT_TXN \| BDB::RECOVER \| BDB::REGISTER \| BDB::USE_ENVIRON \| BDB::CREATE,
65			0666;
66
67			our %DB;
68
69			sub table($) {
70			my $db = $DB{$_[0]} = db_create $db_env;
71			db_open $db, undef, $_[0], undef, BDB::BTREE, BDB::AUTO_COMMIT \| BDB::CREATE, 0666;
72			die if $!;
73			$db
74			}
75
76			our $db_state = table "state"; # global state
77			our $db_info = table "info"; # per torrent state
78
79			our $TXN;
80
81			sub get($$) {
82			db_get $_[0], $TXN, $_[1], my $data, 0;
83			$data # will be undef on error
84			}
85
86			sub put($$$) {
87			db_put $_[0], $TXN, $_[1], $_[2], 0;
88			}
89
90			sub del($$) {
91			db_del $_[0], $TXN, $_[1], 0, sub { };
92			}
93
94			sub sget($) {
95			get $db_state, $_[0]
96			}
97
98			sub sput($$) {
99			put $db_state, $_[0], $_[1];
100			}
101
102			sub iget($) {
103			my $data = get $db_info, $_[0];
104			length $data ? decode_cbor $data : []
105			}
106
107			sub iput($$) {
108			put $db_info, $_[0], encode_cbor $_[1];
109			}
110
111			our $syncer = AE::timer 5, 5, sub {
112			db_env_txn_checkpoint $db_env, 0, 0, 0, sub { };
113			};
114
115			sub dbsync() {
116			db_env_txn_checkpoint $db_env;
117			}
118
119			sub dbflush {
120			return if $::dbflush_called++;
121			db_env_txn_checkpoint $db_env, sub { };
122			BDB::set_sync_prepare undef;
123			db_close $_ for values %DB;
124			db_env_txn_checkpoint $db_env;
125			db_env_close $db_env;
126			}
127
128			END {
129			dbflush;
130			}
131
132			sub quit {
133			EV::unloop;
134			#exit 1;
135			};
136
137			our $sigint = AE::signal INT => \&quit;
138			our $sigterm = AE::signal TERM => \&quit;
139
140			sub dic_compress($$) {
141			my $d = new Compress::Raw::Zlib::Deflate
142			-Level => 9,
143			-WindowBits => -15,
144			-Dictionary => $DICTIONARY[$_[0]],
145			-AppendOutput => 1,
146			;
147
148			my $o;
149
150			$d->deflate ($_[1], $o);
151			$d->flush ($o);
152
153			$o
154			}
155
156			sub dic_decompress($$) {
157			my $i = new Compress::Raw::Zlib::Inflate
158			-WindowBits => -15,
159			-Dictionary => $DICTIONARY[$_[0]],
160			-AppendOutput => 1,
161			-ConsumeInput => 0,
162			;
163
164			$i->inflate ($_[1], my $o, 1);
165
166			$o
167			}
168
169			#############################################################################
170
171			#############################################################################
172
173			sub GET {
174			my ($data, $hdr);
175
176			for (1..10) {
177			http_get $_[0],
178			proxy => undef,
179			recurse => 0,
180			headers => {
181			# "user-agent" => "Mozilla/5.0 (X11; Linux x86_64; rv:20.0) Gecko/20100101 Firefox/20.0 Iceweasel/20.0",
182			"referer" => undef,
183			},
184			Coro::rouse_cb;
185			($data, $hdr) = Coro::rouse_wait;
186
187			$hdr->{Status} = 580
188			if $data =~ /^(?:Upgrading software\|Could not connect to caching server)/;
189
190			$hdr->{Status} = 581
191			unless length $data;
192
193			$data = Compress::Zlib::memGunzip $data
194			if $hdr->{"content-encoding"} eq "gzip";
195
196			last
197			if $hdr->{Status} < 500;
198
199			Coro::AnyEvent::sleep 0.5;
200			++$stat{"retry-$hdr->{Status}"};
201			}
202
203			++$stat{$hdr->{Status}};
204			($data, $hdr)
205			}
206
207			1
208