#include <lib.h>
#include <dirs.h>
#include <cfg.h>
#include <save.h>
#include <daemons.h>
#include <network.h>
#define WGET_D "/secure/daemon/luget"
#ifndef WEB_SOURCE_IP
#define WEB_SOURCE_IP "204.209.44.12"
#endif
#ifndef WEB_SOURCE_NAME
#define WEB_SOURCE_NAME "lpmuds.net"
#endif
#define WEB_SOURCE_PORT 80
#ifndef SAVE_LIVEUPGRADE
#define SAVE_LIVEUPGRADE DIR_SECURE_SAVE "/liveupgrade"
#endif
inherit LIB_DAEMON;
string array allnames = ({});
string array tmpnames;
string reverts_dir, revert_name;
static string upgrade_prefix = "/code/upgrades/"+mudlib_version();
static string reverts_prefix = "/secure/upgrades/reverts/"+mudlib_version();
static string upgrades_txt = "/secure/upgrades/txt";
static string upgrades_files = "/secure/upgrades/files";
static int i = 0;
static int oob = 0;
static object player = 0;
int transver = 0;
static mapping NewFiles = ([]);
void create(){
daemon::create();
SetSaveFile(SAVE_LIVEUPGRADE);
eventRestore(1);
}
int eventBackup(string file){
string tmp, time_str,short_name,filename;
if(!file) return 0;
if(!file_exists(file)) return -1;
if( !(tmp = read_file(file)) ) return -2;
time_str = time()+"."+random_numbers(5);
short_name = last_string_element(file,"/");
if(!revert_name){
revert_name = itoa(time());
}
if(!reverts_prefix){
reverts_prefix = "/secure/upgrades/reverts/"+mudlib_version();
}
if(!directory_exists("/secure/upgrades/reverts/")){
mkdir("/secure/upgrades/reverts");
}
if(!directory_exists(reverts_prefix)){
mkdir(reverts_prefix);
}
if(!directory_exists(reverts_prefix+"/"+revert_name)){
mkdir(reverts_prefix+"/"+revert_name);
}
filename = reverts_prefix+"/"+revert_name+
"/"+short_name+"."+time_str;
write_file(reverts_prefix+"/"+revert_name+"/"+
"/bk.db",short_name+"."+time_str+" : "+file+"\n");
cp(file,filename);
return 1;
}
int eventRevert(string revert_path){
string *files;
if(!revert_path){
return 0;
}
if(!directory_exists(revert_path)){
return -1;
}
if(!file_exists(revert_path+"/bk.db")){
return -2;
}
files = explode(read_file(revert_path+"/bk.db"),"\n");
if(!files || !sizeof(files)){
return -3;
}
foreach(string line in files){
string backup, target;
if(sscanf(line,"%s : %s", backup, target) != 2) continue;
if(!directory_exists(path_prefix(target))){
mkdir_recurse(path_prefix(target));
}
cp(revert_path+"/"+backup, target);
}
return 1;
}
void eventReloads(){
cp("/secure/daemon/update.patch","/secure/daemon/update.c");
reload(UPDATE_D);
reload(WGET_D);
}
int eventCopy(string element){
string path;
if(!NewFiles[element]) return 0;
path = path_prefix(NewFiles[element]);
if(!directory_exists(path)) mkdir_recurse(path);
rename(element, NewFiles[element]);
return 1;
}
mixed cmd(string str) {
string mud = "Dead Souls";
string file;
string orig_str = str;
int foo, tmpint = 0;
object inet = find_object(INET_D);
string *preload_file = explode(read_file(CFG_PRELOAD),"\n");
mixed *socks = socket_status();
NewFiles = ([]);
if(!revert_name){
revert_name = itoa(time());
}
if(!str || str == ""){
write("Try: help liveupgrade.");
return 1;
}
if(str == "alpha"){
if(transver){
transver = 0;
write("Alpha/Stable upgrades disabled.");
return 1;
}
if(!transver){
transver = 1;
write("Alpha/Stable upgrades enabled.");
return 1;
}
}
if(str == "revert"){
string *vers = get_dir("/secure/upgrades/reverts/");
string ver, subver;
if(!vers || !sizeof(vers)){
write("There is no previous backup to revert to.");
return 1;
}
else ver = vers[0];
vers = get_dir("/secure/upgrades/reverts/"+ver+"/");
if(!vers || !sizeof(vers)){
write("There is no backup instance to revert to.");
return 1;
}
else subver = "/secure/upgrades/reverts/"+ver+"/"+vers[0];
eventRevert(subver);
rename(subver,"/secure/upgrades/bak/"+last_string_element(subver,"/"));
rmdir(path_prefix(subver));
write("Reversion complete.");
return 1;
}
foreach(mixed element in socks){
if(element[1] == "DATA_XFER" && element[4] == WEB_SOURCE_IP+"."+WEB_SOURCE_PORT &&
str != "cancel"){
player->eventPrint("A download is still in progress. Please wait until it is complete.");
return 1;
}
}
if(!player && this_player()) player = this_player();
allnames = ({});
if(!player) return 0;
if( !((int)master()->valid_apply(({ "SECURE" }))) )
error("Illegal attempt to access liveupgrade: "+get_stack()+" "+identify(previous_object(-1)));
if(!directory_exists("/secure/upgrades/bak")){
mkdir("/secure/upgrades/bak");
}
if(!directory_exists(reverts_prefix)){
mkdir(reverts_prefix);
}
i = sscanf(mudlib_version(),"2.3a%d",tmpint);
if(i && tmpint < 12) oob = 1;
else {
if(!strsrch(str,"-o ")){
oob = 1;
str = replace_string(str,"-o ","");
}
else oob = 0;
}
i = 0;
foo = sscanf(str,"%s %s",file, mud);
if(!foo || foo < 2) file = str;
if(str == "apply"){
string *files = ({});
player->eventPrint("I hope you backed up...\n");
foreach(string element in get_dir(upgrades_files+"/")){
if(element == "0^0secure0^0sefun0^0mud_info.c"){
object thingy = load_object(upgrades_files+"/"+element);
string vers;
if(thingy){
string current_ver = mudlib_version();
vers = thingy->mudlib_version();
if(((grepp(vers,"a") && !grepp(current_ver, "a")) ||
(!grepp(vers,"a") && grepp(current_ver, "a"))) &&
!transver){
write("This upgrade would cross stable/alpha "
"boundaries, but that has not been enabled "
"with \"liveupgrade alpha\" yet.");
return 1;
}
}
}
files += ({ upgrades_files+"/"+element });
}
foreach(string element in files){
string contents = "";
NewFiles[element] = replace_string(replace_string(element,"0^0","/"),
upgrades_files+"/","");
contents = read_file(element);
if(!contents) contents = "";
if(last(contents,1) != "\n") contents += "\n";
write_file(element, contents, 1);
//eventBackup(NewFiles[element]);
call_out( (: eventBackup :), 0, NewFiles[element]);
if(directory_exists(NewFiles[element])) true();
else {
call_out( (: eventCopy :), 0, element);
//string path = path_prefix(NewFiles[element]);
//if(!directory_exists(path)) mkdir_recurse(path);
//rename(element, NewFiles[element]);
}
}
if(member_array(INET_D,preload_file) == -1 && inet) inet->eventDestruct();
call_out( (: eventReloads :), 10);
RELOAD_D->eventReload(this_object(), 15);
rm("/secure/upgrades/txt/list.txt");
player->eventPrint("\nAlmost done...");
player = 0;
return 1;
}
if(str == "cancel"){
string *files = ({});
foreach(string element in get_dir(upgrades_files+"/")){
files += ({ upgrades_files+"/"+element });
}
foreach(string element in files){
rm(element);
}
rm("/secure/upgrades/txt/list.txt");
player->eventPrint("Cancelled.");
player = 0;
RELOAD_D->eventReload(this_object(), 2);
reload(WGET_D);
return 1;
}
if(oob){
if(!inet){
inet = load_object(INET_D);
player->eventPrint("Starting INET_D.");
if(member_array(INET_D,preload_file) == -1)
player->eventPrint("When you complete the upgrade by using the \"apply\" keyword, the "
"inet daemon will be shut down, since you do not have it enabled by "
"default. Please remember to either apply the upgrades when the downloading "
"is complete, or manually shut down INET_D with the command: mudconfig inet stop\n");
}
if(!inet){
player->eventPrint("There is a problem with INET_D. The upgrade will not proceed.");
return 1;
}
if(!INET_D->GetService("oob")){
player->eventPrint("The OOB service is not enabled. Enabling it now.");
INET_D->AddService("oob",OFFSET_OOB,LIB_OOB,0);
}
if(!INET_D->GetService("oob")){
player->eventPrint("There was a problem enabling the OOB service. The upgrade will not proceed.");
return 1;
}
if(!INET_D->GetServer("oob")){
player->eventPrint("The OOB service is not started. Starting it now.");
INET_D->eventStartServer("oob");
}
if(!INET_D->GetServer("oob")){
player->eventPrint("There was a problem starting the OOB service. The upgrade will not proceed.");
return 1;
}
if(foo < 2) {
mud = LIVEUPGRADE_SERVER;
file = str;
}
if(!file){
return this_object()->GetHelp();
}
mud = INTERMUD_D->GetMudName(mud);
if(!mud){
player->eventPrint("That liveupgrade server appears unavailable.");
return 1;
}
}
if(file == "all"){
string tmp = replace_string(upgrades_txt+"/upgrades.txt","/","0^0");
if(player && this_player() && player != this_player()){
this_player()->eventPrint("This command is currently locked and in use by "+capitalize(player->GetKeyName())+".");
return 1;
}
else if(this_player()) player = this_player();
else player = this_object();
if(WGET_D->GetUpgrading()){
player->eventPrint("An upgrade in already occurring. Please wait for it to complete.");
return 1;
}
if(!file_exists(upgrades_txt+"/list.txt")){
player->eventPrint("Downloading updates table. Please wait...");
rename("/secure/upgrades/files","/secure/upgrades/bak/"+time());
mkdir("/secure/upgrades/files");
if(oob){
OOB_D->GetFile(mud,upgrades_txt+"/upgrades.txt");
}
else {
WGET_D->GetFile(WEB_SOURCE_IP, upgrade_prefix+"/upgrades.txt",WEB_SOURCE_NAME,
"/secure/upgrades/txt/list.txt",WEB_SOURCE_PORT);
}
call_out( (: cmd :), 5, orig_str);
return 1;
}
tmpnames = explode(read_file(upgrades_txt+"/list.txt"),"\n");
foreach(string element in tmpnames){
if(!oob) allnames += ({ "/code/upgrades/"+mudlib_version()+element });
else allnames += ({ element });
}
if(oob){
OOB_D->eventMajorUpgrade(mud, allnames);
}
else {
WGET_D->eventMajorUpgrade(WEB_SOURCE_IP, allnames,WEB_SOURCE_NAME);
}
rm(upgrades_txt+"/list.txt");
player->eventPrint("Full upgrade begun.");
player->eventPrint("Please wait until you receive a completion message, "+
"then issue the command: liveupgrade apply\n\n");
player->eventPrint("%^FLASH%^RED%^WARNING! %^BLACK%^WARNING! %^YELLOW%^WARNING! %^RESET%^WARNING!");
player->eventPrint("You must *always* do a full backup before applying the liveupgrade. "+
"If the liveupgrade screwed up, and you get garbage files because of connection "+
"problems, it may be necessary for you to restore from backup to be able to "+
"start the mud again. You've been warned.");
return 1;
}
if(oob){
OOB_D->GetFile(mud,file);
player->eventPrint("Requesting the file \""+file+"\" from "+INTERMUD_D->GetMudName(mud)+".");
}
else {
player->eventPrint("Requesting the file \""+file+"\" from "+WEB_SOURCE_IP);
WGET_D->GetFile(WEB_SOURCE_IP, upgrade_prefix+file);
}
return 1;
}
void eventReceiveReport(string str){
if(player) player->eventPrint(str);
}
string GetHelp() {
return ("Syntax: liveupgrade all\n"
" liveupgrade apply\n"
" liveupgrade cancel\n"
" liveupgrade revert\n"
" liveupgrade alpha\n"
//"To use oob updates (not recommended), use the -o flag. The default "
//"is currently an http connection to dead-souls.net, which is vastly "
//"faster and more secure than oob.\n"
"To upgrade all files to the next appropriate level for your lib version:\n"
"liveupgrade all\n"
"Wait until you receive the completion message before finalizing the upgrade. "
"You can finalize the upgrade by typing:\n"
"liveupgrade apply\n"
"This will delete your old copies of files and copy the newly downloaded "
"ones in their place.\n"
"NEVER EVER do a liveupgrade without a full backup first.\n"
"To cancel the liveupgrade process:\n"
"liveupgrade cancel\n"
"To restore your mud to the condition it was in prior to the last liveupgrade.\n"
"liveupgrade revert\n"
"To enable liveupgrading between alpha and stable versions:\n"
"liveupgrade alpha\n\n"
"Web proxies are *NOT* supported. OOB is no longer supported.\n");
}
