ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/deliantra/server/lib/cf.pm
(Generate patch)

Comparing deliantra/server/lib/cf.pm (file contents):
Revision 1.203 by root, Thu Feb 1 19:40:42 2007 UTC vs.
Revision 1.215 by root, Thu Feb 15 03:19:02 2007 UTC

15use Coro::Timer; 15use Coro::Timer;
16use Coro::Signal; 16use Coro::Signal;
17use Coro::Semaphore; 17use Coro::Semaphore;
18use Coro::AIO; 18use Coro::AIO;
19 19
20use BDB ();
20use Data::Dumper; 21use Data::Dumper;
21use Digest::MD5; 22use Digest::MD5;
22use Fcntl; 23use Fcntl;
23use IO::AIO 2.32 (); 24use IO::AIO 2.32 ();
24use YAML::Syck (); 25use YAML::Syck ();
25use Time::HiRes; 26use Time::HiRes;
27use Compress::LZF;
28
29Compress::LZF::sfreeze_cr { }; # prime Compress::LZF so it does not use require later
26 30
27use Event; $Event::Eval = 1; # no idea why this is required, but it is 31use Event; $Event::Eval = 1; # no idea why this is required, but it is
28 32
29sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload 33sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload
30 34
45our $LIBDIR = datadir . "/ext"; 49our $LIBDIR = datadir . "/ext";
46 50
47our $TICK = MAX_TIME * 1e-6; 51our $TICK = MAX_TIME * 1e-6;
48our $TICK_WATCHER; 52our $TICK_WATCHER;
49our $AIO_POLL_WATCHER; 53our $AIO_POLL_WATCHER;
50our $WRITE_RUNTIME_WATCHER; 54our $NEXT_RUNTIME_WRITE; # when should the runtime file be written
51our $NEXT_TICK; 55our $NEXT_TICK;
52our $NOW; 56our $NOW;
57our $USE_FSYNC = 1; # use fsync to write maps - default off
58
59our $BDB_POLL_WATCHER;
60our $DB_ENV;
53 61
54our %CFG; 62our %CFG;
55 63
56our $UPTIME; $UPTIME ||= time; 64our $UPTIME; $UPTIME ||= time;
57our $RUNTIME; 65our $RUNTIME;
58 66
59our %PLAYER; # all users 67our %PLAYER; # all users
60our %MAP; # all maps 68our %MAP; # all maps
61our $LINK_MAP; # the special {link} map, which is always available 69our $LINK_MAP; # the special {link} map, which is always available
62our $RANDOM_MAPS = cf::localdir . "/random"; 70our $RANDOM_MAPS = cf::localdir . "/random";
71our $BDB_ENV_DIR = cf::localdir . "/db";
63 72
64our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal; 73our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal;
65our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal; 74our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal;
66 75
67# used to convert map paths into valid unix filenames by replacing / by ∕ 76# used to convert map paths into valid unix filenames by replacing / by ∕
80mkdir cf::localdir; 89mkdir cf::localdir;
81mkdir cf::localdir . "/" . cf::playerdir; 90mkdir cf::localdir . "/" . cf::playerdir;
82mkdir cf::localdir . "/" . cf::tmpdir; 91mkdir cf::localdir . "/" . cf::tmpdir;
83mkdir cf::localdir . "/" . cf::uniquedir; 92mkdir cf::localdir . "/" . cf::uniquedir;
84mkdir $RANDOM_MAPS; 93mkdir $RANDOM_MAPS;
94mkdir $BDB_ENV_DIR;
85 95
86our $EMERGENCY_POSITION; 96our $EMERGENCY_POSITION;
87 97
88sub cf::map::normalise; 98sub cf::map::normalise;
89 99
263sub freeze_mainloop { 273sub freeze_mainloop {
264 return unless $TICK_WATCHER->is_active; 274 return unless $TICK_WATCHER->is_active;
265 275
266 my $guard = Coro::guard { 276 my $guard = Coro::guard {
267 $TICK_WATCHER->start; 277 $TICK_WATCHER->start;
268 $WRITE_RUNTIME_WATCHER->start;
269 }; 278 };
270 $WRITE_RUNTIME_WATCHER->stop;
271 $TICK_WATCHER->stop; 279 $TICK_WATCHER->stop;
272 $guard 280 $guard
273} 281}
274 282
275=item cf::async { BLOCK } 283=item cf::async { BLOCK }
358 # 60 allows for the watchdog to kill the server. 366 # 60 allows for the watchdog to kill the server.
359 367
360 (aio_write $fh, 0, (length $value), $value, 0) <= 0 368 (aio_write $fh, 0, (length $value), $value, 0) <= 0
361 and return; 369 and return;
362 370
371 # always fsync - this file is important
363 aio_fsync $fh 372 aio_fsync $fh
364 and return; 373 and return;
365 374
366 close $fh 375 close $fh
367 or return; 376 or return;
752 $filename, length $$rdata, scalar @$objs; 761 $filename, length $$rdata, scalar @$objs;
753 762
754 if (my $fh = aio_open "$filename~", O_WRONLY | O_CREAT, 0600) { 763 if (my $fh = aio_open "$filename~", O_WRONLY | O_CREAT, 0600) {
755 chmod SAVE_MODE, $fh; 764 chmod SAVE_MODE, $fh;
756 aio_write $fh, 0, (length $$rdata), $$rdata, 0; 765 aio_write $fh, 0, (length $$rdata), $$rdata, 0;
757 aio_fsync $fh; 766 aio_fsync $fh if $cf::USE_FSYNC;
758 close $fh; 767 close $fh;
759 768
760 if (@$objs) { 769 if (@$objs) {
761 if (my $fh = aio_open "$filename.pst~", O_WRONLY | O_CREAT, 0600) { 770 if (my $fh = aio_open "$filename.pst~", O_WRONLY | O_CREAT, 0600) {
762 chmod SAVE_MODE, $fh; 771 chmod SAVE_MODE, $fh;
763 my $data = Storable::nfreeze { version => 1, objs => $objs }; 772 my $data = Storable::nfreeze { version => 1, objs => $objs };
764 aio_write $fh, 0, (length $data), $data, 0; 773 aio_write $fh, 0, (length $data), $data, 0;
765 aio_fsync $fh; 774 aio_fsync $fh if $cf::USE_FSYNC;
766 close $fh; 775 close $fh;
767 aio_rename "$filename.pst~", "$filename.pst"; 776 aio_rename "$filename.pst~", "$filename.pst";
768 } 777 }
769 } else { 778 } else {
770 aio_unlink "$filename.pst"; 779 aio_unlink "$filename.pst";
1598 return if $self->players; 1607 return if $self->players;
1599 return if $self->isa ("ext::map_per_player");#d# 1608 return if $self->isa ("ext::map_per_player");#d#
1600 1609
1601 warn "resetting map ", $self->path;#d# 1610 warn "resetting map ", $self->path;#d#
1602 1611
1612 $self->in_memory (cf::MAP_SWAPPED);
1613
1614 # need to save uniques path
1615 unless ($self->{deny_save}) {
1616 my $uniq = $self->uniq_path; utf8::encode $uniq;
1617
1618 $self->_save_objects ($uniq, cf::IO_UNIQUES)
1619 if $uniq;
1620 }
1621
1603 delete $cf::MAP{$self->path}; 1622 delete $cf::MAP{$self->path};
1604 1623
1605 $self->in_memory (cf::MAP_SWAPPED);
1606 $self->clear; 1624 $self->clear;
1607 1625
1608 $_->clear_links_to ($self) for values %cf::MAP; 1626 $_->clear_links_to ($self) for values %cf::MAP;
1609 1627
1610 $self->unlink_save; 1628 $self->unlink_save;
1800 $map->load; 1818 $map->load;
1801 $map->load_diag; 1819 $map->load_diag;
1802 1820
1803 return unless $self->contr->active; 1821 return unless $self->contr->active;
1804 $self->activate_recursive; 1822 $self->activate_recursive;
1823
1824 local $self->{_prev_pos} = $link_pos; # ugly hack for rent.ext
1805 $self->enter_map ($map, $x, $y); 1825 $self->enter_map ($map, $x, $y);
1806} 1826}
1807 1827
1808cf::player->attach ( 1828cf::player->attach (
1809 on_logout => sub { 1829 on_logout => sub {
1822 # try to abort aborted map switching on player login :) 1842 # try to abort aborted map switching on player login :)
1823 # should happen only on crashes 1843 # should happen only on crashes
1824 if ($pl->ob->{_link_pos}) { 1844 if ($pl->ob->{_link_pos}) {
1825 $pl->ob->enter_link; 1845 $pl->ob->enter_link;
1826 (async { 1846 (async {
1827 # we need this sleep as the login has a concurrent enter_exit running
1828 # and this sleep increases chances of the player not ending up in scorn
1829 $pl->ob->reply (undef, 1847 $pl->ob->reply (undef,
1830 "There was an internal problem at your last logout, " 1848 "There was an internal problem at your last logout, "
1831 . "the server will try to bring you to your intended destination in a second.", 1849 . "the server will try to bring you to your intended destination in a second.",
1832 cf::NDI_RED); 1850 cf::NDI_RED);
1851 # we need this sleep as the login has a concurrent enter_exit running
1852 # and this sleep increases chances of the player not ending up in scorn
1833 Coro::Timer::sleep 1; 1853 Coro::Timer::sleep 1;
1834 $pl->ob->leave_link; 1854 $pl->ob->leave_link;
1835 })->prio (2); 1855 })->prio (2);
1836 } 1856 }
1837 }, 1857 },
2174The parameter C<$family> should best start with the name of the extension 2194The parameter C<$family> should best start with the name of the extension
2175using it, it should be unique. 2195using it, it should be unique.
2176 2196
2177=over 4 2197=over 4
2178 2198
2179=item $hashref = cf::db_get $family
2180
2181Return a hashref for use by the extension C<$family>, which can be
2182modified. After modifications, you have to call C<cf::db_dirty> or
2183C<cf::db_sync>.
2184
2185=item $value = cf::db_get $family => $key 2199=item $value = cf::db_get $family => $key
2186 2200
2187Returns a single value from the database 2201Returns a single value from the database.
2188
2189=item cf::db_put $family => $hashref
2190
2191Stores the given family hashref into the database. Updates are delayed, if
2192you want the data to be synced to disk immediately, use C<cf::db_sync>.
2193 2202
2194=item cf::db_put $family => $key => $value 2203=item cf::db_put $family => $key => $value
2195 2204
2196Stores the given C<$value> in the family hash. Updates are delayed, if you 2205Stores the given C<$value> in the family.
2197want the data to be synced to disk immediately, use C<cf::db_sync>.
2198
2199=item cf::db_dirty
2200
2201Marks the database as dirty, to be updated at a later time.
2202
2203=item cf::db_sync
2204
2205Immediately write the database to disk I<if it is dirty>.
2206 2206
2207=cut 2207=cut
2208 2208
2209our $DB; 2209our $DB;
2210 2210
2211{ 2211sub db_init {
2212 unless ($DB) {
2213 $DB = BDB::db_create $DB_ENV;
2214
2215 cf::sync_job {
2216 eval {
2217 $DB->set_flags (BDB::CHKSUM);
2218
2219 BDB::db_open $DB, undef, "db", undef, BDB::BTREE,
2220 BDB::CREATE | BDB::AUTO_COMMIT, 0666;
2221 cf::cleanup "db_open(db): $!" if $!;
2222 };
2223 cf::cleanup "db_open(db): $@" if $@;
2224 };
2225
2212 my $path = cf::localdir . "/database.pst"; 2226 my $path = cf::localdir . "/database.pst";
2227 if (stat $path) {
2228 cf::sync_job {
2229 my $pst = Storable::retrieve $path;
2213 2230
2214 sub db_load() { 2231 cf::db_put (board => data => $pst->{board});
2215 $DB = stat $path ? Storable::retrieve $path : { }; 2232 cf::db_put (guildrules => data => $pst->{guildrules});
2216 } 2233 cf::db_put (rent => balance => $pst->{rent}{balance});
2234 BDB::db_env_txn_checkpoint $DB_ENV;
2217 2235
2218 my $pid; 2236 unlink $path;
2219
2220 sub db_save() {
2221 waitpid $pid, 0 if $pid;
2222 if (0 == ($pid = fork)) {
2223 $DB->{_meta}{version} = 1;
2224 Storable::nstore $DB, "$path~";
2225 rename "$path~", $path;
2226 cf::_exit 0 if defined $pid;
2227 } 2237 };
2228 }
2229
2230 my $dirty;
2231
2232 sub db_sync() {
2233 db_save if $dirty;
2234 undef $dirty;
2235 }
2236
2237 my $idle = Event->idle (
2238 reentrant => 0,
2239 min => 10,
2240 max => 20,
2241 repeat => 0,
2242 data => WF_AUTOCANCEL,
2243 cb => \&db_sync,
2244 );
2245
2246 sub db_dirty() {
2247 $dirty = 1;
2248 $idle->start;
2249 }
2250
2251 sub db_get($;$) {
2252 @_ >= 2
2253 ? $DB->{$_[0]}{$_[1]}
2254 : ($DB->{$_[0]} ||= { })
2255 }
2256
2257 sub db_put($$;$) {
2258 if (@_ >= 3) {
2259 $DB->{$_[0]}{$_[1]} = $_[2];
2260 } else {
2261 $DB->{$_[0]} = $_[1];
2262 }
2263 db_dirty;
2264 }
2265
2266 cf::global->attach (
2267 prio => 10000,
2268 on_cleanup => sub {
2269 db_sync;
2270 }, 2238 }
2271 ); 2239 }
2240}
2241
2242sub db_get($$) {
2243 my $key = "$_[0]/$_[1]";
2244
2245 cf::sync_job {
2246 BDB::db_get $DB, undef, $key, my $data;
2247
2248 $! ? ()
2249 : Compress::LZF::sthaw $data
2250 }
2251}
2252
2253sub db_put($$$) {
2254 BDB::dbreq_pri 4;
2255 BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { };
2272} 2256}
2273 2257
2274############################################################################# 2258#############################################################################
2275# the server's init and main functions 2259# the server's init and main functions
2276 2260
2312 Event::one_event; 2296 Event::one_event;
2313 })->prio (Coro::PRIO_MAX); 2297 })->prio (Coro::PRIO_MAX);
2314 }; 2298 };
2315 2299
2316 cfg_load; 2300 cfg_load;
2317 db_load; 2301 db_init;
2318 load_extensions; 2302 load_extensions;
2319 2303
2320 $TICK_WATCHER->start; 2304 $TICK_WATCHER->start;
2321 Event::loop; 2305 Event::loop;
2322} 2306}
2361 my $map = $cf::MAP{$path} or next; 2345 my $map = $cf::MAP{$path} or next;
2362 $map->valid or next; 2346 $map->valid or next;
2363 $map->save; 2347 $map->save;
2364 } 2348 }
2365 warn "end emergency map save\n"; 2349 warn "end emergency map save\n";
2350
2351 warn "begin emergency database checkpoint\n";
2352 BDB::db_env_txn_checkpoint $DB_ENV;
2353 warn "end emergency database checkpoint\n";
2366 }; 2354 };
2367 2355
2368 warn "leave emergency perl save\n"; 2356 warn "leave emergency perl save\n";
2357}
2358
2359sub post_cleanup {
2360 my ($make_core) = @_;
2361
2362 warn Carp::longmess "post_cleanup backtrace"
2363 if $make_core;
2369} 2364}
2370 2365
2371sub reload() { 2366sub reload() {
2372 # can/must only be called in main 2367 # can/must only be called in main
2373 if ($Coro::current != $Coro::main) { 2368 if ($Coro::current != $Coro::main) {
2375 return; 2370 return;
2376 } 2371 }
2377 2372
2378 warn "reloading..."; 2373 warn "reloading...";
2379 2374
2380 warn "cancelling server ticker"; 2375 warn "entering sync_job";
2381 $TICK_WATCHER->cancel;
2382 2376
2377 cf::sync_job {
2378 cf::write_runtime; # external watchdog should not bark
2383 cf::emergency_save; 2379 cf::emergency_save;
2380 cf::write_runtime; # external watchdog should not bark
2384 2381
2385 eval { 2382 warn "syncing database to disk";
2383 BDB::db_env_txn_checkpoint $DB_ENV;
2384
2386 # if anything goes wrong in here, we should simply crash as we already saved 2385 # if anything goes wrong in here, we should simply crash as we already saved
2387
2388 warn "syncing database to disk";
2389 cf::db_sync;
2390 2386
2391 warn "cancelling all WF_AUTOCANCEL watchers"; 2387 warn "cancelling all WF_AUTOCANCEL watchers";
2392 for (Event::all_watchers) { 2388 for (Event::all_watchers) {
2393 $_->cancel if $_->data & WF_AUTOCANCEL; 2389 $_->cancel if $_->data & WF_AUTOCANCEL;
2394 } 2390 }
2395 2391
2396 warn "flushing outstanding aio requests"; 2392 warn "flushing outstanding aio requests";
2397 for (;;) { 2393 for (;;) {
2394 BDB::flush;
2398 IO::AIO::flush; 2395 IO::AIO::flush;
2399 Coro::cede; 2396 Coro::cede;
2400 last unless IO::AIO::nreqs; 2397 last unless IO::AIO::nreqs || BDB::nreqs;
2401 warn "iterate..."; 2398 warn "iterate...";
2402 } 2399 }
2403 2400
2404 warn "cancelling all extension coros"; 2401 warn "cancelling all extension coros";
2405 $_->cancel for values %EXT_CORO; 2402 $_->cancel for values %EXT_CORO;
2460 require cf; 2457 require cf;
2461 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt 2458 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt
2462 2459
2463 warn "loading config and database again"; 2460 warn "loading config and database again";
2464 cf::cfg_load; 2461 cf::cfg_load;
2465 cf::db_load;
2466 2462
2467 warn "loading extensions"; 2463 warn "loading extensions";
2468 cf::load_extensions; 2464 cf::load_extensions;
2469 2465
2470 warn "reattaching attachments to objects/players"; 2466 warn "reattaching attachments to objects/players";
2473 reattach $_ for values %MAP; 2469 reattach $_ for values %MAP;
2474 2470
2475 warn "loading reloadable resources"; 2471 warn "loading reloadable resources";
2476 load_resources; 2472 load_resources;
2477 2473
2478 warn "restarting server ticker"; 2474 warn "leaving sync_job";
2479 2475
2480 $TICK_WATCHER->start; 2476 1
2481 }; 2477 } or do {
2482
2483 if ($@) {
2484 warn $@; 2478 warn $@;
2485 warn "error while reloading, exiting."; 2479 warn "error while reloading, exiting.";
2486 exit 1; 2480 exit 1;
2487 } 2481 };
2488 2482
2489 warn "reloaded"; 2483 warn "reloaded";
2490}; 2484};
2491 2485
2492our $RELOAD_WATCHER; # used only during reload 2486our $RELOAD_WATCHER; # used only during reload
2533 2527
2534 cf::server_tick; # one server iteration 2528 cf::server_tick; # one server iteration
2535 $RUNTIME += $TICK; 2529 $RUNTIME += $TICK;
2536 $NEXT_TICK += $TICK; 2530 $NEXT_TICK += $TICK;
2537 2531
2532 if ($NOW >= $NEXT_RUNTIME_WRITE) {
2533 $NEXT_RUNTIME_WRITE = $NOW + 10;
2534 Coro::async_pool {
2535 write_runtime
2536 or warn "ERROR: unable to write runtime file: $!";
2537 };
2538 }
2539
2540
2538 $WAIT_FOR_TICK->broadcast; 2541 $WAIT_FOR_TICK->broadcast;
2539 $WAIT_FOR_TICK_ONE->send if $WAIT_FOR_TICK_ONE->awaited; 2542 $WAIT_FOR_TICK_ONE->send if $WAIT_FOR_TICK_ONE->awaited;
2540 2543
2541 Event::sweep;
2542 Coro::cede_notself;
2543
2544# my $AFTER = Event::time; 2544# my $AFTER = Event::time;
2545# warn $AFTER - $NOW;#d# 2545# warn $AFTER - $NOW;#d#
2546 2546
2547 # if we are delayed by four ticks or more, skip them all 2547 # if we are delayed by four ticks or more, skip them all
2548 $NEXT_TICK = Event::time if Event::time >= $NEXT_TICK + $TICK * 4; 2548 $NEXT_TICK = Event::time if Event::time >= $NEXT_TICK + $TICK * 4;
2550 $TICK_WATCHER->at ($NEXT_TICK); 2550 $TICK_WATCHER->at ($NEXT_TICK);
2551 $TICK_WATCHER->start; 2551 $TICK_WATCHER->start;
2552 }, 2552 },
2553); 2553);
2554 2554
2555{
2556 BDB::max_poll_time $TICK * 0.1;
2557 $BDB_POLL_WATCHER = Event->io (
2558 reentrant => 0,
2559 fd => BDB::poll_fileno,
2560 poll => 'r',
2561 prio => 0,
2562 data => WF_AUTOCANCEL,
2563 cb => \&BDB::poll_cb,
2564 );
2565 BDB::min_parallel 8;
2566
2567 BDB::set_sync_prepare {
2568 my $status;
2569 my $current = $Coro::current;
2570 (
2571 sub {
2572 $status = $!;
2573 $current->ready; undef $current;
2574 },
2575 sub {
2576 Coro::schedule while defined $current;
2577 $! = $status;
2578 },
2579 )
2580 };
2581
2582 unless ($DB_ENV) {
2583 $DB_ENV = BDB::db_env_create;
2584
2585 cf::sync_job {
2586 eval {
2587 BDB::db_env_open
2588 $DB_ENV,
2589 $BDB_ENV_DIR,
2590 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2591 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2592 0666;
2593
2594 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2595
2596 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2597 $DB_ENV->set_lk_detect;
2598 };
2599
2600 cf::cleanup "db_env_open(db): $@" if $@;
2601 };
2602 }
2603}
2604
2605{
2606 IO::AIO::min_parallel 8;
2607
2608 undef $Coro::AIO::WATCHER;
2555IO::AIO::max_poll_time $TICK * 0.1; 2609 IO::AIO::max_poll_time $TICK * 0.1;
2556
2557undef $Coro::AIO::WATCHER;
2558$AIO_POLL_WATCHER = Event->io ( 2610 $AIO_POLL_WATCHER = Event->io (
2559 reentrant => 0, 2611 reentrant => 0,
2612 data => WF_AUTOCANCEL,
2560 fd => IO::AIO::poll_fileno, 2613 fd => IO::AIO::poll_fileno,
2561 poll => 'r', 2614 poll => 'r',
2562 prio => 6, 2615 prio => 6,
2563 data => WF_AUTOCANCEL,
2564 cb => \&IO::AIO::poll_cb, 2616 cb => \&IO::AIO::poll_cb,
2565); 2617 );
2566 2618}
2567$WRITE_RUNTIME_WATCHER = Event->timer (
2568 reentrant => 0,
2569 data => WF_AUTOCANCEL,
2570 after => 1,
2571 interval => 10,
2572 prio => 6, # keep it lowest so it acts like a watchdog
2573 cb => Coro::unblock_sub {
2574 write_runtime
2575 or warn "ERROR: unable to write runtime file: $!";
2576 },
2577);
2578 2619
2579END { cf::emergency_save } 2620END { cf::emergency_save }
2580 2621
25811 26221
2582 2623

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines