ZYM/BH-Bookstore-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

给exit和quit添加校验

Browse Source
This commit is contained in:
ZhuangYumin
2023-12-14 06:40:48 +00:00
parent 7fcaaa534e
commit 8f5279dd9f
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/src/engine.cpp
View File

@ -46,6 +46,10 @@ std::vector<std::string> BookStoreEngineClass::Execute(
std::string head = "";
ss >> head;
if (head == "quit" || head == "exit") {
static std::basic_regex<char> quit_regex(R"(^ *(quit|exit) *$)",
std::regex_constants::optimize);
if (!std::regex_match(cmd, quit_regex))
return std::vector<std::string>({"Invalid"});
while (login_stack.size()) {
login_count[login_stack.top().first]--;
login_stack.pop();
@ -394,9 +398,9 @@ std::vector<std::string> BookStoreEngineClass::ExecuteShowFinance(
std::vector<std::string> BookStoreEngineClass::ExecuteLog(
const std::string &cmd,
std::stack<std::pair<std::string, int>> &login_stack) {
static std::basic_regex<char> log_regex(
R"(^ *log *$)", std::regex_constants::optimize);
if(!std::regex_match(cmd, log_regex))
static std::basic_regex<char> log_regex(R"(^ *log *$)",
std::regex_constants::optimize);
if (!std::regex_match(cmd, log_regex))
return std::vector<std::string>({"Invalid"});
return std::vector<std::string>();
}