Return std::unique_ptr member as const ref

Returning raw pointer has a risk that it may be stolen by
std::shared_ptr in accident.

1 files changed, 3 insertions, 2 deletions

diff --git a/src/DownloadContext.cc b/src/DownloadContext.cc
index 0a76bbf5..6591b3ea 100644
--- a/src/DownloadContext.cc
+++ b/src/DownloadContext.cc
@@ -156,12 +156,13 @@ void DownloadContext::setAttribute
   attrs_[key] = std::move(value);
 }
 
-ContextAttribute* DownloadContext::getAttribute(ContextAttributeType key)
+const std::unique_ptr<ContextAttribute>& DownloadContext::getAttribute
+(ContextAttributeType key)
 {
   assert(key < MAX_CTX_ATTR);
   const std::unique_ptr<ContextAttribute>& attr = attrs_[key];
   if(attr) {
-    return attr.get();
+    return attr;
   } else {
     throw DL_ABORT_EX(fmt("No attribute named %s",
                           strContextAttributeType(key)));