mirror of
https://github.com/asdlokj1qpi233/subconverter.git
synced 2026-11-02 17:52:19 +00:00
Tindy X
8bbc163d08
Add support for applying renames/Emojis to a specific group. Add support for adding stream and expiry info to SSD subscriptions. Optimize codes.
998 lines
26 KiB
C++
998 lines
26 KiB
C++
#include <chrono>
|
|
//#include <regex>
|
|
#include <fstream>
|
|
#include <thread>
|
|
#include <sstream>
|
|
#include <iosfwd>
|
|
#include <iostream>
|
|
//#include <filesystem>
|
|
#include <unistd.h>
|
|
|
|
#ifdef USE_STD_REGEX
|
|
#include <regex>
|
|
#else
|
|
#include <jpcre2.hpp>
|
|
typedef jpcre2::select<char> jp;
|
|
#endif // USE_STD_REGEX
|
|
|
|
#include <rapidjson/document.h>
|
|
|
|
#ifdef USE_MBEDTLS
|
|
#include <mbedtls/md5.h>
|
|
#else
|
|
#include <openssl/md5.h>
|
|
#endif // USE_MBEDTLS
|
|
|
|
#include "misc.h"
|
|
|
|
#ifdef _WIN32
|
|
//#include <io.h>
|
|
#include <windows.h>
|
|
#include <winreg.h>
|
|
#else
|
|
#ifndef __hpux
|
|
#include <sys/select.h>
|
|
#endif /* __hpux */
|
|
#ifndef _access
|
|
#define _access access
|
|
#endif // _access
|
|
#include <sys/socket.h>
|
|
#endif // _WIN32
|
|
|
|
void sleep(int interval)
|
|
{
|
|
/*
|
|
#ifdef _WIN32
|
|
Sleep(interval);
|
|
#else
|
|
// Portable sleep for platforms other than Windows.
|
|
struct timeval wait = { 0, interval * 1000 };
|
|
select(0, NULL, NULL, NULL, &wait);
|
|
#endif
|
|
*/
|
|
//upgrade to c++11 standard
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(interval));
|
|
}
|
|
|
|
// ANSI code page (GBK on 936) to UTF8
|
|
std::string ACPToUTF8(const std::string &str_src)
|
|
{
|
|
#ifdef _WIN32
|
|
const char* strGBK = str_src.c_str();
|
|
int len = MultiByteToWideChar(CP_ACP, 0, strGBK, -1, NULL, 0);
|
|
wchar_t* wstr = new wchar_t[len + 1];
|
|
memset(wstr, 0, len + 1);
|
|
MultiByteToWideChar(CP_ACP, 0, strGBK, -1, wstr, len);
|
|
len = WideCharToMultiByte(CP_UTF8, 0, wstr, -1, NULL, 0, NULL, NULL);
|
|
char* str = new char[len + 1];
|
|
memset(str, 0, len + 1);
|
|
WideCharToMultiByte(CP_UTF8, 0, wstr, -1, str, len, NULL, NULL);
|
|
std::string strTemp = str;
|
|
if(wstr)
|
|
delete[] wstr;
|
|
if(str)
|
|
delete[] str;
|
|
return strTemp;
|
|
#else
|
|
return str_src;
|
|
#endif // _WIN32
|
|
}
|
|
|
|
// UTF8 to ANSI code page (GBK on 936)
|
|
std::string UTF8ToACP(const std::string &str_src)
|
|
{
|
|
#ifdef _WIN32
|
|
const char* strUTF8 = str_src.data();
|
|
int len = MultiByteToWideChar(CP_UTF8, 0, strUTF8, -1, NULL, 0);
|
|
wchar_t* wszGBK = new wchar_t[len + 1];
|
|
memset(wszGBK, 0, len * 2 + 2);
|
|
MultiByteToWideChar(CP_UTF8, 0, strUTF8, -1, wszGBK, len);
|
|
len = WideCharToMultiByte(CP_ACP, 0, wszGBK, -1, NULL, 0, NULL, NULL);
|
|
char* szGBK = new char[len + 1];
|
|
memset(szGBK, 0, len + 1);
|
|
WideCharToMultiByte(CP_ACP, 0, wszGBK, -1, szGBK, len, NULL, NULL);
|
|
std::string strTemp(szGBK);
|
|
if (wszGBK)
|
|
delete[] wszGBK;
|
|
if (szGBK)
|
|
delete[] szGBK;
|
|
return strTemp;
|
|
#else
|
|
return str_src;
|
|
#endif
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
// std::string to wstring
|
|
void StringToWstring(std::wstring& szDst, const std::string &str)
|
|
{
|
|
std::string temp = str;
|
|
int len = MultiByteToWideChar(CP_ACP, 0, (LPCSTR)temp.c_str(), -1, NULL,0);
|
|
wchar_t* wszUtf8 = new wchar_t[len + 1];
|
|
memset(wszUtf8, 0, len * 2 + 2);
|
|
MultiByteToWideChar(CP_ACP, 0, (LPCSTR)temp.c_str(), -1, (LPWSTR)wszUtf8, len);
|
|
szDst = wszUtf8;
|
|
//std::wstring r = wszUtf8;
|
|
delete[] wszUtf8;
|
|
}
|
|
#else
|
|
/* Unimplemented: std::codecvt_utf8 */
|
|
#endif // _WIN32
|
|
|
|
unsigned char ToHex(unsigned char x)
|
|
{
|
|
return x > 9 ? x + 55 : x + 48;
|
|
}
|
|
|
|
unsigned char FromHex(unsigned char x)
|
|
{
|
|
unsigned char y = '\0';
|
|
if (x >= 'A' && x <= 'Z')
|
|
y = x - 'A' + 10;
|
|
else if (x >= 'a' && x <= 'z')
|
|
y = x - 'a' + 10;
|
|
else if (x >= '0' && x <= '9')
|
|
y = x - '0';
|
|
else
|
|
y = x;
|
|
return y;
|
|
}
|
|
|
|
std::string UrlEncode(const std::string& str)
|
|
{
|
|
std::string strTemp = "";
|
|
string_size length = str.length();
|
|
for (string_size i = 0; i < length; i++)
|
|
{
|
|
if (isalnum((unsigned char)str[i]) ||
|
|
(str[i] == '-') ||
|
|
(str[i] == '_') ||
|
|
(str[i] == '.') ||
|
|
(str[i] == '~'))
|
|
strTemp += str[i];
|
|
else
|
|
{
|
|
strTemp += '%';
|
|
strTemp += ToHex((unsigned char)str[i] >> 4);
|
|
strTemp += ToHex((unsigned char)str[i] % 16);
|
|
}
|
|
}
|
|
return strTemp;
|
|
}
|
|
|
|
std::string UrlDecode(const std::string& str)
|
|
{
|
|
std::string strTemp;
|
|
string_size length = str.length();
|
|
for (string_size i = 0; i < length; i++)
|
|
{
|
|
if (str[i] == '+')
|
|
strTemp += ' ';
|
|
else if (str[i] == '%')
|
|
{
|
|
if(i + 2 >= length)
|
|
return strTemp;
|
|
if(isalnum(str[i + 1]) && isalnum(str[i + 2]))
|
|
{
|
|
unsigned char high = FromHex((unsigned char)str[++i]);
|
|
unsigned char low = FromHex((unsigned char)str[++i]);
|
|
strTemp += high * 16 + low;
|
|
}
|
|
else
|
|
strTemp += str[i];
|
|
}
|
|
else
|
|
strTemp += str[i];
|
|
}
|
|
return strTemp;
|
|
}
|
|
|
|
static inline bool is_base64(unsigned char c)
|
|
{
|
|
return (isalnum(c) || (c == '+') || (c == '/'));
|
|
}
|
|
|
|
std::string base64_encode(const std::string &string_to_encode)
|
|
{
|
|
char const* bytes_to_encode = string_to_encode.data();
|
|
unsigned int in_len = string_to_encode.size();
|
|
|
|
std::string ret;
|
|
int i = 0;
|
|
int j = 0;
|
|
unsigned char char_array_3[3];
|
|
unsigned char char_array_4[4];
|
|
|
|
while (in_len--)
|
|
{
|
|
char_array_3[i++] = *(bytes_to_encode++);
|
|
if (i == 3)
|
|
{
|
|
char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
|
|
char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
|
|
char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
|
|
char_array_4[3] = char_array_3[2] & 0x3f;
|
|
|
|
for(i = 0; (i <4) ; i++)
|
|
ret += base64_chars[char_array_4[i]];
|
|
i = 0;
|
|
}
|
|
}
|
|
|
|
if (i)
|
|
{
|
|
for(j = i; j < 3; j++)
|
|
char_array_3[j] = '\0';
|
|
|
|
char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
|
|
char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
|
|
char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
|
|
char_array_4[3] = char_array_3[2] & 0x3f;
|
|
|
|
for (j = 0; (j < i + 1); j++)
|
|
ret += base64_chars[char_array_4[j]];
|
|
|
|
while((i++ < 3))
|
|
ret += '=';
|
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
std::string base64_decode(const std::string &encoded_string, bool accept_urlsafe)
|
|
{
|
|
string_size in_len = encoded_string.size();
|
|
string_size i = 0;
|
|
string_size in_ = 0;
|
|
unsigned char char_array_4[4], char_array_3[3], uchar = 0;
|
|
static unsigned char dtable[256], itable[256], table_ready = 0;
|
|
std::string ret;
|
|
|
|
// Should not need thread_local with the flag...
|
|
if (!table_ready)
|
|
{
|
|
// No memset needed for static/TLS
|
|
for (string_size k = 0; k < base64_chars.length(); k++)
|
|
{
|
|
uchar = base64_chars[k]; // make compiler happy
|
|
dtable[uchar] = k; // decode (find)
|
|
itable[uchar] = 1; // is_base64
|
|
}
|
|
// Add urlsafe table
|
|
dtable['-'] = dtable['+']; itable['-'] = 2;
|
|
dtable['_'] = dtable['/']; itable['_'] = 2;
|
|
table_ready = 1;
|
|
}
|
|
|
|
while (in_len-- && (encoded_string[in_] != '='))
|
|
{
|
|
uchar = encoded_string[in_]; // make compiler happy
|
|
if (!(accept_urlsafe ? itable[uchar] : (itable[uchar] == 1))) // break away from the while condition
|
|
continue;
|
|
char_array_4[i++] = uchar;
|
|
in_++;
|
|
if (i == 4)
|
|
{
|
|
for (string_size j = 0; j < 4; j++)
|
|
char_array_4[j] = dtable[char_array_4[j]];
|
|
|
|
char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
|
|
char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
|
|
char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
|
|
|
|
for (i = 0; (i < 3); i++)
|
|
ret += char_array_3[i];
|
|
i = 0;
|
|
}
|
|
}
|
|
|
|
if (i)
|
|
{
|
|
for (string_size j = i; j <4; j++)
|
|
char_array_4[j] = 0;
|
|
|
|
for (string_size j = 0; j <4; j++)
|
|
char_array_4[j] = dtable[char_array_4[j]];
|
|
|
|
char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
|
|
char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
|
|
char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
|
|
|
|
for (string_size j = 0; (j < i - 1); j++)
|
|
ret += char_array_3[j];
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
std::vector<std::string> split(const std::string &s, const std::string &seperator)
|
|
{
|
|
std::vector<std::string> result;
|
|
typedef std::string::size_type string_size;
|
|
string_size i = 0;
|
|
|
|
while(i != s.size())
|
|
{
|
|
int flag = 0;
|
|
while(i != s.size() && flag == 0)
|
|
{
|
|
flag = 1;
|
|
for(string_size x = 0; x < seperator.size(); ++x)
|
|
if(s[i] == seperator[x])
|
|
{
|
|
++i;
|
|
flag = 0;
|
|
break;
|
|
}
|
|
}
|
|
|
|
flag = 0;
|
|
string_size j = i;
|
|
while(j != s.size() && flag == 0)
|
|
{
|
|
for(string_size x = 0; x < seperator.size(); ++x)
|
|
if(s[j] == seperator[x])
|
|
{
|
|
flag = 1;
|
|
break;
|
|
}
|
|
if(flag == 0)
|
|
++j;
|
|
}
|
|
if(i != j)
|
|
{
|
|
result.push_back(s.substr(i, j-i));
|
|
i = j;
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
std::string GetEnv(const std::string &name)
|
|
{
|
|
std::string retVal;
|
|
#ifdef _WIN32
|
|
char chrData[1024] = {};
|
|
if(GetEnvironmentVariable(name.c_str(), chrData, 1023))
|
|
retVal.assign(chrData);
|
|
#else
|
|
char *env = getenv(name.c_str());
|
|
if(env != NULL)
|
|
retVal.assign(env);
|
|
#endif // _WIN32
|
|
return retVal;
|
|
}
|
|
|
|
std::string getSystemProxy()
|
|
{
|
|
#ifdef _WIN32
|
|
HKEY key;
|
|
auto ret = RegOpenKeyEx(HKEY_CURRENT_USER, R"(Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings)", 0, KEY_ALL_ACCESS, &key);
|
|
if(ret != ERROR_SUCCESS)
|
|
{
|
|
//std::cout << "open failed: " << ret << std::endl;
|
|
return std::string();
|
|
}
|
|
|
|
DWORD values_count, max_value_name_len, max_value_len;
|
|
ret = RegQueryInfoKey(key, NULL, NULL, NULL, NULL, NULL, NULL,
|
|
&values_count, &max_value_name_len, &max_value_len, NULL, NULL);
|
|
if(ret != ERROR_SUCCESS)
|
|
{
|
|
//std::cout << "query failed" << std::endl;
|
|
return std::string();
|
|
}
|
|
|
|
std::vector<std::tuple<std::shared_ptr<char>, DWORD, std::shared_ptr<BYTE>>> values;
|
|
for(DWORD i = 0; i < values_count; i++)
|
|
{
|
|
std::shared_ptr<char> value_name(new char[max_value_name_len + 1],
|
|
std::default_delete<char[]>());
|
|
DWORD value_name_len = max_value_name_len + 1;
|
|
DWORD value_type, value_len;
|
|
RegEnumValue(key, i, value_name.get(), &value_name_len, NULL, &value_type, NULL, &value_len);
|
|
std::shared_ptr<BYTE> value(new BYTE[value_len],
|
|
std::default_delete<BYTE[]>());
|
|
value_name_len = max_value_name_len + 1;
|
|
RegEnumValue(key, i, value_name.get(), &value_name_len, NULL, &value_type, value.get(), &value_len);
|
|
values.push_back(std::make_tuple(value_name, value_type, value));
|
|
}
|
|
|
|
DWORD ProxyEnable = 0;
|
|
for (auto x : values)
|
|
{
|
|
if (strcmp(std::get<0>(x).get(), "ProxyEnable") == 0)
|
|
{
|
|
ProxyEnable = *(DWORD*)(std::get<2>(x).get());
|
|
}
|
|
}
|
|
|
|
if (ProxyEnable)
|
|
{
|
|
for (auto x : values)
|
|
{
|
|
if (strcmp(std::get<0>(x).get(), "ProxyServer") == 0)
|
|
{
|
|
//std::cout << "ProxyServer: " << (char*)(std::get<2>(x).get()) << std::endl;
|
|
return std::string((char*)(std::get<2>(x).get()));
|
|
}
|
|
}
|
|
}
|
|
/*
|
|
else {
|
|
//std::cout << "Proxy not Enabled" << std::endl;
|
|
}
|
|
*/
|
|
//return 0;
|
|
return std::string();
|
|
#else
|
|
char* proxy = getenv("ALL_PROXY");
|
|
if(proxy != NULL)
|
|
return std::string(proxy);
|
|
else
|
|
return std::string();
|
|
#endif // _WIN32
|
|
}
|
|
|
|
std::string trim(const std::string& str)
|
|
{
|
|
std::string::size_type pos = str.find_first_not_of(' ');
|
|
if (pos == std::string::npos)
|
|
{
|
|
return str;
|
|
}
|
|
std::string::size_type pos2 = str.find_last_not_of(' ');
|
|
if (pos2 != std::string::npos)
|
|
{
|
|
return str.substr(pos, pos2 - pos + 1);
|
|
}
|
|
return str.substr(pos);
|
|
}
|
|
|
|
std::string getUrlArg(const std::string &url, const std::string &request)
|
|
{
|
|
//std::smatch result;
|
|
/*
|
|
if (regex_search(url.cbegin(), url.cend(), result, std::regex(request + "=(.*?)&")))
|
|
{
|
|
return result[1];
|
|
}
|
|
else if (regex_search(url.cbegin(), url.cend(), result, std::regex(request + "=(.*)")))
|
|
{
|
|
return result[1];
|
|
}
|
|
else
|
|
{
|
|
return std::string();
|
|
}
|
|
*/
|
|
/*
|
|
std::string::size_type spos = url.find("?");
|
|
if(spos != url.npos)
|
|
url.erase(0, spos + 1);
|
|
|
|
string_array vArray, arglist = split(url, "&");
|
|
for(std::string &x : arglist)
|
|
{
|
|
std::string::size_type epos = x.find("=");
|
|
if(epos != x.npos)
|
|
{
|
|
if(x.substr(0, epos) == request)
|
|
return x.substr(epos + 1);
|
|
}
|
|
}
|
|
*/
|
|
std::string pattern = request + "=";
|
|
std::string::size_type pos = url.rfind(pattern);
|
|
if(pos != url.npos)
|
|
{
|
|
pos += pattern.size();
|
|
return url.substr(pos, url.find("&", pos) - pos);
|
|
}
|
|
return std::string();
|
|
}
|
|
|
|
std::string replace_all_distinct(std::string str, const std::string &old_value, const std::string &new_value)
|
|
{
|
|
for(std::string::size_type pos(0); pos != std::string::npos; pos += new_value.length())
|
|
{
|
|
if((pos = str.find(old_value, pos)) != std::string::npos)
|
|
str.replace(pos, old_value.length(), new_value);
|
|
else
|
|
break;
|
|
}
|
|
return str;
|
|
}
|
|
|
|
#ifdef USE_STD_REGEX
|
|
bool regValid(const std::string ®)
|
|
{
|
|
try
|
|
{
|
|
std::regex r(reg, std::regex::ECMAScript);
|
|
return true;
|
|
}
|
|
catch (std::regex_error &e)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool regFind(const std::string &src, const std::string &target)
|
|
{
|
|
try
|
|
{
|
|
std::regex::flag_type flags = std::regex::extended | std::regex::ECMAScript;
|
|
if(target.find("(?i)") == 0)
|
|
{
|
|
target.erase(0, 4);
|
|
flags |= std::regex::icase;
|
|
}
|
|
std::regex reg(target, flags);
|
|
return regex_search(src, reg);
|
|
}
|
|
catch (std::regex_error &e)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
std::string regReplace(const std::string &src, const std::string &match, const std::string &rep)
|
|
{
|
|
std::string result = "";
|
|
try
|
|
{
|
|
std::regex::flag_type flags = std::regex::extended | std::regex::ECMAScript;
|
|
if(match.find("(?i)") == 0)
|
|
{
|
|
match.erase(0, 4);
|
|
flags |= std::regex::icase;
|
|
}
|
|
std::regex reg(match, flags);
|
|
regex_replace(back_inserter(result), src.begin(), src.end(), reg, rep);
|
|
}
|
|
catch (std::regex_error &e)
|
|
{
|
|
result = src;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool regMatch(const std::string &src, const std::string &match)
|
|
{
|
|
try
|
|
{
|
|
std::regex::flag_type flags = std::regex::extended | std::regex::ECMAScript;
|
|
if(match.find("(?i)") == 0)
|
|
{
|
|
match.erase(0, 4);
|
|
flags |= std::regex::icase;
|
|
}
|
|
std::regex reg(match, flags);
|
|
return regex_match(src, reg);
|
|
}
|
|
catch (std::regex_error &e)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
#else
|
|
|
|
bool regMatch(const std::string &src, const std::string &target)
|
|
{
|
|
jp::Regex reg;
|
|
reg.setPattern(target).addModifier("gm").addPcre2Option(PCRE2_ANCHORED|PCRE2_ENDANCHORED).compile();
|
|
if(!reg)
|
|
return false;
|
|
return reg.match(src);
|
|
}
|
|
|
|
bool regFind(const std::string &src, const std::string &target)
|
|
{
|
|
jp::Regex reg;
|
|
reg.setPattern(target).addModifier("gm").compile();
|
|
if(!reg)
|
|
return false;
|
|
return reg.match(src);
|
|
}
|
|
|
|
std::string regReplace(const std::string &src, const std::string &target, const std::string &rep)
|
|
{
|
|
jp::Regex reg;
|
|
reg.setPattern(target).addModifier("gm").compile();
|
|
if(!reg)
|
|
return src;
|
|
return reg.replace(src, rep);
|
|
}
|
|
|
|
bool regValid(const std::string &target)
|
|
{
|
|
jp::Regex reg(target);
|
|
return !!reg;
|
|
}
|
|
#endif // USE_STD_REGEX
|
|
|
|
std::string speedCalc(double speed)
|
|
{
|
|
if(speed == 0.0)
|
|
return std::string("0.00B");
|
|
char str[10];
|
|
std::string retstr;
|
|
if(speed >= 1073741824.0)
|
|
sprintf(str, "%.2fGB", speed / 1073741824.0);
|
|
else if(speed >= 1048576.0)
|
|
sprintf(str, "%.2fMB", speed / 1048576.0);
|
|
else if(speed >= 1024.0)
|
|
sprintf(str, "%.2fKB", speed / 1024.0);
|
|
else
|
|
sprintf(str, "%.2fB", speed);
|
|
retstr = str;
|
|
return retstr;
|
|
}
|
|
|
|
std::string urlsafe_base64_reverse(const std::string &encoded_string)
|
|
{
|
|
return replace_all_distinct(replace_all_distinct(encoded_string, "-", "+"), "_", "/");
|
|
}
|
|
|
|
std::string urlsafe_base64(const std::string &encoded_string)
|
|
{
|
|
return replace_all_distinct(replace_all_distinct(replace_all_distinct(encoded_string, "+", "-"), "/", "_"), "=", "");
|
|
}
|
|
|
|
std::string urlsafe_base64_decode(const std::string &encoded_string)
|
|
{
|
|
return base64_decode(encoded_string, true);
|
|
}
|
|
|
|
std::string urlsafe_base64_encode(const std::string &string_to_encode)
|
|
{
|
|
return urlsafe_base64(base64_encode(string_to_encode));
|
|
}
|
|
|
|
std::string getMD5(const std::string &data)
|
|
{
|
|
std::string result;
|
|
unsigned int i = 0;
|
|
unsigned char digest[16] = {};
|
|
|
|
#ifdef USE_MBEDTLS
|
|
mbedtls_md5_context ctx;
|
|
|
|
mbedtls_md5_init(&ctx);
|
|
mbedtls_md5_starts_ret(&ctx);
|
|
mbedtls_md5_update_ret(&ctx, reinterpret_cast<const unsigned char*>(data.data()), data.size());
|
|
mbedtls_md5_finish_ret(&ctx, reinterpret_cast<unsigned char*>(&digest));
|
|
mbedtls_md5_free(&ctx);
|
|
#else
|
|
MD5_CTX ctx;
|
|
|
|
MD5_Init(&ctx);
|
|
MD5_Update(&ctx, data.data(), data.size());
|
|
MD5_Final((unsigned char *)&digest, &ctx);
|
|
#endif // USE_MBEDTLS
|
|
|
|
char tmp[3] = {};
|
|
for(i = 0; i < 16; i++)
|
|
{
|
|
snprintf(tmp, 3, "%02x", digest[i]);
|
|
result += tmp;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
// TODO: Add preprocessor option to disable (open web service safety)
|
|
std::string fileGet(const std::string &path, bool binary, bool scope_limit)
|
|
{
|
|
std::ifstream infile;
|
|
std::stringstream strstrm;
|
|
std::ios::openmode mode = binary ? std::ios::binary : std::ios::in;
|
|
|
|
if(scope_limit)
|
|
{
|
|
#ifdef _WIN32
|
|
if(path.find(":\\") != path.npos || path.find("..") != path.npos)
|
|
return std::string();
|
|
#else
|
|
if(path.find("/") == 0 || path.find("..") != path.npos)
|
|
return std::string();
|
|
#endif // _WIN32
|
|
}
|
|
|
|
infile.open(path, mode);
|
|
if(infile)
|
|
{
|
|
strstrm<<infile.rdbuf();
|
|
infile.close();
|
|
return strstrm.str();
|
|
}
|
|
return std::string();
|
|
}
|
|
|
|
bool fileExist(const std::string &path)
|
|
{
|
|
//using c++17 standard, but may cause problem on clang
|
|
//return std::filesystem::exists(path);
|
|
return _access(path.data(), 4) != -1;
|
|
}
|
|
|
|
bool fileCopy(const std::string &source, const std::string &dest)
|
|
{
|
|
std::ifstream infile;
|
|
std::ofstream outfile;
|
|
infile.open(source, std::ios::binary);
|
|
if(!infile)
|
|
return false;
|
|
outfile.open(dest, std::ios::binary);
|
|
if(!outfile)
|
|
return false;
|
|
try
|
|
{
|
|
outfile<<infile.rdbuf();
|
|
}
|
|
catch (std::exception &e)
|
|
{
|
|
return false;
|
|
}
|
|
infile.close();
|
|
outfile.close();
|
|
return true;
|
|
}
|
|
|
|
std::string fileToBase64(const std::string &filepath)
|
|
{
|
|
return base64_encode(fileGet(filepath, true));
|
|
}
|
|
|
|
std::string fileGetMD5(const std::string &filepath)
|
|
{
|
|
return getMD5(fileGet(filepath, true));
|
|
}
|
|
|
|
int fileWrite(const std::string &path, const std::string &content, bool overwrite)
|
|
{
|
|
std::fstream outfile;
|
|
std::ios::openmode mode = overwrite ? std::ios::out : std::ios::app;
|
|
outfile.open(path, mode);
|
|
outfile << content << std::endl;
|
|
outfile.close();
|
|
return 0;
|
|
}
|
|
|
|
bool isIPv4(const std::string &address)
|
|
{
|
|
return regMatch(address, "^(25[0-5]|2[0-4]\\d|[0-1]?\\d?\\d)(\\.(25[0-5]|2[0-4]\\d|[0-1]?\\d?\\d)){3}$");
|
|
}
|
|
|
|
bool isIPv6(const std::string &address)
|
|
{
|
|
std::vector<std::string> regLists = {"^(?:[0-9a-fA-F]{1,4}:){7}[0-9a-fA-F]{1,4}$", "^((?:[0-9A-Fa-f]{1,4}(:[0-9A-Fa-f]{1,4})*)?)::((?:([0-9A-Fa-f]{1,4}:)*[0-9A-Fa-f]{1,4})?)$", "^(::(?:[0-9A-Fa-f]{1,4})(?::[0-9A-Fa-f]{1,4}){5})|((?:[0-9A-Fa-f]{1,4})(?::[0-9A-Fa-f]{1,4}){5}::)$"};
|
|
for(unsigned int i = 0; i < regLists.size(); i++)
|
|
{
|
|
if(regMatch(address, regLists[i]))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::string rand_str(const int len)
|
|
{
|
|
std::string retData;
|
|
srand(time(NULL));
|
|
int cnt = 0;
|
|
while(cnt < len)
|
|
{
|
|
switch((rand() % 3))
|
|
{
|
|
case 1:
|
|
retData += ('A' + rand() % 26);
|
|
break;
|
|
case 2:
|
|
retData += ('a' + rand() % 26);
|
|
break;
|
|
default:
|
|
retData += ('0' + rand() % 10);
|
|
break;
|
|
}
|
|
cnt++;
|
|
}
|
|
return retData;
|
|
}
|
|
|
|
void urlParse(std::string &url, std::string &host, std::string &path, int &port, bool &isTLS)
|
|
{
|
|
std::vector<std::string> args;
|
|
|
|
if(regMatch(url, "^https://(.*)"))
|
|
isTLS = true;
|
|
url = regReplace(url, "^(http|https)://", "");
|
|
if(url.find("/") == url.npos)
|
|
{
|
|
host = url;
|
|
path = "/";
|
|
}
|
|
else
|
|
{
|
|
host = url.substr(0, url.find("/"));
|
|
path = url.substr(url.find("/"));
|
|
}
|
|
if(regFind(host, "\\[(.*)\\]")) //IPv6
|
|
{
|
|
args = split(regReplace(host, "\\[(.*)\\](.*)", "$1,$2"), ",");
|
|
if(args.size() == 2) //with port
|
|
port = to_int(args[1].substr(1));
|
|
host = args[0];
|
|
}
|
|
else if(strFind(host, ":"))
|
|
{
|
|
port = to_int(host.substr(host.rfind(":") + 1));
|
|
host = host.substr(0, host.rfind(":"));
|
|
}
|
|
if(port == 0)
|
|
{
|
|
if(isTLS)
|
|
port = 443;
|
|
else
|
|
port = 80;
|
|
}
|
|
}
|
|
|
|
bool is_str_utf8(const std::string &data)
|
|
{
|
|
const char *str = data.c_str();
|
|
unsigned int nBytes = 0;
|
|
unsigned char chr;
|
|
for (unsigned int i = 0; str[i] != '\0'; ++i)
|
|
{
|
|
chr = *(str + i);
|
|
if (nBytes == 0)
|
|
{
|
|
if (chr >= 0x80)
|
|
{
|
|
if (chr >= 0xFC && chr <= 0xFD)
|
|
nBytes = 6;
|
|
else if (chr >= 0xF8)
|
|
nBytes = 5;
|
|
else if (chr >= 0xF0)
|
|
nBytes = 4;
|
|
else if (chr >= 0xE0)
|
|
nBytes = 3;
|
|
else if (chr >= 0xC0)
|
|
nBytes = 2;
|
|
else
|
|
return false;
|
|
nBytes--;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ((chr & 0xC0) != 0x80)
|
|
return false;
|
|
nBytes--;
|
|
}
|
|
}
|
|
if (nBytes != 0)
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
void removeUTF8BOM(std::string &data)
|
|
{
|
|
if(data.compare(0, 3, "\xEF\xBB\xBF") == 0)
|
|
data = data.substr(3);
|
|
}
|
|
|
|
int shortAssemble(unsigned short num_a, unsigned short num_b)
|
|
{
|
|
return (int)num_b << 16 | num_a;
|
|
}
|
|
|
|
void shortDisassemble(int source, unsigned short &num_a, unsigned short &num_b)
|
|
{
|
|
num_a = (unsigned short)source;
|
|
num_b = (unsigned short)(source >> 16);
|
|
}
|
|
|
|
int to_int(const std::string &str, int def_value)
|
|
{
|
|
int retval = 0;
|
|
char c;
|
|
std::stringstream ss(str);
|
|
if(!(ss >> retval))
|
|
return def_value;
|
|
else if(ss >> c)
|
|
return def_value;
|
|
else
|
|
return retval;
|
|
}
|
|
|
|
std::string getFormData(const std::string &raw_data)
|
|
{
|
|
std::stringstream strstrm;
|
|
std::string line;
|
|
|
|
std::string boundary;
|
|
std::string file; /* actual file content */
|
|
|
|
int i = 0;
|
|
|
|
strstrm<<raw_data;
|
|
|
|
while (std::getline(strstrm, line))
|
|
{
|
|
if(i == 0)
|
|
boundary = line.substr(0, line.length() - 1); // Get boundary
|
|
else if(line.find(boundary) == 0)
|
|
break; // The end
|
|
else if(line.length() == 1)
|
|
{
|
|
// Time to get raw data
|
|
char c;
|
|
int bl = boundary.length();
|
|
bool endfile = false;
|
|
char buffer[256];
|
|
while(!endfile)
|
|
{
|
|
int j = 0;
|
|
int k;
|
|
while(j < 256 && strstrm.get(c) && !endfile)
|
|
{
|
|
buffer[j] = c;
|
|
k = 0;
|
|
// Verify if we are at the end
|
|
while(boundary[bl - 1 - k] == buffer[j - k])
|
|
{
|
|
if(k >= bl - 1)
|
|
{
|
|
// We are at the end of the file
|
|
endfile = true;
|
|
break;
|
|
}
|
|
k++;
|
|
}
|
|
j++;
|
|
}
|
|
file.append(buffer, j);
|
|
j = 0;
|
|
};
|
|
break;
|
|
}
|
|
i++;
|
|
}
|
|
return file;
|
|
}
|
|
|
|
std::string UTF8ToCodePoint(const std::string &data)
|
|
{
|
|
std::stringstream ss;
|
|
int charcode = 0;
|
|
for(std::string::size_type i = 0; i < data.size(); i++)
|
|
{
|
|
charcode = data[i] & 0xff;
|
|
if((charcode >> 7) == 0)
|
|
{
|
|
ss<<data[i];
|
|
}
|
|
else if((charcode >> 5) == 6)
|
|
{
|
|
ss<<"\\u"<<std::hex<<((data[i + 1] & 0x3f) | (data[i] & 0x1f) << 6);
|
|
i++;
|
|
}
|
|
else if((charcode >> 4) == 14)
|
|
{
|
|
ss<<"\\u"<<std::hex<<((data[i + 2] & 0x3f) | (data[i + 1] & 0x3f) << 6 | (data[i] & 0xf) << 12);
|
|
i += 2;
|
|
}
|
|
else if((charcode >> 3) == 30)
|
|
{
|
|
ss<<"\\u"<<std::hex<<((data[i + 3] & 0x3f) | (data[i + 2] & 0x3f) << 6 | (data[i + 1] & 0x3f) << 12 | (data[i] & 0x7) << 18);
|
|
i += 3;
|
|
}
|
|
}
|
|
return ss.str();
|
|
}
|