1*0b57cec5SDimitry Andric //===-- StringSaver.cpp ---------------------------------------------------===// 2*0b57cec5SDimitry Andric // 3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0b57cec5SDimitry Andric // 7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 8*0b57cec5SDimitry Andric 9*0b57cec5SDimitry Andric #include "llvm/Support/StringSaver.h" 10*0b57cec5SDimitry Andric 11*0b57cec5SDimitry Andric using namespace llvm; 12*0b57cec5SDimitry Andric 13*0b57cec5SDimitry Andric StringRef StringSaver::save(StringRef S) { 14*0b57cec5SDimitry Andric char *P = Alloc.Allocate<char>(S.size() + 1); 15*0b57cec5SDimitry Andric if (!S.empty()) 16*0b57cec5SDimitry Andric memcpy(P, S.data(), S.size()); 17*0b57cec5SDimitry Andric P[S.size()] = '\0'; 18*0b57cec5SDimitry Andric return StringRef(P, S.size()); 19*0b57cec5SDimitry Andric } 20*0b57cec5SDimitry Andric 21*0b57cec5SDimitry Andric StringRef UniqueStringSaver::save(StringRef S) { 22*0b57cec5SDimitry Andric auto R = Unique.insert(S); 23*0b57cec5SDimitry Andric if (R.second) // cache miss, need to actually save the string 24*0b57cec5SDimitry Andric *R.first = Strings.save(S); // safe replacement with equal value 25*0b57cec5SDimitry Andric return *R.first; 26*0b57cec5SDimitry Andric } 27