simpleperf: report symbols of native libraries in apk file.

[android-x86/system-extras.git] / simpleperf / utils.cpp

diff --git a/simpleperf/utils.cpp b/simpleperf/utils.cpp
index b212263..2e68767 100644 (file)
--- a/simpleperf/utils.cpp
+++ b/simpleperf/utils.cpp
@@ -18,12 +18,54 @@
 
 #include <dirent.h>
 #include <errno.h>
+#include <fcntl.h>
 #include <stdarg.h>
 #include <stdio.h>
 #include <sys/stat.h>
 #include <unistd.h>
 
-#include <base/logging.h>
+#include <algorithm>
+#include <string>
+
+#include <android-base/file.h>
+#include <android-base/logging.h>
+
+void OneTimeFreeAllocator::Clear() {
+  for (auto& p : v_) {
+    delete[] p;
+  }
+  v_.clear();
+  cur_ = nullptr;
+  end_ = nullptr;
+}
+
+const char* OneTimeFreeAllocator::AllocateString(const std::string& s) {
+  size_t size = s.size() + 1;
+  if (cur_ + size > end_) {
+    size_t alloc_size = std::max(size, unit_size_);
+    char* p = new char[alloc_size];
+    v_.push_back(p);
+    cur_ = p;
+    end_ = p + alloc_size;
+  }
+  strcpy(cur_, s.c_str());
+  const char* result = cur_;
+  cur_ += size;
+  return result;
+}
+
+FileHelper::FileHelper() : fd_(-1) {
+}
+
+FileHelper::FileHelper(const std::string& filename) {
+  fd_ = TEMP_FAILURE_RETRY(open(filename.c_str(), O_RDONLY | O_BINARY));
+}
+
+FileHelper::~FileHelper() {
+  if (fd_ != -1) {
+    close(fd_);
+  }
+}
 
 void PrintIndented(size_t indent, const char* fmt, ...) {
   va_list ap;
@@ -55,7 +97,7 @@ void GetEntriesInDir(const std::string& dirpath, std::vector<std::string>* files
     if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) {
       continue;
     }
-    if (entry->d_type == DT_DIR) {
+    if (IsDir(dirpath + std::string("/") + entry->d_name)) {
       if (subdirs != nullptr) {
         subdirs->push_back(entry->d_name);
       }
@@ -78,17 +120,20 @@ bool IsDir(const std::string& dirpath) {
   return false;
 }
 
-bool RemovePossibleFile(const std::string& filename) {
+bool IsRegularFile(const std::string& filename) {
   struct stat st;
   if (stat(filename.c_str(), &st) == 0) {
-    if (!S_ISREG(st.st_mode)) {
-      LOG(ERROR) << filename << " is not a file.";
-      return false;
-    }
-    if (unlink(filename.c_str()) == -1) {
-      PLOG(ERROR) << "unlink(" << filename << ") failed";
-      return false;
+    if (S_ISREG(st.st_mode)) {
+      return true;
     }
   }
-  return true;
+  return false;
+}
+
+uint64_t GetFileSize(const std::string& filename) {
+  struct stat st;
+  if (stat(filename.c_str(), &st) == 0) {
+    return static_cast<uint64_t>(st.st_size);
+  }
+  return 0;
 }