1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
*
* Copyright (C) 2008 Sun Microsystems, Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 2 of the License.
*
* 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include <config.h>
#include <drizzled/function/str/strfunc.h>
#include <drizzled/function/str/load_file.h>
#include <drizzled/error.h>
#include <drizzled/data_home.h>
#include <drizzled/session.h>
#include <drizzled/internal/my_sys.h>
#include <drizzled/sys_var.h>
#include <drizzled/system_variables.h>
#include <boost/filesystem.hpp>
#include <fcntl.h>
#include <sys/stat.h>
#include <iostream>
namespace fs=boost::filesystem;
using namespace std;
namespace drizzled
{
String *Item_load_file::val_str(String *str)
{
assert(fixed == 1);
String *file_name;
int file;
struct stat stat_info;
if (!(file_name= args[0]->val_str(str)))
{
null_value = 1;
return 0;
}
fs::path target_path(fs::system_complete(getDataHomeCatalog()));
fs::path to_file(file_name->c_ptr());
if (not to_file.has_root_directory())
{
target_path /= to_file;
}
else
{
target_path= to_file;
}
/* Read only allowed from within dir specified by secure_file_priv */
if (not secure_file_priv.string().empty())
{
fs::path secure_file_path(fs::system_complete(secure_file_priv));
if (target_path.file_string().substr(0, secure_file_path.file_string().size()) != secure_file_path.file_string())
{
/* Read only allowed from within dir specified by secure_file_priv */
my_error(ER_OPTION_PREVENTS_STATEMENT, MYF(0), "--secure-file-priv");
null_value = 1;
return 0;
}
}
if (stat(target_path.file_string().c_str(), &stat_info))
{
my_error(ER_TEXTFILE_NOT_READABLE, MYF(0), file_name->c_ptr());
goto err;
}
if (!(stat_info.st_mode & S_IROTH))
{
my_error(ER_TEXTFILE_NOT_READABLE, MYF(0), file_name->c_ptr());
goto err;
}
if (stat_info.st_size > (long) session.variables.max_allowed_packet)
{
push_warning_printf(&session, DRIZZLE_ERROR::WARN_LEVEL_WARN,
ER_WARN_ALLOWED_PACKET_OVERFLOWED,
ER(ER_WARN_ALLOWED_PACKET_OVERFLOWED),
func_name(), session.variables.max_allowed_packet);
goto err;
}
if (stat_info.st_size == 0)
{
goto err;
}
tmp_value.alloc((size_t)stat_info.st_size);
if ((file = internal::my_open(target_path.file_string().c_str(), O_RDONLY, MYF(0))) < 0)
goto err;
if (internal::my_read(file, (unsigned char*) tmp_value.ptr(), (size_t)stat_info.st_size, MYF(MY_NABP)))
{
internal::my_close(file, MYF(0));
goto err;
}
if (strlen(tmp_value.ptr()) == 0)
{
goto err;
}
tmp_value.length((size_t)stat_info.st_size);
internal::my_close(file, MYF(0));
null_value = 0;
return(&tmp_value);
err:
null_value = 1;
return 0;
}
} /* namespace drizzled */
|