main.cpp

#include <iostream>
#include <cstdio>
#include <windows.h>
#include <cstdlib>
#include <ShlObj.h>
#include <fstream>
#include <direct.h>
#include <sstream>
#include <streambuf>
#include <map>
#include <deque>
#include <vector>
#include <time.h>
#include <thread>
#include <conio.h>
#include <mutex>
#include <shellapi.h>
#include <tchar.h>
#include "USN.h"
#include "reg_match.h"
#include "file.h"
#include "screen_settings.h"
using namespace std;
#define ull unsigned long long
#define MAXN 10000000
namespace color
{
	int background = 0;
	int foreground_default = 0xF,
		foreground_chosen = 0x02,
		foreground_folder = 0xB,
		foreground_buffer = 0xB;
} // namespace color
int disk_num = 0, page = 1;
USN last_USN[255];						 //读取到的上一次运行时各个硬盘的最后一个USN编号
USN USN_ID[255];						 //每一个硬盘中的USN日志编号
bool flag_rescanned[30];				 //标记该磁盘是否被重新扫描过
vector<int> rm_pos[30];					 //被删除的文件在文件中的位置
vector<dat> data[30], tmp_dat, crt, rnm; //保存的文件所有信息，创建的文件信息和重命名的信息
vector<string> new_name;				 //保存一个被重命名文件的最终名字
vector<int> new_prnum;                   //保存被重命名后新的ParentReferenceNumber，煞笔windows的剪切和删除操作本质都是重命名
vector<File> result;
vector<string> egg;
bool removed[MAXN], exist[MAXN]; //标记是否被删除，是否已经存在
int renamed[MAXN];				 //被重命名的文件的USN编号->new_name中的编号
char con_buffer[10000];			 //控制台缓冲区
bool is_chinese[10000];
int bufferlen = -1, now_result = 0;
int rpp = 5;			//result per page
int sim_tot = 30;		//模糊搜索的总结果数
double threshold = 0.8; //模糊匹配阈值
project match;
struct Page
{
	int l, r;
	Page()
	{
		l = 0, r = 0;
	}
	Page(int l, int r) : l(l), r(r) {}
};
vector<Page> pa;

void mainwindow();
void save_settings();
void read_egg()
{
	srand(time(0));
	string tmp;
	ifstream fin("egg.db");
	while (getline(fin, tmp))
	{
		egg.emplace_back(tmp);
	}
	fin.close();
	return;
}
void show_egg()
{
	if (egg.size() == 0)
		return;
	if (rand() <= 5000)
	{
		SetTitle(egg[rand() % egg.size()].c_str());
	}
	return;
}
void exit()
{
	printf("Press any key to exit");
	getche();
	exit(0);
}
void admincheck() //检查是否以管理员权限运行
{
	bool flag = IsUserAnAdmin();
	if (!flag)
	{
		SetColor(0x04, color::background);
		printf("Fatal Error : Please run as Administrator.\n");
		SetColor(color::foreground_default, color::background);
		exit();
	}
}
bool check_local()
{
	//检查本地文件是否存在，此处为简便只检查setting和info
	fstream _file;
	_file.open("database\\settings.db", ios::in);
	if (!_file)
		return 1;
	_file.close();
	_file.open("database\\info.db", ios::in);
	if (!_file)
		return 1;
	_file.close();
	return 0;
}

