diff options
author | Austin Wise <AustinWise@gmail.com> | 2021-07-14 01:03:30 -0700 |
---|---|---|
committer | Martin Michlmayr <tbm@cyrius.com> | 2021-08-24 16:32:10 +0800 |
commit | e712ce4e7cb6aac7469682cc05079707cbfb759b (patch) | |
tree | 5e9aafce7c776f8880b5c4229bf2727bfa2353ff /src/textual.cc | |
parent | 71c0903dffeb8e6750abdfba3c2e6e99f38f146c (diff) | |
download | fork-ledger-e712ce4e7cb6aac7469682cc05079707cbfb759b.tar.gz fork-ledger-e712ce4e7cb6aac7469682cc05079707cbfb759b.tar.bz2 fork-ledger-e712ce4e7cb6aac7469682cc05079707cbfb759b.zip |
Fix silent errors when reading lines from input files.
Handle files that don't end with a new line. Throw an error when the buffer
size is exceeded.
Fixes #516
Contributes to #1149
Diffstat (limited to 'src/textual.cc')
-rw-r--r-- | src/textual.cc | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/textual.cc b/src/textual.cc index b8523dbe..3b6ef5f4 100644 --- a/src/textual.cc +++ b/src/textual.cc @@ -311,9 +311,14 @@ std::streamsize instance_t::read_line(char *& line) check_for_signal(); - in.getline(context.linebuf, parse_context_t::MAX_LINE); + const size_t maxLine = parse_context_t::MAX_LINE; + in.getline(context.linebuf, maxLine); std::streamsize len = in.gcount(); + if (in.fail() && len == (parse_context_t::MAX_LINE - 1)) { + throw_(parse_error, _f("Line exceeds %1% characters") % maxLine); + } + if (len > 0) { context.linenum++; @@ -329,7 +334,12 @@ std::streamsize instance_t::read_line(char *& line) line = context.linebuf; } - --len; + if (!in.eof()) { + // if we are not at the end of the file, len includes the new line character, + // even through it does not appear in linebuf + --len; + } + while (len > 0 && std::isspace(line[len - 1])) // strip trailing whitespace line[--len] = '\0'; |