[Ozone-DRI] Do proper bounds checks when moving the cursor

ui/ozone/platform/dri/dri_window_manager.h

 // Copyright 2014 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.
 
 #ifndef UI_OZONE_PLATFORM_DRI_DRI_WINDOW_MANAGER_H_
 #define UI_OZONE_PLATFORM_DRI_DRI_WINDOW_MANAGER_H_
 
-#include "base/containers/scoped_ptr_hash_map.h"
+#include <map>
+
 #include "ui/gfx/native_widget_types.h"
 
 namespace ui {
 
-class DriWindowDelegate;
+class DriWindow;
 
+// Responsible for keeping the mapping between the allocated widgets and
+// windows.
 class DriWindowManager {
  public:
   DriWindowManager();
   ~DriWindowManager();
 
   gfx::AcceleratedWidget NextAcceleratedWidget();
 
-  // Adds a delegate for |widget|. Note: |widget| should not be associated with
-  // a delegate when calling this function.
-  void AddWindowDelegate(gfx::AcceleratedWidget widget,
-                         scoped_ptr<DriWindowDelegate> surface);
+  // Adds a window for |widget|. Note: |widget| should not be associated when
+  // calling this function.
+  void AddWindow(gfx::AcceleratedWidget widget, DriWindow* window);
 
-  // Removes the delegate for |widget|. Note: |widget| must have a delegate
-  // associated with it when calling this function.
-  scoped_ptr<DriWindowDelegate> RemoveWindowDelegate(
-      gfx::AcceleratedWidget widget);
+  // Removes the window association for |widget|. Note: |widget| must be
+  // associated with a window when calling this function.
+  void RemoveWindow(gfx::AcceleratedWidget widget);
 
-  // Returns the delegate associated with |widget|. Note: This function should
-  // be called only if a valid delegate has been associated with |widget|.
-  DriWindowDelegate* GetWindowDelegate(gfx::AcceleratedWidget widget);
-
-  // Check if |widget| has a valid delegate associated with it.
-  bool HasWindowDelegate(gfx::AcceleratedWidget widget);
+  // Returns the window associated with |widget|. Note: This function should
+  // only be called if a valid window has been associated.
+  DriWindow* GetWindow(gfx::AcceleratedWidget widget);
 
  private:
-  typedef base::ScopedPtrHashMap<gfx::AcceleratedWidget, DriWindowDelegate>
-      WidgetToDelegateMap;
+  typedef std::map<gfx::AcceleratedWidget, DriWindow*> WidgetToWindowMap;
 
-  WidgetToDelegateMap delegate_map_;
   gfx::AcceleratedWidget last_allocated_widget_;
+  WidgetToWindowMap window_map_;
 
   DISALLOW_COPY_AND_ASSIGN(DriWindowManager);
 };
 
 }  // namespace ui
 
 #endif  // UI_OZONE_PLATFORM_DRI_DRI_WINDOW_MANAGER_H_