int check_disk()
{ //检查本地磁盘数量
	//可能存在的潜在bug:ntfs格式磁盘不连续时会跳过末尾的磁盘
	int tot = 0;
	for (char dst[] = "C";; dst[0]++)
	{
		if (is_ntfs(dst))
			tot++;
		else
			break;
	}
	return tot;
}
USN write_MFT(const char *disk_path, const char *data_dst, vector<dat> &v)
{
	//输出指定磁盘的所有信息并返回该磁盘USN的ID
	USN now_USN;
	now_USN = GET_MFT(data_dst, disk_path, 0, v);
	return now_USN;
}
void read_info()
{
	ifstream fin("database\\info.db");
	fin >> disk_num;
	for (int i = 0; i < disk_num; i++)
		fin >> USN_ID[i];
	for (int i = 0; i < disk_num; i++)
		fin >> last_USN[i];
	fin.close();
}
bool check_info()
{
	//检查信息是否合理
	bool flag = 0;
	char disk_path[] = "C", data_dst[] = "database\\c.db";
	for (int i = 0; i < disk_num; i++, disk_path[0]++, data_dst[9]++)
	{
		USN now_USN = GET_MFT(data_dst, disk_path, 1, tmp_dat);
		if (now_USN != USN_ID[i])
		{
			//若现有的USN的ID与上一次运行时的ID不符，说明USN日志遭到了改动
			USN_ID[i] = now_USN;
			flag_rescanned[i] = 1;
			SetColor(0x04, color::background);
			printf("Fatal Error: Change in USN Journal of %s detected. Please wait the program to re-scan the disk. This may take several minutes.\n", disk_path);
			SetColor(color::foreground_default, color::background);
			printf("Rescanning %s: please wait.\n", disk_path);
			write_MFT(disk_path, data_dst, data[i]);
			register int *tmp;
			tmp = new int[MAXN];
			for (int j = 0; j < data[i].size(); j++)
				tmp[data[i][j].rnum] = j;
			for (int j = 0; j < data[i].size(); j++)
				data[i][j].ppos = tmp[data[i][j].prnum];
			SetColor(color::foreground_chosen, color::background);
			printf("Successfully rescanned %s.\n", disk_path);
			SetColor(color::foreground_default, color::background);
		}
	}
}
bool update_info()
{
	ofstream fout("database\\info.db");
	fout << disk_num << endl;
	for (int i = 0; i < disk_num; i++)
		fout << USN_ID[i] << endl;
	char disk_path[] = "C";
	for (int i = 0; i < disk_num; i++)
		fout << last_USN[i] << endl;
	fout.close();
}
void first_run()
{
	mkdir("database"); //创建文件夹
	/****************Debug************/
	ofstream fout("database\\settings.db");
	fout.close();
	char disk_path[] = "C", data_dst[] = "database\\c.db";
	disk_num = check_disk();
	for (int i = 0; i < disk_num; i++, disk_path[0]++, data_dst[9]++)
	{
		flag_rescanned[i] = 1;
		USN_ID[i] = write_MFT(disk_path, data_dst, data[i]);
		last_USN[i] = GET_USN(disk_path, 0, 0, removed, tmp_dat, new_name, new_prnum, renamed);
		register int *tmp;
		tmp = new int[MAXN];
		for (int j = 0; j < data[i].size(); j++)
			tmp[data[i][j].rnum] = j;
		for (int j = 0; j < data[i].size(); j++)
			data[i][j].ppos = tmp[data[i][j].prnum];
		SetColor(color::foreground_chosen, color::background);
		printf("Successfully scanned %s:\n", disk_path);
		SetColor(color::foreground_default, color::background);
		delete[] tmp;
	}
}
void update_database()
{
	char disk_path[] = "C";
	char data_dst[] = "database\\c.db";
	for (int x = 0; x < disk_num; x++, disk_path[0]++, data_dst[9]++)
	{
		printf("Now updating disk %s, please wait.\n", disk_path);
		if (flag_rescanned[x])
			continue;
		crt.clear();
		last_USN[x] = GET_USN(disk_path, last_USN[x], 1, removed, crt, new_name, new_prnum, renamed);
		register dat buffer;
		ifstream fin(data_dst, std::ios::binary);
		while (1)
		{
			fin.read((char *)&buffer.rnum, sizeof(int));
			if (buffer.rnum == -1)
				break;
			fin.read((char *)&buffer.prnum, sizeof(int));
			fin.read((char *)&buffer.len, sizeof(int));
			buffer.filename = new char[buffer.len];
			fin.read(buffer.filename, sizeof(char) * buffer.len);
			//cout<<buffer.filename<<endl;
			if (!removed[buffer.rnum])
			{
				data[x].push_back(buffer);
				exist[buffer.rnum] = 1;
			}
		}
		fin.close();
		cout << "read database: " << (double)clock() / CLOCKS_PER_SEC << "s" << endl;
		for (int i = 0; i < crt.size(); i++)
		{
			if (!removed[crt[i].rnum] && !exist[crt[i].rnum])
			{
				data[x].push_back(crt[i]);
				exist[crt[i].rnum] = 1;
			}
		}
		register int *tmp;
		tmp = new int[MAXN];
		memset(tmp, 0, sizeof(tmp));
		cout << "read usn : " << (double)clock() / CLOCKS_PER_SEC << "s" << endl;
		ofstream fout(data_dst, std::ios::binary);
		for (register int i = 0; i < data[x].size(); i++)
		{
			if (renamed[data[x][i].rnum])
			{
				register int nlen = new_name[renamed[data[x][i].rnum] - 1].length();
				delete[] data[x][i].filename;
				data[x][i].filename = new char[nlen + 1];
				data[x][i].len = nlen + 1;
				strcpy(data[x][i].filename, new_name[renamed[data[x][i].rnum] - 1].c_str());
				*(data[x][i].filename + nlen) = '\0';
				data[x][i].prnum = new_prnum[renamed[data[x][i].rnum] - 1];
			}
			fout.write((char *)&data[x][i].rnum, sizeof(int));
			fout.write((char *)&data[x][i].prnum, sizeof(int));
			fout.write((char *)&data[x][i].len, sizeof(int));
			fout.write(data[x][i].filename, sizeof(char) * data[x][i].len);
			tmp[data[x][i].rnum] = i;
		}
		int end = -1;
		fout.write((char *)&end, sizeof(int));
		fout.close();
		cout << "write database : " << (double)clock() / CLOCKS_PER_SEC << "s" << endl;
		/***建立文件索引路径***/
		for (int i = 0; i < data[x].size(); i++)
		{
			data[x][i].ppos = tmp[data[x][i].prnum];
		}
		delete[] tmp;
		/***初始化***/
		crt.clear();
		new_name.clear();
		new_prnum.clear();
		memset(removed, 0, sizeof(removed));
		memset(renamed, 0, sizeof(renamed));
		memset(exist, 0, sizeof(exist));
		SetColor(color::foreground_chosen, color::background);
		printf("Successfully updated disk %s.\n", disk_path);
		SetColor(color::foreground_default, color::background);
	}
}
string get_path(int x, int pos)
{
	string path;
	path = data[x][pos].filename;
	register int fa = data[x][pos].ppos;
	while (1)
	{
		if (fa == 0)
			break;
		//cout<<data[x][fa].filename<<' '<<data[x][fa].rnum<<' '<<data[x][fa].prnum<<endl;
		path = "\\" + path;
		path = data[x][fa].filename + path;
		fa = data[x][fa].ppos;
	}
	return path;
}
void print_res1(const File &tmp, bool selected = 0)
{
	if (selected == 1)
	{
		SetColor(color::foreground_chosen, color::background);
	}
	else if (tmp.filesize == -1)
		SetColor(color::foreground_folder, color::background);
	else
		SetColor(color::foreground_default, color::background);
	cout << "NAME: " << tmp.filepath << endl;
}
void print_res(const File &tmp, bool selected = 0)
{
	if (selected == 1)
	{
		SetColor(color::foreground_chosen, color::background);
	}
	else if (tmp.filesize == -1)
		SetColor(color::foreground_folder, color::background);
	else
		SetColor(color::foreground_default, color::background);
	cout << "NAME: " << tmp.filepath;
	if (tmp.filesize != -1)
		cout << "\nSIZE: " << tmp.filesize << "KB\n";
	else
		cout << '\n';
	cout << "Creat Time             Change Time            Access Time\n";
	print_time(tmp.CreatT);
	cout << "    ";
	print_time(tmp.WriteT);
	cout << "    ";
	print_time(tmp.AccessT);
	cout << "\n\n";
	SetColor(color::foreground_default, color::background);
}
int restart_searching = 0; //是否有字符的变化
recursive_mutex mu;		   //线程互斥锁
bool finished = 0;
bool port_searching = 0, unport_searching = 0;
int cnt = 0, on_screen = 0, new_result = 0; //总结果数，屏幕上的结果数
int page_tot = 0, page_now = 1;				//总页数，当前页
void sear();
thread task01(sear);
void init_files()
{
	admincheck();
	bool is_first_run = check_local(); //若本地文件不存在，则判定为首次运行
	/*初始化信息*/
	//is_first_run=1;
	//cin>>is_first_run;
	if (is_first_run)
	{
		printf("This is your first run, program needs some time to initialize.\n");
		printf("Scanning, please wait several minutes.\n");
		SetColor(0x04, color::background);
		printf("Please close all anti-virus program running background\n");
		SetColor(color::foreground_default, color::background);
	}
	else
	{					 //判定是否出现不合法信息，若出现则当作第一次运行处理
		read_info();	 //读入USN64位ID和上次的最后一个USN
		if (disk_num != check_disk())
		{ //磁盘数量改变属于严重错误，应重新初始化
			SetColor(0x04, color::background);
			cout << "Fatal Error: Change in disk number detected.\n";
			cout << "The program will rescan all your disks, this will take several minutes.\n";
			SetColor(color::foreground_default, color::background);
			is_first_run = 1;
		}
		else
		{
			check_info(); //检查每块磁盘的USN64位ID是否改变，若改变则更新一下
		}
	}
	if (is_first_run)
		first_run();
	else
		update_database(); //先分析USN记录，记录要删除的文件再在读入时标记
	update_info();
	cout << "Totle Time : " << (double)clock() / CLOCKS_PER_SEC << "s" << endl;
	//Sleep(1000);
	cout << "Press any key to continue\n";
	getche();
	system("cls");
	register char c;
	cout << "Input any word to start searching\n";
	/********************************************************/
	//十分重要！首先要把搜索线程分离出去，不然会卡死！！！！！！！！
	task01.detach(); //分离线程
}

