Kiritow/DBHomework
Archived
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update DBHomework

Browse Source
This commit is contained in:
Kirigaya Kazuto 2017-11-20 11:43:49 +08:00
parent 7ecb062364
commit 1031ce7258
5 changed files with 176 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
design.txt
View File

@ -198,10 +198,8 @@
账户或密码错误(包括账户错误)
返回:
当登陆成功时返回一个apikey用于调用以上的API. 此key在一段时间内有效.
当登陆成功时绑定User到Session.
http://booksys.com/api/v1/logout POST
POST:
apikey
http://booksys.com/api/v1/logout GET/POST
提示:退出登陆. 会立刻注销此apikey
提示:退出登陆. 会立刻注销Session.

4
jsonfail.cpp
View File

@ -24,7 +24,9 @@ const char* GetErrMsg(int errcode)
return "Data Error";
case err_session:
return "Session Error";
default:
case err_general:
return "General error";
default:
return "Unknown error";
}
}

1
jsonfail.h
View File

@ -10,5 +10,6 @@
#define err_permission_denied 8
#define err_data 9
#define err_session 10
#define err_general 11
const char* GetErrMsg(int errcode);

175
login.cpp
View File

@ -1,11 +1,12 @@
#include "Request.h"
#include "Response.h"
#include "Session.h"
#include "Util.h"
#include "json.hpp"
#include "jsonfail.h"
using namespace std;
using json=nlohmann::json;
#define postval(NAME) string NAME=req.post[#NAME]
int main()
{
Request req;
@ -13,73 +14,133 @@ int main()
Response res;
json j;
auto jsonfail=[&](int errcode,const std::string& errmsg)
auto jsonfail=[&](int errcode,const string& detail="")
{
j["success"]=0;
j["errcode"]=errcode;
j["errmsg"]=errmsg;
j["errmsg"]=string(GetErrMsg(errcode))+" : "+detail;
};
if(req.requestMethod=="POST")
do
{
if(!se.isReady())
{
jsonfail(err_session);
break;
}
if(!se.isNew()||!se.getUser().empty())
{
/// Logged in...
j["success"]=2;
j["next_url"]="/booksys/dashboard.html";
break;
}
if(req.requestMethod!="POST")
{
jsonfail(err_method_not_supported);
break;
}
if(req.post["username"].empty() ||
req.post["password"].empty())
req.post["password"].empty() )
{
jsonfail(2,"Missing Parameter");
jsonfail(err_missing_parameter);
break;
}
else
postval(username);
postval(password);
DBInfo db;
MySQLConn conn;
if(db.readConfig()<0)
{
DBInfo db;
if(db.readConfig()<0)
{
jsonfail(3,"Failed to read configure");
}
else
{
string uname=req.post["username"];
string upass=req.post["password"];
MySQLConn conn;
if(db.connectProxy(conn)<0)
{
jsonfail(4,"Failed to connect DB");
}
else
{
bool authpass=false;
if(conn.exec(make_str("select count(username) from bs_user where username='",uname,"' and password='",upass,"'"),
[&](MySQLResult& res)
{
res.stepRow([&](char** val,unsigned long* len)
{
if(string(val[0])=="1")
{
authpass=true;
}
});
})<0)
{
jsonfail(5,"Failed to execute SQL.");
}
else
{
/// Try to mark session logged in
if(se.setUser(uname)<0)
{
jsonfail(6,"Failed to set session.");
}
else
{
j["success"]=1;
}
}
}
}
jsonfail(err_config);
break;
}
if(db.connectProxy(conn)<0)
{
jsonfail(err_connect);
break;
}
/// Check if password is correct
bool authpass=false;
if(conn.exec(make_str("select count(username) from bs_user where username='",
username,
"' and password='",
password,
"'"),
[&](MySQLResult& res)
{
res.stepRow([&](char** val,unsigned long* len)
{
if(string(val[0])=="1")
{
authpass=true;
}
});
})<0)
{
jsonfail(err_sql,"Step 1");
break;
}
if(!authpass)
{
jsonfail(err_data,"Auth Failed");
break;
}
/// Check if the user is allowed to login
int account_status;
if(conn.exec(make_str("select account_status from bs_user where username='",
username,
"'"),
[&](MySQLResult& res)
{
res.stepRow([&](char** val,unsigned long* len)
{
account_status=ParseInt(val[0]);
});
})<0)
{
jsonfail(err_sql,"Step 2");
break;
}
if(account_status==0)
{
/// Act as auth failed (username or password wrong) when account is hidden.
jsonfail(err_data,"Auth Failed");
break;
}
else if(account_status==1)
{
/// Banned
jsonfail(err_data,"Your account is banned.");
break;
}
else if(account_status==2)
{
/// Need verify, redirect to account enable.
jsonfail(err_data,"Need verify first.");
j["next_url"]="/booksys/enable.html";
}
/// Try to mark session logged in
int ret;
if((ret=se.setUser(username))<0)
{
jsonfail(err_session,make_str("Failed to set session, ret=",ret));
break;
}
j["success"]=1;
}
else
{
jsonfail(1,"Request Method Not Supported");
}
while(0);
res.content.append(j.dump());
res.show();

51
logout.cpp Normal file
View File

@ -0,0 +1,51 @@
#include "Session.h"
#include "Util.h"
#include "json.hpp"
#include "jsonfail.h"
using namespace std;
using json=nlohmann::json;
#define postval(NAME) string NAME=req.post[#NAME]
int main()
{
Request req;
Session se(req);
Response res;
json j;
auto jsonfail=[&](int errcode,const string& detail="")
{
j["success"]=0;
j["errcode"]=errcode;
j["errmsg"]=string(GetErrMsg(errcode))+" : "+detail;
};
do
{
if(!se.isReady())
{
jsonfail(err_session);
break;
}
if(se.isNew()||se.getUser().empty())
{
jsonfail(err_need_login);
break;
}
int ret;
if((ret=se.setUser(""))<0)
{
jsonfail(err_session,make_str("Failed to logout. Session Module returns ",ret));
break;
}
j["success"]=1;
}while(0);
res.content.append(j.dump());
res.show();
return 0;
}