int main(void)
{
int ostype = get_os();
printf("%s version %d.%d\n",
id_os_name[ostype],
id_os_ver[ostype].maj,
id_os_ver[ostype].min);
return(0);
}
static void display_version(void)
{
printf("QPhotoRec %s, Data Recovery Utility, %s\nChristophe GRENIER <*****@*****.**>\nhttp://www.cgsecurity.org\n",VERSION,TESTDISKDATE);
printf("\n");
printf("Version: %s\n", VERSION);
printf("Compiler: %s\n", get_compiler());
#ifdef RECORD_COMPILATION_DATE
printf("Compilation date: %s\n", get_compilation_date());
#endif
printf("ext2fs lib: %s, ntfs lib: %s, ewf lib: %s, libjpeg: %s\n",
td_ext2fs_version(), td_ntfs_version(), td_ewf_version(), td_jpeg_version());
printf("OS: %s\n" , get_os());
}
uclock_t usec_clock(Uclock_T *now)
{
static int init = 0;
Uclock_T buf, *ptr;
if (NULL == now)
ptr = &buf;
else ptr = now;
if (!init) /* Only reprogram CTC once! */
{
SetMode2_();
init = -1;
}
Uclock_(ptr);
/*
** If Win or OS/2, ticks may lag CTC rollover so wait for
** monotonically increasing readings.
*/
if (DOS != get_os())
{
static Uclock_T last;
while (1)
{
if ((last.ticks > ptr->ticks) ||
((last.ticks == ptr->ticks) &&
(last.count > ptr->count)))
{
++(ptr->ticks);
last.ticks = ptr->ticks;
last.count = ptr->count;
Uclock_(ptr);
}
else
{
last.ticks = ptr->ticks;
last.count = ptr->count;
break;
}
}
}
return ((NULL == now) ? uclock_cnvrt(ptr) : -1.0);
}
/* some constants have different values depending on OS */
static void init_constants(void){
enum os_e os = get_os();
switch(os){
case OS_LINUX:
MP_MAP_ANON = 0x20;
IO_GET = TCGETS;
IO_SET = TCSETS;
ECHO = LINUX_ECHO;
break;
case OS_FREEBSD:
MP_MAP_ANON = 0x1000;
IO_GET = TIOCGETA;
IO_SET = TIOCSETA;
ECHO = FREEBSD_ECHO;
break;
default:
break;
}
}
int main(int argc, char *argv[])
{
int log_errno=0;
time_t my_time;
FILE *log_handle;
#ifdef Q_WS_X11
if(getenv("DISPLAY")==NULL)
{
printf("DISPLAY variable not set. Switching to PhotoRec in text mode.\n");
if(execv("photorec", argv)<0)
{
printf("photorec failed: %s\n", strerror(errno));
}
}
#endif
QApplication a(argc, argv);
log_handle=log_open("qphotorec.log", TD_LOG_CREATE, &log_errno);
#ifdef HAVE_DUP2
if(log_handle)
{
dup2(fileno(log_handle),2);
}
#endif
my_time=time(NULL);
log_info("\n\n%s",ctime(&my_time));
log_info("PhotoRec %s, Data Recovery Utility, %s\nChristophe GRENIER <*****@*****.**>\nhttp://www.cgsecurity.org\n", VERSION, TESTDISKDATE);
log_info("OS: %s\n" , get_os());
log_info("Compiler: %s\n", get_compiler());
log_info("Compilation date: %s\n", get_compilation_date());
log_info("ext2fs lib: %s, ntfs lib: %s, ewf lib: %s, libjpeg: %s\n",
td_ext2fs_version(), td_ntfs_version(), td_ewf_version(), td_jpeg_version());
QPhotorec *p = new QPhotorec();
p->showMaximized();
p->show();
int ret=a.exec();
delete p;
log_close();
return ret;
}
int main(int argc, char **argv)
{
int ifd;
uint32_t checksum;
uint32_t addr;
uint32_t ep;
struct stat sbuf;
unsigned char *ptr;
char *name = "";
cmdname = *argv;
addr = ep = 0;
while (--argc > 0 && **++argv == '-')
{
while (*++*argv)
{
switch (**argv)
{
case 'l':
lflag = 1;
break;
case 'A':
if ((--argc <= 0) || (opt_arch = get_arch(*++argv)) < 0)
usage();
goto NXTARG;
case 'C':
if ((--argc <= 0) || (opt_comp = get_comp(*++argv)) < 0)
usage();
goto NXTARG;
case 'O':
if ((--argc <= 0) || (opt_os = get_os(*++argv)) < 0)
usage();
goto NXTARG;
case 'T':
if ((--argc <= 0) || (opt_type = get_type(*++argv)) < 0)
usage();
goto NXTARG;
case 'a':
if (--argc <= 0)
usage();
addr = strtoul(*++argv, (char **)&ptr, 16);
if (*ptr)
{
fprintf(stderr,
"%s: invalid load address %s\n",
cmdname, *argv);
exit(EXIT_FAILURE);
}
goto NXTARG;
case 'd':
if (--argc <= 0)
usage();
datafile = *++argv;
dflag = 1;
goto NXTARG;
case 'e':
if (--argc <= 0)
usage();
ep = strtoul(*++argv, (char **)&ptr, 16);
if (*ptr)
{
fprintf(stderr,
"%s: invalid entry point %s\n",
cmdname, *argv);
exit(EXIT_FAILURE);
}
eflag = 1;
goto NXTARG;
case 'n':
if (--argc <= 0)
usage();
name = *++argv;
goto NXTARG;
case 'v':
vflag++;
break;
case 'x':
xflag++;
break;
default:
usage();
}
}
NXTARG:
;
}
if ((argc != 1) || ((lflag ^ dflag) == 0))
usage();
if (!eflag)
{
ep = addr;
/* If XIP, entry point must be after the U-Boot header */
if (xflag)
ep += sizeof(image_header_t);
}
/*
* If XIP, ensure the entry point is equal to the load address plus
* the size of the U-Boot header.
*/
if (xflag)
{
if (ep != addr + sizeof(image_header_t))
{
fprintf(stderr,
"%s: For XIP, the entry point must be the load addr + %lu\n",
cmdname, (unsigned long)sizeof(image_header_t));
exit(EXIT_FAILURE);
}
}
imagefile = *argv;
if (lflag)
{
ifd = open(imagefile, O_RDONLY | O_BINARY);
}
else
{
ifd =
open(imagefile, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666);
}
if (ifd < 0)
{
fprintf(stderr, "%s: Can't open %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
if (lflag)
{
int len;
char *data;
/*
* list header information of existing image
*/
if (fstat(ifd, &sbuf) < 0)
{
fprintf(stderr, "%s: Can't stat %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
if ((unsigned)sbuf.st_size < sizeof(image_header_t))
{
fprintf(stderr,
"%s: Bad size: \"%s\" is no valid image\n",
cmdname, imagefile);
exit(EXIT_FAILURE);
}
ptr = (unsigned char *)mmap(0, sbuf.st_size,
PROT_READ, MAP_SHARED, ifd, 0);
if ((caddr_t) ptr == (caddr_t) - 1)
{
fprintf(stderr, "%s: Can't read %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
/*
* create copy of header so that we can blank out the
* checksum field for checking - this can't be done
* on the PROT_READ mapped data.
*/
memcpy(hdr, ptr, sizeof(image_header_t));
if (ntohl(hdr->ih_magic) != IH_MAGIC)
{
fprintf(stderr,
"%s: Bad Magic Number: \"%s\" is no valid image\n",
cmdname, imagefile);
exit(EXIT_FAILURE);
}
data = (char *)hdr;
len = sizeof(image_header_t);
checksum = ntohl(hdr->ih_hcrc);
hdr->ih_hcrc = htonl(0); /* clear for re-calculation */
if (crc32(0, data, len) != checksum)
{
fprintf(stderr,
"*** Warning: \"%s\" has bad header checksum!\n",
imagefile);
}
data = (char *)(ptr + sizeof(image_header_t));
len = sbuf.st_size - sizeof(image_header_t);
if (crc32(0, data, len) != ntohl(hdr->ih_dcrc))
{
fprintf(stderr,
"*** Warning: \"%s\" has corrupted data!\n",
imagefile);
}
/* for multi-file images we need the data part, too */
print_header((image_header_t *) ptr);
(void)munmap((void *)ptr, sbuf.st_size);
(void)close(ifd);
exit(EXIT_SUCCESS);
}
/*
* Must be -w then:
*
* write dummy header, to be fixed later
*/
memset(hdr, 0, sizeof(image_header_t));
if (write(ifd, hdr, sizeof(image_header_t)) != sizeof(image_header_t))
{
fprintf(stderr, "%s: Write error on %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
if (opt_type == IH_TYPE_MULTI || opt_type == IH_TYPE_SCRIPT)
{
char *file = datafile;
unsigned long size;
for (;;)
{
char *sep = NULL;
if (file)
{
if ((sep = strchr(file, ':')) != NULL)
{
*sep = '\0';
}
if (stat(file, &sbuf) < 0)
{
fprintf(stderr, "%s: Can't stat %s: %s\n",
cmdname, file, strerror(errno));
exit(EXIT_FAILURE);
}
size = htonl(sbuf.st_size);
}
else
{
size = 0;
}
if (write(ifd, (char *)&size, sizeof(size)) != sizeof(size))
{
fprintf(stderr, "%s: Write error on %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
if (!file)
{
break;
}
if (sep)
{
*sep = ':';
file = sep + 1;
}
else
{
file = NULL;
}
}
file = datafile;
for (;;)
{
char *sep = strchr(file, ':');
if (sep)
{
*sep = '\0';
copy_file(ifd, file, 1);
*sep++ = ':';
file = sep;
}
else
{
copy_file(ifd, file, 0);
break;
}
}
}
else
{
copy_file(ifd, datafile, 0);
}
/* We're a bit of paranoid */
#if defined(_POSIX_SYNCHRONIZED_IO) && !defined(__sun__) && !defined(__FreeBSD__)
(void)fdatasync(ifd);
#else
(void)fsync(ifd);
#endif
if (fstat(ifd, &sbuf) < 0)
{
fprintf(stderr, "%s: Can't stat %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
ptr = (unsigned char *)mmap(0, sbuf.st_size,
PROT_READ | PROT_WRITE, MAP_SHARED, ifd,
0);
if (ptr == (unsigned char *)MAP_FAILED)
{
fprintf(stderr, "%s: Can't map %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
hdr = (image_header_t *) ptr;
checksum = crc32(0,
(const char *)(ptr + sizeof(image_header_t)),
sbuf.st_size - sizeof(image_header_t));
/* Build new header */
hdr->ih_magic = htonl(IH_MAGIC);
hdr->ih_time = htonl(sbuf.st_mtime);
hdr->ih_size = htonl(sbuf.st_size - sizeof(image_header_t));
hdr->ih_load = htonl(addr);
hdr->ih_ep = htonl(ep);
hdr->ih_dcrc = htonl(checksum);
hdr->ih_os = opt_os;
hdr->ih_arch = opt_arch;
hdr->ih_type = opt_type;
hdr->ih_comp = opt_comp;
strncpy((char *)hdr->ih_name, name, IH_NMLEN);
checksum = crc32(0, (const char *)hdr, sizeof(image_header_t));
hdr->ih_hcrc = htonl(checksum);
print_header(hdr);
(void)munmap((void *)ptr, sbuf.st_size);
/* We're a bit of paranoid */
#if defined(_POSIX_SYNCHRONIZED_IO) && !defined(__sun__) && !defined(__FreeBSD__)
(void)fdatasync(ifd);
#else
(void)fsync(ifd);
#endif
if (close(ifd))
{
fprintf(stderr, "%s: Write error on %s: %s\n",
cmdname, imagefile, strerror(errno));
exit(EXIT_FAILURE);
}
exit(EXIT_SUCCESS);
}
int main( int argc, char **argv )
{
int i;
int use_sudo=0;
int create_log=TD_LOG_NONE;
int run_setlocale=1;
int testdisk_mode=TESTDISK_O_RDONLY|TESTDISK_O_READAHEAD_32K;
list_disk_t *list_disk=NULL;
list_disk_t *element_disk;
const char *logfile="photorec.log";
FILE *log_handle=NULL;
int log_errno=0;
struct ph_options options={
.paranoid=1,
.keep_corrupted_file=0,
.mode_ext2=0,
.expert=0,
.lowmem=0,
.verbose=0,
.list_file_format=list_file_enable
};
struct ph_param params;
params.recup_dir=NULL;
params.cmd_device=NULL;
params.cmd_run=NULL;
params.carve_free_space_only=0;
/* random (weak is ok) is need fot GPT */
srand(time(NULL));
#ifdef HAVE_SIGACTION
/* set up the signal handler for SIGINT & SIGHUP */
sigemptyset(&action.sa_mask);
sigaddset(&action.sa_mask, SIGINT);
sigaddset(&action.sa_mask, SIGHUP);
action.sa_handler = sighup_hdlr;
action.sa_flags = 0;
if(sigaction(SIGINT, &action, NULL)==-1)
{
printf("Error on SIGACTION call\n");
return -1;
}
if(sigaction(SIGHUP, &action, NULL)==-1)
{
printf("Error on SIGACTION call\n");
return -1;
}
#endif
printf("PhotoRec %s, Data Recovery Utility, %s\nChristophe GRENIER <*****@*****.**>\nhttp://www.cgsecurity.org\n",VERSION,TESTDISKDATE);
for(i=1;i<argc;i++)
{
if((strcmp(argv[i],"/logname")==0) ||(strcmp(argv[i],"-logname")==0))
{
if(i+2>=argc)
{
display_help();
free(params.recup_dir);
return 1;
}
logfile=argv[++i];
}
else if((strcmp(argv[i],"/log")==0) ||(strcmp(argv[i],"-log")==0))
{
if(create_log==TD_LOG_NONE)
create_log=TD_LOG_APPEND;
}
else if((strcmp(argv[i],"/debug")==0) || (strcmp(argv[i],"-debug")==0))
{
options.verbose++;
if(create_log==TD_LOG_NONE)
create_log=TD_LOG_APPEND;
}
else if(((strcmp(argv[i],"/d")==0)||(strcmp(argv[i],"-d")==0)) &&(i+1<argc))
{
int len=strlen(argv[i+1]);
if(argv[i+1][len-1]=='\\' || argv[i+1][len-1]=='/')
{
params.recup_dir=(char *)MALLOC(len + strlen(DEFAULT_RECUP_DIR) + 1);
strcpy(params.recup_dir,argv[i+1]);
strcat(params.recup_dir,DEFAULT_RECUP_DIR);
}
else
params.recup_dir=strdup(argv[i+1]);
i++;
}
else if((strcmp(argv[i],"/all")==0) || (strcmp(argv[i],"-all")==0))
testdisk_mode|=TESTDISK_O_ALL;
else if((strcmp(argv[i],"/direct")==0) || (strcmp(argv[i],"-direct")==0))
testdisk_mode|=TESTDISK_O_DIRECT;
else if((strcmp(argv[i],"/help")==0) || (strcmp(argv[i],"-help")==0) || (strcmp(argv[i],"--help")==0) ||
(strcmp(argv[i],"/h")==0) || (strcmp(argv[i],"-h")==0) ||
(strcmp(argv[i],"/?")==0) || (strcmp(argv[i],"-?")==0))
{
display_help();
free(params.recup_dir);
return 0;
}
else if((strcmp(argv[i],"/version")==0) || (strcmp(argv[i],"-version")==0) || (strcmp(argv[i],"--version")==0) ||
(strcmp(argv[i],"/v")==0) || (strcmp(argv[i],"-v")==0))
{
display_version();
free(params.recup_dir);
return 0;
}
else if((strcmp(argv[i],"/nosetlocale")==0) || (strcmp(argv[i],"-nosetlocale")==0))
run_setlocale=0;
else if(strcmp(argv[i],"/cmd")==0)
{
if(i+2>=argc)
{
display_help();
free(params.recup_dir);
return 1;
}
{
disk_t *disk_car;
params.cmd_device=argv[++i];
params.cmd_run=argv[++i];
/* There is no log currently */
disk_car=file_test_availability(params.cmd_device, options.verbose, testdisk_mode);
if(disk_car==NULL)
{
printf("\nUnable to open file or device %s: %s\n", params.cmd_device, strerror(errno));
free(params.recup_dir);
return 1;
}
list_disk=insert_new_disk(list_disk,disk_car);
}
}
else
{
disk_t *disk_car=file_test_availability(argv[i], options.verbose, testdisk_mode);
if(disk_car==NULL)
{
printf("\nUnable to open file or device %s: %s\n", argv[i], strerror(errno));
free(params.recup_dir);
return 1;
}
list_disk=insert_new_disk(list_disk,disk_car);
}
}
#ifdef ENABLE_DFXML
xml_set_command_line(argc, argv);
#endif
if(create_log!=TD_LOG_NONE)
log_handle=log_open(logfile, create_log, &log_errno);
#ifdef HAVE_SETLOCALE
if(run_setlocale>0)
{
const char *locale;
locale = setlocale (LC_ALL, "");
if (locale==NULL) {
locale = setlocale (LC_ALL, NULL);
log_error("Failed to set locale, using default '%s'.\n", locale);
} else {
log_info("Using locale '%s'.\n", locale);
}
}
#endif
if(create_log!=TD_LOG_NONE && log_handle==NULL)
log_handle=log_open_default(logfile, create_log, &log_errno);
#ifdef HAVE_NCURSES
/* ncurses need locale for correct unicode support */
if(start_ncurses("PhotoRec", argv[0]))
{
free(params.recup_dir);
return 1;
}
{
const char*filename=logfile;
while(create_log!=TD_LOG_NONE && log_handle==NULL)
{
filename=ask_log_location(filename, log_errno);
if(filename!=NULL)
log_handle=log_open(filename, create_log, &log_errno);
else
create_log=TD_LOG_NONE;
}
}
aff_copy(stdscr);
wmove(stdscr,5,0);
wprintw(stdscr, "Disk identification, please wait...\n");
wrefresh(stdscr);
#endif
if(log_handle!=NULL)
{
time_t my_time;
#ifdef HAVE_DUP2
dup2(fileno(log_handle),2);
#endif
my_time=time(NULL);
log_info("\n\n%s",ctime(&my_time));
log_info("Command line: PhotoRec");
for(i=1;i<argc;i++)
log_info(" %s", argv[i]);
log_info("\n\n");
}
log_info("PhotoRec %s, Data Recovery Utility, %s\nChristophe GRENIER <*****@*****.**>\nhttp://www.cgsecurity.org\n", VERSION, TESTDISKDATE);
log_info("OS: %s\n" , get_os());
log_info("Compiler: %s\n", get_compiler());
log_info("Compilation date: %s\n", get_compilation_date());
log_info("ext2fs lib: %s, ntfs lib: %s, ewf lib: %s, libjpeg: %s\n",
td_ext2fs_version(), td_ntfs_version(), td_ewf_version(), td_jpeg_version());
#if defined(__CYGWIN__) || defined(__MINGW32__) || defined(DJGPP)
#else
#ifdef HAVE_GETEUID
if(geteuid()!=0)
{
log_warning("User is not root!\n");
}
#endif
#endif
log_flush();
screen_buffer_reset();
/* Scan for available device only if no device or image has been supplied in parameter */
if(list_disk==NULL)
list_disk=hd_parse(list_disk, options.verbose, testdisk_mode);
hd_update_all_geometry(list_disk, options.verbose);
/* Activate the cache, even if photorec has its own */
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
{
element_disk->disk=new_diskcache(element_disk->disk, testdisk_mode);
}
log_disk_list(list_disk);
reset_list_file_enable(options.list_file_format);
file_options_load(options.list_file_format);
#ifdef SUDO_BIN
if(list_disk==NULL)
{
#if defined(__CYGWIN__) || defined(__MINGW32__) || defined(DJGPP)
#else
#ifdef HAVE_GETEUID
if(geteuid()!=0)
use_sudo=2;
#endif
#endif
}
#endif
if(use_sudo==0)
use_sudo=do_curses_photorec(¶ms, &options, list_disk);
#ifdef HAVE_NCURSES
end_ncurses();
#endif
log_info("PhotoRec exited normally.\n");
if(log_close()!=0)
{
printf("PhotoRec: Log file corrupted!\n");
}
else if(params.cmd_run!=NULL && params.cmd_run[0]!='\0')
{
printf("PhotoRec syntax error: %s\n", params.cmd_run);
}
#ifdef SUDO_BIN
if(use_sudo>0)
{
printf("\n");
if(use_sudo>1)
printf("No disk found.\n");
printf("PhotoRec will try to restart itself using the sudo command to get\n");
printf("root (superuser) privileges.\n");
printf("\n");
run_sudo(argc, argv);
}
#endif
delete_list_disk(list_disk);
free(params.recup_dir);
#ifdef ENABLE_DFXML
xml_clear_command_line();
#endif
return 0;
}
static Bool initCitadel(void)
{
if (!read_cfg_messages())
{
#ifdef WINCIT
char Buffer[128];
sprintf(Buffer, getmsg(188), getmsg(671));
MessageBox(NULL, Buffer, NULL, MB_ICONSTOP | MB_OK);
#else
printf(getmsg(188), getmsg(671));
#endif
return (FALSE);
}
checkfiles();
initExtDrivers();
get_os();
cfg.battr = 0xff;
setscreen();
logo(TRUE); // no go for debug version; td32 go crash crash
init_internal_sound();
// some mouse initialization technology!!!!
initMouseHandler();
hideCounter = 1;
if (time(NULL) < 700000000L)
{
#ifdef WINCIT
MessageBox(NULL, getcfgmsg(119), NULL, MB_ICONSTOP | MB_OK);
#else
doccr();
doccr();
cPrintf(getcfgmsg(119));
doccr();
#endif
dump_cfg_messages();
return (FALSE);
}
static char prompt[92];
static char citadel[92];
char *envprompt;
char *citprompt;
envprompt = getenv(getcfgmsg(120));
citprompt = getenv(getcfgmsg(121));
if (citprompt)
{
sprintf(prompt, getcfgmsg(122), citprompt);
}
else if (envprompt)
{
sprintf(prompt, getcfgmsg(123), envprompt);
}
else
{
strcpy(prompt, getcfgmsg(124));
}
putenv(prompt);
sprintf(citadel, getcfgmsg(125), programName, version);
putenv(citadel);
#ifndef WINCIT
OC.whichIO = CONSOLE;
OC.SetOutFlag(OUTOK);
OC.Echo = BOTH;
OC.setio();
#endif
VerifyHeap(1);
// If we aren't reconfiguring, load the tables...
if (!reconfig)
{
// Start by reading ETC.TAB
getcwd(etcpath, 64);
FILE *fd;
if ((fd = fopen(etcTab, FO_RB)) != NULL)
{
if (filelength(fileno(fd)) != (long) sizeof(config) ||
fread(&cfg, 1, sizeof(config), fd) != (long) sizeof(config))
{
memset(&cfg, 0, sizeof(cfg));
reconfig = TRUE;
}
fclose(fd);
unlink(etcTab);
// If ETC.TAB could be loaded, load the rest
if (!reconfig)
{
changedir(cfg.homepath);
allocateTables();
if (!LogTab.Load() || !MessageDat.LoadTable() || !RoomTab.Load())
{
reconfig = TRUE;
}
Cron.ReadTable(WC_TWpn);
}
}
else
{
if (!batchmode)
{
#ifdef WINCIT
MessageBox(NULL, "No ETC.TAB.", NULL, MB_ICONSTOP | MB_OK);
#else
doccr();
discardable *d;
if ((d = readData(6)) != NULL)
{
int i;
for (i = 0; ((char **) d->next->aux)[i][0] != '#'; i++)
{
cPrintf(pcts, ((char **) d->next->aux)[i]);
doccr();
}
doccr();
discardData(d);
}
else
{
cOutOfMemory(28);
}
DeinitializeTimer();
critical(FALSE);
#endif
exit(1);
}
reconfig = TRUE;
}
}
if (reconfig)
{
cfg.attr = 7;
#ifndef WINCIT
pause(200);
cls(SCROLL_SAVE);
cCPrintf(getcfgmsg(126));
doccr();
#endif
if (!configcit())
{
#ifdef WINCIT
MessageBox(NULL, getcfgmsg(127), NULL, MB_ICONSTOP | MB_OK);
#else
doccr();
doccr();
cPrintf(getcfgmsg(127));
doccr();
#endif
dump_cfg_messages();
return (FALSE);
}
#ifndef WINCIT
setdefaultTerm(TT_ANSI);
CurrentUser->SetWidth(80);
#endif
Cron.ReadCronCit(WC_TWpn);
}
else
{
#ifndef WINCIT
if (!CreateScrollBackBuffer())
{
cPrintf(getcfgmsg(60));
doccr();
}
#endif
if (readconfigcit) // forced to read in config.cit
{
if (!readconfig(NULL, 1))
{
#ifdef WINCIT
MessageBox(NULL, getcfgmsg(129), NULL, MB_ICONSTOP | MB_OK);
#else
doccr();
doccr();
cPrintf(getcfgmsg(129));
doccr();
#endif
dump_cfg_messages();
return (FALSE);
}
}
}
VerifyHeap(1);
makeBorders();
readBordersDat();
if (cmd_nobells)
{
cfg.noBells = 2;
}
if (cmd_nochat)
{
cfg.noChat = TRUE;
}
if (cmd_mdata != CERROR)
{
cfg.mdata = cmd_mdata;
}
if (*cfg.f6pass)
{
if (SameString(cfg.f6pass, getmsg(670)))
{
ConsoleLock.LockF6();
}
else
{
ConsoleLock.Lock();
}
}
#ifndef WINCIT
if (cfg.ovrEms)
{
if (_OvrInitEms(0, 0, 0))
{
cPrintf(getcfgmsg(130));
doccr();
pause(200);
}
}
if (cfg.ovrExt)
{
if (_OvrInitExt(0, 0))
{
cPrintf(getcfgmsg(131));
doccr();
pause(200);
}
}
CommPort->Init();
setscreen();
#endif
logo(TRUE); // no go for debug version; td32 go crash crash
#ifndef WINCIT
StatusLine.Update(WC_TWp);
#endif
if (cfg.msgpath[(strlen(cfg.msgpath) - 1)] == '\\')
{
cfg.msgpath[(strlen(cfg.msgpath) - 1)] = '\0';
}
// move to home path
changedir(cfg.homepath);
char FileName[128];
ReIndexFileInfo(); // keep fileinfo.dat nice and pretty
// open message file
if (!MessageDat.OpenMessageFile(cfg.msgpath))
{
illegal(getmsg(78), MessageDat.GetFilename());
}
// Then room file
sprintf(FileName, sbs, cfg.homepath, roomDat);
openFile(FileName, &RoomFile);
citOpen(cfg.trapfile, CO_A, &TrapFile);
initMenus();
dump_cfg_messages();
if(!read_tr_messages())
{
errorDisp(getmsg(172));
}
else
{
#ifdef WINCIT
trap(T_SYSOP, "", gettrmsg(37));
#else
trap(T_SYSOP, gettrmsg(37));
#endif
dump_tr_messages();
}
read_cfg_messages(); // uh-oh!
if (!GroupData.Load())
{
return (FALSE);
}
if (!HallData.Load())
{
return (FALSE);
}
getRoomPos();
if (cfg.accounting)
{
ReadGrpdataCit(WC_TWpn);
}
ReadExternalCit(WC_TWpn);
ReadProtocolCit(WC_TWpn);
ReadMdmresltCit(WC_TWpn);
ReadCommandsCit(WC_TWpn);
#ifndef WINCIT
ReadMCICit(FALSE);
CurrentRoom->Load(LOBBY);
thisRoom = LOBBY;
checkdir();
if (!slv_door)
{
CITWINDOW *w = CitWindowsMsg(NULL, getmsg(19));
Initport();
Initport();
if (w)
{
destroyCitWindow(w, FALSE);
}
}
else
{
CommPort->Enable();
}
OC.whichIO = MODEM;
OC.setio();
#endif
// record when we put system up
time(&uptimestamp);
#ifndef WINCIT
setdefaultconfig(FALSE);
Talley->Fill();
#endif
logo(FALSE);
VerifyHeap(1);
dump_cfg_messages();
compactMemory(1);
return (TRUE);
}
int main( int argc, char **argv )
{
int i;
int use_sudo=0;
int help=0, version=0, verbose=0, dump_ind=0;
int create_log=TD_LOG_NONE;
int do_list=0;
int unit=UNIT_DEFAULT;
int write_used;
int saveheader=0;
int create_backup=0;
int run_setlocale=1;
int done=0;
int safe=0;
int testdisk_mode=TESTDISK_O_RDWR|TESTDISK_O_READAHEAD_8K;
list_disk_t *list_disk=NULL;
list_disk_t *element_disk;
const char *cmd_device=NULL;
char *cmd_run=NULL;
const char *logfile="testdisk.log";
FILE *log_handle=NULL;
int log_errno=0;
/* srand needed for GPT creation (weak is ok) */
srand(time(NULL));
#ifdef HAVE_SIGACTION
/* set up the signal handler for SIGINT & SIGHUP */
sigemptyset(&action.sa_mask);
sigaddset(&action.sa_mask, SIGINT);
sigaddset(&action.sa_mask, SIGHUP);
action.sa_handler = sighup_hdlr;
action.sa_flags = 0;
if(sigaction(SIGINT, &action, NULL)==-1)
{
printf("Error on SIGACTION call\n");
return -1;
}
if(sigaction(SIGHUP, &action, NULL)==-1)
{
printf("Error on SIGACTION call\n");
return -1;
}
#endif
printf("TestDisk %s, Data Recovery Utility, %s\nChristophe GRENIER <*****@*****.**>\nhttp://www.cgsecurity.org\n",VERSION,TESTDISKDATE);
for(i=1;i<argc;i++)
{
if((strcmp(argv[i],"/dump")==0) || (strcmp(argv[i],"-dump")==0))
dump_ind=1;
else if((strcmp(argv[i],"/logname")==0) ||(strcmp(argv[i],"-logname")==0))
{
if(i+2>=argc)
help=1;
else
logfile=argv[++i];
}
else if((strcmp(argv[i],"/log")==0) ||(strcmp(argv[i],"-log")==0))
{
if(create_log==TD_LOG_NONE)
create_log=TD_LOG_APPEND;
if(log_handle==NULL)
log_handle=log_open(logfile, create_log, &log_errno);
}
else if((strcmp(argv[i],"/debug")==0) || (strcmp(argv[i],"-debug")==0))
{
verbose++;
if(create_log==TD_LOG_NONE)
create_log=TD_LOG_APPEND;
if(log_handle==NULL)
log_handle=log_open(logfile, create_log, &log_errno);
}
else if((strcmp(argv[i],"/all")==0) || (strcmp(argv[i],"-all")==0))
testdisk_mode|=TESTDISK_O_ALL;
else if((strcmp(argv[i],"/backup")==0) || (strcmp(argv[i],"-backup")==0))
create_backup=1;
else if((strcmp(argv[i],"/direct")==0) || (strcmp(argv[i],"-direct")==0))
testdisk_mode|=TESTDISK_O_DIRECT;
else if((strcmp(argv[i],"/help")==0) || (strcmp(argv[i],"-help")==0) || (strcmp(argv[i],"--help")==0) ||
(strcmp(argv[i],"/h")==0) || (strcmp(argv[i],"-h")==0) ||
(strcmp(argv[i],"/?")==0) || (strcmp(argv[i],"-?")==0))
help=1;
else if((strcmp(argv[i],"/version")==0) || (strcmp(argv[i],"-version")==0) || (strcmp(argv[i],"--version")==0) ||
(strcmp(argv[i],"/v")==0) || (strcmp(argv[i],"-v")==0))
version=1;
else if(strcmp(argv[i],"/list")==0 || strcmp(argv[i],"-list")==0 || strcmp(argv[i],"-l")==0)
{
do_list=1;
}
else if(strcmp(argv[i],"-lu")==0)
{
do_list=1;
unit=UNIT_SECTOR;
}
else if((strcmp(argv[i],"/nosetlocale")==0) || (strcmp(argv[i],"-nosetlocale")==0))
run_setlocale=0;
else if((strcmp(argv[i],"/safe")==0) || (strcmp(argv[i],"-safe")==0))
safe=1;
else if((strcmp(argv[i],"/saveheader")==0) || (strcmp(argv[i],"-saveheader")==0))
saveheader=1;
else if(strcmp(argv[i],"/cmd")==0)
{
if(i+2>=argc)
help=1;
else
{
disk_t *disk_car;
cmd_device=argv[++i];
cmd_run=argv[++i];
disk_car=file_test_availability(cmd_device, verbose, testdisk_mode);
if(disk_car==NULL)
{
printf("\nUnable to open file or device %s\n",cmd_device);
help=1;
}
else
list_disk=insert_new_disk(list_disk,disk_car);
}
}
else
{
disk_t *disk_car=file_test_availability(argv[i], verbose, testdisk_mode);
if(disk_car==NULL)
{
printf("\nUnable to open file or device %s\n",argv[i]);
help=1;
}
else
list_disk=insert_new_disk(list_disk,disk_car);
}
}
if(version!=0)
{
printf("\n");
printf("Version: %s\n", VERSION);
printf("Compiler: %s\n", get_compiler());
printf("Compilation date: %s\n", get_compilation_date());
printf("ext2fs lib: %s, ntfs lib: %s, reiserfs lib: %s, ewf lib: %s\n",
td_ext2fs_version(), td_ntfs_version(), td_reiserfs_version(), td_ewf_version());
printf("OS: %s\n" , get_os());
return 0;
}
if(help!=0)
{
printf("\n" \
"Usage: testdisk [/log] [/debug] [file.dd|file.e01|device]\n"\
" testdisk /list [/log] [file.dd|file.e01|device]\n" \
" testdisk /version\n" \
"\n" \
"/log : create a testdisk.log file\n" \
"/debug : add debug information\n" \
"/list : display current partitions\n" \
"\n" \
"TestDisk checks and recovers lost partitions\n" \
"It works with :\n" \
"- BeFS (BeOS) - BSD disklabel (Free/Open/Net BSD)\n" \
"- CramFS, Compressed File System - DOS/Windows FAT12, FAT16 and FAT32\n" \
"- XBox FATX - Windows exFAT\n" \
"- HFS, HFS+, Hierarchical File System - JFS, IBM's Journaled File System\n" \
"- Linux btrfs - Linux ext2, ext3 and ext4\n" \
"- Linux GFS2 - Linux LUKS\n" \
"- Linux Raid - Linux Swap\n" \
"- LVM, LVM2, Logical Volume Manager - Netware NSS\n" \
"- Windows NTFS - ReiserFS 3.5, 3.6 and 4\n" \
"- Sun Solaris i386 disklabel - UFS and UFS2 (Sun/BSD/...)\n" \
"- XFS, SGI's Journaled File System - Wii WBFS\n" \
"- Sun ZFS\n" \
"\n" \
"If you have problems with TestDisk or bug reports, please contact me.\n");
return 0;
}
screen_buffer_reset();
if(do_list!=0)
{
printf("Please wait...\n");
/* Scan for available device only if no device or image has been supplied in parameter */
if(list_disk==NULL)
list_disk=hd_parse(list_disk, verbose, testdisk_mode);
/* Activate the cache */
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
element_disk->disk=new_diskcache(element_disk->disk,testdisk_mode);
if(safe==0)
hd_update_all_geometry(list_disk, verbose);
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
{
disk_t *disk=element_disk->disk;
const int hpa_dco=is_hpa_or_dco(disk);
printf("%s\n", disk->description(disk));
printf("Sector size:%u\n", disk->sector_size);
if(disk->model!=NULL)
printf("Model: %s", disk->model);
if(disk->serial_no!=NULL)
printf(", S/N:%s", disk->serial_no);
if(disk->fw_rev!=NULL)
printf(", FW:%s", disk->fw_rev);
printf("\n");
if(hpa_dco!=0)
{
if(disk->sector_size!=0)
printf("size %llu sectors\n", (long long unsigned)(disk->disk_real_size/disk->sector_size));
if(disk->user_max!=0)
printf("user_max %llu sectors\n", (long long unsigned)disk->user_max);
if(disk->native_max!=0)
printf("native_max %llu sectors\n", (long long unsigned)(disk->native_max+1));
if(disk->dco!=0)
printf("dco %llu sectors\n", (long long unsigned)(disk->dco+1));
if(hpa_dco&1)
printf("Host Protected Area (HPA) present.\n");
if(hpa_dco&2)
printf("Device Configuration Overlay (DCO) present.\n");
}
printf("\n");
}
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
{
disk_t *disk=element_disk->disk;
autodetect_arch(disk, NULL);
if(unit==UNIT_DEFAULT)
autoset_unit(disk);
else
disk->unit=unit;
interface_list(disk, verbose, saveheader, create_backup);
printf("\n");
}
delete_list_disk(list_disk);
return 0;
}
#ifdef HAVE_SETLOCALE
if(run_setlocale>0)
{
const char *locale;
locale = setlocale (LC_ALL, "");
if (locale==NULL) {
locale = setlocale (LC_ALL, NULL);
log_error("Failed to set locale, using default '%s'.\n", locale);
} else {
log_info("Using locale '%s'.\n", locale);
}
}
#endif
if(create_log!=TD_LOG_NONE && log_handle==NULL)
log_handle=log_open_default(logfile, create_log, &log_errno);
#ifdef HAVE_NCURSES
/* ncurses need locale for correct unicode support */
if(start_ncurses("TestDisk",argv[0]))
return 1;
if(argc==1 && create_log==TD_LOG_NONE)
{
verbose=1;
create_log=ask_testdisk_log_creation();
if(create_log==TD_LOG_CREATE || create_log==TD_LOG_APPEND)
log_handle=log_open(logfile, create_log, &log_errno);
}
{
const char*filename=logfile;
while(create_log!=TD_LOG_NONE && log_handle==NULL)
{
filename=ask_log_location(filename, log_errno);
if(filename!=NULL)
log_handle=log_open(filename, create_log, &log_errno);
else
create_log=TD_LOG_NONE;
}
}
#endif
if(log_handle!=NULL)
{
time_t my_time;
#ifdef HAVE_DUP2
dup2(fileno(log_handle),2);
#endif
my_time=time(NULL);
log_info("\n\n%s",ctime(&my_time));
log_info("Command line: TestDisk");
for(i=1;i<argc;i++)
log_info(" %s", argv[i]);
log_info("\n\n");
log_flush();
}
log_info("TestDisk %s, Data Recovery Utility, %s\nChristophe GRENIER <*****@*****.**>\nhttp://www.cgsecurity.org\n", VERSION, TESTDISKDATE);
log_info("OS: %s\n" , get_os());
log_info("Compiler: %s\n", get_compiler());
log_info("Compilation date: %s\n", get_compilation_date());
log_info("ext2fs lib: %s, ntfs lib: %s, reiserfs lib: %s, ewf lib: %s\n",
td_ext2fs_version(), td_ntfs_version(), td_reiserfs_version(), td_ewf_version());
#if defined(__CYGWIN__) || defined(__MINGW32__) || defined(DJGPP)
#else
#ifdef HAVE_GETEUID
if(geteuid()!=0)
{
log_warning("User is not root!\n");
}
#endif
#endif
#ifdef HAVE_NCURSES
aff_copy(stdscr);
wmove(stdscr,5,0);
wprintw(stdscr, "Please wait...\n");
wrefresh(stdscr);
#endif
/* Scan for available device only if no device or image has been supplied in parameter */
if(list_disk==NULL)
list_disk=hd_parse(list_disk, verbose, testdisk_mode);
/* Activate the cache */
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
element_disk->disk=new_diskcache(element_disk->disk,testdisk_mode);
#ifdef HAVE_NCURSES
wmove(stdscr,6,0);
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
{
wprintw(stdscr,"%s\n",element_disk->disk->description(element_disk->disk));
}
wrefresh(stdscr);
#endif
if(safe==0)
hd_update_all_geometry(list_disk, verbose);
/* save disk parameters to rapport */
log_info("Hard disk list\n");
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
{
disk_t *disk=element_disk->disk;
log_info("%s, sector size=%u", disk->description(disk), disk->sector_size);
if(disk->model!=NULL)
log_info(" - %s", disk->model);
if(disk->serial_no!=NULL)
log_info(", S/N:%s", disk->serial_no);
if(disk->fw_rev!=NULL)
log_info(", FW:%s", disk->fw_rev);
log_info("\n");
}
log_info("\n");
#ifdef SUDO_BIN
if(list_disk==NULL)
{
#if defined(__CYGWIN__) || defined(__MINGW32__) || defined(DJGPP)
#else
#ifdef HAVE_GETEUID
if(geteuid()!=0)
use_sudo=2;
#endif
#endif
}
#endif
if(use_sudo==0)
use_sudo=do_curses_testdisk(verbose,dump_ind,list_disk,saveheader,cmd_device,&cmd_run);
#ifdef HAVE_NCURSES
end_ncurses();
#endif
log_info("\n");
while(done==0)
{
int command='Q';
if(cmd_run!=NULL)
{
while(cmd_run[0]==',')
(cmd_run)++;
if(strncmp(cmd_run,"list",4)==0)
{
(cmd_run)+=4;
command='L';
}
else if(cmd_run[0]!='\0')
{
log_critical("Syntax error in command line: %s\n",cmd_run);
printf("Syntax error in command line: %s\n",cmd_run);
}
}
switch(command)
{
case 'L':
for(element_disk=list_disk;element_disk!=NULL;element_disk=element_disk->next)
interface_list(element_disk->disk, verbose, saveheader, create_backup);
break;
case 'q':
case 'Q':
done = 1;
break;
}
}
cmd_device=NULL;
cmd_run=NULL;
write_used=delete_list_disk(list_disk);
log_info("TestDisk exited normally.\n");
if(log_close()!=0)
{
printf("TestDisk: Log file corrupted!\n");
}
if(write_used!=0)
{
printf("You have to reboot for the change to take effect.\n");
}
#ifdef SUDO_BIN
if(use_sudo>0)
{
printf("\n");
if(use_sudo>1)
printf("No disk found.\n");
printf("TestDisk will try to restart itself using the sudo command to get\n");
printf("root (superuser) privileges.\n");
printf("\n");
run_sudo(argc, argv);
}
#endif
return 0;
}
