mirror of
https://github.com/irungentoo/toxcore.git
synced 2024-03-22 13:30:51 +08:00
445 lines
13 KiB
C
445 lines
13 KiB
C
/* nTox.c
|
|
*
|
|
* Textual frontend for Tox.
|
|
*
|
|
* Copyright (C) 2013 Tox project All Rights Reserved.
|
|
*
|
|
* This file is part of Tox.
|
|
*
|
|
* Tox is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Tox is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Tox. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
#include "nTox.h"
|
|
#include "misc_tools.h"
|
|
|
|
#include <stdio.h>
|
|
#include <time.h>
|
|
|
|
#ifdef WIN32
|
|
#define c_sleep(x) Sleep(1*x)
|
|
#else
|
|
#include <unistd.h>
|
|
#define c_sleep(x) usleep(1000*x)
|
|
#endif
|
|
|
|
char lines[HISTORY][STRING_LENGTH];
|
|
char line[STRING_LENGTH];
|
|
|
|
char *help = "[i] commands: /f ID (to add friend), /m friendnumber message (to send message), /s status (to change status)\n"
|
|
"[i] /l list (list friends), /h for help, /i for info, /n nick (to change nickname), /q (to quit)";
|
|
int x, y;
|
|
|
|
|
|
uint8_t pending_requests[256][CLIENT_ID_SIZE];
|
|
uint8_t num_requests = 0;
|
|
|
|
void new_lines(char *line)
|
|
{
|
|
int i;
|
|
for (i = HISTORY-1; i > 0; i--)
|
|
strcpy(lines[i], lines[i-1]);
|
|
|
|
strcpy(lines[0], line);
|
|
do_refresh();
|
|
}
|
|
|
|
|
|
void print_friendlist()
|
|
{
|
|
char name[MAX_NAME_LENGTH];
|
|
new_lines("[i] Friend List:");
|
|
uint32_t i;
|
|
for (i = 0; i <= num_requests; i++) {
|
|
printf ("num_resusts: %d\n", num_requests);
|
|
char fstring[128];
|
|
|
|
getname(i, (uint8_t*)name);
|
|
if (strlen(name) <= 0) {
|
|
sprintf(fstring, "[i] Friend: NULL\n\tid: %i", i);
|
|
} else {
|
|
sprintf(fstring, "[i] Friend: %s\n\tid: %i", (uint8_t*)name, i);
|
|
}
|
|
new_lines(fstring);
|
|
}
|
|
}
|
|
|
|
char *format_message(char *message, int friendnum)
|
|
{
|
|
char name[MAX_NAME_LENGTH];
|
|
if (friendnum != -1) {
|
|
getname(friendnum, (uint8_t*)name);
|
|
} else {
|
|
getself_name((uint8_t*)name);
|
|
}
|
|
char *msg = malloc(100+strlen(message)+strlen(name)+1);
|
|
|
|
time_t rawtime;
|
|
struct tm * timeinfo;
|
|
time ( &rawtime );
|
|
timeinfo = localtime ( &rawtime );
|
|
char* time = asctime(timeinfo);
|
|
size_t len = strlen(time);
|
|
time[len-1] = '\0';
|
|
sprintf(msg, "[%d] %s <%s> %s", friendnum, time, name, message); // timestamp
|
|
return msg;
|
|
}
|
|
|
|
void line_eval(char lines[HISTORY][STRING_LENGTH], char *line)
|
|
{
|
|
if (line[0] == '/') {
|
|
char inpt_command = line[1];
|
|
char prompt[STRING_LENGTH + 2] = "> ";
|
|
strcat(prompt, line);
|
|
new_lines(prompt);
|
|
if (inpt_command == 'f') { // add friend command: /f ID
|
|
int i;
|
|
char temp_id[128];
|
|
for (i = 0; i < 128; i++)
|
|
temp_id[i] = line[i+3];
|
|
int num = m_addfriend(hex_string_to_bin(temp_id), (uint8_t*)"Install Gentoo", sizeof("Install Gentoo"));
|
|
char numstring[100];
|
|
sprintf(numstring, "[i] added friend %d", num);
|
|
new_lines(numstring);
|
|
do_refresh();
|
|
}
|
|
else if (inpt_command == 'd') {
|
|
doMessenger();
|
|
}
|
|
else if (inpt_command == 'm') { //message command: /m friendnumber messsage
|
|
size_t len = strlen(line);
|
|
char numstring[len-3];
|
|
char message[len-3];
|
|
int i;
|
|
for (i = 0; i < len; i++) {
|
|
if (line[i+3] != ' ') {
|
|
numstring[i] = line[i+3];
|
|
} else {
|
|
int j;
|
|
for (j = (i+1); j < len; j++)
|
|
message[j-i-1] = line[j+3];
|
|
break;
|
|
}
|
|
}
|
|
int num = atoi(numstring);
|
|
if (m_sendmessage(num, (uint8_t*) message, sizeof(message)) != 1) {
|
|
new_lines("[i] could not send message");
|
|
} else {
|
|
new_lines(format_message(message, -1));
|
|
}
|
|
}
|
|
else if (inpt_command == 'n') {
|
|
uint8_t name[MAX_NAME_LENGTH];
|
|
int i = 0;
|
|
size_t len = strlen(line);
|
|
for (i = 3; i < len; i++) {
|
|
if (line[i] == 0 || line[i] == '\n') break;
|
|
name[i-3] = line[i];
|
|
}
|
|
name[i-3] = 0;
|
|
setname(name, i);
|
|
char numstring[100];
|
|
sprintf(numstring, "[i] changed nick to %s", (char*)name);
|
|
new_lines(numstring);
|
|
}
|
|
else if (inpt_command == 'l') {
|
|
print_friendlist();
|
|
}
|
|
else if (inpt_command == 's') {
|
|
uint8_t status[MAX_USERSTATUS_LENGTH];
|
|
int i = 0;
|
|
size_t len = strlen(line);
|
|
for (i = 3; i < len; i++) {
|
|
if (line[i] == 0 || line[i] == '\n') break;
|
|
status[i-3] = line[i];
|
|
}
|
|
status[i-3] = 0;
|
|
m_set_userstatus(status, strlen((char*)status));
|
|
char numstring[100];
|
|
sprintf(numstring, "[i] changed status to %s", (char*)status);
|
|
new_lines(numstring);
|
|
}
|
|
else if (inpt_command == 'a') {
|
|
uint8_t numf = atoi(line + 3);
|
|
char numchar[100];
|
|
int num = m_addfriend_norequest(pending_requests[numf]);
|
|
if (num != -1) {
|
|
sprintf(numchar, "[i] friend request %u accepted", numf);
|
|
new_lines(numchar);
|
|
sprintf(numchar, "[i] added friendnumber %d", num);
|
|
new_lines(numchar);
|
|
} else {
|
|
sprintf(numchar, "[i] failed to add friend");
|
|
new_lines(numchar);
|
|
}
|
|
do_refresh();
|
|
}
|
|
else if (inpt_command == 'h') { //help
|
|
new_lines("[i] commands: /f ID (to add friend), /m friendnumber message (to send message), /s status (to change status)");
|
|
new_lines("[i] /l list (list friends), /h for help, /i for info, /n nick (to change nickname), /q (to quit)");
|
|
}
|
|
else if (inpt_command == 'i') { //info
|
|
char idstring0[200];
|
|
char idstring1[PUB_KEY_BYTES][5];
|
|
char idstring2[PUB_KEY_BYTES][5];
|
|
int i;
|
|
for (i = 0; i < PUB_KEY_BYTES; i++)
|
|
{
|
|
if (self_public_key[i] < (PUB_KEY_BYTES/2))
|
|
strcpy(idstring1[i],"0");
|
|
else
|
|
strcpy(idstring1[i], "");
|
|
sprintf(idstring2[i], "%hhX", self_public_key[i]);
|
|
}
|
|
//
|
|
strcpy(idstring0,"[i] ID: ");
|
|
int j;
|
|
for (j = 0; j < PUB_KEY_BYTES; j++) {
|
|
strcat(idstring0,idstring1[j]);
|
|
strcat(idstring0,idstring2[j]);
|
|
}
|
|
new_lines(idstring0);
|
|
}
|
|
|
|
else if (inpt_command == 'q') { //exit
|
|
endwin();
|
|
exit(EXIT_SUCCESS);
|
|
} else {
|
|
new_lines("[i] invalid command");
|
|
}
|
|
} else {
|
|
new_lines("[i] invalid command");
|
|
//new_lines(line);
|
|
}
|
|
}
|
|
|
|
void wrap(char output[STRING_LENGTH], char input[STRING_LENGTH], int line_width)
|
|
{
|
|
strcpy(output,input);
|
|
size_t len = strlen(output);
|
|
int i = 0;
|
|
for (i = line_width; i < len; i = i + line_width) {
|
|
while (output[i] != ' ' && i != 0) {
|
|
i--;
|
|
}
|
|
if (i > 0) {
|
|
output[i] = '\n';
|
|
}
|
|
}
|
|
}
|
|
|
|
int count_lines(char *string)
|
|
{
|
|
size_t len = strlen(string);
|
|
int count = 1;
|
|
int i;
|
|
for (i = 0; i < len; i++) {
|
|
if (string[i] == '\n')
|
|
count++;
|
|
}
|
|
return count;
|
|
}
|
|
|
|
char *appender(char *str, const char c)
|
|
{
|
|
size_t len = strlen(str);
|
|
if (len < STRING_LENGTH) {
|
|
str[len+1] = str[len];
|
|
str[len] = c;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
void do_refresh()
|
|
{
|
|
int count=0;
|
|
char wrap_output[STRING_LENGTH];
|
|
int L;
|
|
int i;
|
|
for (i = 0; i < HISTORY; i++) {
|
|
wrap(wrap_output, lines[i], x);
|
|
L = count_lines(wrap_output);
|
|
count = count + L;
|
|
if (count < y) {
|
|
move(y-1-count, 0);
|
|
printw(wrap_output);
|
|
clrtoeol();
|
|
}
|
|
}
|
|
move(y-1, 0);
|
|
clrtoeol();
|
|
printw(">> ");
|
|
printw(line);
|
|
clrtoeol();
|
|
refresh();
|
|
}
|
|
|
|
void print_request(uint8_t *public_key, uint8_t *data, uint16_t length)
|
|
{
|
|
new_lines("[i] received friend request with message:");
|
|
new_lines((char *)data);
|
|
char numchar[100];
|
|
sprintf(numchar, "[i] accept request with /a %u", num_requests);
|
|
new_lines(numchar);
|
|
memcpy(pending_requests[num_requests], public_key, CLIENT_ID_SIZE);
|
|
++num_requests;
|
|
do_refresh();
|
|
}
|
|
|
|
void print_message(int friendnumber, uint8_t * string, uint16_t length)
|
|
{
|
|
char name[MAX_NAME_LENGTH];
|
|
getname(friendnumber, (uint8_t*)name);
|
|
char msg[100+length+strlen(name)+1];
|
|
time_t rawtime;
|
|
struct tm * timeinfo;
|
|
time ( &rawtime );
|
|
timeinfo = localtime ( &rawtime );
|
|
char* temp = asctime(timeinfo);
|
|
size_t len = strlen(temp);
|
|
temp[len-1] = '\0';
|
|
sprintf(msg, "[%d] %s <%s> %s", friendnumber, temp, name, string); // timestamp
|
|
new_lines(format_message((char*)string, friendnumber));
|
|
}
|
|
|
|
void print_nickchange(int friendnumber, uint8_t *string, uint16_t 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);
|
|
new_lines(msg);
|
|
}
|
|
|
|
void print_statuschange(int friendnumber, uint8_t *string, uint16_t 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);
|
|
new_lines(msg);
|
|
}
|
|
|
|
void load_key()
|
|
{
|
|
FILE *data_file = NULL;
|
|
data_file = fopen("data","r");
|
|
if (data_file) {
|
|
//load keys
|
|
fseek(data_file, 0, SEEK_END);
|
|
int size = ftell(data_file);
|
|
fseek(data_file, 0, SEEK_SET);
|
|
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);
|
|
}
|
|
Messenger_load(data, size);
|
|
} else {
|
|
//else save new keys
|
|
int size = Messenger_size();
|
|
uint8_t data[size];
|
|
Messenger_save(data);
|
|
data_file = fopen("data","w");
|
|
if (fwrite(data, sizeof(uint8_t), size, data_file) != size){
|
|
printf("[i] could not write data file\n[i] exiting\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
fclose(data_file);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
if (argc < 4) {
|
|
printf("[!] Usage: %s [IP] [port] [public_key] <nokey>\n", argv[0]);
|
|
exit(0);
|
|
}
|
|
int c;
|
|
int on = 0;
|
|
initMessenger();
|
|
//if keyfiles exist
|
|
if(argc > 4){
|
|
if(strncmp(argv[4], "nokey", 6) < 0){
|
|
//load_key();
|
|
}
|
|
} else {
|
|
load_key();
|
|
}
|
|
m_callback_friendrequest(print_request);
|
|
m_callback_friendmessage(print_message);
|
|
m_callback_namechange(print_nickchange);
|
|
m_callback_userstatus(print_statuschange);
|
|
char idstring0[200];
|
|
char idstring1[PUB_KEY_BYTES][5];
|
|
char idstring2[PUB_KEY_BYTES][5];
|
|
int i;
|
|
for(i = 0; i < PUB_KEY_BYTES; i++)
|
|
{
|
|
if (self_public_key[i] < (PUB_KEY_BYTES / 2))
|
|
strcpy(idstring1[i],"0");
|
|
else
|
|
strcpy(idstring1[i], "");
|
|
sprintf(idstring2[i], "%hhX",self_public_key[i]);
|
|
}
|
|
strcpy(idstring0,"[i] your ID: ");
|
|
int j;
|
|
for (j = 0; j < PUB_KEY_BYTES; j++) {
|
|
strcat(idstring0,idstring1[j]);
|
|
strcat(idstring0,idstring2[j]);
|
|
}
|
|
initscr();
|
|
noecho();
|
|
raw();
|
|
getmaxyx(stdscr, y, x);
|
|
new_lines(idstring0);
|
|
new_lines(help);
|
|
strcpy(line, "");
|
|
IP_Port bootstrap_ip_port;
|
|
bootstrap_ip_port.port = htons(atoi(argv[2]));
|
|
int resolved_address = resolve_addr(argv[1]);
|
|
if (resolved_address != -1)
|
|
bootstrap_ip_port.ip.i = resolved_address;
|
|
else
|
|
exit(1);
|
|
|
|
DHT_bootstrap(bootstrap_ip_port, hex_string_to_bin(argv[3]));
|
|
nodelay(stdscr, TRUE);
|
|
while(true) {
|
|
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;
|
|
}
|