src/ld/InputFiles.h

   1 /* -*- mode: C++; c-basic-offset: 4; tab-width: 4 -*-*
   2  *
   3  * Copyright (c) 2009-2011 Apple Inc. All rights reserved.
   4  *
   5  * @APPLE_LICENSE_HEADER_START@
   6  *
   7  * This file contains Original Code and/or Modifications of Original Code
   8  * as defined in and that are subject to the Apple Public Source License
   9  * Version 2.0 (the 'License'). You may not use this file except in
  10  * compliance with the License. Please obtain a copy of the License at
  11  * http://www.opensource.apple.com/apsl/ and read it before using this
  12  * file.
  13  *
  14  * The Original Code and all software distributed under the License are
  15  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  16  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  17  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  19  * Please see the License for the specific language governing rights and
  20  * limitations under the License.
  21  *
  22  * @APPLE_LICENSE_HEADER_END@
  23  */
  24
  25 #ifndef __INPUT_FILES_H__
  26 #define __INPUT_FILES_H__
  27
  28 #define HAVE_PTHREADS 1
  29
  30 #include <stdlib.h>
  31 #include <sys/types.h>
  32 #include <sys/stat.h>
  33 #include <sys/mman.h>
  34 #include <sys/sysctl.h>
  35 #include <fcntl.h>
  36 #include <errno.h>
  37 #include <limits.h>
  38 #include <unistd.h>
  39 #include <mach/mach_time.h>
  40 #include <mach/vm_statistics.h>
  41 #include <mach/mach_init.h>
  42 #include <mach/mach_host.h>
  43 #include <dlfcn.h>
  44 #include <mach-o/dyld.h>
  45 #if HAVE_PTHREADS
  46 #include <pthread.h>
  47 #endif
  48
  49 #include <vector>
  50
  51 #include "Options.h"
  52 #include "ld.hpp"
  53
  54 namespace ld {
  55 namespace tool {
  56
  57 class InputFiles : public ld::dylib::File::DylibHandler
  58 {
  59 public:
  60                                                                 InputFiles(Options& opts, const char** archName);
  61
  62         // implementation from ld::dylib::File::DylibHandler
  63         virtual ld::dylib::File*        findDylib(const char* installPath, const ld::dylib::File* fromDylib, bool speculative);
  64
  65         // iterates all atoms in initial files
  66         void                                            forEachInitialAtom(ld::File::AtomHandler&, ld::Internal& state);
  67         // searches libraries for name
  68         bool                                            searchLibraries(const char* name, bool searchDylibs, bool searchArchives,
  69                                                                                                                                   bool dataSymbolOnly, ld::File::AtomHandler&) const;
  70         // see if any linked dylibs export a weak def of symbol
  71         bool                                            searchWeakDefInDylib(const char* name) const;
  72         // copy dylibs to link with in command line order
  73         void                                            dylibs(ld::Internal& state);
  74
  75         void                                            archives(ld::Internal& state);
  76
  77         bool                                            inferredArch() const { return _inferredArch; }
  78
  79         void                                            addLinkerOptionLibraries(ld::Internal& state, ld::File::AtomHandler& handler);
  80         void                                            createIndirectDylibs();
  81
  82         // for -print_statistics
  83         volatile int64_t                        _totalObjectSize;
  84         volatile int64_t                        _totalArchiveSize;
  85         volatile int32_t                        _totalObjectLoaded;
  86         volatile int32_t                        _totalArchivesLoaded;
  87         volatile int32_t                        _totalDylibsLoaded;
  88
  89
  90 private:
  91         void                                            inferArchitecture(Options& opts, const char** archName);
  92         const char*                                     fileArch(const uint8_t* p, unsigned len);
  93         ld::File*                                       makeFile(const Options::FileInfo& info, bool indirectDylib);
  94         ld::File*                                       addDylib(ld::dylib::File* f,        const Options::FileInfo& info);
  95         void                                            logTraceInfo (const char* format, ...) const;
  96         void                                            logDylib(ld::File*, bool indirect, bool speculative);
  97         void                                            logArchive(ld::File*) const;
  98         void                                            markExplicitlyLinkedDylibs();
  99         void                                            checkDylibClientRestrictions(ld::dylib::File*);
 100         void                                            createOpaqueFileSections();
 101         bool                                            libraryAlreadyLoaded(const char* path);
 102         bool                                            frameworkAlreadyLoaded(const char* path, const char* frameworkName);
 103
 104         // for pipelined linking
 105     void                                                waitForInputFiles();
 106         static void                                     waitForInputFiles(InputFiles *inputFiles);
 107
 108         // for threaded input file processing
 109         void                                            parseWorkerThread();
 110         static void                                     parseWorkerThread(InputFiles *inputFiles);
 111         void                                            startThread(void (*threadFunc)(InputFiles *)) const;
 112
 113         typedef std::unordered_map<const char*, ld::dylib::File*, CStringHash, CStringEquals>   InstallNameToDylib;
 114
 115         const Options&                          _options;
 116         std::vector<ld::File*>          _inputFiles;
 117         mutable std::set<class ld::File*>       _archiveFilesLogged;
 118         mutable std::vector<std::string>        _archiveFilePaths;
 119         InstallNameToDylib                      _installPathToDylibs;
 120         std::set<ld::dylib::File*>      _allDylibs;
 121         ld::dylib::File*                        _bundleLoader;
 122         bool                                            _inferredArch;
 123     struct strcompclass {
 124         bool operator() (const char *a, const char *b) const { return ::strcmp(a, b) < 0; }
 125     };
 126
 127         // for threaded input file processing
 128 #if HAVE_PTHREADS
 129         pthread_mutex_t                         _parseLock;
 130         pthread_cond_t                          _parseWorkReady;                // used by parse threads to block for work
 131         pthread_cond_t                          _newFileAvailable;              // used by main thread to block for parsed input files
 132         int                                                     _availableWorkers;              // number of remaining unstarted parse threads
 133         int                                                     _idleWorkers;                   // number of running parse threads that are idle
 134         int                                                     _neededFileSlot;                // input file the resolver is currently blocked waiting for
 135         int                                                     _parseCursor;                   // slot to begin searching for a file to parse
 136         int                                                     _availableInputFiles;   // number of input fileinfos with readyToParse==true
 137 #endif
 138         const char *                            _exception;                             // passes an exception message from parse thread to main thread
 139         int                                                     _remainingInputFiles;   // number of input files still to parse
 140
 141         ld::File::Ordinal                       _indirectDylibOrdinal;
 142         ld::File::Ordinal                       _linkerOptionOrdinal;
 143
 144     class LibraryInfo {
 145         ld::File* _lib;
 146         bool      _isDylib;
 147     public:
 148         LibraryInfo(ld::dylib::File* dylib) : _lib(dylib), _isDylib(true) {};
 149         LibraryInfo(ld::archive::File* dylib) : _lib(dylib), _isDylib(false) {};
 150
 151         bool isDylib() const { return _isDylib; }
 152         ld::dylib::File *dylib() const { return (ld::dylib::File*)_lib; }
 153         ld::archive::File *archive() const { return (ld::archive::File*)_lib; }
 154     };
 155     std::vector<LibraryInfo>  _searchLibraries;
 156 };
 157
 158 } // namespace tool
 159 } // namespace ld
 160
 161 #endif // __INPUT_FILES_H__