management.cpp

来自「安全文件传输」· C++ 代码 · 共 72 行

#include "management.h"
/*
1 log
2 reg
3 upload
4 download
*/
management::management():user("server.db")
{
	flag = 0;
		
}

bool management::userManagement(int cmd,char* userName,char* userPwd)
{
	if( 1 == cmd )
	{
		bool temp = user.loginUserDatabase(userName,userPwd);
		if( temp == 1)
		{
			flag = 1;
			string userPath;
			userPath = userPath + userName;
			chdir(userPath.c_str());
			//system(userPath.c_str());
			return true;
		}
		else return false;
	}	
	else if( 2 == cmd )
	{	
		
		bool temp = user.registerUserDatabase(userName,userPwd);
		if(temp == 1)
		{
			string userPath;
			userPath = userPath + "mkdir " + userName;
			system(userPath.c_str());
			//return temp;
			return true;
		}
		else return temp;
		
	}
	else return false;
	
}
bool management::fileManagement(int cmd,char* fileName,char* fileTime,unsigned int fileSize)
{
	if(3 == cmd)
	{
		if(flag == 1)
		{
			cout << "creat filedb"<< endl;
			fileMetadata file("fileDB.db");
			bool temp = file.upLoadMetadata(fileName,fileTime,fileSize);
			return temp;
		}
		return false;
	}
	else if(4 == cmd)
	{
		if(flag == 1)
		{
			fileMetadata file("fileDB.db");
			bool temp = file.downLoadMetadata(fileName);
			return temp;
		}
		return false;
	}
}