Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 1 | package main |
| 2 | |
| 3 | import ( |
| 4 | "bufio" |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 5 | "bytes" |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 6 | "errors" |
Fumitoshi Ukai | f8efa0a | 2015-03-30 18:10:11 +0900 | [diff] [blame] | 7 | "fmt" |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 8 | "io" |
| 9 | "os" |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 10 | "strings" |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 11 | ) |
| 12 | |
| 13 | type Makefile struct { |
| 14 | stmts []AST |
| 15 | } |
| 16 | |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 17 | type ifState struct { |
| 18 | ast *IfAST |
| 19 | in_else bool |
| 20 | } |
| 21 | |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 22 | type parser struct { |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 23 | rd *bufio.Reader |
| 24 | mk Makefile |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 25 | filename string |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 26 | lineno int |
| 27 | elineno int // lineno == elineno unless there is trailing '\'. |
| 28 | unBuf []byte |
| 29 | hasUnBuf bool |
| 30 | done bool |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 31 | outStmts *[]AST |
| 32 | ifStack []ifState |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 33 | } |
| 34 | |
| 35 | func exists(filename string) bool { |
| 36 | f, err := os.Open(filename) |
| 37 | if err != nil { |
| 38 | return false |
| 39 | } |
| 40 | f.Close() |
| 41 | return true |
| 42 | } |
| 43 | |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 44 | func newParser(rd io.Reader, filename string) *parser { |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 45 | p := &parser{ |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 46 | rd: bufio.NewReader(rd), |
| 47 | filename: filename, |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 48 | } |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 49 | p.outStmts = &p.mk.stmts |
| 50 | return p |
| 51 | } |
| 52 | |
Shinichiro Hamaji | ae32b78 | 2015-03-31 14:41:19 +0900 | [diff] [blame^] | 53 | func (p *parser) addStatement(ast AST) { |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 54 | *p.outStmts = append(*p.outStmts, ast) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 55 | } |
| 56 | |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 57 | func (p *parser) readLine() []byte { |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 58 | if p.hasUnBuf { |
| 59 | p.hasUnBuf = false |
| 60 | return p.unBuf |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 61 | } |
| 62 | |
| 63 | p.lineno = p.elineno |
| 64 | line, err := p.rd.ReadBytes('\n') |
| 65 | p.lineno++ |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 66 | p.elineno = p.lineno |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 67 | if err == io.EOF { |
| 68 | p.done = true |
| 69 | } else if err != nil { |
| 70 | panic(err) |
| 71 | } |
| 72 | |
| 73 | if len(line) > 0 { |
| 74 | line = line[0 : len(line)-1] |
| 75 | } |
| 76 | |
| 77 | // TODO: Handle \\ at the end of the line? |
| 78 | for len(line) > 0 && line[len(line)-1] == '\\' { |
| 79 | line = line[:len(line)-1] |
| 80 | nline := p.readLine() |
| 81 | p.elineno++ |
| 82 | line = append(line, nline...) |
| 83 | } |
| 84 | |
| 85 | index := bytes.IndexByte(line, '#') |
| 86 | if index >= 0 { |
| 87 | line = line[:index] |
| 88 | } |
| 89 | |
| 90 | return line |
| 91 | } |
| 92 | |
| 93 | func (p *parser) unreadLine(line []byte) { |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 94 | if p.hasUnBuf { |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 95 | panic("unreadLine twice!") |
| 96 | } |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 97 | p.unBuf = line |
| 98 | p.hasUnBuf = true |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 99 | } |
| 100 | |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 101 | func (p *parser) parseAssign(line []byte, sep, esep int) AST { |
| 102 | Log("parseAssign %s %s", line, line[sep:esep]) |
| 103 | ast := &AssignAST{ |
| 104 | lhs: string(bytes.TrimSpace(line[:sep])), |
| 105 | rhs: string(bytes.TrimLeft(line[esep:], " \t")), |
| 106 | op: string(line[sep:esep]), |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 107 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 108 | ast.filename = p.filename |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 109 | ast.lineno = p.lineno |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 110 | return ast |
| 111 | } |
| 112 | |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 113 | func (p *parser) parseRule(line []byte, sep int) AST { |
| 114 | lhs := string(bytes.TrimSpace(line[:sep])) |
| 115 | rhs := string(bytes.TrimSpace(line[sep+1:])) |
| 116 | ast := &RuleAST{ |
| 117 | lhs: lhs, |
| 118 | rhs: rhs, |
| 119 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 120 | ast.filename = p.filename |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 121 | ast.lineno = p.lineno |
Shinichiro Hamaji | 7c4e325 | 2015-03-30 23:04:25 +0900 | [diff] [blame] | 122 | ast.cmdLineno = p.elineno + 1 |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 123 | for { |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 124 | line := p.readLine() |
| 125 | if len(line) == 0 { |
| 126 | break |
| 127 | } else if line[0] == '\t' { |
| 128 | ast.cmds = append(ast.cmds, string(bytes.TrimSpace(line))) |
| 129 | } else { |
| 130 | p.unreadLine(line) |
| 131 | break |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 132 | } |
| 133 | } |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 134 | return ast |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 135 | } |
| 136 | |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 137 | func (p *parser) parseInclude(line string, oplen int) AST { |
| 138 | ast := &IncludeAST{ |
| 139 | expr: line[oplen+1:], |
| 140 | op: line[:oplen], |
| 141 | } |
| 142 | ast.filename = p.filename |
| 143 | ast.lineno = p.lineno |
| 144 | return ast |
| 145 | } |
| 146 | |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 147 | func (p *parser) parseIfdef(line string, oplen int) AST { |
| 148 | ast := &IfAST{ |
Shinichiro Hamaji | ae32b78 | 2015-03-31 14:41:19 +0900 | [diff] [blame^] | 149 | op: line[:oplen], |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 150 | lhs: strings.TrimSpace(line[oplen+1:]), |
| 151 | } |
| 152 | ast.filename = p.filename |
| 153 | ast.lineno = p.lineno |
| 154 | p.addStatement(ast) |
| 155 | p.ifStack = append(p.ifStack, ifState{ast: ast}) |
| 156 | p.outStmts = &ast.trueStmts |
| 157 | return ast |
| 158 | } |
| 159 | |
Shinichiro Hamaji | af1e816 | 2015-03-31 02:15:37 +0900 | [diff] [blame] | 160 | func (p *parser) parseEq(s string) (string, string, bool) { |
| 161 | if len(s) == 0 || s[0] != '(' { |
| 162 | return "", "", false |
| 163 | } |
| 164 | |
| 165 | // TODO: Double-quotes will not be handled properly. |
| 166 | i := 0 |
| 167 | paren_cnt := 0 |
| 168 | in_rhs := false |
| 169 | var lhs []byte |
| 170 | var rhs []byte |
| 171 | for { |
| 172 | i++ |
| 173 | if i == len(s) { |
| 174 | return "", "", false |
| 175 | } |
| 176 | ch := s[i] |
| 177 | if ch == '(' { |
| 178 | paren_cnt++ |
| 179 | } else if ch == ')' { |
| 180 | paren_cnt-- |
| 181 | if paren_cnt < 0 { |
| 182 | if in_rhs { |
| 183 | break |
| 184 | } else { |
| 185 | return "", "", false |
| 186 | } |
| 187 | } |
| 188 | } else if ch == ',' { |
| 189 | if in_rhs { |
| 190 | return "", "", false |
| 191 | } else { |
| 192 | in_rhs = true |
| 193 | continue |
| 194 | } |
| 195 | } |
| 196 | if in_rhs { |
| 197 | rhs = append(rhs, ch) |
| 198 | } else { |
| 199 | lhs = append(lhs, ch) |
| 200 | } |
| 201 | } |
| 202 | return string(lhs), string(rhs), true |
| 203 | } |
| 204 | |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 205 | func (p *parser) parseIfeq(line string, oplen int) AST { |
Shinichiro Hamaji | af1e816 | 2015-03-31 02:15:37 +0900 | [diff] [blame] | 206 | lhs, rhs, ok := p.parseEq(strings.TrimSpace(line[oplen+1:])) |
| 207 | if !ok { |
| 208 | Error(p.filename, p.lineno, `*** invalid syntax in conditional.`) |
| 209 | } |
| 210 | |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 211 | ast := &IfAST{ |
Shinichiro Hamaji | ae32b78 | 2015-03-31 14:41:19 +0900 | [diff] [blame^] | 212 | op: line[:oplen], |
Shinichiro Hamaji | af1e816 | 2015-03-31 02:15:37 +0900 | [diff] [blame] | 213 | lhs: lhs, |
| 214 | rhs: rhs, |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 215 | } |
| 216 | ast.filename = p.filename |
| 217 | ast.lineno = p.lineno |
| 218 | p.addStatement(ast) |
| 219 | p.ifStack = append(p.ifStack, ifState{ast: ast}) |
| 220 | p.outStmts = &ast.trueStmts |
| 221 | return ast |
| 222 | } |
| 223 | |
| 224 | func (p *parser) checkIfStack(curKeyword string) { |
| 225 | if len(p.ifStack) == 0 { |
| 226 | Error(p.filename, p.lineno, `*** extraneous %q.`, curKeyword) |
| 227 | } |
| 228 | } |
| 229 | |
| 230 | func (p *parser) parseElse(line string) { |
| 231 | p.checkIfStack("else") |
| 232 | state := &p.ifStack[len(p.ifStack)-1] |
| 233 | if state.in_else { |
| 234 | Error(p.filename, p.lineno, `*** only one "else" per conditional.`) |
| 235 | } |
| 236 | state.in_else = true |
| 237 | p.outStmts = &state.ast.falseStmts |
| 238 | } |
| 239 | |
| 240 | func (p *parser) parseEndif(line string) { |
| 241 | p.checkIfStack("endif") |
Shinichiro Hamaji | ae32b78 | 2015-03-31 14:41:19 +0900 | [diff] [blame^] | 242 | p.ifStack = p.ifStack[0 : len(p.ifStack)-1] |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 243 | if len(p.ifStack) == 0 { |
| 244 | p.outStmts = &p.mk.stmts |
| 245 | } else { |
| 246 | state := p.ifStack[len(p.ifStack)-1] |
| 247 | if state.in_else { |
| 248 | p.outStmts = &state.ast.falseStmts |
| 249 | } else { |
| 250 | p.outStmts = &state.ast.trueStmts |
| 251 | } |
| 252 | } |
| 253 | } |
| 254 | |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 255 | func (p *parser) parseLine(line string) AST { |
Shinichiro Hamaji | e3a9463 | 2015-03-31 01:12:52 +0900 | [diff] [blame] | 256 | stripped := strings.TrimLeft(line, " \t") |
| 257 | if strings.HasPrefix(stripped, "include ") { |
| 258 | return p.parseInclude(stripped, len("include")) |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 259 | } |
Shinichiro Hamaji | e3a9463 | 2015-03-31 01:12:52 +0900 | [diff] [blame] | 260 | if strings.HasPrefix(stripped, "-include ") { |
| 261 | return p.parseInclude(stripped, len("-include")) |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 262 | } |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 263 | if strings.HasPrefix(stripped, "ifdef ") { |
| 264 | p.parseIfdef(stripped, len("ifdef")) |
| 265 | return nil |
| 266 | } |
| 267 | if strings.HasPrefix(stripped, "ifndef ") { |
| 268 | p.parseIfdef(stripped, len("ifndef")) |
| 269 | return nil |
| 270 | } |
| 271 | if strings.HasPrefix(stripped, "ifeq ") { |
| 272 | p.parseIfeq(stripped, len("ifeq")) |
| 273 | return nil |
| 274 | } |
| 275 | if strings.HasPrefix(stripped, "ifneq ") { |
| 276 | p.parseIfeq(stripped, len("ifneq")) |
| 277 | return nil |
| 278 | } |
| 279 | if strings.HasPrefix(stripped, "else") { |
| 280 | p.parseElse(stripped) |
| 281 | return nil |
| 282 | } |
| 283 | if strings.HasPrefix(stripped, "endif") { |
| 284 | p.parseEndif(stripped) |
| 285 | return nil |
| 286 | } |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 287 | ast := &RawExprAST{expr: line} |
| 288 | ast.filename = p.filename |
| 289 | ast.lineno = p.lineno |
| 290 | return ast |
| 291 | } |
| 292 | |
Fumitoshi Ukai | f8efa0a | 2015-03-30 18:10:11 +0900 | [diff] [blame] | 293 | func (p *parser) parse() (mk Makefile, err error) { |
| 294 | defer func() { |
| 295 | if r := recover(); r != nil { |
| 296 | err = fmt.Errorf("panic: %v", r) |
| 297 | } |
| 298 | }() |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 299 | for !p.done { |
| 300 | line := p.readLine() |
| 301 | |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 302 | var ast AST |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 303 | for i, ch := range line { |
| 304 | switch ch { |
| 305 | case ':': |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 306 | if i+1 < len(line) && line[i+1] == '=' { |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 307 | ast = p.parseAssign(line, i, i+2) |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 308 | } else { |
| 309 | ast = p.parseRule(line, i) |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 310 | } |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 311 | case '=': |
Fumitoshi Ukai | e1b813c | 2015-03-30 18:38:21 +0900 | [diff] [blame] | 312 | ast = p.parseAssign(line, i, i+1) |
Shinichiro Hamaji | 69b7f65 | 2015-03-31 01:01:59 +0900 | [diff] [blame] | 313 | case '?', '+': |
| 314 | if i+1 < len(line) && line[i+1] == '=' { |
| 315 | ast = p.parseAssign(line, i, i+2) |
| 316 | } |
Shinichiro Hamaji | e184158 | 2015-03-30 17:20:33 +0900 | [diff] [blame] | 317 | } |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 318 | if ast != nil { |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 319 | p.addStatement(ast) |
Shinichiro Hamaji | 7f1ede3 | 2015-03-30 18:05:22 +0900 | [diff] [blame] | 320 | break |
| 321 | } |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 322 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 323 | if ast == nil && len(bytes.TrimSpace(line)) > 0 { |
Shinichiro Hamaji | d7bef60 | 2015-03-30 19:55:32 +0900 | [diff] [blame] | 324 | ast = p.parseLine(string(line)) |
Shinichiro Hamaji | 497754d | 2015-03-31 02:02:11 +0900 | [diff] [blame] | 325 | if ast != nil { |
| 326 | p.addStatement(ast) |
| 327 | } |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 328 | } |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 329 | } |
| 330 | return p.mk, nil |
| 331 | } |
| 332 | |
| 333 | func ParseMakefile(filename string) (Makefile, error) { |
| 334 | f, err := os.Open(filename) |
| 335 | if err != nil { |
| 336 | return Makefile{}, err |
| 337 | } |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 338 | defer f.Close() |
Shinichiro Hamaji | 685fecf | 2015-03-30 18:28:12 +0900 | [diff] [blame] | 339 | parser := newParser(f, filename) |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 340 | return parser.parse() |
| 341 | } |
| 342 | |
| 343 | func ParseDefaultMakefile() (Makefile, error) { |
| 344 | candidates := []string{"GNUmakefile", "makefile", "Makefile"} |
| 345 | for _, filename := range candidates { |
| 346 | if exists(filename) { |
| 347 | return ParseMakefile(filename) |
| 348 | } |
| 349 | } |
Fumitoshi Ukai | cf2b038 | 2015-03-30 17:48:54 +0900 | [diff] [blame] | 350 | return Makefile{}, errors.New("no targets specified and no makefile found.") |
Fumitoshi Ukai | 119dc91 | 2015-03-30 16:52:41 +0900 | [diff] [blame] | 351 | } |