Reland 9034035: Make it possible to have 1 PpapiGlobals per thread.

ppapi/proxy/ppb_var_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <string>
 #include <vector>
 
 #include "base/string_number_conversions.h"
 #include "base/threading/platform_thread.h"
 #include "ppapi/c/pp_var.h"
 #include "ppapi/c/ppb_var.h"
 #include "ppapi/proxy/ppapi_proxy_test.h"
+#include "ppapi/shared_impl/ppapi_globals.h"
 #include "ppapi/shared_impl/ppb_var_shared.h"
 
 namespace {
 std::string VarToString(const PP_Var& var, const PPB_Var* ppb_var) {
   uint32_t len = 0;
   const char* utf8 = ppb_var->VarToUtf8(var, &len);
   return std::string(utf8, len);
 }
 const size_t kNumStrings = 100;
 const size_t kNumThreads = 20;
@@ skipping 66 matching lines @@
 // 3) Spawn kNumThreads 'RemoveVar' threads. Each of these threads releases each
 //    var once. Once all the threads have finished, there should be no vars
 //    left.
 class CreateVarThreadDelegate : public base::PlatformThread::Delegate {
  public:
   // |strings_in|, |vars|, and |strings_out| are arrays, and size is their size.
   // For each |strings_in[i]|, we will set |vars[i]| using that value. Then we
   // read the var back out to |strings_out[i]|.
   CreateVarThreadDelegate(PP_Module pp_module, const std::string* strings_in,
                           PP_Var* vars_out, std::string* strings_out,
-                          size_t size)
+                          size_t size, PpapiGlobals* globals)
       : pp_module_(pp_module), strings_in_(strings_in), vars_out_(vars_out),
-        strings_out_(strings_out), size_(size) {
+        strings_out_(strings_out), size_(size), globals_(globals) {
   }
   virtual ~CreateVarThreadDelegate() {}
   virtual void ThreadMain() {
+    PpapiGlobals::SetPpapiGlobalsOnThreadForTest(globals_);
     const PPB_Var* ppb_var = ppapi::PPB_Var_Shared::GetVarInterface1_1();
     for (size_t i = 0; i < size_; ++i) {
       vars_out_[i] = ppb_var->VarFromUtf8(strings_in_[i].c_str(),
                                           strings_in_[i].length());
       strings_out_[i] = VarToString(vars_out_[i], ppb_var);
     }
   }
  private:
   PP_Module pp_module_;
   const std::string* strings_in_;
   PP_Var* vars_out_;
   std::string* strings_out_;
   size_t size_;
+  PpapiGlobals* globals_;
 };
 
 // A thread that will increment and decrement the reference count of every var
 // multiple times.
 class ChangeRefVarThreadDelegate : public base::PlatformThread::Delegate {
  public:
-  ChangeRefVarThreadDelegate(const std::vector<PP_Var>& vars) : vars_(vars) {
+  ChangeRefVarThreadDelegate(const std::vector<PP_Var>& vars,
+                             PpapiGlobals* globals)
+      : vars_(vars), globals_(globals) {
   }
   virtual ~ChangeRefVarThreadDelegate() {}
   virtual void ThreadMain() {
+    PpapiGlobals::SetPpapiGlobalsOnThreadForTest(globals_);
     const PPB_Var* ppb_var = ppapi::PPB_Var_Shared::GetVarInterface1_1();
     // Increment and decrement the reference count for each var kRefsToAdd
     // times. Note that we always AddRef once before doing the matching Release,
     // to ensure that we never accidentally release the last reference.
     for (int ref = 0; ref < kRefsToAdd; ++ref) {
       for (size_t i = 0; i < kNumStrings; ++i) {
         ppb_var->AddRef(vars_[i]);
         ppb_var->Release(vars_[i]);
       }
     }
     // Now add 1 ref to each Var. The net result is that all Vars will have a
     // ref-count of (kNumThreads + 1) after this. That will allow us to have all
     // threads release all vars later.
     for (size_t i = 0; i < kNumStrings; ++i) {
       ppb_var->AddRef(vars_[i]);
     }
   }
  private:
   std::vector<PP_Var> vars_;
+  PpapiGlobals* globals_;
 };
 
 // A thread that will decrement the reference count of every var once.
 class RemoveRefVarThreadDelegate : public base::PlatformThread::Delegate {
  public:
-  RemoveRefVarThreadDelegate(const std::vector<PP_Var>& vars) : vars_(vars) {
+  RemoveRefVarThreadDelegate(const std::vector<PP_Var>& vars,
+                             PpapiGlobals* globals)
+      : vars_(vars), globals_(globals) {
   }
   virtual ~RemoveRefVarThreadDelegate() {}
   virtual void ThreadMain() {
+    PpapiGlobals::SetPpapiGlobalsOnThreadForTest(globals_);
     const PPB_Var* ppb_var = ppapi::PPB_Var_Shared::GetVarInterface1_1();
     for (size_t i = 0; i < kNumStrings; ++i) {
       ppb_var->Release(vars_[i]);
     }
   }
  private:
   std::vector<PP_Var> vars_;
+  PpapiGlobals* globals_;
 };
 
 }  // namespace
 
 #ifdef ENABLE_PEPPER_THREADING
 TEST_F(PPB_VarTest, Threads) {
 #else
 TEST_F(PPB_VarTest, DISABLED_Threads) {
 #endif
   std::vector<base::PlatformThreadHandle> create_var_threads(kNumThreads);
   std::vector<CreateVarThreadDelegate> create_var_delegates;
   // The strings that the threads will re-extract from Vars (so we can check
   // that they match the original strings).
   std::vector<std::string> strings_out(kNumStrings);
   size_t strings_per_thread = kNumStrings/kNumThreads;
   // Give each thread an equal slice of strings to turn in to vars. (Except the
   // last thread may get fewer if kNumStrings is not evenly divisible by
   // kNumThreads).
   for (size_t slice_start= 0; slice_start < kNumStrings;
        slice_start += strings_per_thread) {
     create_var_delegates.push_back(
         CreateVarThreadDelegate(pp_module(),
                                 &test_strings_[slice_start],
                                 &vars_[slice_start],
                                 &strings_out[slice_start],
                                 std::min(strings_per_thread,
-                                         kNumStrings - slice_start)));
+                                         kNumStrings - slice_start),
+                                GetGlobals()));
   }
   // Now run then join all the threads.
   for (size_t i = 0; i < kNumThreads; ++i)
     base::PlatformThread::Create(0, &create_var_delegates[i],
                                  &create_var_threads[i]);
   for (size_t i = 0; i < kNumThreads; ++i)
     base::PlatformThread::Join(create_var_threads[i]);
   // Now check that the strings have the expected values.
   EXPECT_EQ(test_strings_, strings_out);
 
   // Tinker with the reference counts in a multithreaded way.
   std::vector<base::PlatformThreadHandle> change_ref_var_threads(kNumThreads);
   std::vector<ChangeRefVarThreadDelegate> change_ref_var_delegates;
   for (size_t i = 0; i < kNumThreads; ++i)
-    change_ref_var_delegates.push_back(ChangeRefVarThreadDelegate(vars_));
+    change_ref_var_delegates.push_back(
+        ChangeRefVarThreadDelegate(vars_, GetGlobals()));
   for (size_t i = 0; i < kNumThreads; ++i) {
     base::PlatformThread::Create(0, &change_ref_var_delegates[i],
                                  &change_ref_var_threads[i]);
   }
   for (size_t i = 0; i < kNumThreads; ++i)
     base::PlatformThread::Join(change_ref_var_threads[i]);
 
   // Now each var has a refcount of (kNumThreads + 1). Let's decrement each var
   // once so that every 'RemoveRef' thread (spawned below) owns 1 reference, and
   // when the last one removes a ref, the Var will be deleted.
   for (size_t i = 0; i < kNumStrings; ++i) {
     ppb_var_->Release(vars_[i]);
   }
 
   // Check that all vars are still valid and have the values we expect.
   for (size_t i = 0; i < kNumStrings; ++i)
     EXPECT_EQ(test_strings_[i], VarToString(vars_[i], ppb_var_));
 
   // Remove the last reference counts for all vars.
   std::vector<base::PlatformThreadHandle> remove_ref_var_threads(kNumThreads);
   std::vector<RemoveRefVarThreadDelegate> remove_ref_var_delegates;
   for (size_t i = 0; i < kNumThreads; ++i)
-    remove_ref_var_delegates.push_back(RemoveRefVarThreadDelegate(vars_));
+    remove_ref_var_delegates.push_back(
+        RemoveRefVarThreadDelegate(vars_, GetGlobals()));
   for (size_t i = 0; i < kNumThreads; ++i) {
     base::PlatformThread::Create(0, &remove_ref_var_delegates[i],
                                  &remove_ref_var_threads[i]);
   }
   for (size_t i = 0; i < kNumThreads; ++i)
     base::PlatformThread::Join(remove_ref_var_threads[i]);
 
   // All the vars should no longer represent valid strings.
   for (size_t i = 0; i < kNumStrings; ++i) {
     uint32_t len = 10;
     const char* utf8 = ppb_var_->VarToUtf8(vars_[i], &len);
     EXPECT_EQ(NULL, utf8);
     EXPECT_EQ(0u, len);
   }
 }
 
 }  // namespace proxy
 }  // namespace ppapi