fix for redef'd globals in standalone scripts

src/script_opt/CPP/Driver.cc

@@ -44,7 +44,16 @@ void CPPCompile::Compile(bool report_uncompilable) {
             reporter->FatalError("aborting standalone compilation to C++ due to having to skip some functions");
 
         for ( auto& g : global_scope()->OrderedVars() ) {
-            if ( ! obj_matches_opt_files(g) )
+            bool compiled_global = obj_matches_opt_files(g);
+
+            if ( ! compiled_global )
+                for ( const auto& i_e : g->GetOptInfo()->GetInitExprs() )
+                    if ( obj_matches_opt_files(i_e) ) {
+                        compiled_global = true;
+                        break;
+                    }
+
+            if ( ! compiled_global )
                 continue;
 
             // We will need to generate this global's definition, including
@@ -63,10 +72,10 @@ void CPPCompile::Compile(bool report_uncompilable) {
                     (void)pfs->HashType(t);
                     rep_types.insert(TypeRep(t));
                 }
-		for ( auto& g : pf->Globals() )
-			accessed_globals.insert(g);
-		for ( auto& ag : pf->AllGlobals() )
-			all_accessed_globals.insert(ag);
+                for ( auto& g : pf->Globals() )
+                    accessed_globals.insert(g);
+                for ( auto& ag : pf->AllGlobals() )
+                    all_accessed_globals.insert(ag);
             }
         }
 

src/script_opt/CPP/InitsInfo.cc

@@ -355,14 +355,16 @@ AttrsInfo::AttrsInfo(CPPCompile* _c, const AttributesPtr& _attrs) : CompoundItem
     }
 }
 
-GlobalLookupInitInfo::GlobalLookupInitInfo(CPPCompile* c, const ID* g, string _CPP_name)
+GlobalLookupInitInfo::GlobalLookupInitInfo(CPPCompile* c, const ID* g, string _CPP_name, bool do_init)
     : CPP_InitInfo(g), CPP_name(std::move(_CPP_name)) {
     Zeek_name = g->Name();
+    val = ValElem(c, do_init ? g->GetVal() : nullptr);
 }
 
 void GlobalLookupInitInfo::InitializerVals(std::vector<std::string>& ivs) const {
     ivs.push_back(CPP_name);
     ivs.push_back(string("\"") + Zeek_name + "\"");
+    ivs.push_back(val);
 }
 
 GlobalInitInfo::GlobalInitInfo(CPPCompile* c, const ID* g, string _CPP_name)

src/script_opt/CPP/InitsInfo.h

@@ -482,7 +482,7 @@ public:
 // initialization time but not create it if missing.
 class GlobalLookupInitInfo : public CPP_InitInfo {
 public:
-    GlobalLookupInitInfo(CPPCompile* c, const ID* g, std::string CPP_name);
+    GlobalLookupInitInfo(CPPCompile* c, const ID* g, std::string CPP_name, bool do_init = false);
 
     std::string InitializerType() const override { return "CPP_GlobalLookupInit"; }
     void InitializerVals(std::vector<std::string>& ivs) const override;
@@ -490,6 +490,7 @@ public:
 protected:
     std::string Zeek_name;
     std::string CPP_name;
+    std::string val;
 };
 
 // Information for initializing a Zeek global.

src/script_opt/CPP/RuntimeInits.cc

