BT tuan 32

chat-server.cpp

// tuan32b1_chatserver.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <stdio.h>
#include <winsock2.h>
#include<iostream>
#include<string>
#include<map>

using namespace std;

const char* ask = "Enter name ex: \"client_id: xxxxxxxx\":\n";
int num_clients = 0;

map<int, string > ids;//cặp ánh xạ một mã client -> id client
bool check[64];//check[i] = true nếu client[i] nhập đúng cú pháp
SOCKET clients[64];

char buf[256];
#pragma comment(lib, "ws2_32")

//Hàm kiểm tra client có nhập đúng cú pháp
string validate(string syntax) {
	if (!syntax.substr(0, 11).compare("client_id: ")) {
		return syntax.substr(11);
	}
	else return "";
}

DWORD WINAPI receive(LPVOID p) {
	SOCKET client = *(SOCKET*)p;
	int ret;

	while (1)
	{
		ret = recv(client, buf, sizeof(buf), 0);
		if (ret <= 0)
			break;
		buf[ret-1] = 0;// Bỏ kí tự xuống dòng và thêm kí tự ngắt xâu
		//cout << "received: " << buf << endl;
		if (!check[num_clients])//Kiểm tra nếu client chưa được nhập đúng vào server
		{
			if (validate(buf) == "")//Kiểm tra nếu nhập sai cú pháp
			{
				send(client, ask, strlen(ask), 0);//Hỏi lại client 
			}
			else
			{
				ids.insert(pair<int, string>((int)client, validate(buf)));
				check[num_clients] = true;
			}
		}
		else//Nếu client đã được đăng nhập đúng thì dữ tin nhắn được sau đó được gửi cho các client còn lại
		{
			for (int i = 0; i < num_clients; i++)
			{
				if (clients[i] != client)
				{
					string msg = ids.at((int)client) +":" + buf;
					send(clients[i], msg.c_str(), msg.size(), 0);//hàm gửi tin nhắn
				}
			}
		}
	}

	closesocket(client);
	return 0;
}

int main()
{
	string a = "client_id: xxxxxxxx";

	//cout<<validate(a);
	WSADATA wsa;
	WSAStartup(MAKEWORD(2, 2), &wsa);

	SOCKADDR_IN addr;
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	addr.sin_port = htons(9000);

	SOCKET listener = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	bind(listener, (SOCKADDR*)&addr, sizeof(addr));
	listen(listener, 5);

	// Chap nhan ket noi va truyen nhan du 
	while (1) {
		cout << "Waiting for new client..." << endl;
		SOCKET client = accept(listener, NULL, NULL);
		cout << "Client accepted: " << client << endl;

		clients[num_clients] = client;
		send(client, ask, strlen(ask), 0);//Hỏi tên client

		CreateThread(0, 0, receive, &client, 0, 0);
		num_clients++;
	}

}

telnet-server.cpp

// tuan32b2_telnetserver.cpp : This file contains the 'main' function. Program execution begins and ends there.
//
#include <stdio.h>
#include <winsock2.h>
#include <iostream>
#include<string>
#include<fstream>
#include<map>

CRITICAL_SECTION cs;

using namespace std;
const char* askName = "Enter username:\n";
const char* askPass = "Enter password:\n";
int num_clients = 0;
#pragma comment(lib, "ws2_32")

bool check[64]; // = true khi client duoc dang nhap dung user, pass
bool checkUser = true; // trạng thái đăng nhập, true là đang nhập User, ngược lại là nhập password
string user_tmp;//lưu trữ username tạm thời
SOCKET clients[64];
char buf[256];//dữ liệu nhận được

map<string, string> user_pass;

int read_file() {
	fstream f_user_pass;
	f_user_pass.open("c:\\temp\\userpass.txt");
	char user[100];
	char pass[100];
	if (!f_user_pass.fail())//nếu đọc lỗi thì dừng
	{
		while (!f_user_pass.eof())//cuối file
		{
			if (f_user_pass >> user && f_user_pass >> pass) {
				user_pass.insert(pair<string, string>(user, pass));
			}
			else return -1;
		}
	}
	f_user_pass.close();
}

DWORD WINAPI receive(LPVOID p) {
	SOCKET client = *(SOCKET*)p;
	int ret;

	EnterCriticalSection(&cs);

	while (1)
	{
		ret = recv(client, buf, sizeof(buf), 0);
		if (ret <= 0)
			break;
		buf[ret - 1] = 0; // Bỏ kí tự xuống dòng và thêm kí tự ngắt xâu
		//buf[ret] = 0;
		if (!check[num_clients])//nếu client chưa đc đăng nhập
		{
			if (user_pass.find(buf) != user_pass.end() && checkUser == true)//nhập username
			{
				send(client, askPass, strlen(askPass), 0);//hỏi client Password
				user_tmp = buf;
				checkUser = false;
			}
			else if (checkUser == false)//nhập password
			{
				if (!user_pass.at(user_tmp).compare(buf))//nếu mật khẩu tương ứng với user là đúng
				{
					check[num_clients] = true;
					checkUser = true;
				}
				else
				{
					send(client, "Wrong password!", 15, 0);
					send(client, askPass, strlen(askPass), 0);
				}
			}
			else {//chưa đúng user name
				send(client, "Wrong username!", 15, 0);
				send(client, askName, strlen(askName), 0);
			}
		}
		else//thực hiện yêu cầu của client đã đăng nhập
		{
			string buff(buf);
			buff = buff + " > c:\\temp\\out.txt";
			system(buff.c_str());
		}
	}
	LeaveCriticalSection(&cs);
	closesocket(client);
	return 0;
}



int main()
{
	read_file();

	WSADATA wsa;
	WSAStartup(MAKEWORD(2, 2), &wsa);

	SOCKADDR_IN addr;
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	addr.sin_port = htons(9000);

	SOCKET listener = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	bind(listener, (SOCKADDR*)&addr, sizeof(addr));
	listen(listener, 5);

	InitializeCriticalSection(&cs);

	// Chap nhan ket noi va truyen nhan du 
	while (1) {
		cout << "Waiting for new client..." << endl;
		SOCKET client = accept(listener, NULL, NULL);
		cout << "Client accepted: " << client << endl;

		clients[num_clients] = client;

		//Hỏi username của client
		send(client, askName, strlen(askName), 0);

		//Tạo luộng nhận dữ liệu
		CreateThread(0, 0, receive, &client, 0, 0);
		num_clients++;
	}

	DeleteCriticalSection(&cs);
}