StarMirror/toxcore
1
0
Fork 0
mirror of https://github.com/irungentoo/toxcore.git synced 2024-03-22 13:30:51 +08:00
Code Issues Projects Releases Wiki Activity

cleaned up load_key() and added more error checking

Browse Source
This commit is contained in:
charmlesscoin 2013-08-03 17:57:44 -04:00
parent 09a6d2d351
commit 43f6608530
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
testing/nTox.c
View File

@ -361,16 +361,18 @@ void print_statuschange(int friendnumber, uint8_t *string, uint16_t length)
void load_key(char *path)
{
FILE *data_file = fopen(path, "r");
int size = 0;
if (data_file) {
//load keys
fseek(data_file, 0, SEEK_END);
int size = ftell(data_file);
fseek(data_file, 0, SEEK_SET);
size = ftell(data_file);
rewind(data_file);
uint8_t data[size];
if (fread(data, sizeof(uint8_t), size, data_file) != size){
printf("[i] could not read data file\n[i] exiting\n");
exit(1);
fputs("[!] could not read data file! exiting...\n", stderr);
goto FILE_ERROR;
}
Messenger_load(data, size);
@ -387,11 +389,15 @@ void load_key(char *path)
}
if (fwrite(data, sizeof(uint8_t), size, data_file) != size){
puts("[i] could not write data file! exiting...");
exit(1);
fputs("[!] could not write data file! exiting...", stderr);
goto FILE_ERROR;
}
}
fclose(data_file);
FILE_ERROR:
if(fclose(data_file) < 0)
perror("[!] fclose failed");
exit(1);
}
void print_help(void)