Fix bug in shader and program managers.

gpu/command_buffer/service/shader_manager.cc

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Copyright (c) 2011 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 "gpu/command_buffer/service/shader_manager.h"
 #include "base/logging.h"
 
 namespace gpu {
 namespace gles2 {
 
+ShaderManager::ShaderInfo::ShaderInfo(GLuint service_id, GLenum shader_type)
+      : use_count_(0),
+        service_id_(service_id),
+        shader_type_(shader_type),
+        valid_(false) {
+}
+
+ShaderManager::ShaderInfo::~ShaderInfo() {
+}
+
+void ShaderManager::ShaderInfo::IncUseCount() {
+  ++use_count_;
+}
+
+void ShaderManager::ShaderInfo::DecUseCount() {
+  --use_count_;
+  DCHECK_GE(use_count_, 0);
+}
+
+void ShaderManager::ShaderInfo::MarkAsDeleted() {
+  DCHECK_NE(service_id_, 0u);
+  service_id_ = 0;
+}
+
 void ShaderManager::ShaderInfo::SetStatus(
     bool valid, const char* log, ShaderTranslatorInterface* translator) {
   valid_ = valid;
   log_info_.reset(log ? new std::string(log) : NULL);
   if (translator && valid) {
     attrib_map_ = translator->attrib_map();
     uniform_map_ = translator->uniform_map();
   } else {
     attrib_map_.clear();
     uniform_map_.clear();
@@ skipping 52 matching lines @@
   for (ShaderInfoMap::const_iterator it = shader_infos_.begin();
        it != shader_infos_.end(); ++it) {
     if (it->second->service_id() == service_id) {
       *client_id = it->first;
       return true;
     }
   }
   return false;
 }
 
+bool ShaderManager::IsOwned(ShaderManager::ShaderInfo* info) {
+  for (ShaderInfoMap::iterator it = shader_infos_.begin();
+       it != shader_infos_.end(); ++it) {
+    if (it->second.get() == info) {
+      return true;
+    }
+  }
+  return false;
+}
+
 void ShaderManager::RemoveShaderInfoIfUnused(ShaderManager::ShaderInfo* info) {
   DCHECK(info);
+  DCHECK(IsOwned(info));
   if (info->IsDeleted() && !info->InUse()) {
     for (ShaderInfoMap::iterator it = shader_infos_.begin();
          it != shader_infos_.end(); ++it) {
-      if (it->second->service_id() == info->service_id()) {
+      if (it->second.get() == info) {
         shader_infos_.erase(it);
         return;
       }
     }
     NOTREACHED();
   }
 }
 
 void ShaderManager::MarkAsDeleted(ShaderManager::ShaderInfo* info) {
   DCHECK(info);
+  DCHECK(IsOwned(info));
   info->MarkAsDeleted();
   RemoveShaderInfoIfUnused(info);
 }
 
 void ShaderManager::UseShader(ShaderManager::ShaderInfo* info) {
   DCHECK(info);
+  DCHECK(IsOwned(info));
   info->IncUseCount();
 }
 
 void ShaderManager::UnuseShader(ShaderManager::ShaderInfo* info) {
   DCHECK(info);
+  DCHECK(IsOwned(info));
   info->DecUseCount();
   RemoveShaderInfoIfUnused(info);
 }
 
 }  // namespace gles2
 }  // namespace gpu