(svn r15206) -Fix [FS#2567]: memory corruption due to not properly cleanup up the mess when cancelling a download

This commit is contained in:
rubidium
2009-01-22 10:09:56 +00:00
parent 4d8c467067
commit bf47489d58
5 changed files with 13 additions and 9 deletions

View File

@@ -20,7 +20,14 @@ NetworkTCPSocketHandler::NetworkTCPSocketHandler(SOCKET s) :
NetworkTCPSocketHandler::~NetworkTCPSocketHandler()
{
this->CloseConnection();
if (this->sock != INVALID_SOCKET) closesocket(this->sock);
this->sock = INVALID_SOCKET;
}
NetworkRecvStatus NetworkTCPSocketHandler::CloseConnection()
{
this->writable = false;
this->has_quit = true;
@@ -33,7 +40,7 @@ NetworkTCPSocketHandler::~NetworkTCPSocketHandler()
delete this->packet_recv;
this->packet_recv = NULL;
this->sock = INVALID_SOCKET;
return NETWORK_RECV_STATUS_OKAY;
}
/**

View File

@@ -22,6 +22,7 @@ private:
public:
bool writable; ///< Can we write to this socket?
virtual NetworkRecvStatus CloseConnection();
void Send_Packet(Packet *packet);
bool Send_Packets();
bool IsPacketQueueEmpty();

View File

@@ -50,12 +50,6 @@ bool ContentInfo::IsValid() const
return this->state < ContentInfo::INVALID && this->type >= CONTENT_TYPE_BEGIN && this->type < CONTENT_TYPE_END;
}
NetworkRecvStatus NetworkContentSocketHandler::CloseConnection()
{
this->has_quit = true;
return NETWORK_RECV_STATUS_OKAY;
}
void NetworkContentSocketHandler::Close()
{
CloseConnection();

View File

@@ -102,7 +102,6 @@ struct ContentInfo {
class NetworkContentSocketHandler : public NetworkTCPSocketHandler {
protected:
struct sockaddr_in client_addr; ///< The address we're connected to.
NetworkRecvStatus CloseConnection();
virtual void Close();
/**