swift.cpp

/*
 *  swift.cpp
 *  swift the multiparty transport protocol
 *
 *  Created by Victor Grishchenko on 2/15/10.
 *  Copyright 2009-2012 TECHNISCHE UNIVERSITEIT DELFT. All rights reserved.
 *
 */
#include <stdio.h>
#include <stdlib.h>
#include "compat.h"
#include "swift.h"
#include <cfloat>
#include <sstream>

using namespace swift;


// Local constants
#define RESCAN_DIR_INTERVAL	30 // seconds
#define REPORT_INTERVAL		 4 // seconds

// Local prototypes
#define quit(...) {fprintf(stderr,__VA_ARGS__); exit(1); }
int HandleSwiftFile(std::string filename, Sha1Hash root_hash, std::string trackerargstr, bool printurl, std::string urlfilename, double *maxspeed);
int OpenSwiftFile(std::string filename, const Sha1Hash& hash, Address tracker, bool force_check_diskvshash, uint32_t chunk_size);
int OpenSwiftDirectory(std::string dirname, Address tracker, bool force_check_diskvshash, uint32_t chunk_size);

void ReportCallback(int fd, short event, void *arg);
void EndCallback(int fd, short event, void *arg);
void RescanDirCallback(int fd, short event, void *arg);
int CreateMultifileSpec(std::string specfilename, int argc, char *argv[], int argidx);
void TimerCallback(int fd, short event, void *arg);

// Gateway stuff
bool InstallHTTPGateway(struct event_base *evbase,Address addr,uint32_t chunk_size, double *maxspeed);
bool InstallStatsGateway(struct event_base *evbase,Address addr);
bool InstallCmdGateway (struct event_base *evbase,Address cmdaddr,Address httpaddr);
bool HTTPIsSending();
bool StatsQuit();
void CmdGwUpdateDLStatesCallback();


// Global variables
struct event evreport, evrescan, evend, evtimer;
int single_fd = -1;
bool file_enable_checkpoint = false;
bool file_checkpointed = false;
bool report_progress = false;
bool quiet=false;
bool exitoncomplete=false;
bool httpgw_enabled=false,cmdgw_enabled=false;
// Gertjan fix
bool do_nat_test = false;
bool generate_multifile=false;

std::string scan_dirname="";
uint32_t chunk_size = SWIFT_DEFAULT_CHUNK_SIZE;
Address tracker;

long long int cmdgw_report_counter=0;
long long int cmdgw_report_interval=1; // seconds

