Full support for --sysroot.
[binutils-gdb.git] / gold / fileread.cc
index a92da9d974fc7fd76e5e938493e5ad799d5405c2..1bbd8412a8109cf1e6c051dd0311a50d133b25b9 100644 (file)
@@ -1,12 +1,33 @@
 // fileread.cc -- read files for gold
 
+// Copyright 2006, 2007 Free Software Foundation, Inc.
+// Written by Ian Lance Taylor <iant@google.com>.
+
+// This file is part of gold.
+
+// 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; either version 3 of the License, or
+// (at your option) any later version.
+
+// 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 Street - Fifth Floor, Boston,
+// MA 02110-1301, USA.
+
 #include "gold.h"
 
-#include <cassert>
 #include <cstring>
 #include <cerrno>
 #include <fcntl.h>
 #include <unistd.h>
+#include <sys/mman.h>
+#include "filenames.h"
 
 #include "options.h"
 #include "dirsearch.h"
@@ -19,8 +40,15 @@ namespace gold
 
 File_read::View::~View()
 {
-  assert(!this->is_locked());
-  delete[] this->data_;
+  gold_assert(!this->is_locked());
+  if (!this->mapped_)
+    delete[] this->data_;
+  else
+    {
+      if (::munmap(const_cast<unsigned char*>(this->data_), this->size_) != 0)
+        fprintf(stderr, _("%s: munmap failed: %s\n"),
+                program_name, strerror(errno));
+    }
 }
 
 void
@@ -32,7 +60,7 @@ File_read::View::lock()
 void
 File_read::View::unlock()
 {
-  assert(this->lock_count_ > 0);
+  gold_assert(this->lock_count_ > 0);
   --this->lock_count_;
 }
 
