#include #include #include "http/httplib.h" #include "server/http_server.h" #include "clients/gdrive.h" #include "config.h" #include "fs.h" #include "windows.h" #include "lang.h" #include "system.h" #define SERVER_CERT_FILE "/app0/assets/certs/domain.crt" #define SERVER_PRIVATE_KEY_FILE "/app0/assets/certs/domain.key" #define SERVER_PRIVATE_KEY_PASSWORD "12345678" using namespace httplib; SSLServer *svr; int http_server_port = 8080; namespace HttpServer { std::string dump_headers(const Headers &headers) { std::string s; char buf[BUFSIZ]; for (auto it = headers.begin(); it != headers.end(); ++it) { const auto &x = *it; snprintf(buf, sizeof(buf), "%s: %s\n", x.first.c_str(), x.second.c_str()); s += buf; } return s; } std::string log(const Request &req, const Response &res) { std::string s; char buf[BUFSIZ]; s += "================================\n"; snprintf(buf, sizeof(buf), "%s %s %s", req.method.c_str(), req.version.c_str(), req.path.c_str()); s += buf; std::string query; for (auto it = req.params.begin(); it != req.params.end(); ++it) { const auto &x = *it; snprintf(buf, sizeof(buf), "%c%s=%s", (it == req.params.begin()) ? '?' : '&', x.first.c_str(), x.second.c_str()); query += buf; } snprintf(buf, sizeof(buf), "%s\n", query.c_str()); s += buf; s += dump_headers(req.headers); s += "--------------------------------\n"; snprintf(buf, sizeof(buf), "%d %s\n", res.status, res.version.c_str()); s += buf; s += dump_headers(res.headers); s += "\n"; if (!res.body.empty()) { s += res.body; } s += "\n"; return s; } void *ServerThread(void *argp) { svr->Get("/", [&](const Request & req, Response & res) { res.set_redirect("/mnt/sandbox/pfsmnt/RMTC00001-app0/assets/index.html"); }); svr->Post("/__local__/list", [&](const Request & req, Response & res) { const char *path; bool onlyFolders = false; json_object *jobj = json_tokener_parse(req.body.c_str()); if (jobj != nullptr) { path = json_object_get_string(json_object_object_get(jobj, "path")); const char *onlyFolders_text = json_object_get_string(json_object_object_get(jobj, "onlyFolders")); if (onlyFolders_text != nullptr && strcasecmp(onlyFolders_text, "true")==0) onlyFolders = true; if (path == nullptr) { res.status = 400; res.set_content("path parameter is missing", 25, "text/plain"); return; } } int err; std::vector files = FS::ListDir(path, &err); json_object *json_files = json_object_new_array(); for (std::vector::iterator it = files.begin(); it != files.end();) { if (((onlyFolders && it->isDir) || !onlyFolders) && strcmp(it->name, "..") != 0) { json_object *new_file = json_object_new_object(); json_object_object_add(new_file, "name", json_object_new_string(it->name)); json_object_object_add(new_file, "rights", json_object_new_string(it->isDir ? "drwxrwxrwx" : "rw-rw-rw-")); json_object_object_add(new_file, "date", json_object_new_string(it->display_date)); json_object_object_add(new_file, "size", json_object_new_string(it->isDir ? "" : it->display_size)); json_object_object_add(new_file, "type", json_object_new_string(it->isDir ? "dir" : "file")); json_object_array_add(json_files, new_file); } it++; } json_object *results = json_object_new_object(); json_object_object_add(results, "result", json_files); const char *results_str = json_object_to_json_string(results); res.status = 200; res.set_content(results_str, strlen(results_str), "application/json"); }); svr->Get("/google_auth", [](const Request &req, Response &res) { std::string auth_code = req.get_param_value("code"); Client client(GOOGLE_OAUTH_HOST); client.set_follow_location(true); client.enable_server_certificate_verification(false); std::string url = std::string("/token"); std::string post_data = std::string("code=") + auth_code + "&client_id=" + gg_app.client_id + "&client_secret=" + gg_app.client_secret + "&redirect_uri=https%3A//localhost%3A" + std::to_string(http_server_port) + "/google_auth" "&grant_type=authorization_code"; if (auto result = client.Post(url, post_data.c_str(), post_data.length(), "application/x-www-form-urlencoded")) { if (HTTP_SUCCESS(result->status)) { json_object *jobj = json_tokener_parse(result.value().body.c_str()); enum json_type type; json_object_object_foreach(jobj, key, val) { if (strcmp(key, "access_token")==0) snprintf(remote_settings->gg_account.access_token, 255, "%s", json_object_get_string(val)); else if (strcmp(key, "refresh_token")==0) snprintf(remote_settings->gg_account.refresh_token, 255, "%s", json_object_get_string(val)); else if (strcmp(key, "expires_in")==0) { OrbisTick tick; sceRtcGetCurrentTick(&tick); remote_settings->gg_account.token_expiry = tick.mytick + (json_object_get_uint64(val)*1000000); } } CONFIG::SaveConfig(); login_state = 1; res.set_content(lang_strings[STR_GET_TOKEN_SUCCESS_MSG], "text/plain"); return; } else { login_state = -1; std::string str = std::string(lang_strings[STR_FAIL_GET_TOKEN_MSG]) + " Google"; res.set_content(str.c_str(), "text/plain"); } } login_state = -1; std::string str = std::string(lang_strings[STR_FAIL_GET_TOKEN_MSG]) + " Google"; res.set_content(str.c_str(), "text/plain"); }); svr->Get("/stop", [&](const Request & /*req*/, Response & /*res*/) { svr->stop(); }); svr->set_error_handler([](const Request & /*req*/, Response &res) { const char *fmt = "

Error Status: %d

"; char buf[BUFSIZ]; snprintf(buf, sizeof(buf), fmt, res.status); res.set_content(buf, "text/html"); }); /* svr->set_logger([](const Request &req, const Response &res) { dbglogger_log("%s", log(req, res).c_str()); }); */ svr->set_mount_point("/", "/"); svr->listen("0.0.0.0", http_server_port); return NULL; } void Start() { if (svr == nullptr) svr = new SSLServer(SERVER_CERT_FILE, SERVER_PRIVATE_KEY_FILE, nullptr, nullptr, SERVER_PRIVATE_KEY_PASSWORD); if (!svr->is_valid()) { return; } int ret = pthread_create(&http_server_thid, NULL, ServerThread, NULL); } void Stop() { if (svr != nullptr) svr->stop(); } }