// UNICODE: TODO, convert to std::string carrying UTF-8 arguments. Problem is
// a string based getopt_long type parser.
int utf8main (int argc, char** argv)
{
    static struct option long_options[] =
    {
        {"hash",    required_argument, 0, 'h'},
        {"file",    required_argument, 0, 'f'},
        {"dir",     required_argument, 0, 'd'}, // SEEDDIR reuse
        {"listen",  required_argument, 0, 'l'},
        {"tracker", required_argument, 0, 't'},
        {"debug",   no_argument, 0, 'D'},
        {"progress",no_argument, 0, 'p'},
        {"httpgw",  required_argument, 0, 'g'},
        {"wait",    optional_argument, 0, 'w'},
        {"nat-test",no_argument, 0, 'N'},
        {"statsgw", required_argument, 0, 's'}, // SWIFTPROC
        {"cmdgw",   required_argument, 0, 'c'}, // SWIFTPROC
        {"destdir", required_argument, 0, 'o'}, // SWIFTPROC
        {"uprate",  required_argument, 0, 'u'}, // RATELIMIT
        {"downrate",required_argument, 0, 'y'}, // RATELIMIT
        {"checkpoint",no_argument, 0, 'H'},
        {"chunksize",required_argument, 0, 'z'}, // CHUNKSIZE
        {"printurl", no_argument, 0, 'm'},
        {"urlfile",  required_argument, 0, 'r'},  // should be optional arg to printurl, but win32 getopt don't grok
        {"multifile",required_argument, 0, 'M'}, // MULTIFILE
        {"zerosdir",required_argument, 0, 'e'},  // ZEROSTATE
        {"dummy",no_argument, 0, 'j'},  // WIN32
        {"cmdgwint",required_argument, 0, 'C'}, // SWIFTPROC
        {"filehex",    required_argument, 0, '1'},  // SWIFTPROCUNICODE
        {"urlfilehex",required_argument, 0, '2'},   // SWIFTPROCUNICODE
        {"zerosdirhex",required_argument, 0, '3'},  // SWIFTPROCUNICODE
        {"zerostimeout",required_argument, 0, 'T'},  // ZEROSTATE
        {0, 0, 0, 0}
    };

    Sha1Hash root_hash;
    std::string filename = "",destdir = "", trackerargstr= "", zerostatedir="", urlfilename="";
    bool printurl=false;
    Address bindaddr;
    Address httpaddr;
    Address statsaddr;
    Address cmdaddr;
    tint wait_time = 0;
    double maxspeed[2] = {DBL_MAX,DBL_MAX};
    tint zerostimeout = TINT_NEVER;

    LibraryInit();
    Channel::evbase = event_base_new();

    int c,n;
    while ( -1 != (c = getopt_long (argc, argv, ":h:f:d:l:t:D:pg:s:c:o:u:y:z:wBNHmM:e:r:jC:1:2:3:T:", long_options, 0)) ) {
        switch (c) {
            case 'h':
                if (strlen(optarg)!=40)
                    quit("SHA1 hash must be 40 hex symbols\n");
                root_hash = Sha1Hash(true,optarg); // FIXME ambiguity
                if (root_hash==Sha1Hash::ZERO)
                    quit("SHA1 hash must be 40 hex symbols\n");
                break;
            case 'f':
                filename = strdup(optarg);
                break;
            case 'd':
                scan_dirname = strdup(optarg);
                break;
            case 'l':
                bindaddr = Address(optarg);
                if (bindaddr==Address())
                    quit("address must be hostname:port, ip:port or just port\n");
                wait_time = TINT_NEVER;
                break;
            case 't':
                tracker = Address(optarg);
                trackerargstr = strdup(optarg);
                if (tracker==Address())
                    quit("address must be hostname:port, ip:port or just port\n");
                break;
            case 'D':
                Channel::debug_file = optarg ? fopen_utf8(optarg,"a") : stderr;
                break;
            // Arno hack: get opt diff Win32 doesn't allow -D without arg
            case 'B':
            	fprintf(stderr,"SETTING DEBUG TO STDOUT\n");
                Channel::debug_file = stderr;
                break;
            case 'p':
                report_progress = true;
                break;
            case 'g':
            	httpgw_enabled = true;
                httpaddr = Address(optarg);
				wait_time = TINT_NEVER; // seed
                break;
            case 'w':
                if (optarg) {
                    char unit = 'u';
                    if (sscanf(optarg,"%lli%c",&wait_time,&unit)!=2)
                        quit("time format: 1234[umsMHD], e.g. 1M = one minute\n");

                    switch (unit) {
                        case 'D': wait_time *= 24;
                        case 'H': wait_time *= 60;
                        case 'M': wait_time *= 60;
                        case 's': wait_time *= 1000;
                        case 'm': wait_time *= 1000;
                        case 'u': break;
                        default:  quit("time format: 1234[umsMHD], e.g. 1D = one day\n");
                    }
                } else
                    wait_time = TINT_NEVER;
                break;
            case 'N': // Gertjan fix
                do_nat_test = true;
                break;
            case 's': // SWIFTPROC
                statsaddr = Address(optarg);
                if (statsaddr==Address())
                    quit("address must be hostname:port, ip:port or just port\n");
                break;
            case 'c': // SWIFTPROC
            	cmdgw_enabled = true;
                cmdaddr = Address(optarg);
                if (cmdaddr==Address())
                    quit("address must be hostname:port, ip:port or just port\n");
                wait_time = TINT_NEVER; // seed
                break;
            case 'o': // SWIFTPROC
                destdir = strdup(optarg); // UNICODE
                break;
            case 'u': // RATELIMIT
            	n = sscanf(optarg,"%lf",&maxspeed[DDIR_UPLOAD]);
            	if (n != 1)
            		quit("uprate must be KiB/s as float\n");
            	maxspeed[DDIR_UPLOAD] *= 1024.0;
            	break;
            case 'y': // RATELIMIT
            	n = sscanf(optarg,"%lf",&maxspeed[DDIR_DOWNLOAD]);
            	if (n != 1)
            		quit("downrate must be KiB/s as float\n");
            	maxspeed[DDIR_DOWNLOAD] *= 1024.0;
            	break;
            case 'H': //CHECKPOINT
                file_enable_checkpoint = true;
                break;
            case 'z': // CHUNKSIZE
            	n = sscanf(optarg,"%i",&chunk_size);
            	if (n != 1)
            		quit("chunk size must be bytes as int\n");
            	break;
            case 'm': // printurl
            	printurl = true;
            	quiet = true;
            	wait_time = 0;
            	break;
            case 'r':
           		urlfilename = strdup(optarg);
           		break;
            case 'M': // MULTIFILE
            	filename = strdup(optarg);
            	generate_multifile = true;
            	break;
            case 'e': // ZEROSTATE
                zerostatedir = strdup(optarg); // UNICODE
                wait_time = TINT_NEVER; // seed
                break;
            case 'j': // WIN32
                break;
            case 'C': // SWIFTPROC
                if (sscanf(optarg,"%lli",&cmdgw_report_interval)!=1)
                	quit("report interval must be int\n");
                break;
            case '1': // SWIFTPROCUNICODE
				// Swift on Windows expects command line arguments as UTF-16.
				// When swift is run with Python's popen, however, popen
            	// doesn't allow us to pass params in UTF-16, hence workaround.
            	// Format = hex encoded UTF-8
                filename = hex2bin(strdup(optarg));
                break;
            case '2': // SWIFTPROCUNICODE
           		urlfilename = hex2bin(strdup(optarg));
           		break;
            case '3': // ZEROSTATE // SWIFTPROCUNICODE
                zerostatedir = hex2bin(strdup(optarg));
                break;
            case 'T': // ZEROSTATE
            	double t=0.0;
            	n = sscanf(optarg,"%lf",&t);
            	if (n != 1)
					quit("zerostimeout must be seconds as float\n");
            	zerostimeout = t * TINT_SEC;
                break;
        }

    }   // arguments parsed


	// Change dir to destdir, if set, or to tempdir if HTTPGW
	if (destdir == "") {
		if (httpgw_enabled) {
			std::string dd = gettmpdir_utf8();
			chdir_utf8(dd);
		}
	}
	else
		chdir_utf8(destdir);

	if (httpgw_enabled)
		fprintf(stderr,"CWD %s\n",getcwd_utf8().c_str() );

    if (bindaddr!=Address()) { // seeding
        if (Listen(bindaddr)<=0)
            quit("cant listen to %s\n",bindaddr.str())
    } else if (tracker!=Address() || httpgw_enabled || cmdgw_enabled) { // leeching
    	evutil_socket_t sock = INVALID_SOCKET;
        for (int i=0; i<=10; i++) {
            bindaddr = Address((uint32_t)INADDR_ANY,0);
            sock = Listen(bindaddr);
            if (sock>0)
                break;
            if (i==10)
                quit("cant listen on %s\n",bindaddr.str());
        }
        if (!quiet)
        	fprintf(stderr,"swift: My listen port is %d\n", BoundAddress(sock).port() );
    }

    if (tracker!=Address() && !printurl)
        SetTracker(tracker);

    if (httpgw_enabled)
        InstallHTTPGateway(Channel::evbase,httpaddr,chunk_size,maxspeed);
    if (cmdgw_enabled)
		InstallCmdGateway(Channel::evbase,cmdaddr,httpaddr);

    // TRIALM36: Allow browser to retrieve stats via AJAX and as HTML page
    if (statsaddr != Address())
    	InstallStatsGateway(Channel::evbase,statsaddr);

    // ZEROSTATE
    ZeroState *zs = ZeroState::GetInstance();
    zs->SetContentDir(zerostatedir);
    zs->SetConnectTimeout(zerostimeout);


    if (!cmdgw_enabled)
    {
		int ret = -1;
		if (!generate_multifile)
		{
			if (filename != "" || root_hash != Sha1Hash::ZERO) {

				// Single file
				ret = HandleSwiftFile(filename,root_hash,trackerargstr,printurl,urlfilename,maxspeed);
			}
			else if (scan_dirname != "")
				ret = OpenSwiftDirectory(scan_dirname,Address(),false,chunk_size);
			else
				ret = -1;
		}
		else
		{
			// MULTIFILE
			// Generate multi-file spec
			ret = CreateMultifileSpec(filename,argc,argv,optind); //optind is global var points to first non-opt cmd line argument
			if (ret < 0)
			    quit("Cannot generate multi-file spec")
			else
			    // Calc roothash
			    ret = HandleSwiftFile(filename,root_hash,trackerargstr,printurl,urlfilename,maxspeed);
		}

		// For testing
		if (httpgw_enabled || zerostatedir != "")
			ret = 0;

		// No file/dir nor HTTP gateway nor CMD gateway, will never know what to swarm
		if (ret == -1) {
			fprintf(stderr,"Usage:\n");
			fprintf(stderr,"  -h, --hash\troot Merkle hash for the transmission\n");
			fprintf(stderr,"  -f, --file\tname of file to use (root hash by default)\n");
			fprintf(stderr,"  -l, --listen\t[ip:|host:]port to listen to (default: random)\n");
			fprintf(stderr,"  -t, --tracker\t[ip:|host:]port of the tracker (default: none)\n");
			fprintf(stderr,"  -D, --debug\tfile name for debugging logs (default: stdout)\n");
			fprintf(stderr,"  -B\tdebugging logs to stdout (win32 hack)\n");
			fprintf(stderr,"  -p, --progress\treport transfer progress\n");
			fprintf(stderr,"  -g, --httpgw\t[ip:|host:]port to bind HTTP content gateway to (no default)\n");
			fprintf(stderr,"  -s, --statsgw\t[ip:|host:]port to bind HTTP stats listen socket to (no default)\n");
			fprintf(stderr,"  -c, --cmdgw\t[ip:|host:]port to bind CMD listen socket to (no default)\n");
			fprintf(stderr,"  -o, --destdir\tdirectory for saving data (default: none)\n");
			fprintf(stderr,"  -u, --uprate\tupload rate limit in KiB/s (default: unlimited)\n");
			fprintf(stderr,"  -y, --downrate\tdownload rate limit in KiB/s (default: unlimited)\n");
			fprintf(stderr,"  -w, --wait\tlimit running time, e.g. 1[DHMs] (default: infinite with -l, -g)\n");
			fprintf(stderr,"  -H, --checkpoint\tcreate checkpoint of file when complete for fast restart\n");
			fprintf(stderr,"  -z, --chunksize\tchunk size in bytes (default: %d)\n", SWIFT_DEFAULT_CHUNK_SIZE);
			fprintf(stderr,"  -m, --printurl\tcompose URL from tracker, file and chunksize\n");
			fprintf(stderr,"  -M, --multifile\tcreate multi-file spec with given files\n");
			return 1;
		}
    }

    // Arno, 2012-01-04: Allow download and quit mode
    if (single_fd != -1 && root_hash != Sha1Hash::ZERO && wait_time == 0) {
    	wait_time = TINT_NEVER;
    	exitoncomplete = true;
    }

    // End after wait_time
    if ((long)wait_time > 0) {
    	evtimer_assign(&evend, Channel::evbase, EndCallback, NULL);
    	evtimer_add(&evend, tint2tv(wait_time));
    }

	evtimer_assign(&evtimer, Channel::evbase, TimerCallback, NULL);
	evtimer_add(&evtimer, tint2tv(TIMER_USEC));

    // Enter mainloop, if daemonizing
    if (wait_time == TINT_NEVER || (long)wait_time > 0) {
		// Arno: always, for statsgw, rate control, etc.
		evtimer_assign(&evreport, Channel::evbase, ReportCallback, NULL);
		evtimer_add(&evreport, tint2tv(REPORT_INTERVAL*TINT_SEC));


		// Arno:
		if (scan_dirname != "") {
			evtimer_assign(&evrescan, Channel::evbase, RescanDirCallback, NULL);
			evtimer_add(&evrescan, tint2tv(RESCAN_DIR_INTERVAL*TINT_SEC));
		}


		fprintf(stderr,"swift: Mainloop\n");
		// Enter libevent mainloop
		event_base_dispatch(Channel::evbase);

		// event_base_loopexit() was called, shutting down
    }

    // Arno, 2012-01-03: Close all transfers
	for (int i=0; i<FileTransfer::files.size(); i++) {
		if (FileTransfer::files[i] != NULL)
            Close(FileTransfer::files[i]->fd());
    }

    if (Channel::debug_file)
        fclose(Channel::debug_file);

    swift::Shutdown();

    return 0;
}


