gold/readsyms.cc

   1 // readsyms.cc -- read input file symbols for gold
   2
   3 // Copyright 2006, 2007, 2008, 2009 Free Software Foundation, Inc.
   4 // Written by Ian Lance Taylor <iant@google.com>.
   5
   6 // This file is part of gold.
   7
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 3 of the License, or
  11 // (at your option) any later version.
  12
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the Free Software
  20 // Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston,
  21 // MA 02110-1301, USA.
  22
  23 #include "gold.h"
  24
  25 #include <cstring>
  26
  27 #include "elfcpp.h"
  28 #include "options.h"
  29 #include "dirsearch.h"
  30 #include "symtab.h"
  31 #include "object.h"
  32 #include "archive.h"
  33 #include "script.h"
  34 #include "readsyms.h"
  35 #include "plugin.h"
  36
  37 namespace gold
  38 {
  39
  40 // If we fail to open the object, then we won't create an Add_symbols
  41 // task.  However, we still need to unblock the token, or else the
  42 // link won't proceed to generate more error messages.  We can only
  43 // unblock tokens when the workqueue lock is held, so we need a dummy
  44 // task to do that.  The dummy task has to maintain the right sequence
  45 // of blocks, so we need both this_blocker and next_blocker.
  46
  47 class Unblock_token : public Task
  48 {
  49  public:
  50   Unblock_token(Task_token* this_blocker, Task_token* next_blocker)
  51     : this_blocker_(this_blocker), next_blocker_(next_blocker)
  52   { }
  53
  54   ~Unblock_token()
  55   {
  56     if (this->this_blocker_ != NULL)
  57       delete this->this_blocker_;
  58   }
  59
  60   Task_token*
  61   is_runnable()
  62   {
  63     if (this->this_blocker_ != NULL && this->this_blocker_->is_blocked())
  64       return this->this_blocker_;
  65     return NULL;
  66   }
  67
  68   void
  69   locks(Task_locker* tl)
  70   { tl->add(this, this->next_blocker_); }
  71
  72   void
  73   run(Workqueue*)
  74   { }
  75
  76   std::string
  77   get_name() const
  78   { return "Unblock_token"; }
  79
  80  private:
  81   Task_token* this_blocker_;
  82   Task_token* next_blocker_;
  83 };
  84
  85 // Class read_symbols.
  86
  87 Read_symbols::~Read_symbols()
  88 {
  89   // The this_blocker_ and next_blocker_ pointers are passed on to the
  90   // Add_symbols task.
  91 }
  92
  93 // Return whether a Read_symbols task is runnable.  We can read an
  94 // ordinary input file immediately.  For an archive specified using
  95 // -l, we have to wait until the search path is complete.
  96
  97 Task_token*
  98 Read_symbols::is_runnable()
  99 {
 100   if (this->input_argument_->is_file()
 101       && this->input_argument_->file().may_need_search()
 102       && this->dirpath_->token()->is_blocked())
 103     return this->dirpath_->token();
 104
 105   return NULL;
 106 }
 107
 108 // Return a Task_locker for a Read_symbols task.  We don't need any
 109 // locks here.
 110
 111 void
 112 Read_symbols::locks(Task_locker*)
 113 {
 114 }
 115
 116 // Run a Read_symbols task.
 117
 118 void
 119 Read_symbols::run(Workqueue* workqueue)
 120 {
 121   // If we didn't queue a new task, then we need to explicitly unblock
 122   // the token.
 123   if (!this->do_read_symbols(workqueue))
 124     workqueue->queue_soon(new Unblock_token(this->this_blocker_,
 125                                             this->next_blocker_));
 126 }
 127
 128 // Open the file and read the symbols.  Return true if a new task was
 129 // queued, false if that could not happen due to some error.
 130
 131 bool
 132 Read_symbols::do_read_symbols(Workqueue* workqueue)
 133 {
 134   if (this->input_argument_->is_group())
 135     {
 136       gold_assert(this->input_group_ == NULL);
 137       this->do_group(workqueue);
 138       return true;
 139     }
 140
 141   Input_file* input_file = new Input_file(&this->input_argument_->file());
 142   if (!input_file->open(this->options_, *this->dirpath_, this))
 143     return false;
 144
 145   // Read enough of the file to pick up the entire ELF header.
 146
 147   off_t filesize = input_file->file().filesize();
 148
 149   if (filesize == 0)
 150     {
 151       gold_error(_("%s: file is empty"),
 152                  input_file->file().filename().c_str());
 153       return false;
 154     }
 155
 156   int read_size = elfcpp::Elf_sizes<64>::ehdr_size;
 157   if (filesize < read_size)
 158     read_size = filesize;
 159
 160   const unsigned char* ehdr = input_file->file().get_view(0, 0, read_size,
 161                                                           true, false);
 162
 163   if (read_size >= Archive::sarmag)
 164     {
 165       bool is_thin_archive
 166           = memcmp(ehdr, Archive::armagt, Archive::sarmag) == 0;
 167       if (is_thin_archive
 168           || memcmp(ehdr, Archive::armag, Archive::sarmag) == 0)
 169         {
 170           // This is an archive.
 171           Archive* arch = new Archive(this->input_argument_->file().name(),
 172                                       input_file, is_thin_archive,
 173                                       this->dirpath_, this);
 174           arch->setup(this->input_objects_);
 175
 176           // Unlock the archive so it can be used in the next task.
 177           arch->unlock(this);
 178
 179           workqueue->queue_next(new Add_archive_symbols(this->symtab_,
 180                                                         this->layout_,
 181                                                         this->input_objects_,
 182                                                         this->mapfile_,
 183                                                         arch,
 184                                                         this->input_group_,
 185                                                         this->this_blocker_,
 186                                                         this->next_blocker_));
 187           return true;
 188         }
 189     }
 190
 191   if (parameters->options().has_plugins())
 192     {
 193       Pluginobj* obj = parameters->options().plugins()->claim_file(input_file,
 194                                                                    0, filesize);
 195       if (obj != NULL)
 196         {
 197           // The input file was claimed by a plugin, and its symbols
 198           // have been provided by the plugin.
 199
 200           // We are done with the file at this point, so unlock it.
 201           obj->unlock(this);
 202
 203           workqueue->queue_next(new Add_symbols(this->input_objects_,
 204                                                 this->symtab_,
 205                                                 this->layout_,
 206                                                 obj, NULL,
 207                                                 this->this_blocker_,
 208                                                 this->next_blocker_));
 209           return true;
 210         }
 211     }
 212
 213   if (read_size >= 4)
 214     {
 215       static unsigned char elfmagic[4] =
 216         {
 217           elfcpp::ELFMAG0, elfcpp::ELFMAG1,
 218           elfcpp::ELFMAG2, elfcpp::ELFMAG3
 219         };
 220       if (memcmp(ehdr, elfmagic, 4) == 0)
 221         {
 222           // This is an ELF object.
 223
 224           Object* obj = make_elf_object(input_file->filename(),
 225                                         input_file, 0, ehdr, read_size);
 226           if (obj == NULL)
 227             return false;
 228
 229           Read_symbols_data* sd = new Read_symbols_data;
 230           obj->read_symbols(sd);
 231
 232           // Opening the file locked it, so now we need to unlock it.
 233           // We need to unlock it before queuing the Add_symbols task,
 234           // because the workqueue doesn't know about our lock on the
 235           // file.  If we queue the Add_symbols task first, it will be
 236           // stuck on the end of the file lock, but since the
 237           // workqueue doesn't know about that lock, it will never
 238           // release the Add_symbols task.
 239
 240           input_file->file().unlock(this);
 241
 242           // We use queue_next because everything is cached for this
 243           // task to run right away if possible.
 244
 245           workqueue->queue_next(new Add_symbols(this->input_objects_,
 246                                                 this->symtab_, this->layout_,
 247                                                 obj, sd,
 248                                                 this->this_blocker_,
 249                                                 this->next_blocker_));
 250
 251           return true;
 252         }
 253     }
 254
 255   // Queue up a task to try to parse this file as a script.  We use a
 256   // separate task so that the script will be read in order with other
 257   // objects named on the command line.  Also so that we don't try to
 258   // read multiple scripts simultaneously, which could lead to
 259   // unpredictable changes to the General_options structure.
 260
 261   workqueue->queue_soon(new Read_script(this->options_,
 262                                         this->symtab_,
 263                                         this->layout_,
 264                                         this->dirpath_,
 265                                         this->input_objects_,
 266                                         this->mapfile_,
 267                                         this->input_group_,
 268                                         this->input_argument_,
 269                                         input_file,
 270                                         this->this_blocker_,
 271                                         this->next_blocker_));
 272   return true;
 273 }
 274
 275 // Handle a group.  We need to walk through the arguments over and
 276 // over until we don't see any new undefined symbols.  We do this by
 277 // setting off Read_symbols Tasks as usual, but recording the archive
 278 // entries instead of deleting them.  We also start a Finish_group
 279 // Task which runs after we've read all the symbols.  In that task we
 280 // process the archives in a loop until we are done.
 281
 282 void
 283 Read_symbols::do_group(Workqueue* workqueue)
 284 {
 285   Input_group* input_group = new Input_group();
 286
 287   const Input_file_group* group = this->input_argument_->group();
 288   Task_token* this_blocker = this->this_blocker_;
 289
 290   for (Input_file_group::const_iterator p = group->begin();
 291        p != group->end();
 292        ++p)
 293     {
 294       const Input_argument* arg = &*p;
 295       gold_assert(arg->is_file());
 296
 297       Task_token* next_blocker = new Task_token(true);
 298       next_blocker->add_blocker();
 299       workqueue->queue_soon(new Read_symbols(this->options_,
 300                                              this->input_objects_,
 301                                              this->symtab_, this->layout_,
 302                                              this->dirpath_, this->mapfile_,
 303                                              arg, input_group,
 304                                              this_blocker, next_blocker));
 305       this_blocker = next_blocker;
 306     }
 307
 308   const int saw_undefined = this->symtab_->saw_undefined();
 309   workqueue->queue_soon(new Finish_group(this->input_objects_,
 310                                          this->symtab_,
 311                                          this->layout_,
 312                                          this->mapfile_,
 313                                          input_group,
 314                                          saw_undefined,
 315                                          this_blocker,
 316                                          this->next_blocker_));
 317 }
 318
 319 // Return a debugging name for a Read_symbols task.
 320
 321 std::string
 322 Read_symbols::get_name() const
 323 {
 324   if (!this->input_argument_->is_group())
 325     {
 326       std::string ret("Read_symbols ");
 327       if (this->input_argument_->file().is_lib())
 328         ret += "-l";
 329       ret += this->input_argument_->file().name();
 330       return ret;
 331     }
 332
 333   std::string ret("Read_symbols group (");
 334   bool add_space = false;
 335   const Input_file_group* group = this->input_argument_->group();
 336   for (Input_file_group::const_iterator p = group->begin();
 337        p != group->end();
 338        ++p)
 339     {
 340       if (add_space)
 341         ret += ' ';
 342       ret += p->file().name();
 343       add_space = true;
 344     }
 345   return ret + ')';
 346 }
 347
 348 // Class Add_symbols.
 349
 350 Add_symbols::~Add_symbols()
 351 {
 352   if (this->this_blocker_ != NULL)
 353     delete this->this_blocker_;
 354   // next_blocker_ is deleted by the task associated with the next
 355   // input file.
 356 }
 357
 358 // We are blocked by this_blocker_.  We block next_blocker_.  We also
 359 // lock the file.
 360
 361 Task_token*
 362 Add_symbols::is_runnable()
 363 {
 364   if (this->this_blocker_ != NULL && this->this_blocker_->is_blocked())
 365     return this->this_blocker_;
 366   if (this->object_->is_locked())
 367     return this->object_->token();
 368   return NULL;
 369 }
 370
 371 void
 372 Add_symbols::locks(Task_locker* tl)
 373 {
 374   tl->add(this, this->next_blocker_);
 375   tl->add(this, this->object_->token());
 376 }
 377
 378 // Add the symbols in the object to the symbol table.
 379
 380 void
 381 Add_symbols::run(Workqueue*)
 382 {
 383   Pluginobj* pluginobj = this->object_->pluginobj();
 384   if (pluginobj != NULL)
 385     {
 386       this->object_->add_symbols(this->symtab_, this->sd_, this->layout_);
 387       return;
 388     }
 389
 390   if (!this->input_objects_->add_object(this->object_))
 391     {
 392       // FIXME: We need to close the descriptor here.
 393       delete this->object_;
 394     }
 395   else
 396     {
 397       this->object_->layout(this->symtab_, this->layout_, this->sd_);
 398       this->object_->add_symbols(this->symtab_, this->sd_, this->layout_);
 399       this->object_->release();
 400     }
 401   delete this->sd_;
 402   this->sd_ = NULL;
 403 }
 404
 405 // Class Finish_group.
 406
 407 Finish_group::~Finish_group()
 408 {
 409   if (this->this_blocker_ != NULL)
 410     delete this->this_blocker_;
 411   // next_blocker_ is deleted by the task associated with the next
 412   // input file following the group.
 413 }
 414
 415 // We need to wait for THIS_BLOCKER_ and unblock NEXT_BLOCKER_.
 416
 417 Task_token*
 418 Finish_group::is_runnable()
 419 {
 420   if (this->this_blocker_ != NULL && this->this_blocker_->is_blocked())
 421     return this->this_blocker_;
 422   return NULL;
 423 }
 424
 425 void
 426 Finish_group::locks(Task_locker* tl)
 427 {
 428   tl->add(this, this->next_blocker_);
 429 }
 430
 431 // Loop over the archives until there are no new undefined symbols.
 432
 433 void
 434 Finish_group::run(Workqueue*)
 435 {
 436   int saw_undefined = this->saw_undefined_;
 437   while (saw_undefined != this->symtab_->saw_undefined())
 438     {
 439       saw_undefined = this->symtab_->saw_undefined();
 440
 441       for (Input_group::const_iterator p = this->input_group_->begin();
 442            p != this->input_group_->end();
 443            ++p)
 444         {
 445           Task_lock_obj<Archive> tl(this, *p);
 446
 447           (*p)->add_symbols(this->symtab_, this->layout_,
 448                             this->input_objects_, this->mapfile_);
 449         }
 450     }
 451
 452   // Delete all the archives now that we no longer need them.
 453   for (Input_group::const_iterator p = this->input_group_->begin();
 454        p != this->input_group_->end();
 455        ++p)
 456     delete *p;
 457   delete this->input_group_;
 458 }
 459
 460 // Class Read_script
 461
 462 Read_script::~Read_script()
 463 {
 464   if (this->this_blocker_ != NULL)
 465     delete this->this_blocker_;
 466   // next_blocker_ is deleted by the task associated with the next
 467   // input file.
 468 }
 469
 470 // We are blocked by this_blocker_.
 471
 472 Task_token*
 473 Read_script::is_runnable()
 474 {
 475   if (this->this_blocker_ != NULL && this->this_blocker_->is_blocked())
 476     return this->this_blocker_;
 477   return NULL;
 478 }
 479
 480 // We don't unlock next_blocker_ here.  If the script names any input
 481 // files, then the last file will be responsible for unlocking it.
 482
 483 void
 484 Read_script::locks(Task_locker*)
 485 {
 486 }
 487
 488 // Read the script, if it is a script.
 489
 490 void
 491 Read_script::run(Workqueue* workqueue)
 492 {
 493   bool used_next_blocker;
 494   if (!read_input_script(workqueue, this->options_, this->symtab_,
 495                          this->layout_, this->dirpath_, this->input_objects_,
 496                          this->mapfile_, this->input_group_,
 497                          this->input_argument_, this->input_file_,
 498                          this->next_blocker_, &used_next_blocker))
 499     {
 500       // Here we have to handle any other input file types we need.
 501       gold_error(_("%s: not an object or archive"),
 502                  this->input_file_->file().filename().c_str());
 503     }
 504
 505   if (!used_next_blocker)
 506     {
 507       // Queue up a task to unlock next_blocker.  We can't just unlock
 508       // it here, as we don't hold the workqueue lock.
 509       workqueue->queue_soon(new Unblock_token(NULL, this->next_blocker_));
 510     }
 511 }
 512
 513 // Return a debugging name for a Read_script task.
 514
 515 std::string
 516 Read_script::get_name() const
 517 {
 518   std::string ret("Read_script ");
 519   if (this->input_argument_->file().is_lib())
 520     ret += "-l";
 521   ret += this->input_argument_->file().name();
 522   return ret;
 523 }
 524
 525 } // End namespace gold.