diff options
Diffstat (limited to 'Source/cmFileCommand.cxx')
| -rw-r--r-- | Source/cmFileCommand.cxx | 28 |
1 files changed, 25 insertions, 3 deletions
diff --git a/Source/cmFileCommand.cxx b/Source/cmFileCommand.cxx index d5d1eb2..a97802c 100644 --- a/Source/cmFileCommand.cxx +++ b/Source/cmFileCommand.cxx @@ -190,9 +190,10 @@ bool cmFileCommand::HandleWriteCommand(std::vector<std::string> const& args, //---------------------------------------------------------------------------- bool cmFileCommand::HandleReadCommand(std::vector<std::string> const& args) { - if ( args.size() != 3 ) + if ( args.size() < 3 ) { - this->SetError("READ must be called with two additional arguments"); + this->SetError("READ must be called with at least two additional " + "arguments"); return false; } @@ -214,11 +215,32 @@ bool cmFileCommand::HandleReadCommand(std::vector<std::string> const& args) return false; } + // if there a limit? + long sizeLimit = -1; + if (args.size() >= 5 && args[3] == "LIMIT") + { + sizeLimit = atof(args[4].c_str()); + } + std::string output; std::string line; bool has_newline = false; - while ( cmSystemTools::GetLineFromStream(file, line, &has_newline) ) + while (sizeLimit != 0 && + cmSystemTools::GetLineFromStream(file, line, &has_newline, + sizeLimit) ) { + if (sizeLimit > 0) + { + sizeLimit -= line.size(); + if (has_newline) + { + sizeLimit--; + } + if (sizeLimit < 0) + { + sizeLimit = 0; + } + } output += line; if ( has_newline ) { |