int HandleSwiftFile(std::string filename, Sha1Hash root_hash, std::string trackerargstr, bool printurl, std::string urlfilename, double *maxspeed)
{
	if (root_hash!=Sha1Hash::ZERO && filename == "")
		filename = strdup(root_hash.hex().c_str());

	single_fd = OpenSwiftFile(filename,root_hash,Address(),false,chunk_size);
	if (single_fd < 0)
		quit("cannot open file %s",filename.c_str());
	if (printurl) {

		FILE *fp = stdout;
		if (urlfilename != "")
		{
			fp = fopen_utf8(urlfilename.c_str(),"wb");
		    if (!fp)
			{
				print_error("cannot open file to write tswift URL to");
				quit("cannot open URL file %s",urlfilename.c_str());
			}
		}

		if (swift::Complete(single_fd) == 0)
			quit("cannot open empty file %s",filename.c_str());
	  
                std::ostringstream oss;
	        oss << "tswift:";
		if (trackerargstr != "")
	    	    oss << "//" << trackerargstr;
	        oss << "/" << RootMerkleHash(single_fd).hex();
 		if (chunk_size != SWIFT_DEFAULT_CHUNK_SIZE)
	  	    oss << "$" << chunk_size;
	        oss << "\n";
       	        
	        std::stringbuf *pbuf=oss.rdbuf();
	        if (pbuf == NULL)
	             print_error("cannot create URL");
		int ret = 0;
		ret = fprintf(fp,"%s", pbuf->str().c_str());
		if (ret <0)
			print_error("cannot write URL");

		if (urlfilename != "")
			fclose(fp);
	}
	else
	{
		printf("Root hash: %s\n", RootMerkleHash(single_fd).hex().c_str());
		fflush(stdout); // For testing
	}

	if (printurl || file_enable_checkpoint)
	{
	    // Arno, 2012-01-04: LivingLab: Create checkpoint such that content
	    // can be copied to scanned dir and quickly loaded
	    swift::Checkpoint(single_fd);
	}

	// RATELIMIT
	FileTransfer *ft = FileTransfer::file(single_fd);
	ft->SetMaxSpeed(DDIR_DOWNLOAD,maxspeed[DDIR_DOWNLOAD]);
	ft->SetMaxSpeed(DDIR_UPLOAD,maxspeed[DDIR_UPLOAD]);

	return single_fd;
}