@@ -49,7 +77,7 @@ File_read::View::is_locked()
 
 File_read::~File_read()
 {
-  assert(this->lock_count_ == 0);
+  gold_assert(this->lock_count_ == 0);
   if (this->descriptor_ >= 0)
     {
       if (close(this->descriptor_) < 0)
@@ -61,23 +89,49 @@ File_read::~File_read()
   this->clear_views(true);
 }
 
+// Open the file.
+
 bool
 File_read::open(const std::string& name)
 {
-  assert(this->lock_count_ == 0
-        && this->descriptor_ < 0
-        && this->name_.empty());
+  gold_assert(this->lock_count_ == 0
+             && this->descriptor_ < 0
+             && this->name_.empty());
   this->name_ = name;
+
   this->descriptor_ = ::open(this->name_.c_str(), O_RDONLY);
+
+  if (this->descriptor_ >= 0)
+    {
+      struct stat s;
+      if (::fstat(this->descriptor_, &s) < 0)
+       {
+         fprintf(stderr, _("%s: %s: fstat failed: %s"), program_name,
+                 this->name_.c_str(), strerror(errno));
+         gold_exit(false);
+       }
+      this->size_ = s.st_size;
+    }
+
   ++this->lock_count_;
+
   return this->descriptor_ >= 0;
 }
 
-int
-File_read::get_descriptor()
+// Open the file for testing purposes.
+
+bool
+File_read::open(const std::string& name, const unsigned char* contents,
+               off_t size)
 {
-  assert(this->lock_count_ > 0);
-  return this->descriptor_;
+  gold_assert(this->lock_count_ == 0
+             && this->descriptor_ < 0
+             && this->name_.empty());
+  this->name_ = name;
+  this->contents_ = contents;
+  this->size_ = size;
+  ++this->lock_count_;
+  return true;
 }
 
 void
@@ -89,8 +143,10 @@ File_read::lock()
 void
 File_read::unlock()
 {
-  assert(this->lock_count_ > 0);
+  gold_assert(this->lock_count_ > 0);
   --this->lock_count_;
+  if (this->lock_count_ == 0)
+    this->clear_views(false);
 }
 
 bool
@@ -102,97 +158,143 @@ File_read::is_locked()
 // See if we have a view which covers the file starting at START for
 // SIZE bytes.  Return a pointer to the View if found, NULL if not.
 
-File_read::View*
+inline File_read::View*
 File_read::find_view(off_t start, off_t size)
 {
-  for (std::list<File_read::View*>::iterator p = this->view_list_.begin();
-       p != this->view_list_.end();
-       ++p)
-    {
-      if ((*p)->start() <= start
-         && (*p)->start() + (*p)->size() >= start + size)
-       return *p;
-    }
-  return NULL;
+  off_t page = File_read::page_offset(start);
+  Views::iterator p = this->views_.find(page);
+  if (p == this->views_.end())
+    return NULL;
+  if (p->second->size() - (start - page) < size)
+    return NULL;
+  return p->second;
 }
 
-// Read data from the file.  Return the number of bytes read.  If
-// PBYTES is not NULL, store the number of bytes in *PBYTES, otherwise
-// require that we read exactly the number of bytes requested.
+// Read SIZE bytes from the file starting at offset START.  Read into
+// the buffer at P.
 
-off_t
-File_read::do_read(off_t start, off_t size, void* p, off_t* pbytes)
+void
+File_read::do_read(off_t start, off_t size, void* p)
 {
-  assert(this->lock_count_ > 0);
-  int o = this->descriptor_;
+  gold_assert(this->lock_count_ > 0);
 
-  if (lseek(o, start, SEEK_SET) < 0)
+  off_t bytes;
+  if (this->contents_ != NULL)
     {
-      fprintf(stderr, _("%s: %s: lseek to %lld failed: %s"),
-             program_name, this->filename().c_str(),
-             static_cast<long long>(start),
-             strerror(errno));
-      gold_exit(false);
+      bytes = this->size_ - start;
+      if (bytes >= size)
+       {
+         memcpy(p, this->contents_ + start, size);
+         return;
+       }
     }
-
-  off_t bytes = ::read(o, p, size);
-  if (bytes < 0)
+  else
     {
-      fprintf(stderr, _("%s: %s: read failed: %s\n"),
-             program_name, this->filename().c_str(), strerror(errno));
-      gold_exit(false);
-    }
+      bytes = ::pread(this->descriptor_, p, size, start);
+      if (bytes == size)
+       return;
 
-  if (pbytes != NULL)
-    *pbytes = bytes;
-  else if (bytes != size)
-    {
-      fprintf(stderr,
-             _("%s: %s: file too short: read only %lld of %lld "
-               "bytes at %lld\n"),
-             program_name, this->filename().c_str(),
-             static_cast<long long>(bytes),
-             static_cast<long long>(size),
-             static_cast<long long>(start));
-      gold_exit(false);
+      if (bytes < 0)
+       {
+         fprintf(stderr, _("%s: %s: pread failed: %s\n"),
+                 program_name, this->filename().c_str(), strerror(errno));
+         gold_exit(false);
+       }
     }
 
-  return bytes;
+  fprintf(stderr,
+         _("%s: %s: file too short: read only %lld of %lld bytes at %lld\n"),
+         program_name, this->filename().c_str(),
+         static_cast<long long>(bytes),
+         static_cast<long long>(size),
+         static_cast<long long>(start));
+  gold_exit(false);
 }
 
+// Read data from the file.
+
 void
-File_read::read(off_t start, off_t size, void* p, off_t* pbytes)
+File_read::read(off_t start, off_t size, void* p)
 {
-  assert(this->lock_count_ > 0);
+  gold_assert(this->lock_count_ > 0);
 
   File_read::View* pv = this->find_view(start, size);
   if (pv != NULL)
     {
       memcpy(p, pv->data() + (start - pv->start()), size);
-      if (pbytes != NULL)
-       *pbytes = size;
       return;
     }
 
-  this->do_read(start, size, p, pbytes);
+  this->do_read(start, size, p);
 }
 
 // Find an existing view or make a new one.
 
 File_read::View*
-File_read::find_or_make_view(off_t start, off_t size, off_t* pbytes)
+File_read::find_or_make_view(off_t start, off_t size, bool cache)
 {
-  assert(this->lock_count_ > 0);
+  gold_assert(this->lock_count_ > 0);
 
-  File_read::View* pv = this->find_view(start, size);
-  if (pv != NULL)
-    return pv;
+  off_t poff = File_read::page_offset(start);
 
-  unsigned char* p = new unsigned char[size];
-  off_t bytes = this->do_read(start, size, p, pbytes);
-  pv = new File_read::View(start, bytes, p);
-  this->view_list_.push_back(pv);
-  return pv;
+  File_read::View* const vnull = NULL;
+  std::pair<Views::iterator, bool> ins =
+    this->views_.insert(std::make_pair(poff, vnull));
+
+  if (!ins.second)
+    {
+      // There was an existing view at this offset.
+      File_read::View* v = ins.first->second;
+      if (v->size() - (start - v->start()) >= size)
+       {
+         if (cache)
+           v->set_cache();
+         return v;
+       }
+
+      // This view is not large enough.
+      this->saved_views_.push_back(v);
+    }
+
+  // We need to read data from the file.  We read full pages for
+  // greater efficiency on small files.
+
+  off_t psize = File_read::pages(size + (start - poff));
+
+  if (poff + psize >= this->size_)
+    {
+      psize = this->size_ - poff;
+      gold_assert(psize >= size);
+    }
+
+  File_read::View* v;
+
+  if (this->contents_ != NULL)
+    {
+      unsigned char* p = new unsigned char[psize];
+      this->do_read(poff, psize, p);
+      v = new File_read::View(poff, psize, p, cache, false);
+    }
+  else
+    {
+      void* p = ::mmap(NULL, psize, PROT_READ, MAP_SHARED,
+                       this->descriptor_, poff);
+      if (p == MAP_FAILED)
+        {
+          fprintf(stderr, _("%s: %s: mmap offset %lld size %lld failed: %s\n"),
+                  program_name, this->filename().c_str(),
+                  static_cast<long long>(poff),
+                  static_cast<long long>(psize),
+                  strerror(errno));
+          gold_exit(false);
+        }
+
+      const unsigned char* pbytes = static_cast<const unsigned char*>(p);
+      v = new File_read::View(poff, psize, pbytes, cache, true);
+    }
+
+  ins.first->second = v;
+  return v;
 }
 
 // This implementation of get_view just reads into a memory buffer,
@@ -200,18 +302,18 @@ File_read::find_or_make_view(off_t start, off_t size, off_t* pbytes)
 // mmap.
 
 const unsigned char*
-File_read::get_view(off_t start, off_t size, off_t* pbytes)
+File_read::get_view(off_t start, off_t size, bool cache)
 {
-  assert(this->lock_count_ > 0);
-  File_read::View* pv = this->find_or_make_view(start, size, pbytes);
+  gold_assert(this->lock_count_ > 0);
+  File_read::View* pv = this->find_or_make_view(start, size, cache);
   return pv->data() + (start - pv->start());
 }
 
 File_view*
-File_read::get_lasting_view(off_t start, off_t size, off_t* pbytes)
+File_read::get_lasting_view(off_t start, off_t size, bool cache)
 {
-  assert(this->lock_count_ > 0);
-  File_read::View* pv = this->find_or_make_view(start, size, pbytes);
+  gold_assert(this->lock_count_ > 0);
+  File_read::View* pv = this->find_or_make_view(start, size, cache);
   pv->lock();
   return new File_view(*this, pv, pv->data() + (start - pv->start()));
 }
@@ -221,18 +323,34 @@ File_read::get_lasting_view(off_t start, off_t size, off_t* pbytes)
 void
 File_read::clear_views(bool destroying)
 {
-  std::list<File_read::View*>::iterator p = this->view_list_.begin();
-  while (p != this->view_list_.end())
+  for (Views::iterator p = this->views_.begin();
+       p != this->views_.end();
+       ++p)
     {
-      if ((*p)->is_locked())
+      if (!p->second->is_locked()
+         && (destroying || !p->second->should_cache()))
+       delete p->second;
+      else
        {
-         assert(!destroying);
-         ++p;
+         gold_assert(!destroying);
+         this->saved_views_.push_back(p->second);
        }
-      else
+    }
+  this->views_.clear();
+
+  Saved_views::iterator p = this->saved_views_.begin();
+  while (p != this->saved_views_.end())
+    {
+      if (!(*p)->is_locked()
+         && (destroying || !(*p)->should_cache()))
        {
          delete *p;
-         p = this->view_list_.erase(p);
+         p = this->saved_views_.erase(p);
+       }
+      else
+       {
+         gold_assert(!destroying);
+         ++p;
        }
     }
 }
@@ -241,39 +359,100 @@ File_read::clear_views(bool destroying)
 
 File_view::~File_view()
 {
-  assert(this->file_.is_locked());
+  gold_assert(this->file_.is_locked());
   this->view_->unlock();
 }
 
 // Class Input_file.
 
+// Create a file for testing.
+
+Input_file::Input_file(const char* name, const unsigned char* contents,
+                      off_t size)
+  : file_()
+{
+  this->input_argument_ =
+    new Input_file_argument(name, false, "", Position_dependent_options());
+  bool ok = file_.open(name, contents, size);
+  gold_assert(ok);
+}
+
+// Open the file.
+
+// If the filename is not absolute, we assume it is in the current
+// directory *except* when:
+//    A) input_argument_->is_lib() is true; or
+//    B) input_argument_->extra_search_path() is not empty.
+// In both cases, we look in extra_search_path + library_path to find
+// the file location, rather than the current directory.
+
 void
 Input_file::open(const General_options& options, const Dirsearch& dirpath)
 {
   std::string name;
-  if (!this->input_argument_.is_lib())
-    name = this->input_argument_.name();
-  else
+
+  // Case 1: name is an absolute file, just try to open it
+  // Case 2: name is relative but is_lib is false and extra_search_path
+  //         is empty
+  if (IS_ABSOLUTE_PATH (this->input_argument_->name())
+      || (!this->input_argument_->is_lib()
+         && this->input_argument_->extra_search_path() == NULL))
+    name = this->input_argument_->name();
+
+  // Case 3: is_lib is true
+  else if (this->input_argument_->is_lib())
     {
+      // We don't yet support extra_search_path with -l.
+      gold_assert(this->input_argument_->extra_search_path() == NULL);
       std::string n1("lib");
-      n1 += this->input_argument_.lib_basename();
+      n1 += this->input_argument_->name();
       std::string n2;
       if (options.is_static())
-       n2 = n1 + ".so";
-      n1 += ".a";
-      name = dirpath.find(n1, n2);
+       n1 += ".a";
+      else
+       {
+         n2 = n1 + ".a";
+         n1 += ".so";
+       }
+      name = dirpath.find(n1, n2, &this->is_in_sysroot_);
       if (name.empty())
        {
-         fprintf(stderr, _("%s: cannot find %s"), program_name,
-                 this->input_argument_.name());
+         fprintf(stderr, _("%s: cannot find -l%s\n"), program_name,
+                 this->input_argument_->name());
          gold_exit(false);
        }
     }
 
+  // Case 4: extra_search_path is not empty
+  else
+    {
+      gold_assert(this->input_argument_->extra_search_path() != NULL);
+
+      // First, check extra_search_path.
+      name = this->input_argument_->extra_search_path();
+      if (!IS_DIR_SEPARATOR (name[name.length() - 1]))
+        name += '/';
+      name += this->input_argument_->name();
+      struct stat dummy_stat;
+      if (::stat(name.c_str(), &dummy_stat) < 0)
+        {
+          // extra_search_path failed, so check the normal search-path.
+          name = dirpath.find(this->input_argument_->name(), "",
+                             &this->is_in_sysroot_);
+          if (name.empty())
+            {
+              fprintf(stderr, _("%s: cannot find %s\n"), program_name,
+                      this->input_argument_->name());
+              gold_exit(false);
+            }
+        }
+    }
+
+  // Now that we've figured out where the file lives, try to open it.
   if (!this->file_.open(name))
     {
-      fprintf(stderr, _("%s: cannot open %s: %s"), program_name, name.c_str(),
-             strerror(errno));
+      fprintf(stderr, _("%s: cannot open %s: %s\n"), program_name,
+             name.c_str(), strerror(errno));
       gold_exit(false);
     }
 }