197 lines
7.2 KiB
C
197 lines
7.2 KiB
C
/*
|
|
* Treebird - Lightweight frontend for Pleroma
|
|
* Copyright (C) 2022 Nekobit
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "cookie.h"
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include "env.h"
|
|
|
|
enum cookie_state
|
|
{
|
|
STATE_C_NEUTRAL,
|
|
STATE_C_STARTSP,
|
|
STATE_K_START,
|
|
STATE_V_START,
|
|
};
|
|
|
|
char* read_cookies_env(REQUEST_T req, struct cookie_values* cookies)
|
|
{
|
|
struct http_cookie_info info;
|
|
char* cookies_env = GET_ENV("HTTP_COOKIE", req);
|
|
if (!cookies_env)
|
|
return NULL;
|
|
|
|
char* cookies_str = tb_malloc(strlen(cookies_env) + 1);
|
|
if (!cookies_str)
|
|
{
|
|
perror("malloc");
|
|
return NULL;
|
|
}
|
|
strcpy(cookies_str, cookies_env);
|
|
char* cookies_read = cookies_str;
|
|
|
|
// Will loop through these
|
|
struct key_value_refs refs[] = {
|
|
{ "access_token", &(cookies->access_token), key_string },
|
|
{ "logged_in", &(cookies->logged_in), key_string },
|
|
{ "theme", &(cookies->theme), key_string },
|
|
{ "lang", &(cookies->lang), key_int },
|
|
{ "instance_url", &(cookies->instance_url), key_string },
|
|
{ "background_url", &(cookies->background_url), key_string },
|
|
{ "interact_img", &(cookies->interact_img), key_int },
|
|
{ "client_id", &(cookies->client_id), key_string },
|
|
{ "client_secret", &(cookies->client_secret), key_string },
|
|
{ "themeclr", &(cookies->themeclr), key_int },
|
|
{ "jsactions", &(cookies->jsactions), key_int },
|
|
{ "notifembed", &(cookies->notif_embed), key_int },
|
|
{ "jsreply", &(cookies->jsreply), key_int },
|
|
{ "jslive", &(cookies->jslive), key_int },
|
|
{ "js", &(cookies->js), key_int },
|
|
{ "statattachments", &(cookies->stat_attachments), key_int },
|
|
{ "statgreentexts", &(cookies->stat_greentexts), key_int },
|
|
{ "statdope", &(cookies->stat_dope), key_int },
|
|
{ "statoneclicksoftware", &(cookies->stat_oneclicksoftware), key_int },
|
|
{ "statemojolikes", &(cookies->stat_emojo_likes), key_int },
|
|
{ "sidebaropacity", &(cookies->sidebar_opacity), key_int },
|
|
{ "stathidemuted", &(cookies->stat_hide_muted), key_int },
|
|
{ "instanceshowshoutbox", &(cookies->instance_show_shoutbox), key_int },
|
|
{ "instancepanel", &(cookies->instance_panel), key_int },
|
|
};
|
|
|
|
do
|
|
{
|
|
cookies_read = parse_cookies(cookies_read, &info);
|
|
|
|
if (!(info.key && info.val)) break;
|
|
for (int i = 0; i < (sizeof(refs)/sizeof(refs[0])); ++i)
|
|
{
|
|
if (strcmp(info.key, refs[i].key) == 0)
|
|
{
|
|
refs[i].func(info.val, NULL, refs[i].val);
|
|
refs[i].val->is_set = 1;
|
|
}
|
|
}
|
|
}
|
|
while (cookies_read);
|
|
|
|
// User is responsible for freeing when done!
|
|
return cookies_str;
|
|
}
|
|
|
|
char* parse_cookies(char* begin, struct http_cookie_info* info)
|
|
{
|
|
int keydone = 0;
|
|
enum cookie_state state = STATE_C_STARTSP;
|
|
int end = 0;
|
|
size_t val_s = 0;
|
|
|
|
for (; *begin != ';' && *begin != '\0'; ++begin)
|
|
{
|
|
switch (*begin)
|
|
{
|
|
case '=':
|
|
if (state == STATE_K_START) keydone = 1;
|
|
state = STATE_C_STARTSP;
|
|
*begin = '\0'; // Null ptr
|
|
break;
|
|
case ' ':
|
|
// longkeyval = res;
|
|
// ^ ^
|
|
if (state == STATE_C_STARTSP ||
|
|
state == STATE_K_START)
|
|
break;
|
|
// fall
|
|
default:
|
|
if (state == STATE_C_STARTSP)
|
|
{
|
|
if (keydone) info->val = begin;
|
|
else info->key = begin;
|
|
state = keydone ? STATE_V_START : STATE_K_START;
|
|
}
|
|
if (keydone) ++val_s;
|
|
}
|
|
}
|
|
|
|
// Which character did we stop at?
|
|
if (*begin == '\0')
|
|
end = 1;
|
|
else if (*begin == ';')
|
|
*begin = '\0';
|
|
|
|
// The val length may be large, so strlen can waste time
|
|
info->val_len = val_s;
|
|
|
|
return end ? NULL : begin+1;
|
|
}
|
|
|
|
int cookie_get_val(char* src, char* key, struct http_cookie_info* info)
|
|
{
|
|
struct http_cookie_info read_info;
|
|
char* src_read = src;
|
|
|
|
while (1)
|
|
{
|
|
src_read = parse_cookies(src_read, &read_info);
|
|
|
|
if (!(read_info.key && read_info.val)) break;
|
|
if (strcmp(read_info.key, key) == 0)
|
|
{
|
|
info->key = read_info.key;
|
|
info->val = read_info.val;
|
|
info->val_len = read_info.val_len;
|
|
return 0;
|
|
}
|
|
|
|
if (!src_read) break;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
HV* perlify_cookies(struct cookie_values* cookies)
|
|
{
|
|
HV* ssn_cookies_hv = newHV();
|
|
|
|
hv_stores(ssn_cookies_hv, "lang", newSViv(keyint(cookies->lang)));
|
|
hv_stores(ssn_cookies_hv, "interact_img", newSViv(keyint(cookies->interact_img)));
|
|
hv_stores(ssn_cookies_hv, "themeclr", newSViv(keyint(cookies->themeclr)));
|
|
hv_stores(ssn_cookies_hv, "jsactions", newSViv(keyint(cookies->jsactions)));
|
|
hv_stores(ssn_cookies_hv, "jsreply", newSViv(keyint(cookies->jsreply)));
|
|
hv_stores(ssn_cookies_hv, "jslive", newSViv(keyint(cookies->jslive)));
|
|
hv_stores(ssn_cookies_hv, "js", newSViv(keyint(cookies->js)));
|
|
hv_stores(ssn_cookies_hv, "interact_img", newSViv(keyint(cookies->interact_img)));
|
|
hv_stores(ssn_cookies_hv, "statattachments", newSViv(keyint(cookies->stat_attachments)));
|
|
hv_stores(ssn_cookies_hv, "statgreentexts", newSViv(keyint(cookies->stat_greentexts)));
|
|
hv_stores(ssn_cookies_hv, "statdope", newSViv(keyint(cookies->stat_dope)));
|
|
hv_stores(ssn_cookies_hv, "statoneclicksoftware", newSViv(keyint(cookies->stat_oneclicksoftware)));
|
|
hv_stores(ssn_cookies_hv, "statemojolikes", newSViv(keyint(cookies->stat_emojo_likes)));
|
|
hv_stores(ssn_cookies_hv, "stathidemuted", newSViv(keyint(cookies->stat_hide_muted)));
|
|
hv_stores(ssn_cookies_hv, "instanceshowshoutbox", newSViv(keyint(cookies->instance_show_shoutbox)));
|
|
hv_stores(ssn_cookies_hv, "instancepanel", newSViv(keyint(cookies->instance_panel)));
|
|
hv_stores(ssn_cookies_hv, "notifembed", newSViv(keyint(cookies->notif_embed)));
|
|
hv_stores(ssn_cookies_hv, "access_token", newSVpv(keystr(cookies->access_token), 0));
|
|
hv_stores(ssn_cookies_hv, "logged_in", newSVpv(keystr(cookies->logged_in), 0));
|
|
hv_stores(ssn_cookies_hv, "theme", newSVpv(keystr(cookies->theme), 0));
|
|
hv_stores(ssn_cookies_hv, "instance_url", newSVpv(keystr(cookies->instance_url), 0));
|
|
hv_stores(ssn_cookies_hv, "background_url", newSVpv(keystr(cookies->background_url), 0));
|
|
hv_stores(ssn_cookies_hv, "client_id", newSVpv(keystr(cookies->client_id), 0));
|
|
hv_stores(ssn_cookies_hv, "client_secret", newSVpv(keystr(cookies->client_secret), 0));
|
|
|
|
return ssn_cookies_hv;
|
|
}
|