/*******************************/
//mainwindow.cpp
/*******************************/

const int INFO_MAX_SIZE = 256;
const int FILTERS_MAX_SIZE = 256;
class info
{
public:
	void print();
	void add_string(string adding_str);
	void clear() { size = cur_pt = 0; }
	void setPos(int Posx, int Posy)
	{
		posx = Posx;
		posy = Posy;
	}
	int get_choice()
	{
		if (cur_pt <= 0)
			cur_pt = 1;
		if (cur_pt >= size)
			cur_pt = size;
		return cur_pt;
	}
	void operator+=(int i) { cur_pt += i; }
	void operator-=(int i) { cur_pt -= i; }
	int run();
	info() { size = cur_pt = posx = posy = 0; }

private:
	int size, cur_pt, posx, posy;
	string info_str[INFO_MAX_SIZE];
} window_main, window_filters, window_sorting,
	window_help, window_other_settings;
void info::print()
{
	if (size == 0)
	{
		puts("Error in info::print()");
		return;
	}
	if (cur_pt <= 0)
		cur_pt = 1;
	if (cur_pt >= size)
		cur_pt = size;
	for (int i = 1; i <= size; i++)
	{
		SetPos(posx, posy + i - 1);
		if (cur_pt == i)
		{
			SetColor(color::foreground_buffer, color::background);
			putchar('>');
			cout << info_str[i];
			putchar('<');
			SetColor(color::foreground_default, color::background);
		}
		else
			cout << info_str[i];
		putchar('\n');
	}
}
void info::add_string(string adding_str)
{
	info_str[++size] = adding_str;
}
int info::run()
{
	char ch;
	while (1)
	{
		system("cls");
		print();
		ch = getch();
		if (ch == '\n' || ch == '\r')
			return get_choice();
		if (ch == 80 || ch == 77)
			(*this) += 1;
		if (ch == 72 || ch == 75)
			(*this) -= 1;
	}
}
struct filter
{
	/*
	filter_type = 1. 忽略文件夹
	filter_type = 2. 忽略文件
	filter_type = 3. 屏蔽带有filter_str后缀的文件
	filter_type = 4. 屏蔽filter_str文件目录下的所有文件
	filter_type = 5. 只搜索filter_str文件目录下的文件
*/
	int filter_type;
	string filter_str;
	filter()
	{
		filter_type = 0;
		filter_str.clear();
	}
};
class filters_package
{
public:
	filter filters[FILTERS_MAX_SIZE];
	int size;
	void addfilter(int filter_type, string filter_str)
	{
		if (filter_type == 5)
		{
			for (int i = 1; i <= size; i++)
				if (filters[i].filter_type == 5)
				{
					filters[i].filter_str = filter_str;
					return;
				}
		}
		filters[++size].filter_type = filter_type;
		filters[size].filter_str = filter_str;
	}
	void addfilter(int filter_type)
	{
		for (int i = 1; i <= size; i++)
			if (filters[i].filter_type == filter_type)
				return;
		filters[++size].filter_type = filter_type;
		filters[size].filter_str.clear();
	}
	void print()
	{
		puts("Current Filters:");
		if (size == 0)
			puts("None!");
		for (int i = 1; i <= size; i++)
			switch (filters[i].filter_type)
			{
			case 1:
				puts("Ignore folders.\n");
				break;
			case 2:
				puts("Ignore files.\n");
				break;
			case 3:
				cout << "Ignore suffix: " << filters[i].filter_str << "\n\n";
				break;
			case 4:
				cout << "Ignore path: " << filters[i].filter_str << "\n\n";
				break;
			case 5:
				cout << "Searching path: " << filters[i].filter_str << "\n\n";
				break;
			}
	}
	void file_print(ofstream &fout) {
		fout<<endl<<size<<endl;
		for(int i=1;i<=size;i++) {
			fout<<filters[i].filter_type<<' ';
			if(filters[i].filter_type == 3 || filters[i].filter_type == 4 || filters[i].filter_type == 5)
				fout<<filters[i].filter_str;
			fout<<endl;
		}
	}
	void file_read(ifstream &fin) {
		int i, filter_type; fin>>i;
		string filter_str;
		while(i--) {
			fin>>filter_type;
			if(filter_type == 3 || filter_type == 4 || filter_type == 5) {
				fin>>filter_str;
				addfilter(filter_type, filter_str);
			}
			else
				addfilter(filter_type);
		}
	}
	bool match(File matching_file);
	void clear() { size = 0; }
	filters_package() { size = 0; }
} filters;
int get_public_suffix_length(string x, string y)
{
	int lenx = x.length(), leny = y.length();
	for (int i = 1, maxi = min(lenx, leny); i <= maxi; i++)
		if (x[lenx - i] != y[leny - i])
			return i - 1;
	return min(lenx, leny);
}
int get_public_suffix_length(char *x, string y)
{
	int lenx = strlen(x), leny = y.length();
	for (int i = 1, maxi = min(lenx, leny); i <= maxi; i++)
		if (x[lenx - i] != y[leny - i])
			return i - 1;
	return min(lenx, leny);
}
int get_public_prefix_length(string x, string y)
{
	int lenx = x.length(), leny = y.length();
	for (int i = 1, maxi = min(lenx, leny); i <= maxi; i++)
		if (x[i - 1] != y[i - 1])
			return i - 1;
	return min(lenx, leny);
}
bool filters_package::match(File matching_file)
{
	// cout<<matching_file.filepath<<' '<<matching_file.filename<<' '<<matching_file.filesize<<endl;
	for (int i = 1; i <= size; i++)
	{
		if (filters[i].filter_type == 1)
		{
			if (matching_file.filesize == -1ll)
				return false;
		}
		else if (filters[i].filter_type == 2)
		{
			if (matching_file.filesize != -1ll)
				return false;
		}
		else if (filters[i].filter_type == 3)
		{
			// cout<<"\n\n\n"<<get_public_suffix_length(matching_file.filename,filters[i].filter_str)<<' '<<filters[i].filter_str<<endl;
			if (get_public_suffix_length(matching_file.filename, filters[i].filter_str) == filters[i].filter_str.length())
				return false;
		}
		else if (filters[i].filter_type == 4)
		{
			// cout<<"\n\n\n"<<get_public_prefix_length(matching_file.filepath,filters[i].filter_str)<<' '<<filters[i].filter_str<<endl;
			if (get_public_prefix_length(matching_file.filepath, filters[i].filter_str) == filters[i].filter_str.length())
				return false;
		}
		else
		{
			if (get_public_prefix_length(matching_file.filepath, filters[i].filter_str) != filters[i].filter_str.length())
				return false;
		}
	}
	return true;
}
bool return_to_main = 0, switch_filters = 0, //是否在搜索过程中应用筛选条件
	search_finished = 0;					 //后台搜索是否完成
