main.c

#include "common.h"
#include "ssh.h"
#include "fingerprint.h"
#include "find.h"
#include "files.h"
#include "memcached.h"
#include "check-hash.h"
#include "command-line-args.h"
#include "filesigning.h"
#include "cgi.h"


void HashFromListFile(HashratCtx *Ctx)
{
char *Tempstr=NULL, *HashStr=NULL;
STREAM *S;
struct stat Stat;

if (strcmp(Ctx->ListPath,"-")==0) S=STREAMFromFD(0);
else S=STREAMOpenFile(Ctx->ListPath,SF_RDONLY);
Tempstr=STREAMReadLine(Tempstr, S);
while (Tempstr)
{
	StripTrailingWhitespace(Tempstr);
	if (StatFile(Ctx,Tempstr,&Stat)==0) 
	{
		//HashItem returns '0' on success
		if (HashItem(Ctx, Ctx->HashType, Tempstr, &Stat, &HashStr) ==0)
		{
		HashratOutputInfo(Ctx, Ctx->Out, Tempstr, &Stat, HashStr);
		HashratStoreHash(Ctx, Tempstr, &Stat, HashStr);
		}
		else fprintf(stderr,"ERROR: Failed to open file %s\n",Tempstr);
	}
	else fprintf(stderr,"ERROR: Failed to open file %s\n",Tempstr);
	Tempstr=STREAMReadLine(Tempstr, S);
}

STREAMClose(S);

DestroyString(Tempstr);
DestroyString(HashStr);
}




void HashStdIn(HashratCtx *Ctx)
{
STREAM *In=NULL;
char *Tempstr=NULL, *Hash=NULL;

if (Flags & FLAG_LINEMODE) 
{
	In=STREAMFromFD(0);
	STREAMSetTimeout(In,0);

	if (Flags & FLAG_HIDE_INPUT) Tempstr=TTYReadSecret(Tempstr, In, 0);
	else if (Flags & FLAG_STAR_INPUT) Tempstr=TTYReadSecret(Tempstr, In, TEXT_STARS);
	else Tempstr=STREAMReadLine(Tempstr,In);
	while (Tempstr)
	{
		if (! (Flags & FLAG_RAW)) StripTrailingWhitespace(Tempstr);
		Hash=CopyStr(Hash,"");
		ProcessData(&Hash, Ctx, Tempstr, StrLen(Tempstr));
		STREAMWriteString(Hash,Ctx->Out); STREAMWriteString("\n",Ctx->Out);
		if (Flags & FLAG_HIDE_INPUT) Tempstr=TTYReadSecret(Tempstr, In, 0);
		else if (Flags & FLAG_STAR_INPUT) Tempstr=TTYReadSecret(Tempstr, In, TEXT_STARS);
		else Tempstr=STREAMReadLine(Tempstr,In);
	}
}
else
{
	HashratHashSingleFile(Ctx, Ctx->HashType, 0, "-", NULL, &Hash);
	STREAMWriteString(Hash,Ctx->Out); STREAMWriteString("\n",Ctx->Out);
}
STREAMDisassociateFromFD(In);

DestroyString(Tempstr);
DestroyString(Hash);
}



int ProcessCommandLine(HashratCtx *Ctx, int argc, char *argv[])
{
struct stat Stat;
int i, count=0;

	for (i=1; i < argc; i++)
	{
	if (StrLen(argv[i]))
	{
			if (StatFile(Ctx, argv[i],&Stat)==0)
			{
				ProcessItem(Ctx, argv[i], &Stat);
			}
			else fprintf(stderr,"ERROR: File '%s' not found\n",argv[i]);
			count++;
	}
	}

return(count);
}



main(int argc, char *argv[])
{
char *Tempstr=NULL, *ptr;
int i, result=FALSE, count=0;
struct stat Stat;
HashratCtx *Ctx;	

Tempstr=SetStrLen(Tempstr, HOST_NAME_MAX);
gethostname(Tempstr, HOST_NAME_MAX);
LocalHost=CopyStr(LocalHost, Tempstr);
Tempstr=SetStrLen(Tempstr, HOST_NAME_MAX);
getdomainname(Tempstr, HOST_NAME_MAX);
LocalHost=MCatStr(LocalHost, ".", Tempstr, NULL);

memset(&Stat,0,sizeof(struct stat)); //to keep valgrind happy

Ctx=CommandLineParseArgs(argc,argv);
if (Ctx)
{
MemcachedConnect(GetVar(Ctx->Vars, "Memcached:Server"));

switch (Ctx->Action)
{
	case ACT_HASH:
	count=ProcessCommandLine(Ctx, argc, argv);
	//if we didn't find anything on the command-line then read from stdin
	if (count==0) HashStdIn(Ctx);
	break;

	case ACT_CHECK:
		result=CheckHashesFromList(Ctx);
	break;

	case ACT_CHECK_XATTR:
		ProcessCommandLine(Ctx, argc, argv);
	break;

	case ACT_CHECK_MEMCACHED:
		ProcessCommandLine(Ctx, argc, argv);
	break;

	case ACT_CGI:
		CGIDisplayPage();
	break;

	case ACT_PRINTUSAGE:
		CommandLinePrintUsage();
	break;

	case ACT_HASH_LISTFILE:
		HashFromListFile(Ctx);
	break;

	case ACT_SIGN:
	Ctx->Encoding = ENCODE_BASE64;
	for (i=1; i < argc; i++)
	{
	if (StrLen(argv[i]))
	{
			if (StatFile(Ctx, argv[i],&Stat)==0)
			{
				if (S_ISLNK(Stat.st_mode)) fprintf(stderr,"WARN: Not following symbolic link %s\n",argv[i]);
				else HashratSignFile(argv[i],Ctx);
			}
			else fprintf(stderr,"ERROR: File '%s' not found\n",argv[i]);
			count++;
	}
	}
	break;

	case ACT_CHECKSIGN:
	Ctx->Encoding |= ENCODE_BASE64;
	for (i=1; i < argc; i++)
	{
	if (StrLen(argv[i]))
	{
			if (StatFile(Ctx, argv[i],&Stat)==0)
			{
				if (S_ISLNK(Stat.st_mode)) fprintf(stderr,"WARN: Not following symbolic link %s\n",argv[i]);
				else HashratCheckSignedFile(argv[i], Ctx);
			}
			else fprintf(stderr,"ERROR: File '%s' not found\n",argv[i]);
			count++;
	}
	}
	break;

	case ACT_LOADMATCHES:
	ptr=GetVar(Ctx->Vars, "Memcached:Server");
	if (StrLen(ptr) ==0) printf("ERROR: No memcached server specified. Use the -memcached <server> command-line argument to specify one\n");
	else MatchesLoad(FLAG_MEMCACHED);
	break;

	case ACT_FINDMATCHES:
	if (! MatchesLoad(0))
	{
		printf("No hashes supplied on stdin.\n");
		exit(1);
	}
	ProcessCommandLine(Ctx, argc, argv);
	break;

	case ACT_FINDMATCHES_MEMCACHED:
	case ACT_FINDDUPLICATES:
	ProcessCommandLine(Ctx, argc, argv);
	break;


}

fflush(NULL);
if (Ctx->Out) STREAMClose(Ctx->Out);
if (Ctx->Aux) STREAMClose(Ctx->Aux);
}

DestroyString(Tempstr);

if (result) exit(0);
exit(1);
}