file.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Protocol Buffers - Google's data interchange format 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright 2008 Google Inc. All rights reserved. 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// http://code.google.com/p/protobuf/ 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Redistribution and use in source and binary forms, with or without 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// modification, are permitted provided that the following conditions are 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// met: 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions of source code must retain the above copyright 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// notice, this list of conditions and the following disclaimer. 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions in binary form must reproduce the above 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// copyright notice, this list of conditions and the following disclaimer 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// in the documentation and/or other materials provided with the 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// distribution. 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Neither the name of Google Inc. nor the names of its 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// contributors may be used to endorse or promote products derived from 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// this software without specific prior written permission. 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Author: kenton@google.com (Kenton Varda) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// emulates google3/file/base/file.cc 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/testing/file.h> 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdio.h> 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/stat.h> 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/types.h> 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _MSC_VER 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define WIN32_LEAN_AND_MEAN // yeah, right 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> // Find*File(). :( 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <io.h> 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <direct.h> 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <dirent.h> 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <unistd.h> 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <errno.h> 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace google { 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace protobuf { 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _WIN32 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define mkdir(name, mode) mkdir(name) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Windows doesn't have symbolic links. 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define lstat stat 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef F_OK 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define F_OK 00 // not defined by MSVC for whatever reason 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool File::Exists(const string& name) { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return access(name.c_str(), F_OK) == 0; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool File::ReadFileToString(const string& name, string* output) { 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char buffer[1024]; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE* file = fopen(name.c_str(), "rb"); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (file == NULL) return false; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (true) { 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t n = fread(buffer, 1, sizeof(buffer), file); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (n <= 0) break; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) output->append(buffer, n); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int error = ferror(file); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (fclose(file) != 0) return false; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return error == 0; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void File::ReadFileToStringOrDie(const string& name, string* output) { 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GOOGLE_CHECK(ReadFileToString(name, output)) << "Could not read: " << name; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void File::WriteStringToFileOrDie(const string& contents, const string& name) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE* file = fopen(name.c_str(), "wb"); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GOOGLE_CHECK(file != NULL) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "fopen(" << name << ", \"wb\"): " << strerror(errno); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GOOGLE_CHECK_EQ(fwrite(contents.data(), 1, contents.size(), file), 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) contents.size()) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "fwrite(" << name << "): " << strerror(errno); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GOOGLE_CHECK(fclose(file) == 0) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "fclose(" << name << "): " << strerror(errno); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool File::CreateDir(const string& name, int mode) { 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return mkdir(name.c_str(), mode) == 0; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool File::RecursivelyCreateDir(const string& path, int mode) { 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (CreateDir(path, mode)) return true; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (Exists(path)) return false; 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Try creating the parent. 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string::size_type slashpos = path.find_last_of('/'); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (slashpos == string::npos) { 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // No parent given. 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return RecursivelyCreateDir(path.substr(0, slashpos), mode) && 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CreateDir(path, mode); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void File::DeleteRecursively(const string& name, 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void* dummy1, void* dummy2) { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We don't care too much about error checking here since this is only used 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // in tests to delete temporary directories that are under /tmp anyway. 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _MSC_VER 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This interface is so weird. 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WIN32_FIND_DATA find_data; 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE find_handle = FindFirstFile((name + "/*").c_str(), &find_data); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (find_handle == INVALID_HANDLE_VALUE) { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Just delete it, whatever it is. 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeleteFile(name.c_str()); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveDirectory(name.c_str()); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) do { 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string entry_name = find_data.cFileName; 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (entry_name != "." && entry_name != "..") { 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string path = name + "/" + entry_name; 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (find_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeleteRecursively(path, NULL, NULL); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveDirectory(path.c_str()); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeleteFile(path.c_str()); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } while(FindNextFile(find_handle, &find_data)); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FindClose(find_handle); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RemoveDirectory(name.c_str()); 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Use opendir()! Yay! 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // lstat = Don't follow symbolic links. 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct stat stats; 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (lstat(name.c_str(), &stats) != 0) return; 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (S_ISDIR(stats.st_mode)) { 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DIR* dir = opendir(name.c_str()); 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (dir != NULL) { 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (true) { 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct dirent* entry = readdir(dir); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (entry == NULL) break; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string entry_name = entry->d_name; 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (entry_name != "." && entry_name != "..") { 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DeleteRecursively(name + "/" + entry_name, NULL, NULL); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) closedir(dir); 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) rmdir(name.c_str()); 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (S_ISREG(stats.st_mode)) { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) remove(name.c_str()); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace protobuf 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace google 177