clientmain.cpp

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

#include "client.cpp"
#include <iostream>
using namespace std;

bool doCommand(client& net)
{
	int cmd ;
	char buf[20];
	char temp[20];
	unsigned int fileSize;	
	memset(buf,0,20);
	memset(temp,0,20);
	cin >> cmd;
	cin.clear();
	if(cmd == 1)
	{
		cout << "please input userName userPwd    ";
		cin>>buf>>temp;
		char result = net.userLogin(buf,temp);
		return result;
	}
	else if(cmd == 2)
	{
		cout << "please input userName userPwd    ";
		cin>>buf>>temp;
		char result = net.userRegister(buf,temp);
		return result;
	}
	else if( cmd == 3)
	{
		
		cout << "please input fileName fileTime fileSize    ";
		cin>>buf>>temp>>fileSize;
		char result = net.upLoadFile(buf,temp,fileSize);
		return result;
	}
	else if(cmd ==4)
	{
		
		cout << "please input fileName    ";
		cin>>buf;
		char result = net.downLoadFile(buf);
		return result;
	}
		
}

int main(int argc,char* argv[])
{	
	client net;
//	if(!net.serverIsMaxConnect())
//	{
//		return -1;
//	}
	cout <<"1:userLogin userName&userPwd\n"<<endl;
	cout <<"2:userRegister userName userPwd \n" ;
	cout <<"3:upLoadFile fileName fileTime,fileSize\n";
	cout <<"4:downLoadFile fileName"<< endl;
	while(1)
	{	
		cout << "if you want to quit(q), please choose 'q' or 'c'\n";
		
		char temp;
		cin>>temp;
		cin.clear();
		if(temp == 'q')
		{
			break;
		}
		
		cout <<"please input the number:   ";
		doCommand(net);
	}
}