Fixed a segfault when a piece captured another.

currentPiece was set to the recently deleted captured piece. Operations on currentPiece caused a segfault.

Author: Thumperrr

src/Graphics.cpp

@@ -61,16 +61,16 @@ namespace chesspp
             Piece* pCurrent = b->getCurrent();
             Piece* pSelect  = b->getSelected();
             if (pSelect)
-                for (posList::const_iterator iter = pSelect->getTrajectory().begin(); iter != pSelect->getTrajectory().end(); iter++)
+                for (auto iter = pSelect->getTrajectory().begin(); iter != pSelect->getTrajectory().end(); iter++)
                 {
-                    if(!iter->isValid()) continue;
-                    drawValidMove(iter->getX(), iter->getY());
+                    if(iter->isValid())
+                        drawValidMove(iter->getX(), iter->getY());
                 }
             else if (pCurrent)
-                for (posList::const_iterator iter = pCurrent->getTrajectory().begin(); iter != pCurrent->getTrajectory().end(); iter++)
+                for (auto &iter: pCurrent->getTrajectory())
                 {
-                    if(!iter->isValid()) continue;
-                    drawValidMove(iter->getX(), iter->getY());
+                    if(iter.isValid())
+                        drawValidMove(iter.getX(), iter.getY());
                 }
 
             // Draw the non-selected pieces

src/board/Board.cpp

@@ -227,6 +227,7 @@ bool Board::move(Piece* toMove, int screenX, int screenY)
 		(*iter)->updateTrajectory(this, moveFrom, moveTo);
 	}
 
+    this->currentPiece = NULL; //reset currentPiece to avoid segfault.
 	return true;
 }
 

src/board/Piece.cpp

@@ -26,7 +26,7 @@ namespace chesspp
 	    return color;
     }
 
-    const posList& Piece::getTrajectory(void)
+    const posList& Piece::getTrajectory(void) const
     {
 	    return trajectory;
     }

src/board/Piece.h

@@ -55,7 +55,7 @@ namespace chesspp
 		const Position& getTexturePos(void) const;
 		Color getColor(void) const;
         Type getType(void) const;
-		const posList& getTrajectory(void);
+		const posList& getTrajectory(void) const;
 
 		// We need to know these things :)
 		// If a piece moves, it's trajectory needs to be checked for a king