Merge pull request #97 from nurupo/master

Some cleanup
This commit is contained in:
irungentoo 2013-07-23 10:04:05 -07:00
commit 2b8d86ebe2
2 changed files with 54 additions and 60 deletions

View File

@ -48,10 +48,9 @@ void manage_keys()
if (read_size != KEYS_SIZE) {
printf("Error while reading the key file\nExiting.\n");
exit(1);
} else {
printf("Keys loaded successfully\n");
}
load_keys(keys);
printf("Keys loaded successfully\n");
} else {
//otherwise save new keys
new_keys();
@ -60,9 +59,8 @@ void manage_keys()
if (fwrite(keys, sizeof(uint8_t), KEYS_SIZE, keys_file) != KEYS_SIZE) {
printf("Error while writing the key file.\nExiting.\n");
exit(1);
} else {
printf("Keys saved successfully\n");
}
printf("Keys saved successfully\n");
}
fclose(keys_file);

View File

@ -184,7 +184,7 @@ void print_request(uint8_t * public_key, uint8_t * data, uint16_t length)
}
void print_message(int friendnumber, uint8_t * string, uint16_t length)
{
char *name = malloc(MAX_NAME_LENGTH);
char name[MAX_NAME_LENGTH];
getname(friendnumber, (uint8_t*)name);
char msg[100+length+strlen(name)+1];
time_t rawtime;
@ -195,26 +195,23 @@ void print_message(int friendnumber, uint8_t * string, uint16_t length)
int len = strlen(temp);
temp[len-1]='\0';
sprintf(msg, "[%d] %s <%s> %s", friendnumber, temp, name, string); // someone please fix this
free(name);
new_lines(msg);
}
void print_nickchange(int friendnumber, uint8_t *string, uint16_t length) {
char *name = malloc(MAX_NAME_LENGTH);
char name[MAX_NAME_LENGTH];
getname(friendnumber, (uint8_t*)name);
char msg[100+length];
sprintf(msg, "[i] [%d] %s is now known as %s.", friendnumber, name, string);
free(name);
new_lines(msg);
}
void print_statuschange(int friendnumber, uint8_t *string, uint16_t length) {
char *name = malloc(MAX_NAME_LENGTH);
char name[MAX_NAME_LENGTH];
getname(friendnumber, (uint8_t*)name);
char msg[100+length+strlen(name)+1];
sprintf(msg, "[i] [%d] %s's status changed to %s.", friendnumber, name, string);
free(name);
new_lines(msg);
}
int load_key(){
void load_key(){
FILE *data_file = NULL;
if ((data_file = fopen("data","r"))) {
//load keys
@ -223,8 +220,8 @@ int load_key(){
fseek(data_file, 0, SEEK_SET);
uint8_t data[size];
if(fread(data, sizeof(uint8_t), size, data_file) != size){
printf("Error reading file\n");
exit(0);
printf("Error reading data file\nExiting.\n");
exit(1);
}
Messenger_load(data, size);
} else {
@ -234,12 +231,11 @@ int load_key(){
Messenger_save(data);
data_file = fopen("data","w");
if(fwrite(data, sizeof(uint8_t), size, data_file) != size){
printf("Error writing file\n");
exit(0);
printf("Error writing data file\nExiting.\n");
exit(1);
}
}
fclose(data_file);
return 0;
}
int main(int argc, char *argv[])
{
@ -298,31 +294,31 @@ int main(int argc, char *argv[])
DHT_bootstrap(bootstrap_ip_port, hex_string_to_bin(argv[3]));
nodelay(stdscr, TRUE);
while(true) {
c=getch();
if (c != ERR) {
if (c != 27) {
getmaxyx(stdscr,y,x);
if (c == '\n') {
line_eval(lines, line);
strcpy(line, "");
} else if (c == 127) {
line[strlen(line)-1] = '\0';
} else if (isalnum(c) || ispunct(c) || c == ' ') {
strcpy(line,appender(line, (char) c));
}
}
}
if(on == 0)
{
if(DHT_isconnected())
if (on == 0 && DHT_isconnected())
{
new_lines("[i] connected to DHT\n[i] define username with /n");
on = 1;
}
}
doMessenger();
c_sleep(1);
do_refresh();
c = getch();
if (c == ERR || c == 27)
continue;
getmaxyx(stdscr, y, x);
if (c == '\n') {
line_eval(lines, line);
strcpy(line, "");
} else if (c == 127) {
line[strlen(line) - 1] = '\0';
} else if (isalnum(c) || ispunct(c) || c == ' ') {
strcpy(line, appender(line, (char) c));
}
}
endwin();
return 0;