int OpenSwiftFile(std::string filename, const Sha1Hash& hash, Address tracker, bool force_check_diskvshash, uint32_t chunk_size)
{
	std::string binmap_filename = filename;
	binmap_filename.append(".mbinmap");

	// Arno, 2012-01-03: Hack to discover root hash of a file on disk, such that
	// we don't load it twice while rescanning a dir of content.
	MmapHashTree *ht = new MmapHashTree(true,binmap_filename);

	//	fprintf(stderr,"swift: parsedir: File %s may have hash %s\n", filename, ht->root_hash().hex().c_str() );

	int fd = swift::Find(ht->root_hash());
	delete ht;
	if (fd == -1) {
		if (!quiet)
			fprintf(stderr,"swift: parsedir: Opening %s\n", filename.c_str());

		fd = swift::Open(filename,hash,tracker,force_check_diskvshash,true,chunk_size);
	}
	else if (!quiet)
		fprintf(stderr,"swift: parsedir: Ignoring loaded %s\n", filename.c_str() );
	return fd;
}


int OpenSwiftDirectory(std::string dirname, Address tracker, bool force_check_diskvshash, uint32_t chunk_size)
{
	DirEntry *de = opendir_utf8(dirname);
	if (de == NULL)
		return -1;

	while(1)
	{
		if (!(de->isdir_ || de->filename_.rfind(".mhash") != std::string::npos || de->filename_.rfind(".mbinmap") != std::string::npos))
		{
			// Not dir, or metafile
			std::string path = dirname;
			path.append(FILE_SEP);
			path.append(de->filename_);
			int fd = OpenSwiftFile(path,Sha1Hash::ZERO,tracker,force_check_diskvshash,chunk_size);
			if (fd >= 0)
				Checkpoint(fd);
		}

		DirEntry *newde = readdir_utf8(de);
		delete de;
		de = newde;
		if (de == NULL)
			break;
	}
	return 1;
}



