Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2010, The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #ifndef BCC_SCRIPT_H |
| 18 | #define BCC_SCRIPT_H |
| 19 | |
| 20 | #include <bcc/bcc.h> |
Logan | f340bf7 | 2011-01-14 17:51:40 +0800 | [diff] [blame] | 21 | #include "bcc_internal.h" |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 22 | |
Logan | ecf4cbd | 2011-01-06 05:34:11 +0800 | [diff] [blame] | 23 | #include "Compiler.h" |
| 24 | |
Logan Chien | 7890d43 | 2011-08-03 14:55:17 +0800 | [diff] [blame] | 25 | #include <vector> |
| 26 | #include <string> |
| 27 | |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 28 | #include <stddef.h> |
| 29 | |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 30 | namespace llvm { |
| 31 | class Module; |
Daniel Malea | 094881f | 2011-12-14 17:39:16 -0500 | [diff] [blame] | 32 | class GDBJITRegistrar; |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 33 | } |
| 34 | |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 35 | namespace bcc { |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 36 | class ScriptCompiled; |
| 37 | class ScriptCached; |
Logan | 474cbd2 | 2011-01-31 01:47:44 +0800 | [diff] [blame] | 38 | class SourceInfo; |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 39 | |
| 40 | namespace ScriptStatus { |
| 41 | enum StatusType { |
| 42 | Unknown, |
| 43 | Compiled, |
Logan | 3584900 | 2011-01-15 07:30:43 +0800 | [diff] [blame] | 44 | #if USE_CACHE |
Logan | f7f0ac5 | 2011-01-07 03:53:43 +0800 | [diff] [blame] | 45 | Cached, |
Logan | 3584900 | 2011-01-15 07:30:43 +0800 | [diff] [blame] | 46 | #endif |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 47 | }; |
| 48 | } |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 49 | |
Logan | 0647e9e | 2010-12-29 00:21:31 +0800 | [diff] [blame] | 50 | class Script { |
Logan | 3973641 | 2010-12-29 00:24:04 +0800 | [diff] [blame] | 51 | private: |
Logan | f340bf7 | 2011-01-14 17:51:40 +0800 | [diff] [blame] | 52 | int mErrorCode; |
Logan | 3973641 | 2010-12-29 00:24:04 +0800 | [diff] [blame] | 53 | |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 54 | ScriptStatus::StatusType mStatus; |
| 55 | |
| 56 | union { |
| 57 | ScriptCompiled *mCompiled; |
Logan | 3584900 | 2011-01-15 07:30:43 +0800 | [diff] [blame] | 58 | #if USE_CACHE |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 59 | ScriptCached *mCached; |
Logan | 3584900 | 2011-01-15 07:30:43 +0800 | [diff] [blame] | 60 | #endif |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 61 | }; |
| 62 | |
Logan Chien | 311c26f | 2011-07-11 14:30:34 +0800 | [diff] [blame] | 63 | #if USE_CACHE |
| 64 | std::string mCacheDir; |
| 65 | std::string mCacheName; |
| 66 | #endif |
Logan | ecf4cbd | 2011-01-06 05:34:11 +0800 | [diff] [blame] | 67 | |
Logan | 4259805 | 2011-01-26 22:41:13 +0800 | [diff] [blame] | 68 | bool mIsContextSlotNotAvail; |
| 69 | |
Logan | 474cbd2 | 2011-01-31 01:47:44 +0800 | [diff] [blame] | 70 | // Source List |
| 71 | SourceInfo *mSourceList[2]; |
| 72 | // Note: mSourceList[0] (main source) |
| 73 | // Note: mSourceList[1] (library source) |
| 74 | // TODO(logan): Generalize this, use vector or SmallVector instead! |
Logan | 3133c41 | 2011-01-06 06:15:40 +0800 | [diff] [blame] | 75 | |
Logan Chien | 7890d43 | 2011-08-03 14:55:17 +0800 | [diff] [blame] | 76 | // External Function List |
| 77 | std::vector<char const *> mUserDefinedExternalSymbols; |
| 78 | |
Logan | ecf4cbd | 2011-01-06 05:34:11 +0800 | [diff] [blame] | 79 | // Register Symbol Lookup Function |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 80 | BCCSymbolLookupFn mpExtSymbolLookupFn; |
Logan | f340bf7 | 2011-01-14 17:51:40 +0800 | [diff] [blame] | 81 | void *mpExtSymbolLookupFnContext; |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 82 | |
Stephen Hines | b67c9e7 | 2012-03-22 11:02:48 -0700 | [diff] [blame] | 83 | uint32_t mCompilerVersion; |
| 84 | uint32_t mOptimizationLevel; |
| 85 | |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 86 | public: |
Logan | 033f46e | 2011-01-06 05:51:24 +0800 | [diff] [blame] | 87 | Script() : mErrorCode(BCC_NO_ERROR), mStatus(ScriptStatus::Unknown), |
Logan Chien | 311c26f | 2011-07-11 14:30:34 +0800 | [diff] [blame] | 88 | mIsContextSlotNotAvail(false), |
Stephen Hines | b67c9e7 | 2012-03-22 11:02:48 -0700 | [diff] [blame] | 89 | mpExtSymbolLookupFn(NULL), mpExtSymbolLookupFnContext(NULL), |
| 90 | mCompilerVersion(0), mOptimizationLevel(3) { |
Logan | ecf4cbd | 2011-01-06 05:34:11 +0800 | [diff] [blame] | 91 | Compiler::GlobalInitialization(); |
Logan | 474cbd2 | 2011-01-31 01:47:44 +0800 | [diff] [blame] | 92 | |
| 93 | mSourceList[0] = NULL; |
| 94 | mSourceList[1] = NULL; |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 95 | } |
| 96 | |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 97 | ~Script(); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 98 | |
Logan | 474cbd2 | 2011-01-31 01:47:44 +0800 | [diff] [blame] | 99 | int addSourceBC(size_t idx, |
| 100 | char const *resName, |
| 101 | const char *bitcode, |
| 102 | size_t bitcodeSize, |
| 103 | unsigned long flags); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 104 | |
Logan | 474cbd2 | 2011-01-31 01:47:44 +0800 | [diff] [blame] | 105 | int addSourceModule(size_t idx, |
| 106 | llvm::Module *module, |
| 107 | unsigned long flags); |
Logan | ecf4cbd | 2011-01-06 05:34:11 +0800 | [diff] [blame] | 108 | |
Logan | 474cbd2 | 2011-01-31 01:47:44 +0800 | [diff] [blame] | 109 | int addSourceFile(size_t idx, |
| 110 | char const *path, |
| 111 | unsigned long flags); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 112 | |
Logan Chien | 7890d43 | 2011-08-03 14:55:17 +0800 | [diff] [blame] | 113 | void markExternalSymbol(char const *name) { |
| 114 | mUserDefinedExternalSymbols.push_back(name); |
| 115 | } |
| 116 | |
| 117 | std::vector<char const *> const &getUserDefinedExternalSymbols() const { |
| 118 | return mUserDefinedExternalSymbols; |
| 119 | } |
| 120 | |
Logan Chien | 311c26f | 2011-07-11 14:30:34 +0800 | [diff] [blame] | 121 | int prepareExecutable(char const *cacheDir, |
| 122 | char const *cacheName, |
| 123 | unsigned long flags); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 124 | |
Joseph Wen | 34c600a | 2011-07-25 17:59:17 -0700 | [diff] [blame] | 125 | int prepareSharedObject(char const *cacheDir, |
| 126 | char const *cacheName, |
| 127 | unsigned long flags); |
| 128 | |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 129 | char const *getCompilerErrorMessage(); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 130 | |
Logan | cf3e521 | 2010-12-29 01:44:55 +0800 | [diff] [blame] | 131 | void *lookup(const char *name); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 132 | |
Stephen Hines | b67c9e7 | 2012-03-22 11:02:48 -0700 | [diff] [blame] | 133 | uint32_t getCompilerVersion() const { |
| 134 | return mCompilerVersion; |
| 135 | } |
| 136 | |
| 137 | uint32_t getOptimizationLevel() const { |
| 138 | return mOptimizationLevel; |
| 139 | } |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 140 | |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 141 | size_t getExportVarCount() const; |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 142 | |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 143 | size_t getExportFuncCount() const; |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 144 | |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 145 | size_t getPragmaCount() const; |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 146 | |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 147 | size_t getFuncCount() const; |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 148 | |
Stephen Hines | 071288a | 2011-01-27 14:38:26 -0800 | [diff] [blame] | 149 | size_t getObjectSlotCount() const; |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 150 | |
| 151 | void getExportVarList(size_t size, void **list); |
| 152 | |
| 153 | void getExportFuncList(size_t size, void **list); |
| 154 | |
Joseph Wen | f36637f | 2011-07-06 18:27:12 -0700 | [diff] [blame] | 155 | void getExportVarNameList(std::vector<std::string> &list); |
| 156 | |
| 157 | void getExportFuncNameList(std::vector<std::string> &list); |
| 158 | |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 159 | void getPragmaList(size_t size, |
| 160 | char const **keyList, |
| 161 | char const **valueList); |
| 162 | |
Logan | f340bf7 | 2011-01-14 17:51:40 +0800 | [diff] [blame] | 163 | void getFuncInfoList(size_t size, FuncInfo *list); |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 164 | |
Stephen Hines | 071288a | 2011-01-27 14:38:26 -0800 | [diff] [blame] | 165 | void getObjectSlotList(size_t size, uint32_t *list); |
| 166 | |
Shih-wei Liao | 5e3e0ce | 2011-06-17 13:59:46 -0700 | [diff] [blame] | 167 | size_t getELFSize() const; |
| 168 | |
| 169 | const char *getELF() const; |
Logan | 02286cb | 2011-01-07 00:30:47 +0800 | [diff] [blame] | 170 | |
Shih-wei Liao | ce82d49 | 2011-01-20 12:34:03 -0800 | [diff] [blame] | 171 | int registerSymbolCallback(BCCSymbolLookupFn pFn, void *pContext); |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 172 | |
Logan Chien | d2a5f30 | 2011-07-19 20:32:25 +0800 | [diff] [blame] | 173 | #if USE_OLD_JIT |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 174 | char *getContext(); |
Logan Chien | d2a5f30 | 2011-07-19 20:32:25 +0800 | [diff] [blame] | 175 | #endif |
Logan | be79ada | 2011-01-13 01:33:45 +0800 | [diff] [blame] | 176 | |
Logan | eaa0cc3 | 2010-12-29 01:04:20 +0800 | [diff] [blame] | 177 | |
Logan | f340bf7 | 2011-01-14 17:51:40 +0800 | [diff] [blame] | 178 | void setError(int error) { |
Logan | 3973641 | 2010-12-29 00:24:04 +0800 | [diff] [blame] | 179 | if (mErrorCode == BCC_NO_ERROR && error != BCC_NO_ERROR) { |
| 180 | mErrorCode = error; |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 181 | } |
| 182 | } |
| 183 | |
Logan | f340bf7 | 2011-01-14 17:51:40 +0800 | [diff] [blame] | 184 | int getError() { |
| 185 | int result = mErrorCode; |
Logan | 3973641 | 2010-12-29 00:24:04 +0800 | [diff] [blame] | 186 | mErrorCode = BCC_NO_ERROR; |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 187 | return result; |
| 188 | } |
Logan | 033f46e | 2011-01-06 05:51:24 +0800 | [diff] [blame] | 189 | |
| 190 | private: |
Logan | 3584900 | 2011-01-15 07:30:43 +0800 | [diff] [blame] | 191 | #if USE_CACHE |
Joseph Wen | 34c600a | 2011-07-25 17:59:17 -0700 | [diff] [blame] | 192 | int internalLoadCache(bool checkOnly); |
Logan | 3584900 | 2011-01-15 07:30:43 +0800 | [diff] [blame] | 193 | #endif |
Joseph Wen | 34c600a | 2011-07-25 17:59:17 -0700 | [diff] [blame] | 194 | int internalCompile(bool compileOnly); |
Logan | 3f3d31f | 2010-11-27 13:52:03 +0800 | [diff] [blame] | 195 | }; |
| 196 | |
| 197 | } // namespace bcc |
| 198 | |
| 199 | #endif // BCC_SCRIPT_H |