fix buffer overflow in countlines for files greater than 32k
Signed-off-by: Derek Stevens <nilix@nilfm.cc>
This commit is contained in:
parent
c90b2b5193
commit
d5a33e9150
|
@ -59,7 +59,7 @@ func countLines(r io.Reader) (int, error) {
|
||||||
switch {
|
switch {
|
||||||
case err == io.EOF:
|
case err == io.EOF:
|
||||||
/* handle last line not having a newline at the end */
|
/* handle last line not having a newline at the end */
|
||||||
if bufLen >= 1 && buf[bufLen-1] != '\n' {
|
if bufLen >= 1 && buf[(bufLen-1)%(32*1024)] != '\n' {
|
||||||
count++
|
count++
|
||||||
}
|
}
|
||||||
return count, nil
|
return count, nil
|
||||||
|
|
Loading…
Reference in New Issue