userdatabase.cpp

安全文件传输

#include "userDatabase.h"

userDatabase::userDatabase(char* pathname)
{	
	userResult = NULL;
	zErrMsg = NULL;
	db = NULL;
	zErrMsg = 0;
	int rc = sqlite3_open(pathname,&db);
	
	if( rc )
	{
		cout << "can't open database " << pathname << sqlite3_errmsg(db) << endl;
		sqlite3_close(db);
		exit(1);
	}
	else
		cout << "you have opened a sqlite3 database named "<<pathname << " succesfuly!" << endl ; 	
	

	string sql = "CREATE TABLE userPwd(userName VARCHAR(12),userPassword VARCHAR(12));" ;

	sqlite3_exec(db,sql.c_str(),0,0,&zErrMsg);
}

void userDatabase::insertUserDatabase(char* userName, char* pwd)
{
	string sql = "INSERT INTO \"userPwd\" VALUES (";
	sql = sql +"'"+ userName + "','" + pwd + "');";
	sqlite3_exec(db,sql.c_str(),0,0,&zErrMsg);
	cout << sql << endl;
}

void userDatabase::deleteUserDatabase(char* userName)
{
	string sql = "DELETE FROM userPwd WHERE userName =";
	sql = sql + userName +";";
	sqlite3_exec(db,sql.c_str(),0,0,&zErrMsg);
}
	
bool  userDatabase::loginUserDatabase(char* userName,char* pwd)
{	
	
	int nrow = 0 , ncolumn = 0;
	
	string sql = "SELECT * FROM userPwd";
	sqlite3_get_table(db,sql.c_str(),&userResult,&nrow,&ncolumn,&zErrMsg);
	for(int i =2; i < (nrow+1)*ncolumn;i+=2)
	{
		bool name = !strcmp(userResult[i],userName);
		bool password = !strcmp(userResult[i+1],pwd);
		if(name&&password)
			return true;
	}
	cout << "username or password is failed" << endl;
	return false;
}
bool userDatabase::registerUserDatabase(char* userName,char* pwd)
{
	int nrow = 0 , ncolumn = 0;
        
        string sql = "SELECT * FROM userPwd";
        sqlite3_get_table(db,sql.c_str(),&userResult,&nrow,&ncolumn,&zErrMsg);
        for(int i =2; i < (nrow+1)*ncolumn;i+=2)
        {
               bool name = strcmp(userResult[i],userName);
                
                if(!name)
		{	
			cout << "userName is exist! " << endl;
                        return false;
		}
        }
	insertUserDatabase(userName,pwd);
	return true;
}

userDatabase::~userDatabase()
{	
	if(userResult != NULL)
		sqlite3_free_table(userResult);
	if(db !=NULL)
		sqlite3_close(db);
	zErrMsg = NULL;
}