// Copyright 2021 Google Inc. All rights reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package response import ( "io" "io/ioutil" "strings" "unicode" ) const noQuote = '\x00' // ReadRspFile reads a file in Ninja's response file format and returns its contents. func ReadRspFile(r io.Reader) ([]string, error) { var files []string var file []byte buf, err := ioutil.ReadAll(r) if err != nil { return nil, err } isEscaping := false quotingStart := byte(noQuote) for _, c := range buf { switch { case isEscaping: if quotingStart == '"' { if !(c == '"' || c == '\\') { // '\"' or '\\' will be escaped under double quoting. file = append(file, '\\') } } file = append(file, c) isEscaping = false case c == '\\' && quotingStart != '\'': isEscaping = true case quotingStart == noQuote && (c == '\'' || c == '"'): quotingStart = c case quotingStart != noQuote && c == quotingStart: quotingStart = noQuote case quotingStart == noQuote && unicode.IsSpace(rune(c)): // Current character is a space outside quotes if len(file) != 0 { files = append(files, string(file)) } file = file[:0] default: file = append(file, c) } } if len(file) != 0 { files = append(files, string(file)) } return files, nil } func rspUnsafeChar(r rune) bool { switch { case 'A' <= r && r <= 'Z', 'a' <= r && r <= 'z', '0' <= r && r <= '9', r == '_', r == '+', r == '-', r == '.', r == '/': return false default: return true } } var rspEscaper = strings.NewReplacer(`'`, `'\''`) // WriteRspFile writes a list of files to a file in Ninja's response file format. func WriteRspFile(w io.Writer, files []string) error { for i, f := range files { if i != 0 { _, err := io.WriteString(w, " ") if err != nil { return err } } if strings.IndexFunc(f, rspUnsafeChar) != -1 { f = `'` + rspEscaper.Replace(f) + `'` } _, err := io.WriteString(w, f) if err != nil { return err } } return nil }