int CleanSwiftDirectory(std::string dirname)
{
	std::set<int>	delset;
	std::vector<FileTransfer*>::iterator iter;
	for (iter=FileTransfer::files.begin(); iter!=FileTransfer::files.end(); iter++)
	{
		FileTransfer *ft = *iter;
		if (ft != NULL) {
			std::string filename = ft->GetStorage()->GetOSPathName();
			fprintf(stderr,"swift: clean: Checking %s\n", filename.c_str() );
			int res = file_exists_utf8( filename );
			if (res == 0) {
				fprintf(stderr,"swift: clean: Missing %s\n", filename.c_str() );
				delset.insert(ft->fd());
			}
		}
	}

	std::set<int>::iterator	iiter;
	for (iiter=delset.begin(); iiter!=delset.end(); iiter++)
	{
		int fd = *iiter;
		fprintf(stderr,"swift: clean: Deleting transfer %d\n", fd );
		swift::Close(fd);
	}

	return 1;
}





void ReportCallback(int fd, short event, void *arg) {
	// Called every second to print/calc some stats
	// Arno, 2012-05-24: Why-oh-why, update NOW
	Channel::Time();

	if (single_fd  >= 0)
	{
		if (report_progress) {
			fprintf(stderr,
				"%s %lli of %lli (seq %lli) %lli dgram %lli bytes up, "	\
				"%lli dgram %lli bytes down mptp[send:%lli,%lli;recv:%lli,%lli]\n",
				IsComplete(single_fd ) ? "DONE" : "done",
				Complete(single_fd), Size(single_fd), SeqComplete(single_fd),
				Channel::global_dgrams_up, Channel::global_raw_bytes_up,
				Channel::global_dgrams_down, Channel::global_raw_bytes_down,
				Channel::global_buffers_up, Channel::global_syscalls_up,
				Channel::global_buffers_down, Channel::global_syscalls_down);
		}

        FileTransfer *ft = FileTransfer::file(single_fd);
        if (report_progress) { // TODO: move up
        	fprintf(stderr,"upload %lf\n",ft->GetCurrentSpeed(DDIR_UPLOAD));
        	fprintf(stderr,"dwload %lf\n",ft->GetCurrentSpeed(DDIR_DOWNLOAD) );
        	//fprintf(stderr,"npeers %d\n",ft->GetNumLeechers()+ft->GetNumSeeders() );
        }
        // Update speed measurements such that they decrease when DL/UL stops
        // Always
    	ft->OnRecvData(0);
    	ft->OnSendData(0);

    	// CHECKPOINT
    	if (file_enable_checkpoint && !file_checkpointed && IsComplete(single_fd))
    	{
    		std::string binmap_filename = ft->GetStorage()->GetOSPathName();
    		binmap_filename.append(".mbinmap");
    		fprintf(stderr,"swift: Complete, checkpointing %s\n", binmap_filename.c_str() );

    		if (swift::Checkpoint(single_fd) >= 0)
    			file_checkpointed = true;
    	}


    	if (exitoncomplete && IsComplete(single_fd))
    		// Download and stop mode
    	    event_base_loopexit(Channel::evbase, NULL);

	}
    if (httpgw_enabled)
    {
        //fprintf(stderr,".");

        // ARNOSMPTODO: Restore fail behaviour when used in SwarmPlayer 3000.
        if (!HTTPIsSending()) {
        	// TODO
        	//event_base_loopexit(Channel::evbase, NULL);
            return;
        }
    }
    if (StatsQuit())
    {
    	// SwarmPlayer 3000: User click "Quit" button in webUI.
    	struct timeval tv;
    	tv.tv_sec = 1;
    	int ret = event_base_loopexit(Channel::evbase,&tv);
    }
	// SWIFTPROC
    if (cmdgw_report_interval == 1 || ((cmdgw_report_counter % cmdgw_report_interval) == 0))
    	CmdGwUpdateDLStatesCallback();

	cmdgw_report_counter++;

	// Gertjan fix
	// Arno, 2011-10-04: Temp disable
    //if (do_nat_test)
    //     nat_test_update();

	evtimer_add(&evreport, tint2tv(REPORT_INTERVAL*TINT_SEC));
}