@@ -457,6 +457,9 @@ int CPP_EnumMapping::ComputeOffset(InitsManager* im) const {
 
 void CPP_GlobalLookupInit::Generate(InitsManager* im, std::vector<void*>& /* inits_vec */, int /* offset */) const {
     global = find_global__CPP(name);
+    if ( val >= 0 )
+        // Have explicit initialization value.
+        global->SetVal(im->ConstVals(val));
 }
 
 void CPP_GlobalInit::Generate(InitsManager* im, std::vector<void*>& /* inits_vec */, int /* offset */) const {

src/script_opt/CPP/RuntimeInits.h

@@ -383,13 +383,15 @@ public:
 // vector, so we use void* as the underlying type.
 class CPP_GlobalLookupInit : public CPP_Init<void*> {
 public:
-    CPP_GlobalLookupInit(IDPtr& _global, const char* _name) : CPP_Init<void*>(), global(_global), name(_name) {}
+    CPP_GlobalLookupInit(IDPtr& _global, const char* _name, int _val)
+        : CPP_Init<void*>(), global(_global), name(_name), val(_val) {}
 
     void Generate(InitsManager* im, std::vector<void*>& /* inits_vec */, int /* offset */) const override;
 
 protected:
     IDPtr& global;
     const char* name;
+    int val;
 };
 
 class CPP_GlobalInit : public CPP_Init<void*> {

src/script_opt/CPP/Vars.cc

@@ -1,6 +1,7 @@
 // See the file "COPYING" in the main distribution directory for copyright.
 
 #include "zeek/script_opt/CPP/Compile.h"
+#include "zeek/script_opt/IDOptInfo.h"
 #include "zeek/script_opt/ProfileFunc.h"
 
 namespace zeek::detail {
@@ -32,12 +33,7 @@ void CPPCompile::CreateGlobal(const ID* g) {
             // This is an event that's also used as a variable.
             Emit("EventHandlerPtr %s_ev;", globals[gn]);
 
-        shared_ptr<CPP_InitInfo> gi;
-        if ( standalone )
-            gi = make_shared<GlobalInitInfo>(this, g, globals[gn]);
-        else
-            gi = make_shared<GlobalLookupInitInfo>(this, g, globals[gn]);
-
+        auto gi = GenerateGlobalInit(g);
         global_id_info->AddInstance(gi);
         global_gis[g] = gi;
     }
@@ -69,18 +65,36 @@ std::shared_ptr<CPP_InitInfo> CPPCompile::RegisterGlobal(const ID* g) {
             return gg->second;
     }
 
-    shared_ptr<CPP_InitInfo> gi;
-    if ( standalone )
-        gi = make_shared<GlobalInitInfo>(this, g, globals[gn]);
-    else
-        gi = make_shared<GlobalLookupInitInfo>(this, g, globals[gn]);
-
+    auto gi = GenerateGlobalInit(g);
     global_id_info->AddInstance(gi);
     global_gis[g] = gi;
 
     return gi;
 }
 
+std::shared_ptr<CPP_InitInfo> CPPCompile::GenerateGlobalInit(const ID* g) {
+    auto gn = string(g->Name());
+    if ( ! standalone )
+        return make_shared<GlobalLookupInitInfo>(this, g, globals[gn]);
+
+    if ( obj_matches_opt_files(g) )
+        return make_shared<GlobalInitInfo>(this, g, globals[gn]);
+
+    // It's not a global that's created by the scripts we're compiling,
+    // but it might have a redef in those scripts, in which case we need
+    // to generate an initializer that will both look it up and then assign
+    // it to that value.
+    bool needs_redef = false;
+
+    for ( const auto& i_e : g->GetOptInfo()->GetInitExprs() )
+        if ( obj_matches_opt_files(i_e) ) {
+            needs_redef = true;
+            break;
+        }
+
+    return make_shared<GlobalLookupInitInfo>(this, g, globals[gn], needs_redef);
+}
+
 void CPPCompile::AddBiF(const ID* b, bool is_var) {
     auto bn = b->Name();
     auto n = string(bn);

src/script_opt/CPP/Vars.h

@@ -14,6 +14,10 @@ private:
 // as a variable (not just as a function being called), track it as such.
 void CreateGlobal(const ID* g);
 
+// Low-level function for generating an initializer for a global. Takes
+// into account differences for standalone-compilation.
+std::shared_ptr<CPP_InitInfo> GenerateGlobalInit(const ID* g);
+
 // Register the given identifier as a BiF.  If is_var is true then the BiF
 // is also used in a non-call context.
 void AddBiF(const ID* b, bool is_var);