/*
This is a sample program that uses several Chirp API calls to list a file server,
write data to a file, and then checksum that file.  You can use this file as 
a starting point for writing more complex programs.
*/

#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>

#include "chirp_reli.h"

#include "debug.h"
#include "auth_all.h"
#include "md5.h"

static int timeout=60;

static void show_help( const char *cmd )
{
	printf("use: %s [options]\n",cmd);
	printf("where options are:\n");
	printf(" -a <flag>  Require this authentication mode.\n");
	printf(" -d <flag>  Enable debugging for this subsystem.\n");
	printf(" -t <time>  Timeout for failure. (default is %ds)\n",timeout);
	printf(" -h         This message.\n");
}

int main( int argc, char *argv[] )
{	
	const char *hostname = "ccl00.cse.nd.edu";
	const char *filename = "/datafile";
	const char *line = "Hello Chirp!\n";

	struct chirp_file *file;
	struct chirp_dir *dir;
	struct chirp_dirent *d;

	unsigned char digest[MD5_DIGEST_LENGTH];
	int did_explicit_auth = 0;
	time_t stoptime;
	INT64_T result;
	char c;

	debug_config(argv[0]);

	while((c=getopt(argc,argv,"a:d:t:h"))!=(char)-1) {
		switch(c) {
			case 'a':
				auth_register_byname(optarg);
				did_explicit_auth = 1;
				break;
			case 'd':
				debug_flags_set(optarg);
				break;
			case 't':
				timeout = atoi(optarg);
				break;
			case 'h':
				show_help(argv[0]);
				exit(0);
				break;
		}
	}
 
	if(!did_explicit_auth) auth_register_all();

	stoptime = time(0) + timeout;

	printf("reading directory of host %s:\n",hostname);

	dir = chirp_reli_opendir(hostname,"/",stoptime);
	if(!dir) {
		fprintf(stderr,"couldn't read directory: %s\n",strerror(errno));
		return 1;
	}

	while(1) {
		d = chirp_reli_readdir(dir);
		if(!d) break;
		printf("%8lld %s\n",d->info.cst_size,d->name);
	}

	chirp_reli_closedir(dir);

	file = chirp_reli_open(hostname,filename,O_WRONLY|O_TRUNC|O_CREAT,0700,stoptime);
	if(!file) {
		fprintf(stderr,"couldn't open %s for writing: %s\n",filename,strerror(errno));
		return 1;
	}
	
	result = chirp_reli_pwrite(file,line,strlen(line),0,stoptime);
	if(result<0) {
		fprintf(stderr,"couldn't write to %s: %s\n",filename,strerror(errno));
		return 1;
	}
	
	printf("wrote %lld bytes to %s\n",result,filename);

	chirp_reli_close(file,stoptime);

	result = chirp_reli_md5(hostname,filename,digest,stoptime);
	if(result<0) {
		fprintf(stderr,"couldn't checksum %s: %s\n",filename,strerror(errno));
		return 1;
	}
	
	printf("checksum of %s is %s\n",filename,md5_string(digest));

	return 0;
}