Fix use-after-free when switching blitters in SDL and Win32 modes.

This mostly occurs when disabling pallete animation when fast-forward
is enabled.
This commit is contained in:
Jonathan G Rennison
2017-03-07 19:12:15 +00:00
parent 68c280bef8
commit 2d473c05b0
6 changed files with 45 additions and 4 deletions

View File

@@ -23,6 +23,7 @@
#include "clear_map.h"
#include "clear_func.h"
#include "tree_map.h"
#include "scope.h"
#include "table/tree_land.h"
#include "blitter/32bpp_base.hpp"
@@ -331,6 +332,11 @@ static bool SwitchNewGRFBlitter()
const bool animation_wanted = HasBit(_display_opt, DO_FULL_ANIMATION);
const char *cur_blitter = BlitterFactory::GetCurrentBlitter()->GetName();
VideoDriver::GetInstance()->AcquireBlitterLock();
auto guard = scope_guard([&]() {
VideoDriver::GetInstance()->ReleaseBlitterLock();
});
for (uint i = 0; i < lengthof(replacement_blitters); i++) {
if (animation_wanted && (replacement_blitters[i].animation == 0)) continue;
if (!animation_wanted && (replacement_blitters[i].animation == 1)) continue;

View File

@@ -830,11 +830,18 @@ bool VideoDriver_SDL::ToggleFullscreen(bool fullscreen)
}
bool VideoDriver_SDL::AfterBlitterChange()
{
return CreateMainSurface(_screen.width, _screen.height);
}
void VideoDriver_SDL::AcquireBlitterLock()
{
if (_draw_mutex != NULL) _draw_mutex->BeginCritical(true);
bool ret = CreateMainSurface(_screen.width, _screen.height);
}
void VideoDriver_SDL::ReleaseBlitterLock()
{
if (_draw_mutex != NULL) _draw_mutex->EndCritical(true);
return ret;
}
#endif /* WITH_SDL */

View File

@@ -31,6 +31,10 @@ public:
/* virtual */ bool AfterBlitterChange();
/* virtual */ void AcquireBlitterLock();
/* virtual */ void ReleaseBlitterLock();
/* virtual */ bool ClaimMousePointer();
/* virtual */ const char *GetName() const { return "sdl"; }

View File

@@ -49,6 +49,7 @@ public:
/**
* Callback invoked after the blitter was changed.
* This may only be called between AcquireBlitterLock and ReleaseBlitterLock.
* @return True if no error.
*/
virtual bool AfterBlitterChange()
@@ -56,6 +57,18 @@ public:
return true;
}
/**
* Acquire any lock(s) required to be held when changing blitters.
* These lock(s) may not be acquired recursively.
*/
virtual void AcquireBlitterLock() { }
/**
* Release any lock(s) required to be held when changing blitters.
* These lock(s) may not be acquired recursively.
*/
virtual void ReleaseBlitterLock() { }
virtual bool ClaimMousePointer()
{
return true;

View File

@@ -1333,11 +1333,18 @@ bool VideoDriver_Win32::ToggleFullscreen(bool full_screen)
}
bool VideoDriver_Win32::AfterBlitterChange()
{
return AllocateDibSection(_screen.width, _screen.height, true) && this->MakeWindow(_fullscreen);
}
void VideoDriver_Win32::AcquireBlitterLock()
{
if (_draw_mutex != NULL) _draw_mutex->BeginCritical(true);
bool ret = AllocateDibSection(_screen.width, _screen.height, true) && this->MakeWindow(_fullscreen);
}
void VideoDriver_Win32::ReleaseBlitterLock()
{
if (_draw_mutex != NULL) _draw_mutex->EndCritical(true);
return ret;
}
void VideoDriver_Win32::EditBoxLostFocus()

View File

@@ -31,6 +31,10 @@ public:
/* virtual */ bool AfterBlitterChange();
/* virtual */ void AcquireBlitterLock();
/* virtual */ void ReleaseBlitterLock();
/* virtual */ bool ClaimMousePointer();
/* virtual */ void EditBoxLostFocus();