Cleanup some memory issues

4 files changed, 19 insertions, 6 deletions

diff --git a/src/gui/widgets/dropdown.cpp b/src/gui/widgets/dropdown.cpp
index b8f5b895..80c3747e 100644
--- a/src/gui/widgets/dropdown.cpp
+++ b/src/gui/widgets/dropdown.cpp
@@ -43,10 +43,10 @@ Image *DropDown::buttons[2][2];
 ImageRect DropDown::skin;
 float DropDown::mAlpha = 1.0;
 
-DropDown::DropDown(gcn::ListModel *listModel):
+DropDown::DropDown(gcn::ListModel *listModel, bool deleteModel):
     gcn::DropDown::DropDown(listModel,
                             new ScrollArea,
-                            new ListBox(listModel))
+                            new ListBox(listModel, deleteModel))
 {
     setFrameSize(2);
 
diff --git a/src/gui/widgets/dropdown.h b/src/gui/widgets/dropdown.h
index f4ab4c93..eda7b1fa 100644
--- a/src/gui/widgets/dropdown.h
+++ b/src/gui/widgets/dropdown.h
@@ -45,7 +45,7 @@ class DropDown : public gcn::DropDown
          * @param listBox the listBox to use.
          * @see ListModel, ScrollArea, ListBox.
          */
-        DropDown(gcn::ListModel *listModel = NULL);
+        DropDown(gcn::ListModel *listModel = NULL, bool deleteModel = false);
 
         ~DropDown();
 
diff --git a/src/gui/widgets/listbox.cpp b/src/gui/widgets/listbox.cpp
index 7f96ad3e..e97f1d63 100644
--- a/src/gui/widgets/listbox.cpp
+++ b/src/gui/widgets/listbox.cpp
@@ -34,11 +34,20 @@
 
 float ListBox::mAlpha = 1.0;
 
-ListBox::ListBox(gcn::ListModel *listModel):
-    gcn::ListBox(listModel)
+ListBox::ListBox(gcn::ListModel *listModel, bool deleteModel):
+    gcn::ListBox(listModel),
+    mDeleteModel(deleteModel)
 {
 }
 
+ListBox::~ListBox()
+{
+    if (mDeleteModel)
+    {
+        delete mListModel;
+    }
+}
+
 void ListBox::updateAlpha()
 {
     float alpha = std::max(config.getValue("guialpha", 0.8),
diff --git a/src/gui/widgets/listbox.h b/src/gui/widgets/listbox.h
index d5798d66..c0662ed2 100644
--- a/src/gui/widgets/listbox.h
+++ b/src/gui/widgets/listbox.h
@@ -39,7 +39,9 @@ class ListBox : public gcn::ListBox
         /**
          * Constructor.
          */
-        ListBox(gcn::ListModel *listModel);
+        ListBox(gcn::ListModel *listModel, bool deleteModel = false);
+
+        ~ListBox();
 
         /**
          * Draws the list box.
@@ -65,6 +67,8 @@ class ListBox : public gcn::ListBox
 
     private:
         static float mAlpha;
+
+        bool mDeleteModel;
 };
 
 #endif