expand/CCash
1
0
Fork 0
mirror of https://github.com/Expand-sys/CCash synced 2026-03-22 20:47:10 +11:00
Code Issues Projects Releases Packages Wiki Activity Actions

admin filter header

Browse source
This commit is contained in:
EntireTwix 2021-06-27 17:10:51 -07:00
parent 371cd16e38
commit 7dec3bea72
4 changed files with 20 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
CMakeLists.txt
View file

@ -13,6 +13,7 @@ set(CMAKE_CXX_FLAGS_RELEASE "-O3")
find_package(Threads REQUIRED) find_package(Threads REQUIRED)
add_executable(${PROJECT_NAME} main.cpp ) add_executable(${PROJECT_NAME} main.cpp )
target_sources(${PROJECT_NAME} PRIVATE target_sources(${PROJECT_NAME} PRIVATE
src/admin_filter.cpp
src/bank_api.cpp src/bank_api.cpp
src/bank.cpp src/bank.cpp
src/log.cpp src/log.cpp

1
include/bank_api.h
View file

@ -1,5 +1,6 @@
#pragma once #pragma once
#include <drogon/HttpController.h> #include <drogon/HttpController.h>
#include "admin_filter.h"
#include "user_filter.h" #include "user_filter.h"
using namespace drogon; using namespace drogon;

2
main.cpp
View file

@ -87,6 +87,7 @@ int main(int argc, char **argv)
//endpoints //endpoints
auto APIv1 = std::make_shared<v1::api>(bank); //v1 auto APIv1 = std::make_shared<v1::api>(bank); //v1
auto user_filter = std::make_shared<UserFilter>(bank); auto user_filter = std::make_shared<UserFilter>(bank);
auto admin_filter = std::make_shared<AdminFilter>(bank);
app().registerPostHandlingAdvice( app().registerPostHandlingAdvice(
[](const drogon::HttpRequestPtr &req, const drogon::HttpResponsePtr &resp) { [](const drogon::HttpRequestPtr &req, const drogon::HttpResponsePtr &resp) {
@ -95,6 +96,7 @@ int main(int argc, char **argv)
app() app()
.loadConfigFile(config_location) .loadConfigFile(config_location)
.registerFilter(user_filter) .registerFilter(user_filter)
.registerFilter(admin_filter)
.registerController(APIv1) .registerController(APIv1)
.setThreadNum(get_nprocs()) .setThreadNum(get_nprocs())
.run(); .run();

28
src/admin_filter.cpp
View file

@ -1,6 +1,6 @@
#include "user_filter.h" #include "admin_filter.h"
static char DecodeChar(const char ch) static char DecodeChar2(const char ch)
{ {
if (ch >= 'A' && ch <= 'Z') if (ch >= 'A' && ch <= 'Z')
{ {
@ -17,7 +17,7 @@ static char DecodeChar(const char ch)
return 63 - (ch == '-'); return 63 - (ch == '-');
} }
char *DecodeBase64(const char *string) char *DecodeBase642(const char *string)
{ {
char *output; char *output;
size_t length = strlen(string); size_t length = strlen(string);
@ -30,10 +30,10 @@ char *DecodeBase64(const char *string)
uint32_t storage = 0; uint32_t storage = 0;
while (string[4]) while (string[4])
{ {
storage |= DecodeChar(*string++) << 18; storage |= DecodeChar2(*string++) << 18;
storage |= DecodeChar(*string++) << 12; storage |= DecodeChar2(*string++) << 12;
storage |= DecodeChar(*string++) << 6; storage |= DecodeChar2(*string++) << 6;
storage |= DecodeChar(*string++); storage |= DecodeChar2(*string++);
output[index++] = storage >> 16; output[index++] = storage >> 16;
output[index++] = (char)(storage >> 8); output[index++] = (char)(storage >> 8);
@ -42,8 +42,8 @@ char *DecodeBase64(const char *string)
storage = 0; storage = 0;
} }
storage |= DecodeChar(*string++) << 18; storage |= DecodeChar2(*string++) << 18;
storage |= DecodeChar(*string++) << 12; storage |= DecodeChar2(*string++) << 12;
output[index++] = storage >> 16; output[index++] = storage >> 16;
if (*string == '=') if (*string == '=')
@ -51,7 +51,7 @@ char *DecodeBase64(const char *string)
output[index] = '\0'; output[index] = '\0';
return output; return output;
} }
storage |= DecodeChar(*string++) << 6; storage |= DecodeChar2(*string++) << 6;
output[index++] = (char)(storage >> 8); output[index++] = (char)(storage >> 8);
if (*string == '=') if (*string == '=')
@ -59,16 +59,16 @@ char *DecodeBase64(const char *string)
output[index] = '\0'; output[index] = '\0';
return output; return output;
} }
storage |= DecodeChar(*string); storage |= DecodeChar2(*string);
output[index++] = (char)storage; output[index++] = (char)storage;
output[index] = '\0'; output[index] = '\0';
return output; return output;
} }
UserFilter::UserFilter(Bank &b) : bank(b) {} AdminFilter::AdminFilter(Bank &b) : bank(b) {}
void UserFilter::doFilter(const HttpRequestPtr &req, void AdminFilter::doFilter(const HttpRequestPtr &req,
FilterCallback &&fcb, FilterCallback &&fcb,
FilterChainCallback &&fccb) FilterChainCallback &&fccb)
{ {
@ -77,7 +77,7 @@ void UserFilter::doFilter(const HttpRequestPtr &req,
{ {
if (auth_header.substr(0, 6) == "Basic ") if (auth_header.substr(0, 6) == "Basic ")
{ {
std::stringstream ss(DecodeBase64(auth_header.substr(6).c_str())); std::stringstream ss(DecodeBase642(auth_header.substr(6).c_str()));
std::string username, password; std::string username, password;
std::getline(ss, username, ':'); std::getline(ss, username, ':');
std::getline(ss, password); std::getline(ss, password);