Merge pull request #3 from tetsuok/fix-stack-overflow

Fix stack-overflow in StringPiece
This commit is contained in:
Taku Kudo 2017-03-08 17:21:44 +09:00 коммит произвёл GitHub
Родитель f4d062f6e0 9e238d496d
Коммит 8ba8c63cd3
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -17,6 +17,7 @@
#include <cstring>
#include <string>
#include <ostream>
namespace sentencepiece {
@ -215,8 +216,7 @@ inline int StringPiece::compare(StringPiece b) const {
}
inline std::ostream &operator<<(std::ostream &o, StringPiece piece) {
o << piece.data();
return o;
return o.write(piece.data(), static_cast<std::streamsize>(piece.size()));
}
struct StringPieceHash {