#include<iostream>
#include<list>
#include<map>
#include<queue>
#include<cstdlib>
#include<thread>
#include<algorithm>
#include<mutex>
#include<netinet/in.h>
#include<sys/socket.h>
#include <sys/types.h>
#include <fstream>
#include<vector>
#include <unistd.h>
#include <errno.h>
#define MAX 256
using namespace std;
/*
Struktura opisująca wątek klienta
*/
struct client_threads{
int id_soc; // Numer gniazda
int id_user; // Numer użytkownika
struct sockaddr_in addrx; // Adres Gniazda
};
/*
Struktura opisująca wiadomosc przesyłaną miedzy klientem a serwerem.
Podczas przesyłu znakiem konca wiadomosci jest ^
*/
struct mess{
int op; // numer operacji (3 pierwsze / znaki 0-2)
int id; // miejsce na id uzytkownika (3 nastepne /znaki 3-5)
string text; // wiadomosc (od 6 do znaku konca "^")
};
mutex mtx; //Semafor dostępu do listy klientów
mutex mtx_write; //Semafor do operacji na plikach
vector<client_threads*> client_list; //Lista klientów w systemie
vector<int> users_list; // Lista istniejacych użytkowników naszego chatu
int generateID();
void loadUsers();
void service(int my_soc);
void disconnect (int id);
string checkifDigit(string check);
string getHistory(string id);
int find_client(int id_name);
int find_clients(int id_name);
string getFriends(string id);
bool checkUser(string user);
string getPassword(string id);
struct mess recive(int client);
void set_name(int soc, int name);
bool checkFriend(string id,string id_friend);
void createUser(string id, string password);
void addFriend(string id, string id_friend);
string getHistory(string id_from,string id_to);
bool checkifFriend(string user_id, string friend_id);
void SaveAndclearFile(string id_from,string id_to,string text);
void send_message(int fd,string op, string id, string text);
void makeHistory(string id_from,string id_to, string text);
int main(int argc, char** argv){
struct sockaddr_in server_addr;
socklen_t socket_len;
int server_socket;
loadUsers();
// Tworzenie gniazda
if((server_socket = socket(PF_INET, SOCK_STREAM, 0)) != -1)
cout << "Utworzona gniazdo serwera... " << endl ;
else
cout << "Błąd tworzenia gniazda serwera... " << endl ;
int value = 1;
// Czy port jest zajety
if(setsockopt(server_socket, SOL_SOCKET, SO_REUSEADDR, &value, sizeof(int) ) < 0)
cout << "Błąd setsockopt..." << endl;
server_addr.sin_family = PF_INET;
server_addr.sin_port = htons(9004);
server_addr.sin_addr.s_addr = INADDR_ANY;
// Wązanie gniazda
if(bind (server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr)) != -1)
cout << "Powiązano gniazdo serwera... " << endl ;
else
cout << "Błąd powiązania gniazda serwera... " << endl ;
// Gniazdo oczekuje nanpolaczenia
if(listen(server_socket,10) != -1)
cout << "Gniazdo serwera w trybie pasywnym... " << endl ;
else
cout << "Błąd przechodzenia w stan pasywny gniazda serwera... " << endl ;
while(true){
//Tworzenie struktury nowego klienta i akceptacja połączenie
client_threads* new_client = new client_threads;
socket_len = sizeof(new_client->addrx);
if ((new_client->id_soc = accept(server_socket, (struct sockaddr*)&new_client->addrx, &socket_len)) != -1)
cout << "Zaakceptowano nowe połączenie... " << endl ;
else
cout << "Błąd nowego połączenia... " << endl ;
//Dosanie nowego klienta do listy klientów w systemie
mtx.lock();
client_list.push_back(new_client);
mtx.unlock();
//Procedura obsługi klienta
thread client_thread(service,new_client->id_soc);
client_thread.detach();
}
return EXIT_SUCCESS;
}
/*
Funkcja obsuługująca klientów
*/
void service(int my_soc){
string send_from; //Login użytkownika obsugiwanego przez wątek (jeśli jest zalogowany)
struct mess recived_message;//Wiadomość otrzymana od klienta
while(true){
recived_message = recive(my_soc);//Odczyt wiadomosci
if(recived_message.op == -1){//Zakończenie Połączenia
cout << "Brak Połączenia z użytownikiem" << endl;
break;
}
switch(recived_message.op){//Obsługa poszczególnych operacji
case 0 : { //Log in - Logowanie do serwisu
send_from = to_string(recived_message.id);
if((find_clients(recived_message.id) == -1) & (recived_message.text.compare(getPassword(to_string(recived_message.id))) == 0)){
set_name(my_soc, recived_message.id);
send_message(my_soc,"000",to_string(recived_message.id),"ok");
cout << "Pomyslne logowanie uzytkownika: " << recived_message.id << endl;
}
else{
send_message(my_soc,"000",to_string(recived_message.id),"0");
cout << "Błąd logowania uzytkownika: " << recived_message.id << endl;
}
break;
}
case 1 : { //Friends_Send - Wysłanie znajomych użytkownika
send_message(my_soc,"001",send_from, getFriends(to_string(recived_message.id)));
break;
}
case 2 : { //History_Send - Wysłanie historii konwersacji
send_message(my_soc,"002",to_string(recived_message.id), getHistory(send_from,to_string(recived_message.id)));
break;
}
/*
Wiadmość wysyłana jeśli obaj użytkownicy mają siebie w liście znajomych.
Jeśli znajomy jest niedostępny wiadomośc jest dopisywana do jego historii
*/
case 3 : { //Send Message - Przesłanie wiadomości
int send_to;
if(checkifFriend(send_from,to_string(recived_message.id)))
if((send_to = find_client(recived_message.id)) != -1)
send_message(send_to,"004",send_from, recived_message.text);
else
makeHistory(send_from, to_string(recived_message.id), send_from + ": " + recived_message.text);
else
send_message(my_soc,"004",to_string(recived_message.id), "SERVER: NIE MAM CIE W ZNAJOMYCH");
;
break;
}
case 4 : { //Add New User - Tworzy nowego uzytkownika
int i;
if ((i = generateID()) != -1){
createUser(to_string(i), recived_message.text);
send_message(my_soc,"004",to_string(i), recived_message.text);
}
else
send_message(find_client(recived_message.id),"004","0-1", "Brak Miejsca");
break;
}
case 5 : { // History_Make - Po przy zamykaniu konwersacji klient wysła cała historie która jest zapisywana
SaveAndclearFile(send_from, to_string(recived_message.id),recived_message.text);
break;
}
case 6 : { // Friend - Dodaj znajomego do listy
if(checkUser(to_string(recived_message.id))){
addFriend(send_from, to_string(recived_message.id));
send_message(my_soc,"006",to_string(recived_message.id), "1");
}
else
send_message(my_soc,"006",to_string(recived_message.id), "0");
break;
}
default : { // Not defined operation
cout << "Nieznany kod operacji" << endl;
break;
}
}
}
disconnect (my_soc);
close(my_soc);
}
/*
Nadpisuje całą romowy użytkownika id_from z użytkownikiem id_to
*/
void SaveAndclearFile(string id_from,string id_to, string text){
ofstream currfile;
mtx_write.lock();
currfile.open(id_from + "/" + id_to +".txt", ios_base::trunc);
currfile << text;
currfile.close();
mtx_write.unlock();
}
/*
Sprawdza czy użytkownik friend_id ma w znajomych użytkownika user_id
*/
bool checkifFriend(string user_id, string friend_id){
string line;
mtx_write.lock();
ifstream currfile(friend_id + "/Friends.txt");
while(currfile >> line)
if (user_id.compare(line) == 0){
cout << "foud user " << line << endl;
mtx_write.unlock();
return true;
}
currfile.close();
mtx_write.unlock();
return false;
}
/*
Sprawdza czy użytkownik istnieje
*/
bool checkUser(string user){
string line;
ifstream currfile("Users.txt");
while(currfile >> line)
if (user.compare(line) == 0){
cout << "found user " << line << endl;
return true;
}
currfile.close();
return false;
}
/*
Dodaj użutkownika id_friend do znajomych
*/
void addFriend(string id, string id_friend){
ofstream currfile;
mtx_write.lock();
currfile.open(id + "/Friends.txt", ios_base::app);
currfile << id_friend + "\n";
cout << "Added " << id_friend << " to " << id << "/Friends.txt" << endl;
currfile.close();
mtx_write.unlock();
}
/*
Dopisz wiadomość do historii użytkownika niedostępnego
*/
void makeHistory(string id_from,string id_to, string text){
ofstream currfile;
mtx_write.lock();
currfile.open(id_to + "/" + id_from +".txt", ios_base::app);
currfile << text << "\n";
currfile.close();
mtx_write.unlock();
}
/*
Uzyskaj historię rozmowy użytkonika id_form z użytkownikiem id_to
*/
string getHistory(string id_from,string id_to){
string history(id_from + "/" + id_to + ".txt"), text, line;
mtx_write.lock();
ifstream currfile(history);
while(getline(currfile,line))
text.append(line +"\n");
currfile.close();
mtx_write.unlock();
if(!text.empty())
return text;
else
return "0";
}
/*
Uzyskaj listę znajomych
*/
string getFriends(string id){
string friends(id + "/Friends.txt"), text, line;
ifstream currfile(friends);
while(currfile >> line)
text.append(line +" ");
currfile.close();
return text;
}
/*
Uzyskaj hasło
*/
string getPassword(string id){
string line;
string password(id + "/Haslo.txt");
ifstream currfile(password);
getline (currfile,line);
currfile.close();
return line;
}
/*
Stwórz nowego użytkownika
*/
void createUser(string id, string password){
string dir("mkdir " + id);
ofstream currfile;
system(dir.c_str());
currfile.open("Users.txt", ios_base::app);
currfile << id + "\n";
currfile.close();
currfile.open(id + "/Haslo.txt");
currfile << password;
currfile.close();
currfile.open(id + "/Friends.txt");
currfile.close();
}
/*
Wczytaj użytkowników
*/
void loadUsers(){
string line;
ifstream currfile("Users.txt");
while(currfile >> line)
users_list.push_back(stoi(line));
currfile.close();
}
/*
Generuj pierwsze wolne id z przedziału 100-998
*/
int generateID(){
for (int i = 100; i < 999; ++i)
if(!(find(users_list.begin(), users_list.end(), i) != users_list.end())){
users_list.push_back(i);
return i;
}
return -1;
}
/*
Usuń klienta z listy podłączonych klientów
*/
void disconnect (int soc) {
mtx.lock();
for(size_t i=0; i<client_list.size(); ++i)
if (client_list[i]->id_soc == soc){
delete client_list[i];
client_list.erase(client_list.begin() + i);
mtx.unlock();
return;
}
mtx.unlock();
}
/*
Znajdź numer gniazda pod którym znajduje się użytkownik o danym id
*/
int find_client(int id_name){
for(size_t i=0; i<client_list.size(); ++i)
if (client_list[i]->id_user == id_name){
return client_list[i]->id_soc;
}
return -1;
}
/*
Czy dany użytkownik jest zalogowany
*/
int find_clients(int id_name){
for(size_t i=0; i<client_list.size(); ++i)
if (client_list[i]->id_user == id_name){
return 1;
}
return -1;
}
/*
Logowanie - przypisanie użytkownika do danego gniazda
*/
void set_name(int soc, int name){
mtx.lock();
for(size_t i=0; i<client_list.size(); ++i)
if (client_list[i]->id_soc == soc){
client_list[i]->id_user = name;
mtx.unlock();
break;
}
mtx.unlock();
}
/*
Wysłanie wiadomości w ustalonym formacie
*/
void send_message(int fd,string op, string id, string text){
string message = op + id + text + "^";
cout << "Send: " << message << endl;
write(fd, message.c_str(), message.size());
}
/*
Odbiór wiadomości i przetworzenie jej do danej struktury
*/
struct mess recive(int client){
char bufor[MAX];
string zdanie;
size_t fnd;
struct mess recived;
int size;
do{
size = read(client, &bufor, 256);
if(size == 0){// Jeśli 0 to klient się rozłączył
recived.op = -1;
return recived;
}
if(size == -1){// Jeśli 0 to klient się rozłączył
recived.op = -1;
return recived;
}
//Jeśli znaleziono znak końca wiadomości to break i koniec czytania wiadomości
string wczytane(bufor);
fnd = wczytane.find_first_of("^");//fnd reprezentuje pozycję znaku w stringu
if(fnd != string::npos){// Jeśli nie znajdzie znaku w stringu przyjmuje wartość nops czyli wiadmość się nie skończyła
if (fnd == 0)
break;
wczytane = wczytane.substr(0,fnd);
zdanie.append(wczytane);
break;
}
zdanie.append(wczytane);
}while(fnd == string::npos);// Czytaj dopóki nie wytąpi znak końca
//Otrzymana wiadmość
cout << "Recv: " << zdanie << endl;
//Sprawdzanie czy wiadomość jest zgodna z ustaloną formą i przekształcenie do struktury
if (zdanie.size() >= 6){
recived.op = stoi(checkifDigit(zdanie.substr(0,3)));
recived.id = stoi(checkifDigit(zdanie.substr(3,3)));
recived.text = zdanie.substr(6);
}
else{
cout << "Nieodpowiednia ramka" << endl;
recived.op = -2;
return recived;
}
return recived;
}
/*
Sprawdza czy string składa się wyłącznie z cyfr jeśli nie to ustal na -1 tzn. rozłącz
*/
string checkifDigit(string check){
for(size_t i=0; i<check.size(); ++i)
if (!isdigit(check[i])){
return "0-1";
cout << "Złe Dane " << endl;
}
return check;
}
