Utils.cpp revision cdbf6fe94f1907e29aa816471c9d093375ccba2f
15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/* 25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * Copyright (C) 2016 The Android Open Source Project 35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * 45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * Licensed under the Apache License, Version 2.0 (the "License"); 50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner * you may not use this file except in compliance with the License. 60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner * You may obtain a copy of the License at 75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * 85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * http://www.apache.org/licenses/LICENSE-2.0 9010bfc253050626435f006386c0bee4b55329687James Dennett * 10010bfc253050626435f006386c0bee4b55329687James Dennett * Unless required by applicable law or agreed to in writing, software 11010bfc253050626435f006386c0bee4b55329687James Dennett * distributed under the License is distributed on an "AS IS" BASIS, 12010bfc253050626435f006386c0bee4b55329687James Dennett * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * See the License for the specific language governing permissions and 145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer * limitations under the License. 155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer */ 165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "Utils.h" 18478851c3ed6bd784e7377dffd8e57b200c1b9ba9Benjamin Kramer 194967a710c84587c654b56c828382219c3937dacbPirama Arumuga Nainar#include <err.h> 205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include <fts.h> 210e2c34f92f00628d48968dfea096d36381f494cbStephen Hines#include <string.h> 22d5532b6cfff2977e0c59fa6ead7f7973984a620dJohn McCall#include <unistd.h> 2330a2e16f6c27f888dd11eba6bbbae1e980078fcbChandler Carruth 2401d08018b7cf5ce1601707cfd7a84d22015fc04eDouglas Gregor#include <sstream> 2501d08018b7cf5ce1601707cfd7a84d22015fc04eDouglas Gregor#include <string> 26aa49a7d70e58dac2aeb40664ba16d2ea571b8c95Daniel Dunbar#include <vector> 275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include <android-base/strings.h> 295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 30aa165f8458b51c546bebff947343e1a36f3594cbDouglas Gregor#include "DeclarationDatabase.h" 3176da55d3a49e1805f51b1ced7c5da5bcd7f759d8John McCall 32aa165f8458b51c546bebff947343e1a36f3594cbDouglas Gregorstd::string getWorkingDir() { 332fa67efeaf66a9332c30a026dc1c21bef6c33a6cBenjamin Kramer char buf[PATH_MAX]; 344c5d320a7581f4b80b151630c91cea5727fa9923Sebastian Redl if (!getcwd(buf, sizeof(buf))) { 351060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek err(1, "getcwd failed"); 361060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek } 371060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek return buf; 381060aff23f72135f8b50034a1e80f16725ebc56cTed Kremenek} 393fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor 403fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregorstd::vector<std::string> collectHeaders(const std::string& directory) { 413fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor std::vector<std::string> headers; 423fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor 433fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor char* dir_argv[2] = { const_cast<char*>(directory.c_str()), nullptr }; 443fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor std::unique_ptr<FTS, decltype(&fts_close)> fts( 453fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor fts_open(dir_argv, FTS_LOGICAL | FTS_NOCHDIR, nullptr), fts_close); 463fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor 473fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor if (!fts) { 483fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor err(1, "failed to open directory '%s'", directory.c_str()); 493fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor } 503fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor 513fd95ce225393fe4a3623e429766a8c3f487ff9dDouglas Gregor while (FTSENT* ent = fts_read(fts.get())) { 52b4609806e9232593ece09ce08b630836e825865cDouglas Gregor if (ent->fts_info & (FTS_D | FTS_DP)) { 53063daf6e196c51f162e0485478355d8e280eef5cDouglas Gregor continue; 54063daf6e196c51f162e0485478355d8e280eef5cDouglas Gregor } 554548ca2912e5f2b78a20e50c58d8a1a9c5e9e67cArgyrios Kyrtzidis 56063daf6e196c51f162e0485478355d8e280eef5cDouglas Gregor if (!android::base::EndsWith(ent->fts_path, ".h")) { 57be9af1288881110e406b87914162eaa59f1e5918Lang Hames continue; 58be9af1288881110e406b87914162eaa59f1e5918Lang Hames } 59be9af1288881110e406b87914162eaa59f1e5918Lang Hames 60be9af1288881110e406b87914162eaa59f1e5918Lang Hames headers.push_back(ent->fts_path); 61be9af1288881110e406b87914162eaa59f1e5918Lang Hames } 624548ca2912e5f2b78a20e50c58d8a1a9c5e9e67cArgyrios Kyrtzidis 63b4609806e9232593ece09ce08b630836e825865cDouglas Gregor return headers; 641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump} 653b6bef9a213249c6ab6d67c07b1ac6380961be3eBenjamin Kramer 66be9af1288881110e406b87914162eaa59f1e5918Lang Hamesllvm::StringRef StripPrefix(llvm::StringRef string, llvm::StringRef prefix) { 674967a710c84587c654b56c828382219c3937dacbPirama Arumuga Nainar if (string.startswith(prefix)) { 68be9af1288881110e406b87914162eaa59f1e5918Lang Hames return string.drop_front(prefix.size()); 694548ca2912e5f2b78a20e50c58d8a1a9c5e9e67cArgyrios Kyrtzidis } 704548ca2912e5f2b78a20e50c58d8a1a9c5e9e67cArgyrios Kyrtzidis return string; 711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump} 72ba0a9006dbc4814e1e35f82812cb5a1dad65e8b8Argyrios Kyrtzidis