int sorting_mode = 1;						 //0-不排序(default) 1&-1:大小 2&-2:创建时间 3&-3:修改时间 4&-4:访问时间
void sear()
{
FINISH:
	search_finished = 1;
	// cout<<"FINISHED"<<endl;
	while (!restart_searching); //开始时等待信号
	// cout<<"START"<<endl;
RESTART:
	search_finished = 0;
	return_to_main = 0;
	// cout<<"RESTART "<<con_buffer<<endl;
	// cout<<bufferlen<<' '<<(int)con_buffer[bufferlen+1]<<' '<<strlen(con_buffer)<<endl;
	/**********初始化************/
	restart_searching = 0;
	result.clear();		   //清空一下现有的结果
	pa.clear();			   //清空现有的页码记录
	pa.emplace_back(0, 0); //从1开始
	cnt = 0, on_screen = 0;
	page_now = 1, page_tot = 0;
	new_result = 0;
	/***************************/
	if (!match.read_reg(con_buffer, 0))
		goto FINISH;
	string disk = "C:\\", filepath;
	for (int x = 0; x < disk_num; x++, disk[0]++)
	{
		for (int i = 0; i < data[x].size(); i++)
		{
			if (restart_searching)
			{ //每次判断是否有字符的增加或删除，若有，则重新搜索
				goto RESTART;
			}
			if (return_to_main)
			{
				return_to_main = 0; //返回主界面
				goto FINISH;
			}
			if (new_result == rpp)
			{ //第一次输出特殊判断一下
				new_result = 0;
				page_tot++;
				pa.emplace_back(cnt - rpp, cnt);
			}
			if (match.match(data[x][i].filename))
			{
				File tmp(data[x][i], x, disk + get_path(x, i));
				if (switch_filters && !filters.match(tmp))
					continue;
				cnt++;
				result.push_back(tmp);
				if (on_screen < rpp)
					print_res(tmp);
				on_screen++;
				new_result++;
			}
		}
	}
	if (cnt != pa[page_tot].r)
	{ //最后一页可能有不完整的结果，特殊判断
		page_tot++;
		pa.emplace_back(pa[page_tot - 1].r, cnt);
	}
	goto FINISH; //万一搜索完了就返回第一行，不能退出！
}

