| // 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 <cstring> |
| #include <cerrno> |
| #include <fcntl.h> |
| #include <unistd.h> |
| #include <sys/mman.h> |
| #include "filenames.h" |
| |
| #include "options.h" |
| #include "dirsearch.h" |
| #include "fileread.h" |
| |
| namespace gold |
| { |
| |
| // Class File_read::View. |
| |
| File_read::View::~View() |
| { |
| gold_assert(!this->is_locked()); |
| if (!this->mapped_) |
| delete[] this->data_; |
| else |
| { |
| if (::munmap(const_cast<unsigned char*>(this->data_), this->size_) != 0) |
| gold_warning(_("munmap failed: %s"), strerror(errno)); |
| |
| File_read::current_mapped_bytes -= this->size_; |
| } |
| } |
| |
| void |
| File_read::View::lock() |
| { |
| ++this->lock_count_; |
| } |
| |
| void |
| File_read::View::unlock() |
| { |
| gold_assert(this->lock_count_ > 0); |
| --this->lock_count_; |
| } |
| |
| bool |
| File_read::View::is_locked() |
| { |
| return this->lock_count_ > 0; |
| } |
| |
| // Class File_read. |
| |
| // The File_read static variables. |
| unsigned long long File_read::total_mapped_bytes; |
| unsigned long long File_read::current_mapped_bytes; |
| unsigned long long File_read::maximum_mapped_bytes; |
| |
| // The File_read class is designed to support file descriptor caching, |
| // but this is not currently implemented. |
| |
| File_read::~File_read() |
| { |
| gold_assert(this->lock_count_ == 0); |
| if (this->descriptor_ >= 0) |
| { |
| if (close(this->descriptor_) < 0) |
| gold_warning(_("close of %s failed: %s"), |
| this->name_.c_str(), strerror(errno)); |
| this->descriptor_ = -1; |
| } |
| this->name_.clear(); |
| this->clear_views(true); |
| } |
| |
| // Open the file. |
| |
| bool |
| File_read::open(const std::string& name) |
| { |
| 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) |
| gold_error(_("%s: fstat failed: %s"), |
| this->name_.c_str(), strerror(errno)); |
| this->size_ = s.st_size; |
| } |
| |
| ++this->lock_count_; |
| |
| return this->descriptor_ >= 0; |
| } |
| |
| // Open the file for testing purposes. |
| |
| bool |
| File_read::open(const std::string& name, const unsigned char* contents, |
| off_t size) |
| { |
| 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 |
| File_read::lock() |
| { |
| ++this->lock_count_; |
| } |
| |
| void |
| File_read::unlock() |
| { |
| gold_assert(this->lock_count_ > 0); |
| --this->lock_count_; |
| if (this->lock_count_ == 0) |
| { |
| File_read::total_mapped_bytes += this->mapped_bytes_; |
| File_read::current_mapped_bytes += this->mapped_bytes_; |
| this->mapped_bytes_ = 0; |
| if (File_read::current_mapped_bytes > File_read::maximum_mapped_bytes) |
| File_read::maximum_mapped_bytes = File_read::current_mapped_bytes; |
| |
| this->clear_views(false); |
| } |
| } |
| |
| bool |
| File_read::is_locked() |
| { |
| return this->lock_count_ > 0; |
| } |
| |
| // 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. |
| |
| inline File_read::View* |
| File_read::find_view(off_t start, off_t size) |
| { |
| 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 SIZE bytes from the file starting at offset START. Read into |
| // the buffer at P. |
| |
| void |
| File_read::do_read(off_t start, off_t size, void* p) |
| { |
| gold_assert(this->lock_count_ > 0); |
| |
| off_t bytes; |
| if (this->contents_ != NULL) |
| { |
| bytes = this->size_ - start; |
| if (bytes >= size) |
| { |
| memcpy(p, this->contents_ + start, size); |
| return; |
| } |
| } |
| else |
| { |
| bytes = ::pread(this->descriptor_, p, size, start); |
| if (bytes == size) |
| return; |
| |
| if (bytes < 0) |
| { |
| gold_fatal(_("%s: pread failed: %s"), |
| this->filename().c_str(), strerror(errno)); |
| return; |
| } |
| } |
| |
| gold_fatal(_("%s: file too short: read only %lld of %lld bytes at %lld"), |
| this->filename().c_str(), |
| static_cast<long long>(bytes), |
| static_cast<long long>(size), |
| static_cast<long long>(start)); |
| } |
| |
| // Read data from the file. |
| |
| void |
| File_read::read(off_t start, off_t size, void* p) |
| { |
| 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); |
| return; |
| } |
| |
| 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, bool cache) |
| { |
| gold_assert(this->lock_count_ > 0); |
| |
| off_t poff = File_read::page_offset(start); |
| |
| 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) |
| gold_fatal(_("%s: mmap offset %lld size %lld failed: %s"), |
| this->filename().c_str(), |
| static_cast<long long>(poff), |
| static_cast<long long>(psize), |
| strerror(errno)); |
| |
| this->mapped_bytes_ += psize; |
| |
| 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, |
| // which we store on view_list_. At some point we should support |
| // mmap. |
| |
| const unsigned char* |
| File_read::get_view(off_t start, off_t size, bool cache) |
| { |
| 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, bool cache) |
| { |
| 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())); |
| } |
| |
| // Remove all the file views. |
| |
| void |
| File_read::clear_views(bool destroying) |
| { |
| for (Views::iterator p = this->views_.begin(); |
| p != this->views_.end(); |
| ++p) |
| { |
| if (!p->second->is_locked() |
| && (destroying || !p->second->should_cache())) |
| delete p->second; |
| else |
| { |
| gold_assert(!destroying); |
| this->saved_views_.push_back(p->second); |
| } |
| } |
| 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->saved_views_.erase(p); |
| } |
| else |
| { |
| gold_assert(!destroying); |
| ++p; |
| } |
| } |
| } |
| |
| // Print statistical information to stderr. This is used for --stats. |
| |
| void |
| File_read::print_stats() |
| { |
| fprintf(stderr, _("%s: total bytes mapped for read: %llu\n"), |
| program_name, File_read::total_mapped_bytes); |
| fprintf(stderr, _("%s: maximum bytes mapped for read at one time: %llu\n"), |
| program_name, File_read::maximum_mapped_bytes); |
| } |
| |
| // Class File_view. |
| |
| File_view::~File_view() |
| { |
| 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. |
| |
| bool |
| Input_file::open(const General_options& options, const Dirsearch& dirpath) |
| { |
| std::string name; |
| |
| // 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(); |
| this->found_name_ = 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_->name(); |
| std::string n2; |
| if (options.is_static() |
| || this->input_argument_->options().do_static_search()) |
| n1 += ".a"; |
| else |
| { |
| n2 = n1 + ".a"; |
| n1 += ".so"; |
| } |
| name = dirpath.find(n1, n2, &this->is_in_sysroot_); |
| if (name.empty()) |
| { |
| gold_error(_("cannot find -l%s"), |
| this->input_argument_->name()); |
| return false; |
| } |
| if (n2.empty() || name[name.length() - 1] == 'o') |
| this->found_name_ = n1; |
| else |
| this->found_name_ = n2; |
| } |
| // 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()) |
| { |
| gold_error(_("cannot find %s"), |
| this->input_argument_->name()); |
| return false; |
| } |
| } |
| this->found_name_ = this->input_argument_->name(); |
| } |
| |
| // Now that we've figured out where the file lives, try to open it. |
| if (!this->file_.open(name)) |
| { |
| gold_error(_("cannot open %s: %s"), |
| name.c_str(), strerror(errno)); |
| return false; |
| } |
| |
| return true; |
| } |
| |
| } // End namespace gold. |