Factor out value_sett::get_entry_for_symbol

Make this a separate function so it can be used from other places.
No change in behaviour.

Author: Owen

src/pointer-analysis/value_set.cpp

@@ -367,6 +367,45 @@ static std::string strip_first_field_from_suffix(
   return suffix.substr(field.length() + 1);
 }
 
+const value_sett::entryt *value_sett::get_entry_for_symbol(
+  const irep_idt identifier,
+  const typet &expr_type,
+  const std::string &suffix) const
+{
+  // We only deal with pointers, integers, arrays and structs
+  if(
+    expr_type.id() != ID_pointer && expr_type.id() != ID_signedbv &&
+    expr_type.id() != ID_unsignedbv && expr_type.id() != ID_struct &&
+    expr_type.id() != ID_union && expr_type.id() != ID_array)
+  {
+    return nullptr;
+  }
+
+  // look it up
+  const value_sett::entryt *entry = find_entry(id2string(identifier) + suffix);
+
+  // try first component name as suffix if not yet found
+  if(!entry && (expr_type.id() == ID_struct || expr_type.id() == ID_union))
+  {
+    const struct_union_typet &struct_union_type =
+      to_struct_union_type(expr_type);
+
+    const irep_idt &first_component_name =
+      struct_union_type.components().front().get_name();
+
+    entry = find_entry(
+      id2string(identifier) + "." + id2string(first_component_name) + suffix);
+  }
+
+  if(!entry)
+  {
+    // not found? try without suffix
+    entry = find_entry(identifier);
+  }
+
+  return entry;
+}
+
 void value_sett::get_value_set_rec(
   const exprt &expr,
   object_mapt &dest,
@@ -414,45 +453,17 @@ void value_sett::get_value_set_rec(
   }
   else if(expr.id()==ID_symbol)
   {
-    irep_idt identifier=to_symbol_expr(expr).get_identifier();
+    const entryt *entry = get_entry_for_symbol(
+      to_symbol_expr(expr).get_identifier(), expr_type, suffix);
 
-    // is it a pointer, integer, array or struct?
-    if(expr_type.id()==ID_pointer ||
-       expr_type.id()==ID_signedbv ||
-       expr_type.id()==ID_unsignedbv ||
-       expr_type.id()==ID_struct ||
-       expr_type.id()==ID_union ||
-       expr_type.id()==ID_array)
+    if(entry)
     {
-      // look it up
-      const entryt *entry =
-        find_entry(id2string(identifier) + suffix);
-
-      // try first component name as suffix if not yet found
-      if(!entry && (expr_type.id() == ID_struct || expr_type.id() == ID_union))
-      {
-        const struct_union_typet &struct_union_type=
-          to_struct_union_type(expr_type);
-
-        const irep_idt &first_component_name =
-          struct_union_type.components().front().get_name();
-
-        entry = find_entry(
-          id2string(identifier) + "." + id2string(first_component_name) +
-          suffix);
-      }
-
-      // not found? try without suffix
-      if(!entry)
-        entry = find_entry(identifier);
-
-      if(entry)
-        make_union(dest, entry->object_map);
-      else
-        insert(dest, exprt(ID_unknown, original_type));
+      make_union(dest, entry->object_map);
     }
     else
+    {
       insert(dest, exprt(ID_unknown, original_type));
+    }
   }
   else if(expr.id()==ID_if)
   {

src/pointer-analysis/value_set.h

@@ -447,6 +447,11 @@ class value_sett
     exprt &expr,
     const namespacet &ns) const;
 
+  const value_sett::entryt *get_entry_for_symbol(
+    const irep_idt identifier,
+    const typet &expr_type,
+    const std::string &suffix) const;
+
 protected:
   /// Reads the set of objects pointed to by `expr`, including making
   /// recursive lookups for dereference operations etc.