/***************模糊搜索*************/
struct link
{
	dat info;
	int num, disk;
	double similarity;
	link *l, *r;
	link()
	{
		num = 0;
		disk = 0;
		similarity = 0;
		l = NULL;
		r = NULL;
	}
};
int nod_cnt = 0;						   //总的节点数
link *link_head = NULL, *link_tail = NULL; //链表的头尾
void init_sim_sort()
{ //清空链表
	link *now = link_head;
	while (now != NULL)
	{
		link *nex = now->r;
		delete now;
		now = nex;
	}
}
link *add_nod(link *pos, dat x, double similarity, int num, int disk)
{ //在pos之后添加一个
	link *tmp = new link;
	tmp->info = x;
	tmp->similarity = similarity;
	tmp->num = num;
	tmp->disk = disk;
	if (pos->r != NULL)
	{
		tmp->r = pos->r;
		(pos->r)->l = tmp;
	}
	pos->r = tmp;
	tmp->l = pos;
	return tmp;
}
void del_nod(link *pos)
{ //删除pos
	if (pos->l != NULL)
		(pos->l)->r = pos->r;
	if (pos->r != NULL)
		(pos->r)->l = pos->l;
	delete pos;
}
int Stop_sim_sort = 0;
void sim_sort()
{
	nod_cnt = 0;
	link_head = new link; //新建一个空的头
	link_head->similarity = 2;
	link_tail = new link; //新建一个空的尾
	link_tail->r = NULL;
	link_tail->l = link_head;
	link_head->r = link_tail;
	link_head->l = NULL;
	string disk = "C:\\";
	for (int x = 0; x < disk_num; x++)
	{
		for (int i = 0; i < data[x].size(); i++)
		{
			//cout<<i<<endl;
			if (Stop_sim_sort)
			{ //停止了
				init_sim_sort();
				return;
			}
			//cout<<i<<endl;
			//cout<<"********"<<endl;
			//cout<<data[i].filename<<endl;
			double similarity = data[x][i].Cosine_Similarity(con_buffer);
			if (similarity < threshold)
				continue; //匹配度低于阈值，跳过
			//cout<<"fuck "<<data[x][i].filename<<' '<<similarity<<endl;
			if (nod_cnt < sim_tot || link_tail->similarity < similarity || (link_tail->similarity == similarity && link_tail->info.len > data[x][i].len))
			{
				if (nod_cnt == 0)
				{ //第一个要特判
					link *tmp = add_nod(link_head, data[x][i], similarity, i, x);
					delete link_tail;
					link_tail = tmp;
					link_tail->r = NULL;
					nod_cnt++;
				}
				else
				{
					if (nod_cnt == sim_tot)
					{
						//cout<<"delete!!!! "<<link_tail<<endl;
						link *tmp = link_tail->l; //删除tail
						del_nod(link_tail);
						link_tail = tmp;
						//cout<<link_tail<<endl;
						nod_cnt--;
					}
					link *now = link_tail;
					while (now->similarity < similarity || (now->similarity == similarity && now->info.len > data[x][i].len))
					{
						now = now->l;
					}
					bool flag = 0;
					if (now->r == NULL)
						flag = 1;
					//cout<<now<<' '<<now->r<<endl;
					link *tmp = add_nod(now, data[x][i], similarity, i, x);
					if (flag)
					{
						link_tail = tmp;
					}
					nod_cnt++;
				}
				link *now = link_head->r;
				system("cls");
				SetColor(color::foreground_chosen, color::background);
				cout << con_buffer << endl;
				SetColor(color::foreground_default, color::background);
				//cout<<link_head<<endl;
				while (true)
				{
					string tmp1 = disk;
					tmp1[0] += now->disk;
					File tmp(now->info, now->disk, tmp1 + get_path(now->disk, now->num));
					//cout<<now->similarity<<endl;
					print_res1(tmp);
					//cout<<now->info.filename<<' '<<now->similarity<<' '<<now<<' '<<now->l<<' '<<now->r<<endl;
					if (now == link_tail)
						break;
					now = now->r;
				}
				//cout<<"fuck"<<endl;
				//getchar();
			}
		}
	}
}
void start_searching()
{
	bool F5_flag = 0, F3_flag = 0;
	port_searching = 0;
	unport_searching = 1;
	char c;
	page = 1;
	system("cls");
	SetColor(color::foreground_buffer, color::background);
	cout << "Input any word to start searching\n";
	SetColor(color::foreground_default, color::background);
	while (1)
	{
		c = getch();
		show_egg();
		bool is_changed = 0;
		if (c == 0)
		{ //按Fn键组切换排序方式与筛选器开关
			int fn_valume = getch() - 58;
			// cout<<"DDDD   "<<fn_valume<<endl;
			switch (fn_valume)
			{
			case 3:
			{
				system("cls");
				SetColor(color::foreground_chosen, color::background);
				cout << con_buffer << endl;
				SetColor(color::foreground_default, color::background);
				return_to_main = 1; //停止搜索进程
				Stop_sim_sort = 0;
				thread task02(sim_sort); //新建搜索线程
				task02.detach();		 //分离出线程
				char cc;
				int is_print_info=0;
				while (1)
				{
					cc=getche();
					if (cc == 27)
					{
						Stop_sim_sort = 1;
						goto QUIT_SORTING;
					}
					else if(!is_print_info){
						is_print_info=1;
						cout<<"Program is still running backwards, please wait for the program to get results.\n";
						cout<<"If you want to quit sorting, please press ESC.\n";
					}
				}
				break;
			}
			case 2:
				sorting_mode = window_sorting.run()-1;
				break;
			case 10:
				if (sorting_mode != 0)
				{ //满足条件时对结果进行排序
					fileh_switch_debug = 1;
					if (!search_finished)
					{
						cout << "The program is still searching results, please wait.\n";
						cout << "If you want to quit sorting, please press ESC button\n";
						char cc;
						while (1)
						{
							if (search_finished)
								break;
							if (!kbhit())
								continue;
							cc = getche();
							if (cc == 0 || cc == -32)
								cc = getche(); //艹可能会有输入奇妙的符号的操作
							if (cc == 27)
							{
								goto QUIT_SORTING;
							}
						}
					}
					system("cls");
					puts("Sorting begin!");
					switch (sorting_mode)
					{
					case 1:
						merge_sort(result, Name_cmp_s);
					case 2:
						merge_sort(result, size_cmp_s);
						break;
					case 3:
						merge_sort(result, size_cmp_l);
						break;
					case 4:
						merge_sort(result, CreatT_cmp_s);
						break;
					case 5:
						merge_sort(result, CreatT_cmp_l);
						break;
					case 6:
						merge_sort(result, WriteT_cmp_s);
						break;
					case 7:
						merge_sort(result, WriteT_cmp_l);
						break;
					case 8:
						merge_sort(result, AccessT_cmp_s);
						break;
					case 9:
						merge_sort(result, AccessT_cmp_l);
						break;
					}
					puts("Sorting completed!");
					fileh_switch_debug = 0;
				}
				else if (!search_finished)
					puts("Search hasn't finished. Please wait...");
				break;
			case 5:
				if (!F5_flag)
				{ //上次如果按了F5就不要再输出了
					F5_flag = 1;
					cout << "Sorting mode = " << sorting_mode << " ;" << (switch_filters ? "Filters activated" : "Filters disabled") << endl;
					cout << "Press F2 to select sorting mode.\n";
					cout << "If you want to set filters, please press ESC to return to main menu.\n";
					cout << "For more information, press F1 to open manual.\n";
				}
				continue;
			case 1: //help
				ShellExecute(NULL, _T("open"), _T("help.txt"), NULL, NULL, SW_SHOW);
				break;
			}
			F5_flag = 0;
			now_result = 0;
			if (fn_valume == 10)
				page_now = 1;
			system("cls");
			SetColor(color::foreground_chosen, color::background);
			cout << con_buffer << endl;
			SetColor(color::foreground_default, color::background);
			cout << "PAGE: " << page_now << endl;
			if (pa.size() != 1)
				for (int i = pa[page_now].l; i < pa[page_now].r; i++)
				{
					print_res(result[i]);
				}
			SetColor(color::foreground_default, color::background);
			if (search_finished)
				cout << "#Search completed: Total " << result.size() << " results!\n\n";
			continue;
		}
		if (c == 13)
		{
			if (now_result == 0)
				continue;
			else
			{
				string tmp_path = result[pa[page_now].l + now_result - 1].filepath;
				if (result[pa[page_now].l + now_result - 1].filesize == -1)
				{
					tmp_path += "\\";
					ShellExecute(NULL, _T("explore"), _T(tmp_path.c_str()), NULL, NULL, SW_SHOW); //打开文件夹
				}
				else
				{
					ShellExecute(NULL, _T("open"), _T(tmp_path.c_str()), NULL, NULL, SW_SHOW); //默认方式运行
				}
			}
			continue;
			//continue;//忽略回车
		}
		if (c == 27)
		{ //esc 返回主菜单
			return_to_main = 1;
			memset(con_buffer, 0, sizeof(con_buffer));
			bufferlen = -1;
			return;
		}
		if (c == 8 && bufferlen != -1)
		{ //删除
			return_to_main=1;
			con_buffer[bufferlen] = '\0';
			bufferlen--;
			page = 1;
			if (bufferlen >= 0)
			{ //中文类型需要特殊判断，完整删除两个ASCII
				if (is_chinese[bufferlen])
				{
					is_chinese[bufferlen] = 0;
					con_buffer[bufferlen] = '\0';
					bufferlen--;
				}
			}
			if (bufferlen == -1)
			{ //删干净了
				page = 1;
				system("cls");
				SetColor(color::foreground_buffer, color::background);
				cout << "Input any word to start searching\n";
				SetColor(color::foreground_default, color::background);
				continue;
			}
			else
			{
				page = 1;
				system("cls");
				SetColor(color::foreground_chosen, color::background);
				cout << con_buffer << endl;
				SetColor(color::foreground_default, color::background);
				cout << "PAGE: " << page << endl;
				restart_searching = 1;
			}
			continue;
		}
		else if (c == 8 && bufferlen == -1)
		{ //防止溢出
			continue;
		}
		else if (c == -32)
		{ //翻页操作
			char direction = getch();
			if (bufferlen == -1)
				continue;
			if (direction == 77)
			{ //向下翻页
				if (page_now == page_tot)
					continue; //当前是最后一页时忽略该命令
				else
				{ //翻页操作
					now_result = 0;
					if (pa.size() != 1)
						page_now++;
					system("cls");
					SetColor(color::foreground_chosen, color::background);
					cout << con_buffer << endl;
					SetColor(color::foreground_default, color::background);
					cout << "PAGE: " << page_now << endl;
					if (pa.size() != 1)
						for (int i = pa[page_now].l; i < pa[page_now].r; i++)
						{
							print_res(result[i]);
						}
					SetColor(color::foreground_default, color::background);
					if (search_finished)
						cout << "#Search completed: Total " << result.size() << " results!\n\n";
					continue;
				}
			}
			else if (direction == 75)
			{ //向上翻页
				if (page_now != 1)
				{ //当前页不是第一页才能执行翻页操作
					now_result = 0;
					system("cls");
					SetColor(color::foreground_chosen, color::background);
					cout << con_buffer << endl;
					SetColor(color::foreground_default, color::background);
					if (pa.size() != 1)
						page_now--;
					cout << "PAGE: " << page_now << endl;
					if (pa.size() != 1)
						for (int i = pa[page_now].l; i < pa[page_now].r; i++)
						{
							print_res(result[i]);
						}
					SetColor(color::foreground_default, color::background);
					if (search_finished)
						cout << "#Search completed: Total " << result.size() << " results!\n\n";
					continue;
				}
				else
					continue; //当前是第一页时忽略翻页操作
			}
			else if (direction == 80)
			{ //向下选择结果
				if(pa.size()==1)continue;
				register bool selected = 0;
				if (now_result < pa[page_now].r - pa[page_now].l)
					now_result++;
				else
					continue;
				system("cls");
				SetColor(color::foreground_chosen, color::background);
				cout << con_buffer << endl;
				SetColor(color::foreground_default, color::background);
				cout << "PAGE: " << page_now << endl;
				for (int i = pa[page_now].l; i < pa[page_now].r; i++)
				{
					selected = 0;
					if (i - pa[page_now].l + 1 == now_result)
						selected = 1;
					print_res(result[i], selected);
				}
				if (search_finished)
					cout << "#Search completed: Total " << result.size() << " results!\n\n";
			}
			else if (direction == 72)
			{ //向上选择结果
				if(pa.size()==1)continue;
				register bool selected = 0;
				if (now_result > 1)
					now_result--;
				else
					continue;
				system("cls");
				SetColor(color::foreground_chosen, color::background);
				cout << con_buffer << endl;
				SetColor(color::foreground_default, color::background);
				cout << "PAGE: " << page_now << endl;
				for (int i = pa[page_now].l; i < pa[page_now].r; i++)
				{
					selected = 0;
					if (i - pa[page_now].l + 1 == now_result)
						selected = 1;
					print_res(result[i], selected);
				}
				if (search_finished)
					cout << "#Search completed: Total " << result.size() << " results!\n\n";
			}
		}
		else
		{ //更新关键字重新搜索
			if (c == 22)
			{ //读取剪切板
				char *ClipBoard = GetClipboard();
				if (ClipBoard == NULL)
					continue;
				for (int i = 0; ClipBoard[i]; i++)
				{
					con_buffer[++bufferlen] = ClipBoard[i];
				}
			}
			else
			{
				con_buffer[++bufferlen] = c;
			}
			page_now = 1;
			if (c < 0)
			{ //若当前输入的是中文，要同时读进两个字符，减少bug可能
				c = getch();
				is_chinese[bufferlen] = 1;
				con_buffer[++bufferlen] = c;
			}
		QUIT_SORTING:
			system("cls");
			SetColor(color::foreground_chosen, color::background);
			cout << con_buffer<< endl;
			SetColor(color::foreground_default, color::background);
			cout << "PAGE: " << page_now << endl;
			if(bufferlen==2&&con_buffer[0]=='T'&&con_buffer[1]=='H'&&con_buffer[2]=='U'){
				SetColor(color::foreground_chosen, color::background);
				printf("Thank you for using!!\n");
				SetColor(color::foreground_default, color::background);
			}
			now_result = 0;
			restart_searching = 1;
		}
	}
	unport_searching = 0;
	port_searching = 1;
}
void set_searching_filter()
{
	int window_result;
	string tmp;
	while (1)
	{
		save_settings();
		window_result = window_filters.run();
		if (window_result == 1 || window_result == 2)
			filters.addfilter(window_result);
		else if (window_result == 3 || window_result == 4 || window_result == 5)
		{
			puts("Please input the corresponding string..");
			cin >> tmp;
			filters.addfilter(window_result, tmp);
		}
		else if (window_result == 6)
			filters.clear();
		else if (window_result == 7)
			switch_filters = !switch_filters;
		else
			return;
		filters.print();
		cout << (switch_filters ? "Filters activated!" : "Filters disabled.") << endl;
		puts("Press any botton to continue...");
		getch();
	}
}
void other_settings()
{
	int window_result;
	char ch;
	int tmpa, tmpb, tmpc, tmpd, tmpe;
	while (true)
	{
		save_settings();
		window_result = window_other_settings.run();
		if(window_result == 1) {
			double _threshold;
			int _sim_tot;
			puts("Please input the thresold(double) between 0~1 and number of results per page of simsort(int) between 1~50");
			cin>>_threshold>>_sim_tot;
			if(_threshold <= 0 || _threshold >= 1)
				puts("Fail: Threshold should be in the range of (0,1)");
			if(_sim_tot<=0 || _sim_tot>50)
				puts("Fail: Results per page should be in the range of [1,50]");
			if(_threshold <= 0 || _threshold >= 1 || _sim_tot<=0 || _sim_tot>50) {
				getch();
				continue;
			}
			threshold = _threshold;
			sim_tot = _sim_tot;
		}
		else if (window_result == 2)
		{
			RE_INPUT:
			puts("Please input a valid integer in range of 4~10");
			cin >> rpp;
			if(rpp<4||rpp>10)goto RE_INPUT;
		}
		else if (window_result == 3)
		{
			while (true)
			{
				system("cls");
				SetColor(color::foreground_default, color::background);
				cout << color::foreground_default << ' ';
				puts("Normal text.      Input number 1 to change..");
				SetColor(color::foreground_chosen, color::background);
				cout << color::foreground_chosen << ' ';
				puts(">Chosen text.<       Input number 2 to change..");
				SetColor(color::foreground_folder, color::background);
				cout << color::foreground_folder << ' ';
				puts("Folder.       Input number 3 to change..");
				SetColor(color::foreground_buffer, color::background);
				cout << color::foreground_buffer << ' ';
				puts("Searching buffer.       Input number 4 to change..");
				SetColor(color::foreground_default, color::background);
				SetColor(color::foreground_default, color::background);
				cout << color::background << ' ';
				puts("Background.      Input number 5 to change..");
				SetColor(color::foreground_chosen, color::background);
				puts("Input any corresponding integer and the ID of color you want to change to (integer, [0,14]) to change color.");
				puts("Input 0 0 to return to last menu\n");
				cin >> tmpa >> tmpb;
				if(tmpb<0 || tmpb>15) {
					puts("Invalid color ID!");
					continue ;
				}
				if(tmpa == 1)
					color::foreground_default = tmpb;
				else if(tmpa == 2)
					color::foreground_chosen = tmpb;
				else if(tmpa == 3)
					color::foreground_folder = tmpb;
				else if(tmpa == 4)
					color::foreground_buffer = tmpb;
				else if(tmpa == 5)
					color::background = tmpb;
				else
					SetColor(color::foreground_default, color::background);
					break;
			}
		}
		else
			break;
	}
}
void init_windows()
{
	SetTitle("A GOOD LOCAL FILE SEARCHER");
	read_egg();
	//SetSize(140,40);
	HideConsoleCursor();
	// main window:
	window_main.add_string("Start Searching");
	window_main.add_string("Set Searching Fliter");
	window_main.add_string("Other Settings");
	window_main.add_string("Save & Quit");
	window_main.setPos(60, 14);
	// filters settings window:
	window_filters.add_string("Ignore Folders");
	window_filters.add_string("Ignore Files");
	window_filters.add_string("Ignore a Specific Suffix");
	window_filters.add_string("Ignore a Specific Path");
	window_filters.add_string("Only search in a Specific Path(will cover the last filter!)");
	window_filters.add_string("Erase all");
	window_filters.add_string("Active/Disable the filters while searching");
	window_filters.add_string("Back");
	window_filters.setPos(60, 14);
	// Sorting mode setting window:
	window_sorting.add_string("Close the sorting function");
	window_sorting.add_string("Name - Ascending order");
	window_sorting.add_string("Size - Ascending order");
	window_sorting.add_string("Size - Descending order");
	window_sorting.add_string("Create time - Ascending order");
	window_sorting.add_string("Create time - Descending order");
	window_sorting.add_string("Write time - Ascending order");
	window_sorting.add_string("Write time - Descending order");
	window_sorting.add_string("Access time - Ascending order");
	window_sorting.add_string("Access time - Descending order");
	window_sorting.setPos(60, 14);
	// Other settings window:
	window_other_settings.add_string("Simsort settings");
	window_other_settings.add_string("Set results per page");
	window_other_settings.add_string("Set theme and colors");
	window_other_settings.add_string("Back");
	window_other_settings.setPos(60, 14);
}
void read_settings() {
	ifstream fin("Settings.db");
	if(!fin.is_open())
		return ;
	fin>>color::foreground_default>>color::foreground_chosen>>
		color::foreground_folder>>color::foreground_buffer>>color::background;
	fin>>threshold>>sim_tot>>rpp>>sorting_mode>>switch_filters;
	filters.file_read(fin);
	fin.close();
}
void save_settings() {
	ofstream fout;
	fout.open("Settings.db",ios::trunc);
	fout<<color::foreground_default<<' '<<color::foreground_chosen<<' '<<
		color::foreground_folder<<' '<<color::foreground_buffer<<' '<<color::background<<endl;
	fout<<threshold<<' '<<sim_tot<<' '<<rpp<<' '<<sorting_mode<<' '<<switch_filters<<endl;
	filters.file_print(fout);
	fout.close();
}
void mainwindow()
{
	int window_result;
	read_settings();
	save_settings();
	while (1)
	{
		window_result = window_main.run();
		switch (window_result)
		{
		case 1:
			start_searching();
			break;
		case 2:
			set_searching_filter();
			save_settings();
			break;
		case 3:
			other_settings();
			save_settings();
			break;
		default:
			save_settings();
			return;
		}
	}
}

signed main()
{
	// string a,b;while(1)
	// cin>>a>>b,
	// cout<<get_public_suffix_length(a,b)<<' '<<get_public_prefix_length(a,b)<<endl;

	init_windows();
	init_files();
	mainwindow();
}