/** print_and_clean was supposed to make the binary somewhat smaller but
I think I failed and have not bothered to investigate it further (like
checking if the function is inlined or if one can force it to not be
inline). As this stuff is pretty boring and not critical to performance I
would prefer that all the boring stuff between benchmarks would be as
small as possible.
*/
static void
print_and_clean(const char *str, unsigned int nrprocessed,
const clock_t *start, const clock_t *end) {
printf("%s\t%.1f\n", str,
nrprocessed/(((double)(*end-*start))/CLOCKS_PER_SEC));
cleanPDFCrack();
finished = false;
}
static void print_and_clean_wall_time(const char *str, unsigned int nrprocessed, struct timeval start,
struct timeval end) {
// pthread is NOT working with multiple threads
// use gettimeofday() from sys/time.h instead for wall time measurements
double diffSec = ((double) end.tv_sec - (double) start.tv_sec);
double diffUSec = ((double) end.tv_usec - (double) start.tv_usec);
diffSec += diffUSec / 1000000;
printf("%s\t%.1f\n", str, (double) nrprocessed / diffSec);
cleanPDFCrack();
finished = false;
}
int
main(int argc, char** argv) {
int ret = 0, minpw = 0, maxpw = 32;
struct sigaction act1, act2;
FILE *file = NULL, *wordlist = NULL;
bool recovery = false, quiet = false,
work_with_user = true, permutation = false;
uint8_t *userpassword = NULL;
char *charset = NULL, *inputfile = NULL, *wordlistfile = NULL;
EncData *e;
/** Parse arguments */
while(true) {
int c, option_index;
static struct option long_options[] = {
{"bench", no_argument , 0, 'b'},
{"charset", required_argument, 0, 'c'},
{"file", required_argument, 0, 'f'},
{"loadState",required_argument, 0, 'l'},
{"maxpw", required_argument, 0, 'm'},
{"minpw", required_argument, 0, 'n'},
{"owner", no_argument , 0, 'o'},
{"password", required_argument, 0, 'p'},
{"quiet", required_argument, 0, 'q'},
{"permutate",no_argument, 0, 's'},
{"user", no_argument , 0, 'u'},
{"wordlist", required_argument, 0, 'w'},
{"version", no_argument , 0, 'v'},
{0, 0, 0, 0}};
/* getopt_long stores the option index here. */
option_index = 0;
c = getopt_long(argc, argv, "bc:f:l:m:n:op:qsuw:v",
long_options, &option_index);
/* Detect the end of the options. */
if(c == -1)
break;
switch(c) {
case 'b':
runBenchmark();
return 0;
case 'c':
if(charset)
fprintf(stderr,"Charset already set\n");
else
charset = strdup(optarg);
break;
case 'f':
if(!recovery)
inputfile = strdup(optarg);
break;
case 'l':
if(inputfile) {
free(inputfile);
inputfile = NULL;
}
inputfile = strdup(optarg);
recovery = true;
break;
case 'm':
maxpw = atoi(optarg);
break;
case 'n':
minpw = atoi(optarg);
break;
case 'o':
work_with_user = false;
break;
case 'p':
userpassword = (uint8_t*)strdup(optarg);
work_with_user = false;
break;
case 'q':
quiet = true;
break;
case 'u':
if(!work_with_user) {
fprintf(stderr, "Cannot work with both user- and owner-password\n");
exit(1);
}
work_with_user = true;
break;
case 's':
permutation = true;
break;
case 'w':
if(wordlistfile)
fprintf(stderr, "Wordlist already set\n");
else
wordlistfile = strdup(optarg);
break;
case 'v':
printf("pdfcrack version %d.%d\n", VERSION_MAJOR, VERSION_MINOR);
return 0;
default:
printHelp(argv[0]);
ret = 1;
goto out3;
}
}
/** If there are any unhandled arguments and the filename and/or wordlist is
not set: try to match the first as the filename and the second as
wordlist.
*/
{
int i = optind;
if(i > 0) {
if(i < argc && !inputfile)
inputfile = strdup(argv[i++]);
if(i < argc && !wordlistfile)
wordlistfile = strdup(argv[i++]);
if(i < argc)
while(i<argc) {
fprintf(stderr,"Non-option argument %s\n", argv[i++]);
}
}
}
if(!inputfile || minpw < 0 || maxpw < 0) {
printHelp(argv[0]);
ret = 1;
goto out3;
}
if((file = fopen(inputfile, "r")) == 0) {
fprintf(stderr,"Error: file %s not found\n", inputfile);
ret = 2;
goto out3;
}
e = calloc(1,sizeof(EncData));
if(recovery) {
if(wordlistfile) {
free(wordlistfile);
wordlistfile = NULL;
}
if(!loadState(file, e, &wordlistfile, &work_with_user)) {
fprintf(stderr, "Error: Not a savefile or savefile is damaged\n");
ret = 3;
goto out1;
}
if(!quiet)
printf("Loaded state for %s\n", inputfile);
}
else { /** !recovery */
if(!openPDF(file,e)) {
fprintf(stderr, "Error: Not a valid PDF\n");
ret = 3;
goto out1;
}
ret = getEncryptedInfo(file, e);
if(ret) {
if(ret == EENCNF)
fprintf(stderr, "Error: Could not extract encryption information\n");
else if(ret == ETRANF || ret == ETRENF || ret == ETRINF)
fprintf(stderr, "Error: Encryption not detected (is the document password protected?)\n");
ret = 4;
goto out1;
}
else if(e->revision < 2 || (strcmp(e->s_handler,"Standard") != 0)) {
fprintf(stderr, "The specific version is not supported (%s - %d)\n", e->s_handler, e->revision);
ret = 5;
goto out1;
}
}
if(fclose(file)) {
fprintf(stderr, "Error: closing file %s\n", inputfile);
}
if(minpw > maxpw) {
fprintf(stderr, "Warning: minimum pw-length bigger than max\n");
}
if(wordlistfile) {
if((wordlist = fopen(wordlistfile, "r")) == 0) {
fprintf(stderr,"Error: file %s not found\n", wordlistfile);
ret = 6;
goto out2;
}
}
act2.sa_handler = autoSave;
sigfillset(&act2.sa_mask);
act2.sa_flags = 0;
sigaction(SIGINT, &act2, 0);
if(!quiet) {
printEncData(e);
act1.sa_handler = alarmInterrupt;
sigemptyset(&act1.sa_mask);
act1.sa_flags = 0;
sigaction(SIGALRM, &act1, 0);
alarm(PRINTERVAL);
}
/** Try to initialize the cracking-engine */
if(!initPDFCrack(e, userpassword, work_with_user, wordlistfile,
wordlistfile?Wordlist:Generative, wordlist, charset,
(unsigned int)minpw, (unsigned int)maxpw, permutation)) {
cleanPDFCrack();
fprintf(stderr, "Wrong userpassword, '%s'\n", userpassword);
ret = 7;
goto out2;
}
/** Do the actual crunching */
runCrack();
/** cleanup */
cleanPDFCrack();
freeEncData(e);
if(wordlistfile) {
fclose(wordlist);
free(wordlistfile);
}
if(inputfile)
free(inputfile);
if(charset)
free(charset);
if(userpassword)
free(userpassword);
return 0;
out1:
if(fclose(file))
fprintf(stderr, "Error: closing file %s\n", inputfile);
out2:
freeEncData(e);
out3:
if(inputfile)
free(inputfile);
if(charset)
free(charset);
if(userpassword)
free(userpassword);
exit(ret);
}
