rstudio/src/cpp/core/FileUtils.cpp at master · pearsonca/rstudio · GitHub

History

220 lines (187 loc) · 5.54 KB

Raw

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

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

* FileUtils.cpp

* Unless you have received this program directly from RStudio pursuant

* to the terms of a commercial license agreement with RStudio, then

* this program is licensed to you under the terms of version 3 of the

* GNU Affero General Public License. This program is distributed WITHOUT

* ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,

* MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the

* AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.

#include <fstream>

#include <iostream>

#include <boost/bind.hpp>

#include <core/FileUtils.hpp>

#include <core/FileSerializer.hpp>

#include <shared_core/FilePath.hpp>

#include <core/StringUtils.hpp>

#include <core/system/System.hpp>

#ifndef _WIN32

#include <core/system/PosixUser.hpp>

#endif

namespace rstudio {

namespace core {

namespace file_utils {

namespace {

bool copySourceFile(const FilePath& sourceDir,

const FilePath& destDir,

const FilePath& sourceFilePath)

{

// compute the target path

std::string relativePath = sourceFilePath.getRelativePath(sourceDir);

FilePath targetPath = destDir.completePath(relativePath);

// if the copy item is a directory just create it

if (sourceFilePath.isDirectory())

{

Error error = targetPath.ensureDirectory();

if (error)

LOG_ERROR(error);

}

// otherwise copy it

else

{

Error error = sourceFilePath.copy(targetPath);

if (error)

LOG_ERROR(error);

}

return true;

}

} // anonymous namespace

FilePath uniqueFilePath(const FilePath& parent, const std::string& prefix)

{

// try up to 100 times then fallback to a uuid

for (int i=0; i<100; i++)

{

// get a shortened uuid

std::string shortentedUuid = core::system::generateShortenedUuid();

// form full path

FilePath uniqueDir = parent.completeChildPath(prefix + shortentedUuid);

// return if it doesn't exist

if (!uniqueDir.exists())

return uniqueDir;

}

// if we didn't succeed then return prefix + uuid

return parent.completeChildPath(prefix + core::system::generateUuid(false));

}

std::string readFile(const FilePath& filePath)

{

std::ifstream stream(

filePath.getAbsolutePath().c_str(),

std::ios::in | std::ios::binary);

std::string content;

if (stream)

{

stream.seekg(0, std::ios::end);

std::streamsize size = stream.tellg();

content.resize(static_cast<size_t>(size));

stream.seekg(0, std::ios::beg);

stream.read(&content[0], size);

stream.close();

}

return content;

}

#ifdef _WIN32

// test a filename to see if it corresponds to a reserved device name on

// Windows

bool isWindowsReservedName(const std::string& name)

{

const char* reserved[] = { "con", "prn", "aux", "nul", "com1", "com2",

"com3", "com4", "com5", "com6", "com7",

"com8", "com9", "lpt1", "lpt2", "lpt3",

"lpt4", "lpt5", "lpt6", "lpt7", "lpt8",

"lpt9" };

std::string lowerName = string_utils::toLower(name);

for (int i = 0; i < sizeof(reserved)/sizeof(char*); i++)

{

if (lowerName == reserved[i])

{

return true;

}

return false;

}

#endif

Error copyDirectory(const FilePath& sourceDirectory,

const FilePath& targetDirectory)

{

// create the target directory

Error error = targetDirectory.ensureDirectory();

if (error)

return error ;

// iterate over the source

return sourceDirectory.getChildrenRecursive(

boost::bind(copySourceFile, sourceDirectory, targetDirectory, _2));

}

bool isDirectoryWriteable(const FilePath& directory)

{

std::string prefix(

#ifndef _WIN32

"."

#endif

"write-test-");

FilePath testFile = directory.completePath(prefix + core::system::generateUuid());

Error error = core::writeStringToFile(testFile, "test");

if (error)

{

return false;

}

else

{

error = testFile.removeIfExists();

if (error)

LOG_ERROR(error);

return true;

}

#ifndef _WIN32

Error changeOwnership(const FilePath& file,

const std::string& owner,

bool recursive,

const FilePath::RecursiveIterationFunction& shouldChown)

{

// changes ownership of file to the server user

core::system::User user;

Error error = core::system::User::getUserFromIdentifier(owner, user);

if (error)

return error;

auto chown = [&](const FilePath& f)

{

return core::system::posixCall<int>(

boost::bind(::chown,

f.getAbsolutePath().c_str(),

user.getUserId(),

user.getGroupId()),

ERROR_LOCATION);

};

error = chown(file);

if (error)

{

error.addProperty("path", file.getAbsolutePath());

return error;

}

if (!recursive)

return Success();

// recurse into subdirectories

if (file.isDirectory())

{

file.getChildrenRecursive([&](int depth, const FilePath& child)

{

if (shouldChown && !shouldChown(depth, child))

return true;

error = chown(child);

if (error)

error.addProperty("path", child.getAbsolutePath());

// if there was an error, stop iterating

return !error;

});

}

return error;

}

#endif

} // namespace file_utils

} // namespace core

} // namespace rstudio

rstudio/src/cpp/core/FileUtils.cpp at master · pearsonca/rstudio · GitHub

Expand file tree

Breadcrumbs

FileUtils.cpp

Latest commit

History

Breadcrumbs

FileUtils.cpp

File metadata and controls