[C++] Introduce ScopedTerminator

1 files changed, 9 insertions, 0 deletions

diff --git a/strutil.cc b/strutil.cc
index d09abda..316a657 100644
--- a/strutil.cc
+++ b/strutil.cc
@@ -77,6 +77,15 @@ void WordWriter::Write(StringPiece s) {
   AppendString(s, out_);
 }
 
+ScopedTerminator::ScopedTerminator(StringPiece s)
+    : s_(s), c_(s[s.size()]) {
+  const_cast<char*>(s_.data())[s_.size()] = '\0';
+}
+
+ScopedTerminator::~ScopedTerminator() {
+  const_cast<char*>(s_.data())[s_.size()] = c_;
+}
+
 static unordered_map<StringPiece, char*>* g_symtab;
 
 void InitSymtab() {