blob: 7d50219fc83b831bcc103047684c1a7f8aba16dc [file] [log] [blame]
Shinichiro Hamaji1d545aa2015-06-23 15:29:13 +09001// 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 Ukai744bb2b2015-06-25 00:10:52 +090015// +build ignore
16
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090017#include "func.h"
18
Shinichiro Hamaji62b16e72015-07-02 01:42:31 +090019#include <errno.h>
Dan Willemsenf06d8012016-10-03 00:16:07 -070020#include <fcntl.h>
Shinichiro Hamajid5271452015-06-17 18:50:03 +090021#include <limits.h>
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090022#include <stdio.h>
Shinichiro Hamajid5271452015-06-17 18:50:03 +090023#include <stdlib.h>
Dan Willemsenf06d8012016-10-03 00:16:07 -070024#include <sys/stat.h>
Shinichiro Hamaji62b16e72015-07-02 01:42:31 +090025#include <unistd.h>
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090026
Shinichiro Hamajid87e59e2015-06-17 18:18:34 +090027#include <algorithm>
Shinichiro Hamajid5271452015-06-17 18:50:03 +090028#include <iterator>
Shinichiro Hamajicf0cd682015-06-18 16:18:13 +090029#include <memory>
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090030#include <unordered_map>
31
Shinichiro Hamaji9619b362015-06-16 16:13:25 +090032#include "eval.h"
Shinichiro Hamaji94d6f2a2015-07-05 05:32:25 +090033#include "fileutil.h"
Shinichiro Hamaji5f57a992015-06-30 19:39:39 +090034#include "find.h"
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090035#include "log.h"
Shinichiro Hamaji80456fb2015-06-18 14:56:10 +090036#include "parser.h"
Shinichiro Hamaji0d8e79b2015-06-30 03:29:35 +090037#include "stats.h"
Shinichiro Hamaji645cca72015-09-24 17:04:21 +090038#include "stmt.h"
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090039#include "strutil.h"
Shinichiro Hamajie7992752015-06-29 18:38:35 +090040#include "symtab.h"
Shinichiro Hamajicf0cd682015-06-18 16:18:13 +090041#include "var.h"
Shinichiro Hamaji776ca302015-06-06 03:52:48 +090042
43namespace {
44
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090045// TODO: This code is very similar to
46// NinjaGenerator::TranslateCommand. Factor them out.
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090047void StripShellComment(string* cmd) {
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090048 if (cmd->find('#') == string::npos)
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090049 return;
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090050
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090051 string res;
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090052 bool prev_backslash = false;
53 // Set space as an initial value so the leading comment will be
54 // stripped out.
55 char prev_char = ' ';
56 char quote = 0;
57 bool done = false;
58 const char* in = cmd->c_str();
59 for (; *in && !done; in++) {
60 switch (*in) {
61 case '#':
62 if (quote == 0 && isspace(prev_char)) {
Shinichiro Hamaji212b7a52015-10-05 17:02:29 +090063 while (in[1] && *in != '\n')
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090064 in++;
65 break;
66 }
67
68 case '\'':
69 case '"':
70 case '`':
71 if (quote) {
72 if (quote == *in)
73 quote = 0;
74 } else if (!prev_backslash) {
75 quote = *in;
76 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090077 res += *in;
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090078 break;
79
80 case '\\':
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090081 res += '\\';
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090082 break;
83
84 default:
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090085 res += *in;
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090086 }
87
88 if (*in == '\\') {
89 prev_backslash = !prev_backslash;
90 } else {
91 prev_backslash = false;
92 }
93
94 prev_char = *in;
95 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +090096 cmd->swap(res);
Shinichiro Hamajifc14d5f2015-07-28 17:07:57 +090097}
98
Shinichiro Hamaji2e6cbfc2015-06-16 18:46:50 +090099void PatsubstFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900100 const string&& pat_str = args[0]->Eval(ev);
101 const string&& repl = args[1]->Eval(ev);
102 const string&& str = args[2]->Eval(ev);
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900103 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900104 Pattern pat(pat_str);
105 for (StringPiece tok : WordScanner(str)) {
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900106 ww.MaybeAddWhitespace();
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900107 pat.AppendSubst(tok, repl, s);
Shinichiro Hamaji2e6cbfc2015-06-16 18:46:50 +0900108 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900109}
110
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900111void StripFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900112 const string&& str = args[0]->Eval(ev);
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900113 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900114 for (StringPiece tok : WordScanner(str)) {
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900115 ww.Write(tok);
116 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900117}
118
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900119void SubstFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900120 const string&& pat = args[0]->Eval(ev);
121 const string&& repl = args[1]->Eval(ev);
122 const string&& str = args[2]->Eval(ev);
123 if (pat.empty()) {
124 *s += str;
125 *s += repl;
Shinichiro Hamaji5af931d2015-07-06 15:37:59 +0900126 return;
127 }
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900128 size_t index = 0;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900129 while (index < str.size()) {
130 size_t found = str.find(pat, index);
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900131 if (found == string::npos)
132 break;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900133 AppendString(StringPiece(str).substr(index, found - index), s);
134 AppendString(repl, s);
135 index = found + pat.size();
Shinichiro Hamaji37591ce2015-06-16 19:36:05 +0900136 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900137 AppendString(StringPiece(str).substr(index), s);
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900138}
139
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900140void FindstringFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900141 const string&& find = args[0]->Eval(ev);
142 const string&& in = args[1]->Eval(ev);
143 if (in.find(find) != string::npos)
144 AppendString(find, s);
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900145}
146
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900147void FilterFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900148 const string&& pat_buf = args[0]->Eval(ev);
149 const string&& text = args[1]->Eval(ev);
Shinichiro Hamajia6a17a42015-06-18 20:11:19 +0900150 vector<Pattern> pats;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900151 for (StringPiece pat : WordScanner(pat_buf)) {
Shinichiro Hamajia6a17a42015-06-18 20:11:19 +0900152 pats.push_back(Pattern(pat));
153 }
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900154 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900155 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamajia6a17a42015-06-18 20:11:19 +0900156 for (const Pattern& pat : pats) {
157 if (pat.Match(tok)) {
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900158 ww.Write(tok);
159 break;
160 }
161 }
162 }
163}
164
165void FilterOutFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900166 const string&& pat_buf = args[0]->Eval(ev);
167 const string&& text = args[1]->Eval(ev);
Shinichiro Hamajia6a17a42015-06-18 20:11:19 +0900168 vector<Pattern> pats;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900169 for (StringPiece pat : WordScanner(pat_buf)) {
Shinichiro Hamajia6a17a42015-06-18 20:11:19 +0900170 pats.push_back(Pattern(pat));
171 }
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900172 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900173 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900174 bool matched = false;
Shinichiro Hamajia6a17a42015-06-18 20:11:19 +0900175 for (const Pattern& pat : pats) {
176 if (pat.Match(tok)) {
Shinichiro Hamaji00cc6582015-06-17 18:12:46 +0900177 matched = true;
178 break;
179 }
180 }
181 if (!matched)
182 ww.Write(tok);
183 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900184}
185
Shinichiro Hamajid87e59e2015-06-17 18:18:34 +0900186void SortFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamaji14436892016-02-12 16:32:42 +0900187 string list;
188 args[0]->Eval(ev, &list);
189 COLLECT_STATS("func sort time");
Shinichiro Hamaji2d353a02016-02-15 18:24:42 +0900190 // TODO(hamaji): Probably we could use a faster string-specific sort
191 // algorithm.
Shinichiro Hamajid87e59e2015-06-17 18:18:34 +0900192 vector<StringPiece> toks;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900193 WordScanner(list).Split(&toks);
Shinichiro Hamaji14436892016-02-12 16:32:42 +0900194 stable_sort(toks.begin(), toks.end());
Shinichiro Hamajid87e59e2015-06-17 18:18:34 +0900195 WordWriter ww(s);
196 StringPiece prev;
197 for (StringPiece tok : toks) {
198 if (prev != tok) {
199 ww.Write(tok);
200 prev = tok;
201 }
202 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900203}
204
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900205static int GetNumericValueForFunc(const string& buf) {
206 StringPiece s = TrimLeftSpace(buf);
207 char* end;
208 long n = strtol(s.data(), &end, 10);
209 if (n < 0 || n == LONG_MAX || s.data() + s.size() != end) {
210 return -1;
211 }
212 return n;
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900213}
214
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900215void WordFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900216 const string&& n_str = args[0]->Eval(ev);
217 int n = GetNumericValueForFunc(n_str);
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900218 if (n < 0) {
219 ev->Error(StringPrintf(
220 "*** non-numeric first argument to `word' function: '%s'.",
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900221 n_str.c_str()));
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900222 }
223 if (n == 0) {
224 ev->Error("*** first argument to `word' function must be greater than 0.");
225 }
226
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900227 const string&& text = args[1]->Eval(ev);
228 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900229 n--;
230 if (n == 0) {
231 AppendString(tok, s);
232 break;
233 }
234 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900235}
236
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900237void WordlistFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900238 const string&& s_str = args[0]->Eval(ev);
239 int si = GetNumericValueForFunc(s_str);
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900240 if (si < 0) {
241 ev->Error(StringPrintf(
242 "*** non-numeric first argument to `wordlist' function: '%s'.",
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900243 s_str.c_str()));
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900244 }
245 if (si == 0) {
246 ev->Error(StringPrintf(
247 "*** invalid first argument to `wordlist' function: %s`",
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900248 s_str.c_str()));
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900249 }
250
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900251 const string&& e_str = args[1]->Eval(ev);
252 int ei = GetNumericValueForFunc(e_str);
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900253 if (ei < 0) {
254 ev->Error(StringPrintf(
255 "*** non-numeric second argument to `wordlist' function: '%s'.",
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900256 e_str.c_str()));
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900257 }
258
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900259 const string&& text = args[2]->Eval(ev);
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900260 int i = 0;
261 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900262 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900263 i++;
264 if (si <= i && i <= ei) {
265 ww.Write(tok);
266 }
267 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900268}
269
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900270void WordsFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900271 const string&& text = args[0]->Eval(ev);
272 WordScanner ws(text);
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900273 int n = 0;
274 for (auto iter = ws.begin(); iter != ws.end(); ++iter)
275 n++;
276 char buf[32];
277 sprintf(buf, "%d", n);
278 *s += buf;
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900279}
280
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900281void FirstwordFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900282 const string&& text = args[0]->Eval(ev);
283 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900284 AppendString(tok, s);
285 return;
286 }
287}
288
289void LastwordFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900290 const string&& text = args[0]->Eval(ev);
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900291 StringPiece last;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900292 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamajid5271452015-06-17 18:50:03 +0900293 last = tok;
294 }
295 AppendString(last, s);
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900296}
297
Shinichiro Hamaji30b8e602015-06-17 19:28:54 +0900298void JoinFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900299 const string&& list1 = args[0]->Eval(ev);
300 const string&& list2 = args[1]->Eval(ev);
301 WordScanner ws1(list1);
302 WordScanner ws2(list2);
Shinichiro Hamaji30b8e602015-06-17 19:28:54 +0900303 WordWriter ww(s);
Shinichiro Hamaji71e79342016-06-07 14:05:48 +0900304 WordScanner::Iterator iter1, iter2;
305 for (iter1 = ws1.begin(), iter2 = ws2.begin();
Shinichiro Hamaji30b8e602015-06-17 19:28:54 +0900306 iter1 != ws1.end() && iter2 != ws2.end();
307 ++iter1, ++iter2) {
308 ww.Write(*iter1);
309 // Use |AppendString| not to append extra ' '.
310 AppendString(*iter2, s);
311 }
Shinichiro Hamaji71e79342016-06-07 14:05:48 +0900312 for (; iter1 != ws1.end(); ++iter1)
313 ww.Write(*iter1);
314 for (; iter2 != ws2.end(); ++iter2)
315 ww.Write(*iter2);
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900316}
317
Shinichiro Hamajifcf1b762015-06-18 03:43:54 +0900318void WildcardFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900319 const string&& pat = args[0]->Eval(ev);
Shinichiro Hamajib67a2992015-11-07 14:11:09 +0900320 COLLECT_STATS("func wildcard time");
Shinichiro Hamaji7409aee2015-07-28 14:52:37 +0900321 // Note GNU make does not delay the execution of $(wildcard) so we
322 // do not need to check avoid_io here.
Shinichiro Hamajifcf1b762015-06-18 03:43:54 +0900323 WordWriter ww(s);
Shinichiro Hamaji0e3873a2015-07-05 15:48:28 +0900324 vector<string>* files;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900325 for (StringPiece tok : WordScanner(pat)) {
Shinichiro Hamaji8f68bd32015-06-18 11:01:51 +0900326 ScopedTerminator st(tok);
Shinichiro Hamaji0e3873a2015-07-05 15:48:28 +0900327 Glob(tok.data(), &files);
Shinichiro Hamaji0e3873a2015-07-05 15:48:28 +0900328 for (const string& file : *files) {
Shinichiro Hamajife97c412015-06-29 16:57:57 +0900329 ww.Write(file);
330 }
Shinichiro Hamaji284f3d12015-06-17 19:29:01 +0900331 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900332}
333
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900334void DirFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900335 const string&& text = args[0]->Eval(ev);
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900336 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900337 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900338 ww.Write(Dirname(tok));
Shinichiro Hamaji55906852015-06-29 16:40:33 +0900339 s->push_back('/');
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900340 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900341}
342
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900343void NotdirFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900344 const string&& text = args[0]->Eval(ev);
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900345 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900346 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900347 if (tok == "/") {
Shinichiro Hamaji388e8582015-07-03 16:51:46 +0900348 ww.Write(StringPiece(""));
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900349 } else {
350 ww.Write(Basename(tok));
351 }
352 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900353}
354
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900355void SuffixFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900356 const string&& text = args[0]->Eval(ev);
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900357 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900358 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900359 StringPiece suf = GetExt(tok);
360 if (!suf.empty())
361 ww.Write(suf);
362 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900363}
364
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900365void BasenameFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900366 const string&& text = args[0]->Eval(ev);
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900367 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900368 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji67f9a702015-06-18 06:00:57 +0900369 ww.Write(StripExt(tok));
370 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900371}
372
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900373void AddsuffixFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900374 const string&& suf = args[0]->Eval(ev);
375 const string&& text = args[1]->Eval(ev);
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900376 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900377 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900378 ww.Write(tok);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900379 *s += suf;
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900380 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900381}
382
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900383void AddprefixFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900384 const string&& pre = args[0]->Eval(ev);
385 const string&& text = args[1]->Eval(ev);
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900386 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900387 for (StringPiece tok : WordScanner(text)) {
388 ww.Write(pre);
Shinichiro Hamaji5d694f02015-06-18 06:05:36 +0900389 AppendString(tok, s);
390 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900391}
392
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900393void RealpathFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900394 const string&& text = args[0]->Eval(ev);
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900395 if (ev->avoid_io()) {
Shinichiro Hamaji71cf60b2015-10-08 18:00:07 +0900396 *s += "$(";
397 string kati_binary;
398 GetExecutablePath(&kati_binary);
399 *s += kati_binary;
400 *s += " --realpath ";
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900401 *s += text;
Shinichiro Hamaji65dce542015-07-28 15:39:03 +0900402 *s += " 2> /dev/null)";
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900403 return;
404 }
405
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900406 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900407 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji8f68bd32015-06-18 11:01:51 +0900408 ScopedTerminator st(tok);
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900409 char buf[PATH_MAX];
410 if (realpath(tok.data(), buf))
Shinichiro Hamajicb4724e2015-10-08 17:52:49 +0900411 ww.Write(buf);
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900412 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900413}
414
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900415void AbspathFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900416 const string&& text = args[0]->Eval(ev);
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900417 WordWriter ww(s);
418 string buf;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900419 for (StringPiece tok : WordScanner(text)) {
Shinichiro Hamaji8a963582015-06-18 07:05:58 +0900420 AbsPath(tok, &buf);
421 ww.Write(buf);
422 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900423}
424
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900425void IfFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900426 const string&& cond = args[0]->Eval(ev);
427 if (cond.empty()) {
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900428 if (args.size() > 2)
429 args[2]->Eval(ev, s);
430 } else {
431 args[1]->Eval(ev, s);
432 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900433}
434
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900435void AndFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900436 string cond;
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900437 for (Value* a : args) {
438 cond = a->Eval(ev);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900439 if (cond.empty())
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900440 return;
441 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900442 if (!cond.empty()) {
443 *s += cond;
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900444 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900445}
446
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900447void OrFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
448 for (Value* a : args) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900449 const string&& cond = a->Eval(ev);
450 if (!cond.empty()) {
451 *s += cond;
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900452 return;
453 }
454 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900455}
456
Shinichiro Hamaji81699be2015-06-22 18:07:38 +0900457void ValueFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900458 const string&& var_name = args[0]->Eval(ev);
459 Var* var = ev->LookupVar(Intern(var_name));
Shinichiro Hamaji81699be2015-06-22 18:07:38 +0900460 AppendString(var->String().as_string(), s);
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900461}
462
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900463void EvalFunc(const vector<Value*>& args, Evaluator* ev, string*) {
Shinichiro Hamaji76ff9832015-06-18 17:11:22 +0900464 // TODO: eval leaks everything... for now.
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900465 //const string text = args[0]->Eval(ev);
Shinichiro Hamaji76ff9832015-06-18 17:11:22 +0900466 string* text = new string;
467 args[0]->Eval(ev, text);
Shinichiro Hamaji644d6b92015-11-17 14:47:56 +0900468 if (ev->avoid_io()) {
Dan Willemsene41c7552017-02-22 14:31:16 -0800469 KATI_WARN_LOC(ev->loc(), "*warning*: $(eval) in a recipe is not recommended: %s",
470 text->c_str());
Shinichiro Hamaji644d6b92015-11-17 14:47:56 +0900471 }
Shinichiro Hamaji645cca72015-09-24 17:04:21 +0900472 vector<Stmt*> stmts;
473 Parse(*text, ev->loc(), &stmts);
474 for (Stmt* stmt : stmts) {
475 LOG("%s", stmt->DebugString().c_str());
476 stmt->Eval(ev);
477 //delete stmt;
Shinichiro Hamaji80456fb2015-06-18 14:56:10 +0900478 }
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900479}
480
Shinichiro Hamaji5f57a992015-06-30 19:39:39 +0900481//#define TEST_FIND_EMULATOR
482
Shinichiro Hamaji68e712b2015-07-17 06:11:08 +0900483// A hack for Android build. We need to evaluate things like $((3+4))
484// when we emit ninja file, because the result of such expressions
485// will be passed to other make functions.
486// TODO: Maybe we should introduce a helper binary which evaluate
487// make expressions at ninja-time.
488static bool HasNoIoInShellScript(const string& cmd) {
489 if (cmd.empty())
490 return true;
491 if (HasPrefix(cmd, "echo $((") && cmd[cmd.size()-1] == ')')
492 return true;
493 return false;
494}
495
Dan Willemsen064be222016-09-30 20:17:14 -0700496static void ShellFuncImpl(const string& shell, const string& shellflag,
Dan Willemsen692e64e2017-02-22 15:38:33 -0800497 const string& cmd, const Loc& loc, string* s,
498 FindCommand** fc) {
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900499 LOG("ShellFunc: %s", cmd.c_str());
500
501#ifdef TEST_FIND_EMULATOR
502 bool need_check = false;
503 string out2;
504#endif
505 if (FindEmulator::Get()) {
506 *fc = new FindCommand();
507 if ((*fc)->Parse(cmd)) {
508#ifdef TEST_FIND_EMULATOR
Dan Willemsen692e64e2017-02-22 15:38:33 -0800509 if (FindEmulator::Get()->HandleFind(cmd, **fc, loc, &out2)) {
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900510 need_check = true;
511 }
512#else
Dan Willemsen692e64e2017-02-22 15:38:33 -0800513 if (FindEmulator::Get()->HandleFind(cmd, **fc, loc, s)) {
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900514 return;
515 }
516#endif
517 }
518 delete *fc;
519 *fc = NULL;
520 }
521
522 COLLECT_STATS_WITH_SLOW_REPORT("func shell time", cmd.c_str());
Dan Willemsen064be222016-09-30 20:17:14 -0700523 RunCommand(shell, shellflag, cmd, RedirectStderr::NONE, s);
Dan Willemsen48d6e8c2015-08-05 14:38:34 -0700524 FormatForCommandSubstitution(s);
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900525
526#ifdef TEST_FIND_EMULATOR
527 if (need_check) {
Dan Willemsen48d6e8c2015-08-05 14:38:34 -0700528 if (*s != out2) {
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900529 ERROR("FindEmulator is broken: %s\n%s\nvs\n%s",
Dan Willemsen48d6e8c2015-08-05 14:38:34 -0700530 cmd.c_str(), s->c_str(), out2.c_str());
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900531 }
532 }
533#endif
534}
535
Shinichiro Hamaji4db9edc2015-08-13 16:43:02 +0900536static vector<CommandResult*> g_command_results;
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900537
Shinichiro Hamaji180b4092015-08-13 17:11:18 +0900538bool ShouldStoreCommandResult(StringPiece cmd) {
Dan Willemsen40e5a3d2016-09-16 19:35:32 -0700539 // We really just want to ignore this one, or remove BUILD_DATETIME from
540 // Android completely
541 if (cmd == "date +%s")
Shinichiro Hamaji180b4092015-08-13 17:11:18 +0900542 return false;
Colin Crossf23ae8c2015-11-12 17:05:32 -0800543
544 Pattern pat(g_flags.ignore_dirty_pattern);
545 Pattern nopat(g_flags.no_ignore_dirty_pattern);
546 for (StringPiece tok : WordScanner(cmd)) {
547 if (pat.Match(tok) && !nopat.Match(tok)) {
548 return false;
Shinichiro Hamaji180b4092015-08-13 17:11:18 +0900549 }
550 }
Colin Crossf23ae8c2015-11-12 17:05:32 -0800551
Shinichiro Hamaji180b4092015-08-13 17:11:18 +0900552 return true;
553}
554
Shinichiro Hamajifcf1b762015-06-18 03:43:54 +0900555void ShellFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900556 string cmd = args[0]->Eval(ev);
557 if (ev->avoid_io() && !HasNoIoInShellScript(cmd)) {
Shinichiro Hamaji28da2372015-11-30 19:03:53 +0900558 if (ev->eval_depth() > 1) {
Dan Willemsene41c7552017-02-22 14:31:16 -0800559 ERROR_LOC(ev->loc(), "kati doesn't support passing results of $(shell) "
560 "to other make constructs: %s",
561 cmd.c_str());
Shinichiro Hamaji28da2372015-11-30 19:03:53 +0900562 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900563 StripShellComment(&cmd);
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900564 *s += "$(";
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900565 *s += cmd;
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900566 *s += ")";
567 return;
568 }
569
Dan Willemsen064be222016-09-30 20:17:14 -0700570 const string&& shell = ev->GetShell();
571 const string&& shellflag = ev->GetShellFlag();
Shinichiro Hamajifcf1b762015-06-18 03:43:54 +0900572
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900573 string out;
574 FindCommand* fc = NULL;
Dan Willemsen692e64e2017-02-22 15:38:33 -0800575 ShellFuncImpl(shell, shellflag, cmd, ev->loc(), &out, &fc);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900576 if (ShouldStoreCommandResult(cmd)) {
Shinichiro Hamaji4db9edc2015-08-13 16:43:02 +0900577 CommandResult* cr = new CommandResult();
Dan Willemsenf06d8012016-10-03 00:16:07 -0700578 cr->op = (fc == NULL) ? CommandOp::SHELL : CommandOp::FIND,
Stefan Beckerd4f28712016-04-07 13:29:36 +0300579 cr->shell = shell;
Dan Willemsen064be222016-09-30 20:17:14 -0700580 cr->shellflag = shellflag;
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900581 cr->cmd = cmd;
Shinichiro Hamaji4db9edc2015-08-13 16:43:02 +0900582 cr->find.reset(fc);
583 cr->result = out;
584 g_command_results.push_back(cr);
Shinichiro Hamajifcf1b762015-06-18 03:43:54 +0900585 }
586 *s += out;
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900587}
588
Shinichiro Hamaji3064f1f2015-06-18 16:32:09 +0900589void CallFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajif3c9bbc2016-02-08 18:47:26 +0900590 static const Symbol tmpvar_names[] = {
591 Intern("0"), Intern("1"), Intern("2"), Intern("3"), Intern("4"),
592 Intern("5"), Intern("6"), Intern("7"), Intern("8"), Intern("9")
Shinichiro Hamaji3064f1f2015-06-18 16:32:09 +0900593 };
594
Shinichiro Hamajid8d43ea2016-04-27 16:10:47 +0900595 const string&& func_name_buf = args[0]->Eval(ev);
596 const StringPiece func_name = TrimSpace(func_name_buf);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900597 Var* func = ev->LookupVar(Intern(func_name));
Shinichiro Hamaji433ad992015-11-17 15:43:07 +0900598 if (!func->IsDefined()) {
Dan Willemsene41c7552017-02-22 14:31:16 -0800599 KATI_WARN_LOC(ev->loc(), "*warning*: undefined user function: %s",
600 func_name.as_string().c_str());
Shinichiro Hamaji433ad992015-11-17 15:43:07 +0900601 }
Shinichiro Hamaji3064f1f2015-06-18 16:32:09 +0900602 vector<unique_ptr<SimpleVar>> av;
Shinichiro Hamaji3064f1f2015-06-18 16:32:09 +0900603 for (size_t i = 1; i < args.size(); i++) {
Shinichiro Hamajif62e9a72015-06-26 04:18:21 +0900604 unique_ptr<SimpleVar> s(
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900605 new SimpleVar(args[i]->Eval(ev), VarOrigin::AUTOMATIC));
Shinichiro Hamaji3064f1f2015-06-18 16:32:09 +0900606 av.push_back(move(s));
607 }
Shinichiro Hamajic9b9e5e2016-02-18 18:18:54 +0900608 vector<unique_ptr<ScopedGlobalVar>> sv;
Dan Willemsenb4467962015-08-06 23:59:22 -0700609 for (size_t i = 1; ; i++) {
Colin Cross580cc1f2015-07-27 13:18:18 -0700610 string s;
Shinichiro Hamajif3c9bbc2016-02-08 18:47:26 +0900611 Symbol tmpvar_name_sym(Symbol::IsUninitialized{});
Colin Cross580cc1f2015-07-27 13:18:18 -0700612 if (i < sizeof(tmpvar_names)/sizeof(tmpvar_names[0])) {
Shinichiro Hamajif3c9bbc2016-02-08 18:47:26 +0900613 tmpvar_name_sym = tmpvar_names[i];
Colin Cross580cc1f2015-07-27 13:18:18 -0700614 } else {
615 s = StringPrintf("%d", i);
Shinichiro Hamajif3c9bbc2016-02-08 18:47:26 +0900616 tmpvar_name_sym = Intern(s);
Colin Cross580cc1f2015-07-27 13:18:18 -0700617 }
Dan Willemsenb4467962015-08-06 23:59:22 -0700618 if (i < args.size()) {
Shinichiro Hamajic9b9e5e2016-02-18 18:18:54 +0900619 sv.emplace_back(new ScopedGlobalVar(tmpvar_name_sym, av[i-1].get()));
Dan Willemsenb4467962015-08-06 23:59:22 -0700620 } else {
621 // We need to blank further automatic vars
Shinichiro Hamajif3c9bbc2016-02-08 18:47:26 +0900622 Var *v = ev->LookupVar(tmpvar_name_sym);
Dan Willemsenb4467962015-08-06 23:59:22 -0700623 if (!v->IsDefined()) break;
624 if (v->Origin() != VarOrigin::AUTOMATIC) break;
625
Shinichiro Hamajie978a892015-08-17 16:53:40 +0900626 av.emplace_back(new SimpleVar("", VarOrigin::AUTOMATIC));
Shinichiro Hamajic9b9e5e2016-02-18 18:18:54 +0900627 sv.emplace_back(new ScopedGlobalVar(tmpvar_name_sym, av[i-1].get()));
Dan Willemsenb4467962015-08-06 23:59:22 -0700628 }
Shinichiro Hamaji5d53bc72015-06-26 08:31:54 +0900629 }
Shinichiro Hamaji28da2372015-11-30 19:03:53 +0900630
631 ev->DecrementEvalDepth();
Shinichiro Hamaji3064f1f2015-06-18 16:32:09 +0900632 func->Eval(ev, s);
Shinichiro Hamaji28da2372015-11-30 19:03:53 +0900633 ev->IncrementEvalDepth();
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900634}
635
Shinichiro Hamajicf0cd682015-06-18 16:18:13 +0900636void ForeachFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900637 const string&& varname = args[0]->Eval(ev);
638 const string&& list = args[1]->Eval(ev);
Shinichiro Hamaji28da2372015-11-30 19:03:53 +0900639 ev->DecrementEvalDepth();
Shinichiro Hamajicf0cd682015-06-18 16:18:13 +0900640 WordWriter ww(s);
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900641 for (StringPiece tok : WordScanner(list)) {
Shinichiro Hamajicf0cd682015-06-18 16:18:13 +0900642 unique_ptr<SimpleVar> v(new SimpleVar(
Shinichiro Hamaji5081c712015-08-14 16:49:20 +0900643 tok.as_string(), VarOrigin::AUTOMATIC));
Shinichiro Hamajic9b9e5e2016-02-18 18:18:54 +0900644 ScopedGlobalVar sv(Intern(varname), v.get());
Shinichiro Hamajicf0cd682015-06-18 16:18:13 +0900645 ww.MaybeAddWhitespace();
646 args[2]->Eval(ev, s);
647 }
Shinichiro Hamaji28da2372015-11-30 19:03:53 +0900648 ev->IncrementEvalDepth();
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900649}
650
Shinichiro Hamajic22fdb42015-06-22 18:10:40 +0900651void OriginFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900652 const string&& var_name = args[0]->Eval(ev);
653 Var* var = ev->LookupVar(Intern(var_name));
Shinichiro Hamajif62e9a72015-06-26 04:18:21 +0900654 *s += GetOriginStr(var->Origin());
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900655}
656
Shinichiro Hamajic22fdb42015-06-22 18:10:40 +0900657void FlavorFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900658 const string&& var_name = args[0]->Eval(ev);
659 Var* var = ev->LookupVar(Intern(var_name));
Shinichiro Hamajic22fdb42015-06-22 18:10:40 +0900660 *s += var->Flavor();
Shinichiro Hamaji4f22f5c2015-06-16 16:28:25 +0900661}
662
Shinichiro Hamaji86e11332015-07-28 15:24:17 +0900663void InfoFunc(const vector<Value*>& args, Evaluator* ev, string*) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900664 const string&& a = args[0]->Eval(ev);
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900665 if (ev->avoid_io()) {
Dan Willemsene6f68582015-08-21 11:13:28 -0700666 ev->add_delayed_output_command(StringPrintf("echo -e \"%s\"", EchoEscape(a).c_str()));
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900667 return;
668 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900669 printf("%s\n", a.c_str());
Shinichiro Hamaji776ca302015-06-06 03:52:48 +0900670 fflush(stdout);
671}
672
Shinichiro Hamaji86e11332015-07-28 15:24:17 +0900673void WarningFunc(const vector<Value*>& args, Evaluator* ev, string*) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900674 const string&& a = args[0]->Eval(ev);
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900675 if (ev->avoid_io()) {
Shinichiro Hamaji86e11332015-07-28 15:24:17 +0900676 ev->add_delayed_output_command(
Dan Willemsene6f68582015-08-21 11:13:28 -0700677 StringPrintf("echo -e \"%s:%d: %s\" 2>&1", LOCF(ev->loc()), EchoEscape(a).c_str()));
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900678 return;
679 }
Dan Willemsene41c7552017-02-22 14:31:16 -0800680 WARN_LOC(ev->loc(), "%s", a.c_str());
Shinichiro Hamaji9619b362015-06-16 16:13:25 +0900681}
682
Shinichiro Hamaji86e11332015-07-28 15:24:17 +0900683void ErrorFunc(const vector<Value*>& args, Evaluator* ev, string*) {
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900684 const string&& a = args[0]->Eval(ev);
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900685 if (ev->avoid_io()) {
Shinichiro Hamaji86e11332015-07-28 15:24:17 +0900686 ev->add_delayed_output_command(
Dan Willemsene6f68582015-08-21 11:13:28 -0700687 StringPrintf("echo -e \"%s:%d: *** %s.\" 2>&1 && false",
688 LOCF(ev->loc()), EchoEscape(a).c_str()));
Shinichiro Hamajidf1fc8b2015-06-29 15:45:21 +0900689 return;
690 }
Shinichiro Hamajifb415ad2015-08-14 17:19:34 +0900691 ev->Error(StringPrintf("*** %s.", a.c_str()));
Shinichiro Hamaji9619b362015-06-16 16:13:25 +0900692}
693
Dan Willemsenf06d8012016-10-03 00:16:07 -0700694static void FileReadFunc(Evaluator* ev, const string& filename, string* s) {
695 int fd = open(filename.c_str(), O_RDONLY);
696 if (fd < 0) {
697 if (errno == ENOENT) {
698 if (ShouldStoreCommandResult(filename)) {
699 CommandResult* cr = new CommandResult();
700 cr->op = CommandOp::READ_MISSING;
701 cr->cmd = filename;
702 g_command_results.push_back(cr);
703 }
704 return;
705 } else {
706 ev->Error("*** open failed.");
707 }
708 }
709
710 struct stat st;
711 if (fstat(fd, &st) < 0) {
712 ev->Error("*** fstat failed.");
713 }
714
715 size_t len = st.st_size;
716 string out;
717 out.resize(len);
718 ssize_t r = HANDLE_EINTR(read(fd, &out[0], len));
719 if (r != static_cast<ssize_t>(len)) {
720 ev->Error("*** read failed.");
721 }
722
723 if (close(fd) < 0) {
724 ev->Error("*** close failed.");
725 }
726
727 if (out.back() == '\n') {
728 out.pop_back();
729 }
730
731 if (ShouldStoreCommandResult(filename)) {
732 CommandResult* cr = new CommandResult();
733 cr->op = CommandOp::READ;
734 cr->cmd = filename;
735 g_command_results.push_back(cr);
736 }
737 *s += out;
738}
739
740static void FileWriteFunc(Evaluator* ev, const string& filename, bool append, string text) {
741 FILE* f = fopen(filename.c_str(), append ? "ab" : "wb");
742 if (f == NULL) {
743 ev->Error("*** fopen failed.");
744 }
745
746 if (fwrite(&text[0], text.size(), 1, f) != 1) {
747 ev->Error("*** fwrite failed.");
748 }
749
750 if (fclose(f) != 0) {
751 ev->Error("*** fclose failed.");
752 }
753
754 if (ShouldStoreCommandResult(filename)) {
755 CommandResult* cr = new CommandResult();
756 cr->op = CommandOp::WRITE;
757 cr->cmd = filename;
758 cr->result = text;
759 g_command_results.push_back(cr);
760 }
761}
762
763void FileFunc(const vector<Value*>& args, Evaluator* ev, string* s) {
764 if (ev->avoid_io()) {
765 ev->Error("*** $(file ...) is not supported in rules.");
766 }
767
768 string arg = args[0]->Eval(ev);
769 StringPiece filename = TrimSpace(arg);
770
771 if (filename.size() <= 1) {
772 ev->Error("*** Missing filename");
773 }
774
775 if (filename[0] == '<') {
776 filename = TrimLeftSpace(filename.substr(1));
777 if (!filename.size()) {
778 ev->Error("*** Missing filename");
779 }
780 if (args.size() > 1) {
781 ev->Error("*** invalid argument");
782 }
783
784 FileReadFunc(ev, filename.as_string(), s);
785 } else if (filename[0] == '>') {
786 bool append = false;
787 if (filename[1] == '>') {
788 append = true;
789 filename = filename.substr(2);
790 } else {
791 filename = filename.substr(1);
792 }
793 filename = TrimLeftSpace(filename);
794 if (!filename.size()) {
795 ev->Error("*** Missing filename");
796 }
797
798 string text;
799 if (args.size() > 1) {
800 text = args[1]->Eval(ev);
801 if (text.size() == 0 || text.back() != '\n') {
802 text.push_back('\n');
803 }
804 }
805
806 FileWriteFunc(ev, filename.as_string(), append, text);
807 } else {
808 ev->Error(StringPrintf("*** Invalid file operation: %s. Stop.", filename.as_string().c_str()));
809 }
810}
811
Shinichiro Hamaji776ca302015-06-06 03:52:48 +0900812FuncInfo g_func_infos[] = {
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900813 { "patsubst", &PatsubstFunc, 3, 3, false, false },
814 { "strip", &StripFunc, 1, 1, false, false },
815 { "subst", &SubstFunc, 3, 3, false, false },
816 { "findstring", &FindstringFunc, 2, 2, false, false },
817 { "filter", &FilterFunc, 2, 2, false, false },
818 { "filter-out", &FilterOutFunc, 2, 2, false, false },
819 { "sort", &SortFunc, 1, 1, false, false },
820 { "word", &WordFunc, 2, 2, false, false },
821 { "wordlist", &WordlistFunc, 3, 3, false, false },
822 { "words", &WordsFunc, 1, 1, false, false },
823 { "firstword", &FirstwordFunc, 1, 1, false, false },
824 { "lastword", &LastwordFunc, 1, 1, false, false },
Shinichiro Hamaji2e6cbfc2015-06-16 18:46:50 +0900825
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900826 { "join", &JoinFunc, 2, 2, false, false },
827 { "wildcard", &WildcardFunc, 1, 1, false, false },
828 { "dir", &DirFunc, 1, 1, false, false },
829 { "notdir", &NotdirFunc, 1, 1, false, false },
830 { "suffix", &SuffixFunc, 1, 1, false, false },
831 { "basename", &BasenameFunc, 1, 1, false, false },
832 { "addsuffix", &AddsuffixFunc, 2, 2, false, false },
833 { "addprefix", &AddprefixFunc, 2, 2, false, false },
834 { "realpath", &RealpathFunc, 1, 1, false, false },
835 { "abspath", &AbspathFunc, 1, 1, false, false },
Shinichiro Hamaji30b8e602015-06-17 19:28:54 +0900836
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900837 { "if", &IfFunc, 3, 2, false, true },
838 { "and", &AndFunc, 0, 0, true, false },
839 { "or", &OrFunc, 0, 0, true, false },
Shinichiro Hamajie22fe8e2015-06-18 07:11:54 +0900840
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900841 { "value", &ValueFunc, 1, 1, false, false },
842 { "eval", &EvalFunc, 1, 1, false, false },
843 { "shell", &ShellFunc, 1, 1, false, false },
844 { "call", &CallFunc, 0, 0, false, false },
845 { "foreach", &ForeachFunc, 3, 3, false, false },
Shinichiro Hamajie22fe8e2015-06-18 07:11:54 +0900846
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900847 { "origin", &OriginFunc, 1, 1, false, false },
848 { "flavor", &FlavorFunc, 1, 1, false, false },
Shinichiro Hamajie22fe8e2015-06-18 07:11:54 +0900849
Shinichiro Hamajifead3b72015-06-18 15:31:15 +0900850 { "info", &InfoFunc, 1, 1, false, false },
851 { "warning", &WarningFunc, 1, 1, false, false },
852 { "error", &ErrorFunc, 1, 1, false, false },
Dan Willemsenf06d8012016-10-03 00:16:07 -0700853
854 { "file", &FileFunc, 2, 1, false, false },
Shinichiro Hamaji776ca302015-06-06 03:52:48 +0900855};
856
857unordered_map<StringPiece, FuncInfo*>* g_func_info_map;
858
859} // namespace
860
861void InitFuncTable() {
862 g_func_info_map = new unordered_map<StringPiece, FuncInfo*>;
863 for (size_t i = 0; i < sizeof(g_func_infos) / sizeof(g_func_infos[0]); i++) {
864 FuncInfo* fi = &g_func_infos[i];
Shinichiro Hamajie7992752015-06-29 18:38:35 +0900865 bool ok = g_func_info_map->emplace(fi->name, fi).second;
Shinichiro Hamaji776ca302015-06-06 03:52:48 +0900866 CHECK(ok);
867 }
868}
869
870void QuitFuncTable() {
871 delete g_func_info_map;
872}
873
874FuncInfo* GetFuncInfo(StringPiece name) {
875 auto found = g_func_info_map->find(name);
876 if (found == g_func_info_map->end())
877 return NULL;
878 return found->second;
879}
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900880
Shinichiro Hamaji4db9edc2015-08-13 16:43:02 +0900881const vector<CommandResult*>& GetShellCommandResults() {
882 return g_command_results;
Shinichiro Hamajic9b0aca2015-07-31 16:47:56 +0900883}