void TimerCallback(int fd, short event, void *arg) {
	Channel::messageQueue.Flush();
	evtimer_add(&evtimer, tint2tv(TIMER_USEC));
}

void EndCallback(int fd, short event, void *arg) {
	// Called when wait timer expires == fixed time daemon
    event_base_loopexit(Channel::evbase, NULL);
}


void RescanDirCallback(int fd, short event, void *arg) {

	// SEEDDIR
	// Rescan dir: CAREFUL: this is blocking, better prepare .m* files first
	// by running swift separately and then copy content + *.m* to scanned dir,
	// such that a fast restore from checkpoint is done.
	//
	OpenSwiftDirectory(scan_dirname,tracker,false,chunk_size);

	CleanSwiftDirectory(scan_dirname);

	evtimer_add(&evrescan, tint2tv(RESCAN_DIR_INTERVAL*TINT_SEC));
}


#include <iostream>

// MULTIFILE
typedef std::vector<std::pair<std::string,int64_t> >	filelist_t;
int CreateMultifileSpec(std::string specfilename, int argc, char *argv[], int argidx)
{
	fprintf(stderr,"CreateMultiFileSpec: %s nfiles %d\n", specfilename.c_str(), argc-argidx );

	filelist_t	filelist;


	// MULTIFILE TODO: if arg is a directory, include all files


	// 1. Make list of files
	for (int i=argidx; i<argc; i++)
	{
		std::string pathname = argv[i];
		int64_t fsize = file_size_by_path_utf8(pathname);
		if( fsize < 0)
		{
			fprintf(stderr,"cannot open file in multi-spec list: %s\n", pathname.c_str() );
			print_error("cannot open file in multi-spec list" );
			return fsize;
		}

		// TODO: strip off common path from source pathnames
		// TODO: convert path separator to standard
		std::string pathstr = pathname; // TODO: UTF8-encode
		filelist.push_back(std::make_pair(pathstr,fsize));
	}

	// 2. Files in multi-file spec must be sorted, such that creating a swarm
	// from the same set of files results in the same swarm.
	sort(filelist.begin(), filelist.end());


	// 3. Create spec body
	std::ostringstream specbody;

	filelist_t::iterator iter;
	for (iter = filelist.begin(); iter < filelist.end(); iter++)
	{
		specbody << Storage::os2specpn( (*iter).first );
		specbody << " ";
		specbody << (*iter).second << "\n";
	}

	// 4. Calc specsize
	int specsize = Storage::MULTIFILE_PATHNAME.size()+1+0+1+specbody.str().size();
	char numstr[100];
	sprintf(numstr,"%d",specsize);
	char numstr2[100];
	sprintf(numstr2,"%d",specsize+strlen(numstr));
	if (strlen(numstr) == strlen(numstr2))
		specsize += strlen(numstr);
	else
		specsize += strlen(numstr)+(strlen(numstr2)-strlen(numstr));

	// 5. Create spec as string
	std::ostringstream spec;
	spec << Storage::MULTIFILE_PATHNAME;
	spec << " ";
	spec << specsize;
	spec << "\n";
	spec << specbody.str();

	fprintf(stderr,"spec: <%s>\n", spec.str().c_str() );

	// 6. Write to specfile
	FILE *fp = fopen_utf8(specfilename.c_str(),"wb");
	int ret = fwrite(spec.str().c_str(),sizeof(char),spec.str().length(),fp);
	if (ret < 0)
		print_error("cannot write multi-file spec");
	fclose(fp);

	return ret;
}


#ifdef _WIN32

// UTF-16 version of app entry point for console Windows-apps
int wmain( int wargc, wchar_t *wargv[ ], wchar_t *envp[ ] )
{
	char **utf8args = (char **)malloc(wargc*sizeof(char *));
	for (int i=0; i<wargc; i++)
	{
		//std::wcerr << "wmain: orig " << wargv[i] << std::endl;
		std::string utf8c = utf16to8(wargv[i]);
		utf8args[i] = strdup(utf8c.c_str());
	}
	return utf8main(wargc,utf8args);
}

// UTF-16 version of app entry point for non-console Windows apps
int WINAPI wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PWSTR pCmdLine, int nCmdShow)
{
	int wargc=0;
	fprintf(stderr,"wWinMain: enter\n");
	// Arno, 2012-05-30: TODO: add dummy first arg, because getopt eats the first
	// the argument when it is a non-console app. Currently done with -j dummy arg.
	LPWSTR* wargv = CommandLineToArgvW(pCmdLine, &wargc );
    return wmain(wargc,wargv,NULL);
}

#else

// UNIX version of app entry point for console apps
int main(int argc, char *argv[])
{
	// TODO: Convert to UTF-8 if locale not UTF-8
	return utf8main(argc,argv);
}

#endif