Shinichiro Hamaji | 1d545aa | 2015-06-23 15:29:13 +0900 | [diff] [blame] | 1 | // Copyright 2015 Google Inc. All rights reserved |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
Fumitoshi Ukai | 744bb2b | 2015-06-25 00:10:52 +0900 | [diff] [blame] | 15 | // +build ignore |
| 16 | |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 17 | #include <limits.h> |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 18 | #include <stdio.h> |
| 19 | #include <string.h> |
Shinichiro Hamaji | 087cecd | 2015-07-06 19:51:58 +0900 | [diff] [blame^] | 20 | #include <stdlib.h> |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 21 | #include <unistd.h> |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 22 | |
| 23 | #include "ast.h" |
| 24 | #include "dep.h" |
| 25 | #include "eval.h" |
| 26 | #include "exec.h" |
| 27 | #include "file.h" |
| 28 | #include "file_cache.h" |
| 29 | #include "fileutil.h" |
Shinichiro Hamaji | 5f57a99 | 2015-06-30 19:39:39 +0900 | [diff] [blame] | 30 | #include "find.h" |
Shinichiro Hamaji | 08808d3 | 2015-06-26 08:02:45 +0900 | [diff] [blame] | 31 | #include "flags.h" |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 32 | #include "func.h" |
| 33 | #include "log.h" |
Shinichiro Hamaji | 5f86e1a | 2015-06-29 14:25:39 +0900 | [diff] [blame] | 34 | #include "ninja.h" |
Shinichiro Hamaji | 42b625f | 2015-06-16 23:07:21 +0900 | [diff] [blame] | 35 | #include "parser.h" |
Shinichiro Hamaji | 0d8e79b | 2015-06-30 03:29:35 +0900 | [diff] [blame] | 36 | #include "stats.h" |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 37 | #include "string_piece.h" |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 38 | #include "stringprintf.h" |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 39 | #include "strutil.h" |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 40 | #include "symtab.h" |
Shinichiro Hamaji | cbd34cd | 2015-07-05 02:53:04 +0900 | [diff] [blame] | 41 | #include "timeutil.h" |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 42 | #include "var.h" |
| 43 | |
| 44 | static const char* g_makefile; |
Shinichiro Hamaji | 64cd806 | 2015-06-27 16:58:36 +0900 | [diff] [blame] | 45 | static bool g_is_syntax_check_only; |
Shinichiro Hamaji | 5f86e1a | 2015-06-29 14:25:39 +0900 | [diff] [blame] | 46 | static bool g_generate_ninja; |
Shinichiro Hamaji | 8bddb46 | 2015-07-06 18:55:47 +0900 | [diff] [blame] | 47 | static const char* g_ninja_suffix; |
Shinichiro Hamaji | 4e950e6 | 2015-07-02 02:59:55 +0900 | [diff] [blame] | 48 | static bool g_use_find_emulator; |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 49 | |
Shinichiro Hamaji | 94d7a61 | 2015-06-29 18:16:11 +0900 | [diff] [blame] | 50 | static bool ParseCommandLineOptionWithArg(StringPiece option, |
| 51 | char* argv[], |
| 52 | int* index, |
| 53 | const char** out_arg) { |
| 54 | const char* arg = argv[*index]; |
| 55 | if (!HasPrefix(arg, option)) |
| 56 | return false; |
| 57 | if (arg[option.size()] == '\0') { |
| 58 | ++*index; |
| 59 | *out_arg = argv[*index]; |
| 60 | return true; |
| 61 | } |
| 62 | if (arg[option.size()] == '=') { |
| 63 | *out_arg = arg + option.size() + 1; |
| 64 | return true; |
| 65 | } |
Shinichiro Hamaji | 087cecd | 2015-07-06 19:51:58 +0900 | [diff] [blame^] | 66 | // E.g, -j999 |
| 67 | if (option.size() == 2) { |
| 68 | *out_arg = arg + option.size(); |
| 69 | return true; |
| 70 | } |
Shinichiro Hamaji | 94d7a61 | 2015-06-29 18:16:11 +0900 | [diff] [blame] | 71 | return false; |
| 72 | } |
| 73 | |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 74 | static void ParseCommandLine(int argc, char* argv[], |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 75 | vector<Symbol>* targets, |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 76 | vector<StringPiece>* cl_vars) { |
Shinichiro Hamaji | 087cecd | 2015-07-06 19:51:58 +0900 | [diff] [blame^] | 77 | // TODO: Decide the appropriate number based on the number of cores. |
| 78 | g_num_jobs = 32; |
| 79 | const char* num_jobs_str; |
| 80 | |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 81 | for (int i = 1; i < argc; i++) { |
| 82 | const char* arg = argv[i]; |
| 83 | if (!strcmp(arg, "-f")) { |
| 84 | g_makefile = argv[++i]; |
Shinichiro Hamaji | 08808d3 | 2015-06-26 08:02:45 +0900 | [diff] [blame] | 85 | } else if (!strcmp(arg, "-c")) { |
| 86 | g_is_syntax_check_only = true; |
Shinichiro Hamaji | 64cd806 | 2015-06-27 16:58:36 +0900 | [diff] [blame] | 87 | } else if (!strcmp(arg, "-i")) { |
| 88 | g_is_dry_run = true; |
Shinichiro Hamaji | 54e52dd | 2015-06-27 17:14:06 +0900 | [diff] [blame] | 89 | } else if (!strcmp(arg, "--kati_stats")) { |
| 90 | g_enable_stat_logs = true; |
Shinichiro Hamaji | 5f86e1a | 2015-06-29 14:25:39 +0900 | [diff] [blame] | 91 | } else if (!strcmp(arg, "--ninja")) { |
| 92 | g_generate_ninja = true; |
Shinichiro Hamaji | 8bddb46 | 2015-07-06 18:55:47 +0900 | [diff] [blame] | 93 | } else if (ParseCommandLineOptionWithArg( |
Shinichiro Hamaji | 087cecd | 2015-07-06 19:51:58 +0900 | [diff] [blame^] | 94 | "-j", argv, &i, &num_jobs_str)) { |
| 95 | g_num_jobs = strtol(num_jobs_str, NULL, 10); |
| 96 | if (g_num_jobs <= 0) { |
| 97 | ERROR("Invalid -j flag: %s", num_jobs_str); |
| 98 | } |
| 99 | } else if (ParseCommandLineOptionWithArg( |
Shinichiro Hamaji | 8bddb46 | 2015-07-06 18:55:47 +0900 | [diff] [blame] | 100 | "--ninja_suffix", argv, &i, &g_ninja_suffix)) { |
Shinichiro Hamaji | 4e950e6 | 2015-07-02 02:59:55 +0900 | [diff] [blame] | 101 | } else if (!strcmp(arg, "--use_find_emulator")) { |
| 102 | g_use_find_emulator = true; |
Shinichiro Hamaji | 94d7a61 | 2015-06-29 18:16:11 +0900 | [diff] [blame] | 103 | } else if (ParseCommandLineOptionWithArg( |
Shinichiro Hamaji | 9facae2 | 2015-07-03 17:16:36 +0900 | [diff] [blame] | 104 | "--goma_dir", argv, &i, &g_goma_dir)) { |
| 105 | } else if (ParseCommandLineOptionWithArg( |
Shinichiro Hamaji | 94d7a61 | 2015-06-29 18:16:11 +0900 | [diff] [blame] | 106 | "--ignore_optional_include", |
| 107 | argv, &i, &g_ignore_optional_include_pattern)) { |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 108 | } else if (arg[0] == '-') { |
| 109 | ERROR("Unknown flag: %s", arg); |
| 110 | } else { |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 111 | if (strchr(arg, '=')) { |
| 112 | cl_vars->push_back(arg); |
| 113 | } else { |
| 114 | targets->push_back(Intern(arg)); |
| 115 | } |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 116 | } |
| 117 | } |
| 118 | } |
| 119 | |
| 120 | static void Init() { |
| 121 | InitSymtab(); |
| 122 | InitFuncTable(); |
| 123 | InitDepNodePool(); |
Shinichiro Hamaji | 42b625f | 2015-06-16 23:07:21 +0900 | [diff] [blame] | 124 | InitParser(); |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 125 | |
| 126 | if (g_makefile == NULL) { |
| 127 | if (Exists("GNUmakefile")) { |
| 128 | g_makefile = "GNUmakefile"; |
| 129 | } else if (Exists("makefile")) { |
| 130 | g_makefile = "makefile"; |
| 131 | } else if (Exists("Makefile")) { |
| 132 | g_makefile = "Makefile"; |
| 133 | } else { |
| 134 | ERROR("*** No targets specified and no makefile found."); |
| 135 | } |
| 136 | } |
| 137 | } |
| 138 | |
| 139 | static void Quit() { |
Shinichiro Hamaji | 0d8e79b | 2015-06-30 03:29:35 +0900 | [diff] [blame] | 140 | ReportAllStats(); |
| 141 | |
Shinichiro Hamaji | 42b625f | 2015-06-16 23:07:21 +0900 | [diff] [blame] | 142 | QuitParser(); |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 143 | QuitDepNodePool(); |
| 144 | QuitFuncTable(); |
| 145 | QuitSymtab(); |
| 146 | } |
| 147 | |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 148 | static void ReadBootstrapMakefile(const vector<Symbol>& targets, |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 149 | vector<AST*>* asts) { |
| 150 | string bootstrap = ( |
| 151 | "CC:=cc\n" |
| 152 | "CXX:=g++\n" |
| 153 | "AR:=ar\n" |
| 154 | "MAKE:=kati\n" |
| 155 | // Pretend to be GNU make 3.81, for compatibility. |
| 156 | "MAKE_VERSION:=3.81\n" |
| 157 | "SHELL:=/bin/sh\n" |
| 158 | // TODO: Add more builtin vars. |
| 159 | |
| 160 | // http://www.gnu.org/software/make/manual/make.html#Catalogue-of-Rules |
| 161 | // The document above is actually not correct. See default.c: |
| 162 | // http://git.savannah.gnu.org/cgit/make.git/tree/default.c?id=4.1 |
| 163 | ".c.o:\n" |
| 164 | "\t$(CC) $(CFLAGS) $(CPPFLAGS) $(TARGET_ARCH) -c -o $@ $<\n" |
| 165 | ".cc.o:\n" |
| 166 | "\t$(CXX) $(CXXFLAGS) $(CPPFLAGS) $(TARGET_ARCH) -c -o $@ $<\n" |
| 167 | // TODO: Add more builtin rules. |
| 168 | ); |
| 169 | bootstrap += StringPrintf("MAKECMDGOALS:=%s\n", |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 170 | JoinSymbols(targets, " ").c_str()); |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 171 | |
| 172 | char cwd[PATH_MAX]; |
| 173 | if (!getcwd(cwd, PATH_MAX)) { |
| 174 | fprintf(stderr, "getcwd failed\n"); |
| 175 | CHECK(false); |
| 176 | } |
| 177 | bootstrap += StringPrintf("CURDIR:=%s\n", cwd); |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 178 | Parse(Intern(bootstrap).str(), Loc("*bootstrap*", 0), asts); |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 179 | } |
| 180 | |
Shinichiro Hamaji | f62e9a7 | 2015-06-26 04:18:21 +0900 | [diff] [blame] | 181 | static void SetVar(StringPiece l, VarOrigin origin, Vars* vars) { |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 182 | size_t found = l.find('='); |
| 183 | CHECK(found != string::npos); |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 184 | Symbol lhs = Intern(l.substr(0, found)); |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 185 | StringPiece rhs = l.substr(found + 1); |
Shinichiro Hamaji | 81699be | 2015-06-22 18:07:38 +0900 | [diff] [blame] | 186 | vars->Assign(lhs, |
| 187 | new RecursiveVar(NewLiteral(rhs.data()), origin, rhs.data())); |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 188 | } |
| 189 | |
| 190 | extern "C" char** environ; |
| 191 | static void FillDefaultVars(const vector<StringPiece>& cl_vars, Vars* vars) { |
| 192 | for (char** p = environ; *p; p++) { |
Shinichiro Hamaji | f62e9a7 | 2015-06-26 04:18:21 +0900 | [diff] [blame] | 193 | SetVar(*p, VarOrigin::ENVIRONMENT, vars); |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 194 | } |
| 195 | for (StringPiece l : cl_vars) { |
Shinichiro Hamaji | f62e9a7 | 2015-06-26 04:18:21 +0900 | [diff] [blame] | 196 | SetVar(l, VarOrigin::COMMAND_LINE, vars); |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 197 | } |
| 198 | } |
| 199 | |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 200 | static int Run(const vector<Symbol>& targets, |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 201 | const vector<StringPiece>& cl_vars) { |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 202 | MakefileCacheManager* cache_mgr = NewMakefileCacheManager(); |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 203 | |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 204 | Vars* vars = new Vars(); |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 205 | FillDefaultVars(cl_vars, vars); |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 206 | Evaluator* ev = new Evaluator(vars); |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 207 | |
| 208 | vector<AST*> bootstrap_asts; |
| 209 | ReadBootstrapMakefile(targets, &bootstrap_asts); |
| 210 | ev->set_is_bootstrap(true); |
| 211 | for (AST* ast : bootstrap_asts) { |
| 212 | LOG("%s", ast->DebugString().c_str()); |
| 213 | ast->Eval(ev); |
| 214 | } |
| 215 | ev->set_is_bootstrap(false); |
| 216 | |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 217 | vars->Assign(Intern("MAKEFILE_LIST"), |
Shinichiro Hamaji | b74b890 | 2015-06-22 18:22:30 +0900 | [diff] [blame] | 218 | new SimpleVar(make_shared<string>( |
Shinichiro Hamaji | f62e9a7 | 2015-06-26 04:18:21 +0900 | [diff] [blame] | 219 | StringPrintf(" %s", g_makefile)), VarOrigin::FILE)); |
Shinichiro Hamaji | b74b890 | 2015-06-22 18:22:30 +0900 | [diff] [blame] | 220 | |
Shinichiro Hamaji | 54e52dd | 2015-06-27 17:14:06 +0900 | [diff] [blame] | 221 | { |
| 222 | ScopedTimeReporter tr("eval time"); |
| 223 | Makefile* mk = cache_mgr->ReadMakefile(g_makefile); |
| 224 | for (AST* ast : mk->asts()) { |
| 225 | LOG("%s", ast->DebugString().c_str()); |
| 226 | ast->Eval(ev); |
| 227 | } |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 228 | } |
| 229 | |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 230 | vector<DepNode*> nodes; |
Shinichiro Hamaji | 54e52dd | 2015-06-27 17:14:06 +0900 | [diff] [blame] | 231 | { |
| 232 | ScopedTimeReporter tr("make dep time"); |
| 233 | MakeDep(ev, ev->rules(), ev->rule_vars(), targets, &nodes); |
| 234 | } |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 235 | |
Shinichiro Hamaji | 45a0c76 | 2015-06-26 06:47:10 +0900 | [diff] [blame] | 236 | for (const auto& p : ev->exports()) { |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 237 | const Symbol name = p.first; |
Shinichiro Hamaji | 45a0c76 | 2015-06-26 06:47:10 +0900 | [diff] [blame] | 238 | if (p.second) { |
| 239 | Var* v = ev->LookupVar(name); |
| 240 | shared_ptr<string> value = v->Eval(ev); |
| 241 | LOG("setenv(%s, %s)", name.c_str(), value->c_str()); |
| 242 | setenv(name.c_str(), value->c_str(), 1); |
| 243 | } else { |
| 244 | LOG("unsetenv(%s)", name.c_str()); |
| 245 | unsetenv(name.c_str()); |
| 246 | } |
| 247 | } |
| 248 | |
Shinichiro Hamaji | 64cd806 | 2015-06-27 16:58:36 +0900 | [diff] [blame] | 249 | if (g_is_syntax_check_only) |
| 250 | return 0; |
| 251 | |
Shinichiro Hamaji | 5f86e1a | 2015-06-29 14:25:39 +0900 | [diff] [blame] | 252 | if (g_generate_ninja) { |
| 253 | ScopedTimeReporter tr("generate ninja time"); |
Shinichiro Hamaji | 8bddb46 | 2015-07-06 18:55:47 +0900 | [diff] [blame] | 254 | GenerateNinja(g_ninja_suffix, nodes, ev); |
Shinichiro Hamaji | 5f86e1a | 2015-06-29 14:25:39 +0900 | [diff] [blame] | 255 | return 0; |
| 256 | } |
| 257 | |
Shinichiro Hamaji | 54e52dd | 2015-06-27 17:14:06 +0900 | [diff] [blame] | 258 | { |
| 259 | ScopedTimeReporter tr("exec time"); |
| 260 | Exec(nodes, ev); |
| 261 | } |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 262 | |
Shinichiro Hamaji | 0e74c54 | 2015-06-22 16:17:08 +0900 | [diff] [blame] | 263 | for (AST* ast : bootstrap_asts) |
| 264 | delete ast; |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 265 | delete ev; |
| 266 | delete vars; |
| 267 | delete cache_mgr; |
| 268 | |
Shinichiro Hamaji | 64cd806 | 2015-06-27 16:58:36 +0900 | [diff] [blame] | 269 | return 0; |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 270 | } |
| 271 | |
| 272 | int main(int argc, char* argv[]) { |
| 273 | Init(); |
Shinichiro Hamaji | e799275 | 2015-06-29 18:38:35 +0900 | [diff] [blame] | 274 | vector<Symbol> targets; |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 275 | vector<StringPiece> cl_vars; |
| 276 | ParseCommandLine(argc, argv, &targets, &cl_vars); |
Shinichiro Hamaji | 5f57a99 | 2015-06-30 19:39:39 +0900 | [diff] [blame] | 277 | // This depends on command line flags. |
Shinichiro Hamaji | 4e950e6 | 2015-07-02 02:59:55 +0900 | [diff] [blame] | 278 | if (g_use_find_emulator) |
| 279 | InitFindEmulator(); |
Shinichiro Hamaji | 9db4cb8 | 2015-06-22 16:51:04 +0900 | [diff] [blame] | 280 | int r = Run(targets, cl_vars); |
Shinichiro Hamaji | 776ca30 | 2015-06-06 03:52:48 +0900 | [diff] [blame] | 281 | Quit(); |
| 282 | return r; |
| 283 | } |