thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-19 14:11:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make filepath check done but with infile func, need to modify the ld path

Browse Source
This commit is contained in:
keenwolf 2021-01-16 14:55:59 +08:00
parent 2d22772455
commit 29aa0da28c
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/Functions/FunctionFile.cpp
View File

@ -6,7 +6,8 @@
#include <Poco/File.h>
#include <Poco/Path.h>
#include <Interpreters/Context.h>
#include <Storages/StorageFile.h>
#include <Common/StringUtils/StringUtils.h>
namespace DB
{
@ -20,6 +21,25 @@ namespace DB
void checkCreationIsAllowed(const Context & context_global, const std::string & db_dir_path, const std::string & table_path);
inline bool startsWith2(const std::string & s, const std::string & prefix)
{
return s.size() >= prefix.size() && 0 == memcmp(s.data(), prefix.data(), prefix.size());
}
void checkCreationIsAllowed(const Context & context_global, const std::string & db_dir_path, const std::string & table_path)
{
if (context_global.getApplicationType() != Context::ApplicationType::SERVER)
return;
/// "/dev/null" is allowed for perf testing
if (!startsWith2(table_path, db_dir_path) && table_path != "/dev/null")
throw Exception("File is not inside " + db_dir_path, 9);
Poco::File table_path_poco_file = Poco::File(table_path);
if (table_path_poco_file.exists() && table_path_poco_file.isDirectory())
throw Exception("File must not be a directory", 9);
}
/** A function to read file as a string.
*/
class FunctionFile : public IFunction